67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
# Copyright (c) Twisted Matrix Laboratories.
|
|
# See LICENSE for details.
|
|
|
|
"""
|
|
Helper classes for twisted.test.test_ssl.
|
|
|
|
They are in a separate module so they will not prevent test_ssl importing if
|
|
pyOpenSSL is unavailable.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from OpenSSL import SSL
|
|
|
|
from twisted.internet import ssl
|
|
from twisted.python.compat import nativeString
|
|
from twisted.python.filepath import FilePath
|
|
|
|
certPath = nativeString(FilePath(__file__.encode("utf-8")).sibling(b"server.pem").path)
|
|
|
|
|
|
class ClientTLSContext(ssl.ClientContextFactory):
|
|
"""
|
|
SSL Context Factory for client-side connections.
|
|
"""
|
|
|
|
isClient = 1
|
|
|
|
def getContext(self) -> SSL.Context:
|
|
"""
|
|
Return an L{SSL.Context} to be use for client-side connections.
|
|
|
|
Will not return a cached context.
|
|
This is done to improve the test coverage as most implementation
|
|
are caching the context.
|
|
"""
|
|
return SSL.Context(SSL.SSLv23_METHOD)
|
|
|
|
|
|
class ServerTLSContext:
|
|
"""
|
|
SSL Context Factory for server-side connections.
|
|
"""
|
|
|
|
isClient = 0
|
|
|
|
def __init__(
|
|
self, filename: str | bytes = certPath, method: int | None = None
|
|
) -> None:
|
|
self.filename = filename
|
|
if method is None:
|
|
method = SSL.SSLv23_METHOD
|
|
|
|
self._method = method
|
|
|
|
def getContext(self) -> SSL.Context:
|
|
"""
|
|
Return an L{SSL.Context} to be use for server-side connections.
|
|
|
|
Will not return a cached context.
|
|
This is done to improve the test coverage as most implementation
|
|
are caching the context.
|
|
"""
|
|
ctx = SSL.Context(self._method)
|
|
ctx.use_certificate_file(self.filename)
|
|
ctx.use_privatekey_file(self.filename)
|
|
return ctx
|