root/trunk/flowmon-web/db_handler.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>
#


// initiate the database connection
function init_db() {
    global $dbh, $db_host, $db_name, $db_user, $db_pass;
    try {
        if ($database = 'mysql') {
            $dbh = new PDO('mysql:host='.$db_host.';dbname='.$db_name, $db_user, $db_pass);
        } else if ($database = 'postgresql') {
            $dbh = new PDO('pgsql:host='.$db_host.';dbname='.$db_name, $db_user, $db_pass);
        }
        $dbh->setAttribute( PDO::ATTR_ERRMODE, PDO::ERRMODE_WARNING );
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }
}

// return the number of reporting probes
function get_num_probes() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT COUNT(FlowSrc) FROM FlowSources;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the number of archived flows
function get_num_flows() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT COUNT(FlowSrc) FROM Flows;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the number of systems in the databse
function get_num_systems() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT COUNT(DISTINCT IPSrc) FROM Flows;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the oldest flow in the db as a unix timestamp
function get_oldest_flow_ts() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT MIN(TimeStart) FROM Flows;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the oldest flow in the db as a unix timestamp
function get_newest_flow_ts() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT MAX(TimeStop) FROM Flows;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the total traffic in the archive
function get_bytes() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT SUM(Bytes) FROM Flows;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the total traffic in the archive
function get_pkts() {
    global $dbh;
    try {
        $q = $dbh->query("SELECT SUM(Pakets) FROM Flows;");
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }

    return $r[0];
}

// return the database size
function get_db_size() {
    global $dbh;
    try {
        $size = 0;
        foreach($dbh->query("SHOW TABLE STATUS") as $row) {
            $size += $row['Data_length'] + $row['Index_length'];
        }
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }
    return $size;
}

// return existing FlowProbes
function get_probes() {
    global $dbh;
    $probes = array();
    try {
        foreach($dbh->query("SELECT FlowSrc, Hostname FROM FlowSources") as $row) {
            // No hostname found in DB, retrieve it from the DNS
            $ip = long2ip($row['FlowSrc']);
            if (empty($row['Hostname'])) {
                $row['Hostname'] = gethostbyaddr($ip);
            }
            $probes[$ip] = $row['Hostname'];
        }
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }
    return $probes;
}

// check if a probe definition is valid
// expects an ip address
function isvalid_probe($ip) {
    global $dbh;
    $probes = array();
    try {
        $q = $dbh->prepare("SELECT FlowSrc FROM FlowSources WHERE FlowSrc = ?;");
        $q->bindParam(1, sprintf("%u", ip2long($ip)));
        $q->execute();
        $r = $q->fetch();
    } catch (PDOException $e) {
        print "Error!: " . $e->getMessage() . "<br/>";
        die();
    }
    if ($ip == long2ip($r[0])) {
        return True;
    }
    return False;
}

//$dbh = '';
//init_db();