|
|
@@ -37,6 +37,7 @@
|
|
|
#include "url.h"
|
|
|
#include "select.h"
|
|
|
#include "progress.h"
|
|
|
+#include "cfilters.h"
|
|
|
#include "connect.h"
|
|
|
#include "curlx.h"
|
|
|
#include "vtls/vtls.h"
|
|
|
@@ -48,179 +49,192 @@
|
|
|
#include "curl_memory.h"
|
|
|
#include "memdebug.h"
|
|
|
|
|
|
-/*
|
|
|
- * Perform SSL initialization for HTTPS proxy. Sets
|
|
|
- * proxy_ssl_connected connection bit when complete. Can be
|
|
|
- * called multiple times.
|
|
|
- */
|
|
|
-static CURLcode https_proxy_connect(struct Curl_easy *data, int sockindex)
|
|
|
+typedef enum {
|
|
|
+ TUNNEL_INIT, /* init/default/no tunnel state */
|
|
|
+ TUNNEL_CONNECT, /* CONNECT request is being send */
|
|
|
+ TUNNEL_RECEIVE, /* CONNECT answer is being received */
|
|
|
+ TUNNEL_RESPONSE, /* CONNECT response received completely */
|
|
|
+ TUNNEL_ESTABLISHED,
|
|
|
+ TUNNEL_FAILED
|
|
|
+} tunnel_state;
|
|
|
+
|
|
|
+/* struct for HTTP CONNECT tunneling */
|
|
|
+struct tunnel_state {
|
|
|
+ int sockindex;
|
|
|
+ const char *hostname;
|
|
|
+ int remote_port;
|
|
|
+ struct HTTP http_proxy;
|
|
|
+ struct HTTP *prot_save;
|
|
|
+ struct dynbuf rcvbuf;
|
|
|
+ struct dynbuf req;
|
|
|
+ size_t nsend;
|
|
|
+ size_t headerlines;
|
|
|
+ enum keeponval {
|
|
|
+ KEEPON_DONE,
|
|
|
+ KEEPON_CONNECT,
|
|
|
+ KEEPON_IGNORE
|
|
|
+ } keepon;
|
|
|
+ curl_off_t cl; /* size of content to read and ignore */
|
|
|
+ tunnel_state tunnel_state;
|
|
|
+ BIT(chunked_encoding);
|
|
|
+ BIT(close_connection);
|
|
|
+};
|
|
|
+
|
|
|
+
|
|
|
+static bool tunnel_is_established(struct tunnel_state *ts)
|
|
|
{
|
|
|
-#ifdef USE_SSL
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
- CURLcode result = CURLE_OK;
|
|
|
- DEBUGASSERT(conn->http_proxy.proxytype == CURLPROXY_HTTPS);
|
|
|
- if(!conn->bits.proxy_ssl_connected[sockindex]) {
|
|
|
- /* perform SSL initialization for this socket */
|
|
|
- result =
|
|
|
- Curl_ssl_connect_nonblocking(data, conn, TRUE, sockindex,
|
|
|
- &conn->bits.proxy_ssl_connected[sockindex]);
|
|
|
- if(result)
|
|
|
- /* a failed connection is marked for closure to prevent (bad) re-use or
|
|
|
- similar */
|
|
|
- connclose(conn, "TLS handshake failed");
|
|
|
- }
|
|
|
- return result;
|
|
|
-#else
|
|
|
- (void) data;
|
|
|
- (void) sockindex;
|
|
|
- return CURLE_NOT_BUILT_IN;
|
|
|
-#endif
|
|
|
+ return ts && (ts->tunnel_state == TUNNEL_ESTABLISHED);
|
|
|
}
|
|
|
|
|
|
-CURLcode Curl_proxy_connect(struct Curl_easy *data, int sockindex)
|
|
|
+static bool tunnel_is_failed(struct tunnel_state *ts)
|
|
|
{
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
- if(conn->http_proxy.proxytype == CURLPROXY_HTTPS) {
|
|
|
- const CURLcode result = https_proxy_connect(data, sockindex);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
- if(!conn->bits.proxy_ssl_connected[sockindex])
|
|
|
- return result; /* wait for HTTPS proxy SSL initialization to complete */
|
|
|
- }
|
|
|
-
|
|
|
- if(conn->bits.tunnel_proxy && conn->bits.httpproxy) {
|
|
|
-#ifndef CURL_DISABLE_PROXY
|
|
|
- /* for [protocol] tunneled through HTTP proxy */
|
|
|
- const char *hostname;
|
|
|
- int remote_port;
|
|
|
- CURLcode result;
|
|
|
-
|
|
|
- /* We want "seamless" operations through HTTP proxy tunnel */
|
|
|
-
|
|
|
- /* for the secondary socket (FTP), use the "connect to host"
|
|
|
- * but ignore the "connect to port" (use the secondary port)
|
|
|
- */
|
|
|
-
|
|
|
- if(conn->bits.conn_to_host)
|
|
|
- hostname = conn->conn_to_host.name;
|
|
|
- else if(sockindex == SECONDARYSOCKET)
|
|
|
- hostname = conn->secondaryhostname;
|
|
|
- else
|
|
|
- hostname = conn->host.name;
|
|
|
-
|
|
|
- if(sockindex == SECONDARYSOCKET)
|
|
|
- remote_port = conn->secondary_port;
|
|
|
- else if(conn->bits.conn_to_port)
|
|
|
- remote_port = conn->conn_to_port;
|
|
|
- else
|
|
|
- remote_port = conn->remote_port;
|
|
|
-
|
|
|
- result = Curl_proxyCONNECT(data, sockindex, hostname, remote_port);
|
|
|
- if(CURLE_OK != result)
|
|
|
- return result;
|
|
|
- Curl_safefree(data->state.aptr.proxyuserpwd);
|
|
|
-#else
|
|
|
- return CURLE_NOT_BUILT_IN;
|
|
|
-#endif
|
|
|
- }
|
|
|
- /* no HTTP tunnel proxy, just return */
|
|
|
- return CURLE_OK;
|
|
|
+ return ts && (ts->tunnel_state == TUNNEL_FAILED);
|
|
|
}
|
|
|
|
|
|
-bool Curl_connect_complete(struct connectdata *conn)
|
|
|
+static CURLcode tunnel_reinit(struct tunnel_state *ts,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct Curl_easy *data)
|
|
|
{
|
|
|
- return !conn->connect_state ||
|
|
|
- (conn->connect_state->tunnel_state >= TUNNEL_COMPLETE);
|
|
|
-}
|
|
|
+ (void)data;
|
|
|
+ DEBUGASSERT(ts);
|
|
|
+ Curl_dyn_reset(&ts->rcvbuf);
|
|
|
+ Curl_dyn_reset(&ts->req);
|
|
|
+ ts->tunnel_state = TUNNEL_INIT;
|
|
|
+ ts->keepon = KEEPON_CONNECT;
|
|
|
+ ts->cl = 0;
|
|
|
+ ts->close_connection = FALSE;
|
|
|
+
|
|
|
+ if(conn->bits.conn_to_host)
|
|
|
+ ts->hostname = conn->conn_to_host.name;
|
|
|
+ else if(ts->sockindex == SECONDARYSOCKET)
|
|
|
+ ts->hostname = conn->secondaryhostname;
|
|
|
+ else
|
|
|
+ ts->hostname = conn->host.name;
|
|
|
+
|
|
|
+ if(ts->sockindex == SECONDARYSOCKET)
|
|
|
+ ts->remote_port = conn->secondary_port;
|
|
|
+ else if(conn->bits.conn_to_port)
|
|
|
+ ts->remote_port = conn->conn_to_port;
|
|
|
+ else
|
|
|
+ ts->remote_port = conn->remote_port;
|
|
|
|
|
|
-bool Curl_connect_ongoing(struct connectdata *conn)
|
|
|
-{
|
|
|
- return conn->connect_state &&
|
|
|
- (conn->connect_state->tunnel_state <= TUNNEL_COMPLETE);
|
|
|
+ return CURLE_OK;
|
|
|
}
|
|
|
|
|
|
-/* when we've 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're still sending the request, wait for write. */
|
|
|
-int Curl_connect_getsock(struct connectdata *conn)
|
|
|
+static CURLcode tunnel_init(struct tunnel_state **pts,
|
|
|
+ struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ int sockindex)
|
|
|
{
|
|
|
- struct HTTP *http;
|
|
|
- DEBUGASSERT(conn);
|
|
|
- DEBUGASSERT(conn->connect_state);
|
|
|
- http = &conn->connect_state->http_proxy;
|
|
|
-
|
|
|
- if(http->sending == HTTPSEND_REQUEST)
|
|
|
- return GETSOCK_WRITESOCK(0);
|
|
|
-
|
|
|
- return GETSOCK_READSOCK(0);
|
|
|
-}
|
|
|
+ struct tunnel_state *ts;
|
|
|
+ CURLcode result;
|
|
|
|
|
|
-static CURLcode connect_init(struct Curl_easy *data, bool reinit)
|
|
|
-{
|
|
|
- struct http_connect_state *s;
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
if(conn->handler->flags & PROTOPT_NOTCPPROXY) {
|
|
|
failf(data, "%s cannot be done over CONNECT", conn->handler->scheme);
|
|
|
return CURLE_UNSUPPORTED_PROTOCOL;
|
|
|
}
|
|
|
- if(!reinit) {
|
|
|
- CURLcode result;
|
|
|
- DEBUGASSERT(!conn->connect_state);
|
|
|
- /* we might need the upload buffer for streaming a partial request */
|
|
|
- result = Curl_get_upload_buffer(data);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
|
|
|
- s = calloc(1, sizeof(struct http_connect_state));
|
|
|
- if(!s)
|
|
|
- return CURLE_OUT_OF_MEMORY;
|
|
|
- infof(data, "allocate connect buffer");
|
|
|
- conn->connect_state = s;
|
|
|
- Curl_dyn_init(&s->rcvbuf, DYN_PROXY_CONNECT_HEADERS);
|
|
|
-
|
|
|
- /* Curl_proxyCONNECT is based on a pointer to a struct HTTP at the
|
|
|
- * member conn->proto.http; we want [protocol] through HTTP and we have
|
|
|
- * to change the member temporarily for connecting to the HTTP
|
|
|
- * proxy. After Curl_proxyCONNECT we have to set back the member to the
|
|
|
- * original pointer
|
|
|
- *
|
|
|
- * This function might be called several times in the multi interface case
|
|
|
- * if the proxy's CONNECT response is not instant.
|
|
|
- */
|
|
|
- s->prot_save = data->req.p.http;
|
|
|
- data->req.p.http = &s->http_proxy;
|
|
|
- connkeep(conn, "HTTP proxy CONNECT");
|
|
|
- }
|
|
|
- else {
|
|
|
- DEBUGASSERT(conn->connect_state);
|
|
|
- s = conn->connect_state;
|
|
|
- Curl_dyn_reset(&s->rcvbuf);
|
|
|
- }
|
|
|
- s->tunnel_state = TUNNEL_INIT;
|
|
|
- s->keepon = KEEPON_CONNECT;
|
|
|
- s->cl = 0;
|
|
|
- s->close_connection = FALSE;
|
|
|
- return CURLE_OK;
|
|
|
+ /* we might need the upload buffer for streaming a partial request */
|
|
|
+ result = Curl_get_upload_buffer(data);
|
|
|
+ if(result)
|
|
|
+ return result;
|
|
|
+
|
|
|
+ ts = calloc(1, sizeof(*ts));
|
|
|
+ if(!ts)
|
|
|
+ return CURLE_OUT_OF_MEMORY;
|
|
|
+
|
|
|
+ ts->sockindex = sockindex;
|
|
|
+ infof(data, "allocate connect buffer");
|
|
|
+
|
|
|
+ Curl_dyn_init(&ts->rcvbuf, DYN_PROXY_CONNECT_HEADERS);
|
|
|
+ Curl_dyn_init(&ts->req, DYN_HTTP_REQUEST);
|
|
|
+
|
|
|
+ /* Curl_proxyCONNECT is based on a pointer to a struct HTTP at the
|
|
|
+ * member conn->proto.http; we want [protocol] through HTTP and we have
|
|
|
+ * to change the member temporarily for connecting to the HTTP
|
|
|
+ * proxy. After Curl_proxyCONNECT we have to set back the member to the
|
|
|
+ * original pointer
|
|
|
+ *
|
|
|
+ * This function might be called several times in the multi interface case
|
|
|
+ * if the proxy's CONNECT response is not instant.
|
|
|
+ */
|
|
|
+ ts->prot_save = data->req.p.http;
|
|
|
+ data->req.p.http = &ts->http_proxy;
|
|
|
+ *pts = ts;
|
|
|
+ connkeep(conn, "HTTP proxy CONNECT");
|
|
|
+ return tunnel_reinit(ts, conn, data);
|
|
|
}
|
|
|
|
|
|
-void Curl_connect_done(struct Curl_easy *data)
|
|
|
+static void tunnel_go_state(struct Curl_cfilter *cf,
|
|
|
+ struct tunnel_state *ts,
|
|
|
+ tunnel_state new_state,
|
|
|
+ struct Curl_easy *data)
|
|
|
{
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
- struct http_connect_state *s = conn->connect_state;
|
|
|
- if(s && (s->tunnel_state != TUNNEL_EXIT)) {
|
|
|
- s->tunnel_state = TUNNEL_EXIT;
|
|
|
- Curl_dyn_free(&s->rcvbuf);
|
|
|
- Curl_dyn_free(&s->req);
|
|
|
+ if(ts->tunnel_state == new_state)
|
|
|
+ return;
|
|
|
+ /* leaving this one */
|
|
|
+ switch(ts->tunnel_state) {
|
|
|
+ case TUNNEL_CONNECT:
|
|
|
+ data->req.ignorebody = FALSE;
|
|
|
+ break;
|
|
|
+ default:
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ /* entering this one */
|
|
|
+ switch(new_state) {
|
|
|
+ case TUNNEL_INIT:
|
|
|
+ tunnel_reinit(ts, cf->conn, data);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case TUNNEL_CONNECT:
|
|
|
+ ts->tunnel_state = TUNNEL_CONNECT;
|
|
|
+ ts->keepon = KEEPON_CONNECT;
|
|
|
+ Curl_dyn_reset(&ts->rcvbuf);
|
|
|
+ break;
|
|
|
+
|
|
|
+ case TUNNEL_RECEIVE:
|
|
|
+ ts->tunnel_state = TUNNEL_RECEIVE;
|
|
|
+ break;
|
|
|
+
|
|
|
+ case TUNNEL_RESPONSE:
|
|
|
+ ts->tunnel_state = TUNNEL_RESPONSE;
|
|
|
+ break;
|
|
|
|
|
|
+ case TUNNEL_ESTABLISHED:
|
|
|
+ infof(data, "CONNECT phase completed");
|
|
|
+ data->state.authproxy.done = TRUE;
|
|
|
+ data->state.authproxy.multipass = FALSE;
|
|
|
+ /* FALLTHROUGH */
|
|
|
+ case TUNNEL_FAILED:
|
|
|
+ ts->tunnel_state = new_state;
|
|
|
+ Curl_dyn_reset(&ts->rcvbuf);
|
|
|
+ Curl_dyn_reset(&ts->req);
|
|
|
/* restore the protocol pointer */
|
|
|
- data->req.p.http = s->prot_save;
|
|
|
+ data->req.p.http = ts->prot_save;
|
|
|
data->info.httpcode = 0; /* clear it as it might've been used for the
|
|
|
proxy */
|
|
|
- data->req.ignorebody = FALSE;
|
|
|
+ /* If a proxy-authorization header was used for the proxy, then we should
|
|
|
+ make sure that it isn't accidentally used for the document request
|
|
|
+ after we've connected. So let's free and clear it here. */
|
|
|
+ Curl_safefree(data->state.aptr.proxyuserpwd);
|
|
|
+ data->state.aptr.proxyuserpwd = NULL;
|
|
|
#ifdef USE_HYPER
|
|
|
data->state.hconnect = FALSE;
|
|
|
#endif
|
|
|
- infof(data, "CONNECT phase completed");
|
|
|
+ break;
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static void tunnel_free(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data)
|
|
|
+{
|
|
|
+ struct tunnel_state *ts = cf->ctx;
|
|
|
+ if(ts) {
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_FAILED, data);
|
|
|
+ Curl_dyn_free(&ts->rcvbuf);
|
|
|
+ Curl_dyn_free(&ts->req);
|
|
|
+ free(ts);
|
|
|
+ cf->ctx = NULL;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@@ -257,821 +271,1013 @@ static CURLcode CONNECT_host(struct Curl_easy *data,
|
|
|
}
|
|
|
|
|
|
#ifndef USE_HYPER
|
|
|
-static CURLcode CONNECT(struct Curl_easy *data,
|
|
|
- int sockindex,
|
|
|
- const char *hostname,
|
|
|
- int remote_port)
|
|
|
+static CURLcode start_CONNECT(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct tunnel_state *ts)
|
|
|
{
|
|
|
- int subversion = 0;
|
|
|
- struct SingleRequest *k = &data->req;
|
|
|
+ char *hostheader = NULL;
|
|
|
+ char *host = NULL;
|
|
|
+ const char *httpv;
|
|
|
CURLcode result;
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
- curl_socket_t tunnelsocket = conn->sock[sockindex];
|
|
|
- struct http_connect_state *s = conn->connect_state;
|
|
|
- struct HTTP *http = data->req.p.http;
|
|
|
- char *linep;
|
|
|
- size_t perline;
|
|
|
|
|
|
-#define SELECT_OK 0
|
|
|
-#define SELECT_ERROR 1
|
|
|
+ infof(data, "Establish HTTP proxy tunnel to %s:%d",
|
|
|
+ ts->hostname, ts->remote_port);
|
|
|
+
|
|
|
+ /* This only happens if we've looped here due to authentication
|
|
|
+ reasons, and we don't really use the newly cloned URL here
|
|
|
+ then. Just free() it. */
|
|
|
+ Curl_safefree(data->req.newurl);
|
|
|
+
|
|
|
+ result = CONNECT_host(data, conn,
|
|
|
+ ts->hostname, ts->remote_port,
|
|
|
+ &hostheader, &host);
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ /* Setup the proxy-authorization header, if any */
|
|
|
+ result = Curl_http_output_auth(data, conn, "CONNECT", HTTPREQ_GET,
|
|
|
+ hostheader, TRUE);
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ httpv = (conn->http_proxy.proxytype == CURLPROXY_HTTP_1_0) ? "1.0" : "1.1";
|
|
|
+
|
|
|
+ result =
|
|
|
+ Curl_dyn_addf(&ts->req,
|
|
|
+ "CONNECT %s HTTP/%s\r\n"
|
|
|
+ "%s" /* Host: */
|
|
|
+ "%s", /* Proxy-Authorization */
|
|
|
+ hostheader,
|
|
|
+ httpv,
|
|
|
+ host?host:"",
|
|
|
+ data->state.aptr.proxyuserpwd?
|
|
|
+ data->state.aptr.proxyuserpwd:"");
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ if(!Curl_checkProxyheaders(data, conn, STRCONST("User-Agent"))
|
|
|
+ && data->set.str[STRING_USERAGENT])
|
|
|
+ result = Curl_dyn_addf(&ts->req, "User-Agent: %s\r\n",
|
|
|
+ data->set.str[STRING_USERAGENT]);
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ if(!Curl_checkProxyheaders(data, conn, STRCONST("Proxy-Connection")))
|
|
|
+ result = Curl_dyn_addn(&ts->req,
|
|
|
+ STRCONST("Proxy-Connection: Keep-Alive\r\n"));
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ result = Curl_add_custom_headers(data, TRUE, &ts->req);
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ /* CRLF terminate the request */
|
|
|
+ result = Curl_dyn_addn(&ts->req, STRCONST("\r\n"));
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+
|
|
|
+ /* Send the connect request to the proxy */
|
|
|
+ result = Curl_buffer_send(&ts->req, data, &data->info.request_size, 0,
|
|
|
+ ts->sockindex);
|
|
|
+ ts->headerlines = 0;
|
|
|
+
|
|
|
+out:
|
|
|
+ if(result)
|
|
|
+ failf(data, "Failed sending CONNECT to proxy");
|
|
|
+ free(host);
|
|
|
+ free(hostheader);
|
|
|
+ return result;
|
|
|
+}
|
|
|
|
|
|
- if(Curl_connect_complete(conn))
|
|
|
- return CURLE_OK; /* CONNECT is already completed */
|
|
|
+static CURLcode send_CONNECT(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct tunnel_state *ts,
|
|
|
+ bool *done)
|
|
|
+{
|
|
|
+ struct SingleRequest *k = &data->req;
|
|
|
+ struct HTTP *http = data->req.p.http;
|
|
|
+ CURLcode result = CURLE_OK;
|
|
|
|
|
|
- conn->bits.proxy_connect_closed = FALSE;
|
|
|
+ if(http->sending != HTTPSEND_REQUEST)
|
|
|
+ goto out;
|
|
|
|
|
|
- do {
|
|
|
- timediff_t check;
|
|
|
- if(TUNNEL_INIT == s->tunnel_state) {
|
|
|
- /* BEGIN CONNECT PHASE */
|
|
|
- struct dynbuf *req = &s->req;
|
|
|
- char *hostheader = NULL;
|
|
|
- char *host = NULL;
|
|
|
+ if(!ts->nsend) {
|
|
|
+ size_t fillcount;
|
|
|
+ k->upload_fromhere = data->state.ulbuf;
|
|
|
+ result = Curl_fillreadbuffer(data, data->set.upload_buffer_size,
|
|
|
+ &fillcount);
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+ ts->nsend = fillcount;
|
|
|
+ }
|
|
|
+ if(ts->nsend) {
|
|
|
+ ssize_t bytes_written;
|
|
|
+ /* write to socket (send away data) */
|
|
|
+ result = Curl_write(data,
|
|
|
+ conn->writesockfd, /* socket to send to */
|
|
|
+ k->upload_fromhere, /* buffer pointer */
|
|
|
+ ts->nsend, /* buffer size */
|
|
|
+ &bytes_written); /* actually sent */
|
|
|
+ if(result)
|
|
|
+ goto out;
|
|
|
+ /* send to debug callback! */
|
|
|
+ Curl_debug(data, CURLINFO_HEADER_OUT,
|
|
|
+ k->upload_fromhere, bytes_written);
|
|
|
|
|
|
- infof(data, "Establish HTTP proxy tunnel to %s:%d",
|
|
|
- hostname, remote_port);
|
|
|
+ ts->nsend -= bytes_written;
|
|
|
+ k->upload_fromhere += bytes_written;
|
|
|
+ }
|
|
|
+ if(!ts->nsend)
|
|
|
+ http->sending = HTTPSEND_NADA;
|
|
|
|
|
|
- /* This only happens if we've looped here due to authentication
|
|
|
- reasons, and we don't really use the newly cloned URL here
|
|
|
- then. Just free() it. */
|
|
|
- Curl_safefree(data->req.newurl);
|
|
|
+out:
|
|
|
+ if(result)
|
|
|
+ failf(data, "Failed sending CONNECT to proxy");
|
|
|
+ *done = (http->sending != HTTPSEND_REQUEST);
|
|
|
+ return result;
|
|
|
+}
|
|
|
|
|
|
- /* initialize send-buffer */
|
|
|
- Curl_dyn_init(req, DYN_HTTP_REQUEST);
|
|
|
+static CURLcode on_resp_header(struct Curl_easy *data,
|
|
|
+ struct tunnel_state *ts,
|
|
|
+ const char *header)
|
|
|
+{
|
|
|
+ CURLcode result = CURLE_OK;
|
|
|
+ struct SingleRequest *k = &data->req;
|
|
|
+ int subversion = 0;
|
|
|
|
|
|
- result = CONNECT_host(data, conn,
|
|
|
- hostname, remote_port, &hostheader, &host);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
+ if((checkprefix("WWW-Authenticate:", header) &&
|
|
|
+ (401 == k->httpcode)) ||
|
|
|
+ (checkprefix("Proxy-authenticate:", header) &&
|
|
|
+ (407 == k->httpcode))) {
|
|
|
|
|
|
- /* Setup the proxy-authorization header, if any */
|
|
|
- result = Curl_http_output_auth(data, conn, "CONNECT", HTTPREQ_GET,
|
|
|
- hostheader, TRUE);
|
|
|
-
|
|
|
- if(!result) {
|
|
|
- const char *httpv =
|
|
|
- (conn->http_proxy.proxytype == CURLPROXY_HTTP_1_0) ? "1.0" : "1.1";
|
|
|
-
|
|
|
- result =
|
|
|
- Curl_dyn_addf(req,
|
|
|
- "CONNECT %s HTTP/%s\r\n"
|
|
|
- "%s" /* Host: */
|
|
|
- "%s", /* Proxy-Authorization */
|
|
|
- hostheader,
|
|
|
- httpv,
|
|
|
- host?host:"",
|
|
|
- data->state.aptr.proxyuserpwd?
|
|
|
- data->state.aptr.proxyuserpwd:"");
|
|
|
-
|
|
|
- if(!result && !Curl_checkProxyheaders(data,
|
|
|
- conn, STRCONST("User-Agent")) &&
|
|
|
- data->set.str[STRING_USERAGENT])
|
|
|
- result = Curl_dyn_addf(req, "User-Agent: %s\r\n",
|
|
|
- data->set.str[STRING_USERAGENT]);
|
|
|
-
|
|
|
- if(!result && !Curl_checkProxyheaders(data, conn,
|
|
|
- STRCONST("Proxy-Connection")))
|
|
|
- result = Curl_dyn_addn(req,
|
|
|
- STRCONST("Proxy-Connection: Keep-Alive\r\n"));
|
|
|
-
|
|
|
- if(!result)
|
|
|
- result = Curl_add_custom_headers(data, TRUE, req);
|
|
|
-
|
|
|
- if(!result)
|
|
|
- /* CRLF terminate the request */
|
|
|
- result = Curl_dyn_addn(req, STRCONST("\r\n"));
|
|
|
-
|
|
|
- if(!result) {
|
|
|
- /* Send the connect request to the proxy */
|
|
|
- result = Curl_buffer_send(req, data, &data->info.request_size, 0,
|
|
|
- sockindex);
|
|
|
- s->headerlines = 0;
|
|
|
- }
|
|
|
- if(result)
|
|
|
- failf(data, "Failed sending CONNECT to proxy");
|
|
|
- }
|
|
|
- free(host);
|
|
|
- free(hostheader);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
+ bool proxy = (k->httpcode == 407) ? TRUE : FALSE;
|
|
|
+ char *auth = Curl_copy_header_value(header);
|
|
|
+ if(!auth)
|
|
|
+ return CURLE_OUT_OF_MEMORY;
|
|
|
|
|
|
- s->tunnel_state = TUNNEL_CONNECT;
|
|
|
- } /* END CONNECT PHASE */
|
|
|
+ DEBUGF(infof(data, "CONNECT: fwd auth header '%s'",
|
|
|
+ header));
|
|
|
+ result = Curl_http_input_auth(data, proxy, auth);
|
|
|
|
|
|
- check = Curl_timeleft(data, NULL, TRUE);
|
|
|
- if(check <= 0) {
|
|
|
- failf(data, "Proxy CONNECT aborted due to timeout");
|
|
|
- return CURLE_OPERATION_TIMEDOUT;
|
|
|
+ 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;
|
|
|
+ /* init our chunky engine */
|
|
|
+ Curl_httpchunk_init(data);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ else if(Curl_compareheader(header,
|
|
|
+ STRCONST("Proxy-Connection:"),
|
|
|
+ STRCONST("close")))
|
|
|
+ ts->close_connection = TRUE;
|
|
|
+ else if(2 == sscanf(header, "HTTP/1.%d %d",
|
|
|
+ &subversion,
|
|
|
+ &k->httpcode)) {
|
|
|
+ /* store the HTTP code from the proxy */
|
|
|
+ data->info.httpproxycode = k->httpcode;
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+}
|
|
|
+
|
|
|
+static CURLcode recv_CONNECT_resp(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct tunnel_state *ts,
|
|
|
+ bool *done)
|
|
|
+{
|
|
|
+ CURLcode result = CURLE_OK;
|
|
|
+ struct SingleRequest *k = &data->req;
|
|
|
+ curl_socket_t tunnelsocket = conn->sock[ts->sockindex];
|
|
|
+ char *linep;
|
|
|
+ size_t perline;
|
|
|
+ int error;
|
|
|
+
|
|
|
+#define SELECT_OK 0
|
|
|
+#define SELECT_ERROR 1
|
|
|
+
|
|
|
+ error = SELECT_OK;
|
|
|
+ *done = FALSE;
|
|
|
+
|
|
|
+ if(!Curl_conn_data_pending(data, ts->sockindex))
|
|
|
+ return CURLE_OK;
|
|
|
+
|
|
|
+ while(ts->keepon) {
|
|
|
+ ssize_t gotbytes;
|
|
|
+ char byte;
|
|
|
|
|
|
- if(!Curl_conn_data_pending(conn, sockindex) && !http->sending)
|
|
|
- /* return so we'll be called again polling-style */
|
|
|
+ /* Read one byte at a time to avoid a race condition. Wait at most one
|
|
|
+ second before looping to ensure continuous pgrsUpdates. */
|
|
|
+ result = Curl_read(data, tunnelsocket, &byte, 1, &gotbytes);
|
|
|
+ if(result == CURLE_AGAIN)
|
|
|
+ /* socket buffer drained, return */
|
|
|
return CURLE_OK;
|
|
|
|
|
|
- /* at this point, the tunnel_connecting phase is over. */
|
|
|
-
|
|
|
- if(http->sending == HTTPSEND_REQUEST) {
|
|
|
- if(!s->nsend) {
|
|
|
- size_t fillcount;
|
|
|
- k->upload_fromhere = data->state.ulbuf;
|
|
|
- result = Curl_fillreadbuffer(data, data->set.upload_buffer_size,
|
|
|
- &fillcount);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
- s->nsend = fillcount;
|
|
|
+ if(Curl_pgrsUpdate(data))
|
|
|
+ return CURLE_ABORTED_BY_CALLBACK;
|
|
|
+
|
|
|
+ if(result) {
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ if(gotbytes <= 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");
|
|
|
}
|
|
|
- if(s->nsend) {
|
|
|
- ssize_t bytes_written;
|
|
|
- /* write to socket (send away data) */
|
|
|
- result = Curl_write(data,
|
|
|
- conn->writesockfd, /* socket to send to */
|
|
|
- k->upload_fromhere, /* buffer pointer */
|
|
|
- s->nsend, /* buffer size */
|
|
|
- &bytes_written); /* actually sent */
|
|
|
-
|
|
|
- if(!result)
|
|
|
- /* send to debug callback! */
|
|
|
- Curl_debug(data, CURLINFO_HEADER_OUT,
|
|
|
- k->upload_fromhere, bytes_written);
|
|
|
-
|
|
|
- s->nsend -= bytes_written;
|
|
|
- k->upload_fromhere += bytes_written;
|
|
|
- return result;
|
|
|
+ else {
|
|
|
+ error = SELECT_ERROR;
|
|
|
+ failf(data, "Proxy CONNECT aborted");
|
|
|
}
|
|
|
- http->sending = HTTPSEND_NADA;
|
|
|
- /* if nothing left to send, continue */
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
+ break;
|
|
|
}
|
|
|
- { /* READING RESPONSE PHASE */
|
|
|
- int error = SELECT_OK;
|
|
|
-
|
|
|
- while(s->keepon) {
|
|
|
- ssize_t gotbytes;
|
|
|
- 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_read(data, tunnelsocket, &byte, 1, &gotbytes);
|
|
|
- if(result == CURLE_AGAIN)
|
|
|
- /* socket buffer drained, return */
|
|
|
- return CURLE_OK;
|
|
|
|
|
|
- if(Curl_pgrsUpdate(data))
|
|
|
- return CURLE_ABORTED_BY_CALLBACK;
|
|
|
+ if(ts->keepon == KEEPON_IGNORE) {
|
|
|
+ /* This means we are currently ignoring a response-body */
|
|
|
|
|
|
- if(result) {
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
+ if(ts->cl) {
|
|
|
+ /* A Content-Length based body: simply count down the counter
|
|
|
+ and make sure to break out of the loop when we're done! */
|
|
|
+ ts->cl--;
|
|
|
+ if(ts->cl <= 0) {
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
break;
|
|
|
}
|
|
|
- else if(gotbytes <= 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 */
|
|
|
- conn->bits.proxy_connect_closed = TRUE;
|
|
|
- infof(data, "Proxy CONNECT connection closed");
|
|
|
- }
|
|
|
- else {
|
|
|
- error = SELECT_ERROR;
|
|
|
- failf(data, "Proxy CONNECT aborted");
|
|
|
- }
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
- break;
|
|
|
- }
|
|
|
-
|
|
|
- if(s->keepon == KEEPON_IGNORE) {
|
|
|
- /* This means we are currently ignoring a response-body */
|
|
|
-
|
|
|
- if(s->cl) {
|
|
|
- /* A Content-Length based body: simply count down the counter
|
|
|
- and make sure to break out of the loop when we're done! */
|
|
|
- s->cl--;
|
|
|
- if(s->cl <= 0) {
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
- s->tunnel_state = TUNNEL_COMPLETE;
|
|
|
- break;
|
|
|
- }
|
|
|
- }
|
|
|
- else {
|
|
|
- /* chunked-encoded body, so we need to do the chunked dance
|
|
|
- properly to know when the end of the body is reached */
|
|
|
- CHUNKcode r;
|
|
|
- CURLcode extra;
|
|
|
- ssize_t tookcareof = 0;
|
|
|
-
|
|
|
- /* now parse the chunked piece of data so that we can
|
|
|
- properly tell when the stream ends */
|
|
|
- r = Curl_httpchunk_read(data, &byte, 1, &tookcareof, &extra);
|
|
|
- if(r == CHUNKE_STOP) {
|
|
|
- /* we're done reading chunks! */
|
|
|
- infof(data, "chunk reading DONE");
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
- /* we did the full CONNECT treatment, go COMPLETE */
|
|
|
- s->tunnel_state = TUNNEL_COMPLETE;
|
|
|
- }
|
|
|
- }
|
|
|
- continue;
|
|
|
- }
|
|
|
-
|
|
|
- if(Curl_dyn_addn(&s->rcvbuf, &byte, 1)) {
|
|
|
- failf(data, "CONNECT response too large");
|
|
|
- return CURLE_RECV_ERROR;
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ /* chunked-encoded body, so we need to do the chunked dance
|
|
|
+ properly to know when the end of the body is reached */
|
|
|
+ CHUNKcode r;
|
|
|
+ CURLcode extra;
|
|
|
+ ssize_t tookcareof = 0;
|
|
|
+
|
|
|
+ /* now parse the chunked piece of data so that we can
|
|
|
+ properly tell when the stream ends */
|
|
|
+ r = Curl_httpchunk_read(data, &byte, 1, &tookcareof, &extra);
|
|
|
+ if(r == CHUNKE_STOP) {
|
|
|
+ /* we're done reading chunks! */
|
|
|
+ infof(data, "chunk reading DONE");
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
}
|
|
|
+ }
|
|
|
+ continue;
|
|
|
+ }
|
|
|
|
|
|
- /* if this is not the end of a header line then continue */
|
|
|
- if(byte != 0x0a)
|
|
|
- continue;
|
|
|
-
|
|
|
- s->headerlines++;
|
|
|
- linep = Curl_dyn_ptr(&s->rcvbuf);
|
|
|
- perline = Curl_dyn_len(&s->rcvbuf); /* amount of bytes in this line */
|
|
|
-
|
|
|
- /* output debug if that is requested */
|
|
|
- Curl_debug(data, CURLINFO_HEADER_IN, linep, perline);
|
|
|
+ if(Curl_dyn_addn(&ts->rcvbuf, &byte, 1)) {
|
|
|
+ failf(data, "CONNECT response too large");
|
|
|
+ return CURLE_RECV_ERROR;
|
|
|
+ }
|
|
|
|
|
|
- if(!data->set.suppress_connect_headers) {
|
|
|
- /* send the header to the callback */
|
|
|
- int writetype = CLIENTWRITE_HEADER | CLIENTWRITE_CONNECT |
|
|
|
- (data->set.include_header ? CLIENTWRITE_BODY : 0) |
|
|
|
- (s->headerlines == 1 ? CLIENTWRITE_STATUS : 0);
|
|
|
+ /* if this is not the end of a header line then continue */
|
|
|
+ if(byte != 0x0a)
|
|
|
+ continue;
|
|
|
|
|
|
- result = Curl_client_write(data, writetype, linep, perline);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
- }
|
|
|
+ ts->headerlines++;
|
|
|
+ linep = Curl_dyn_ptr(&ts->rcvbuf);
|
|
|
+ perline = Curl_dyn_len(&ts->rcvbuf); /* amount of bytes in this line */
|
|
|
|
|
|
- data->info.header_size += (long)perline;
|
|
|
-
|
|
|
- /* Newlines are CRLF, so the CR is ignored as the line isn't
|
|
|
- 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 */
|
|
|
- s->keepon = KEEPON_IGNORE;
|
|
|
-
|
|
|
- if(s->cl) {
|
|
|
- infof(data, "Ignore %" CURL_FORMAT_CURL_OFF_T
|
|
|
- " bytes of response-body", s->cl);
|
|
|
- }
|
|
|
- else if(s->chunked_encoding) {
|
|
|
- CHUNKcode r;
|
|
|
- CURLcode extra;
|
|
|
-
|
|
|
- infof(data, "Ignore chunked response-body");
|
|
|
-
|
|
|
- /* We set ignorebody true here since the chunked decoder
|
|
|
- function will acknowledge that. Pay attention so that this is
|
|
|
- cleared again when this function returns! */
|
|
|
- k->ignorebody = TRUE;
|
|
|
-
|
|
|
- if(linep[1] == '\n')
|
|
|
- /* this can only be a LF if the letter at index 0 was a CR */
|
|
|
- linep++;
|
|
|
-
|
|
|
- /* now parse the chunked piece of data so that we can properly
|
|
|
- tell when the stream ends */
|
|
|
- r = Curl_httpchunk_read(data, linep + 1, 1, &gotbytes,
|
|
|
- &extra);
|
|
|
- if(r == CHUNKE_STOP) {
|
|
|
- /* we're done reading chunks! */
|
|
|
- infof(data, "chunk reading DONE");
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
- /* we did the full CONNECT treatment, go to COMPLETE */
|
|
|
- s->tunnel_state = TUNNEL_COMPLETE;
|
|
|
- }
|
|
|
- }
|
|
|
- else {
|
|
|
- /* without content-length or chunked encoding, we
|
|
|
- can't keep the connection alive since the close is
|
|
|
- the end signal so we bail out at once instead */
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
- }
|
|
|
- }
|
|
|
- else
|
|
|
- s->keepon = KEEPON_DONE;
|
|
|
+ /* output debug if that is requested */
|
|
|
+ Curl_debug(data, CURLINFO_HEADER_IN, linep, perline);
|
|
|
|
|
|
- if(s->keepon == KEEPON_DONE && !s->cl)
|
|
|
- /* we did the full CONNECT treatment, go to COMPLETE */
|
|
|
- s->tunnel_state = TUNNEL_COMPLETE;
|
|
|
+ if(!data->set.suppress_connect_headers) {
|
|
|
+ /* send the header to the callback */
|
|
|
+ int writetype = CLIENTWRITE_HEADER | CLIENTWRITE_CONNECT |
|
|
|
+ (data->set.include_header ? CLIENTWRITE_BODY : 0) |
|
|
|
+ (ts->headerlines == 1 ? CLIENTWRITE_STATUS : 0);
|
|
|
|
|
|
- DEBUGASSERT(s->keepon == KEEPON_IGNORE || s->keepon == KEEPON_DONE);
|
|
|
- continue;
|
|
|
- }
|
|
|
+ result = Curl_client_write(data, writetype, linep, perline);
|
|
|
+ if(result)
|
|
|
+ return result;
|
|
|
+ }
|
|
|
|
|
|
- if((checkprefix("WWW-Authenticate:", linep) &&
|
|
|
- (401 == k->httpcode)) ||
|
|
|
- (checkprefix("Proxy-authenticate:", linep) &&
|
|
|
- (407 == k->httpcode))) {
|
|
|
+ data->info.header_size += (long)perline;
|
|
|
|
|
|
- bool proxy = (k->httpcode == 407) ? TRUE : FALSE;
|
|
|
- char *auth = Curl_copy_header_value(linep);
|
|
|
- if(!auth)
|
|
|
- return CURLE_OUT_OF_MEMORY;
|
|
|
+ /* Newlines are CRLF, so the CR is ignored as the line isn't
|
|
|
+ really terminated until the LF comes. Treat a following CR
|
|
|
+ as end-of-headers as well.*/
|
|
|
|
|
|
- result = Curl_http_input_auth(data, proxy, auth);
|
|
|
+ if(('\r' == linep[0]) ||
|
|
|
+ ('\n' == linep[0])) {
|
|
|
+ /* end of response-headers from the proxy */
|
|
|
|
|
|
- free(auth);
|
|
|
+ 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(result)
|
|
|
- return result;
|
|
|
+ if(ts->cl) {
|
|
|
+ infof(data, "Ignore %" CURL_FORMAT_CURL_OFF_T
|
|
|
+ " bytes of response-body", ts->cl);
|
|
|
}
|
|
|
- else if(checkprefix("Content-Length:", linep)) {
|
|
|
- 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(linep +
|
|
|
- strlen("Content-Length:"), NULL, 10, &s->cl);
|
|
|
+ else if(ts->chunked_encoding) {
|
|
|
+ CHUNKcode r;
|
|
|
+ CURLcode extra;
|
|
|
+
|
|
|
+ infof(data, "Ignore chunked response-body");
|
|
|
+
|
|
|
+ /* We set ignorebody true here since the chunked decoder
|
|
|
+ function will acknowledge that. Pay attention so that this is
|
|
|
+ cleared again when this function returns! */
|
|
|
+ k->ignorebody = TRUE;
|
|
|
+
|
|
|
+ if(linep[1] == '\n')
|
|
|
+ /* this can only be a LF if the letter at index 0 was a CR */
|
|
|
+ linep++;
|
|
|
+
|
|
|
+ /* now parse the chunked piece of data so that we can properly
|
|
|
+ tell when the stream ends */
|
|
|
+ r = Curl_httpchunk_read(data, linep + 1, 1, &gotbytes,
|
|
|
+ &extra);
|
|
|
+ if(r == CHUNKE_STOP) {
|
|
|
+ /* we're done reading chunks! */
|
|
|
+ infof(data, "chunk reading DONE");
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
}
|
|
|
}
|
|
|
- else if(Curl_compareheader(linep,
|
|
|
- STRCONST("Connection:"), STRCONST("close")))
|
|
|
- s->close_connection = TRUE;
|
|
|
- else if(checkprefix("Transfer-Encoding:", linep)) {
|
|
|
- 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(linep,
|
|
|
- STRCONST("Transfer-Encoding:"),
|
|
|
- STRCONST("chunked"))) {
|
|
|
- infof(data, "CONNECT responded chunked");
|
|
|
- s->chunked_encoding = TRUE;
|
|
|
- /* init our chunky engine */
|
|
|
- Curl_httpchunk_init(data);
|
|
|
- }
|
|
|
+ else {
|
|
|
+ /* without content-length or chunked encoding, we
|
|
|
+ can't keep the connection alive since the close is
|
|
|
+ the end signal so we bail out at once instead */
|
|
|
+ DEBUGF(infof(data, "CONNECT: no content-length or chunked"));
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
}
|
|
|
- else if(Curl_compareheader(linep,
|
|
|
- STRCONST("Proxy-Connection:"),
|
|
|
- STRCONST("close")))
|
|
|
- s->close_connection = TRUE;
|
|
|
- else if(2 == sscanf(linep, "HTTP/1.%d %d",
|
|
|
- &subversion,
|
|
|
- &k->httpcode)) {
|
|
|
- /* store the HTTP code from the proxy */
|
|
|
- data->info.httpproxycode = k->httpcode;
|
|
|
- }
|
|
|
-
|
|
|
- Curl_dyn_reset(&s->rcvbuf);
|
|
|
- } /* while there's buffer left and loop is requested */
|
|
|
-
|
|
|
- if(Curl_pgrsUpdate(data))
|
|
|
- return CURLE_ABORTED_BY_CALLBACK;
|
|
|
-
|
|
|
- if(error)
|
|
|
- return CURLE_RECV_ERROR;
|
|
|
-
|
|
|
- if(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);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
-
|
|
|
- if(conn->bits.close)
|
|
|
- /* the connection has been marked for closure, most likely in the
|
|
|
- Curl_http_auth_act() function and thus we can kill it at once
|
|
|
- below */
|
|
|
- s->close_connection = TRUE;
|
|
|
}
|
|
|
-
|
|
|
- if(s->close_connection && data->req.newurl) {
|
|
|
- /* Connection closed by server. Don't use it anymore */
|
|
|
- Curl_closesocket(data, conn, conn->sock[sockindex]);
|
|
|
- conn->sock[sockindex] = CURL_SOCKET_BAD;
|
|
|
- break;
|
|
|
+ else {
|
|
|
+ ts->keepon = KEEPON_DONE;
|
|
|
}
|
|
|
- } /* END READING RESPONSE PHASE */
|
|
|
|
|
|
- /* If we are supposed to continue and request a new URL, which basically
|
|
|
- * means the HTTP authentication is still going on so if the tunnel
|
|
|
- * is complete we start over in INIT state */
|
|
|
- if(data->req.newurl && (TUNNEL_COMPLETE == s->tunnel_state)) {
|
|
|
- connect_init(data, TRUE); /* reinit */
|
|
|
+ DEBUGASSERT(ts->keepon == KEEPON_IGNORE
|
|
|
+ || ts->keepon == KEEPON_DONE);
|
|
|
+ continue;
|
|
|
}
|
|
|
|
|
|
- } while(data->req.newurl);
|
|
|
-
|
|
|
- if(data->info.httpproxycode/100 != 2) {
|
|
|
- if(s->close_connection && data->req.newurl) {
|
|
|
- conn->bits.proxy_connect_closed = TRUE;
|
|
|
- infof(data, "Connect me again please");
|
|
|
- Curl_connect_done(data);
|
|
|
- }
|
|
|
- else {
|
|
|
- free(data->req.newurl);
|
|
|
- data->req.newurl = NULL;
|
|
|
- /* failure, close this connection to avoid re-use */
|
|
|
- streamclose(conn, "proxy CONNECT failure");
|
|
|
- }
|
|
|
+ result = on_resp_header(data, ts, linep);
|
|
|
+ if(result)
|
|
|
+ return result;
|
|
|
|
|
|
- /* to back to init state */
|
|
|
- s->tunnel_state = TUNNEL_INIT;
|
|
|
+ Curl_dyn_reset(&ts->rcvbuf);
|
|
|
+ } /* while there's buffer left and loop is requested */
|
|
|
|
|
|
- if(conn->bits.proxy_connect_closed)
|
|
|
- /* this is not an error, just part of the connection negotiation */
|
|
|
- return CURLE_OK;
|
|
|
- Curl_dyn_free(&s->rcvbuf);
|
|
|
- failf(data, "Received HTTP code %d from proxy after CONNECT",
|
|
|
- data->req.httpcode);
|
|
|
- return CURLE_RECV_ERROR;
|
|
|
+ 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);
|
|
|
}
|
|
|
-
|
|
|
- s->tunnel_state = TUNNEL_COMPLETE;
|
|
|
-
|
|
|
- /* If a proxy-authorization header was used for the proxy, then we should
|
|
|
- make sure that it isn't accidentally used for the document request
|
|
|
- after we've connected. So let's free and clear it here. */
|
|
|
- Curl_safefree(data->state.aptr.proxyuserpwd);
|
|
|
- data->state.aptr.proxyuserpwd = NULL;
|
|
|
-
|
|
|
- data->state.authproxy.done = TRUE;
|
|
|
- data->state.authproxy.multipass = FALSE;
|
|
|
-
|
|
|
- infof(data, "Proxy replied %d to CONNECT request",
|
|
|
- data->info.httpproxycode);
|
|
|
- data->req.ignorebody = FALSE; /* put it (back) to non-ignore state */
|
|
|
- conn->bits.rewindaftersend = FALSE; /* make sure this isn't set for the
|
|
|
- document request */
|
|
|
- Curl_dyn_free(&s->rcvbuf);
|
|
|
- return CURLE_OK;
|
|
|
+ return result;
|
|
|
}
|
|
|
-#else
|
|
|
+
|
|
|
+#else /* USE_HYPER */
|
|
|
/* The Hyper version of CONNECT */
|
|
|
-static CURLcode CONNECT(struct Curl_easy *data,
|
|
|
- int sockindex,
|
|
|
- const char *hostname,
|
|
|
- int remote_port)
|
|
|
+static CURLcode start_CONNECT(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct tunnel_state *ts)
|
|
|
{
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
struct hyptransfer *h = &data->hyp;
|
|
|
- curl_socket_t tunnelsocket = conn->sock[sockindex];
|
|
|
- struct http_connect_state *s = conn->connect_state;
|
|
|
- CURLcode result = CURLE_OUT_OF_MEMORY;
|
|
|
+ curl_socket_t tunnelsocket = conn->sock[ts->sockindex];
|
|
|
hyper_io *io = NULL;
|
|
|
hyper_request *req = NULL;
|
|
|
hyper_headers *headers = NULL;
|
|
|
hyper_clientconn_options *options = NULL;
|
|
|
hyper_task *handshake = NULL;
|
|
|
hyper_task *task = NULL; /* for the handshake */
|
|
|
- hyper_task *sendtask = NULL; /* for the send */
|
|
|
hyper_clientconn *client = NULL;
|
|
|
- hyper_error *hypererr = NULL;
|
|
|
+ hyper_task *sendtask = NULL; /* for the send */
|
|
|
char *hostheader = NULL; /* for CONNECT */
|
|
|
char *host = NULL; /* Host: */
|
|
|
+ CURLcode result = CURLE_OUT_OF_MEMORY;
|
|
|
|
|
|
- if(Curl_connect_complete(conn))
|
|
|
- return CURLE_OK; /* CONNECT is already completed */
|
|
|
+ io = hyper_io_new();
|
|
|
+ if(!io) {
|
|
|
+ failf(data, "Couldn't create hyper IO");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ /* tell Hyper how to read/write network data */
|
|
|
+ hyper_io_set_userdata(io, data);
|
|
|
+ hyper_io_set_read(io, Curl_hyper_recv);
|
|
|
+ hyper_io_set_write(io, Curl_hyper_send);
|
|
|
+ conn->sockfd = tunnelsocket;
|
|
|
+
|
|
|
+ data->state.hconnect = TRUE;
|
|
|
+
|
|
|
+ /* create an executor to poll futures */
|
|
|
+ if(!h->exec) {
|
|
|
+ h->exec = hyper_executor_new();
|
|
|
+ if(!h->exec) {
|
|
|
+ failf(data, "Couldn't create hyper executor");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
- conn->bits.proxy_connect_closed = FALSE;
|
|
|
+ options = hyper_clientconn_options_new();
|
|
|
+ hyper_clientconn_options_set_preserve_header_case(options, 1);
|
|
|
+ hyper_clientconn_options_set_preserve_header_order(options, 1);
|
|
|
|
|
|
- do {
|
|
|
- switch(s->tunnel_state) {
|
|
|
- case TUNNEL_INIT:
|
|
|
- /* BEGIN CONNECT PHASE */
|
|
|
- io = hyper_io_new();
|
|
|
- if(!io) {
|
|
|
- failf(data, "Couldn't create hyper IO");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- /* tell Hyper how to read/write network data */
|
|
|
- hyper_io_set_userdata(io, data);
|
|
|
- hyper_io_set_read(io, Curl_hyper_recv);
|
|
|
- hyper_io_set_write(io, Curl_hyper_send);
|
|
|
- conn->sockfd = tunnelsocket;
|
|
|
-
|
|
|
- data->state.hconnect = TRUE;
|
|
|
-
|
|
|
- /* create an executor to poll futures */
|
|
|
- if(!h->exec) {
|
|
|
- h->exec = hyper_executor_new();
|
|
|
- if(!h->exec) {
|
|
|
- failf(data, "Couldn't create hyper executor");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- }
|
|
|
+ if(!options) {
|
|
|
+ failf(data, "Couldn't create hyper client options");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
|
|
|
- options = hyper_clientconn_options_new();
|
|
|
- hyper_clientconn_options_set_preserve_header_case(options, 1);
|
|
|
- hyper_clientconn_options_set_preserve_header_order(options, 1);
|
|
|
+ hyper_clientconn_options_exec(options, h->exec);
|
|
|
|
|
|
- if(!options) {
|
|
|
- failf(data, "Couldn't create hyper client options");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
+ /* "Both the `io` and the `options` are consumed in this function
|
|
|
+ call" */
|
|
|
+ handshake = hyper_clientconn_handshake(io, options);
|
|
|
+ if(!handshake) {
|
|
|
+ failf(data, "Couldn't create hyper client handshake");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ io = NULL;
|
|
|
+ options = NULL;
|
|
|
|
|
|
- hyper_clientconn_options_exec(options, h->exec);
|
|
|
+ if(HYPERE_OK != hyper_executor_push(h->exec, handshake)) {
|
|
|
+ failf(data, "Couldn't hyper_executor_push the handshake");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ handshake = NULL; /* ownership passed on */
|
|
|
|
|
|
- /* "Both the `io` and the `options` are consumed in this function
|
|
|
- call" */
|
|
|
- handshake = hyper_clientconn_handshake(io, options);
|
|
|
- if(!handshake) {
|
|
|
- failf(data, "Couldn't create hyper client handshake");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- io = NULL;
|
|
|
- options = NULL;
|
|
|
+ task = hyper_executor_poll(h->exec);
|
|
|
+ if(!task) {
|
|
|
+ failf(data, "Couldn't hyper_executor_poll the handshake");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
|
|
|
- if(HYPERE_OK != hyper_executor_push(h->exec, handshake)) {
|
|
|
- failf(data, "Couldn't hyper_executor_push the handshake");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- handshake = NULL; /* ownership passed on */
|
|
|
+ client = hyper_task_value(task);
|
|
|
+ hyper_task_free(task);
|
|
|
+ req = hyper_request_new();
|
|
|
+ if(!req) {
|
|
|
+ failf(data, "Couldn't hyper_request_new");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ if(hyper_request_set_method(req, (uint8_t *)"CONNECT",
|
|
|
+ strlen("CONNECT"))) {
|
|
|
+ failf(data, "error setting method");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
|
|
|
- task = hyper_executor_poll(h->exec);
|
|
|
- if(!task) {
|
|
|
- failf(data, "Couldn't hyper_executor_poll the handshake");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
+ infof(data, "Establish HTTP proxy tunnel to %s:%d",
|
|
|
+ ts->hostname, ts->remote_port);
|
|
|
|
|
|
- client = hyper_task_value(task);
|
|
|
- hyper_task_free(task);
|
|
|
- req = hyper_request_new();
|
|
|
- if(!req) {
|
|
|
- failf(data, "Couldn't hyper_request_new");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- if(hyper_request_set_method(req, (uint8_t *)"CONNECT",
|
|
|
- strlen("CONNECT"))) {
|
|
|
- failf(data, "error setting method");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
+ /* This only happens if we've looped here due to authentication
|
|
|
+ reasons, and we don't really use the newly cloned URL here
|
|
|
+ then. Just free() it. */
|
|
|
+ Curl_safefree(data->req.newurl);
|
|
|
|
|
|
- infof(data, "Establish HTTP proxy tunnel to %s:%d",
|
|
|
- hostname, remote_port);
|
|
|
+ result = CONNECT_host(data, conn, ts->hostname, ts->remote_port,
|
|
|
+ &hostheader, &host);
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
|
|
|
- /* This only happens if we've looped here due to authentication
|
|
|
- reasons, and we don't really use the newly cloned URL here
|
|
|
- then. Just free() it. */
|
|
|
- Curl_safefree(data->req.newurl);
|
|
|
+ if(hyper_request_set_uri(req, (uint8_t *)hostheader,
|
|
|
+ strlen(hostheader))) {
|
|
|
+ failf(data, "error setting path");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ if(data->set.verbose) {
|
|
|
+ char *se = aprintf("CONNECT %s HTTP/1.1\r\n", hostheader);
|
|
|
+ if(!se) {
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ Curl_debug(data, CURLINFO_HEADER_OUT, se, strlen(se));
|
|
|
+ free(se);
|
|
|
+ }
|
|
|
+ /* Setup the proxy-authorization header, if any */
|
|
|
+ result = Curl_http_output_auth(data, conn, "CONNECT", HTTPREQ_GET,
|
|
|
+ hostheader, TRUE);
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+ Curl_safefree(hostheader);
|
|
|
+
|
|
|
+ /* default is 1.1 */
|
|
|
+ if((conn->http_proxy.proxytype == CURLPROXY_HTTP_1_0) &&
|
|
|
+ (HYPERE_OK != hyper_request_set_version(req,
|
|
|
+ HYPER_HTTP_VERSION_1_0))) {
|
|
|
+ failf(data, "error setting HTTP version");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
|
|
|
- result = CONNECT_host(data, conn, hostname, remote_port,
|
|
|
- &hostheader, &host);
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
+ headers = hyper_request_headers(req);
|
|
|
+ if(!headers) {
|
|
|
+ failf(data, "hyper_request_headers");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+ if(host) {
|
|
|
+ result = Curl_hyper_header(data, headers, host);
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+ Curl_safefree(host);
|
|
|
+ }
|
|
|
|
|
|
- if(hyper_request_set_uri(req, (uint8_t *)hostheader,
|
|
|
- strlen(hostheader))) {
|
|
|
- failf(data, "error setting path");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- if(data->set.verbose) {
|
|
|
- char *se = aprintf("CONNECT %s HTTP/1.1\r\n", hostheader);
|
|
|
- if(!se) {
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
- Curl_debug(data, CURLINFO_HEADER_OUT, se, strlen(se));
|
|
|
- free(se);
|
|
|
- }
|
|
|
- /* Setup the proxy-authorization header, if any */
|
|
|
- result = Curl_http_output_auth(data, conn, "CONNECT", HTTPREQ_GET,
|
|
|
- hostheader, TRUE);
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- Curl_safefree(hostheader);
|
|
|
+ if(data->state.aptr.proxyuserpwd) {
|
|
|
+ result = Curl_hyper_header(data, headers,
|
|
|
+ data->state.aptr.proxyuserpwd);
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
|
|
|
- /* default is 1.1 */
|
|
|
- if((conn->http_proxy.proxytype == CURLPROXY_HTTP_1_0) &&
|
|
|
- (HYPERE_OK != hyper_request_set_version(req,
|
|
|
- HYPER_HTTP_VERSION_1_0))) {
|
|
|
- failf(data, "error setting HTTP version");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
+ if(!Curl_checkProxyheaders(data, conn, STRCONST("User-Agent")) &&
|
|
|
+ data->set.str[STRING_USERAGENT]) {
|
|
|
+ struct dynbuf ua;
|
|
|
+ Curl_dyn_init(&ua, DYN_HTTP_REQUEST);
|
|
|
+ result = Curl_dyn_addf(&ua, "User-Agent: %s\r\n",
|
|
|
+ data->set.str[STRING_USERAGENT]);
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+ result = Curl_hyper_header(data, headers, Curl_dyn_ptr(&ua));
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+ Curl_dyn_free(&ua);
|
|
|
+ }
|
|
|
+
|
|
|
+ if(!Curl_checkProxyheaders(data, conn, STRCONST("Proxy-Connection"))) {
|
|
|
+ result = Curl_hyper_header(data, headers,
|
|
|
+ "Proxy-Connection: Keep-Alive");
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+
|
|
|
+ result = Curl_add_custom_headers(data, TRUE, headers);
|
|
|
+ if(result)
|
|
|
+ goto error;
|
|
|
+
|
|
|
+ sendtask = hyper_clientconn_send(client, req);
|
|
|
+ if(!sendtask) {
|
|
|
+ failf(data, "hyper_clientconn_send");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
+
|
|
|
+ if(HYPERE_OK != hyper_executor_push(h->exec, sendtask)) {
|
|
|
+ failf(data, "Couldn't hyper_executor_push the send");
|
|
|
+ result = CURLE_OUT_OF_MEMORY;
|
|
|
+ goto error;
|
|
|
+ }
|
|
|
|
|
|
- headers = hyper_request_headers(req);
|
|
|
- if(!headers) {
|
|
|
- failf(data, "hyper_request_headers");
|
|
|
+error:
|
|
|
+ free(host);
|
|
|
+ free(hostheader);
|
|
|
+ if(io)
|
|
|
+ hyper_io_free(io);
|
|
|
+ if(options)
|
|
|
+ hyper_clientconn_options_free(options);
|
|
|
+ if(handshake)
|
|
|
+ hyper_task_free(handshake);
|
|
|
+ if(client)
|
|
|
+ hyper_clientconn_free(client);
|
|
|
+ return result;
|
|
|
+}
|
|
|
+
|
|
|
+static CURLcode send_CONNECT(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct tunnel_state *ts,
|
|
|
+ bool *done)
|
|
|
+{
|
|
|
+ struct hyptransfer *h = &data->hyp;
|
|
|
+ hyper_task *task = NULL;
|
|
|
+ hyper_error *hypererr = NULL;
|
|
|
+ CURLcode result = CURLE_OK;
|
|
|
+
|
|
|
+ (void)ts;
|
|
|
+ (void)conn;
|
|
|
+ do {
|
|
|
+ task = hyper_executor_poll(h->exec);
|
|
|
+ if(task) {
|
|
|
+ bool error = hyper_task_type(task) == HYPER_TASK_ERROR;
|
|
|
+ if(error)
|
|
|
+ hypererr = hyper_task_value(task);
|
|
|
+ hyper_task_free(task);
|
|
|
+ if(error) {
|
|
|
+ /* this could probably use a better error code? */
|
|
|
result = CURLE_OUT_OF_MEMORY;
|
|
|
goto error;
|
|
|
}
|
|
|
- if(host) {
|
|
|
- result = Curl_hyper_header(data, headers, host);
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- Curl_safefree(host);
|
|
|
- }
|
|
|
+ }
|
|
|
+ } while(task);
|
|
|
+error:
|
|
|
+ *done = (result == CURLE_OK);
|
|
|
+ if(hypererr) {
|
|
|
+ uint8_t errbuf[256];
|
|
|
+ size_t errlen = hyper_error_print(hypererr, errbuf, sizeof(errbuf));
|
|
|
+ failf(data, "Hyper: %.*s", (int)errlen, errbuf);
|
|
|
+ hyper_error_free(hypererr);
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+}
|
|
|
|
|
|
- if(data->state.aptr.proxyuserpwd) {
|
|
|
- result = Curl_hyper_header(data, headers,
|
|
|
- data->state.aptr.proxyuserpwd);
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- }
|
|
|
+static CURLcode recv_CONNECT_resp(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ struct tunnel_state *ts,
|
|
|
+ bool *done)
|
|
|
+{
|
|
|
+ struct hyptransfer *h = &data->hyp;
|
|
|
+ CURLcode result;
|
|
|
+ int didwhat;
|
|
|
+
|
|
|
+ (void)ts;
|
|
|
+ *done = FALSE;
|
|
|
+ result = Curl_hyper_stream(data, conn, &didwhat, done,
|
|
|
+ CURL_CSELECT_IN | CURL_CSELECT_OUT);
|
|
|
+ if(result || !*done)
|
|
|
+ return result;
|
|
|
+ if(h->exec) {
|
|
|
+ hyper_executor_free(h->exec);
|
|
|
+ h->exec = NULL;
|
|
|
+ }
|
|
|
+ if(h->read_waker) {
|
|
|
+ hyper_waker_free(h->read_waker);
|
|
|
+ h->read_waker = NULL;
|
|
|
+ }
|
|
|
+ if(h->write_waker) {
|
|
|
+ hyper_waker_free(h->write_waker);
|
|
|
+ h->write_waker = NULL;
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+}
|
|
|
|
|
|
- if(!Curl_checkProxyheaders(data, conn, STRCONST("User-Agent")) &&
|
|
|
- data->set.str[STRING_USERAGENT]) {
|
|
|
- struct dynbuf ua;
|
|
|
- Curl_dyn_init(&ua, DYN_HTTP_REQUEST);
|
|
|
- result = Curl_dyn_addf(&ua, "User-Agent: %s\r\n",
|
|
|
- data->set.str[STRING_USERAGENT]);
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- result = Curl_hyper_header(data, headers, Curl_dyn_ptr(&ua));
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- Curl_dyn_free(&ua);
|
|
|
- }
|
|
|
+#endif /* USE_HYPER */
|
|
|
|
|
|
- if(!Curl_checkProxyheaders(data, conn, STRCONST("Proxy-Connection"))) {
|
|
|
- result = Curl_hyper_header(data, headers,
|
|
|
- "Proxy-Connection: Keep-Alive");
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- }
|
|
|
+static CURLcode CONNECT(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data,
|
|
|
+ struct 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;
|
|
|
|
|
|
- result = Curl_add_custom_headers(data, TRUE, headers);
|
|
|
+ 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 TUNNEL_INIT:
|
|
|
+ /* Prepare the CONNECT request and make a first attempt to send. */
|
|
|
+ result = start_CONNECT(data, cf->conn, ts);
|
|
|
if(result)
|
|
|
- goto error;
|
|
|
+ goto out;
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_CONNECT, data);
|
|
|
+ /* FALLTHROUGH */
|
|
|
|
|
|
- sendtask = hyper_clientconn_send(client, req);
|
|
|
- if(!sendtask) {
|
|
|
- failf(data, "hyper_clientconn_send");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
+ case TUNNEL_CONNECT:
|
|
|
+ /* see that the request is completely sent */
|
|
|
+ result = send_CONNECT(data, cf->conn, ts, &done);
|
|
|
+ if(result || !done)
|
|
|
+ goto out;
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_RECEIVE, data);
|
|
|
+ /* FALLTHROUGH */
|
|
|
|
|
|
- if(HYPERE_OK != hyper_executor_push(h->exec, sendtask)) {
|
|
|
- failf(data, "Couldn't hyper_executor_push the send");
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
+ case TUNNEL_RECEIVE:
|
|
|
+ /* read what is there */
|
|
|
+ result = recv_CONNECT_resp(data, cf->conn, 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 */
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_RESPONSE, data);
|
|
|
+ /* FALLTHROUGH */
|
|
|
|
|
|
- hyper_clientconn_free(client);
|
|
|
-
|
|
|
- do {
|
|
|
- task = hyper_executor_poll(h->exec);
|
|
|
- if(task) {
|
|
|
- bool error = hyper_task_type(task) == HYPER_TASK_ERROR;
|
|
|
- if(error)
|
|
|
- hypererr = hyper_task_value(task);
|
|
|
- hyper_task_free(task);
|
|
|
- if(error) {
|
|
|
- /* this could probably use a better error code? */
|
|
|
- result = CURLE_OUT_OF_MEMORY;
|
|
|
- goto error;
|
|
|
- }
|
|
|
+ case TUNNEL_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. */
|
|
|
+ if(ts->close_connection || conn->bits.close) {
|
|
|
+ /* Close the filter chain and trigger connect, non-blocking
|
|
|
+ * again, so the process is ongoing. This will
|
|
|
+ * a) the close resets our tunnel state
|
|
|
+ * b) the connect makes sure that there will be a socket
|
|
|
+ * to select on again.
|
|
|
+ * We return and expect to be called again. */
|
|
|
+ infof(data, "Connect me again please");
|
|
|
+ Curl_conn_close(data, cf->sockindex);
|
|
|
+ result = cf->next->cft->connect(cf->next, data, FALSE, &done);
|
|
|
+ goto out;
|
|
|
}
|
|
|
- } while(task);
|
|
|
- s->tunnel_state = TUNNEL_CONNECT;
|
|
|
- /* FALLTHROUGH */
|
|
|
- case TUNNEL_CONNECT: {
|
|
|
- int didwhat;
|
|
|
- bool done = FALSE;
|
|
|
- result = Curl_hyper_stream(data, conn, &didwhat, &done,
|
|
|
- CURL_CSELECT_IN | CURL_CSELECT_OUT);
|
|
|
- if(result)
|
|
|
- goto error;
|
|
|
- if(!done)
|
|
|
- break;
|
|
|
- s->tunnel_state = TUNNEL_COMPLETE;
|
|
|
- if(h->exec) {
|
|
|
- hyper_executor_free(h->exec);
|
|
|
- h->exec = NULL;
|
|
|
+ /* staying on this connection, reset state */
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_INIT, data);
|
|
|
}
|
|
|
- if(h->read_waker) {
|
|
|
- hyper_waker_free(h->read_waker);
|
|
|
- h->read_waker = NULL;
|
|
|
- }
|
|
|
- if(h->write_waker) {
|
|
|
- hyper_waker_free(h->write_waker);
|
|
|
- h->write_waker = NULL;
|
|
|
- }
|
|
|
- }
|
|
|
- break;
|
|
|
-
|
|
|
- default:
|
|
|
break;
|
|
|
- }
|
|
|
|
|
|
- if(conn->bits.close && data->req.newurl) {
|
|
|
- /* Connection closed by server. Don't use it anymore */
|
|
|
- Curl_closesocket(data, conn, conn->sock[sockindex]);
|
|
|
- conn->sock[sockindex] = CURL_SOCKET_BAD;
|
|
|
+ default:
|
|
|
break;
|
|
|
}
|
|
|
|
|
|
- /* If we are supposed to continue and request a new URL, which basically
|
|
|
- * means the HTTP authentication is still going on so if the tunnel
|
|
|
- * is complete we start over in INIT state */
|
|
|
- if(data->req.newurl && (TUNNEL_COMPLETE == s->tunnel_state)) {
|
|
|
- infof(data, "CONNECT request done, loop to make another");
|
|
|
- connect_init(data, TRUE); /* reinit */
|
|
|
- }
|
|
|
} while(data->req.newurl);
|
|
|
|
|
|
+ DEBUGASSERT(ts->tunnel_state == TUNNEL_RESPONSE);
|
|
|
+ if(data->info.httpproxycode/100 != 2) {
|
|
|
+ /* a non-2xx response and we have no next url to try. */
|
|
|
+ free(data->req.newurl);
|
|
|
+ data->req.newurl = NULL;
|
|
|
+ /* failure, close this connection to avoid re-use */
|
|
|
+ streamclose(conn, "proxy CONNECT failure");
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_FAILED, data);
|
|
|
+ failf(data, "CONNECT tunnel failed, response %d", data->req.httpcode);
|
|
|
+ return CURLE_RECV_ERROR;
|
|
|
+ }
|
|
|
+ /* 2xx response, SUCCESS! */
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_ESTABLISHED, data);
|
|
|
+ infof(data, "CONNECT tunnel established, response %d",
|
|
|
+ data->info.httpproxycode);
|
|
|
result = CURLE_OK;
|
|
|
- if(s->tunnel_state == TUNNEL_COMPLETE) {
|
|
|
- if(data->info.httpproxycode/100 != 2) {
|
|
|
- if(conn->bits.close && data->req.newurl) {
|
|
|
- conn->bits.proxy_connect_closed = TRUE;
|
|
|
- infof(data, "Connect me again please");
|
|
|
- Curl_connect_done(data);
|
|
|
- }
|
|
|
- else {
|
|
|
- free(data->req.newurl);
|
|
|
- data->req.newurl = NULL;
|
|
|
- /* failure, close this connection to avoid re-use */
|
|
|
- streamclose(conn, "proxy CONNECT failure");
|
|
|
- Curl_closesocket(data, conn, conn->sock[sockindex]);
|
|
|
- conn->sock[sockindex] = CURL_SOCKET_BAD;
|
|
|
- }
|
|
|
|
|
|
- /* to back to init state */
|
|
|
- s->tunnel_state = TUNNEL_INIT;
|
|
|
+out:
|
|
|
+ if(result)
|
|
|
+ tunnel_go_state(cf, ts, TUNNEL_FAILED, data);
|
|
|
+ return result;
|
|
|
+}
|
|
|
|
|
|
- if(!conn->bits.proxy_connect_closed) {
|
|
|
- failf(data, "Received HTTP code %d from proxy after CONNECT",
|
|
|
- data->req.httpcode);
|
|
|
- result = CURLE_RECV_ERROR;
|
|
|
- }
|
|
|
- }
|
|
|
+static CURLcode http_proxy_cf_connect(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data,
|
|
|
+ bool blocking, bool *done)
|
|
|
+{
|
|
|
+ CURLcode result;
|
|
|
+ struct tunnel_state *ts = cf->ctx;
|
|
|
+
|
|
|
+ if(cf->connected) {
|
|
|
+ *done = TRUE;
|
|
|
+ return CURLE_OK;
|
|
|
}
|
|
|
- error:
|
|
|
- free(host);
|
|
|
- free(hostheader);
|
|
|
- if(io)
|
|
|
- hyper_io_free(io);
|
|
|
|
|
|
- if(options)
|
|
|
- hyper_clientconn_options_free(options);
|
|
|
+ result = cf->next->cft->connect(cf->next, data, blocking, done);
|
|
|
+ if(result || !*done)
|
|
|
+ return result;
|
|
|
|
|
|
- if(handshake)
|
|
|
- hyper_task_free(handshake);
|
|
|
+ /* TODO: can we do blocking? */
|
|
|
+ /* We want "seamless" operations through HTTP proxy tunnel */
|
|
|
|
|
|
- if(hypererr) {
|
|
|
- uint8_t errbuf[256];
|
|
|
- size_t errlen = hyper_error_print(hypererr, errbuf, sizeof(errbuf));
|
|
|
- failf(data, "Hyper: %.*s", (int)errlen, errbuf);
|
|
|
- hyper_error_free(hypererr);
|
|
|
+ /* for the secondary socket (FTP), use the "connect to host"
|
|
|
+ * but ignore the "connect to port" (use the secondary port)
|
|
|
+ */
|
|
|
+ *done = FALSE;
|
|
|
+ if(!ts) {
|
|
|
+ result = tunnel_init(&ts, data, cf->conn, cf->sockindex);
|
|
|
+ if(result)
|
|
|
+ return result;
|
|
|
+ cf->ctx = ts;
|
|
|
+ }
|
|
|
+
|
|
|
+ result = 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;
|
|
|
+ tunnel_free(cf, data);
|
|
|
}
|
|
|
return result;
|
|
|
}
|
|
|
-#endif
|
|
|
|
|
|
-void Curl_connect_free(struct Curl_easy *data)
|
|
|
+static void http_proxy_cf_get_host(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data,
|
|
|
+ const char **phost,
|
|
|
+ const char **pdisplay_host,
|
|
|
+ int *pport)
|
|
|
+{
|
|
|
+ (void)data;
|
|
|
+ if(!cf->connected) {
|
|
|
+ *phost = cf->conn->http_proxy.host.name;
|
|
|
+ *pdisplay_host = cf->conn->http_proxy.host.dispname;
|
|
|
+ *pport = (int)cf->conn->http_proxy.port;
|
|
|
+ }
|
|
|
+ else {
|
|
|
+ cf->next->cft->get_host(cf->next, data, phost, pdisplay_host, pport);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static int http_proxy_cf_get_select_socks(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data,
|
|
|
+ curl_socket_t *socks)
|
|
|
+{
|
|
|
+ struct tunnel_state *ts = cf->ctx;
|
|
|
+ struct connectdata *conn = cf->conn;
|
|
|
+ int fds;
|
|
|
+
|
|
|
+ DEBUGASSERT(conn);
|
|
|
+ fds = cf->next->cft->get_select_socks(cf->next, data, socks);
|
|
|
+ if(!fds && cf->next->connected && !cf->connected) {
|
|
|
+ /* If we are not connected, but the filter "below" is
|
|
|
+ * and not waiting on something, we are tunneling. */
|
|
|
+ socks[0] = conn->sock[cf->sockindex];
|
|
|
+ if(ts) {
|
|
|
+ /* when we've 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're still sending the request, wait
|
|
|
+ for write. */
|
|
|
+ if(ts->http_proxy.sending == HTTPSEND_REQUEST)
|
|
|
+ return GETSOCK_WRITESOCK(0);
|
|
|
+ return GETSOCK_READSOCK(0);
|
|
|
+ }
|
|
|
+ return GETSOCK_WRITESOCK(0);
|
|
|
+ }
|
|
|
+ return fds;
|
|
|
+}
|
|
|
+
|
|
|
+static void http_proxy_cf_detach_data(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data)
|
|
|
+{
|
|
|
+ if(cf->ctx) {
|
|
|
+ tunnel_free(cf, data);
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+static void http_proxy_cf_destroy(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data)
|
|
|
+{
|
|
|
+ http_proxy_cf_detach_data(cf, data);
|
|
|
+}
|
|
|
+
|
|
|
+static void http_proxy_cf_close(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data)
|
|
|
{
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
- struct http_connect_state *s = conn->connect_state;
|
|
|
- if(s) {
|
|
|
- free(s);
|
|
|
- conn->connect_state = NULL;
|
|
|
+ DEBUGASSERT(cf->next);
|
|
|
+ cf->connected = FALSE;
|
|
|
+ cf->next->cft->close(cf->next, data);
|
|
|
+ if(cf->ctx) {
|
|
|
+ tunnel_go_state(cf, cf->ctx, TUNNEL_INIT, data);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
-/*
|
|
|
- * Curl_proxyCONNECT() requires that we're connected to a HTTP proxy. This
|
|
|
- * function will issue the necessary commands to get a seamless tunnel through
|
|
|
- * this proxy. After that, the socket can be used just as a normal socket.
|
|
|
- */
|
|
|
|
|
|
-CURLcode Curl_proxyCONNECT(struct Curl_easy *data,
|
|
|
- int sockindex,
|
|
|
- const char *hostname,
|
|
|
- int remote_port)
|
|
|
+static const struct Curl_cftype cft_http_proxy = {
|
|
|
+ "HTTP-PROXY",
|
|
|
+ CF_TYPE_IP_CONNECT,
|
|
|
+ http_proxy_cf_destroy,
|
|
|
+ Curl_cf_def_setup,
|
|
|
+ http_proxy_cf_connect,
|
|
|
+ http_proxy_cf_close,
|
|
|
+ http_proxy_cf_get_host,
|
|
|
+ http_proxy_cf_get_select_socks,
|
|
|
+ Curl_cf_def_data_pending,
|
|
|
+ Curl_cf_def_send,
|
|
|
+ Curl_cf_def_recv,
|
|
|
+ Curl_cf_def_attach_data,
|
|
|
+ http_proxy_cf_detach_data,
|
|
|
+};
|
|
|
+
|
|
|
+CURLcode Curl_conn_http_proxy_add(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ int sockindex)
|
|
|
{
|
|
|
+ struct Curl_cfilter *cf;
|
|
|
CURLcode result;
|
|
|
- struct connectdata *conn = data->conn;
|
|
|
- if(!conn->connect_state) {
|
|
|
- result = connect_init(data, FALSE);
|
|
|
- if(result)
|
|
|
- return result;
|
|
|
+
|
|
|
+ result = Curl_cf_create(&cf, &cft_http_proxy, NULL);
|
|
|
+ if(!result)
|
|
|
+ Curl_conn_cf_add(data, conn, sockindex, cf);
|
|
|
+ return result;
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+static CURLcode send_haproxy_header(struct Curl_cfilter*cf,
|
|
|
+ struct Curl_easy *data)
|
|
|
+{
|
|
|
+ struct dynbuf req;
|
|
|
+ CURLcode result;
|
|
|
+ const char *tcp_version;
|
|
|
+ Curl_dyn_init(&req, DYN_HAXPROXY);
|
|
|
+
|
|
|
+#ifdef USE_UNIX_SOCKETS
|
|
|
+ if(cf->conn->unix_domain_socket)
|
|
|
+ /* the buffer is large enough to hold this! */
|
|
|
+ result = Curl_dyn_addn(&req, STRCONST("PROXY UNKNOWN\r\n"));
|
|
|
+ else {
|
|
|
+#endif /* USE_UNIX_SOCKETS */
|
|
|
+ /* Emit the correct prefix for IPv6 */
|
|
|
+ tcp_version = cf->conn->bits.ipv6 ? "TCP6" : "TCP4";
|
|
|
+
|
|
|
+ result = Curl_dyn_addf(&req, "PROXY %s %s %s %i %i\r\n",
|
|
|
+ tcp_version,
|
|
|
+ data->info.conn_local_ip,
|
|
|
+ data->info.conn_primary_ip,
|
|
|
+ data->info.conn_local_port,
|
|
|
+ data->info.conn_primary_port);
|
|
|
+
|
|
|
+#ifdef USE_UNIX_SOCKETS
|
|
|
}
|
|
|
- result = CONNECT(data, sockindex, hostname, remote_port);
|
|
|
+#endif /* USE_UNIX_SOCKETS */
|
|
|
|
|
|
- if(result || Curl_connect_complete(conn))
|
|
|
- Curl_connect_done(data);
|
|
|
+ if(!result)
|
|
|
+ result = Curl_buffer_send(&req, data, &data->info.request_size,
|
|
|
+ 0, FIRSTSOCKET);
|
|
|
+ return result;
|
|
|
+}
|
|
|
+
|
|
|
+static CURLcode haproxy_cf_connect(struct Curl_cfilter *cf,
|
|
|
+ struct Curl_easy *data,
|
|
|
+ bool blocking, bool *done)
|
|
|
+{
|
|
|
+ CURLcode result;
|
|
|
+
|
|
|
+ if(cf->connected) {
|
|
|
+ *done = TRUE;
|
|
|
+ return CURLE_OK;
|
|
|
+ }
|
|
|
+
|
|
|
+ result = cf->next->cft->connect(cf->next, data, blocking, done);
|
|
|
+ if(result || !*done)
|
|
|
+ return result;
|
|
|
|
|
|
+ result = send_haproxy_header(cf, data);
|
|
|
+ *done = (!result);
|
|
|
+ cf->connected = *done;
|
|
|
return result;
|
|
|
}
|
|
|
|
|
|
-#else
|
|
|
-void Curl_connect_free(struct Curl_easy *data)
|
|
|
+static const struct Curl_cftype cft_haproxy = {
|
|
|
+ "HAPROXY",
|
|
|
+ 0,
|
|
|
+ Curl_cf_def_destroy_this,
|
|
|
+ Curl_cf_def_setup,
|
|
|
+ haproxy_cf_connect,
|
|
|
+ Curl_cf_def_close,
|
|
|
+ Curl_cf_def_get_host,
|
|
|
+ Curl_cf_def_get_select_socks,
|
|
|
+ Curl_cf_def_data_pending,
|
|
|
+ Curl_cf_def_send,
|
|
|
+ Curl_cf_def_recv,
|
|
|
+ Curl_cf_def_attach_data,
|
|
|
+ Curl_cf_def_detach_data,
|
|
|
+};
|
|
|
+
|
|
|
+CURLcode Curl_conn_haproxy_add(struct Curl_easy *data,
|
|
|
+ struct connectdata *conn,
|
|
|
+ int sockindex)
|
|
|
{
|
|
|
- (void)data;
|
|
|
+ struct Curl_cfilter *cf;
|
|
|
+ CURLcode result;
|
|
|
+
|
|
|
+ result = Curl_cf_create(&cf, &cft_haproxy, NULL);
|
|
|
+ if(!result)
|
|
|
+ Curl_conn_cf_add(data, conn, sockindex, cf);
|
|
|
+ return result;
|
|
|
}
|
|
|
|
|
|
-#endif /* CURL_DISABLE_PROXY */
|
|
|
+#endif /* !CURL_DISABLE_PROXY &6 ! CURL_DISABLE_HTTP */
|