#!/usr/bin/perl

######
# ssrotate: start the following month (crontab script)
# Author: Jimmy Scott
# Nick: Sick`
# Mail: jimmy *at* inet-solutions.be
# Address:      Jimmy Scott
#               Goordijk 66
#               2930 BRASSCHAAT
#               BELGIUM
# source: http://pub.devbox.be
###
#
# Copyright (C) 2003 Jimmy Scott <http://www.devbox.be>
#
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#   1. Redistributions of source code must retain the above copyright notice,
#      this list of conditions and the following disclaimer.
#
#   2. Redistributions in binary form must reproduce the above copyright
#      notice, this list of conditions and the following disclaimer in the
#      documentation and/or other materials provided with the distribution.
#
#   3. The names of the authors may not be used to endorse or promote products
#      derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS "AS IS" AND ANY
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
# THE POSSIBILITY OF SUCH DAMAGE.
#
######

use Fcntl qw(:DEFAULT :flock);

######
# config options
###

$basedir="/home/jimmy/snmpdev/ss";
$ssreset="/home/jimmy/snmpdev/ssreset.pl";
$max=4294967296;

######
# getting month
###

($sec,$min,$hour,$mday,$mon,$year,$wday,$yday,$isdst) = localtime(time);
$mon++;
$year+=1900;

$mon--;
if ( $mon < 1 )
{
  $mon=12;
}

######
# checking and building hier
###

sub newfile
{
   my $file = shift || die "wrong usage of the function &newfile()\n";
   open NF, ">$file" || die "could not create $file\n";
   print NF "0";
   close NF;
}

######
# create missing month files
###

for ($i=1;$i<=24;$i++)
{
   for ($m=1;$m<=12;$m++)
   {
      if (! -f "$basedir/$i.in.m$m" )
      {
         &newfile("$basedir/$i.in.m$m");
      }
      if (! -f "$basedir/$i.out.m$m" )
      {
         &newfile("$basedir/$i.out.m$m");
      }
   }
}

######
# getting the lock
###

open LOCK, "$basedir/ss.lock" || die "can't open lockfile $basedir/ss.lock\n";
flock LOCK, LOCK_EX || die "can't lock lockfile $basedir/ss.lock\n";

######
# doing rotation
###

for ($i=1;$i<=24;$i++)
{
   my ($if,$of,$ic,$oc,$io,$oo,$it,$ot);

   ######
   # updating in
   ###

   # flow
   open FLOW, "$basedir/$i.in.f" || die "can't open file $basedir/$i.in.f\n";
   $if=<FLOW>;
   close FLOW;

   # count
   open FLOW, "$basedir/$i.in.c" || die "can't open file $basedir/$i.in.c\n";
   $ic=<FLOW>;
   close FLOW;

   # offset
   open FLOW, "$basedir/$i.in.o" || die "can't open file $basedir/$i.in.o\n";
   $io=<FLOW>;
   close FLOW;

   # total
   $it=$if + ($ic * $max) - $io;

   open TOT, ">$basedir/$i.in.m$mon" || die "can't open file >$basedir/$i.in.m$mon\n";
   print TOT "$it";
   close TOT;

   ######
   # updating out
   ###

   # flow
   open FLOW, "$basedir/$i.out.f" || die "can't open file $basedir/$i.out.f\n";
   $of=<FLOW>;
   close FLOW;

   # count
   open FLOW, "$basedir/$i.out.c" || die "can't open file $basedir/$i.out.c\n";
   $oc=<FLOW>;
   close FLOW;

   # offset
   open FLOW, "$basedir/$i.out.o" || die "can't open file $basedir/$i.out.o\n";
   $oo=<FLOW>;
   close FLOW;

   # total
   $ot=$of + ($oc * $max) - $oo;

   open TOT, ">$basedir/$i.out.m$mon" || die "can't open file >$basedir/$i.out.m$mon\n";
   print TOT "$ot";
   close TOT;
}

######
# running reset
###

close LOCK;
system("$ssreset");

system("date > \"$basedir/last.rotate\"");

######
# releasing the lock
###

exit;