Home | History | Annotate | Line # | Download | only in mail
mime_attach.c revision 1.6
      1  1.6  christos /*	$NetBSD: mime_attach.c,v 1.6 2007/10/29 23:20:38 christos Exp $	*/
      2  1.1  christos 
      3  1.1  christos /*-
      4  1.1  christos  * Copyright (c) 2006 The NetBSD Foundation, Inc.
      5  1.1  christos  * All rights reserved.
      6  1.1  christos  *
      7  1.1  christos  * This code is derived from software contributed to The NetBSD Foundation
      8  1.1  christos  * by Anon Ymous.
      9  1.1  christos  *
     10  1.1  christos  * Redistribution and use in source and binary forms, with or without
     11  1.1  christos  * modification, are permitted provided that the following conditions
     12  1.1  christos  * are met:
     13  1.1  christos  * 1. Redistributions of source code must retain the above copyright
     14  1.1  christos  *    notice, this list of conditions and the following disclaimer.
     15  1.1  christos  * 2. Redistributions in binary form must reproduce the above copyright
     16  1.1  christos  *    notice, this list of conditions and the following disclaimer in the
     17  1.1  christos  *    documentation and/or other materials provided with the distribution.
     18  1.1  christos  * 3. All advertising materials mentioning features or use of this software
     19  1.1  christos  *    must display the following acknowledgement:
     20  1.1  christos  *        This product includes software developed by the NetBSD
     21  1.1  christos  *        Foundation, Inc. and its contributors.
     22  1.1  christos  * 4. Neither the name of The NetBSD Foundation nor the names of its
     23  1.1  christos  *    contributors may be used to endorse or promote products derived
     24  1.1  christos  *    from this software without specific prior written permission.
     25  1.1  christos  *
     26  1.1  christos  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     27  1.1  christos  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     28  1.1  christos  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     29  1.1  christos  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     30  1.1  christos  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     31  1.1  christos  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     32  1.1  christos  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     33  1.1  christos  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     34  1.1  christos  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     35  1.1  christos  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     36  1.1  christos  * POSSIBILITY OF SUCH DAMAGE.
     37  1.1  christos  */
     38  1.1  christos 
     39  1.1  christos #ifdef MIME_SUPPORT
     40  1.1  christos 
     41  1.1  christos #include <sys/cdefs.h>
     42  1.1  christos #ifndef __lint__
     43  1.6  christos __RCSID("$NetBSD: mime_attach.c,v 1.6 2007/10/29 23:20:38 christos Exp $");
     44  1.1  christos #endif /* not __lint__ */
     45  1.1  christos 
     46  1.1  christos #include <assert.h>
     47  1.1  christos #include <err.h>
     48  1.1  christos #include <fcntl.h>
     49  1.1  christos #include <libgen.h>
     50  1.1  christos #include <magic.h>
     51  1.1  christos #include <setjmp.h>
     52  1.1  christos #include <signal.h>
     53  1.1  christos #include <stdio.h>
     54  1.1  christos #include <stdlib.h>
     55  1.1  christos #include <string.h>
     56  1.1  christos #include <unistd.h>
     57  1.1  christos #include <util.h>
     58  1.1  christos 
     59  1.1  christos #include "def.h"
     60  1.1  christos #include "extern.h"
     61  1.1  christos #ifdef USE_EDITLINE
     62  1.1  christos #include "complete.h"
     63  1.1  christos #endif
     64  1.1  christos #ifdef MIME_SUPPORT
     65  1.1  christos #include "mime.h"
     66  1.1  christos #include "mime_codecs.h"
     67  1.1  christos #include "mime_child.h"
     68  1.1  christos #endif
     69  1.1  christos #include "glob.h"
     70  1.1  christos 
     71  1.1  christos #if 0
     72  1.1  christos #ifndef __lint__
     73  1.1  christos /*
     74  1.1  christos  * XXX - This block for debugging only and eventually should go away.
     75  1.1  christos  */
     76  1.1  christos static void
     77  1.1  christos show_name(const char *prefix, struct name *np)
     78  1.1  christos {
     79  1.1  christos 	int i;
     80  1.1  christos 
     81  1.1  christos 	i = 0;
     82  1.5  christos 	for (/*EMPTY*/; np; np = np->n_flink) {
     83  1.1  christos 		(void)printf("%s[%d]: %s\n", prefix, i, np->n_name);
     84  1.1  christos 		i++;
     85  1.1  christos 	}
     86  1.1  christos }
     87  1.1  christos 
     88  1.2  christos static void fput_mime_content(FILE *fp, struct Content *Cp);
     89  1.2  christos 
     90  1.1  christos PUBLIC void
     91  1.1  christos show_attach(const char *prefix, struct attachment *ap)
     92  1.1  christos {
     93  1.1  christos 	int i;
     94  1.1  christos 	i = 1;
     95  1.5  christos 	for (/*EMPTY*/; ap; ap = ap->a_flink) {
     96  1.1  christos 		(void)printf("%s[%d]:\n", prefix, i);
     97  1.1  christos 		fput_mime_content(stdout, &ap->a_Content);
     98  1.1  christos 		i++;
     99  1.1  christos 	}
    100  1.1  christos }
    101  1.1  christos 
    102  1.1  christos PUBLIC void
    103  1.1  christos show_header(struct header *hp)
    104  1.1  christos {
    105  1.1  christos 	show_name("TO", hp->h_to);
    106  1.1  christos 	(void)printf("SUBJECT: %s\n", hp->h_subject);
    107  1.1  christos 	show_name("CC", hp->h_cc);
    108  1.1  christos 	show_name("BCC", hp->h_bcc);
    109  1.1  christos 	show_name("SMOPTS", hp->h_smopts);
    110  1.1  christos 	show_attach("ATTACH", hp->h_attach);
    111  1.1  christos }
    112  1.1  christos #endif	/* __lint__ */
    113  1.1  christos #endif
    114  1.1  christos 
    115  1.1  christos /***************************
    116  1.1  christos  * boundary string routines
    117  1.1  christos  */
    118  1.1  christos static char *
    119  1.1  christos getrandstring(size_t length)
    120  1.1  christos {
    121  1.1  christos 	void *vbin;
    122  1.1  christos 	uint32_t *bin;
    123  1.1  christos 	size_t binlen;
    124  1.1  christos 	size_t i;
    125  1.1  christos 	char *b64;
    126  1.1  christos 
    127  1.1  christos 	/* XXX - check this stuff again!!! */
    128  1.1  christos 
    129  1.1  christos 	binlen = 3 * roundup(length, 4) / 4;	/* bytes of binary to encode base64 */
    130  1.1  christos 	bin = vbin = salloc(roundup(binlen, 4));
    131  1.1  christos 	for (i = 0; i < roundup(binlen, 4) / 4; i++)
    132  1.1  christos 		bin[i] = arc4random();
    133  1.1  christos 
    134  1.1  christos 	b64 = salloc(roundup(length, 4));
    135  1.1  christos 	mime_bintob64(b64, vbin, binlen);
    136  1.1  christos 	b64[length] = '\0';
    137  1.1  christos 
    138  1.1  christos 	return b64;
    139  1.1  christos }
    140  1.1  christos 
    141  1.1  christos /*
    142  1.1  christos  * Generate a boundary for MIME multipart messages.
    143  1.1  christos  */
    144  1.1  christos static char *
    145  1.1  christos make_boundary(void)
    146  1.1  christos {
    147  1.1  christos #define BOUND_LEN 70	/* maximum length is 70 characters: RFC2046 sec 5.1.1 */
    148  1.1  christos 
    149  1.1  christos 	char *bound;
    150  1.1  christos 	time_t	now;
    151  1.1  christos 
    152  1.1  christos 	(void)time(&now);
    153  1.1  christos 	bound = salloc(BOUND_LEN);
    154  1.1  christos 	(void)snprintf(bound, BOUND_LEN, "=_%08lx.%s",
    155  1.1  christos 	    (long)now, getrandstring(BOUND_LEN - 12));
    156  1.1  christos 	return bound;
    157  1.1  christos 
    158  1.1  christos #undef BOUND_LEN
    159  1.1  christos }
    160  1.1  christos 
    161  1.1  christos /***************************
    162  1.1  christos  * Transfer coding routines
    163  1.1  christos  */
    164  1.1  christos /*
    165  1.1  christos  * We determine the recommended transfer encoding type for a file as
    166  1.1  christos  * follows:
    167  1.1  christos  *
    168  1.1  christos  * 1) If there is a NULL byte or a stray CR (not in a CRLF
    169  1.1  christos  *    combination) in the file, play it safe and use base64.
    170  1.1  christos  *
    171  1.1  christos  * 2) If any high bit is set, use quoted-printable if the content type
    172  1.1  christos  *    is "text" and base64 otherwise.
    173  1.1  christos  *
    174  1.1  christos  * 3) Otherwise:
    175  1.1  christos  *    a) use quoted-printable if there are any long lines, control
    176  1.1  christos  *       chars (including CR), end-of-line blank space, or a missing
    177  1.1  christos  *       terminating NL.
    178  1.1  christos  *    b) use 7bit in all remaining case, including an empty file.
    179  1.1  christos  *
    180  1.1  christos  * NOTE: This means that CRLF text (MSDOS) files will be encoded
    181  1.1  christos  * quoted-printable.
    182  1.5  christos  */
    183  1.1  christos /*
    184  1.1  christos  * RFC 821 imposes the following line length limit:
    185  1.1  christos  *  The maximum total length of a text line including the
    186  1.1  christos  *  <CRLF> is 1000 characters (but not counting the leading
    187  1.1  christos  *  dot duplicated for transparency).
    188  1.1  christos  */
    189  1.1  christos #define MIME_UNENCODED_LINE_MAX	(1000 - 2)
    190  1.1  christos static size_t
    191  1.1  christos line_limit(void)
    192  1.1  christos {
    193  1.1  christos 	int limit;
    194  1.1  christos 	const char *cp;
    195  1.1  christos 	limit = -1;
    196  1.1  christos 
    197  1.1  christos 	if ((cp = value(ENAME_MIME_UNENC_LINE_MAX)) != NULL)
    198  1.1  christos 		limit = atoi(cp);
    199  1.1  christos 
    200  1.1  christos 	if (limit < 0 || limit > MIME_UNENCODED_LINE_MAX)
    201  1.1  christos 		limit = MIME_UNENCODED_LINE_MAX;
    202  1.1  christos 
    203  1.1  christos 	return (size_t)limit;
    204  1.1  christos }
    205  1.1  christos 
    206  1.3  christos static inline int
    207  1.3  christos is_text(const char *ctype)
    208  1.1  christos {
    209  1.1  christos 	return ctype &&
    210  1.1  christos 	    strncasecmp(ctype, "text/", sizeof("text/") - 1) == 0;
    211  1.1  christos }
    212  1.1  christos 
    213  1.1  christos static const char *
    214  1.1  christos content_encoding_core(void *fh, const char *ctype)
    215  1.1  christos {
    216  1.1  christos 	int c, lastc;
    217  1.1  christos 	int ctrlchar, endwhite;
    218  1.1  christos 	size_t curlen, maxlen;
    219  1.5  christos 
    220  1.1  christos 	curlen = 0;
    221  1.1  christos 	maxlen = 0;
    222  1.1  christos 	ctrlchar = 0;
    223  1.1  christos 	endwhite = 0;
    224  1.1  christos 	lastc = EOF;
    225  1.1  christos 	while ((c = fgetc(fh)) != EOF) {
    226  1.1  christos 		curlen++;
    227  1.5  christos 
    228  1.1  christos 		if (c == '\0' || (lastc == '\r' && c != '\n'))
    229  1.1  christos 			return MIME_TRANSFER_BASE64;
    230  1.1  christos 
    231  1.1  christos 		if (c > 0x7f) {
    232  1.1  christos 			if (is_text(ctype))
    233  1.1  christos 				return MIME_TRANSFER_QUOTED;
    234  1.1  christos 			else
    235  1.1  christos 				return MIME_TRANSFER_BASE64;
    236  1.1  christos 		}
    237  1.1  christos 		if (c == '\n') {
    238  1.5  christos 			if (is_WSP(lastc))
    239  1.1  christos 				endwhite = 1;
    240  1.1  christos 			if (curlen > maxlen)
    241  1.1  christos 				maxlen = curlen;
    242  1.1  christos 			curlen = 0;
    243  1.1  christos 		}
    244  1.1  christos 		else if ((c < 0x20 && c != '\t') || c == 0x7f)
    245  1.1  christos 			ctrlchar = 1;
    246  1.5  christos 
    247  1.1  christos 		lastc = c;
    248  1.1  christos 	}
    249  1.1  christos 	if (lastc == EOF) /* no characters read */
    250  1.1  christos 		return MIME_TRANSFER_7BIT;
    251  1.1  christos 
    252  1.1  christos 	if (lastc != '\n' || ctrlchar || endwhite || maxlen > line_limit())
    253  1.1  christos 		return MIME_TRANSFER_QUOTED;
    254  1.1  christos 
    255  1.1  christos 	return MIME_TRANSFER_7BIT;
    256  1.1  christos }
    257  1.1  christos 
    258  1.1  christos static const char *
    259  1.1  christos content_encoding_by_name(const char *filename, const char *ctype)
    260  1.1  christos {
    261  1.1  christos 	FILE *fp;
    262  1.1  christos 	const char *enc;
    263  1.1  christos 	fp = fopen(filename, "r");
    264  1.1  christos 	if (fp == NULL) {
    265  1.1  christos 		warn("content_encoding_by_name: %s", filename);
    266  1.1  christos 		return MIME_TRANSFER_BASE64;	/* safe */
    267  1.1  christos 	}
    268  1.1  christos 	enc = content_encoding_core(fp, ctype);
    269  1.1  christos 	(void)fclose(fp);
    270  1.1  christos 	return enc;
    271  1.1  christos }
    272  1.1  christos 
    273  1.1  christos static const char *
    274  1.1  christos content_encoding_by_fileno(int fd, const char *ctype)
    275  1.1  christos {
    276  1.1  christos 	FILE *fp;
    277  1.1  christos 	const char *encoding;
    278  1.1  christos 	off_t cur_pos;
    279  1.1  christos 
    280  1.1  christos 	cur_pos = lseek(fd, (off_t)0, SEEK_CUR);
    281  1.1  christos 	fp = fdopen(fd, "r");
    282  1.1  christos 	if (fp == NULL) {
    283  1.1  christos 		warn("content_encoding_by_fileno");
    284  1.1  christos 		return MIME_TRANSFER_BASE64;
    285  1.1  christos 	}
    286  1.1  christos 	encoding = content_encoding_core(fp, ctype);
    287  1.1  christos 	(void)lseek(fd, cur_pos, SEEK_SET);
    288  1.1  christos 	return encoding;
    289  1.1  christos }
    290  1.1  christos 
    291  1.1  christos static const char *
    292  1.1  christos content_encoding(struct attachment *attach, const char *ctype)
    293  1.1  christos {
    294  1.1  christos 	switch (attach->a_type) {
    295  1.1  christos 	case ATTACH_FNAME:
    296  1.1  christos 		return content_encoding_by_name(attach->a_name, ctype);
    297  1.1  christos 	case ATTACH_MSG:
    298  1.1  christos 		errx(EXIT_FAILURE, "msgno not supported yet\n");
    299  1.1  christos 		/* NOTREACHED */
    300  1.1  christos 	case ATTACH_FILENO:
    301  1.1  christos 		return content_encoding_by_fileno(attach->a_fileno, ctype);
    302  1.1  christos 	default:
    303  1.1  christos 		errx(EXIT_FAILURE, "invalid attach type: %d\n", attach->a_type);
    304  1.1  christos 	}
    305  1.1  christos 	/* NOTREACHED */
    306  1.1  christos }
    307  1.1  christos 
    308  1.1  christos /************************
    309  1.1  christos  * Content type routines
    310  1.1  christos  */
    311  1.1  christos /*
    312  1.1  christos  * We use libmagic(3) to get the content type, except in the case of a
    313  1.4  christos  * 0 or 1 byte file where libmagic gives rather useless results.
    314  1.1  christos  */
    315  1.1  christos static const char *
    316  1.4  christos content_type_by_name(char *filename)
    317  1.1  christos {
    318  1.1  christos 	const char *cp;
    319  1.4  christos 	char *cp2;
    320  1.1  christos 	magic_t magic;
    321  1.1  christos 	struct stat sb;
    322  1.1  christos 
    323  1.1  christos 	/*
    324  1.1  christos 	 * libmagic produces annoying results on very short files.
    325  1.4  christos 	 * The common case is with mime-encode-message defined and an
    326  1.4  christos 	 * empty message body.
    327  1.4  christos 	 *
    328  1.4  christos 	 * Note: a 1-byte message body always consists of a newline,
    329  1.4  christos 	 * so size determines all there.  However, 1-byte attachments
    330  1.4  christos 	 * (filename != NULL) could be anything, so check those.
    331  1.1  christos 	 */
    332  1.1  christos 	if ((filename != NULL && stat(filename, &sb) == 0) ||
    333  1.4  christos 	    (filename == NULL && fstat(0, &sb) == 0)) {
    334  1.4  christos 		if (sb.st_size < 2 && S_ISREG(sb.st_mode)) {
    335  1.4  christos 			FILE *fp;
    336  1.4  christos 			int ch;
    337  1.4  christos 			if (sb.st_size == 0 || filename == NULL ||
    338  1.4  christos 			    (fp = fopen(filename, "r")) == NULL)
    339  1.4  christos 				return "text/plain";
    340  1.4  christos 
    341  1.4  christos 			ch = fgetc(fp);
    342  1.4  christos 			(void)fclose(fp);
    343  1.4  christos 
    344  1.4  christos 			return isprint(ch) || isspace(ch) ?
    345  1.4  christos 			    "text/plain" : "application/octet-stream";
    346  1.4  christos 		}
    347  1.4  christos 	}
    348  1.1  christos 	magic = magic_open(MAGIC_MIME);
    349  1.1  christos 	if (magic == NULL) {
    350  1.1  christos 		warn("magic_open: %s", magic_error(magic));
    351  1.1  christos 		return NULL;
    352  1.1  christos 	}
    353  1.1  christos 	if (magic_load(magic, NULL) != 0) {
    354  1.1  christos 		warn("magic_load: %s", magic_error(magic));
    355  1.1  christos 		return NULL;
    356  1.1  christos 	}
    357  1.1  christos 	cp = magic_file(magic, filename);
    358  1.1  christos 	if (cp == NULL) {
    359  1.1  christos 		warn("magic_load: %s", magic_error(magic));
    360  1.1  christos 		return NULL;
    361  1.1  christos 	}
    362  1.4  christos 	if (filename &&
    363  1.4  christos 	    sasprintf(&cp2, "%s; name=\"%s\"", cp, basename(filename)) != -1)
    364  1.4  christos 		cp = cp2;
    365  1.4  christos 	else
    366  1.4  christos 		cp = savestr(cp);
    367  1.1  christos 	magic_close(magic);
    368  1.1  christos 	return cp;
    369  1.1  christos }
    370  1.1  christos 
    371  1.1  christos static const char *
    372  1.1  christos content_type_by_fileno(int fd)
    373  1.1  christos {
    374  1.1  christos 	const char *cp;
    375  1.1  christos 	off_t cur_pos;
    376  1.1  christos 	int ofd;
    377  1.1  christos 
    378  1.1  christos 	cur_pos = lseek(fd, (off_t)0, SEEK_CUR);
    379  1.1  christos 
    380  1.1  christos 	ofd = dup(0);		/* save stdin */
    381  1.1  christos 	if (dup2(fd, 0) == -1)	/* become stdin */
    382  1.1  christos 		warn("dup2");
    383  1.1  christos 
    384  1.1  christos 	cp = content_type_by_name(NULL);
    385  1.1  christos 
    386  1.1  christos 	if (dup2(ofd, 0) == -1)	/* restore stdin */
    387  1.1  christos 		warn("dup2");
    388  1.1  christos 	(void)close(ofd);	/* close the copy */
    389  1.1  christos 
    390  1.5  christos 	(void)lseek(fd, cur_pos, SEEK_SET);
    391  1.1  christos 	return cp;
    392  1.1  christos }
    393  1.1  christos 
    394  1.1  christos static const char *
    395  1.1  christos content_type(struct attachment *attach)
    396  1.1  christos {
    397  1.1  christos 	switch (attach->a_type) {
    398  1.1  christos 	case ATTACH_FNAME:
    399  1.1  christos 		return content_type_by_name(attach->a_name);
    400  1.1  christos 	case ATTACH_MSG:
    401  1.1  christos 		return "message/rfc822";
    402  1.1  christos 	case ATTACH_FILENO:
    403  1.1  christos 		return content_type_by_fileno(attach->a_fileno);
    404  1.1  christos 	default:
    405  1.1  christos 		/* This is a coding error! */
    406  1.1  christos 		assert(/* CONSTCOND */ 0);
    407  1.1  christos 		return NULL;
    408  1.1  christos 	}
    409  1.1  christos }
    410  1.1  christos 
    411  1.1  christos /*************************
    412  1.1  christos  * Other content routines
    413  1.1  christos  */
    414  1.1  christos 
    415  1.1  christos static const char *
    416  1.1  christos content_disposition(struct attachment *ap)
    417  1.1  christos {
    418  1.1  christos 	switch (ap->a_type) {
    419  1.1  christos 	case ATTACH_FNAME: {
    420  1.1  christos 		char *disp;
    421  1.3  christos 		(void)sasprintf(&disp, "attachment; filename=\"%s\"", basename(ap->a_name));
    422  1.1  christos 		return disp;
    423  1.1  christos 	}
    424  1.1  christos 	case ATTACH_MSG:
    425  1.1  christos 	case ATTACH_FILENO:
    426  1.1  christos 		return "inline";
    427  1.5  christos 
    428  1.1  christos 	default:
    429  1.1  christos 		return NULL;
    430  1.1  christos 	}
    431  1.1  christos }
    432  1.1  christos 
    433  1.5  christos /*ARGSUSED*/
    434  1.1  christos static const char *
    435  1.1  christos content_id(struct attachment *attach __unused)
    436  1.1  christos {
    437  1.1  christos 	/* XXX - to be written. */
    438  1.5  christos 
    439  1.1  christos 	return NULL;
    440  1.1  christos }
    441  1.1  christos 
    442  1.1  christos static const char *
    443  1.1  christos content_description(struct attachment *attach, int attach_num)
    444  1.1  christos {
    445  1.1  christos 	if (attach_num) {
    446  1.1  christos 		char *description;
    447  1.3  christos 		(void)sasprintf(&description, "attachment %d", attach_num);
    448  1.1  christos 		return description;
    449  1.1  christos 	}
    450  1.1  christos 	else
    451  1.1  christos 		return attach->a_Content.C_description;
    452  1.1  christos }
    453  1.1  christos 
    454  1.1  christos /*******************************************
    455  1.1  christos  * Routines to get the MIME content strings.
    456  1.1  christos  */
    457  1.1  christos static struct Content
    458  1.1  christos get_mime_content(struct attachment *ap, int i)
    459  1.1  christos {
    460  1.1  christos 	struct Content Cp;
    461  1.1  christos 
    462  1.1  christos 	Cp.C_type	 = content_type(ap);
    463  1.1  christos 	Cp.C_encoding	 = content_encoding(ap, Cp.C_type);
    464  1.1  christos 	Cp.C_disposition = content_disposition(ap);
    465  1.1  christos 	Cp.C_id		 = content_id(ap);
    466  1.1  christos 	Cp.C_description = content_description(ap, i);
    467  1.1  christos 
    468  1.1  christos 	return Cp;
    469  1.1  christos }
    470  1.1  christos 
    471  1.1  christos /******************
    472  1.1  christos  * Output routines
    473  1.1  christos  */
    474  1.1  christos static void
    475  1.1  christos fput_mime_content(FILE *fp, struct Content *Cp)
    476  1.1  christos {
    477  1.1  christos 	(void)fprintf(fp, MIME_HDR_TYPE ": %s\n", Cp->C_type);
    478  1.1  christos 	(void)fprintf(fp, MIME_HDR_ENCODING ": %s\n", Cp->C_encoding);
    479  1.1  christos 	if (Cp->C_disposition)
    480  1.1  christos 		(void)fprintf(fp, MIME_HDR_DISPOSITION ": %s\n",
    481  1.1  christos 		    Cp->C_disposition);
    482  1.1  christos 	if (Cp->C_id)
    483  1.1  christos 		(void)fprintf(fp, MIME_HDR_ID ": %s\n", Cp->C_id);
    484  1.1  christos 	if (Cp->C_description)
    485  1.1  christos 		(void)fprintf(fp, MIME_HDR_DESCRIPTION ": %s\n",
    486  1.1  christos 		    Cp->C_description);
    487  1.1  christos }
    488  1.1  christos 
    489  1.1  christos static void
    490  1.1  christos fput_body(FILE *fi, FILE *fo, struct Content *Cp)
    491  1.1  christos {
    492  1.1  christos 	mime_codec_t enc;
    493  1.1  christos 
    494  1.1  christos 	enc = mime_fio_encoder(Cp->C_encoding);
    495  1.1  christos 	if (enc == NULL)
    496  1.1  christos 		warnx("unknown transfer encoding type: %s\n", Cp->C_encoding);
    497  1.1  christos 	else
    498  1.1  christos 		enc(fi, fo, 0);
    499  1.1  christos }
    500  1.1  christos 
    501  1.1  christos static void
    502  1.1  christos fput_attachment(FILE *fo, struct attachment *ap)
    503  1.1  christos {
    504  1.1  christos 	FILE *fi;
    505  1.1  christos 	struct Content *Cp = &ap->a_Content;
    506  1.1  christos 
    507  1.1  christos 	fput_mime_content(fo, &ap->a_Content);
    508  1.1  christos 	(void)putc('\n', fo);
    509  1.1  christos 
    510  1.1  christos 	switch (ap->a_type) {
    511  1.1  christos 	case ATTACH_FNAME:
    512  1.1  christos 		fi = fopen(ap->a_name, "r");
    513  1.1  christos 		if (fi == NULL)
    514  1.1  christos 			err(EXIT_FAILURE, "fopen: %s", ap->a_name);
    515  1.1  christos 		break;
    516  1.1  christos 
    517  1.1  christos 	case ATTACH_FILENO:
    518  1.1  christos 		fi = fdopen(ap->a_fileno, "r");
    519  1.1  christos 		if (fi == NULL)
    520  1.1  christos 			err(EXIT_FAILURE, "fdopen: %d", ap->a_fileno);
    521  1.1  christos 		break;
    522  1.1  christos 
    523  1.1  christos 	case ATTACH_MSG:
    524  1.1  christos 	default:
    525  1.1  christos 		errx(EXIT_FAILURE, "unsupported attachment type");
    526  1.1  christos 	}
    527  1.1  christos 
    528  1.1  christos 	fput_body(fi, fo, Cp);
    529  1.1  christos 
    530  1.1  christos 	if (ap->a_type == ATTACH_FNAME)
    531  1.1  christos 		(void)fclose(fi);
    532  1.1  christos }
    533  1.1  christos 
    534  1.1  christos /***********************************
    535  1.1  christos  * Higher level attachment routines.
    536  1.1  christos  */
    537  1.1  christos 
    538  1.1  christos static int
    539  1.1  christos mktemp_file(FILE **nfo, FILE **nfi, const char *hint)
    540  1.1  christos {
    541  1.1  christos 	char tempname[PATHSIZE];
    542  1.1  christos 	int fd, fd2;
    543  1.1  christos 	(void)snprintf(tempname, sizeof(tempname), "%s/%sXXXXXXXXXX",
    544  1.1  christos 	    tmpdir, hint);
    545  1.1  christos 	if ((fd = mkstemp(tempname)) == -1 ||
    546  1.1  christos 	    (*nfo = Fdopen(fd, "w")) == NULL) {
    547  1.1  christos 		if (fd != -1)
    548  1.1  christos 			(void)close(fd);
    549  1.1  christos 		warn("%s", tempname);
    550  1.1  christos 		return -1;
    551  1.1  christos 	}
    552  1.1  christos 	(void)rm(tempname);
    553  1.1  christos 	if ((fd2 = dup(fd)) == -1 ||
    554  1.1  christos 	    (*nfi = Fdopen(fd2, "r")) == NULL) {
    555  1.1  christos 		warn("%s", tempname);
    556  1.1  christos 		(void)Fclose(*nfo);
    557  1.1  christos 		return -1;
    558  1.1  christos 	}
    559  1.1  christos 	return 0;
    560  1.1  christos }
    561  1.1  christos 
    562  1.1  christos /*
    563  1.1  christos  * Repackage the mail as a multipart MIME message.  This should always
    564  1.1  christos  * be called whenever there are attachments, but might be called even
    565  1.1  christos  * if there are none if we want to wrap the message in a MIME package.
    566  1.1  christos  */
    567  1.1  christos PUBLIC FILE *
    568  1.1  christos mime_encode(FILE *fi, struct header *header)
    569  1.1  christos {
    570  1.1  christos 	struct attachment map;	/* fake structure for the message body */
    571  1.1  christos 	struct attachment *attach;
    572  1.1  christos 	struct attachment *ap;
    573  1.1  christos 	FILE *nfi, *nfo;
    574  1.1  christos 
    575  1.1  christos 	attach = header->h_attach;
    576  1.1  christos 
    577  1.1  christos 	/*
    578  1.1  christos 	 * Make new phantom temporary file with read and write file
    579  1.1  christos 	 * handles: nfi and nfo, resp.
    580  1.1  christos 	 */
    581  1.1  christos 	if (mktemp_file(&nfo, &nfi, "mail.Rs") != 0)
    582  1.1  christos 		return fi;
    583  1.1  christos 
    584  1.1  christos 	(void)memset(&map, 0, sizeof(map));
    585  1.1  christos 	map.a_type = ATTACH_FILENO;
    586  1.1  christos 	map.a_fileno = fileno(fi);
    587  1.1  christos 
    588  1.1  christos  	map.a_Content = get_mime_content(&map, 0);
    589  1.1  christos 
    590  1.1  christos 	if (attach) {
    591  1.1  christos 		/* Multi-part message:
    592  1.1  christos 		 * Make an attachment structure for the body message
    593  1.1  christos 		 * and make that the first element in the attach list.
    594  1.1  christos 		 */
    595  1.1  christos 		if (fsize(fi)) {
    596  1.1  christos 			map.a_flink = attach;
    597  1.1  christos 			attach->a_blink = &map;
    598  1.1  christos 			attach = &map;
    599  1.1  christos 		}
    600  1.1  christos 
    601  1.1  christos 		/* Construct our MIME boundary string - used by mime_putheader() */
    602  1.1  christos 		header->h_mime_boundary = make_boundary();
    603  1.5  christos 
    604  1.1  christos 		(void)fprintf(nfo, "This is a multi-part message in MIME format.\n");
    605  1.5  christos 
    606  1.1  christos 		for (ap = attach; ap; ap = ap->a_flink) {
    607  1.1  christos 			(void)fprintf(nfo, "\n--%s\n", header->h_mime_boundary);
    608  1.1  christos 			fput_attachment(nfo, ap);
    609  1.1  christos 		}
    610  1.5  christos 
    611  1.1  christos 		/* the final boundary with two attached dashes */
    612  1.1  christos 		(void)fprintf(nfo, "\n--%s--\n", header->h_mime_boundary);
    613  1.1  christos 	}
    614  1.1  christos 	else {
    615  1.1  christos 		/* Single-part message (no attachments):
    616  1.1  christos 		 * Update header->h_Content (used by mime_putheader()).
    617  1.1  christos 		 * Output the body contents.
    618  1.1  christos 		 */
    619  1.1  christos 		char *encoding;
    620  1.1  christos 
    621  1.1  christos 		header->h_Content = map.a_Content;
    622  1.5  christos 
    623  1.1  christos 		/* check for an encoding override */
    624  1.1  christos 		if ((encoding = value(ENAME_MIME_ENCODE_MSG)) && *encoding)
    625  1.1  christos 			header->h_Content.C_encoding = encoding;
    626  1.1  christos 
    627  1.1  christos 		fput_body(fi, nfo, &header->h_Content);
    628  1.1  christos 	}
    629  1.1  christos 	(void)Fclose(fi);
    630  1.1  christos 	(void)Fclose(nfo);
    631  1.1  christos 	rewind(nfi);
    632  1.3  christos 	return nfi;
    633  1.1  christos }
    634  1.1  christos 
    635  1.1  christos static char*
    636  1.1  christos check_filename(char *filename, char *canon_name)
    637  1.1  christos {
    638  1.1  christos 	int fd;
    639  1.1  christos 	struct stat sb;
    640  1.1  christos 	char *fname = filename;
    641  1.3  christos 
    642  1.2  christos 	/* We need to expand '~' if we got here from '~@'.  The shell
    643  1.2  christos 	 * does this otherwise.
    644  1.2  christos 	 */
    645  1.1  christos 	if (fname[0] == '~' && fname[1] == '/') {
    646  1.2  christos 		if (homedir && homedir[0] != '~')
    647  1.2  christos 			(void)easprintf(&fname, "%s/%s",
    648  1.2  christos 			    homedir, fname + 2);
    649  1.1  christos 	}
    650  1.1  christos 	if (realpath(fname, canon_name) == NULL) {
    651  1.1  christos 		warn("realpath: %s", filename);
    652  1.1  christos 		canon_name = NULL;
    653  1.1  christos 		goto done;
    654  1.1  christos 	}
    655  1.1  christos 	fd = open(canon_name, O_RDONLY, 0);
    656  1.1  christos 	if (fd == -1) {
    657  1.1  christos 		warnx("open: cannot read %s", filename);
    658  1.1  christos 		canon_name = NULL;
    659  1.1  christos 		goto done;
    660  1.1  christos 	}
    661  1.1  christos 	if (fstat(fd, &sb) == -1) {
    662  1.1  christos 		warn("stat: %s", canon_name);
    663  1.1  christos 		canon_name = NULL;
    664  1.1  christos 		goto do_close;
    665  1.1  christos 	}
    666  1.1  christos 	if (!S_ISREG(sb.st_mode)) {
    667  1.1  christos 		warnx("stat: %s is not a file", filename);
    668  1.1  christos 		canon_name = NULL;
    669  1.2  christos 	     /*	goto do_close; */
    670  1.1  christos 	}
    671  1.1  christos  do_close:
    672  1.1  christos 	(void)close(fd);
    673  1.1  christos  done:
    674  1.1  christos 	if (fname != filename)
    675  1.1  christos 		free(fname);
    676  1.1  christos 
    677  1.1  christos 	return canon_name;
    678  1.1  christos }
    679  1.1  christos 
    680  1.1  christos static struct attachment *
    681  1.1  christos attach_one_file(struct attachment *attach, char *filename, int attach_num)
    682  1.1  christos {
    683  1.1  christos 	char canon_name[MAXPATHLEN];
    684  1.1  christos 	struct attachment *ap, *nap;
    685  1.1  christos 
    686  1.3  christos 	/*
    687  1.3  christos 	 * 1) check that filename is really a readable file.
    688  1.3  christos 	 * 2) allocate an attachment structure.
    689  1.3  christos 	 * 3) save cananonical name for filename, so cd won't screw things later.
    690  1.3  christos 	 * 4) add the structure to the end of the chain.
    691  1.3  christos 	 */
    692  1.1  christos 	if (check_filename(filename, canon_name) == NULL)
    693  1.1  christos 		return NULL;
    694  1.5  christos 
    695  1.1  christos 	nap = csalloc(1, sizeof(*nap));
    696  1.1  christos 	nap->a_type = ATTACH_FNAME;
    697  1.1  christos 	nap->a_name = savestr(canon_name);
    698  1.1  christos 
    699  1.1  christos 	if (attach == NULL)
    700  1.1  christos 		attach = nap;
    701  1.5  christos 	else {
    702  1.1  christos 		for (ap = attach; ap->a_flink != NULL; ap = ap->a_flink)
    703  1.1  christos 			continue;
    704  1.1  christos 		ap->a_flink = nap;
    705  1.1  christos 		nap->a_blink = ap;
    706  1.1  christos 	}
    707  1.1  christos 
    708  1.1  christos 	if (attach_num)
    709  1.1  christos 		nap->a_Content = get_mime_content(nap, attach_num);
    710  1.1  christos 
    711  1.1  christos 	return attach;
    712  1.1  christos }
    713  1.1  christos 
    714  1.1  christos static char *
    715  1.1  christos get_line(el_mode_t *em, const char *pr, const char *str, int i)
    716  1.1  christos {
    717  1.3  christos 	char *prompt;
    718  1.1  christos 	char *line;
    719  1.1  christos 
    720  1.3  christos 	/*
    721  1.3  christos 	 * Don't use a '\t' in the format string here as completion
    722  1.3  christos 	 * seems to handle it badly.
    723  1.3  christos 	 */
    724  1.1  christos 	(void)easprintf(&prompt, "#%-8d%s: ", i, pr);
    725  1.3  christos 	line = my_getline(em, prompt, __UNCONST(str));
    726  1.1  christos 	/* LINTED */
    727  1.1  christos 	line = line ? savestr(line) : __UNCONST("");
    728  1.1  christos 	free(prompt);
    729  1.1  christos 
    730  1.3  christos 	return line;
    731  1.1  christos }
    732  1.1  christos 
    733  1.1  christos static void
    734  1.1  christos sget_line(el_mode_t *em, const char *pr, const char **str, int i)
    735  1.1  christos {
    736  1.1  christos 	char *line;
    737  1.1  christos 	line = get_line(em, pr, *str, i);
    738  1.1  christos 	if (strcmp(line, *str) != 0)
    739  1.1  christos 		*str = savestr(line);
    740  1.1  christos }
    741  1.1  christos 
    742  1.1  christos static void
    743  1.1  christos sget_encoding(const char **str, const char *filename, const char *ctype, int num)
    744  1.1  christos {
    745  1.1  christos 	const char *ename;
    746  1.1  christos 	const char *defename;
    747  1.1  christos 
    748  1.1  christos 	defename = NULL;
    749  1.1  christos 	ename = *str;
    750  1.1  christos 	for (;;) {
    751  1.1  christos 		ename = get_line(&elm.mime_enc, "encoding", ename, num);
    752  1.1  christos 
    753  1.1  christos 		if (*ename == '\0') {
    754  1.1  christos 			if (defename == NULL)
    755  1.1  christos 				defename = content_encoding_by_name(filename, ctype);
    756  1.1  christos 			ename = defename;
    757  1.1  christos 		}
    758  1.1  christos 		else if (mime_fio_encoder(ename) == NULL) {
    759  1.1  christos 			const void *cookie;
    760  1.1  christos 			(void)printf("Sorry: valid encoding modes are: ");
    761  1.1  christos 			cookie = NULL;
    762  1.1  christos 			ename = mime_next_encoding_name(&cookie);
    763  1.1  christos 			for (;;) {
    764  1.1  christos 				(void)printf("%s", ename);
    765  1.1  christos 				ename = mime_next_encoding_name(&cookie);
    766  1.1  christos 				if (ename == NULL)
    767  1.1  christos 					break;
    768  1.1  christos 				(void)fputc(',', stdout);
    769  1.1  christos 			}
    770  1.1  christos 			ename = *str;
    771  1.1  christos 		}
    772  1.1  christos 		else {
    773  1.1  christos 			if (strcmp(ename, *str) != 0)
    774  1.1  christos 				*str = savestr(ename);
    775  1.1  christos 			break;
    776  1.1  christos 		}
    777  1.1  christos 	}
    778  1.1  christos }
    779  1.1  christos 
    780  1.1  christos static struct attachment *
    781  1.1  christos edit_attachments(struct attachment *attach)
    782  1.1  christos {
    783  1.1  christos 	char canon_name[MAXPATHLEN];
    784  1.1  christos 	struct attachment *ap;
    785  1.1  christos 	char *line;
    786  1.2  christos 	int attach_num;
    787  1.1  christos 
    788  1.1  christos 	(void)printf("Attachments:\n");
    789  1.1  christos 
    790  1.2  christos 	attach_num = 1;
    791  1.1  christos 	ap = attach;
    792  1.1  christos 	while (ap) {
    793  1.2  christos 		line = get_line(&elm.filec, "filename", ap->a_name, attach_num);
    794  1.1  christos 		if (*line == '\0') {	/* omit this attachment */
    795  1.1  christos 			if (ap->a_blink)
    796  1.1  christos 				ap->a_blink->a_flink = ap->a_flink;
    797  1.1  christos 			else
    798  1.1  christos 				attach = ap->a_flink;
    799  1.1  christos 		}
    800  1.1  christos 		else {
    801  1.1  christos 			if (strcmp(line, ap->a_name) != 0) { /* new filename */
    802  1.1  christos 				if (check_filename(line, canon_name) == NULL)
    803  1.1  christos 					continue;
    804  1.1  christos 				ap->a_name = savestr(canon_name);
    805  1.1  christos 				ap->a_Content = get_mime_content(ap, 0);
    806  1.1  christos 			}
    807  1.2  christos 			sget_line(&elm.string, "description",
    808  1.2  christos 			    &ap->a_Content.C_description, attach_num);
    809  1.2  christos 			sget_encoding(&ap->a_Content.C_encoding, ap->a_name,
    810  1.2  christos 			    ap->a_Content.C_type, attach_num);
    811  1.1  christos 		}
    812  1.2  christos 		attach_num++;
    813  1.1  christos 		if (ap->a_flink == NULL)
    814  1.1  christos 			break;
    815  1.1  christos 
    816  1.1  christos 		ap = ap->a_flink;
    817  1.1  christos 	}
    818  1.1  christos 
    819  1.1  christos 	do {
    820  1.1  christos 		struct attachment *nap;
    821  1.1  christos 
    822  1.2  christos 		line = get_line(&elm.filec, "filename", "", attach_num);
    823  1.1  christos 		if (*line == '\0')
    824  1.1  christos 			break;
    825  1.1  christos 
    826  1.2  christos 		nap = attach_one_file(ap, line, attach_num);
    827  1.1  christos 		if (nap == NULL)
    828  1.1  christos 			continue;
    829  1.1  christos 
    830  1.1  christos 		if (ap)
    831  1.1  christos 			ap = ap->a_flink;
    832  1.1  christos 		else
    833  1.1  christos 			ap = attach = nap;
    834  1.1  christos 
    835  1.2  christos 		sget_line(&elm.string, "description",
    836  1.2  christos 		    &ap->a_Content.C_description, attach_num);
    837  1.2  christos 		sget_encoding(&ap->a_Content.C_encoding, ap->a_name,
    838  1.2  christos 		    ap->a_Content.C_type, attach_num);
    839  1.2  christos 		attach_num++;
    840  1.1  christos 
    841  1.1  christos 	} while (ap);
    842  1.1  christos 
    843  1.1  christos 	return attach;
    844  1.1  christos }
    845  1.1  christos 
    846  1.1  christos /*
    847  1.2  christos  * Hook used by the '~@' escape to attach files.
    848  1.1  christos  */
    849  1.1  christos PUBLIC struct attachment*
    850  1.2  christos mime_attach_files(struct attachment *attach, char *linebuf)
    851  1.1  christos {
    852  1.1  christos 	struct attachment *ap;
    853  1.1  christos 	char *argv[MAXARGC];
    854  1.1  christos 	int argc;
    855  1.1  christos 	int attach_num;
    856  1.1  christos 
    857  1.2  christos 	argc = getrawlist(linebuf, argv, sizeofarray(argv));
    858  1.2  christos 	attach_num = 1;
    859  1.1  christos 	for (ap = attach; ap && ap->a_flink; ap = ap->a_flink)
    860  1.1  christos 			attach_num++;
    861  1.1  christos 
    862  1.1  christos 	if (argc) {
    863  1.1  christos 		int i;
    864  1.1  christos 		for (i = 0; i < argc; i++) {
    865  1.2  christos 			struct attachment *ap2;
    866  1.2  christos 			ap2 = attach_one_file(ap, argv[i], attach_num);
    867  1.2  christos 			if (ap2 != NULL) {
    868  1.2  christos 				ap = ap2;
    869  1.2  christos 				if (attach == NULL)
    870  1.2  christos 					attach = ap;
    871  1.2  christos 				attach_num++;
    872  1.2  christos 			}
    873  1.1  christos 		}
    874  1.1  christos 	}
    875  1.1  christos 	else {
    876  1.1  christos 		attach = edit_attachments(attach);
    877  1.1  christos 		(void)printf("--- end attachments ---\n");
    878  1.1  christos 	}
    879  1.1  christos 
    880  1.1  christos 	return attach;
    881  1.1  christos }
    882  1.1  christos 
    883  1.1  christos /*
    884  1.2  christos  * Hook called in main() to attach files registered by the '-a' flag.
    885  1.2  christos  */
    886  1.2  christos PUBLIC struct attachment *
    887  1.2  christos mime_attach_optargs(struct name *optargs)
    888  1.2  christos {
    889  1.2  christos 	struct attachment *attach;
    890  1.2  christos 	struct attachment *ap;
    891  1.2  christos 	struct name *np;
    892  1.2  christos 	char *expand_optargs;
    893  1.2  christos 	int attach_num;
    894  1.2  christos 
    895  1.2  christos 	expand_optargs = value(ENAME_MIME_ATTACH_LIST);
    896  1.2  christos 	attach_num = 1;
    897  1.2  christos 	ap = NULL;
    898  1.2  christos 	attach = NULL;
    899  1.2  christos 	for (np = optargs; np; np = np->n_flink) {
    900  1.2  christos 		char *argv[MAXARGC];
    901  1.2  christos 		int argc;
    902  1.2  christos 		int i;
    903  1.2  christos 
    904  1.2  christos 		if (expand_optargs != NULL)
    905  1.2  christos 			argc = getrawlist(np->n_name, argv, sizeofarray(argv));
    906  1.2  christos 		else {
    907  1.2  christos 			if (np->n_name == '\0')
    908  1.2  christos 				argc = 0;
    909  1.2  christos 			else {
    910  1.2  christos 				argc = 1;
    911  1.2  christos 				argv[0] = np->n_name;
    912  1.2  christos 			}
    913  1.2  christos 			argv[argc] = NULL;/* be consistent with getrawlist() */
    914  1.2  christos 		}
    915  1.2  christos 		for (i = 0; i < argc; i++) {
    916  1.2  christos 			struct attachment *ap2;
    917  1.2  christos 			char *filename;
    918  1.5  christos 
    919  1.2  christos 			if (argv[i][0] == '/')	/* an absolute path */
    920  1.2  christos 				(void)easprintf(&filename, "%s", argv[i]);
    921  1.2  christos 			else
    922  1.2  christos 				(void)easprintf(&filename, "%s/%s",
    923  1.2  christos 				    origdir, argv[i]);
    924  1.2  christos 
    925  1.2  christos 			ap2 = attach_one_file(ap, filename, attach_num);
    926  1.2  christos 			free(filename);
    927  1.2  christos 			if (ap2 != NULL) {
    928  1.2  christos 				ap = ap2;
    929  1.2  christos 				if (attach == NULL)
    930  1.2  christos 					attach = ap;
    931  1.2  christos 				attach_num++;
    932  1.2  christos 			}
    933  1.2  christos 		}
    934  1.2  christos 	}
    935  1.2  christos 	return attach;
    936  1.2  christos }
    937  1.2  christos 
    938  1.2  christos /*
    939  1.1  christos  * Output MIME header strings as specified in the header structure.
    940  1.1  christos  */
    941  1.1  christos PUBLIC void
    942  1.1  christos mime_putheader(FILE *fp, struct header *header)
    943  1.1  christos {
    944  1.1  christos 	(void)fprintf(fp, MIME_HDR_VERSION ": " MIME_VERSION "\n");
    945  1.1  christos 	if (header->h_attach) {
    946  1.1  christos 		(void)fprintf(fp, MIME_HDR_TYPE ": multipart/mixed;\n");
    947  1.1  christos 		(void)fprintf(fp, "\tboundary=\"%s\"\n", header->h_mime_boundary);
    948  1.1  christos 	}
    949  1.1  christos 	else {
    950  1.1  christos 		fput_mime_content(fp, &header->h_Content);
    951  1.1  christos 	}
    952  1.1  christos }
    953  1.1  christos 
    954  1.1  christos #endif /* MIME_SUPPORT */
    955