mirror of https://github.com/MISP/PyMISP
				
				
				
			
		
			
				
	
	
		
			286 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
			
		
		
	
	
			286 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
| 
 | |
| # -*- coding: utf-8 -*-
 | |
| 
 | |
| import os
 | |
| 
 | |
| from .. import MISPEvent
 | |
| try:
 | |
|     from bs4 import BeautifulSoup
 | |
|     has_bs4 = True
 | |
| except ImportError:
 | |
|     has_bs4 = False
 | |
| 
 | |
| iocMispMapping = {
 | |
|     'CookieHistoryItem/HostName': {'type': 'hostname', 'comment': 'CookieHistory.'},
 | |
| 
 | |
|     'DriverItem/DriverName': {'category': 'Artifacts dropped', 'type': 'other', 'comment': 'DriverName.'},
 | |
|     'DriverItem/CertificateIssuer': {'category': 'Artifacts dropped', 'type': 'other', 'comment': 'DriverCertificateIssuer.'},
 | |
|     'DriverItem/DeviceItem/AttachedDeviceName': {'category': 'Artifacts dropped', 'type': 'other', 'comment': 'DriverDeviceName. '},
 | |
| 
 | |
|     'DnsEntryItem/Host': {'type': 'domain'},
 | |
|     'DnsEntryItem/RecordName': {'type': 'domain'},
 | |
| 
 | |
|     'Email/To': {'type': 'target-email'},
 | |
|     'Email/Date': {'type': 'comment', 'comment': 'EmailDate.'},
 | |
|     'Email/Body': {'type': 'email-subject'},
 | |
|     'Email/From': {'type': 'email-dst'},
 | |
|     'Email/Subject': {'type': 'email-subject'},
 | |
|     'Email/Attachment/Name': {'type': 'email-attachment'},
 | |
| 
 | |
|     'FileItem/Md5sum': {'type': 'md5'},
 | |
|     'FileItem/Sha1sum': {'type': 'sha1'},
 | |
|     'FileItem/FileName': {'type': 'filename'},
 | |
|     'FileItem/FullPath': {'type': 'filename'},
 | |
|     'FileItem/FilePath': {'type': 'filename'},
 | |
|     'FileItem/Sha256sum': {'type': 'sha256'},
 | |
|     'FileItem/DevicePath': {'type': 'comment', 'comment': 'DevicePath. '},
 | |
|     'FileItem/SizeInBytes': {'type': 'size-in-bytes'},
 | |
|     'FileItem/PEInfo/Type': {'type': 'comment', 'comment': 'PE Type. '},
 | |
|     'FileItem/FileExtension': {'type': 'comment', 'comment': 'FileExtension. '},
 | |
|     'FileItem/FilenameCreated': {'type': 'filename', 'comment': 'FilenameCreated. '},
 | |
|     'FileItem/StringList/string': {'type': 'pattern-in-file', 'comment': 'string list. '},
 | |
|     'FileItem/PEInfo/PETimeStamp': {'type': 'pattern-in-file', 'comment': 'PE TimeStamp. '},
 | |
|     'FileItem/PEInfo/Exports/DllName': {'type': 'pattern-in-memory', 'comment': 'PE export DllName. '},
 | |
|     'FileItem/PEInfo/Sections/Section/Name': {'type': 'pattern-in-memory', 'comment': 'PE SectionName. '},
 | |
|     'FileItem/PEInfo/DetectedAnomalies/string': {'type': 'pattern-in-file', 'comment': 'PE DEtected AnomaliesString. '},
 | |
|     'FileItem/PEInfo/Exports/NumberOfFunctions': {'type': 'pattern-in-file', 'comment': 'PE Export NumberOfFunctions. '},
 | |
|     'FileItem/PEInfo/ImportedModules/Module/Name': {'type': 'pattern-in-file', 'comment': 'PE ImportedModulesName. '},
 | |
|     'FileItem/PEInfo/DigitalSignature/Description': {'type': 'comment', 'comment': 'PE DigitalSignatureDescription. '},
 | |
|     'FileItem/PEInfo/DigitalSignature/SignatureExists': {'type': 'comment', 'comment': 'PE SignatureExists. '},
 | |
|     'FileItem/PEInfo/Exports/ExportedFunctions/string': {'type': 'comment', 'comment': 'PE ExportedFunctions. '},
 | |
|     'FileItem/PEInfo/DigitalSignature/CertificateIssuer': {'type': 'comment', 'comment': 'PE SignatureCertificateIssuer. '},
 | |
|     'FileItem/PEInfo/DigitalSignature/SignatureVerified': {'type': 'comment', 'comment': 'PE SignatureVerified. '},
 | |
|     'FileItem/PEInfo/DigitalSignature/CertificateSubject': {'type': 'other', 'comment': 'PE CertificateDigitalSignatureSubject. '},
 | |
|     'FileItem/PEInfo/ResourceInfoList/ResourceInfoItem/Name': {'type': 'comment', 'comment': 'PE ResourceName. '},
 | |
|     'FileItem/PEInfo/ResourceInfoList/ResourceInfoItem/Type': {'type': 'comment', 'comment': 'PE ResourceType. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/Language': {'type': 'pattern-in-file', 'comment': 'PE LanguageVersion. '},
 | |
|     'FileItem/PEInfo/ResourceInfoList/ResourceInfoItem/Language': {'type': 'pattern-in-file', 'comment': 'PE LanguageResource. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/CompanyName': {'type': 'pattern-in-file', 'comment': 'PE versionInfo CompanyName. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/FileVersion': {'type': 'pattern-in-file', 'comment': 'PE Version. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/ProductName': {'type': 'pattern-in-file', 'comment': 'PE ProductName. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/InternalName': {'type': 'pattern-in-file', 'comment': 'PE InternalName. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/LegalCopyright': {'type': 'pattern-in-file', 'comment': 'PE LegalCopyright. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/ProductVersion': {'type': 'pattern-in-file', 'comment': 'PE ProductVersion. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/FileDescription': {'type': 'comment', 'comment': 'PE FileDescription .'},
 | |
|     'FileItem/PEInfo/ImportedModules/Module/ImportedFunctions/string': {'type': 'pattern-in-file', 'comment': 'PE ImportedModules. '},
 | |
|     'FileItem/PEInfo/VersionInfoList/VersionInfoItem/OriginalFilename': {'type': 'pattern-in-file', 'comment': 'OriginalFilename of PE. '},
 | |
| 
 | |
|     'FormHistoryItem/HostName': {'type': 'hostname', 'comment': 'FormHistory. '},
 | |
| 
 | |
|     'Network/URI': {'type': 'uri'},
 | |
|     'Network/DNS': {'type': 'domain'},
 | |
|     'Network/String': {'type': 'url'},
 | |
|     'Network/IPRange': {'type': 'ip-dst'},
 | |
|     'Network/UserAgent': {'type': 'user-agent'},
 | |
| 
 | |
|     'PortItem/localIP': {'type': 'ip-src'},
 | |
|     'PortItem/remoteIP': {'type': 'ip-dst'},
 | |
|     'PortItem/remotePort': {'type': 'pattern-in-traffic', 'comment': 'RemotePort. '},
 | |
| 
 | |
|     'ProcessItem/name': {'type': 'pattern-in-memory', 'comment': 'ProcessName. '},
 | |
|     'ProcessItem/path': {'type': 'pattern-in-memory', 'comment': 'ProcessPath. '},
 | |
|     'ProcessItem/Mutex': {'type': 'mutex', 'comment': 'mutex'},
 | |
|     'ProcessItem/arguments': {'type': 'pattern-in-memory', 'comment': 'ProcessArguments. '},
 | |
|     'ProcessItem/NamedPipe': {'type': 'named pipe'},
 | |
|     'ProcessItem/Pipe/Name': {'type': 'named pipe'},
 | |
|     'ProcessItem/Mutex/Name': {'type': 'mutex', 'comment': 'MutexName. '},
 | |
|     'ProcessItem/Event/Name': {'type': 'pattern-in-memory', 'comment': 'ProcessEventName. '},
 | |
|     'ProcessItem/StringList/string': {'type': 'pattern-in-memory', 'comment': 'StringlistName. '},
 | |
|     'ProcessItem/HandleList/Handle/Name': {'type': 'pattern-in-memory', 'comment': 'ProcessHandleListName'},
 | |
|     'ProcessItem/HandleList/Handle/Type': {'type': 'pattern-in-memory', 'comment': 'ProcessHandleType'},
 | |
|     'ProcessItem/SectionList/MemorySection/Name': {'type': 'pattern-in-memory', 'comment': 'ProcessSectionMemoryName'},
 | |
|     'ProcessItem/SectionList/MemorySection/PEInfo/Exports/DllName': {'type': 'pattern-in-memory', 'comment': 'ProcessMemoryPEExportsDllName'},
 | |
|     'ProcessItem/SectionList/MemorySection/PEInfo/Sections/Section/Name': {'type': 'pattern-in-memory', 'comment': 'Section name from PE in process memory section'},
 | |
| 
 | |
|     'RegistryItem/Text': {'type': 'regkey', 'comment': 'RegistryText. '},
 | |
|     'RegistryItem/Path': {'type': 'regkey', 'comment': 'RegistryPath. '},
 | |
|     'RegistryItem/Value': {'type': 'regkey', 'comment': 'RegistryValue. '},
 | |
|     'RegistryItem/KeyPath': {'type': 'regkey', 'comment': 'RegistryKeyPath. '},
 | |
|     'RegistryItem/ValueName': {'type': 'regkey', 'comment': 'RegistryValueName. '},
 | |
| 
 | |
|     'RouteEntryItem/Destination': {'type': 'ip-dst'},
 | |
|     'RouteEntryItem/Destination/IP': {'type': 'ip-dst', 'comment': 'RouteDestination. '},
 | |
|     'RouteEntryItem/Destination/string': {'type': 'hostname', 'comment': 'RouteDestination. '},
 | |
| 
 | |
| 
 | |
|     'ServiceItem/name': {'type': 'windows-service-name'},
 | |
|     'ServiceItem/type': {'type': 'pattern-in-memory', 'comment': 'ServiceType. '},
 | |
|     'ServiceItem/startedAs': {'type': 'pattern-in-memory', 'comment': 'ServiceStartedAs. '},
 | |
|     'ServiceItem/serviceDLL': {'type': 'pattern-in-memory', 'comment': 'ServiceDll. '},
 | |
|     'ServiceItem/description': {'type': 'comment', 'comment': 'ServiceDescription. '},
 | |
|     'ServiceItem/descriptiveName': {'type': 'windows-service-displayname'},
 | |
|     'ServiceItem/serviceDLLmd5sum': {'type': 'md5', 'comment': 'ServiceDLL. '},
 | |
|     'ServiceItem/serviceDLLsha1sum': {'type': 'sha1', 'comment': 'ServiceDLL. '},
 | |
|     'ServiceItem/serviceDLLsha256sum': {'type': 'sha256', 'comment': 'ServiceDLL. '},
 | |
|     'ServiceItem/serviceDLLSignatureVerified': {'type': 'pattern-in-memory', 'comment': 'ServiceDllSignatureVerified. '},
 | |
| 
 | |
|     'Snort/Snort': {'type': 'snort'},
 | |
| 
 | |
|     'SystemInfoItem/HostName': {'type': 'hostname', 'comment': 'SystemInfo. '},
 | |
| 
 | |
|     'TaskItem/Name': {'type': 'windows-scheduled-task', 'comment': 'TaskName. '},
 | |
|     'TaskItem/sha1sum': {'type': 'windows-scheduled-task', 'comment': 'TashSha1. '},
 | |
|     'TaskItem/sha256sum': {'type': 'windows-scheduled-task', 'comment': 'TashSha256. '},
 | |
|     'TaskItem/AccountName': {'type': 'windows-scheduled-task', 'comment': 'TaskAccountName'},
 | |
|     'TaskItem/ActionList/Action/ExecProgramPath': {'type': 'windows-scheduled-task', 'comment': 'TaskExecProgramPath. '},
 | |
|     'TaskItem/TriggerList/Trigger/TriggerFrequency': {'type': 'windows-scheduled-task', 'comment': 'TaskTriggerFrequency. '},
 | |
| 
 | |
|     'UrlHistoryItem/URL': {'type': 'url', 'comment': 'UrlHistory. '},
 | |
|     'UrlHistoryItem/HostName': {'type': 'hostname', 'comment': 'UrlHistory. '},
 | |
| 
 | |
|     'Yara/Yara': {'type': 'yara'}
 | |
| }
 | |
| 
 | |
| iocMispCompositeMapping = {
 | |
|     # mapping for composite object
 | |
|     # maybe later filename|sizeinbyte
 | |
|     'FileItem/FileName|FileItem/Md5sum': {'type': 'filename|md5'},
 | |
|     'FileItem/FileName|FileItem/Sha1sum': {'type': 'filename|sha1'},
 | |
|     'FileItem/FileName|FileItem/Sha256sum': {'type': 'filename|sha256'},
 | |
|     'Network/DNS|PortItem/remoteIP': {'type': 'domain|ip'},
 | |
|     'PortItem/remoteIP|PortItem/remotePort': {'type': 'ip-dst|port'},
 | |
|     'RegistryItem/Path|RegistryItem/Value': {'type': 'regkey|value'},
 | |
|     'RegistryItem/KeyPath|RegistryItem/Value': {'type': 'regkey|value'},
 | |
|     'RegistryItem/Path|RegistryItem/Text': {'type': 'regkey|value'}
 | |
| }
 | |
| 
 | |
| 
 | |
| def extract_field(report, field_name):
 | |
|     if report:
 | |
|         data = report.find(field_name.lower())
 | |
|         if data and hasattr(data, 'text'):
 | |
|             return data.text
 | |
|     return ''
 | |
| 
 | |
| 
 | |
| def load_openioc_file(openioc_path):
 | |
|     if not os.path.exists(openioc_path):
 | |
|         raise Exception("Path doesn't exists.")
 | |
|     with open(openioc_path, 'r') as f:
 | |
|         return load_openioc(f)
 | |
| 
 | |
| 
 | |
| def load_openioc(openioc):
 | |
|     # Takes a opened file, or a string
 | |
|     if not has_bs4:
 | |
|         raise Exception('You need to install BeautifulSoup: pip install bs4')
 | |
|     misp_event = MISPEvent()
 | |
|     iocreport = BeautifulSoup(openioc, "html.parser")
 | |
|     # Set event fields
 | |
|     info = extract_field(iocreport, 'short_description')
 | |
|     if info:
 | |
|         misp_event.info = info
 | |
|     date = extract_field(iocreport, 'authored_date')
 | |
|     if date:
 | |
|         misp_event.set_date(date)
 | |
|     # Set special attributes
 | |
|     description = extract_field(iocreport, 'description')
 | |
|     if description:
 | |
|         if not misp_event.info:
 | |
|             misp_event.info = description
 | |
|         else:
 | |
|             misp_event.add_attribute('comment', description)
 | |
|     if not misp_event.info:
 | |
|         misp_event.info = 'OpenIOC import'
 | |
|     author = extract_field(iocreport, 'authored_by')
 | |
|     if author:
 | |
|         misp_event.add_attribute('comment', author)
 | |
|     misp_event = set_all_attributes(iocreport, misp_event)
 | |
|     return misp_event
 | |
| 
 | |
| 
 | |
| def get_mapping(openioc_type, mappingDict=iocMispMapping):
 | |
|     t = openioc_type.lower()
 | |
|     for k, v in mappingDict.items():
 | |
|         if k.lower() == t:
 | |
|             return v
 | |
|     return False
 | |
| 
 | |
| 
 | |
| def set_values(value1, value2=None):
 | |
|     attribute_values = {}
 | |
| 
 | |
|     if value2 is not None:
 | |
|         # construct attribut composite value
 | |
|         value = "{}|{}".format(extract_field(value1, 'Content'),
 | |
|                                extract_field(value2, 'Content')
 | |
|                                )
 | |
|     else:
 | |
|         value = extract_field(value1, 'Content')
 | |
| 
 | |
|     if value:
 | |
|         attribute_values['value'] = value
 | |
|     else:
 | |
|         return None
 | |
| 
 | |
|     if value2 is not None:
 | |
|         # construct attribut composite type
 | |
|         compositeMapping = '{}|{}'.format(value1.find('context')['search'], value2.find('context')['search'])
 | |
|         mapping = get_mapping(compositeMapping, mappingDict=iocMispCompositeMapping)
 | |
|     else:
 | |
|         context_search = value1.find('context')['search']
 | |
|         content_type = value1.find('content').get('type', None)
 | |
|         if "RouteEntryItem/Destination" in context_search and content_type:
 | |
|             mapping = get_mapping(context_search + '/' + content_type)
 | |
|         else:
 | |
|             mapping = get_mapping(context_search)
 | |
| 
 | |
|     if mapping:
 | |
|         attribute_values.update(mapping)
 | |
|     else:
 | |
|         # Unknown mapping, assign to default
 | |
|         attribute_values['category'] = 'External analysis'
 | |
|         attribute_values['type'] = 'other'
 | |
| 
 | |
|     # change value to composite
 | |
|     # 127.0.0.1:80 ip-* to 127.0.0.1|80 ip-*|port
 | |
|     if attribute_values['type'] in ['ip-src', 'ip-dst'] and attribute_values['value'].count(':') == 1:
 | |
|         attribute_values['type'] = attribute_values['type'] + '|port'
 | |
|         attribute_values['value'] = attribute_values['value'].replace(':', '|')
 | |
| 
 | |
|     attribute_values["comment"] = '{}{}'.format(extract_field(value1, 'Comment'),
 | |
|                                                 extract_field(value2, 'Comment')
 | |
|                                                 )
 | |
| 
 | |
|     return attribute_values
 | |
| 
 | |
| 
 | |
| def set_all_attributes(openioc, misp_event):
 | |
|     processed = set()
 | |
| 
 | |
|     # check for composite item
 | |
|     for composite in openioc.find_all("indicator", operator="AND"):
 | |
| 
 | |
|         # check for composite number under
 | |
|         childs = composite.find_all('indicatoritem')
 | |
| 
 | |
|         if len(childs) == 2:
 | |
|             childList = [child.find('context')['search'] for child in childs]
 | |
| 
 | |
|             def check_and_add(value1, value2):
 | |
|                 if (value1 and value2) in childList:
 | |
|                     if childs[0].find('context')['search'] == value1:
 | |
|                         attribute_values = set_values(childs[0], childs[1])
 | |
|                     else:
 | |
|                         attribute_values = set_values(childs[1], childs[0])
 | |
| 
 | |
|                     misp_event.add_attribute(**attribute_values)
 | |
|                     processed.add(childs[0]['id'])
 | |
|                     processed.add(childs[1]['id'])
 | |
| 
 | |
|             for k in iocMispCompositeMapping:
 | |
|                 check_and_add(k.split('|')[0], k.split('|')[1])
 | |
| 
 | |
|     for item in openioc.find_all("indicatoritem"):
 | |
|         # check if id in processed list
 | |
|         if item['id'] in processed:
 | |
|             continue
 | |
|         attribute_values = set_values(item)
 | |
| 
 | |
|         if attribute_values is None:
 | |
|             continue
 | |
| 
 | |
|         misp_event.add_attribute(**attribute_values)
 | |
| 
 | |
|     return misp_event
 |