From f78b6a957627a320bbceccd9066a22449f5b5724 Mon Sep 17 00:00:00 2001
From: Ramon Novoa <rnovoa@artica.es>
Date: Wed, 6 Sep 2017 17:08:55 +0200
Subject: [PATCH] Change pandora_db.pl to work with the Elastic log collector.

---
 pandora_server/util/pandora_db.pl | 77 ++++---------------------------
 1 file changed, 9 insertions(+), 68 deletions(-)

diff --git a/pandora_server/util/pandora_db.pl b/pandora_server/util/pandora_db.pl
index 71b4805eda..786025d314 100644
--- a/pandora_server/util/pandora_db.pl
+++ b/pandora_server/util/pandora_db.pl
@@ -407,36 +407,15 @@ sub pandora_purgedb ($$) {
 	
 	# Delete old log data
 	log_message ('PURGE', "Deleting old log data.");
-	if (! defined ($conf->{'_log_dir'}) || ! -d $conf->{'_log_dir'}) {
-		log_message ('!', "Log data directory does not exist, skipping.");
+	if (!defined ($conf->{'logstash_host'}) || $conf->{'logstash_host'} eq '') {
+		log_message ('!', "Log collection disabled.");
 	}
-	elsif ($conf->{'_log_max_lifetime'} > 0) {
-		log_message ('PURGE', 'Deleting log data older than ' . $conf->{'_log_max_lifetime'} . ' days.');
-		
-		# Calculate the limit date
-		my ($sec,$min,$hour,$mday,$mon,$year) = localtime(time() - $conf->{'_log_max_lifetime'} * 86400); 
-		
-		# Fix the year
-		$year += 1900;
-		
-		# Fix the month
-		$mon += 1;
-		$mon = sprintf("%02d", $mon);
-		
-		# Fix the day
-		$mday = sprintf("%02d", $mday);
-		
-		# Fix the hour
-		$hour = sprintf("%02d", $hour);
-		
-		# Set the per-depth limits
-		my $limits = [$year, $mon, $mday, $hour];
-		
-		# Purge the log dir
-		pandora_purge_log_dir ($conf->{'_log_dir'}, $limits);
+	elsif (defined($conf->{'_days_purge_old_information'}) && $conf->{'_days_purge_old_information'} > 0) {
+		log_message ('PURGE', 'Deleting log data older than ' . $conf->{'_days_purge_old_information'} . ' days.');
+    	enterprise_hook ('pandora_purge_logs', [$dbh, $conf]);
 	}
 	else {
-		log_message ('PURGE', 'log_max_lifetime is set to 0. Old log data will not be deleted.');
+		log_message ('PURGE', 'days_purge_old_data is set to 0. Old log data will not be deleted.');
 	}
 
 	# Delete old special days
@@ -454,45 +433,6 @@ sub pandora_purgedb ($$) {
 	}
 }
 
-########################################################################
-# Recursively delete old log files by sub directory.
-########################################################################
-sub pandora_purge_log_dir ($$;$) {
-	my ($dir, $limits, $depth) = @_;
-	
-	# Initial call
-	if (! defined ($depth)) {
-		$depth = 0;
-	}
-
-	# No limit for this depth
-	if (! defined ($limits->[$depth])) {
-		return;
-	}
-
-	# Open the dir
-	my $dir_dh;
-	if (! opendir($dir_dh, $dir)) {
-		return;
-	}
-
-	# Purge sub dirs
-	while (my $sub_dir = readdir ($dir_dh)) {
-
-		next if ($sub_dir eq '.' || $sub_dir eq '..' || ! -d $dir . '/' . $sub_dir);
-				
-		# Sub dirs have names that represent a year, month, day or hour
-		if ($sub_dir < $limits->[$depth]) {
-			rmtree ($dir . '/' . $sub_dir);
-		} elsif ($sub_dir == $limits->[$depth]) {
-			&pandora_purge_log_dir ($dir . '/' . $sub_dir, $limits, $depth + 1)
-		}
-	}
-	
-	# Close the dir
-	closedir ($dir_dh);
-}
-
 ###############################################################################
 # Compact agent data.
 ###############################################################################
@@ -715,13 +655,14 @@ sub pandora_load_config ($) {
 	$conf->{'_netflow_max_lifetime'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'netflow_max_lifetime'");
 	$conf->{'_netflow_nfexpire'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'netflow_nfexpire'");
    	$conf->{'_netflow_path'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'netflow_path'");
-   	$conf->{'_log_dir'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'log_dir'");
-   	$conf->{'_log_max_lifetime'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'log_max_lifetime'");
 	$conf->{'_delete_notinit'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'delete_notinit'");
 
 	$conf->{'_big_operation_step_datos_purge'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'big_operation_step_datos_purge'");
 	$conf->{'_small_operation_step_datos_purge'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'small_operation_step_datos_purge'");
 	$conf->{'_days_autodisable_deletion'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'days_autodisable_deletion'");
+	$conf->{'_days_purge_old_information'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'Days_purge_old_information'");
+	$conf->{'_elasticsearch_ip'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'elasticsearch_ip'");
+	$conf->{'_elasticsearch_port'} = get_db_value ($dbh, "SELECT value FROM tconfig WHERE token = 'elasticsearch_port'");
 
 	$BIG_OPERATION_STEP = $conf->{'_big_operation_step_datos_purge'}
 					if ( $conf->{'_big_operation_step_datos_purge'} );