xref: /openbmc/qemu/block/curl.c (revision 9abaf9fc474c3dd53e8e119326abc774c977c331)
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"
280b8fa32fSMarkus Armbruster #include "qemu/module.h"
29922a01a0SMarkus Armbruster #include "qemu/option.h"
30737e150eSPaolo Bonzini #include "block/block_int.h"
31452fcdbcSMarkus Armbruster #include "qapi/qmp/qdict.h"
32d49b6836SMarkus Armbruster #include "qapi/qmp/qstring.h"
331bff9606SDaniel P. Berrange #include "crypto/secret.h"
34769ce76dSAlexander Graf #include <curl/curl.h>
35f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
36ed2a66deSLaurent Vivier #include "trace.h"
37769ce76dSAlexander Graf 
38769ce76dSAlexander Graf // #define DEBUG_VERBOSE
39769ce76dSAlexander Graf 
40031fd1beSPeter Maydell #if LIBCURL_VERSION_NUM >= 0x071000
41031fd1beSPeter Maydell /* The multi interface timer callback was introduced in 7.16.0 */
42031fd1beSPeter Maydell #define NEED_CURL_TIMER_CALLBACK
439aedd5a5SMatthew Booth #define HAVE_SOCKET_ACTION
449aedd5a5SMatthew Booth #endif
459aedd5a5SMatthew Booth 
469aedd5a5SMatthew Booth #ifndef HAVE_SOCKET_ACTION
479aedd5a5SMatthew Booth /* If curl_multi_socket_action isn't available, define it statically here in
489aedd5a5SMatthew Booth  * terms of curl_multi_socket. Note that ev_bitmask will be ignored, which is
499aedd5a5SMatthew Booth  * less efficient but still safe. */
509aedd5a5SMatthew Booth static CURLMcode __curl_multi_socket_action(CURLM *multi_handle,
519aedd5a5SMatthew Booth                                             curl_socket_t sockfd,
529aedd5a5SMatthew Booth                                             int ev_bitmask,
539aedd5a5SMatthew Booth                                             int *running_handles)
549aedd5a5SMatthew Booth {
559aedd5a5SMatthew Booth     return curl_multi_socket(multi_handle, sockfd, running_handles);
569aedd5a5SMatthew Booth }
579aedd5a5SMatthew Booth #define curl_multi_socket_action __curl_multi_socket_action
58031fd1beSPeter Maydell #endif
59031fd1beSPeter Maydell 
60fb6d1bbdSStefan Hajnoczi #define PROTOCOLS (CURLPROTO_HTTP | CURLPROTO_HTTPS | \
6123dce387SMax Reitz                    CURLPROTO_FTP | CURLPROTO_FTPS)
62fb6d1bbdSStefan Hajnoczi 
63769ce76dSAlexander Graf #define CURL_NUM_STATES 8
64769ce76dSAlexander Graf #define CURL_NUM_ACB    8
65f76faedaSRichard W.M. Jones #define CURL_TIMEOUT_MAX 10000
66769ce76dSAlexander Graf 
67e3542c67SMatthew Booth #define CURL_BLOCK_OPT_URL       "url"
68e3542c67SMatthew Booth #define CURL_BLOCK_OPT_READAHEAD "readahead"
6997a3ea57SMatthew Booth #define CURL_BLOCK_OPT_SSLVERIFY "sslverify"
70212aefaaSDaniel Henrique Barboza #define CURL_BLOCK_OPT_TIMEOUT "timeout"
71a94f83d9SRichard W.M. Jones #define CURL_BLOCK_OPT_COOKIE    "cookie"
72327c8ebdSPeter Krempa #define CURL_BLOCK_OPT_COOKIE_SECRET "cookie-secret"
731bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_USERNAME "username"
741bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PASSWORD_SECRET "password-secret"
751bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_USERNAME "proxy-username"
761bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET "proxy-password-secret"
77e3542c67SMatthew Booth 
78712b64e8SMax Reitz #define CURL_BLOCK_OPT_READAHEAD_DEFAULT (256 * 1024)
79712b64e8SMax Reitz #define CURL_BLOCK_OPT_SSLVERIFY_DEFAULT true
80712b64e8SMax Reitz #define CURL_BLOCK_OPT_TIMEOUT_DEFAULT 5
81712b64e8SMax Reitz 
82769ce76dSAlexander Graf struct BDRVCURLState;
8304878616SMax Reitz struct CURLState;
84769ce76dSAlexander Graf 
852d25964dSJeff Cody static bool libcurl_initialized;
862d25964dSJeff Cody 
87769ce76dSAlexander Graf typedef struct CURLAIOCB {
8828256d82SPaolo Bonzini     Coroutine *co;
89769ce76dSAlexander Graf     QEMUIOVector *qiov;
90363c3c85SNick Thomas 
912125e5eaSPaolo Bonzini     uint64_t offset;
922125e5eaSPaolo Bonzini     uint64_t bytes;
9328256d82SPaolo Bonzini     int ret;
94363c3c85SNick Thomas 
95769ce76dSAlexander Graf     size_t start;
96769ce76dSAlexander Graf     size_t end;
97769ce76dSAlexander Graf } CURLAIOCB;
98769ce76dSAlexander Graf 
99ff5ca166SMax Reitz typedef struct CURLSocket {
100ff5ca166SMax Reitz     int fd;
10104878616SMax Reitz     struct CURLState *state;
102ff5ca166SMax Reitz     QLIST_ENTRY(CURLSocket) next;
103ff5ca166SMax Reitz } CURLSocket;
104ff5ca166SMax Reitz 
105769ce76dSAlexander Graf typedef struct CURLState
106769ce76dSAlexander Graf {
107769ce76dSAlexander Graf     struct BDRVCURLState *s;
108769ce76dSAlexander Graf     CURLAIOCB *acb[CURL_NUM_ACB];
109769ce76dSAlexander Graf     CURL *curl;
110ff5ca166SMax Reitz     QLIST_HEAD(, CURLSocket) sockets;
111769ce76dSAlexander Graf     char *orig_buf;
1122125e5eaSPaolo Bonzini     uint64_t buf_start;
113769ce76dSAlexander Graf     size_t buf_off;
114769ce76dSAlexander Graf     size_t buf_len;
115769ce76dSAlexander Graf     char range[128];
116769ce76dSAlexander Graf     char errmsg[CURL_ERROR_SIZE];
117769ce76dSAlexander Graf     char in_use;
118769ce76dSAlexander Graf } CURLState;
119769ce76dSAlexander Graf 
120769ce76dSAlexander Graf typedef struct BDRVCURLState {
121769ce76dSAlexander Graf     CURLM *multi;
122031fd1beSPeter Maydell     QEMUTimer timer;
1232125e5eaSPaolo Bonzini     uint64_t len;
124769ce76dSAlexander Graf     CURLState states[CURL_NUM_STATES];
125769ce76dSAlexander Graf     char *url;
126c76f4952SNolan     size_t readahead_size;
12797a3ea57SMatthew Booth     bool sslverify;
128f76faedaSRichard W.M. Jones     uint64_t timeout;
129a94f83d9SRichard W.M. Jones     char *cookie;
1303494d650SFam Zheng     bool accept_range;
13163f0f45fSStefan Hajnoczi     AioContext *aio_context;
132ba3186c4SPaolo Bonzini     QemuMutex mutex;
133709f2132SPaolo Bonzini     CoQueue free_state_waitq;
1341bff9606SDaniel P. Berrange     char *username;
1351bff9606SDaniel P. Berrange     char *password;
1361bff9606SDaniel P. Berrange     char *proxyusername;
1371bff9606SDaniel P. Berrange     char *proxypassword;
138769ce76dSAlexander Graf } BDRVCURLState;
139769ce76dSAlexander Graf 
140769ce76dSAlexander Graf static void curl_clean_state(CURLState *s);
141769ce76dSAlexander Graf static void curl_multi_do(void *arg);
142769ce76dSAlexander Graf 
143031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
14434db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
145031fd1beSPeter Maydell static int curl_timer_cb(CURLM *multi, long timeout_ms, void *opaque)
146031fd1beSPeter Maydell {
147031fd1beSPeter Maydell     BDRVCURLState *s = opaque;
148031fd1beSPeter Maydell 
149ed2a66deSLaurent Vivier     trace_curl_timer_cb(timeout_ms);
150031fd1beSPeter Maydell     if (timeout_ms == -1) {
151031fd1beSPeter Maydell         timer_del(&s->timer);
152031fd1beSPeter Maydell     } else {
153031fd1beSPeter Maydell         int64_t timeout_ns = (int64_t)timeout_ms * 1000 * 1000;
154031fd1beSPeter Maydell         timer_mod(&s->timer,
155031fd1beSPeter Maydell                   qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + timeout_ns);
156031fd1beSPeter Maydell     }
157031fd1beSPeter Maydell     return 0;
158031fd1beSPeter Maydell }
159031fd1beSPeter Maydell #endif
160031fd1beSPeter Maydell 
16134db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
162769ce76dSAlexander Graf static int curl_sock_cb(CURL *curl, curl_socket_t fd, int action,
16363f0f45fSStefan Hajnoczi                         void *userp, void *sp)
164769ce76dSAlexander Graf {
16563f0f45fSStefan Hajnoczi     BDRVCURLState *s;
166838ef602SMatthew Booth     CURLState *state = NULL;
167ff5ca166SMax Reitz     CURLSocket *socket;
168ff5ca166SMax Reitz 
169838ef602SMatthew Booth     curl_easy_getinfo(curl, CURLINFO_PRIVATE, (char **)&state);
17063f0f45fSStefan Hajnoczi     s = state->s;
171838ef602SMatthew Booth 
172ff5ca166SMax Reitz     QLIST_FOREACH(socket, &state->sockets, next) {
173ff5ca166SMax Reitz         if (socket->fd == fd) {
174ff5ca166SMax Reitz             break;
175ff5ca166SMax Reitz         }
176ff5ca166SMax Reitz     }
177ff5ca166SMax Reitz     if (!socket) {
178ff5ca166SMax Reitz         socket = g_new0(CURLSocket, 1);
179ff5ca166SMax Reitz         socket->fd = fd;
18004878616SMax Reitz         socket->state = state;
181ff5ca166SMax Reitz         QLIST_INSERT_HEAD(&state->sockets, socket, next);
182ff5ca166SMax Reitz     }
183ff5ca166SMax Reitz 
184ed2a66deSLaurent Vivier     trace_curl_sock_cb(action, (int)fd);
185769ce76dSAlexander Graf     switch (action) {
186769ce76dSAlexander Graf         case CURL_POLL_IN:
187dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
1889dbad87dSMax Reitz                                curl_multi_do, NULL, NULL, socket);
189769ce76dSAlexander Graf             break;
190769ce76dSAlexander Graf         case CURL_POLL_OUT:
191dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
1929dbad87dSMax Reitz                                NULL, curl_multi_do, NULL, socket);
193769ce76dSAlexander Graf             break;
194769ce76dSAlexander Graf         case CURL_POLL_INOUT:
195dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
1969dbad87dSMax Reitz                                curl_multi_do, curl_multi_do, NULL, socket);
197769ce76dSAlexander Graf             break;
198769ce76dSAlexander Graf         case CURL_POLL_REMOVE:
199dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
200f6a51c84SStefan Hajnoczi                                NULL, NULL, NULL, NULL);
201769ce76dSAlexander Graf             break;
202769ce76dSAlexander Graf     }
203769ce76dSAlexander Graf 
204007f339bSMax Reitz     if (action == CURL_POLL_REMOVE) {
205007f339bSMax Reitz         QLIST_REMOVE(socket, next);
206007f339bSMax Reitz         g_free(socket);
207007f339bSMax Reitz     }
208007f339bSMax Reitz 
209769ce76dSAlexander Graf     return 0;
210769ce76dSAlexander Graf }
211769ce76dSAlexander Graf 
21234db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
2133494d650SFam Zheng static size_t curl_header_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
214769ce76dSAlexander Graf {
2153494d650SFam Zheng     BDRVCURLState *s = opaque;
216769ce76dSAlexander Graf     size_t realsize = size * nmemb;
2173494d650SFam Zheng     const char *accept_line = "Accept-Ranges: bytes";
218769ce76dSAlexander Graf 
2193494d650SFam Zheng     if (realsize >= strlen(accept_line)
2203494d650SFam Zheng         && strncmp((char *)ptr, accept_line, strlen(accept_line)) == 0) {
2213494d650SFam Zheng         s->accept_range = true;
2220bfcd599SBlue Swirl     }
223769ce76dSAlexander Graf 
224769ce76dSAlexander Graf     return realsize;
225769ce76dSAlexander Graf }
226769ce76dSAlexander Graf 
22734db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
228769ce76dSAlexander Graf static size_t curl_read_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
229769ce76dSAlexander Graf {
230769ce76dSAlexander Graf     CURLState *s = ((CURLState*)opaque);
231769ce76dSAlexander Graf     size_t realsize = size * nmemb;
232769ce76dSAlexander Graf     int i;
233769ce76dSAlexander Graf 
234ed2a66deSLaurent Vivier     trace_curl_read_cb(realsize);
235769ce76dSAlexander Graf 
2364e767657SMax Reitz     if (!s || !s->orig_buf) {
2374e767657SMax Reitz         goto read_end;
2384e767657SMax Reitz     }
239769ce76dSAlexander Graf 
2406d4b9e55SFam Zheng     if (s->buf_off >= s->buf_len) {
2416d4b9e55SFam Zheng         /* buffer full, read nothing */
2424e767657SMax Reitz         goto read_end;
2436d4b9e55SFam Zheng     }
2446d4b9e55SFam Zheng     realsize = MIN(realsize, s->buf_len - s->buf_off);
245769ce76dSAlexander Graf     memcpy(s->orig_buf + s->buf_off, ptr, realsize);
246769ce76dSAlexander Graf     s->buf_off += realsize;
247769ce76dSAlexander Graf 
248769ce76dSAlexander Graf     for(i=0; i<CURL_NUM_ACB; i++) {
249769ce76dSAlexander Graf         CURLAIOCB *acb = s->acb[i];
250769ce76dSAlexander Graf 
251769ce76dSAlexander Graf         if (!acb)
252769ce76dSAlexander Graf             continue;
253769ce76dSAlexander Graf 
254769ce76dSAlexander Graf         if ((s->buf_off >= acb->end)) {
2552125e5eaSPaolo Bonzini             size_t request_length = acb->bytes;
2564e504535SMax Reitz 
25703396148SMichael Tokarev             qemu_iovec_from_buf(acb->qiov, 0, s->orig_buf + acb->start,
258769ce76dSAlexander Graf                                 acb->end - acb->start);
2594e504535SMax Reitz 
2604e504535SMax Reitz             if (acb->end - acb->start < request_length) {
2614e504535SMax Reitz                 size_t offset = acb->end - acb->start;
2624e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, offset, 0,
2634e504535SMax Reitz                                   request_length - offset);
2644e504535SMax Reitz             }
2654e504535SMax Reitz 
26628256d82SPaolo Bonzini             acb->ret = 0;
267769ce76dSAlexander Graf             s->acb[i] = NULL;
26828256d82SPaolo Bonzini             qemu_mutex_unlock(&s->s->mutex);
26928256d82SPaolo Bonzini             aio_co_wake(acb->co);
27028256d82SPaolo Bonzini             qemu_mutex_lock(&s->s->mutex);
271769ce76dSAlexander Graf         }
272769ce76dSAlexander Graf     }
273769ce76dSAlexander Graf 
2744e767657SMax Reitz read_end:
2754e767657SMax Reitz     /* curl will error out if we do not return this value */
2764e767657SMax Reitz     return size * nmemb;
277769ce76dSAlexander Graf }
278769ce76dSAlexander Graf 
279456af346SPaolo Bonzini /* Called with s->mutex held.  */
28028256d82SPaolo Bonzini static bool curl_find_buf(BDRVCURLState *s, uint64_t start, uint64_t len,
281769ce76dSAlexander Graf                           CURLAIOCB *acb)
282769ce76dSAlexander Graf {
283769ce76dSAlexander Graf     int i;
2842125e5eaSPaolo Bonzini     uint64_t end = start + len;
2852125e5eaSPaolo Bonzini     uint64_t clamped_end = MIN(end, s->len);
2862125e5eaSPaolo Bonzini     uint64_t clamped_len = clamped_end - start;
287769ce76dSAlexander Graf 
288769ce76dSAlexander Graf     for (i=0; i<CURL_NUM_STATES; i++) {
289769ce76dSAlexander Graf         CURLState *state = &s->states[i];
2902125e5eaSPaolo Bonzini         uint64_t buf_end = (state->buf_start + state->buf_off);
2912125e5eaSPaolo Bonzini         uint64_t buf_fend = (state->buf_start + state->buf_len);
292769ce76dSAlexander Graf 
293769ce76dSAlexander Graf         if (!state->orig_buf)
294769ce76dSAlexander Graf             continue;
295769ce76dSAlexander Graf         if (!state->buf_off)
296769ce76dSAlexander Graf             continue;
297769ce76dSAlexander Graf 
298769ce76dSAlexander Graf         // Does the existing buffer cover our section?
299769ce76dSAlexander Graf         if ((start >= state->buf_start) &&
300769ce76dSAlexander Graf             (start <= buf_end) &&
3014e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3024e504535SMax Reitz             (clamped_end <= buf_end))
303769ce76dSAlexander Graf         {
304769ce76dSAlexander Graf             char *buf = state->orig_buf + (start - state->buf_start);
305769ce76dSAlexander Graf 
3064e504535SMax Reitz             qemu_iovec_from_buf(acb->qiov, 0, buf, clamped_len);
3074e504535SMax Reitz             if (clamped_len < len) {
3084e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, clamped_len, 0, len - clamped_len);
3094e504535SMax Reitz             }
31028256d82SPaolo Bonzini             acb->ret = 0;
31128256d82SPaolo Bonzini             return true;
312769ce76dSAlexander Graf         }
313769ce76dSAlexander Graf 
314769ce76dSAlexander Graf         // Wait for unfinished chunks
315b7079df4SMatthew Booth         if (state->in_use &&
316b7079df4SMatthew Booth             (start >= state->buf_start) &&
317769ce76dSAlexander Graf             (start <= buf_fend) &&
3184e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3194e504535SMax Reitz             (clamped_end <= buf_fend))
320769ce76dSAlexander Graf         {
321769ce76dSAlexander Graf             int j;
322769ce76dSAlexander Graf 
323769ce76dSAlexander Graf             acb->start = start - state->buf_start;
3244e504535SMax Reitz             acb->end = acb->start + clamped_len;
325769ce76dSAlexander Graf 
326769ce76dSAlexander Graf             for (j=0; j<CURL_NUM_ACB; j++) {
327769ce76dSAlexander Graf                 if (!state->acb[j]) {
328769ce76dSAlexander Graf                     state->acb[j] = acb;
32928256d82SPaolo Bonzini                     return true;
330769ce76dSAlexander Graf                 }
331769ce76dSAlexander Graf             }
332769ce76dSAlexander Graf         }
333769ce76dSAlexander Graf     }
334769ce76dSAlexander Graf 
33528256d82SPaolo Bonzini     return false;
336769ce76dSAlexander Graf }
337769ce76dSAlexander Graf 
338ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
339838ef602SMatthew Booth static void curl_multi_check_completion(BDRVCURLState *s)
340769ce76dSAlexander Graf {
341769ce76dSAlexander Graf     int msgs_in_queue;
342769ce76dSAlexander Graf 
343769ce76dSAlexander Graf     /* Try to find done transfers, so we can free the easy
344769ce76dSAlexander Graf      * handle again. */
3451f2cead3SMatthew Booth     for (;;) {
346769ce76dSAlexander Graf         CURLMsg *msg;
347769ce76dSAlexander Graf         msg = curl_multi_info_read(s->multi, &msgs_in_queue);
348769ce76dSAlexander Graf 
3491f2cead3SMatthew Booth         /* Quit when there are no more completions */
350769ce76dSAlexander Graf         if (!msg)
351769ce76dSAlexander Graf             break;
352769ce76dSAlexander Graf 
3531f2cead3SMatthew Booth         if (msg->msg == CURLMSG_DONE) {
354769ce76dSAlexander Graf             CURLState *state = NULL;
355f6246509SMatthew Booth             curl_easy_getinfo(msg->easy_handle, CURLINFO_PRIVATE,
356f6246509SMatthew Booth                               (char **)&state);
357f785a5aeSNicholas Thomas 
358f785a5aeSNicholas Thomas             /* ACBs for successful messages get completed in curl_read_cb */
359f785a5aeSNicholas Thomas             if (msg->data.result != CURLE_OK) {
360f785a5aeSNicholas Thomas                 int i;
361796a060bSRichard W.M. Jones                 static int errcount = 100;
362796a060bSRichard W.M. Jones 
363796a060bSRichard W.M. Jones                 /* Don't lose the original error message from curl, since
364796a060bSRichard W.M. Jones                  * it contains extra data.
365796a060bSRichard W.M. Jones                  */
366796a060bSRichard W.M. Jones                 if (errcount > 0) {
367796a060bSRichard W.M. Jones                     error_report("curl: %s", state->errmsg);
368796a060bSRichard W.M. Jones                     if (--errcount == 0) {
369796a060bSRichard W.M. Jones                         error_report("curl: further errors suppressed");
370796a060bSRichard W.M. Jones                     }
371796a060bSRichard W.M. Jones                 }
372796a060bSRichard W.M. Jones 
373f785a5aeSNicholas Thomas                 for (i = 0; i < CURL_NUM_ACB; i++) {
374f785a5aeSNicholas Thomas                     CURLAIOCB *acb = state->acb[i];
375f785a5aeSNicholas Thomas 
376f785a5aeSNicholas Thomas                     if (acb == NULL) {
377f785a5aeSNicholas Thomas                         continue;
378f785a5aeSNicholas Thomas                     }
379f785a5aeSNicholas Thomas 
38028256d82SPaolo Bonzini                     acb->ret = -EIO;
381f785a5aeSNicholas Thomas                     state->acb[i] = NULL;
38228256d82SPaolo Bonzini                     qemu_mutex_unlock(&s->mutex);
38328256d82SPaolo Bonzini                     aio_co_wake(acb->co);
38428256d82SPaolo Bonzini                     qemu_mutex_lock(&s->mutex);
385f785a5aeSNicholas Thomas                 }
386f785a5aeSNicholas Thomas             }
387f785a5aeSNicholas Thomas 
388769ce76dSAlexander Graf             curl_clean_state(state);
389769ce76dSAlexander Graf             break;
390769ce76dSAlexander Graf         }
391769ce76dSAlexander Graf     }
392769ce76dSAlexander Graf }
393769ce76dSAlexander Graf 
394ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
395*9abaf9fcSMax Reitz static void curl_multi_do_locked(CURLSocket *socket)
396031fd1beSPeter Maydell {
397*9abaf9fcSMax Reitz     BDRVCURLState *s = socket->state->s;
398031fd1beSPeter Maydell     int running;
399031fd1beSPeter Maydell     int r;
400031fd1beSPeter Maydell 
401*9abaf9fcSMax Reitz     if (!s->multi) {
402031fd1beSPeter Maydell         return;
403031fd1beSPeter Maydell     }
404031fd1beSPeter Maydell 
405031fd1beSPeter Maydell     do {
406*9abaf9fcSMax Reitz         r = curl_multi_socket_action(s->multi, socket->fd, 0, &running);
407031fd1beSPeter Maydell     } while (r == CURLM_CALL_MULTI_PERFORM);
408ff5ca166SMax Reitz }
409838ef602SMatthew Booth 
4109d456654SPaolo Bonzini static void curl_multi_do(void *arg)
4119d456654SPaolo Bonzini {
4129dbad87dSMax Reitz     CURLSocket *socket = arg;
4139dbad87dSMax Reitz     BDRVCURLState *s = socket->state->s;
4149d456654SPaolo Bonzini 
4159dbad87dSMax Reitz     qemu_mutex_lock(&s->mutex);
4169dbad87dSMax Reitz     curl_multi_do_locked(socket);
4179dbad87dSMax Reitz     curl_multi_check_completion(s);
4189dbad87dSMax Reitz     qemu_mutex_unlock(&s->mutex);
419031fd1beSPeter Maydell }
420031fd1beSPeter Maydell 
421031fd1beSPeter Maydell static void curl_multi_timeout_do(void *arg)
422031fd1beSPeter Maydell {
423031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
424031fd1beSPeter Maydell     BDRVCURLState *s = (BDRVCURLState *)arg;
425031fd1beSPeter Maydell     int running;
426031fd1beSPeter Maydell 
427031fd1beSPeter Maydell     if (!s->multi) {
428031fd1beSPeter Maydell         return;
429031fd1beSPeter Maydell     }
430031fd1beSPeter Maydell 
431ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
432031fd1beSPeter Maydell     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
433031fd1beSPeter Maydell 
434838ef602SMatthew Booth     curl_multi_check_completion(s);
435ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
436031fd1beSPeter Maydell #else
437031fd1beSPeter Maydell     abort();
438031fd1beSPeter Maydell #endif
439031fd1beSPeter Maydell }
440031fd1beSPeter Maydell 
441456af346SPaolo Bonzini /* Called with s->mutex held.  */
4423ce6a729SPaolo Bonzini static CURLState *curl_find_state(BDRVCURLState *s)
443769ce76dSAlexander Graf {
444769ce76dSAlexander Graf     CURLState *state = NULL;
4453ce6a729SPaolo Bonzini     int i;
446769ce76dSAlexander Graf 
447769ce76dSAlexander Graf     for (i = 0; i < CURL_NUM_STATES; i++) {
4483ce6a729SPaolo Bonzini         if (!s->states[i].in_use) {
449769ce76dSAlexander Graf             state = &s->states[i];
450769ce76dSAlexander Graf             state->in_use = 1;
451769ce76dSAlexander Graf             break;
452769ce76dSAlexander Graf         }
453769ce76dSAlexander Graf     }
4543ce6a729SPaolo Bonzini     return state;
4553ce6a729SPaolo Bonzini }
456769ce76dSAlexander Graf 
4573ce6a729SPaolo Bonzini static int curl_init_state(BDRVCURLState *s, CURLState *state)
4583ce6a729SPaolo Bonzini {
4599e550b32SMatthew Booth     if (!state->curl) {
460769ce76dSAlexander Graf         state->curl = curl_easy_init();
4619e550b32SMatthew Booth         if (!state->curl) {
4623ce6a729SPaolo Bonzini             return -EIO;
4639e550b32SMatthew Booth         }
464769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_URL, s->url);
46597a3ea57SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYPEER,
46697a3ea57SMatthew Booth                          (long) s->sslverify);
467637fa44aSRichard W.M. Jones         curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYHOST,
468637fa44aSRichard W.M. Jones                          s->sslverify ? 2L : 0L);
469a94f83d9SRichard W.M. Jones         if (s->cookie) {
470a94f83d9SRichard W.M. Jones             curl_easy_setopt(state->curl, CURLOPT_COOKIE, s->cookie);
471a94f83d9SRichard W.M. Jones         }
472f76faedaSRichard W.M. Jones         curl_easy_setopt(state->curl, CURLOPT_TIMEOUT, (long)s->timeout);
4739e550b32SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_WRITEFUNCTION,
4749e550b32SMatthew Booth                          (void *)curl_read_cb);
475769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_WRITEDATA, (void *)state);
476769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_PRIVATE, (void *)state);
477769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_AUTOREFERER, 1);
478769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_FOLLOWLOCATION, 1);
479769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_NOSIGNAL, 1);
480769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_ERRORBUFFER, state->errmsg);
481f785a5aeSNicholas Thomas         curl_easy_setopt(state->curl, CURLOPT_FAILONERROR, 1);
482769ce76dSAlexander Graf 
4831bff9606SDaniel P. Berrange         if (s->username) {
4841bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_USERNAME, s->username);
4851bff9606SDaniel P. Berrange         }
4861bff9606SDaniel P. Berrange         if (s->password) {
4871bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_PASSWORD, s->password);
4881bff9606SDaniel P. Berrange         }
4891bff9606SDaniel P. Berrange         if (s->proxyusername) {
4901bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
4911bff9606SDaniel P. Berrange                              CURLOPT_PROXYUSERNAME, s->proxyusername);
4921bff9606SDaniel P. Berrange         }
4931bff9606SDaniel P. Berrange         if (s->proxypassword) {
4941bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
4951bff9606SDaniel P. Berrange                              CURLOPT_PROXYPASSWORD, s->proxypassword);
4961bff9606SDaniel P. Berrange         }
4971bff9606SDaniel P. Berrange 
498fb6d1bbdSStefan Hajnoczi         /* Restrict supported protocols to avoid security issues in the more
499fb6d1bbdSStefan Hajnoczi          * obscure protocols.  For example, do not allow POP3/SMTP/IMAP see
500fb6d1bbdSStefan Hajnoczi          * CVE-2013-0249.
5018a8f5840SStefan Hajnoczi          *
5028a8f5840SStefan Hajnoczi          * Restricting protocols is only supported from 7.19.4 upwards.
503fb6d1bbdSStefan Hajnoczi          */
5048a8f5840SStefan Hajnoczi #if LIBCURL_VERSION_NUM >= 0x071304
505fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_PROTOCOLS, PROTOCOLS);
506fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_REDIR_PROTOCOLS, PROTOCOLS);
5078a8f5840SStefan Hajnoczi #endif
508fb6d1bbdSStefan Hajnoczi 
509769ce76dSAlexander Graf #ifdef DEBUG_VERBOSE
510769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_VERBOSE, 1);
511769ce76dSAlexander Graf #endif
5129e550b32SMatthew Booth     }
513769ce76dSAlexander Graf 
514ff5ca166SMax Reitz     QLIST_INIT(&state->sockets);
515769ce76dSAlexander Graf     state->s = s;
516769ce76dSAlexander Graf 
5173ce6a729SPaolo Bonzini     return 0;
518769ce76dSAlexander Graf }
519769ce76dSAlexander Graf 
520456af346SPaolo Bonzini /* Called with s->mutex held.  */
521769ce76dSAlexander Graf static void curl_clean_state(CURLState *s)
522769ce76dSAlexander Graf {
523675a7756SPaolo Bonzini     int j;
524675a7756SPaolo Bonzini     for (j = 0; j < CURL_NUM_ACB; j++) {
525675a7756SPaolo Bonzini         assert(!s->acb[j]);
526675a7756SPaolo Bonzini     }
527675a7756SPaolo Bonzini 
528769ce76dSAlexander Graf     if (s->s->multi)
529769ce76dSAlexander Graf         curl_multi_remove_handle(s->s->multi, s->curl);
530ff5ca166SMax Reitz 
531ff5ca166SMax Reitz     while (!QLIST_EMPTY(&s->sockets)) {
532ff5ca166SMax Reitz         CURLSocket *socket = QLIST_FIRST(&s->sockets);
533ff5ca166SMax Reitz 
534ff5ca166SMax Reitz         QLIST_REMOVE(socket, next);
535ff5ca166SMax Reitz         g_free(socket);
536ff5ca166SMax Reitz     }
537ff5ca166SMax Reitz 
538769ce76dSAlexander Graf     s->in_use = 0;
5392bb5c936SPaolo Bonzini 
540709f2132SPaolo Bonzini     qemu_co_enter_next(&s->s->free_state_waitq, &s->s->mutex);
541769ce76dSAlexander Graf }
542769ce76dSAlexander Graf 
5438e6d58cdSKevin Wolf static void curl_parse_filename(const char *filename, QDict *options,
5448e6d58cdSKevin Wolf                                 Error **errp)
545769ce76dSAlexander Graf {
54646f5ac20SEric Blake     qdict_put_str(options, CURL_BLOCK_OPT_URL, filename);
5478e6d58cdSKevin Wolf }
5488e6d58cdSKevin Wolf 
54963f0f45fSStefan Hajnoczi static void curl_detach_aio_context(BlockDriverState *bs)
55063f0f45fSStefan Hajnoczi {
55163f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
55263f0f45fSStefan Hajnoczi     int i;
55363f0f45fSStefan Hajnoczi 
554456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
55563f0f45fSStefan Hajnoczi     for (i = 0; i < CURL_NUM_STATES; i++) {
55663f0f45fSStefan Hajnoczi         if (s->states[i].in_use) {
55763f0f45fSStefan Hajnoczi             curl_clean_state(&s->states[i]);
55863f0f45fSStefan Hajnoczi         }
55963f0f45fSStefan Hajnoczi         if (s->states[i].curl) {
56063f0f45fSStefan Hajnoczi             curl_easy_cleanup(s->states[i].curl);
56163f0f45fSStefan Hajnoczi             s->states[i].curl = NULL;
56263f0f45fSStefan Hajnoczi         }
56363f0f45fSStefan Hajnoczi         g_free(s->states[i].orig_buf);
56463f0f45fSStefan Hajnoczi         s->states[i].orig_buf = NULL;
56563f0f45fSStefan Hajnoczi     }
56663f0f45fSStefan Hajnoczi     if (s->multi) {
56763f0f45fSStefan Hajnoczi         curl_multi_cleanup(s->multi);
56863f0f45fSStefan Hajnoczi         s->multi = NULL;
56963f0f45fSStefan Hajnoczi     }
570456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
57163f0f45fSStefan Hajnoczi 
57263f0f45fSStefan Hajnoczi     timer_del(&s->timer);
57363f0f45fSStefan Hajnoczi }
57463f0f45fSStefan Hajnoczi 
57563f0f45fSStefan Hajnoczi static void curl_attach_aio_context(BlockDriverState *bs,
57663f0f45fSStefan Hajnoczi                                     AioContext *new_context)
57763f0f45fSStefan Hajnoczi {
57863f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
57963f0f45fSStefan Hajnoczi 
58063f0f45fSStefan Hajnoczi     aio_timer_init(new_context, &s->timer,
58163f0f45fSStefan Hajnoczi                    QEMU_CLOCK_REALTIME, SCALE_NS,
58263f0f45fSStefan Hajnoczi                    curl_multi_timeout_do, s);
58363f0f45fSStefan Hajnoczi 
58463f0f45fSStefan Hajnoczi     assert(!s->multi);
58563f0f45fSStefan Hajnoczi     s->multi = curl_multi_init();
58663f0f45fSStefan Hajnoczi     s->aio_context = new_context;
58763f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_SOCKETFUNCTION, curl_sock_cb);
58863f0f45fSStefan Hajnoczi #ifdef NEED_CURL_TIMER_CALLBACK
58963f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERDATA, s);
59063f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERFUNCTION, curl_timer_cb);
59163f0f45fSStefan Hajnoczi #endif
59263f0f45fSStefan Hajnoczi }
59363f0f45fSStefan Hajnoczi 
5948e6d58cdSKevin Wolf static QemuOptsList runtime_opts = {
5958e6d58cdSKevin Wolf     .name = "curl",
5968e6d58cdSKevin Wolf     .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
5978e6d58cdSKevin Wolf     .desc = {
5988e6d58cdSKevin Wolf         {
599e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_URL,
6008e6d58cdSKevin Wolf             .type = QEMU_OPT_STRING,
6018e6d58cdSKevin Wolf             .help = "URL to open",
6028e6d58cdSKevin Wolf         },
6038e6d58cdSKevin Wolf         {
604e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_READAHEAD,
6058e6d58cdSKevin Wolf             .type = QEMU_OPT_SIZE,
6068e6d58cdSKevin Wolf             .help = "Readahead size",
6078e6d58cdSKevin Wolf         },
60897a3ea57SMatthew Booth         {
60997a3ea57SMatthew Booth             .name = CURL_BLOCK_OPT_SSLVERIFY,
61097a3ea57SMatthew Booth             .type = QEMU_OPT_BOOL,
61197a3ea57SMatthew Booth             .help = "Verify SSL certificate"
61297a3ea57SMatthew Booth         },
613212aefaaSDaniel Henrique Barboza         {
614212aefaaSDaniel Henrique Barboza             .name = CURL_BLOCK_OPT_TIMEOUT,
615212aefaaSDaniel Henrique Barboza             .type = QEMU_OPT_NUMBER,
616212aefaaSDaniel Henrique Barboza             .help = "Curl timeout"
617212aefaaSDaniel Henrique Barboza         },
618a94f83d9SRichard W.M. Jones         {
619a94f83d9SRichard W.M. Jones             .name = CURL_BLOCK_OPT_COOKIE,
620a94f83d9SRichard W.M. Jones             .type = QEMU_OPT_STRING,
621a94f83d9SRichard W.M. Jones             .help = "Pass the cookie or list of cookies with each request"
622a94f83d9SRichard W.M. Jones         },
6231bff9606SDaniel P. Berrange         {
624327c8ebdSPeter Krempa             .name = CURL_BLOCK_OPT_COOKIE_SECRET,
625327c8ebdSPeter Krempa             .type = QEMU_OPT_STRING,
626327c8ebdSPeter Krempa             .help = "ID of secret used as cookie passed with each request"
627327c8ebdSPeter Krempa         },
628327c8ebdSPeter Krempa         {
6291bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_USERNAME,
6301bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6311bff9606SDaniel P. Berrange             .help = "Username for HTTP auth"
6321bff9606SDaniel P. Berrange         },
6331bff9606SDaniel P. Berrange         {
6341bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PASSWORD_SECRET,
6351bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6361bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP auth",
6371bff9606SDaniel P. Berrange         },
6381bff9606SDaniel P. Berrange         {
6391bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_USERNAME,
6401bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6411bff9606SDaniel P. Berrange             .help = "Username for HTTP proxy auth"
6421bff9606SDaniel P. Berrange         },
6431bff9606SDaniel P. Berrange         {
6441bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET,
6451bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6461bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP proxy auth",
6471bff9606SDaniel P. Berrange         },
6488e6d58cdSKevin Wolf         { /* end of list */ }
6498e6d58cdSKevin Wolf     },
6508e6d58cdSKevin Wolf };
6518e6d58cdSKevin Wolf 
6521bff9606SDaniel P. Berrange 
653015a1036SMax Reitz static int curl_open(BlockDriverState *bs, QDict *options, int flags,
654015a1036SMax Reitz                      Error **errp)
6558e6d58cdSKevin Wolf {
6568e6d58cdSKevin Wolf     BDRVCURLState *s = bs->opaque;
6578e6d58cdSKevin Wolf     CURLState *state = NULL;
6588e6d58cdSKevin Wolf     QemuOpts *opts;
6598e6d58cdSKevin Wolf     Error *local_err = NULL;
6608e6d58cdSKevin Wolf     const char *file;
661a94f83d9SRichard W.M. Jones     const char *cookie;
662327c8ebdSPeter Krempa     const char *cookie_secret;
6638e6d58cdSKevin Wolf     double d;
6641bff9606SDaniel P. Berrange     const char *secretid;
66534634ca2SMax Reitz     const char *protocol_delimiter;
6662d25964dSJeff Cody     int ret;
6678e6d58cdSKevin Wolf 
6686ceef36aSKevin Wolf     ret = bdrv_apply_auto_read_only(bs, "curl driver does not support writes",
6696ceef36aSKevin Wolf                                     errp);
6706ceef36aSKevin Wolf     if (ret < 0) {
6716ceef36aSKevin Wolf         return ret;
672a7cea2baSRichard W.M. Jones     }
673a7cea2baSRichard W.M. Jones 
6742d25964dSJeff Cody     if (!libcurl_initialized) {
6752d25964dSJeff Cody         ret = curl_global_init(CURL_GLOBAL_ALL);
6762d25964dSJeff Cody         if (ret) {
6772d25964dSJeff Cody             error_setg(errp, "libcurl initialization failed with %d", ret);
6782d25964dSJeff Cody             return -EIO;
6792d25964dSJeff Cody         }
6802d25964dSJeff Cody         libcurl_initialized = true;
6812d25964dSJeff Cody     }
6822d25964dSJeff Cody 
683456af346SPaolo Bonzini     qemu_mutex_init(&s->mutex);
68487ea75d5SPeter Crosthwaite     opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
6858e6d58cdSKevin Wolf     qemu_opts_absorb_qdict(opts, options, &local_err);
68684d18f06SMarkus Armbruster     if (local_err) {
6872a94fee3SPaolo Bonzini         error_propagate(errp, local_err);
6888e6d58cdSKevin Wolf         goto out_noclean;
6898e6d58cdSKevin Wolf     }
6908e6d58cdSKevin Wolf 
691e3542c67SMatthew Booth     s->readahead_size = qemu_opt_get_size(opts, CURL_BLOCK_OPT_READAHEAD,
692712b64e8SMax Reitz                                           CURL_BLOCK_OPT_READAHEAD_DEFAULT);
693c76f4952SNolan     if ((s->readahead_size & 0x1ff) != 0) {
6942a94fee3SPaolo Bonzini         error_setg(errp, "HTTP_READAHEAD_SIZE %zd is not a multiple of 512",
695c76f4952SNolan                    s->readahead_size);
696c76f4952SNolan         goto out_noclean;
697c76f4952SNolan     }
698c76f4952SNolan 
699212aefaaSDaniel Henrique Barboza     s->timeout = qemu_opt_get_number(opts, CURL_BLOCK_OPT_TIMEOUT,
700712b64e8SMax Reitz                                      CURL_BLOCK_OPT_TIMEOUT_DEFAULT);
701f76faedaSRichard W.M. Jones     if (s->timeout > CURL_TIMEOUT_MAX) {
702f76faedaSRichard W.M. Jones         error_setg(errp, "timeout parameter is too large or negative");
703f76faedaSRichard W.M. Jones         goto out_noclean;
704f76faedaSRichard W.M. Jones     }
705212aefaaSDaniel Henrique Barboza 
706712b64e8SMax Reitz     s->sslverify = qemu_opt_get_bool(opts, CURL_BLOCK_OPT_SSLVERIFY,
707712b64e8SMax Reitz                                      CURL_BLOCK_OPT_SSLVERIFY_DEFAULT);
70897a3ea57SMatthew Booth 
709a94f83d9SRichard W.M. Jones     cookie = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE);
710327c8ebdSPeter Krempa     cookie_secret = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE_SECRET);
711327c8ebdSPeter Krempa 
712327c8ebdSPeter Krempa     if (cookie && cookie_secret) {
713327c8ebdSPeter Krempa         error_setg(errp,
714327c8ebdSPeter Krempa                    "curl driver cannot handle both cookie and cookie secret");
715327c8ebdSPeter Krempa         goto out_noclean;
716327c8ebdSPeter Krempa     }
717327c8ebdSPeter Krempa 
718327c8ebdSPeter Krempa     if (cookie_secret) {
719327c8ebdSPeter Krempa         s->cookie = qcrypto_secret_lookup_as_utf8(cookie_secret, errp);
720327c8ebdSPeter Krempa         if (!s->cookie) {
721327c8ebdSPeter Krempa             goto out_noclean;
722327c8ebdSPeter Krempa         }
723327c8ebdSPeter Krempa     } else {
724a94f83d9SRichard W.M. Jones         s->cookie = g_strdup(cookie);
725327c8ebdSPeter Krempa     }
726a94f83d9SRichard W.M. Jones 
727e3542c67SMatthew Booth     file = qemu_opt_get(opts, CURL_BLOCK_OPT_URL);
7288e6d58cdSKevin Wolf     if (file == NULL) {
7292a94fee3SPaolo Bonzini         error_setg(errp, "curl block driver requires an 'url' option");
7308e6d58cdSKevin Wolf         goto out_noclean;
7318e6d58cdSKevin Wolf     }
7328e6d58cdSKevin Wolf 
73334634ca2SMax Reitz     if (!strstart(file, bs->drv->protocol_name, &protocol_delimiter) ||
73434634ca2SMax Reitz         !strstart(protocol_delimiter, "://", NULL))
73534634ca2SMax Reitz     {
73634634ca2SMax Reitz         error_setg(errp, "%s curl driver cannot handle the URL '%s' (does not "
73734634ca2SMax Reitz                    "start with '%s://')", bs->drv->protocol_name, file,
73834634ca2SMax Reitz                    bs->drv->protocol_name);
73934634ca2SMax Reitz         goto out_noclean;
74034634ca2SMax Reitz     }
74134634ca2SMax Reitz 
7421bff9606SDaniel P. Berrange     s->username = g_strdup(qemu_opt_get(opts, CURL_BLOCK_OPT_USERNAME));
7431bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PASSWORD_SECRET);
7441bff9606SDaniel P. Berrange 
7451bff9606SDaniel P. Berrange     if (secretid) {
7461bff9606SDaniel P. Berrange         s->password = qcrypto_secret_lookup_as_utf8(secretid, errp);
7471bff9606SDaniel P. Berrange         if (!s->password) {
7481bff9606SDaniel P. Berrange             goto out_noclean;
7491bff9606SDaniel P. Berrange         }
7501bff9606SDaniel P. Berrange     }
7511bff9606SDaniel P. Berrange 
7521bff9606SDaniel P. Berrange     s->proxyusername = g_strdup(
7531bff9606SDaniel P. Berrange         qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_USERNAME));
7541bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET);
7551bff9606SDaniel P. Berrange     if (secretid) {
7561bff9606SDaniel P. Berrange         s->proxypassword = qcrypto_secret_lookup_as_utf8(secretid, errp);
7571bff9606SDaniel P. Berrange         if (!s->proxypassword) {
7581bff9606SDaniel P. Berrange             goto out_noclean;
7591bff9606SDaniel P. Berrange         }
7601bff9606SDaniel P. Berrange     }
7611bff9606SDaniel P. Berrange 
762ed2a66deSLaurent Vivier     trace_curl_open(file);
763709f2132SPaolo Bonzini     qemu_co_queue_init(&s->free_state_waitq);
76463f0f45fSStefan Hajnoczi     s->aio_context = bdrv_get_aio_context(bs);
7658e6d58cdSKevin Wolf     s->url = g_strdup(file);
766456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
7673ce6a729SPaolo Bonzini     state = curl_find_state(s);
768456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
7693ce6a729SPaolo Bonzini     if (!state) {
770769ce76dSAlexander Graf         goto out_noclean;
7713ce6a729SPaolo Bonzini     }
772769ce76dSAlexander Graf 
773769ce76dSAlexander Graf     // Get file size
774769ce76dSAlexander Graf 
7753ce6a729SPaolo Bonzini     if (curl_init_state(s, state) < 0) {
7763ce6a729SPaolo Bonzini         goto out;
7773ce6a729SPaolo Bonzini     }
7783ce6a729SPaolo Bonzini 
7793494d650SFam Zheng     s->accept_range = false;
780769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_NOBODY, 1);
7813494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERFUNCTION,
7823494d650SFam Zheng                      curl_header_cb);
7833494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERDATA, s);
784769ce76dSAlexander Graf     if (curl_easy_perform(state->curl))
785769ce76dSAlexander Graf         goto out;
786a41c4578STomáš Golembiovský     if (curl_easy_getinfo(state->curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d)) {
787769ce76dSAlexander Graf         goto out;
788a41c4578STomáš Golembiovský     }
789a41c4578STomáš Golembiovský     /* Prior CURL 7.19.4 return value of 0 could mean that the file size is not
790a41c4578STomáš Golembiovský      * know or the size is zero. From 7.19.4 CURL returns -1 if size is not
79150d6a8a3SStefan Weil      * known and zero if it is really zero-length file. */
792a41c4578STomáš Golembiovský #if LIBCURL_VERSION_NUM >= 0x071304
793a41c4578STomáš Golembiovský     if (d < 0) {
794a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
795a41c4578STomáš Golembiovský                 "Server didn't report file size.");
796a41c4578STomáš Golembiovský         goto out;
797a41c4578STomáš Golembiovský     }
798a41c4578STomáš Golembiovský #else
799a41c4578STomáš Golembiovský     if (d <= 0) {
800a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
801a41c4578STomáš Golembiovský                 "Unknown file size or zero-length file.");
802a41c4578STomáš Golembiovský         goto out;
803a41c4578STomáš Golembiovský     }
804a41c4578STomáš Golembiovský #endif
805a41c4578STomáš Golembiovský 
8062125e5eaSPaolo Bonzini     s->len = d;
807a41c4578STomáš Golembiovský 
8083494d650SFam Zheng     if ((!strncasecmp(s->url, "http://", strlen("http://"))
8093494d650SFam Zheng         || !strncasecmp(s->url, "https://", strlen("https://")))
8103494d650SFam Zheng         && !s->accept_range) {
8113494d650SFam Zheng         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
8123494d650SFam Zheng                 "Server does not support 'range' (byte ranges).");
8133494d650SFam Zheng         goto out;
8143494d650SFam Zheng     }
815ed2a66deSLaurent Vivier     trace_curl_open_size(s->len);
816769ce76dSAlexander Graf 
817456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
818769ce76dSAlexander Graf     curl_clean_state(state);
819456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
820769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
821769ce76dSAlexander Graf     state->curl = NULL;
822769ce76dSAlexander Graf 
82363f0f45fSStefan Hajnoczi     curl_attach_aio_context(bs, bdrv_get_aio_context(bs));
824769ce76dSAlexander Graf 
8258e6d58cdSKevin Wolf     qemu_opts_del(opts);
826769ce76dSAlexander Graf     return 0;
827769ce76dSAlexander Graf 
828769ce76dSAlexander Graf out:
829acd7fdc6SMaria Kustova     error_setg(errp, "CURL: Error opening file: %s", state->errmsg);
830769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
831769ce76dSAlexander Graf     state->curl = NULL;
832769ce76dSAlexander Graf out_noclean:
833456af346SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
834a94f83d9SRichard W.M. Jones     g_free(s->cookie);
8358e6d58cdSKevin Wolf     g_free(s->url);
836996922deSJeff Cody     g_free(s->username);
837996922deSJeff Cody     g_free(s->proxyusername);
838996922deSJeff Cody     g_free(s->proxypassword);
8398e6d58cdSKevin Wolf     qemu_opts_del(opts);
840769ce76dSAlexander Graf     return -EINVAL;
841769ce76dSAlexander Graf }
842769ce76dSAlexander Graf 
84328256d82SPaolo Bonzini static void curl_setup_preadv(BlockDriverState *bs, CURLAIOCB *acb)
844769ce76dSAlexander Graf {
845769ce76dSAlexander Graf     CURLState *state;
846b69cdef8SMatthew Booth     int running;
847769ce76dSAlexander Graf 
8481919631eSPaolo Bonzini     BDRVCURLState *s = bs->opaque;
849769ce76dSAlexander Graf 
8502125e5eaSPaolo Bonzini     uint64_t start = acb->offset;
8512125e5eaSPaolo Bonzini     uint64_t end;
852769ce76dSAlexander Graf 
853ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
8541919631eSPaolo Bonzini 
855769ce76dSAlexander Graf     // In case we have the requested data already (e.g. read-ahead),
856769ce76dSAlexander Graf     // we can just call the callback and be done.
85728256d82SPaolo Bonzini     if (curl_find_buf(s, start, acb->bytes, acb)) {
85834db05e7SPaolo Bonzini         goto out;
859769ce76dSAlexander Graf     }
860769ce76dSAlexander Graf 
861769ce76dSAlexander Graf     // No cache found, so let's start a new request
8623ce6a729SPaolo Bonzini     for (;;) {
8633ce6a729SPaolo Bonzini         state = curl_find_state(s);
8643ce6a729SPaolo Bonzini         if (state) {
8653ce6a729SPaolo Bonzini             break;
8663ce6a729SPaolo Bonzini         }
867709f2132SPaolo Bonzini         qemu_co_queue_wait(&s->free_state_waitq, &s->mutex);
8683ce6a729SPaolo Bonzini     }
8693ce6a729SPaolo Bonzini 
8703ce6a729SPaolo Bonzini     if (curl_init_state(s, state) < 0) {
8713ce6a729SPaolo Bonzini         curl_clean_state(state);
87228256d82SPaolo Bonzini         acb->ret = -EIO;
8731919631eSPaolo Bonzini         goto out;
874363c3c85SNick Thomas     }
875769ce76dSAlexander Graf 
876769ce76dSAlexander Graf     acb->start = 0;
8772125e5eaSPaolo Bonzini     acb->end = MIN(acb->bytes, s->len - start);
878769ce76dSAlexander Graf 
879769ce76dSAlexander Graf     state->buf_off = 0;
8807267c094SAnthony Liguori     g_free(state->orig_buf);
881769ce76dSAlexander Graf     state->buf_start = start;
8824e504535SMax Reitz     state->buf_len = MIN(acb->end + s->readahead_size, s->len - start);
8834e504535SMax Reitz     end = start + state->buf_len - 1;
8848dc7a772SKevin Wolf     state->orig_buf = g_try_malloc(state->buf_len);
8858dc7a772SKevin Wolf     if (state->buf_len && state->orig_buf == NULL) {
8868dc7a772SKevin Wolf         curl_clean_state(state);
88728256d82SPaolo Bonzini         acb->ret = -ENOMEM;
8881919631eSPaolo Bonzini         goto out;
8898dc7a772SKevin Wolf     }
890769ce76dSAlexander Graf     state->acb[0] = acb;
891769ce76dSAlexander Graf 
8922125e5eaSPaolo Bonzini     snprintf(state->range, 127, "%" PRIu64 "-%" PRIu64, start, end);
893ed2a66deSLaurent Vivier     trace_curl_setup_preadv(acb->bytes, start, state->range);
894769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_RANGE, state->range);
895769ce76dSAlexander Graf 
896769ce76dSAlexander Graf     curl_multi_add_handle(s->multi, state->curl);
897769ce76dSAlexander Graf 
898b69cdef8SMatthew Booth     /* Tell curl it needs to kick things off */
899b69cdef8SMatthew Booth     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
9001919631eSPaolo Bonzini 
9011919631eSPaolo Bonzini out:
902ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
903363c3c85SNick Thomas }
904363c3c85SNick Thomas 
90528256d82SPaolo Bonzini static int coroutine_fn curl_co_preadv(BlockDriverState *bs,
90628256d82SPaolo Bonzini         uint64_t offset, uint64_t bytes, QEMUIOVector *qiov, int flags)
907363c3c85SNick Thomas {
90828256d82SPaolo Bonzini     CURLAIOCB acb = {
90928256d82SPaolo Bonzini         .co = qemu_coroutine_self(),
91028256d82SPaolo Bonzini         .ret = -EINPROGRESS,
91128256d82SPaolo Bonzini         .qiov = qiov,
91228256d82SPaolo Bonzini         .offset = offset,
91328256d82SPaolo Bonzini         .bytes = bytes
91428256d82SPaolo Bonzini     };
915363c3c85SNick Thomas 
91628256d82SPaolo Bonzini     curl_setup_preadv(bs, &acb);
91728256d82SPaolo Bonzini     while (acb.ret == -EINPROGRESS) {
91828256d82SPaolo Bonzini         qemu_coroutine_yield();
91928256d82SPaolo Bonzini     }
92028256d82SPaolo Bonzini     return acb.ret;
921769ce76dSAlexander Graf }
922769ce76dSAlexander Graf 
923769ce76dSAlexander Graf static void curl_close(BlockDriverState *bs)
924769ce76dSAlexander Graf {
925769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
926769ce76dSAlexander Graf 
927ed2a66deSLaurent Vivier     trace_curl_close();
92863f0f45fSStefan Hajnoczi     curl_detach_aio_context(bs);
929ba3186c4SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
930031fd1beSPeter Maydell 
931a94f83d9SRichard W.M. Jones     g_free(s->cookie);
93245724d6dSStefan Weil     g_free(s->url);
933996922deSJeff Cody     g_free(s->username);
934996922deSJeff Cody     g_free(s->proxyusername);
935996922deSJeff Cody     g_free(s->proxypassword);
936769ce76dSAlexander Graf }
937769ce76dSAlexander Graf 
938769ce76dSAlexander Graf static int64_t curl_getlength(BlockDriverState *bs)
939769ce76dSAlexander Graf {
940769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
941769ce76dSAlexander Graf     return s->len;
942769ce76dSAlexander Graf }
943769ce76dSAlexander Graf 
944937c007bSMax Reitz static void curl_refresh_filename(BlockDriverState *bs)
945937c007bSMax Reitz {
946937c007bSMax Reitz     BDRVCURLState *s = bs->opaque;
947937c007bSMax Reitz 
948937c007bSMax Reitz     /* "readahead" and "timeout" do not change the guest-visible data,
949937c007bSMax Reitz      * so ignore them */
950937c007bSMax Reitz     if (s->sslverify != CURL_BLOCK_OPT_SSLVERIFY_DEFAULT ||
951937c007bSMax Reitz         s->cookie || s->username || s->password || s->proxyusername ||
952937c007bSMax Reitz         s->proxypassword)
953937c007bSMax Reitz     {
954937c007bSMax Reitz         return;
955937c007bSMax Reitz     }
956937c007bSMax Reitz 
957937c007bSMax Reitz     pstrcpy(bs->exact_filename, sizeof(bs->exact_filename), s->url);
958937c007bSMax Reitz }
959937c007bSMax Reitz 
960937c007bSMax Reitz 
9612654267cSMax Reitz static const char *const curl_strong_runtime_opts[] = {
9622654267cSMax Reitz     CURL_BLOCK_OPT_URL,
9632654267cSMax Reitz     CURL_BLOCK_OPT_SSLVERIFY,
9642654267cSMax Reitz     CURL_BLOCK_OPT_COOKIE,
9652654267cSMax Reitz     CURL_BLOCK_OPT_COOKIE_SECRET,
9662654267cSMax Reitz     CURL_BLOCK_OPT_USERNAME,
9672654267cSMax Reitz     CURL_BLOCK_OPT_PASSWORD_SECRET,
9682654267cSMax Reitz     CURL_BLOCK_OPT_PROXY_USERNAME,
9692654267cSMax Reitz     CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET,
9702654267cSMax Reitz 
9712654267cSMax Reitz     NULL
9722654267cSMax Reitz };
9732654267cSMax Reitz 
974769ce76dSAlexander Graf static BlockDriver bdrv_http = {
975769ce76dSAlexander Graf     .format_name                = "http",
976769ce76dSAlexander Graf     .protocol_name              = "http",
977769ce76dSAlexander Graf 
978769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9798e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
98066f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
981769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
982769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
983769ce76dSAlexander Graf 
98428256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
98563f0f45fSStefan Hajnoczi 
98663f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
98763f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
9882654267cSMax Reitz 
989937c007bSMax Reitz     .bdrv_refresh_filename      = curl_refresh_filename,
9902654267cSMax Reitz     .strong_runtime_opts        = curl_strong_runtime_opts,
991769ce76dSAlexander Graf };
992769ce76dSAlexander Graf 
993769ce76dSAlexander Graf static BlockDriver bdrv_https = {
994769ce76dSAlexander Graf     .format_name                = "https",
995769ce76dSAlexander Graf     .protocol_name              = "https",
996769ce76dSAlexander Graf 
997769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9988e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
99966f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
1000769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
1001769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
1002769ce76dSAlexander Graf 
100328256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
100463f0f45fSStefan Hajnoczi 
100563f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
100663f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
10072654267cSMax Reitz 
1008937c007bSMax Reitz     .bdrv_refresh_filename      = curl_refresh_filename,
10092654267cSMax Reitz     .strong_runtime_opts        = curl_strong_runtime_opts,
1010769ce76dSAlexander Graf };
1011769ce76dSAlexander Graf 
1012769ce76dSAlexander Graf static BlockDriver bdrv_ftp = {
1013769ce76dSAlexander Graf     .format_name                = "ftp",
1014769ce76dSAlexander Graf     .protocol_name              = "ftp",
1015769ce76dSAlexander Graf 
1016769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
10178e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
101866f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
1019769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
1020769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
1021769ce76dSAlexander Graf 
102228256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
102363f0f45fSStefan Hajnoczi 
102463f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
102563f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
10262654267cSMax Reitz 
1027937c007bSMax Reitz     .bdrv_refresh_filename      = curl_refresh_filename,
10282654267cSMax Reitz     .strong_runtime_opts        = curl_strong_runtime_opts,
1029769ce76dSAlexander Graf };
1030769ce76dSAlexander Graf 
1031769ce76dSAlexander Graf static BlockDriver bdrv_ftps = {
1032769ce76dSAlexander Graf     .format_name                = "ftps",
1033769ce76dSAlexander Graf     .protocol_name              = "ftps",
1034769ce76dSAlexander Graf 
1035769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
10368e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
103766f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
1038769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
1039769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
1040769ce76dSAlexander Graf 
104128256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
104263f0f45fSStefan Hajnoczi 
104363f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
104463f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
10452654267cSMax Reitz 
1046937c007bSMax Reitz     .bdrv_refresh_filename      = curl_refresh_filename,
10472654267cSMax Reitz     .strong_runtime_opts        = curl_strong_runtime_opts,
1048769ce76dSAlexander Graf };
1049769ce76dSAlexander Graf 
1050769ce76dSAlexander Graf static void curl_block_init(void)
1051769ce76dSAlexander Graf {
1052769ce76dSAlexander Graf     bdrv_register(&bdrv_http);
1053769ce76dSAlexander Graf     bdrv_register(&bdrv_https);
1054769ce76dSAlexander Graf     bdrv_register(&bdrv_ftp);
1055769ce76dSAlexander Graf     bdrv_register(&bdrv_ftps);
1056769ce76dSAlexander Graf }
1057769ce76dSAlexander Graf 
1058769ce76dSAlexander Graf block_init(curl_block_init);
1059