View file File name : rbldstats Content :#!/usr/bin/perl ########################################################################## # rbldstats - A script to query rbld and print out statistics # Copyright 2006, Bluehost, Inc. # # Authors and Contributers: # # Spencer Candland <spencer@bluehost.com> # Erick Cantwell <erick@bluehost.com> # # http://www.bluehost.com # https://github.com/bluehost/rbld # ########################################################################## # # This file is part of rbld # # Rbld 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; either version 2 of the License, or # (at your option) any later version. # # 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., 59 Temple Place, Suite 330, Boston, MA02111-1307, USA. # ########################################################################## use strict; use warnings; use IO::Socket; use IO::Select; use Data::Dumper; use YAML::Syck qw(LoadFile);; use Getopt::Long; my $config_file = '/etc/rbld.conf'; my $socket_path; my $settings; # Get cli options GetOptions ( 'c|config=s' => \$config_file, 's|socket=s' => \$socket_path, 'h|help' => \&help, ) || &help; $Data::Dumper::Indent = 1; $Data::Dumper::Useqq = 1; sub main { if ($socket_path) { $settings->{sock_path} = $socket_path; } else { $settings = LoadFile($config_file); } unless ($settings->{sock_path}) { print STDERR "No socket path available to query. Aborting...\n"; exit 1; } my ($msg, $data, $buffer, $sock); $msg = "STATS\n"; $sock = IO::Socket::UNIX->new($settings->{sock_path}) or die "Socket: $!\n"; eval { syswrite($sock, $msg, length($msg)); $data = ''; $data .= $buffer while sysread($sock, $buffer, 8192); $sock->shutdown(2); }; if ($@) { print $@; } print "RECV> $data\n"; } main(); sub help { print <<EOF; $0: Use this script to show RBLD stats usage: -c|--config Path to rbld configuration file to load socket path (default is /etc/rbld.conf) -s|--socket Alternatively, you can specify the socket path directly -h|--help Help EOF exit; }