pandorafms/pandora_console/extensions/plugin_registration.php

<?php

// Pandora FMS - http://pandorafms.com
// ==================================================
// Copyright (c) 2005-2011 Artica Soluciones Tecnologicas
// Please see http://pandorafms.org for full contribution list

// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; version 2

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

function pluginreg_extension_main () {
	global $config;

	if (! check_acl ($config['id_user'], 0, "PM") && ! is_user_admin ($config['id_user'])) {
		pandora_audit("ACL Violation", "Trying to access Setup Management");
		require ("general/noaccess.php");
		return;
	}

	ui_print_page_header (__('Plugin registration'), "images/extensions.png", false, "", true, "" );

	echo "<div class=notify>";
	printf(__("This extension makes registration of server plugins more easy. Here you can upload a server plugin in Pandora FMS 3.x zipped format (.pspz). Please refer to documentation on how to obtain and use Pandora FMS Server Plugins.<br><br>You can get more plugins in our <a href='%s'>Public Resource Library</a>") , "http://pandorafms.org/index.php?sec=community&sec2=repository&lng=en");
	echo "</div>";
	
	echo "<br><br>";

	if (!isset ($_FILES['plugin_upload']['tmp_name'])){
		// Upload form
		echo "<form name='submit_plugin' method='post' enctype='multipart/form-data'>";
		echo '<table class="databox" id="table1" width="50%" border="0" cellpadding="4" cellspacing="4">';
		echo "<tr><td class='datos'><input type='file' name='plugin_upload' />";
		echo "<td class='datos'><input type='submit' class='sub next' value='".__('Upload')."' />";
		echo "</form></table>";
		
		return;
	} 	
	
	$config["plugin_store"] = $config["attachment_store"] . "/plugin";
	$zip = zip_open($_FILES['plugin_upload']['tmp_name']);

	if ($zip) {
		while ($zip_entry = zip_read($zip)) {
			if (zip_entry_open($zip, $zip_entry, "r")) {
				if (zip_entry_name($zip_entry) == "plugin_definition.ini") {
					$basepath = $config["attachment_store"];
				}
				else {
					$basepath = $config["plugin_store"];
				}
				$filename = $basepath . "/". zip_entry_name($zip_entry);
				$fp = fopen($filename, 'w');
				$buf = zip_entry_read($zip_entry, zip_entry_filesize($zip_entry));
					fwrite($fp, $buf);
					fclose($fp);
					chmod ($filename, 0755);
				zip_entry_close($zip_entry);
			}
		}
		zip_close($zip);
	}

	// Parse with sections
	if (! $ini_array = parse_ini_file($config["attachment_store"] . "/plugin_definition.ini", true)){
		echo "<h2 class=error>".__("Cannot load INI file")."</h2>";
		return;
	}

	$exec_path = $config["plugin_store"] . "/" . $ini_array["plugin_definition"]["filename"];
	
	$file_exec_path = $exec_path;
	
	if (isset($ini_array["plugin_definition"]["execution_command"]) && ($ini_array["plugin_definition"]["execution_command"] != "")){
		$exec_path = $ini_array["plugin_definition"]["execution_command"] . " " . $config["plugin_store"] . "/" . $ini_array["plugin_definition"]["filename"];
	}
	
	if (isset($ini_array["plugin_definition"]["execution_postcommand"]) && ($ini_array["plugin_definition"]["execution_postcommand"] != "")){
		$exec_path = $exec_path . " " .$ini_array["plugin_definition"]["execution_postcommand"];
	}
	
	if (!file_exists($file_exec_path)){
		echo "<h2 class=error>".__("Plugin exec not found. Aborting!")."</h2>";
		unlink ($config["attachment_store"] . "/plugin_definition.ini");
		return;
	}

	// Verify if a plugin with the same name is already registered

	$sql0 = "SELECT COUNT(*) FROM tplugin WHERE name = '" . safe_input ($ini_array["plugin_definition"]["name"]) . "'";
	$result = get_db_sql ($sql0);
	
	
	if ($result> 0) {
		echo "<h2 class=error>".__("Plugin already registered. Aborting!")."</h2>";
		unlink ($config["attachment_store"] . "/plugin_definition.ini");
		return;;
	}

	$values = array(
		'name' => safe_input ($ini_array["plugin_definition"]["name"]),
		'description' => safe_input ($ini_array["plugin_definition"]["description"]),
		'max_timeout' => $ini_array["plugin_definition"]["timeout"],
		'execute' => safe_input ($exec_path),
		'net_dst_opt' => $ini_array["plugin_definition"]["ip_opt"],
		'net_port_opt' => $ini_array["plugin_definition"]["port_opt"],
		'user_opt' => $ini_array["plugin_definition"]["user_opt"],
		'pass_opt' => $ini_array["plugin_definition"]["pass_opt"],
		'plugin_type' => $ini_array["plugin_definition"]["plugin_type"]);
	
	$create_id = process_sql_insert('tplugin', $values);
	
	$values = array(
		'name' => safe_input ($ini_array["plugin_definition"]["name"]),
		'description' => safe_input ($ini_array["plugin_definition"]["description"]),
		'max_timeout' => $ini_array["plugin_definition"]["timeout"],
		'execute' => safe_input ($exec_path),
		'net_dst_opt' => $ini_array["plugin_definition"]["ip_opt"],
		'net_port_opt' => $ini_array["plugin_definition"]["port_opt"],
		'user_opt' => $ini_array["plugin_definition"]["user_opt"],
		'pass_opt' => $ini_array["plugin_definition"]["pass_opt"],
		'plugin_type' => $ini_array["plugin_definition"]["plugin_type"]);
	$create_id = process_sql_insert('tplugin', $values);

	for ($ax=1; $ax <= $ini_array["plugin_definition"]["total_modules_provided"]; $ax++){
		$label = "module".$ax;
		
		$values = array(
			'name' => safe_input ($ini_array[$label]["name"]),
			'description' => safe_input ($ini_array[$label]["description"]),
			'id_group' => $ini_array[$label]["id_group"],
			'type' => $ini_array[$label]["type"],
			'max' => $ini_array[$label]["max"],
			'min' => $ini_array[$label]["min"],
			'module_interval' => $ini_array[$label]["module_interval"],
			'id_module_group' => $ini_array[$label]["id_module_group"],
			'id_modulo' => $ini_array[$label]["id_modulo"], 
			'plugin_user' => safe_input ($ini_array[$label]["plugin_user"]),
			'plugin_pass' => safe_input ($ini_array[$label]["plugin_pass"]),
			'plugin_parameter' => safe_input ($ini_array[$label]["plugin_parameter"]),
			'max_timeout' => $ini_array[$label]["max_timeout"],
			'history_data' => $ini_array[$label]["history_data"],
			'min_warning' => $ini_array[$label]["min_warning"],
			'min_critical' => $ini_array[$label]["min_critical"],
			'min_ff_event' => $ini_array[$label]["min_ff_event"],
			'tcp_port' => $ini_array[$label]["tcp_port"],
			'id_plugin' => $create_id);
		process_sql_insert('tnetwork_component', $values);
		
		echo "<h3 class=suc>".__("Module plugin registered"). " : ". $ini_array[$label]["name"] ."</h2>";
	}
	
	echo "<h2 class=suc>".__("Plugin"). " ". $ini_array["plugin_definition"]["name"] . " ". __("Registered successfully")."</h2>";
	unlink ($config["attachment_store"] . "/plugin_definition.ini");	

}

add_godmode_menu_option (__('Register plugin'), 'PM','gservers','');
add_extension_godmode_function('pluginreg_extension_main');

?>