diff options
| author | Casey Deccio <casey@deccio.net> | 2016-12-21 23:14:54 +0300 |
|---|---|---|
| committer | Casey Deccio <casey@deccio.net> | 2017-06-15 23:34:47 +0300 |
| commit | 6564ccd30a71a1116b38831a6d2e1432c38e2765 (patch) | |
| tree | 5ca762a545f8de345614c0712573bbc7f8325ff0 | |
| parent | 4f47d39e71cdd2a40812e84e805a6cba7abb10c2 (diff) | |
Fix frame length in WebSocket header
| -rw-r--r-- | dnsviz/transport.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/dnsviz/transport.py b/dnsviz/transport.py index 51b53c5..833b910 100644 --- a/dnsviz/transport.py +++ b/dnsviz/transport.py @@ -864,7 +864,7 @@ class DNSQueryTransportHandlerWebSocket(DNSQueryTransportHandlerMulti): elif l <= 0xffff: header += struct.pack(b'!BH', 126, l) else: # 0xffff < len <= 2^63 - header += struct.pack(b'!BL', 127, l) + header += struct.pack(b'!BLL', 127, 0, l) self.req = header + data self.req_len = len(self.req) self.req_index = 0 @@ -913,7 +913,7 @@ class DNSQueryTransportHandlerWebSocket(DNSQueryTransportHandlerMulti): self.res_len = byte1b elif byte1b == 126: self.res_len = struct.unpack(b'!H', self.res_buf[2:4])[0] - elif byte1b == 127: + else: # byte1b == 127: self.res_len = struct.unpack(b'!Q', self.res_buf[2:10])[0] # handle mask |