p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Improved stix2 core import statement

This commit is contained in:
VakarisZ 2020-04-08 16:06:02 +03:00
parent a55812edbe
commit 90e655d242
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/monkey_island/cc/services/attack/mitre_api_interface.py
View File

@ -1,7 +1,6 @@
from typing import List, Dict
from stix2 import FileSystemSource, Filter, CourseOfAction, AttackPattern
from stix2.core import STIXDomainObject
from stix2 import FileSystemSource, Filter, CourseOfAction, AttackPattern, core
class MitreApiInterface:
@ -33,14 +32,14 @@ class MitreApiInterface:
return all_techniques
@staticmethod
def get_stix2_external_reference_id(stix2_data: STIXDomainObject) -> str:
def get_stix2_external_reference_id(stix2_data: core.STIXDomainObject) -> str:
for reference in stix2_data['external_references']:
if reference['source_name'] == "mitre-attack" and 'external_id' in reference:
return reference['external_id']
return ''
@staticmethod
def get_stix2_external_reference_url(stix2_data: STIXDomainObject) -> str:
def get_stix2_external_reference_url(stix2_data: core.STIXDomainObject) -> str:
for reference in stix2_data['external_references']:
if 'url' in reference:
return reference['url']