1 /*
2 * Copyright (c) 1980, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 * @(#)quit.c 8.2 (Berkeley) 4/28/95
30 * $FreeBSD: src/usr.bin/mail/quit.c,v 1.2.6.3 2003/01/06 05:46:03 mikeh Exp $
31 * $DragonFly: src/usr.bin/mail/quit.c,v 1.5 2004/09/08 03:01:11 joerg Exp $
32 */
33
34 #include "rcv.h"
35 #include <fcntl.h>
36 #include "extern.h"
37
38 static void edstop(void);
39 static int writeback(FILE *res);
40
41 /*
42 * Rcv -- receive mail rationally.
43 *
44 * Termination processing.
45 */
46
47 /*
48 * The "quit" command.
49 */
50 int
quitcmd(void)51 quitcmd(void)
52 {
53 /*
54 * If we are sourcing, then return 1 so execute() can handle it.
55 * Otherwise, return -1 to abort command loop.
56 */
57 if (sourcing)
58 return (1);
59 return (-1);
60 }
61
62 /*
63 * Save all of the undetermined messages at the top of "mbox"
64 * Save all untouched messages back in the system mailbox.
65 * Remove the system mailbox, if none saved there.
66 */
67 void
quit(void)68 quit(void)
69 {
70 int mcount, p, modify, autohold, anystat, holdbit, nohold;
71 FILE *ibuf, *obuf, *fbuf, *rbuf, *readstat, *abuf;
72 struct message *mp;
73 int c, fd;
74 struct stat minfo;
75 char *mbox, tempname[PATHSIZE];
76
77 /*
78 * If we are read only, we can't do anything,
79 * so just return quickly.
80 */
81 if (readonly)
82 return;
83 /*
84 * If editing (not reading system mail box), then do the work
85 * in edstop()
86 */
87 if (edit) {
88 edstop();
89 return;
90 }
91
92 /*
93 * See if there any messages to save in mbox. If no, we
94 * can save copying mbox to /tmp and back.
95 *
96 * Check also to see if any files need to be preserved.
97 * Delete all untouched messages to keep them out of mbox.
98 * If all the messages are to be preserved, just exit with
99 * a message.
100 */
101
102 fbuf = Fopen(mailname, "r");
103 if (fbuf == NULL)
104 goto newmail;
105 flock(fileno(fbuf), LOCK_EX);
106 rbuf = NULL;
107 if (fstat(fileno(fbuf), &minfo) >= 0 && minfo.st_size > mailsize) {
108 printf("New mail has arrived.\n");
109 snprintf(tempname, sizeof(tempname), "%s/mail.RqXXXXXXXXXX",
110 tmpdir);
111 if ((fd = mkstemp(tempname)) == -1 ||
112 (rbuf = Fdopen(fd, "w")) == NULL)
113 goto newmail;
114 #ifdef APPEND
115 fseeko(fbuf, mailsize, SEEK_SET);
116 while ((c = getc(fbuf)) != EOF)
117 putc(c, rbuf);
118 #else
119 p = minfo.st_size - mailsize;
120 while (p-- > 0) {
121 c = getc(fbuf);
122 if (c == EOF)
123 goto newmail;
124 putc(c, rbuf);
125 }
126 #endif
127 Fclose(rbuf);
128 if ((rbuf = Fopen(tempname, "r")) == NULL)
129 goto newmail;
130 rm(tempname);
131 }
132
133 /*
134 * Adjust the message flags in each message.
135 */
136
137 anystat = 0;
138 autohold = value("hold") != NULL;
139 holdbit = autohold ? MPRESERVE : MBOX;
140 nohold = MBOX|MSAVED|MDELETED|MPRESERVE;
141 if (value("keepsave") != NULL)
142 nohold &= ~MSAVED;
143 for (mp = &message[0]; mp < &message[msgCount]; mp++) {
144 if (mp->m_flag & MNEW) {
145 mp->m_flag &= ~MNEW;
146 mp->m_flag |= MSTATUS;
147 }
148 if (mp->m_flag & MSTATUS)
149 anystat++;
150 if ((mp->m_flag & MTOUCH) == 0)
151 mp->m_flag |= MPRESERVE;
152 if ((mp->m_flag & nohold) == 0)
153 mp->m_flag |= holdbit;
154 }
155 modify = 0;
156 if (Tflag != NULL) {
157 if ((readstat = Fopen(Tflag, "w")) == NULL)
158 Tflag = NULL;
159 }
160 for (c = 0, p = 0, mp = &message[0]; mp < &message[msgCount]; mp++) {
161 if (mp->m_flag & MBOX)
162 c++;
163 if (mp->m_flag & MPRESERVE)
164 p++;
165 if (mp->m_flag & MODIFY)
166 modify++;
167 if (Tflag != NULL && (mp->m_flag & (MREAD|MDELETED)) != 0) {
168 char *id;
169
170 if ((id = hfield("article-id", mp)) != NULL)
171 fprintf(readstat, "%s\n", id);
172 }
173 }
174 if (Tflag != NULL)
175 Fclose(readstat);
176 if (p == msgCount && !modify && !anystat) {
177 printf("Held %d message%s in %s\n",
178 p, p == 1 ? "" : "s", mailname);
179 Fclose(fbuf);
180 return;
181 }
182 if (c == 0) {
183 if (p != 0) {
184 writeback(rbuf);
185 Fclose(fbuf);
186 return;
187 }
188 goto cream;
189 }
190
191 /*
192 * Create another temporary file and copy user's mbox file
193 * darin. If there is no mbox, copy nothing.
194 * If he has specified "append" don't copy his mailbox,
195 * just copy saveable entries at the end.
196 */
197
198 mbox = expand("&");
199 mcount = c;
200 if (value("append") == NULL) {
201 snprintf(tempname, sizeof(tempname), "%s/mail.RmXXXXXXXXXX",
202 tmpdir);
203 if ((fd = mkstemp(tempname)) == -1 ||
204 (obuf = Fdopen(fd, "w")) == NULL) {
205 warn("%s", tempname);
206 Fclose(fbuf);
207 return;
208 }
209 if ((ibuf = Fopen(tempname, "r")) == NULL) {
210 warn("%s", tempname);
211 rm(tempname);
212 Fclose(obuf);
213 Fclose(fbuf);
214 return;
215 }
216 rm(tempname);
217 if ((abuf = Fopen(mbox, "r")) != NULL) {
218 while ((c = getc(abuf)) != EOF)
219 putc(c, obuf);
220 Fclose(abuf);
221 }
222 if (ferror(obuf)) {
223 warnx("%s", tempname);
224 Fclose(ibuf);
225 Fclose(obuf);
226 Fclose(fbuf);
227 return;
228 }
229 Fclose(obuf);
230 close(open(mbox, O_CREAT | O_TRUNC | O_WRONLY, 0600));
231 if ((obuf = Fopen(mbox, "r+")) == NULL) {
232 warn("%s", mbox);
233 Fclose(ibuf);
234 Fclose(fbuf);
235 return;
236 }
237 }
238 if (value("append") != NULL) {
239 if ((obuf = Fopen(mbox, "a")) == NULL) {
240 warn("%s", mbox);
241 Fclose(fbuf);
242 return;
243 }
244 fchmod(fileno(obuf), 0600);
245 }
246 for (mp = &message[0]; mp < &message[msgCount]; mp++)
247 if (mp->m_flag & MBOX)
248 if (sendmessage(mp, obuf, saveignore, NULL) < 0) {
249 warnx("%s", mbox);
250 Fclose(ibuf);
251 Fclose(obuf);
252 Fclose(fbuf);
253 return;
254 }
255
256 /*
257 * Copy the user's old mbox contents back
258 * to the end of the stuff we just saved.
259 * If we are appending, this is unnecessary.
260 */
261
262 if (value("append") == NULL) {
263 rewind(ibuf);
264 c = getc(ibuf);
265 while (c != EOF) {
266 putc(c, obuf);
267 if (ferror(obuf))
268 break;
269 c = getc(ibuf);
270 }
271 Fclose(ibuf);
272 }
273 fflush(obuf);
274 trunc(obuf);
275 if (ferror(obuf)) {
276 warn("%s", mbox);
277 Fclose(obuf);
278 Fclose(fbuf);
279 return;
280 }
281 Fclose(obuf);
282 if (mcount == 1)
283 printf("Saved 1 message in mbox\n");
284 else
285 printf("Saved %d messages in mbox\n", mcount);
286
287 /*
288 * Now we are ready to copy back preserved files to
289 * the system mailbox, if any were requested.
290 */
291
292 if (p != 0) {
293 writeback(rbuf);
294 Fclose(fbuf);
295 return;
296 }
297
298 /*
299 * Finally, remove his /var/mail file.
300 * If new mail has arrived, copy it back.
301 */
302
303 cream:
304 if (rbuf != NULL) {
305 abuf = Fopen(mailname, "r+");
306 if (abuf == NULL)
307 goto newmail;
308 while ((c = getc(rbuf)) != EOF)
309 putc(c, abuf);
310 Fclose(rbuf);
311 trunc(abuf);
312 Fclose(abuf);
313 alter(mailname);
314 Fclose(fbuf);
315 return;
316 }
317 demail();
318 Fclose(fbuf);
319 return;
320
321 newmail:
322 printf("Thou hast new mail.\n");
323 if (fbuf != NULL)
324 Fclose(fbuf);
325 }
326
327 /*
328 * Preserve all the appropriate messages back in the system
329 * mailbox, and print a nice message indicated how many were
330 * saved. On any error, just return -1. Else return 0.
331 * Incorporate the any new mail that we found.
332 */
333 static int
writeback(FILE * res)334 writeback(FILE *res)
335 {
336 struct message *mp;
337 int p, c;
338 FILE *obuf;
339
340 p = 0;
341 if ((obuf = Fopen(mailname, "r+")) == NULL) {
342 warn("%s", mailname);
343 return (-1);
344 }
345 #ifndef APPEND
346 if (res != NULL)
347 while ((c = getc(res)) != EOF)
348 putc(c, obuf);
349 #endif
350 for (mp = &message[0]; mp < &message[msgCount]; mp++)
351 if ((mp->m_flag&MPRESERVE)||(mp->m_flag&MTOUCH)==0) {
352 p++;
353 if (sendmessage(mp, obuf, NULL, NULL) < 0) {
354 warnx("%s", mailname);
355 Fclose(obuf);
356 return (-1);
357 }
358 }
359 #ifdef APPEND
360 if (res != NULL)
361 while ((c = getc(res)) != EOF)
362 putc(c, obuf);
363 #endif
364 fflush(obuf);
365 trunc(obuf);
366 if (ferror(obuf)) {
367 warn("%s", mailname);
368 Fclose(obuf);
369 return (-1);
370 }
371 if (res != NULL)
372 Fclose(res);
373 Fclose(obuf);
374 alter(mailname);
375 if (p == 1)
376 printf("Held 1 message in %s\n", mailname);
377 else
378 printf("Held %d messages in %s\n", p, mailname);
379 return (0);
380 }
381
382 /*
383 * Terminate an editing session by attempting to write out the user's
384 * file from the temporary. Save any new stuff appended to the file.
385 */
386 static void
edstop(void)387 edstop(void)
388 {
389 int gotcha, c;
390 struct message *mp;
391 FILE *obuf, *ibuf, *readstat;
392 struct stat statb;
393 char tempname[PATHSIZE];
394
395 if (readonly)
396 return;
397 holdsigs();
398 if (Tflag != NULL) {
399 if ((readstat = Fopen(Tflag, "w")) == NULL)
400 Tflag = NULL;
401 }
402 for (mp = &message[0], gotcha = 0; mp < &message[msgCount]; mp++) {
403 if (mp->m_flag & MNEW) {
404 mp->m_flag &= ~MNEW;
405 mp->m_flag |= MSTATUS;
406 }
407 if (mp->m_flag & (MODIFY|MDELETED|MSTATUS))
408 gotcha++;
409 if (Tflag != NULL && (mp->m_flag & (MREAD|MDELETED)) != 0) {
410 char *id;
411
412 if ((id = hfield("article-id", mp)) != NULL)
413 fprintf(readstat, "%s\n", id);
414 }
415 }
416 if (Tflag != NULL)
417 Fclose(readstat);
418 if (!gotcha || Tflag != NULL)
419 goto done;
420 ibuf = NULL;
421 if (stat(mailname, &statb) >= 0 && statb.st_size > mailsize) {
422 int fd;
423
424 snprintf(tempname, sizeof(tempname), "%s/mbox.XXXXXXXXXX",
425 tmpdir);
426 if ((fd = mkstemp(tempname)) == -1 ||
427 (obuf = Fdopen(fd, "w")) == NULL) {
428 warn("%s", tempname);
429 relsesigs();
430 reset(0);
431 }
432 if ((ibuf = Fopen(mailname, "r")) == NULL) {
433 warn("%s", mailname);
434 Fclose(obuf);
435 rm(tempname);
436 relsesigs();
437 reset(0);
438 }
439 fseeko(ibuf, mailsize, SEEK_SET);
440 while ((c = getc(ibuf)) != EOF)
441 putc(c, obuf);
442 Fclose(ibuf);
443 Fclose(obuf);
444 if ((ibuf = Fopen(tempname, "r")) == NULL) {
445 warn("%s", tempname);
446 rm(tempname);
447 relsesigs();
448 reset(0);
449 }
450 rm(tempname);
451 }
452 printf("\"%s\" ", mailname);
453 fflush(stdout);
454 if ((obuf = Fopen(mailname, "r+")) == NULL) {
455 warn("%s", mailname);
456 relsesigs();
457 reset(0);
458 }
459 trunc(obuf);
460 c = 0;
461 for (mp = &message[0]; mp < &message[msgCount]; mp++) {
462 if ((mp->m_flag & MDELETED) != 0)
463 continue;
464 c++;
465 if (sendmessage(mp, obuf, NULL, NULL) < 0) {
466 warnx("%s", mailname);
467 relsesigs();
468 reset(0);
469 }
470 }
471 gotcha = (c == 0 && ibuf == NULL);
472 if (ibuf != NULL) {
473 while ((c = getc(ibuf)) != EOF)
474 putc(c, obuf);
475 Fclose(ibuf);
476 }
477 fflush(obuf);
478 if (ferror(obuf)) {
479 warn("%s", mailname);
480 relsesigs();
481 reset(0);
482 }
483 Fclose(obuf);
484 if (gotcha) {
485 rm(mailname);
486 printf("removed\n");
487 } else
488 printf("complete\n");
489 fflush(stdout);
490
491 done:
492 relsesigs();
493 }
494