RAHHH

backend/lib/python3.12/site-packages/twisted/test/ssl_helpers.py

@@ -0,0 +1,66 @@
+# 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