mirror of https://github.com/CIRCL/AIL-framework
255 lines
9.4 KiB
Python
255 lines
9.4 KiB
Python
#!/usr/bin/env python3
|
|
# -*-coding:UTF-8 -*
|
|
|
|
'''
|
|
Blueprint Flask: crawler splash endpoints: dashboard, onion crawler ...
|
|
'''
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
|
|
from flask import Flask, render_template, jsonify, request, Blueprint, redirect, url_for, Response, abort
|
|
from flask_login import login_required, current_user, login_user, logout_user
|
|
|
|
sys.path.append('modules')
|
|
import Flask_config
|
|
|
|
# Import Role_Manager
|
|
from Role_Manager import login_admin, login_analyst, login_read_only
|
|
|
|
|
|
sys.path.append(os.environ['AIL_BIN'])
|
|
##################################
|
|
# Import Project packages
|
|
##################################
|
|
from lib.objects import ail_objects
|
|
from lib import Tag
|
|
|
|
bootstrap_label = Flask_config.bootstrap_label
|
|
vt_enabled = Flask_config.vt_enabled
|
|
|
|
# ============ BLUEPRINT ============
|
|
correlation = Blueprint('correlation', __name__,
|
|
template_folder=os.path.join(os.environ['AIL_FLASK'], 'templates/correlation'))
|
|
|
|
# ============ VARIABLES ============
|
|
|
|
# ============ FUNCTIONS ============
|
|
|
|
def sanitise_graph_mode(graph_mode):
|
|
if graph_mode not in ('inter', 'union'):
|
|
return 'union'
|
|
else:
|
|
return graph_mode
|
|
|
|
def sanitise_nb_max_nodes(nb_max_nodes):
|
|
try:
|
|
nb_max_nodes = int(nb_max_nodes)
|
|
if nb_max_nodes < 2:
|
|
nb_max_nodes = 300
|
|
except (TypeError, ValueError):
|
|
nb_max_nodes = 300
|
|
return nb_max_nodes
|
|
|
|
def sanitise_level(level):
|
|
try:
|
|
level = int(level)
|
|
if level < 0:
|
|
level = 2
|
|
except (TypeError, ValueError):
|
|
level = 2
|
|
return level
|
|
|
|
# ============= ROUTES ==============
|
|
@correlation.route('/correlation/show', methods=['GET', 'POST'])
|
|
@login_required
|
|
@login_read_only
|
|
def show_correlation():
|
|
if request.method == 'POST':
|
|
object_type = request.form.get('obj_type')
|
|
subtype = request.form.get('subtype')
|
|
obj_id = request.form.get('obj_id')
|
|
max_nodes = request.form.get('max_nb_nodes_in')
|
|
mode = request.form.get('mode')
|
|
if mode:
|
|
mode = 'inter'
|
|
else:
|
|
mode = 'union'
|
|
level = sanitise_level(request.form.get('level'))
|
|
|
|
## get all selected correlations
|
|
filter_types = []
|
|
correl_option = request.form.get('CveCheck')
|
|
if correl_option:
|
|
filter_types.append('cve')
|
|
correl_option = request.form.get('CryptocurrencyCheck')
|
|
if correl_option:
|
|
filter_types.append('cryptocurrency')
|
|
correl_option = request.form.get('PgpCheck')
|
|
if correl_option:
|
|
filter_types.append('pgp')
|
|
correl_option = request.form.get('UsernameCheck')
|
|
if correl_option:
|
|
filter_types.append('username')
|
|
correl_option = request.form.get('DecodedCheck')
|
|
if correl_option:
|
|
filter_types.append('decoded')
|
|
correl_option = request.form.get('ScreenshotCheck')
|
|
if correl_option:
|
|
filter_types.append('screenshot')
|
|
# correlation_objects
|
|
correl_option = request.form.get('DomainCheck')
|
|
if correl_option:
|
|
filter_types.append('domain')
|
|
correl_option = request.form.get('ItemCheck')
|
|
if correl_option:
|
|
filter_types.append('item')
|
|
correl_option = request.form.get('TitleCheck')
|
|
if correl_option:
|
|
filter_types.append('title')
|
|
|
|
# list as params
|
|
filter_types = ",".join(filter_types)
|
|
|
|
# redirect to keep history and bookmark
|
|
return redirect(url_for('correlation.show_correlation', type=object_type, subtype=subtype, id=obj_id, mode=mode,
|
|
max_nodes=max_nodes, level=level, filter=filter_types))
|
|
|
|
# request.method == 'GET'
|
|
else:
|
|
obj_type = request.args.get('type')
|
|
subtype = request.args.get('subtype', '')
|
|
obj_id = request.args.get('id')
|
|
max_nodes = sanitise_nb_max_nodes(request.args.get('max_nodes'))
|
|
mode = sanitise_graph_mode(request.args.get('mode'))
|
|
level = sanitise_level(request.args.get('level'))
|
|
|
|
related_btc = bool(request.args.get('related_btc', False))
|
|
|
|
filter_types = ail_objects.sanitize_objs_types(request.args.get('filter', '').split(','))
|
|
|
|
# check if obj_id exist
|
|
if not ail_objects.exists_obj(obj_type, subtype, obj_id):
|
|
return abort(404)
|
|
# object exist
|
|
else:
|
|
dict_object = {"object_type": obj_type,
|
|
"correlation_id": obj_id,
|
|
"max_nodes": max_nodes, "mode": mode, "level": level,
|
|
"filter": filter_types, "filter_str": ",".join(filter_types),
|
|
"metadata": ail_objects.get_object_meta(obj_type, subtype, obj_id,
|
|
options={'tags'}, flask_context=True),
|
|
"nb_correl": ail_objects.get_obj_nb_correlations(obj_type, subtype, obj_id)
|
|
}
|
|
if subtype:
|
|
dict_object["metadata"]['type_id'] = subtype
|
|
dict_object["metadata_card"] = ail_objects.get_object_card_meta(obj_type, subtype, obj_id, related_btc=related_btc)
|
|
return render_template("show_correlation.html", dict_object=dict_object, bootstrap_label=bootstrap_label,
|
|
tags_selector_data=Tag.get_tags_selector_data())
|
|
|
|
@correlation.route('/correlation/get/description')
|
|
@login_required
|
|
@login_read_only
|
|
def get_description():
|
|
object_id = request.args.get('object_id')
|
|
object_id = object_id.split(';')
|
|
# unpack object_id # # TODO: put me in lib
|
|
if len(object_id) == 3:
|
|
object_type = object_id[0]
|
|
type_id = object_id[1]
|
|
correlation_id = object_id[2]
|
|
elif len(object_id) == 2:
|
|
object_type = object_id[0]
|
|
type_id = None
|
|
correlation_id = object_id[1]
|
|
else:
|
|
return jsonify({})
|
|
|
|
# check if correlation_id exist
|
|
# # TODO: return error json
|
|
if not ail_objects.exists_obj(object_type, type_id, correlation_id):
|
|
return Response(json.dumps({"status": "error", "reason": "404 Not Found"}, indent=2, sort_keys=True), mimetype='application/json'), 404
|
|
# object exist
|
|
else:
|
|
res = ail_objects.get_object_meta(object_type, type_id, correlation_id, options={'tags', 'tags_safe'},
|
|
flask_context=True)
|
|
if 'tags' in res:
|
|
res['tags'] = list(res['tags'])
|
|
return jsonify(res)
|
|
|
|
@correlation.route('/correlation/graph_node_json')
|
|
@login_required
|
|
@login_read_only
|
|
def graph_node_json():
|
|
obj_id = request.args.get('id')
|
|
subtype = request.args.get('subtype')
|
|
obj_type = request.args.get('type')
|
|
max_nodes = sanitise_nb_max_nodes(request.args.get('max_nodes'))
|
|
level = sanitise_level(request.args.get('level'))
|
|
|
|
filter_types = ail_objects.sanitize_objs_types(request.args.get('filter', '').split(','))
|
|
|
|
json_graph = ail_objects.get_correlations_graph_node(obj_type, subtype, obj_id, filter_types=filter_types, max_nodes=max_nodes, level=level, flask_context=True)
|
|
#json_graph = Correlate_object.get_graph_node_object_correlation(obj_type, obj_id, 'union', correlation_names, correlation_objects, requested_correl_type=subtype, max_nodes=max_nodes)
|
|
return jsonify(json_graph)
|
|
|
|
@correlation.route('/correlation/delete', methods=['GET'])
|
|
@login_required
|
|
@login_admin
|
|
def correlation_delete():
|
|
obj_type = request.args.get('type')
|
|
subtype = request.args.get('subtype', '')
|
|
obj_id = request.args.get('id')
|
|
|
|
if not ail_objects.exists_obj(obj_type, subtype, obj_id):
|
|
return abort(404)
|
|
|
|
ail_objects.delete_obj_correlations(obj_type, subtype, obj_id)
|
|
return redirect(url_for('correlation.show_correlation', type=obj_type, subtype=subtype, id=obj_id))
|
|
|
|
@correlation.route('/correlation/tags/add', methods=['POST'])
|
|
@login_required
|
|
@login_analyst
|
|
def correlation_tags_add():
|
|
obj_id = request.form.get('tag_obj_id')
|
|
subtype = request.form.get('tag_subtype', '')
|
|
obj_type = request.form.get('tag_obj_type')
|
|
nb_max = sanitise_nb_max_nodes(request.form.get('tag_nb_max'))
|
|
level = sanitise_level(request.form.get('tag_level'))
|
|
filter_types = ail_objects.sanitize_objs_types(request.form.get('tag_filter', '').split(','))
|
|
|
|
if not ail_objects.exists_obj(obj_type, subtype, obj_id):
|
|
return abort(404)
|
|
|
|
# tags
|
|
taxonomies_tags = request.form.get('taxonomies_tags')
|
|
if taxonomies_tags:
|
|
try:
|
|
taxonomies_tags = json.loads(taxonomies_tags)
|
|
except Exception:
|
|
taxonomies_tags = []
|
|
else:
|
|
taxonomies_tags = []
|
|
galaxies_tags = request.form.get('galaxies_tags')
|
|
if galaxies_tags:
|
|
try:
|
|
galaxies_tags = json.loads(galaxies_tags)
|
|
except Exception:
|
|
galaxies_tags = []
|
|
if taxonomies_tags or galaxies_tags:
|
|
if not Tag.is_valid_tags_taxonomies_galaxy(taxonomies_tags, galaxies_tags):
|
|
return {'error': 'Invalid tag(s)'}, 400
|
|
tags = taxonomies_tags + galaxies_tags
|
|
else:
|
|
tags = []
|
|
|
|
if tags:
|
|
ail_objects.obj_correlations_objs_add_tags(obj_type, subtype, obj_id, tags, filter_types=filter_types,
|
|
lvl=level + 1, nb_max=nb_max)
|
|
return redirect(url_for('correlation.show_correlation',
|
|
type=obj_type, subtype=subtype, id=obj_id,
|
|
level=level,
|
|
max_nodes=nb_max,
|
|
filter=",".join(filter_types)))
|