扭曲高速公路websocket被wss初始化两次

Question

我有一些使用Twisted实现的websocket协议，当我使用"ws"连接时，它们工作得很好，但是当我启用安全websockets时，__init__方法被调用两次。更具体地说，它被调用一次，然后连接显然失败，调用connectionLost，然后再次调用__init__，这次连接保持打开状态。

下面的代码就是一个例子。当我与wss连接时，websocket协议的__init__中的日志行被调用两次，但这不会发生在普通的websockets中。

进口套接字从datetime导入datetime从twisted.internet导入反应器

from twisted.internet.ssl import DefaultOpenSSLContextFactory
from autobahn.twisted.websocket import WebSocketServerProtocol, WebSocketServerFactory, listenWS
import txaio
txaio.use_twisted()

CERT_KEY = "certificate.key"
CERT_PATH = "certificate.crt"

def log(msg):
    print("{}: {}".format(str(datetime.now()), msg))

class TestProtocol(WebSocketServerProtocol):
    def __init__(self):
        super(TestProtocol, self).__init__()
        log("Test protocol init")
    def connectionLost(self, reason):
        WebSocketServerProtocol.connectionLost(self, reason)
        log("Connection closed: Reason is {}".format(reason))

class TestProtocolFactory(WebSocketServerFactory):
    protocol = TestProtocol

def init_websocket_protocol(factory_cls, port):
    try:
        key, crt = CERT_KEY, CERT_PATH
        context_factory = DefaultOpenSSLContextFactory(key, crt)
        connection_string = "wss://localhost:{}".format(str(port))
        factory = factory_cls(connection_string)
        listenWS(factory, contextFactory=context_factory)
        log("Port {} bound to test websocket server".format(str(port)))
    except socket.error as e:
        log("Server was unable to bind to a new port: ".format(str(e)))

def main():
    init_websocket_protocol(TestProtocolFactory, 9000)
    reactor.run()

if __name__ == '__main__':
    main()

Answer 1

目前推荐的API是使用端点。此外，twisted.internet.ssl.CertificateOptions是TLS连接的首选API。因此，通过这些更改，上面的代码看起来像这样:

from datetime import datetime
from autobahn.twisted.websocket import WebSocketServerProtocol, WebSocketServerFactory
from twisted.internet.ssl import CertificateOptions, PrivateCertificate, Certificate, KeyPair
from twisted.internet.endpoints import SSL4ServerEndpoint
from twisted.internet.task import react
from OpenSSL import crypto

CERT_KEY = "certificate.key"
CERT_PATH = "certificate.crt"

def log(msg):
    print("{}: {}".format(str(datetime.now()), msg))

class TestProtocol(WebSocketServerProtocol):
    def __init__(self):
        super(TestProtocol, self).__init__()
        log("Test protocol init")
    def connectionLost(self, reason):
        WebSocketServerProtocol.connectionLost(self, reason)
        log("Connection closed: Reason is {}".format(reason))

class TestProtocolFactory(WebSocketServerFactory):
    protocol = TestProtocol

def init_websocket_protocol(reactor, port):
    with open(CERT_KEY) as key_file, open(CERT_PATH) as cert_file:
        key = KeyPair.load(key_file.read(), crypto.FILETYPE_PEM).original
        cert = Certificate.loadPEM(cert_file.read()).original
    ctx = CertificateOptions(
        privateKey=key,
        certificate=cert,
    )
    return SSL4ServerEndpoint(reactor, port, ctx)

def main(reactor):
    ep = init_websocket_protocol(reactor, 9000)
    ep.listen(TestProtocolFactory())
    reactor.run()

if __name__ == '__main__':
    react(main)

当我运行这段代码并将Firefox指向它时，它会连接一次。你使用的浏览器端代码是什么样子的?