Annotation of embedaddon/curl/lib/transfer.c, revision 1.1
1.1 ! misho 1: /***************************************************************************
! 2: * _ _ ____ _
! 3: * Project ___| | | | _ \| |
! 4: * / __| | | | |_) | |
! 5: * | (__| |_| | _ <| |___
! 6: * \___|\___/|_| \_\_____|
! 7: *
! 8: * Copyright (C) 1998 - 2020, Daniel Stenberg, <daniel@haxx.se>, 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:
! 23: #include "curl_setup.h"
! 24: #include "strtoofft.h"
! 25:
! 26: #ifdef HAVE_NETINET_IN_H
! 27: #include <netinet/in.h>
! 28: #endif
! 29: #ifdef HAVE_NETDB_H
! 30: #include <netdb.h>
! 31: #endif
! 32: #ifdef HAVE_ARPA_INET_H
! 33: #include <arpa/inet.h>
! 34: #endif
! 35: #ifdef HAVE_NET_IF_H
! 36: #include <net/if.h>
! 37: #endif
! 38: #ifdef HAVE_SYS_IOCTL_H
! 39: #include <sys/ioctl.h>
! 40: #endif
! 41: #ifdef HAVE_SIGNAL_H
! 42: #include <signal.h>
! 43: #endif
! 44:
! 45: #ifdef HAVE_SYS_PARAM_H
! 46: #include <sys/param.h>
! 47: #endif
! 48:
! 49: #ifdef HAVE_SYS_SELECT_H
! 50: #include <sys/select.h>
! 51: #elif defined(HAVE_UNISTD_H)
! 52: #include <unistd.h>
! 53: #endif
! 54:
! 55: #ifndef HAVE_SOCKET
! 56: #error "We can't compile without socket() support!"
! 57: #endif
! 58:
! 59: #include "urldata.h"
! 60: #include <curl/curl.h>
! 61: #include "netrc.h"
! 62:
! 63: #include "content_encoding.h"
! 64: #include "hostip.h"
! 65: #include "transfer.h"
! 66: #include "sendf.h"
! 67: #include "speedcheck.h"
! 68: #include "progress.h"
! 69: #include "http.h"
! 70: #include "url.h"
! 71: #include "getinfo.h"
! 72: #include "vtls/vtls.h"
! 73: #include "select.h"
! 74: #include "multiif.h"
! 75: #include "connect.h"
! 76: #include "non-ascii.h"
! 77: #include "http2.h"
! 78: #include "mime.h"
! 79: #include "strcase.h"
! 80: #include "urlapi-int.h"
! 81:
! 82: /* The last 3 #include files should be in this order */
! 83: #include "curl_printf.h"
! 84: #include "curl_memory.h"
! 85: #include "memdebug.h"
! 86:
! 87: #if !defined(CURL_DISABLE_HTTP) || !defined(CURL_DISABLE_SMTP) || \
! 88: !defined(CURL_DISABLE_IMAP)
! 89: /*
! 90: * checkheaders() checks the linked list of custom headers for a
! 91: * particular header (prefix). Provide the prefix without colon!
! 92: *
! 93: * Returns a pointer to the first matching header or NULL if none matched.
! 94: */
! 95: char *Curl_checkheaders(const struct connectdata *conn,
! 96: const char *thisheader)
! 97: {
! 98: struct curl_slist *head;
! 99: size_t thislen = strlen(thisheader);
! 100: struct Curl_easy *data = conn->data;
! 101:
! 102: for(head = data->set.headers; head; head = head->next) {
! 103: if(strncasecompare(head->data, thisheader, thislen) &&
! 104: Curl_headersep(head->data[thislen]) )
! 105: return head->data;
! 106: }
! 107:
! 108: return NULL;
! 109: }
! 110: #endif
! 111:
! 112: CURLcode Curl_get_upload_buffer(struct Curl_easy *data)
! 113: {
! 114: if(!data->state.ulbuf) {
! 115: data->state.ulbuf = malloc(data->set.upload_buffer_size);
! 116: if(!data->state.ulbuf)
! 117: return CURLE_OUT_OF_MEMORY;
! 118: }
! 119: return CURLE_OK;
! 120: }
! 121:
! 122: #ifndef CURL_DISABLE_HTTP
! 123: /*
! 124: * This function will be called to loop through the trailers buffer
! 125: * until no more data is available for sending.
! 126: */
! 127: static size_t Curl_trailers_read(char *buffer, size_t size, size_t nitems,
! 128: void *raw)
! 129: {
! 130: struct Curl_easy *data = (struct Curl_easy *)raw;
! 131: Curl_send_buffer *trailers_buf = data->state.trailers_buf;
! 132: size_t bytes_left = trailers_buf->size_used-data->state.trailers_bytes_sent;
! 133: size_t to_copy = (size*nitems < bytes_left) ? size*nitems : bytes_left;
! 134: if(to_copy) {
! 135: memcpy(buffer,
! 136: &trailers_buf->buffer[data->state.trailers_bytes_sent],
! 137: to_copy);
! 138: data->state.trailers_bytes_sent += to_copy;
! 139: }
! 140: return to_copy;
! 141: }
! 142:
! 143: static size_t Curl_trailers_left(void *raw)
! 144: {
! 145: struct Curl_easy *data = (struct Curl_easy *)raw;
! 146: Curl_send_buffer *trailers_buf = data->state.trailers_buf;
! 147: return trailers_buf->size_used - data->state.trailers_bytes_sent;
! 148: }
! 149: #endif
! 150:
! 151: /*
! 152: * This function will call the read callback to fill our buffer with data
! 153: * to upload.
! 154: */
! 155: CURLcode Curl_fillreadbuffer(struct connectdata *conn, size_t bytes,
! 156: size_t *nreadp)
! 157: {
! 158: struct Curl_easy *data = conn->data;
! 159: size_t buffersize = bytes;
! 160: size_t nread;
! 161:
! 162: curl_read_callback readfunc = NULL;
! 163: void *extra_data = NULL;
! 164:
! 165: #ifdef CURL_DOES_CONVERSIONS
! 166: bool sending_http_headers = FALSE;
! 167:
! 168: if(conn->handler->protocol&(PROTO_FAMILY_HTTP|CURLPROTO_RTSP)) {
! 169: const struct HTTP *http = data->req.protop;
! 170:
! 171: if(http->sending == HTTPSEND_REQUEST)
! 172: /* We're sending the HTTP request headers, not the data.
! 173: Remember that so we don't re-translate them into garbage. */
! 174: sending_http_headers = TRUE;
! 175: }
! 176: #endif
! 177:
! 178: #ifndef CURL_DISABLE_HTTP
! 179: if(data->state.trailers_state == TRAILERS_INITIALIZED) {
! 180: struct curl_slist *trailers = NULL;
! 181: CURLcode result;
! 182: int trailers_ret_code;
! 183:
! 184: /* at this point we already verified that the callback exists
! 185: so we compile and store the trailers buffer, then proceed */
! 186: infof(data,
! 187: "Moving trailers state machine from initialized to sending.\n");
! 188: data->state.trailers_state = TRAILERS_SENDING;
! 189: data->state.trailers_buf = Curl_add_buffer_init();
! 190: if(!data->state.trailers_buf) {
! 191: failf(data, "Unable to allocate trailing headers buffer !");
! 192: return CURLE_OUT_OF_MEMORY;
! 193: }
! 194: data->state.trailers_bytes_sent = 0;
! 195: Curl_set_in_callback(data, true);
! 196: trailers_ret_code = data->set.trailer_callback(&trailers,
! 197: data->set.trailer_data);
! 198: Curl_set_in_callback(data, false);
! 199: if(trailers_ret_code == CURL_TRAILERFUNC_OK) {
! 200: result = Curl_http_compile_trailers(trailers, &data->state.trailers_buf,
! 201: data);
! 202: }
! 203: else {
! 204: failf(data, "operation aborted by trailing headers callback");
! 205: *nreadp = 0;
! 206: result = CURLE_ABORTED_BY_CALLBACK;
! 207: }
! 208: if(result) {
! 209: Curl_add_buffer_free(&data->state.trailers_buf);
! 210: curl_slist_free_all(trailers);
! 211: return result;
! 212: }
! 213: infof(data, "Successfully compiled trailers.\r\n");
! 214: curl_slist_free_all(trailers);
! 215: }
! 216: #endif
! 217:
! 218: /* if we are transmitting trailing data, we don't need to write
! 219: a chunk size so we skip this */
! 220: if(data->req.upload_chunky &&
! 221: data->state.trailers_state == TRAILERS_NONE) {
! 222: /* if chunked Transfer-Encoding */
! 223: buffersize -= (8 + 2 + 2); /* 32bit hex + CRLF + CRLF */
! 224: data->req.upload_fromhere += (8 + 2); /* 32bit hex + CRLF */
! 225: }
! 226:
! 227: #ifndef CURL_DISABLE_HTTP
! 228: if(data->state.trailers_state == TRAILERS_SENDING) {
! 229: /* if we're here then that means that we already sent the last empty chunk
! 230: but we didn't send a final CR LF, so we sent 0 CR LF. We then start
! 231: pulling trailing data until we have no more at which point we
! 232: simply return to the previous point in the state machine as if
! 233: nothing happened.
! 234: */
! 235: readfunc = Curl_trailers_read;
! 236: extra_data = (void *)data;
! 237: }
! 238: else
! 239: #endif
! 240: {
! 241: readfunc = data->state.fread_func;
! 242: extra_data = data->state.in;
! 243: }
! 244:
! 245: Curl_set_in_callback(data, true);
! 246: nread = readfunc(data->req.upload_fromhere, 1,
! 247: buffersize, extra_data);
! 248: Curl_set_in_callback(data, false);
! 249:
! 250: if(nread == CURL_READFUNC_ABORT) {
! 251: failf(data, "operation aborted by callback");
! 252: *nreadp = 0;
! 253: return CURLE_ABORTED_BY_CALLBACK;
! 254: }
! 255: if(nread == CURL_READFUNC_PAUSE) {
! 256: struct SingleRequest *k = &data->req;
! 257:
! 258: if(conn->handler->flags & PROTOPT_NONETWORK) {
! 259: /* protocols that work without network cannot be paused. This is
! 260: actually only FILE:// just now, and it can't pause since the transfer
! 261: isn't done using the "normal" procedure. */
! 262: failf(data, "Read callback asked for PAUSE when not supported!");
! 263: return CURLE_READ_ERROR;
! 264: }
! 265:
! 266: /* CURL_READFUNC_PAUSE pauses read callbacks that feed socket writes */
! 267: k->keepon |= KEEP_SEND_PAUSE; /* mark socket send as paused */
! 268: if(data->req.upload_chunky) {
! 269: /* Back out the preallocation done above */
! 270: data->req.upload_fromhere -= (8 + 2);
! 271: }
! 272: *nreadp = 0;
! 273:
! 274: return CURLE_OK; /* nothing was read */
! 275: }
! 276: else if(nread > buffersize) {
! 277: /* the read function returned a too large value */
! 278: *nreadp = 0;
! 279: failf(data, "read function returned funny value");
! 280: return CURLE_READ_ERROR;
! 281: }
! 282:
! 283: if(!data->req.forbidchunk && data->req.upload_chunky) {
! 284: /* if chunked Transfer-Encoding
! 285: * build chunk:
! 286: *
! 287: * <HEX SIZE> CRLF
! 288: * <DATA> CRLF
! 289: */
! 290: /* On non-ASCII platforms the <DATA> may or may not be
! 291: translated based on set.prefer_ascii while the protocol
! 292: portion must always be translated to the network encoding.
! 293: To further complicate matters, line end conversion might be
! 294: done later on, so we need to prevent CRLFs from becoming
! 295: CRCRLFs if that's the case. To do this we use bare LFs
! 296: here, knowing they'll become CRLFs later on.
! 297: */
! 298:
! 299: bool added_crlf = FALSE;
! 300: int hexlen = 0;
! 301: const char *endofline_native;
! 302: const char *endofline_network;
! 303:
! 304: if(
! 305: #ifdef CURL_DO_LINEEND_CONV
! 306: (data->set.prefer_ascii) ||
! 307: #endif
! 308: (data->set.crlf)) {
! 309: /* \n will become \r\n later on */
! 310: endofline_native = "\n";
! 311: endofline_network = "\x0a";
! 312: }
! 313: else {
! 314: endofline_native = "\r\n";
! 315: endofline_network = "\x0d\x0a";
! 316: }
! 317:
! 318: /* if we're not handling trailing data, proceed as usual */
! 319: if(data->state.trailers_state != TRAILERS_SENDING) {
! 320: char hexbuffer[11] = "";
! 321: hexlen = msnprintf(hexbuffer, sizeof(hexbuffer),
! 322: "%zx%s", nread, endofline_native);
! 323:
! 324: /* move buffer pointer */
! 325: data->req.upload_fromhere -= hexlen;
! 326: nread += hexlen;
! 327:
! 328: /* copy the prefix to the buffer, leaving out the NUL */
! 329: memcpy(data->req.upload_fromhere, hexbuffer, hexlen);
! 330:
! 331: /* always append ASCII CRLF to the data unless
! 332: we have a valid trailer callback */
! 333: #ifndef CURL_DISABLE_HTTP
! 334: if((nread-hexlen) == 0 &&
! 335: data->set.trailer_callback != NULL &&
! 336: data->state.trailers_state == TRAILERS_NONE) {
! 337: data->state.trailers_state = TRAILERS_INITIALIZED;
! 338: }
! 339: else
! 340: #endif
! 341: {
! 342: memcpy(data->req.upload_fromhere + nread,
! 343: endofline_network,
! 344: strlen(endofline_network));
! 345: added_crlf = TRUE;
! 346: }
! 347: }
! 348:
! 349: #ifdef CURL_DOES_CONVERSIONS
! 350: {
! 351: CURLcode result;
! 352: size_t length;
! 353: if(data->set.prefer_ascii)
! 354: /* translate the protocol and data */
! 355: length = nread;
! 356: else
! 357: /* just translate the protocol portion */
! 358: length = hexlen;
! 359: if(length) {
! 360: result = Curl_convert_to_network(data, data->req.upload_fromhere,
! 361: length);
! 362: /* Curl_convert_to_network calls failf if unsuccessful */
! 363: if(result)
! 364: return result;
! 365: }
! 366: }
! 367: #endif /* CURL_DOES_CONVERSIONS */
! 368:
! 369: #ifndef CURL_DISABLE_HTTP
! 370: if(data->state.trailers_state == TRAILERS_SENDING &&
! 371: !Curl_trailers_left(data)) {
! 372: Curl_add_buffer_free(&data->state.trailers_buf);
! 373: data->state.trailers_state = TRAILERS_DONE;
! 374: data->set.trailer_data = NULL;
! 375: data->set.trailer_callback = NULL;
! 376: /* mark the transfer as done */
! 377: data->req.upload_done = TRUE;
! 378: infof(data, "Signaling end of chunked upload after trailers.\n");
! 379: }
! 380: else
! 381: #endif
! 382: if((nread - hexlen) == 0 &&
! 383: data->state.trailers_state != TRAILERS_INITIALIZED) {
! 384: /* mark this as done once this chunk is transferred */
! 385: data->req.upload_done = TRUE;
! 386: infof(data,
! 387: "Signaling end of chunked upload via terminating chunk.\n");
! 388: }
! 389:
! 390: if(added_crlf)
! 391: nread += strlen(endofline_network); /* for the added end of line */
! 392: }
! 393: #ifdef CURL_DOES_CONVERSIONS
! 394: else if((data->set.prefer_ascii) && (!sending_http_headers)) {
! 395: CURLcode result;
! 396: result = Curl_convert_to_network(data, data->req.upload_fromhere, nread);
! 397: /* Curl_convert_to_network calls failf if unsuccessful */
! 398: if(result)
! 399: return result;
! 400: }
! 401: #endif /* CURL_DOES_CONVERSIONS */
! 402:
! 403: *nreadp = nread;
! 404:
! 405: return CURLE_OK;
! 406: }
! 407:
! 408:
! 409: /*
! 410: * Curl_readrewind() rewinds the read stream. This is typically used for HTTP
! 411: * POST/PUT with multi-pass authentication when a sending was denied and a
! 412: * resend is necessary.
! 413: */
! 414: CURLcode Curl_readrewind(struct connectdata *conn)
! 415: {
! 416: struct Curl_easy *data = conn->data;
! 417: curl_mimepart *mimepart = &data->set.mimepost;
! 418:
! 419: conn->bits.rewindaftersend = FALSE; /* we rewind now */
! 420:
! 421: /* explicitly switch off sending data on this connection now since we are
! 422: about to restart a new transfer and thus we want to avoid inadvertently
! 423: sending more data on the existing connection until the next transfer
! 424: starts */
! 425: data->req.keepon &= ~KEEP_SEND;
! 426:
! 427: /* We have sent away data. If not using CURLOPT_POSTFIELDS or
! 428: CURLOPT_HTTPPOST, call app to rewind
! 429: */
! 430: if(conn->handler->protocol & PROTO_FAMILY_HTTP) {
! 431: struct HTTP *http = data->req.protop;
! 432:
! 433: if(http->sendit)
! 434: mimepart = http->sendit;
! 435: }
! 436: if(data->set.postfields)
! 437: ; /* do nothing */
! 438: else if(data->set.httpreq == HTTPREQ_POST_MIME ||
! 439: data->set.httpreq == HTTPREQ_POST_FORM) {
! 440: if(Curl_mime_rewind(mimepart)) {
! 441: failf(data, "Cannot rewind mime/post data");
! 442: return CURLE_SEND_FAIL_REWIND;
! 443: }
! 444: }
! 445: else {
! 446: if(data->set.seek_func) {
! 447: int err;
! 448:
! 449: Curl_set_in_callback(data, true);
! 450: err = (data->set.seek_func)(data->set.seek_client, 0, SEEK_SET);
! 451: Curl_set_in_callback(data, false);
! 452: if(err) {
! 453: failf(data, "seek callback returned error %d", (int)err);
! 454: return CURLE_SEND_FAIL_REWIND;
! 455: }
! 456: }
! 457: else if(data->set.ioctl_func) {
! 458: curlioerr err;
! 459:
! 460: Curl_set_in_callback(data, true);
! 461: err = (data->set.ioctl_func)(data, CURLIOCMD_RESTARTREAD,
! 462: data->set.ioctl_client);
! 463: Curl_set_in_callback(data, false);
! 464: infof(data, "the ioctl callback returned %d\n", (int)err);
! 465:
! 466: if(err) {
! 467: failf(data, "ioctl callback returned error %d", (int)err);
! 468: return CURLE_SEND_FAIL_REWIND;
! 469: }
! 470: }
! 471: else {
! 472: /* If no CURLOPT_READFUNCTION is used, we know that we operate on a
! 473: given FILE * stream and we can actually attempt to rewind that
! 474: ourselves with fseek() */
! 475: if(data->state.fread_func == (curl_read_callback)fread) {
! 476: if(-1 != fseek(data->state.in, 0, SEEK_SET))
! 477: /* successful rewind */
! 478: return CURLE_OK;
! 479: }
! 480:
! 481: /* no callback set or failure above, makes us fail at once */
! 482: failf(data, "necessary data rewind wasn't possible");
! 483: return CURLE_SEND_FAIL_REWIND;
! 484: }
! 485: }
! 486: return CURLE_OK;
! 487: }
! 488:
! 489: static int data_pending(const struct Curl_easy *data)
! 490: {
! 491: struct connectdata *conn = data->conn;
! 492: /* in the case of libssh2, we can never be really sure that we have emptied
! 493: its internal buffers so we MUST always try until we get EAGAIN back */
! 494: return conn->handler->protocol&(CURLPROTO_SCP|CURLPROTO_SFTP) ||
! 495: #if defined(USE_NGHTTP2)
! 496: Curl_ssl_data_pending(conn, FIRSTSOCKET) ||
! 497: /* For HTTP/2, we may read up everything including response body
! 498: with header fields in Curl_http_readwrite_headers. If no
! 499: content-length is provided, curl waits for the connection
! 500: close, which we emulate it using conn->proto.httpc.closed =
! 501: TRUE. The thing is if we read everything, then http2_recv won't
! 502: be called and we cannot signal the HTTP/2 stream has closed. As
! 503: a workaround, we return nonzero here to call http2_recv. */
! 504: ((conn->handler->protocol&PROTO_FAMILY_HTTP) && conn->httpversion >= 20);
! 505: #elif defined(ENABLE_QUIC)
! 506: Curl_ssl_data_pending(conn, FIRSTSOCKET) || Curl_quic_data_pending(data);
! 507: #else
! 508: Curl_ssl_data_pending(conn, FIRSTSOCKET);
! 509: #endif
! 510: }
! 511:
! 512: /*
! 513: * Check to see if CURLOPT_TIMECONDITION was met by comparing the time of the
! 514: * remote document with the time provided by CURLOPT_TIMEVAL
! 515: */
! 516: bool Curl_meets_timecondition(struct Curl_easy *data, time_t timeofdoc)
! 517: {
! 518: if((timeofdoc == 0) || (data->set.timevalue == 0))
! 519: return TRUE;
! 520:
! 521: switch(data->set.timecondition) {
! 522: case CURL_TIMECOND_IFMODSINCE:
! 523: default:
! 524: if(timeofdoc <= data->set.timevalue) {
! 525: infof(data,
! 526: "The requested document is not new enough\n");
! 527: data->info.timecond = TRUE;
! 528: return FALSE;
! 529: }
! 530: break;
! 531: case CURL_TIMECOND_IFUNMODSINCE:
! 532: if(timeofdoc >= data->set.timevalue) {
! 533: infof(data,
! 534: "The requested document is not old enough\n");
! 535: data->info.timecond = TRUE;
! 536: return FALSE;
! 537: }
! 538: break;
! 539: }
! 540:
! 541: return TRUE;
! 542: }
! 543:
! 544: /*
! 545: * Go ahead and do a read if we have a readable socket or if
! 546: * the stream was rewound (in which case we have data in a
! 547: * buffer)
! 548: *
! 549: * return '*comeback' TRUE if we didn't properly drain the socket so this
! 550: * function should get called again without select() or similar in between!
! 551: */
! 552: static CURLcode readwrite_data(struct Curl_easy *data,
! 553: struct connectdata *conn,
! 554: struct SingleRequest *k,
! 555: int *didwhat, bool *done,
! 556: bool *comeback)
! 557: {
! 558: CURLcode result = CURLE_OK;
! 559: ssize_t nread; /* number of bytes read */
! 560: size_t excess = 0; /* excess bytes read */
! 561: bool readmore = FALSE; /* used by RTP to signal for more data */
! 562: int maxloops = 100;
! 563:
! 564: *done = FALSE;
! 565: *comeback = FALSE;
! 566:
! 567: /* This is where we loop until we have read everything there is to
! 568: read or we get a CURLE_AGAIN */
! 569: do {
! 570: bool is_empty_data = FALSE;
! 571: size_t buffersize = data->set.buffer_size;
! 572: size_t bytestoread = buffersize;
! 573: #ifdef USE_NGHTTP2
! 574: bool is_http2 = ((conn->handler->protocol & PROTO_FAMILY_HTTP) &&
! 575: (conn->httpversion == 20));
! 576: #endif
! 577:
! 578: if(
! 579: #ifdef USE_NGHTTP2
! 580: /* For HTTP/2, read data without caring about the content
! 581: length. This is safe because body in HTTP/2 is always
! 582: segmented thanks to its framing layer. Meanwhile, we have to
! 583: call Curl_read to ensure that http2_handle_stream_close is
! 584: called when we read all incoming bytes for a particular
! 585: stream. */
! 586: !is_http2 &&
! 587: #endif
! 588: k->size != -1 && !k->header) {
! 589: /* make sure we don't read too much */
! 590: curl_off_t totalleft = k->size - k->bytecount;
! 591: if(totalleft < (curl_off_t)bytestoread)
! 592: bytestoread = (size_t)totalleft;
! 593: }
! 594:
! 595: if(bytestoread) {
! 596: /* receive data from the network! */
! 597: result = Curl_read(conn, conn->sockfd, k->buf, bytestoread, &nread);
! 598:
! 599: /* read would've blocked */
! 600: if(CURLE_AGAIN == result)
! 601: break; /* get out of loop */
! 602:
! 603: if(result>0)
! 604: return result;
! 605: }
! 606: else {
! 607: /* read nothing but since we wanted nothing we consider this an OK
! 608: situation to proceed from */
! 609: DEBUGF(infof(data, "readwrite_data: we're done!\n"));
! 610: nread = 0;
! 611: }
! 612:
! 613: if(!k->bytecount) {
! 614: Curl_pgrsTime(data, TIMER_STARTTRANSFER);
! 615: if(k->exp100 > EXP100_SEND_DATA)
! 616: /* set time stamp to compare with when waiting for the 100 */
! 617: k->start100 = Curl_now();
! 618: }
! 619:
! 620: *didwhat |= KEEP_RECV;
! 621: /* indicates data of zero size, i.e. empty file */
! 622: is_empty_data = ((nread == 0) && (k->bodywrites == 0)) ? TRUE : FALSE;
! 623:
! 624: /* NUL terminate, allowing string ops to be used */
! 625: if(0 < nread || is_empty_data) {
! 626: k->buf[nread] = 0;
! 627: }
! 628: else {
! 629: /* if we receive 0 or less here, either the http2 stream is closed or the
! 630: server closed the connection and we bail out from this! */
! 631: #ifdef USE_NGHTTP2
! 632: if(is_http2 && !nread)
! 633: DEBUGF(infof(data, "nread == 0, stream closed, bailing\n"));
! 634: else
! 635: #endif
! 636: DEBUGF(infof(data, "nread <= 0, server closed connection, bailing\n"));
! 637: k->keepon &= ~KEEP_RECV;
! 638: break;
! 639: }
! 640:
! 641: /* Default buffer to use when we write the buffer, it may be changed
! 642: in the flow below before the actual storing is done. */
! 643: k->str = k->buf;
! 644:
! 645: if(conn->handler->readwrite) {
! 646: result = conn->handler->readwrite(data, conn, &nread, &readmore);
! 647: if(result)
! 648: return result;
! 649: if(readmore)
! 650: break;
! 651: }
! 652:
! 653: #ifndef CURL_DISABLE_HTTP
! 654: /* Since this is a two-state thing, we check if we are parsing
! 655: headers at the moment or not. */
! 656: if(k->header) {
! 657: /* we are in parse-the-header-mode */
! 658: bool stop_reading = FALSE;
! 659: result = Curl_http_readwrite_headers(data, conn, &nread, &stop_reading);
! 660: if(result)
! 661: return result;
! 662:
! 663: if(conn->handler->readwrite &&
! 664: (k->maxdownload <= 0 && nread > 0)) {
! 665: result = conn->handler->readwrite(data, conn, &nread, &readmore);
! 666: if(result)
! 667: return result;
! 668: if(readmore)
! 669: break;
! 670: }
! 671:
! 672: if(stop_reading) {
! 673: /* We've stopped dealing with input, get out of the do-while loop */
! 674:
! 675: if(nread > 0) {
! 676: infof(data,
! 677: "Excess found:"
! 678: " excess = %zd"
! 679: " url = %s (zero-length body)\n",
! 680: nread, data->state.up.path);
! 681: }
! 682:
! 683: break;
! 684: }
! 685: }
! 686: #endif /* CURL_DISABLE_HTTP */
! 687:
! 688:
! 689: /* This is not an 'else if' since it may be a rest from the header
! 690: parsing, where the beginning of the buffer is headers and the end
! 691: is non-headers. */
! 692: if(k->str && !k->header && (nread > 0 || is_empty_data)) {
! 693:
! 694: if(data->set.opt_no_body) {
! 695: /* data arrives although we want none, bail out */
! 696: streamclose(conn, "ignoring body");
! 697: *done = TRUE;
! 698: return CURLE_WEIRD_SERVER_REPLY;
! 699: }
! 700:
! 701: #ifndef CURL_DISABLE_HTTP
! 702: if(0 == k->bodywrites && !is_empty_data) {
! 703: /* These checks are only made the first time we are about to
! 704: write a piece of the body */
! 705: if(conn->handler->protocol&(PROTO_FAMILY_HTTP|CURLPROTO_RTSP)) {
! 706: /* HTTP-only checks */
! 707:
! 708: if(data->req.newurl) {
! 709: if(conn->bits.close) {
! 710: /* Abort after the headers if "follow Location" is set
! 711: and we're set to close anyway. */
! 712: k->keepon &= ~KEEP_RECV;
! 713: *done = TRUE;
! 714: return CURLE_OK;
! 715: }
! 716: /* We have a new url to load, but since we want to be able
! 717: to re-use this connection properly, we read the full
! 718: response in "ignore more" */
! 719: k->ignorebody = TRUE;
! 720: infof(data, "Ignoring the response-body\n");
! 721: }
! 722: if(data->state.resume_from && !k->content_range &&
! 723: (data->set.httpreq == HTTPREQ_GET) &&
! 724: !k->ignorebody) {
! 725:
! 726: if(k->size == data->state.resume_from) {
! 727: /* The resume point is at the end of file, consider this fine
! 728: even if it doesn't allow resume from here. */
! 729: infof(data, "The entire document is already downloaded");
! 730: connclose(conn, "already downloaded");
! 731: /* Abort download */
! 732: k->keepon &= ~KEEP_RECV;
! 733: *done = TRUE;
! 734: return CURLE_OK;
! 735: }
! 736:
! 737: /* we wanted to resume a download, although the server doesn't
! 738: * seem to support this and we did this with a GET (if it
! 739: * wasn't a GET we did a POST or PUT resume) */
! 740: failf(data, "HTTP server doesn't seem to support "
! 741: "byte ranges. Cannot resume.");
! 742: return CURLE_RANGE_ERROR;
! 743: }
! 744:
! 745: if(data->set.timecondition && !data->state.range) {
! 746: /* A time condition has been set AND no ranges have been
! 747: requested. This seems to be what chapter 13.3.4 of
! 748: RFC 2616 defines to be the correct action for a
! 749: HTTP/1.1 client */
! 750:
! 751: if(!Curl_meets_timecondition(data, k->timeofdoc)) {
! 752: *done = TRUE;
! 753: /* We're simulating a http 304 from server so we return
! 754: what should have been returned from the server */
! 755: data->info.httpcode = 304;
! 756: infof(data, "Simulate a HTTP 304 response!\n");
! 757: /* we abort the transfer before it is completed == we ruin the
! 758: re-use ability. Close the connection */
! 759: connclose(conn, "Simulated 304 handling");
! 760: return CURLE_OK;
! 761: }
! 762: } /* we have a time condition */
! 763:
! 764: } /* this is HTTP or RTSP */
! 765: } /* this is the first time we write a body part */
! 766: #endif /* CURL_DISABLE_HTTP */
! 767:
! 768: k->bodywrites++;
! 769:
! 770: /* pass data to the debug function before it gets "dechunked" */
! 771: if(data->set.verbose) {
! 772: if(k->badheader) {
! 773: Curl_debug(data, CURLINFO_DATA_IN, data->state.headerbuff,
! 774: (size_t)k->hbuflen);
! 775: if(k->badheader == HEADER_PARTHEADER)
! 776: Curl_debug(data, CURLINFO_DATA_IN,
! 777: k->str, (size_t)nread);
! 778: }
! 779: else
! 780: Curl_debug(data, CURLINFO_DATA_IN,
! 781: k->str, (size_t)nread);
! 782: }
! 783:
! 784: #ifndef CURL_DISABLE_HTTP
! 785: if(k->chunk) {
! 786: /*
! 787: * Here comes a chunked transfer flying and we need to decode this
! 788: * properly. While the name says read, this function both reads
! 789: * and writes away the data. The returned 'nread' holds the number
! 790: * of actual data it wrote to the client.
! 791: */
! 792: CURLcode extra;
! 793: CHUNKcode res =
! 794: Curl_httpchunk_read(conn, k->str, nread, &nread, &extra);
! 795:
! 796: if(CHUNKE_OK < res) {
! 797: if(CHUNKE_PASSTHRU_ERROR == res) {
! 798: failf(data, "Failed reading the chunked-encoded stream");
! 799: return extra;
! 800: }
! 801: failf(data, "%s in chunked-encoding", Curl_chunked_strerror(res));
! 802: return CURLE_RECV_ERROR;
! 803: }
! 804: if(CHUNKE_STOP == res) {
! 805: size_t dataleft;
! 806: /* we're done reading chunks! */
! 807: k->keepon &= ~KEEP_RECV; /* read no more */
! 808:
! 809: /* There are now possibly N number of bytes at the end of the
! 810: str buffer that weren't written to the client.
! 811: Push it back to be read on the next pass. */
! 812:
! 813: dataleft = conn->chunk.dataleft;
! 814: if(dataleft != 0) {
! 815: infof(conn->data, "Leftovers after chunking: %zu bytes\n",
! 816: dataleft);
! 817: }
! 818: }
! 819: /* If it returned OK, we just keep going */
! 820: }
! 821: #endif /* CURL_DISABLE_HTTP */
! 822:
! 823: /* Account for body content stored in the header buffer */
! 824: if((k->badheader == HEADER_PARTHEADER) && !k->ignorebody) {
! 825: DEBUGF(infof(data, "Increasing bytecount by %zu from hbuflen\n",
! 826: k->hbuflen));
! 827: k->bytecount += k->hbuflen;
! 828: }
! 829:
! 830: if((-1 != k->maxdownload) &&
! 831: (k->bytecount + nread >= k->maxdownload)) {
! 832:
! 833: excess = (size_t)(k->bytecount + nread - k->maxdownload);
! 834: if(excess > 0 && !k->ignorebody) {
! 835: infof(data,
! 836: "Excess found in a read:"
! 837: " excess = %zu"
! 838: ", size = %" CURL_FORMAT_CURL_OFF_T
! 839: ", maxdownload = %" CURL_FORMAT_CURL_OFF_T
! 840: ", bytecount = %" CURL_FORMAT_CURL_OFF_T "\n",
! 841: excess, k->size, k->maxdownload, k->bytecount);
! 842: }
! 843:
! 844: nread = (ssize_t) (k->maxdownload - k->bytecount);
! 845: if(nread < 0) /* this should be unusual */
! 846: nread = 0;
! 847:
! 848: k->keepon &= ~KEEP_RECV; /* we're done reading */
! 849: }
! 850:
! 851: k->bytecount += nread;
! 852:
! 853: Curl_pgrsSetDownloadCounter(data, k->bytecount);
! 854:
! 855: if(!k->chunk && (nread || k->badheader || is_empty_data)) {
! 856: /* If this is chunky transfer, it was already written */
! 857:
! 858: if(k->badheader && !k->ignorebody) {
! 859: /* we parsed a piece of data wrongly assuming it was a header
! 860: and now we output it as body instead */
! 861:
! 862: /* Don't let excess data pollute body writes */
! 863: if(k->maxdownload == -1 || (curl_off_t)k->hbuflen <= k->maxdownload)
! 864: result = Curl_client_write(conn, CLIENTWRITE_BODY,
! 865: data->state.headerbuff,
! 866: k->hbuflen);
! 867: else
! 868: result = Curl_client_write(conn, CLIENTWRITE_BODY,
! 869: data->state.headerbuff,
! 870: (size_t)k->maxdownload);
! 871:
! 872: if(result)
! 873: return result;
! 874: }
! 875: if(k->badheader < HEADER_ALLBAD) {
! 876: /* This switch handles various content encodings. If there's an
! 877: error here, be sure to check over the almost identical code
! 878: in http_chunks.c.
! 879: Make sure that ALL_CONTENT_ENCODINGS contains all the
! 880: encodings handled here. */
! 881: if(conn->data->set.http_ce_skip || !k->writer_stack) {
! 882: if(!k->ignorebody) {
! 883: #ifndef CURL_DISABLE_POP3
! 884: if(conn->handler->protocol & PROTO_FAMILY_POP3)
! 885: result = Curl_pop3_write(conn, k->str, nread);
! 886: else
! 887: #endif /* CURL_DISABLE_POP3 */
! 888: result = Curl_client_write(conn, CLIENTWRITE_BODY, k->str,
! 889: nread);
! 890: }
! 891: }
! 892: else if(!k->ignorebody)
! 893: result = Curl_unencode_write(conn, k->writer_stack, k->str, nread);
! 894: }
! 895: k->badheader = HEADER_NORMAL; /* taken care of now */
! 896:
! 897: if(result)
! 898: return result;
! 899: }
! 900:
! 901: } /* if(!header and data to read) */
! 902:
! 903: if(conn->handler->readwrite && excess) {
! 904: /* Parse the excess data */
! 905: k->str += nread;
! 906:
! 907: if(&k->str[excess] > &k->buf[data->set.buffer_size]) {
! 908: /* the excess amount was too excessive(!), make sure
! 909: it doesn't read out of buffer */
! 910: excess = &k->buf[data->set.buffer_size] - k->str;
! 911: }
! 912: nread = (ssize_t)excess;
! 913:
! 914: result = conn->handler->readwrite(data, conn, &nread, &readmore);
! 915: if(result)
! 916: return result;
! 917:
! 918: if(readmore)
! 919: k->keepon |= KEEP_RECV; /* we're not done reading */
! 920: break;
! 921: }
! 922:
! 923: if(is_empty_data) {
! 924: /* if we received nothing, the server closed the connection and we
! 925: are done */
! 926: k->keepon &= ~KEEP_RECV;
! 927: }
! 928:
! 929: if(k->keepon & KEEP_RECV_PAUSE) {
! 930: /* this is a paused transfer */
! 931: break;
! 932: }
! 933:
! 934: } while(data_pending(data) && maxloops--);
! 935:
! 936: if(maxloops <= 0) {
! 937: /* we mark it as read-again-please */
! 938: conn->cselect_bits = CURL_CSELECT_IN;
! 939: *comeback = TRUE;
! 940: }
! 941:
! 942: if(((k->keepon & (KEEP_RECV|KEEP_SEND)) == KEEP_SEND) &&
! 943: conn->bits.close) {
! 944: /* When we've read the entire thing and the close bit is set, the server
! 945: may now close the connection. If there's now any kind of sending going
! 946: on from our side, we need to stop that immediately. */
! 947: infof(data, "we are done reading and this is set to close, stop send\n");
! 948: k->keepon &= ~KEEP_SEND; /* no writing anymore either */
! 949: }
! 950:
! 951: return CURLE_OK;
! 952: }
! 953:
! 954: CURLcode Curl_done_sending(struct connectdata *conn,
! 955: struct SingleRequest *k)
! 956: {
! 957: k->keepon &= ~KEEP_SEND; /* we're done writing */
! 958:
! 959: /* These functions should be moved into the handler struct! */
! 960: Curl_http2_done_sending(conn);
! 961: Curl_quic_done_sending(conn);
! 962:
! 963: if(conn->bits.rewindaftersend) {
! 964: CURLcode result = Curl_readrewind(conn);
! 965: if(result)
! 966: return result;
! 967: }
! 968: return CURLE_OK;
! 969: }
! 970:
! 971: #if defined(WIN32) && !defined(USE_LWIPSOCK)
! 972: #ifndef SIO_IDEAL_SEND_BACKLOG_QUERY
! 973: #define SIO_IDEAL_SEND_BACKLOG_QUERY 0x4004747B
! 974: #endif
! 975:
! 976: static void win_update_buffer_size(curl_socket_t sockfd)
! 977: {
! 978: int result;
! 979: ULONG ideal;
! 980: DWORD ideallen;
! 981: result = WSAIoctl(sockfd, SIO_IDEAL_SEND_BACKLOG_QUERY, 0, 0,
! 982: &ideal, sizeof(ideal), &ideallen, 0, 0);
! 983: if(result == 0) {
! 984: setsockopt(sockfd, SOL_SOCKET, SO_SNDBUF,
! 985: (const char *)&ideal, sizeof(ideal));
! 986: }
! 987: }
! 988: #else
! 989: #define win_update_buffer_size(x)
! 990: #endif
! 991:
! 992: /*
! 993: * Send data to upload to the server, when the socket is writable.
! 994: */
! 995: static CURLcode readwrite_upload(struct Curl_easy *data,
! 996: struct connectdata *conn,
! 997: int *didwhat)
! 998: {
! 999: ssize_t i, si;
! 1000: ssize_t bytes_written;
! 1001: CURLcode result;
! 1002: ssize_t nread; /* number of bytes read */
! 1003: bool sending_http_headers = FALSE;
! 1004: struct SingleRequest *k = &data->req;
! 1005:
! 1006: if((k->bytecount == 0) && (k->writebytecount == 0))
! 1007: Curl_pgrsTime(data, TIMER_STARTTRANSFER);
! 1008:
! 1009: *didwhat |= KEEP_SEND;
! 1010:
! 1011: do {
! 1012: /* only read more data if there's no upload data already
! 1013: present in the upload buffer */
! 1014: if(0 == k->upload_present) {
! 1015: result = Curl_get_upload_buffer(data);
! 1016: if(result)
! 1017: return result;
! 1018: /* init the "upload from here" pointer */
! 1019: k->upload_fromhere = data->state.ulbuf;
! 1020:
! 1021: if(!k->upload_done) {
! 1022: /* HTTP pollution, this should be written nicer to become more
! 1023: protocol agnostic. */
! 1024: size_t fillcount;
! 1025: struct HTTP *http = k->protop;
! 1026:
! 1027: if((k->exp100 == EXP100_SENDING_REQUEST) &&
! 1028: (http->sending == HTTPSEND_BODY)) {
! 1029: /* If this call is to send body data, we must take some action:
! 1030: We have sent off the full HTTP 1.1 request, and we shall now
! 1031: go into the Expect: 100 state and await such a header */
! 1032: k->exp100 = EXP100_AWAITING_CONTINUE; /* wait for the header */
! 1033: k->keepon &= ~KEEP_SEND; /* disable writing */
! 1034: k->start100 = Curl_now(); /* timeout count starts now */
! 1035: *didwhat &= ~KEEP_SEND; /* we didn't write anything actually */
! 1036: /* set a timeout for the multi interface */
! 1037: Curl_expire(data, data->set.expect_100_timeout, EXPIRE_100_TIMEOUT);
! 1038: break;
! 1039: }
! 1040:
! 1041: if(conn->handler->protocol&(PROTO_FAMILY_HTTP|CURLPROTO_RTSP)) {
! 1042: if(http->sending == HTTPSEND_REQUEST)
! 1043: /* We're sending the HTTP request headers, not the data.
! 1044: Remember that so we don't change the line endings. */
! 1045: sending_http_headers = TRUE;
! 1046: else
! 1047: sending_http_headers = FALSE;
! 1048: }
! 1049:
! 1050: result = Curl_fillreadbuffer(conn, data->set.upload_buffer_size,
! 1051: &fillcount);
! 1052: if(result)
! 1053: return result;
! 1054:
! 1055: nread = fillcount;
! 1056: }
! 1057: else
! 1058: nread = 0; /* we're done uploading/reading */
! 1059:
! 1060: if(!nread && (k->keepon & KEEP_SEND_PAUSE)) {
! 1061: /* this is a paused transfer */
! 1062: break;
! 1063: }
! 1064: if(nread <= 0) {
! 1065: result = Curl_done_sending(conn, k);
! 1066: if(result)
! 1067: return result;
! 1068: break;
! 1069: }
! 1070:
! 1071: /* store number of bytes available for upload */
! 1072: k->upload_present = nread;
! 1073:
! 1074: /* convert LF to CRLF if so asked */
! 1075: if((!sending_http_headers) && (
! 1076: #ifdef CURL_DO_LINEEND_CONV
! 1077: /* always convert if we're FTPing in ASCII mode */
! 1078: (data->set.prefer_ascii) ||
! 1079: #endif
! 1080: (data->set.crlf))) {
! 1081: /* Do we need to allocate a scratch buffer? */
! 1082: if(!data->state.scratch) {
! 1083: data->state.scratch = malloc(2 * data->set.upload_buffer_size);
! 1084: if(!data->state.scratch) {
! 1085: failf(data, "Failed to alloc scratch buffer!");
! 1086:
! 1087: return CURLE_OUT_OF_MEMORY;
! 1088: }
! 1089: }
! 1090:
! 1091: /*
! 1092: * ASCII/EBCDIC Note: This is presumably a text (not binary)
! 1093: * transfer so the data should already be in ASCII.
! 1094: * That means the hex values for ASCII CR (0x0d) & LF (0x0a)
! 1095: * must be used instead of the escape sequences \r & \n.
! 1096: */
! 1097: for(i = 0, si = 0; i < nread; i++, si++) {
! 1098: if(k->upload_fromhere[i] == 0x0a) {
! 1099: data->state.scratch[si++] = 0x0d;
! 1100: data->state.scratch[si] = 0x0a;
! 1101: if(!data->set.crlf) {
! 1102: /* we're here only because FTP is in ASCII mode...
! 1103: bump infilesize for the LF we just added */
! 1104: if(data->state.infilesize != -1)
! 1105: data->state.infilesize++;
! 1106: }
! 1107: }
! 1108: else
! 1109: data->state.scratch[si] = k->upload_fromhere[i];
! 1110: }
! 1111:
! 1112: if(si != nread) {
! 1113: /* only perform the special operation if we really did replace
! 1114: anything */
! 1115: nread = si;
! 1116:
! 1117: /* upload from the new (replaced) buffer instead */
! 1118: k->upload_fromhere = data->state.scratch;
! 1119:
! 1120: /* set the new amount too */
! 1121: k->upload_present = nread;
! 1122: }
! 1123: }
! 1124:
! 1125: #ifndef CURL_DISABLE_SMTP
! 1126: if(conn->handler->protocol & PROTO_FAMILY_SMTP) {
! 1127: result = Curl_smtp_escape_eob(conn, nread);
! 1128: if(result)
! 1129: return result;
! 1130: }
! 1131: #endif /* CURL_DISABLE_SMTP */
! 1132: } /* if 0 == k->upload_present */
! 1133: else {
! 1134: /* We have a partial buffer left from a previous "round". Use
! 1135: that instead of reading more data */
! 1136: }
! 1137:
! 1138: /* write to socket (send away data) */
! 1139: result = Curl_write(conn,
! 1140: conn->writesockfd, /* socket to send to */
! 1141: k->upload_fromhere, /* buffer pointer */
! 1142: k->upload_present, /* buffer size */
! 1143: &bytes_written); /* actually sent */
! 1144: if(result)
! 1145: return result;
! 1146:
! 1147: win_update_buffer_size(conn->writesockfd);
! 1148:
! 1149: if(data->set.verbose)
! 1150: /* show the data before we change the pointer upload_fromhere */
! 1151: Curl_debug(data, CURLINFO_DATA_OUT, k->upload_fromhere,
! 1152: (size_t)bytes_written);
! 1153:
! 1154: k->writebytecount += bytes_written;
! 1155: Curl_pgrsSetUploadCounter(data, k->writebytecount);
! 1156:
! 1157: if((!k->upload_chunky || k->forbidchunk) &&
! 1158: (k->writebytecount == data->state.infilesize)) {
! 1159: /* we have sent all data we were supposed to */
! 1160: k->upload_done = TRUE;
! 1161: infof(data, "We are completely uploaded and fine\n");
! 1162: }
! 1163:
! 1164: if(k->upload_present != bytes_written) {
! 1165: /* we only wrote a part of the buffer (if anything), deal with it! */
! 1166:
! 1167: /* store the amount of bytes left in the buffer to write */
! 1168: k->upload_present -= bytes_written;
! 1169:
! 1170: /* advance the pointer where to find the buffer when the next send
! 1171: is to happen */
! 1172: k->upload_fromhere += bytes_written;
! 1173: }
! 1174: else {
! 1175: /* we've uploaded that buffer now */
! 1176: result = Curl_get_upload_buffer(data);
! 1177: if(result)
! 1178: return result;
! 1179: k->upload_fromhere = data->state.ulbuf;
! 1180: k->upload_present = 0; /* no more bytes left */
! 1181:
! 1182: if(k->upload_done) {
! 1183: result = Curl_done_sending(conn, k);
! 1184: if(result)
! 1185: return result;
! 1186: }
! 1187: }
! 1188:
! 1189:
! 1190: } while(0); /* just to break out from! */
! 1191:
! 1192: return CURLE_OK;
! 1193: }
! 1194:
! 1195: /*
! 1196: * Curl_readwrite() is the low-level function to be called when data is to
! 1197: * be read and written to/from the connection.
! 1198: *
! 1199: * return '*comeback' TRUE if we didn't properly drain the socket so this
! 1200: * function should get called again without select() or similar in between!
! 1201: */
! 1202: CURLcode Curl_readwrite(struct connectdata *conn,
! 1203: struct Curl_easy *data,
! 1204: bool *done,
! 1205: bool *comeback)
! 1206: {
! 1207: struct SingleRequest *k = &data->req;
! 1208: CURLcode result;
! 1209: int didwhat = 0;
! 1210:
! 1211: curl_socket_t fd_read;
! 1212: curl_socket_t fd_write;
! 1213: int select_res = conn->cselect_bits;
! 1214:
! 1215: conn->cselect_bits = 0;
! 1216:
! 1217: /* only use the proper socket if the *_HOLD bit is not set simultaneously as
! 1218: then we are in rate limiting state in that transfer direction */
! 1219:
! 1220: if((k->keepon & KEEP_RECVBITS) == KEEP_RECV)
! 1221: fd_read = conn->sockfd;
! 1222: else
! 1223: fd_read = CURL_SOCKET_BAD;
! 1224:
! 1225: if((k->keepon & KEEP_SENDBITS) == KEEP_SEND)
! 1226: fd_write = conn->writesockfd;
! 1227: else
! 1228: fd_write = CURL_SOCKET_BAD;
! 1229:
! 1230: if(conn->data->state.drain) {
! 1231: select_res |= CURL_CSELECT_IN;
! 1232: DEBUGF(infof(data, "Curl_readwrite: forcibly told to drain data\n"));
! 1233: }
! 1234:
! 1235: if(!select_res) /* Call for select()/poll() only, if read/write/error
! 1236: status is not known. */
! 1237: select_res = Curl_socket_check(fd_read, CURL_SOCKET_BAD, fd_write, 0);
! 1238:
! 1239: if(select_res == CURL_CSELECT_ERR) {
! 1240: failf(data, "select/poll returned error");
! 1241: return CURLE_SEND_ERROR;
! 1242: }
! 1243:
! 1244: /* We go ahead and do a read if we have a readable socket or if
! 1245: the stream was rewound (in which case we have data in a
! 1246: buffer) */
! 1247: if((k->keepon & KEEP_RECV) && (select_res & CURL_CSELECT_IN)) {
! 1248: result = readwrite_data(data, conn, k, &didwhat, done, comeback);
! 1249: if(result || *done)
! 1250: return result;
! 1251: }
! 1252:
! 1253: /* If we still have writing to do, we check if we have a writable socket. */
! 1254: if((k->keepon & KEEP_SEND) && (select_res & CURL_CSELECT_OUT)) {
! 1255: /* write */
! 1256:
! 1257: result = readwrite_upload(data, conn, &didwhat);
! 1258: if(result)
! 1259: return result;
! 1260: }
! 1261:
! 1262: k->now = Curl_now();
! 1263: if(didwhat) {
! 1264: ;
! 1265: }
! 1266: else {
! 1267: /* no read no write, this is a timeout? */
! 1268: if(k->exp100 == EXP100_AWAITING_CONTINUE) {
! 1269: /* This should allow some time for the header to arrive, but only a
! 1270: very short time as otherwise it'll be too much wasted time too
! 1271: often. */
! 1272:
! 1273: /* Quoting RFC2616, section "8.2.3 Use of the 100 (Continue) Status":
! 1274:
! 1275: Therefore, when a client sends this header field to an origin server
! 1276: (possibly via a proxy) from which it has never seen a 100 (Continue)
! 1277: status, the client SHOULD NOT wait for an indefinite period before
! 1278: sending the request body.
! 1279:
! 1280: */
! 1281:
! 1282: timediff_t ms = Curl_timediff(k->now, k->start100);
! 1283: if(ms >= data->set.expect_100_timeout) {
! 1284: /* we've waited long enough, continue anyway */
! 1285: k->exp100 = EXP100_SEND_DATA;
! 1286: k->keepon |= KEEP_SEND;
! 1287: Curl_expire_done(data, EXPIRE_100_TIMEOUT);
! 1288: infof(data, "Done waiting for 100-continue\n");
! 1289: }
! 1290: }
! 1291: }
! 1292:
! 1293: if(Curl_pgrsUpdate(conn))
! 1294: result = CURLE_ABORTED_BY_CALLBACK;
! 1295: else
! 1296: result = Curl_speedcheck(data, k->now);
! 1297: if(result)
! 1298: return result;
! 1299:
! 1300: if(k->keepon) {
! 1301: if(0 > Curl_timeleft(data, &k->now, FALSE)) {
! 1302: if(k->size != -1) {
! 1303: failf(data, "Operation timed out after %" CURL_FORMAT_TIMEDIFF_T
! 1304: " milliseconds with %" CURL_FORMAT_CURL_OFF_T " out of %"
! 1305: CURL_FORMAT_CURL_OFF_T " bytes received",
! 1306: Curl_timediff(k->now, data->progress.t_startsingle),
! 1307: k->bytecount, k->size);
! 1308: }
! 1309: else {
! 1310: failf(data, "Operation timed out after %" CURL_FORMAT_TIMEDIFF_T
! 1311: " milliseconds with %" CURL_FORMAT_CURL_OFF_T " bytes received",
! 1312: Curl_timediff(k->now, data->progress.t_startsingle),
! 1313: k->bytecount);
! 1314: }
! 1315: return CURLE_OPERATION_TIMEDOUT;
! 1316: }
! 1317: }
! 1318: else {
! 1319: /*
! 1320: * The transfer has been performed. Just make some general checks before
! 1321: * returning.
! 1322: */
! 1323:
! 1324: if(!(data->set.opt_no_body) && (k->size != -1) &&
! 1325: (k->bytecount != k->size) &&
! 1326: #ifdef CURL_DO_LINEEND_CONV
! 1327: /* Most FTP servers don't adjust their file SIZE response for CRLFs,
! 1328: so we'll check to see if the discrepancy can be explained
! 1329: by the number of CRLFs we've changed to LFs.
! 1330: */
! 1331: (k->bytecount != (k->size + data->state.crlf_conversions)) &&
! 1332: #endif /* CURL_DO_LINEEND_CONV */
! 1333: !k->newurl) {
! 1334: failf(data, "transfer closed with %" CURL_FORMAT_CURL_OFF_T
! 1335: " bytes remaining to read", k->size - k->bytecount);
! 1336: return CURLE_PARTIAL_FILE;
! 1337: }
! 1338: if(!(data->set.opt_no_body) && k->chunk &&
! 1339: (conn->chunk.state != CHUNK_STOP)) {
! 1340: /*
! 1341: * In chunked mode, return an error if the connection is closed prior to
! 1342: * the empty (terminating) chunk is read.
! 1343: *
! 1344: * The condition above used to check for
! 1345: * conn->proto.http->chunk.datasize != 0 which is true after reading
! 1346: * *any* chunk, not just the empty chunk.
! 1347: *
! 1348: */
! 1349: failf(data, "transfer closed with outstanding read data remaining");
! 1350: return CURLE_PARTIAL_FILE;
! 1351: }
! 1352: if(Curl_pgrsUpdate(conn))
! 1353: return CURLE_ABORTED_BY_CALLBACK;
! 1354: }
! 1355:
! 1356: /* Now update the "done" boolean we return */
! 1357: *done = (0 == (k->keepon&(KEEP_RECV|KEEP_SEND|
! 1358: KEEP_RECV_PAUSE|KEEP_SEND_PAUSE))) ? TRUE : FALSE;
! 1359:
! 1360: return CURLE_OK;
! 1361: }
! 1362:
! 1363: /*
! 1364: * Curl_single_getsock() gets called by the multi interface code when the app
! 1365: * has requested to get the sockets for the current connection. This function
! 1366: * will then be called once for every connection that the multi interface
! 1367: * keeps track of. This function will only be called for connections that are
! 1368: * in the proper state to have this information available.
! 1369: */
! 1370: int Curl_single_getsock(const struct connectdata *conn,
! 1371: curl_socket_t *sock)
! 1372: {
! 1373: const struct Curl_easy *data = conn->data;
! 1374: int bitmap = GETSOCK_BLANK;
! 1375: unsigned sockindex = 0;
! 1376:
! 1377: if(conn->handler->perform_getsock)
! 1378: return conn->handler->perform_getsock(conn, sock);
! 1379:
! 1380: /* don't include HOLD and PAUSE connections */
! 1381: if((data->req.keepon & KEEP_RECVBITS) == KEEP_RECV) {
! 1382:
! 1383: DEBUGASSERT(conn->sockfd != CURL_SOCKET_BAD);
! 1384:
! 1385: bitmap |= GETSOCK_READSOCK(sockindex);
! 1386: sock[sockindex] = conn->sockfd;
! 1387: }
! 1388:
! 1389: /* don't include HOLD and PAUSE connections */
! 1390: if((data->req.keepon & KEEP_SENDBITS) == KEEP_SEND) {
! 1391:
! 1392: if((conn->sockfd != conn->writesockfd) ||
! 1393: bitmap == GETSOCK_BLANK) {
! 1394: /* only if they are not the same socket and we have a readable
! 1395: one, we increase index */
! 1396: if(bitmap != GETSOCK_BLANK)
! 1397: sockindex++; /* increase index if we need two entries */
! 1398:
! 1399: DEBUGASSERT(conn->writesockfd != CURL_SOCKET_BAD);
! 1400:
! 1401: sock[sockindex] = conn->writesockfd;
! 1402: }
! 1403:
! 1404: bitmap |= GETSOCK_WRITESOCK(sockindex);
! 1405: }
! 1406:
! 1407: return bitmap;
! 1408: }
! 1409:
! 1410: /* Curl_init_CONNECT() gets called each time the handle switches to CONNECT
! 1411: which means this gets called once for each subsequent redirect etc */
! 1412: void Curl_init_CONNECT(struct Curl_easy *data)
! 1413: {
! 1414: data->state.fread_func = data->set.fread_func_set;
! 1415: data->state.in = data->set.in_set;
! 1416: }
! 1417:
! 1418: /*
! 1419: * Curl_pretransfer() is called immediately before a transfer starts, and only
! 1420: * once for one transfer no matter if it has redirects or do multi-pass
! 1421: * authentication etc.
! 1422: */
! 1423: CURLcode Curl_pretransfer(struct Curl_easy *data)
! 1424: {
! 1425: CURLcode result;
! 1426:
! 1427: if(!data->change.url && !data->set.uh) {
! 1428: /* we can't do anything without URL */
! 1429: failf(data, "No URL set!");
! 1430: return CURLE_URL_MALFORMAT;
! 1431: }
! 1432:
! 1433: /* since the URL may have been redirected in a previous use of this handle */
! 1434: if(data->change.url_alloc) {
! 1435: /* the already set URL is allocated, free it first! */
! 1436: Curl_safefree(data->change.url);
! 1437: data->change.url_alloc = FALSE;
! 1438: }
! 1439:
! 1440: if(!data->change.url && data->set.uh) {
! 1441: CURLUcode uc;
! 1442: uc = curl_url_get(data->set.uh,
! 1443: CURLUPART_URL, &data->set.str[STRING_SET_URL], 0);
! 1444: if(uc) {
! 1445: failf(data, "No URL set!");
! 1446: return CURLE_URL_MALFORMAT;
! 1447: }
! 1448: }
! 1449:
! 1450: data->change.url = data->set.str[STRING_SET_URL];
! 1451:
! 1452: /* Init the SSL session ID cache here. We do it here since we want to do it
! 1453: after the *_setopt() calls (that could specify the size of the cache) but
! 1454: before any transfer takes place. */
! 1455: result = Curl_ssl_initsessions(data, data->set.general_ssl.max_ssl_sessions);
! 1456: if(result)
! 1457: return result;
! 1458:
! 1459: data->state.wildcardmatch = data->set.wildcard_enabled;
! 1460: data->set.followlocation = 0; /* reset the location-follow counter */
! 1461: data->state.this_is_a_follow = FALSE; /* reset this */
! 1462: data->state.errorbuf = FALSE; /* no error has occurred */
! 1463: data->state.httpversion = 0; /* don't assume any particular server version */
! 1464:
! 1465: data->state.authproblem = FALSE;
! 1466: data->state.authhost.want = data->set.httpauth;
! 1467: data->state.authproxy.want = data->set.proxyauth;
! 1468: Curl_safefree(data->info.wouldredirect);
! 1469: data->info.wouldredirect = NULL;
! 1470:
! 1471: if(data->set.httpreq == HTTPREQ_PUT)
! 1472: data->state.infilesize = data->set.filesize;
! 1473: else if((data->set.httpreq != HTTPREQ_GET) &&
! 1474: (data->set.httpreq != HTTPREQ_HEAD)) {
! 1475: data->state.infilesize = data->set.postfieldsize;
! 1476: if(data->set.postfields && (data->state.infilesize == -1))
! 1477: data->state.infilesize = (curl_off_t)strlen(data->set.postfields);
! 1478: }
! 1479: else
! 1480: data->state.infilesize = 0;
! 1481:
! 1482: /* If there is a list of cookie files to read, do it now! */
! 1483: if(data->change.cookielist)
! 1484: Curl_cookie_loadfiles(data);
! 1485:
! 1486: /* If there is a list of host pairs to deal with */
! 1487: if(data->change.resolve)
! 1488: result = Curl_loadhostpairs(data);
! 1489:
! 1490: if(!result) {
! 1491: /* Allow data->set.use_port to set which port to use. This needs to be
! 1492: * disabled for example when we follow Location: headers to URLs using
! 1493: * different ports! */
! 1494: data->state.allow_port = TRUE;
! 1495:
! 1496: #if defined(HAVE_SIGNAL) && defined(SIGPIPE) && !defined(HAVE_MSG_NOSIGNAL)
! 1497: /*************************************************************
! 1498: * Tell signal handler to ignore SIGPIPE
! 1499: *************************************************************/
! 1500: if(!data->set.no_signal)
! 1501: data->state.prev_signal = signal(SIGPIPE, SIG_IGN);
! 1502: #endif
! 1503:
! 1504: Curl_initinfo(data); /* reset session-specific information "variables" */
! 1505: Curl_pgrsResetTransferSizes(data);
! 1506: Curl_pgrsStartNow(data);
! 1507:
! 1508: /* In case the handle is re-used and an authentication method was picked
! 1509: in the session we need to make sure we only use the one(s) we now
! 1510: consider to be fine */
! 1511: data->state.authhost.picked &= data->state.authhost.want;
! 1512: data->state.authproxy.picked &= data->state.authproxy.want;
! 1513:
! 1514: #ifndef CURL_DISABLE_FTP
! 1515: if(data->state.wildcardmatch) {
! 1516: struct WildcardData *wc = &data->wildcard;
! 1517: if(wc->state < CURLWC_INIT) {
! 1518: result = Curl_wildcard_init(wc); /* init wildcard structures */
! 1519: if(result)
! 1520: return CURLE_OUT_OF_MEMORY;
! 1521: }
! 1522: }
! 1523: #endif
! 1524: Curl_http2_init_state(&data->state);
! 1525: }
! 1526:
! 1527: return result;
! 1528: }
! 1529:
! 1530: /*
! 1531: * Curl_posttransfer() is called immediately after a transfer ends
! 1532: */
! 1533: CURLcode Curl_posttransfer(struct Curl_easy *data)
! 1534: {
! 1535: #if defined(HAVE_SIGNAL) && defined(SIGPIPE) && !defined(HAVE_MSG_NOSIGNAL)
! 1536: /* restore the signal handler for SIGPIPE before we get back */
! 1537: if(!data->set.no_signal)
! 1538: signal(SIGPIPE, data->state.prev_signal);
! 1539: #else
! 1540: (void)data; /* unused parameter */
! 1541: #endif
! 1542:
! 1543: return CURLE_OK;
! 1544: }
! 1545:
! 1546: /*
! 1547: * Curl_follow() handles the URL redirect magic. Pass in the 'newurl' string
! 1548: * as given by the remote server and set up the new URL to request.
! 1549: *
! 1550: * This function DOES NOT FREE the given url.
! 1551: */
! 1552: CURLcode Curl_follow(struct Curl_easy *data,
! 1553: char *newurl, /* the Location: string */
! 1554: followtype type) /* see transfer.h */
! 1555: {
! 1556: #ifdef CURL_DISABLE_HTTP
! 1557: (void)data;
! 1558: (void)newurl;
! 1559: (void)type;
! 1560: /* Location: following will not happen when HTTP is disabled */
! 1561: return CURLE_TOO_MANY_REDIRECTS;
! 1562: #else
! 1563:
! 1564: /* Location: redirect */
! 1565: bool disallowport = FALSE;
! 1566: bool reachedmax = FALSE;
! 1567: CURLUcode uc;
! 1568:
! 1569: if(type == FOLLOW_REDIR) {
! 1570: if((data->set.maxredirs != -1) &&
! 1571: (data->set.followlocation >= data->set.maxredirs)) {
! 1572: reachedmax = TRUE;
! 1573: type = FOLLOW_FAKE; /* switch to fake to store the would-be-redirected
! 1574: to URL */
! 1575: }
! 1576: else {
! 1577: /* mark the next request as a followed location: */
! 1578: data->state.this_is_a_follow = TRUE;
! 1579:
! 1580: data->set.followlocation++; /* count location-followers */
! 1581:
! 1582: if(data->set.http_auto_referer) {
! 1583: /* We are asked to automatically set the previous URL as the referer
! 1584: when we get the next URL. We pick the ->url field, which may or may
! 1585: not be 100% correct */
! 1586:
! 1587: if(data->change.referer_alloc) {
! 1588: Curl_safefree(data->change.referer);
! 1589: data->change.referer_alloc = FALSE;
! 1590: }
! 1591:
! 1592: data->change.referer = strdup(data->change.url);
! 1593: if(!data->change.referer)
! 1594: return CURLE_OUT_OF_MEMORY;
! 1595: data->change.referer_alloc = TRUE; /* yes, free this later */
! 1596: }
! 1597: }
! 1598: }
! 1599:
! 1600: if(Curl_is_absolute_url(newurl, NULL, MAX_SCHEME_LEN))
! 1601: /* This is an absolute URL, don't allow the custom port number */
! 1602: disallowport = TRUE;
! 1603:
! 1604: DEBUGASSERT(data->state.uh);
! 1605: uc = curl_url_set(data->state.uh, CURLUPART_URL, newurl,
! 1606: (type == FOLLOW_FAKE) ? CURLU_NON_SUPPORT_SCHEME :
! 1607: ((type == FOLLOW_REDIR) ? CURLU_URLENCODE : 0) );
! 1608: if(uc) {
! 1609: if(type != FOLLOW_FAKE)
! 1610: return Curl_uc_to_curlcode(uc);
! 1611:
! 1612: /* the URL could not be parsed for some reason, but since this is FAKE
! 1613: mode, just duplicate the field as-is */
! 1614: newurl = strdup(newurl);
! 1615: if(!newurl)
! 1616: return CURLE_OUT_OF_MEMORY;
! 1617: }
! 1618: else {
! 1619:
! 1620: uc = curl_url_get(data->state.uh, CURLUPART_URL, &newurl, 0);
! 1621: if(uc)
! 1622: return Curl_uc_to_curlcode(uc);
! 1623: }
! 1624:
! 1625: if(type == FOLLOW_FAKE) {
! 1626: /* we're only figuring out the new url if we would've followed locations
! 1627: but now we're done so we can get out! */
! 1628: data->info.wouldredirect = newurl;
! 1629:
! 1630: if(reachedmax) {
! 1631: failf(data, "Maximum (%ld) redirects followed", data->set.maxredirs);
! 1632: return CURLE_TOO_MANY_REDIRECTS;
! 1633: }
! 1634: return CURLE_OK;
! 1635: }
! 1636:
! 1637: if(disallowport)
! 1638: data->state.allow_port = FALSE;
! 1639:
! 1640: if(data->change.url_alloc)
! 1641: Curl_safefree(data->change.url);
! 1642:
! 1643: data->change.url = newurl;
! 1644: data->change.url_alloc = TRUE;
! 1645:
! 1646: infof(data, "Issue another request to this URL: '%s'\n", data->change.url);
! 1647:
! 1648: /*
! 1649: * We get here when the HTTP code is 300-399 (and 401). We need to perform
! 1650: * differently based on exactly what return code there was.
! 1651: *
! 1652: * News from 7.10.6: we can also get here on a 401 or 407, in case we act on
! 1653: * a HTTP (proxy-) authentication scheme other than Basic.
! 1654: */
! 1655: switch(data->info.httpcode) {
! 1656: /* 401 - Act on a WWW-Authenticate, we keep on moving and do the
! 1657: Authorization: XXXX header in the HTTP request code snippet */
! 1658: /* 407 - Act on a Proxy-Authenticate, we keep on moving and do the
! 1659: Proxy-Authorization: XXXX header in the HTTP request code snippet */
! 1660: /* 300 - Multiple Choices */
! 1661: /* 306 - Not used */
! 1662: /* 307 - Temporary Redirect */
! 1663: default: /* for all above (and the unknown ones) */
! 1664: /* Some codes are explicitly mentioned since I've checked RFC2616 and they
! 1665: * seem to be OK to POST to.
! 1666: */
! 1667: break;
! 1668: case 301: /* Moved Permanently */
! 1669: /* (quote from RFC7231, section 6.4.2)
! 1670: *
! 1671: * Note: For historical reasons, a user agent MAY change the request
! 1672: * method from POST to GET for the subsequent request. If this
! 1673: * behavior is undesired, the 307 (Temporary Redirect) status code
! 1674: * can be used instead.
! 1675: *
! 1676: * ----
! 1677: *
! 1678: * Many webservers expect this, so these servers often answers to a POST
! 1679: * request with an error page. To be sure that libcurl gets the page that
! 1680: * most user agents would get, libcurl has to force GET.
! 1681: *
! 1682: * This behaviour is forbidden by RFC1945 and the obsolete RFC2616, and
! 1683: * can be overridden with CURLOPT_POSTREDIR.
! 1684: */
! 1685: if((data->set.httpreq == HTTPREQ_POST
! 1686: || data->set.httpreq == HTTPREQ_POST_FORM
! 1687: || data->set.httpreq == HTTPREQ_POST_MIME)
! 1688: && !(data->set.keep_post & CURL_REDIR_POST_301)) {
! 1689: infof(data, "Switch from POST to GET\n");
! 1690: data->set.httpreq = HTTPREQ_GET;
! 1691: }
! 1692: break;
! 1693: case 302: /* Found */
! 1694: /* (quote from RFC7231, section 6.4.3)
! 1695: *
! 1696: * Note: For historical reasons, a user agent MAY change the request
! 1697: * method from POST to GET for the subsequent request. If this
! 1698: * behavior is undesired, the 307 (Temporary Redirect) status code
! 1699: * can be used instead.
! 1700: *
! 1701: * ----
! 1702: *
! 1703: * Many webservers expect this, so these servers often answers to a POST
! 1704: * request with an error page. To be sure that libcurl gets the page that
! 1705: * most user agents would get, libcurl has to force GET.
! 1706: *
! 1707: * This behaviour is forbidden by RFC1945 and the obsolete RFC2616, and
! 1708: * can be overridden with CURLOPT_POSTREDIR.
! 1709: */
! 1710: if((data->set.httpreq == HTTPREQ_POST
! 1711: || data->set.httpreq == HTTPREQ_POST_FORM
! 1712: || data->set.httpreq == HTTPREQ_POST_MIME)
! 1713: && !(data->set.keep_post & CURL_REDIR_POST_302)) {
! 1714: infof(data, "Switch from POST to GET\n");
! 1715: data->set.httpreq = HTTPREQ_GET;
! 1716: }
! 1717: break;
! 1718:
! 1719: case 303: /* See Other */
! 1720: /* 'See Other' location is not the resource but a substitute for the
! 1721: * resource. In this case we switch the method to GET/HEAD, unless the
! 1722: * method is POST and the user specified to keep it as POST.
! 1723: * https://github.com/curl/curl/issues/5237#issuecomment-614641049
! 1724: */
! 1725: if(data->set.httpreq != HTTPREQ_GET &&
! 1726: ((data->set.httpreq != HTTPREQ_POST &&
! 1727: data->set.httpreq != HTTPREQ_POST_FORM &&
! 1728: data->set.httpreq != HTTPREQ_POST_MIME) ||
! 1729: !(data->set.keep_post & CURL_REDIR_POST_303))) {
! 1730: data->set.httpreq = HTTPREQ_GET;
! 1731: data->set.upload = false;
! 1732: infof(data, "Switch to %s\n",
! 1733: data->set.opt_no_body?"HEAD":"GET");
! 1734: }
! 1735: break;
! 1736: case 304: /* Not Modified */
! 1737: /* 304 means we did a conditional request and it was "Not modified".
! 1738: * We shouldn't get any Location: header in this response!
! 1739: */
! 1740: break;
! 1741: case 305: /* Use Proxy */
! 1742: /* (quote from RFC2616, section 10.3.6):
! 1743: * "The requested resource MUST be accessed through the proxy given
! 1744: * by the Location field. The Location field gives the URI of the
! 1745: * proxy. The recipient is expected to repeat this single request
! 1746: * via the proxy. 305 responses MUST only be generated by origin
! 1747: * servers."
! 1748: */
! 1749: break;
! 1750: }
! 1751: Curl_pgrsTime(data, TIMER_REDIRECT);
! 1752: Curl_pgrsResetTransferSizes(data);
! 1753:
! 1754: return CURLE_OK;
! 1755: #endif /* CURL_DISABLE_HTTP */
! 1756: }
! 1757:
! 1758: /* Returns CURLE_OK *and* sets '*url' if a request retry is wanted.
! 1759:
! 1760: NOTE: that the *url is malloc()ed. */
! 1761: CURLcode Curl_retry_request(struct connectdata *conn,
! 1762: char **url)
! 1763: {
! 1764: struct Curl_easy *data = conn->data;
! 1765: bool retry = FALSE;
! 1766: *url = NULL;
! 1767:
! 1768: /* if we're talking upload, we can't do the checks below, unless the protocol
! 1769: is HTTP as when uploading over HTTP we will still get a response */
! 1770: if(data->set.upload &&
! 1771: !(conn->handler->protocol&(PROTO_FAMILY_HTTP|CURLPROTO_RTSP)))
! 1772: return CURLE_OK;
! 1773:
! 1774: if((data->req.bytecount + data->req.headerbytecount == 0) &&
! 1775: conn->bits.reuse &&
! 1776: (!data->set.opt_no_body
! 1777: || (conn->handler->protocol & PROTO_FAMILY_HTTP)) &&
! 1778: (data->set.rtspreq != RTSPREQ_RECEIVE))
! 1779: /* We got no data, we attempted to re-use a connection. For HTTP this
! 1780: can be a retry so we try again regardless if we expected a body.
! 1781: For other protocols we only try again only if we expected a body.
! 1782:
! 1783: This might happen if the connection was left alive when we were
! 1784: done using it before, but that was closed when we wanted to read from
! 1785: it again. Bad luck. Retry the same request on a fresh connect! */
! 1786: retry = TRUE;
! 1787: else if(data->state.refused_stream &&
! 1788: (data->req.bytecount + data->req.headerbytecount == 0) ) {
! 1789: /* This was sent on a refused stream, safe to rerun. A refused stream
! 1790: error can typically only happen on HTTP/2 level if the stream is safe
! 1791: to issue again, but the nghttp2 API can deliver the message to other
! 1792: streams as well, which is why this adds the check the data counters
! 1793: too. */
! 1794: infof(conn->data, "REFUSED_STREAM, retrying a fresh connect\n");
! 1795: data->state.refused_stream = FALSE; /* clear again */
! 1796: retry = TRUE;
! 1797: }
! 1798: if(retry) {
! 1799: #define CONN_MAX_RETRIES 5
! 1800: if(conn->retrycount++ >= CONN_MAX_RETRIES) {
! 1801: failf(data, "Connection died, tried %d times before giving up",
! 1802: CONN_MAX_RETRIES);
! 1803: return CURLE_SEND_ERROR;
! 1804: }
! 1805: infof(conn->data, "Connection died, retrying a fresh connect\n");
! 1806: *url = strdup(conn->data->change.url);
! 1807: if(!*url)
! 1808: return CURLE_OUT_OF_MEMORY;
! 1809:
! 1810: connclose(conn, "retry"); /* close this connection */
! 1811: conn->bits.retry = TRUE; /* mark this as a connection we're about
! 1812: to retry. Marking it this way should
! 1813: prevent i.e HTTP transfers to return
! 1814: error just because nothing has been
! 1815: transferred! */
! 1816:
! 1817:
! 1818: if(conn->handler->protocol&PROTO_FAMILY_HTTP) {
! 1819: if(data->req.writebytecount) {
! 1820: CURLcode result = Curl_readrewind(conn);
! 1821: if(result) {
! 1822: Curl_safefree(*url);
! 1823: return result;
! 1824: }
! 1825: }
! 1826: }
! 1827: }
! 1828: return CURLE_OK;
! 1829: }
! 1830:
! 1831: /*
! 1832: * Curl_setup_transfer() is called to setup some basic properties for the
! 1833: * upcoming transfer.
! 1834: */
! 1835: void
! 1836: Curl_setup_transfer(
! 1837: struct Curl_easy *data, /* transfer */
! 1838: int sockindex, /* socket index to read from or -1 */
! 1839: curl_off_t size, /* -1 if unknown at this point */
! 1840: bool getheader, /* TRUE if header parsing is wanted */
! 1841: int writesockindex /* socket index to write to, it may very well be
! 1842: the same we read from. -1 disables */
! 1843: )
! 1844: {
! 1845: struct SingleRequest *k = &data->req;
! 1846: struct connectdata *conn = data->conn;
! 1847: struct HTTP *http = data->req.protop;
! 1848: bool httpsending = ((conn->handler->protocol&PROTO_FAMILY_HTTP) &&
! 1849: (http->sending == HTTPSEND_REQUEST));
! 1850: DEBUGASSERT(conn != NULL);
! 1851: DEBUGASSERT((sockindex <= 1) && (sockindex >= -1));
! 1852:
! 1853: if(conn->bits.multiplex || conn->httpversion == 20 || httpsending) {
! 1854: /* when multiplexing, the read/write sockets need to be the same! */
! 1855: conn->sockfd = sockindex == -1 ?
! 1856: ((writesockindex == -1 ? CURL_SOCKET_BAD : conn->sock[writesockindex])) :
! 1857: conn->sock[sockindex];
! 1858: conn->writesockfd = conn->sockfd;
! 1859: if(httpsending)
! 1860: /* special and very HTTP-specific */
! 1861: writesockindex = FIRSTSOCKET;
! 1862: }
! 1863: else {
! 1864: conn->sockfd = sockindex == -1 ?
! 1865: CURL_SOCKET_BAD : conn->sock[sockindex];
! 1866: conn->writesockfd = writesockindex == -1 ?
! 1867: CURL_SOCKET_BAD:conn->sock[writesockindex];
! 1868: }
! 1869: k->getheader = getheader;
! 1870:
! 1871: k->size = size;
! 1872:
! 1873: /* The code sequence below is placed in this function just because all
! 1874: necessary input is not always known in do_complete() as this function may
! 1875: be called after that */
! 1876:
! 1877: if(!k->getheader) {
! 1878: k->header = FALSE;
! 1879: if(size > 0)
! 1880: Curl_pgrsSetDownloadSize(data, size);
! 1881: }
! 1882: /* we want header and/or body, if neither then don't do this! */
! 1883: if(k->getheader || !data->set.opt_no_body) {
! 1884:
! 1885: if(sockindex != -1)
! 1886: k->keepon |= KEEP_RECV;
! 1887:
! 1888: if(writesockindex != -1) {
! 1889: /* HTTP 1.1 magic:
! 1890:
! 1891: Even if we require a 100-return code before uploading data, we might
! 1892: need to write data before that since the REQUEST may not have been
! 1893: finished sent off just yet.
! 1894:
! 1895: Thus, we must check if the request has been sent before we set the
! 1896: state info where we wait for the 100-return code
! 1897: */
! 1898: if((data->state.expect100header) &&
! 1899: (conn->handler->protocol&PROTO_FAMILY_HTTP) &&
! 1900: (http->sending == HTTPSEND_BODY)) {
! 1901: /* wait with write until we either got 100-continue or a timeout */
! 1902: k->exp100 = EXP100_AWAITING_CONTINUE;
! 1903: k->start100 = Curl_now();
! 1904:
! 1905: /* Set a timeout for the multi interface. Add the inaccuracy margin so
! 1906: that we don't fire slightly too early and get denied to run. */
! 1907: Curl_expire(data, data->set.expect_100_timeout, EXPIRE_100_TIMEOUT);
! 1908: }
! 1909: else {
! 1910: if(data->state.expect100header)
! 1911: /* when we've sent off the rest of the headers, we must await a
! 1912: 100-continue but first finish sending the request */
! 1913: k->exp100 = EXP100_SENDING_REQUEST;
! 1914:
! 1915: /* enable the write bit when we're not waiting for continue */
! 1916: k->keepon |= KEEP_SEND;
! 1917: }
! 1918: } /* if(writesockindex != -1) */
! 1919: } /* if(k->getheader || !data->set.opt_no_body) */
! 1920:
! 1921: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to transfer.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / curl / lib