Проверка квоты пользователя на этапе RCPT

На одном из почтовых серверов появилась необходимость отшибать письма (до приема содержимого письма) для пользователей, у которых исчерпан отведенный им лимит дискового пространства. В качестве MTA на сервере используется Exim. Как известно стандартными средствами данного MTA такое не сделать, но в силу гибкости Exim,а можно выйти из положения написав скрипт, который будет проверять лимиты пользователей и посредством сокетов взаимодействовать с ним (см. в доках readsocket). Что и было реализовано в скрипте, приведенном немного ниже. Скрипт запускается в фоне и обслуживает запросы Exim,а на проверку превышения квоты пользователем. На входе скрипт имеет полный почтовый адрес пользователя, а на выход выдает одну из заранее определенных констант:

QUOTA_OK — квота не исчерпана;
QUOTA_EXCEEDED — квота исчерпана;
QUOTA_UNKNOWN — по какой-либо причине не удалось установить состояние квоты;
QUOTA_ERROR — произошла ошибка на стороне скрипта.

Вот собственно код скрипта на Perl:

#!/usr/bin/env perl
 
#
# Для работы требуется модуль DBD::Pg
#
 
use strict;
use warnings;
use Socket;
use POSIX;
use DBI;
 
my $work_dir = "/path/to/workdir";
my $sock_name = "/tmp/ckmq.sock";
my $log_file = "/path/to/log/ckmq_daemon.log";
my $pid_file = "/var/run/ckmq.pid";
 
my $SOCK_FD = -1;
my $srv_status = 1;
my $sock_addr = sockaddr_un($sock_name);
my $max_forks = 15;
my $fork_count = 0;
my $pid = -1;
my $dbh = -1;
 
my $db_host = '127.0.0.1';
my $db_user = 'dbuser';
my $db_pass = 'dbpass';
my $db_name = 'dbname';
 
#--------------------------------------------------------------------
 
# Переключаемся в режим демона
defined($pid = fork())          || die "can't fork: $!\n";
exit(0) if $pid;
(setsid() != -1)                || die "can't start new session: $!\n";
 
open(PID_FILE, '>', $pid_file)  || die "cat't write pid: $!\n";
print(PID_FILE $$);
close(PID_FILE);
 
open(LOG_FILE, '>>', $log_file) || die "can't write $log_file: $!\n";
open(STDIN, '<', '/dev/null')   || die "can't read /dev/null: $!\n";
open(STDOUT, '>', '/dev/null')  || die "can't write /dev/null: $!\n";
open(STDERR, '>', '/dev/null')  || die "can't write /dev/null: $!\n";
 
$SIG{INT} = \&sig_int;
$SIG{TERM} = \&sig_int;
$SIG{CHLD} = \&sig_chld;
 
socket(SOCK_FD, AF_UNIX, SOCK_STREAM, 0) || &error_log("$!");
bind(SOCK_FD, $sock_addr)                || &error_log("$!");
listen(SOCK_FD, SOMAXCONN)               || &error_log("$!");
chmod(0777, $sock_name)                  || &error_log("can't chmod: $!");
chdir($work_dir)                         || &error_log("can't chdir: $!");
 
&notice_log("check quota daemon started (pid: $$)");
while($srv_status){
  my $CLIENT_FD = -1;
 
  if (!accept(CLIENT_FD, SOCK_FD)){ next; }
 
  if ($fork_count >= $max_forks){
    send(CLIENT_FD, "QUOTA_ERROR", 0);
    shutdown(CLIENT_FD, 2);
    close(CLIENT_FD);
    &notice_log('fork_max exceeded');
    next;
  }
 
  $fork_count++;
  if (fork() == 0){
    close(SOCK_FD);
 
    $SIG{INT} = 'IGNORE';
    $SIG{TERM} = 'DEFAULT';
    $SIG{CHLD} = 'IGNORE';
    setsockopt(CLIENT_FD, SOL_SOCKET, SO_RCVTIMEO, pack('L!L!', 5, 0));
    &client_work($CLIENT_FD);
 
    shutdown(CLIENT_FD, 2);
    close(CLIENT_FD);
        
    exit(0);
  }
  close(CLIENT_FD);
}
 
if ($SOCK_FD != -1){
  shutdown(SOCK_FD, 2);
  close(SOCK_FD);
}
unlink($sock_name);
 
&notice_log('check quota daemon stopped');
 
exit(0);
 
#--------------------------------------------------------------------
 
sub error_log {
  my $message = shift;
  my ($sec, $min, $hour, $mday, $mon) = localtime();
  print(LOG_FILE "$mon $mday $hour:$min:$sec error: $message\n");
 
  if ($SOCK_FD != -1){
    shutdown(SOCK_FD, 2);
    close(SOCK_FD);
  }
 
  exit(0);
}
 
sub notice_log {
  my $message = shift;
  my ($sec, $min, $hour, $mday, $mon) = localtime();
  print(LOG_FILE "$mon $mday $hour:$min:$sec notice: $message\n");
}
 
sub sig_int {
  $srv_status = 0;
  shutdown(SOCK_FD, 2);
  close(SOCK_FD);
  $SOCK_FD = -1;
}
 
sub sig_chld {
  my $pid = -1;
 
  while (($pid = waitpid(-1, WNOHANG)) > 0){
    $fork_count--;
  }
}
 
sub client_work {
  my $CLIENT_FD = shift;
  my $buffer = '';
  my $user_name = '';
  my @email = ();
  my ($quota_size, $mail_root) = (0,0);
 
  if (!defined(recv(CLIENT_FD, $buffer, 255, 0))){ return; }
  if (!$buffer){ return; }
 
  &db_connect();
 
  chomp($buffer);
  @email = split(/@/, $buffer);
  #print("Name: ", $email[0], "\nDomain: ", $email[1], "\n");
  if (defined($buffer = &db_get_user_from_alias($email[0], $email[1]))){
    @email = split(/@/, $buffer);
  }
 
  ($quota_size, $mail_root) = &db_get_user_quota($email[0], $email[1]);
  $quota_size = $quota_size * 1024;
 
  $buffer = "$mail_root/$email[1]/$email[0]/Maildir/maildirsize";
  if (open(MDSF, "<", $buffer)){
    my $total_size = 0;
 
    $buffer = <MDSF>;
    while (<MDSF>){
        my ($tmp) = split(/ /);
        $total_size += $tmp;
    }
    #&notice_log("Quota size: $quota_size");
    #&notice_log("Total maildir size: $total_size");
    send(CLIENT_FD, ($quota_size == 0) || ($total_size < $quota_size) ? 'QUOTA_OK' : 'QUOTA_EXCEEDED', 0);
 
    close(MDSF);
  } else {
    send(CLIENT_FD, 'QUOTA_UNKNOWN', 0);
  }
 
  &db_disconnect();
 
  undef;
}
 
 
#--------------------------------------------------------------------
sub db_connect {
  $dbh = DBI->connect("dbi:Pg:host=$db_host;dbname=$db_name", $db_user, $db_pass, {AutoCommit => 0});
}
 
sub db_disconnect {
  $dbh->disconnect();
}
 
# Функция возвращает массив из двух элементов: путь до почтового ящика и размер квоты
sub db_get_user_data {
  my ($uname, $dname) = @_;
  my $stm = -1;
  my @row = ();
 
  @row = $dbh->selectrow_array('SELECT "users_tb"."quota", "users_tb"."homedir" FROM "users_tb"
    INNER JOIN "domains_tb" ON ("users_tb"."domain_id" = "domains_tb"."id")
    WHERE "username" = $$' . $uname . '$$ AND
      "domains_tb"."domainname" = $$' . $dname . '$$ LIMIT 1;');
 
  return ($#row == 2) ? @row : -1;
}
 
# Функция преобразовывает алиас в реальное имя пользователя
sub db_get_user_from_alias {
  my ($uname, $dname) = @_;
  my $stm = -1;
  my @row = ();
 
  @row = $dbh->selectrow_array('SELECT "aliases_tb"."mailaddr" FROM "aliases_tb"
      INNER JOIN "domains_tb" ON ("aliases_tb"."domain_id" = "domains_tb"."id")
      WHERE "aliases_tb"."aliasname" = $$'.$uname.'$$ AND
        "domains_tb"."domainname" = $$'.$dname.'$$ AND
        "aliases_tb"."active" = $$true$$ AND "domains_tb"."active" = $$true$$');
 
  return $row[0];
}

Данный скрипт написан для почтовой системы, описанной в данной статье. Думаю, переделать под какую-либо другую конфигурацию не составит особого труда. Для автоматического запуска во время старта системы можно добавить его в автозагрузку, например, создать файл в /usr/local/etc/rc.d с таким содержимым:

#!/bin/sh
 
# PROVIDE: ckmq
# BEFORE: exim
# KEYWORD: shutdown
#
# Add the following lines to /etc/rc.conf to enable Check Mail Quota Daemon:
# ckmq_enable (bool):    Set to "NO" by default.
#                       Set it to "YES" to enable ckmq.
#
 
. /etc/rc.subr
 
name="ckmq"
rcvar=ckmq_enable
 
load_rc_config ${name}
 
: ${ckmq_enable="NO"}
 
pidfile="/var/run/${name}.pid"
command="/path/to/script"
 
run_rc_command "$1"

Правило для Exim,а будет выглядить так:

deny message = Quota size exceeded
       domains = +local_domains
       condition = ${if eq{QUOTA_EXCEEDED}{${readsocket{/tmp/ckmq.sock}{$local_part@$domain}{3s}{}{false}}}{yes}{no}}

Поскольку скрипт не может узнать размер принимаемого сообщения, то в MTA или MDA необходимо отключить проверку квот, чтобы допустить переполнение почтового ящика. В остальном скрипт довольно прост для понимания, поэтому я не буду подробно расписывать каждую строчку. Если есть вопросы по работе скрипта, то задавайте их в комменты или в форум.

Проверка квоты пользователя на этапе RCPT

Добавить комментарий