qmail-badmimetypes.c

Go to the documentation of this file.

#include <unistd.h>
#include <sys/stat.h>
#include "logmsg.h"
#include "stralloc.h"
#include "buffer.h"
#include "getln.h"
#include "exit.h"
#include "open.h"
#include "auto_qmail.h"
#include "cdbmake.h"
#include "qmail.h"
 
#define WHO "qmail-badmimetypes"
#define MIMETYPE_LEN 9
 
int rename(const char *,const char *); // stdio.h
 
static void die_read()
{
  logmsg(WHO,111,FATAL,"unable to read control/badmimetypes");
}
static void die_write()
{
  logmsg(WHO,111,FATAL,"unable to write to control/badmimetypes.tmp");
}
 
char inbuf[BUFSIZE_LINE];
buffer b;
 
int fd;
int fdtemp;
 
struct cdb_make cdb;
stralloc line = {0};
int match;
 
int main()
{
  umask(033);
  if (chdir(auto_qmail) == -1)
    logmsg(WHO,111,FATAL,B("unable to chdir to: ",auto_qmail));
 
  fd = open_read("control/badmimetypes");
  if (fd == -1) die_read();
 
  buffer_init(&b,read,fd,inbuf,sizeof(inbuf));
 
  fdtemp = open_trunc("control/badmimetypes.tmp");
  if (fdtemp == -1) die_write();
 
  if (cdb_make_start(&cdb,fdtemp) == -1) die_write();
 
  for (;;) {
    if (getln(&b,&line,&match,'\n') != 0) die_read();
    if (line.s[0] != '#' && line.len > MIMETYPE_LEN)
      if (cdb_make_add(&cdb,line.s,MIMETYPE_LEN,"",0) == -1)
        die_write();
    if (!match) break;
  }
 
  if (cdb_make_finish(&cdb) == -1) die_write();
  if (fsync(fdtemp) == -1) die_write();
  if (close(fdtemp) == -1) die_write(); /* NFS stupidity */
  if (rename("control/badmimetypes.tmp","control/badmimetypes.cdb") == -1)
    logmsg(WHO,111,FATAL,"unable to move control/badmimetypes.tmp to control/badmimetypes.cdb");
 
  _exit(0);
}