209 lines
6.6 KiB
Perl
209 lines
6.6 KiB
Perl
#!/usr/bin/perl -w
|
|
#==============================================================================
|
|
# lat-quota
|
|
# =========
|
|
# 0.9.0 (2004-09-08)
|
|
# (c)2003-2004 Altiplano bvba
|
|
#==============================================================================
|
|
package esmith;
|
|
use strict;
|
|
use esmith::db;
|
|
use Getopt::Long;
|
|
use Pod::Usage;
|
|
my %conf;
|
|
my %accounts;
|
|
tie %conf, 'esmith::config';
|
|
tie %accounts, 'esmith::config', '/home/e-smith/db/accounts';
|
|
my ($Hlp, $Cml, $Inp);
|
|
|
|
#==============================================================================
|
|
# Main
|
|
#==============================================================================
|
|
# Analyze commandline options
|
|
GetOptions ("help" => \$Hlp,
|
|
"command-line=s" => \$Cml,
|
|
"input-file=s" => \$Inp);
|
|
if ( $Hlp ) { &PrintPod(9); exit; }
|
|
|
|
# We need one argument or the other, but not both
|
|
if (($Cml && $Inp) || (! $Cml && ! $Inp))
|
|
{ &PrintPod(1); exit; }
|
|
|
|
my @records;
|
|
if ($Inp) {
|
|
open(LIST,"< $Inp") || die "Can't find $Inp.\n";
|
|
@records = grep(!/(^\s*#)|(^\s*$)/,<LIST>);
|
|
close(LIST); }
|
|
elsif ($Cml) { @records=($Cml); }
|
|
else { &PrintPod(1); exit; }
|
|
|
|
&ExpandWildCard; # Check for wildcards and expand if necessary
|
|
|
|
# Process each user
|
|
foreach my $record (@records)
|
|
{
|
|
my @fields=split(/\|/,$record);
|
|
for (my $cnt=0; $cnt <= $#fields; ++$cnt) { for ($fields[$cnt]) { s/^\s+//; s/\s+$//; }}
|
|
|
|
if ( @fields == 3) # All three arguments must be given
|
|
{
|
|
my $Soft = &CalculateKB($fields[1]);
|
|
my $Hard = &CalculateKB($fields[2]);
|
|
if ((db_get(\%accounts, $fields[0])) &&
|
|
(db_get_type(\%accounts, $fields[0]) eq "user"))
|
|
{
|
|
print "Setting quota for user $fields[0] ($fields[1]/$fields[2])\n";
|
|
db_set_prop(\%accounts, $fields[0], "MaxBlocksSoftLim", $Soft);
|
|
db_set_prop(\%accounts, $fields[0], "MaxBlocks", $Hard);
|
|
system("/sbin/e-smith/signal-event", "user-modify", $fields[0]) == 0
|
|
or die ("Error occurred while updating quota values for user $fields[0].\n");
|
|
}
|
|
else { print "User $fields[0] does not exist on this server.\n"; }
|
|
}
|
|
else { print "Illegal number of arguments for user $fields[0]\n\a";}
|
|
}
|
|
|
|
#==============================================================================
|
|
# Subroutines
|
|
#==============================================================================
|
|
# Convert from Gigabytes or Megabytes to true Kilobytes
|
|
sub CalculateKB {
|
|
if ( $_[0] =~ m/M$|G$/ ) {
|
|
if ($& eq "M") { return $`*1024 }
|
|
if ($& eq "G") { return $`*1024*1024 }
|
|
}
|
|
else { return $_[0]; }
|
|
}
|
|
#==============================================================================
|
|
# Test for wildcards in the username. If any wildecards are found, the array
|
|
# @records is expanded with the user names that meet the conditions.
|
|
sub ExpandWildCard {
|
|
my $ctrec = 0;
|
|
foreach my $record (@records)
|
|
{
|
|
my @fld=split(/\|/,$record);
|
|
for (my $cnt=0; $cnt <= $#fld; ++$cnt) { for ($fld[$cnt]) { s/^\s+//; s/\s+$//; }}
|
|
|
|
if ($fld[0] =~ /\*|\?/) { # Does it contain the wildcards?
|
|
$fld[0] =~ s/\*/\.\*/g; # Replace * with .* to allow for grep.
|
|
$fld[0] =~ s/\?/\./g; # Replace ? with . to allow for grep.
|
|
|
|
open USRS, "</home/e-smith/db/accounts" or die "Can't open /home/e-smith/db/accounts: $!";
|
|
my @match = grep /^$fld[0]\=user\|/i, <USRS>;
|
|
close(USRS);
|
|
|
|
my $cu = 0;
|
|
foreach my $tst (@match) {
|
|
$tst =~ /\=/; $tst = $`;
|
|
for (my $cnt=1; $cnt <= $#fld; ++$cnt) { $tst = $tst." | ".$fld[$cnt]; };
|
|
if ($cu == 0 ) {
|
|
$records[$ctrec] = $tst;
|
|
$cu =1;
|
|
}
|
|
else {
|
|
push(@records, $tst);
|
|
}
|
|
}
|
|
}
|
|
++$ctrec;
|
|
}
|
|
}
|
|
#==============================================================================
|
|
# Print the pod text as a help screen
|
|
sub PrintPod {
|
|
my ($verbose, $message) = @_;
|
|
pod2usage(-verbose => $verbose, -message => $message, -exitval => 64);
|
|
}
|
|
|
|
#==============================================================================
|
|
|
|
=pod
|
|
|
|
=head1 NAME
|
|
|
|
B<lat-quota> - The lazy administrator's tool to set user quota
|
|
|
|
=head1 DESCRIPTION
|
|
|
|
Sets the disk quota for individual users on Mitel's SME servers (5.x/6.x).
|
|
This tool is functionally equivalent to the 'Quota management' option in the server-manager, but can be run from the command line or called from an other script.
|
|
It allows you, for example, to set the disk quota for a large number of accounts in a batch process or on a remote machine via an ssh console.
|
|
|
|
See F</usr/doc/lazy-admin-tools/example.quota> for the format of the input file.
|
|
|
|
=head1 SYNOPSIS
|
|
|
|
B<lat-quota> -c "user | softlimit | hardlimit"
|
|
|
|
B<lat-quota> -i /path/to/quota.list
|
|
|
|
=head1 OPTIONS
|
|
|
|
The following options are supported:
|
|
|
|
=over 4
|
|
|
|
=item B<-c "Arguments">, B<--command-line="Arguments">
|
|
|
|
Take arguments from the command line.
|
|
See the 'Arguments' section below for the various arguments that are accepted.
|
|
|
|
=item B<-h>, B<--help>
|
|
|
|
Extended help for this tool
|
|
|
|
=item B<-i FILE>, B<--input-file=FILE>
|
|
|
|
Use the information from F<FILE> to set the disk quota for individual users.
|
|
|
|
=back
|
|
|
|
=head2 Arguments:
|
|
|
|
user* : Must be an existing user name. Wildcards (* and ?)
|
|
are accepted.
|
|
softlimit* : Limit with grace period in kilobytes, unless
|
|
followed by M (megabytes) or G (gigabytes).
|
|
hardlimit* : Absolute limit in kilobytes, unless followed by
|
|
M (megabytes) or G (gigabytes).
|
|
* mandatory field
|
|
|
|
=head1 EXAMPLES
|
|
|
|
B<lat-quota -c "harry | 800M | 1G">
|
|
|
|
Sets the quota for user 'harry' to 800 MByte (soft limit) and 1 GByte (hard limit).
|
|
|
|
B<lat-quota -i /root/quota.list>
|
|
|
|
Sets the quota for the users defined in F</root/quota.list>.
|
|
Refer to F</usr/doc/lazy-admin-tools/example.quota> for an example of an input file.
|
|
|
|
B<lat-quota -c "* | 100M | 120M">
|
|
|
|
Sets the qquota for all users to 100M / 120M.
|
|
|
|
B<lat-quota -c "hermione | 0 | 0">
|
|
|
|
Turns off quota for user 'hermione'.
|
|
|
|
=head1 SEE ALSO
|
|
|
|
lat-group(8), lat-pseudonyms(8), lat-ibays(8), lat-users(8), lat-domains(8), lat-hosts(8), lat-procmail(8), lat-pptp(8), lat-dump(8)
|
|
|
|
=head1 VERSION
|
|
|
|
Version 0.9.0 (2004-09-08). The latest version is hosted at B<http://www.contribs.org/contribs/mblotwijk/>
|
|
|
|
=head1 COPYRIGHT
|
|
|
|
(c)2003-2004, Altiplano bvba (B<http://www.altiplano.be>). Released under the terms of the GNU license.
|
|
|
|
=head1 BUGS
|
|
|
|
Please report bugs to <Bugs@Altiplano.Be>
|
|
|
|
=cut
|
|
|
|
#==============================================================================
|