[client] Pass the token when sending message

Author: Samuel Hassine

pycti/api/opencti_api_client.py

@@ -71,6 +71,7 @@ def __init__(self, url, token, log_level="info", ssl_verify=False):
         logging.basicConfig(level=numeric_level)
 
         # Define API
+        self.api_token = token
         self.api_url = url + "/graphql"
         self.request_headers = {"Authorization": "Bearer " + token}
 
@@ -110,6 +111,12 @@ def __init__(self, url, token, log_level="info", ssl_verify=False):
                 "OpenCTI API is not reachable. Waiting for OpenCTI API to start or check your configuration..."
             )
 
+    def get_token(self):
+        return self.api_token
+
+    def set_token(self, token):
+        self.request_headers = {"Authorization": "Bearer " + token}
+
     def query(self, query, variables={}):
         query_var = {}
         files_vars = []

pycti/connector/opencti_connector_helper.py

@@ -286,6 +286,7 @@ def _send_bundle(self, channel, bundle, entities_types=None, update=False):
             "job_id": job_id,
             "entities_types": entities_types,
             "update": update,
+            "token": self.opencti_token,
             "content": base64.b64encode(bundle.encode("utf-8")).decode("utf-8"),
         }