[connector] Add entities types in the message to the workers

Author: Samuel Hassine

examples/stix2/import.py

@@ -9,7 +9,7 @@
 config = yaml.load(open(os.path.dirname(__file__) + '/../config.yml'))
 
 # File to import
-file_to_import = '/home/oxid/Dropbox/Developpement/Associations/Luatix/Produits/OpenCTI/cti/enterprise-attack/enterprise-attack.json'
+file_to_import = './enterprise-attack.json'
 
 # OpenCTI initialization
 opencti_api_client = OpenCTIApiClient(config['opencti']['url'], config['opencti']['token'])

pycti/opencti_connector_helper.py

@@ -71,22 +71,24 @@ def _reconnect(self):
         self.connection = self._connect()
         self.channel = self._create_channel()
 
-    def send_stix2_bundle(self, bundle):
+    def send_stix2_bundle(self, bundle, entities_types=[]):
         """
             This method send a STIX2 bundle to RabbitMQ to be consumed by workers
             :param bundle: A valid STIX2 bundle
+            :param entities_types: Entities types to ingest
         """
         if not self.channel.is_open:
             self._reconnect()
 
         # Validate the STIX 2 bundle
-        #validation = validate_string(bundle)
-        #if not validation.is_valid:
-            #raise ValueError('The bundle is not a valid STIX2 JSON:' + bundle)
+        # validation = validate_string(bundle)
+        # if not validation.is_valid:
+        # raise ValueError('The bundle is not a valid STIX2 JSON:' + bundle)
 
         # Prepare the message
         message = {
             'type': 'stix2-bundle',
+            'entities_types': entities_types,
             'content': base64.b64encode(bundle.encode('utf-8')).decode('utf-8')
         }