checkquota.in 3.26 KB
Newer Older
1 2 3
#!/usr/bin/perl -wT
#
# Copyright (c) 2000-2005 University of Utah and the Flux Group.
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# 
# {{{EMULAB-LICENSE
# 
# This file is part of the Emulab network testbed software.
# 
# This file is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or (at
# your option) any later version.
# 
# This file 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 Affero General Public
# License for more details.
# 
# You should have received a copy of the GNU Affero General Public License
# along with this file.  If not, see <http://www.gnu.org/licenses/>.
# 
# }}}
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
#
use English;
use Getopt::Std;
    
#
# Parse an ns file. Since the parser runs arbitrary NS file for the user,
# this cannot be safely done on boss without jumping through huge hoops
# to secure tcl and the DB. Yuck! So, instead of running the parser on boss,
# we run it over on ops. This first version operates like this:
#
# NB: This script is setuid.
#
sub usage()
{
    print STDOUT "Usage: checkquota [-d] <user>\n";
    exit(-1);
}
my $optlist  = "d";
my $debug    = 0;

#
# Configure variables
#
my $TB       = "@prefix@";
my $TBOPS    = "@TBOPSEMAIL@";
48
my $FS       = "@FSNODE@";
49
my $TESTMODE = @TESTMODE@;
50
my $FSLIST   = "@FS_WITH_QUOTAS@";
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71

my $QUOTACMD = "/usr/bin/quota";
my $SSHTB    = "/usr/testbed/bin/sshtb";

# Locals
my $user;
my $dbuid;
my $overquota = 0;

#
# Turn off line buffering on output
#
$| = 1;

#
# Untaint the path
# 
$ENV{'PATH'} = "$TB/bin:$TB/sbin:/bin:/usr/bin:/sbin:/usr/sbin";
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

exit(0)
72
    if ($TESTMODE || $FSLIST eq "");
73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119

if ($EUID != 0) {
    # We don't want to run this script unless its the real version.
    die("*** $0:\n".
	"    Must be root! Maybe its a development version?\n");
}

# This script is setuid, so please do not run it as root. Hard to track
# what has happened.
if ($UID == 0) {
    die("*** $0:\n".
	"    Please do not run this as root! Its already setuid!\n");
}

#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use libdb;
use libtestbed;

#
# Parse command arguments. Once we return from getopts, all that should
# left are the required arguments.
#
%options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (defined($options{"d"})) {
    $debug = 1;
}
if (@ARGV != 1) {
    usage();
}
$user = $ARGV[0];

#
# Must taint check!
#
if ($user =~ /^([-\w]+)$/) {
    $user = $1;
}
else {
    die("Bad data in argument: $user");
}
   
120 121 122 123 124
#
# Convert to a uid since fs node may not have our user names
#
my $userid = getpwnam($user);

125 126 127 128 129 130 131 132
#
# Must flip to real root for the ssh.
# 
$UID = 0;

#
# We invoke the quota command and look at the results.
# 
133
open(QUOTA, "$SSHTB -host $FS $QUOTACMD -q -l $userid |") or
134
    die("*** $0:\n".
135
	"    Could not invoke $QUOTACMD on $FS!\n");
136 137

while (<QUOTA>) {
138 139 140 141
    if (($_ =~ /limit reached on ([-\w\/]*)$/) ||
	($_ =~ /grace period on ([-\w\/]*)$/) ||
	($_ =~ /Over file quota on ([-\w\/]*)$/) ||
	($_ =~ /Over block quota on ([-\w\/]*)$/)) {
142 143 144 145 146 147
	print STDOUT "*** Disk Quota exceeded on $1\n";
	$overquota++;
    }
}
close(QUOTA);
exit($overquota);