[Libguestfs] [PATCH nbdkit 4/6] curl: Introduce concept of getting/putting handles from a common pool

Introduce the concept of a pool of curl handles from which the main code has to take a handle (get_handle()), and later return it to the pool (put_handle()). When a handle is taken from the pool it can be used exclusively by the main code until it is returned. The "pool" only has at most one handle, so this is actually a pessimization from the situation before because all NBD connections share a single curl handle. In a future commit we will add support for multiple handles in the pool. A note about the change to tests/test-retry-request-mirror.c: This test depended implicitly on a new NBD connection opening a new libcurl handle (this no longer happens), and when opening a new libcurl handle that causes a HEAD request to be made. We have to adjust the test to remove this assumption. --- plugins/curl/Makefile.am | 5 +- plugins/curl/curldefs.h | 5 + plugins/curl/curl.c | 430 +++----------------------- plugins/curl/pool.c | 482 ++++++++++++++++++++++++++++++ tests/test-retry-request-mirror.c | 9 - 5 files changed, 534 insertions(+), 397 deletions(-) diff --git a/plugins/curl/Makefile.am b/plugins/curl/Makefile.am index f08923e77..d6b97a821 100644 --- a/plugins/curl/Makefile.am +++ b/plugins/curl/Makefile.am @@ -1,5 +1,5 @@ # nbdkit -# Copyright (C) 2014-2022 Red Hat Inc. +# Copyright (C) 2014-2023 Red Hat Inc. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions are @@ -39,8 +39,9 @@ plugin_LTLIBRARIES = nbdkit-curl-plugin.la nbdkit_curl_plugin_la_SOURCES = \ curldefs.h \ - scripts.c \ curl.c \ + pool.c \ + scripts.c \ $(top_srcdir)/include/nbdkit-plugin.h \ $(NULL) diff --git a/plugins/curl/curldefs.h b/plugins/curl/curldefs.h index 36f5fd5bb..fabd1e537 100644 --- a/plugins/curl/curldefs.h +++ b/plugins/curl/curldefs.h @@ -100,6 +100,11 @@ struct curl_handle { struct curl_slist *headers_copy; }; +/* pool.c */ +extern struct curl_handle *get_handle (void); +extern void put_handle (struct curl_handle *ch); +extern void free_all_handles (void); + /* scripts.c */ extern int do_scripts (struct curl_handle *ch); extern void scripts_unload (void); diff --git a/plugins/curl/curl.c b/plugins/curl/curl.c index 86d953432..105de29c6 100644 --- a/plugins/curl/curl.c +++ b/plugins/curl/curl.c @@ -48,8 +48,6 @@ #include <nbdkit-plugin.h> -#include "ascii-ctype.h" -#include "ascii-string.h" #include "cleanup.h" #include "curldefs.h" @@ -112,6 +110,7 @@ curl_unload (void) free (password); free (proxy_password); scripts_unload (); + free_all_handles (); curl_global_cleanup (); } @@ -433,23 +432,11 @@ curl_config_complete (void) curl_easy_strerror ((r)), (ch)->errbuf); \ } while (0) -static int debug_cb (CURL *handle, curl_infotype type, - const char *data, size_t size, void *); -static size_t header_cb (void *ptr, size_t size, size_t nmemb, void *opaque); -static size_t write_cb (char *ptr, size_t size, size_t nmemb, void *opaque); -static size_t read_cb (void *ptr, size_t size, size_t nmemb, void *opaque); - /* Create the per-connection handle. */ static void * curl_open (int readonly) { struct handle *h; - CURLcode r; -#ifdef HAVE_CURLINFO_CONTENT_LENGTH_DOWNLOAD_T - curl_off_t o; -#else - double d; -#endif h = calloc (1, sizeof *h); if (h == NULL) { @@ -458,306 +445,7 @@ curl_open (int readonly) } h->readonly = readonly; - h->ch = calloc (1, sizeof *h->ch); - if (h->ch == NULL) { - nbdkit_error ("calloc: %m"); - free (h); - return NULL; - } - - h->ch->c = curl_easy_init (); - if (h->ch->c == NULL) { - nbdkit_error ("curl_easy_init: failed: %m"); - goto err; - } - - if (curl_debug_verbose) { - /* NB: Constants must be explicitly long because the parameter is - * varargs. - */ - curl_easy_setopt (h->ch->c, CURLOPT_VERBOSE, 1L); - curl_easy_setopt (h->ch->c, CURLOPT_DEBUGFUNCTION, debug_cb); - } - - curl_easy_setopt (h->ch->c, CURLOPT_ERRORBUFFER, h->ch->errbuf); - - r = CURLE_OK; - if (unix_socket_path) { -#if HAVE_CURLOPT_UNIX_SOCKET_PATH - r = curl_easy_setopt (h->ch->c, CURLOPT_UNIX_SOCKET_PATH, unix_socket_path); -#else - r = CURLE_UNKNOWN_OPTION; -#endif - } - if (r != CURLE_OK) { - display_curl_error (h->ch, r, "curl_easy_setopt: CURLOPT_UNIX_SOCKET_PATH"); - goto err; - } - - /* Set the URL. */ - r = curl_easy_setopt (h->ch->c, CURLOPT_URL, url); - if (r != CURLE_OK) { - display_curl_error (h->ch, r, "curl_easy_setopt: CURLOPT_URL [%s]", url); - goto err; - } - - /* Various options we always set. - * - * NB: Both here and below constants must be explicitly long because - * the parameter is varargs. - */ - curl_easy_setopt (h->ch->c, CURLOPT_AUTOREFERER, 1L); - if (followlocation) - curl_easy_setopt (h->ch->c, CURLOPT_FOLLOWLOCATION, 1L); - curl_easy_setopt (h->ch->c, CURLOPT_FAILONERROR, 1L); - - /* Options. */ - if (cainfo) { - if (strlen (cainfo) == 0) - curl_easy_setopt (h->ch->c, CURLOPT_CAINFO, NULL); - else - curl_easy_setopt (h->ch->c, CURLOPT_CAINFO, cainfo); - } - if (capath) - curl_easy_setopt (h->ch->c, CURLOPT_CAPATH, capath); - if (cookie) - curl_easy_setopt (h->ch->c, CURLOPT_COOKIE, cookie); - if (cookiefile) - curl_easy_setopt (h->ch->c, CURLOPT_COOKIEFILE, cookiefile); - if (cookiejar) - curl_easy_setopt (h->ch->c, CURLOPT_COOKIEJAR, cookiejar); - if (headers) - curl_easy_setopt (h->ch->c, CURLOPT_HTTPHEADER, headers); - if (password) - curl_easy_setopt (h->ch->c, CURLOPT_PASSWORD, password); -#ifndef HAVE_CURLOPT_PROTOCOLS_STR - if (protocols != CURLPROTO_ALL) { - curl_easy_setopt (h->ch->c, CURLOPT_PROTOCOLS, protocols); - curl_easy_setopt (h->ch->c, CURLOPT_REDIR_PROTOCOLS, protocols); - } -#else /* HAVE_CURLOPT_PROTOCOLS_STR (new in 7.85.0) */ - if (protocols) { - curl_easy_setopt (h->ch->c, CURLOPT_PROTOCOLS_STR, protocols); - curl_easy_setopt (h->ch->c, CURLOPT_REDIR_PROTOCOLS_STR, protocols); - } -#endif /* HAVE_CURLOPT_PROTOCOLS_STR */ - if (proxy) - curl_easy_setopt (h->ch->c, CURLOPT_PROXY, proxy); - if (proxy_password) - curl_easy_setopt (h->ch->c, CURLOPT_PROXYPASSWORD, proxy_password); - if (proxy_user) - curl_easy_setopt (h->ch->c, CURLOPT_PROXYUSERNAME, proxy_user); - if (!sslverify) { - curl_easy_setopt (h->ch->c, CURLOPT_SSL_VERIFYPEER, 0L); - curl_easy_setopt (h->ch->c, CURLOPT_SSL_VERIFYHOST, 0L); - } - if (ssl_version) { - if (strcmp (ssl_version, "tlsv1") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1); - else if (strcmp (ssl_version, "sslv2") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_SSLv2); - else if (strcmp (ssl_version, "sslv3") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_SSLv3); - else if (strcmp (ssl_version, "tlsv1.0") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_0); - else if (strcmp (ssl_version, "tlsv1.1") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_1); - else if (strcmp (ssl_version, "tlsv1.2") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_2); - else if (strcmp (ssl_version, "tlsv1.3") == 0) - curl_easy_setopt (h->ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_3); - else { - display_curl_error (h->ch, r, "curl_easy_setopt: CURLOPT_SSLVERSION [%s]", - ssl_version); - goto err; - } - - } - if (ssl_cipher_list) - curl_easy_setopt (h->ch->c, CURLOPT_SSL_CIPHER_LIST, ssl_cipher_list); - if (tls13_ciphers) { -#if (LIBCURL_VERSION_MAJOR > 7) || \ - (LIBCURL_VERSION_MAJOR == 7 && LIBCURL_VERSION_MINOR >= 61) - curl_easy_setopt (h->ch->c, CURLOPT_TLS13_CIPHERS, tls13_ciphers); -#else - /* This is not available before curl-7.61 */ - nbdkit_error ("tls13-ciphers is not supported in this build of " - "nbdkit-curl-plugin"); - goto err; -#endif - } - if (tcp_keepalive) - curl_easy_setopt (h->ch->c, CURLOPT_TCP_KEEPALIVE, 1L); - if (!tcp_nodelay) - curl_easy_setopt (h->ch->c, CURLOPT_TCP_NODELAY, 0L); - if (timeout > 0) - /* NB: The cast is required here because the parameter is varargs - * treated as long, and not type safe. - */ - curl_easy_setopt (h->ch->c, CURLOPT_TIMEOUT, (long) timeout); - if (user) - curl_easy_setopt (h->ch->c, CURLOPT_USERNAME, user); - if (user_agent) - curl_easy_setopt (h->ch->c, CURLOPT_USERAGENT, user_agent); - - /* Get the file size and also whether the remote HTTP server - * supports byte ranges. - * - * We must run the scripts if necessary and set headers in the - * handle. - */ - if (do_scripts (h->ch) == -1) goto err; - h->ch->accept_range = false; - curl_easy_setopt (h->ch->c, CURLOPT_NOBODY, 1L); /* No Body, not nobody! */ - curl_easy_setopt (h->ch->c, CURLOPT_HEADERFUNCTION, header_cb); - curl_easy_setopt (h->ch->c, CURLOPT_HEADERDATA, h->ch); - r = curl_easy_perform (h->ch->c); - if (r != CURLE_OK) { - display_curl_error (h->ch, r, - "problem doing HEAD request to fetch size of URL [%s]", - url); - goto err; - } - -#ifdef HAVE_CURLINFO_CONTENT_LENGTH_DOWNLOAD_T - r = curl_easy_getinfo (h->ch->c, CURLINFO_CONTENT_LENGTH_DOWNLOAD_T, &o); - if (r != CURLE_OK) { - display_curl_error (h->ch, r, - "could not get length of remote file [%s]", url); - goto err; - } - - if (o == -1) { - nbdkit_error ("could not get length of remote file [%s], " - "is the URL correct?", url); - goto err; - } - - h->ch->exportsize = o; -#else - r = curl_easy_getinfo (h->ch->c, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d); - if (r != CURLE_OK) { - display_curl_error (h->ch, r, - "could not get length of remote file [%s]", url); - goto err; - } - - if (d == -1) { - nbdkit_error ("could not get length of remote file [%s], " - "is the URL correct?", url); - goto err; - } - - h->ch->exportsize = d; -#endif - nbdkit_debug ("content length: %" PRIi64, h->ch->exportsize); - - if (ascii_strncasecmp (url, "http://", strlen ("http://")) == 0 || - ascii_strncasecmp (url, "https://", strlen ("https://")) == 0) { - if (!h->ch->accept_range) { - nbdkit_error ("server does not support 'range' (byte range) requests"); - goto err; - } - - nbdkit_debug ("accept range supported (for HTTP/HTTPS)"); - } - - /* Get set up for reading and writing. */ - curl_easy_setopt (h->ch->c, CURLOPT_HEADERFUNCTION, NULL); - curl_easy_setopt (h->ch->c, CURLOPT_HEADERDATA, NULL); - curl_easy_setopt (h->ch->c, CURLOPT_WRITEFUNCTION, write_cb); - curl_easy_setopt (h->ch->c, CURLOPT_WRITEDATA, h->ch); - if (!readonly) { - curl_easy_setopt (h->ch->c, CURLOPT_READFUNCTION, read_cb); - curl_easy_setopt (h->ch->c, CURLOPT_READDATA, h->ch); - } - return h; - - err: - if (h->ch->c) - curl_easy_cleanup (h->ch->c); - free (h->ch); - free (h); - return NULL; -} - -/* When using CURLOPT_VERBOSE, this callback is used to redirect - * messages to nbdkit_debug (instead of stderr). - */ -static int -debug_cb (CURL *handle, curl_infotype type, - const char *data, size_t size, void *opaque) -{ - size_t origsize = size; - CLEANUP_FREE char *str; - - /* The data parameter passed is NOT \0-terminated, but also it may - * have \n or \r\n line endings. The only sane way to deal with - * this is to copy the string. (The data strings may also be - * multi-line, but we don't deal with that here). - */ - str = malloc (size + 1); - if (str == NULL) - goto out; - memcpy (str, data, size); - str[size] = '\0'; - - while (size > 0 && (str[size-1] == '\n' || str[size-1] == '\r')) { - str[size-1] = '\0'; - size--; - } - - switch (type) { - case CURLINFO_TEXT: - nbdkit_debug ("%s", str); - break; - case CURLINFO_HEADER_IN: - nbdkit_debug ("S: %s", str); - break; - case CURLINFO_HEADER_OUT: - nbdkit_debug ("C: %s", str); - break; - default: - /* Assume everything else is binary data that we cannot print. */ - nbdkit_debug ("<data with size=%zu>", origsize); - } - - out: - return 0; -} - -static size_t -header_cb (void *ptr, size_t size, size_t nmemb, void *opaque) -{ - struct curl_handle *ch = opaque; - size_t realsize = size * nmemb; - const char *header = ptr; - const char *end = header + realsize; - const char *accept_ranges = "accept-ranges:"; - const char *bytes = "bytes"; - - if (realsize >= strlen (accept_ranges) && - ascii_strncasecmp (header, accept_ranges, strlen (accept_ranges)) == 0) { - const char *p = strchr (header, ':') + 1; - - /* Skip whitespace between the header name and value. */ - while (p < end && *p && ascii_isspace (*p)) - p++; - - if (end - p >= strlen (bytes) - && strncmp (p, bytes, strlen (bytes)) == 0) { - /* Check that there is nothing but whitespace after the value. */ - p += strlen (bytes); - while (p < end && *p && ascii_isspace (*p)) - p++; - - if (p == end || !*p) - ch->accept_range = true; - } - } - - return realsize; } /* Free up the per-connection handle. */ @@ -766,22 +454,35 @@ curl_close (void *handle) { struct handle *h = handle; - curl_easy_cleanup (h->ch->c); - if (h->ch->headers_copy) - curl_slist_free_all (h->ch->headers_copy); - free (h->ch); free (h); } #define THREAD_MODEL NBDKIT_THREAD_MODEL_SERIALIZE_REQUESTS +/* Calls get_handle() ... put_handle() to get a handle for the length + * of the current scope. + */ +#define GET_HANDLE_FOR_CURRENT_SCOPE(ch) \ + CLEANUP_PUT_HANDLE struct curl_handle *ch = get_handle (); +#define CLEANUP_PUT_HANDLE __attribute__((cleanup (cleanup_put_handle))) +static void +cleanup_put_handle (void *chp) +{ + struct curl_handle *ch = * (struct curl_handle **) chp; + + if (ch != NULL) + put_handle (ch); +} + /* Get the file size. */ static int64_t curl_get_size (void *handle) { - struct handle *h = handle; + GET_HANDLE_FOR_CURRENT_SCOPE (ch); + if (ch == NULL) + return -1; - return h->ch->exportsize; + return ch->exportsize; } /* Multi-conn is safe for read-only connections, but HTTP does not @@ -796,42 +497,37 @@ curl_can_multi_conn (void *handle) return !! h->readonly; } -/* NB: The terminology used by libcurl is confusing! - * - * WRITEFUNCTION / write_cb is used when reading from the remote server - * READFUNCTION / read_cb is used when writing to the remote server. - * - * We use the same terminology as libcurl here. - */ - /* Read data from the remote server. */ static int curl_pread (void *handle, void *buf, uint32_t count, uint64_t offset) { - struct handle *h = handle; CURLcode r; char range[128]; + GET_HANDLE_FOR_CURRENT_SCOPE (ch); + if (ch == NULL) + return -1; + /* Run the scripts if necessary and set headers in the handle. */ - if (do_scripts (h->ch) == -1) return -1; + if (do_scripts (ch) == -1) return -1; /* Tell the write_cb where we want the data to be written. write_cb * will update this if the data comes in multiple sections. */ - h->ch->write_buf = buf; - h->ch->write_count = count; + ch->write_buf = buf; + ch->write_count = count; - curl_easy_setopt (h->ch->c, CURLOPT_HTTPGET, 1L); + curl_easy_setopt (ch->c, CURLOPT_HTTPGET, 1L); /* Make an HTTP range request. */ snprintf (range, sizeof range, "%" PRIu64 "-%" PRIu64, offset, offset + count); - curl_easy_setopt (h->ch->c, CURLOPT_RANGE, range); + curl_easy_setopt (ch->c, CURLOPT_RANGE, range); /* The assumption here is that curl will look after timeouts. */ - r = curl_easy_perform (h->ch->c); + r = curl_easy_perform (ch->c); if (r != CURLE_OK) { - display_curl_error (h->ch, r, "pread: curl_easy_perform"); + display_curl_error (ch, r, "pread: curl_easy_perform"); return -1; } @@ -840,62 +536,42 @@ curl_pread (void *handle, void *buf, uint32_t count, uint64_t offset) */ /* As far as I understand the cURL API, this should never happen. */ - assert (h->ch->write_count == 0); + assert (ch->write_count == 0); return 0; } -static size_t -write_cb (char *ptr, size_t size, size_t nmemb, void *opaque) -{ - struct curl_handle *ch = opaque; - size_t orig_realsize = size * nmemb; - size_t realsize = orig_realsize; - - assert (ch->write_buf); - - /* Don't read more than the requested amount of data, even if the - * server or libcurl sends more. - */ - if (realsize > ch->write_count) - realsize = ch->write_count; - - memcpy (ch->write_buf, ptr, realsize); - - ch->write_count -= realsize; - ch->write_buf += realsize; - - return orig_realsize; -} - /* Write data to the remote server. */ static int curl_pwrite (void *handle, const void *buf, uint32_t count, uint64_t offset) { - struct handle *h = handle; CURLcode r; char range[128]; + GET_HANDLE_FOR_CURRENT_SCOPE (ch); + if (ch == NULL) + return -1; + /* Run the scripts if necessary and set headers in the handle. */ - if (do_scripts (h->ch) == -1) return -1; + if (do_scripts (ch) == -1) return -1; /* Tell the read_cb where we want the data to be read from. read_cb * will update this if the data comes in multiple sections. */ - h->ch->read_buf = buf; - h->ch->read_count = count; + ch->read_buf = buf; + ch->read_count = count; - curl_easy_setopt (h->ch->c, CURLOPT_UPLOAD, 1L); + curl_easy_setopt (ch->c, CURLOPT_UPLOAD, 1L); /* Make an HTTP range request. */ snprintf (range, sizeof range, "%" PRIu64 "-%" PRIu64, offset, offset + count); - curl_easy_setopt (h->ch->c, CURLOPT_RANGE, range); + curl_easy_setopt (ch->c, CURLOPT_RANGE, range); /* The assumption here is that curl will look after timeouts. */ - r = curl_easy_perform (h->ch->c); + r = curl_easy_perform (ch->c); if (r != CURLE_OK) { - display_curl_error (h->ch, r, "pwrite: curl_easy_perform"); + display_curl_error (ch, r, "pwrite: curl_easy_perform"); return -1; } @@ -904,29 +580,11 @@ curl_pwrite (void *handle, const void *buf, uint32_t count, uint64_t offset) */ /* As far as I understand the cURL API, this should never happen. */ - assert (h->ch->read_count == 0); + assert (ch->read_count == 0); return 0; } -static size_t -read_cb (void *ptr, size_t size, size_t nmemb, void *opaque) -{ - struct curl_handle *ch = opaque; - size_t realsize = size * nmemb; - - assert (ch->read_buf); - if (realsize > ch->read_count) - realsize = ch->read_count; - - memcpy (ptr, ch->read_buf, realsize); - - ch->read_count -= realsize; - ch->read_buf += realsize; - - return realsize; -} - static struct nbdkit_plugin plugin = { .name = "curl", .version = PACKAGE_VERSION, diff --git a/plugins/curl/pool.c b/plugins/curl/pool.c new file mode 100644 index 000000000..8fd1dc147 --- /dev/null +++ b/plugins/curl/pool.c @@ -0,0 +1,482 @@ +/* nbdkit + * Copyright (C) 2014-2023 Red Hat Inc. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are + * met: + * + * * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * * Neither the name of Red Hat nor the names of its contributors may be + * used to endorse or promote products derived from this software without + * specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, + * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A + * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR + * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF + * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT + * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +/* Curl handle pool. + * + * To get a libcurl handle, call get_handle(). When you hold the + * handle, it is yours exclusively to use. After you have finished + * with the handle, put it back into the pool by calling put_handle(). + */ + +#include <config.h> + +#include <stdio.h> +#include <stdlib.h> +#include <stdbool.h> +#include <stdint.h> +#include <inttypes.h> +#include <string.h> +#include <assert.h> +#include <pthread.h> + +#include <curl/curl.h> + +#include <nbdkit-plugin.h> + +#include "ascii-ctype.h" +#include "ascii-string.h" +#include "cleanup.h" + +#include "curldefs.h" + +/* Translate CURLcode to nbdkit_error. */ +#define display_curl_error(ch, r, fs, ...) \ + do { \ + nbdkit_error ((fs ": %s: %s"), ## __VA_ARGS__, \ + curl_easy_strerror ((r)), (ch)->errbuf); \ + } while (0) + +static struct curl_handle *allocate_handle (void); +static int debug_cb (CURL *handle, curl_infotype type, + const char *data, size_t size, void *); +static size_t header_cb (void *ptr, size_t size, size_t nmemb, void *opaque); +static size_t write_cb (char *ptr, size_t size, size_t nmemb, void *opaque); +static size_t read_cb (void *ptr, size_t size, size_t nmemb, void *opaque); + +/* In the current implementation there is only one handle. This lock + * prevents it from being used multiple times. + */ +static pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER; + +/* The single curl handle. NULL means not yet allocated. */ +static struct curl_handle *the_ch; + +/* Close and free all handles in the pool. */ +void +free_all_handles (void) +{ + if (the_ch) { + curl_easy_cleanup (the_ch->c); + if (the_ch->headers_copy) + curl_slist_free_all (the_ch->headers_copy); + free (the_ch); + } +} + +/* Get a handle from the pool. + * + * It is owned exclusively by the caller until they call put_handle. + */ +struct curl_handle * +get_handle (void) +{ + int r; + + r = pthread_mutex_lock (&lock); + assert (r == 0); + if (!the_ch) { + the_ch = allocate_handle (); + if (!the_ch) { + pthread_mutex_unlock (&lock); + return NULL; + } + } + return the_ch; +} + +/* Return the handle to the pool. */ +void +put_handle (struct curl_handle *ch) +{ + pthread_mutex_unlock (&lock); +} + +/* Allocate and initialize a new libcurl handle. */ +static struct curl_handle * +allocate_handle (void) +{ + struct curl_handle *ch; + CURLcode r; +#ifdef HAVE_CURLINFO_CONTENT_LENGTH_DOWNLOAD_T + curl_off_t o; +#else + double d; +#endif + + ch = calloc (1, sizeof *ch); + if (ch == NULL) { + nbdkit_error ("calloc: %m"); + free (ch); + return NULL; + } + + ch->c = curl_easy_init (); + if (ch->c == NULL) { + nbdkit_error ("curl_easy_init: failed: %m"); + goto err; + } + + if (curl_debug_verbose) { + /* NB: Constants must be explicitly long because the parameter is + * varargs. + */ + curl_easy_setopt (ch->c, CURLOPT_VERBOSE, 1L); + curl_easy_setopt (ch->c, CURLOPT_DEBUGFUNCTION, debug_cb); + } + + curl_easy_setopt (ch->c, CURLOPT_ERRORBUFFER, ch->errbuf); + + r = CURLE_OK; + if (unix_socket_path) { +#if HAVE_CURLOPT_UNIX_SOCKET_PATH + r = curl_easy_setopt (ch->c, CURLOPT_UNIX_SOCKET_PATH, unix_socket_path); +#else + r = CURLE_UNKNOWN_OPTION; +#endif + } + if (r != CURLE_OK) { + display_curl_error (ch, r, "curl_easy_setopt: CURLOPT_UNIX_SOCKET_PATH"); + goto err; + } + + /* Set the URL. */ + r = curl_easy_setopt (ch->c, CURLOPT_URL, url); + if (r != CURLE_OK) { + display_curl_error (ch, r, "curl_easy_setopt: CURLOPT_URL [%s]", url); + goto err; + } + + /* Various options we always set. + * + * NB: Both here and below constants must be explicitly long because + * the parameter is varargs. + */ + curl_easy_setopt (ch->c, CURLOPT_AUTOREFERER, 1L); + if (followlocation) + curl_easy_setopt (ch->c, CURLOPT_FOLLOWLOCATION, 1L); + curl_easy_setopt (ch->c, CURLOPT_FAILONERROR, 1L); + + /* Options. */ + if (cainfo) { + if (strlen (cainfo) == 0) + curl_easy_setopt (ch->c, CURLOPT_CAINFO, NULL); + else + curl_easy_setopt (ch->c, CURLOPT_CAINFO, cainfo); + } + if (capath) + curl_easy_setopt (ch->c, CURLOPT_CAPATH, capath); + if (cookie) + curl_easy_setopt (ch->c, CURLOPT_COOKIE, cookie); + if (cookiefile) + curl_easy_setopt (ch->c, CURLOPT_COOKIEFILE, cookiefile); + if (cookiejar) + curl_easy_setopt (ch->c, CURLOPT_COOKIEJAR, cookiejar); + if (headers) + curl_easy_setopt (ch->c, CURLOPT_HTTPHEADER, headers); + if (password) + curl_easy_setopt (ch->c, CURLOPT_PASSWORD, password); +#ifndef HAVE_CURLOPT_PROTOCOLS_STR + if (protocols != CURLPROTO_ALL) { + curl_easy_setopt (ch->c, CURLOPT_PROTOCOLS, protocols); + curl_easy_setopt (ch->c, CURLOPT_REDIR_PROTOCOLS, protocols); + } +#else /* HAVE_CURLOPT_PROTOCOLS_STR (new in 7.85.0) */ + if (protocols) { + curl_easy_setopt (ch->c, CURLOPT_PROTOCOLS_STR, protocols); + curl_easy_setopt (ch->c, CURLOPT_REDIR_PROTOCOLS_STR, protocols); + } +#endif /* HAVE_CURLOPT_PROTOCOLS_STR */ + if (proxy) + curl_easy_setopt (ch->c, CURLOPT_PROXY, proxy); + if (proxy_password) + curl_easy_setopt (ch->c, CURLOPT_PROXYPASSWORD, proxy_password); + if (proxy_user) + curl_easy_setopt (ch->c, CURLOPT_PROXYUSERNAME, proxy_user); + if (!sslverify) { + curl_easy_setopt (ch->c, CURLOPT_SSL_VERIFYPEER, 0L); + curl_easy_setopt (ch->c, CURLOPT_SSL_VERIFYHOST, 0L); + } + if (ssl_version) { + if (strcmp (ssl_version, "tlsv1") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1); + else if (strcmp (ssl_version, "sslv2") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_SSLv2); + else if (strcmp (ssl_version, "sslv3") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_SSLv3); + else if (strcmp (ssl_version, "tlsv1.0") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_0); + else if (strcmp (ssl_version, "tlsv1.1") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_1); + else if (strcmp (ssl_version, "tlsv1.2") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_2); + else if (strcmp (ssl_version, "tlsv1.3") == 0) + curl_easy_setopt (ch->c, CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1_3); + else { + display_curl_error (ch, r, "curl_easy_setopt: CURLOPT_SSLVERSION [%s]", + ssl_version); + goto err; + } + + } + if (ssl_cipher_list) + curl_easy_setopt (ch->c, CURLOPT_SSL_CIPHER_LIST, ssl_cipher_list); + if (tls13_ciphers) { +#if (LIBCURL_VERSION_MAJOR > 7) || \ + (LIBCURL_VERSION_MAJOR == 7 && LIBCURL_VERSION_MINOR >= 61) + curl_easy_setopt (ch->c, CURLOPT_TLS13_CIPHERS, tls13_ciphers); +#else + /* This is not available before curl-7.61 */ + nbdkit_error ("tls13-ciphers is not supported in this build of " + "nbdkit-curl-plugin"); + goto err; +#endif + } + if (tcp_keepalive) + curl_easy_setopt (ch->c, CURLOPT_TCP_KEEPALIVE, 1L); + if (!tcp_nodelay) + curl_easy_setopt (ch->c, CURLOPT_TCP_NODELAY, 0L); + if (timeout > 0) + /* NB: The cast is required here because the parameter is varargs + * treated as long, and not type safe. + */ + curl_easy_setopt (ch->c, CURLOPT_TIMEOUT, (long) timeout); + if (user) + curl_easy_setopt (ch->c, CURLOPT_USERNAME, user); + if (user_agent) + curl_easy_setopt (ch->c, CURLOPT_USERAGENT, user_agent); + + /* Get the file size and also whether the remote HTTP server + * supports byte ranges. + * + * We must run the scripts if necessary and set headers in the + * handle. + */ + if (do_scripts (ch) == -1) goto err; + ch->accept_range = false; + curl_easy_setopt (ch->c, CURLOPT_NOBODY, 1L); /* No Body, not nobody! */ + curl_easy_setopt (ch->c, CURLOPT_HEADERFUNCTION, header_cb); + curl_easy_setopt (ch->c, CURLOPT_HEADERDATA, ch); + r = curl_easy_perform (ch->c); + if (r != CURLE_OK) { + display_curl_error (ch, r, + "problem doing HEAD request to fetch size of URL [%s]", + url); + goto err; + } + +#ifdef HAVE_CURLINFO_CONTENT_LENGTH_DOWNLOAD_T + r = curl_easy_getinfo (ch->c, CURLINFO_CONTENT_LENGTH_DOWNLOAD_T, &o); + if (r != CURLE_OK) { + display_curl_error (ch, r, + "could not get length of remote file [%s]", url); + goto err; + } + + if (o == -1) { + nbdkit_error ("could not get length of remote file [%s], " + "is the URL correct?", url); + goto err; + } + + ch->exportsize = o; +#else + r = curl_easy_getinfo (ch->c, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d); + if (r != CURLE_OK) { + display_curl_error (ch, r, + "could not get length of remote file [%s]", url); + goto err; + } + + if (d == -1) { + nbdkit_error ("could not get length of remote file [%s], " + "is the URL correct?", url); + goto err; + } + + ch->exportsize = d; +#endif + nbdkit_debug ("content length: %" PRIi64, ch->exportsize); + + if (ascii_strncasecmp (url, "http://", strlen ("http://")) == 0 || + ascii_strncasecmp (url, "https://", strlen ("https://")) == 0) { + if (!ch->accept_range) { + nbdkit_error ("server does not support 'range' (byte range) requests"); + goto err; + } + + nbdkit_debug ("accept range supported (for HTTP/HTTPS)"); + } + + /* Get set up for reading and writing. */ + curl_easy_setopt (ch->c, CURLOPT_HEADERFUNCTION, NULL); + curl_easy_setopt (ch->c, CURLOPT_HEADERDATA, NULL); + curl_easy_setopt (ch->c, CURLOPT_WRITEFUNCTION, write_cb); + curl_easy_setopt (ch->c, CURLOPT_WRITEDATA, ch); + /* These are only used if !readonly but we always register them. */ + curl_easy_setopt (ch->c, CURLOPT_READFUNCTION, read_cb); + curl_easy_setopt (ch->c, CURLOPT_READDATA, ch); + + return ch; + + err: + if (ch->c) + curl_easy_cleanup (ch->c); + free (ch); + return NULL; +} + +/* When using CURLOPT_VERBOSE, this callback is used to redirect + * messages to nbdkit_debug (instead of stderr). + */ +static int +debug_cb (CURL *handle, curl_infotype type, + const char *data, size_t size, void *opaque) +{ + size_t origsize = size; + CLEANUP_FREE char *str; + + /* The data parameter passed is NOT \0-terminated, but also it may + * have \n or \r\n line endings. The only sane way to deal with + * this is to copy the string. (The data strings may also be + * multi-line, but we don't deal with that here). + */ + str = malloc (size + 1); + if (str == NULL) + goto out; + memcpy (str, data, size); + str[size] = '\0'; + + while (size > 0 && (str[size-1] == '\n' || str[size-1] == '\r')) { + str[size-1] = '\0'; + size--; + } + + switch (type) { + case CURLINFO_TEXT: + nbdkit_debug ("%s", str); + break; + case CURLINFO_HEADER_IN: + nbdkit_debug ("S: %s", str); + break; + case CURLINFO_HEADER_OUT: + nbdkit_debug ("C: %s", str); + break; + default: + /* Assume everything else is binary data that we cannot print. */ + nbdkit_debug ("<data with size=%zu>", origsize); + } + + out: + return 0; +} + +static size_t +header_cb (void *ptr, size_t size, size_t nmemb, void *opaque) +{ + struct curl_handle *ch = opaque; + size_t realsize = size * nmemb; + const char *header = ptr; + const char *end = header + realsize; + const char *accept_ranges = "accept-ranges:"; + const char *bytes = "bytes"; + + if (realsize >= strlen (accept_ranges) && + ascii_strncasecmp (header, accept_ranges, strlen (accept_ranges)) == 0) { + const char *p = strchr (header, ':') + 1; + + /* Skip whitespace between the header name and value. */ + while (p < end && *p && ascii_isspace (*p)) + p++; + + if (end - p >= strlen (bytes) + && strncmp (p, bytes, strlen (bytes)) == 0) { + /* Check that there is nothing but whitespace after the value. */ + p += strlen (bytes); + while (p < end && *p && ascii_isspace (*p)) + p++; + + if (p == end || !*p) + ch->accept_range = true; + } + } + + return realsize; +} + +/* NB: The terminology used by libcurl is confusing! + * + * WRITEFUNCTION / write_cb is used when reading from the remote server + * READFUNCTION / read_cb is used when writing to the remote server. + * + * We use the same terminology as libcurl here. + */ + +static size_t +write_cb (char *ptr, size_t size, size_t nmemb, void *opaque) +{ + struct curl_handle *ch = opaque; + size_t orig_realsize = size * nmemb; + size_t realsize = orig_realsize; + + assert (ch->write_buf); + + /* Don't read more than the requested amount of data, even if the + * server or libcurl sends more. + */ + if (realsize > ch->write_count) + realsize = ch->write_count; + + memcpy (ch->write_buf, ptr, realsize); + + ch->write_count -= realsize; + ch->write_buf += realsize; + + return orig_realsize; +} + +static size_t +read_cb (void *ptr, size_t size, size_t nmemb, void *opaque) +{ + struct curl_handle *ch = opaque; + size_t realsize = size * nmemb; + + assert (ch->read_buf); + if (realsize > ch->read_count) + realsize = ch->read_count; + + memcpy (ptr, ch->read_buf, realsize); + + ch->read_count -= realsize; + ch->read_buf += realsize; + + return realsize; +} diff --git a/tests/test-retry-request-mirror.c b/tests/test-retry-request-mirror.c index b1eb8af02..e7157184f 100644 --- a/tests/test-retry-request-mirror.c +++ b/tests/test-retry-request-mirror.c @@ -135,15 +135,6 @@ main (int argc, char *argv[]) } nbd_close (nbd); - - /* Reconnection in the next iteration of the loop will read from - * the mirror (because the curl plugin always makes a HEAD request - * in curl_open to read the size) which will flip the state, so we - * have to do it here. - */ - state++; - if (state == 3) - state = 1; } exit (EXIT_SUCCESS); -- 2.39.0