Autobahn | Python Twisted服务器，用于检查API密钥并断开客户端连接

Question

我想向Autobahn Python WebSocket服务器添加一个简单的API密钥检查。 服务器应检查客户端HTTP标头中的密钥，并断开没有正确密钥的客户端。

我已经找到了解决方案，但是我不确定这是最好的解决方案（见下文）。 如果有人提出建议，我将不胜感激。

Answer 1

从API文档中的onConnect方法：

当您不想接受WebSocket连接请求时，抛出autobahn.websocket.types.ConnectionDeny。

你可以看到的例子之一的117线完成这件事在这里 。

我已经对此进行了测试，它不能干净地关闭连接。 但是，您将终止与未经身份验证的客户端的连接，因此您不希望进行关闭握手。

onClose回调函数使用wasClean参数，该参数可让您区分干净连接和不干净连接的关闭。

Answer 2

我的解决方案是在客户端连接到服务器后检查HTTP标头，如果客户端没有有效的API密钥，则关闭连接。

MY_API_KEY = u'12345'

class MyServerProtocol(WebSocketServerProtocol):

    def onConnect(self, request):
        print("Client connecting: {}".format(request.peer))

    def onOpen(self):
        # Check API Key
        if 'my-api-key' not in self.http_headers or\
            self.http_headers['my-api-key'] != MY_API_KEY:
            # Disconnect the client
            print('Missing/Invalid Key')
            self.sendClose( 4000, u'Missing/Invalid Key')

        # Register client
        self.factory.register(self)

我发现如果关闭onConnect内的连接，则会收到一条错误消息，提示我无法关闭尚未连接的连接。 上面的解决方案在客户端完全关闭，但在服务器端却表现异常。 日志输出为

dropping connection: None
Connection to/from tcp4:127.0.0.1:51967 was aborted locally
_connectionLost: [Failure instance: Traceback (failure with no frames): <class 'twisted.internet.error.ConnectionAborted'>: Connection was aborted locally, using.
    ]
WebSocket connection closed: None

服务器关闭消息为“无”的原因是服务器关闭了连接并且客户端没有发回邮件吗？ 有一个更好的方法吗？

更新：我接受了亨利·希思的回答，因为它似乎是官方支持的解决方案，即使它不能干净地关闭连接。 使用autobahn.websocket.types.ConnectionDeny ，解决方案变为

from autobahn.websocket.types import ConnectionDeny
MY_API_KEY = u'12345'

class MyServerProtocol(WebSocketServerProtocol):

    def onConnect(self, request):
        print("Client connecting: {}".format(request.peer))
        # Check API Key
        if 'my-api-key' not in request.headers or\
            request.headers['my-api-key'] != MY_API_KEY:
            # Disconnect the client
            print('Missing/Invalid Key')
            raise ConnectionDeny( 4000, u'Missing/Invalid Key')

    def onOpen(self):
        # Register client
        self.factory.register(self)

请注意，在onConnect中，HTTP标头可通过request.headers访问，而在onOpen中，它们可通过self.http_headers访问。