faeez/mxmbsocket
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Repo. Project Releases Wiki Activity

[WebSocket] RFC Masking

Browse Source 
Full un/masking capabilities on RFC6455 Frames
This commit is contained in:
Chris Boden 2012-06-03 11:55:35 -04:00
parent 9f0e29fe7f
commit 3a530c8c24
2 changed files with 32 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
src/Ratchet/WebSocket/Version/RFC6455/Frame.php
View File

@ -179,13 +179,7 @@ class Frame implements FrameInterface {
throw new \InvalidArgumentException("Masking key MUST be ASCII");
}
if (!$this->isCoalesced()) {
throw new \UnderflowException('Frame must be coalesced to apply a mask');
}
if ($this->isMasked()) {
$this->unMaskPayload();
}
$this->unMaskPayload();
$byte = sprintf('%08b', ord(substr($this->data, 1, 1)));
@ -193,16 +187,7 @@ class Frame implements FrameInterface {
$this->data = substr_replace($this->data, $maskingKey, $this->getNumPayloadBytes() + 1, 0);
$this->_bytes_rec += static::MASK_LENGTH;
$plLen = $this->getPayloadLength();
$start = $this->getPayloadStartingByte();
$maskedPl = '';
for ($i = 0; $i < $plLen; $i++) {
$maskedPl .= substr($this->data, $i + $start, 1) ^ substr($maskingKey, $i % static::MASK_LENGTH, 1);
}
$this->data = substr_replace($this->data, $maskedPl, $start, $plLen);
$this->data = substr_replace($this->data, $this->applyMaskToPayload($maskingKey), $this->getPayloadStartingByte(), $this->getPayloadLength());
return $this;
}
@ -217,18 +202,33 @@ class Frame implements FrameInterface {
return $this;
}
$maskingKey = $this->getMaskingKey();
$byte = sprintf('%08b', ord(substr($this->data, 1, 1)));
$this->data = substr_replace($this->data, static::encode(substr_replace($byte, '0', 0, 1)), 1, 1);
$this->data = substr_replace($this->data, '', $this->getNumPayloadBytes() + 1, static::MASK_LENGTH);
$this->_bytes_rec -= static::MASK_LENGTH;
$this->data = substr_replace($this->data, $this->applyMaskToPayload($maskingKey), $this->getPayloadStartingByte(), $this->getPayloadLength());
return $this;
}
protected function applyMaskToPayload($maskingKey) {
if (!$this->isCoalesced()) {
throw new \UnderflowException('Frame must be coalesced to apply a mask');
}
$maskingKey = $this->getMaskingKey();
$plLen = $this->getPayloadLength();
$start = $this->getPayloadStartingByte();
$applied = '';
// set the indicator bit to 0
// remove the masking key
// get the masking key
// unmask the payload
for ($i = 0; $i < $plLen; $i++) {
$applied .= substr($this->data, $i + $start, 1) ^ substr($maskingKey, $i % static::MASK_LENGTH, 1);
}
return $this;
return $applied;
}
/**
@ -343,17 +343,11 @@ class Frame implements FrameInterface {
throw new \UnderflowException('Can not return partial message');
}
$payload = '';
$length = $this->getPayloadLength();
$start = $this->getPayloadStartingByte();
if ($this->isMasked()) {
$mask = $this->getMaskingKey();
for ($i = 0; $i < $length; $i++) {
// Double check the RFC - is the masking byte level or character level?
$payload .= substr($this->data, $i + $start, 1) ^ substr($mask, $i % static::MASK_LENGTH, 1);
}
$payload = $this->applyMaskToPayload($this->getMaskingKey());
} else {
$payload = substr($this->data, $start, $this->getPayloadLength());
}

8
tests/Ratchet/Tests/WebSocket/Version/RFC6455/FrameTest.php
View File

@ -306,6 +306,14 @@ class FrameTest extends \PHPUnit_Framework_TestCase {
$this->assertEquals($msg, $frame->getPayload());
}
public function testUnMaskPayload() {
$string = $this->generateRandomString();
$frame = Frame::create($string)->maskPayload()->unMaskPayload();
$this->assertFalse($frame->isMasked());
$this->assertEquals($string, $frame->getPayload());
}
protected function generateRandomString($length = 10, $addSpaces = true, $addNumbers = true) {
$characters = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"$%&/()=[]{}'; // ยง