mirror of https://github.com/MISP/misp-modules
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
"""vt_graph_parser.helpers.wrappers.
|
|
|
|
This module provides a Python object wrapper for MISP objects.
|
|
"""
|
|
|
|
|
|
class MispAttribute(object):
|
|
"""Python object wrapper for MISP attribute.
|
|
|
|
Attributes:
|
|
type (str): VirusTotal node type.
|
|
category (str): MISP attribute category.
|
|
value (str): node id.
|
|
label (str): node name.
|
|
misp_type (str): MISP node type.
|
|
"""
|
|
|
|
MISP_TYPES_REFERENCE = {
|
|
"hostname": "domain",
|
|
"domain": "domain",
|
|
"ip-src": "ip_address",
|
|
"ip-dst": "ip_address",
|
|
"url": "url",
|
|
"filename|X": "file",
|
|
"filename": "file",
|
|
"md5": "file",
|
|
"sha1": "file",
|
|
"sha256": "file",
|
|
"target-user": "victim",
|
|
"target-email": "email"
|
|
}
|
|
|
|
def __init__(self, misp_type, category, value, label=""):
|
|
"""Constructor for a MispAttribute.
|
|
|
|
Args:
|
|
misp_type (str): MISP type attribute.
|
|
category (str): MISP category attribute.
|
|
value (str): attribute value.
|
|
label (str): attribute label.
|
|
"""
|
|
if misp_type.startswith("filename|"):
|
|
label, value = value.split("|")
|
|
misp_type = "filename|X"
|
|
if misp_type == "filename":
|
|
label = value
|
|
|
|
self.type = self.MISP_TYPES_REFERENCE.get(misp_type)
|
|
self.category = category
|
|
self.value = value
|
|
self.label = label
|
|
self.misp_type = misp_type
|
|
|
|
def __eq__(self, other):
|
|
return (isinstance(other, self.__class__) and self.value == other.value and
|
|
self.type == other.type)
|
|
|
|
def __repr__(self):
|
|
return 'MispAttribute("{type}", "{category}", "{value}")'.format(type=self.type, category=self.category, value=self.value)
|