Commit | Line | Data |
---|---|---|
2117e02e MW |
1 | #include "readwrite.h" |
2 | #include "prioq.h" | |
3 | #include "env.h" | |
4 | #include "stralloc.h" | |
5 | #include "subfd.h" | |
6 | #include "substdio.h" | |
7 | #include "getln.h" | |
8 | #include "error.h" | |
9 | #include "open.h" | |
10 | #include "lock.h" | |
11 | #include "gfrom.h" | |
12 | #include "str.h" | |
13 | #include "exit.h" | |
14 | #include "myctime.h" | |
15 | #include "maildir.h" | |
16 | ||
17 | char *mbox; | |
18 | char *mboxtmp; | |
19 | ||
20 | stralloc filenames = {0}; | |
21 | prioq pq = {0}; | |
22 | prioq pq2 = {0}; | |
23 | ||
24 | stralloc line = {0}; | |
25 | ||
26 | stralloc ufline = {0}; | |
27 | ||
28 | char inbuf[SUBSTDIO_INSIZE]; | |
29 | char outbuf[SUBSTDIO_OUTSIZE]; | |
30 | ||
31 | #define FATAL "maildir2mbox: fatal: " | |
32 | #define WARNING "maildir2mbox: warning: " | |
33 | ||
34 | void die_nomem() { strerr_die2x(111,FATAL,"out of memory"); } | |
35 | ||
36 | void main() | |
37 | { | |
38 | substdio ssin; | |
39 | substdio ssout; | |
40 | struct prioq_elt pe; | |
41 | int fdoldmbox; | |
42 | int fdnewmbox; | |
43 | int fd; | |
44 | int match; | |
45 | int fdlock; | |
46 | ||
47 | umask(077); | |
48 | ||
49 | mbox = env_get("MAIL"); | |
50 | if (!mbox) strerr_die2x(111,FATAL,"MAIL not set"); | |
51 | mboxtmp = env_get("MAILTMP"); | |
52 | if (!mboxtmp) strerr_die2x(111,FATAL,"MAILTMP not set"); | |
53 | ||
54 | if (maildir_chdir() == -1) | |
55 | strerr_die1(111,FATAL,&maildir_chdir_err); | |
56 | maildir_clean(&filenames); | |
57 | if (maildir_scan(&pq,&filenames,1,1) == -1) | |
58 | strerr_die1(111,FATAL,&maildir_scan_err); | |
59 | ||
60 | if (!prioq_min(&pq,&pe)) _exit(0); /* nothing new */ | |
61 | ||
62 | fdlock = open_append(mbox); | |
63 | if (fdlock == -1) | |
64 | strerr_die4sys(111,FATAL,"unable to lock ",mbox,": "); | |
65 | if (lock_ex(fdlock) == -1) | |
66 | strerr_die4sys(111,FATAL,"unable to lock ",mbox,": "); | |
67 | ||
68 | fdoldmbox = open_read(mbox); | |
69 | if (fdoldmbox == -1) | |
70 | strerr_die4sys(111,FATAL,"unable to read ",mbox,": "); | |
71 | ||
72 | fdnewmbox = open_trunc(mboxtmp); | |
73 | if (fdnewmbox == -1) | |
74 | strerr_die4sys(111,FATAL,"unable to create ",mboxtmp,": "); | |
75 | ||
76 | substdio_fdbuf(&ssin,read,fdoldmbox,inbuf,sizeof(inbuf)); | |
77 | substdio_fdbuf(&ssout,write,fdnewmbox,outbuf,sizeof(outbuf)); | |
78 | ||
79 | switch(substdio_copy(&ssout,&ssin)) | |
80 | { | |
81 | case -2: strerr_die4sys(111,FATAL,"unable to read ",mbox,": "); | |
82 | case -3: strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
83 | } | |
84 | ||
85 | while (prioq_min(&pq,&pe)) | |
86 | { | |
87 | prioq_delmin(&pq); | |
88 | if (!prioq_insert(&pq2,&pe)) die_nomem(); | |
89 | ||
90 | fd = open_read(filenames.s + pe.id); | |
91 | if (fd == -1) | |
92 | strerr_die4sys(111,FATAL,"unable to read $MAILDIR/",filenames.s + pe.id,": "); | |
93 | substdio_fdbuf(&ssin,read,fd,inbuf,sizeof(inbuf)); | |
94 | ||
95 | if (getln(&ssin,&line,&match,'\n') != 0) | |
96 | strerr_die4sys(111,FATAL,"unable to read $MAILDIR/",filenames.s + pe.id,": "); | |
97 | ||
98 | if (!stralloc_copys(&ufline,"From XXX ")) die_nomem(); | |
99 | if (match) | |
100 | if (stralloc_starts(&line,"Return-Path: <")) | |
101 | { | |
102 | if (line.s[14] == '>') | |
103 | { | |
104 | if (!stralloc_copys(&ufline,"From MAILER-DAEMON ")) die_nomem(); | |
105 | } | |
106 | else | |
107 | { | |
108 | int i; | |
109 | if (!stralloc_ready(&ufline,line.len)) die_nomem(); | |
110 | if (!stralloc_copys(&ufline,"From ")) die_nomem(); | |
111 | for (i = 14;i < line.len - 2;++i) | |
112 | if ((line.s[i] == ' ') || (line.s[i] == '\t')) | |
113 | ufline.s[ufline.len++] = '-'; | |
114 | else | |
115 | ufline.s[ufline.len++] = line.s[i]; | |
116 | if (!stralloc_cats(&ufline," ")) die_nomem(); | |
117 | } | |
118 | } | |
119 | if (!stralloc_cats(&ufline,myctime(pe.dt))) die_nomem(); | |
120 | if (substdio_put(&ssout,ufline.s,ufline.len) == -1) | |
121 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
122 | ||
123 | while (match && line.len) | |
124 | { | |
125 | if (gfrom(line.s,line.len)) | |
126 | if (substdio_puts(&ssout,">") == -1) | |
127 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
128 | if (substdio_put(&ssout,line.s,line.len) == -1) | |
129 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
130 | if (!match) | |
131 | { | |
132 | if (substdio_puts(&ssout,"\n") == -1) | |
133 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
134 | break; | |
135 | } | |
136 | if (getln(&ssin,&line,&match,'\n') != 0) | |
137 | strerr_die4sys(111,FATAL,"unable to read $MAILDIR/",filenames.s + pe.id,": "); | |
138 | } | |
139 | if (substdio_puts(&ssout,"\n")) | |
140 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
141 | ||
142 | close(fd); | |
143 | } | |
144 | ||
145 | if (substdio_flush(&ssout) == -1) | |
146 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
147 | if (fsync(fdnewmbox) == -1) | |
148 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
149 | if (close(fdnewmbox) == -1) /* NFS dorks */ | |
150 | strerr_die4sys(111,FATAL,"unable to write to ",mboxtmp,": "); | |
151 | if (rename(mboxtmp,mbox) == -1) | |
152 | strerr_die6(111,FATAL,"unable to move ",mboxtmp," to ",mbox,": ",&strerr_sys); | |
153 | ||
154 | while (prioq_min(&pq2,&pe)) | |
155 | { | |
156 | prioq_delmin(&pq2); | |
157 | if (unlink(filenames.s + pe.id) == -1) | |
158 | strerr_warn4(WARNING,"$MAILDIR/",filenames.s + pe.id," will be delivered twice; unable to unlink: ",&strerr_sys); | |
159 | } | |
160 | ||
161 | _exit(0); | |
162 | } |