wanderer/ fastnetmon-ng
1
1
Fork 0
Code Issues 2 Pull Requests Projects 1 Releases Activity

Add optional redis prefix keys feature

This commit is contained in:
Vicente De Luca 2015-08-17 18:50:06 +01:00
parent d1d6d2bcd9
commit c96379e4b2
2 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/fastnetmon.conf
View File

@ -155,6 +155,7 @@ redis_enabled = no
# Reddis configuration
redis_port = 6379
redis_host = 127.0.0.1
redis_prefix = off
# ExaBGP could announce blocked IPs with BGP protocol
exabgp = off

17
src/fastnetmon.cpp
View File

@ -137,6 +137,7 @@ u_int32_t ndpi_size_id_struct = 0;
#ifdef REDIS
unsigned int redis_port = 6379;
std::string redis_host = "127.0.0.1";
std::string redis_prefix = NULL;
// because it's additional and very specific feature we should disable it by default
bool redis_enabled = false;
@ -1044,6 +1045,10 @@ bool load_configuration_file() {
redis_host = configuration_map["redis_host"];
}
if (configuration_map.count("redis_prefix") != "off") {
redis_host = configuration_map["redis_prefix"];
}
if (configuration_map.count("redis_enabled") != 0) {
// We use yes and on because it's stupid typo :(
if (configuration_map["redis_enabled"] == "on" or configuration_map["redis_enabled"] == "yes") {
@ -2862,6 +2867,10 @@ void call_ban_handlers(uint32_t client_ip, attack_details& current_attack, std::
if (redis_enabled) {
std::string redis_key_name = client_ip_as_string + "_information";
if (redis_prefix != NULL) {
std::string redis_key_name = redis_prefix + "_" + client_ip_as_string + "_information";
}
logger << log4cpp::Priority::INFO << "Start data save in redis in key: " << redis_key_name;
boost::thread redis_store_thread(store_data_in_redis, redis_key_name, basic_attack_information_in_json);
redis_store_thread.detach();
@ -2872,6 +2881,10 @@ void call_ban_handlers(uint32_t client_ip, attack_details& current_attack, std::
if (redis_enabled && !flow_attack_details.empty()) {
std::string redis_key_name = client_ip_as_string + "_flow_dump";
if (redis_prefix != NULL) {
std::string redis_key_name = redis_prefix + "_" + client_ip_as_string + "_flow_dump";
}
logger << log4cpp::Priority::INFO << "Start data save in redis in key: " << redis_key_name;
boost::thread redis_store_thread(store_data_in_redis, redis_key_name, flow_attack_details);
redis_store_thread.detach();
@ -3434,6 +3447,10 @@ void call_attack_details_handlers(uint32_t client_ip, attack_details& current_at
if (redis_enabled) {
std::string redis_key_name = client_ip_as_string + "_packets_dump";
if (redis_prefix != NULL) {
std::string redis_key_name = redis_prefix + "_" + client_ip_as_string + "_packets_dump";
}
logger << log4cpp::Priority::INFO << "Start data save in redis for key: " << redis_key_name;
boost::thread redis_store_thread(store_data_in_redis, redis_key_name, attack_fingerprint);
redis_store_thread.detach();