Merge pull request #97 from JohnDoe117/master

Add the `destroySession` method and support the `with` syntax.

Author: FlorianREGAZ

tls_client/cffi.py

@@ -26,4 +26,8 @@
 
 freeMemory = library.freeMemory
 freeMemory.argtypes = [ctypes.c_char_p]
-freeMemory.restype = ctypes.c_char_p
+freeMemory.restype = ctypes.c_char_p
+
+destroySession = library.destroySession
+destroySession.argtypes = [ctypes.c_char_p]
+destroySession.restype = ctypes.c_char_p

tls_client/dependencies/tls-client-arm64_test.dylib

@@ -1,4 +1,4 @@
-from .cffi import request, freeMemory
+from .cffi import request, freeMemory, destroySession
 from .cookies import cookiejar_from_dict, get_cookie_header, merge_cookies, extract_cookies_to_jar
 from .exceptions import TLSClientExeption
 from .response import build_response
@@ -271,6 +271,31 @@ def __init__(
         # debugging
         self.debug = debug
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        self.close()
+
+    def close(self):
+        destroySessionPayload = {
+            "sessionId": self._session_id
+        }
+
+        destroySessionResponse = destroySession(dumps(destroySessionPayload).encode('utf-8'))
+        # we dereference the pointer to a byte array
+        destroySessionResponse_bytes = ctypes.string_at(destroySessionResponse)
+        # convert our byte array to a string (tls client returns json)
+        destroySessionResponse_string = destroySessionResponse_bytes.decode('utf-8')
+        # convert response string to json
+        destroySessionResponse_object = loads(destroySessionResponse_string)
+
+        # print out output
+        # print(destroySessionResponse_object)
+        freeMemory(destroySessionResponse_object['id'].encode('utf-8'))
+
+        return destroySessionResponse_string
+
     def execute_request(
         self,
         method: str,