mirror of https://github.com/D4-project/d4-core
715 lines
28 KiB
Python
Executable File
715 lines
28 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*-coding:UTF-8 -*
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
import uuid
|
|
import time
|
|
import json
|
|
import redis
|
|
import flask
|
|
import datetime
|
|
import ipaddress
|
|
|
|
import subprocess
|
|
|
|
from flask import Flask, render_template, jsonify, request, Blueprint, redirect, url_for
|
|
|
|
baseUrl = ''
|
|
if baseUrl != '':
|
|
baseUrl = '/'+baseUrl
|
|
|
|
host_redis_stream = "localhost"
|
|
port_redis_stream = 6379
|
|
|
|
default_max_entries_by_stream = 10000
|
|
analyzer_list_max_default_size = 10000
|
|
|
|
default_analyzer_max_line_len = 3000
|
|
|
|
json_type_description_path = os.path.join(os.environ['D4_HOME'], 'web/static/json/type.json')
|
|
|
|
redis_server_stream = redis.StrictRedis(
|
|
host=host_redis_stream,
|
|
port=port_redis_stream,
|
|
db=0,
|
|
decode_responses=True)
|
|
|
|
host_redis_metadata = "localhost"
|
|
port_redis_metadata= 6380
|
|
|
|
redis_server_metadata = redis.StrictRedis(
|
|
host=host_redis_metadata,
|
|
port=port_redis_metadata,
|
|
db=0,
|
|
decode_responses=True)
|
|
|
|
redis_server_analyzer = redis.StrictRedis(
|
|
host=host_redis_metadata,
|
|
port=port_redis_metadata,
|
|
db=2,
|
|
decode_responses=True)
|
|
|
|
with open(json_type_description_path, 'r') as f:
|
|
json_type = json.loads(f.read())
|
|
json_type_description = {}
|
|
for type_info in json_type:
|
|
json_type_description[type_info['type']] = type_info
|
|
|
|
app = Flask(__name__, static_url_path=baseUrl+'/static/')
|
|
app.config['MAX_CONTENT_LENGTH'] = 900 * 1024 * 1024
|
|
|
|
# ========== FUNCTIONS ============
|
|
def is_valid_uuid_v4(header_uuid):
|
|
try:
|
|
header_uuid=header_uuid.replace('-', '')
|
|
uuid_test = uuid.UUID(hex=header_uuid, version=4)
|
|
return uuid_test.hex == header_uuid
|
|
except:
|
|
return False
|
|
|
|
def is_valid_ip(ip):
|
|
try:
|
|
ipaddress.ip_address(ip)
|
|
return True
|
|
except ValueError:
|
|
return False
|
|
|
|
def is_valid_network(ip_network):
|
|
try:
|
|
ipaddress.ip_network(ip_network)
|
|
return True
|
|
except ValueError:
|
|
return False
|
|
|
|
# server_management input handler
|
|
def get_server_management_input_handler_value(value):
|
|
if value is not None:
|
|
if value !="0":
|
|
try:
|
|
value=int(value)
|
|
except:
|
|
value=0
|
|
else:
|
|
value=0
|
|
return value
|
|
|
|
def get_json_type_description():
|
|
return json_type_description
|
|
|
|
def get_whois_ouput(ip):
|
|
if is_valid_ip(ip):
|
|
process = subprocess.run(["whois", ip], stdout=subprocess.PIPE)
|
|
return re.sub(r"#.*\n?", '', process.stdout.decode()).lstrip('\n').rstrip('\n')
|
|
else:
|
|
return ''
|
|
|
|
# ========== ERRORS ============
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(e):
|
|
return render_template('404.html'), 404
|
|
|
|
# ========== ROUTES ============
|
|
@app.route('/')
|
|
def index():
|
|
date = datetime.datetime.now().strftime("%Y/%m/%d")
|
|
return render_template("index.html", date=date)
|
|
|
|
@app.route('/_json_daily_uuid_stats')
|
|
def _json_daily_uuid_stats():
|
|
date = datetime.datetime.now().strftime("%Y%m%d")
|
|
daily_uuid = redis_server_metadata.zrange('daily_uuid:{}'.format(date), 0, -1, withscores=True)
|
|
|
|
data_daily_uuid = []
|
|
for result in daily_uuid:
|
|
data_daily_uuid.append({"key": result[0], "value": int(result[1])})
|
|
|
|
return jsonify(data_daily_uuid)
|
|
|
|
@app.route('/_json_daily_type_stats')
|
|
def _json_daily_type_stats():
|
|
date = datetime.datetime.now().strftime("%Y%m%d")
|
|
daily_uuid = redis_server_metadata.zrange('daily_type:{}'.format(date), 0, -1, withscores=True)
|
|
json_type_description = get_json_type_description()
|
|
|
|
data_daily_uuid = []
|
|
for result in daily_uuid:
|
|
try:
|
|
type_description = json_type_description[int(result[0])]['description']
|
|
except:
|
|
type_description = 'Please update your web server'
|
|
data_daily_uuid.append({"key": '{}: {}'.format(result[0], type_description), "value": int(result[1])})
|
|
|
|
return jsonify(data_daily_uuid)
|
|
|
|
@app.route('/sensors_status')
|
|
def sensors_status():
|
|
active_connection_filter = request.args.get('active_connection_filter')
|
|
if active_connection_filter is None:
|
|
active_connection_filter = False
|
|
else:
|
|
if active_connection_filter=='True':
|
|
active_connection_filter = True
|
|
else:
|
|
active_connection_filter = False
|
|
|
|
date = datetime.datetime.now().strftime("%Y%m%d")
|
|
|
|
if not active_connection_filter:
|
|
daily_uuid = redis_server_metadata.zrange('daily_uuid:{}'.format(date), 0, -1)
|
|
else:
|
|
daily_uuid = redis_server_stream.smembers('active_connection')
|
|
|
|
status_daily_uuid = []
|
|
for result in daily_uuid:
|
|
first_seen = redis_server_metadata.hget('metadata_uuid:{}'.format(result), 'first_seen')
|
|
first_seen_gmt = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(first_seen)))
|
|
last_seen = redis_server_metadata.hget('metadata_uuid:{}'.format(result), 'last_seen')
|
|
last_seen_gmt = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(last_seen)))
|
|
if redis_server_metadata.sismember('blacklist_ip_by_uuid', result):
|
|
Error = "All IP using this UUID are Blacklisted"
|
|
elif redis_server_metadata.sismember('blacklist_uuid', result):
|
|
Error = "Blacklisted UUID"
|
|
else:
|
|
Error = redis_server_metadata.hget('metadata_uuid:{}'.format(result), 'Error')
|
|
if redis_server_stream.sismember('active_connection', result):
|
|
active_connection = True
|
|
else:
|
|
active_connection = False
|
|
|
|
if first_seen is not None and last_seen is not None:
|
|
status_daily_uuid.append({"uuid": result,"first_seen": first_seen, "last_seen": last_seen,
|
|
"active_connection": active_connection,
|
|
"first_seen_gmt": first_seen_gmt, "last_seen_gmt": last_seen_gmt, "Error": Error})
|
|
|
|
return render_template("sensors_status.html", status_daily_uuid=status_daily_uuid,
|
|
active_connection_filter=active_connection_filter)
|
|
|
|
@app.route('/show_active_uuid')
|
|
def show_active_uuid():
|
|
#swap switch value
|
|
active_connection_filter = request.args.get('show_active_connection')
|
|
if active_connection_filter is None:
|
|
active_connection_filter = True
|
|
else:
|
|
if active_connection_filter=='True':
|
|
active_connection_filter = False
|
|
else:
|
|
active_connection_filter = True
|
|
|
|
return redirect(url_for('sensors_status', active_connection_filter=active_connection_filter))
|
|
|
|
@app.route('/server_management')
|
|
def server_management():
|
|
blacklisted_ip = request.args.get('blacklisted_ip')
|
|
unblacklisted_ip = request.args.get('unblacklisted_ip')
|
|
blacklisted_uuid = request.args.get('blacklisted_uuid')
|
|
unblacklisted_uuid = request.args.get('unblacklisted_uuid')
|
|
|
|
blacklisted_ip = get_server_management_input_handler_value(blacklisted_ip)
|
|
unblacklisted_ip = get_server_management_input_handler_value(unblacklisted_ip)
|
|
blacklisted_uuid = get_server_management_input_handler_value(blacklisted_uuid)
|
|
unblacklisted_uuid = get_server_management_input_handler_value(unblacklisted_uuid)
|
|
|
|
json_type_description = get_json_type_description()
|
|
|
|
list_accepted_types = []
|
|
list_analyzer_types = []
|
|
for type in redis_server_metadata.smembers('server:accepted_type'):
|
|
try:
|
|
description = json_type_description[int(type)]['description']
|
|
except:
|
|
description = 'Please update your web server'
|
|
|
|
list_analyzer_uuid = []
|
|
for analyzer_uuid in redis_server_metadata.smembers('analyzer:{}'.format(type)):
|
|
size_limit = redis_server_metadata.hget('analyzer:{}'.format(analyzer_uuid), 'max_size')
|
|
if size_limit is None:
|
|
size_limit = analyzer_list_max_default_size
|
|
last_updated = redis_server_metadata.hget('analyzer:{}'.format(analyzer_uuid), 'last_updated')
|
|
if last_updated is None:
|
|
last_updated = 'Never'
|
|
else:
|
|
last_updated = datetime.datetime.fromtimestamp(float(last_updated)).strftime('%Y-%m-%d %H:%M:%S')
|
|
description_analyzer = redis_server_metadata.hget('analyzer:{}'.format(analyzer_uuid), 'description')
|
|
if description_analyzer is None:
|
|
description_analyzer = ''
|
|
len_queue = redis_server_analyzer.llen('analyzer:{}:{}'.format(type, analyzer_uuid))
|
|
if len_queue is None:
|
|
len_queue = 0
|
|
list_analyzer_uuid.append({'uuid': analyzer_uuid, 'description': description_analyzer, 'size_limit': size_limit,'last_updated': last_updated, 'length': len_queue})
|
|
|
|
list_accepted_types.append({"id": int(type), "description": description, 'list_analyzer_uuid': list_analyzer_uuid})
|
|
|
|
list_accepted_extended_types = []
|
|
for extended_type in redis_server_metadata.smembers('server:accepted_extended_type'):
|
|
|
|
list_analyzer_uuid = []
|
|
for analyzer_uuid in redis_server_metadata.smembers('analyzer:254:{}'.format(extended_type)):
|
|
size_limit = redis_server_metadata.hget('analyzer:{}'.format(analyzer_uuid), 'max_size')
|
|
if size_limit is None:
|
|
size_limit = analyzer_list_max_default_size
|
|
last_updated = redis_server_metadata.hget('analyzer:{}'.format(analyzer_uuid), 'last_updated')
|
|
if last_updated is None:
|
|
last_updated = 'Never'
|
|
else:
|
|
last_updated = datetime.datetime.fromtimestamp(float(last_updated)).strftime('%Y-%m-%d %H:%M:%S')
|
|
description_analyzer = redis_server_metadata.hget('analyzer:{}'.format(analyzer_uuid), 'description')
|
|
if description_analyzer is None:
|
|
description_analyzer = ''
|
|
len_queue = redis_server_analyzer.llen('analyzer:{}:{}'.format(extended_type, analyzer_uuid))
|
|
if len_queue is None:
|
|
len_queue = 0
|
|
list_analyzer_uuid.append({'uuid': analyzer_uuid, 'description': description_analyzer, 'size_limit': size_limit,'last_updated': last_updated, 'length': len_queue})
|
|
|
|
list_accepted_extended_types.append({"name": extended_type, 'list_analyzer_uuid': list_analyzer_uuid})
|
|
|
|
return render_template("server_management.html", list_accepted_types=list_accepted_types, list_accepted_extended_types=list_accepted_extended_types,
|
|
default_analyzer_max_line_len=default_analyzer_max_line_len,
|
|
blacklisted_ip=blacklisted_ip, unblacklisted_ip=unblacklisted_ip,
|
|
blacklisted_uuid=blacklisted_uuid, unblacklisted_uuid=unblacklisted_uuid)
|
|
|
|
@app.route('/uuid_management')
|
|
def uuid_management():
|
|
uuid_sensor = request.args.get('uuid')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
|
|
first_seen = redis_server_metadata.hget('metadata_uuid:{}'.format(uuid_sensor), 'first_seen')
|
|
first_seen_gmt = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(first_seen)))
|
|
last_seen = redis_server_metadata.hget('metadata_uuid:{}'.format(uuid_sensor), 'last_seen')
|
|
last_seen_gmt = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(last_seen)))
|
|
Error = redis_server_metadata.hget('metadata_uuid:{}'.format(uuid_sensor), 'Error')
|
|
if redis_server_stream.exists('temp_blacklist_uuid:{}'.format(uuid_sensor)):
|
|
temp_blacklist_uuid = True
|
|
else:
|
|
temp_blacklist_uuid = False
|
|
if redis_server_metadata.sismember('blacklist_uuid', uuid_sensor):
|
|
blacklisted_uuid = True
|
|
Error = "Blacklisted UUID"
|
|
else:
|
|
blacklisted_uuid = False
|
|
if redis_server_metadata.sismember('blacklist_ip_by_uuid', uuid_sensor):
|
|
blacklisted_ip_by_uuid = True
|
|
Error = "All IP using this UUID are Blacklisted"
|
|
else:
|
|
blacklisted_ip_by_uuid = False
|
|
data_uuid= {"first_seen": first_seen, "last_seen": last_seen,
|
|
"temp_blacklist_uuid": temp_blacklist_uuid,
|
|
"blacklisted_uuid": blacklisted_uuid, "blacklisted_ip_by_uuid": blacklisted_ip_by_uuid,
|
|
"first_seen_gmt": first_seen_gmt, "last_seen_gmt": last_seen_gmt, "Error": Error}
|
|
|
|
if redis_server_stream.sismember('active_connection', uuid_sensor):
|
|
active_connection = True
|
|
else:
|
|
active_connection = False
|
|
|
|
max_uuid_stream = redis_server_metadata.hget('stream_max_size_by_uuid', uuid_sensor)
|
|
if max_uuid_stream is not None:
|
|
max_uuid_stream = int(max_uuid_stream)
|
|
else:
|
|
max_uuid_stream = default_max_entries_by_stream
|
|
|
|
uuid_key = redis_server_metadata.hget('metadata_uuid:{}'.format(uuid_sensor), 'hmac_key')
|
|
if uuid_key is None:
|
|
uuid_key = redis_server_metadata.get('server:hmac_default_key')
|
|
|
|
list_ip = redis_server_metadata.lrange('list_uuid_ip:{}'.format(uuid_sensor), 0, -1)
|
|
all_ip = []
|
|
for elem in list_ip:
|
|
ip, d_time = elem.split('-')
|
|
all_ip.append({'ip': ip,'datetime': '{}/{}/{} - {}:{}.{}'.format(d_time[0:4], d_time[5:6], d_time[6:8], d_time[8:10], d_time[10:12], d_time[12:14])})
|
|
|
|
return render_template("uuid_management.html", uuid_sensor=uuid_sensor, active_connection=active_connection,
|
|
uuid_key=uuid_key, data_uuid=data_uuid, max_uuid_stream=max_uuid_stream, all_ip=all_ip)
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/blacklisted_ip')
|
|
def blacklisted_ip():
|
|
blacklisted_ip = request.args.get('blacklisted_ip')
|
|
unblacklisted_ip = request.args.get('unblacklisted_ip')
|
|
try:
|
|
page = int(request.args.get('page'))
|
|
except:
|
|
page = 1
|
|
if page <= 0:
|
|
page = 1
|
|
nb_page_max = redis_server_metadata.scard('blacklist_ip')/(1000*2)
|
|
if isinstance(nb_page_max, float):
|
|
nb_page_max = int(nb_page_max)+1
|
|
if page > nb_page_max:
|
|
page = nb_page_max
|
|
start = 1000*(page -1)
|
|
stop = 1000*page
|
|
|
|
list_blacklisted_ip = list(redis_server_metadata.smembers('blacklist_ip'))
|
|
list_blacklisted_ip_1 = list_blacklisted_ip[start:stop]
|
|
list_blacklisted_ip_2 = list_blacklisted_ip[stop:stop+1000]
|
|
return render_template("blacklisted_ip.html", list_blacklisted_ip_1=list_blacklisted_ip_1, list_blacklisted_ip_2=list_blacklisted_ip_2,
|
|
page=page, nb_page_max=nb_page_max,
|
|
unblacklisted_ip=unblacklisted_ip, blacklisted_ip=blacklisted_ip)
|
|
|
|
@app.route('/blacklisted_uuid')
|
|
def blacklisted_uuid():
|
|
blacklisted_uuid = request.args.get('blacklisted_uuid')
|
|
unblacklisted_uuid = request.args.get('unblacklisted_uuid')
|
|
try:
|
|
page = int(request.args.get('page'))
|
|
except:
|
|
page = 1
|
|
if page <= 0:
|
|
page = 1
|
|
nb_page_max = redis_server_metadata.scard('blacklist_uuid')/(1000*2)
|
|
if isinstance(nb_page_max, float):
|
|
nb_page_max = int(nb_page_max)+1
|
|
if page > nb_page_max:
|
|
page = nb_page_max
|
|
start = 1000*(page -1)
|
|
stop = 1000*page
|
|
|
|
list_blacklisted_uuid = list(redis_server_metadata.smembers('blacklist_uuid'))
|
|
list_blacklisted_uuid_1 = list_blacklisted_uuid[start:stop]
|
|
list_blacklisted_uuid_2 = list_blacklisted_uuid[stop:stop+1000]
|
|
return render_template("blacklisted_uuid.html", list_blacklisted_uuid_1=list_blacklisted_uuid_1, list_blacklisted_uuid_2=list_blacklisted_uuid_2,
|
|
page=page, nb_page_max=nb_page_max,
|
|
unblacklisted_uuid=unblacklisted_uuid, blacklisted_uuid=blacklisted_uuid)
|
|
|
|
|
|
@app.route('/uuid_change_stream_max_size')
|
|
def uuid_change_stream_max_size():
|
|
uuid_sensor = request.args.get('uuid')
|
|
user = request.args.get('redirect')
|
|
max_uuid_stream = request.args.get('max_uuid_stream')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
try:
|
|
max_uuid_stream = int(max_uuid_stream)
|
|
if max_uuid_stream < 0:
|
|
return 'stream max size, Invalid Integer'
|
|
except:
|
|
return 'stream max size, Invalid Integer'
|
|
redis_server_metadata.hset('stream_max_size_by_uuid', uuid_sensor, max_uuid_stream)
|
|
if user:
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
# # TODO: check analyser uuid dont exist
|
|
@app.route('/add_new_analyzer')
|
|
def add_new_analyzer():
|
|
type = request.args.get('type')
|
|
user = request.args.get('redirect')
|
|
metatype_name = request.args.get('metatype_name')
|
|
analyzer_description = request.args.get('analyzer_description')
|
|
analyzer_uuid = request.args.get('analyzer_uuid')
|
|
if is_valid_uuid_v4(analyzer_uuid):
|
|
try:
|
|
type = int(type)
|
|
if type < 0:
|
|
return 'type, Invalid Integer'
|
|
except:
|
|
return 'type, Invalid Integer'
|
|
if type == 254:
|
|
# # TODO: check metatype_name
|
|
redis_server_metadata.sadd('analyzer:{}:{}'.format(type, metatype_name), analyzer_uuid)
|
|
else:
|
|
redis_server_metadata.sadd('analyzer:{}'.format(type), analyzer_uuid)
|
|
if redis_server_metadata.exists('analyzer:{}:{}'.format(type, metatype_name)) or redis_server_metadata.exists('analyzer:{}'.format(type)):
|
|
redis_server_metadata.hset('analyzer:{}'.format(analyzer_uuid), 'description', analyzer_description)
|
|
if user:
|
|
return redirect(url_for('server_management'))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/empty_analyzer_queue')
|
|
def empty_analyzer_queue():
|
|
analyzer_uuid = request.args.get('analyzer_uuid')
|
|
type = request.args.get('type')
|
|
metatype_name = request.args.get('metatype_name')
|
|
user = request.args.get('redirect')
|
|
if is_valid_uuid_v4(analyzer_uuid):
|
|
try:
|
|
type = int(type)
|
|
if type < 0:
|
|
return 'type, Invalid Integer'
|
|
except:
|
|
return 'type, Invalid Integer'
|
|
if type == 254:
|
|
redis_server_analyzer.delete('analyzer:{}:{}'.format(metatype_name, analyzer_uuid))
|
|
else:
|
|
redis_server_analyzer.delete('analyzer:{}:{}'.format(type, analyzer_uuid))
|
|
if user:
|
|
return redirect(url_for('server_management'))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/remove_analyzer')
|
|
def remove_analyzer():
|
|
analyzer_uuid = request.args.get('analyzer_uuid')
|
|
type = request.args.get('type')
|
|
metatype_name = request.args.get('metatype_name')
|
|
user = request.args.get('redirect')
|
|
if is_valid_uuid_v4(analyzer_uuid):
|
|
try:
|
|
type = int(type)
|
|
if type < 0:
|
|
return 'type, Invalid Integer'
|
|
except:
|
|
return 'type, Invalid Integer'
|
|
if type == 254:
|
|
redis_server_metadata.srem('analyzer:{}:{}'.format(type, metatype_name), analyzer_uuid)
|
|
redis_server_analyzer.delete('analyzer:{}:{}'.format(metatype_name, analyzer_uuid))
|
|
else:
|
|
redis_server_metadata.srem('analyzer:{}'.format(type), analyzer_uuid)
|
|
redis_server_analyzer.delete('analyzer:{}:{}'.format(type, analyzer_uuid))
|
|
redis_server_metadata.delete('analyzer:{}'.format(analyzer_uuid))
|
|
if user:
|
|
return redirect(url_for('server_management'))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/analyzer_change_max_size')
|
|
def analyzer_change_max_size():
|
|
analyzer_uuid = request.args.get('analyzer_uuid')
|
|
user = request.args.get('redirect')
|
|
max_size_analyzer = request.args.get('max_size_analyzer')
|
|
if is_valid_uuid_v4(analyzer_uuid):
|
|
try:
|
|
max_size_analyzer = int(max_size_analyzer)
|
|
if max_size_analyzer < 0:
|
|
return 'analyzer max size, Invalid Integer'
|
|
except:
|
|
return 'analyzer max size, Invalid Integer'
|
|
redis_server_metadata.hset('analyzer:{}'.format(analyzer_uuid), 'max_size', max_size_analyzer)
|
|
if user:
|
|
return redirect(url_for('server_management'))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/kick_uuid')
|
|
def kick_uuid():
|
|
uuid_sensor = request.args.get('uuid')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
redis_server_stream.sadd('server:sensor_to_kick', uuid_sensor)
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/blacklist_uuid')
|
|
def blacklist_uuid():
|
|
uuid_sensor = request.args.get('uuid')
|
|
user = request.args.get('redirect')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
res = redis_server_metadata.sadd('blacklist_uuid', uuid_sensor)
|
|
if user=="0":
|
|
if res==0:
|
|
return redirect(url_for('server_management', blacklisted_uuid=2))
|
|
else:
|
|
return redirect(url_for('server_management', blacklisted_uuid=1))
|
|
elif user=="1":
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
else:
|
|
return "404"
|
|
else:
|
|
if user=="0":
|
|
return redirect(url_for('server_management', blacklisted_uuid=0))
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/unblacklist_uuid')
|
|
def unblacklist_uuid():
|
|
uuid_sensor = request.args.get('uuid')
|
|
user = request.args.get('redirect')
|
|
page = request.args.get('page')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
res = redis_server_metadata.srem('blacklist_uuid', uuid_sensor)
|
|
if page:
|
|
return redirect(url_for('blacklisted_uuid', page=page))
|
|
if user=="0":
|
|
if res==0:
|
|
return redirect(url_for('server_management', unblacklisted_uuid=2))
|
|
else:
|
|
return redirect(url_for('server_management', unblacklisted_uuid=1))
|
|
elif user=="1":
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
else:
|
|
return "404"
|
|
else:
|
|
if user=="0":
|
|
return redirect(url_for('server_management', unblacklisted_uuid=0))
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/blacklist_ip')
|
|
def blacklist_ip():
|
|
ip = request.args.get('ip')
|
|
user = request.args.get('redirect')
|
|
|
|
if is_valid_ip(ip):
|
|
res = redis_server_metadata.sadd('blacklist_ip', ip)
|
|
if user:
|
|
if res==0:
|
|
return redirect(url_for('server_management', blacklisted_ip=2))
|
|
else:
|
|
return redirect(url_for('server_management', blacklisted_ip=1))
|
|
elif is_valid_network(ip):
|
|
for addr in ipaddress.ip_network(ip):
|
|
res = redis_server_metadata.sadd('blacklist_ip', str(addr))
|
|
if user:
|
|
if res==0:
|
|
return redirect(url_for('server_management', blacklisted_ip=2))
|
|
else:
|
|
return redirect(url_for('server_management', blacklisted_ip=1))
|
|
else:
|
|
if user:
|
|
return redirect(url_for('server_management', blacklisted_ip=0))
|
|
return 'Invalid ip'
|
|
|
|
@app.route('/unblacklist_ip')
|
|
def unblacklist_ip():
|
|
ip = request.args.get('ip')
|
|
user = request.args.get('redirect')
|
|
page = request.args.get('page')
|
|
if is_valid_ip(ip):
|
|
res = redis_server_metadata.srem('blacklist_ip', ip)
|
|
if page:
|
|
return redirect(url_for('blacklisted_ip', page=page))
|
|
if user:
|
|
if res==0:
|
|
return redirect(url_for('server_management', unblacklisted_ip=2))
|
|
else:
|
|
return redirect(url_for('server_management', unblacklisted_ip=1))
|
|
elif is_valid_network(ip):
|
|
for addr in ipaddress.ip_network(ip):
|
|
res = redis_server_metadata.srem('blacklist_ip', str(addr))
|
|
if user:
|
|
if res==0:
|
|
return redirect(url_for('server_management', unblacklisted_ip=2))
|
|
else:
|
|
return redirect(url_for('server_management', unblacklisted_ip=1))
|
|
else:
|
|
if user:
|
|
return redirect(url_for('server_management', unblacklisted_ip=0))
|
|
return 'Invalid ip'
|
|
|
|
@app.route('/blacklist_ip_by_uuid')
|
|
def blacklist_ip_by_uuid():
|
|
uuid_sensor = request.args.get('uuid')
|
|
user = request.args.get('redirect')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
redis_server_metadata.sadd('blacklist_ip_by_uuid', uuid_sensor)
|
|
if user:
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/unblacklist_ip_by_uuid')
|
|
def unblacklist_ip_by_uuid():
|
|
uuid_sensor = request.args.get('uuid')
|
|
user = request.args.get('redirect')
|
|
if is_valid_uuid_v4(uuid_sensor):
|
|
redis_server_metadata.srem('blacklist_ip_by_uuid', uuid_sensor)
|
|
if user:
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
else:
|
|
return 'Invalid uuid'
|
|
|
|
@app.route('/add_accepted_type')
|
|
def add_accepted_type():
|
|
type = request.args.get('type')
|
|
extended_type_name = request.args.get('extended_type_name')
|
|
user = request.args.get('redirect')
|
|
json_type_description = get_json_type_description()
|
|
try:
|
|
type = int(type)
|
|
except:
|
|
return 'Invalid type'
|
|
if json_type_description[int(type)]:
|
|
redis_server_metadata.sadd('server:accepted_type', type)
|
|
if type == 254:
|
|
redis_server_metadata.sadd('server:accepted_extended_type', extended_type_name)
|
|
if user:
|
|
return redirect(url_for('server_management'))
|
|
else:
|
|
return 'Invalid type'
|
|
|
|
@app.route('/remove_accepted_type')
|
|
def remove_accepted_type():
|
|
type = request.args.get('type')
|
|
user = request.args.get('redirect')
|
|
json_type_description = get_json_type_description()
|
|
if json_type_description[int(type)]:
|
|
redis_server_metadata.srem('server:accepted_type', type)
|
|
if user:
|
|
return redirect(url_for('server_management'))
|
|
else:
|
|
return 'Invalid type'
|
|
|
|
@app.route('/remove_accepted_extended_type')
|
|
def remove_accepted_extended_type():
|
|
type_name = request.args.get('type_name')
|
|
redis_server_metadata.srem('server:accepted_extended_type', type_name)
|
|
return redirect(url_for('server_management'))
|
|
|
|
# demo function
|
|
@app.route('/delete_data')
|
|
def delete_data():
|
|
date = datetime.datetime.now().strftime("%Y%m%d")
|
|
redis_server_metadata.delete('daily_type:{}'.format(date))
|
|
redis_server_metadata.delete('daily_uuid:{}'.format(date))
|
|
return render_template("index.html")
|
|
|
|
# demo function
|
|
@app.route('/set_uuid_hmac_key')
|
|
def set_uuid_hmac_key():
|
|
uuid_sensor = request.args.get('uuid')
|
|
user = request.args.get('redirect')
|
|
key = request.args.get('key')
|
|
redis_server_metadata.hset('metadata_uuid:{}'.format(uuid_sensor), 'hmac_key', key)
|
|
if user:
|
|
return redirect(url_for('uuid_management', uuid=uuid_sensor))
|
|
|
|
|
|
# demo function
|
|
@app.route('/whois_data')
|
|
def whois_data():
|
|
ip = request.args.get('ip')
|
|
if is_valid_ip:
|
|
return jsonify(get_whois_ouput(ip))
|
|
else:
|
|
return 'Invalid IP'
|
|
|
|
@app.route('/generate_uuid')
|
|
def generate_uuid():
|
|
new_uuid = uuid.uuid4()
|
|
return jsonify({'uuid': new_uuid})
|
|
|
|
@app.route('/get_analyser_sample')
|
|
def get_analyser_sample():
|
|
type = request.args.get('type')
|
|
analyzer_uuid = request.args.get('analyzer_uuid')
|
|
max_line_len = request.args.get('max_line_len')
|
|
# get max_line_len
|
|
if max_line_len is not None and max_line_len!= 'undefined':
|
|
try:
|
|
max_line_len = int(max_line_len)
|
|
except:
|
|
max_line_len = default_analyzer_max_line_len
|
|
if max_line_len < 1:
|
|
max_line_len = default_analyzer_max_line_len
|
|
else:
|
|
max_line_len = default_analyzer_max_line_len
|
|
if is_valid_uuid_v4(analyzer_uuid):
|
|
list_queue = redis_server_analyzer.lrange('analyzer:{}:{}'.format(type, analyzer_uuid), 0 ,10)
|
|
list_queue_res = []
|
|
for res in list_queue:
|
|
#limit line len
|
|
if len(res) > max_line_len:
|
|
res = '{} [...]'.format(res[:max_line_len])
|
|
list_queue_res.append('{}\n'.format(res))
|
|
return jsonify(''.join(list_queue_res))
|
|
else:
|
|
return jsonify('Incorrect UUID')
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host='0.0.0.0', port=7000, threaded=True)
|