TLS connectivity

Author: Alex Gaetano Padula

cursusdb-py/main.py

@@ -32,6 +32,11 @@ def __init__(self, host, port, username, password, tls):
 
     # Instance method
     def connect(self):
+        if self.tls:
+            context = ssl.create_default_context()
+            context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
+            self.sock = context.wrap_socket(socket.socket(socket.AF_INET), server_hostname=self.host)
+            
         self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self.sock.connect((self.host, self.port))
         auth_str = f"{self.username}\\0{self.password}".encode("utf8")