diff --git a/websockify/websocket.py b/websockify/websocket.py index 7adfca8..4f298bc 100644 --- a/websockify/websocket.py +++ b/websockify/websocket.py @@ -424,7 +424,7 @@ Sec-WebSocket-Accept: %s\r while self.send_parts: # Send pending frames buf = self.send_parts.pop(0) - sent = self.client.send(buf) + sent = self.request.send(buf) if sent == len(buf): self.traffic("<") @@ -446,7 +446,7 @@ Sec-WebSocket-Accept: %s\r bufs = [] tdelta = int(time.time()*1000) - self.start_time - buf = self.client.recv(self.buffer_size) + buf = self.request.recv(self.buffer_size) if len(buf) == 0: closed = {'code': 1000, 'reason': "Client closed abruptly"} return bufs, closed @@ -501,7 +501,7 @@ Sec-WebSocket-Accept: %s\r msg = pack(">H%ds" % len(reason), code, reason) buf, h, t = self.encode_hybi(msg, opcode=0x08, base64=False) - self.client.send(buf) + self.request.send(buf) def do_websocket_handshake(self, headers, path): h = self.headers = headers @@ -689,7 +689,7 @@ Sec-WebSocket-Accept: %s\r # handler process try: try: - self.client = self.do_handshake(startsock, address) + self.request = self.do_handshake(startsock, address) if self.record: # Record raw frame data as JavaScript array @@ -708,7 +708,7 @@ Sec-WebSocket-Accept: %s\r except self.CClose: # Close the client _, exc, _ = sys.exc_info() - if self.client: + if self.request: self.send_close(exc.args[0], exc.args[1]) except self.EClose: _, exc, _ = sys.exc_info() @@ -725,10 +725,10 @@ Sec-WebSocket-Accept: %s\r self.rec.write("'EOF'];\n") self.rec.close() - if self.client and self.client != startsock: + if self.request and self.request != startsock: # Close the SSL wrapped socket # Original socket closed by caller - self.client.close() + self.request.close() def new_client(self): """ Do something with a WebSockets client connection. """ @@ -758,7 +758,7 @@ Sec-WebSocket-Accept: %s\r while True: try: try: - self.client = None + self.request = None startsock = None pid = err = 0 child_count = 0 diff --git a/websockify/websocketproxy.py b/websockify/websocketproxy.py index 1154d92..52b46b4 100755 --- a/websockify/websocketproxy.py +++ b/websockify/websocketproxy.py @@ -242,23 +242,23 @@ Traffic Legend: cqueue = [] c_pend = 0 tqueue = [] - rlist = [self.client, target] + rlist = [self.request, target] while True: wlist = [] if tqueue: wlist.append(target) - if cqueue or c_pend: wlist.append(self.client) + if cqueue or c_pend: wlist.append(self.request) ins, outs, excepts = select(rlist, wlist, [], 1) if excepts: raise Exception("Socket exception") - if self.client in outs: + if self.request in outs: # Send queued target data to the client c_pend = self.send_frames(cqueue) cqueue = [] - if self.client in ins: + if self.request in ins: # Receive client data, decode it, and queue for target bufs, closed = self.recv_frames() tqueue.extend(bufs)