vdelivermail.c

相当优秀的 UNIX 进程管理工具

/*
 * $Id: vdelivermail.c,v 1.11.2.1 2004/05/27 00:31:09 tomcollins Exp $
 * Copyright (C) 1999-2003 Inter7 Internet Technologies, Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
 */

/* include files */
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <fcntl.h>
#include <dirent.h>
#include <errno.h>
#include <time.h>
#include <signal.h>
#include "config.h"
#include "vpopmail.h"
#include "vauth.h"
#include "maildirquota.h"
#ifdef MAKE_SEEKABLE
#include "seek.h"
#endif

/* Globals */
#define AUTH_SIZE 300
char TheUser[AUTH_SIZE];
char TheUserFull[AUTH_SIZE];
char TheDomain[AUTH_SIZE];
char TheDomainDir[AUTH_SIZE];
uid_t TheDomainUid;
gid_t TheDomainGid;
char TheDir[AUTH_SIZE];
char CurrentDir[AUTH_SIZE];
char DeliveredTo[AUTH_SIZE];
struct vqpasswd *vpw;
off_t message_size = 0;
char bounce[AUTH_SIZE];
int CurrentQuotaSizeFd;

#ifdef QMAIL_EXT
/* the User with '-' and following chars out if any */
char TheUserExt[AUTH_SIZE]; 
#endif

#define FILE_SIZE 156
char hostname[FILE_SIZE];
char loop_buf[FILE_SIZE];

#define MSG_BUF_SIZE 5000
char msgbuf[MSG_BUF_SIZE];

#define BUFF_SIZE 300
int fdm;

#define QUOTA_WARN_PERCENT 90

/* from qmail's wait.h for run_command() */
#define wait_exitcode(w) ((w) >> 8)

/* Forward declarations */
int process_valias(void);
int is_delete(char *deliverto);
int is_bounce(char *deliverto);
void get_arguments(int argc, char **argv);
off_t get_message_size();
int deliver_mail(char *address, char *quota);
int check_forward_deliver(char *dir);
off_t count_dir(char *dir_name);
int is_looping( char *address );
void run_command(char *prog);
void checkuser(void);
void usernotfound(void);
int is_loop_match( char *dt, char *address);
int deliver_quota_warning(const char *dir, const char *q);

static char local_file[156];
static char local_file_new[156];

/* 
 * The email message comes in on file descriptor 0 - stanard in
 * The user to deliver the email to is in the EXT environment variable
 * The domain to deliver the email to is in the HOST environment variable
 */
int main(int argc, char **argv)
{
    /* get the arguments to the program and setup things */
    get_arguments(argc, argv);
 
#ifdef VALIAS
    /* process valiases if configured */
    if ( process_valias() == 1 ) {
        printf("vdelivermail: valiases processed\n");
        vexit(0);

    /* if the database is down, deferr */
    } else if ( verrori == VA_NO_AUTH_CONNECTION ) {
        vexit(111);
    }

#endif

    /* get the user from vpopmail database */
    if ((vpw=vauth_getpw(TheUser, TheDomain)) != NULL ) {
        checkuser();
    } 

#ifdef QMAIL_EXT
    /* try and find user that matches the QmailEXT address if: no user found,
     * and the QmailEXT address is different, meaning there was an extension 
     */
    else if ( strncmp(TheUser, TheUserExt, AUTH_SIZE) != 0 ) {
        /* get the user from vpopmail database */
        if ((vpw=vauth_getpw(TheUserExt, TheDomain)) != NULL ) {
            checkuser();
        } else {
            usernotfound();
        }
    }
#endif

   /* Additions by Steve Fulton - [e-mail] 
    * - added error code handling for error codes generated by Autoresponder
    * 99 = deliver mail
    * 100 = bounce mail
    */ 

    else if ( verrori == 100) 
	vexit(100);	

    else if ( verrori == 99) 
	vexit(0);

   /* End of additions by Steve Fulton */

    else {
        if ( verrori != 0 ) {
            vexit(111);
        }
        usernotfound();
    } 

    /* exit successfully and have qmail delete the email */
    return(vexit(0));
            
}

/* 
 * Get the command line arguments and the environment variables.
 * Force addresses to be lower case and set the default domain
 */
void get_arguments(int argc, char **argv)
{
 char *tmpstr; 
 int i;

    if (argc != 3) {
        printf("vdelivermail: invalid syntax in .qmail-default file\n");

        /* exit with temporary error */
        vexit(111);
    }

    /* get the last parameter in the .qmail-default file */
    strncpy(bounce, argv[2], sizeof(bounce)); 

    if ((tmpstr=getenv("EXT")) == NULL ) {
        printf("vdelivermail: no EXT environment varilable\n");

        /* exit and bounce the email back */
        vexit(100);
    } else {
        for(i=0;*tmpstr!=0&&i<AUTH_SIZE;++i,++tmpstr) {
            TheUser[i] = *tmpstr;
        }
        TheUser[i] = 0;
    }

    if ((tmpstr=getenv("HOST")) == NULL ) {
        printf("vdelivermail: no HOST environment varilable\n");

        /* exit and bounce the email back */
        vexit(100);
    } else {
        for(i=0;*tmpstr!=0&&i<AUTH_SIZE;++i,++tmpstr) {
            TheDomain[i] = *tmpstr;
        }
        TheDomain[i] = 0;
    }

    lowerit(TheUser);
    lowerit(TheDomain);

    if ( is_username_valid(TheUser) != 0 ) vexit(100);
    if ( is_domain_valid(TheDomain) != 0 ) vexit(100);
    strncpy(TheUserFull, TheUser, AUTH_SIZE);

#ifdef QMAIL_EXT 
    /* delete the '-' and following chars if any and store in TheUserExt */
    for(i = 0; TheUser[i] != 0; i++) {
        if (TheUser[i] == '-' ) {
            break;
        }
        TheUserExt[i] = TheUser[i];
    }

    TheUserExt[i] = 0;
    if ( is_username_valid(TheUserExt) != 0 ) {
        vexit(100);
    }

#endif

    vget_assign(TheDomain,TheDomainDir,156,&TheDomainUid,&TheDomainGid);

}

#ifdef VALIAS
/* 
 * Process any valiases for this user@domain
 * 
 * This will look up any valiases in vpopmail and
 * deliver the email to the entries
 *
 * Return 1 if aliases found
 * Return 0 if no aliases found 
 */
int process_valias(void)
{
 int found = 0;
 int deliver_err;
 char *tmpstr;
 char tmpuser[350];
 char def[150];
 int i;
 int j=0;

    /* Get the first alias for this user@domain */
    tmpstr = valias_select( TheUser, TheDomain );

    /* check for wildcard if there's no match */
    if(tmpstr == NULL) {
        for(i=strlen(TheUser);i >= 0 && j != 1;--i) {
            if(TheUser[i-1]=='-') {
                tmpuser[0] = '\0';
                strncat(tmpuser,TheUser,i); 
                strcat(tmpuser, "default");
                tmpstr = valias_select( tmpuser, TheDomain );
                if(tmpstr != NULL) {
                    sprintf(def, "DEFAULT=%s", &TheUser[i]);
                    putenv(def);
                    j = 1;
                }
            }
        }
    }

    /* tmpstr will be NULL if there are no more aliases */
    while (tmpstr != NULL ) {

        /* We found one */
        found = 1;

        /* deliver the mail */
        deliver_err = deliver_mail(tmpstr, "NOQUOTA");
        if (deliver_err == -2) {
            printf("system error\n");
            vexit(111);
        } else if (deliver_err == -3) {
            printf("mail is looping");
            vexit(111);
        }

        /* Get the next alias for this user@domain */
        tmpstr = valias_select_next();
    }

#ifdef QMAIL_EXT 
    /* try and find alias that matches the QmailEXT address 
     * if: no alias found, 
     * and the QmailEXT address is different, meaning there was an extension 
     */
    if ( (!found) && ( strncmp(TheUser, TheUserExt, AUTH_SIZE) != 0 )  ) {
        /* Get the first alias for this user@domain */
        tmpstr = valias_select( TheUserExt, TheDomain );

        /* tmpstr will be NULL if there are no more aliases */
        while (tmpstr != NULL ) {

            /* We found one */
            found = 1;

            /* deliver the mail */
            deliver_err = deliver_mail(tmpstr, "NOQUOTA");
            if (deliver_err == -2) {
                printf("system error\n");
                vexit(111);
            } else if (deliver_err == -3) {
                printf("mail is looping\n");
                vexit(111);
            }

            /* Get the next alias for this user@domain */
            tmpstr = valias_select_next();
        } 
    }    
#endif

    /* Return whether we found an alias or not */
    return(found);
}
#endif

/* If the .qmail-default file has bounce all in it
 * Then return 1
 * otherwise return 0
 */
int is_bounce(char *deliverto)
{
    if ( strcmp( deliverto, BOUNCE_ALL ) == 0 ) {
        return(1);
    } else {
        return(0);
    }
}

/* If the .qmail-default file has delete all in it
 * Then return 1
 * otherwise return 0
 */
int is_delete(char *deliverto)
{
    if ( strcmp( deliverto, DELETE_ALL ) == 0 ) {
        return(1);
    } else {
        return(0);
    }
}



/* cound the directory */
off_t count_dir(char *dir_name)
{
 DIR *mydir;
 struct dirent *mydirent;
 struct stat statbuf;
 off_t file_size = 0;
 char *tmpstr;

    if ( dir_name == NULL ) return(0);
    if (chdir(dir_name) == -1) return(0);
    if ( (mydir = opendir(".")) == NULL )  return(0);

    while( (mydirent=readdir(mydir)) != NULL ) {
        if ( strcmp( mydirent->d_name, "..") == 0 ) continue;
        if ( strcmp( mydirent->d_name, ".") == 0 ) continue;

        if ( (tmpstr=strstr(mydirent->d_name, ",S="))!=NULL) {
            tmpstr += 3;
            file_size += atoi(tmpstr);
        } else if (stat(mydirent->d_name,&statbuf)==0 && 
                   (statbuf.st_mode & S_IFDIR) ) {
            file_size +=  count_dir(mydirent->d_name);
        }
    }
    closedir(mydir);
    if ( dir_name != NULL && strcmp(dir_name, ".." )!=0 && 
                             strcmp(dir_name, "." )!=0) {
        chdir("..");
    }
    return(file_size);
}

long unsigned qmail_inject_open(char *address)
{
 int pim[2];
 long unsigned pid;
 int i=0;
 static char *in_address;
 static char *binqqargs[4];

    in_address = malloc(strlen(address)+1);  
    strcpy(in_address, address);

    /* skip over an & sign if there */
    if (in_address[i] == '&') ++i;

    if ( pipe(pim) == -1) return(-1);

    switch(pid=vfork()){
      case -1:
        close(pim[0]);
        close(pim[1]);
        return(-1);
      case 0:
        close(pim[1]);
        if (vfd_move(0,pim[0]) == -1 ) _exit(-1);
        binqqargs[0] = QMAILINJECT;
        binqqargs[1] = "--";
        binqqargs[2] = &in_address[i];
        execv(*binqqargs, binqqargs);
    }
    fdm = pim[1];
    close(pim[0]);
    free(in_address);
    return(pid);
}



/* 
 * Deliver an email to an address
 * Return 0 on success
 * Return less than zero on failure
 * 
 * -1 = user is over quota
 * -2 and below are system failures
 * -3 mail is looping