xref: /dragonfly/usr.bin/mail/fio.c (revision 6e285212)
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. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#)fio.c	8.2 (Berkeley) 4/20/95
34  * $FreeBSD: src/usr.bin/mail/fio.c,v 1.2.6.4 2003/01/06 05:46:03 mikeh Exp $
35  * $DragonFly: src/usr.bin/mail/fio.c,v 1.2 2003/06/17 04:29:28 dillon Exp $
36  */
37 
38 #include "rcv.h"
39 #include <sys/file.h>
40 #include <sys/wait.h>
41 
42 #include <unistd.h>
43 #include <paths.h>
44 #include <errno.h>
45 #include "extern.h"
46 
47 /*
48  * Mail -- a mail program
49  *
50  * File I/O.
51  */
52 
53 extern int wait_status;
54 
55 /*
56  * Set up the input pointers while copying the mail file into /tmp.
57  */
58 void
59 setptr(ibuf, offset)
60 	FILE *ibuf;
61 	off_t offset;
62 {
63 	int c, count;
64 	char *cp, *cp2;
65 	struct message this;
66 	FILE *mestmp;
67 	int maybe, inhead;
68 	char linebuf[LINESIZE], pathbuf[PATHSIZE];
69 	int omsgCount;
70 
71 	/* Get temporary file. */
72 	(void)snprintf(pathbuf, sizeof(pathbuf), "%s/mail.XXXXXXXXXX", tmpdir);
73 	if ((c = mkstemp(pathbuf)) == -1 || (mestmp = Fdopen(c, "r+")) == NULL)
74 		err(1, "can't open %s", pathbuf);
75 	(void)rm(pathbuf);
76 
77 	if (offset == 0) {
78 		 msgCount = 0;
79 	} else {
80 		/* Seek into the file to get to the new messages */
81 		(void)fseeko(ibuf, offset, SEEK_SET);
82 		/*
83 		 * We need to make "offset" a pointer to the end of
84 		 * the temp file that has the copy of the mail file.
85 		 * If any messages have been edited, this will be
86 		 * different from the offset into the mail file.
87 		 */
88 		(void)fseeko(otf, (off_t)0, SEEK_END);
89 		offset = ftello(otf);
90 	}
91 	omsgCount = msgCount;
92 	maybe = 1;
93 	inhead = 0;
94 	this.m_flag = MUSED|MNEW;
95 	this.m_size = 0;
96 	this.m_lines = 0;
97 	this.m_block = 0;
98 	this.m_offset = 0;
99 	for (;;) {
100 		if (fgets(linebuf, sizeof(linebuf), ibuf) == NULL) {
101 			if (append(&this, mestmp))
102 				errx(1, "temporary file");
103 			makemessage(mestmp, omsgCount);
104 			return;
105 		}
106 		count = strlen(linebuf);
107 		/*
108 		 * Transforms lines ending in <CR><LF> to just <LF>.
109 		 * This allows mail to be able to read Eudora mailboxes.
110 		 */
111 		if (count >= 2 && linebuf[count - 1] == '\n' &&
112 		    linebuf[count - 2] == '\r') {
113 			count--;
114 			linebuf[count - 1] = '\n';
115 		}
116 
117 		(void)fwrite(linebuf, sizeof(*linebuf), count, otf);
118 		if (ferror(otf))
119 			errx(1, "/tmp");
120 		if (count)
121 			linebuf[count - 1] = '\0';
122 		if (maybe && linebuf[0] == 'F' && ishead(linebuf)) {
123 			msgCount++;
124 			if (append(&this, mestmp))
125 				errx(1, "temporary file");
126 			this.m_flag = MUSED|MNEW;
127 			this.m_size = 0;
128 			this.m_lines = 0;
129 			this.m_block = blockof(offset);
130 			this.m_offset = boffsetof(offset);
131 			inhead = 1;
132 		} else if (linebuf[0] == 0) {
133 			inhead = 0;
134 		} else if (inhead) {
135 			for (cp = linebuf, cp2 = "status";; cp++) {
136 				if ((c = *cp2++) == '\0') {
137 					while (isspace((unsigned char)*cp++))
138 						;
139 					if (cp[-1] != ':')
140 						break;
141 					while ((c = *cp++) != '\0')
142 						if (c == 'R')
143 							this.m_flag |= MREAD;
144 						else if (c == 'O')
145 							this.m_flag &= ~MNEW;
146 					inhead = 0;
147 					break;
148 				}
149 				if (*cp != c && *cp != toupper((unsigned char)c))
150 					break;
151 			}
152 		}
153 		offset += count;
154 		this.m_size += count;
155 		this.m_lines++;
156 		maybe = linebuf[0] == 0;
157 	}
158 }
159 
160 /*
161  * Drop the passed line onto the passed output buffer.
162  * If a write error occurs, return -1, else the count of
163  * characters written, including the newline if requested.
164  */
165 int
166 putline(obuf, linebuf, outlf)
167 	FILE *obuf;
168 	char *linebuf;
169 	int outlf;
170 {
171 	int c;
172 
173 	c = strlen(linebuf);
174 	(void)fwrite(linebuf, sizeof(*linebuf), c, obuf);
175 	if (outlf) {
176 		fprintf(obuf, "\n");
177 		c++;
178 	}
179 	if (ferror(obuf))
180 		return (-1);
181 	return (c);
182 }
183 
184 /*
185  * Read up a line from the specified input into the line
186  * buffer.  Return the number of characters read.  Do not
187  * include the newline (or carriage return) at the end.
188  */
189 int
190 readline(ibuf, linebuf, linesize)
191 	FILE *ibuf;
192 	char *linebuf;
193 	int linesize;
194 {
195 	int n;
196 
197 	clearerr(ibuf);
198 	if (fgets(linebuf, linesize, ibuf) == NULL)
199 		return (-1);
200 	n = strlen(linebuf);
201 	if (n > 0 && linebuf[n - 1] == '\n')
202 		linebuf[--n] = '\0';
203 	if (n > 0 && linebuf[n - 1] == '\r')
204 		linebuf[--n] = '\0';
205 	return (n);
206 }
207 
208 /*
209  * Return a file buffer all ready to read up the
210  * passed message pointer.
211  */
212 FILE *
213 setinput(mp)
214 	struct message *mp;
215 {
216 
217 	(void)fflush(otf);
218 	if (fseeko(itf,
219 		   positionof(mp->m_block, mp->m_offset), SEEK_SET) < 0)
220 		err(1, "fseeko");
221 	return (itf);
222 }
223 
224 /*
225  * Take the data out of the passed ghost file and toss it into
226  * a dynamically allocated message structure.
227  */
228 void
229 makemessage(f, omsgCount)
230 	FILE *f;
231 	int omsgCount;
232 {
233 	size_t size;
234 	struct message *nmessage;
235 
236 	size = (msgCount + 1) * sizeof(struct message);
237 	nmessage = (struct message *)realloc(message, size);
238 	if (nmessage == NULL)
239 		errx(1, "Insufficient memory for %d messages\n",
240 		    msgCount);
241 	if (omsgCount == 0 || message == NULL)
242 		dot = nmessage;
243 	else
244 		dot = nmessage + (dot - message);
245 	message = nmessage;
246 	size -= (omsgCount + 1) * sizeof(struct message);
247 	(void)fflush(f);
248 	(void)lseek(fileno(f), (off_t)sizeof(*message), 0);
249 	if (read(fileno(f), (char *)&message[omsgCount], size) != size)
250 		errx(1, "Message temporary file corrupted");
251 	message[msgCount].m_size = 0;
252 	message[msgCount].m_lines = 0;
253 	(void)Fclose(f);
254 }
255 
256 /*
257  * Append the passed message descriptor onto the temp file.
258  * If the write fails, return 1, else 0
259  */
260 int
261 append(mp, f)
262 	struct message *mp;
263 	FILE *f;
264 {
265 	return (fwrite((char *)mp, sizeof(*mp), 1, f) != 1);
266 }
267 
268 /*
269  * Delete a file, but only if the file is a plain file.
270  */
271 int
272 rm(name)
273 	char *name;
274 {
275 	struct stat sb;
276 
277 	if (stat(name, &sb) < 0)
278 		return (-1);
279 	if (!S_ISREG(sb.st_mode)) {
280 		errno = EISDIR;
281 		return (-1);
282 	}
283 	return (unlink(name));
284 }
285 
286 static int sigdepth;		/* depth of holdsigs() */
287 static sigset_t nset, oset;
288 /*
289  * Hold signals SIGHUP, SIGINT, and SIGQUIT.
290  */
291 void
292 holdsigs()
293 {
294 
295 	if (sigdepth++ == 0) {
296 		(void)sigemptyset(&nset);
297 		(void)sigaddset(&nset, SIGHUP);
298 		(void)sigaddset(&nset, SIGINT);
299 		(void)sigaddset(&nset, SIGQUIT);
300 		(void)sigprocmask(SIG_BLOCK, &nset, &oset);
301 	}
302 }
303 
304 /*
305  * Release signals SIGHUP, SIGINT, and SIGQUIT.
306  */
307 void
308 relsesigs()
309 {
310 
311 	if (--sigdepth == 0)
312 		(void)sigprocmask(SIG_SETMASK, &oset, NULL);
313 }
314 
315 /*
316  * Determine the size of the file possessed by
317  * the passed buffer.
318  */
319 off_t
320 fsize(iob)
321 	FILE *iob;
322 {
323 	struct stat sbuf;
324 
325 	if (fstat(fileno(iob), &sbuf) < 0)
326 		return (0);
327 	return (sbuf.st_size);
328 }
329 
330 /*
331  * Evaluate the string given as a new mailbox name.
332  * Supported meta characters:
333  *	%	for my system mail box
334  *	%user	for user's system mail box
335  *	#	for previous file
336  *	&	invoker's mbox file
337  *	+file	file in folder directory
338  *	any shell meta character
339  * Return the file name as a dynamic string.
340  */
341 char *
342 expand(name)
343 	char *name;
344 {
345 	char xname[PATHSIZE];
346 	char cmdbuf[PATHSIZE];		/* also used for file names */
347 	int pid, l;
348 	char *cp, *sh;
349 	int pivec[2];
350 	struct stat sbuf;
351 
352 	/*
353 	 * The order of evaluation is "%" and "#" expand into constants.
354 	 * "&" can expand into "+".  "+" can expand into shell meta characters.
355 	 * Shell meta characters expand into constants.
356 	 * This way, we make no recursive expansion.
357 	 */
358 	switch (*name) {
359 	case '%':
360 		findmail(name[1] ? name + 1 : myname, xname, sizeof(xname));
361 		return (savestr(xname));
362 	case '#':
363 		if (name[1] != 0)
364 			break;
365 		if (prevfile[0] == 0) {
366 			printf("No previous file\n");
367 			return (NULL);
368 		}
369 		return (savestr(prevfile));
370 	case '&':
371 		if (name[1] == 0 && (name = value("MBOX")) == NULL)
372 			name = "~/mbox";
373 		/* fall through */
374 	}
375 	if (name[0] == '+' && getfold(cmdbuf, sizeof(cmdbuf)) >= 0) {
376 		(void)snprintf(xname, sizeof(xname), "%s/%s", cmdbuf, name + 1);
377 		name = savestr(xname);
378 	}
379 	/* catch the most common shell meta character */
380 	if (name[0] == '~' && homedir != NULL &&
381 	    (name[1] == '/' || name[1] == '\0')) {
382 		(void)snprintf(xname, sizeof(xname), "%s%s", homedir, name + 1);
383 		name = savestr(xname);
384 	}
385 	if (!strpbrk(name, "~{[*?$`'\"\\"))
386 		return (name);
387 	if (pipe(pivec) < 0) {
388 		warn("pipe");
389 		return (name);
390 	}
391 	(void)snprintf(cmdbuf, sizeof(cmdbuf), "echo %s", name);
392 	if ((sh = value("SHELL")) == NULL)
393 		sh = _PATH_CSHELL;
394 	pid = start_command(sh, 0, -1, pivec[1], "-c", cmdbuf, NULL);
395 	if (pid < 0) {
396 		(void)close(pivec[0]);
397 		(void)close(pivec[1]);
398 		return (NULL);
399 	}
400 	(void)close(pivec[1]);
401 	l = read(pivec[0], xname, BUFSIZ);
402 	(void)close(pivec[0]);
403 	if (wait_child(pid) < 0 && WIFSIGNALED(wait_status) &&
404 	    WTERMSIG(wait_status) != SIGPIPE) {
405 		fprintf(stderr, "\"%s\": Expansion failed.\n", name);
406 		return (NULL);
407 	}
408 	if (l < 0) {
409 		warn("read");
410 		return (NULL);
411 	}
412 	if (l == 0) {
413 		fprintf(stderr, "\"%s\": No match.\n", name);
414 		return (NULL);
415 	}
416 	if (l == BUFSIZ) {
417 		fprintf(stderr, "\"%s\": Expansion buffer overflow.\n", name);
418 		return (NULL);
419 	}
420 	xname[l] = '\0';
421 	for (cp = &xname[l-1]; *cp == '\n' && cp > xname; cp--)
422 		;
423 	cp[1] = '\0';
424 	if (strchr(xname, ' ') && stat(xname, &sbuf) < 0) {
425 		fprintf(stderr, "\"%s\": Ambiguous.\n", name);
426 		return (NULL);
427 	}
428 	return (savestr(xname));
429 }
430 
431 /*
432  * Determine the current folder directory name.
433  */
434 int
435 getfold(name, namelen)
436 	char *name;
437 	int namelen;
438 {
439 	char *folder;
440 	int copylen;
441 
442 	if ((folder = value("folder")) == NULL)
443 		return (-1);
444 	if (*folder == '/')
445 		copylen = strlcpy(name, folder, namelen);
446 	else
447 		copylen = snprintf(name, namelen, "%s/%s",
448 		    homedir ? homedir : ".", folder);
449 	return (copylen < 0 || copylen >= namelen ? (-1) : (0));
450 }
451 
452 /*
453  * Return the name of the dead.letter file.
454  */
455 char *
456 getdeadletter()
457 {
458 	char *cp;
459 
460 	if ((cp = value("DEAD")) == NULL || (cp = expand(cp)) == NULL)
461 		cp = expand("~/dead.letter");
462 	else if (*cp != '/') {
463 		char buf[PATHSIZE];
464 
465 		(void)snprintf(buf, sizeof(buf), "~/%s", cp);
466 		cp = expand(buf);
467 	}
468 	return (cp);
469 }
470