def test_tuple_to_address_invalid_arg(self): """ Tests the tuple to address conversion when a wrong argument is passed """ with self.assertRaises(ValueError): try: tuple_to_address("WRONG") except ValueError as e: self.assertTrue(str(e).startswith("too many values to unpack")) raise e
def test_tuple_to_address_invalid_arg(self): """ Tests the tuple to address conversion when a wrong argument is passed """ with self.assertRaises(ValueError): try: tuple_to_address("WRONG") except ValueError as e: self.assertTrue(str(e).startswith("too many values to unpack")) raise e
def test_tuple_to_address_empty_tuple(self): """ Tests the tuple to address conversion when a tuple with empty values is passed """ with self.assertRaises(ValueError): try: tuple_to_address((None, None)) except ValueError as e: self.assertEqual("invalid argument passed: (None, None)", str(e)) raise e
def handle_stream(self, stream, address): """ Handle a new client connection with a proxy over websocket """ logger.info("Got connection from %s on %s" % (tuple_to_address(stream.socket.getpeername()), tuple_to_address(stream.socket.getsockname()))) self.ws_conn = WebSocketProxyConnection(self.ws_url, stream, address, filters=self.filters, ws_options=self.ws_options) self.ws_conn.connect()
def test_tuple_to_address_empty_tuple(self): """ Tests the tuple to address conversion when a tuple with empty values is passed """ with self.assertRaises(ValueError): try: tuple_to_address((None, None)) except ValueError as e: self.assertEqual("invalid argument passed: (None, None)", str(e)) raise e
def handle_stream(self, stream, address): """ Handle a new client connection with a proxy over websocket """ logger.info("Got connection from %s on %s" % (tuple_to_address(stream.socket.getpeername()), tuple_to_address(stream.socket.getsockname()))) self.ws_conn = WebSocketProxyConnection(self.ws_url, stream, address, filters=self.filters, ws_options=self.ws_options) self.ws_conn.connect()
def open(self): """ Open the connection to the service when the WebSocket connection has been established """ logger.info("Forwarding connection to server %s" % tuple_to_address(self.remote_address)) self.io_stream.connect(self.remote_address, self.on_connect)
def test_address_to_tuple(self): """ Tests the address to tuple conversions """ addr = "127.0.0.1:443" t = address_to_tuple(addr) self.assertEqual(t, ("127.0.0.1", 443)) self.assertEqual(addr, tuple_to_address(t))
def test_address_to_tuple(self): """ Tests the address to tuple conversions """ addr = "127.0.0.1:443" t = address_to_tuple(addr) self.assertEqual(t, ("127.0.0.1", 443)) self.assertEqual(addr, tuple_to_address(t))
def test_address_to_tuple_missing_port(self): """ Tests the address to tuple conversions when port is missing """ addr = "127.0.0.1" t = address_to_tuple(addr) self.assertEqual(t, ("127.0.0.1", None)) self.assertEqual(addr, tuple_to_address(t))
def test_address_to_tuple_missing_host(self): """ Tests the address to tuple conversions when host is missing """ addr = 443 t = address_to_tuple(addr) self.assertEqual(t, (None, 443)) self.assertEqual(addr, tuple_to_address(t))
def test_address_to_tuple_missing_port(self): """ Tests the address to tuple conversions when port is missing """ addr = "127.0.0.1" t = address_to_tuple(addr) self.assertEqual(t, ("127.0.0.1", None)) self.assertEqual(addr, tuple_to_address(t))
def test_address_to_tuple_missing_host(self): """ Tests the address to tuple conversions when host is missing """ addr = 443 t = address_to_tuple(addr) self.assertEqual(t, (None, 443)) self.assertEqual(addr, tuple_to_address(t))
def on_connect(self): """ Callback invoked on connection with mapped service """ logger.info("Connection established with peer at %s" % tuple_to_address(self.remote_address)) for k in range(len(self.filters)): self.filters[k] = copy.deepcopy(self.filters[k]) self.filters[k].startup() self.io_stream.read_until_close(self.on_close, self.on_peer_message)
def on_close(self, *args, **kwargs): """ When web socket gets closed, close the connection to the service too """ logger.info("Closing connection with peer at %s" % tuple_to_address(self.remote_address)) logger.debug("Received args %s and %s", args, kwargs) #if not self.io_stream._closed: for message in args: self.on_peer_message(message) if not self.io_stream.closed(): self.io_stream.close() self.close()
def on_connect(self): """ Callback invoked on connection with mapped service """ logger.info("Connection established with peer at %s" % tuple_to_address(self.remote_address)) self.io_stream.read_until_close(self.on_close, self.on_peer_message)
def open(self): """ Open the connection to the service when the WebSocket connection has been established """ logger.info("Forwarding connection to server %s" % tuple_to_address(self.remote_address)) self.io_stream.connect(self.remote_address, self.on_connect)