ezmlm-limit.c

Go to the documentation of this file.

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include "stralloc.h"
#include "buffer.h"
#include "readwrite.h"
#include "sig.h"
#include "lock.h"
#include "getconf.h"
#include "fmt.h"
#include "now.h"
#include "getoptb.h"
#include "error.h"
#include "scan.h"
#include "open.h"
#include "errtxt.h"
#include "idx.h"
#include "logmsg.h"
#include "lockfile.h"
 
#define WHO "ezmlm-limit"

 
unsigned long deltasecs = LIMSECS;     /* interval to test over (seconds) */
unsigned long deltanum = LIMMSG;       /* max no messages in interval */
                                       /* see idx.h. Usually 30 msg/3600 secs*/
int flagd = 0;                         /* =0 create modpost, =1 ignore,  excess, =2 defer excess */
int flagmod;                           /* list moderated */
int flagloop;
const char *fn = TXT_LOOPNUM;
 
static void die_usage() { logmsg(WHO,100,USAGE,"ezmlm-limit [-f file] [-dDF] [-n messages]" " [-t secs] dir"); }
static void die_nomem() { logmsg(WHO,111,FATAL,ERR_NOMEM); }
static void die_new() { logmsg(WHO,111,FATAL,B(ERR_WRITE,"fn")); }
 
stralloc line = {0};
 
buffer bn;
char newbuf[16];
 
char strnum[FMT_ULONG];
 
int main(int argc,char **argv)
{
  char *dir;
  int opt;
  unsigned int pos;
  unsigned long num, loopnum, when;
  unsigned long loopwhen = 0L;
  int fd, fdlock;
 
  umask(022);
  sig_pipeignore();
  when = (unsigned long) now();
 
  while ((opt = getoptb(argc,argv,"dDf:Fn:t:")) != opteof)
    switch(opt) {
      case 'd': flagd = 1; break;
      case 'D': flagd = 0; break;
      case 'f': if (optarg && *optarg) fn = optarg; break;
      case 'F': fn = TXT_LOOPNUM;
      case 'n': if (optarg) scan_ulong(optarg,&deltanum); break;
      case 't': if (optarg) scan_ulong(optarg,&deltasecs); break;
      default:  die_usage();
  }
 
  dir = argv[optind++];
  if (!dir) die_usage();
 
  if (chdir(dir) == -1)
    logmsg(WHO,111,FATAL,B(ERR_SWITCH,dir));
 
  if (argv[optind])
    die_usage();                       /* avoid common error of putting options after dir */
  if ((flagmod = getconf_line(&line,"modpost",0,dir)))
    _exit(0);                          /* already mod */
  
  fdlock = lockfile("lock");           /* lock for num and for writing loopnum */
 
  if (!getconf_line(&line,"num",0,dir))
    _exit(99);                         /* no msgs */
 
  if (!stralloc_0(&line)) die_nomem();
  pos = scan_ulong(line.s,&num);       /* current msg */
  if ((flagloop = getconf_line(&line,fn,0,dir))) {
    if (!stralloc_0(&line)) die_nomem();
    pos = scan_ulong(line.s,&loopnum); /* msg when written */
    if (line.s[pos] == ':')
      scan_ulong(line.s+pos+1,&loopwhen);/* time written */
  }
  if (!flagloop || loopwhen + deltasecs < when || loopwhen > when) { /* loopnum too old, bad or not there */
    fd = open_trunc(fn);               /* no need to write crash-proof */
    if (fd == -1) die_new();
    buffer_init(&bn,buffer_unixwrite,fd,newbuf,sizeof(newbuf));
    if (buffer_put(&bn,strnum,fmt_ulong(strnum,num)) == -1) die_new();
    if (buffer_puts(&bn,":") == -1) die_new();
    if (buffer_put(&bn,strnum,fmt_ulong(strnum,when)) == -1) die_new();
    if (buffer_puts(&bn,"\n") == -1) die_new();
    if (buffer_flush(&bn) == -1) die_new();
    close(fd);
  } else if (num >= loopnum + deltanum) {       /* excess messages */
    if (!flagd) {
      if ((fd = open_append("modpost")) == -1)  /* create dir/modpost */
      logmsg(WHO,111,FATAL,B(ERR_WRITE,"subpost"));
      else {
        close(fd);
        unlink(fn);
        logmsg(WHO,0,INFO,ERR_EXCESS_MOD);
      }
    } else
        logmsg(WHO,111,FATAL,ERR_EXCESS_DEFER);
  }
  _exit(0);
 
  return 0;
}