def test_ssl_proxy_with_decoder(self): fix_sslwrap() self.test_input = 'Hiya, this is a test'.encode('utf-8') keyfile = os.path.join(package_directory, 'templates/default/ssl/ssl.key') certfile = os.path.join(package_directory, 'templates/default/ssl/ssl.crt') mock_service = StreamServer(('127.0.0.1', 0), self.echo_server, keyfile=keyfile, certfile=certfile) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy( 'proxy', '127.0.0.1', mock_service.server_port, decoder='conpot.protocols.misc.ascii_decoder.AsciiDecoder', keyfile=keyfile, certfile=certfile) server = proxy.get_server('127.0.0.1', 0) gevent.spawn(server.start) gevent.sleep(1) s = wrap_socket(socket(), keyfile, certfile) s.connect(('127.0.0.1', server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_ssl_proxy(self): self.test_input = 'Hiya, this is a test' keyfile = os.path.join(package_directory, 'templates/example_ssl.key') certfile = os.path.join(package_directory, 'templates/example_ssl.crt') mock_service = StreamServer(('127.0.0.1', 0), self.echo_server, keyfile=keyfile, certfile=certfile) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy('proxy', '127.0.0.1', mock_service.server_port, keyfile=keyfile, certfile=certfile) server = proxy.get_server('127.0.0.1', 0) gevent.spawn(server.start) gevent.sleep(1) s = wrap_socket(socket(), keyfile, certfile) s.connect(('127.0.0.1', server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_ssl_proxy(self): fix_sslwrap() self.test_input = "Hiya, this is a test".encode("utf-8") keyfile = os.path.join(package_directory, "templates/default/ssl/ssl.key") certfile = os.path.join(package_directory, "templates/default/ssl/ssl.crt") mock_service = StreamServer(("127.0.0.1", 0), self.echo_server, keyfile=keyfile, certfile=certfile) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy( "proxy", "127.0.0.1", mock_service.server_port, keyfile=keyfile, certfile=certfile, ) server = proxy.get_server("127.0.0.1", 0) gevent.spawn(server.start) gevent.sleep(1) s = wrap_socket(socket(), keyfile=keyfile, certfile=certfile) s.connect(("127.0.0.1", server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_proxy(self): self.test_input = 'Hiya, this is a test' mock_service = StreamServer(('127.0.0.1', 0), self.echo_server) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy('proxy', '127.0.0.1', mock_service.server_port) server = proxy.get_server('127.0.0.1', 0) gevent.spawn(server.start) gevent.sleep(1) s = socket() s.connect(('127.0.0.1', server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_proxy(self): self.test_input = 'Hiya, this is a test'.encode('utf-8') mock_service = StreamServer(('127.0.0.1', 0), self.echo_server) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy('proxy', '127.0.0.1', mock_service.server_port) server = proxy.get_server('127.0.0.1', 0) gevent.spawn(server.start) gevent.sleep(1) s = socket() s.connect(('127.0.0.1', server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_proxy(self): self.test_input = "Hiya, this is a test".encode("utf-8") mock_service = StreamServer(("127.0.0.1", 0), self.echo_server) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy("proxy", "127.0.0.1", mock_service.server_port) server = proxy.get_server("127.0.0.1", 0) gevent.spawn(server.start) gevent.sleep(1) s = socket() s.connect(("127.0.0.1", server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_proxy(self): self.test_input = "Hiya, this is a test" mock_service = StreamServer(("127.0.0.1", 0), self.echo_server) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy("proxy", "127.0.0.1", mock_service.server_port) server = proxy.get_server("127.0.0.1", 0) gevent.spawn(server.start) gevent.sleep(1) s = socket() s.connect(("127.0.0.1", server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_proxy_with_decoder(self): self.test_input = 'Hiya, this is a test'.encode('utf-8') mock_service = StreamServer(('127.0.0.1', 0), self.echo_server) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy( 'proxy', '127.0.0.1', mock_service.server_port, decoder='conpot.protocols.misc.ascii_decoder.AsciiDecoder') server = proxy.get_server('127.0.0.1', 0) gevent.spawn(server.start) gevent.sleep(1) s = socket() s.connect(('127.0.0.1', server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_ssl_proxy(self): fix_sslwrap() self.test_input = "Hiya, this is a test" keyfile = os.path.join(package_directory, "templates/default/ssl/ssl.key") certfile = os.path.join(package_directory, "templates/default/ssl/ssl.crt") mock_service = StreamServer(("127.0.0.1", 0), self.echo_server, keyfile=keyfile, certfile=certfile) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy("proxy", "127.0.0.1", mock_service.server_port, keyfile=keyfile, certfile=certfile) server = proxy.get_server("127.0.0.1", 0) gevent.spawn(server.start) gevent.sleep(1) s = wrap_socket(socket(), keyfile, certfile) s.connect(("127.0.0.1", server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)
def test_ssl_proxy_with_decoder(self): fix_sslwrap() self.test_input = 'Hiya, this is a test'.encode('utf-8') keyfile = os.path.join(package_directory, 'templates/default/ssl/ssl.key') certfile = os.path.join(package_directory, 'templates/default/ssl/ssl.crt') mock_service = StreamServer(('127.0.0.1', 0), self.echo_server, keyfile=keyfile, certfile=certfile) gevent.spawn(mock_service.start) gevent.sleep(1) proxy = Proxy('proxy', '127.0.0.1', mock_service.server_port, decoder='conpot.protocols.misc.ascii_decoder.AsciiDecoder', keyfile=keyfile, certfile=certfile) server = proxy.get_server('127.0.0.1', 0) gevent.spawn(server.start) gevent.sleep(1) s = wrap_socket(socket(), keyfile, certfile) s.connect(('127.0.0.1', server.server_port)) s.sendall(self.test_input) received = s.recv(len(self.test_input)) self.assertEqual(self.test_input, received) mock_service.stop(1)