postfix milter实例

如果收件人超过3个且信件内容大于10个字节，则把来信静悄悄丢掉。

需要安装 sendmail-devel。

main.cf 配置如下：
smtpd_milters = inet:127.0.0.1:5888

hdmilter.c 如下：

# define SMFI_VERSION   2

#include <fcntl.h>
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sysexits.h>
#include <termios.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <syslog.h>

#include <libmilter/mfapi.h>

struct mlfiPriv {
    int nrcpts;
    size_t bsize;
};


#define VERTEXT "HD-Filter-Version"
#define VERSION "0.1"

#define MLFIPRIV ((struct mlfiPriv *) smfi_getpriv(ctx))

extern sfsistat mlfi_cleanup(SMFICTX *, sfsistat);


sfsistat mlfi_envfrom(SMFICTX *ctx, char **envfrom) {
    struct mlfiPriv *priv;

    syslog(LOG_DAEMON|LOG_INFO, "----------------------------------------------");

    priv = (struct mlfiPriv *) malloc(sizeof *priv);
    if (!priv) {
       
        syslog(LOG_DAEMON|LOG_INFO, "envfrom: %s", "oom");
        return SMFIS_TEMPFAIL;
    }
    priv->nrcpts = 0;
    priv->bsize = 0;

   
    smfi_setpriv(ctx, priv);

   
    {
        char **env = envfrom;
        while (*env) {
            syslog(LOG_DAEMON|LOG_INFO, "envfrom: %s", *env);
            env++;
        }
    }
    return SMFIS_CONTINUE;
}

sfsistat mlfi_envrcpt(SMFICTX *ctx, char **envrcpt) {
   struct mlfiPriv *priv = MLFIPRIV;

    priv->nrcpts++;
    {
        char **env = envrcpt;
        while (*env) {
            syslog(LOG_DAEMON|LOG_INFO, "envrcpt: %s", *env);
            env++;
        }
    }
    return SMFIS_CONTINUE;
}


sfsistat mlfi_header(SMFICTX *ctx, char *headerf, char *headerv) {
    struct mlfiPriv *priv = MLFIPRIV;
    syslog(LOG_DAEMON|LOG_INFO, "header: %s => %s", headerf, headerv);
    return SMFIS_CONTINUE;
}

sfsistat mlfi_eoh(SMFICTX *ctx) {
    syslog(LOG_DAEMON|LOG_INFO, "eoh: %s", "...");
    return SMFIS_CONTINUE;
}

sfsistat mlfi_body(SMFICTX *ctx, u_char *bodyp, size_t bodylen) {
    struct mlfiPriv *priv = MLFIPRIV;
    priv->bsize += bodylen;

   
    syslog(LOG_DAEMON|LOG_INFO, "body: +len %d", bodylen);
    return SMFIS_CONTINUE;
}

sfsistat mlfi_eom(SMFICTX *ctx) {
    struct mlfiPriv *priv = MLFIPRIV;

    syslog(LOG_DAEMON|LOG_INFO, "eom: nrcpts %d bsize %d",
        priv->nrcpts, priv->bsize);

    if (priv->nrcpts > 3 && priv->bsize > 10)
        return mlfi_cleanup(ctx, SMFIS_DISCARD);
    else
        return mlfi_cleanup(ctx, SMFIS_CONTINUE);
}

sfsistat mlfi_close(SMFICTX *ctx) {
    syslog(LOG_DAEMON|LOG_INFO, "close: ...");
    return mlfi_cleanup(ctx, SMFIS_ACCEPT);
}

sfsistat mlfi_abort(SMFICTX *ctx) {
    syslog(LOG_DAEMON|LOG_INFO, "abort: ...");
    return mlfi_cleanup(ctx, SMFIS_CONTINUE);
}

sfsistat mlfi_cleanup(SMFICTX *ctx, sfsistat rc) {
    struct mlfiPriv *priv = MLFIPRIV;

    if (priv) {
        free(priv);
        smfi_setpriv(ctx, NULL);
    }

    return(rc);
}

struct smfiDesc smfilter = {
    "HDFilter",   
    SMFI_VERSION,   
    SMFIF_NONE,   
    NULL,       
    NULL,       
    mlfi_envfrom,   
    mlfi_envrcpt,   
    mlfi_header,   
    mlfi_eoh,   
    mlfi_body,   
    mlfi_eom,   
    mlfi_abort,   
    mlfi_close   
};


int main(int argc, char *argv[]) {
    int c, fd;
    int fg = 0;
    const char *args = "p:f";

   
    while ((c = getopt(argc, argv, args)) != -1) {
        switch (c) {
          case 'p':
            if (optarg == NULL || *optarg == '\0') {
                (void) fprintf(stderr, "Illegal conn: %s\n",
                           optarg);
                exit(EX_USAGE);
            }
            (void) smfi_setconn(optarg);
            break;

        case 'f':
            fg = 1;
            break;

        }
    }

    if (smfi_register(smfilter) == MI_FAILURE) {
        fprintf(stderr, "smfi_register failed\n");
        exit(EX_UNAVAILABLE);
    }

    if (fg) {
        return smfi_main();
    }

    if (fork() == 0) {
        close(STDIN_FILENO);
        close(STDOUT_FILENO);
        close(STDERR_FILENO);

        fd = open("/dev/tty", O_RDWR);
        if (fd >= 0) {
            ioctl(fd, TIOCNOTTY, 0);
            close(fd);
        }

        return smfi_main();
    }
    return 0;
}

Makefile如下：

all: hdmilter

hdmilter: hdmilter.c
    gcc -g -o hdmilter hdmilter.c /usr/lib64/libmilter.a /usr/lib64/libsm.a /usr/lib64/libsmutil.a -lpthread

clean:
    -@rm hdmilter

test:
    ./hdmilter -p inet:5888

注意priv结构体必须在 envfrom 分配，如果在helo或connect阶段分配，在close阶段释放，则可能会导致其他回调函数拿不到priv结构体，因为同一个tcp连接可能发送多个消息，helo/connect只回调一次，而 envfrom/envrcpt/.../close过程会因消息的个数而回调多次