Commit | Line | Data |
---|---|---|
2117e02e MW |
1 | #include "stralloc.h" |
2 | #include "subfd.h" | |
3 | #include "getln.h" | |
4 | #include "substdio.h" | |
5 | #include "cdbmss.h" | |
6 | #include "exit.h" | |
7 | #include "readwrite.h" | |
8 | #include "open.h" | |
9 | #include "error.h" | |
10 | #include "case.h" | |
11 | #include "auto_qmail.h" | |
12 | ||
13 | void die_temp() { _exit(111); } | |
14 | ||
15 | void die_chdir() | |
16 | { | |
17 | substdio_putsflush(subfderr,"qmail-newu: fatal: unable to chdir\n"); | |
18 | die_temp(); | |
19 | } | |
20 | void die_nomem() | |
21 | { | |
22 | substdio_putsflush(subfderr,"qmail-newu: fatal: out of memory\n"); | |
23 | die_temp(); | |
24 | } | |
25 | void die_opena() | |
26 | { | |
27 | substdio_putsflush(subfderr,"qmail-newu: fatal: unable to open users/assign\n"); | |
28 | die_temp(); | |
29 | } | |
30 | void die_reada() | |
31 | { | |
32 | substdio_putsflush(subfderr,"qmail-newu: fatal: unable to read users/assign\n"); | |
33 | die_temp(); | |
34 | } | |
35 | void die_format() | |
36 | { | |
37 | substdio_putsflush(subfderr,"qmail-newu: fatal: bad format in users/assign\n"); | |
38 | die_temp(); | |
39 | } | |
40 | void die_opent() | |
41 | { | |
42 | substdio_putsflush(subfderr,"qmail-newu: fatal: unable to open users/cdb.tmp\n"); | |
43 | die_temp(); | |
44 | } | |
45 | void die_writet() | |
46 | { | |
47 | substdio_putsflush(subfderr,"qmail-newu: fatal: unable to write users/cdb.tmp\n"); | |
48 | die_temp(); | |
49 | } | |
50 | void die_rename() | |
51 | { | |
52 | substdio_putsflush(subfderr,"qmail-newu: fatal: unable to move users/cdb.tmp to users/cdb\n"); | |
53 | die_temp(); | |
54 | } | |
55 | ||
56 | struct cdbmss cdbmss; | |
57 | stralloc key = {0}; | |
58 | stralloc data = {0}; | |
59 | ||
60 | char inbuf[1024]; | |
61 | substdio ssin; | |
62 | ||
63 | int fd; | |
64 | int fdtemp; | |
65 | ||
66 | stralloc line = {0}; | |
67 | int match; | |
68 | ||
69 | stralloc wildchars = {0}; | |
70 | ||
71 | void main() | |
72 | { | |
73 | int i; | |
74 | int numcolons; | |
75 | ||
76 | umask(033); | |
77 | if (chdir(auto_qmail) == -1) die_chdir(); | |
78 | ||
79 | fd = open_read("users/assign"); | |
80 | if (fd == -1) die_opena(); | |
81 | ||
82 | substdio_fdbuf(&ssin,read,fd,inbuf,sizeof(inbuf)); | |
83 | ||
84 | fdtemp = open_trunc("users/cdb.tmp"); | |
85 | if (fdtemp == -1) die_opent(); | |
86 | ||
87 | if (cdbmss_start(&cdbmss,fdtemp) == -1) die_writet(); | |
88 | ||
89 | if (!stralloc_copys(&wildchars,"")) die_nomem(); | |
90 | ||
91 | for (;;) { | |
92 | if (getln(&ssin,&line,&match,'\n') != 0) die_reada(); | |
93 | if (line.len && (line.s[0] == '.')) break; | |
94 | if (!match) die_format(); | |
95 | ||
96 | if (byte_chr(line.s,line.len,'\0') < line.len) die_format(); | |
97 | i = byte_chr(line.s,line.len,':'); | |
98 | if (i == line.len) die_format(); | |
99 | if (i == 0) die_format(); | |
100 | if (!stralloc_copys(&key,"!")) die_nomem(); | |
101 | if (line.s[0] == '+') { | |
102 | if (!stralloc_catb(&key,line.s + 1,i - 1)) die_nomem(); | |
103 | case_lowerb(key.s,key.len); | |
104 | if (i >= 2) | |
105 | if (byte_chr(wildchars.s,wildchars.len,line.s[i - 1]) == wildchars.len) | |
106 | if (!stralloc_append(&wildchars,line.s + i - 1)) die_nomem(); | |
107 | } | |
108 | else { | |
109 | if (!stralloc_catb(&key,line.s + 1,i - 1)) die_nomem(); | |
110 | if (!stralloc_0(&key)) die_nomem(); | |
111 | case_lowerb(key.s,key.len); | |
112 | } | |
113 | ||
114 | if (!stralloc_copyb(&data,line.s + i + 1,line.len - i - 1)) die_nomem(); | |
115 | ||
116 | numcolons = 0; | |
117 | for (i = 0;i < data.len;++i) | |
118 | if (data.s[i] == ':') { | |
119 | data.s[i] = 0; | |
120 | if (++numcolons == 6) | |
121 | break; | |
122 | } | |
123 | if (numcolons < 6) die_format(); | |
124 | data.len = i; | |
125 | ||
126 | if (cdbmss_add(&cdbmss,key.s,key.len,data.s,data.len) == -1) die_writet(); | |
127 | } | |
128 | ||
129 | if (cdbmss_add(&cdbmss,"",0,wildchars.s,wildchars.len) == -1) die_writet(); | |
130 | ||
131 | if (cdbmss_finish(&cdbmss) == -1) die_writet(); | |
132 | if (fsync(fdtemp) == -1) die_writet(); | |
133 | if (close(fdtemp) == -1) die_writet(); /* NFS stupidity */ | |
134 | if (rename("users/cdb.tmp","users/cdb") == -1) die_rename(); | |
135 | ||
136 | _exit(0); | |
137 | } |