123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765 |
- /***************************************************************************
- * _ _ ____ _
- * 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"
- #if !defined(CURL_DISABLE_PROXY) && !defined(CURL_DISABLE_HTTP)
- #include <curl/curl.h>
- #include "urldata.h"
- #include "dynbuf.h"
- #include "sendf.h"
- #include "http.h"
- #include "http1.h"
- #include "http_proxy.h"
- #include "url.h"
- #include "select.h"
- #include "progress.h"
- #include "cfilters.h"
- #include "cf-h1-proxy.h"
- #include "connect.h"
- #include "curl_trc.h"
- #include "curlx.h"
- #include "vtls/vtls.h"
- #include "transfer.h"
- #include "multiif.h"
- /* The last 3 #include files should be in this order */
- #include "curl_printf.h"
- #include "curl_memory.h"
- #include "memdebug.h"
- typedef enum {
- H1_TUNNEL_INIT, /* init/default/no tunnel state */
- H1_TUNNEL_CONNECT, /* CONNECT request is being send */
- H1_TUNNEL_RECEIVE, /* CONNECT answer is being received */
- H1_TUNNEL_RESPONSE, /* CONNECT response received completely */
- H1_TUNNEL_ESTABLISHED,
- H1_TUNNEL_FAILED
- } h1_tunnel_state;
- /* struct for HTTP CONNECT tunneling */
- struct h1_tunnel_state {
- struct dynbuf rcvbuf;
- struct dynbuf request_data;
- size_t nsent;
- size_t headerlines;
- struct Curl_chunker ch;
- enum keeponval {
- KEEPON_DONE,
- KEEPON_CONNECT,
- KEEPON_IGNORE
- } keepon;
- curl_off_t cl; /* size of content to read and ignore */
- h1_tunnel_state tunnel_state;
- BIT(chunked_encoding);
- BIT(close_connection);
- };
- static bool tunnel_is_established(struct h1_tunnel_state *ts)
- {
- return ts && (ts->tunnel_state == H1_TUNNEL_ESTABLISHED);
- }
- static bool tunnel_is_failed(struct h1_tunnel_state *ts)
- {
- return ts && (ts->tunnel_state == H1_TUNNEL_FAILED);
- }
- static CURLcode tunnel_reinit(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state *ts)
- {
- (void)data;
- (void)cf;
- DEBUGASSERT(ts);
- Curl_dyn_reset(&ts->rcvbuf);
- Curl_dyn_reset(&ts->request_data);
- ts->tunnel_state = H1_TUNNEL_INIT;
- ts->keepon = KEEPON_CONNECT;
- ts->cl = 0;
- ts->close_connection = FALSE;
- return CURLE_OK;
- }
- static CURLcode tunnel_init(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state **pts)
- {
- struct h1_tunnel_state *ts;
- if(cf->conn->handler->flags & PROTOPT_NOTCPPROXY) {
- failf(data, "%s cannot be done over CONNECT", cf->conn->handler->scheme);
- return CURLE_UNSUPPORTED_PROTOCOL;
- }
- ts = calloc(1, sizeof(*ts));
- if(!ts)
- return CURLE_OUT_OF_MEMORY;
- infof(data, "allocate connect buffer");
- Curl_dyn_init(&ts->rcvbuf, DYN_PROXY_CONNECT_HEADERS);
- Curl_dyn_init(&ts->request_data, DYN_HTTP_REQUEST);
- Curl_httpchunk_init(data, &ts->ch, TRUE);
- *pts = ts;
- connkeep(cf->conn, "HTTP proxy CONNECT");
- return tunnel_reinit(cf, data, ts);
- }
- static void h1_tunnel_go_state(struct Curl_cfilter *cf,
- struct h1_tunnel_state *ts,
- h1_tunnel_state new_state,
- struct Curl_easy *data)
- {
- if(ts->tunnel_state == new_state)
- return;
- /* entering this one */
- switch(new_state) {
- case H1_TUNNEL_INIT:
- CURL_TRC_CF(data, cf, "new tunnel state 'init'");
- tunnel_reinit(cf, data, ts);
- break;
- case H1_TUNNEL_CONNECT:
- CURL_TRC_CF(data, cf, "new tunnel state 'connect'");
- ts->tunnel_state = H1_TUNNEL_CONNECT;
- ts->keepon = KEEPON_CONNECT;
- Curl_dyn_reset(&ts->rcvbuf);
- break;
- case H1_TUNNEL_RECEIVE:
- CURL_TRC_CF(data, cf, "new tunnel state 'receive'");
- ts->tunnel_state = H1_TUNNEL_RECEIVE;
- break;
- case H1_TUNNEL_RESPONSE:
- CURL_TRC_CF(data, cf, "new tunnel state 'response'");
- ts->tunnel_state = H1_TUNNEL_RESPONSE;
- break;
- case H1_TUNNEL_ESTABLISHED:
- CURL_TRC_CF(data, cf, "new tunnel state 'established'");
- infof(data, "CONNECT phase completed");
- data->state.authproxy.done = TRUE;
- data->state.authproxy.multipass = FALSE;
- FALLTHROUGH();
- case H1_TUNNEL_FAILED:
- if(new_state == H1_TUNNEL_FAILED)
- CURL_TRC_CF(data, cf, "new tunnel state 'failed'");
- ts->tunnel_state = new_state;
- Curl_dyn_reset(&ts->rcvbuf);
- Curl_dyn_reset(&ts->request_data);
- /* restore the protocol pointer */
- data->info.httpcode = 0; /* clear it as it might've been used for the
- proxy */
- /* If a proxy-authorization header was used for the proxy, then we should
- make sure that it is not accidentally used for the document request
- after we have connected. So let's free and clear it here. */
- Curl_safefree(data->state.aptr.proxyuserpwd);
- break;
- }
- }
- static void tunnel_free(struct Curl_cfilter *cf,
- struct Curl_easy *data)
- {
- if(cf) {
- struct h1_tunnel_state *ts = cf->ctx;
- if(ts) {
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_FAILED, data);
- Curl_dyn_free(&ts->rcvbuf);
- Curl_dyn_free(&ts->request_data);
- Curl_httpchunk_free(data, &ts->ch);
- free(ts);
- cf->ctx = NULL;
- }
- }
- }
- static bool tunnel_want_send(struct h1_tunnel_state *ts)
- {
- return (ts->tunnel_state == H1_TUNNEL_CONNECT);
- }
- static CURLcode start_CONNECT(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state *ts)
- {
- struct httpreq *req = NULL;
- int http_minor;
- CURLcode result;
- /* This only happens if we have looped here due to authentication
- reasons, and we do not really use the newly cloned URL here
- then. Just free() it. */
- Curl_safefree(data->req.newurl);
- result = Curl_http_proxy_create_CONNECT(&req, cf, data, 1);
- if(result)
- goto out;
- infof(data, "Establish HTTP proxy tunnel to %s", req->authority);
- Curl_dyn_reset(&ts->request_data);
- ts->nsent = 0;
- ts->headerlines = 0;
- http_minor = (cf->conn->http_proxy.proxytype == CURLPROXY_HTTP_1_0) ? 0 : 1;
- result = Curl_h1_req_write_head(req, http_minor, &ts->request_data);
- if(!result)
- result = Curl_creader_set_null(data);
- out:
- if(result)
- failf(data, "Failed sending CONNECT to proxy");
- if(req)
- Curl_http_req_free(req);
- return result;
- }
- static CURLcode send_CONNECT(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state *ts,
- bool *done)
- {
- char *buf = Curl_dyn_ptr(&ts->request_data);
- size_t request_len = Curl_dyn_len(&ts->request_data);
- size_t blen = request_len;
- CURLcode result = CURLE_OK;
- ssize_t nwritten;
- if(blen <= ts->nsent)
- goto out; /* we are done */
- blen -= ts->nsent;
- buf += ts->nsent;
- nwritten = cf->next->cft->do_send(cf->next, data, buf, blen, FALSE, &result);
- if(nwritten < 0) {
- if(result == CURLE_AGAIN) {
- result = CURLE_OK;
- }
- goto out;
- }
- DEBUGASSERT(blen >= (size_t)nwritten);
- ts->nsent += (size_t)nwritten;
- Curl_debug(data, CURLINFO_HEADER_OUT, buf, (size_t)nwritten);
- out:
- if(result)
- failf(data, "Failed sending CONNECT to proxy");
- *done = (!result && (ts->nsent >= request_len));
- return result;
- }
- static CURLcode on_resp_header(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state *ts,
- const char *header)
- {
- CURLcode result = CURLE_OK;
- struct SingleRequest *k = &data->req;
- (void)cf;
- if((checkprefix("WWW-Authenticate:", header) &&
- (401 == k->httpcode)) ||
- (checkprefix("Proxy-authenticate:", header) &&
- (407 == k->httpcode))) {
- bool proxy = (k->httpcode == 407);
- char *auth = Curl_copy_header_value(header);
- if(!auth)
- return CURLE_OUT_OF_MEMORY;
- CURL_TRC_CF(data, cf, "CONNECT: fwd auth header '%s'", header);
- result = Curl_http_input_auth(data, proxy, auth);
- free(auth);
- if(result)
- return result;
- }
- else if(checkprefix("Content-Length:", header)) {
- if(k->httpcode/100 == 2) {
- /* A client MUST ignore any Content-Length or Transfer-Encoding
- header fields received in a successful response to CONNECT.
- "Successful" described as: 2xx (Successful). RFC 7231 4.3.6 */
- infof(data, "Ignoring Content-Length in CONNECT %03d response",
- k->httpcode);
- }
- else {
- (void)curlx_strtoofft(header + strlen("Content-Length:"),
- NULL, 10, &ts->cl);
- }
- }
- else if(Curl_compareheader(header,
- STRCONST("Connection:"), STRCONST("close")))
- ts->close_connection = TRUE;
- else if(checkprefix("Transfer-Encoding:", header)) {
- if(k->httpcode/100 == 2) {
- /* A client MUST ignore any Content-Length or Transfer-Encoding
- header fields received in a successful response to CONNECT.
- "Successful" described as: 2xx (Successful). RFC 7231 4.3.6 */
- infof(data, "Ignoring Transfer-Encoding in "
- "CONNECT %03d response", k->httpcode);
- }
- else if(Curl_compareheader(header,
- STRCONST("Transfer-Encoding:"),
- STRCONST("chunked"))) {
- infof(data, "CONNECT responded chunked");
- ts->chunked_encoding = TRUE;
- /* reset our chunky engine */
- Curl_httpchunk_reset(data, &ts->ch, TRUE);
- }
- }
- else if(Curl_compareheader(header,
- STRCONST("Proxy-Connection:"),
- STRCONST("close")))
- ts->close_connection = TRUE;
- else if(!strncmp(header, "HTTP/1.", 7) &&
- ((header[7] == '0') || (header[7] == '1')) &&
- (header[8] == ' ') &&
- ISDIGIT(header[9]) && ISDIGIT(header[10]) && ISDIGIT(header[11]) &&
- !ISDIGIT(header[12])) {
- /* store the HTTP code from the proxy */
- data->info.httpproxycode = k->httpcode = (header[9] - '0') * 100 +
- (header[10] - '0') * 10 + (header[11] - '0');
- }
- return result;
- }
- static CURLcode recv_CONNECT_resp(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state *ts,
- bool *done)
- {
- CURLcode result = CURLE_OK;
- struct SingleRequest *k = &data->req;
- char *linep;
- size_t line_len;
- int error, writetype;
- #define SELECT_OK 0
- #define SELECT_ERROR 1
- error = SELECT_OK;
- *done = FALSE;
- if(!Curl_conn_data_pending(data, cf->sockindex))
- return CURLE_OK;
- while(ts->keepon) {
- ssize_t nread;
- char byte;
- /* Read one byte at a time to avoid a race condition. Wait at most one
- second before looping to ensure continuous pgrsUpdates. */
- result = Curl_conn_recv(data, cf->sockindex, &byte, 1, &nread);
- if(result == CURLE_AGAIN)
- /* socket buffer drained, return */
- return CURLE_OK;
- if(Curl_pgrsUpdate(data))
- return CURLE_ABORTED_BY_CALLBACK;
- if(result) {
- ts->keepon = KEEPON_DONE;
- break;
- }
- if(nread <= 0) {
- if(data->set.proxyauth && data->state.authproxy.avail &&
- data->state.aptr.proxyuserpwd) {
- /* proxy auth was requested and there was proxy auth available,
- then deem this as "mere" proxy disconnect */
- ts->close_connection = TRUE;
- infof(data, "Proxy CONNECT connection closed");
- }
- else {
- error = SELECT_ERROR;
- failf(data, "Proxy CONNECT aborted");
- }
- ts->keepon = KEEPON_DONE;
- break;
- }
- if(ts->keepon == KEEPON_IGNORE) {
- /* This means we are currently ignoring a response-body */
- if(ts->cl) {
- /* A Content-Length based body: simply count down the counter
- and make sure to break out of the loop when we are done! */
- ts->cl--;
- if(ts->cl <= 0) {
- ts->keepon = KEEPON_DONE;
- break;
- }
- }
- else if(ts->chunked_encoding) {
- /* chunked-encoded body, so we need to do the chunked dance
- properly to know when the end of the body is reached */
- size_t consumed = 0;
- /* now parse the chunked piece of data so that we can
- properly tell when the stream ends */
- result = Curl_httpchunk_read(data, &ts->ch, &byte, 1, &consumed);
- if(result)
- return result;
- if(Curl_httpchunk_is_done(data, &ts->ch)) {
- /* we are done reading chunks! */
- infof(data, "chunk reading DONE");
- ts->keepon = KEEPON_DONE;
- }
- }
- continue;
- }
- if(Curl_dyn_addn(&ts->rcvbuf, &byte, 1)) {
- failf(data, "CONNECT response too large");
- return CURLE_RECV_ERROR;
- }
- /* if this is not the end of a header line then continue */
- if(byte != 0x0a)
- continue;
- ts->headerlines++;
- linep = Curl_dyn_ptr(&ts->rcvbuf);
- line_len = Curl_dyn_len(&ts->rcvbuf); /* amount of bytes in this line */
- /* output debug if that is requested */
- Curl_debug(data, CURLINFO_HEADER_IN, linep, line_len);
- /* send the header to the callback */
- writetype = CLIENTWRITE_HEADER | CLIENTWRITE_CONNECT |
- (ts->headerlines == 1 ? CLIENTWRITE_STATUS : 0);
- result = Curl_client_write(data, writetype, linep, line_len);
- if(result)
- return result;
- result = Curl_bump_headersize(data, line_len, TRUE);
- if(result)
- return result;
- /* Newlines are CRLF, so the CR is ignored as the line is not
- really terminated until the LF comes. Treat a following CR
- as end-of-headers as well.*/
- if(('\r' == linep[0]) ||
- ('\n' == linep[0])) {
- /* end of response-headers from the proxy */
- if((407 == k->httpcode) && !data->state.authproblem) {
- /* If we get a 407 response code with content length
- when we have no auth problem, we must ignore the
- whole response-body */
- ts->keepon = KEEPON_IGNORE;
- if(ts->cl) {
- infof(data, "Ignore %" FMT_OFF_T " bytes of response-body", ts->cl);
- }
- else if(ts->chunked_encoding) {
- infof(data, "Ignore chunked response-body");
- }
- else {
- /* without content-length or chunked encoding, we
- cannot keep the connection alive since the close is
- the end signal so we bail out at once instead */
- CURL_TRC_CF(data, cf, "CONNECT: no content-length or chunked");
- ts->keepon = KEEPON_DONE;
- }
- }
- else {
- ts->keepon = KEEPON_DONE;
- }
- DEBUGASSERT(ts->keepon == KEEPON_IGNORE
- || ts->keepon == KEEPON_DONE);
- continue;
- }
- result = on_resp_header(cf, data, ts, linep);
- if(result)
- return result;
- Curl_dyn_reset(&ts->rcvbuf);
- } /* while there is buffer left and loop is requested */
- if(error)
- result = CURLE_RECV_ERROR;
- *done = (ts->keepon == KEEPON_DONE);
- if(!result && *done && data->info.httpproxycode/100 != 2) {
- /* Deal with the possibly already received authenticate
- headers. 'newurl' is set to a new URL if we must loop. */
- result = Curl_http_auth_act(data);
- }
- return result;
- }
- static CURLcode H1_CONNECT(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct h1_tunnel_state *ts)
- {
- struct connectdata *conn = cf->conn;
- CURLcode result;
- bool done;
- if(tunnel_is_established(ts))
- return CURLE_OK;
- if(tunnel_is_failed(ts))
- return CURLE_RECV_ERROR; /* Need a cfilter close and new bootstrap */
- do {
- timediff_t check;
- check = Curl_timeleft(data, NULL, TRUE);
- if(check <= 0) {
- failf(data, "Proxy CONNECT aborted due to timeout");
- result = CURLE_OPERATION_TIMEDOUT;
- goto out;
- }
- switch(ts->tunnel_state) {
- case H1_TUNNEL_INIT:
- /* Prepare the CONNECT request and make a first attempt to send. */
- CURL_TRC_CF(data, cf, "CONNECT start");
- result = start_CONNECT(cf, data, ts);
- if(result)
- goto out;
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_CONNECT, data);
- FALLTHROUGH();
- case H1_TUNNEL_CONNECT:
- /* see that the request is completely sent */
- CURL_TRC_CF(data, cf, "CONNECT send");
- result = send_CONNECT(cf, data, ts, &done);
- if(result || !done)
- goto out;
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_RECEIVE, data);
- FALLTHROUGH();
- case H1_TUNNEL_RECEIVE:
- /* read what is there */
- CURL_TRC_CF(data, cf, "CONNECT receive");
- result = recv_CONNECT_resp(cf, data, ts, &done);
- if(Curl_pgrsUpdate(data)) {
- result = CURLE_ABORTED_BY_CALLBACK;
- goto out;
- }
- /* error or not complete yet. return for more multi-multi */
- if(result || !done)
- goto out;
- /* got it */
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_RESPONSE, data);
- FALLTHROUGH();
- case H1_TUNNEL_RESPONSE:
- CURL_TRC_CF(data, cf, "CONNECT response");
- if(data->req.newurl) {
- /* not the "final" response, we need to do a follow up request.
- * If the other side indicated a connection close, or if someone
- * else told us to close this connection, do so now.
- */
- Curl_req_soft_reset(&data->req, data);
- if(ts->close_connection || conn->bits.close) {
- /* Close this filter and the sub-chain, re-connect the
- * sub-chain and continue. Closing this filter will
- * reset our tunnel state. To avoid recursion, we return
- * and expect to be called again.
- */
- CURL_TRC_CF(data, cf, "CONNECT need to close+open");
- infof(data, "Connect me again please");
- Curl_conn_cf_close(cf, data);
- connkeep(conn, "HTTP proxy CONNECT");
- result = Curl_conn_cf_connect(cf->next, data, FALSE, &done);
- goto out;
- }
- else {
- /* staying on this connection, reset state */
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_INIT, data);
- }
- }
- break;
- default:
- break;
- }
- } while(data->req.newurl);
- DEBUGASSERT(ts->tunnel_state == H1_TUNNEL_RESPONSE);
- if(data->info.httpproxycode/100 != 2) {
- /* a non-2xx response and we have no next URL to try. */
- Curl_safefree(data->req.newurl);
- /* failure, close this connection to avoid reuse */
- streamclose(conn, "proxy CONNECT failure");
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_FAILED, data);
- failf(data, "CONNECT tunnel failed, response %d", data->req.httpcode);
- return CURLE_RECV_ERROR;
- }
- /* 2xx response, SUCCESS! */
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_ESTABLISHED, data);
- infof(data, "CONNECT tunnel established, response %d",
- data->info.httpproxycode);
- result = CURLE_OK;
- out:
- if(result)
- h1_tunnel_go_state(cf, ts, H1_TUNNEL_FAILED, data);
- return result;
- }
- static CURLcode cf_h1_proxy_connect(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- bool blocking, bool *done)
- {
- CURLcode result;
- struct h1_tunnel_state *ts = cf->ctx;
- if(cf->connected) {
- *done = TRUE;
- return CURLE_OK;
- }
- CURL_TRC_CF(data, cf, "connect");
- result = cf->next->cft->do_connect(cf->next, data, blocking, done);
- if(result || !*done)
- return result;
- *done = FALSE;
- if(!ts) {
- result = tunnel_init(cf, data, &ts);
- if(result)
- return result;
- cf->ctx = ts;
- }
- /* TODO: can we do blocking? */
- /* We want "seamless" operations through HTTP proxy tunnel */
- result = H1_CONNECT(cf, data, ts);
- if(result)
- goto out;
- Curl_safefree(data->state.aptr.proxyuserpwd);
- out:
- *done = (result == CURLE_OK) && tunnel_is_established(cf->ctx);
- if(*done) {
- cf->connected = TRUE;
- /* The real request will follow the CONNECT, reset request partially */
- Curl_req_soft_reset(&data->req, data);
- Curl_client_reset(data);
- Curl_pgrsSetUploadCounter(data, 0);
- Curl_pgrsSetDownloadCounter(data, 0);
- tunnel_free(cf, data);
- }
- return result;
- }
- static void cf_h1_proxy_adjust_pollset(struct Curl_cfilter *cf,
- struct Curl_easy *data,
- struct easy_pollset *ps)
- {
- struct h1_tunnel_state *ts = cf->ctx;
- if(!cf->connected) {
- /* If we are not connected, but the filter "below" is
- * and not waiting on something, we are tunneling. */
- curl_socket_t sock = Curl_conn_cf_get_socket(cf, data);
- if(ts) {
- /* when we have sent a CONNECT to a proxy, we should rather either
- wait for the socket to become readable to be able to get the
- response headers or if we are still sending the request, wait
- for write. */
- if(tunnel_want_send(ts))
- Curl_pollset_set_out_only(data, ps, sock);
- else
- Curl_pollset_set_in_only(data, ps, sock);
- }
- else
- Curl_pollset_set_out_only(data, ps, sock);
- }
- }
- static void cf_h1_proxy_destroy(struct Curl_cfilter *cf,
- struct Curl_easy *data)
- {
- CURL_TRC_CF(data, cf, "destroy");
- tunnel_free(cf, data);
- }
- static void cf_h1_proxy_close(struct Curl_cfilter *cf,
- struct Curl_easy *data)
- {
- CURL_TRC_CF(data, cf, "close");
- if(cf) {
- cf->connected = FALSE;
- if(cf->ctx) {
- h1_tunnel_go_state(cf, cf->ctx, H1_TUNNEL_INIT, data);
- }
- if(cf->next)
- cf->next->cft->do_close(cf->next, data);
- }
- }
- struct Curl_cftype Curl_cft_h1_proxy = {
- "H1-PROXY",
- CF_TYPE_IP_CONNECT|CF_TYPE_PROXY,
- 0,
- cf_h1_proxy_destroy,
- cf_h1_proxy_connect,
- cf_h1_proxy_close,
- Curl_cf_def_shutdown,
- Curl_cf_http_proxy_get_host,
- cf_h1_proxy_adjust_pollset,
- Curl_cf_def_data_pending,
- Curl_cf_def_send,
- Curl_cf_def_recv,
- Curl_cf_def_cntrl,
- Curl_cf_def_conn_is_alive,
- Curl_cf_def_conn_keep_alive,
- Curl_cf_def_query,
- };
- CURLcode Curl_cf_h1_proxy_insert_after(struct Curl_cfilter *cf_at,
- struct Curl_easy *data)
- {
- struct Curl_cfilter *cf;
- CURLcode result;
- (void)data;
- result = Curl_cf_create(&cf, &Curl_cft_h1_proxy, NULL);
- if(!result)
- Curl_conn_cf_insert_after(cf_at, cf);
- return result;
- }
- #endif /* !CURL_DISABLE_PROXY && ! CURL_DISABLE_HTTP */
|