<?php
#
# Copyright (C) 2008 Red Hat, Inc.
# Author: Andreas Thienemann <athienem@redhat.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Library General Public License as published by
# the Free Software Foundation; version 2 only
#
# 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 Library General Public License for more details.
#
# You should have received a copy of the GNU Library General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
# Copyright 2004, 2005 Red Hat, Inc.
#
# AUTHOR: Andreas Thienemann <athienem@redhat.com>
#

require_once('stacked-bars.inc.php');

// Get the data from the table, format it and write it out to a file
function chart_prepare($flowsrc, $starttime, $endtime) {
	global $date_fmt_str;
	global $dbh;


	// Define the plot properties
	$style = 'Top Source Ports';
	$chart_title = $style.' '.$flowsrc.' - '.strftime($date_fmt_str, $starttime).' to '.strftime($date_fmt_str, $endtime);
	$ytitle = 'Bytes';


	// Grab the data from the database
	// get top-talkers for the timespan: top 9 hosts plus rest-traffic
	$q = $dbh->prepare('SELECT PortSrc, SUM(Bytes) AS TrafficSum
			FROM Flows
			WHERE TimeStop BETWEEN ? AND ?
			AND FlowSrc = ?
			GROUP BY PortSrc
			ORDER BY TrafficSum DESC
			LIMIT 0, 9');
	$q->bindParam(1, $starttime, PDO::PARAM_INT);
	$q->bindParam(2, $endtime, PDO::PARAM_INT);
	$q->bindParam(3, sprintf('%u', ip2long($flowsrc)), PDO::PARAM_INT);
	$q->execute();

	// Fill the legend build the WHERE limit
	$legend = $q->fetchAll(PDO::FETCH_COLUMN, 0);
	$limit = '';
	for ($i = 0; $i < count($legend); $i++) {
		$limit .= @$legend[$i].', ';
		if (array_key_exists($i, $legend)) {
			$legend[$i] = $legend[$i];
		} else {
			$legend[$i] = '';
		}
	}
	$legend[] = 'rest';
	$limit = substr($limit, 0, (strlen($limit) - 2));

	// divide the timespan into 5min intervals
	// query the traffic per port in 5min intervals
	// Initialize the data_table
	$data = array();
	$steps = 0;
	$rest_table = array();
	for ($time1 = $starttime; $time1 < $endtime; $time1 += (5 * 60)) {
		$time2 = ($time1 + (5 * 60));
		$steps++;
		// Add the column header (time) to the data array
		$data[] = array(strftime('%H:%M', $time1));
		$q = $dbh->prepare('SELECT PortSrc, SUM(Bytes) AS TrafficSum
				FROM Flows
				WHERE TimeStop BETWEEN ? AND ?
				AND PortSrc IN ('.$limit.')
				AND FlowSrc = ?
				GROUP BY PortSrc
				ORDER BY TrafficSum DESC
				LIMIT 0, 9');
		$q->bindParam(1, $time1, PDO::PARAM_INT);
		$q->bindParam(2, $time2, PDO::PARAM_INT);
		$q->bindParam(3, sprintf('%u', ip2long($flowsrc)), PDO::PARAM_INT);
		$q->execute();
		$r = $q->fetchAll();
		$i = (count($data) - 1);
		foreach ($r as $row) {
			$tmpdata[$i][$row[0]] = $row[1];
		}

		// Get the rest of the aggregated traffic to fill up the bars
		$q = $dbh->prepare('SELECT SUM(Bytes) AS TrafficSum
				FROM Flows
				WHERE TimeStop BETWEEN ? AND ?
				AND PortSrc NOT IN ('.$limit.')
				AND FlowSrc = ?');
		$q->bindParam(1, $time1, PDO::PARAM_INT);
		$q->bindParam(2, $time2, PDO::PARAM_INT);
		$q->bindParam(3, sprintf('%u', ip2long($flowsrc)), PDO::PARAM_INT);
		$q->execute();
		$rest_table[] = $q->fetch();
	}

	// Iterate through the legend and fill the data array from the temporary data array
	// This function makes sure that each top talker is always at the same position
	// in the multidimensional data table.
	// This is necessary for phpplot to assign the same color to each host on
	// sequential columns
	// $ip is the key for the temp_data lookup
	foreach ($legend as $port) {
		for($i = 0; $i < $steps; $i++) {
			if (array_key_exists($i, $tmpdata)) {
			// The "column" exists, let's sort the rows
				$key = sprintf('%u', $port);
				if (array_key_exists($key, $tmpdata[$i])) {
					$data[$i][] = $tmpdata[$i][$key];
				} else {
					$data[$i][] = '';
				}
			} else {
				$data[$i][] = '';
			}
		}
	}


	// Append the data from the rest_table to the data array
	for($i = 0; $i < $steps; $i++) {
		$data[$i][(count($data[$i]) - 1)] = $rest_table[$i][0];
	}

	return draw_stackedbar($chart_title, $legend, $ytitle, $data);

}