blob: 4c0d2eebabdb8b7401a4414bfea608e465f061de [file] [log] [blame]
Alex Deymo486467e2017-12-19 19:04:07 +01001/***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
Elliott Hughes0128fe42018-02-27 14:57:55 -08008 * Copyright (C) 1998 - 2018, Daniel Stenberg, <daniel@haxx.se>, et al.
Alex Deymo486467e2017-12-19 19:04:07 +01009 *
10 * This software is licensed as described in the file COPYING, which
11 * you should have received as part of this distribution. The terms
12 * are also available at https://curl.haxx.se/docs/copyright.html.
13 *
14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15 * copies of the Software, and permit persons to whom the Software is
16 * furnished to do so, under the terms of the COPYING file.
17 *
18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19 * KIND, either express or implied.
20 *
21 ***************************************************************************/
22
23#include "curl_setup.h"
24
25#include <curl/curl.h>
26
27#include "mime.h"
28#include "non-ascii.h"
29#include "urldata.h"
30#include "sendf.h"
31
32#if !defined(CURL_DISABLE_HTTP) || !defined(CURL_DISABLE_SMTP) || \
33 !defined(CURL_DISABLE_IMAP)
34
35#if defined(HAVE_LIBGEN_H) && defined(HAVE_BASENAME)
36#include <libgen.h>
37#endif
38
39#include "rand.h"
40#include "slist.h"
41#include "strcase.h"
42/* The last 3 #include files should be in this order */
43#include "curl_printf.h"
44#include "curl_memory.h"
45#include "memdebug.h"
46
47#ifdef WIN32
48# ifndef R_OK
49# define R_OK 4
50# endif
51#endif
52
53
Alex Deymo486467e2017-12-19 19:04:07 +010054#define READ_ERROR ((size_t) -1)
55
56/* Encoders. */
57static size_t encoder_nop_read(char *buffer, size_t size, bool ateof,
58 curl_mimepart *part);
59static curl_off_t encoder_nop_size(curl_mimepart *part);
60static size_t encoder_7bit_read(char *buffer, size_t size, bool ateof,
61 curl_mimepart *part);
62static size_t encoder_base64_read(char *buffer, size_t size, bool ateof,
63 curl_mimepart *part);
64static curl_off_t encoder_base64_size(curl_mimepart *part);
65static size_t encoder_qp_read(char *buffer, size_t size, bool ateof,
66 curl_mimepart *part);
67static curl_off_t encoder_qp_size(curl_mimepart *part);
68
69static const mime_encoder encoders[] = {
70 {"binary", encoder_nop_read, encoder_nop_size},
71 {"8bit", encoder_nop_read, encoder_nop_size},
72 {"7bit", encoder_7bit_read, encoder_nop_size},
73 {"base64", encoder_base64_read, encoder_base64_size},
74 {"quoted-printable", encoder_qp_read, encoder_qp_size},
75 {ZERO_NULL, ZERO_NULL, ZERO_NULL}
76};
77
78/* Base64 encoding table */
79static const char base64[] =
80 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
81
82/* Quoted-printable character class table.
83 *
84 * We cannot rely on ctype functions since quoted-printable input data
85 * is assumed to be ascii-compatible, even on non-ascii platforms. */
86#define QP_OK 1 /* Can be represented by itself. */
87#define QP_SP 2 /* Space or tab. */
88#define QP_CR 3 /* Carriage return. */
89#define QP_LF 4 /* Line-feed. */
90static const unsigned char qp_class[] = {
91 0, 0, 0, 0, 0, 0, 0, 0, /* 00 - 07 */
92 0, QP_SP, QP_LF, 0, 0, QP_CR, 0, 0, /* 08 - 0F */
93 0, 0, 0, 0, 0, 0, 0, 0, /* 10 - 17 */
94 0, 0, 0, 0, 0, 0, 0, 0, /* 18 - 1F */
95 QP_SP, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 20 - 27 */
96 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 28 - 2F */
97 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 30 - 37 */
98 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, 0 , QP_OK, QP_OK, /* 38 - 3F */
99 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 40 - 47 */
100 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 48 - 4F */
101 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 50 - 57 */
102 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 58 - 5F */
103 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 60 - 67 */
104 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 68 - 6F */
105 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, /* 70 - 77 */
106 QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, QP_OK, 0, /* 78 - 7F */
107 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 80 - 8F */
108 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 90 - 9F */
109 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* A0 - AF */
110 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* B0 - BF */
111 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* C0 - CF */
112 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* D0 - DF */
113 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* E0 - EF */
114 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 /* F0 - FF */
115};
116
117
118/* Binary --> hexadecimal ASCII table. */
119static const char aschex[] =
120 "\x30\x31\x32\x33\x34\x35\x36\x37\x38\x39\x41\x42\x43\x44\x45\x46";
121
122
123
124#ifndef __VMS
125#define filesize(name, stat_data) (stat_data.st_size)
126#define fopen_read fopen
127
128#else
129
130#include <fabdef.h>
131/*
132 * get_vms_file_size does what it takes to get the real size of the file
133 *
134 * For fixed files, find out the size of the EOF block and adjust.
135 *
136 * For all others, have to read the entire file in, discarding the contents.
137 * Most posted text files will be small, and binary files like zlib archives
138 * and CD/DVD images should be either a STREAM_LF format or a fixed format.
139 *
140 */
141curl_off_t VmsRealFileSize(const char *name,
142 const struct_stat *stat_buf)
143{
144 char buffer[8192];
145 curl_off_t count;
146 int ret_stat;
147 FILE * file;
148
149 file = fopen(name, FOPEN_READTEXT); /* VMS */
150 if(file == NULL)
151 return 0;
152
153 count = 0;
154 ret_stat = 1;
155 while(ret_stat > 0) {
156 ret_stat = fread(buffer, 1, sizeof(buffer), file);
157 if(ret_stat != 0)
158 count += ret_stat;
159 }
160 fclose(file);
161
162 return count;
163}
164
165/*
166 *
167 * VmsSpecialSize checks to see if the stat st_size can be trusted and
168 * if not to call a routine to get the correct size.
169 *
170 */
171static curl_off_t VmsSpecialSize(const char *name,
172 const struct_stat *stat_buf)
173{
174 switch(stat_buf->st_fab_rfm) {
175 case FAB$C_VAR:
176 case FAB$C_VFC:
177 return VmsRealFileSize(name, stat_buf);
178 break;
179 default:
180 return stat_buf->st_size;
181 }
182}
183
184#define filesize(name, stat_data) VmsSpecialSize(name, &stat_data)
185
186/*
187 * vmsfopenread
188 *
189 * For upload to work as expected on VMS, different optional
190 * parameters must be added to the fopen command based on
191 * record format of the file.
192 *
193 */
194static FILE * vmsfopenread(const char *file, const char *mode)
195{
196 struct_stat statbuf;
197 int result;
198
199 result = stat(file, &statbuf);
200
201 switch(statbuf.st_fab_rfm) {
202 case FAB$C_VAR:
203 case FAB$C_VFC:
204 case FAB$C_STMCR:
205 return fopen(file, FOPEN_READTEXT); /* VMS */
206 break;
207 default:
208 return fopen(file, FOPEN_READTEXT, "rfm=stmlf", "ctx=stm");
209 }
210}
211
212#define fopen_read vmsfopenread
213#endif
214
215
216#ifndef HAVE_BASENAME
217/*
218 (Quote from The Open Group Base Specifications Issue 6 IEEE Std 1003.1, 2004
219 Edition)
220
221 The basename() function shall take the pathname pointed to by path and
222 return a pointer to the final component of the pathname, deleting any
223 trailing '/' characters.
224
225 If the string pointed to by path consists entirely of the '/' character,
226 basename() shall return a pointer to the string "/". If the string pointed
227 to by path is exactly "//", it is implementation-defined whether '/' or "//"
228 is returned.
229
230 If path is a null pointer or points to an empty string, basename() shall
231 return a pointer to the string ".".
232
233 The basename() function may modify the string pointed to by path, and may
234 return a pointer to static storage that may then be overwritten by a
235 subsequent call to basename().
236
237 The basename() function need not be reentrant. A function that is not
238 required to be reentrant is not required to be thread-safe.
239
240*/
241static char *Curl_basename(char *path)
242{
243 /* Ignore all the details above for now and make a quick and simple
Elliott Hughes1ef06ba2018-05-30 15:43:58 -0700244 implementation here */
Alex Deymo486467e2017-12-19 19:04:07 +0100245 char *s1;
246 char *s2;
247
248 s1 = strrchr(path, '/');
249 s2 = strrchr(path, '\\');
250
251 if(s1 && s2) {
252 path = (s1 > s2? s1 : s2) + 1;
253 }
254 else if(s1)
255 path = s1 + 1;
256 else if(s2)
257 path = s2 + 1;
258
259 return path;
260}
261
262#define basename(x) Curl_basename((x))
263#endif
264
265
266/* Set readback state. */
267static void mimesetstate(mime_state *state, enum mimestate tok, void *ptr)
268{
269 state->state = tok;
270 state->ptr = ptr;
271 state->offset = 0;
272}
273
274
275/* Escape header string into allocated memory. */
276static char *escape_string(const char *src)
277{
278 size_t bytecount = 0;
279 size_t i;
280 char *dst;
281
282 for(i = 0; src[i]; i++)
283 if(src[i] == '"' || src[i] == '\\')
284 bytecount++;
285
286 bytecount += i;
287 dst = malloc(bytecount + 1);
288 if(!dst)
289 return NULL;
290
291 for(i = 0; *src; src++) {
292 if(*src == '"' || *src == '\\')
293 dst[i++] = '\\';
294 dst[i++] = *src;
295 }
296
297 dst[i] = '\0';
298 return dst;
299}
300
301/* Check if header matches. */
302static char *match_header(struct curl_slist *hdr, const char *lbl, size_t len)
303{
304 char *value = NULL;
305
306 if(strncasecompare(hdr->data, lbl, len) && hdr->data[len] == ':')
307 for(value = hdr->data + len + 1; *value == ' '; value++)
308 ;
309 return value;
310}
311
312/* Get a header from an slist. */
313static char *search_header(struct curl_slist *hdrlist, const char *hdr)
314{
315 size_t len = strlen(hdr);
316 char *value = NULL;
317
318 for(; !value && hdrlist; hdrlist = hdrlist->next)
319 value = match_header(hdrlist, hdr, len);
320
321 return value;
322}
323
324static char *strippath(const char *fullfile)
325{
326 char *filename;
327 char *base;
328 filename = strdup(fullfile); /* duplicate since basename() may ruin the
329 buffer it works on */
330 if(!filename)
331 return NULL;
332 base = strdup(basename(filename));
333
334 free(filename); /* free temporary buffer */
335
336 return base; /* returns an allocated string or NULL ! */
337}
338
339/* Initialize data encoder state. */
340static void cleanup_encoder_state(mime_encoder_state *p)
341{
342 p->pos = 0;
343 p->bufbeg = 0;
344 p->bufend = 0;
345}
346
347
348/* Dummy encoder. This is used for 8bit and binary content encodings. */
349static size_t encoder_nop_read(char *buffer, size_t size, bool ateof,
350 curl_mimepart *part)
351{
352 mime_encoder_state *st = &part->encstate;
353 size_t insize = st->bufend - st->bufbeg;
354
355 (void) ateof;
356
357 if(size > insize)
358 size = insize;
359 if(size)
360 memcpy(buffer, st->buf, size);
361 st->bufbeg += size;
362 return size;
363}
364
365static curl_off_t encoder_nop_size(curl_mimepart *part)
366{
367 return part->datasize;
368}
369
370
371/* 7bit encoder: the encoder is just a data validity check. */
372static size_t encoder_7bit_read(char *buffer, size_t size, bool ateof,
373 curl_mimepart *part)
374{
375 mime_encoder_state *st = &part->encstate;
376 size_t cursize = st->bufend - st->bufbeg;
377
378 (void) ateof;
379
380 if(size > cursize)
381 size = cursize;
382
383 for(cursize = 0; cursize < size; cursize++) {
384 *buffer = st->buf[st->bufbeg];
385 if(*buffer++ & 0x80)
386 return cursize? cursize: READ_ERROR;
387 st->bufbeg++;
388 }
389
390 return cursize;
391}
392
393
394/* Base64 content encoder. */
395static size_t encoder_base64_read(char *buffer, size_t size, bool ateof,
396 curl_mimepart *part)
397{
398 mime_encoder_state *st = &part->encstate;
399 size_t cursize = 0;
400 int i;
401 char *ptr = buffer;
402
403 while(st->bufbeg < st->bufend) {
404 /* Line full ? */
405 if(st->pos > MAX_ENCODED_LINE_LENGTH - 4) {
406 /* Yes, we need 2 characters for CRLF. */
407 if(size < 2)
408 break;
409 *ptr++ = '\r';
410 *ptr++ = '\n';
411 st->pos = 0;
412 cursize += 2;
413 size -= 2;
414 }
415
416 /* Be sure there is enough space and input data for a base64 group. */
417 if(size < 4 || st->bufend - st->bufbeg < 3)
418 break;
419
420 /* Encode three bytes as four characters. */
421 i = st->buf[st->bufbeg++] & 0xFF;
422 i = (i << 8) | (st->buf[st->bufbeg++] & 0xFF);
423 i = (i << 8) | (st->buf[st->bufbeg++] & 0xFF);
424 *ptr++ = base64[(i >> 18) & 0x3F];
425 *ptr++ = base64[(i >> 12) & 0x3F];
426 *ptr++ = base64[(i >> 6) & 0x3F];
427 *ptr++ = base64[i & 0x3F];
428 cursize += 4;
429 st->pos += 4;
430 size -= 4;
431 }
432
433 /* If at eof, we have to flush the buffered data. */
434 if(ateof && size >= 4) {
435 /* Buffered data size can only be 0, 1 or 2. */
436 ptr[2] = ptr[3] = '=';
437 i = 0;
438 switch(st->bufend - st->bufbeg) {
439 case 2:
440 i = (st->buf[st->bufbeg + 1] & 0xFF) << 8;
441 /* FALLTHROUGH */
442 case 1:
443 i |= (st->buf[st->bufbeg] & 0xFF) << 16;
444 ptr[0] = base64[(i >> 18) & 0x3F];
445 ptr[1] = base64[(i >> 12) & 0x3F];
446 if(++st->bufbeg != st->bufend) {
447 ptr[2] = base64[(i >> 6) & 0x3F];
448 st->bufbeg++;
449 }
450 cursize += 4;
451 st->pos += 4;
452 break;
453 }
454 }
455
456#ifdef CURL_DOES_CONVERSIONS
457 /* This is now textual data, Convert character codes. */
458 if(part->easy && cursize) {
459 CURLcode result = Curl_convert_to_network(part->easy, buffer, cursize);
460 if(result)
461 return READ_ERROR;
462 }
463#endif
464
465 return cursize;
466}
467
468static curl_off_t encoder_base64_size(curl_mimepart *part)
469{
470 curl_off_t size = part->datasize;
471
472 if(size <= 0)
473 return size; /* Unknown size or no data. */
474
475 /* Compute base64 character count. */
476 size = 4 * (1 + (size - 1) / 3);
477
478 /* Effective character count must include CRLFs. */
479 return size + 2 * ((size - 1) / MAX_ENCODED_LINE_LENGTH);
480}
481
482
483/* Quoted-printable lookahead.
484 *
485 * Check if a CRLF or end of data is in input buffer at current position + n.
486 * Return -1 if more data needed, 1 if CRLF or end of data, else 0.
487 */
488static int qp_lookahead_eol(mime_encoder_state *st, int ateof, size_t n)
489{
490 n += st->bufbeg;
491 if(n >= st->bufend && ateof)
492 return 1;
493 if(n + 2 > st->bufend)
494 return ateof? 0: -1;
495 if(qp_class[st->buf[n] & 0xFF] == QP_CR &&
496 qp_class[st->buf[n + 1] & 0xFF] == QP_LF)
497 return 1;
498 return 0;
499}
500
501/* Quoted-printable encoder. */
502static size_t encoder_qp_read(char *buffer, size_t size, bool ateof,
503 curl_mimepart *part)
504{
505 mime_encoder_state *st = &part->encstate;
506 char *ptr = buffer;
507 size_t cursize = 0;
508 int i;
509 size_t len;
510 size_t consumed;
511 int softlinebreak;
512 char buf[4];
513
514 /* On all platforms, input is supposed to be ASCII compatible: for this
515 reason, we use hexadecimal ASCII codes in this function rather than
516 character constants that can be interpreted as non-ascii on some
517 platforms. Preserve ASCII encoding on output too. */
518 while(st->bufbeg < st->bufend) {
519 len = 1;
520 consumed = 1;
521 i = st->buf[st->bufbeg];
522 buf[0] = (char) i;
523 buf[1] = aschex[(i >> 4) & 0xF];
524 buf[2] = aschex[i & 0xF];
525
526 switch(qp_class[st->buf[st->bufbeg] & 0xFF]) {
527 case QP_OK: /* Not a special character. */
528 break;
529 case QP_SP: /* Space or tab. */
530 /* Spacing must be escaped if followed by CRLF. */
531 switch(qp_lookahead_eol(st, ateof, 1)) {
532 case -1: /* More input data needed. */
533 return cursize;
534 case 0: /* No encoding needed. */
535 break;
536 default: /* CRLF after space or tab. */
537 buf[0] = '\x3D'; /* '=' */
538 len = 3;
539 break;
540 }
541 break;
542 case QP_CR: /* Carriage return. */
543 /* If followed by a line-feed, output the CRLF pair.
544 Else escape it. */
545 switch(qp_lookahead_eol(st, ateof, 0)) {
546 case -1: /* Need more data. */
547 return cursize;
548 case 1: /* CRLF found. */
549 buf[len++] = '\x0A'; /* Append '\n'. */
550 consumed = 2;
551 break;
552 default: /* Not followed by LF: escape. */
553 buf[0] = '\x3D'; /* '=' */
554 len = 3;
555 break;
556 }
557 break;
558 default: /* Character must be escaped. */
559 buf[0] = '\x3D'; /* '=' */
560 len = 3;
561 break;
562 }
563
564 /* Be sure the encoded character fits within maximum line length. */
565 if(buf[len - 1] != '\x0A') { /* '\n' */
566 softlinebreak = st->pos + len > MAX_ENCODED_LINE_LENGTH;
567 if(!softlinebreak && st->pos + len == MAX_ENCODED_LINE_LENGTH) {
568 /* We may use the current line only if end of data or followed by
569 a CRLF. */
570 switch(qp_lookahead_eol(st, ateof, consumed)) {
571 case -1: /* Need more data. */
572 return cursize;
573 break;
574 case 0: /* Not followed by a CRLF. */
575 softlinebreak = 1;
576 break;
577 }
578 }
579 if(softlinebreak) {
580 strcpy(buf, "\x3D\x0D\x0A"); /* "=\r\n" */
581 len = 3;
582 consumed = 0;
583 }
584 }
585
586 /* If the output buffer would overflow, do not store. */
587 if(len > size)
588 break;
589
590 /* Append to output buffer. */
591 memcpy(ptr, buf, len);
592 cursize += len;
593 ptr += len;
594 size -= len;
595 st->pos += len;
596 if(buf[len - 1] == '\x0A') /* '\n' */
597 st->pos = 0;
598 st->bufbeg += consumed;
599 }
600
601 return cursize;
602}
603
604static curl_off_t encoder_qp_size(curl_mimepart *part)
605{
606 /* Determining the size can only be done by reading the data: unless the
607 data size is 0, we return it as unknown (-1). */
608 return part->datasize? -1: 0;
609}
610
611
612/* In-memory data callbacks. */
613/* Argument is a pointer to the mime part. */
614static size_t mime_mem_read(char *buffer, size_t size, size_t nitems,
615 void *instream)
616{
617 curl_mimepart *part = (curl_mimepart *) instream;
618 size_t sz = (size_t) part->datasize - part->state.offset;
619 (void) size; /* Always 1.*/
620
621 if(sz > nitems)
622 sz = nitems;
623
624 if(sz)
625 memcpy(buffer, (char *) &part->data[part->state.offset], sz);
626
627 part->state.offset += sz;
628 return sz;
629}
630
631static int mime_mem_seek(void *instream, curl_off_t offset, int whence)
632{
633 curl_mimepart *part = (curl_mimepart *) instream;
634
635 switch(whence) {
636 case SEEK_CUR:
637 offset += part->state.offset;
638 break;
639 case SEEK_END:
640 offset += part->datasize;
641 break;
642 }
643
644 if(offset < 0 || offset > part->datasize)
645 return CURL_SEEKFUNC_FAIL;
646
647 part->state.offset = (size_t) offset;
648 return CURL_SEEKFUNC_OK;
649}
650
651static void mime_mem_free(void *ptr)
652{
653 Curl_safefree(((curl_mimepart *) ptr)->data);
654}
655
656
657/* Named file callbacks. */
658/* Argument is a pointer to the mime part. */
659static int mime_open_file(curl_mimepart * part)
660{
661 /* Open a MIMEKIND_FILE part. */
662
663 if(part->fp)
664 return 0;
665 part->fp = fopen_read(part->data, "rb");
666 return part->fp? 0: -1;
667}
668
669static size_t mime_file_read(char *buffer, size_t size, size_t nitems,
670 void *instream)
671{
672 curl_mimepart *part = (curl_mimepart *) instream;
673
674 if(mime_open_file(part))
675 return READ_ERROR;
676
677 return fread(buffer, size, nitems, part->fp);
678}
679
680static int mime_file_seek(void *instream, curl_off_t offset, int whence)
681{
682 curl_mimepart *part = (curl_mimepart *) instream;
683
684 if(whence == SEEK_SET && !offset && !part->fp)
685 return CURL_SEEKFUNC_OK; /* Not open: implicitly already at BOF. */
686
687 if(mime_open_file(part))
688 return CURL_SEEKFUNC_FAIL;
689
690 return fseek(part->fp, (long) offset, whence)?
691 CURL_SEEKFUNC_CANTSEEK: CURL_SEEKFUNC_OK;
692}
693
694static void mime_file_free(void *ptr)
695{
696 curl_mimepart *part = (curl_mimepart *) ptr;
697
698 if(part->fp) {
699 fclose(part->fp);
700 part->fp = NULL;
701 }
702 Curl_safefree(part->data);
703 part->data = NULL;
704}
705
706
707/* Subparts callbacks. */
708/* Argument is a pointer to the mime structure. */
709
710/* Readback a byte string segment. */
711static size_t readback_bytes(mime_state *state,
712 char *buffer, size_t bufsize,
713 const char *bytes, size_t numbytes,
714 const char *trail)
715{
716 size_t sz;
717
718 if(numbytes > state->offset) {
719 sz = numbytes - state->offset;
720 bytes += state->offset;
721 }
722 else {
723 size_t tsz = strlen(trail);
724
725 sz = state->offset - numbytes;
726 if(sz >= tsz)
727 return 0;
728 bytes = trail + sz;
729 sz = tsz - sz;
730 }
731
732 if(sz > bufsize)
733 sz = bufsize;
734
735 memcpy(buffer, bytes, sz);
736 state->offset += sz;
737 return sz;
738}
739
740/* Read a non-encoded part content. */
741static size_t read_part_content(curl_mimepart *part,
742 char *buffer, size_t bufsize)
743{
744 size_t sz = 0;
745
746 if(part->readfunc)
747 sz = part->readfunc(buffer, 1, bufsize, part->arg);
748 return sz;
749}
750
751/* Read and encode part content. */
752static size_t read_encoded_part_content(curl_mimepart *part,
753 char *buffer, size_t bufsize)
754{
755 mime_encoder_state *st = &part->encstate;
756 size_t cursize = 0;
757 size_t sz;
758 bool ateof = FALSE;
759
760 while(bufsize) {
761 if(st->bufbeg < st->bufend || ateof) {
762 /* Encode buffered data. */
763 sz = part->encoder->encodefunc(buffer, bufsize, ateof, part);
764 switch(sz) {
765 case 0:
766 if(ateof)
767 return cursize;
768 break;
769 case CURL_READFUNC_ABORT:
770 case CURL_READFUNC_PAUSE:
771 case READ_ERROR:
772 return cursize? cursize: sz;
773 default:
774 cursize += sz;
775 buffer += sz;
776 bufsize -= sz;
777 continue;
778 }
779 }
780
781 /* We need more data in input buffer. */
782 if(st->bufbeg) {
783 size_t len = st->bufend - st->bufbeg;
784
785 if(len)
786 memmove(st->buf, st->buf + st->bufbeg, len);
787 st->bufbeg = 0;
788 st->bufend = len;
789 }
790 if(st->bufend >= sizeof st->buf)
791 return cursize? cursize: READ_ERROR; /* Buffer full. */
792 sz = read_part_content(part, st->buf + st->bufend,
793 sizeof st->buf - st->bufend);
794 switch(sz) {
795 case 0:
796 ateof = TRUE;
797 break;
798 case CURL_READFUNC_ABORT:
799 case CURL_READFUNC_PAUSE:
800 case READ_ERROR:
801 return cursize? cursize: sz;
802 default:
803 st->bufend += sz;
804 break;
805 }
806 }
807
808 return cursize;
809}
810
811/* Readback a mime part. */
812static size_t readback_part(curl_mimepart *part,
813 char *buffer, size_t bufsize)
814{
815 size_t cursize = 0;
816 size_t sz;
817 struct curl_slist *hdr;
818#ifdef CURL_DOES_CONVERSIONS
819 char *convbuf = buffer;
820#endif
821
822 /* Readback from part. */
823
824 while(bufsize) {
825 sz = 0;
826 hdr = (struct curl_slist *) part->state.ptr;
827 switch(part->state.state) {
828 case MIMESTATE_BEGIN:
829 mimesetstate(&part->state, part->flags & MIME_BODY_ONLY? MIMESTATE_BODY:
830 MIMESTATE_CURLHEADERS, part->curlheaders);
831 break;
832 case MIMESTATE_USERHEADERS:
833 if(!hdr) {
834 mimesetstate(&part->state, MIMESTATE_EOH, NULL);
835 break;
836 }
837 if(match_header(hdr, "Content-Type", 12)) {
838 mimesetstate(&part->state, MIMESTATE_USERHEADERS, hdr->next);
839 break;
840 }
841 /* FALLTHROUGH */
842 case MIMESTATE_CURLHEADERS:
843 if(!hdr)
844 mimesetstate(&part->state, MIMESTATE_USERHEADERS, part->userheaders);
845 else {
846 sz = readback_bytes(&part->state, buffer, bufsize,
847 hdr->data, strlen(hdr->data), "\r\n");
848 if(!sz)
849 mimesetstate(&part->state, part->state.state, hdr->next);
850 }
851 break;
852 case MIMESTATE_EOH:
853 sz = readback_bytes(&part->state, buffer, bufsize, "\r\n", 2, "");
854 if(!sz)
855 mimesetstate(&part->state, MIMESTATE_BODY, NULL);
856 break;
857 case MIMESTATE_BODY:
858#ifdef CURL_DOES_CONVERSIONS
859 if(part->easy && convbuf < buffer) {
860 CURLcode result = Curl_convert_to_network(part->easy, convbuf,
861 buffer - convbuf);
862 if(result)
863 return READ_ERROR;
864 convbuf = buffer;
865 }
866#endif
867 cleanup_encoder_state(&part->encstate);
868 mimesetstate(&part->state, MIMESTATE_CONTENT, NULL);
869 break;
870 case MIMESTATE_CONTENT:
871 if(part->encoder)
872 sz = read_encoded_part_content(part, buffer, bufsize);
873 else
874 sz = read_part_content(part, buffer, bufsize);
875 switch(sz) {
876 case 0:
877 mimesetstate(&part->state, MIMESTATE_END, NULL);
878 /* Try sparing open file descriptors. */
879 if(part->kind == MIMEKIND_FILE && part->fp) {
880 fclose(part->fp);
881 part->fp = NULL;
882 }
883 /* FALLTHROUGH */
884 case CURL_READFUNC_ABORT:
885 case CURL_READFUNC_PAUSE:
886 case READ_ERROR:
887 return cursize? cursize: sz;
888 }
889 break;
890 case MIMESTATE_END:
891 return cursize;
892 default:
893 break; /* Other values not in part state. */
894 }
895
896 /* Bump buffer and counters according to read size. */
897 cursize += sz;
898 buffer += sz;
899 bufsize -= sz;
900 }
901
902#ifdef CURL_DOES_CONVERSIONS
903 if(part->easy && convbuf < buffer &&
904 part->state.state < MIMESTATE_BODY) {
905 CURLcode result = Curl_convert_to_network(part->easy, convbuf,
906 buffer - convbuf);
907 if(result)
908 return READ_ERROR;
909 }
910#endif
911
912 return cursize;
913}
914
915/* Readback from mime. */
916static size_t mime_subparts_read(char *buffer, size_t size, size_t nitems,
917 void *instream)
918{
919 curl_mime *mime = (curl_mime *) instream;
920 size_t cursize = 0;
921 size_t sz;
922 curl_mimepart *part;
923#ifdef CURL_DOES_CONVERSIONS
924 char *convbuf = buffer;
925#endif
926
927 (void) size; /* Always 1. */
928
929 while(nitems) {
930 sz = 0;
931 part = mime->state.ptr;
932 switch(mime->state.state) {
933 case MIMESTATE_BEGIN:
934 case MIMESTATE_BODY:
935#ifdef CURL_DOES_CONVERSIONS
936 convbuf = buffer;
937#endif
938 mimesetstate(&mime->state, MIMESTATE_BOUNDARY1, mime->firstpart);
939 /* The first boundary always follows the header termination empty line,
940 so is always preceded by a CRLK. We can then spare 2 characters
941 by skipping the leading CRLF in boundary. */
942 mime->state.offset += 2;
943 break;
944 case MIMESTATE_BOUNDARY1:
945 sz = readback_bytes(&mime->state, buffer, nitems, "\r\n--", 4, "");
946 if(!sz)
947 mimesetstate(&mime->state, MIMESTATE_BOUNDARY2, part);
948 break;
949 case MIMESTATE_BOUNDARY2:
950 sz = readback_bytes(&mime->state, buffer, nitems, mime->boundary,
951 strlen(mime->boundary), part? "\r\n": "--\r\n");
952 if(!sz) {
953#ifdef CURL_DOES_CONVERSIONS
954 if(mime->easy && convbuf < buffer) {
955 CURLcode result = Curl_convert_to_network(mime->easy, convbuf,
956 buffer - convbuf);
957 if(result)
958 return READ_ERROR;
959 convbuf = buffer;
960 }
961#endif
962 mimesetstate(&mime->state, MIMESTATE_CONTENT, part);
963 }
964 break;
965 case MIMESTATE_CONTENT:
966 if(!part) {
967 mimesetstate(&mime->state, MIMESTATE_END, NULL);
968 break;
969 }
970 sz = readback_part(part, buffer, nitems);
971 switch(sz) {
972 case CURL_READFUNC_ABORT:
973 case CURL_READFUNC_PAUSE:
974 case READ_ERROR:
975 return cursize? cursize: sz;
976 case 0:
977#ifdef CURL_DOES_CONVERSIONS
978 convbuf = buffer;
979#endif
980 mimesetstate(&mime->state, MIMESTATE_BOUNDARY1, part->nextpart);
981 break;
982 }
983 break;
984 case MIMESTATE_END:
985 return cursize;
986 default:
987 break; /* other values not used in mime state. */
988 }
989
990 /* Bump buffer and counters according to read size. */
991 cursize += sz;
992 buffer += sz;
993 nitems -= sz;
994 }
995
996#ifdef CURL_DOES_CONVERSIONS
997 if(mime->easy && convbuf < buffer &&
998 mime->state.state <= MIMESTATE_CONTENT) {
999 CURLcode result = Curl_convert_to_network(mime->easy, convbuf,
1000 buffer - convbuf);
1001 if(result)
1002 return READ_ERROR;
1003 }
1004#endif
1005
1006 return cursize;
1007}
1008
1009static int mime_part_rewind(curl_mimepart *part)
1010{
1011 int res = CURL_SEEKFUNC_OK;
1012 enum mimestate targetstate = MIMESTATE_BEGIN;
1013
1014 if(part->flags & MIME_BODY_ONLY)
1015 targetstate = MIMESTATE_BODY;
1016 cleanup_encoder_state(&part->encstate);
1017 if(part->state.state > targetstate) {
1018 res = CURL_SEEKFUNC_CANTSEEK;
1019 if(part->seekfunc) {
1020 res = part->seekfunc(part->arg, (curl_off_t) 0, SEEK_SET);
1021 switch(res) {
1022 case CURL_SEEKFUNC_OK:
1023 case CURL_SEEKFUNC_FAIL:
1024 case CURL_SEEKFUNC_CANTSEEK:
1025 break;
1026 case -1: /* For fseek() error. */
1027 res = CURL_SEEKFUNC_CANTSEEK;
1028 break;
1029 default:
1030 res = CURL_SEEKFUNC_FAIL;
1031 break;
1032 }
1033 }
1034 }
1035
1036 if(res == CURL_SEEKFUNC_OK)
1037 mimesetstate(&part->state, targetstate, NULL);
1038
1039 return res;
1040}
1041
1042static int mime_subparts_seek(void *instream, curl_off_t offset, int whence)
1043{
1044 curl_mime *mime = (curl_mime *) instream;
1045 curl_mimepart *part;
1046 int result = CURL_SEEKFUNC_OK;
1047 int res;
1048
1049 if(whence != SEEK_SET || offset)
1050 return CURL_SEEKFUNC_CANTSEEK; /* Only support full rewind. */
1051
1052 if(mime->state.state == MIMESTATE_BEGIN)
1053 return CURL_SEEKFUNC_OK; /* Already rewound. */
1054
1055 for(part = mime->firstpart; part; part = part->nextpart) {
1056 res = mime_part_rewind(part);
1057 if(res != CURL_SEEKFUNC_OK)
1058 result = res;
1059 }
1060
1061 if(result == CURL_SEEKFUNC_OK)
1062 mimesetstate(&mime->state, MIMESTATE_BEGIN, NULL);
1063
1064 return result;
1065}
1066
1067/* Release part content. */
1068static void cleanup_part_content(curl_mimepart *part)
1069{
1070 if(part->freefunc)
1071 part->freefunc(part->arg);
1072
1073 part->readfunc = NULL;
1074 part->seekfunc = NULL;
1075 part->freefunc = NULL;
1076 part->arg = (void *) part; /* Defaults to part itself. */
1077 part->data = NULL;
1078 part->fp = NULL;
1079 part->datasize = (curl_off_t) 0; /* No size yet. */
1080 cleanup_encoder_state(&part->encstate);
1081 part->kind = MIMEKIND_NONE;
1082}
1083
1084static void mime_subparts_free(void *ptr)
1085{
1086 curl_mime *mime = (curl_mime *) ptr;
1087
1088 if(mime && mime->parent) {
1089 mime->parent->freefunc = NULL; /* Be sure we won't be called again. */
1090 cleanup_part_content(mime->parent); /* Avoid dangling pointer in part. */
1091 }
1092 curl_mime_free(mime);
1093}
1094
1095/* Do not free subparts: unbind them. This is used for the top level only. */
1096static void mime_subparts_unbind(void *ptr)
1097{
1098 curl_mime *mime = (curl_mime *) ptr;
1099
1100 if(mime && mime->parent) {
1101 mime->parent->freefunc = NULL; /* Be sure we won't be called again. */
1102 cleanup_part_content(mime->parent); /* Avoid dangling pointer in part. */
1103 mime->parent = NULL;
1104 }
1105}
1106
1107
1108void Curl_mime_cleanpart(curl_mimepart *part)
1109{
1110 cleanup_part_content(part);
1111 curl_slist_free_all(part->curlheaders);
1112 if(part->flags & MIME_USERHEADERS_OWNER)
1113 curl_slist_free_all(part->userheaders);
1114 Curl_safefree(part->mimetype);
1115 Curl_safefree(part->name);
1116 Curl_safefree(part->filename);
1117 Curl_mime_initpart(part, part->easy);
1118}
1119
1120/* Recursively delete a mime handle and its parts. */
1121void curl_mime_free(curl_mime *mime)
1122{
1123 curl_mimepart *part;
1124
1125 if(mime) {
1126 mime_subparts_unbind(mime); /* Be sure it's not referenced anymore. */
1127 while(mime->firstpart) {
1128 part = mime->firstpart;
1129 mime->firstpart = part->nextpart;
1130 Curl_mime_cleanpart(part);
1131 free(part);
1132 }
1133
1134 free(mime->boundary);
1135 free(mime);
1136 }
1137}
1138
Elliott Hughes0128fe42018-02-27 14:57:55 -08001139CURLcode Curl_mime_duppart(curl_mimepart *dst, const curl_mimepart *src)
1140{
1141 curl_mime *mime;
1142 curl_mimepart *d;
1143 const curl_mimepart *s;
1144 CURLcode res = CURLE_OK;
1145
1146 /* Duplicate content. */
1147 switch(src->kind) {
1148 case MIMEKIND_NONE:
1149 break;
1150 case MIMEKIND_DATA:
1151 res = curl_mime_data(dst, src->data, (size_t) src->datasize);
1152 break;
1153 case MIMEKIND_FILE:
1154 res = curl_mime_filedata(dst, src->data);
1155 /* Do not abort duplication if file is not readable. */
1156 if(res == CURLE_READ_ERROR)
1157 res = CURLE_OK;
1158 break;
1159 case MIMEKIND_CALLBACK:
1160 res = curl_mime_data_cb(dst, src->datasize, src->readfunc,
1161 src->seekfunc, src->freefunc, src->arg);
1162 break;
1163 case MIMEKIND_MULTIPART:
1164 /* No one knows about the cloned subparts, thus always attach ownership
1165 to the part. */
1166 mime = curl_mime_init(dst->easy);
1167 res = mime? curl_mime_subparts(dst, mime): CURLE_OUT_OF_MEMORY;
1168
1169 /* Duplicate subparts. */
1170 for(s = ((curl_mime *) src->arg)->firstpart; !res && s; s = s->nextpart) {
1171 d = curl_mime_addpart(mime);
1172 res = d? Curl_mime_duppart(d, s): CURLE_OUT_OF_MEMORY;
1173 }
1174 break;
1175 default: /* Invalid kind: should not occur. */
1176 res = CURLE_BAD_FUNCTION_ARGUMENT; /* Internal error? */
1177 break;
1178 }
1179
1180 /* Duplicate headers. */
1181 if(!res && src->userheaders) {
1182 struct curl_slist *hdrs = Curl_slist_duplicate(src->userheaders);
1183
1184 if(!hdrs)
1185 res = CURLE_OUT_OF_MEMORY;
1186 else {
1187 /* No one but this procedure knows about the new header list,
1188 so always take ownership. */
1189 res = curl_mime_headers(dst, hdrs, TRUE);
1190 if(res)
1191 curl_slist_free_all(hdrs);
1192 }
1193 }
1194
1195 /* Duplicate other fields. */
Elliott Hughes1ef06ba2018-05-30 15:43:58 -07001196 if(dst != NULL)
1197 dst->encoder = src->encoder;
1198 else
1199 res = CURLE_WRITE_ERROR;
Elliott Hughes0128fe42018-02-27 14:57:55 -08001200 if(!res)
1201 res = curl_mime_type(dst, src->mimetype);
1202 if(!res)
1203 res = curl_mime_name(dst, src->name);
1204 if(!res)
1205 res = curl_mime_filename(dst, src->filename);
1206
1207 /* If an error occurred, rollback. */
Elliott Hughes1ef06ba2018-05-30 15:43:58 -07001208 if(res && dst)
Elliott Hughes0128fe42018-02-27 14:57:55 -08001209 Curl_mime_cleanpart(dst);
1210
1211 return res;
1212}
1213
Alex Deymo486467e2017-12-19 19:04:07 +01001214/*
1215 * Mime build functions.
1216 */
1217
1218/* Create a mime handle. */
1219curl_mime *curl_mime_init(struct Curl_easy *easy)
1220{
1221 curl_mime *mime;
1222
1223 mime = (curl_mime *) malloc(sizeof *mime);
1224
1225 if(mime) {
1226 mime->easy = easy;
1227 mime->parent = NULL;
1228 mime->firstpart = NULL;
1229 mime->lastpart = NULL;
1230
1231 /* Get a part boundary. */
1232 mime->boundary = malloc(24 + MIME_RAND_BOUNDARY_CHARS + 1);
1233 if(!mime->boundary) {
1234 free(mime);
1235 return NULL;
1236 }
1237
1238 memset(mime->boundary, '-', 24);
1239 Curl_rand_hex(easy, (unsigned char *) mime->boundary + 24,
1240 MIME_RAND_BOUNDARY_CHARS + 1);
1241 mimesetstate(&mime->state, MIMESTATE_BEGIN, NULL);
1242 }
1243
1244 return mime;
1245}
1246
1247/* Initialize a mime part. */
1248void Curl_mime_initpart(curl_mimepart *part, struct Curl_easy *easy)
1249{
1250 memset((char *) part, 0, sizeof *part);
1251 part->easy = easy;
1252 mimesetstate(&part->state, MIMESTATE_BEGIN, NULL);
1253}
1254
1255/* Create a mime part and append it to a mime handle's part list. */
1256curl_mimepart *curl_mime_addpart(curl_mime *mime)
1257{
1258 curl_mimepart *part;
1259
1260 if(!mime)
1261 return NULL;
1262
1263 part = (curl_mimepart *) malloc(sizeof *part);
1264
1265 if(part) {
1266 Curl_mime_initpart(part, mime->easy);
1267 part->parent = mime;
1268
1269 if(mime->lastpart)
1270 mime->lastpart->nextpart = part;
1271 else
1272 mime->firstpart = part;
1273
1274 mime->lastpart = part;
1275 }
1276
1277 return part;
1278}
1279
1280/* Set mime part name. */
1281CURLcode curl_mime_name(curl_mimepart *part, const char *name)
1282{
1283 if(!part)
1284 return CURLE_BAD_FUNCTION_ARGUMENT;
1285
1286 Curl_safefree(part->name);
1287 part->name = NULL;
1288
1289 if(name) {
1290 part->name = strdup(name);
1291 if(!part->name)
1292 return CURLE_OUT_OF_MEMORY;
1293 }
1294
1295 return CURLE_OK;
1296}
1297
1298/* Set mime part remote file name. */
1299CURLcode curl_mime_filename(curl_mimepart *part, const char *filename)
1300{
1301 if(!part)
1302 return CURLE_BAD_FUNCTION_ARGUMENT;
1303
1304 Curl_safefree(part->filename);
1305 part->filename = NULL;
1306
1307 if(filename) {
1308 part->filename = strdup(filename);
1309 if(!part->filename)
1310 return CURLE_OUT_OF_MEMORY;
1311 }
1312
1313 return CURLE_OK;
1314}
1315
1316/* Set mime part content from memory data. */
1317CURLcode curl_mime_data(curl_mimepart *part,
1318 const char *data, size_t datasize)
1319{
1320 if(!part)
1321 return CURLE_BAD_FUNCTION_ARGUMENT;
1322
1323 cleanup_part_content(part);
1324
1325 if(data) {
1326 if(datasize == CURL_ZERO_TERMINATED)
1327 datasize = strlen(data);
1328
1329 part->data = malloc(datasize + 1);
1330 if(!part->data)
1331 return CURLE_OUT_OF_MEMORY;
1332
1333 part->datasize = datasize;
1334
1335 if(datasize)
1336 memcpy(part->data, data, datasize);
1337 part->data[datasize] = '\0'; /* Set a nul terminator as sentinel. */
1338
1339 part->readfunc = mime_mem_read;
1340 part->seekfunc = mime_mem_seek;
1341 part->freefunc = mime_mem_free;
1342 part->kind = MIMEKIND_DATA;
1343 }
1344
1345 return CURLE_OK;
1346}
1347
1348/* Set mime part content from named local file. */
1349CURLcode curl_mime_filedata(curl_mimepart *part, const char *filename)
1350{
1351 CURLcode result = CURLE_OK;
1352 char *base;
1353
1354 if(!part)
1355 return CURLE_BAD_FUNCTION_ARGUMENT;
1356
1357 cleanup_part_content(part);
1358
1359 if(filename) {
1360 struct_stat sbuf;
1361
1362 if(stat(filename, &sbuf) || access(filename, R_OK))
1363 result = CURLE_READ_ERROR;
1364
1365 part->data = strdup(filename);
1366 if(!part->data)
1367 result = CURLE_OUT_OF_MEMORY;
1368
1369 part->datasize = -1;
1370 if(!result && S_ISREG(sbuf.st_mode)) {
1371 part->datasize = filesize(filename, sbuf);
1372 part->seekfunc = mime_file_seek;
1373 }
1374
1375 part->readfunc = mime_file_read;
1376 part->freefunc = mime_file_free;
1377 part->kind = MIMEKIND_FILE;
1378
1379 /* As a side effect, set the filename to the current file's base name.
1380 It is possible to withdraw this by explicitly calling
1381 curl_mime_filename() with a NULL filename argument after the current
1382 call. */
1383 base = strippath(filename);
1384 if(!base)
1385 result = CURLE_OUT_OF_MEMORY;
1386 else {
1387 CURLcode res = curl_mime_filename(part, base);
1388
1389 if(res)
1390 result = res;
1391 free(base);
1392 }
1393 }
1394 return result;
1395}
1396
1397/* Set mime part type. */
1398CURLcode curl_mime_type(curl_mimepart *part, const char *mimetype)
1399{
1400 if(!part)
1401 return CURLE_BAD_FUNCTION_ARGUMENT;
1402
1403 Curl_safefree(part->mimetype);
1404 part->mimetype = NULL;
1405
1406 if(mimetype) {
1407 part->mimetype = strdup(mimetype);
1408 if(!part->mimetype)
1409 return CURLE_OUT_OF_MEMORY;
1410 }
1411
1412 return CURLE_OK;
1413}
1414
1415/* Set mime data transfer encoder. */
1416CURLcode curl_mime_encoder(curl_mimepart *part, const char *encoding)
1417{
1418 CURLcode result = CURLE_BAD_FUNCTION_ARGUMENT;
1419 const mime_encoder *mep;
1420
1421 if(!part)
1422 return result;
1423
1424 part->encoder = NULL;
1425
1426 if(!encoding)
1427 return CURLE_OK; /* Removing current encoder. */
1428
1429 for(mep = encoders; mep->name; mep++)
1430 if(strcasecompare(encoding, mep->name)) {
1431 part->encoder = mep;
1432 result = CURLE_OK;
1433 }
1434
1435 return result;
1436}
1437
1438/* Set mime part headers. */
1439CURLcode curl_mime_headers(curl_mimepart *part,
1440 struct curl_slist *headers, int take_ownership)
1441{
1442 if(!part)
1443 return CURLE_BAD_FUNCTION_ARGUMENT;
1444
1445 if(part->flags & MIME_USERHEADERS_OWNER) {
1446 if(part->userheaders != headers) /* Allow setting twice the same list. */
1447 curl_slist_free_all(part->userheaders);
1448 part->flags &= ~MIME_USERHEADERS_OWNER;
1449 }
1450 part->userheaders = headers;
1451 if(headers && take_ownership)
1452 part->flags |= MIME_USERHEADERS_OWNER;
1453 return CURLE_OK;
1454}
1455
1456/* Set mime part content from callback. */
1457CURLcode curl_mime_data_cb(curl_mimepart *part, curl_off_t datasize,
1458 curl_read_callback readfunc,
1459 curl_seek_callback seekfunc,
1460 curl_free_callback freefunc, void *arg)
1461{
1462 if(!part)
1463 return CURLE_BAD_FUNCTION_ARGUMENT;
1464
1465 cleanup_part_content(part);
1466
1467 if(readfunc) {
1468 part->readfunc = readfunc;
1469 part->seekfunc = seekfunc;
1470 part->freefunc = freefunc;
1471 part->arg = arg;
1472 part->datasize = datasize;
1473 part->kind = MIMEKIND_CALLBACK;
1474 }
1475
1476 return CURLE_OK;
1477}
1478
1479/* Set mime part content from subparts. */
1480CURLcode Curl_mime_set_subparts(curl_mimepart *part,
1481 curl_mime *subparts, int take_ownership)
1482{
1483 curl_mime *root;
1484
1485 if(!part)
1486 return CURLE_BAD_FUNCTION_ARGUMENT;
1487
1488 /* Accept setting twice the same subparts. */
1489 if(part->kind == MIMEKIND_MULTIPART && part->arg == subparts)
1490 return CURLE_OK;
1491
1492 cleanup_part_content(part);
1493
1494 if(subparts) {
1495 /* Must belong to the same data handle. */
1496 if(part->easy && subparts->easy && part->easy != subparts->easy)
1497 return CURLE_BAD_FUNCTION_ARGUMENT;
1498
1499 /* Should not have been attached already. */
1500 if(subparts->parent)
1501 return CURLE_BAD_FUNCTION_ARGUMENT;
1502
1503 /* Should not be the part's root. */
1504 root = part->parent;
1505 if(root) {
1506 while(root->parent && root->parent->parent)
1507 root = root->parent->parent;
1508 if(subparts == root) {
1509 if(part->easy)
1510 failf(part->easy, "Can't add itself as a subpart!");
1511 return CURLE_BAD_FUNCTION_ARGUMENT;
1512 }
1513 }
1514
1515 subparts->parent = part;
1516 part->readfunc = mime_subparts_read;
1517 part->seekfunc = mime_subparts_seek;
1518 part->freefunc = take_ownership? mime_subparts_free: mime_subparts_unbind;
1519 part->arg = subparts;
1520 part->datasize = -1;
1521 part->kind = MIMEKIND_MULTIPART;
1522 }
1523
1524 return CURLE_OK;
1525}
1526
1527CURLcode curl_mime_subparts(curl_mimepart *part, curl_mime *subparts)
1528{
1529 return Curl_mime_set_subparts(part, subparts, TRUE);
1530}
1531
1532
1533/* Readback from top mime. */
1534/* Argument is the dummy top part. */
1535size_t Curl_mime_read(char *buffer, size_t size, size_t nitems, void *instream)
1536{
1537 curl_mimepart *part = (curl_mimepart *) instream;
1538
1539 (void) size; /* Always 1. */
1540 return readback_part(part, buffer, nitems);
1541}
1542
1543/* Rewind mime stream. */
1544CURLcode Curl_mime_rewind(curl_mimepart *part)
1545{
1546 return mime_part_rewind(part) == CURL_SEEKFUNC_OK?
1547 CURLE_OK: CURLE_SEND_FAIL_REWIND;
1548}
1549
1550/* Compute header list size. */
1551static size_t slist_size(struct curl_slist *s,
1552 size_t overhead, const char *skip)
1553{
1554 size_t size = 0;
1555 size_t skiplen = skip? strlen(skip): 0;
1556
1557 for(; s; s = s->next)
1558 if(!skip || !match_header(s, skip, skiplen))
1559 size += strlen(s->data) + overhead;
1560 return size;
1561}
1562
1563/* Get/compute multipart size. */
1564static curl_off_t multipart_size(curl_mime *mime)
1565{
1566 curl_off_t size;
1567 curl_off_t sz;
1568 size_t boundarysize;
1569 curl_mimepart *part;
1570
1571 if(!mime)
1572 return 0; /* Not present -> empty. */
1573
1574 boundarysize = 4 + strlen(mime->boundary) + 2;
1575 size = boundarysize; /* Final boundary - CRLF after headers. */
1576
1577 for(part = mime->firstpart; part; part = part->nextpart) {
1578 sz = Curl_mime_size(part);
1579
1580 if(sz < 0)
1581 size = sz;
1582
1583 if(size >= 0)
1584 size += boundarysize + sz;
1585 }
1586
1587 return size;
1588}
1589
1590/* Get/compute mime size. */
1591curl_off_t Curl_mime_size(curl_mimepart *part)
1592{
1593 curl_off_t size;
1594
1595 if(part->kind == MIMEKIND_MULTIPART)
1596 part->datasize = multipart_size(part->arg);
1597
1598 size = part->datasize;
1599
1600 if(part->encoder)
1601 size = part->encoder->sizefunc(part);
1602
1603 if(size >= 0 && !(part->flags & MIME_BODY_ONLY)) {
1604 /* Compute total part size. */
1605 size += slist_size(part->curlheaders, 2, NULL);
1606 size += slist_size(part->userheaders, 2, "Content-Type");
1607 size += 2; /* CRLF after headers. */
1608 }
1609 return size;
1610}
1611
1612/* Add a header. */
1613/* VARARGS2 */
1614CURLcode Curl_mime_add_header(struct curl_slist **slp, const char *fmt, ...)
1615{
1616 struct curl_slist *hdr = NULL;
1617 char *s = NULL;
1618 va_list ap;
1619
1620 va_start(ap, fmt);
1621 s = curl_mvaprintf(fmt, ap);
1622 va_end(ap);
1623
1624 if(s) {
1625 hdr = Curl_slist_append_nodup(*slp, s);
1626 if(hdr)
1627 *slp = hdr;
1628 else
1629 free(s);
1630 }
1631
1632 return hdr? CURLE_OK: CURLE_OUT_OF_MEMORY;
1633}
1634
1635/* Add a content type header. */
1636static CURLcode add_content_type(struct curl_slist **slp,
1637 const char *type, const char *boundary)
1638{
1639 return Curl_mime_add_header(slp, "Content-Type: %s%s%s", type,
1640 boundary? "; boundary=": "",
1641 boundary? boundary: "");
1642}
1643
Elliott Hughescac39802018-04-27 16:19:43 -07001644const char *Curl_mime_contenttype(const char *filename)
Alex Deymo486467e2017-12-19 19:04:07 +01001645{
1646 unsigned int i;
1647
1648 /*
1649 * If no content type was specified, we scan through a few well-known
1650 * extensions and pick the first we match!
1651 */
1652 struct ContentType {
1653 const char *extension;
1654 const char *type;
1655 };
1656 static const struct ContentType ctts[] = {
1657 {".gif", "image/gif"},
1658 {".jpg", "image/jpeg"},
1659 {".jpeg", "image/jpeg"},
1660 {".png", "image/png"},
1661 {".svg", "image/svg+xml"},
1662 {".txt", "text/plain"},
1663 {".htm", "text/html"},
1664 {".html", "text/html"},
1665 {".pdf", "application/pdf"},
1666 {".xml", "application/xml"}
1667 };
1668
1669 if(filename) {
1670 size_t len1 = strlen(filename);
1671 const char *nameend = filename + len1;
1672
1673 for(i = 0; i < sizeof ctts / sizeof ctts[0]; i++) {
1674 size_t len2 = strlen(ctts[i].extension);
1675
1676 if(len1 >= len2 && strcasecompare(nameend - len2, ctts[i].extension))
1677 return ctts[i].type;
1678 }
1679 }
1680 return NULL;
1681}
1682
1683CURLcode Curl_mime_prepare_headers(curl_mimepart *part,
1684 const char *contenttype,
1685 const char *disposition,
1686 enum mimestrategy strategy)
1687{
1688 curl_mime *mime = NULL;
1689 const char *boundary = NULL;
1690 char *customct;
1691 const char *cte = NULL;
1692 CURLcode ret = CURLE_OK;
1693
1694 /* Get rid of previously prepared headers. */
1695 curl_slist_free_all(part->curlheaders);
1696 part->curlheaders = NULL;
1697
1698 /* Be sure we won't access old headers later. */
1699 if(part->state.state == MIMESTATE_CURLHEADERS)
1700 mimesetstate(&part->state, MIMESTATE_CURLHEADERS, NULL);
1701
1702 /* Check if content type is specified. */
1703 customct = part->mimetype;
1704 if(!customct)
1705 customct = search_header(part->userheaders, "Content-Type");
1706 if(customct)
1707 contenttype = customct;
1708
1709 /* If content type is not specified, try to determine it. */
1710 if(!contenttype) {
1711 switch(part->kind) {
1712 case MIMEKIND_MULTIPART:
1713 contenttype = MULTIPART_CONTENTTYPE_DEFAULT;
1714 break;
1715 case MIMEKIND_FILE:
Elliott Hughescac39802018-04-27 16:19:43 -07001716 contenttype = Curl_mime_contenttype(part->filename);
Alex Deymo486467e2017-12-19 19:04:07 +01001717 if(!contenttype)
Elliott Hughescac39802018-04-27 16:19:43 -07001718 contenttype = Curl_mime_contenttype(part->data);
Alex Deymo486467e2017-12-19 19:04:07 +01001719 if(!contenttype && part->filename)
1720 contenttype = FILE_CONTENTTYPE_DEFAULT;
1721 break;
1722 default:
Elliott Hughescac39802018-04-27 16:19:43 -07001723 contenttype = Curl_mime_contenttype(part->filename);
Alex Deymo486467e2017-12-19 19:04:07 +01001724 break;
1725 }
1726 }
1727
1728 if(part->kind == MIMEKIND_MULTIPART) {
1729 mime = (curl_mime *) part->arg;
1730 if(mime)
1731 boundary = mime->boundary;
1732 }
1733 else if(contenttype && !customct &&
1734 strcasecompare(contenttype, "text/plain"))
1735 if(strategy == MIMESTRATEGY_MAIL || !part->filename)
1736 contenttype = NULL;
1737
1738 /* Issue content-disposition header only if not already set by caller. */
1739 if(!search_header(part->userheaders, "Content-Disposition")) {
1740 if(!disposition)
1741 if(part->filename || part->name ||
1742 (contenttype && !strncasecompare(contenttype, "multipart/", 10)))
1743 disposition = DISPOSITION_DEFAULT;
1744 if(disposition && curl_strequal(disposition, "attachment") &&
1745 !part->name && !part->filename)
1746 disposition = NULL;
1747 if(disposition) {
1748 char *name = NULL;
1749 char *filename = NULL;
1750
1751 if(part->name) {
1752 name = escape_string(part->name);
1753 if(!name)
1754 ret = CURLE_OUT_OF_MEMORY;
1755 }
1756 if(!ret && part->filename) {
1757 filename = escape_string(part->filename);
1758 if(!filename)
1759 ret = CURLE_OUT_OF_MEMORY;
1760 }
1761 if(!ret)
1762 ret = Curl_mime_add_header(&part->curlheaders,
1763 "Content-Disposition: %s%s%s%s%s%s%s",
1764 disposition,
1765 name? "; name=\"": "",
1766 name? name: "",
1767 name? "\"": "",
1768 filename? "; filename=\"": "",
1769 filename? filename: "",
1770 filename? "\"": "");
1771 Curl_safefree(name);
1772 Curl_safefree(filename);
1773 if(ret)
1774 return ret;
1775 }
1776 }
1777
1778 /* Issue Content-Type header. */
1779 if(contenttype) {
1780 ret = add_content_type(&part->curlheaders, contenttype, boundary);
1781 if(ret)
1782 return ret;
1783 }
1784
1785 /* Content-Transfer-Encoding header. */
1786 if(!search_header(part->userheaders, "Content-Transfer-Encoding")) {
1787 if(part->encoder)
1788 cte = part->encoder->name;
1789 else if(contenttype && strategy == MIMESTRATEGY_MAIL &&
1790 part->kind != MIMEKIND_MULTIPART)
1791 cte = "8bit";
1792 if(cte) {
1793 ret = Curl_mime_add_header(&part->curlheaders,
1794 "Content-Transfer-Encoding: %s", cte);
1795 if(ret)
1796 return ret;
1797 }
1798 }
1799
1800 /* If we were reading curl-generated headers, restart with new ones (this
1801 should not occur). */
1802 if(part->state.state == MIMESTATE_CURLHEADERS)
1803 mimesetstate(&part->state, MIMESTATE_CURLHEADERS, part->curlheaders);
1804
1805 /* Process subparts. */
1806 if(part->kind == MIMEKIND_MULTIPART && mime) {
1807 curl_mimepart *subpart;
1808
1809 disposition = NULL;
1810 if(strcasecompare(contenttype, "multipart/form-data"))
1811 disposition = "form-data";
1812 for(subpart = mime->firstpart; subpart; subpart = subpart->nextpart) {
1813 ret = Curl_mime_prepare_headers(subpart, NULL, disposition, strategy);
1814 if(ret)
1815 return ret;
1816 }
1817 }
1818 return ret;
1819}
1820
1821#else /* !CURL_DISABLE_HTTP || !CURL_DISABLE_SMTP || !CURL_DISABLE_IMAP */
1822
1823/* Mime not compiled in: define stubs for externally-referenced functions. */
1824curl_mime *curl_mime_init(CURL *easy)
1825{
1826 (void) easy;
1827 return NULL;
1828}
1829
1830void curl_mime_free(curl_mime *mime)
1831{
1832 (void) mime;
1833}
1834
1835curl_mimepart *curl_mime_addpart(curl_mime *mime)
1836{
1837 (void) mime;
1838 return NULL;
1839}
1840
1841CURLcode curl_mime_name(curl_mimepart *part, const char *name)
1842{
1843 (void) part;
1844 (void) name;
1845 return CURLE_NOT_BUILT_IN;
1846}
1847
1848CURLcode curl_mime_filename(curl_mimepart *part, const char *filename)
1849{
1850 (void) part;
1851 (void) filename;
1852 return CURLE_NOT_BUILT_IN;
1853}
1854
1855CURLcode curl_mime_type(curl_mimepart *part, const char *mimetype)
1856{
1857 (void) part;
1858 (void) mimetype;
1859 return CURLE_NOT_BUILT_IN;
1860}
1861
1862CURLcode curl_mime_encoder(curl_mimepart *part, const char *encoding)
1863{
1864 (void) part;
1865 (void) encoding;
1866 return CURLE_NOT_BUILT_IN;
1867}
1868
1869CURLcode curl_mime_data(curl_mimepart *part,
1870 const char *data, size_t datasize)
1871{
1872 (void) part;
1873 (void) data;
1874 (void) datasize;
1875 return CURLE_NOT_BUILT_IN;
1876}
1877
1878CURLcode curl_mime_filedata(curl_mimepart *part, const char *filename)
1879{
1880 (void) part;
1881 (void) filename;
1882 return CURLE_NOT_BUILT_IN;
1883}
1884
1885CURLcode curl_mime_data_cb(curl_mimepart *part,
1886 curl_off_t datasize,
1887 curl_read_callback readfunc,
1888 curl_seek_callback seekfunc,
1889 curl_free_callback freefunc,
1890 void *arg)
1891{
1892 (void) part;
1893 (void) datasize;
1894 (void) readfunc;
1895 (void) seekfunc;
1896 (void) freefunc;
1897 (void) arg;
1898 return CURLE_NOT_BUILT_IN;
1899}
1900
1901CURLcode curl_mime_subparts(curl_mimepart *part, curl_mime *subparts)
1902{
1903 (void) part;
1904 (void) subparts;
1905 return CURLE_NOT_BUILT_IN;
1906}
1907
1908CURLcode curl_mime_headers(curl_mimepart *part,
1909 struct curl_slist *headers, int take_ownership)
1910{
1911 (void) part;
1912 (void) headers;
1913 (void) take_ownership;
1914 return CURLE_NOT_BUILT_IN;
1915}
1916
1917void Curl_mime_initpart(curl_mimepart *part, struct Curl_easy *easy)
1918{
1919 (void) part;
1920 (void) easy;
1921}
1922
1923void Curl_mime_cleanpart(curl_mimepart *part)
1924{
1925 (void) part;
1926}
1927
Elliott Hughes0128fe42018-02-27 14:57:55 -08001928CURLcode Curl_mime_duppart(curl_mimepart *dst, const curl_mimepart *src)
1929{
1930 (void) dst;
1931 (void) src;
1932 return CURLE_OK; /* Nothing to duplicate: always succeed. */
1933}
1934
Alex Deymo486467e2017-12-19 19:04:07 +01001935CURLcode Curl_mime_set_subparts(curl_mimepart *part,
1936 curl_mime *subparts, int take_ownership)
1937{
1938 (void) part;
1939 (void) subparts;
1940 (void) take_ownership;
1941 return CURLE_NOT_BUILT_IN;
1942}
1943
1944CURLcode Curl_mime_prepare_headers(curl_mimepart *part,
1945 const char *contenttype,
1946 const char *disposition,
1947 enum mimestrategy strategy)
1948{
1949 (void) part;
1950 (void) contenttype;
1951 (void) disposition;
1952 (void) strategy;
1953 return CURLE_NOT_BUILT_IN;
1954}
1955
1956curl_off_t Curl_mime_size(curl_mimepart *part)
1957{
1958 (void) part;
1959 return (curl_off_t) -1;
1960}
1961
1962size_t Curl_mime_read(char *buffer, size_t size, size_t nitems, void *instream)
1963{
1964 (void) buffer;
1965 (void) size;
1966 (void) nitems;
1967 (void) instream;
1968 return 0;
1969}
1970
1971CURLcode Curl_mime_rewind(curl_mimepart *part)
1972{
1973 (void) part;
1974 return CURLE_NOT_BUILT_IN;
1975}
1976
1977/* VARARGS2 */
1978CURLcode Curl_mime_add_header(struct curl_slist **slp, const char *fmt, ...)
1979{
1980 (void) slp;
1981 (void) fmt;
1982 return CURLE_NOT_BUILT_IN;
1983}
1984
1985#endif /* !CURL_DISABLE_HTTP || !CURL_DISABLE_SMTP || !CURL_DISABLE_IMAP */