1 files changed, 113 insertions, 0 deletions
diff --git a/src/modules/rlm_counter/rad_counter b/src/modules/rlm_counter/rad_counter
new file mode 100755
index 0000000..8de31a1
--- /dev/null
+++ b/src/modules/rlm_counter/rad_counter
@@ -0,0 +1,113 @@
+#!/usr/bin/perl
+#
+#	$Id$
+#
+use warnings ;
+use GDBM_File ;
+use Fcntl ;
+use Getopt::Long;
+use File::Basename;
+
+my $user = '';
+my $divisor = 1;
+my $reset = 0;
+my $match = '.*';
+my $help = 0;
+
+#
+#  This should be fixed...
+#
+$filename = '';
+
+sub show_help {
+    my $progname = basename($0);
+    print <<EOF;
+Usage: $progname --file=<counter filename> [OPTION...]
+Query and maintain FreeRADIUS rlm_counter DB file.
+
+Arguments:
+--file=<filename>               Counter DB filename.
+
+Options:
+--user=<username>               Information for specific user.
+--match=<regexp>                Information for matching users.
+--reset=<number>                Reset counter to <number>.
+                                If divisor is set use it,
+                                else <number> means seconds.
+--help                          Show this help screen.
+--(hours|minutes|seconds)       Specify information divisor.
+EOF
+    exit 0;
+}
+
+#
+#  Print out only one user,
+#
+#  Or specify printing in hours, minutes, or seconds (default)
+#
+GetOptions ('user=s'  => \$user,
+	    'match=s' => \$match,
+	    'file=s'  => \$filename,
+	    'reset=i' => \$reset,
+	    'help'    => \$help,
+	    'hours'   => sub { $divisor = 3600 },
+	    'minutes' => sub { $divisor = 60 },
+	    'seconds' => sub { $divisor = 1 } );
+
+show_help if ($help || $filename eq '');
+
+#
+#  Open the file.
+#
+if ($reset){
+    my $db = tie(%hash, 'GDBM_File', $filename, O_RDWR, 0666) or die "Cannot open $filename: $!\n";
+}else{
+    my $db = tie(%hash, 'GDBM_File', $filename, O_RDONLY, 0666) or die "Cannot open $filename: $!\n";
+}
+
+#
+#  If given one name, give the seconds
+#
+if ($user ne '') {
+    if (defined($hash{$user})){
+	print $user, "\t\t", int ( unpack('L',$hash{$user}) / $divisor), "\n";
+	if ($reset){
+	    my $uniqueid = (unpack('L A32',$hash{$user}))[1];
+	    $hash{$user} = pack('L A32',$reset * $divisor,$uniqueid);
+	    print $user, "\t\t", "Counter reset to ", $reset * $divisor, "\n";
+	}
+    }else{
+	print $user, "\t\t", "Not found\n";
+    }
+
+    undef $db;
+    untie %hash;
+    exit 0;
+}
+
+#
+#  This may be faster, but unordered.
+#while (($key,$val) = each %hash) {
+#
+foreach $key (sort keys %hash) {
+    #
+    #  These are special.
+    next if ($key eq "DEFAULT1");
+    next if ($key eq "DEFAULT2");
+
+    #
+    #  Allow user names matching a regex.
+    #
+    next if ($key !~ /$match/);
+
+    #
+    #  Print out the names...
+    print $key, "\t\t", int ( unpack('L',$hash{$key}) / $divisor), "\n";
+    if ($reset){
+	my $uniqueid = (unpack('L A32',$hash{$key}))[1];
+	$hash{$key} = pack('L A32',$reset * $divisor,$uniqueid);
+	print $key, "\t\t", "Counter reset to ", $reset * $divisor, "\n";
+    }
+}
+undef $db;
+untie %hash;