source: MondoRescue/branches/3.3/mindi-busybox/mailutils/reformime.c@ 3621

Last change on this file since 3621 was 3621, checked in by Bruno Cornec, 7 years ago

New 3?3 banch for incorporation of latest busybox 1.25. Changing minor version to handle potential incompatibilities.
  • Property svn:eol-style set to native
File size: 7.1 KB
Line 
1/* vi: set sw=4 ts=4: */
2/*
3 * reformime: parse MIME-encoded message
4 *
5 * Copyright (C) 2008 by Vladimir Dronnikov <dronnikov@gmail.com>
6 *
7 * Licensed under GPLv2, see file LICENSE in this source tree.
8 */
9
10//kbuild:lib-$(CONFIG_REFORMIME) += reformime.o mail.o
11
12#include "libbb.h"
13#include "mail.h"
14
15#if 0
16# define dbg_error_msg(...) bb_error_msg(__VA_ARGS__)
17#else
18# define dbg_error_msg(...) ((void)0)
19#endif
20
21static const char *find_token(const char *const string_array[], const char *key, const char *defvalue)
22{
23 const char *r = NULL;
24 int i;
25 for (i = 0; string_array[i] != NULL; i++) {
26 if (strcasecmp(string_array[i], key) == 0) {
27 r = (char *)string_array[i+1];
28 break;
29 }
30 }
31 return (r) ? r : defvalue;
32}
33
34static const char *xfind_token(const char *const string_array[], const char *key)
35{
36 const char *r = find_token(string_array, key, NULL);
37 if (r)
38 return r;
39 bb_error_msg_and_die("not found: '%s'", key);
40}
41
42enum {
43 OPT_x = 1 << 0,
44 OPT_X = 1 << 1,
45#if ENABLE_FEATURE_REFORMIME_COMPAT
46 OPT_d = 1 << 2,
47 OPT_e = 1 << 3,
48 OPT_i = 1 << 4,
49 OPT_s = 1 << 5,
50 OPT_r = 1 << 6,
51 OPT_c = 1 << 7,
52 OPT_m = 1 << 8,
53 OPT_h = 1 << 9,
54 OPT_o = 1 << 10,
55 OPT_O = 1 << 11,
56#endif
57};
58
59static int parse(const char *boundary, char **argv)
60{
61 int boundary_len = strlen(boundary);
62 char uniq[sizeof("%%llu.%u") + sizeof(int)*3];
63
64 dbg_error_msg("BOUNDARY[%s]", boundary);
65
66 // prepare unique string pattern
67 sprintf(uniq, "%%llu.%u", (unsigned)getpid());
68 dbg_error_msg("UNIQ[%s]", uniq);
69
70 while (1) {
71 char *header;
72 const char *tokens[32]; /* 32 is enough */
73 const char *type;
74
75 /* Read the header (everything up to two \n) */
76 {
77 unsigned header_idx = 0;
78 int last_ch = 0;
79 header = NULL;
80 while (1) {
81 int ch = fgetc(stdin);
82 if (ch == '\r') /* Support both line endings */
83 continue;
84 if (ch == EOF)
85 break;
86 if (ch == '\n' && last_ch == ch)
87 break;
88 if (!(header_idx & 0xff))
89 header = xrealloc(header, header_idx + 0x101);
90 header[header_idx++] = last_ch = ch;
91 }
92 if (!header) {
93 dbg_error_msg("EOF");
94 break;
95 }
96 header[header_idx] = '\0';
97 dbg_error_msg("H:'%s'", p);
98 }
99
100 /* Split to tokens */
101 {
102 char *s, *p;
103 unsigned ntokens;
104 const char *delims = ";=\" \t\n";
105
106 /* Skip to last Content-Type: */
107 s = p = header;
108 while ((p = strchr(p, '\n')) != NULL) {
109 p++;
110 if (strncasecmp(p, "Content-Type:", sizeof("Content-Type:")-1) == 0)
111 s = p;
112 }
113 dbg_error_msg("L:'%s'", p);
114 ntokens = 0;
115 s = strtok(s, delims);
116 while (s) {
117 tokens[ntokens] = s;
118 if (ntokens < ARRAY_SIZE(tokens) - 1)
119 ntokens++;
120 dbg_error_msg("L[%d]='%s'", ntokens, s);
121 s = strtok(NULL, delims);
122 }
123 tokens[ntokens] = NULL;
124 dbg_error_msg("EMPTYLINE, ntokens:%d", ntokens);
125 if (ntokens == 0)
126 break;
127 }
128
129 /* Is it multipart? */
130 type = find_token(tokens, "Content-Type:", "text/plain");
131 dbg_error_msg("TYPE:'%s'", type);
132 if (0 == strncasecmp(type, "multipart/", 10)) {
133 /* Yes, recurse */
134 if (strcasecmp(type + 10, "mixed") != 0)
135 bb_error_msg_and_die("no support of content type '%s'", type);
136 parse(xfind_token(tokens, "boundary"), argv);
137 } else {
138 /* No, process one non-multipart section */
139 char *end;
140 pid_t pid = pid;
141 FILE *fp;
142
143 const char *charset = find_token(tokens, "charset", CONFIG_FEATURE_MIME_CHARSET);
144 const char *encoding = find_token(tokens, "Content-Transfer-Encoding:", "7bit");
145
146 /* Compose target filename */
147 char *filename = (char *)find_token(tokens, "filename", NULL);
148 if (!filename)
149 filename = xasprintf(uniq, monotonic_us());
150 else
151 filename = bb_get_last_path_component_strip(xstrdup(filename));
152
153 if (opts & OPT_X) {
154 int fd[2];
155
156 /* start external helper */
157 xpipe(fd);
158 pid = vfork();
159 if (0 == pid) {
160 /* child reads from fd[0] */
161 close(fd[1]);
162 xmove_fd(fd[0], STDIN_FILENO);
163 xsetenv("CONTENT_TYPE", type);
164 xsetenv("CHARSET", charset);
165 xsetenv("ENCODING", encoding);
166 xsetenv("FILENAME", filename);
167 BB_EXECVP_or_die(argv);
168 }
169 /* parent will write to fd[1] */
170 close(fd[0]);
171 fp = xfdopen_for_write(fd[1]);
172 signal(SIGPIPE, SIG_IGN);
173 } else {
174 /* write to file */
175 char *fname = xasprintf("%s%s", *argv, filename);
176 fp = xfopen_for_write(fname);
177 free(fname);
178 }
179 free(filename);
180
181 /* write to fp */
182 end = NULL;
183 if (0 == strcasecmp(encoding, "base64")) {
184 read_base64(stdin, fp, '-');
185 } else
186 if (0 != strcasecmp(encoding, "7bit")
187 && 0 != strcasecmp(encoding, "8bit")
188 ) {
189 /* quoted-printable, binary, user-defined are unsupported so far */
190 bb_error_msg_and_die("encoding '%s' not supported", encoding);
191 } else {
192 /* plain 7bit or 8bit */
193 while ((end = xmalloc_fgets(stdin)) != NULL) {
194 if ('-' == end[0]
195 && '-' == end[1]
196 && strncmp(end + 2, boundary, boundary_len) == 0
197 ) {
198 break;
199 }
200 fputs(end, fp);
201 }
202 }
203 fclose(fp);
204
205 /* Wait for child */
206 if (opts & OPT_X) {
207 int rc;
208 signal(SIGPIPE, SIG_DFL);
209 rc = (wait4pid(pid) & 0xff);
210 if (rc != 0)
211 return rc + 20;
212 }
213
214 /* Multipart ended? */
215 if (end && '-' == end[2 + boundary_len] && '-' == end[2 + boundary_len + 1]) {
216 dbg_error_msg("FINISHED MPART:'%s'", end);
217 break;
218 }
219 dbg_error_msg("FINISHED:'%s'", end);
220 free(end);
221 } /* end of "handle one non-multipart block" */
222
223 free(header);
224 } /* while (1) */
225
226 dbg_error_msg("ENDPARSE[%s]", boundary);
227
228 return EXIT_SUCCESS;
229}
230
231//usage:#define reformime_trivial_usage
232//usage: "[OPTIONS]"
233//usage:#define reformime_full_usage "\n\n"
234//usage: "Parse MIME-encoded message on stdin\n"
235//usage: "\n -x PREFIX Extract content of MIME sections to files"
236//usage: "\n -X PROG ARGS Filter content of MIME sections through PROG"
237//usage: "\n Must be the last option"
238//usage: "\n"
239//usage: "\nOther options are silently ignored"
240
241/*
242Usage: reformime [options]
243 -d - parse a delivery status notification.
244 -e - extract contents of MIME section.
245 -x - extract MIME section to a file.
246 -X - pipe MIME section to a program.
247 -i - show MIME info.
248 -s n.n.n.n - specify MIME section.
249 -r - rewrite message, filling in missing MIME headers.
250 -r7 - also convert 8bit/raw encoding to quoted-printable, if possible.
251 -r8 - also convert quoted-printable encoding to 8bit, if possible.
252 -c charset - default charset for rewriting, -o, and -O.
253 -m [file] [file]... - create a MIME message digest.
254 -h "header" - decode RFC 2047-encoded header.
255 -o "header" - encode unstructured header using RFC 2047.
256 -O "header" - encode address list header using RFC 2047.
257*/
258
259int reformime_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
260int reformime_main(int argc UNUSED_PARAM, char **argv)
261{
262 const char *opt_prefix = "";
263
264 INIT_G();
265
266 // parse options
267 // N.B. only -x and -X are supported so far
268 opt_complementary = "x--X:X--x" IF_FEATURE_REFORMIME_COMPAT(":m::");
269 opts = getopt32(argv,
270 "x:X" IF_FEATURE_REFORMIME_COMPAT("deis:r:c:m:h:o:O:"),
271 &opt_prefix
272 IF_FEATURE_REFORMIME_COMPAT(, NULL, NULL, &G.opt_charset, NULL, NULL, NULL, NULL)
273 );
274 argv += optind;
275
276 return parse("", (opts & OPT_X) ? argv : (char **)&opt_prefix);
277}
Note: See TracBrowser for help on using the repository browser.