1769ce76dSAlexander Graf /* 2769ce76dSAlexander Graf * QEMU Block driver for CURL images 3769ce76dSAlexander Graf * 4769ce76dSAlexander Graf * Copyright (c) 2009 Alexander Graf <agraf@suse.de> 5769ce76dSAlexander Graf * 6769ce76dSAlexander Graf * Permission is hereby granted, free of charge, to any person obtaining a copy 7769ce76dSAlexander Graf * of this software and associated documentation files (the "Software"), to deal 8769ce76dSAlexander Graf * in the Software without restriction, including without limitation the rights 9769ce76dSAlexander Graf * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 10769ce76dSAlexander Graf * copies of the Software, and to permit persons to whom the Software is 11769ce76dSAlexander Graf * furnished to do so, subject to the following conditions: 12769ce76dSAlexander Graf * 13769ce76dSAlexander Graf * The above copyright notice and this permission notice shall be included in 14769ce76dSAlexander Graf * all copies or substantial portions of the Software. 15769ce76dSAlexander Graf * 16769ce76dSAlexander Graf * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 17769ce76dSAlexander Graf * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 18769ce76dSAlexander Graf * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 19769ce76dSAlexander Graf * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 20769ce76dSAlexander Graf * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 21769ce76dSAlexander Graf * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 22769ce76dSAlexander Graf * THE SOFTWARE. 23769ce76dSAlexander Graf */ 24452fcdbcSMarkus Armbruster 2580c71a24SPeter Maydell #include "qemu/osdep.h" 26da34e65cSMarkus Armbruster #include "qapi/error.h" 27796a060bSRichard W.M. Jones #include "qemu/error-report.h" 28922a01a0SMarkus Armbruster #include "qemu/option.h" 29737e150eSPaolo Bonzini #include "block/block_int.h" 30452fcdbcSMarkus Armbruster #include "qapi/qmp/qdict.h" 31d49b6836SMarkus Armbruster #include "qapi/qmp/qstring.h" 321bff9606SDaniel P. Berrange #include "crypto/secret.h" 33769ce76dSAlexander Graf #include <curl/curl.h> 34f348b6d1SVeronia Bahaa #include "qemu/cutils.h" 35ed2a66deSLaurent Vivier #include "trace.h" 36769ce76dSAlexander Graf 37769ce76dSAlexander Graf // #define DEBUG_VERBOSE 38769ce76dSAlexander Graf 39031fd1beSPeter Maydell #if LIBCURL_VERSION_NUM >= 0x071000 40031fd1beSPeter Maydell /* The multi interface timer callback was introduced in 7.16.0 */ 41031fd1beSPeter Maydell #define NEED_CURL_TIMER_CALLBACK 429aedd5a5SMatthew Booth #define HAVE_SOCKET_ACTION 439aedd5a5SMatthew Booth #endif 449aedd5a5SMatthew Booth 459aedd5a5SMatthew Booth #ifndef HAVE_SOCKET_ACTION 469aedd5a5SMatthew Booth /* If curl_multi_socket_action isn't available, define it statically here in 479aedd5a5SMatthew Booth * terms of curl_multi_socket. Note that ev_bitmask will be ignored, which is 489aedd5a5SMatthew Booth * less efficient but still safe. */ 499aedd5a5SMatthew Booth static CURLMcode __curl_multi_socket_action(CURLM *multi_handle, 509aedd5a5SMatthew Booth curl_socket_t sockfd, 519aedd5a5SMatthew Booth int ev_bitmask, 529aedd5a5SMatthew Booth int *running_handles) 539aedd5a5SMatthew Booth { 549aedd5a5SMatthew Booth return curl_multi_socket(multi_handle, sockfd, running_handles); 559aedd5a5SMatthew Booth } 569aedd5a5SMatthew Booth #define curl_multi_socket_action __curl_multi_socket_action 57031fd1beSPeter Maydell #endif 58031fd1beSPeter Maydell 59fb6d1bbdSStefan Hajnoczi #define PROTOCOLS (CURLPROTO_HTTP | CURLPROTO_HTTPS | \ 6023dce387SMax Reitz CURLPROTO_FTP | CURLPROTO_FTPS) 61fb6d1bbdSStefan Hajnoczi 62769ce76dSAlexander Graf #define CURL_NUM_STATES 8 63769ce76dSAlexander Graf #define CURL_NUM_ACB 8 64f76faedaSRichard W.M. Jones #define CURL_TIMEOUT_MAX 10000 65769ce76dSAlexander Graf 66e3542c67SMatthew Booth #define CURL_BLOCK_OPT_URL "url" 67e3542c67SMatthew Booth #define CURL_BLOCK_OPT_READAHEAD "readahead" 6897a3ea57SMatthew Booth #define CURL_BLOCK_OPT_SSLVERIFY "sslverify" 69212aefaaSDaniel Henrique Barboza #define CURL_BLOCK_OPT_TIMEOUT "timeout" 70a94f83d9SRichard W.M. Jones #define CURL_BLOCK_OPT_COOKIE "cookie" 71327c8ebdSPeter Krempa #define CURL_BLOCK_OPT_COOKIE_SECRET "cookie-secret" 721bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_USERNAME "username" 731bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PASSWORD_SECRET "password-secret" 741bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_USERNAME "proxy-username" 751bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET "proxy-password-secret" 76e3542c67SMatthew Booth 77712b64e8SMax Reitz #define CURL_BLOCK_OPT_READAHEAD_DEFAULT (256 * 1024) 78712b64e8SMax Reitz #define CURL_BLOCK_OPT_SSLVERIFY_DEFAULT true 79712b64e8SMax Reitz #define CURL_BLOCK_OPT_TIMEOUT_DEFAULT 5 80712b64e8SMax Reitz 81769ce76dSAlexander Graf struct BDRVCURLState; 82769ce76dSAlexander Graf 832d25964dSJeff Cody static bool libcurl_initialized; 842d25964dSJeff Cody 85769ce76dSAlexander Graf typedef struct CURLAIOCB { 8628256d82SPaolo Bonzini Coroutine *co; 87769ce76dSAlexander Graf QEMUIOVector *qiov; 88363c3c85SNick Thomas 892125e5eaSPaolo Bonzini uint64_t offset; 902125e5eaSPaolo Bonzini uint64_t bytes; 9128256d82SPaolo Bonzini int ret; 92363c3c85SNick Thomas 93769ce76dSAlexander Graf size_t start; 94769ce76dSAlexander Graf size_t end; 95769ce76dSAlexander Graf } CURLAIOCB; 96769ce76dSAlexander Graf 97ff5ca166SMax Reitz typedef struct CURLSocket { 98ff5ca166SMax Reitz int fd; 99ff5ca166SMax Reitz QLIST_ENTRY(CURLSocket) next; 100ff5ca166SMax Reitz } CURLSocket; 101ff5ca166SMax Reitz 102769ce76dSAlexander Graf typedef struct CURLState 103769ce76dSAlexander Graf { 104769ce76dSAlexander Graf struct BDRVCURLState *s; 105769ce76dSAlexander Graf CURLAIOCB *acb[CURL_NUM_ACB]; 106769ce76dSAlexander Graf CURL *curl; 107ff5ca166SMax Reitz QLIST_HEAD(, CURLSocket) sockets; 108769ce76dSAlexander Graf char *orig_buf; 1092125e5eaSPaolo Bonzini uint64_t buf_start; 110769ce76dSAlexander Graf size_t buf_off; 111769ce76dSAlexander Graf size_t buf_len; 112769ce76dSAlexander Graf char range[128]; 113769ce76dSAlexander Graf char errmsg[CURL_ERROR_SIZE]; 114769ce76dSAlexander Graf char in_use; 115769ce76dSAlexander Graf } CURLState; 116769ce76dSAlexander Graf 117769ce76dSAlexander Graf typedef struct BDRVCURLState { 118769ce76dSAlexander Graf CURLM *multi; 119031fd1beSPeter Maydell QEMUTimer timer; 1202125e5eaSPaolo Bonzini uint64_t len; 121769ce76dSAlexander Graf CURLState states[CURL_NUM_STATES]; 122769ce76dSAlexander Graf char *url; 123c76f4952SNolan size_t readahead_size; 12497a3ea57SMatthew Booth bool sslverify; 125f76faedaSRichard W.M. Jones uint64_t timeout; 126a94f83d9SRichard W.M. Jones char *cookie; 1273494d650SFam Zheng bool accept_range; 12863f0f45fSStefan Hajnoczi AioContext *aio_context; 129ba3186c4SPaolo Bonzini QemuMutex mutex; 130709f2132SPaolo Bonzini CoQueue free_state_waitq; 1311bff9606SDaniel P. Berrange char *username; 1321bff9606SDaniel P. Berrange char *password; 1331bff9606SDaniel P. Berrange char *proxyusername; 1341bff9606SDaniel P. Berrange char *proxypassword; 135769ce76dSAlexander Graf } BDRVCURLState; 136769ce76dSAlexander Graf 137769ce76dSAlexander Graf static void curl_clean_state(CURLState *s); 138769ce76dSAlexander Graf static void curl_multi_do(void *arg); 139838ef602SMatthew Booth static void curl_multi_read(void *arg); 140769ce76dSAlexander Graf 141031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK 14234db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held. */ 143031fd1beSPeter Maydell static int curl_timer_cb(CURLM *multi, long timeout_ms, void *opaque) 144031fd1beSPeter Maydell { 145031fd1beSPeter Maydell BDRVCURLState *s = opaque; 146031fd1beSPeter Maydell 147ed2a66deSLaurent Vivier trace_curl_timer_cb(timeout_ms); 148031fd1beSPeter Maydell if (timeout_ms == -1) { 149031fd1beSPeter Maydell timer_del(&s->timer); 150031fd1beSPeter Maydell } else { 151031fd1beSPeter Maydell int64_t timeout_ns = (int64_t)timeout_ms * 1000 * 1000; 152031fd1beSPeter Maydell timer_mod(&s->timer, 153031fd1beSPeter Maydell qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + timeout_ns); 154031fd1beSPeter Maydell } 155031fd1beSPeter Maydell return 0; 156031fd1beSPeter Maydell } 157031fd1beSPeter Maydell #endif 158031fd1beSPeter Maydell 15934db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held. */ 160769ce76dSAlexander Graf static int curl_sock_cb(CURL *curl, curl_socket_t fd, int action, 16163f0f45fSStefan Hajnoczi void *userp, void *sp) 162769ce76dSAlexander Graf { 16363f0f45fSStefan Hajnoczi BDRVCURLState *s; 164838ef602SMatthew Booth CURLState *state = NULL; 165ff5ca166SMax Reitz CURLSocket *socket; 166ff5ca166SMax Reitz 167838ef602SMatthew Booth curl_easy_getinfo(curl, CURLINFO_PRIVATE, (char **)&state); 16863f0f45fSStefan Hajnoczi s = state->s; 169838ef602SMatthew Booth 170ff5ca166SMax Reitz QLIST_FOREACH(socket, &state->sockets, next) { 171ff5ca166SMax Reitz if (socket->fd == fd) { 172ff5ca166SMax Reitz if (action == CURL_POLL_REMOVE) { 173ff5ca166SMax Reitz QLIST_REMOVE(socket, next); 174ff5ca166SMax Reitz g_free(socket); 175ff5ca166SMax Reitz } 176ff5ca166SMax Reitz break; 177ff5ca166SMax Reitz } 178ff5ca166SMax Reitz } 179ff5ca166SMax Reitz if (!socket) { 180ff5ca166SMax Reitz socket = g_new0(CURLSocket, 1); 181ff5ca166SMax Reitz socket->fd = fd; 182ff5ca166SMax Reitz QLIST_INSERT_HEAD(&state->sockets, socket, next); 183ff5ca166SMax Reitz } 184ff5ca166SMax Reitz socket = NULL; 185ff5ca166SMax Reitz 186ed2a66deSLaurent Vivier trace_curl_sock_cb(action, (int)fd); 187769ce76dSAlexander Graf switch (action) { 188769ce76dSAlexander Graf case CURL_POLL_IN: 189dca21ef2SFam Zheng aio_set_fd_handler(s->aio_context, fd, false, 190f6a51c84SStefan Hajnoczi curl_multi_read, NULL, NULL, state); 191769ce76dSAlexander Graf break; 192769ce76dSAlexander Graf case CURL_POLL_OUT: 193dca21ef2SFam Zheng aio_set_fd_handler(s->aio_context, fd, false, 194f6a51c84SStefan Hajnoczi NULL, curl_multi_do, NULL, state); 195769ce76dSAlexander Graf break; 196769ce76dSAlexander Graf case CURL_POLL_INOUT: 197dca21ef2SFam Zheng aio_set_fd_handler(s->aio_context, fd, false, 198f6a51c84SStefan Hajnoczi curl_multi_read, curl_multi_do, NULL, state); 199769ce76dSAlexander Graf break; 200769ce76dSAlexander Graf case CURL_POLL_REMOVE: 201dca21ef2SFam Zheng aio_set_fd_handler(s->aio_context, fd, false, 202f6a51c84SStefan Hajnoczi NULL, NULL, NULL, NULL); 203769ce76dSAlexander Graf break; 204769ce76dSAlexander Graf } 205769ce76dSAlexander Graf 206769ce76dSAlexander Graf return 0; 207769ce76dSAlexander Graf } 208769ce76dSAlexander Graf 20934db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held. */ 2103494d650SFam Zheng static size_t curl_header_cb(void *ptr, size_t size, size_t nmemb, void *opaque) 211769ce76dSAlexander Graf { 2123494d650SFam Zheng BDRVCURLState *s = opaque; 213769ce76dSAlexander Graf size_t realsize = size * nmemb; 2143494d650SFam Zheng const char *accept_line = "Accept-Ranges: bytes"; 215769ce76dSAlexander Graf 2163494d650SFam Zheng if (realsize >= strlen(accept_line) 2173494d650SFam Zheng && strncmp((char *)ptr, accept_line, strlen(accept_line)) == 0) { 2183494d650SFam Zheng s->accept_range = true; 2190bfcd599SBlue Swirl } 220769ce76dSAlexander Graf 221769ce76dSAlexander Graf return realsize; 222769ce76dSAlexander Graf } 223769ce76dSAlexander Graf 22434db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held. */ 225769ce76dSAlexander Graf static size_t curl_read_cb(void *ptr, size_t size, size_t nmemb, void *opaque) 226769ce76dSAlexander Graf { 227769ce76dSAlexander Graf CURLState *s = ((CURLState*)opaque); 228769ce76dSAlexander Graf size_t realsize = size * nmemb; 229769ce76dSAlexander Graf int i; 230769ce76dSAlexander Graf 231ed2a66deSLaurent Vivier trace_curl_read_cb(realsize); 232769ce76dSAlexander Graf 2334e767657SMax Reitz if (!s || !s->orig_buf) { 2344e767657SMax Reitz goto read_end; 2354e767657SMax Reitz } 236769ce76dSAlexander Graf 2376d4b9e55SFam Zheng if (s->buf_off >= s->buf_len) { 2386d4b9e55SFam Zheng /* buffer full, read nothing */ 2394e767657SMax Reitz goto read_end; 2406d4b9e55SFam Zheng } 2416d4b9e55SFam Zheng realsize = MIN(realsize, s->buf_len - s->buf_off); 242769ce76dSAlexander Graf memcpy(s->orig_buf + s->buf_off, ptr, realsize); 243769ce76dSAlexander Graf s->buf_off += realsize; 244769ce76dSAlexander Graf 245769ce76dSAlexander Graf for(i=0; i<CURL_NUM_ACB; i++) { 246769ce76dSAlexander Graf CURLAIOCB *acb = s->acb[i]; 247769ce76dSAlexander Graf 248769ce76dSAlexander Graf if (!acb) 249769ce76dSAlexander Graf continue; 250769ce76dSAlexander Graf 251769ce76dSAlexander Graf if ((s->buf_off >= acb->end)) { 2522125e5eaSPaolo Bonzini size_t request_length = acb->bytes; 2534e504535SMax Reitz 25403396148SMichael Tokarev qemu_iovec_from_buf(acb->qiov, 0, s->orig_buf + acb->start, 255769ce76dSAlexander Graf acb->end - acb->start); 2564e504535SMax Reitz 2574e504535SMax Reitz if (acb->end - acb->start < request_length) { 2584e504535SMax Reitz size_t offset = acb->end - acb->start; 2594e504535SMax Reitz qemu_iovec_memset(acb->qiov, offset, 0, 2604e504535SMax Reitz request_length - offset); 2614e504535SMax Reitz } 2624e504535SMax Reitz 26328256d82SPaolo Bonzini acb->ret = 0; 264769ce76dSAlexander Graf s->acb[i] = NULL; 26528256d82SPaolo Bonzini qemu_mutex_unlock(&s->s->mutex); 26628256d82SPaolo Bonzini aio_co_wake(acb->co); 26728256d82SPaolo Bonzini qemu_mutex_lock(&s->s->mutex); 268769ce76dSAlexander Graf } 269769ce76dSAlexander Graf } 270769ce76dSAlexander Graf 2714e767657SMax Reitz read_end: 2724e767657SMax Reitz /* curl will error out if we do not return this value */ 2734e767657SMax Reitz return size * nmemb; 274769ce76dSAlexander Graf } 275769ce76dSAlexander Graf 276456af346SPaolo Bonzini /* Called with s->mutex held. */ 27728256d82SPaolo Bonzini static bool curl_find_buf(BDRVCURLState *s, uint64_t start, uint64_t len, 278769ce76dSAlexander Graf CURLAIOCB *acb) 279769ce76dSAlexander Graf { 280769ce76dSAlexander Graf int i; 2812125e5eaSPaolo Bonzini uint64_t end = start + len; 2822125e5eaSPaolo Bonzini uint64_t clamped_end = MIN(end, s->len); 2832125e5eaSPaolo Bonzini uint64_t clamped_len = clamped_end - start; 284769ce76dSAlexander Graf 285769ce76dSAlexander Graf for (i=0; i<CURL_NUM_STATES; i++) { 286769ce76dSAlexander Graf CURLState *state = &s->states[i]; 2872125e5eaSPaolo Bonzini uint64_t buf_end = (state->buf_start + state->buf_off); 2882125e5eaSPaolo Bonzini uint64_t buf_fend = (state->buf_start + state->buf_len); 289769ce76dSAlexander Graf 290769ce76dSAlexander Graf if (!state->orig_buf) 291769ce76dSAlexander Graf continue; 292769ce76dSAlexander Graf if (!state->buf_off) 293769ce76dSAlexander Graf continue; 294769ce76dSAlexander Graf 295769ce76dSAlexander Graf // Does the existing buffer cover our section? 296769ce76dSAlexander Graf if ((start >= state->buf_start) && 297769ce76dSAlexander Graf (start <= buf_end) && 2984e504535SMax Reitz (clamped_end >= state->buf_start) && 2994e504535SMax Reitz (clamped_end <= buf_end)) 300769ce76dSAlexander Graf { 301769ce76dSAlexander Graf char *buf = state->orig_buf + (start - state->buf_start); 302769ce76dSAlexander Graf 3034e504535SMax Reitz qemu_iovec_from_buf(acb->qiov, 0, buf, clamped_len); 3044e504535SMax Reitz if (clamped_len < len) { 3054e504535SMax Reitz qemu_iovec_memset(acb->qiov, clamped_len, 0, len - clamped_len); 3064e504535SMax Reitz } 30728256d82SPaolo Bonzini acb->ret = 0; 30828256d82SPaolo Bonzini return true; 309769ce76dSAlexander Graf } 310769ce76dSAlexander Graf 311769ce76dSAlexander Graf // Wait for unfinished chunks 312b7079df4SMatthew Booth if (state->in_use && 313b7079df4SMatthew Booth (start >= state->buf_start) && 314769ce76dSAlexander Graf (start <= buf_fend) && 3154e504535SMax Reitz (clamped_end >= state->buf_start) && 3164e504535SMax Reitz (clamped_end <= buf_fend)) 317769ce76dSAlexander Graf { 318769ce76dSAlexander Graf int j; 319769ce76dSAlexander Graf 320769ce76dSAlexander Graf acb->start = start - state->buf_start; 3214e504535SMax Reitz acb->end = acb->start + clamped_len; 322769ce76dSAlexander Graf 323769ce76dSAlexander Graf for (j=0; j<CURL_NUM_ACB; j++) { 324769ce76dSAlexander Graf if (!state->acb[j]) { 325769ce76dSAlexander Graf state->acb[j] = acb; 32628256d82SPaolo Bonzini return true; 327769ce76dSAlexander Graf } 328769ce76dSAlexander Graf } 329769ce76dSAlexander Graf } 330769ce76dSAlexander Graf } 331769ce76dSAlexander Graf 33228256d82SPaolo Bonzini return false; 333769ce76dSAlexander Graf } 334769ce76dSAlexander Graf 335ba3186c4SPaolo Bonzini /* Called with s->mutex held. */ 336838ef602SMatthew Booth static void curl_multi_check_completion(BDRVCURLState *s) 337769ce76dSAlexander Graf { 338769ce76dSAlexander Graf int msgs_in_queue; 339769ce76dSAlexander Graf 340769ce76dSAlexander Graf /* Try to find done transfers, so we can free the easy 341769ce76dSAlexander Graf * handle again. */ 3421f2cead3SMatthew Booth for (;;) { 343769ce76dSAlexander Graf CURLMsg *msg; 344769ce76dSAlexander Graf msg = curl_multi_info_read(s->multi, &msgs_in_queue); 345769ce76dSAlexander Graf 3461f2cead3SMatthew Booth /* Quit when there are no more completions */ 347769ce76dSAlexander Graf if (!msg) 348769ce76dSAlexander Graf break; 349769ce76dSAlexander Graf 3501f2cead3SMatthew Booth if (msg->msg == CURLMSG_DONE) { 351769ce76dSAlexander Graf CURLState *state = NULL; 352f6246509SMatthew Booth curl_easy_getinfo(msg->easy_handle, CURLINFO_PRIVATE, 353f6246509SMatthew Booth (char **)&state); 354f785a5aeSNicholas Thomas 355f785a5aeSNicholas Thomas /* ACBs for successful messages get completed in curl_read_cb */ 356f785a5aeSNicholas Thomas if (msg->data.result != CURLE_OK) { 357f785a5aeSNicholas Thomas int i; 358796a060bSRichard W.M. Jones static int errcount = 100; 359796a060bSRichard W.M. Jones 360796a060bSRichard W.M. Jones /* Don't lose the original error message from curl, since 361796a060bSRichard W.M. Jones * it contains extra data. 362796a060bSRichard W.M. Jones */ 363796a060bSRichard W.M. Jones if (errcount > 0) { 364796a060bSRichard W.M. Jones error_report("curl: %s", state->errmsg); 365796a060bSRichard W.M. Jones if (--errcount == 0) { 366796a060bSRichard W.M. Jones error_report("curl: further errors suppressed"); 367796a060bSRichard W.M. Jones } 368796a060bSRichard W.M. Jones } 369796a060bSRichard W.M. Jones 370f785a5aeSNicholas Thomas for (i = 0; i < CURL_NUM_ACB; i++) { 371f785a5aeSNicholas Thomas CURLAIOCB *acb = state->acb[i]; 372f785a5aeSNicholas Thomas 373f785a5aeSNicholas Thomas if (acb == NULL) { 374f785a5aeSNicholas Thomas continue; 375f785a5aeSNicholas Thomas } 376f785a5aeSNicholas Thomas 37728256d82SPaolo Bonzini acb->ret = -EIO; 378f785a5aeSNicholas Thomas state->acb[i] = NULL; 37928256d82SPaolo Bonzini qemu_mutex_unlock(&s->mutex); 38028256d82SPaolo Bonzini aio_co_wake(acb->co); 38128256d82SPaolo Bonzini qemu_mutex_lock(&s->mutex); 382f785a5aeSNicholas Thomas } 383f785a5aeSNicholas Thomas } 384f785a5aeSNicholas Thomas 385769ce76dSAlexander Graf curl_clean_state(state); 386769ce76dSAlexander Graf break; 387769ce76dSAlexander Graf } 388769ce76dSAlexander Graf } 389769ce76dSAlexander Graf } 390769ce76dSAlexander Graf 391ba3186c4SPaolo Bonzini /* Called with s->mutex held. */ 3929d456654SPaolo Bonzini static void curl_multi_do_locked(CURLState *s) 393031fd1beSPeter Maydell { 394ff5ca166SMax Reitz CURLSocket *socket, *next_socket; 395031fd1beSPeter Maydell int running; 396031fd1beSPeter Maydell int r; 397031fd1beSPeter Maydell 398838ef602SMatthew Booth if (!s->s->multi) { 399031fd1beSPeter Maydell return; 400031fd1beSPeter Maydell } 401031fd1beSPeter Maydell 402ff5ca166SMax Reitz /* Need to use _SAFE because curl_multi_socket_action() may trigger 403ff5ca166SMax Reitz * curl_sock_cb() which might modify this list */ 404ff5ca166SMax Reitz QLIST_FOREACH_SAFE(socket, &s->sockets, next, next_socket) { 405031fd1beSPeter Maydell do { 406ff5ca166SMax Reitz r = curl_multi_socket_action(s->s->multi, socket->fd, 0, &running); 407031fd1beSPeter Maydell } while (r == CURLM_CALL_MULTI_PERFORM); 408ff5ca166SMax Reitz } 409838ef602SMatthew Booth } 410838ef602SMatthew Booth 4119d456654SPaolo Bonzini static void curl_multi_do(void *arg) 4129d456654SPaolo Bonzini { 4139d456654SPaolo Bonzini CURLState *s = (CURLState *)arg; 4149d456654SPaolo Bonzini 415ba3186c4SPaolo Bonzini qemu_mutex_lock(&s->s->mutex); 4169d456654SPaolo Bonzini curl_multi_do_locked(s); 417ba3186c4SPaolo Bonzini qemu_mutex_unlock(&s->s->mutex); 4189d456654SPaolo Bonzini } 4199d456654SPaolo Bonzini 420838ef602SMatthew Booth static void curl_multi_read(void *arg) 421838ef602SMatthew Booth { 422838ef602SMatthew Booth CURLState *s = (CURLState *)arg; 423838ef602SMatthew Booth 424ba3186c4SPaolo Bonzini qemu_mutex_lock(&s->s->mutex); 4259d456654SPaolo Bonzini curl_multi_do_locked(s); 426838ef602SMatthew Booth curl_multi_check_completion(s->s); 427ba3186c4SPaolo Bonzini qemu_mutex_unlock(&s->s->mutex); 428031fd1beSPeter Maydell } 429031fd1beSPeter Maydell 430031fd1beSPeter Maydell static void curl_multi_timeout_do(void *arg) 431031fd1beSPeter Maydell { 432031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK 433031fd1beSPeter Maydell BDRVCURLState *s = (BDRVCURLState *)arg; 434031fd1beSPeter Maydell int running; 435031fd1beSPeter Maydell 436031fd1beSPeter Maydell if (!s->multi) { 437031fd1beSPeter Maydell return; 438031fd1beSPeter Maydell } 439031fd1beSPeter Maydell 440ba3186c4SPaolo Bonzini qemu_mutex_lock(&s->mutex); 441031fd1beSPeter Maydell curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running); 442031fd1beSPeter Maydell 443838ef602SMatthew Booth curl_multi_check_completion(s); 444ba3186c4SPaolo Bonzini qemu_mutex_unlock(&s->mutex); 445031fd1beSPeter Maydell #else 446031fd1beSPeter Maydell abort(); 447031fd1beSPeter Maydell #endif 448031fd1beSPeter Maydell } 449031fd1beSPeter Maydell 450456af346SPaolo Bonzini /* Called with s->mutex held. */ 4513ce6a729SPaolo Bonzini static CURLState *curl_find_state(BDRVCURLState *s) 452769ce76dSAlexander Graf { 453769ce76dSAlexander Graf CURLState *state = NULL; 4543ce6a729SPaolo Bonzini int i; 455769ce76dSAlexander Graf 456769ce76dSAlexander Graf for (i = 0; i < CURL_NUM_STATES; i++) { 4573ce6a729SPaolo Bonzini if (!s->states[i].in_use) { 458769ce76dSAlexander Graf state = &s->states[i]; 459769ce76dSAlexander Graf state->in_use = 1; 460769ce76dSAlexander Graf break; 461769ce76dSAlexander Graf } 462769ce76dSAlexander Graf } 4633ce6a729SPaolo Bonzini return state; 4643ce6a729SPaolo Bonzini } 465769ce76dSAlexander Graf 4663ce6a729SPaolo Bonzini static int curl_init_state(BDRVCURLState *s, CURLState *state) 4673ce6a729SPaolo Bonzini { 4689e550b32SMatthew Booth if (!state->curl) { 469769ce76dSAlexander Graf state->curl = curl_easy_init(); 4709e550b32SMatthew Booth if (!state->curl) { 4713ce6a729SPaolo Bonzini return -EIO; 4729e550b32SMatthew Booth } 473769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_URL, s->url); 47497a3ea57SMatthew Booth curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYPEER, 47597a3ea57SMatthew Booth (long) s->sslverify); 476637fa44aSRichard W.M. Jones curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYHOST, 477637fa44aSRichard W.M. Jones s->sslverify ? 2L : 0L); 478a94f83d9SRichard W.M. Jones if (s->cookie) { 479a94f83d9SRichard W.M. Jones curl_easy_setopt(state->curl, CURLOPT_COOKIE, s->cookie); 480a94f83d9SRichard W.M. Jones } 481f76faedaSRichard W.M. Jones curl_easy_setopt(state->curl, CURLOPT_TIMEOUT, (long)s->timeout); 4829e550b32SMatthew Booth curl_easy_setopt(state->curl, CURLOPT_WRITEFUNCTION, 4839e550b32SMatthew Booth (void *)curl_read_cb); 484769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_WRITEDATA, (void *)state); 485769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_PRIVATE, (void *)state); 486769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_AUTOREFERER, 1); 487769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_FOLLOWLOCATION, 1); 488769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_NOSIGNAL, 1); 489769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_ERRORBUFFER, state->errmsg); 490f785a5aeSNicholas Thomas curl_easy_setopt(state->curl, CURLOPT_FAILONERROR, 1); 491769ce76dSAlexander Graf 4921bff9606SDaniel P. Berrange if (s->username) { 4931bff9606SDaniel P. Berrange curl_easy_setopt(state->curl, CURLOPT_USERNAME, s->username); 4941bff9606SDaniel P. Berrange } 4951bff9606SDaniel P. Berrange if (s->password) { 4961bff9606SDaniel P. Berrange curl_easy_setopt(state->curl, CURLOPT_PASSWORD, s->password); 4971bff9606SDaniel P. Berrange } 4981bff9606SDaniel P. Berrange if (s->proxyusername) { 4991bff9606SDaniel P. Berrange curl_easy_setopt(state->curl, 5001bff9606SDaniel P. Berrange CURLOPT_PROXYUSERNAME, s->proxyusername); 5011bff9606SDaniel P. Berrange } 5021bff9606SDaniel P. Berrange if (s->proxypassword) { 5031bff9606SDaniel P. Berrange curl_easy_setopt(state->curl, 5041bff9606SDaniel P. Berrange CURLOPT_PROXYPASSWORD, s->proxypassword); 5051bff9606SDaniel P. Berrange } 5061bff9606SDaniel P. Berrange 507fb6d1bbdSStefan Hajnoczi /* Restrict supported protocols to avoid security issues in the more 508fb6d1bbdSStefan Hajnoczi * obscure protocols. For example, do not allow POP3/SMTP/IMAP see 509fb6d1bbdSStefan Hajnoczi * CVE-2013-0249. 5108a8f5840SStefan Hajnoczi * 5118a8f5840SStefan Hajnoczi * Restricting protocols is only supported from 7.19.4 upwards. 512fb6d1bbdSStefan Hajnoczi */ 5138a8f5840SStefan Hajnoczi #if LIBCURL_VERSION_NUM >= 0x071304 514fb6d1bbdSStefan Hajnoczi curl_easy_setopt(state->curl, CURLOPT_PROTOCOLS, PROTOCOLS); 515fb6d1bbdSStefan Hajnoczi curl_easy_setopt(state->curl, CURLOPT_REDIR_PROTOCOLS, PROTOCOLS); 5168a8f5840SStefan Hajnoczi #endif 517fb6d1bbdSStefan Hajnoczi 518769ce76dSAlexander Graf #ifdef DEBUG_VERBOSE 519769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_VERBOSE, 1); 520769ce76dSAlexander Graf #endif 5219e550b32SMatthew Booth } 522769ce76dSAlexander Graf 523ff5ca166SMax Reitz QLIST_INIT(&state->sockets); 524769ce76dSAlexander Graf state->s = s; 525769ce76dSAlexander Graf 5263ce6a729SPaolo Bonzini return 0; 527769ce76dSAlexander Graf } 528769ce76dSAlexander Graf 529456af346SPaolo Bonzini /* Called with s->mutex held. */ 530769ce76dSAlexander Graf static void curl_clean_state(CURLState *s) 531769ce76dSAlexander Graf { 532675a7756SPaolo Bonzini int j; 533675a7756SPaolo Bonzini for (j = 0; j < CURL_NUM_ACB; j++) { 534675a7756SPaolo Bonzini assert(!s->acb[j]); 535675a7756SPaolo Bonzini } 536675a7756SPaolo Bonzini 537769ce76dSAlexander Graf if (s->s->multi) 538769ce76dSAlexander Graf curl_multi_remove_handle(s->s->multi, s->curl); 539ff5ca166SMax Reitz 540ff5ca166SMax Reitz while (!QLIST_EMPTY(&s->sockets)) { 541ff5ca166SMax Reitz CURLSocket *socket = QLIST_FIRST(&s->sockets); 542ff5ca166SMax Reitz 543ff5ca166SMax Reitz QLIST_REMOVE(socket, next); 544ff5ca166SMax Reitz g_free(socket); 545ff5ca166SMax Reitz } 546ff5ca166SMax Reitz 547769ce76dSAlexander Graf s->in_use = 0; 5482bb5c936SPaolo Bonzini 549709f2132SPaolo Bonzini qemu_co_enter_next(&s->s->free_state_waitq, &s->s->mutex); 550769ce76dSAlexander Graf } 551769ce76dSAlexander Graf 5528e6d58cdSKevin Wolf static void curl_parse_filename(const char *filename, QDict *options, 5538e6d58cdSKevin Wolf Error **errp) 554769ce76dSAlexander Graf { 55546f5ac20SEric Blake qdict_put_str(options, CURL_BLOCK_OPT_URL, filename); 5568e6d58cdSKevin Wolf } 5578e6d58cdSKevin Wolf 55863f0f45fSStefan Hajnoczi static void curl_detach_aio_context(BlockDriverState *bs) 55963f0f45fSStefan Hajnoczi { 56063f0f45fSStefan Hajnoczi BDRVCURLState *s = bs->opaque; 56163f0f45fSStefan Hajnoczi int i; 56263f0f45fSStefan Hajnoczi 563456af346SPaolo Bonzini qemu_mutex_lock(&s->mutex); 56463f0f45fSStefan Hajnoczi for (i = 0; i < CURL_NUM_STATES; i++) { 56563f0f45fSStefan Hajnoczi if (s->states[i].in_use) { 56663f0f45fSStefan Hajnoczi curl_clean_state(&s->states[i]); 56763f0f45fSStefan Hajnoczi } 56863f0f45fSStefan Hajnoczi if (s->states[i].curl) { 56963f0f45fSStefan Hajnoczi curl_easy_cleanup(s->states[i].curl); 57063f0f45fSStefan Hajnoczi s->states[i].curl = NULL; 57163f0f45fSStefan Hajnoczi } 57263f0f45fSStefan Hajnoczi g_free(s->states[i].orig_buf); 57363f0f45fSStefan Hajnoczi s->states[i].orig_buf = NULL; 57463f0f45fSStefan Hajnoczi } 57563f0f45fSStefan Hajnoczi if (s->multi) { 57663f0f45fSStefan Hajnoczi curl_multi_cleanup(s->multi); 57763f0f45fSStefan Hajnoczi s->multi = NULL; 57863f0f45fSStefan Hajnoczi } 579456af346SPaolo Bonzini qemu_mutex_unlock(&s->mutex); 58063f0f45fSStefan Hajnoczi 58163f0f45fSStefan Hajnoczi timer_del(&s->timer); 58263f0f45fSStefan Hajnoczi } 58363f0f45fSStefan Hajnoczi 58463f0f45fSStefan Hajnoczi static void curl_attach_aio_context(BlockDriverState *bs, 58563f0f45fSStefan Hajnoczi AioContext *new_context) 58663f0f45fSStefan Hajnoczi { 58763f0f45fSStefan Hajnoczi BDRVCURLState *s = bs->opaque; 58863f0f45fSStefan Hajnoczi 58963f0f45fSStefan Hajnoczi aio_timer_init(new_context, &s->timer, 59063f0f45fSStefan Hajnoczi QEMU_CLOCK_REALTIME, SCALE_NS, 59163f0f45fSStefan Hajnoczi curl_multi_timeout_do, s); 59263f0f45fSStefan Hajnoczi 59363f0f45fSStefan Hajnoczi assert(!s->multi); 59463f0f45fSStefan Hajnoczi s->multi = curl_multi_init(); 59563f0f45fSStefan Hajnoczi s->aio_context = new_context; 59663f0f45fSStefan Hajnoczi curl_multi_setopt(s->multi, CURLMOPT_SOCKETFUNCTION, curl_sock_cb); 59763f0f45fSStefan Hajnoczi #ifdef NEED_CURL_TIMER_CALLBACK 59863f0f45fSStefan Hajnoczi curl_multi_setopt(s->multi, CURLMOPT_TIMERDATA, s); 59963f0f45fSStefan Hajnoczi curl_multi_setopt(s->multi, CURLMOPT_TIMERFUNCTION, curl_timer_cb); 60063f0f45fSStefan Hajnoczi #endif 60163f0f45fSStefan Hajnoczi } 60263f0f45fSStefan Hajnoczi 6038e6d58cdSKevin Wolf static QemuOptsList runtime_opts = { 6048e6d58cdSKevin Wolf .name = "curl", 6058e6d58cdSKevin Wolf .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head), 6068e6d58cdSKevin Wolf .desc = { 6078e6d58cdSKevin Wolf { 608e3542c67SMatthew Booth .name = CURL_BLOCK_OPT_URL, 6098e6d58cdSKevin Wolf .type = QEMU_OPT_STRING, 6108e6d58cdSKevin Wolf .help = "URL to open", 6118e6d58cdSKevin Wolf }, 6128e6d58cdSKevin Wolf { 613e3542c67SMatthew Booth .name = CURL_BLOCK_OPT_READAHEAD, 6148e6d58cdSKevin Wolf .type = QEMU_OPT_SIZE, 6158e6d58cdSKevin Wolf .help = "Readahead size", 6168e6d58cdSKevin Wolf }, 61797a3ea57SMatthew Booth { 61897a3ea57SMatthew Booth .name = CURL_BLOCK_OPT_SSLVERIFY, 61997a3ea57SMatthew Booth .type = QEMU_OPT_BOOL, 62097a3ea57SMatthew Booth .help = "Verify SSL certificate" 62197a3ea57SMatthew Booth }, 622212aefaaSDaniel Henrique Barboza { 623212aefaaSDaniel Henrique Barboza .name = CURL_BLOCK_OPT_TIMEOUT, 624212aefaaSDaniel Henrique Barboza .type = QEMU_OPT_NUMBER, 625212aefaaSDaniel Henrique Barboza .help = "Curl timeout" 626212aefaaSDaniel Henrique Barboza }, 627a94f83d9SRichard W.M. Jones { 628a94f83d9SRichard W.M. Jones .name = CURL_BLOCK_OPT_COOKIE, 629a94f83d9SRichard W.M. Jones .type = QEMU_OPT_STRING, 630a94f83d9SRichard W.M. Jones .help = "Pass the cookie or list of cookies with each request" 631a94f83d9SRichard W.M. Jones }, 6321bff9606SDaniel P. Berrange { 633327c8ebdSPeter Krempa .name = CURL_BLOCK_OPT_COOKIE_SECRET, 634327c8ebdSPeter Krempa .type = QEMU_OPT_STRING, 635327c8ebdSPeter Krempa .help = "ID of secret used as cookie passed with each request" 636327c8ebdSPeter Krempa }, 637327c8ebdSPeter Krempa { 6381bff9606SDaniel P. Berrange .name = CURL_BLOCK_OPT_USERNAME, 6391bff9606SDaniel P. Berrange .type = QEMU_OPT_STRING, 6401bff9606SDaniel P. Berrange .help = "Username for HTTP auth" 6411bff9606SDaniel P. Berrange }, 6421bff9606SDaniel P. Berrange { 6431bff9606SDaniel P. Berrange .name = CURL_BLOCK_OPT_PASSWORD_SECRET, 6441bff9606SDaniel P. Berrange .type = QEMU_OPT_STRING, 6451bff9606SDaniel P. Berrange .help = "ID of secret used as password for HTTP auth", 6461bff9606SDaniel P. Berrange }, 6471bff9606SDaniel P. Berrange { 6481bff9606SDaniel P. Berrange .name = CURL_BLOCK_OPT_PROXY_USERNAME, 6491bff9606SDaniel P. Berrange .type = QEMU_OPT_STRING, 6501bff9606SDaniel P. Berrange .help = "Username for HTTP proxy auth" 6511bff9606SDaniel P. Berrange }, 6521bff9606SDaniel P. Berrange { 6531bff9606SDaniel P. Berrange .name = CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET, 6541bff9606SDaniel P. Berrange .type = QEMU_OPT_STRING, 6551bff9606SDaniel P. Berrange .help = "ID of secret used as password for HTTP proxy auth", 6561bff9606SDaniel P. Berrange }, 6578e6d58cdSKevin Wolf { /* end of list */ } 6588e6d58cdSKevin Wolf }, 6598e6d58cdSKevin Wolf }; 6608e6d58cdSKevin Wolf 6611bff9606SDaniel P. Berrange 662015a1036SMax Reitz static int curl_open(BlockDriverState *bs, QDict *options, int flags, 663015a1036SMax Reitz Error **errp) 6648e6d58cdSKevin Wolf { 6658e6d58cdSKevin Wolf BDRVCURLState *s = bs->opaque; 6668e6d58cdSKevin Wolf CURLState *state = NULL; 6678e6d58cdSKevin Wolf QemuOpts *opts; 6688e6d58cdSKevin Wolf Error *local_err = NULL; 6698e6d58cdSKevin Wolf const char *file; 670a94f83d9SRichard W.M. Jones const char *cookie; 671327c8ebdSPeter Krempa const char *cookie_secret; 6728e6d58cdSKevin Wolf double d; 6731bff9606SDaniel P. Berrange const char *secretid; 67434634ca2SMax Reitz const char *protocol_delimiter; 6752d25964dSJeff Cody int ret; 6768e6d58cdSKevin Wolf 6776ceef36aSKevin Wolf ret = bdrv_apply_auto_read_only(bs, "curl driver does not support writes", 6786ceef36aSKevin Wolf errp); 6796ceef36aSKevin Wolf if (ret < 0) { 6806ceef36aSKevin Wolf return ret; 681a7cea2baSRichard W.M. Jones } 682a7cea2baSRichard W.M. Jones 6832d25964dSJeff Cody if (!libcurl_initialized) { 6842d25964dSJeff Cody ret = curl_global_init(CURL_GLOBAL_ALL); 6852d25964dSJeff Cody if (ret) { 6862d25964dSJeff Cody error_setg(errp, "libcurl initialization failed with %d", ret); 6872d25964dSJeff Cody return -EIO; 6882d25964dSJeff Cody } 6892d25964dSJeff Cody libcurl_initialized = true; 6902d25964dSJeff Cody } 6912d25964dSJeff Cody 692456af346SPaolo Bonzini qemu_mutex_init(&s->mutex); 69387ea75d5SPeter Crosthwaite opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort); 6948e6d58cdSKevin Wolf qemu_opts_absorb_qdict(opts, options, &local_err); 69584d18f06SMarkus Armbruster if (local_err) { 6962a94fee3SPaolo Bonzini error_propagate(errp, local_err); 6978e6d58cdSKevin Wolf goto out_noclean; 6988e6d58cdSKevin Wolf } 6998e6d58cdSKevin Wolf 700e3542c67SMatthew Booth s->readahead_size = qemu_opt_get_size(opts, CURL_BLOCK_OPT_READAHEAD, 701712b64e8SMax Reitz CURL_BLOCK_OPT_READAHEAD_DEFAULT); 702c76f4952SNolan if ((s->readahead_size & 0x1ff) != 0) { 7032a94fee3SPaolo Bonzini error_setg(errp, "HTTP_READAHEAD_SIZE %zd is not a multiple of 512", 704c76f4952SNolan s->readahead_size); 705c76f4952SNolan goto out_noclean; 706c76f4952SNolan } 707c76f4952SNolan 708212aefaaSDaniel Henrique Barboza s->timeout = qemu_opt_get_number(opts, CURL_BLOCK_OPT_TIMEOUT, 709712b64e8SMax Reitz CURL_BLOCK_OPT_TIMEOUT_DEFAULT); 710f76faedaSRichard W.M. Jones if (s->timeout > CURL_TIMEOUT_MAX) { 711f76faedaSRichard W.M. Jones error_setg(errp, "timeout parameter is too large or negative"); 712f76faedaSRichard W.M. Jones goto out_noclean; 713f76faedaSRichard W.M. Jones } 714212aefaaSDaniel Henrique Barboza 715712b64e8SMax Reitz s->sslverify = qemu_opt_get_bool(opts, CURL_BLOCK_OPT_SSLVERIFY, 716712b64e8SMax Reitz CURL_BLOCK_OPT_SSLVERIFY_DEFAULT); 71797a3ea57SMatthew Booth 718a94f83d9SRichard W.M. Jones cookie = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE); 719327c8ebdSPeter Krempa cookie_secret = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE_SECRET); 720327c8ebdSPeter Krempa 721327c8ebdSPeter Krempa if (cookie && cookie_secret) { 722327c8ebdSPeter Krempa error_setg(errp, 723327c8ebdSPeter Krempa "curl driver cannot handle both cookie and cookie secret"); 724327c8ebdSPeter Krempa goto out_noclean; 725327c8ebdSPeter Krempa } 726327c8ebdSPeter Krempa 727327c8ebdSPeter Krempa if (cookie_secret) { 728327c8ebdSPeter Krempa s->cookie = qcrypto_secret_lookup_as_utf8(cookie_secret, errp); 729327c8ebdSPeter Krempa if (!s->cookie) { 730327c8ebdSPeter Krempa goto out_noclean; 731327c8ebdSPeter Krempa } 732327c8ebdSPeter Krempa } else { 733a94f83d9SRichard W.M. Jones s->cookie = g_strdup(cookie); 734327c8ebdSPeter Krempa } 735a94f83d9SRichard W.M. Jones 736e3542c67SMatthew Booth file = qemu_opt_get(opts, CURL_BLOCK_OPT_URL); 7378e6d58cdSKevin Wolf if (file == NULL) { 7382a94fee3SPaolo Bonzini error_setg(errp, "curl block driver requires an 'url' option"); 7398e6d58cdSKevin Wolf goto out_noclean; 7408e6d58cdSKevin Wolf } 7418e6d58cdSKevin Wolf 74234634ca2SMax Reitz if (!strstart(file, bs->drv->protocol_name, &protocol_delimiter) || 74334634ca2SMax Reitz !strstart(protocol_delimiter, "://", NULL)) 74434634ca2SMax Reitz { 74534634ca2SMax Reitz error_setg(errp, "%s curl driver cannot handle the URL '%s' (does not " 74634634ca2SMax Reitz "start with '%s://')", bs->drv->protocol_name, file, 74734634ca2SMax Reitz bs->drv->protocol_name); 74834634ca2SMax Reitz goto out_noclean; 74934634ca2SMax Reitz } 75034634ca2SMax Reitz 7511bff9606SDaniel P. Berrange s->username = g_strdup(qemu_opt_get(opts, CURL_BLOCK_OPT_USERNAME)); 7521bff9606SDaniel P. Berrange secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PASSWORD_SECRET); 7531bff9606SDaniel P. Berrange 7541bff9606SDaniel P. Berrange if (secretid) { 7551bff9606SDaniel P. Berrange s->password = qcrypto_secret_lookup_as_utf8(secretid, errp); 7561bff9606SDaniel P. Berrange if (!s->password) { 7571bff9606SDaniel P. Berrange goto out_noclean; 7581bff9606SDaniel P. Berrange } 7591bff9606SDaniel P. Berrange } 7601bff9606SDaniel P. Berrange 7611bff9606SDaniel P. Berrange s->proxyusername = g_strdup( 7621bff9606SDaniel P. Berrange qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_USERNAME)); 7631bff9606SDaniel P. Berrange secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET); 7641bff9606SDaniel P. Berrange if (secretid) { 7651bff9606SDaniel P. Berrange s->proxypassword = qcrypto_secret_lookup_as_utf8(secretid, errp); 7661bff9606SDaniel P. Berrange if (!s->proxypassword) { 7671bff9606SDaniel P. Berrange goto out_noclean; 7681bff9606SDaniel P. Berrange } 7691bff9606SDaniel P. Berrange } 7701bff9606SDaniel P. Berrange 771ed2a66deSLaurent Vivier trace_curl_open(file); 772709f2132SPaolo Bonzini qemu_co_queue_init(&s->free_state_waitq); 77363f0f45fSStefan Hajnoczi s->aio_context = bdrv_get_aio_context(bs); 7748e6d58cdSKevin Wolf s->url = g_strdup(file); 775456af346SPaolo Bonzini qemu_mutex_lock(&s->mutex); 7763ce6a729SPaolo Bonzini state = curl_find_state(s); 777456af346SPaolo Bonzini qemu_mutex_unlock(&s->mutex); 7783ce6a729SPaolo Bonzini if (!state) { 779769ce76dSAlexander Graf goto out_noclean; 7803ce6a729SPaolo Bonzini } 781769ce76dSAlexander Graf 782769ce76dSAlexander Graf // Get file size 783769ce76dSAlexander Graf 7843ce6a729SPaolo Bonzini if (curl_init_state(s, state) < 0) { 7853ce6a729SPaolo Bonzini goto out; 7863ce6a729SPaolo Bonzini } 7873ce6a729SPaolo Bonzini 7883494d650SFam Zheng s->accept_range = false; 789769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_NOBODY, 1); 7903494d650SFam Zheng curl_easy_setopt(state->curl, CURLOPT_HEADERFUNCTION, 7913494d650SFam Zheng curl_header_cb); 7923494d650SFam Zheng curl_easy_setopt(state->curl, CURLOPT_HEADERDATA, s); 793769ce76dSAlexander Graf if (curl_easy_perform(state->curl)) 794769ce76dSAlexander Graf goto out; 795a41c4578STomáš Golembiovský if (curl_easy_getinfo(state->curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d)) { 796769ce76dSAlexander Graf goto out; 797a41c4578STomáš Golembiovský } 798a41c4578STomáš Golembiovský /* Prior CURL 7.19.4 return value of 0 could mean that the file size is not 799a41c4578STomáš Golembiovský * know or the size is zero. From 7.19.4 CURL returns -1 if size is not 80050d6a8a3SStefan Weil * known and zero if it is really zero-length file. */ 801a41c4578STomáš Golembiovský #if LIBCURL_VERSION_NUM >= 0x071304 802a41c4578STomáš Golembiovský if (d < 0) { 803a41c4578STomáš Golembiovský pstrcpy(state->errmsg, CURL_ERROR_SIZE, 804a41c4578STomáš Golembiovský "Server didn't report file size."); 805a41c4578STomáš Golembiovský goto out; 806a41c4578STomáš Golembiovský } 807a41c4578STomáš Golembiovský #else 808a41c4578STomáš Golembiovský if (d <= 0) { 809a41c4578STomáš Golembiovský pstrcpy(state->errmsg, CURL_ERROR_SIZE, 810a41c4578STomáš Golembiovský "Unknown file size or zero-length file."); 811a41c4578STomáš Golembiovský goto out; 812a41c4578STomáš Golembiovský } 813a41c4578STomáš Golembiovský #endif 814a41c4578STomáš Golembiovský 8152125e5eaSPaolo Bonzini s->len = d; 816a41c4578STomáš Golembiovský 8173494d650SFam Zheng if ((!strncasecmp(s->url, "http://", strlen("http://")) 8183494d650SFam Zheng || !strncasecmp(s->url, "https://", strlen("https://"))) 8193494d650SFam Zheng && !s->accept_range) { 8203494d650SFam Zheng pstrcpy(state->errmsg, CURL_ERROR_SIZE, 8213494d650SFam Zheng "Server does not support 'range' (byte ranges)."); 8223494d650SFam Zheng goto out; 8233494d650SFam Zheng } 824ed2a66deSLaurent Vivier trace_curl_open_size(s->len); 825769ce76dSAlexander Graf 826456af346SPaolo Bonzini qemu_mutex_lock(&s->mutex); 827769ce76dSAlexander Graf curl_clean_state(state); 828456af346SPaolo Bonzini qemu_mutex_unlock(&s->mutex); 829769ce76dSAlexander Graf curl_easy_cleanup(state->curl); 830769ce76dSAlexander Graf state->curl = NULL; 831769ce76dSAlexander Graf 83263f0f45fSStefan Hajnoczi curl_attach_aio_context(bs, bdrv_get_aio_context(bs)); 833769ce76dSAlexander Graf 8348e6d58cdSKevin Wolf qemu_opts_del(opts); 835769ce76dSAlexander Graf return 0; 836769ce76dSAlexander Graf 837769ce76dSAlexander Graf out: 838acd7fdc6SMaria Kustova error_setg(errp, "CURL: Error opening file: %s", state->errmsg); 839769ce76dSAlexander Graf curl_easy_cleanup(state->curl); 840769ce76dSAlexander Graf state->curl = NULL; 841769ce76dSAlexander Graf out_noclean: 842456af346SPaolo Bonzini qemu_mutex_destroy(&s->mutex); 843a94f83d9SRichard W.M. Jones g_free(s->cookie); 8448e6d58cdSKevin Wolf g_free(s->url); 845996922deSJeff Cody g_free(s->username); 846996922deSJeff Cody g_free(s->proxyusername); 847996922deSJeff Cody g_free(s->proxypassword); 8488e6d58cdSKevin Wolf qemu_opts_del(opts); 849769ce76dSAlexander Graf return -EINVAL; 850769ce76dSAlexander Graf } 851769ce76dSAlexander Graf 85228256d82SPaolo Bonzini static void curl_setup_preadv(BlockDriverState *bs, CURLAIOCB *acb) 853769ce76dSAlexander Graf { 854769ce76dSAlexander Graf CURLState *state; 855b69cdef8SMatthew Booth int running; 856769ce76dSAlexander Graf 8571919631eSPaolo Bonzini BDRVCURLState *s = bs->opaque; 858769ce76dSAlexander Graf 8592125e5eaSPaolo Bonzini uint64_t start = acb->offset; 8602125e5eaSPaolo Bonzini uint64_t end; 861769ce76dSAlexander Graf 862ba3186c4SPaolo Bonzini qemu_mutex_lock(&s->mutex); 8631919631eSPaolo Bonzini 864769ce76dSAlexander Graf // In case we have the requested data already (e.g. read-ahead), 865769ce76dSAlexander Graf // we can just call the callback and be done. 86628256d82SPaolo Bonzini if (curl_find_buf(s, start, acb->bytes, acb)) { 86734db05e7SPaolo Bonzini goto out; 868769ce76dSAlexander Graf } 869769ce76dSAlexander Graf 870769ce76dSAlexander Graf // No cache found, so let's start a new request 8713ce6a729SPaolo Bonzini for (;;) { 8723ce6a729SPaolo Bonzini state = curl_find_state(s); 8733ce6a729SPaolo Bonzini if (state) { 8743ce6a729SPaolo Bonzini break; 8753ce6a729SPaolo Bonzini } 876709f2132SPaolo Bonzini qemu_co_queue_wait(&s->free_state_waitq, &s->mutex); 8773ce6a729SPaolo Bonzini } 8783ce6a729SPaolo Bonzini 8793ce6a729SPaolo Bonzini if (curl_init_state(s, state) < 0) { 8803ce6a729SPaolo Bonzini curl_clean_state(state); 88128256d82SPaolo Bonzini acb->ret = -EIO; 8821919631eSPaolo Bonzini goto out; 883363c3c85SNick Thomas } 884769ce76dSAlexander Graf 885769ce76dSAlexander Graf acb->start = 0; 8862125e5eaSPaolo Bonzini acb->end = MIN(acb->bytes, s->len - start); 887769ce76dSAlexander Graf 888769ce76dSAlexander Graf state->buf_off = 0; 8897267c094SAnthony Liguori g_free(state->orig_buf); 890769ce76dSAlexander Graf state->buf_start = start; 8914e504535SMax Reitz state->buf_len = MIN(acb->end + s->readahead_size, s->len - start); 8924e504535SMax Reitz end = start + state->buf_len - 1; 8938dc7a772SKevin Wolf state->orig_buf = g_try_malloc(state->buf_len); 8948dc7a772SKevin Wolf if (state->buf_len && state->orig_buf == NULL) { 8958dc7a772SKevin Wolf curl_clean_state(state); 89628256d82SPaolo Bonzini acb->ret = -ENOMEM; 8971919631eSPaolo Bonzini goto out; 8988dc7a772SKevin Wolf } 899769ce76dSAlexander Graf state->acb[0] = acb; 900769ce76dSAlexander Graf 9012125e5eaSPaolo Bonzini snprintf(state->range, 127, "%" PRIu64 "-%" PRIu64, start, end); 902ed2a66deSLaurent Vivier trace_curl_setup_preadv(acb->bytes, start, state->range); 903769ce76dSAlexander Graf curl_easy_setopt(state->curl, CURLOPT_RANGE, state->range); 904769ce76dSAlexander Graf 905769ce76dSAlexander Graf curl_multi_add_handle(s->multi, state->curl); 906769ce76dSAlexander Graf 907b69cdef8SMatthew Booth /* Tell curl it needs to kick things off */ 908b69cdef8SMatthew Booth curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running); 9091919631eSPaolo Bonzini 9101919631eSPaolo Bonzini out: 911ba3186c4SPaolo Bonzini qemu_mutex_unlock(&s->mutex); 912363c3c85SNick Thomas } 913363c3c85SNick Thomas 91428256d82SPaolo Bonzini static int coroutine_fn curl_co_preadv(BlockDriverState *bs, 91528256d82SPaolo Bonzini uint64_t offset, uint64_t bytes, QEMUIOVector *qiov, int flags) 916363c3c85SNick Thomas { 91728256d82SPaolo Bonzini CURLAIOCB acb = { 91828256d82SPaolo Bonzini .co = qemu_coroutine_self(), 91928256d82SPaolo Bonzini .ret = -EINPROGRESS, 92028256d82SPaolo Bonzini .qiov = qiov, 92128256d82SPaolo Bonzini .offset = offset, 92228256d82SPaolo Bonzini .bytes = bytes 92328256d82SPaolo Bonzini }; 924363c3c85SNick Thomas 92528256d82SPaolo Bonzini curl_setup_preadv(bs, &acb); 92628256d82SPaolo Bonzini while (acb.ret == -EINPROGRESS) { 92728256d82SPaolo Bonzini qemu_coroutine_yield(); 92828256d82SPaolo Bonzini } 92928256d82SPaolo Bonzini return acb.ret; 930769ce76dSAlexander Graf } 931769ce76dSAlexander Graf 932769ce76dSAlexander Graf static void curl_close(BlockDriverState *bs) 933769ce76dSAlexander Graf { 934769ce76dSAlexander Graf BDRVCURLState *s = bs->opaque; 935769ce76dSAlexander Graf 936ed2a66deSLaurent Vivier trace_curl_close(); 93763f0f45fSStefan Hajnoczi curl_detach_aio_context(bs); 938ba3186c4SPaolo Bonzini qemu_mutex_destroy(&s->mutex); 939031fd1beSPeter Maydell 940a94f83d9SRichard W.M. Jones g_free(s->cookie); 94145724d6dSStefan Weil g_free(s->url); 942996922deSJeff Cody g_free(s->username); 943996922deSJeff Cody g_free(s->proxyusername); 944996922deSJeff Cody g_free(s->proxypassword); 945769ce76dSAlexander Graf } 946769ce76dSAlexander Graf 947769ce76dSAlexander Graf static int64_t curl_getlength(BlockDriverState *bs) 948769ce76dSAlexander Graf { 949769ce76dSAlexander Graf BDRVCURLState *s = bs->opaque; 950769ce76dSAlexander Graf return s->len; 951769ce76dSAlexander Graf } 952769ce76dSAlexander Graf 953*937c007bSMax Reitz static void curl_refresh_filename(BlockDriverState *bs) 954*937c007bSMax Reitz { 955*937c007bSMax Reitz BDRVCURLState *s = bs->opaque; 956*937c007bSMax Reitz 957*937c007bSMax Reitz /* "readahead" and "timeout" do not change the guest-visible data, 958*937c007bSMax Reitz * so ignore them */ 959*937c007bSMax Reitz if (s->sslverify != CURL_BLOCK_OPT_SSLVERIFY_DEFAULT || 960*937c007bSMax Reitz s->cookie || s->username || s->password || s->proxyusername || 961*937c007bSMax Reitz s->proxypassword) 962*937c007bSMax Reitz { 963*937c007bSMax Reitz return; 964*937c007bSMax Reitz } 965*937c007bSMax Reitz 966*937c007bSMax Reitz pstrcpy(bs->exact_filename, sizeof(bs->exact_filename), s->url); 967*937c007bSMax Reitz } 968*937c007bSMax Reitz 969*937c007bSMax Reitz 9702654267cSMax Reitz static const char *const curl_strong_runtime_opts[] = { 9712654267cSMax Reitz CURL_BLOCK_OPT_URL, 9722654267cSMax Reitz CURL_BLOCK_OPT_SSLVERIFY, 9732654267cSMax Reitz CURL_BLOCK_OPT_COOKIE, 9742654267cSMax Reitz CURL_BLOCK_OPT_COOKIE_SECRET, 9752654267cSMax Reitz CURL_BLOCK_OPT_USERNAME, 9762654267cSMax Reitz CURL_BLOCK_OPT_PASSWORD_SECRET, 9772654267cSMax Reitz CURL_BLOCK_OPT_PROXY_USERNAME, 9782654267cSMax Reitz CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET, 9792654267cSMax Reitz 9802654267cSMax Reitz NULL 9812654267cSMax Reitz }; 9822654267cSMax Reitz 983769ce76dSAlexander Graf static BlockDriver bdrv_http = { 984769ce76dSAlexander Graf .format_name = "http", 985769ce76dSAlexander Graf .protocol_name = "http", 986769ce76dSAlexander Graf 987769ce76dSAlexander Graf .instance_size = sizeof(BDRVCURLState), 9888e6d58cdSKevin Wolf .bdrv_parse_filename = curl_parse_filename, 98966f82ceeSKevin Wolf .bdrv_file_open = curl_open, 990769ce76dSAlexander Graf .bdrv_close = curl_close, 991769ce76dSAlexander Graf .bdrv_getlength = curl_getlength, 992769ce76dSAlexander Graf 99328256d82SPaolo Bonzini .bdrv_co_preadv = curl_co_preadv, 99463f0f45fSStefan Hajnoczi 99563f0f45fSStefan Hajnoczi .bdrv_detach_aio_context = curl_detach_aio_context, 99663f0f45fSStefan Hajnoczi .bdrv_attach_aio_context = curl_attach_aio_context, 9972654267cSMax Reitz 998*937c007bSMax Reitz .bdrv_refresh_filename = curl_refresh_filename, 9992654267cSMax Reitz .strong_runtime_opts = curl_strong_runtime_opts, 1000769ce76dSAlexander Graf }; 1001769ce76dSAlexander Graf 1002769ce76dSAlexander Graf static BlockDriver bdrv_https = { 1003769ce76dSAlexander Graf .format_name = "https", 1004769ce76dSAlexander Graf .protocol_name = "https", 1005769ce76dSAlexander Graf 1006769ce76dSAlexander Graf .instance_size = sizeof(BDRVCURLState), 10078e6d58cdSKevin Wolf .bdrv_parse_filename = curl_parse_filename, 100866f82ceeSKevin Wolf .bdrv_file_open = curl_open, 1009769ce76dSAlexander Graf .bdrv_close = curl_close, 1010769ce76dSAlexander Graf .bdrv_getlength = curl_getlength, 1011769ce76dSAlexander Graf 101228256d82SPaolo Bonzini .bdrv_co_preadv = curl_co_preadv, 101363f0f45fSStefan Hajnoczi 101463f0f45fSStefan Hajnoczi .bdrv_detach_aio_context = curl_detach_aio_context, 101563f0f45fSStefan Hajnoczi .bdrv_attach_aio_context = curl_attach_aio_context, 10162654267cSMax Reitz 1017*937c007bSMax Reitz .bdrv_refresh_filename = curl_refresh_filename, 10182654267cSMax Reitz .strong_runtime_opts = curl_strong_runtime_opts, 1019769ce76dSAlexander Graf }; 1020769ce76dSAlexander Graf 1021769ce76dSAlexander Graf static BlockDriver bdrv_ftp = { 1022769ce76dSAlexander Graf .format_name = "ftp", 1023769ce76dSAlexander Graf .protocol_name = "ftp", 1024769ce76dSAlexander Graf 1025769ce76dSAlexander Graf .instance_size = sizeof(BDRVCURLState), 10268e6d58cdSKevin Wolf .bdrv_parse_filename = curl_parse_filename, 102766f82ceeSKevin Wolf .bdrv_file_open = curl_open, 1028769ce76dSAlexander Graf .bdrv_close = curl_close, 1029769ce76dSAlexander Graf .bdrv_getlength = curl_getlength, 1030769ce76dSAlexander Graf 103128256d82SPaolo Bonzini .bdrv_co_preadv = curl_co_preadv, 103263f0f45fSStefan Hajnoczi 103363f0f45fSStefan Hajnoczi .bdrv_detach_aio_context = curl_detach_aio_context, 103463f0f45fSStefan Hajnoczi .bdrv_attach_aio_context = curl_attach_aio_context, 10352654267cSMax Reitz 1036*937c007bSMax Reitz .bdrv_refresh_filename = curl_refresh_filename, 10372654267cSMax Reitz .strong_runtime_opts = curl_strong_runtime_opts, 1038769ce76dSAlexander Graf }; 1039769ce76dSAlexander Graf 1040769ce76dSAlexander Graf static BlockDriver bdrv_ftps = { 1041769ce76dSAlexander Graf .format_name = "ftps", 1042769ce76dSAlexander Graf .protocol_name = "ftps", 1043769ce76dSAlexander Graf 1044769ce76dSAlexander Graf .instance_size = sizeof(BDRVCURLState), 10458e6d58cdSKevin Wolf .bdrv_parse_filename = curl_parse_filename, 104666f82ceeSKevin Wolf .bdrv_file_open = curl_open, 1047769ce76dSAlexander Graf .bdrv_close = curl_close, 1048769ce76dSAlexander Graf .bdrv_getlength = curl_getlength, 1049769ce76dSAlexander Graf 105028256d82SPaolo Bonzini .bdrv_co_preadv = curl_co_preadv, 105163f0f45fSStefan Hajnoczi 105263f0f45fSStefan Hajnoczi .bdrv_detach_aio_context = curl_detach_aio_context, 105363f0f45fSStefan Hajnoczi .bdrv_attach_aio_context = curl_attach_aio_context, 10542654267cSMax Reitz 1055*937c007bSMax Reitz .bdrv_refresh_filename = curl_refresh_filename, 10562654267cSMax Reitz .strong_runtime_opts = curl_strong_runtime_opts, 1057769ce76dSAlexander Graf }; 1058769ce76dSAlexander Graf 1059769ce76dSAlexander Graf static void curl_block_init(void) 1060769ce76dSAlexander Graf { 1061769ce76dSAlexander Graf bdrv_register(&bdrv_http); 1062769ce76dSAlexander Graf bdrv_register(&bdrv_https); 1063769ce76dSAlexander Graf bdrv_register(&bdrv_ftp); 1064769ce76dSAlexander Graf bdrv_register(&bdrv_ftps); 1065769ce76dSAlexander Graf } 1066769ce76dSAlexander Graf 1067769ce76dSAlexander Graf block_init(curl_block_init); 1068