123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443 |
- /***************************************************************************
- * _ _ ____ _
- * Project ___| | | | _ \| |
- * / __| | | | |_) | |
- * | (__| |_| | _ <| |___
- * \___|\___/|_| \_\_____|
- *
- * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
- *
- * This software is licensed as described in the file COPYING, which
- * you should have received as part of this distribution. The terms
- * are also available at https://curl.se/docs/copyright.html.
- *
- * You may opt to use, copy, modify, merge, publish, distribute and/or sell
- * copies of the Software, and permit persons to whom the Software is
- * furnished to do so, under the terms of the COPYING file.
- *
- * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
- * KIND, either express or implied.
- *
- * SPDX-License-Identifier: curl
- *
- ***************************************************************************/
- #include "curl_setup.h"
- #ifdef HAVE_NETINET_IN_H
- #include <netinet/in.h>
- #endif
- #ifdef HAVE_LINUX_TCP_H
- #include <linux/tcp.h>
- #elif defined(HAVE_NETINET_TCP_H)
- #include <netinet/tcp.h>
- #endif
- #include <curl/curl.h>
- #include "urldata.h"
- #include "sendf.h"
- #include "cfilters.h"
- #include "connect.h"
- #include "content_encoding.h"
- #include "cw-out.h"
- #include "vtls/vtls.h"
- #include "vssh/ssh.h"
- #include "easyif.h"
- #include "multiif.h"
- #include "strerror.h"
- #include "select.h"
- #include "strdup.h"
- #include "http2.h"
- #include "progress.h"
- #include "warnless.h"
- #include "ws.h"
- /* The last 3 #include files should be in this order */
- #include "curl_printf.h"
- #include "curl_memory.h"
- #include "memdebug.h"
- static CURLcode do_init_writer_stack(struct Curl_easy *data);
- /* Curl_client_write() sends data to the write callback(s)
- The bit pattern defines to what "streams" to write to. Body and/or header.
- The defines are in sendf.h of course.
- */
- CURLcode Curl_client_write(struct Curl_easy *data,
- int type, const char *buf, size_t blen)
- {
- CURLcode result;
- /* it is one of those, at least */
- DEBUGASSERT(type & (CLIENTWRITE_BODY|CLIENTWRITE_HEADER|CLIENTWRITE_INFO));
- /* BODY is only BODY (with optional EOS) */
- DEBUGASSERT(!(type & CLIENTWRITE_BODY) ||
- ((type & ~(CLIENTWRITE_BODY|CLIENTWRITE_EOS)) == 0));
- /* INFO is only INFO (with optional EOS) */
- DEBUGASSERT(!(type & CLIENTWRITE_INFO) ||
- ((type & ~(CLIENTWRITE_INFO|CLIENTWRITE_EOS)) == 0));
- if(!data->req.writer_stack) {
- result = do_init_writer_stack(data);
- if(result)
- return result;
- DEBUGASSERT(data->req.writer_stack);
- }
- result = Curl_cwriter_write(data, data->req.writer_stack, type, buf, blen);
- CURL_TRC_WRITE(data, "client_write(type=%x, len=%zu) -> %d",
- type, blen, result);
- return result;
- }
- static void cl_reset_writer(struct Curl_easy *data)
- {
- struct Curl_cwriter *writer = data->req.writer_stack;
- while(writer) {
- data->req.writer_stack = writer->next;
- writer->cwt->do_close(data, writer);
- free(writer);
- writer = data->req.writer_stack;
- }
- }
- static void cl_reset_reader(struct Curl_easy *data)
- {
- struct Curl_creader *reader = data->req.reader_stack;
- while(reader) {
- data->req.reader_stack = reader->next;
- reader->crt->do_close(data, reader);
- free(reader);
- reader = data->req.reader_stack;
- }
- }
- void Curl_client_cleanup(struct Curl_easy *data)
- {
- cl_reset_reader(data);
- cl_reset_writer(data);
- data->req.bytecount = 0;
- data->req.headerline = 0;
- }
- void Curl_client_reset(struct Curl_easy *data)
- {
- if(data->req.rewind_read) {
- /* already requested */
- CURL_TRC_READ(data, "client_reset, will rewind reader");
- }
- else {
- CURL_TRC_READ(data, "client_reset, clear readers");
- cl_reset_reader(data);
- }
- cl_reset_writer(data);
- data->req.bytecount = 0;
- data->req.headerline = 0;
- }
- CURLcode Curl_client_start(struct Curl_easy *data)
- {
- if(data->req.rewind_read) {
- struct Curl_creader *r = data->req.reader_stack;
- CURLcode result = CURLE_OK;
- CURL_TRC_READ(data, "client start, rewind readers");
- while(r) {
- result = r->crt->rewind(data, r);
- if(result) {
- failf(data, "rewind of client reader '%s' failed: %d",
- r->crt->name, result);
- return result;
- }
- r = r->next;
- }
- data->req.rewind_read = FALSE;
- cl_reset_reader(data);
- }
- return CURLE_OK;
- }
- bool Curl_creader_will_rewind(struct Curl_easy *data)
- {
- return data->req.rewind_read;
- }
- void Curl_creader_set_rewind(struct Curl_easy *data, bool enable)
- {
- data->req.rewind_read = !!enable;
- }
- /* Write data using an unencoding writer stack. */
- CURLcode Curl_cwriter_write(struct Curl_easy *data,
- struct Curl_cwriter *writer, int type,
- const char *buf, size_t nbytes)
- {
- if(!writer)
- return CURLE_WRITE_ERROR;
- return writer->cwt->do_write(data, writer, type, buf, nbytes);
- }
- CURLcode Curl_cwriter_def_init(struct Curl_easy *data,
- struct Curl_cwriter *writer)
- {
- (void)data;
- (void)writer;
- return CURLE_OK;
- }
- CURLcode Curl_cwriter_def_write(struct Curl_easy *data,
- struct Curl_cwriter *writer, int type,
- const char *buf, size_t nbytes)
- {
- return Curl_cwriter_write(data, writer->next, type, buf, nbytes);
- }
- void Curl_cwriter_def_close(struct Curl_easy *data,
- struct Curl_cwriter *writer)
- {
- (void) data;
- (void) writer;
- }
- static size_t get_max_body_write_len(struct Curl_easy *data, curl_off_t limit)
- {
- if(limit != -1) {
- /* How much more are we allowed to write? */
- curl_off_t remain_diff;
- remain_diff = limit - data->req.bytecount;
- if(remain_diff < 0) {
- /* already written too much! */
- return 0;
- }
- #if SIZEOF_CURL_OFF_T > SIZEOF_SIZE_T
- else if(remain_diff > SSIZE_T_MAX) {
- return SIZE_T_MAX;
- }
- #endif
- else {
- return (size_t)remain_diff;
- }
- }
- return SIZE_T_MAX;
- }
- struct cw_download_ctx {
- struct Curl_cwriter super;
- BIT(started_response);
- };
- /* Download client writer in phase CURL_CW_PROTOCOL that
- * sees the "real" download body data. */
- static CURLcode cw_download_write(struct Curl_easy *data,
- struct Curl_cwriter *writer, int type,
- const char *buf, size_t nbytes)
- {
- struct cw_download_ctx *ctx = writer->ctx;
- CURLcode result;
- size_t nwrite, excess_len = 0;
- bool is_connect = !!(type & CLIENTWRITE_CONNECT);
- if(!is_connect && !ctx->started_response) {
- Curl_pgrsTime(data, TIMER_STARTTRANSFER);
- ctx->started_response = TRUE;
- }
- if(!(type & CLIENTWRITE_BODY)) {
- if(is_connect && data->set.suppress_connect_headers)
- return CURLE_OK;
- result = Curl_cwriter_write(data, writer->next, type, buf, nbytes);
- CURL_TRC_WRITE(data, "download_write header(type=%x, blen=%zu) -> %d",
- type, nbytes, result);
- return result;
- }
- /* Here, we deal with REAL BODY bytes. All filtering and transfer
- * encodings have been applied and only the true content, e.g. BODY,
- * bytes are passed here.
- * This allows us to check sizes, update stats, etc. independent
- * from the protocol in play. */
- if(data->req.no_body && nbytes > 0) {
- /* BODY arrives although we want none, bail out */
- streamclose(data->conn, "ignoring body");
- CURL_TRC_WRITE(data, "download_write body(type=%x, blen=%zu), "
- "did not want a BODY", type, nbytes);
- data->req.download_done = TRUE;
- if(data->info.header_size)
- /* if headers have been received, this is fine */
- return CURLE_OK;
- return CURLE_WEIRD_SERVER_REPLY;
- }
- /* Determine if we see any bytes in excess to what is allowed.
- * We write the allowed bytes and handle excess further below.
- * This gives deterministic BODY writes on varying buffer receive
- * lengths. */
- nwrite = nbytes;
- if(-1 != data->req.maxdownload) {
- size_t wmax = get_max_body_write_len(data, data->req.maxdownload);
- if(nwrite > wmax) {
- excess_len = nbytes - wmax;
- nwrite = wmax;
- }
- if(nwrite == wmax) {
- data->req.download_done = TRUE;
- }
- if((type & CLIENTWRITE_EOS) && !data->req.no_body &&
- (data->req.maxdownload > data->req.bytecount)) {
- failf(data, "end of response with %" FMT_OFF_T " bytes missing",
- data->req.maxdownload - data->req.bytecount);
- return CURLE_PARTIAL_FILE;
- }
- }
- /* Error on too large filesize is handled below, after writing
- * the permitted bytes */
- if(data->set.max_filesize && !data->req.ignorebody) {
- size_t wmax = get_max_body_write_len(data, data->set.max_filesize);
- if(nwrite > wmax) {
- nwrite = wmax;
- }
- }
- if(!data->req.ignorebody && (nwrite || (type & CLIENTWRITE_EOS))) {
- result = Curl_cwriter_write(data, writer->next, type, buf, nwrite);
- CURL_TRC_WRITE(data, "download_write body(type=%x, blen=%zu) -> %d",
- type, nbytes, result);
- if(result)
- return result;
- }
- /* Update stats, write and report progress */
- data->req.bytecount += nwrite;
- #ifdef USE_HYPER
- data->req.bodywritten = TRUE;
- #endif
- result = Curl_pgrsSetDownloadCounter(data, data->req.bytecount);
- if(result)
- return result;
- if(excess_len) {
- if(!data->req.ignorebody) {
- infof(data,
- "Excess found writing body:"
- " excess = %zu"
- ", size = %" FMT_OFF_T
- ", maxdownload = %" FMT_OFF_T
- ", bytecount = %" FMT_OFF_T,
- excess_len, data->req.size, data->req.maxdownload,
- data->req.bytecount);
- connclose(data->conn, "excess found in a read");
- }
- }
- else if((nwrite < nbytes) && !data->req.ignorebody) {
- failf(data, "Exceeded the maximum allowed file size "
- "(%" FMT_OFF_T ") with %" FMT_OFF_T " bytes",
- data->set.max_filesize, data->req.bytecount);
- return CURLE_FILESIZE_EXCEEDED;
- }
- return CURLE_OK;
- }
- static const struct Curl_cwtype cw_download = {
- "protocol",
- NULL,
- Curl_cwriter_def_init,
- cw_download_write,
- Curl_cwriter_def_close,
- sizeof(struct cw_download_ctx)
- };
- /* RAW client writer in phase CURL_CW_RAW that
- * enabled tracing of raw data. */
- static CURLcode cw_raw_write(struct Curl_easy *data,
- struct Curl_cwriter *writer, int type,
- const char *buf, size_t nbytes)
- {
- if(type & CLIENTWRITE_BODY && data->set.verbose && !data->req.ignorebody) {
- Curl_debug(data, CURLINFO_DATA_IN, (char *)buf, nbytes);
- }
- return Curl_cwriter_write(data, writer->next, type, buf, nbytes);
- }
- static const struct Curl_cwtype cw_raw = {
- "raw",
- NULL,
- Curl_cwriter_def_init,
- cw_raw_write,
- Curl_cwriter_def_close,
- sizeof(struct Curl_cwriter)
- };
- /* Create an unencoding writer stage using the given handler. */
- CURLcode Curl_cwriter_create(struct Curl_cwriter **pwriter,
- struct Curl_easy *data,
- const struct Curl_cwtype *cwt,
- Curl_cwriter_phase phase)
- {
- struct Curl_cwriter *writer = NULL;
- CURLcode result = CURLE_OUT_OF_MEMORY;
- void *p;
- DEBUGASSERT(cwt->cwriter_size >= sizeof(struct Curl_cwriter));
- p = calloc(1, cwt->cwriter_size);
- if(!p)
- goto out;
- writer = (struct Curl_cwriter *)p;
- writer->cwt = cwt;
- writer->ctx = p;
- writer->phase = phase;
- result = cwt->do_init(data, writer);
- out:
- *pwriter = result ? NULL : writer;
- if(result)
- free(writer);
- return result;
- }
- void Curl_cwriter_free(struct Curl_easy *data,
- struct Curl_cwriter *writer)
- {
- if(writer) {
- writer->cwt->do_close(data, writer);
- free(writer);
- }
- }
- size_t Curl_cwriter_count(struct Curl_easy *data, Curl_cwriter_phase phase)
- {
- struct Curl_cwriter *w;
- size_t n = 0;
- for(w = data->req.writer_stack; w; w = w->next) {
- if(w->phase == phase)
- ++n;
- }
- return n;
- }
- static CURLcode do_init_writer_stack(struct Curl_easy *data)
- {
- struct Curl_cwriter *writer;
- CURLcode result;
- DEBUGASSERT(!data->req.writer_stack);
- result = Curl_cwriter_create(&data->req.writer_stack,
- data, &Curl_cwt_out, CURL_CW_CLIENT);
- if(result)
- return result;
- result = Curl_cwriter_create(&writer, data, &cw_download, CURL_CW_PROTOCOL);
- if(result)
- return result;
- result = Curl_cwriter_add(data, writer);
- if(result) {
- Curl_cwriter_free(data, writer);
- }
- result = Curl_cwriter_create(&writer, data, &cw_raw, CURL_CW_RAW);
- if(result)
- return result;
- result = Curl_cwriter_add(data, writer);
- if(result) {
- Curl_cwriter_free(data, writer);
- }
- return result;
- }
- CURLcode Curl_cwriter_add(struct Curl_easy *data,
- struct Curl_cwriter *writer)
- {
- CURLcode result;
- struct Curl_cwriter **anchor = &data->req.writer_stack;
- if(!*anchor) {
- result = do_init_writer_stack(data);
- if(result)
- return result;
- }
- /* Insert the writer as first in its phase.
- * Skip existing writers of lower phases. */
- while(*anchor && (*anchor)->phase < writer->phase)
- anchor = &((*anchor)->next);
- writer->next = *anchor;
- *anchor = writer;
- return CURLE_OK;
- }
- struct Curl_cwriter *Curl_cwriter_get_by_name(struct Curl_easy *data,
- const char *name)
- {
- struct Curl_cwriter *writer;
- for(writer = data->req.writer_stack; writer; writer = writer->next) {
- if(!strcmp(name, writer->cwt->name))
- return writer;
- }
- return NULL;
- }
- struct Curl_cwriter *Curl_cwriter_get_by_type(struct Curl_easy *data,
- const struct Curl_cwtype *cwt)
- {
- struct Curl_cwriter *writer;
- for(writer = data->req.writer_stack; writer; writer = writer->next) {
- if(writer->cwt == cwt)
- return writer;
- }
- return NULL;
- }
- void Curl_cwriter_remove_by_name(struct Curl_easy *data,
- const char *name)
- {
- struct Curl_cwriter **anchor = &data->req.writer_stack;
- while(*anchor) {
- if(!strcmp(name, (*anchor)->cwt->name)) {
- struct Curl_cwriter *w = (*anchor);
- *anchor = w->next;
- Curl_cwriter_free(data, w);
- continue;
- }
- anchor = &((*anchor)->next);
- }
- }
- bool Curl_cwriter_is_paused(struct Curl_easy *data)
- {
- return Curl_cw_out_is_paused(data);
- }
- CURLcode Curl_cwriter_unpause(struct Curl_easy *data)
- {
- return Curl_cw_out_unpause(data);
- }
- CURLcode Curl_creader_read(struct Curl_easy *data,
- struct Curl_creader *reader,
- char *buf, size_t blen, size_t *nread, bool *eos)
- {
- *nread = 0;
- *eos = FALSE;
- if(!reader)
- return CURLE_READ_ERROR;
- return reader->crt->do_read(data, reader, buf, blen, nread, eos);
- }
- CURLcode Curl_creader_def_init(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- (void)data;
- (void)reader;
- return CURLE_OK;
- }
- void Curl_creader_def_close(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- (void)data;
- (void)reader;
- }
- CURLcode Curl_creader_def_read(struct Curl_easy *data,
- struct Curl_creader *reader,
- char *buf, size_t blen,
- size_t *nread, bool *eos)
- {
- if(reader->next)
- return reader->next->crt->do_read(data, reader->next, buf, blen,
- nread, eos);
- else {
- *nread = 0;
- *eos = FALSE;
- return CURLE_READ_ERROR;
- }
- }
- bool Curl_creader_def_needs_rewind(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- (void)data;
- (void)reader;
- return FALSE;
- }
- curl_off_t Curl_creader_def_total_length(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- return reader->next ?
- reader->next->crt->total_length(data, reader->next) : -1;
- }
- CURLcode Curl_creader_def_resume_from(struct Curl_easy *data,
- struct Curl_creader *reader,
- curl_off_t offset)
- {
- (void)data;
- (void)reader;
- (void)offset;
- return CURLE_READ_ERROR;
- }
- CURLcode Curl_creader_def_rewind(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- (void)data;
- (void)reader;
- return CURLE_OK;
- }
- CURLcode Curl_creader_def_unpause(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- (void)data;
- (void)reader;
- return CURLE_OK;
- }
- bool Curl_creader_def_is_paused(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- (void)data;
- (void)reader;
- return FALSE;
- }
- void Curl_creader_def_done(struct Curl_easy *data,
- struct Curl_creader *reader, int premature)
- {
- (void)data;
- (void)reader;
- (void)premature;
- }
- struct cr_in_ctx {
- struct Curl_creader super;
- curl_read_callback read_cb;
- void *cb_user_data;
- curl_off_t total_len;
- curl_off_t read_len;
- CURLcode error_result;
- BIT(seen_eos);
- BIT(errored);
- BIT(has_used_cb);
- BIT(is_paused);
- };
- static CURLcode cr_in_init(struct Curl_easy *data, struct Curl_creader *reader)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- (void)data;
- ctx->read_cb = data->state.fread_func;
- ctx->cb_user_data = data->state.in;
- ctx->total_len = -1;
- ctx->read_len = 0;
- return CURLE_OK;
- }
- /* Real client reader to installed client callbacks. */
- static CURLcode cr_in_read(struct Curl_easy *data,
- struct Curl_creader *reader,
- char *buf, size_t blen,
- size_t *pnread, bool *peos)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- size_t nread;
- ctx->is_paused = FALSE;
- /* Once we have errored, we will return the same error forever */
- if(ctx->errored) {
- *pnread = 0;
- *peos = FALSE;
- return ctx->error_result;
- }
- if(ctx->seen_eos) {
- *pnread = 0;
- *peos = TRUE;
- return CURLE_OK;
- }
- /* respect length limitations */
- if(ctx->total_len >= 0) {
- curl_off_t remain = ctx->total_len - ctx->read_len;
- if(remain <= 0)
- blen = 0;
- else if(remain < (curl_off_t)blen)
- blen = (size_t)remain;
- }
- nread = 0;
- if(ctx->read_cb && blen) {
- Curl_set_in_callback(data, true);
- nread = ctx->read_cb(buf, 1, blen, ctx->cb_user_data);
- Curl_set_in_callback(data, false);
- ctx->has_used_cb = TRUE;
- }
- switch(nread) {
- case 0:
- if((ctx->total_len >= 0) && (ctx->read_len < ctx->total_len)) {
- failf(data, "client read function EOF fail, "
- "only %"FMT_OFF_T"/%"FMT_OFF_T " of needed bytes read",
- ctx->read_len, ctx->total_len);
- return CURLE_READ_ERROR;
- }
- *pnread = 0;
- *peos = TRUE;
- ctx->seen_eos = TRUE;
- break;
- case CURL_READFUNC_ABORT:
- failf(data, "operation aborted by callback");
- *pnread = 0;
- *peos = FALSE;
- ctx->errored = TRUE;
- ctx->error_result = CURLE_ABORTED_BY_CALLBACK;
- return CURLE_ABORTED_BY_CALLBACK;
- case CURL_READFUNC_PAUSE:
- if(data->conn->handler->flags & PROTOPT_NONETWORK) {
- /* protocols that work without network cannot be paused. This is
- actually only FILE:// just now, and it cannot pause since the transfer
- is not done using the "normal" procedure. */
- failf(data, "Read callback asked for PAUSE when not supported");
- return CURLE_READ_ERROR;
- }
- /* CURL_READFUNC_PAUSE pauses read callbacks that feed socket writes */
- CURL_TRC_READ(data, "cr_in_read, callback returned CURL_READFUNC_PAUSE");
- ctx->is_paused = TRUE;
- data->req.keepon |= KEEP_SEND_PAUSE; /* mark socket send as paused */
- *pnread = 0;
- *peos = FALSE;
- break; /* nothing was read */
- default:
- if(nread > blen) {
- /* the read function returned a too large value */
- failf(data, "read function returned funny value");
- *pnread = 0;
- *peos = FALSE;
- ctx->errored = TRUE;
- ctx->error_result = CURLE_READ_ERROR;
- return CURLE_READ_ERROR;
- }
- ctx->read_len += nread;
- if(ctx->total_len >= 0)
- ctx->seen_eos = (ctx->read_len >= ctx->total_len);
- *pnread = nread;
- *peos = ctx->seen_eos;
- break;
- }
- CURL_TRC_READ(data, "cr_in_read(len=%zu, total=%"FMT_OFF_T
- ", read=%"FMT_OFF_T") -> %d, nread=%zu, eos=%d",
- blen, ctx->total_len, ctx->read_len, CURLE_OK,
- *pnread, *peos);
- return CURLE_OK;
- }
- static bool cr_in_needs_rewind(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- (void)data;
- return ctx->has_used_cb;
- }
- static curl_off_t cr_in_total_length(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- (void)data;
- return ctx->total_len;
- }
- static CURLcode cr_in_resume_from(struct Curl_easy *data,
- struct Curl_creader *reader,
- curl_off_t offset)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- int seekerr = CURL_SEEKFUNC_CANTSEEK;
- DEBUGASSERT(data->conn);
- /* already started reading? */
- if(ctx->read_len)
- return CURLE_READ_ERROR;
- if(data->set.seek_func) {
- Curl_set_in_callback(data, true);
- seekerr = data->set.seek_func(data->set.seek_client, offset, SEEK_SET);
- Curl_set_in_callback(data, false);
- }
- if(seekerr != CURL_SEEKFUNC_OK) {
- curl_off_t passed = 0;
- if(seekerr != CURL_SEEKFUNC_CANTSEEK) {
- failf(data, "Could not seek stream");
- return CURLE_READ_ERROR;
- }
- /* when seekerr == CURL_SEEKFUNC_CANTSEEK (cannot seek to offset) */
- do {
- char scratch[4*1024];
- size_t readthisamountnow =
- (offset - passed > (curl_off_t)sizeof(scratch)) ?
- sizeof(scratch) :
- curlx_sotouz(offset - passed);
- size_t actuallyread;
- Curl_set_in_callback(data, true);
- actuallyread = ctx->read_cb(scratch, 1, readthisamountnow,
- ctx->cb_user_data);
- Curl_set_in_callback(data, false);
- passed += actuallyread;
- if((actuallyread == 0) || (actuallyread > readthisamountnow)) {
- /* this checks for greater-than only to make sure that the
- CURL_READFUNC_ABORT return code still aborts */
- failf(data, "Could only read %" FMT_OFF_T " bytes from the input",
- passed);
- return CURLE_READ_ERROR;
- }
- } while(passed < offset);
- }
- /* now, decrease the size of the read */
- if(ctx->total_len > 0) {
- ctx->total_len -= offset;
- if(ctx->total_len <= 0) {
- failf(data, "File already completely uploaded");
- return CURLE_PARTIAL_FILE;
- }
- }
- /* we have passed, proceed as normal */
- return CURLE_OK;
- }
- static CURLcode cr_in_rewind(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- /* If we never invoked the callback, there is noting to rewind */
- if(!ctx->has_used_cb)
- return CURLE_OK;
- if(data->set.seek_func) {
- int err;
- Curl_set_in_callback(data, true);
- err = (data->set.seek_func)(data->set.seek_client, 0, SEEK_SET);
- Curl_set_in_callback(data, false);
- CURL_TRC_READ(data, "cr_in, rewind via set.seek_func -> %d", err);
- if(err) {
- failf(data, "seek callback returned error %d", (int)err);
- return CURLE_SEND_FAIL_REWIND;
- }
- }
- else if(data->set.ioctl_func) {
- curlioerr err;
- Curl_set_in_callback(data, true);
- err = (data->set.ioctl_func)(data, CURLIOCMD_RESTARTREAD,
- data->set.ioctl_client);
- Curl_set_in_callback(data, false);
- CURL_TRC_READ(data, "cr_in, rewind via set.ioctl_func -> %d", (int)err);
- if(err) {
- failf(data, "ioctl callback returned error %d", (int)err);
- return CURLE_SEND_FAIL_REWIND;
- }
- }
- else {
- /* If no CURLOPT_READFUNCTION is used, we know that we operate on a
- given FILE * stream and we can actually attempt to rewind that
- ourselves with fseek() */
- if(data->state.fread_func == (curl_read_callback)fread) {
- int err = fseek(data->state.in, 0, SEEK_SET);
- CURL_TRC_READ(data, "cr_in, rewind via fseek -> %d(%d)",
- (int)err, (int)errno);
- if(-1 != err)
- /* successful rewind */
- return CURLE_OK;
- }
- /* no callback set or failure above, makes us fail at once */
- failf(data, "necessary data rewind was not possible");
- return CURLE_SEND_FAIL_REWIND;
- }
- return CURLE_OK;
- }
- static CURLcode cr_in_unpause(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- (void)data;
- ctx->is_paused = FALSE;
- return CURLE_OK;
- }
- static bool cr_in_is_paused(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_in_ctx *ctx = reader->ctx;
- (void)data;
- return ctx->is_paused;
- }
- static const struct Curl_crtype cr_in = {
- "cr-in",
- cr_in_init,
- cr_in_read,
- Curl_creader_def_close,
- cr_in_needs_rewind,
- cr_in_total_length,
- cr_in_resume_from,
- cr_in_rewind,
- cr_in_unpause,
- cr_in_is_paused,
- Curl_creader_def_done,
- sizeof(struct cr_in_ctx)
- };
- CURLcode Curl_creader_create(struct Curl_creader **preader,
- struct Curl_easy *data,
- const struct Curl_crtype *crt,
- Curl_creader_phase phase)
- {
- struct Curl_creader *reader = NULL;
- CURLcode result = CURLE_OUT_OF_MEMORY;
- void *p;
- DEBUGASSERT(crt->creader_size >= sizeof(struct Curl_creader));
- p = calloc(1, crt->creader_size);
- if(!p)
- goto out;
- reader = (struct Curl_creader *)p;
- reader->crt = crt;
- reader->ctx = p;
- reader->phase = phase;
- result = crt->do_init(data, reader);
- out:
- *preader = result ? NULL : reader;
- if(result)
- free(reader);
- return result;
- }
- void Curl_creader_free(struct Curl_easy *data, struct Curl_creader *reader)
- {
- if(reader) {
- reader->crt->do_close(data, reader);
- free(reader);
- }
- }
- struct cr_lc_ctx {
- struct Curl_creader super;
- struct bufq buf;
- BIT(read_eos); /* we read an EOS from the next reader */
- BIT(eos); /* we have returned an EOS */
- BIT(prev_cr); /* the last byte was a CR */
- };
- static CURLcode cr_lc_init(struct Curl_easy *data, struct Curl_creader *reader)
- {
- struct cr_lc_ctx *ctx = reader->ctx;
- (void)data;
- Curl_bufq_init2(&ctx->buf, (16 * 1024), 1, BUFQ_OPT_SOFT_LIMIT);
- return CURLE_OK;
- }
- static void cr_lc_close(struct Curl_easy *data, struct Curl_creader *reader)
- {
- struct cr_lc_ctx *ctx = reader->ctx;
- (void)data;
- Curl_bufq_free(&ctx->buf);
- }
- /* client reader doing line end conversions. */
- static CURLcode cr_lc_read(struct Curl_easy *data,
- struct Curl_creader *reader,
- char *buf, size_t blen,
- size_t *pnread, bool *peos)
- {
- struct cr_lc_ctx *ctx = reader->ctx;
- CURLcode result;
- size_t nread, i, start, n;
- bool eos;
- if(ctx->eos) {
- *pnread = 0;
- *peos = TRUE;
- return CURLE_OK;
- }
- if(Curl_bufq_is_empty(&ctx->buf)) {
- if(ctx->read_eos) {
- ctx->eos = TRUE;
- *pnread = 0;
- *peos = TRUE;
- return CURLE_OK;
- }
- /* Still getting data form the next reader, ctx->buf is empty */
- result = Curl_creader_read(data, reader->next, buf, blen, &nread, &eos);
- if(result)
- return result;
- ctx->read_eos = eos;
- if(!nread || !memchr(buf, '\n', nread)) {
- /* nothing to convert, return this right away */
- if(ctx->read_eos)
- ctx->eos = TRUE;
- *pnread = nread;
- *peos = ctx->eos;
- goto out;
- }
- /* at least one \n might need conversion to '\r\n', place into ctx->buf */
- for(i = start = 0; i < nread; ++i) {
- /* if this byte is not an LF character, or if the preceding character is
- a CR (meaning this already is a CRLF pair), go to next */
- if((buf[i] != '\n') || ctx->prev_cr) {
- ctx->prev_cr = (buf[i] == '\r');
- continue;
- }
- ctx->prev_cr = false;
- /* on a soft limit bufq, we do not need to check length */
- result = Curl_bufq_cwrite(&ctx->buf, buf + start, i - start, &n);
- if(!result)
- result = Curl_bufq_cwrite(&ctx->buf, STRCONST("\r\n"), &n);
- if(result)
- return result;
- start = i + 1;
- if(!data->set.crlf && (data->state.infilesize != -1)) {
- /* we are here only because FTP is in ASCII mode...
- bump infilesize for the LF we just added */
- data->state.infilesize++;
- /* comment: this might work for FTP, but in HTTP we could not change
- * the content length after having started the request... */
- }
- }
- if(start < i) { /* leftover */
- result = Curl_bufq_cwrite(&ctx->buf, buf + start, i - start, &n);
- if(result)
- return result;
- }
- }
- DEBUGASSERT(!Curl_bufq_is_empty(&ctx->buf));
- *peos = FALSE;
- result = Curl_bufq_cread(&ctx->buf, buf, blen, pnread);
- if(!result && ctx->read_eos && Curl_bufq_is_empty(&ctx->buf)) {
- /* no more data, read all, done. */
- ctx->eos = TRUE;
- *peos = TRUE;
- }
- out:
- CURL_TRC_READ(data, "cr_lc_read(len=%zu) -> %d, nread=%zu, eos=%d",
- blen, result, *pnread, *peos);
- return result;
- }
- static curl_off_t cr_lc_total_length(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- /* this reader changes length depending on input */
- (void)data;
- (void)reader;
- return -1;
- }
- static const struct Curl_crtype cr_lc = {
- "cr-lineconv",
- cr_lc_init,
- cr_lc_read,
- cr_lc_close,
- Curl_creader_def_needs_rewind,
- cr_lc_total_length,
- Curl_creader_def_resume_from,
- Curl_creader_def_rewind,
- Curl_creader_def_unpause,
- Curl_creader_def_is_paused,
- Curl_creader_def_done,
- sizeof(struct cr_lc_ctx)
- };
- static CURLcode cr_lc_add(struct Curl_easy *data)
- {
- struct Curl_creader *reader = NULL;
- CURLcode result;
- result = Curl_creader_create(&reader, data, &cr_lc,
- CURL_CR_CONTENT_ENCODE);
- if(!result)
- result = Curl_creader_add(data, reader);
- if(result && reader)
- Curl_creader_free(data, reader);
- return result;
- }
- static CURLcode do_init_reader_stack(struct Curl_easy *data,
- struct Curl_creader *r)
- {
- CURLcode result = CURLE_OK;
- curl_off_t clen;
- DEBUGASSERT(r);
- DEBUGASSERT(r->crt);
- DEBUGASSERT(r->phase == CURL_CR_CLIENT);
- DEBUGASSERT(!data->req.reader_stack);
- data->req.reader_stack = r;
- clen = r->crt->total_length(data, r);
- /* if we do not have 0 length init, and crlf conversion is wanted,
- * add the reader for it */
- if(clen && (data->set.crlf
- #ifdef CURL_PREFER_LF_LINEENDS
- || data->state.prefer_ascii
- #endif
- )) {
- result = cr_lc_add(data);
- if(result)
- return result;
- }
- return result;
- }
- CURLcode Curl_creader_set_fread(struct Curl_easy *data, curl_off_t len)
- {
- CURLcode result;
- struct Curl_creader *r;
- struct cr_in_ctx *ctx;
- result = Curl_creader_create(&r, data, &cr_in, CURL_CR_CLIENT);
- if(result)
- goto out;
- ctx = r->ctx;
- ctx->total_len = len;
- cl_reset_reader(data);
- result = do_init_reader_stack(data, r);
- out:
- CURL_TRC_READ(data, "add fread reader, len=%"FMT_OFF_T " -> %d",
- len, result);
- return result;
- }
- CURLcode Curl_creader_add(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- CURLcode result;
- struct Curl_creader **anchor = &data->req.reader_stack;
- if(!*anchor) {
- result = Curl_creader_set_fread(data, data->state.infilesize);
- if(result)
- return result;
- }
- /* Insert the writer as first in its phase.
- * Skip existing readers of lower phases. */
- while(*anchor && (*anchor)->phase < reader->phase)
- anchor = &((*anchor)->next);
- reader->next = *anchor;
- *anchor = reader;
- return CURLE_OK;
- }
- CURLcode Curl_creader_set(struct Curl_easy *data, struct Curl_creader *r)
- {
- CURLcode result;
- DEBUGASSERT(r);
- DEBUGASSERT(r->crt);
- DEBUGASSERT(r->phase == CURL_CR_CLIENT);
- cl_reset_reader(data);
- result = do_init_reader_stack(data, r);
- if(result)
- Curl_creader_free(data, r);
- return result;
- }
- CURLcode Curl_client_read(struct Curl_easy *data, char *buf, size_t blen,
- size_t *nread, bool *eos)
- {
- CURLcode result;
- DEBUGASSERT(buf);
- DEBUGASSERT(blen);
- DEBUGASSERT(nread);
- DEBUGASSERT(eos);
- if(!data->req.reader_stack) {
- result = Curl_creader_set_fread(data, data->state.infilesize);
- if(result)
- return result;
- DEBUGASSERT(data->req.reader_stack);
- }
- result = Curl_creader_read(data, data->req.reader_stack, buf, blen,
- nread, eos);
- CURL_TRC_READ(data, "client_read(len=%zu) -> %d, nread=%zu, eos=%d",
- blen, result, *nread, *eos);
- return result;
- }
- bool Curl_creader_needs_rewind(struct Curl_easy *data)
- {
- struct Curl_creader *reader = data->req.reader_stack;
- while(reader) {
- if(reader->crt->needs_rewind(data, reader)) {
- CURL_TRC_READ(data, "client reader needs rewind before next request");
- return TRUE;
- }
- reader = reader->next;
- }
- return FALSE;
- }
- static CURLcode cr_null_read(struct Curl_easy *data,
- struct Curl_creader *reader,
- char *buf, size_t blen,
- size_t *pnread, bool *peos)
- {
- (void)data;
- (void)reader;
- (void)buf;
- (void)blen;
- *pnread = 0;
- *peos = TRUE;
- return CURLE_OK;
- }
- static curl_off_t cr_null_total_length(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- /* this reader changes length depending on input */
- (void)data;
- (void)reader;
- return 0;
- }
- static const struct Curl_crtype cr_null = {
- "cr-null",
- Curl_creader_def_init,
- cr_null_read,
- Curl_creader_def_close,
- Curl_creader_def_needs_rewind,
- cr_null_total_length,
- Curl_creader_def_resume_from,
- Curl_creader_def_rewind,
- Curl_creader_def_unpause,
- Curl_creader_def_is_paused,
- Curl_creader_def_done,
- sizeof(struct Curl_creader)
- };
- CURLcode Curl_creader_set_null(struct Curl_easy *data)
- {
- struct Curl_creader *r;
- CURLcode result;
- result = Curl_creader_create(&r, data, &cr_null, CURL_CR_CLIENT);
- if(result)
- return result;
- cl_reset_reader(data);
- return do_init_reader_stack(data, r);
- }
- struct cr_buf_ctx {
- struct Curl_creader super;
- const char *buf;
- size_t blen;
- size_t index;
- };
- static CURLcode cr_buf_read(struct Curl_easy *data,
- struct Curl_creader *reader,
- char *buf, size_t blen,
- size_t *pnread, bool *peos)
- {
- struct cr_buf_ctx *ctx = reader->ctx;
- size_t nread = ctx->blen - ctx->index;
- (void)data;
- if(!nread || !ctx->buf) {
- *pnread = 0;
- *peos = TRUE;
- }
- else {
- if(nread > blen)
- nread = blen;
- memcpy(buf, ctx->buf + ctx->index, nread);
- *pnread = nread;
- ctx->index += nread;
- *peos = (ctx->index == ctx->blen);
- }
- CURL_TRC_READ(data, "cr_buf_read(len=%zu) -> 0, nread=%zu, eos=%d",
- blen, *pnread, *peos);
- return CURLE_OK;
- }
- static bool cr_buf_needs_rewind(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_buf_ctx *ctx = reader->ctx;
- (void)data;
- return ctx->index > 0;
- }
- static curl_off_t cr_buf_total_length(struct Curl_easy *data,
- struct Curl_creader *reader)
- {
- struct cr_buf_ctx *ctx = reader->ctx;
- (void)data;
- return (curl_off_t)ctx->blen;
- }
- static CURLcode cr_buf_resume_from(struct Curl_easy *data,
- struct Curl_creader *reader,
- curl_off_t offset)
- {
- struct cr_buf_ctx *ctx = reader->ctx;
- size_t boffset;
- (void)data;
- DEBUGASSERT(data->conn);
- /* already started reading? */
- if(ctx->index)
- return CURLE_READ_ERROR;
- if(offset <= 0)
- return CURLE_OK;
- boffset = (size_t)offset;
- if(boffset > ctx->blen)
- return CURLE_READ_ERROR;
- ctx->buf += boffset;
- ctx->blen -= boffset;
- return CURLE_OK;
- }
- static const struct Curl_crtype cr_buf = {
- "cr-buf",
- Curl_creader_def_init,
- cr_buf_read,
- Curl_creader_def_close,
- cr_buf_needs_rewind,
- cr_buf_total_length,
- cr_buf_resume_from,
- Curl_creader_def_rewind,
- Curl_creader_def_unpause,
- Curl_creader_def_is_paused,
- Curl_creader_def_done,
- sizeof(struct cr_buf_ctx)
- };
- CURLcode Curl_creader_set_buf(struct Curl_easy *data,
- const char *buf, size_t blen)
- {
- CURLcode result;
- struct Curl_creader *r;
- struct cr_buf_ctx *ctx;
- result = Curl_creader_create(&r, data, &cr_buf, CURL_CR_CLIENT);
- if(result)
- goto out;
- ctx = r->ctx;
- ctx->buf = buf;
- ctx->blen = blen;
- ctx->index = 0;
- cl_reset_reader(data);
- result = do_init_reader_stack(data, r);
- out:
- CURL_TRC_READ(data, "add buf reader, len=%zu -> %d", blen, result);
- return result;
- }
- curl_off_t Curl_creader_total_length(struct Curl_easy *data)
- {
- struct Curl_creader *r = data->req.reader_stack;
- return r ? r->crt->total_length(data, r) : -1;
- }
- curl_off_t Curl_creader_client_length(struct Curl_easy *data)
- {
- struct Curl_creader *r = data->req.reader_stack;
- while(r && r->phase != CURL_CR_CLIENT)
- r = r->next;
- return r ? r->crt->total_length(data, r) : -1;
- }
- CURLcode Curl_creader_resume_from(struct Curl_easy *data, curl_off_t offset)
- {
- struct Curl_creader *r = data->req.reader_stack;
- while(r && r->phase != CURL_CR_CLIENT)
- r = r->next;
- return r ? r->crt->resume_from(data, r, offset) : CURLE_READ_ERROR;
- }
- CURLcode Curl_creader_unpause(struct Curl_easy *data)
- {
- struct Curl_creader *reader = data->req.reader_stack;
- CURLcode result = CURLE_OK;
- while(reader) {
- result = reader->crt->unpause(data, reader);
- if(result)
- break;
- reader = reader->next;
- }
- return result;
- }
- bool Curl_creader_is_paused(struct Curl_easy *data)
- {
- struct Curl_creader *reader = data->req.reader_stack;
- while(reader) {
- if(reader->crt->is_paused(data, reader))
- return TRUE;
- reader = reader->next;
- }
- return FALSE;
- }
- void Curl_creader_done(struct Curl_easy *data, int premature)
- {
- struct Curl_creader *reader = data->req.reader_stack;
- while(reader) {
- reader->crt->done(data, reader, premature);
- reader = reader->next;
- }
- }
- struct Curl_creader *Curl_creader_get_by_type(struct Curl_easy *data,
- const struct Curl_crtype *crt)
- {
- struct Curl_creader *r;
- for(r = data->req.reader_stack; r; r = r->next) {
- if(r->crt == crt)
- return r;
- }
- return NULL;
- }
|