<?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/                                                   |
 +-------------------------------------------------------------------------+
*/

/* nth_percentile - given a data source, calculate the Nth percentile for a given
     time period
   @arg $local_data_id - the data source to perform the Nth percentile calculation
   @arg $start_seconds - start seconds of time range
   @arg $stop_seconds - stop seconds of time range
   @arg $percentile - Nth Percentile to calculate, integer between 1 and 99
   @arg $resolution - the accuracy of the data measured in seconds
   @returns - (array) an array containing each data source item, and its 95th percentile */
function nth_percentile($local_data_id, $start_seconds, $end_seconds, $percentile = 95, $resolution = 0) {
	global $config;

	include_once($config['library_path'] . '/rrd.php');

	$values_array = array();

	/* Check to see if we have array, if array, we are doing complex 95th calculations */
	if (is_array($local_data_id)) {
		$sum_array = array();
		$fetch_array = array();
		$percentile_array = array();
		$num_ds = 0;

		$i = 0;
		/* do a fetch for each data source */
		while (list($ldi, $ldi_value) = each($local_data_id)) {
			$fetch_array[$i] = @rrdtool_function_fetch($ldi, $start_seconds, $end_seconds, $resolution);
			/* clean up unwanted data source items */
			if (! empty($fetch_array[$i])) {
				while (list($id, $name) = each($fetch_array[$i]['data_source_names'])) {
					/* get rid of the Nth max for now since we'll need to re-calculate it later */
					if ($name == 'nth_percentile_maximum') {
						unset($fetch_array[$i]['data_source_names'][$id]);
						unset($fetch_array[$i]['values'][$id]);
					}
					/* clean up DS items that aren't defined on the graph */
					if (! in_array($name, $local_data_id[$ldi])) {
						unset($fetch_array[$i]['data_source_names'][$id]);
						unset($fetch_array[$i]['values'][$id]);
					}
				}
				$i++;
			}
		}

		/* Create our array for working  */
		if (empty($fetch_array[0]['data_source_names'])) {
			/* here to avoid warning on non-exist file */
			$fetch_array = array();
		}else{
			$sum_array['data_source_names'] = $fetch_array[0]['data_source_names'];
		}

		if (sizeof($fetch_array) > 0) {
			/* Create hash of ds_item_name => array or fetch_array indexes */
			/* this is used to later sum, max and total the data sources used on the graph */
			/* Loop fetch array index  */
			$dsi_name_to_id = array();
			reset($fetch_array);
			for ($i=0; $i<count($fetch_array); $i++) {
				/* Go through data souce names */
				reset($fetch_array[$i]['data_source_names']);
				foreach ($fetch_array[$i]['data_source_names'] as $ds_name) {
					$dsi_name_to_id[$ds_name][] = $i;
				}
			}

			/* Sum up the like data sources */
			$sum_array = array();
			$i = 0;
			foreach ( $dsi_name_to_id as $ds_name => $id_array ) {
				$sum_array['data_source_names'][$i] = $ds_name;
				foreach ($id_array as $id) {
					$fetch_id = array_search($ds_name,$fetch_array[$id]['data_source_names']);
						/* Sum up like ds names */
						for ($j=0; $j<count($fetch_array[$id]['values'][$fetch_id]); $j++) {
							if (isset($fetch_array[$id]['values'][$fetch_id][$j])) {
								$value = $fetch_array[$id]['values'][$fetch_id][$j];
							}else{
								$value = 0;
							}
							if (isset($sum_array['values'][$i][$j])) {
								$sum_array['values'][$i][$j] += $value;
							}else{
								$sum_array['values'][$i][$j] = $value;
							}
						}
				}
				$i++;
			}
		}

		/* calculate extra data, max, sum */
		if (isset($sum_array['values'])) {
			$num_ds = count($sum_array['values']);
			$total_ds = count($sum_array['values']);
			for ($j=0; $j<$total_ds; $j++) { /* each data source item */
				for ($k=0; $k<count($sum_array['values'][$j]); $k++) { /* each rrd row */
					/* now we must re-calculate the 95th max */
					$value = 0;
					if (isset($sum_array['values'][$j][$k])) {
						$value = $sum_array['values'][$j][$k];
					}
					if (isset($sum_array['values'][$num_ds][$k])) {
						$sum_array['values'][$num_ds][$k] = max($value, $sum_array['values'][$num_ds][$k]);
					}else{
						$sum_array['values'][$num_ds][$k] = max($value, 0);
					}
					/* sum of all ds rows */
					$value = 0;
					if (isset($sum_array['values'][$j][$k])) {
						$value = $sum_array['values'][$j][$k];
					}

					if (isset($sum_array['values'][$num_ds + 1][$k])) {
						$sum_array['values'][$num_ds + 1][$k] += $value;
					}else{
						$sum_array['values'][$num_ds + 1][$k] = $value;
					}

				}
			}

			$sum_array['data_source_names'][$num_ds] = 'nth_percentile_aggregate_max';
			$sum_array['data_source_names'][$num_ds + 1] = 'nth_percentile_aggregate_sum';

		}

		$fetch_array = $sum_array;
	}else{
		/* No array, just calculate the 95th for the data source */
		$fetch_array = @rrdtool_function_fetch($local_data_id, $start_seconds, $end_seconds, $resolution);
	}

	/* loop through each data source */
	if (empty($fetch_array['data_source_names'])) {
		$return_array = array();
	}else{
		for ($i=0; $i<count($fetch_array['data_source_names']); $i++) {
			if (isset($fetch_array['values'][$i])) {
				$values_array = $fetch_array['values'][$i];

				/* sort the array in descending order */
				rsort($values_array, SORT_NUMERIC);
			}

			/* grab the N% row (or 1 - N% in reverse) and use that as our Nth percentile
			value */
			$inverse_percentile = 1 - ($percentile / 100);
			$target = ((count($values_array) + 1) * $inverse_percentile);
			$target = sprintf('%d', $target);

			if (empty($values_array[$target])) { $values_array[$target] = 0; }

			/* collect Nth percentile values in this array so we can return them */
			$return_array{$fetch_array['data_source_names'][$i]} = $values_array[$target];

			/* get max Nth calculation for aggregate */
			if (($fetch_array['data_source_names'][$i] != 'nth_percentile_aggregate_max') &&
				($fetch_array['data_source_names'][$i] != 'nth_percentile_aggregate_sum') &&
				($fetch_array['data_source_names'][$i] != 'nth_percentile_maximum')) {
				if (isset($return_array{'nth_percentile_aggregate_total'})) {
					if (($return_array{'nth_percentile_aggregate_total'} < $values_array[$target])) {
						$return_array{'nth_percentile_aggregate_total'} = $values_array[$target];
					}
				}else{
					$return_array{'nth_percentile_aggregate_total'} = $values_array[$target];
				}
			}
		}
	}

	if (isset($return_array)) {
		return $return_array;
	}

}



/* bandwidth_summation - given a data source, sums all data in the rrd for a given
     time period
   @arg $local_data_id - the data source to perform the summation for
   @arg $start_time - the start time to use for the data calculation. this value can
     either be absolute (unix timestamp) or relative (to now)
   @arg $end_time - the end time to use for the data calculation. this value can
     either be absolute (unix timestamp) or relative (to now)
   @arg $resolution - the accuracy of the data measured in seconds
   @arg $rra_steps - how many periods each sample in the RRA counts for, values above '1'
     result in an averaged summation
   @arg $ds_steps - how many seconds each period represents
   @returns - (array) an array containing each data source item, and its sum */
function bandwidth_summation($local_data_id, $start_time, $end_time, $rra_steps, $ds_steps) {
	$fetch_array = @rrdtool_function_fetch($local_data_id, $start_time, $end_time, $rra_steps * $ds_steps);

	if ((!isset($fetch_array['data_source_names'])) || (count($fetch_array['data_source_names']) == 0)) {
		return;
	}

	$return_array = array();

	/* loop through each regexp determined above (or each data source) */
	for ($i=0;$i<count($fetch_array['data_source_names']);$i++) {
		$sum = 0;

		if (isset($fetch_array['values'][$i])) {
			$values_array = $fetch_array['values'][$i];

			foreach ($fetch_array['values'][$i] as $value) {
				$sum += $value;
			}

			if (count($fetch_array['values'][$i]) != 0) {
				$sum = ($sum * $ds_steps * $rra_steps);
			}else{
				$sum = 0;
			}

			/* collect summation alues in this array so we can return them */
			$return_array[$fetch_array['data_source_names'][$i]] = $sum;
		}
	}

	return $return_array;
}

/* variable_nth_percentile - given a Nth percentile variable, calculate the Nth percentile
     and format it for display on the graph
   @arg $regexp_match_array - the array that contains each argument in the Nth percentile variable. it
     should be formatted like so:
       $arr[0] // full variable string
       $arr[1] // Nth percentile
       $arr[2] // bits or bytes
       $arr[3] // power of 10 divisor
       $arr[4] // current, total, max, total_peak, all_max_current, all_max_peak
       $arr[5] // digits of floating point precision
   @arg $graph_item - an array that contains the current graph item
   @arg $graph_items - an array that contains all graph items
   @arg $graph_start - the start time to use for the data calculation. this value can
     either be absolute (unix timestamp) or relative (to now)
   @arg $graph_end - the end time to use for the data calculation. this value can
     either be absolute (unix timestamp) or relative (to now)
   @arg $seconds_between_graph_updates - the number of seconds between each update on the graph which
     varies depending on the RRA in use
   @returns - a string containg the Nth percentile suitable for placing on the graph */
function variable_nth_percentile(&$regexp_match_array, &$graph_item, &$graph_items, $graph_start, $graph_end) {
	global $graph_item_types;

	if (sizeof($regexp_match_array) == 0) {
		return 0;
	}

	if (($regexp_match_array[1] < 1) || ($regexp_match_array[1] > 99)) {
		/* error Nth Percentile variable is incorrect */
		return -1;
	}

	/* Get the Nth percentile values */
	if (($regexp_match_array[4] == 'current') || ($regexp_match_array[4] == 'max')) {
		$nth_cache{$graph_item['local_data_id']} = nth_percentile($graph_item['local_data_id'], $graph_start, $graph_end, $regexp_match_array[1]);
	}elseif (($regexp_match_array[4] == 'total') || ($regexp_match_array[4] == 'total_peak') || ($regexp_match_array[4] == 'all_max_current') || ($regexp_match_array[4] == 'all_max_peak')) {
		for ($t=0;($t<count($graph_items));$t++) {
			if (!empty($graph_items[$t]['local_data_id'])) {
				$nth_cache{$graph_items[$t]['local_data_id']} = nth_percentile($graph_items[$t]['local_data_id'], $graph_start, $graph_end, $regexp_match_array[1]);
			}
		}
	}elseif (($regexp_match_array[4] == 'aggregate') || ($regexp_match_array[4] == 'aggregate_sum') || ($regexp_match_array[4] == 'aggregate_max')) {
		$local_data_array = array();
		for ($t=0;($t<count($graph_items));$t++) {
			if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) && (!empty($graph_items[$t]['data_template_rrd_id']))) {
				$local_data_array[$graph_items[$t]['local_data_id']][] = $graph_items[$t]['data_source_name'];
			}
		}
                $nth_cache{0} = nth_percentile($local_data_array, $graph_start, $graph_end, $regexp_match_array[1]);
	} elseif ($regexp_match_array[4] == 'aggregate_current') {
		$local_data_array = array();
		if (!empty($graph_item['data_source_name'])) {
			for ($t=0;($t<count($graph_items));$t++) {
				if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) &&
					(!empty($graph_items[$t]['data_template_rrd_id'])) &&
					($graph_item['data_source_name'] == $graph_items[$t]['data_source_name'])) {
					$local_data_array[$graph_items[$t]['local_data_id']][] = $graph_items[$t]['data_source_name'];
				}
			}
			$nth_cache{0} = nth_percentile($local_data_array, $graph_start, $graph_end, $regexp_match_array[1]);
		}
	}

	$nth = 0;

	/* format the output according to args passed to the variable */
	if ($regexp_match_array[4] == 'current') {
		if (! empty($nth_cache{$graph_item['local_data_id']}{$graph_item['data_source_name']})) {
			$nth = $nth_cache{$graph_item['local_data_id']}{$graph_item['data_source_name']};
			$nth = ($regexp_match_array[2] == 'bits') ? $nth * 8 : $nth;
			$nth /= pow(10,intval($regexp_match_array[3]));
		}
	}elseif ($regexp_match_array[4] == 'total') {
		for ($t=0;($t<count($graph_items));$t++) {
			if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) && (!empty($graph_items[$t]['data_template_rrd_id']))) {
				if (! empty($nth_cache{$graph_items[$t]['local_data_id']}{$graph_items[$t]['data_source_name']})) {
					$local_nth = $nth_cache{$graph_items[$t]['local_data_id']}{$graph_items[$t]['data_source_name']};
					$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
					$local_nth /= pow(10,intval($regexp_match_array[3]));

					$nth += $local_nth;
				}

			}
		}
	}elseif ($regexp_match_array[4] == 'max') {
		if (! empty($nth_cache{$graph_item['local_data_id']}['nth_percentile_maximum'])) {
			$nth = $nth_cache{$graph_item['local_data_id']}['nth_percentile_maximum'];
			$nth = ($regexp_match_array[2] == 'bits') ? $nth * 8 : $nth;
			$nth /= pow(10,intval($regexp_match_array[3]));
		}
	}elseif ($regexp_match_array[4] == 'total_peak') {
		for ($t=0;($t<count($graph_items));$t++) {
			if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) && (!empty($graph_items[$t]['data_template_rrd_id']))) {
				if (! empty($nth_cache{$graph_items[$t]['local_data_id']}['nth_percentile_maximum'])) {
					$local_nth = $nth_cache{$graph_items[$t]['local_data_id']}['nth_percentile_maximum'];
					$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
					$local_nth /= pow(10,intval($regexp_match_array[3]));

					$nth += $local_nth;
				}
			}
		}
	}elseif ($regexp_match_array[4] == 'all_max_current') {
		for ($t=0;($t<count($graph_items));$t++) {
			if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) && (!empty($graph_items[$t]['data_template_rrd_id']))) {
				if (! empty($ninety_fifth_cache{$graph_items[$t]['local_data_id']}{$graph_items[$t]['data_source_name']})) {
					$local_nth = $ninety_fifth_cache{$graph_items[$t]['local_data_id']}{$graph_items[$t]['data_source_name']};
					$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
					$local_nth /= pow(10,intval($regexp_match_array[3]));

					if ($local_nth > $nth) {
						$nth = $local_nth;
					}
				}
			}
		}
	}elseif ($regexp_match_array[4] == 'all_max_peak') {
		for ($t=0;($t<count($graph_items));$t++) {
			if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) && (!empty($graph_items[$t]['data_template_rrd_id']))) {
				if (! empty($nth_cache{$graph_items[$t]['local_data_id']}['nth_percentile_maximum'])) {
					$local_nth = $nth_cache{$graph_items[$t]['local_data_id']}['nth_percentile_maximum'];
					$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
					$local_nth /= pow(10,intval($regexp_match_array[3]));

					if ($local_nth > $nth) {
						$nth = $local_nth;
					}
				}
			}
		}
	}elseif (($regexp_match_array[4] == 'aggregate') || ($regexp_match_array[4] == 'aggregate_current')) {
		if (! empty($nth_cache{0}['nth_percentile_aggregate_total'])) {
			$local_nth = $nth_cache{0}['nth_percentile_aggregate_total'];
			$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
			$local_nth /= pow(10,intval($regexp_match_array[3]));
			$nth = $local_nth;
		}
	}elseif ($regexp_match_array[4] == 'aggregate_max') {
		if (! empty($nth_cache{0}['nth_percentile_aggregate_max'])) {
			$local_nth = $nth_cache{0}['nth_percentile_aggregate_max'];
			$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
			$local_nth /= pow(10,intval($regexp_match_array[3]));
			$nth = $local_nth;
		}
	}elseif ($regexp_match_array[4] == 'aggregate_sum') {
		if (! empty($nth_cache{0}['nth_percentile_aggregate_sum'])) {
			$local_nth = $nth_cache{0}['nth_percentile_aggregate_sum'];
			$local_nth = ($regexp_match_array[2] == 'bits') ? $local_nth * 8 : $local_nth;
			$local_nth /= pow(10,intval($regexp_match_array[3]));
			$nth = $local_nth;
		}
	}

	/* determine the floating point precision */
	if (is_numeric($regexp_match_array[5])) {
		$round_to = $regexp_match_array[5];
	}else{
		$round_to = 2;
	}

	/* return the final result and round off to two decimal digits */
	return round($nth, $round_to);
}

/* variable_bandwidth_summation - given a bandwidth summation variable, calculate the summation
     and format it for display on the graph
   @arg $regexp_match_array - the array that contains each argument in the bandwidth summation variable. it
     should be formatted like so:
       $arr[0] // full variable string
       $arr[1] // power of 10 divisor or 'auto'
       $arr[2] // current, total
       $arr[3] // digits of floating point precision
       $arr[4] // seconds to perform the calculation for or 'auto'
   @arg $graph_item - an array that contains the current graph item
   @arg $graph_items - an array that contains all graph items
   @arg $graph_start - the start time to use for the data calculation. this value can
     either be absolute (unix timestamp) or relative (to now)
   @arg $graph_end - the end time to use for the data calculation. this value can
     either be absolute (unix timestamp) or relative (to now)
   @arg $seconds_between_graph_updates - the number of seconds between each update on the graph which
     varies depending on the RRA in use
   @arg $rra_step - how many periods each sample in the RRA counts for, values above '1' result in an
     averaged summation
   @arg $ds_step - how many seconds each period represents
   @returns - a string containg the bandwidth summation suitable for placing on the graph */
function variable_bandwidth_summation(&$regexp_match_array, &$graph_item, &$graph_items, $graph_start, $graph_end, $rra_step, $ds_step) {
	global $graph_item_types;

	if (sizeof($regexp_match_array) == 0) {
		return 0;
	}

	if (is_numeric($regexp_match_array[4])) {
		$summation_timespan_start = -$regexp_match_array[4];
	}else{
		$summation_timespan_start = $graph_start;
	}

	if ($regexp_match_array[2] == 'current') {
		$summation_cache{$graph_item['local_data_id']} = bandwidth_summation($graph_item['local_data_id'], $summation_timespan_start, $graph_end, $rra_step, $ds_step);
	}elseif ($regexp_match_array[2] == 'total') {
		for ($t=0;($t<count($graph_items));$t++) {
			if (!empty($graph_items[$t]['local_data_id'])) {
				$summation_cache{$graph_items[$t]['local_data_id']} = bandwidth_summation($graph_items[$t]['local_data_id'], $summation_timespan_start, $graph_end, $rra_step, $ds_step);
			}
		}
	}elseif ($regexp_match_array[2] == 'atomic') {
		$summation_cache{$graph_item['local_data_id']} = bandwidth_summation($graph_item['local_data_id'], $summation_timespan_start, $graph_end, $rra_step, 1);
	}

	$summation = 0;

	/* format the output according to args passed to the variable */
	if (($regexp_match_array[2] == 'current') || ($regexp_match_array[2] == 'atomic')) {
		if (isset($summation_cache{$graph_item['local_data_id']}{$graph_item['data_source_name']})) {
			$summation = $summation_cache{$graph_item['local_data_id']}{$graph_item['data_source_name']};
		}else{
			$summation = 0;
		}
	}elseif ($regexp_match_array[2] == 'total') {
		for ($t=0;($t<count($graph_items));$t++) {
			if ((preg_match('/(AREA|STACK|LINE[123])/', $graph_item_types{$graph_items[$t]['graph_type_id']})) && (!empty($graph_items[$t]['data_template_rrd_id']))) {
				if (isset($summation_cache{$graph_items[$t]['local_data_id']}{$graph_items[$t]['data_source_name']})) {
					$summation += $summation_cache{$graph_items[$t]['local_data_id']}{$graph_items[$t]['data_source_name']};
				}
			}
		}
	}

	if (preg_match('/\d+/', $regexp_match_array[1])) {
		$summation /= pow(10,intval($regexp_match_array[1]));
	}elseif ($regexp_match_array[1] == 'auto') {
		if ($summation < 1000) {
			$summation_label = 'B';
		}elseif ($summation < 1000000) {
			$summation_label = 'KB';
			$summation /= 1000;
		}elseif ($summation < 1000000000) {
			$summation_label = 'MB';
			$summation /= 1000000;
		}elseif ($summation < 1000000000000) {
			$summation_label = 'GB';
			$summation /= 1000000000;
		}else{
			$summation_label = 'TB';
			$summation /= 1000000000000;
		}
	}

	/* determine the floating point precision */
	if (is_numeric($regexp_match_array[3])) {
		$round_to = $regexp_match_array[3];
	}else{
		$round_to = 2;
	}

	/* substitute in the final result and round off to two decimal digits */
	if (isset($summation_label)) {
		return round($summation, $round_to) . " $summation_label";
	}else{
		return round($summation, $round_to);
	}
}

