root/trunk/flowmon-web/charts/top-src.inc.php

<?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('phplot-5.0.5/phplot.php');

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

    if ($timeframe != False) {
        $starttime = (time() - $timeframe);
        $endtime = time();
    }

    // Define the plot properties
    $style = 'Top Sources';
    $chart_title = $style.' '.$flowsrc.' - '.strftime($date_fmt_str, $starttime).' to '.strftime($date_fmt_str, $endtime);
    $ytitle = 'Bytes';
    $output = tempnam('tmp/', 'plot');
    $data_colors = array('SkyBlue', 'green', 'orange', 'blue', 'purple', 'red', 'violet', 'azure1', 'yellow', 'DarkGreen');


    // Grab the data from the database
    // get top-talkers for the timespan: top 9 hosts plus rest-traffic
    $q = $dbh->prepare('SELECT IPSrc, SUM(Bytes) AS TrafficSum
            FROM Flows
            WHERE TimeStop BETWEEN ? AND ?
            AND FlowSrc = ?
            GROUP BY IPSrc
            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] = long2ip($legend[$i]);
        } else {
            $legend[$i] = '';
        }
    }
    $legend[] = 'rest';
    $limit = substr($limit, 0, (strlen($limit) - 2));

    // divide the timespan into 5min intervals
    // query the traffic per talker 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 IPSrc, SUM(Bytes) AS TrafficSum
                FROM Flows
                WHERE TimeStop BETWEEN ? AND ?
                AND IPSrc IN ('.$limit.')
                AND FlowSrc = ?
                GROUP BY IPSrc
                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 IPSrc 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 $ip) {
        for($i = 0; $i < $steps; $i++) {
            if (array_key_exists($i, $tmpdata)) {
            // The "column" exists, let's sort the rows
                $key = sprintf('%u', ip2long($ip));
                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];
    }

    // Create the chart, write it to tmp/output.png
    $plot = new PHPlot(900, 400);
    $plot->SetIsInline(True);
    //$plot->SetPrintImage(False);
    $plot->SetImageBorderType('plain');
    $plot->SetDataColors($data_colors);


    $plot->SetPlotType('stackedbars');
    $plot->SetDataType('text-data');
    $plot->SetDataValues($data);

    $plot->SetTitle($chart_title);
    $plot->SetYTitle($ytitle);
    $plot->SetLegend($legend);

    $plot->SetXTickLabelPos('none');
    $plot->SetXTickPos('none');

    $plot->SetLegendPixels(760, 40);
    $plot->SetMarginsPixels(100, 150, 50, 50);


    $plot->SetNumberFormat(',', '.');
    $plot->SetPrecisionY(0);
    $plot->SetYLabelType('data');

    $plot->SetOutputFile($output);
    $plot->DrawGraph();

    chmod($output, 0644);
    $target =  dirname($output).'/'.basename($output).'.png';
    if (rename($output, $target)) {
        return 'tmp/'.basename($target);
    } else {
        return False;
    }
}