mirror of https://github.com/MISP/PyMISP
40 lines
1.6 KiB
Python
Executable File
40 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import json
|
|
from pymisp import PyMISP
|
|
from pymisp.tools import GenericObjectGenerator
|
|
from keys import misp_url, misp_key, misp_verifycert
|
|
import argparse
|
|
|
|
"""
|
|
Sample usage:
|
|
./add_generic_object.py -e 5065 -t email -l '[{"to": "undisclosed@ppp.com"}, {"to": "second.to@mail.com"}]'
|
|
"""
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Create a MISP Object selectable by type starting from a dictionary')
|
|
parser.add_argument("-e", "--event", required=True, help="Event ID to update")
|
|
parser.add_argument("-t", "--type", required=True, help="Type of the generic object")
|
|
parser.add_argument("-l", "--attr_list", required=True, help="List of attributes")
|
|
args = parser.parse_args()
|
|
|
|
pymisp = PyMISP(misp_url, misp_key, misp_verifycert)
|
|
template = pymisp.get_object_templates_list()
|
|
if 'response' in template.keys():
|
|
template = template['response']
|
|
try:
|
|
template_ids = [x['ObjectTemplate']['id'] for x in template if x['ObjectTemplate']['name'] == args.type]
|
|
if len(template_ids) > 0:
|
|
template_id = template_ids[0]
|
|
else:
|
|
raise IndexError
|
|
except IndexError:
|
|
valid_types = ", ".join([x['ObjectTemplate']['name'] for x in template])
|
|
print ("Template for type %s not found! Valid types are: %s" % (args.type, valid_types))
|
|
exit()
|
|
|
|
misp_object = GenericObjectGenerator(args.type.replace("|", "-"))
|
|
misp_object.generate_attributes(json.loads(args.attr_list))
|
|
r = pymisp.add_object(args.event, template_id, misp_object)
|