ezmlm-tstdig.c

Go to the documentation of this file.

#include <sys/types.h>
#include <sys/stat.h>
#include <stdio.h>
#include <unistd.h>
#include "stralloc.h"
#include "buffer.h"
#include "readwrite.h"
#include "sig.h"
#include "getconf.h"
#include "env.h"
#include "fmt.h"
#include "now.h"
#include "lock.h"
#include "getoptb.h"
#include "errtxt.h"
#include "scan.h"
#include "case.h"
#include "str.h"
#include "open.h"
#include "idx.h"
#include "auto_version.h"
#include "logmsg.h"
#include "lockfile.h"
 
#define WHO "ezmlm-tstdig"

 
 
static void die_usage() { logmsg(WHO,100,USAGE,"ezmlm-tstdig [-k kbytes] [-m messages] [-t hours] dir"); }
static void die_nomem() { logmsg(WHO,111,FATAL,ERR_NOMEM); }
 
stralloc line = {0};
 
buffer bn;
char numbuf[16];
 
char strnum[FMT_ULONG];
 
int flaglocal = 0;
 
int main(int argc,char **argv)
{
  char *dir;
  char *local;
  char *def;
  int opt;
  unsigned int pos;
  unsigned long num, digsize, dignum;
  unsigned long cumsize = 0L;
  unsigned long deltanum = 0L;
  unsigned long deltawhen = 0L;
  unsigned long deltasize = 0L;
  unsigned long when, tsttime, digwhen;
  int fd, fdlock;
 
  umask(022);
  sig_pipeignore();
  when = (unsigned long) now();
 
  while ((opt = getoptb(argc,argv,"k:t:m:vV")) != opteof)
    switch(opt) {
      case 'k': if (optarg)
                  scan_ulong(optarg,&deltasize);
                break;
      case 't': if (optarg)    /* hours */
                  scan_ulong(optarg,&deltawhen);
                break;
      case 'm': if (optarg)
                  scan_ulong(optarg,&deltanum);
                break;
      case 'v':
      case 'V': logmsg(WHO,0,VERSION,auto_version);
      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 (!getconf_line(&line,"num",0,dir))
    _exit(99);               /* no msgs no shirt -> no digest */
  if(!stralloc_0(&line)) die_nomem();
  pos = scan_ulong(line.s,&num);
  if (line.s[pos] == ':')
    scan_ulong(line.s+pos+1,&cumsize);
 
  if (getconf_line(&line,"dignum",0,dir)) {
    if(!stralloc_0(&line)) die_nomem();
    pos = scan_ulong(line.s,&dignum);
    if (line.s[pos] == ':')
      pos += 1 + scan_ulong(line.s + pos + 1,&digsize);
    if (line.s[pos] == ':')
      scan_ulong(line.s +pos + 1,&digwhen);
  } else {
    dignum = 0L;             /* no file, not done any digest */
    digsize = 0L;            /* nothing digested */
    digwhen = 0L;            /* will force a digest, but the last one was eons ago. */
                             /* ezmlm-get sends it out only if there are messages. */
                             /* This is as it should for new lists.   */
  }
  local = env_get("LOCAL");
  if (local && *local) {     /* in editor or manager */
    def = env_get("DEFAULT");
    if (def && *def) {       /* in manager */
      if (!case_starts(def,"dig") || case_starts(def,"digest-"))
        _exit(0);
    } else {                 /* in editor */
      flaglocal = 1;
    }
  }
 
  if (!deltawhen && !deltasize && !deltanum) _exit(0);
  if ((deltawhen && ((digwhen + deltawhen * 3600L) <= when)) ||
      (deltasize && ((digsize + (deltasize << 2)) <= cumsize)) ||
      (deltanum && ((dignum + deltanum) <= num))) {        /* digest! */
    if (flaglocal) {                                       /* avoid multiple digests. Of course, ezmlm-tstdig*/
                                                           /* belongs in ezmlm-digest, but it's too late ....*/
      fdlock = lockfile("lock");
      getconf_line(&line,"tstdig",0,dir);
      if (!stralloc_0(&line)) die_nomem();
      scan_ulong(line.s,&tsttime);                         /* give digest 1 h to complete */
                                                           /* nobody does digests more often */
      if ((tsttime + 3600L < when) || (tsttime <= digwhen)) {
        fd = open_trunc("tstdign");
        if (fd == -1)
          logmsg(WHO,111,FATAL,B(ERR_CREATE,dir,"/","tstdign"));
        buffer_init(&bn,buffer_unixwrite,fd,numbuf,sizeof(numbuf));
        if (buffer_put(&bn,strnum,fmt_ulong(strnum,when)) == -1)
          logmsg(WHO,111,FATAL,B(ERR_WRITE,dir,"/","tstdign"));
        if (buffer_puts(&bn,"\n") == -1)
          logmsg(WHO,111,FATAL,B(ERR_WRITE,dir,"/","tstdign"));
        if (buffer_flush(&bn) == -1)
          logmsg(WHO,111,FATAL,B(ERR_FLUSH,dir,"/","tstdign"));
        if (fsync(fd) == -1)
          logmsg(WHO,111,FATAL,B(ERR_SYNC,dir,"/","tstdign"));
        if (close(fd) == -1)
          logmsg(WHO,111,FATAL,B(ERR_CLOSE,dir,"/","tstdign"));
        if (rename("tstdign","tstdig") == -1)
          logmsg(WHO,111,FATAL,B(ERR_MOVE,"tstdign"));
        _exit(0);
      }
    } else
        _exit(0);
  }
  _exit(99);
 
  return 0;
}