mirror of https://github.com/D4-project/d4-core
chg: [UI uuid_management] show stats: files save on disk by types
TODO: split 254 by extended typepull/23/head
Terrtia
parent
ae2adfe4d6
commit
3a22c250ee
|
|
@ -11,6 +11,7 @@ import redis
|
|||
import flask
|
||||
import datetime
|
||||
import ipaddress
|
||||
import configparser
|
||||
|
||||
import subprocess
|
||||
|
||||
|
|
@ -30,6 +31,19 @@ default_analyzer_max_line_len = 3000
|
|||
|
||||
json_type_description_path = os.path.join(os.environ['D4_HOME'], 'web/static/json/type.json')
|
||||
|
||||
# get file config
|
||||
config_file_server = os.path.join(os.environ['D4_HOME'], 'configs/server.conf')
|
||||
config_server = configparser.ConfigParser()
|
||||
config_server.read(config_file_server)
|
||||
|
||||
# get data directory
|
||||
use_default_save_directory = config_server['Save_Directories'].getboolean('use_default_save_directory')
|
||||
# check if field is None
|
||||
if use_default_save_directory:
|
||||
data_directory = os.path.join(os.environ['D4_HOME'], 'data')
|
||||
else:
|
||||
data_directory = config_server['Save_Directories'].get('save_directory')
|
||||
|
||||
redis_server_stream = redis.StrictRedis(
|
||||
host=host_redis_stream,
|
||||
port=port_redis_stream,
|
||||
|
|
@ -117,6 +131,66 @@ def get_substract_date_range(num_day, date_from=None):
|
|||
l_date.append( date.strftime('%Y%m%d') )
|
||||
return list(reversed(l_date))
|
||||
|
||||
def get_uuid_all_types_disk(uuid_name):
|
||||
uuid_data_directory = os.path.join(data_directory, uuid_name)
|
||||
all_types_on_disk = []
|
||||
# Get all types save on disk
|
||||
for file in os.listdir(uuid_data_directory):
|
||||
uuid_type_path = os.path.join(uuid_data_directory, file)
|
||||
if os.path.isdir(uuid_type_path):
|
||||
all_types_on_disk.append(file)
|
||||
return all_types_on_disk
|
||||
|
||||
def get_uuid_disk_statistics(uuid_name, date_day='', type='', all_types_on_disk=[], all_stats=True):
|
||||
# # TODO: escape uuid_name
|
||||
|
||||
stat_disk_uuid = {}
|
||||
uuid_data_directory = os.path.join(data_directory, uuid_name)
|
||||
if date_day:
|
||||
directory_date = os.path.join(date_day[0:4], date_day[4:6], date_day[6:8])
|
||||
all_types_on_disk = {}
|
||||
|
||||
if all_types_on_disk:
|
||||
for type in all_types_on_disk:
|
||||
if date_day:
|
||||
uuid_type_path = os.path.join(uuid_data_directory, type, directory_date)
|
||||
else:
|
||||
uuid_type_path = os.path.join(uuid_data_directory, type)
|
||||
all_types_on_disk[type] = uuid_type_path
|
||||
else:
|
||||
# Get all types save on disk
|
||||
for file in os.listdir(uuid_data_directory):
|
||||
if date_day:
|
||||
uuid_type_path = os.path.join(uuid_data_directory, file, directory_date)
|
||||
else:
|
||||
uuid_type_path = os.path.join(uuid_data_directory, file)
|
||||
if os.path.isdir(uuid_type_path):
|
||||
all_types_on_disk[file] = uuid_type_path
|
||||
|
||||
nb_file = 0
|
||||
total_size = 0
|
||||
|
||||
for uuid_type in all_types_on_disk:
|
||||
nb_file_type = 0
|
||||
total_size_type = 0
|
||||
for dirpath, dirnames, filenames in os.walk(all_types_on_disk[uuid_type]):
|
||||
stat_disk_uuid[uuid_type] = {}
|
||||
for f in filenames:
|
||||
fp = os.path.join(dirpath, f)
|
||||
file_size = os.path.getsize(fp)
|
||||
total_size_type += file_size
|
||||
total_size += file_size
|
||||
nb_file_type += 1
|
||||
nb_file += 1
|
||||
stat_disk_uuid[uuid_type]['nb_files'] = nb_file_type
|
||||
stat_disk_uuid[uuid_type]['total_size'] = total_size_type
|
||||
if all_stats:
|
||||
stat_all = {}
|
||||
stat_all['nb_files'] = nb_file
|
||||
stat_all['total_size'] = total_size
|
||||
stat_disk_uuid['All'] = stat_all
|
||||
return stat_disk_uuid
|
||||
|
||||
# ========== ERRORS ============
|
||||
|
||||
@app.errorhandler(404)
|
||||
|
|
@ -124,6 +198,11 @@ def page_not_found(e):
|
|||
return render_template('404.html'), 404
|
||||
|
||||
# ========== ROUTES ============
|
||||
@app.route('/test')
|
||||
def test():
|
||||
print(get_uuid_disk_statistics('fae58cdc30024239874f4c7ce53fbf4d', date_day='20190527'))
|
||||
return 'test'
|
||||
|
||||
@app.route('/')
|
||||
def index():
|
||||
date = datetime.datetime.now().strftime("%Y/%m/%d")
|
||||
|
|
@ -288,10 +367,13 @@ def uuid_management():
|
|||
uuid_sensor = request.args.get('uuid')
|
||||
if is_valid_uuid_v4(uuid_sensor):
|
||||
|
||||
disk_stats = get_uuid_disk_statistics(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)))
|
||||
description = redis_server_metadata.hget('metadata_uuid:{}'.format(uuid_sensor), 'description')
|
||||
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
|
||||
|
|
@ -308,6 +390,7 @@ def uuid_management():
|
|||
else:
|
||||
blacklisted_ip_by_uuid = False
|
||||
data_uuid= {"first_seen": first_seen, "last_seen": last_seen,
|
||||
"description": description,
|
||||
"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}
|
||||
|
|
@ -346,6 +429,7 @@ def uuid_management():
|
|||
|
||||
return render_template("uuid_management.html", uuid_sensor=uuid_sensor, active_connection=active_connection,
|
||||
uuid_key=uuid_key, data_uuid=data_uuid, uuid_all_type=uuid_all_type_list,
|
||||
disk_stats=disk_stats,
|
||||
max_uuid_stream=max_uuid_stream, all_ip=all_ip)
|
||||
else:
|
||||
return 'Invalid uuid'
|
||||
|
|
@ -764,7 +848,35 @@ def get_uuid_type_history_json():
|
|||
else:
|
||||
return jsonify('Incorrect UUID')
|
||||
|
||||
@app.route('/get_uuid_stats_history_json')
|
||||
def get_uuid_stats_history_json():
|
||||
uuid_sensor = request.args.get('uuid_sensor')
|
||||
stats = request.args.get('stats')
|
||||
if is_valid_uuid_v4(uuid_sensor):
|
||||
if stats not in ['nb_files', 'total_size']:
|
||||
stats = 'nb_files'
|
||||
|
||||
num_day_type = 7
|
||||
date_range = get_substract_date_range(num_day_type)
|
||||
stat_type_history = []
|
||||
range_decoder = []
|
||||
all_type = get_uuid_all_types_disk(uuid_sensor)
|
||||
|
||||
default_dict_type = {}
|
||||
for type in all_type:
|
||||
default_dict_type[type] = 0
|
||||
|
||||
for date in date_range:
|
||||
day_type = default_dict_type.copy()
|
||||
daily_stat = get_uuid_disk_statistics(uuid_sensor, date, all_types_on_disk=all_type, all_stats=False)
|
||||
day_type['date']= date[0:4] + '-' + date[4:6] + '-' + date[6:8]
|
||||
for type_key in daily_stat:
|
||||
day_type[type_key] += daily_stat[type_key][stats]
|
||||
stat_type_history.append(day_type)
|
||||
|
||||
return jsonify(stat_type_history)
|
||||
else:
|
||||
return jsonify('Incorrect UUID')
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
|
|||
|
|
@ -43,6 +43,7 @@
|
|||
UUID: {{uuid_sensor}}
|
||||
</div>
|
||||
<div class="card-body">
|
||||
{{data_uuid['description']}}
|
||||
<div class="card-group">
|
||||
<div class="card">
|
||||
<div class="card-header bg-info text-white">
|
||||
|
|
@ -147,7 +148,7 @@
|
|||
Types Used:
|
||||
</div>
|
||||
<div class="row ml-0 mr-0">
|
||||
<div class="col-lg-4">
|
||||
<div class="col-xl-4">
|
||||
<div class="mt-2">
|
||||
<table class="table table-striped table-bordered table-hover" id="myTable_1">
|
||||
<thead class="thead-dark">
|
||||
|
|
@ -169,7 +170,7 @@
|
|||
</table>
|
||||
</div>
|
||||
</div>
|
||||
<div class="col-lg-8">
|
||||
<div class="col-xl-8">
|
||||
<div id="barchart_type">
|
||||
</div>
|
||||
</div>
|
||||
|
|
@ -177,6 +178,47 @@
|
|||
</div>
|
||||
</div>
|
||||
|
||||
<div>
|
||||
<div class="card text-center mt-3 mx-3">
|
||||
<div class="card-header bg-dark text-white">
|
||||
Data Saved:
|
||||
</div>
|
||||
<div class="row ml-0 mr-0">
|
||||
<div class="col-xl-4">
|
||||
<div class="mt-2">
|
||||
<table class="table table-striped table-bordered table-hover" id="myTable_2">
|
||||
<thead class="thead-dark">
|
||||
<tr>
|
||||
<th>Type</th>
|
||||
<th style="max-width: 800px;">Size (Ko)</th>
|
||||
<th style="max-width: 800px;">Nb Files</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{% for type_stats in disk_stats %}
|
||||
<tr>
|
||||
<td>{{type_stats}}</td>
|
||||
<td>{{disk_stats[type_stats]['total_size']}}</td>
|
||||
<td>{{disk_stats[type_stats]['nb_files']}}</td>
|
||||
</tr>
|
||||
{% endfor %}
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
</div>
|
||||
<div class="col-xl-8">
|
||||
<input value="nb_files" id="type_stats_disk" hidden></input>
|
||||
<h4 id="stats_disk_title">Number of files save on disk :</h4>
|
||||
<div id="barchart_type_disk">
|
||||
</div>
|
||||
<button type="button" id="stats_disk_btn" class="btn btn-outline-secondary mt-1" onclick="swap_stats_type();">
|
||||
Show Size Chart
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="row ml-0 mr-0">
|
||||
<div class="col-lg-6">
|
||||
<div class="card text-center mt-3">
|
||||
|
|
@ -215,8 +257,16 @@ $(document).ready(function(){
|
|||
"iDisplayLength": 10,
|
||||
"order": [[ 0, "asc" ]]
|
||||
}
|
||||
);
|
||||
chart.stackBarChart =barchart_type_stack("{{ url_for('get_uuid_type_history_json') }}?uuid_sensor={{uuid_sensor}}", 'id');
|
||||
);
|
||||
table = $('#myTable_2').DataTable(
|
||||
{
|
||||
"aLengthMenu": [[5, 10, 15, 20, -1], [5, 10, 15, 20, "All"]],
|
||||
"iDisplayLength": 10,
|
||||
"order": [[ 0, "asc" ]]
|
||||
}
|
||||
);
|
||||
chart.stackBarChart1 =barchart_type_stack("{{ url_for('get_uuid_type_history_json') }}?uuid_sensor={{uuid_sensor}}", '#barchart_type');
|
||||
chart.stackBarChart2 =barchart_type_stack("{{ url_for('get_uuid_stats_history_json') }}?uuid_sensor={{uuid_sensor}}", '#barchart_type_disk');
|
||||
|
||||
chart.onResize();
|
||||
$(window).on("resize", function() {
|
||||
|
|
@ -231,14 +281,34 @@ $(document).ready(function(){
|
|||
|
||||
});
|
||||
|
||||
function get_whois_data(ip){
|
||||
function get_whois_data(ip){
|
||||
|
||||
$.getJSON( "{{url_for('whois_data')}}?ip="+ip, function( data ) {
|
||||
$( "#whois_data" ).removeClass( "d-none" );
|
||||
$( "#whois_output" ).text(data);
|
||||
$.getJSON( "{{url_for('whois_data')}}?ip="+ip, function( data ) {
|
||||
$( "#whois_data" ).removeClass( "d-none" );
|
||||
$( "#whois_output" ).text(data);
|
||||
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
function swap_stats_type(){
|
||||
var stats_value = $('#type_stats_disk').val();
|
||||
if(stats_value==='nb_files'){
|
||||
$('#type_stats_disk').val('total_size');
|
||||
$('#stats_disk_title').text('Size of files save on disk :');
|
||||
$('#stats_disk_btn').text('Show # Files Chart');
|
||||
stats_value = 'total_size';
|
||||
} else {
|
||||
$('#type_stats_disk').val('nb_files');
|
||||
$('#stats_disk_title').text('Number of files save on disk :');
|
||||
$('#stats_disk_btn').text('Show Size Chart');
|
||||
stats_value = 'nb_files';
|
||||
}
|
||||
|
||||
$('#barchart_type_disk').children().remove();
|
||||
url_json_stats = "{{ url_for('get_uuid_stats_history_json') }}?uuid_sensor={{uuid_sensor}}&stats=" + stats_value;
|
||||
chart.stackBarChart2 =barchart_type_stack(url_json_stats, '#barchart_type_disk');
|
||||
chart.onResize();
|
||||
}
|
||||
</script>
|
||||
|
||||
<script>
|
||||
|
|
@ -246,27 +316,27 @@ var margin = {top: 20, right: 90, bottom: 55, left: 0},
|
|||
width = parseInt(d3.select('#barchart_type').style('width'), 10);
|
||||
width = 1000 - margin.left - margin.right,
|
||||
height = 500 - margin.top - margin.bottom;
|
||||
var x = d3.scaleBand().rangeRound([0, width]).padding(0.1);
|
||||
|
||||
var y = d3.scaleLinear().rangeRound([height, 0]);
|
||||
|
||||
var xAxis = d3.axisBottom(x);
|
||||
|
||||
var yAxis = d3.axisLeft(y);
|
||||
|
||||
var color = d3.scaleOrdinal(d3.schemeSet3);
|
||||
|
||||
var svg = d3.select("#barchart_type").append("svg")
|
||||
.attr("id", "thesvg")
|
||||
.attr("viewBox", "0 0 "+width+" 500")
|
||||
.attr("width", width + margin.left + margin.right)
|
||||
.attr("height", height + margin.top + margin.bottom)
|
||||
.append("g")
|
||||
.attr("transform", "translate(" + margin.left + "," + margin.top + ")");
|
||||
|
||||
|
||||
function barchart_type_stack(url, id) {
|
||||
|
||||
var x = d3.scaleBand().rangeRound([0, width]).padding(0.1);
|
||||
|
||||
var y = d3.scaleLinear().rangeRound([height, 0]);
|
||||
|
||||
var xAxis = d3.axisBottom(x);
|
||||
|
||||
var yAxis = d3.axisLeft(y);
|
||||
|
||||
var color = d3.scaleOrdinal(d3.schemeSet3);
|
||||
|
||||
var svg = d3.select(id).append("svg")
|
||||
.attr("id", "thesvg")
|
||||
.attr("viewBox", "0 0 "+width+" 500")
|
||||
.attr("width", width + margin.left + margin.right)
|
||||
.attr("height", height + margin.top + margin.bottom)
|
||||
.append("g")
|
||||
.attr("transform", "translate(" + margin.left + "," + margin.top + ")");
|
||||
|
||||
d3.json(url)
|
||||
.then(function(data){
|
||||
|
||||
|
|
@ -346,29 +416,43 @@ function barchart_type_stack(url, id) {
|
|||
drawLegend(varNames);
|
||||
});
|
||||
|
||||
}
|
||||
function drawLegend (varNames) {
|
||||
var legend = svg.selectAll(".legend")
|
||||
.data(varNames.slice().reverse())
|
||||
.enter().append("g")
|
||||
.attr("class", "legend")
|
||||
.attr("transform", function (d, i) { return "translate(0," + i * 20 + ")"; });
|
||||
|
||||
function drawLegend (varNames) {
|
||||
var legend = svg.selectAll(".legend")
|
||||
.data(varNames.slice().reverse())
|
||||
.enter().append("g")
|
||||
.attr("class", "legend")
|
||||
.attr("transform", function (d, i) { return "translate(0," + i * 20 + ")"; });
|
||||
legend.append("rect")
|
||||
.attr("x", 943)
|
||||
.attr("width", 10)
|
||||
.attr("height", 10)
|
||||
.style("fill", color)
|
||||
.style("stroke", "grey");
|
||||
|
||||
legend.append("rect")
|
||||
.attr("x", 943)
|
||||
.attr("width", 10)
|
||||
.attr("height", 10)
|
||||
.style("fill", color)
|
||||
.style("stroke", "grey");
|
||||
legend.append("text")
|
||||
.attr("class", "svgText")
|
||||
.attr("x", 941)
|
||||
.attr("y", 6)
|
||||
.attr("dy", ".35em")
|
||||
.style("text-anchor", "end")
|
||||
.text(function (d) { return d; });
|
||||
}
|
||||
|
||||
function showPopover (d) {
|
||||
$(this).popover({
|
||||
title: d.name,
|
||||
placement: 'top',
|
||||
container: 'body',
|
||||
trigger: 'manual',
|
||||
html : true,
|
||||
content: function() {
|
||||
return d.label +
|
||||
"<br/>num: " + d3.format(",")(d.value ? d.value: d.y1 - d.y0); }
|
||||
});
|
||||
$(this).popover('show')
|
||||
}
|
||||
|
||||
legend.append("text")
|
||||
.attr("class", "svgText")
|
||||
.attr("x", 941)
|
||||
.attr("y", 6)
|
||||
.attr("dy", ".35em")
|
||||
.style("text-anchor", "end")
|
||||
.text(function (d) { return d; });
|
||||
}
|
||||
|
||||
function removePopovers () {
|
||||
|
|
@ -377,25 +461,16 @@ function removePopovers () {
|
|||
});
|
||||
}
|
||||
|
||||
function showPopover (d) {
|
||||
$(this).popover({
|
||||
title: d.name,
|
||||
placement: 'top',
|
||||
container: 'body',
|
||||
trigger: 'manual',
|
||||
html : true,
|
||||
content: function() {
|
||||
return d.label +
|
||||
"<br/>num: " + d3.format(",")(d.value ? d.value: d.y1 - d.y0); }
|
||||
});
|
||||
$(this).popover('show')
|
||||
function resize_chart_by_id(id_chart) {
|
||||
var aspect = width / height, chart = $(id_chart).children();
|
||||
var targetWidth = chart.parent().width();
|
||||
chart.attr("width", targetWidth);
|
||||
chart.attr("height", targetWidth / 2);
|
||||
}
|
||||
|
||||
chart.onResize = function () {
|
||||
var aspect = width / height, chart = $("#thesvg");
|
||||
var targetWidth = chart.parent().width();
|
||||
chart.attr("width", targetWidth);
|
||||
chart.attr("height", targetWidth / 2);
|
||||
resize_chart_by_id("#barchart_type");
|
||||
resize_chart_by_id("#barchart_type_disk");
|
||||
}
|
||||
|
||||
window.chart = chart;
|
||||
|
|
|
|||
Loading…
Reference in New Issue