#!/usr/bin/perl
# -*- perl -*-
#
# Copyright (C) 2004 Rune Nordbe Skillingstad
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; version 2 dated June,
# 1991.
#
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
#
# $Log$
#
#%# family=snmpauto
#%# capabilities=snmpconf

use strict;
use Net::SNMP;

my $DEBUG = 0;

my $host      = $ENV{host}      || undef;
my $port      = $ENV{port}      || 161;
my $community = $ENV{community} || "public";
my $timeout   = $ENV{timeout}   || 30;

if(defined $ARGV[0] and $ARGV[0] eq "snmpconf") {
	print "require 1.3.6.1.2.1.25.4.2.1.2. rdpclip\\.exe\n";
	exit 0;
}

if($0 =~ /^(?:|.*\/)snmp_([^_]+)_rdp_users$/) {
    $host  = $1;
    if($host =~ /^([^:]+):(\d+)$/) {
	$host = $1;
	$port = $2;
    }
} elsif(!defined($host)) {
    print "# Debug: $0 -- $1\n" if $DEBUG;
    die "# Error: couldn't understand what I'm supposed to monitor.";
}

my($session, $error) = Net::SNMP->session(
    -hostname  => $host,
    -community => $community,
    -port      => $port,
    -timeout   => $timeout
);

if(!defined($session)) {
    die "Croaking: $error";
}

if(defined $ARGV[0] and $ARGV[0] eq "config") {
    print "host_name $host\n";
    print "graph_title Number of RDP users
graph_args --base 1000 -l 0 
graph_vlabel number of RDP users
graph_scale no
graph_category system
graph_info This graph shows the number of remote users on the Windows system.
users.label users
users.draw LINE2
users.info Number of users logged in.
";
    exit 0;
}

print "users.value ", &get_users($session), "\n";

sub get_users {
    my $handle = shift;
    my $oid    = "1.3.6.1.2.1.25.4.2.1.2";
    
    print "# Getting single $oid...\n" if $DEBUG;
    
    my $response = $handle->get_table($oid);
    my $count = 0;
    
    if(!defined($response)) {
	return "";
    } else {
	foreach my $key (keys(%{$response})) {
	    if($response->{$key} =~ /^rdpclip\.exe$/i) {
		$count++;
	    }
	}
	return $count;
    }
}
