Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1 | /*************************************************************************** |
| 2 | * _ _ ____ _ |
| 3 | * Project ___| | | | _ \| | |
| 4 | * / __| | | | |_) | | |
| 5 | * | (__| |_| | _ <| |___ |
| 6 | * \___|\___/|_| \_\_____| |
| 7 | * |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 8 | * Copyright (C) 1998 - 2018, Daniel Stenberg, <daniel@haxx.se>, et al. |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 9 | * |
| 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 Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 54 | #define READ_ERROR ((size_t) -1) |
| 55 | |
| 56 | /* Encoders. */ |
| 57 | static size_t encoder_nop_read(char *buffer, size_t size, bool ateof, |
| 58 | curl_mimepart *part); |
| 59 | static curl_off_t encoder_nop_size(curl_mimepart *part); |
| 60 | static size_t encoder_7bit_read(char *buffer, size_t size, bool ateof, |
| 61 | curl_mimepart *part); |
| 62 | static size_t encoder_base64_read(char *buffer, size_t size, bool ateof, |
| 63 | curl_mimepart *part); |
| 64 | static curl_off_t encoder_base64_size(curl_mimepart *part); |
| 65 | static size_t encoder_qp_read(char *buffer, size_t size, bool ateof, |
| 66 | curl_mimepart *part); |
| 67 | static curl_off_t encoder_qp_size(curl_mimepart *part); |
| 68 | |
| 69 | static 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 */ |
| 79 | static 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. */ |
| 90 | static 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. */ |
| 119 | static 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 | */ |
| 141 | curl_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 | */ |
| 171 | static 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 | */ |
| 194 | static 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 | */ |
| 241 | static char *Curl_basename(char *path) |
| 242 | { |
| 243 | /* Ignore all the details above for now and make a quick and simple |
Elliott Hughes | 1ef06ba | 2018-05-30 15:43:58 -0700 | [diff] [blame] | 244 | implementation here */ |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 245 | 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. */ |
| 267 | static 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. */ |
| 276 | static 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. */ |
| 302 | static 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. */ |
| 313 | static 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 | |
| 324 | static 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. */ |
| 340 | static 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. */ |
| 349 | static 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 | |
| 365 | static 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. */ |
| 372 | static 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. */ |
| 395 | static 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 | |
| 468 | static 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 | */ |
| 488 | static 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. */ |
| 502 | static 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 | |
| 604 | static 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. */ |
| 614 | static 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 | |
| 631 | static 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 | |
| 651 | static 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. */ |
| 659 | static 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 | |
| 669 | static 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 | |
| 680 | static 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 | |
| 694 | static 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. */ |
| 711 | static 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. */ |
| 741 | static 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. */ |
| 752 | static 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. */ |
| 812 | static 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. */ |
| 916 | static 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 | |
| 1009 | static 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 | |
| 1042 | static 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. */ |
| 1068 | static 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 | |
| 1084 | static 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. */ |
| 1096 | static 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 | |
| 1108 | void 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. */ |
| 1121 | void 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 Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 1139 | CURLcode 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 Hughes | 1ef06ba | 2018-05-30 15:43:58 -0700 | [diff] [blame] | 1196 | if(dst != NULL) |
| 1197 | dst->encoder = src->encoder; |
| 1198 | else |
| 1199 | res = CURLE_WRITE_ERROR; |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 1200 | 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 Hughes | 1ef06ba | 2018-05-30 15:43:58 -0700 | [diff] [blame] | 1208 | if(res && dst) |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 1209 | Curl_mime_cleanpart(dst); |
| 1210 | |
| 1211 | return res; |
| 1212 | } |
| 1213 | |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1214 | /* |
| 1215 | * Mime build functions. |
| 1216 | */ |
| 1217 | |
| 1218 | /* Create a mime handle. */ |
| 1219 | curl_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. */ |
| 1248 | void 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. */ |
| 1256 | curl_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. */ |
| 1281 | CURLcode 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. */ |
| 1299 | CURLcode 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. */ |
| 1317 | CURLcode 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. */ |
| 1349 | CURLcode 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. */ |
| 1398 | CURLcode 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. */ |
| 1416 | CURLcode 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. */ |
| 1439 | CURLcode 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. */ |
| 1457 | CURLcode 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. */ |
| 1480 | CURLcode 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 | |
| 1527 | CURLcode 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. */ |
| 1535 | size_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. */ |
| 1544 | CURLcode 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. */ |
| 1551 | static 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. */ |
| 1564 | static 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. */ |
| 1591 | curl_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 */ |
| 1614 | CURLcode 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. */ |
| 1636 | static 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 Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 1644 | const char *Curl_mime_contenttype(const char *filename) |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1645 | { |
| 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 | |
| 1683 | CURLcode 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 Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 1716 | contenttype = Curl_mime_contenttype(part->filename); |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1717 | if(!contenttype) |
Elliott Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 1718 | contenttype = Curl_mime_contenttype(part->data); |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1719 | if(!contenttype && part->filename) |
| 1720 | contenttype = FILE_CONTENTTYPE_DEFAULT; |
| 1721 | break; |
| 1722 | default: |
Elliott Hughes | cac3980 | 2018-04-27 16:19:43 -0700 | [diff] [blame] | 1723 | contenttype = Curl_mime_contenttype(part->filename); |
Alex Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1724 | 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. */ |
| 1824 | curl_mime *curl_mime_init(CURL *easy) |
| 1825 | { |
| 1826 | (void) easy; |
| 1827 | return NULL; |
| 1828 | } |
| 1829 | |
| 1830 | void curl_mime_free(curl_mime *mime) |
| 1831 | { |
| 1832 | (void) mime; |
| 1833 | } |
| 1834 | |
| 1835 | curl_mimepart *curl_mime_addpart(curl_mime *mime) |
| 1836 | { |
| 1837 | (void) mime; |
| 1838 | return NULL; |
| 1839 | } |
| 1840 | |
| 1841 | CURLcode curl_mime_name(curl_mimepart *part, const char *name) |
| 1842 | { |
| 1843 | (void) part; |
| 1844 | (void) name; |
| 1845 | return CURLE_NOT_BUILT_IN; |
| 1846 | } |
| 1847 | |
| 1848 | CURLcode curl_mime_filename(curl_mimepart *part, const char *filename) |
| 1849 | { |
| 1850 | (void) part; |
| 1851 | (void) filename; |
| 1852 | return CURLE_NOT_BUILT_IN; |
| 1853 | } |
| 1854 | |
| 1855 | CURLcode curl_mime_type(curl_mimepart *part, const char *mimetype) |
| 1856 | { |
| 1857 | (void) part; |
| 1858 | (void) mimetype; |
| 1859 | return CURLE_NOT_BUILT_IN; |
| 1860 | } |
| 1861 | |
| 1862 | CURLcode curl_mime_encoder(curl_mimepart *part, const char *encoding) |
| 1863 | { |
| 1864 | (void) part; |
| 1865 | (void) encoding; |
| 1866 | return CURLE_NOT_BUILT_IN; |
| 1867 | } |
| 1868 | |
| 1869 | CURLcode 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 | |
| 1878 | CURLcode curl_mime_filedata(curl_mimepart *part, const char *filename) |
| 1879 | { |
| 1880 | (void) part; |
| 1881 | (void) filename; |
| 1882 | return CURLE_NOT_BUILT_IN; |
| 1883 | } |
| 1884 | |
| 1885 | CURLcode 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 | |
| 1901 | CURLcode curl_mime_subparts(curl_mimepart *part, curl_mime *subparts) |
| 1902 | { |
| 1903 | (void) part; |
| 1904 | (void) subparts; |
| 1905 | return CURLE_NOT_BUILT_IN; |
| 1906 | } |
| 1907 | |
| 1908 | CURLcode 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 | |
| 1917 | void Curl_mime_initpart(curl_mimepart *part, struct Curl_easy *easy) |
| 1918 | { |
| 1919 | (void) part; |
| 1920 | (void) easy; |
| 1921 | } |
| 1922 | |
| 1923 | void Curl_mime_cleanpart(curl_mimepart *part) |
| 1924 | { |
| 1925 | (void) part; |
| 1926 | } |
| 1927 | |
Elliott Hughes | 0128fe4 | 2018-02-27 14:57:55 -0800 | [diff] [blame] | 1928 | CURLcode 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 Deymo | 486467e | 2017-12-19 19:04:07 +0100 | [diff] [blame] | 1935 | CURLcode 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 | |
| 1944 | CURLcode 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 | |
| 1956 | curl_off_t Curl_mime_size(curl_mimepart *part) |
| 1957 | { |
| 1958 | (void) part; |
| 1959 | return (curl_off_t) -1; |
| 1960 | } |
| 1961 | |
| 1962 | size_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 | |
| 1971 | CURLcode Curl_mime_rewind(curl_mimepart *part) |
| 1972 | { |
| 1973 | (void) part; |
| 1974 | return CURLE_NOT_BUILT_IN; |
| 1975 | } |
| 1976 | |
| 1977 | /* VARARGS2 */ |
| 1978 | CURLcode 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 */ |