mime.c 50 KB

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