Fixed wrap_socket

qeeqbox · Sep 12, 2024 · d452b94 · d452b94
1 parent cf07da5
commit d452b94
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/honeypots/elastic_server.py b/honeypots/elastic_server.py
@@ -2,7 +2,7 @@
 from contextlib import suppress
 from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
 from json import dumps
-from ssl import wrap_socket
+from ssl import SSLContext, PROTOCOL_TLS_SERVER
 from urllib.parse import urlparse
 from zlib import compressobj, DEFLATED
 
@@ -312,10 +312,10 @@ def get_auth_key(self):
         with create_certificate() as (cert, key):
             server = CustomElasticServer((self.ip, self.port))
             server.set_auth_key(self.username, self.password)
-            server.socket = wrap_socket(
+            ctx = SSLContext(PROTOCOL_TLS_SERVER)
+            ctx.load_cert_chain(certfile=cert, keyfile=key)
+            server.socket = ctx.wrap_socket(
                 server.socket,
-                keyfile=key,
-                certfile=cert,
                 server_side=True,
             )
             server.serve_forever()