File cyrus_ldap_quota.pl of Package cyrus-imapd

File cyrus_ldap_quota.pl of Package cyrus-imapd (Revision 12)

Currently displaying revision 12 , Show latest

#!/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.

# 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;
}