12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115 |
- /***************************************************************************
- * _ _ ____ _
- * 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 USE_MSH3
- #include "urldata.h"
- #include "hash.h"
- #include "timeval.h"
- #include "multiif.h"
- #include "sendf.h"
- #include "curl_trc.h"
- #include "cfilters.h"
- #include "cf-socket.h"
- #include "connect.h"
- #include "progress.h"
- #include "http1.h"
- #include "curl_msh3.h"
- #include "socketpair.h"
- #include "vtls/vtls.h"
- #include "vquic/vquic.h"
- /* The last 3 #include files should be in this order */
- #include "curl_printf.h"
- #include "curl_memory.h"
- #include "memdebug.h"
- #ifdef CURL_DISABLE_SOCKETPAIR
- #error "MSH3 cannot be build with CURL_DISABLE_SOCKETPAIR set"
- #endif
- #define H3_STREAM_WINDOW_SIZE (128 * 1024)
- #define H3_STREAM_CHUNK_SIZE (16 * 1024)
- #define H3_STREAM_RECV_CHUNKS \
- (H3_STREAM_WINDOW_SIZE / H3_STREAM_CHUNK_SIZE)
- #ifdef _WIN32
- #define msh3_lock CRITICAL_SECTION
- #define msh3_lock_initialize(lock) InitializeCriticalSection(lock)
- #define msh3_lock_uninitialize(lock) DeleteCriticalSection(lock)
- #define msh3_lock_acquire(lock) EnterCriticalSection(lock)
- #define msh3_lock_release(lock) LeaveCriticalSection(lock)
- #else /* !_WIN32 */
- #include <pthread.h>
- #define msh3_lock pthread_mutex_t
- #define msh3_lock_initialize(lock) do { \
- pthread_mutexattr_t attr; \
- pthread_mutexattr_init(&attr); \
- pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE); \
- pthread_mutex_init(lock, &attr); \
- pthread_mutexattr_destroy(&attr); \
- }while(0)
- #define msh3_lock_uninitialize(lock) pthread_mutex_destroy(lock)
- #define msh3_lock_acquire(lock) pthread_mutex_lock(lock)
- #define msh3_lock_release(lock) pthread_mutex_unlock(lock)
- #endif /* _WIN32 */
- static void MSH3_CALL msh3_conn_connected(MSH3_CONNECTION *Connection,
- void *IfContext);
- static void MSH3_CALL msh3_conn_shutdown_complete(MSH3_CONNECTION *Connection,
- void *IfContext);
- static void MSH3_CALL msh3_conn_new_request(MSH3_CONNECTION *Connection,
- void *IfContext,
- MSH3_REQUEST *Request);
- static void MSH3_CALL msh3_header_received(MSH3_REQUEST *Request,
- void *IfContext,
- const MSH3_HEADER *Header);
- static bool MSH3_CALL msh3_data_received(MSH3_REQUEST *Request,
- void *IfContext, uint32_t *Length,
- const uint8_t *Data);
- static void MSH3_CALL msh3_complete(MSH3_REQUEST *Request, void *IfContext,
- bool Aborted, uint64_t AbortError);
- static void MSH3_CALL msh3_shutdown_complete(MSH3_REQUEST *Request,
- void *IfContext);
- static void MSH3_CALL msh3_data_sent(MSH3_REQUEST *Request,
- void *IfContext, void *SendContext);
- void Curl_msh3_ver(char *p, size_t len)
- {
- uint32_t v[4];
- MsH3Version(v);
- (void)msnprintf(p, len, "msh3/%d.%d.%d.%d", v[0], v[1], v[2], v[3]);
- }
- #define SP_LOCAL 0
- #define SP_REMOTE 1
- struct cf_msh3_ctx {
- MSH3_API *api;
- MSH3_CONNECTION *qconn;
- struct Curl_sockaddr_ex addr;
- curl_socket_t sock[2]; /* fake socket pair until we get support in msh3 */
- char l_ip[MAX_IPADR_LEN]; /* local IP as string */
- int l_port; /* local port number */
- struct cf_call_data call_data;
- struct curltime connect_started; /* time the current attempt started */
- struct curltime handshake_at; /* time connect handshake finished */
- struct Curl_hash streams; /* hash `data->id` to `stream_ctx` */
- /* Flags written by msh3/msquic thread */
- bool handshake_complete;
- bool handshake_succeeded;
- bool connected;
- /* Flags written by curl thread */
- BIT(verbose);
- BIT(active);
- };
- static struct cf_msh3_ctx *h3_get_msh3_ctx(struct Curl_easy *data);
- /* How to access `call_data` from a cf_msh3 filter */
- #undef CF_CTX_CALL_DATA
- #define CF_CTX_CALL_DATA(cf) \
- ((struct cf_msh3_ctx *)(cf)->ctx)->call_data
- /**
- * All about the H3 internals of a stream
- */
- struct stream_ctx {
- struct MSH3_REQUEST *req;
- struct bufq recvbuf; /* h3 response */
- #ifdef _WIN32
- CRITICAL_SECTION recv_lock;
- #else /* !_WIN32 */
- pthread_mutex_t recv_lock;
- #endif /* _WIN32 */
- uint64_t error3; /* HTTP/3 stream error code */
- int status_code; /* HTTP status code */
- CURLcode recv_error;
- bool closed;
- bool reset;
- bool upload_done;
- bool firstheader; /* FALSE until headers arrive */
- bool recv_header_complete;
- };
- #define H3_STREAM_CTX(ctx,data) ((struct stream_ctx *)((data && ctx)? \
- Curl_hash_offt_get(&(ctx)->streams, (data)->id) : NULL))
- static void h3_stream_ctx_free(struct stream_ctx *stream)
- {
- Curl_bufq_free(&stream->recvbuf);
- free(stream);
- }
- static void h3_stream_hash_free(void *stream)
- {
- DEBUGASSERT(stream);
- h3_stream_ctx_free((struct stream_ctx *)stream);
- }
- static CURLcode h3_data_setup(struct Curl_cfilter *cf,
- struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- if(stream)
- return CURLE_OK;
- stream = calloc(1, sizeof(*stream));
- if(!stream)
- return CURLE_OUT_OF_MEMORY;
- stream->req = ZERO_NULL;
- msh3_lock_initialize(&stream->recv_lock);
- Curl_bufq_init2(&stream->recvbuf, H3_STREAM_CHUNK_SIZE,
- H3_STREAM_RECV_CHUNKS, BUFQ_OPT_SOFT_LIMIT);
- CURL_TRC_CF(data, cf, "data setup");
- if(!Curl_hash_offt_set(&ctx->streams, data->id, stream)) {
- h3_stream_ctx_free(stream);
- return CURLE_OUT_OF_MEMORY;
- }
- return CURLE_OK;
- }
- static void h3_data_done(struct Curl_cfilter *cf, struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- (void)cf;
- if(stream) {
- CURL_TRC_CF(data, cf, "easy handle is done");
- Curl_hash_offt_remove(&ctx->streams, data->id);
- }
- }
- static void drain_stream_from_other_thread(struct Curl_easy *data,
- struct stream_ctx *stream)
- {
- unsigned char bits;
- /* risky */
- bits = CURL_CSELECT_IN;
- if(stream && !stream->upload_done)
- bits |= CURL_CSELECT_OUT;
- if(data->state.select_bits != bits) {
- data->state.select_bits = bits;
- /* cannot expire from other thread */
- }
- }
- static void drain_stream(struct Curl_cfilter *cf,
- struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- unsigned char bits;
- (void)cf;
- bits = CURL_CSELECT_IN;
- if(stream && !stream->upload_done)
- bits |= CURL_CSELECT_OUT;
- if(data->state.select_bits != bits) {
- data->state.select_bits = bits;
- Curl_expire(data, 0, EXPIRE_RUN_NOW);
- }
- }
- static const MSH3_CONNECTION_IF msh3_conn_if = {
- msh3_conn_connected,
- msh3_conn_shutdown_complete,
- msh3_conn_new_request
- };
- static void MSH3_CALL msh3_conn_connected(MSH3_CONNECTION *Connection,
- void *IfContext)
- {
- struct Curl_cfilter *cf = IfContext;
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct Curl_easy *data = CF_DATA_CURRENT(cf);
- (void)Connection;
- CURL_TRC_CF(data, cf, "[MSH3] connected");
- ctx->handshake_succeeded = true;
- ctx->connected = true;
- ctx->handshake_complete = true;
- }
- static void MSH3_CALL msh3_conn_shutdown_complete(MSH3_CONNECTION *Connection,
- void *IfContext)
- {
- struct Curl_cfilter *cf = IfContext;
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct Curl_easy *data = CF_DATA_CURRENT(cf);
- (void)Connection;
- CURL_TRC_CF(data, cf, "[MSH3] shutdown complete");
- ctx->connected = false;
- ctx->handshake_complete = true;
- }
- static void MSH3_CALL msh3_conn_new_request(MSH3_CONNECTION *Connection,
- void *IfContext,
- MSH3_REQUEST *Request)
- {
- (void)Connection;
- (void)IfContext;
- (void)Request;
- }
- static const MSH3_REQUEST_IF msh3_request_if = {
- msh3_header_received,
- msh3_data_received,
- msh3_complete,
- msh3_shutdown_complete,
- msh3_data_sent
- };
- /* Decode HTTP status code. Returns -1 if no valid status code was
- decoded. (duplicate from http2.c) */
- static int decode_status_code(const char *value, size_t len)
- {
- int i;
- int res;
- if(len != 3) {
- return -1;
- }
- res = 0;
- for(i = 0; i < 3; ++i) {
- char c = value[i];
- if(c < '0' || c > '9') {
- return -1;
- }
- res *= 10;
- res += c - '0';
- }
- return res;
- }
- /*
- * write_resp_raw() copies response data in raw format to the `data`'s
- * receive buffer. If not enough space is available, it appends to the
- * `data`'s overflow buffer.
- */
- static CURLcode write_resp_raw(struct Curl_easy *data,
- const void *mem, size_t memlen)
- {
- struct cf_msh3_ctx *ctx = h3_get_msh3_ctx(data);
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- CURLcode result = CURLE_OK;
- ssize_t nwritten;
- if(!stream)
- return CURLE_RECV_ERROR;
- nwritten = Curl_bufq_write(&stream->recvbuf, mem, memlen, &result);
- if(nwritten < 0) {
- return result;
- }
- if((size_t)nwritten < memlen) {
- /* This MUST not happen. Our recbuf is dimensioned to hold the
- * full max_stream_window and then some for this very reason. */
- DEBUGASSERT(0);
- return CURLE_RECV_ERROR;
- }
- return result;
- }
- static void MSH3_CALL msh3_header_received(MSH3_REQUEST *Request,
- void *userp,
- const MSH3_HEADER *hd)
- {
- struct Curl_easy *data = userp;
- struct cf_msh3_ctx *ctx = h3_get_msh3_ctx(data);
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- CURLcode result;
- (void)Request;
- DEBUGF(infof(data, "[MSH3] header received, stream=%d", !!stream));
- if(!stream || stream->recv_header_complete) {
- return;
- }
- msh3_lock_acquire(&stream->recv_lock);
- if((hd->NameLength == 7) &&
- !strncmp(HTTP_PSEUDO_STATUS, (char *)hd->Name, 7)) {
- char line[14]; /* status line is always 13 characters long */
- size_t ncopy;
- DEBUGASSERT(!stream->firstheader);
- stream->status_code = decode_status_code(hd->Value, hd->ValueLength);
- DEBUGASSERT(stream->status_code != -1);
- ncopy = msnprintf(line, sizeof(line), "HTTP/3 %03d \r\n",
- stream->status_code);
- result = write_resp_raw(data, line, ncopy);
- if(result)
- stream->recv_error = result;
- stream->firstheader = TRUE;
- }
- else {
- /* store as an HTTP1-style header */
- DEBUGASSERT(stream->firstheader);
- result = write_resp_raw(data, hd->Name, hd->NameLength);
- if(!result)
- result = write_resp_raw(data, ": ", 2);
- if(!result)
- result = write_resp_raw(data, hd->Value, hd->ValueLength);
- if(!result)
- result = write_resp_raw(data, "\r\n", 2);
- if(result) {
- stream->recv_error = result;
- }
- }
- drain_stream_from_other_thread(data, stream);
- msh3_lock_release(&stream->recv_lock);
- }
- static bool MSH3_CALL msh3_data_received(MSH3_REQUEST *Request,
- void *IfContext, uint32_t *buflen,
- const uint8_t *buf)
- {
- struct Curl_easy *data = IfContext;
- struct cf_msh3_ctx *ctx = h3_get_msh3_ctx(data);
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- CURLcode result;
- bool rv = FALSE;
- /* TODO: we would like to limit the amount of data we are buffer here.
- * There seems to be no mechanism in msh3 to adjust flow control and
- * it is undocumented what happens if we return FALSE here or less
- * length (buflen is an inout parameter).
- */
- (void)Request;
- if(!stream)
- return FALSE;
- msh3_lock_acquire(&stream->recv_lock);
- if(!stream->recv_header_complete) {
- result = write_resp_raw(data, "\r\n", 2);
- if(result) {
- stream->recv_error = result;
- goto out;
- }
- stream->recv_header_complete = true;
- }
- result = write_resp_raw(data, buf, *buflen);
- if(result) {
- stream->recv_error = result;
- }
- rv = TRUE;
- out:
- msh3_lock_release(&stream->recv_lock);
- return rv;
- }
- static void MSH3_CALL msh3_complete(MSH3_REQUEST *Request, void *IfContext,
- bool aborted, uint64_t error)
- {
- struct Curl_easy *data = IfContext;
- struct cf_msh3_ctx *ctx = h3_get_msh3_ctx(data);
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- (void)Request;
- if(!stream)
- return;
- msh3_lock_acquire(&stream->recv_lock);
- stream->closed = TRUE;
- stream->recv_header_complete = true;
- if(error)
- stream->error3 = error;
- if(aborted)
- stream->reset = TRUE;
- msh3_lock_release(&stream->recv_lock);
- }
- static void MSH3_CALL msh3_shutdown_complete(MSH3_REQUEST *Request,
- void *IfContext)
- {
- struct Curl_easy *data = IfContext;
- struct cf_msh3_ctx *ctx = h3_get_msh3_ctx(data);
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- if(!stream)
- return;
- (void)Request;
- (void)stream;
- }
- static void MSH3_CALL msh3_data_sent(MSH3_REQUEST *Request,
- void *IfContext, void *SendContext)
- {
- struct Curl_easy *data = IfContext;
- struct cf_msh3_ctx *ctx = h3_get_msh3_ctx(data);
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- if(!stream)
- return;
- (void)Request;
- (void)stream;
- (void)SendContext;
- }
- static ssize_t recv_closed_stream(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- CURLcode *err)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- ssize_t nread = -1;
- if(!stream) {
- *err = CURLE_RECV_ERROR;
- return -1;
- }
- (void)cf;
- if(stream->reset) {
- failf(data, "HTTP/3 stream reset by server");
- *err = CURLE_PARTIAL_FILE;
- CURL_TRC_CF(data, cf, "cf_recv, was reset -> %d", *err);
- goto out;
- }
- else if(stream->error3) {
- failf(data, "HTTP/3 stream was not closed cleanly: (error %zd)",
- (ssize_t)stream->error3);
- *err = CURLE_HTTP3;
- CURL_TRC_CF(data, cf, "cf_recv, closed uncleanly -> %d", *err);
- goto out;
- }
- else {
- CURL_TRC_CF(data, cf, "cf_recv, closed ok -> %d", *err);
- }
- *err = CURLE_OK;
- nread = 0;
- out:
- return nread;
- }
- static void set_quic_expire(struct Curl_cfilter *cf, struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- /* we have no indication from msh3 when it would be a good time
- * to juggle the connection again. So, we compromise by calling
- * us again every some milliseconds. */
- (void)cf;
- if(stream && stream->req && !stream->closed) {
- Curl_expire(data, 10, EXPIRE_QUIC);
- }
- else {
- Curl_expire(data, 50, EXPIRE_QUIC);
- }
- }
- static ssize_t cf_msh3_recv(struct Curl_cfilter *cf, struct Curl_easy *data,
- char *buf, size_t len, CURLcode *err)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- ssize_t nread = -1;
- struct cf_call_data save;
- CURL_TRC_CF(data, cf, "cf_recv(len=%zu), stream=%d", len, !!stream);
- if(!stream) {
- *err = CURLE_RECV_ERROR;
- return -1;
- }
- CF_DATA_SAVE(save, cf, data);
- msh3_lock_acquire(&stream->recv_lock);
- if(stream->recv_error) {
- failf(data, "request aborted");
- *err = stream->recv_error;
- goto out;
- }
- *err = CURLE_OK;
- if(!Curl_bufq_is_empty(&stream->recvbuf)) {
- nread = Curl_bufq_read(&stream->recvbuf,
- (unsigned char *)buf, len, err);
- CURL_TRC_CF(data, cf, "read recvbuf(len=%zu) -> %zd, %d",
- len, nread, *err);
- if(nread < 0)
- goto out;
- if(stream->closed)
- drain_stream(cf, data);
- }
- else if(stream->closed) {
- nread = recv_closed_stream(cf, data, err);
- goto out;
- }
- else {
- CURL_TRC_CF(data, cf, "req: nothing here, call again");
- *err = CURLE_AGAIN;
- }
- out:
- msh3_lock_release(&stream->recv_lock);
- set_quic_expire(cf, data);
- CF_DATA_RESTORE(cf, save);
- return nread;
- }
- static ssize_t cf_msh3_send(struct Curl_cfilter *cf, struct Curl_easy *data,
- const void *buf, size_t len, CURLcode *err)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- struct h1_req_parser h1;
- struct dynhds h2_headers;
- MSH3_HEADER *nva = NULL;
- size_t nheader, i;
- ssize_t nwritten = -1;
- struct cf_call_data save;
- bool eos;
- CF_DATA_SAVE(save, cf, data);
- Curl_h1_req_parse_init(&h1, H1_PARSE_DEFAULT_MAX_LINE_LEN);
- Curl_dynhds_init(&h2_headers, 0, DYN_HTTP_REQUEST);
- /* Sizes must match for cast below to work" */
- DEBUGASSERT(stream);
- CURL_TRC_CF(data, cf, "req: send %zu bytes", len);
- if(!stream->req) {
- /* The first send on the request contains the headers and possibly some
- data. Parse out the headers and create the request, then if there is
- any data left over go ahead and send it too. */
- nwritten = Curl_h1_req_parse_read(&h1, buf, len, NULL, 0, err);
- if(nwritten < 0)
- goto out;
- DEBUGASSERT(h1.done);
- DEBUGASSERT(h1.req);
- *err = Curl_http_req_to_h2(&h2_headers, h1.req, data);
- if(*err) {
- nwritten = -1;
- goto out;
- }
- nheader = Curl_dynhds_count(&h2_headers);
- nva = malloc(sizeof(MSH3_HEADER) * nheader);
- if(!nva) {
- *err = CURLE_OUT_OF_MEMORY;
- nwritten = -1;
- goto out;
- }
- for(i = 0; i < nheader; ++i) {
- struct dynhds_entry *e = Curl_dynhds_getn(&h2_headers, i);
- nva[i].Name = e->name;
- nva[i].NameLength = e->namelen;
- nva[i].Value = e->value;
- nva[i].ValueLength = e->valuelen;
- }
- switch(data->state.httpreq) {
- case HTTPREQ_POST:
- case HTTPREQ_POST_FORM:
- case HTTPREQ_POST_MIME:
- case HTTPREQ_PUT:
- /* known request body size or -1 */
- eos = FALSE;
- break;
- default:
- /* there is not request body */
- eos = TRUE;
- stream->upload_done = TRUE;
- break;
- }
- CURL_TRC_CF(data, cf, "req: send %zu headers", nheader);
- stream->req = MsH3RequestOpen(ctx->qconn, &msh3_request_if, data,
- nva, nheader,
- eos ? MSH3_REQUEST_FLAG_FIN :
- MSH3_REQUEST_FLAG_NONE);
- if(!stream->req) {
- failf(data, "request open failed");
- *err = CURLE_SEND_ERROR;
- goto out;
- }
- *err = CURLE_OK;
- nwritten = len;
- goto out;
- }
- else {
- /* request is open */
- CURL_TRC_CF(data, cf, "req: send %zu body bytes", len);
- if(len > 0xFFFFFFFF) {
- len = 0xFFFFFFFF;
- }
- if(!MsH3RequestSend(stream->req, MSH3_REQUEST_FLAG_NONE, buf,
- (uint32_t)len, stream)) {
- *err = CURLE_SEND_ERROR;
- goto out;
- }
- /* TODO - msh3/msquic will hold onto this memory until the send complete
- event. How do we make sure curl doesn't free it until then? */
- *err = CURLE_OK;
- nwritten = len;
- }
- out:
- set_quic_expire(cf, data);
- free(nva);
- Curl_h1_req_parse_free(&h1);
- Curl_dynhds_free(&h2_headers);
- CF_DATA_RESTORE(cf, save);
- return nwritten;
- }
- static void cf_msh3_adjust_pollset(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct easy_pollset *ps)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- struct cf_call_data save;
- CF_DATA_SAVE(save, cf, data);
- if(stream && ctx->sock[SP_LOCAL] != CURL_SOCKET_BAD) {
- if(stream->recv_error) {
- Curl_pollset_add_in(data, ps, ctx->sock[SP_LOCAL]);
- drain_stream(cf, data);
- }
- else if(stream->req) {
- Curl_pollset_add_out(data, ps, ctx->sock[SP_LOCAL]);
- drain_stream(cf, data);
- }
- }
- }
- static bool cf_msh3_data_pending(struct Curl_cfilter *cf,
- const struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- struct cf_call_data save;
- bool pending = FALSE;
- CF_DATA_SAVE(save, cf, data);
- (void)cf;
- if(stream && stream->req) {
- msh3_lock_acquire(&stream->recv_lock);
- CURL_TRC_CF((struct Curl_easy *)data, cf, "data pending = %zu",
- Curl_bufq_len(&stream->recvbuf));
- pending = !Curl_bufq_is_empty(&stream->recvbuf);
- msh3_lock_release(&stream->recv_lock);
- if(pending)
- drain_stream(cf, (struct Curl_easy *)data);
- }
- CF_DATA_RESTORE(cf, save);
- return pending;
- }
- static CURLcode h3_data_pause(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- bool pause)
- {
- if(!pause) {
- drain_stream(cf, data);
- Curl_expire(data, 0, EXPIRE_RUN_NOW);
- }
- return CURLE_OK;
- }
- static CURLcode cf_msh3_data_event(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- int event, int arg1, void *arg2)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct stream_ctx *stream = H3_STREAM_CTX(ctx, data);
- struct cf_call_data save;
- CURLcode result = CURLE_OK;
- CF_DATA_SAVE(save, cf, data);
- (void)arg1;
- (void)arg2;
- switch(event) {
- case CF_CTRL_DATA_SETUP:
- result = h3_data_setup(cf, data);
- break;
- case CF_CTRL_DATA_PAUSE:
- result = h3_data_pause(cf, data, (arg1 != 0));
- break;
- case CF_CTRL_DATA_DONE:
- h3_data_done(cf, data);
- break;
- case CF_CTRL_DATA_DONE_SEND:
- CURL_TRC_CF(data, cf, "req: send done");
- if(stream) {
- stream->upload_done = TRUE;
- if(stream->req) {
- char buf[1];
- if(!MsH3RequestSend(stream->req, MSH3_REQUEST_FLAG_FIN,
- buf, 0, data)) {
- result = CURLE_SEND_ERROR;
- }
- }
- }
- break;
- default:
- break;
- }
- CF_DATA_RESTORE(cf, save);
- return result;
- }
- static CURLcode cf_connect_start(struct Curl_cfilter *cf,
- struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct ssl_primary_config *conn_config;
- MSH3_ADDR addr = {0};
- CURLcode result;
- bool verify;
- Curl_hash_offt_init(&ctx->streams, 63, h3_stream_hash_free);
- conn_config = Curl_ssl_cf_get_primary_config(cf);
- if(!conn_config)
- return CURLE_FAILED_INIT;
- verify = !!conn_config->verifypeer;
- memcpy(&addr, &ctx->addr.sa_addr, ctx->addr.addrlen);
- MSH3_SET_PORT(&addr, (uint16_t)cf->conn->remote_port);
- if(verify && (conn_config->CAfile || conn_config->CApath)) {
- /* TODO: need a way to provide trust anchors to MSH3 */
- #ifdef DEBUGBUILD
- /* we need this for our test cases to run */
- CURL_TRC_CF(data, cf, "non-standard CA not supported, "
- "switching off verifypeer in DEBUG mode");
- verify = 0;
- #else
- CURL_TRC_CF(data, cf, "non-standard CA not supported, "
- "attempting with built-in verification");
- #endif
- }
- CURL_TRC_CF(data, cf, "connecting to %s:%d (verify=%d)",
- cf->conn->host.name, (int)cf->conn->remote_port, verify);
- ctx->api = MsH3ApiOpen();
- if(!ctx->api) {
- failf(data, "can't create msh3 api");
- return CURLE_FAILED_INIT;
- }
- ctx->qconn = MsH3ConnectionOpen(ctx->api,
- &msh3_conn_if,
- cf,
- cf->conn->host.name,
- &addr,
- !verify);
- if(!ctx->qconn) {
- failf(data, "can't create msh3 connection");
- if(ctx->api) {
- MsH3ApiClose(ctx->api);
- ctx->api = NULL;
- }
- return CURLE_FAILED_INIT;
- }
- result = h3_data_setup(cf, data);
- if(result)
- return result;
- return CURLE_OK;
- }
- static CURLcode cf_msh3_connect(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- bool blocking, bool *done)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct cf_call_data save;
- CURLcode result = CURLE_OK;
- (void)blocking;
- if(cf->connected) {
- *done = TRUE;
- return CURLE_OK;
- }
- CF_DATA_SAVE(save, cf, data);
- if(ctx->sock[SP_LOCAL] == CURL_SOCKET_BAD) {
- if(Curl_socketpair(AF_UNIX, SOCK_STREAM, 0, &ctx->sock[0]) < 0) {
- ctx->sock[SP_LOCAL] = CURL_SOCKET_BAD;
- ctx->sock[SP_REMOTE] = CURL_SOCKET_BAD;
- return CURLE_COULDNT_CONNECT;
- }
- }
- *done = FALSE;
- if(!ctx->qconn) {
- ctx->connect_started = Curl_now();
- result = cf_connect_start(cf, data);
- if(result)
- goto out;
- }
- if(ctx->handshake_complete) {
- ctx->handshake_at = Curl_now();
- if(ctx->handshake_succeeded) {
- CURL_TRC_CF(data, cf, "handshake succeeded");
- cf->conn->bits.multiplex = TRUE; /* at least potentially multiplexed */
- cf->conn->httpversion = 30;
- cf->conn->bundle->multiuse = BUNDLE_MULTIPLEX;
- cf->connected = TRUE;
- cf->conn->alpn = CURL_HTTP_VERSION_3;
- *done = TRUE;
- connkeep(cf->conn, "HTTP/3 default");
- Curl_pgrsTime(data, TIMER_APPCONNECT);
- }
- else {
- failf(data, "failed to connect, handshake failed");
- result = CURLE_COULDNT_CONNECT;
- }
- }
- out:
- CF_DATA_RESTORE(cf, save);
- return result;
- }
- static void cf_msh3_close(struct Curl_cfilter *cf, struct Curl_easy *data)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- struct cf_call_data save;
- (void)data;
- CF_DATA_SAVE(save, cf, data);
- if(ctx) {
- CURL_TRC_CF(data, cf, "destroying");
- if(ctx->qconn) {
- MsH3ConnectionClose(ctx->qconn);
- ctx->qconn = NULL;
- }
- if(ctx->api) {
- MsH3ApiClose(ctx->api);
- ctx->api = NULL;
- }
- Curl_hash_destroy(&ctx->streams);
- if(ctx->active) {
- /* We share our socket at cf->conn->sock[cf->sockindex] when active.
- * If it is no longer there, someone has stolen (and hopefully
- * closed it) and we just forget about it.
- */
- ctx->active = FALSE;
- if(ctx->sock[SP_LOCAL] == cf->conn->sock[cf->sockindex]) {
- CURL_TRC_CF(data, cf, "cf_msh3_close(%d) active",
- (int)ctx->sock[SP_LOCAL]);
- cf->conn->sock[cf->sockindex] = CURL_SOCKET_BAD;
- }
- else {
- CURL_TRC_CF(data, cf, "cf_socket_close(%d) no longer at "
- "conn->sock[], discarding", (int)ctx->sock[SP_LOCAL]);
- ctx->sock[SP_LOCAL] = CURL_SOCKET_BAD;
- }
- if(cf->sockindex == FIRSTSOCKET)
- cf->conn->remote_addr = NULL;
- }
- if(ctx->sock[SP_LOCAL] != CURL_SOCKET_BAD) {
- sclose(ctx->sock[SP_LOCAL]);
- }
- if(ctx->sock[SP_REMOTE] != CURL_SOCKET_BAD) {
- sclose(ctx->sock[SP_REMOTE]);
- }
- ctx->sock[SP_LOCAL] = CURL_SOCKET_BAD;
- ctx->sock[SP_REMOTE] = CURL_SOCKET_BAD;
- }
- CF_DATA_RESTORE(cf, save);
- }
- static void cf_msh3_destroy(struct Curl_cfilter *cf, struct Curl_easy *data)
- {
- struct cf_call_data save;
- CF_DATA_SAVE(save, cf, data);
- cf_msh3_close(cf, data);
- free(cf->ctx);
- cf->ctx = NULL;
- /* no CF_DATA_RESTORE(cf, save); its gone */
- }
- static CURLcode cf_msh3_query(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- int query, int *pres1, void *pres2)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- switch(query) {
- case CF_QUERY_MAX_CONCURRENT: {
- /* TODO: we do not have access to this so far, fake it */
- (void)ctx;
- *pres1 = 100;
- return CURLE_OK;
- }
- case CF_QUERY_TIMER_CONNECT: {
- struct curltime *when = pres2;
- /* we do not know when the first byte arrived */
- if(cf->connected)
- *when = ctx->handshake_at;
- return CURLE_OK;
- }
- case CF_QUERY_TIMER_APPCONNECT: {
- struct curltime *when = pres2;
- if(cf->connected)
- *when = ctx->handshake_at;
- return CURLE_OK;
- }
- default:
- break;
- }
- return cf->next?
- cf->next->cft->query(cf->next, data, query, pres1, pres2) :
- CURLE_UNKNOWN_OPTION;
- }
- static bool cf_msh3_conn_is_alive(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- bool *input_pending)
- {
- struct cf_msh3_ctx *ctx = cf->ctx;
- (void)data;
- *input_pending = FALSE;
- return ctx && ctx->sock[SP_LOCAL] != CURL_SOCKET_BAD && ctx->qconn &&
- ctx->connected;
- }
- struct Curl_cftype Curl_cft_http3 = {
- "HTTP/3",
- CF_TYPE_IP_CONNECT | CF_TYPE_SSL | CF_TYPE_MULTIPLEX,
- 0,
- cf_msh3_destroy,
- cf_msh3_connect,
- cf_msh3_close,
- Curl_cf_def_get_host,
- cf_msh3_adjust_pollset,
- cf_msh3_data_pending,
- cf_msh3_send,
- cf_msh3_recv,
- cf_msh3_data_event,
- cf_msh3_conn_is_alive,
- Curl_cf_def_conn_keep_alive,
- cf_msh3_query,
- };
- static struct cf_msh3_ctx *h3_get_msh3_ctx(struct Curl_easy *data)
- {
- if(data && data->conn) {
- struct Curl_cfilter *cf = data->conn->cfilter[FIRSTSOCKET];
- while(cf) {
- if(cf->cft == &Curl_cft_http3)
- return cf->ctx;
- cf = cf->next;
- }
- }
- DEBUGF(infof(data, "no filter context found"));
- return NULL;
- }
- CURLcode Curl_cf_msh3_create(struct Curl_cfilter **pcf,
- struct Curl_easy *data,
- struct connectdata *conn,
- const struct Curl_addrinfo *ai)
- {
- struct cf_msh3_ctx *ctx = NULL;
- struct Curl_cfilter *cf = NULL;
- CURLcode result;
- (void)data;
- (void)conn;
- (void)ai; /* TODO: msh3 resolves itself? */
- ctx = calloc(1, sizeof(*ctx));
- if(!ctx) {
- result = CURLE_OUT_OF_MEMORY;
- goto out;
- }
- Curl_sock_assign_addr(&ctx->addr, ai, TRNSPRT_QUIC);
- ctx->sock[SP_LOCAL] = CURL_SOCKET_BAD;
- ctx->sock[SP_REMOTE] = CURL_SOCKET_BAD;
- result = Curl_cf_create(&cf, &Curl_cft_http3, ctx);
- out:
- *pcf = (!result)? cf : NULL;
- if(result) {
- Curl_safefree(cf);
- Curl_safefree(ctx);
- }
- return result;
- }
- bool Curl_conn_is_msh3(const struct Curl_easy *data,
- const struct connectdata *conn,
- int sockindex)
- {
- struct Curl_cfilter *cf = conn? conn->cfilter[sockindex] : NULL;
- (void)data;
- for(; cf; cf = cf->next) {
- if(cf->cft == &Curl_cft_http3)
- return TRUE;
- if(cf->cft->flags & CF_TYPE_IP_CONNECT)
- return FALSE;
- }
- return FALSE;
- }
- #endif /* USE_MSH3 */
|