#!/usr/bin/perl

# ircstats2.pl v1.3 06 Apr 2001 Lucas Nussbaum <lucas@schickler.net>

# More info can be found on http://www.schickler.net/lucas/ircstats2.php

# based on ircstats from Matt Ramsey (MR227) mjr@geeks.org, mjr@nin.com

# Used to generate stats on the Kewl.Org IRC Network

# If you see something that should be changed, please notify me !

use Socket;

# pseudo-client parameters

$nick = "StatsMaker234";

$ident = "Stats";

$realname = "http://www.schickler.net/lucas/";

# server and port to connect to

$ircserver = "127.0.0.1";

$ircport = "6667";

# Where to store the output data (directory MUST exist)

# The number of users on each server will be in a file named like the server.

# An additionnal file named 'global' will be created, containing the sum of

# all servers' users.

# An example configuration file is included.

$datapath = "/home/lucas/ircstats2/temp";

# Where to put the PID file

$pidfile = "/home/lucas/ircstats2/ircstats2.pid";

# NOTE: Set those variables to 0 to enable, put a # in front of the 

# lines to disable.

# Count the global number of users in "globalusersmrtg" file (MRTG format)

$globalusersmrtg=0;

# Count the channels in "channelsmrtg" file (MRTG format)

$channelsmrtg=0;

# Count the global number of ircops in "ircopsmrtg" file (MRTG format)

$ircopsmrtg=0;

# Count the number of servers in "serversmrtg" file (MRTG format)

$serversmrtg=0;

# Count the global number of users in "globalusersplain" file

# (Plain text, to be used with others scripts)

$globalusersplain=0;

# Count the channels in "channelsplain" file (Plain text, 

# to be used with others scripts)

$channelsplain=0;

# Count the global number of ircops in "ircopsplain" file

# (Plain text, to be used with others scripts)

$ircopsplain=0;

# Count the number of servers in "serversmrtg file (MRTG format)

$serversplain=0;

# ircops number offset (not to count services, for example)

$ircopsoffset=0;

# servers number offset (not to count services, for example)

$serversoffset=0;

# Servers to poll

$server{"cybernet.langochat.net"}=0;

$server{"sicfa.langochat.net"}=0;

$server{"free.langochat.net"}=0;

$server{"jeuxgroup.langochat.net"}=0;

$server{"diligo.langochat.net"}=0;

##############################################

### DO NOT MODIFY ANYTHING BELOW THIS LINE ###

##############################################

print "Now daemonizing ...\n";

&daemonize;

if (!(($globalusersmrtg eq "")&&($globalusersplain eq "")&&($channelsmrtg eq "")&&($channelsplain eq "")&&($ircopsmrtg eq "")&&($ircopsplain eq "")&&($serversplain eq "")&&($serversmrtg eq ""))) {

	$needlusers=1;

}

($g, $g, $proto) = getprotobyname("tcp");

while (true) {

	($g, $g, $g, $g, $rawserver) = gethostbyname($ircserver);

	if ($rawserver) {

		$serveraddr = pack("Sna4x8", 2, $ircport, $rawserver);

		socket(SOCKET, AF_INET, SOCK_STREAM, $proto) || die "No socket: $!";

		if (connect(SOCKET, $serveraddr)){

			select(SOCKET); $| = 1;

			select(STDOUT); $| = 1; 

			print SOCKET "USER $ident a b :$realname\n";

			print SOCKET "NICK $nick\n";

			while (<SOCKET>){

				@i = split(" ",$_);

				if ($i[1] eq "433") {print SOCKET "NICK ".$nick.time()."\n";}

				elsif ($i[1] eq "437") {print SOCKET "NICK ".$nick.time()."\n";}

				elsif ($i[1] eq "376") {&getinfo;}

				elsif ($i[1] eq "402") {$server{lc($i[3])}=0;}

				elsif ($i[1] eq "265") {

					$i[0]=~s/\://;

					$server{lc($i[0])}=$i[6];

				}

				elsif ($i[1] eq "266") {

					if ($globalusersmrtg ne "") {

						$globalusersmrtg=$i[6];

					}

					if ($globalusersplain ne "") {

						$globalusersplain=$i[6];

					}

				}

				elsif ($i[1] eq "254") {

					if ($channelsmrtg ne "") {

						$channelsmrtg=$i[3];

					}

					if ($channelsplain ne "") {

						$channelsplain=$i[3];

					}

				}

				elsif ($i[1] eq "251") {

					if ($serversmrtg ne "") {

						$serversmrtg=$i[11]-$serversoffset;

					}

					if ($serversplain ne "") {

						$serversplain=$i[11]-$serversoffset;

					}

				}

				elsif ($i[1] eq "252") {

					if ($ircopsmrtg ne "") {

						$ircopsmrtg=$i[3]-$ircopsoffset;

					}

					if ($ircopsplain ne "") {

						$ircopsplain=$i[3]-$ircopsoffset;

					}

				}

				elsif ($i[0] eq "PING") {

					&saveinfo;

					print SOCKET "PONG $i[1]\n";

					&getinfo;

				}

			}

		}

		close(SOCKET);

	}

	sleep(120);

}

sub saveinfo {

	foreach $s (keys %server){

		open(OUTPUT,">$datapath/$s");

		print OUTPUT "$server{$s}\n$server{$s}\n$time\n";

		close(OUTPUT); 

	}

	if ($globalusersmrtg ne "") {

		open(OUTPUT, ">$datapath/globalusersmrtg");

		print OUTPUT "$globalusersmrtg\n$globalusersmrtg\n$time\n";

		close(OUTPUT);

	}

	if ($channelsmrtg ne "") {

		open(OUTPUT, ">$datapath/channelsmrtg");

		print OUTPUT "$channelsmrtg\n$channelsmrtg\n$time\n";

		close(OUTPUT);

	}

	if ($serversmrtg ne "") {

		open(OUTPUT, ">$datapath/serversmrtg");

		print OUTPUT "$serversmrtg\n$serversmrtg\n$time\n";

		close(OUTPUT);

	}

	if ($ircopsmrtg ne "") {

		open(OUTPUT, ">$datapath/ircopsmrtg");

		print OUTPUT "$ircopsmrtg\n$ircopsmrtg\n$time\n";

		close(OUTPUT);

		$ircopsmrtg=0; # raw 252 pas disp si pas opers

	}

	if ($globalusersplain ne "") {

		open(OUTPUT,">$datapath/globalusersplain");

		print OUTPUT "$globalusersplain";

		close(OUTPUT);

	}

	if ($channelsplain ne "") {

		open(OUTPUT, ">$datapath/channelsplain");

		print OUTPUT "$channelsplain";

		close(OUTPUT);

	}

	if ($serversplain ne "") {

		open(OUTPUT, ">$datapath/serversplain");

		print OUTPUT "$serversplain";

		close(OUTPUT);

	}

	if ($ircopsplain ne "") {

		open(OUTPUT, ">$datapath/ircopsplain");

		print OUTPUT "$ircopsplain";

		close(OUTPUT);

		$ircopsplain=0; # raw 252 pas disp si pas opers

	}

}

sub getinfo {

	foreach $s (keys %server) {

		print SOCKET "lusers * $s\n";

	}

	$time=time();

	if ($needlusers==1) {

		print SOCKET "lusers\n";

	}

}

sub daemonize () {

	chdir "/" or die "Can't chdir to /: $!";

	open STDIN, "/dev/null" or die "Can't read /dev/null: $!";

	open STDOUT, ">/dev/null" or die "Can't write to /dev/null: $!";

	defined (my $pid = fork) or die "Can't fork: $!";

	if ($pid) {

		if ($pidfile) {

			open(PIDFILE,">$pidfile");         

			print PIDFILE "$pid\n";                           

			close(PIDFILE);

		}

		exit;

	}

	setsid or die "Can't start a new session: $!";

	open STDERR, ">&STDOUT" or die "Can't dup stdout: $!";

}