128 lines
4.0 KiB
Perl
128 lines
4.0 KiB
Perl
#!/usr/bin/perl
|
|
use Cyrus::IMAP::Admin;
|
|
|
|
# This script was created by Kevin J. Menard, Jr. <kmenard@wpi.edu>.
|
|
# It requires root privileges to write to a file in /etc. Best use is
|
|
# to set this up as cron job. Works for me. Hope it does for you.
|
|
# Any questions/complaints/praise/whatever, send 'em to the address
|
|
# above. -- 08/16/2001
|
|
|
|
|
|
# These are the variables you might want to tweak.
|
|
my $quota_attr = "mailQuota";
|
|
my $mail_attr = "mailRoutingAddress";
|
|
my $user = "cyrus";
|
|
my $passwd = "blah";
|
|
|
|
# These are the ones that you shouldn't have to.
|
|
my @entries = ();
|
|
my $index = 0;
|
|
my $counter = 0;
|
|
my $old_timestamp = 0;
|
|
my $timestamp = "199412161032Z";
|
|
|
|
# Open the /etc/cyrus_ldap_quota_time file; it's a long name, but
|
|
# shouldn't interfere with existing files :) This file contains 1 line,
|
|
# the generalized time format of the last time the script ran. This is
|
|
# used for the search, so we only update quotas that have been modified
|
|
# since then.
|
|
|
|
{
|
|
if (-e "/etc/cyrus_ldap_quota_time")
|
|
{
|
|
open(TIME_FILE, "/etc/cyrus_ldap_quota_time") or die "could not
|
|
open the time file: $!\n";
|
|
|
|
while(<TIME_FILE>) { $old_timestamp = $_; }
|
|
|
|
close(TIME_FILE);
|
|
}
|
|
|
|
# Now we deal with the case where the file doesn't exist, that is to
|
|
# say the first time the script was run.
|
|
|
|
unless ($old_timestamp == 0) { $timestamp = $old_timestamp; }
|
|
|
|
|
|
# Now that we have that information, we can overwrite the file with
|
|
# the new timestamp. Maybe this overkill, but this is only a
|
|
# temporary solution anyway.
|
|
|
|
open(TIME_FILE, ">/etc/cyrus_ldap_quota_time") or die "could not
|
|
create file: $!\n";
|
|
|
|
my @time = (localtime)[0..5];
|
|
|
|
printf TIME_FILE $time[5] + 1900;
|
|
printf TIME_FILE "%02d", $time[4] + 1;
|
|
for (my $i = 3; $i >= 0; $i--) { printf TIME_FILE "%02d", $time[$i];}
|
|
print TIME_FILE 'Z';
|
|
|
|
close(TIME_FILE);
|
|
}
|
|
|
|
|
|
# This is where we do the search and then parse the results into a
|
|
# useable form. In this case, an arry of hash entries.
|
|
{
|
|
# Okay, this very ugly line sets up the LDAP search, and the strips
|
|
# away the meaningless stuff. This could be prettier, but I didn't
|
|
# want to add a dependency for an LDAP module, and everyone should
|
|
# have LDAP search. The greps are just to make things simpler.
|
|
|
|
(my $query = "ldapsearch -x '(&(modifyTimestamp>=$timestamp)($quota_attr=*))' $quota_attr $mail_attr
|
|
| grep -v ^# | grep -v ^dn | grep -v ^version | grep -v ^search | grep -v ^result | grep -v ^\$") =~ s!\n!!;
|
|
|
|
# Now actually do the commands in the above line.
|
|
my $result = `$query`;
|
|
|
|
|
|
# Split the output into an array, one entry per line.
|
|
my @output = split(/\n/, $result);
|
|
|
|
# Now go through each line . . .
|
|
foreach (@output)
|
|
{
|
|
# Split on the attribute:value boundary.
|
|
(my $key, my $value) = split(/: /);
|
|
|
|
# Handle mailRoutingAddress; strip away everything after '@'.
|
|
if ($value =~ m!@!)
|
|
{
|
|
($value, undef) = split (/@/, $value);
|
|
}
|
|
|
|
# Add each doctored up attribute:value pair to the entries array.
|
|
$entries[$index]{$key} = $value;
|
|
|
|
# A crude hack to make sure each of the two attributes makes it
|
|
# into one of the entries array element.
|
|
if ($counter % 2)
|
|
{
|
|
$index++;
|
|
}
|
|
|
|
$counter++;
|
|
}
|
|
}
|
|
|
|
# Now here's the actual interaction with Cyrus IMAPd. It's all pretty
|
|
# self-explanatory.
|
|
{
|
|
my $imap = Cyrus::IMAP::Admin->new('localhost') or die "imap:
|
|
cannot connect to server: $!\n";
|
|
|
|
$imap->send(undef, undef, "LOGIN %s %s", $user, $passwd) or die
|
|
"could not send user:pass to the server: $!\n";
|
|
|
|
for (my $i = 0; $i <= $#entries; $i++)
|
|
{
|
|
$imap->setquota("user." . $entries[$i]{$mail_attr}, "STORAGE",
|
|
$entries[$i]{$quota_attr})
|
|
or die "imap: could not set quota for
|
|
user.$entries[$i]{$mail_attr}: $!\n";
|
|
}
|
|
|
|
$imap=undef;
|
|
}
|