mirror of https://github.com/MISP/mail_to_misp
71 lines
2.3 KiB
Python
Executable File
71 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import sys
|
|
import argparse
|
|
import syslog
|
|
from pathlib import Path
|
|
from io import BytesIO
|
|
import importlib
|
|
|
|
from mail2misp import Mail2MISP
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description='Push a Mail into a MISP instance')
|
|
parser.add_argument("-r", "--read", help="Read from tempfile.")
|
|
parser.add_argument("-t", "--trap", action='store_true', default=False, help="Import the Email as-is.")
|
|
parser.add_argument('infile', nargs='?', type=argparse.FileType('rb'))
|
|
args = parser.parse_args()
|
|
|
|
syslog.openlog(logoption=syslog.LOG_PID, facility=syslog.LOG_USER)
|
|
syslog.syslog("Job started.")
|
|
|
|
os.chdir(Path(__file__).parent)
|
|
|
|
configmodule = Path(__file__).name.replace('.py', '_config')
|
|
if Path(f'{configmodule}.py').exists():
|
|
config = importlib.import_module(configmodule)
|
|
try:
|
|
misp_url = config.misp_url
|
|
misp_key = config.misp_key
|
|
misp_verifycert = config.misp_verifycert
|
|
debug = config.debug
|
|
except Exception as e:
|
|
syslog.syslog(str(e))
|
|
print("There is a problem with the configuration. A mandatory configuration variable is not set.")
|
|
print("Did you just update? mail_to_misp might have new configuration variables.")
|
|
print("Please compare with the configuration example.")
|
|
print("\nTrace:")
|
|
print(e)
|
|
sys.exit(-2)
|
|
else:
|
|
print("Couldn't locate config file {0}".format(f'{configmodule}.py'))
|
|
sys.exit(-1)
|
|
|
|
if args.infile:
|
|
pseudofile = BytesIO(args.infile.read().encode('utf8', 'surrogateescape'))
|
|
elif args.read:
|
|
# read from tempfile
|
|
with open(args.read, 'rb') as f:
|
|
pseudofile = BytesIO(f.read())
|
|
else:
|
|
# receive data and subject through arguments
|
|
raise Exception('This is not implemented anymore.')
|
|
|
|
mail2misp = Mail2MISP(misp_url, misp_key, misp_verifycert, config=config)
|
|
mail2misp.load_email(pseudofile)
|
|
|
|
if debug:
|
|
syslog.syslog(f'Working on {mail2misp.subject}')
|
|
|
|
if args.trap or config.spamtrap:
|
|
mail2misp.email_from_spamtrap()
|
|
else:
|
|
mail2misp.process_email_body()
|
|
|
|
mail2misp.process_body_iocs()
|
|
|
|
mail2misp.add_event()
|
|
syslog.syslog("Job finished.")
|