[英]twisted autobahn websocket being initialized twice with wss
我有一些通过Twisted实现的websocket协议,当我使用“ ws”连接时,它们可以正常工作,但是当我启用安全的websocket时, __init__
方法被调用了两次。 更具体地说,它被调用一次,然后连接显然失败,并调用connectionLost,然后再次调用__init__
,这一次连接保持打开状态。
下面的代码说明了这一点。 当我与wss连接时,websocket协议的__init__
中的日志行被调用了两次,但是普通的websocket不会发生这种情况。
从datetime导入socket从twisted.internet导入反应器导入datetime
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()
这些天推荐的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指向它时,它会连接一次。 您正在使用的浏览器端代码是什么样的?
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.