def __init__(self, *args, **opts): super(Telnet, self).__init__() self.args = args self.opts = opts if len(args) == 1: if os.path.exists(args[0]): UNIXClient(channel=self.channel).register(self) host = dest = port = args[0] dest = (dest, ) else: raise OSError("Path %s not found" % args[0]) else: if not opts["udp"]: TCPClient(channel=self.channel).register(self) else: UDPClient(0, channel=self.channel).register(self) host, port = args port = int(port) dest = host, port self.host = host self.port = port print("Trying %s ..." % host) if not opts["udp"]: self.fire(connect(*dest, secure=opts["secure"])) else: self.fire(write((host, port), b"\x00"))
def test_basic(Poller, ipv6): m = Manager() + Poller() if ipv6: udp_server = UDP6Server(("::1", 0)) udp_client = UDP6Client(("::1", 0), channel="client") else: udp_server = UDPServer(0) udp_client = UDPClient(0, channel="client") server = Server() + udp_server client = Client() + udp_client server.register(m) client.register(m) m.start() try: assert pytest.wait_for(server, "ready") assert pytest.wait_for(client, "ready") wait_host(server) client.fire(write((server.host, server.port), b"foo")) assert pytest.wait_for(server, "data", b"foo") client.fire(close()) assert pytest.wait_for(client, "closed") server.fire(close()) assert pytest.wait_for(server, "closed") finally: m.stop()
def __init__(self, *args): """ Initialize the bus repeater component. :param args: """ super(BusRepeater, self).__init__('BUSREPEATER', *args) self._tcp_socket = self._udp_socket = None self._connected_tcp_endpoints = [] if self.config.tcp_port != 0 and \ self.config.tcp_ip is not None and \ self.config.tcp_enabled is True: address = self.config.tcp_ip + ':' + str(self.config.tcp_port) self.log('Opening listening socket on', address, lvl=debug) self._tcp_socket = TCPServer(address, channel=self.channel + '_tcp').register(self) if len(self.config.udp_endpoints) > 0 and \ self.config.udp_enabled is True: address = self.config.udp_ip + ':' + str(self.config.udp_port) self.log('Registering udp socket on', address, lvl=debug) self._udp_socket = UDPClient(0, channel=self.channel + '_udp').register(self)
def init(self, server, port): self.server = server self.port = int(port) self.transport = UDPClient(("127.0.0.1", 0), channel=self.channel).register(self) self.protocol = DNS(channel=self.channel).register(self)
def init(self, server, port, channel=channel): self.server = server self.port = int(port) self.transport = UDPClient(0, channel=self.channel).register(self) self.protocol = DNS(channel=self.channel).register(self) self.handler = ReturnResponse(channel=self.channel).register(self)
def init(self, server, port, query, verbose=False): self.server = server self.port = int(port) self.query = query if verbose: Debugger().register(self) self.transport = UDPClient(0).register(self) self.protocol = DNS().register(self) self.dummy = Dummy().register(self)