80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
"""Python APIs for STIX 2."""
|
|
|
|
# flake8: noqa
|
|
|
|
from .bundle import Bundle
|
|
from .observables import Artifact, AutonomousSystem, EmailAddress, \
|
|
EmailMessage, File
|
|
from .other import ExternalReference, KillChainPhase, MarkingDefinition, \
|
|
GranularMarking, StatementMarking, TLPMarking
|
|
from .sdo import AttackPattern, Campaign, CourseOfAction, Identity, Indicator, \
|
|
IntrusionSet, Malware, ObservedData, Report, ThreatActor, Tool, \
|
|
Vulnerability
|
|
from .sro import Relationship, Sighting
|
|
from .utils import get_dict
|
|
from . import exceptions
|
|
|
|
|
|
OBJ_MAP = {
|
|
'attack-pattern': AttackPattern,
|
|
'campaign': Campaign,
|
|
'course-of-action': CourseOfAction,
|
|
'identity': Identity,
|
|
'indicator': Indicator,
|
|
'intrusion-set': IntrusionSet,
|
|
'malware': Malware,
|
|
'marking-definition': MarkingDefinition,
|
|
'observed-data': ObservedData,
|
|
'report': Report,
|
|
'relationship': Relationship,
|
|
'threat-actor': ThreatActor,
|
|
'tool': Tool,
|
|
'sighting': Sighting,
|
|
'vulnerability': Vulnerability,
|
|
}
|
|
|
|
OBJ_MAP_OBSERVABLE = {
|
|
'artifact': Artifact,
|
|
'autonomous-system': AutonomousSystem,
|
|
'email-address': EmailAddress,
|
|
'email-message': EmailMessage,
|
|
'file': File,
|
|
}
|
|
|
|
|
|
def parse(data):
|
|
"""Deserialize a string or file-like object into a STIX object"""
|
|
|
|
obj = get_dict(data)
|
|
|
|
if 'type' not in obj:
|
|
# TODO parse external references, kill chain phases, and granular markings
|
|
pass
|
|
else:
|
|
try:
|
|
obj_class = OBJ_MAP[obj['type']]
|
|
except KeyError:
|
|
# TODO handle custom objects
|
|
raise ValueError("Can't parse unknown object type '%s'!" % obj['type'])
|
|
return obj_class(**obj)
|
|
|
|
return obj
|
|
|
|
|
|
def parse_observable(data, _valid_refs):
|
|
"""Deserialize a string or file-like object into a STIX Cyber Observable
|
|
object.
|
|
"""
|
|
|
|
obj = get_dict(data)
|
|
obj['_valid_refs'] = _valid_refs
|
|
|
|
if 'type' not in obj:
|
|
raise ValueError("'type' is a required field!")
|
|
try:
|
|
obj_class = OBJ_MAP_OBSERVABLE[obj['type']]
|
|
except KeyError:
|
|
# TODO handle custom objects
|
|
raise ValueError("Can't parse unknown object type '%s'!" % obj['type'])
|
|
return obj_class(**obj)
|