Files
cacti/lib/data_query.php
T
cigamit 4fd458069f Use preg_split over explode for issue #865
Plus use an additional trim on the string to prevent out of order
arguments
2017-07-26 21:33:34 -05:00

1527 lines
57 KiB
PHP

<?php
/*
+-------------------------------------------------------------------------+
| Copyright (C) 2004-2017 The Cacti Group |
| |
| 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; either version 2 |
| of the License, or (at your option) any later version. |
| |
| 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. |
+-------------------------------------------------------------------------+
| Cacti: The Complete RRDTool-based Graphing Solution |
+-------------------------------------------------------------------------+
| This code is designed, written, and maintained by the Cacti Group. See |
| about.php and/or the AUTHORS file for specific developer information. |
+-------------------------------------------------------------------------+
| http://www.cacti.net/ |
+-------------------------------------------------------------------------+
*/
function run_data_query($host_id, $snmp_query_id) {
global $config, $input_types;
/* don't run/rerun the query if the host is down, or disabled */
$status = db_fetch_row_prepared('SELECT status, disabled, poller_id FROM host WHERE id = ?', array($host_id));
if (!sizeof($status)) {
return false;
} elseif ($status['status'] == HOST_DOWN || $status['disabled'] == 'on') {
return true;
}
$poller_id = $status['poller_id'];
if ($poller_id != $config['poller_id']) {
$hostname = db_fetch_cell_prepared('SELECT hostname FROM poller WHERE id = ?', array($poller_id));
$response = file_get_contents(get_url_type() . '://' . $hostname . $config['url_path'] . '/remote_agent.php?action=runquery&host_id=' . $host_id . '&data_query_id=' . $snmp_query_id);
if ($response != '') {
$response = json_decode($response, true);
$_SESSION['debug_log']['data_query'] = $response['data_query'];
if (isset($_SESSION['debug_log']['response'])) {
$result = $_SESSION['debug_log']['response'];
unset($_SESSION['debug_log']['response']);
return $result;
}
}
return false;
}
query_debug_timer_start();
include_once($config['library_path'] . '/poller.php');
include_once($config['library_path'] . '/api_data_source.php');
include_once($config['library_path'] . '/utility.php');
query_debug_timer_offset('data_query', "Running data query [$snmp_query_id].");
$type_id = db_fetch_cell_prepared('SELECT data_input.type_id
FROM snmp_query
INNER JOIN data_input
ON snmp_query.data_input_id=data_input.id
WHERE snmp_query.id = ?', array($snmp_query_id));
if (isset($input_types[$type_id])) {
query_debug_timer_offset('data_query', "Found type = '" . $type_id . "' [" . $input_types[$type_id] . "].");
}
if ($type_id == DATA_INPUT_TYPE_SNMP_QUERY) {
$result = query_snmp_host($host_id, $snmp_query_id);
} elseif ($type_id == DATA_INPUT_TYPE_SCRIPT_QUERY) {
$result = query_script_host($host_id, $snmp_query_id);
} elseif ($type_id == DATA_INPUT_TYPE_QUERY_SCRIPT_SERVER) {
$result = query_script_host($host_id, $snmp_query_id);
} else {
$result = false;
$arguments = array(
'result' => $result,
'host_id' => $host_id,
'snmp_query_id' => $snmp_query_id
);
$arguments = api_plugin_hook_function('run_data_query', $arguments);
if (isset($arguments['result']) && $arguments['result'] !== false) {
$result = $arguments['result'];
} else {
query_debug_timer_offset('data_query', "Unknown type = '$type_id'");
unset($result);
}
}
/* update the sort cache */
update_data_query_sort_cache($host_id, $snmp_query_id);
query_debug_timer_offset('data_query', 'Update data query sort cache complete');
/* recalculate/change sort order */
$data_queries = db_fetch_assoc_prepared('SELECT dl.host_id, dl.snmp_query_id,
dl.snmp_index, dtd.local_data_id, dtd.data_input_id,
did.data_template_data_id, did.data_input_field_id, did.value
FROM data_local AS dl
INNER JOIN data_template_data AS dtd
ON dl.id = dtd.local_data_id
INNER JOIN data_input_fields AS dif
ON dtd.data_input_id = dif.data_input_id
LEFT JOIN data_input_data AS did
ON dtd.id = did.data_template_data_id
AND dif.id = did.data_input_field_id
WHERE dif.type_code = "output_type"
AND dl.snmp_query_id = ?
AND dl.host_id = ?',
array($host_id, $snmp_query_id));
$bestDataQueryIndexType = get_best_data_query_index_type($host_id, $snmp_query_id);
if (sizeof($data_queries)) {
foreach($data_queries as $data_query) {
/* build array required for function call */
$data_query['snmp_index_on'] = $bestDataQueryIndexType;
/* as we request the output_type, 'value' gives the snmp_query_graph_id */
$data_query['snmp_query_graph_id'] = $data_query['value'];
update_snmp_index_order($data_query);
}
}
query_debug_timer_offset('data_query', 'Updated data query index ordering');
/* update the auto reindex cache */
update_reindex_cache($host_id, $snmp_query_id);
query_debug_timer_offset('data_query', 'Update re-index cache complete');
/* update the the 'local' data query cache */
update_data_query_cache($host_id, $snmp_query_id);
query_debug_timer_offset('data_query', 'Update data query cache complete');
/* update the poller cache */
update_poller_cache_from_query($host_id, $snmp_query_id);
query_debug_timer_offset('data_query', 'Update poller cache from query complete');
if ($config['poller_id'] == 1) {
/* perform any automation on reindex */
automation_execute_data_query($host_id, $snmp_query_id);
query_debug_timer_offset('data_query', 'Automation execute data query complete');
api_plugin_hook_function('run_data_query', array('host_id' => $host_id, 'snmp_query_id' => $snmp_query_id));
query_debug_timer_offset('data_query', 'Plugin hooks complete');
} else {
if ($config['connection'] == 'online') {
automation_execute_data_query($host_id, $snmp_query_id);
query_debug_timer_offset('data_query', 'Automation execute data query complete');
api_plugin_hook_function('run_data_query', array('host_id' => $host_id, 'snmp_query_id' => $snmp_query_id));
query_debug_timer_offset('data_query', 'Plugin hooks complete');
}
if (!isset($_SESSION)) {
$config['debug_log']['result'] = $result;
print json_encode($config['debug_log']);
} else {
$_SESSION['debug_log']['result'] = $result;
print json_encode($_SESSION['debug_log']);
kill_session_var('debug_log');
}
return true;
}
return (isset($result) ? $result : true);
}
function get_data_query_array($snmp_query_id) {
global $config, $data_query_xml_arrays;
include_once($config['library_path'] . '/xml.php');
/* load the array into memory if it hasn't been done yet */
if (!isset($data_query_xml_arrays[$snmp_query_id])) {
$xml_file_path = db_fetch_cell_prepared('SELECT xml_path FROM snmp_query WHERE id = ?', array($snmp_query_id));
$search = array('<path_cacti>', '<path_snmpget>', '<path_php_binary>');
$replace = array($config['base_path'], read_config_option('path_snmpget'), read_config_option('path_php_binary'));
$xml_file_path = str_replace($search, $replace, $xml_file_path);
if (!file_exists($xml_file_path)) {
query_debug_timer_offset('data_query', "Could not find data query XML file at '$xml_file_path'");
return false;
}
query_debug_timer_offset('data_query', "Found data query XML file at '$xml_file_path'");
$data = implode('',file($xml_file_path));
$xml_data = xml2array($data);
/* store the array value to the global array for future reference */
$data_query_xml_arrays[$snmp_query_id] = $xml_data;
}
return $data_query_xml_arrays[$snmp_query_id];
}
function query_script_host($host_id, $snmp_query_id) {
$script_queries = get_data_query_array($snmp_query_id);
/* invalid xml check */
if ((!is_array($script_queries)) || (sizeof($script_queries) == 0)) {
query_debug_timer_offset('data_query', 'Error parsing XML file into an array.');
return false;
}
query_debug_timer_offset('data_query', 'XML file parsed ok.');
/* are we talking to script server? */
if (isset($script_queries['script_server'])) {
$script_queries['script_path'] = "\"|path_php_binary|\" -q " . $script_queries['script_path'];
}
if (!verify_index_order($script_queries)) {
query_debug_timer_offset('data_query', 'Invalid field &lt;index_order&gt;' . $script_queries['index_order'] . '&lt;/index_order&gt;');
query_debug_timer_offset('data_query', 'Must contain &lt;direction&gt;input&lt;/direction&gt; or &lt;direction&gt;input-output&lt;/direction&gt; fields only');
return false;
}
/* provide data for arg_num_indexes, if given */
if (isset($script_queries['arg_num_indexes'])) {
$script_path = get_script_query_path((isset($script_queries['arg_prepend']) ? $script_queries['arg_prepend'] . ' ': '') . $script_queries['arg_num_indexes'], $script_queries['script_path'], $host_id);
/* fetch specified index at specified OID */
$script_num_index_array = exec_into_array($script_path);
query_debug_timer_offset('data_query', 'Executing script for num of indexes' . " '$script_path'");
foreach($script_num_index_array as $element) {
query_debug_timer_offset('data_query', 'Found number of indexes: ' . $element);
}
} else {
if (isset($script_queries['script_server'])) {
query_debug_timer_offset('data_query', "&lt;arg_num_indexes&gt; missing in XML file, 'Index Count Changed' not supported");
} else {
query_debug_timer_offset('data_query', "&lt;arg_num_indexes&gt; missing in XML file, 'Index Count Changed' emulated by counting arg_index entries");
}
}
/* provide data for index, mandatory */
$script_path = get_script_query_path((isset($script_queries['arg_prepend']) ? $script_queries['arg_prepend'] . ' ': '') . $script_queries['arg_index'], $script_queries['script_path'], $host_id);
/* fetch specified index */
$script_index_array = exec_into_array($script_path);
query_debug_timer_offset('data_query', 'Executing script for list of indexes' . " '$script_path' " . 'Index Count: ' . sizeof($script_index_array));
debug_log_insert_section_start('data_query', "Click to show data query output for 'index'");
foreach($script_index_array as $element) {
debug_log_insert('data_query', 'Found index: ' . $element);
}
debug_log_insert_section_end('data_query');
/* set an array to host all updates */
$output_array = array();
foreach ($script_queries['fields'] as $field_name => $field_array) {
if ($field_array['direction'] == 'input' || $field_array['direction'] == 'input-output') {
$rewrite_value = isset($field_array['rewrite_value']) ? $field_array['rewrite_value'] : NULL;
$script_path = get_script_query_path((isset($script_queries['arg_prepend']) ? $script_queries['arg_prepend'] . ' ': '') . $script_queries['arg_query'] . ' ' . $field_array['query_name'], $script_queries['script_path'], $host_id);
debug_log_insert_section_start('data_query', "Click to show data query output for field '$field_name'");
$script_data_array = exec_into_array($script_path);
debug_log_insert('data_query', "Executing script query '$script_path'");
foreach($script_data_array as $element) {
if (preg_match("/(.*?)" . preg_quote($script_queries['output_delimeter']) . "(.*)/", $element, $matches)) {
$script_index = $matches[1];
$field_value = $matches[2];
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $field_value, $script_index, '');
debug_log_insert('data_query', "Found item [$field_name='$field_value'] index: $script_index");
}
}
debug_log_insert_section_end('data_query');
}
}
if (sizeof($output_array)) {
data_query_update_host_cache_from_buffer($host_id, $snmp_query_id, $output_array);
}
return true;
}
function query_debug_timer_start() {
global $query_debug_timer, $query_debug_start;
/* record the start time */
$query_debug_timer = microtime(true);
$query_debug_start = $query_debug_timer;
}
function query_debug_timer_offset($section, $message) {
global $query_debug_timer, $query_debug_start;
if (empty($query_debug_timer)) {
query_debug_timer_start();
}
/* record the start time */
$cur_time = microtime(true);
$delta = $cur_time - $query_debug_timer;
$total = $cur_time - $query_debug_start;
$query_debug_timer = $cur_time;
debug_log_insert($section, 'Total: ' . round($total, 2) . ', Delta: ' . round($delta, 2) . ', ' . $message);
return $delta;
}
function query_debug_timer_stop($sction, $message) {
global $query_debug_timer, $query_debug_start;
/* record the start time */
$cur_time = microtime(true);
$delta = $cur_time - $query_debug_timer;
$total = $cur_time - $query_debug_start;
unset($query_debug_timer);
unset($query_debug_start);
debug_log_insert($section, 'Total: ' . round($total, 2) . ', Delta: ' . round($delta, 2) . ', ' . $message);
return $delta;
}
function query_snmp_host($host_id, $snmp_query_id) {
global $config, $data_query_rewrite_indexes_cache;
include_once($config['library_path'] . '/snmp.php');
$host = db_fetch_row_prepared('SELECT hostname, snmp_community, snmp_version,
snmp_username, snmp_password, snmp_auth_protocol, snmp_priv_passphrase,
snmp_priv_protocol, snmp_context, snmp_engine_id, snmp_port, snmp_timeout,
ping_retries, max_oids
FROM host
WHERE id = ?', array($host_id));
$snmp_queries = get_data_query_array($snmp_query_id);
if (!sizeof($host) || $host['hostname'] == '') {
query_debug_timer_offset('data_query', "Invalid host_id: $host_id");
return false;
}
/* invalid xml check */
if ((!is_array($snmp_queries)) || (sizeof($snmp_queries) == 0)) {
query_debug_timer_offset('data_query', 'Error parsing XML file into an array.');
return false;
}
query_debug_timer_offset('data_query', 'XML file parsed ok.');
if (!verify_index_order($snmp_queries)) {
query_debug_timer_offset('data_query', 'Invalid field &lt;index_order&gt;' . $snmp_queries['index_order'] . '&lt;/index_order&gt;');
query_debug_timer_offset('data_query', 'Must contain &lt;direction&gt;input&lt;/direction&gt; or &lt;direction&gt;input-output&lt;/direction&gt; fields only');
return false;
}
$session = cacti_snmp_session($host['hostname'], $host['snmp_community'],
$host['snmp_version'], $host['snmp_username'], $host['snmp_password'],
$host['snmp_auth_protocol'], $host['snmp_priv_passphrase'], $host['snmp_priv_protocol'],
$host['snmp_context'], $host['snmp_engine_id'], $host['snmp_port'],
$host['snmp_timeout'], $host['ping_retries'], $host['max_oids']);
if ($session === FALSE) {
debug_log_insert('data_query', 'Failed to load SNMP session.');
return false;
}
/* provide data for oid_num_indexes, if given */
if (isset($snmp_queries['oid_num_indexes'])) {
$snmp_num_indexes = cacti_snmp_session_get($session, $snmp_queries['oid_num_indexes']);
query_debug_timer_offset('data_query', "Executing SNMP get for num of indexes @ '" . $snmp_queries['oid_num_indexes'] . "' Index Count: " . $snmp_num_indexes);
} else {
query_debug_timer_offset('data_query', "&lt;oid_num_indexes&gt; missing in XML file, 'Index Count Changed' emulated by counting oid_index entries");
}
/* fetch specified index at specified OID */
$snmp_indexes = cacti_snmp_session_walk($session, $snmp_queries['oid_index']);
query_debug_timer_offset('data_query', "Executing SNMP walk for list of indexes @ '" . $snmp_queries['oid_index'] . "' Index Count: " . sizeof($snmp_indexes));
/* no data found; get out */
if (!sizeof($snmp_indexes)) {
query_debug_timer_offset('data_query', 'No SNMP data returned');
return false;
} else {
/* show list of indices found */
foreach($snmp_indexes as $oid => $value) {
query_debug_timer_offset('data_query', "Index found at OID: '" . $oid . "' value: '" . $value . "'");
}
}
/* the last octet of the oid is the index by default */
$index_parse_regexp = '/.*\.([0-9]+)$/';
/* parse the index if required */
if (isset($snmp_queries['oid_index_parse'])) {
$index_parse_regexp = '/' . str_replace('OID/REGEXP:', '', $snmp_queries['oid_index_parse']) . '/';
foreach($snmp_indexes as $oid => $value) {
$snmp_indexes[$oid] = preg_replace($index_parse_regexp, "\\1", $oid);
query_debug_timer_offset('data_query', "index_parse at OID: '" . $oid . "' results: '" . $snmp_indexes[$oid] . "'");
}
}
/* set an array to host all updates */
$output_array = array();
/* invalidate rewrite_index cache */
$data_query_rewrite_indexes_cache = array();
$fields_processed = array();
rewrite_snmp_enum_value(NULL);
foreach ($snmp_queries['fields'] as $field_name => $field_array) {
if ($field_array['source'] != 'index' && ($field_array['direction'] == 'input' || $field_array['direction'] == 'input-output') && $field_array['method'] != 'get' &&
(isset($field_array['rewrite_index']) || isset($field_array['oid_suffix']))) {
$field_array['method'] = 'get';
debug_log_insert('data_query', "Fixing wrong 'method' field for '$field_name' since 'rewrite_index' or 'oid_suffix' is defined");
}
$rewrite_value = isset($field_array['rewrite_value']) ? $field_array['rewrite_value'] : NULL;
if ((!isset($field_array['oid'])) && ($field_array['source'] == 'index')) {
foreach($snmp_indexes as $oid => $value) {
query_debug_timer_offset('data_query', "Inserting index data for field '" . $field_name . "' [value='" . $value . "']");
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $value, $value, '');
}
} elseif (($field_array['method'] == 'get') && ($field_array['direction'] == 'input' || $field_array['direction'] == 'input-output')) {
query_debug_timer_offset('data_query', "Located input field '$field_name' [get]");
if ($field_array['source'] == 'value' && !isset($field_array['rewrite_index'])) {
$oid_rewrite_pattern = null;
$oid_rewrite_replacement = null;
if (isset($field_array['oid_rewrite_pattern']) && isset($field_array['oid_rewrite_replacement'])) {
$oid_rewrite_pattern = '/' . str_replace('OID/REGEXP:', '', $field_array['oid_rewrite_pattern']) . '/';
$oid_rewrite_replacement = $field_array['oid_rewrite_replacement'];
query_debug_timer_offset('data_query', "Found OID rewrite rule: 's/$oid_rewrite_pattern/$oid_rewrite_replacement/'");
}
foreach($snmp_indexes as $oid => $index) {
$oid = $field_array['oid'] . '.' . $index;
$oid .= isset($field_array['oid_suffix']) ? ('.' . $field_array['oid_suffix']) : '';
/* rewrite the oid if required */
if (isset($oid_rewrite_pattern)) {
$orig_oid = $oid;
$oid = preg_replace($oid_rewrite_pattern, $oid_rewrite_replacement, $oid);
query_debug_timer_offset('data_query', "oid_rewrite at OID: '" . $orig_oid . "' new OID: '" . $oid ."'");
}
if (isset($field_array['output_format'])) {
if ($field_array['output_format'] == 'hex') {
$value_output_format = SNMP_STRING_OUTPUT_HEX;
} elseif ($field_array['output_format'] == 'ascii') {
$value_output_format = SNMP_STRING_OUTPUT_ASCII;
} else {
$value_output_format = SNMP_STRING_OUTPUT_GUESS;
}
$value = cacti_snmp_get($host['hostname'], $host['snmp_community'], $oid, $host['snmp_version'],
$host['snmp_username'], $host['snmp_password'], $host['snmp_auth_protocol'],
$host['snmp_priv_passphrase'], $host['snmp_priv_protocol'], $host['snmp_context'],
$host['snmp_port'], $host['snmp_timeout'], SNMP_POLLER, $host['snmp_engine_id'],
$value_output_format);
}else{
$value = cacti_snmp_session_get($session, $oid);
}
query_debug_timer_offset('data_query', "Executing SNMP get for data @ '$oid' [value='$value']");
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $value, $index, $oid);
}
} elseif (isset($field_array['rewrite_index'])) {
$rewritten_indexes = array();
if (isset($field_array['rewrite_index'])){
$rewritten_indexes = data_query_rewrite_indexes($errmsg, $host_id, $snmp_query_id, $field_array['rewrite_index'], $snmp_indexes, $fields_processed);
if (sizeof($errmsg)){
foreach($errmsg as $message){
debug_log_insert('data_query', "Field '$field_name'" . $message);
}
}
}
$values = array();
foreach ($snmp_indexes as $index_oid => $index) {
$oid = $field_array['oid'];
if (isset($field_array['rewrite_index'])){
if (isset($rewritten_indexes[$index])){
$oid_suffix = $rewritten_indexes[$index];
} else {
// we failed to build rewritten index. warnings are sent already, just skip this index silently
continue;
}
$oid .= '.' . $oid_suffix;
} else {
$oid .= '.' . $index;
}
$oid .= isset($field_array['oid_suffix']) ? ('.' . $field_array['oid_suffix']) : '';
$value = NULL;
if (substr($field_array['source'], 0, 11) == 'OID/REGEXP:'){
$value = preg_replace('/' . str_replace('OID/REGEXP:', '', $field_array['source']) . '/', "\\1", $oid);
}
$values[] = array('value' => $value, 'index' => $index, 'oid' => $oid);
}
if (sizeof($values) > 0 && (substr($field_array['source'], 0, 13) == 'VALUE/REGEXP:' || $field_array['source'] == 'value')) {
$oids = array();
foreach ($values as $key => $value){
$oids[] = $value['oid'];
}
debug_log_insert('data_query', 'Executing SNMP get for ' . count($oids) . ' oids');
if (isset($field_array['output_format'])) {
if ($field_array['output_format'] == 'hex') {
$value_output_format = SNMP_STRING_OUTPUT_HEX;
} elseif ($field_array['output_format'] == 'ascii') {
$value_output_format = SNMP_STRING_OUTPUT_ASCII;
} else {
$value_output_format = SNMP_STRING_OUTPUT_GUESS;
}
foreach($oids as $oid) {
$results[$oid] = cacti_snmp_get($host['hostname'], $host['snmp_community'], $oid, $host['snmp_version'],
$host['snmp_username'], $host['snmp_password'], $host['snmp_auth_protocol'],
$host['snmp_priv_passphrase'], $host['snmp_priv_protocol'], $host['snmp_context'],
$host['snmp_port'], $host['snmp_timeout'], SNMP_POLLER, $host['snmp_engine_id'],
$value_output_format);
}
}else{
$results = cacti_snmp_session_walk($session, $oids);
}
foreach (array_keys($values) as $key){
if (isset($results[$values[$key]['oid']])) {
$values[$key]['value'] = $results[$values[$key]['oid']];
}
}
foreach ($values as $key => $value) {
if (substr($field_array['source'], 0, 13) == 'VALUE/REGEXP:') {
$values[$key]['value'] = preg_replace('/' . str_replace('VALUE/REGEXP:', '', $field_array['source']) . '/', "\\1", $values[$key]['value']);
}
}
}
foreach ($values as $item) {
debug_log_insert('data_query', "Got SNMP get result for data @ '" . $item['oid'] . "' [value='" . $item['value'] . "'] (index: " . $item['index'] . ')');
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $item['value'], $item['index'], $item['oid']);
}
$values = NULL;
} elseif (substr($field_array['source'], 0, 13) == 'VALUE/REGEXP:') {
foreach($snmp_indexes as $oid => $index) {
$oid = $field_array['oid'] . '.' . $index;
$oid .= isset($field_array['oid_suffix']) ? ('.' . $field_array['oid_suffix']) : '';
if (isset($field_array['output_format'])) {
if ($field_array['output_format'] == 'hex') {
$value_output_format = SNMP_STRING_OUTPUT_HEX;
} elseif ($field_array['output_format'] == 'ascii') {
$value_output_format = SNMP_STRING_OUTPUT_ASCII;
} else {
$value_output_format = SNMP_STRING_OUTPUT_GUESS;
}
$value = cacti_snmp_get($host['hostname'], $host['snmp_community'], $oid, $host['snmp_version'],
$host['snmp_username'], $host['snmp_password'], $host['snmp_auth_protocol'],
$host['snmp_priv_passphrase'], $host['snmp_priv_protocol'], $host['snmp_context'],
$host['snmp_port'], $host['snmp_timeout'], SNMP_POLLER, $host['snmp_engine_id'],
$value_output_format);
}else{
$value = cacti_snmp_session_get($session, $oid);
}
$value = preg_replace('/' . str_replace('VALUE/REGEXP:', '', $field_array['source']) . '/', "\\1", $value);
query_debug_timer_offset('data_query', "Executing SNMP get for data @ '$oid' [value='$value']");
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $value, $index, $oid);
}
}
} elseif ($field_array['method'] == 'walk' && ($field_array['direction'] == 'input' || $field_array['direction'] == 'input-output')) {
debug_log_insert_section_start('data_query', "Click to show data query output for field '$field_name'");
query_debug_timer_offset('data_query', "Located input field '$field_name' [walk]");
if (isset($field_array['output_format'])) {
if ($field_array['output_format'] == 'hex') {
$value_output_format = SNMP_STRING_OUTPUT_HEX;
} elseif ($field_array['output_format'] == 'ascii') {
$value_output_format = SNMP_STRING_OUTPUT_ASCII;
} else {
$value_output_format = SNMP_STRING_OUTPUT_GUESS;
}
$snmp_data = cacti_snmp_walk($host['hostname'], $host['snmp_community'], $field_array['oid'], $host['snmp_version'],
$host['snmp_username'], $host['snmp_password'], $host['snmp_auth_protocol'],
$host['snmp_priv_passphrase'], $host['snmp_priv_protocol'], $host['snmp_context'],
$host['snmp_port'], $host['snmp_timeout'], $host['ping_retries'], $host['max_oids'], SNMP_POLLER,
$host['snmp_engine_id'], $value_output_format);
}else{
$snmp_data = cacti_snmp_session_walk($session, $field_array['oid']);
}
query_debug_timer_offset('data_query', "Executing SNMP walk for data @ '" . $field_array['oid'] . "'");
if (preg_match('/^VALUE\/TABLE:(.*)/',$field_array['source'],$matches)) {
preg_match_all('/([^:]+):([^:]+)/',$matches[1],$match_temp);
$const_table = array_combine($match_temp[1],$match_temp[2]);
}
if (preg_match('/^value/i',$field_array['source'])) {
foreach ($snmp_data as $oid => $value) {
$index_regex = (isset($field_array['oid_index_parse']) ? $field_array['oid_index_parse'] : $index_parse_regexp) . (isset($field_array['oid_suffix']) ? ('.' . $field_array['oid_suffix']) : '');
if (!preg_match($index_regex, $oid)) {
continue;
}
$snmp_index = preg_replace($index_regex,"\\1", $oid);
$oid = $field_array['oid'] . ".$snmp_index" . (isset($field_array['oid_suffix']) ? ('.' . $field_array['oid_suffix']) : '');
if ($field_name == 'ifOperStatus') {
switch(true) {
case preg_match('/^(down|2)/i',$value):
$value = 'Down';
break;
case preg_match('/^(up|1)/i',$value):
$value = 'Up';
break;
case preg_match('/^(notpresent|6)/i',$value):
$value = 'notPresent';
break;
default:
$value = 'Testing';
}
$mode = 'value';
} elseif (preg_match('/VALUE\/REGEXP:(.*)/',$field_array['source'],$matches)) {
$modified_value = preg_replace('/' . $matches[1] . '/', "\\1", $value);
$mode = 'regexp value parse';
} elseif (preg_match('/^VALUE\/TEST:(.*):(.*):(.*)/',$field_array['source'],$matches)) {
$modified_value = (strcmp($matches[1],$value) !== 0)?$matches[3]:$matches[2];
$mode = 'test value parse';
} elseif (preg_match('/^VALUE\/TABLE:(.*)/',$field_array['source'])) {
$modified_value = (array_key_exists($value,$const_table))?$const_table[$value]:'N/A';
$mode = 'table value parse';
} elseif (preg_match('/^VALUE\/HEX2IP:(\d+):(\d+)/',$field_array['source'],$matches)) {
#$modified_value = substr(str_replace(':','',$value),$matches[1],$matches[2]);
$modified_value = substr(preg_replace('/(:|HEX-)/','',$value),$matches[1],$matches[2]);
$modified_value = implode('.',unpack ('C*',pack('H*', $modified_value)));
$mode = 'hex2ip value parse';
} else {
$mode = 'value';
}
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, isset($modified_value)?$modified_value:$value , $snmp_index, $oid);
debug_log_insert('data_query', sprintf("Found item [%s='%s'] index: %s [from %s]",$field_name,isset($modified_value)?"$modified_value ($value)":$value,$snmp_index,$mode));
unset($modified_value);
}
} elseif (substr($field_array['source'], 0, 11) == 'OID/REGEXP:') {
foreach($snmp_data as $oid => $value) {
$parse_value = preg_replace('/' . str_replace('OID/REGEXP:', '', $field_array['source']) . '/', "\\1", $oid);
if (isset($snmp_queries['oid_index_parse'])) {
$snmp_index = preg_replace($index_parse_regexp, "\\1", $oid);
} elseif ((isset($value)) && ($value != '')) {
$snmp_index = $value;
}
/* correct bogus index value */
/* found in some devices such as an EMC Cellera */
if ($snmp_index == 0) {
$snmp_index = 1;
}
$oid = $field_array['oid'] . '.' . $parse_value;
/* rewrite octet strings */
if (preg_match('/^\d{1,3}(\.\d{1,3}){2,}$/', $parse_value)) {
$octets = explode('.', $parse_value);
$size = array_shift($octets);
if (count($octets) == $size) {
$decoded = '';
$isascii = true;
foreach($octets as $octet) {
if (($octet <= 31) || ($octet >= 127)) {
$isascii = false;
} else {
$decoded .= chr($octet);
}
}
if ($isascii) {
query_debug_timer_offset('data_query', "Found OCTET STRING '$parse_value' decoded value: '$decoded'");
$parse_value = $decoded;
}
}
}
debug_log_insert('data_query', "Found item [$field_name='$parse_value'] index: $snmp_index [from regexp oid parse]");
$output_array[] = data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $parse_value, $snmp_index, $oid);
}
}
debug_log_insert_section_end('data_query');
}
$fields_processed[] = $field_name;
}
$session->close();
if (sizeof($output_array)) {
data_query_update_host_cache_from_buffer($host_id, $snmp_query_id, $output_array);
}
return true;
}
function data_query_format_record($host_id, $snmp_query_id, $field_name, $rewrite_value, $value, $snmp_index, $oid) {
global $data_query_rewrite_indexes_cache;
if ($rewrite_value !== NULL) {
$value = rewrite_snmp_enum_value($field_name, $value, $rewrite_value);
}
$hash = "$host_id@$snmp_query_id@$field_name";
if (!isset($data_query_rewrite_indexes_cache[$hash])) {
$data_query_rewrite_indexes_cache[$hash] = array();
}
$data_query_rewrite_indexes_cache[$hash][$snmp_index] = $value;
if (ctype_print($value) === FALSE) {
$value = bin2hex($value);
}
return "($host_id, $snmp_query_id, " . db_qstr($field_name) . ', ' . db_qstr($value) . ', ' . db_qstr($snmp_index) . ', ' . db_qstr($oid) . ', 1)';
}
function data_query_update_host_cache_from_buffer($host_id, $snmp_query_id, &$output_array) {
/* set all fields present value to 0, to mark the outliers when we are all done */
db_execute_prepared('UPDATE host_snmp_cache SET present=0 WHERE host_id = ? AND snmp_query_id = ?', array($host_id, $snmp_query_id));
/* setup the database call */
$sql_prefix = 'INSERT INTO host_snmp_cache (host_id, snmp_query_id, field_name, field_value, snmp_index, oid, present) VALUES';
$sql_suffix = ' ON DUPLICATE KEY UPDATE field_value=VALUES(field_value), oid=VALUES(oid), present=VALUES(present)';
/* use a reasonable insert buffer, the default is 1MByte */
$max_packet = 256000;
/* setup somme defaults */
$overhead = strlen($sql_prefix) + strlen($sql_suffix);
$buf_len = 0;
$buf_count = 0;
$buffer = '';
foreach($output_array as $record) {
if ($buf_count == 0) {
$delim = ' ';
} else {
$delim = ', ';
}
$buffer .= $delim . $record;
$buf_len += strlen($record);
if (($overhead + $buf_len) > ($max_packet - 1024)) {
db_execute($sql_prefix . $buffer . $sql_suffix);
$buffer = '';
$buf_len = 0;
$buf_count = 0;
} else {
$buf_count++;
}
}
if ($buf_count > 0) {
db_execute($sql_prefix . $buffer . $sql_suffix);
}
/* remove stale records from the host cache */
db_execute_prepared("DELETE FROM host_snmp_cache WHERE host_id = ? AND snmp_query_id = ? AND present='0'", array($host_id, $snmp_query_id));
}
/* data_query_rewrite_indexes - returns array of rewritten indexes
@arg $errmsg array that will contain warnings if any
@arg $host_id
@arg $snmp_query_id
@arg $rewrite_index - value of <rewrite_index> from data query XML
@arg $snmp_indexes - array of snmp indexes as it used in query_snmp_host() or single index
@arg $fields_processed - array of field names that are already processed in query_snmp_host(),
refusing non-processed (e.g. stale) fields to be used as index rewrite source
@returns - (array) of original snmp indexes associated with rewritten ones
*/
function data_query_rewrite_indexes(&$errmsg, $host_id, $snmp_query_id, $rewrite_index, $snmp_indexes, $fields_processed = FALSE){
global $data_query_rewrite_indexes_cache;
$errmsg = array();
$oid_items = explode('.', $rewrite_index);
$chain_indexes = array();
if (!is_array($data_query_rewrite_indexes_cache))
$data_query_rewrite_indexes_cache = array();
if (sizeof($oid_items)){
foreach($oid_items as $item){
$matches = array();
if (preg_match('/^\|query_([^|]+)\|$/', $item, $matches)){
$iv = $matches[1];
if (is_array($fields_processed) && !in_array($iv, $fields_processed)){
$errmsg[] = "rewrite_index='$rewrite_index': '$iv' is not processed yet, could not use it as index source";
continue;
}
if (!isset($chain_indexes[$iv])) {
$hash = "$host_id@$snmp_query_id@$iv";
if (!isset($data_query_rewrite_indexes_cache[$hash])) {
$data_query_rewrite_indexes_cache[$hash] = array();
$iv = db_qstr($iv);
$field_values = db_fetch_assoc_prepared("SELECT snmp_index, field_value
FROM host_snmp_cache
WHERE host_id = ?
AND snmp_query_id = ?
AND field_name = ?",
array($host_id, $snmp_query_id, $iv));
if (sizeof($field_values)){
foreach($field_values as $item){
$data_query_rewrite_indexes_cache[$hash][$item['snmp_index']] = $item['field_value'];
}
} else {
$errmsg[] = " field '$iv' was not found either in cache nor in DB";
}
}
$chain_indexes[$iv] = $data_query_rewrite_indexes_cache[$hash];
}
}
}
}
$out = array();
$numeric_output = FALSE;
if (!is_array($snmp_indexes)){
$snmp_indexes = array($snmp_indexes);
$numeric_output = TRUE;
}
foreach ($snmp_indexes as $oid => $num_index) {
$index = $rewrite_index;
foreach($chain_indexes as $key => $values){
if (isset($values[$num_index]) && preg_match('/^[0-9.]+$/', $values[$num_index])){
$index = str_replace("|query_$key|", trim($values[$num_index]), $index);
} else {
$errmsg[] = "@'" . $num_index . "': could not load value of '$key'";
}
}
$index = str_replace('|index|', trim($num_index), $index);
if (!preg_match('/^[0-9.]*$/', $index)){
$errmsg[] = "@'" . $num_index ."': some portions of rewrite_index field were not parsed: '$index'";
continue;
}
if ($numeric_output){
return $index;
}
$out[$num_index] = $index;
}
if ($numeric_output){
return NULL;
}
return $out;
}
/* rewrite_snmp_enum_value - returns rewritten $value based on rewrite map
@arg $field_name - name of field being rewritten, used for cache purpuses
@arg $value - value to be translated
@arg $map - translation map in serialize()/array form
@returns - rewritten value if possible, original one otherwise*/
function rewrite_snmp_enum_value($field_name, $value=NULL, $map=NULL){
static $mapcache = array();
if ($field_name === NULL){
$mapcache = array();
return NULL;
}
if (is_array($map)){ # produced from XML, needs to be reformatted
$newmap = array();
foreach($map as $index => $item){
if (!isset($item['match']) || !isset($item['replace'])){
debug_log_insert('data_query', "Bogus rewrite_value item found, index='$index'");
continue;
}
$newmap[$item['match']] = $item['replace'];
}
$map = $newmap;
} else {
$map = unserialize($map);
}
if ($map === FALSE || !is_array($map)){
debug_log_insert('data_query', 'Could not parse translation map (rewrite_value)');
return $value;
}
if (!isset($mapcache[$field_name])){
$mapcache[$field_name] = array();
foreach($map as $src => $dst){
if (preg_match('/^REGEXP(NC)?:(.*)$/', $src, $matches)){
if ($matches[1] == 'NC') {
$src = '/' . str_replace('/', '\/', $matches[2]) . '/i';
} else {
$src = '/' . str_replace('/', '\/', $matches[1]) . '/';
}
} else {
$src = '/^' . str_replace('/^', '\/', $src) . '$/';
}
$mapcache[$field_name][$src] = $dst;
}
}
foreach($mapcache[$field_name] as $src => $dst){
if (preg_match($src, $value)){
$nvalue = preg_replace($src, $dst, $value);
debug_log_insert('data_query', "rewrite_value: '$value' => '$nvalue'");
$value = $nvalue;
break;
}
}
return $value;
}
/* data_query_index - returns an array containing the data query ID and index value given
a data query index type/value combination and a host ID
@arg $index_type - the name of the index to match
@arg $index_value - the value of the index to match
@arg $host_id - (int) the host ID to match
@arg $data_query_id - (int) the data query ID to match
@returns - (array) the data query ID and index that matches the three arguments */
function data_query_index($index_type, $index_value, $host_id, $data_query_id) {
return db_fetch_cell_prepared("SELECT
host_snmp_cache.snmp_index
FROM host_snmp_cache
WHERE host_snmp_cache.field_name = ?
AND host_snmp_cache.field_value = ?
AND host_snmp_cache.host_id = ?
AND host_snmp_cache.snmp_query_id = ?",
array($index_type, $index_value, $host_id, $data_query_id));
}
/* data_query_field_list - returns an array containing data query information for a given data source
@arg $data_template_data_id - the ID of the data source to retrieve information for
@returns - (array) an array that looks like:
Array
(
[index_type] => ifIndex
[index_value] => 3
[output_type] => 13
) */
function data_query_field_list($data_template_data_id) {
if (!is_numeric($data_template_data_id)) {
return 0;
}
$field = db_fetch_assoc_prepared("SELECT dif.type_code, did.value
FROM data_input_fields AS dif
INNER JOIN data_input_data AS did
ON dif.id=did.data_input_field_id
WHERE did.data_template_data_id = ?
AND dif.type_code IN ('index_type','index_value','output_type')", array($data_template_data_id));
$field = array_rekey($field, 'type_code', 'value');
if ((!isset($field['index_type'])) || (!isset($field['index_value'])) || (!isset($field['output_type']))) {
return 0;
} else {
return $field;
}
}
/* encode_data_query_index - encodes a data query index value so that it can be included
inside of a form
@arg $index - the index name to encode
@returns - the encoded data query index */
function encode_data_query_index($index) {
return md5($index);
}
/* decode_data_query_index - decodes a data query index value so that it can be read from
a form
@arg $encoded_index - the index that was encoded with encode_data_query_index()
@arg $data_query_id - the id of the data query that this index belongs to
@arg $encoded_index - the id of the host that this index belongs to
@returns - the decoded data query index */
function decode_data_query_index($encoded_index, $data_query_id, $host_id) {
/* yes, i know MySQL has a MD5() function that would make this a bit quicker. however i would like to
keep things abstracted for now so Cacti works with ADODB fully when i get around to porting my db calls */
$indexes = db_fetch_assoc_prepared('SELECT snmp_index
FROM host_snmp_cache
WHERE host_id = ?
AND snmp_query_id = ?
GROUP BY snmp_index', array($host_id, $data_query_id));
if (sizeof($indexes)) {
foreach ($indexes as $index) {
if (encode_data_query_index($index['snmp_index']) == $encoded_index) {
return $index['snmp_index'];
}
}
}
}
/* update_data_query_cache - updates the local data query cache for each graph AND data
source tied to this host/data query
@arg $host_id - the id of the host to refresh
@arg $data_query_id - the id of the data query to refresh */
function update_data_query_cache($host_id, $data_query_id) {
$graphs = db_fetch_assoc_prepared('SELECT *
FROM graph_local
WHERE host_id = ?
AND snmp_query_id = ?',
array($host_id, $data_query_id));
if (sizeof($graphs)) {
foreach ($graphs as $graph) {
update_graph_data_query_cache($graph['id'], $host_id, $data_query_id, $graph['snmp_index']);
}
}
query_debug_timer_offset('data_query', 'Update graph data query cache complete');
$data_sources = db_fetch_assoc_prepared('SELECT *
FROM data_local
WHERE host_id = ?
AND snmp_query_id = ?',
array($host_id, $data_query_id));
if (sizeof($data_sources) > 0) {
foreach ($data_sources as $data_source) {
$changed = update_data_source_data_query_cache($data_source['id'], $host_id, $data_query_id, $data_source['snmp_index']);
if ($changed) {
update_poller_cache($data_source, true);
}
}
}
query_debug_timer_offset('data_query', 'Update data source data query cache complete');
}
/* update_graph_data_query_cache - updates the local data query cache for a particular
graph
@arg $local_graph_id - the id of the graph to update the data query cache for */
function update_graph_data_query_cache($local_graph_id, $host_id = '', $data_query_id = '', $data_query_index = '') {
global $data_query_id_cache;
if ((!is_array($data_query_id_cache)) || sizeof($data_query_id_cache) == 0) {
$data_query_id_cache = array_rekey(db_fetch_assoc('SELECT id, snmp_query_id FROM snmp_query_graph'), 'id', 'snmp_query_id');
}
if (empty($host_id)) {
$host_id = db_fetch_cell_prepared('SELECT host_id
FROM graph_local
WHERE id = ?',
array($local_graph_id));
}
$field = data_query_field_list(db_fetch_cell_prepared('SELECT
data_template_data.id
FROM (graph_templates_item,data_template_rrd,data_template_data)
WHERE graph_templates_item.task_item_id=data_template_rrd.id
AND data_template_rrd.local_data_id=data_template_data.local_data_id
AND graph_templates_item.local_graph_id = ?
LIMIT 1',
array($local_graph_id)));
if (empty($field)) { return; }
if (empty($data_query_id)) {
$data_query_id = db_fetch_cell_prepared('SELECT snmp_query_id
FROM snmp_query_graph
WHERE id = ?',
array($field['output_type']));
}
$index = data_query_index($field['index_type'], $field['index_value'], $host_id, $data_query_id);
if ($data_query_id != 0 && $index != '' && $index != $data_query_index) {
db_execute_prepared('UPDATE graph_local
SET snmp_query_id = ?,
snmp_index = ?
WHERE id = ?',
array($data_query_id, $index, $local_graph_id));
/* update graph title cache */
update_graph_title_cache($local_graph_id);
}
}
/* update_data_source_data_query_cache - updates the local data query cache for a particular
data source
@arg $local_data_id - the id of the data source to update the data query cache for */
function update_data_source_data_query_cache($local_data_id, $host_id = '', $data_query_id = '', $data_query_index = '') {
global $data_query_id_cache;
if ((!is_array($data_query_id_cache)) || sizeof($data_query_id_cache) == 0) {
$data_query_id_cache = array_rekey(db_fetch_assoc('SELECT id, snmp_query_id FROM snmp_query_graph'), 'id', 'snmp_query_id');
}
if (empty($host_id)) {
$host_id = db_fetch_cell_prepared('SELECT host_id FROM data_local WHERE id = ?', array($local_data_id));
}
$field = data_query_field_list(db_fetch_cell_prepared('SELECT
data_template_data.id
FROM data_template_data
WHERE data_template_data.local_data_id = ?', array($local_data_id)));
if (empty($field)) {
return;
}
if (empty($data_query_id)) {
$data_query_id = db_fetch_cell_prepared('SELECT snmp_query_id
FROM snmp_query_graph
WHERE id = ?',
array($field['output_type']));
}
$index = data_query_index($field['index_type'], $field['index_value'], $host_id, $data_query_id);
if ($data_query_id != 0 && $index != '' && $index != $data_query_index) {
db_execute_prepared('UPDATE data_local
SET snmp_query_id = ?,
snmp_index = ?
WHERE id = ?',
array($data_query_id, $index, $local_data_id));
/* update data source title cache */
update_data_source_title_cache($local_data_id);
return true;
}
return false;
}
/* get_formatted_data_query_indexes - obtains a list of indexes for a host/data query that
is sorted by the chosen index field and formatted using the data query index title
format
@arg $host_id - the id of the host which contains the data query
@arg $data_query_id - the id of the data query to retrieve a list of indexes for
@returns - an array formatted like the following:
$arr[snmp_index] = 'formatted data query index string' */
function get_formatted_data_query_indexes($host_id, $data_query_id) {
global $config;
include_once($config['library_path'] . '/sort.php');
if (empty($data_query_id)) {
return array('' => 'Unknown Index');
}
/* from the xml; cached in 'host_snmp_query' */
$sort_cache = db_fetch_row_prepared('SELECT sort_field, title_format
FROM host_snmp_query
WHERE host_id = ?
AND snmp_query_id = ?',
array($host_id, $data_query_id));
/* in case no unique index is available, fallback to first field in XML */
if ($sort_cache['sort_field'] == ''){
$snmp_queries = get_data_query_array($data_query_id);
if (isset($snmp_queries['index_order'])){
$i = explode(':', $snmp_queries['index_order']);
if (sizeof($i) > 0){
$sort_cache['sort_field'] = array_shift($i);
}
}
}
/* get a list of data query indexes AND the field value that we are supposed
to sort */
$sort_field_data = array_rekey(db_fetch_assoc_prepared('SELECT gl.snmp_index, hsc.field_value
FROM graph_local AS gl
INNER JOIN host_snmp_cache AS hsc
ON gl.host_id=hsc.host_id
AND gl.snmp_query_id=hsc.snmp_query_id
AND gl.snmp_index=hsc.snmp_index
WHERE gl.snmp_query_id = ?
AND gl.host_id = ?
AND hsc.field_name = ?
GROUP BY gl.snmp_index',
array($data_query_id, $host_id, $sort_cache['sort_field'])), 'snmp_index', 'field_value');
/* sort the data using the 'data query index' sort algorithm */
uasort($sort_field_data, 'usort_data_query_index');
$sorted_results = array();
foreach ($sort_field_data as $snmp_index => $sort_field_value) {
$sorted_results[$snmp_index] = substitute_snmp_query_data($sort_cache['title_format'], $host_id, $data_query_id, $snmp_index);
}
return $sorted_results;
}
/* get_formatted_data_query_index - obtains a single index for a host/data query/data query
index that is formatted using the data query index title format
@arg $host_id - the id of the host which contains the data query
@arg $data_query_id - the id of the data query which contains the data query index
@arg $data_query_index - the index to retrieve the formatted name for
@returns - a string containing the formatted name for the given data query index */
function get_formatted_data_query_index($host_id, $data_query_id, $data_query_index) {
/* from the xml; cached in 'host_snmp_query' */
$sort_cache = db_fetch_row_prepared('SELECT sort_field, title_format
FROM host_snmp_query
WHERE host_id = ?
AND snmp_query_id = ?',
array($host_id, $data_query_id));
return substitute_snmp_query_data($sort_cache['title_format'], $host_id, $data_query_id, $data_query_index);
}
/* get_ordered_index_type_list - builds an ordered list of data query index types that are
valid given a list of data query indexes that will be checked against the data query
cache
@arg $host_id - the id of the host which contains the data query
@arg $data_query_id - the id of the data query to build the type list from
@arg $data_query_index_array - an array containing each data query index to use when checking
each data query type for validity. a valid data query type will contain no empty or duplicate
values for each row in the cache that matches one of the $data_query_index_array
@returns - an array of data query types either ordered or unordered depending on whether
the xml file has a manual ordering preference specified */
function get_ordered_index_type_list($host_id, $data_query_id, $data_query_index_array = array()) {
$raw_xml = get_data_query_array($data_query_id);
/* invalid xml check */
if ((!is_array($raw_xml)) || (sizeof($raw_xml) == 0)) {
return array();
}
$xml_outputs = array();
/* create an SQL string that contains each index in this snmp_index_id */
$sql_or = array_to_sql_or($data_query_index_array, 'snmp_index');
/* check for nonunique query parameter, set value */
if (isset($raw_xml['index_type'])) {
if ($raw_xml['index_type'] == 'nonunique') {
$nonunique = 1;
} else {
$nonunique = 0;
}
} else {
$nonunique = 0;
}
/* list each of the input fields for this snmp query */
foreach ($raw_xml['fields'] as $field_name => $field_array) {
if ($field_array['direction'] == 'input' || $field_array['direction'] == 'input-output') {
/* create a list of all values for this index */
if (sizeof($data_query_index_array) == 0) {
$field_values = db_fetch_assoc_prepared('SELECT field_value
FROM host_snmp_cache
WHERE host_id = ?
AND snmp_query_id = ?
AND field_name = ?',
array($host_id, $data_query_id, $field_name));
} else {
$field_values = db_fetch_assoc_prepared("SELECT field_value
FROM host_snmp_cache
WHERE host_id = ?
AND snmp_query_id = ?
AND field_name = ?
AND $sql_or",
array($host_id, $data_query_id, $field_name));
}
/* aggregate the above list so there is no duplicates */
$aggregate_field_values = array_rekey($field_values, 'field_value', 'field_value');
/* fields that contain duplicate or empty values are not suitable to index off of */
if (!((sizeof($aggregate_field_values) < sizeof($field_values)) || (in_array('', $aggregate_field_values) == true) || (sizeof($aggregate_field_values) == 0)) || ($nonunique)) {
array_push($xml_outputs, $field_name);
}
}
}
$return_array = array();
/* the xml file contains an ordered list of 'indexable' fields */
if (isset($raw_xml['index_order'])) {
$index_order_array = explode(':', $raw_xml['index_order']);
foreach($index_order_array as $element) {
if (in_array($element, $xml_outputs)) {
$return_array[] = $element;
}
}
/* the xml file does not contain a field list, ignore the order */
} else {
foreach($xml_outputs as $output) {
$return_array[] = $output;
}
}
return $return_array;
}
/* update_data_query_sort_cache - updates the sort cache for a particular host/data query
combination. this works by fetching a list of valid data query index types and choosing
the first one in the list. the user can optionally override how the cache is updated
in the data query xml file
@arg $host_id - the id of the host which contains the data query
@arg $data_query_id - the id of the data query update the sort cache for */
function update_data_query_sort_cache($host_id, $data_query_id) {
$raw_xml = get_data_query_array($data_query_id);
/* get a list of valid data query types */
$valid_index_types = get_ordered_index_type_list($host_id, $data_query_id);
/* something is probably wrong with the data query */
if (sizeof($valid_index_types) == 0) {
$sort_field = '';
} else {
/* grab the first field off the list */
$sort_field = $valid_index_types[0];
}
/* substitute variables */
if (isset($raw_xml['index_title_format'])) {
$title_format = str_replace('|chosen_order_field|', "|query_$sort_field|", $raw_xml['index_title_format']);
} else {
$title_format = "|query_$sort_field|";
}
/* update the cache */
/* TODO: if both $sort field and $title_format are empty, this yields funny results */
db_execute_prepared('UPDATE host_snmp_query
SET sort_field = ?,
title_format = ?
WHERE host_id = ?
AND snmp_query_id = ?',
array($sort_field, $title_format, $host_id, $data_query_id));
}
/* update_data_query_sort_cache_by_host - updates the sort cache for all data queries associated
with a particular host. see update_data_query_sort_cache() for details about updating the cache
@arg $host_id - the id of the host to update the cache for */
function update_data_query_sort_cache_by_host($host_id) {
$data_queries = db_fetch_assoc_prepared('SELECT snmp_query_id
FROM host_snmp_query
WHERE host_id = ?',
array($host_id));
if (sizeof($data_queries) > 0) {
foreach ($data_queries as $data_query) {
update_data_query_sort_cache($host_id, $data_query['snmp_query_id']);
}
}
}
/* get_best_data_query_index_type - returns the best available data query index type using the
sort cache
@arg $host_id - the id of the host which contains the data query
@arg $data_query_id - the id of the data query to fetch the best data query index type for
@returns - a string containing containing best data query index type. this will be one of the
valid input field names as specified in the data query xml file */
function get_best_data_query_index_type($host_id, $data_query_id) {
return db_fetch_cell_prepared('SELECT sort_field
FROM host_snmp_query
WHERE host_id = ?
AND snmp_query_id = ?',
array($host_id, $data_query_id));
}
/* get_script_query_path - builds the complete script query executable path
@arg $args - the variable that contains any arguments to be appended to the argument
list (variables will be substituted in this function)
@arg $script_path - the path on the disk to the script file
@arg $host_id - the id of the host that this script query belongs to
@returns - a full path to the script query script containing all arguments */
function get_script_query_path($args, $script_path, $host_id) {
global $config;
include_once($config['library_path'] . '/variables.php');
/* get any extra arguments that need to be passed to the script */
if ($args != '') {
$parts = preg_split('/\s+/', $args);
$extra_arguments = '';
foreach($parts as $index => $part) {
$extra_arguments .= ($index > 0 ? ' ':'') . cacti_escapeshellarg(substitute_host_data($part, '|', '|', $host_id));
}
} else {
$extra_arguments = '';
}
/* get a complete path for out target script */
return substitute_script_query_path($script_path) . ' ' . $extra_arguments;
}
/**
* verify a given index_order
* @param array $raw_xml - parsed XML array
* @return bool - index_order field valid
*/
function verify_index_order($raw_xml) {
/* invalid xml check */
if ((!is_array($raw_xml)) || (sizeof($raw_xml) == 0)) {
query_debug_timer_offset('data_query', 'Error parsing XML file into an array.');
return false;
}
$xml_inputs = array();
/* list each of the input fields for this snmp query */
foreach ($raw_xml['fields'] as $field_name => $field_array) {
if ($field_array['direction'] == 'input' || $field_array['direction'] == 'input-output') {
/* create a list of all values for this index */
array_push($xml_inputs, $field_name);
}
}
$all_index_order_fields_found = true;
/* the xml file contains an ordered list of 'indexable' fields */
if (isset($raw_xml['index_order'])) {
$index_order_array = explode(':', $raw_xml['index_order']);
foreach($index_order_array as $element) {
$all_index_order_fields_found = $all_index_order_fields_found && (in_array($element, $xml_inputs));
}
} else {
/* the xml file does not contain an index order */
}
return $all_index_order_fields_found;
}
/**
* perform sql updates for all required tables for new index_sort_order
* @arg array $snmp_query_array
* host_id, snmp_query_id, snmp_index_on, snmp_query_graph_id,
* snmp_index,data_template_data_id, local_data_id
*
* this code stems from lib/template.php, function create_complete_graph_from_template
*/
function update_snmp_index_order($data_query) {
if (sizeof($data_query)) {
$data_input_field = array_rekey(db_fetch_assoc_prepared("SELECT
data_input_fields.id,
data_input_fields.type_code
FROM snmp_query
INNER JOIN data_input
ON snmp_query.data_input_id = data_input.id
INNER JOIN data_input_fields
ON data_input.id = data_input_fields.data_input_id
WHERE data_input_fields.type_code IN('index_type', 'index_value', 'output_type')
AND snmp_query.id = ?",
array($data_query['snmp_query_id'])), 'type_code', 'id');
$snmp_cache_value = db_fetch_cell_prepared('SELECT field_value
FROM host_snmp_cache
WHERE host_id = ?
AND snmp_query_id = ?
AND field_name = ?
AND snmp_index = ?',
array($data_query['host_id'], $data_query['snmp_query_id'], $data_query['snmp_index_on'], $data_query['snmp_index']));
db_execute_prepared('REPLACE INTO data_input_data
(data_input_field_id, data_template_data_id, t_value, value)
VALUES
(?, ?, "", ?),
(?, ?, "", ?),
(?, ?, "", ?)',
array(
/* save the value to index on (ie. ifindex, ifip, etc) */
$data_input_field['index_type'], $data_query['data_template_data_id'], $data_query['snmp_index_on'],
/* save the actual value (ie. 3, 192.168.1.101, etc) */
$data_input_field['index_value'], $data_query['data_template_data_id'], $snmp_cache_value,
/* set the expected output type (ie. bytes, errors, packets) */
$data_input_field['output_type'], $data_query['data_template_data_id'], $data_query['snmp_query_graph_id']
)
);
/* now that we have put data into the 'data_input_data' table, update the snmp cache for ds's */
update_data_source_data_query_cache($data_query['local_data_id']);
}
}