示例#1
0
    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"))
示例#2
0
文件: test_udp.py 项目: yws/circuits
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()
示例#3
0
    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)
示例#4
0
文件: client.py 项目: mlytics/udns
    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)
示例#5
0
    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)
示例#6
0
    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)