mailutils/mime.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * makemime: create MIME-encoded message
   4  * reformime: parse MIME-encoded message
   5  *
   6  * Copyright (C) 2008 by Vladimir Dronnikov <dronnikov@gmail.com>
   7  *
   8  * Licensed under GPLv2, see file LICENSE in this tarball for details.
   9  */
  10 #include "libbb.h"
  11 #include "mail.h"
  12
  13 /*
  14   makemime -c type [-o file] [-e encoding] [-C charset] [-N name] \
  15                    [-a "Header: Contents"] file
  16            -m [ type ] [-o file] [-e encoding] [-a "Header: Contents"] file
  17            -j [-o file] file1 file2
  18            @file
  19
  20    file:  filename    - read or write from filename
  21           -           - read or write from stdin or stdout
  22           &n          - read or write from file descriptor n
  23           \( opts \)  - read from child process, that generates [ opts ]
  24
  25 Options:
  26
  27   -c type         - create a new MIME section from "file" with this
  28                     Content-Type: (default is application/octet-stream).
  29   -C charset      - MIME charset of a new text/plain section.
  30   -N name         - MIME content name of the new mime section.
  31   -m [ type ]     - create a multipart mime section from "file" of this
  32                     Content-Type: (default is multipart/mixed).
  33   -e encoding     - use the given encoding (7bit, 8bit, quoted-printable,
  34                     or base64), instead of guessing.  Omit "-e" and use
  35                     -c auto to set Content-Type: to text/plain or
  36                     application/octet-stream based on picked encoding.
  37   -j file1 file2  - join mime section file2 to multipart section file1.
  38   -o file         - write ther result to file, instead of stdout (not
  39                     allowed in child processes).
  40   -a header       - prepend an additional header to the output.
  41
  42   @file - read all of the above options from file, one option or
  43           value on each line.
  44 */
  45
  46 int makemime_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  47 int makemime_main(int argc UNUSED_PARAM, char **argv)
  48 {
  49         llist_t *opt_headers = NULL, *l;
  50         const char *opt_output;
  51 #define boundary opt_output
  52
  53         enum {
  54                 OPT_c = 1 << 0,         // Content-Type:
  55                 OPT_e = 1 << 1,         // Content-Transfer-Encoding. Ignored. Assumed base64
  56                 OPT_o = 1 << 2,         // output to
  57                 OPT_C = 1 << 3,         // charset
  58                 OPT_N = 1 << 4,         // COMPAT
  59                 OPT_a = 1 << 5,         // additional headers
  60                 OPT_m = 1 << 6,         // COMPAT
  61                 OPT_j = 1 << 7,         // COMPAT
  62         };
  63
  64         INIT_G();
  65
  66         // parse options
  67         opt_complementary = "a::";
  68         opts = getopt32(argv,
  69                 "c:e:o:C:N:a:m:j:",
  70                 &G.content_type, NULL, &opt_output, &G.opt_charset, NULL, &opt_headers, NULL, NULL
  71         );
  72         //argc -= optind;
  73         argv += optind;
  74
  75         // respect -o output
  76         if (opts & OPT_o)
  77                 freopen(opt_output, "w", stdout);
  78
  79         // no files given on command line? -> use stdin
  80         if (!*argv)
  81                 *--argv = (char *)"-";
  82
  83         // put additional headers
  84         for (l = opt_headers; l; l = l->link)
  85                 puts(l->data);
  86
  87         // make a random string -- it will delimit message parts
  88         srand(monotonic_us());
  89         boundary = xasprintf("%d-%d-%d", rand(), rand(), rand());
  90
  91         // put multipart header
  92         printf(
  93                 "Mime-Version: 1.0\n"
  94                 "Content-Type: multipart/mixed; boundary=\"%s\"\n"
  95                 , boundary
  96         );
  97
  98         // put attachments
  99         while (*argv) {
 100                 printf(
 101                         "\n--%s\n"
 102                         "Content-Type: %s; charset=%s\n"
 103                         "Content-Disposition: inline; filename=\"%s\"\n"
 104                         "Content-Transfer-Encoding: base64\n"
 105                         , boundary
 106                         , G.content_type
 107                         , G.opt_charset
 108                         , bb_get_last_path_component_strip(*argv)
 109                 );
 110                 encode_base64(*argv++, (const char *)stdin, "");
 111         }
 112
 113         // put multipart footer
 114         printf("\n--%s--\n" "\n", boundary);
 115
 116         return EXIT_SUCCESS;
 117 #undef boundary
 118 }
 119
 120 static const char *find_token(const char *const string_array[], const char *key, const char *defvalue)
 121 {
 122         const char *r = NULL;
 123         for (int i = 0; string_array[i] != 0; i++) {
 124                 if (strcasecmp(string_array[i], key) == 0) {
 125                         r = (char *)string_array[i+1];
 126                         break;
 127                 }
 128         }
 129         return (r) ? r : defvalue;
 130 }
 131
 132 static const char *xfind_token(const char *const string_array[], const char *key)
 133 {
 134         const char *r = find_token(string_array, key, NULL);
 135         if (r)
 136                 return r;
 137         bb_error_msg_and_die("header: %s", key);
 138 }
 139
 140 enum {
 141         OPT_x = 1 << 0,
 142         OPT_X = 1 << 1,
 143 #if ENABLE_FEATURE_REFORMIME_COMPAT
 144         OPT_d = 1 << 2,
 145         OPT_e = 1 << 3,
 146         OPT_i = 1 << 4,
 147         OPT_s = 1 << 5,
 148         OPT_r = 1 << 6,
 149         OPT_c = 1 << 7,
 150         OPT_m = 1 << 8,
 151         OPT_h = 1 << 9,
 152         OPT_o = 1 << 10,
 153         OPT_O = 1 << 11,
 154 #endif
 155 };
 156
 157 static int parse(const char *boundary, char **argv)
 158 {
 159         char *line, *s, *p;
 160         const char *type;
 161         int boundary_len = strlen(boundary);
 162         const char *delims = " ;\"\t\r\n";
 163         const char *uniq;
 164         int ntokens;
 165         const char *tokens[32]; // 32 is enough
 166
 167         // prepare unique string pattern
 168         uniq = xasprintf("%%llu.%u.%s", (unsigned)getpid(), safe_gethostname());
 169
 170 //bb_info_msg("PARSE[%s]", terminator);
 171
 172         while ((line = xmalloc_fgets_str(stdin, "\r\n\r\n")) != NULL) {
 173
 174                 // seek to start of MIME section
 175                 // N.B. to avoid false positives let us seek to the _last_ occurance
 176                 p = NULL;
 177                 s = line;
 178                 while ((s=strcasestr(s, "Content-Type:")) != NULL)
 179                         p = s++;
 180                 if (!p)
 181                         goto next;
 182 //bb_info_msg("L[%s]", p);
 183
 184                 // split to tokens
 185                 // TODO: strip of comments which are of form: (comment-text)
 186                 ntokens = 0;
 187                 tokens[ntokens] = NULL;
 188                 for (s = strtok(p, delims); s; s = strtok(NULL, delims)) {
 189                         tokens[ntokens] = s;
 190                         if (ntokens < ARRAY_SIZE(tokens) - 1)
 191                                 ntokens++;
 192 //bb_info_msg("L[%d][%s]", ntokens, s);
 193                 }
 194                 tokens[ntokens] = NULL;
 195 //bb_info_msg("N[%d]", ntokens);
 196
 197                 // analyse tokens
 198                 type = find_token(tokens, "Content-Type:", "text/plain");
 199 //bb_info_msg("T[%s]", type);
 200                 if (0 == strncasecmp(type, "multipart/", 10)) {
 201                         if (0 == strcasecmp(type+10, "mixed")) {
 202                                 parse(xfind_token(tokens, "boundary="), argv);
 203                         } else
 204                                 bb_error_msg_and_die("no support of content type '%s'", type);
 205                 } else {
 206                         pid_t pid = pid;
 207                         int rc;
 208                         FILE *fp;
 209                         // fetch charset
 210                         const char *charset = find_token(tokens, "charset=", CONFIG_FEATURE_MIME_CHARSET);
 211                         // fetch encoding
 212                         const char *encoding = find_token(tokens, "Content-Transfer-Encoding:", "7bit");
 213                         // compose target filename
 214                         char *filename = (char *)find_token(tokens, "filename=", NULL);
 215                         if (!filename)
 216                                 filename = xasprintf(uniq, monotonic_us());
 217                         else
 218                                 filename = bb_get_last_path_component_strip(xstrdup(filename));
 219
 220                         // start external helper, if any
 221                         if (opts & OPT_X) {
 222                                 int fd[2];
 223                                 xpipe(fd);
 224                                 pid = vfork();
 225                                 if (0 == pid) {
 226                                         // child reads from fd[0]
 227                                         xdup2(fd[0], STDIN_FILENO);
 228                                         close(fd[0]); close(fd[1]);
 229                                         xsetenv("CONTENT_TYPE", type);
 230                                         xsetenv("CHARSET", charset);
 231                                         xsetenv("ENCODING", encoding);
 232                                         xsetenv("FILENAME", filename);
 233                                         BB_EXECVP(*argv, argv);
 234                                         _exit(EXIT_FAILURE);
 235                                 }
 236                                 // parent dumps to fd[1]
 237                                 close(fd[0]);
 238                                 fp = fdopen(fd[1], "w");
 239                                 signal(SIGPIPE, SIG_IGN); // ignore EPIPE
 240                         // or create a file for dump
 241                         } else {
 242                                 char *fname = xasprintf("%s%s", *argv, filename);
 243                                 fp = xfopen_for_write(fname);
 244                                 free(fname);
 245                         }
 246
 247                         // housekeeping
 248                         free(filename);
 249
 250                         // dump to fp
 251                         if (0 == strcasecmp(encoding, "base64")) {
 252                                 decode_base64(stdin, fp);
 253                         } else if (0 != strcasecmp(encoding, "7bit")
 254                                 && 0 != strcasecmp(encoding, "8bit")) {
 255                                 // quoted-printable, binary, user-defined are unsupported so far
 256                                 bb_error_msg_and_die("no support of encoding '%s'", encoding);
 257                         } else {
 258                                 // N.B. we have written redundant \n. so truncate the file
 259                                 // The following weird 2-tacts reading technique is due to
 260                                 // we have to not write extra \n at the end of the file
 261                                 // In case of -x option we could truncate the resulting file as
 262                                 // fseek(fp, -1, SEEK_END);
 263                                 // if (ftruncate(fileno(fp), ftell(fp)))
 264                                 //      bb_perror_msg("ftruncate");
 265                                 // But in case of -X we have to be much more careful. There is
 266                                 // no means to truncate what we already have sent to the helper.
 267                                 p = xmalloc_fgets_str(stdin, "\r\n");
 268                                 while (p) {
 269                                         if ((s = xmalloc_fgets_str(stdin, "\r\n")) == NULL)
 270                                                 break;
 271                                         if ('-' == s[0] && '-' == s[1]
 272                                                 && 0 == strncmp(s+2, boundary, boundary_len))
 273                                                 break;
 274                                         fputs(p, fp);
 275                                         p = s;
 276                                 }
 277
 278 /*
 279                                 while ((s = xmalloc_fgetline_str(stdin, "\r\n")) != NULL) {
 280                                         if ('-' == s[0] && '-' == s[1]
 281                                                 && 0 == strncmp(s+2, boundary, boundary_len))
 282                                                 break;
 283                                         fprintf(fp, "%s\n", s);
 284                                 }
 285                                 // N.B. we have written redundant \n. so truncate the file
 286                                 fseek(fp, -1, SEEK_END);
 287                                 if (ftruncate(fileno(fp), ftell(fp)))
 288                                         bb_perror_msg("ftruncate");
 289 */
 290                         }
 291                         fclose(fp);
 292
 293                         // finalize helper
 294                         if (opts & OPT_X) {
 295                                 signal(SIGPIPE, SIG_DFL);
 296                                 // exit if helper exited >0
 297                                 rc = wait4pid(pid);
 298                                 if (rc)
 299                                         return rc+20;
 300                         }
 301
 302                         // check multipart finalized
 303                         if (s && '-' == s[2+boundary_len] && '-' == s[2+boundary_len+1]) {
 304                                 free(line);
 305                                 break;
 306                         }
 307                 }
 308  next:
 309                 free(line);
 310         }
 311
 312 //bb_info_msg("ENDPARSE[%s]", boundary);
 313
 314         return EXIT_SUCCESS;
 315 }
 316
 317 /*
 318 Usage: reformime [options]
 319     -d - parse a delivery status notification.
 320     -e - extract contents of MIME section.
 321     -x - extract MIME section to a file.
 322     -X - pipe MIME section to a program.
 323     -i - show MIME info.
 324     -s n.n.n.n - specify MIME section.
 325     -r - rewrite message, filling in missing MIME headers.
 326     -r7 - also convert 8bit/raw encoding to quoted-printable, if possible.
 327     -r8 - also convert quoted-printable encoding to 8bit, if possible.
 328     -c charset - default charset for rewriting, -o, and -O.
 329     -m [file] [file]... - create a MIME message digest.
 330     -h "header" - decode RFC 2047-encoded header.
 331     -o "header" - encode unstructured header using RFC 2047.
 332     -O "header" - encode address list header using RFC 2047.
 333 */
 334
 335 int reformime_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 336 int reformime_main(int argc UNUSED_PARAM, char **argv)
 337 {
 338         const char *opt_prefix = "";
 339
 340         INIT_G();
 341
 342         // parse options
 343         // N.B. only -x and -X are supported so far
 344         opt_complementary = "x--X:X--x" USE_FEATURE_REFORMIME_COMPAT(":m::");
 345         opts = getopt32(argv,
 346                 "x:X" USE_FEATURE_REFORMIME_COMPAT("deis:r:c:m:h:o:O:"),
 347                 &opt_prefix
 348                 USE_FEATURE_REFORMIME_COMPAT(, NULL, NULL, &G.opt_charset, NULL, NULL, NULL, NULL)
 349         );
 350         //argc -= optind;
 351         argv += optind;
 352
 353         return parse("", (opts & OPT_X) ? argv : (char **)&opt_prefix);
 354 }