Python 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()

当您从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,现在您就知道它来自哪里了。希望这有帮助/有效!

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

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

最新更新