From 182df1c12da90bb6e52cf3adcc95794a34625aa2 Mon Sep 17 00:00:00 2001
From: fbsanchez <fborja.sanchez@artica.es>
Date: Thu, 1 Oct 2020 10:01:08 +0200
Subject: [PATCH] WIP omnishell

---
 pandora_agents/unix/pandora_agent          | 23 ++++++--------------
 pandora_server/lib/PandoraFMS/Omnishell.pm | 25 ++++++++++++++++++++++
 2 files changed, 31 insertions(+), 17 deletions(-)

diff --git a/pandora_agents/unix/pandora_agent b/pandora_agents/unix/pandora_agent
index 50195aea2a..51005a3f66 100755
--- a/pandora_agents/unix/pandora_agent
+++ b/pandora_agents/unix/pandora_agent
@@ -273,20 +273,6 @@ sub error ($) {
 	exit 1;
 }
 
-################################################################################
-# Try to load extra libraries.c
-################################################################################
-sub load_libraries() {
-	# Dynamic load. Avoid unwanted behaviour.
-	eval {eval 'require YAML::Tiny;1' or die('YAML::Tiny lib not found, commands feature won\'t be available');};
-	if ($@) {
-		$YAML = 0;
-		print STDERR $@;
-	} else {
-		$YAML = 1;
-	}
-}
-
 ################################################################################
 # Erase blank spaces before and after the string
 ################################################################################
@@ -3095,8 +3081,6 @@ my $iter_base_time = time();
 $LogFileIdx = -1;
 # Loop
 while (1) {
-	load_libraries();
-
 	if (-e $Conf{'logfile'} && (stat($Conf{'logfile'}))[7] > $Conf{'logsize'}) {
 		rotate_log();
 	}
@@ -3114,7 +3098,12 @@ while (1) {
 	if ($Conf{'debug'} ne '1') {
 		# Check scheduled commands	
 		my $omni = new PandoraFMS::Omnishell(\%Conf);
-		$omni->prepare_commands();
+		log_message('log', "Running omnishell process");
+		if (!$omni->run()) {
+			log_message('error', "Failed to run omnishell process: ".$omni->get_last_error());
+		} else {
+			log_message('log', "Omnishell process completed.");
+		}
 	}
 
 	# Launch broker agents
diff --git a/pandora_server/lib/PandoraFMS/Omnishell.pm b/pandora_server/lib/PandoraFMS/Omnishell.pm
index b47955470c..3a46a81b49 100644
--- a/pandora_server/lib/PandoraFMS/Omnishell.pm
+++ b/pandora_server/lib/PandoraFMS/Omnishell.pm
@@ -40,6 +40,21 @@ sub get_last_error {
   return '';
 }
 
+################################################################################
+# Try to load extra libraries.c
+################################################################################
+sub load_libraries {
+	my $self = shift;
+
+	# Dynamic load. Avoid unwanted behaviour.
+	eval {eval 'require YAML::Tiny;1' or die('YAML::Tiny lib not found, commands feature won\'t be available');};
+	if ($@) {
+		$self->set_last_error($@);
+		return 0;
+	} else {
+		return 1;
+	}
+}
 
 ################################################################################
 # Create new omnishell handler.
@@ -62,6 +77,16 @@ sub new {
   return $self;
 }
 
+################################################################################
+# Run process.
+################################################################################
+sub run {
+	my ($self) = @_;
+
+	if($self->load_libraries()) {
+		$self->prepare_commands();
+	}
+}
 
 ################################################################################
 # Check for remote commands defined.