#!/usr/bin/perl

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

#

# Copyright (c) 1997 by Balthasar Indermuehle <balthasar@indermuehle.com>

# Currently supports Linux 2.0.25

# All Rights Reserved.

# 

# See the file COPYRIGHT in the distribution for the exact terms.

# Based on the mrtg-ping-probe script Copyright 1997 by Peter W. Osel <pwo@guug.de>

#

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

require 5.003;

use Getopt::Std;

use File::Basename;

use Config;

$ProgName = "mrtg-blast";

$Usage = "Usage: $ProgName hostname portnumber packets\nTypically: $ProgName my.host.com 9 500\n";

# Parse Command Line:

die $Usage unless getopts('');

if (@ARGV > 3) {

	print STDERR "$ProgName: WARNING: ignoring but the first three arguments\n";

	}

if (@ARGV < 3) {

	print STDERR "$ProgName ERROR:\nPlease enter a hostname port number and packets to blast (e.g. 9 or 21) as argument\n";

	print STDERR "$Usage";

	exit(1);

	}

($HostToBlast, $PortToBlast, $Packets) = @ARGV;

($blastres) = blast($HostToBlast, $PortToBlast, $Packets);

# The external mrtg probe returns up to 4 lines of output:

#	1. Line: current state of the 'incoming bytes counter'

#	2. Line: current state of the 'outgoing bytes counter'

#	3. Line: string, telling the uptime of the target.

#	4. Line: telling the name of the target.

# We leave out line 3, and 4.

print "$blastres\n$blastres\n";

exit(0);

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

# Do the actual blasting 

#

sub blast {

	my($host, $port, $packets) = @_;

	# The following path to tcpblast needs to be adjusted for your system

	unless (open(BLAST, "/home/netadmin/mrtg-2.1/tcpblast -r -t $host $packets -p$port|")) {

		print STDERR "${ProgName}: FATAL: Can't open tcpblast: $!";

		exit(1);

		}

	while (<BLAST>) {

		# Handle output from 'real' tcpblast which looks like this:

		# 100 KB in 1979 msec  =  413946.4 b/s  =  51743.3 B/s  =  50.5 KB/s

		close(BLAST), return("$4.$5") if /^(\d+) KB in (\d+) msec  =  (\d+)\.\d b\/s  =  (\d+)\.\d B\/s  =  (\d+)\.\d KB\/s/;

		}

	#Could not get blast results, link may be down, return 0

	close(BLAST), return(0);

}