Nagios Support Forum

Code: Select all

register_custom_api_callback('custom', 'commandtest', 'custom_commandtest');

function custom_commandtest($args) {
	
	global $cfg;
    global $ccm;

	// Initialization stuff
	pre_init();

	// Grab GET or POST variables and check pre-reqs
	grab_request_vars();
	check_prereqs();

	// Verify access
	if (!user_can_access_ccm()) {
		die(_('You do not have access to this page.'));
	}

	// Main includes
	require_once(dirname(__FILE__).'/../ccm/includes/common_functions.inc.php'); 
	require_once(dirname(__FILE__).'/../ccm/includes/language.inc.php'); 
	require_once(dirname(__FILE__).'/../ccm/includes/auth.inc.php'); 
	require_once(dirname(__FILE__).'/../ccm/includes/hidden_overlay_functions.inc.php'); 

	// CCM classes
	require_once(dirname(__FILE__).'/../ccm/classes/config.class.php'); 
	require_once(dirname(__FILE__).'/../ccm/classes/data.class.php');
	require_once(dirname(__FILE__).'/../ccm/classes/import.class.php');
	require_once(dirname(__FILE__).'/../ccm/classes/db.class.php');
	require_once(dirname(__FILE__).'/../ccm/classes/form.class.php');

	// CCM main includes 
	require_once(dirname(__FILE__).'/../ccm/page_templates/ccm_table.php');
	require_once(dirname(__FILE__).'/../ccm/includes/page_router.inc.php');

	// Currently we only support single domain configs 
	$_SESSION['domain'] = 1;

	// Create connections
	$ccm = new stdClass();
	$ccm->db = new DB($cfg['db_info']['nagiosql']);
	$ccm->import = new Import();
	$ccm->config = new Config();
	$ccm->data = new Data();


    // Get all necessary parameters for a check command
    $cid = intval(ccm_grab_request_var('cid')); 
    $address = ccm_grab_request_var('address', ''); 
    $host = ccm_grab_array_var($_REQUEST, 'host', '');
    $arg1 = ccm_grab_array_var($_REQUEST, 'arg1', '');
    $arg2 = ccm_grab_array_var($_REQUEST, 'arg2', '');
    $arg3 = ccm_grab_array_var($_REQUEST, 'arg3', '');
    $arg4 = ccm_grab_array_var($_REQUEST, 'arg4', '');
    $arg5 = ccm_grab_array_var($_REQUEST, 'arg5', '');
    $arg6 = ccm_grab_array_var($_REQUEST, 'arg6', '');
    $arg7 = ccm_grab_array_var($_REQUEST, 'arg7', '');
    $arg8 = ccm_grab_array_var($_REQUEST, 'arg8', '');

	// Grab the command we were sent from the database
    if ($cid != "") {
		$query = "SELECT `command_name`, `command_line` FROM tbl_command WHERE `id` = '$cid' && command_type = 1 LIMIT 1";
	}
	else {
		$query = "SELECT `id`, `command_type`, `active`, `command_name`, `command_line` FROM tbl_command WHERE command_type = 1";
	}
	$command = $ccm->db->query($query); 
	
	if ($cid != "") {
		// If the command isn't in the database then fail now
		if (!isset($command[0]['command_name'])) {
			print "ERROR: Unable to locate the command in the database";
			exit(); 
		}
		
		// Create all the database variables and then replace all variables that need to be replaced
		// before running the actual command and echoing the output to the screen.
		$name = $command[0]['command_name'];
		$cmd_line = $command[0]['command_line'];
		$haystack = array($cfg['component_info']['nagioscore']['plugin_dir'], $host, $address, $arg1, $arg2, $arg3, $arg4, $arg5, $arg6, $arg7, $arg8);
		$needles = array('$USER1$', '$HOSTNAME$', '$HOSTADDRESS$', '$ARG1$', '$ARG2$', '$ARG3$', '$ARG4$', '$ARG5$', '$ARG6$', '$ARG7$', '$ARG8$');

		$fullcommand = str_replace($needles, $haystack, $cmd_line);

		// break apart everything after an unescaped semi-colon
		$fullcommand = str_replace("\;", "%%%%%", $fullcommand);
		$fullcommand_array = explode(';', $fullcommand);
		$fullcommand = $fullcommand_array[0];
		$fullcommand = str_replace("%%%%%", "\;", $fullcommand);    

		// Grab the current value of $fullcommand and store it to display in the run check command results box in order to maintain obfuscation
		$displaycommand = $fullcommand;

		$fullcommand = nagiosccm_replace_user_macros($fullcommand);

		$id = submit_command(COMMAND_RUN_CHECK_CMD, $fullcommand);

		if ($id <= 0) {
			$output = _("Error submitting command.");
		}
		else {
			for ($x = 0; $x < 40; $x++) {
				usleep(500000);
				$args = array(
					"command_id" => $id
				);
				$xml = get_command_status_xml($args);
				if ($xml) {
					if ($xml->command[0]) {
						if (intval($xml->command[0]->status_code) == 2) {
							$output = $xml->command[0]->result;
							break;
						}
					}
				}
			}
		}

		$hn = (function_exists('gethostname')) ? gethostname() : php_uname('n');
		if (stripos('.', $hn)) {
			$hna = explode('.', $hn);
			$hn = $hna[0];
		}
		$response = $output;
	}
	else {
		$response = $command;
	}

	return $response;
}