src/scripts/qosd

   1 #!/usr/bin/perl -w
   2 use strict;
   3
   4 ##########################################
   5 ##
   6 ## DESCRIPTION
   7 ##
   8 ##   The tc-graph daemon script: "tc-collector"
   9 ##   Which is part of the ADSL-optimizer.
  10 ##
  11 ##   The script will become a daemon and periodically collect data
  12 ##   from the Linux traffic control system.  The collected data is
  13 ##   stored in some RRD-data files, which is created automatically by
  14 ##   the script if they don't exist.
  15 ##
  16 ## GRAPHs
  17 ##
  18 ##   How the RRD-data is displayed as graphs is not part of the
  19 ##   tc-collector tool.  But we recommend using the RRD-frontend 'ddraw'.
  20 ##   We have included some 'ddraw' examples (which is hardcoded to use
  21 ##   files from '/var/spool/rrdqueues').
  22 ##
  23 ##      drraw:  http://web.taranis.org/drraw/
  24 ##
  25 ##
  26 ## REQUIRES
  27 ##
  28 ##   RRDtools Perl interface RRDs
  29 ##   The "tc" command.
  30 ##
  31 ##
  32 ## AUTHOR
  33 ##   Jesper Dangaard Brouer <hawk@diku.dk>, d.16/4-2004
  34 ##
  35 ## CHANGELOG
  36 ##   2004-04-16:  Initial version.
  37 ##   2004-05-27:  Daemon version.
  38 ##
  39 ## $Id: tc-collector.pl,v 1.12 2005/03/19 19:31:08 hawk Exp $
  40 ##########################################
  41
  42 # TODO:
  43 #  * Calc time used to parse, use to make time steps more precise
  44 #  * Device list support
  45 #  * Detecting the correct devices
  46
  47 # Configuration options:
  48 #
  49 my  $device        = "$ARGV[0]";
  50 our $rrd_datadir   = "/var/log/rrd/";
  51 our $event_datadir = $rrd_datadir;
  52 our $STEP          = 10;
  53 our $tc_command    = "/sbin/tc";
  54
  55 # A trick is to set the environment PERL5LIB to include $GRAPHDIR
  56 #  This is done by the init-script
  57 # ($GRAPHDIR is obtained from /usr/local/etc/ADSL-optimizer.conf)
  58 my $include_dir = '/var/ipfire/qos/bin';
  59
  60
  61 # Create the $rrd_datadir if it doesn't exists
  62 if ( ! -d $rrd_datadir ) {
  63     print "RRD-datadir not found, creating it: $rrd_datadir \n";
  64     my $status = system("mkdir $rrd_datadir");
  65     die "\nERROR cannot create \"$rrd_datadir\"\n" unless $status == 0;
  66 }
  67
  68 # use POSIX;
  69 #
  70 #POSIX::setsid()
  71 #    or die "Can't become a daemon: $!";
  72
  73 # The init scripts will do the right "daemon" thing...
  74 # Become a daemon
  75 print "Becoming a daemon...\n";
  76 my $pid = fork;
  77 exit if $pid;
  78 die "Couldn't fork: $!" unless defined($pid);
  79
  80 my $time_to_die = 0;
  81 sub signal_handler {
  82     $time_to_die = 1;
  83 }
  84 # Trap signals
  85 $SIG{INT} = $SIG{TERM} = $SIG{HUP} = \&signal_handler;
  86 $SIG{PIPE} = 'IGNORE';
  87
  88 our %classes_data;
  89 our %classes_info;
  90 require "$include_dir/parse-func.pl";
  91 require "$include_dir/RRD-func.pl";
  92
  93 until ($time_to_die) {
  94
  95     #print "Parsing tc statistics on $device\n";
  96     my $res = parse_class($device);
  97     if ( ! $res ) {
  98         print " Error when parsing classes on $device\n";
  99     }
 100
 101     #print "Updating RRD data-files\n";
 102     $res = update_rrds();
 103     #if ( $res ) {
 104     #   print " Error updating RRDs: \"$res\"\n";
 105     #}
 106
 107 #    my $timestamp = time;
 108 #    print "$timestamp\n";
 109
 110     sleep($STEP);
 111 }
 112
 113 print "tc-collector daemon exiting ... bye bye!\n";