python asyncore跟踪客户

时间:2012-05-15 16:31:34

标签: python sockets asyncore

我正在编写一个简单的套接字服务器,我想跟踪客户端状态(身份验证和内容)。每次调用handle_read()时,我都不知道该特定客户端。如果我知道客户的ID或其他东西会有所帮助。这是我到目前为止所做的:

import asyncore
import socket

class EchoHandler(asyncore.dispatcher_with_send):

    def handle_read(self):
        data = self.recv(8192)
        self.send(data)

class EchoServer(asyncore.dispatcher):

    def __init__(self, host, port):
        asyncore.dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.set_reuse_addr()
        self.bind((host, port))
        self.listen(5)

    def handle_accept(self):
        pair = self.accept()
        if pair is None:
            pass
        else:
            sock, addr = pair
            print 'Incoming connection from %s' % repr(addr)
            handler = EchoHandler(sock)

server = EchoServer('localhost', 8080)
asyncore.loop()

2 个答案:

答案 0 :(得分:4)

当您从repr(addr)收到连接时,您确实知道客户端的ID。返回的元组是该客户端的IP和唯一编号,可以在您希望将数据发送回该客户端时使用。如果您想保留一个客户列表,您应该考虑一个字典,当收到传入连接时,它会存储该客户端的信息。

如果您希望将这些信息传递给handle_read函数,那么您的代码将如下所示:

class EchoHandler(asyncore.dispatcher_with_send):
    def setAddr(self, addr):
        self.addr = addr

    def handle_read(self):
        data = self.recv(8192)
        print '%s from %s' % (data, self.addr)
        self.send(data)

class EchoServer(asyncore.dispatcher):

    def __init__(self, host, port):
        asyncore.dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.set_reuse_addr()
        self.bind((host, port))
        self.listen(5)

    def handle_accept(self):
        pair = self.accept()
        if pair is None:
            pass
        else:
            sock, addr = pair
            print 'Incoming connection from %s' % repr(addr)
            handler = EchoHandler(sock)
            handler.setAddr(addr) #Set the address of the sender in the EchoHandler

只需将客户端的地址发送给EchoHandler,现在就知道它来自哪里了。希望这有助于/有效!

答案 1 :(得分:1)

您可以直接从套接字获取的唯一“id”是对套接字和发送方地址的引用。其他所有内容都基于您收到的数据 - 您的客户可以例如将其id作为前16个字节发送,之后的任何内容都是数据。这意味着有效地实施满足您要求的协议。

更简单,更好的身份验证解决方案是在python标准库中使用某些库,如sslssl库以易于与python套接字一起使用的形式提供加密和身份验证机制。如果您需要安全身份验证,我强烈建议您使用现有解决方案。