summaryrefslogtreecommitdiff
path: root/src/qmail-badmimetypes.c
blob: 314d1425522d0985e268aad84df847bcd4b55b02 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#include <sys/stat.h>
#include <unistd.h>

#include <stdio.h>

#include "buffer.h"
#include "cdbmake.h"
#include "exit.h"
#include "getln.h"
#include "logmsg.h"
#include "open.h"
#include "stralloc.h"

#include "auto_qmail.h"

#define WHO          "qmail-badmimetypes"
#define MIMETYPE_LEN 9

void die_read()
{
  logmsg(WHO, 111, FATAL, "unable to read control/badmimetypes");
}

void die_write()
{
  logmsg(WHO, 111, FATAL, "unable to write to control/badmimetypes.tmp");
}

char inbuf[1024];
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);
}