mirror of https://github.com/MISP/misp-modules
77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
import os
|
|
import random
|
|
import uuid
|
|
import json
|
|
import requests
|
|
# import jsonschema
|
|
from conf.config import Config
|
|
from pathlib import Path
|
|
import configparser
|
|
config = configparser.ConfigParser()
|
|
CONF_PATH = os.path.join(os.getcwd(), "conf", "config.cfg")
|
|
config.read(CONF_PATH)
|
|
|
|
MODULES = []
|
|
|
|
def query_get_module(headers={'Content-type': 'application/json'}):
|
|
global MODULES
|
|
if not MODULES:
|
|
try:
|
|
r = requests.get(f"http://{Config.MISP_MODULE}/modules", headers=headers)
|
|
except ConnectionError:
|
|
return {"message": "Instance of misp-modules is unreachable"}
|
|
except Exception as e:
|
|
return {"message": e}
|
|
MODULES = r.json()
|
|
return r.json()
|
|
else:
|
|
return MODULES
|
|
|
|
def query_post_query(data, headers={'Content-type': 'application/json'}):
|
|
try:
|
|
r = requests.post(f"http://{Config.MISP_MODULE}/query", data=json.dumps(data), headers=headers)
|
|
except ConnectionError:
|
|
return {"message": "Instance of misp-modules is unreachable"}
|
|
except Exception as e:
|
|
return {"message": e}
|
|
return r.json()
|
|
|
|
|
|
def isUUID(uid):
|
|
try:
|
|
uuid.UUID(str(uid))
|
|
return True
|
|
except ValueError:
|
|
return False
|
|
|
|
def get_object(obj_name):
|
|
path = Path(os.getcwd())
|
|
parent_path = path.parent.absolute()
|
|
loc_path = os.path.join(parent_path, "misp_modules", "lib", "misp-objects", "objects")
|
|
if os.path.isdir(loc_path):
|
|
with open(os.path.join(loc_path, obj_name, "definition.json"), "r") as read_json:
|
|
loc_json = json.load(read_json)
|
|
return loc_json
|
|
return False
|
|
|
|
|
|
def admin_user_active():
|
|
config.read(CONF_PATH)
|
|
return config.getboolean("ADMIN", "ADMIN_USER")
|
|
|
|
def admin_password():
|
|
return config["ADMIN"]["ADMIN_PASSWORD"]
|
|
|
|
def gen_admin_password():
|
|
if not admin_password():
|
|
chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!@$%#[]+-:;_&*().,?0123456789'
|
|
password = ''
|
|
for _ in range(20):
|
|
password += random.choice(chars)
|
|
print(f"##########################\n## Admin password ##\n## {password} ##\n##########################")
|
|
config["ADMIN"]["ADMIN_PASSWORD"] = password
|
|
with open(CONF_PATH, "w") as conffile:
|
|
config.write(conffile)
|
|
|
|
def get_limit_queries():
|
|
return Config.QUERIES_LIMIT |