xref: /openbmc/qemu/block/curl.c (revision 2d25964d1831c99d54981e8b615eba5dd6a63e36)
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  */
2480c71a24SPeter Maydell #include "qemu/osdep.h"
25da34e65cSMarkus Armbruster #include "qapi/error.h"
26769ce76dSAlexander Graf #include "qemu-common.h"
27796a060bSRichard W.M. Jones #include "qemu/error-report.h"
28737e150eSPaolo Bonzini #include "block/block_int.h"
2997a3ea57SMatthew Booth #include "qapi/qmp/qbool.h"
30d49b6836SMarkus Armbruster #include "qapi/qmp/qstring.h"
311bff9606SDaniel P. Berrange #include "crypto/secret.h"
32769ce76dSAlexander Graf #include <curl/curl.h>
33f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
34769ce76dSAlexander Graf 
3541c23467SRichard W.M. Jones // #define DEBUG_CURL
36769ce76dSAlexander Graf // #define DEBUG_VERBOSE
37769ce76dSAlexander Graf 
38769ce76dSAlexander Graf #ifdef DEBUG_CURL
39ed79f37dSZhou Jie #define DEBUG_CURL_PRINT 1
40769ce76dSAlexander Graf #else
41ed79f37dSZhou Jie #define DEBUG_CURL_PRINT 0
42769ce76dSAlexander Graf #endif
43ed79f37dSZhou Jie #define DPRINTF(fmt, ...)                                            \
44ed79f37dSZhou Jie     do {                                                             \
45ed79f37dSZhou Jie         if (DEBUG_CURL_PRINT) {                                      \
46ed79f37dSZhou Jie             fprintf(stderr, fmt, ## __VA_ARGS__);                    \
47ed79f37dSZhou Jie         }                                                            \
48ed79f37dSZhou Jie     } while (0)
49769ce76dSAlexander Graf 
50031fd1beSPeter Maydell #if LIBCURL_VERSION_NUM >= 0x071000
51031fd1beSPeter Maydell /* The multi interface timer callback was introduced in 7.16.0 */
52031fd1beSPeter Maydell #define NEED_CURL_TIMER_CALLBACK
539aedd5a5SMatthew Booth #define HAVE_SOCKET_ACTION
549aedd5a5SMatthew Booth #endif
559aedd5a5SMatthew Booth 
569aedd5a5SMatthew Booth #ifndef HAVE_SOCKET_ACTION
579aedd5a5SMatthew Booth /* If curl_multi_socket_action isn't available, define it statically here in
589aedd5a5SMatthew Booth  * terms of curl_multi_socket. Note that ev_bitmask will be ignored, which is
599aedd5a5SMatthew Booth  * less efficient but still safe. */
609aedd5a5SMatthew Booth static CURLMcode __curl_multi_socket_action(CURLM *multi_handle,
619aedd5a5SMatthew Booth                                             curl_socket_t sockfd,
629aedd5a5SMatthew Booth                                             int ev_bitmask,
639aedd5a5SMatthew Booth                                             int *running_handles)
649aedd5a5SMatthew Booth {
659aedd5a5SMatthew Booth     return curl_multi_socket(multi_handle, sockfd, running_handles);
669aedd5a5SMatthew Booth }
679aedd5a5SMatthew Booth #define curl_multi_socket_action __curl_multi_socket_action
68031fd1beSPeter Maydell #endif
69031fd1beSPeter Maydell 
70fb6d1bbdSStefan Hajnoczi #define PROTOCOLS (CURLPROTO_HTTP | CURLPROTO_HTTPS | \
7123dce387SMax Reitz                    CURLPROTO_FTP | CURLPROTO_FTPS)
72fb6d1bbdSStefan Hajnoczi 
73769ce76dSAlexander Graf #define CURL_NUM_STATES 8
74769ce76dSAlexander Graf #define CURL_NUM_ACB    8
75e3542c67SMatthew Booth #define READ_AHEAD_DEFAULT (256 * 1024)
76212aefaaSDaniel Henrique Barboza #define CURL_TIMEOUT_DEFAULT 5
77f76faedaSRichard W.M. Jones #define CURL_TIMEOUT_MAX 10000
78769ce76dSAlexander Graf 
79e3542c67SMatthew Booth #define CURL_BLOCK_OPT_URL       "url"
80e3542c67SMatthew Booth #define CURL_BLOCK_OPT_READAHEAD "readahead"
8197a3ea57SMatthew Booth #define CURL_BLOCK_OPT_SSLVERIFY "sslverify"
82212aefaaSDaniel Henrique Barboza #define CURL_BLOCK_OPT_TIMEOUT "timeout"
83a94f83d9SRichard W.M. Jones #define CURL_BLOCK_OPT_COOKIE    "cookie"
84327c8ebdSPeter Krempa #define CURL_BLOCK_OPT_COOKIE_SECRET "cookie-secret"
851bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_USERNAME "username"
861bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PASSWORD_SECRET "password-secret"
871bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_USERNAME "proxy-username"
881bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET "proxy-password-secret"
89e3542c67SMatthew Booth 
90769ce76dSAlexander Graf struct BDRVCURLState;
91769ce76dSAlexander Graf 
92*2d25964dSJeff Cody static bool libcurl_initialized;
93*2d25964dSJeff Cody 
94769ce76dSAlexander Graf typedef struct CURLAIOCB {
9528256d82SPaolo Bonzini     Coroutine *co;
96769ce76dSAlexander Graf     QEMUIOVector *qiov;
97363c3c85SNick Thomas 
982125e5eaSPaolo Bonzini     uint64_t offset;
992125e5eaSPaolo Bonzini     uint64_t bytes;
10028256d82SPaolo Bonzini     int ret;
101363c3c85SNick Thomas 
102769ce76dSAlexander Graf     size_t start;
103769ce76dSAlexander Graf     size_t end;
1042bb5c936SPaolo Bonzini 
1052bb5c936SPaolo Bonzini     QSIMPLEQ_ENTRY(CURLAIOCB) next;
106769ce76dSAlexander Graf } CURLAIOCB;
107769ce76dSAlexander Graf 
108ff5ca166SMax Reitz typedef struct CURLSocket {
109ff5ca166SMax Reitz     int fd;
110ff5ca166SMax Reitz     QLIST_ENTRY(CURLSocket) next;
111ff5ca166SMax Reitz } CURLSocket;
112ff5ca166SMax Reitz 
113769ce76dSAlexander Graf typedef struct CURLState
114769ce76dSAlexander Graf {
115769ce76dSAlexander Graf     struct BDRVCURLState *s;
116769ce76dSAlexander Graf     CURLAIOCB *acb[CURL_NUM_ACB];
117769ce76dSAlexander Graf     CURL *curl;
118ff5ca166SMax Reitz     QLIST_HEAD(, CURLSocket) sockets;
119769ce76dSAlexander Graf     char *orig_buf;
1202125e5eaSPaolo Bonzini     uint64_t buf_start;
121769ce76dSAlexander Graf     size_t buf_off;
122769ce76dSAlexander Graf     size_t buf_len;
123769ce76dSAlexander Graf     char range[128];
124769ce76dSAlexander Graf     char errmsg[CURL_ERROR_SIZE];
125769ce76dSAlexander Graf     char in_use;
126769ce76dSAlexander Graf } CURLState;
127769ce76dSAlexander Graf 
128769ce76dSAlexander Graf typedef struct BDRVCURLState {
129769ce76dSAlexander Graf     CURLM *multi;
130031fd1beSPeter Maydell     QEMUTimer timer;
1312125e5eaSPaolo Bonzini     uint64_t len;
132769ce76dSAlexander Graf     CURLState states[CURL_NUM_STATES];
133769ce76dSAlexander Graf     char *url;
134c76f4952SNolan     size_t readahead_size;
13597a3ea57SMatthew Booth     bool sslverify;
136f76faedaSRichard W.M. Jones     uint64_t timeout;
137a94f83d9SRichard W.M. Jones     char *cookie;
1383494d650SFam Zheng     bool accept_range;
13963f0f45fSStefan Hajnoczi     AioContext *aio_context;
140ba3186c4SPaolo Bonzini     QemuMutex mutex;
1412bb5c936SPaolo Bonzini     QSIMPLEQ_HEAD(, CURLAIOCB) free_state_waitq;
1421bff9606SDaniel P. Berrange     char *username;
1431bff9606SDaniel P. Berrange     char *password;
1441bff9606SDaniel P. Berrange     char *proxyusername;
1451bff9606SDaniel P. Berrange     char *proxypassword;
146769ce76dSAlexander Graf } BDRVCURLState;
147769ce76dSAlexander Graf 
148769ce76dSAlexander Graf static void curl_clean_state(CURLState *s);
149769ce76dSAlexander Graf static void curl_multi_do(void *arg);
150838ef602SMatthew Booth static void curl_multi_read(void *arg);
151769ce76dSAlexander Graf 
152031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
15334db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
154031fd1beSPeter Maydell static int curl_timer_cb(CURLM *multi, long timeout_ms, void *opaque)
155031fd1beSPeter Maydell {
156031fd1beSPeter Maydell     BDRVCURLState *s = opaque;
157031fd1beSPeter Maydell 
158031fd1beSPeter Maydell     DPRINTF("CURL: timer callback timeout_ms %ld\n", timeout_ms);
159031fd1beSPeter Maydell     if (timeout_ms == -1) {
160031fd1beSPeter Maydell         timer_del(&s->timer);
161031fd1beSPeter Maydell     } else {
162031fd1beSPeter Maydell         int64_t timeout_ns = (int64_t)timeout_ms * 1000 * 1000;
163031fd1beSPeter Maydell         timer_mod(&s->timer,
164031fd1beSPeter Maydell                   qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + timeout_ns);
165031fd1beSPeter Maydell     }
166031fd1beSPeter Maydell     return 0;
167031fd1beSPeter Maydell }
168031fd1beSPeter Maydell #endif
169031fd1beSPeter Maydell 
17034db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
171769ce76dSAlexander Graf static int curl_sock_cb(CURL *curl, curl_socket_t fd, int action,
17263f0f45fSStefan Hajnoczi                         void *userp, void *sp)
173769ce76dSAlexander Graf {
17463f0f45fSStefan Hajnoczi     BDRVCURLState *s;
175838ef602SMatthew Booth     CURLState *state = NULL;
176ff5ca166SMax Reitz     CURLSocket *socket;
177ff5ca166SMax Reitz 
178838ef602SMatthew Booth     curl_easy_getinfo(curl, CURLINFO_PRIVATE, (char **)&state);
17963f0f45fSStefan Hajnoczi     s = state->s;
180838ef602SMatthew Booth 
181ff5ca166SMax Reitz     QLIST_FOREACH(socket, &state->sockets, next) {
182ff5ca166SMax Reitz         if (socket->fd == fd) {
183ff5ca166SMax Reitz             if (action == CURL_POLL_REMOVE) {
184ff5ca166SMax Reitz                 QLIST_REMOVE(socket, next);
185ff5ca166SMax Reitz                 g_free(socket);
186ff5ca166SMax Reitz             }
187ff5ca166SMax Reitz             break;
188ff5ca166SMax Reitz         }
189ff5ca166SMax Reitz     }
190ff5ca166SMax Reitz     if (!socket) {
191ff5ca166SMax Reitz         socket = g_new0(CURLSocket, 1);
192ff5ca166SMax Reitz         socket->fd = fd;
193ff5ca166SMax Reitz         QLIST_INSERT_HEAD(&state->sockets, socket, next);
194ff5ca166SMax Reitz     }
195ff5ca166SMax Reitz     socket = NULL;
196ff5ca166SMax Reitz 
19792b6a160SFam Zheng     DPRINTF("CURL (AIO): Sock action %d on fd %d\n", action, (int)fd);
198769ce76dSAlexander Graf     switch (action) {
199769ce76dSAlexander Graf         case CURL_POLL_IN:
200dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
201f6a51c84SStefan Hajnoczi                                curl_multi_read, NULL, NULL, state);
202769ce76dSAlexander Graf             break;
203769ce76dSAlexander Graf         case CURL_POLL_OUT:
204dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
205f6a51c84SStefan Hajnoczi                                NULL, curl_multi_do, NULL, state);
206769ce76dSAlexander Graf             break;
207769ce76dSAlexander Graf         case CURL_POLL_INOUT:
208dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
209f6a51c84SStefan Hajnoczi                                curl_multi_read, curl_multi_do, NULL, state);
210769ce76dSAlexander Graf             break;
211769ce76dSAlexander Graf         case CURL_POLL_REMOVE:
212dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
213f6a51c84SStefan Hajnoczi                                NULL, NULL, NULL, NULL);
214769ce76dSAlexander Graf             break;
215769ce76dSAlexander Graf     }
216769ce76dSAlexander Graf 
217769ce76dSAlexander Graf     return 0;
218769ce76dSAlexander Graf }
219769ce76dSAlexander Graf 
22034db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
2213494d650SFam Zheng static size_t curl_header_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
222769ce76dSAlexander Graf {
2233494d650SFam Zheng     BDRVCURLState *s = opaque;
224769ce76dSAlexander Graf     size_t realsize = size * nmemb;
2253494d650SFam Zheng     const char *accept_line = "Accept-Ranges: bytes";
226769ce76dSAlexander Graf 
2273494d650SFam Zheng     if (realsize >= strlen(accept_line)
2283494d650SFam Zheng         && strncmp((char *)ptr, accept_line, strlen(accept_line)) == 0) {
2293494d650SFam Zheng         s->accept_range = true;
2300bfcd599SBlue Swirl     }
231769ce76dSAlexander Graf 
232769ce76dSAlexander Graf     return realsize;
233769ce76dSAlexander Graf }
234769ce76dSAlexander Graf 
23534db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
236769ce76dSAlexander Graf static size_t curl_read_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
237769ce76dSAlexander Graf {
238769ce76dSAlexander Graf     CURLState *s = ((CURLState*)opaque);
239769ce76dSAlexander Graf     size_t realsize = size * nmemb;
240769ce76dSAlexander Graf     int i;
241769ce76dSAlexander Graf 
2420bfcd599SBlue Swirl     DPRINTF("CURL: Just reading %zd bytes\n", realsize);
243769ce76dSAlexander Graf 
2444e767657SMax Reitz     if (!s || !s->orig_buf) {
2454e767657SMax Reitz         goto read_end;
2464e767657SMax Reitz     }
247769ce76dSAlexander Graf 
2486d4b9e55SFam Zheng     if (s->buf_off >= s->buf_len) {
2496d4b9e55SFam Zheng         /* buffer full, read nothing */
2504e767657SMax Reitz         goto read_end;
2516d4b9e55SFam Zheng     }
2526d4b9e55SFam Zheng     realsize = MIN(realsize, s->buf_len - s->buf_off);
253769ce76dSAlexander Graf     memcpy(s->orig_buf + s->buf_off, ptr, realsize);
254769ce76dSAlexander Graf     s->buf_off += realsize;
255769ce76dSAlexander Graf 
256769ce76dSAlexander Graf     for(i=0; i<CURL_NUM_ACB; i++) {
257769ce76dSAlexander Graf         CURLAIOCB *acb = s->acb[i];
258769ce76dSAlexander Graf 
259769ce76dSAlexander Graf         if (!acb)
260769ce76dSAlexander Graf             continue;
261769ce76dSAlexander Graf 
262769ce76dSAlexander Graf         if ((s->buf_off >= acb->end)) {
2632125e5eaSPaolo Bonzini             size_t request_length = acb->bytes;
2644e504535SMax Reitz 
26503396148SMichael Tokarev             qemu_iovec_from_buf(acb->qiov, 0, s->orig_buf + acb->start,
266769ce76dSAlexander Graf                                 acb->end - acb->start);
2674e504535SMax Reitz 
2684e504535SMax Reitz             if (acb->end - acb->start < request_length) {
2694e504535SMax Reitz                 size_t offset = acb->end - acb->start;
2704e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, offset, 0,
2714e504535SMax Reitz                                   request_length - offset);
2724e504535SMax Reitz             }
2734e504535SMax Reitz 
27428256d82SPaolo Bonzini             acb->ret = 0;
275769ce76dSAlexander Graf             s->acb[i] = NULL;
27628256d82SPaolo Bonzini             qemu_mutex_unlock(&s->s->mutex);
27728256d82SPaolo Bonzini             aio_co_wake(acb->co);
27828256d82SPaolo Bonzini             qemu_mutex_lock(&s->s->mutex);
279769ce76dSAlexander Graf         }
280769ce76dSAlexander Graf     }
281769ce76dSAlexander Graf 
2824e767657SMax Reitz read_end:
2834e767657SMax Reitz     /* curl will error out if we do not return this value */
2844e767657SMax Reitz     return size * nmemb;
285769ce76dSAlexander Graf }
286769ce76dSAlexander Graf 
287456af346SPaolo Bonzini /* Called with s->mutex held.  */
28828256d82SPaolo Bonzini static bool curl_find_buf(BDRVCURLState *s, uint64_t start, uint64_t len,
289769ce76dSAlexander Graf                           CURLAIOCB *acb)
290769ce76dSAlexander Graf {
291769ce76dSAlexander Graf     int i;
2922125e5eaSPaolo Bonzini     uint64_t end = start + len;
2932125e5eaSPaolo Bonzini     uint64_t clamped_end = MIN(end, s->len);
2942125e5eaSPaolo Bonzini     uint64_t clamped_len = clamped_end - start;
295769ce76dSAlexander Graf 
296769ce76dSAlexander Graf     for (i=0; i<CURL_NUM_STATES; i++) {
297769ce76dSAlexander Graf         CURLState *state = &s->states[i];
2982125e5eaSPaolo Bonzini         uint64_t buf_end = (state->buf_start + state->buf_off);
2992125e5eaSPaolo Bonzini         uint64_t buf_fend = (state->buf_start + state->buf_len);
300769ce76dSAlexander Graf 
301769ce76dSAlexander Graf         if (!state->orig_buf)
302769ce76dSAlexander Graf             continue;
303769ce76dSAlexander Graf         if (!state->buf_off)
304769ce76dSAlexander Graf             continue;
305769ce76dSAlexander Graf 
306769ce76dSAlexander Graf         // Does the existing buffer cover our section?
307769ce76dSAlexander Graf         if ((start >= state->buf_start) &&
308769ce76dSAlexander Graf             (start <= buf_end) &&
3094e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3104e504535SMax Reitz             (clamped_end <= buf_end))
311769ce76dSAlexander Graf         {
312769ce76dSAlexander Graf             char *buf = state->orig_buf + (start - state->buf_start);
313769ce76dSAlexander Graf 
3144e504535SMax Reitz             qemu_iovec_from_buf(acb->qiov, 0, buf, clamped_len);
3154e504535SMax Reitz             if (clamped_len < len) {
3164e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, clamped_len, 0, len - clamped_len);
3174e504535SMax Reitz             }
31828256d82SPaolo Bonzini             acb->ret = 0;
31928256d82SPaolo Bonzini             return true;
320769ce76dSAlexander Graf         }
321769ce76dSAlexander Graf 
322769ce76dSAlexander Graf         // Wait for unfinished chunks
323b7079df4SMatthew Booth         if (state->in_use &&
324b7079df4SMatthew Booth             (start >= state->buf_start) &&
325769ce76dSAlexander Graf             (start <= buf_fend) &&
3264e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3274e504535SMax Reitz             (clamped_end <= buf_fend))
328769ce76dSAlexander Graf         {
329769ce76dSAlexander Graf             int j;
330769ce76dSAlexander Graf 
331769ce76dSAlexander Graf             acb->start = start - state->buf_start;
3324e504535SMax Reitz             acb->end = acb->start + clamped_len;
333769ce76dSAlexander Graf 
334769ce76dSAlexander Graf             for (j=0; j<CURL_NUM_ACB; j++) {
335769ce76dSAlexander Graf                 if (!state->acb[j]) {
336769ce76dSAlexander Graf                     state->acb[j] = acb;
33728256d82SPaolo Bonzini                     return true;
338769ce76dSAlexander Graf                 }
339769ce76dSAlexander Graf             }
340769ce76dSAlexander Graf         }
341769ce76dSAlexander Graf     }
342769ce76dSAlexander Graf 
34328256d82SPaolo Bonzini     return false;
344769ce76dSAlexander Graf }
345769ce76dSAlexander Graf 
346ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
347838ef602SMatthew Booth static void curl_multi_check_completion(BDRVCURLState *s)
348769ce76dSAlexander Graf {
349769ce76dSAlexander Graf     int msgs_in_queue;
350769ce76dSAlexander Graf 
351769ce76dSAlexander Graf     /* Try to find done transfers, so we can free the easy
352769ce76dSAlexander Graf      * handle again. */
3531f2cead3SMatthew Booth     for (;;) {
354769ce76dSAlexander Graf         CURLMsg *msg;
355769ce76dSAlexander Graf         msg = curl_multi_info_read(s->multi, &msgs_in_queue);
356769ce76dSAlexander Graf 
3571f2cead3SMatthew Booth         /* Quit when there are no more completions */
358769ce76dSAlexander Graf         if (!msg)
359769ce76dSAlexander Graf             break;
360769ce76dSAlexander Graf 
3611f2cead3SMatthew Booth         if (msg->msg == CURLMSG_DONE) {
362769ce76dSAlexander Graf             CURLState *state = NULL;
363f6246509SMatthew Booth             curl_easy_getinfo(msg->easy_handle, CURLINFO_PRIVATE,
364f6246509SMatthew Booth                               (char **)&state);
365f785a5aeSNicholas Thomas 
366f785a5aeSNicholas Thomas             /* ACBs for successful messages get completed in curl_read_cb */
367f785a5aeSNicholas Thomas             if (msg->data.result != CURLE_OK) {
368f785a5aeSNicholas Thomas                 int i;
369796a060bSRichard W.M. Jones                 static int errcount = 100;
370796a060bSRichard W.M. Jones 
371796a060bSRichard W.M. Jones                 /* Don't lose the original error message from curl, since
372796a060bSRichard W.M. Jones                  * it contains extra data.
373796a060bSRichard W.M. Jones                  */
374796a060bSRichard W.M. Jones                 if (errcount > 0) {
375796a060bSRichard W.M. Jones                     error_report("curl: %s", state->errmsg);
376796a060bSRichard W.M. Jones                     if (--errcount == 0) {
377796a060bSRichard W.M. Jones                         error_report("curl: further errors suppressed");
378796a060bSRichard W.M. Jones                     }
379796a060bSRichard W.M. Jones                 }
380796a060bSRichard W.M. Jones 
381f785a5aeSNicholas Thomas                 for (i = 0; i < CURL_NUM_ACB; i++) {
382f785a5aeSNicholas Thomas                     CURLAIOCB *acb = state->acb[i];
383f785a5aeSNicholas Thomas 
384f785a5aeSNicholas Thomas                     if (acb == NULL) {
385f785a5aeSNicholas Thomas                         continue;
386f785a5aeSNicholas Thomas                     }
387f785a5aeSNicholas Thomas 
38828256d82SPaolo Bonzini                     acb->ret = -EIO;
389f785a5aeSNicholas Thomas                     state->acb[i] = NULL;
39028256d82SPaolo Bonzini                     qemu_mutex_unlock(&s->mutex);
39128256d82SPaolo Bonzini                     aio_co_wake(acb->co);
39228256d82SPaolo Bonzini                     qemu_mutex_lock(&s->mutex);
393f785a5aeSNicholas Thomas                 }
394f785a5aeSNicholas Thomas             }
395f785a5aeSNicholas Thomas 
396769ce76dSAlexander Graf             curl_clean_state(state);
397769ce76dSAlexander Graf             break;
398769ce76dSAlexander Graf         }
399769ce76dSAlexander Graf     }
400769ce76dSAlexander Graf }
401769ce76dSAlexander Graf 
402ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
4039d456654SPaolo Bonzini static void curl_multi_do_locked(CURLState *s)
404031fd1beSPeter Maydell {
405ff5ca166SMax Reitz     CURLSocket *socket, *next_socket;
406031fd1beSPeter Maydell     int running;
407031fd1beSPeter Maydell     int r;
408031fd1beSPeter Maydell 
409838ef602SMatthew Booth     if (!s->s->multi) {
410031fd1beSPeter Maydell         return;
411031fd1beSPeter Maydell     }
412031fd1beSPeter Maydell 
413ff5ca166SMax Reitz     /* Need to use _SAFE because curl_multi_socket_action() may trigger
414ff5ca166SMax Reitz      * curl_sock_cb() which might modify this list */
415ff5ca166SMax Reitz     QLIST_FOREACH_SAFE(socket, &s->sockets, next, next_socket) {
416031fd1beSPeter Maydell         do {
417ff5ca166SMax Reitz             r = curl_multi_socket_action(s->s->multi, socket->fd, 0, &running);
418031fd1beSPeter Maydell         } while (r == CURLM_CALL_MULTI_PERFORM);
419ff5ca166SMax Reitz     }
420838ef602SMatthew Booth }
421838ef602SMatthew Booth 
4229d456654SPaolo Bonzini static void curl_multi_do(void *arg)
4239d456654SPaolo Bonzini {
4249d456654SPaolo Bonzini     CURLState *s = (CURLState *)arg;
4259d456654SPaolo Bonzini 
426ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->s->mutex);
4279d456654SPaolo Bonzini     curl_multi_do_locked(s);
428ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->s->mutex);
4299d456654SPaolo Bonzini }
4309d456654SPaolo Bonzini 
431838ef602SMatthew Booth static void curl_multi_read(void *arg)
432838ef602SMatthew Booth {
433838ef602SMatthew Booth     CURLState *s = (CURLState *)arg;
434838ef602SMatthew Booth 
435ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->s->mutex);
4369d456654SPaolo Bonzini     curl_multi_do_locked(s);
437838ef602SMatthew Booth     curl_multi_check_completion(s->s);
438ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->s->mutex);
439031fd1beSPeter Maydell }
440031fd1beSPeter Maydell 
441031fd1beSPeter Maydell static void curl_multi_timeout_do(void *arg)
442031fd1beSPeter Maydell {
443031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
444031fd1beSPeter Maydell     BDRVCURLState *s = (BDRVCURLState *)arg;
445031fd1beSPeter Maydell     int running;
446031fd1beSPeter Maydell 
447031fd1beSPeter Maydell     if (!s->multi) {
448031fd1beSPeter Maydell         return;
449031fd1beSPeter Maydell     }
450031fd1beSPeter Maydell 
451ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
452031fd1beSPeter Maydell     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
453031fd1beSPeter Maydell 
454838ef602SMatthew Booth     curl_multi_check_completion(s);
455ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
456031fd1beSPeter Maydell #else
457031fd1beSPeter Maydell     abort();
458031fd1beSPeter Maydell #endif
459031fd1beSPeter Maydell }
460031fd1beSPeter Maydell 
461456af346SPaolo Bonzini /* Called with s->mutex held.  */
4623ce6a729SPaolo Bonzini static CURLState *curl_find_state(BDRVCURLState *s)
463769ce76dSAlexander Graf {
464769ce76dSAlexander Graf     CURLState *state = NULL;
4653ce6a729SPaolo Bonzini     int i;
466769ce76dSAlexander Graf 
467769ce76dSAlexander Graf     for (i = 0; i < CURL_NUM_STATES; i++) {
4683ce6a729SPaolo Bonzini         if (!s->states[i].in_use) {
469769ce76dSAlexander Graf             state = &s->states[i];
470769ce76dSAlexander Graf             state->in_use = 1;
471769ce76dSAlexander Graf             break;
472769ce76dSAlexander Graf         }
473769ce76dSAlexander Graf     }
4743ce6a729SPaolo Bonzini     return state;
4753ce6a729SPaolo Bonzini }
476769ce76dSAlexander Graf 
4773ce6a729SPaolo Bonzini static int curl_init_state(BDRVCURLState *s, CURLState *state)
4783ce6a729SPaolo Bonzini {
4799e550b32SMatthew Booth     if (!state->curl) {
480769ce76dSAlexander Graf         state->curl = curl_easy_init();
4819e550b32SMatthew Booth         if (!state->curl) {
4823ce6a729SPaolo Bonzini             return -EIO;
4839e550b32SMatthew Booth         }
484769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_URL, s->url);
48597a3ea57SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYPEER,
48697a3ea57SMatthew Booth                          (long) s->sslverify);
487a94f83d9SRichard W.M. Jones         if (s->cookie) {
488a94f83d9SRichard W.M. Jones             curl_easy_setopt(state->curl, CURLOPT_COOKIE, s->cookie);
489a94f83d9SRichard W.M. Jones         }
490f76faedaSRichard W.M. Jones         curl_easy_setopt(state->curl, CURLOPT_TIMEOUT, (long)s->timeout);
4919e550b32SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_WRITEFUNCTION,
4929e550b32SMatthew Booth                          (void *)curl_read_cb);
493769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_WRITEDATA, (void *)state);
494769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_PRIVATE, (void *)state);
495769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_AUTOREFERER, 1);
496769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_FOLLOWLOCATION, 1);
497769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_NOSIGNAL, 1);
498769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_ERRORBUFFER, state->errmsg);
499f785a5aeSNicholas Thomas         curl_easy_setopt(state->curl, CURLOPT_FAILONERROR, 1);
500769ce76dSAlexander Graf 
5011bff9606SDaniel P. Berrange         if (s->username) {
5021bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_USERNAME, s->username);
5031bff9606SDaniel P. Berrange         }
5041bff9606SDaniel P. Berrange         if (s->password) {
5051bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_PASSWORD, s->password);
5061bff9606SDaniel P. Berrange         }
5071bff9606SDaniel P. Berrange         if (s->proxyusername) {
5081bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
5091bff9606SDaniel P. Berrange                              CURLOPT_PROXYUSERNAME, s->proxyusername);
5101bff9606SDaniel P. Berrange         }
5111bff9606SDaniel P. Berrange         if (s->proxypassword) {
5121bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
5131bff9606SDaniel P. Berrange                              CURLOPT_PROXYPASSWORD, s->proxypassword);
5141bff9606SDaniel P. Berrange         }
5151bff9606SDaniel P. Berrange 
516fb6d1bbdSStefan Hajnoczi         /* Restrict supported protocols to avoid security issues in the more
517fb6d1bbdSStefan Hajnoczi          * obscure protocols.  For example, do not allow POP3/SMTP/IMAP see
518fb6d1bbdSStefan Hajnoczi          * CVE-2013-0249.
5198a8f5840SStefan Hajnoczi          *
5208a8f5840SStefan Hajnoczi          * Restricting protocols is only supported from 7.19.4 upwards.
521fb6d1bbdSStefan Hajnoczi          */
5228a8f5840SStefan Hajnoczi #if LIBCURL_VERSION_NUM >= 0x071304
523fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_PROTOCOLS, PROTOCOLS);
524fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_REDIR_PROTOCOLS, PROTOCOLS);
5258a8f5840SStefan Hajnoczi #endif
526fb6d1bbdSStefan Hajnoczi 
527769ce76dSAlexander Graf #ifdef DEBUG_VERBOSE
528769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_VERBOSE, 1);
529769ce76dSAlexander Graf #endif
5309e550b32SMatthew Booth     }
531769ce76dSAlexander Graf 
532ff5ca166SMax Reitz     QLIST_INIT(&state->sockets);
533769ce76dSAlexander Graf     state->s = s;
534769ce76dSAlexander Graf 
5353ce6a729SPaolo Bonzini     return 0;
536769ce76dSAlexander Graf }
537769ce76dSAlexander Graf 
538456af346SPaolo Bonzini /* Called with s->mutex held.  */
539769ce76dSAlexander Graf static void curl_clean_state(CURLState *s)
540769ce76dSAlexander Graf {
5412bb5c936SPaolo Bonzini     CURLAIOCB *next;
542675a7756SPaolo Bonzini     int j;
543675a7756SPaolo Bonzini     for (j = 0; j < CURL_NUM_ACB; j++) {
544675a7756SPaolo Bonzini         assert(!s->acb[j]);
545675a7756SPaolo Bonzini     }
546675a7756SPaolo Bonzini 
547769ce76dSAlexander Graf     if (s->s->multi)
548769ce76dSAlexander Graf         curl_multi_remove_handle(s->s->multi, s->curl);
549ff5ca166SMax Reitz 
550ff5ca166SMax Reitz     while (!QLIST_EMPTY(&s->sockets)) {
551ff5ca166SMax Reitz         CURLSocket *socket = QLIST_FIRST(&s->sockets);
552ff5ca166SMax Reitz 
553ff5ca166SMax Reitz         QLIST_REMOVE(socket, next);
554ff5ca166SMax Reitz         g_free(socket);
555ff5ca166SMax Reitz     }
556ff5ca166SMax Reitz 
557769ce76dSAlexander Graf     s->in_use = 0;
5582bb5c936SPaolo Bonzini 
5592bb5c936SPaolo Bonzini     next = QSIMPLEQ_FIRST(&s->s->free_state_waitq);
5602bb5c936SPaolo Bonzini     if (next) {
5612bb5c936SPaolo Bonzini         QSIMPLEQ_REMOVE_HEAD(&s->s->free_state_waitq, next);
5622bb5c936SPaolo Bonzini         qemu_mutex_unlock(&s->s->mutex);
5632bb5c936SPaolo Bonzini         aio_co_wake(next->co);
5642bb5c936SPaolo Bonzini         qemu_mutex_lock(&s->s->mutex);
5652bb5c936SPaolo Bonzini     }
566769ce76dSAlexander Graf }
567769ce76dSAlexander Graf 
5688e6d58cdSKevin Wolf static void curl_parse_filename(const char *filename, QDict *options,
5698e6d58cdSKevin Wolf                                 Error **errp)
570769ce76dSAlexander Graf {
57146f5ac20SEric Blake     qdict_put_str(options, CURL_BLOCK_OPT_URL, filename);
5728e6d58cdSKevin Wolf }
5738e6d58cdSKevin Wolf 
57463f0f45fSStefan Hajnoczi static void curl_detach_aio_context(BlockDriverState *bs)
57563f0f45fSStefan Hajnoczi {
57663f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
57763f0f45fSStefan Hajnoczi     int i;
57863f0f45fSStefan Hajnoczi 
579456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
58063f0f45fSStefan Hajnoczi     for (i = 0; i < CURL_NUM_STATES; i++) {
58163f0f45fSStefan Hajnoczi         if (s->states[i].in_use) {
58263f0f45fSStefan Hajnoczi             curl_clean_state(&s->states[i]);
58363f0f45fSStefan Hajnoczi         }
58463f0f45fSStefan Hajnoczi         if (s->states[i].curl) {
58563f0f45fSStefan Hajnoczi             curl_easy_cleanup(s->states[i].curl);
58663f0f45fSStefan Hajnoczi             s->states[i].curl = NULL;
58763f0f45fSStefan Hajnoczi         }
58863f0f45fSStefan Hajnoczi         g_free(s->states[i].orig_buf);
58963f0f45fSStefan Hajnoczi         s->states[i].orig_buf = NULL;
59063f0f45fSStefan Hajnoczi     }
59163f0f45fSStefan Hajnoczi     if (s->multi) {
59263f0f45fSStefan Hajnoczi         curl_multi_cleanup(s->multi);
59363f0f45fSStefan Hajnoczi         s->multi = NULL;
59463f0f45fSStefan Hajnoczi     }
595456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
59663f0f45fSStefan Hajnoczi 
59763f0f45fSStefan Hajnoczi     timer_del(&s->timer);
59863f0f45fSStefan Hajnoczi }
59963f0f45fSStefan Hajnoczi 
60063f0f45fSStefan Hajnoczi static void curl_attach_aio_context(BlockDriverState *bs,
60163f0f45fSStefan Hajnoczi                                     AioContext *new_context)
60263f0f45fSStefan Hajnoczi {
60363f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
60463f0f45fSStefan Hajnoczi 
60563f0f45fSStefan Hajnoczi     aio_timer_init(new_context, &s->timer,
60663f0f45fSStefan Hajnoczi                    QEMU_CLOCK_REALTIME, SCALE_NS,
60763f0f45fSStefan Hajnoczi                    curl_multi_timeout_do, s);
60863f0f45fSStefan Hajnoczi 
60963f0f45fSStefan Hajnoczi     assert(!s->multi);
61063f0f45fSStefan Hajnoczi     s->multi = curl_multi_init();
61163f0f45fSStefan Hajnoczi     s->aio_context = new_context;
61263f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_SOCKETFUNCTION, curl_sock_cb);
61363f0f45fSStefan Hajnoczi #ifdef NEED_CURL_TIMER_CALLBACK
61463f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERDATA, s);
61563f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERFUNCTION, curl_timer_cb);
61663f0f45fSStefan Hajnoczi #endif
61763f0f45fSStefan Hajnoczi }
61863f0f45fSStefan Hajnoczi 
6198e6d58cdSKevin Wolf static QemuOptsList runtime_opts = {
6208e6d58cdSKevin Wolf     .name = "curl",
6218e6d58cdSKevin Wolf     .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
6228e6d58cdSKevin Wolf     .desc = {
6238e6d58cdSKevin Wolf         {
624e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_URL,
6258e6d58cdSKevin Wolf             .type = QEMU_OPT_STRING,
6268e6d58cdSKevin Wolf             .help = "URL to open",
6278e6d58cdSKevin Wolf         },
6288e6d58cdSKevin Wolf         {
629e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_READAHEAD,
6308e6d58cdSKevin Wolf             .type = QEMU_OPT_SIZE,
6318e6d58cdSKevin Wolf             .help = "Readahead size",
6328e6d58cdSKevin Wolf         },
63397a3ea57SMatthew Booth         {
63497a3ea57SMatthew Booth             .name = CURL_BLOCK_OPT_SSLVERIFY,
63597a3ea57SMatthew Booth             .type = QEMU_OPT_BOOL,
63697a3ea57SMatthew Booth             .help = "Verify SSL certificate"
63797a3ea57SMatthew Booth         },
638212aefaaSDaniel Henrique Barboza         {
639212aefaaSDaniel Henrique Barboza             .name = CURL_BLOCK_OPT_TIMEOUT,
640212aefaaSDaniel Henrique Barboza             .type = QEMU_OPT_NUMBER,
641212aefaaSDaniel Henrique Barboza             .help = "Curl timeout"
642212aefaaSDaniel Henrique Barboza         },
643a94f83d9SRichard W.M. Jones         {
644a94f83d9SRichard W.M. Jones             .name = CURL_BLOCK_OPT_COOKIE,
645a94f83d9SRichard W.M. Jones             .type = QEMU_OPT_STRING,
646a94f83d9SRichard W.M. Jones             .help = "Pass the cookie or list of cookies with each request"
647a94f83d9SRichard W.M. Jones         },
6481bff9606SDaniel P. Berrange         {
649327c8ebdSPeter Krempa             .name = CURL_BLOCK_OPT_COOKIE_SECRET,
650327c8ebdSPeter Krempa             .type = QEMU_OPT_STRING,
651327c8ebdSPeter Krempa             .help = "ID of secret used as cookie passed with each request"
652327c8ebdSPeter Krempa         },
653327c8ebdSPeter Krempa         {
6541bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_USERNAME,
6551bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6561bff9606SDaniel P. Berrange             .help = "Username for HTTP auth"
6571bff9606SDaniel P. Berrange         },
6581bff9606SDaniel P. Berrange         {
6591bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PASSWORD_SECRET,
6601bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6611bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP auth",
6621bff9606SDaniel P. Berrange         },
6631bff9606SDaniel P. Berrange         {
6641bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_USERNAME,
6651bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6661bff9606SDaniel P. Berrange             .help = "Username for HTTP proxy auth"
6671bff9606SDaniel P. Berrange         },
6681bff9606SDaniel P. Berrange         {
6691bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET,
6701bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6711bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP proxy auth",
6721bff9606SDaniel P. Berrange         },
6738e6d58cdSKevin Wolf         { /* end of list */ }
6748e6d58cdSKevin Wolf     },
6758e6d58cdSKevin Wolf };
6768e6d58cdSKevin Wolf 
6771bff9606SDaniel P. Berrange 
678015a1036SMax Reitz static int curl_open(BlockDriverState *bs, QDict *options, int flags,
679015a1036SMax Reitz                      Error **errp)
6808e6d58cdSKevin Wolf {
6818e6d58cdSKevin Wolf     BDRVCURLState *s = bs->opaque;
6828e6d58cdSKevin Wolf     CURLState *state = NULL;
6838e6d58cdSKevin Wolf     QemuOpts *opts;
6848e6d58cdSKevin Wolf     Error *local_err = NULL;
6858e6d58cdSKevin Wolf     const char *file;
686a94f83d9SRichard W.M. Jones     const char *cookie;
687327c8ebdSPeter Krempa     const char *cookie_secret;
6888e6d58cdSKevin Wolf     double d;
6891bff9606SDaniel P. Berrange     const char *secretid;
69034634ca2SMax Reitz     const char *protocol_delimiter;
691*2d25964dSJeff Cody     int ret;
6928e6d58cdSKevin Wolf 
6938e6d58cdSKevin Wolf 
694a7cea2baSRichard W.M. Jones     if (flags & BDRV_O_RDWR) {
6952a94fee3SPaolo Bonzini         error_setg(errp, "curl block device does not support writes");
696a7cea2baSRichard W.M. Jones         return -EROFS;
697a7cea2baSRichard W.M. Jones     }
698a7cea2baSRichard W.M. Jones 
699*2d25964dSJeff Cody     if (!libcurl_initialized) {
700*2d25964dSJeff Cody         ret = curl_global_init(CURL_GLOBAL_ALL);
701*2d25964dSJeff Cody         if (ret) {
702*2d25964dSJeff Cody             error_setg(errp, "libcurl initialization failed with %d", ret);
703*2d25964dSJeff Cody             return -EIO;
704*2d25964dSJeff Cody         }
705*2d25964dSJeff Cody         libcurl_initialized = true;
706*2d25964dSJeff Cody     }
707*2d25964dSJeff Cody 
708456af346SPaolo Bonzini     qemu_mutex_init(&s->mutex);
70987ea75d5SPeter Crosthwaite     opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
7108e6d58cdSKevin Wolf     qemu_opts_absorb_qdict(opts, options, &local_err);
71184d18f06SMarkus Armbruster     if (local_err) {
7122a94fee3SPaolo Bonzini         error_propagate(errp, local_err);
7138e6d58cdSKevin Wolf         goto out_noclean;
7148e6d58cdSKevin Wolf     }
7158e6d58cdSKevin Wolf 
716e3542c67SMatthew Booth     s->readahead_size = qemu_opt_get_size(opts, CURL_BLOCK_OPT_READAHEAD,
717e3542c67SMatthew Booth                                           READ_AHEAD_DEFAULT);
718c76f4952SNolan     if ((s->readahead_size & 0x1ff) != 0) {
7192a94fee3SPaolo Bonzini         error_setg(errp, "HTTP_READAHEAD_SIZE %zd is not a multiple of 512",
720c76f4952SNolan                    s->readahead_size);
721c76f4952SNolan         goto out_noclean;
722c76f4952SNolan     }
723c76f4952SNolan 
724212aefaaSDaniel Henrique Barboza     s->timeout = qemu_opt_get_number(opts, CURL_BLOCK_OPT_TIMEOUT,
725212aefaaSDaniel Henrique Barboza                                      CURL_TIMEOUT_DEFAULT);
726f76faedaSRichard W.M. Jones     if (s->timeout > CURL_TIMEOUT_MAX) {
727f76faedaSRichard W.M. Jones         error_setg(errp, "timeout parameter is too large or negative");
728f76faedaSRichard W.M. Jones         goto out_noclean;
729f76faedaSRichard W.M. Jones     }
730212aefaaSDaniel Henrique Barboza 
73197a3ea57SMatthew Booth     s->sslverify = qemu_opt_get_bool(opts, CURL_BLOCK_OPT_SSLVERIFY, true);
73297a3ea57SMatthew Booth 
733a94f83d9SRichard W.M. Jones     cookie = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE);
734327c8ebdSPeter Krempa     cookie_secret = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE_SECRET);
735327c8ebdSPeter Krempa 
736327c8ebdSPeter Krempa     if (cookie && cookie_secret) {
737327c8ebdSPeter Krempa         error_setg(errp,
738327c8ebdSPeter Krempa                    "curl driver cannot handle both cookie and cookie secret");
739327c8ebdSPeter Krempa         goto out_noclean;
740327c8ebdSPeter Krempa     }
741327c8ebdSPeter Krempa 
742327c8ebdSPeter Krempa     if (cookie_secret) {
743327c8ebdSPeter Krempa         s->cookie = qcrypto_secret_lookup_as_utf8(cookie_secret, errp);
744327c8ebdSPeter Krempa         if (!s->cookie) {
745327c8ebdSPeter Krempa             goto out_noclean;
746327c8ebdSPeter Krempa         }
747327c8ebdSPeter Krempa     } else {
748a94f83d9SRichard W.M. Jones         s->cookie = g_strdup(cookie);
749327c8ebdSPeter Krempa     }
750a94f83d9SRichard W.M. Jones 
751e3542c67SMatthew Booth     file = qemu_opt_get(opts, CURL_BLOCK_OPT_URL);
7528e6d58cdSKevin Wolf     if (file == NULL) {
7532a94fee3SPaolo Bonzini         error_setg(errp, "curl block driver requires an 'url' option");
7548e6d58cdSKevin Wolf         goto out_noclean;
7558e6d58cdSKevin Wolf     }
7568e6d58cdSKevin Wolf 
75734634ca2SMax Reitz     if (!strstart(file, bs->drv->protocol_name, &protocol_delimiter) ||
75834634ca2SMax Reitz         !strstart(protocol_delimiter, "://", NULL))
75934634ca2SMax Reitz     {
76034634ca2SMax Reitz         error_setg(errp, "%s curl driver cannot handle the URL '%s' (does not "
76134634ca2SMax Reitz                    "start with '%s://')", bs->drv->protocol_name, file,
76234634ca2SMax Reitz                    bs->drv->protocol_name);
76334634ca2SMax Reitz         goto out_noclean;
76434634ca2SMax Reitz     }
76534634ca2SMax Reitz 
7661bff9606SDaniel P. Berrange     s->username = g_strdup(qemu_opt_get(opts, CURL_BLOCK_OPT_USERNAME));
7671bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PASSWORD_SECRET);
7681bff9606SDaniel P. Berrange 
7691bff9606SDaniel P. Berrange     if (secretid) {
7701bff9606SDaniel P. Berrange         s->password = qcrypto_secret_lookup_as_utf8(secretid, errp);
7711bff9606SDaniel P. Berrange         if (!s->password) {
7721bff9606SDaniel P. Berrange             goto out_noclean;
7731bff9606SDaniel P. Berrange         }
7741bff9606SDaniel P. Berrange     }
7751bff9606SDaniel P. Berrange 
7761bff9606SDaniel P. Berrange     s->proxyusername = g_strdup(
7771bff9606SDaniel P. Berrange         qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_USERNAME));
7781bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET);
7791bff9606SDaniel P. Berrange     if (secretid) {
7801bff9606SDaniel P. Berrange         s->proxypassword = qcrypto_secret_lookup_as_utf8(secretid, errp);
7811bff9606SDaniel P. Berrange         if (!s->proxypassword) {
7821bff9606SDaniel P. Berrange             goto out_noclean;
7831bff9606SDaniel P. Berrange         }
7841bff9606SDaniel P. Berrange     }
7851bff9606SDaniel P. Berrange 
786d0f2c4c6Smalc     DPRINTF("CURL: Opening %s\n", file);
7872bb5c936SPaolo Bonzini     QSIMPLEQ_INIT(&s->free_state_waitq);
78863f0f45fSStefan Hajnoczi     s->aio_context = bdrv_get_aio_context(bs);
7898e6d58cdSKevin Wolf     s->url = g_strdup(file);
790456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
7913ce6a729SPaolo Bonzini     state = curl_find_state(s);
792456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
7933ce6a729SPaolo Bonzini     if (!state) {
794769ce76dSAlexander Graf         goto out_noclean;
7953ce6a729SPaolo Bonzini     }
796769ce76dSAlexander Graf 
797769ce76dSAlexander Graf     // Get file size
798769ce76dSAlexander Graf 
7993ce6a729SPaolo Bonzini     if (curl_init_state(s, state) < 0) {
8003ce6a729SPaolo Bonzini         goto out;
8013ce6a729SPaolo Bonzini     }
8023ce6a729SPaolo Bonzini 
8033494d650SFam Zheng     s->accept_range = false;
804769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_NOBODY, 1);
8053494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERFUNCTION,
8063494d650SFam Zheng                      curl_header_cb);
8073494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERDATA, s);
808769ce76dSAlexander Graf     if (curl_easy_perform(state->curl))
809769ce76dSAlexander Graf         goto out;
810a41c4578STomáš Golembiovský     if (curl_easy_getinfo(state->curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d)) {
811769ce76dSAlexander Graf         goto out;
812a41c4578STomáš Golembiovský     }
813a41c4578STomáš Golembiovský     /* Prior CURL 7.19.4 return value of 0 could mean that the file size is not
814a41c4578STomáš Golembiovský      * know or the size is zero. From 7.19.4 CURL returns -1 if size is not
815a41c4578STomáš Golembiovský      * known and zero if it is realy zero-length file. */
816a41c4578STomáš Golembiovský #if LIBCURL_VERSION_NUM >= 0x071304
817a41c4578STomáš Golembiovský     if (d < 0) {
818a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
819a41c4578STomáš Golembiovský                 "Server didn't report file size.");
820a41c4578STomáš Golembiovský         goto out;
821a41c4578STomáš Golembiovský     }
822a41c4578STomáš Golembiovský #else
823a41c4578STomáš Golembiovský     if (d <= 0) {
824a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
825a41c4578STomáš Golembiovský                 "Unknown file size or zero-length file.");
826a41c4578STomáš Golembiovský         goto out;
827a41c4578STomáš Golembiovský     }
828a41c4578STomáš Golembiovský #endif
829a41c4578STomáš Golembiovský 
8302125e5eaSPaolo Bonzini     s->len = d;
831a41c4578STomáš Golembiovský 
8323494d650SFam Zheng     if ((!strncasecmp(s->url, "http://", strlen("http://"))
8333494d650SFam Zheng         || !strncasecmp(s->url, "https://", strlen("https://")))
8343494d650SFam Zheng         && !s->accept_range) {
8353494d650SFam Zheng         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
8363494d650SFam Zheng                 "Server does not support 'range' (byte ranges).");
8373494d650SFam Zheng         goto out;
8383494d650SFam Zheng     }
8392125e5eaSPaolo Bonzini     DPRINTF("CURL: Size = %" PRIu64 "\n", s->len);
840769ce76dSAlexander Graf 
841456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
842769ce76dSAlexander Graf     curl_clean_state(state);
843456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
844769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
845769ce76dSAlexander Graf     state->curl = NULL;
846769ce76dSAlexander Graf 
84763f0f45fSStefan Hajnoczi     curl_attach_aio_context(bs, bdrv_get_aio_context(bs));
848769ce76dSAlexander Graf 
8498e6d58cdSKevin Wolf     qemu_opts_del(opts);
850769ce76dSAlexander Graf     return 0;
851769ce76dSAlexander Graf 
852769ce76dSAlexander Graf out:
853acd7fdc6SMaria Kustova     error_setg(errp, "CURL: Error opening file: %s", state->errmsg);
854769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
855769ce76dSAlexander Graf     state->curl = NULL;
856769ce76dSAlexander Graf out_noclean:
857456af346SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
858a94f83d9SRichard W.M. Jones     g_free(s->cookie);
8598e6d58cdSKevin Wolf     g_free(s->url);
8608e6d58cdSKevin Wolf     qemu_opts_del(opts);
861769ce76dSAlexander Graf     return -EINVAL;
862769ce76dSAlexander Graf }
863769ce76dSAlexander Graf 
86428256d82SPaolo Bonzini static void curl_setup_preadv(BlockDriverState *bs, CURLAIOCB *acb)
865769ce76dSAlexander Graf {
866769ce76dSAlexander Graf     CURLState *state;
867b69cdef8SMatthew Booth     int running;
868769ce76dSAlexander Graf 
8691919631eSPaolo Bonzini     BDRVCURLState *s = bs->opaque;
870769ce76dSAlexander Graf 
8712125e5eaSPaolo Bonzini     uint64_t start = acb->offset;
8722125e5eaSPaolo Bonzini     uint64_t end;
873769ce76dSAlexander Graf 
874ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
8751919631eSPaolo Bonzini 
876769ce76dSAlexander Graf     // In case we have the requested data already (e.g. read-ahead),
877769ce76dSAlexander Graf     // we can just call the callback and be done.
87828256d82SPaolo Bonzini     if (curl_find_buf(s, start, acb->bytes, acb)) {
87934db05e7SPaolo Bonzini         goto out;
880769ce76dSAlexander Graf     }
881769ce76dSAlexander Graf 
882769ce76dSAlexander Graf     // No cache found, so let's start a new request
8833ce6a729SPaolo Bonzini     for (;;) {
8843ce6a729SPaolo Bonzini         state = curl_find_state(s);
8853ce6a729SPaolo Bonzini         if (state) {
8863ce6a729SPaolo Bonzini             break;
8873ce6a729SPaolo Bonzini         }
8882bb5c936SPaolo Bonzini         QSIMPLEQ_INSERT_TAIL(&s->free_state_waitq, acb, next);
8893ce6a729SPaolo Bonzini         qemu_mutex_unlock(&s->mutex);
8902bb5c936SPaolo Bonzini         qemu_coroutine_yield();
8913ce6a729SPaolo Bonzini         qemu_mutex_lock(&s->mutex);
8923ce6a729SPaolo Bonzini     }
8933ce6a729SPaolo Bonzini 
8943ce6a729SPaolo Bonzini     if (curl_init_state(s, state) < 0) {
8953ce6a729SPaolo Bonzini         curl_clean_state(state);
89628256d82SPaolo Bonzini         acb->ret = -EIO;
8971919631eSPaolo Bonzini         goto out;
898363c3c85SNick Thomas     }
899769ce76dSAlexander Graf 
900769ce76dSAlexander Graf     acb->start = 0;
9012125e5eaSPaolo Bonzini     acb->end = MIN(acb->bytes, s->len - start);
902769ce76dSAlexander Graf 
903769ce76dSAlexander Graf     state->buf_off = 0;
9047267c094SAnthony Liguori     g_free(state->orig_buf);
905769ce76dSAlexander Graf     state->buf_start = start;
9064e504535SMax Reitz     state->buf_len = MIN(acb->end + s->readahead_size, s->len - start);
9074e504535SMax Reitz     end = start + state->buf_len - 1;
9088dc7a772SKevin Wolf     state->orig_buf = g_try_malloc(state->buf_len);
9098dc7a772SKevin Wolf     if (state->buf_len && state->orig_buf == NULL) {
9108dc7a772SKevin Wolf         curl_clean_state(state);
91128256d82SPaolo Bonzini         acb->ret = -ENOMEM;
9121919631eSPaolo Bonzini         goto out;
9138dc7a772SKevin Wolf     }
914769ce76dSAlexander Graf     state->acb[0] = acb;
915769ce76dSAlexander Graf 
9162125e5eaSPaolo Bonzini     snprintf(state->range, 127, "%" PRIu64 "-%" PRIu64, start, end);
9172125e5eaSPaolo Bonzini     DPRINTF("CURL (AIO): Reading %" PRIu64 " at %" PRIu64 " (%s)\n",
9182125e5eaSPaolo Bonzini             acb->bytes, start, state->range);
919769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_RANGE, state->range);
920769ce76dSAlexander Graf 
921769ce76dSAlexander Graf     curl_multi_add_handle(s->multi, state->curl);
922769ce76dSAlexander Graf 
923b69cdef8SMatthew Booth     /* Tell curl it needs to kick things off */
924b69cdef8SMatthew Booth     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
9251919631eSPaolo Bonzini 
9261919631eSPaolo Bonzini out:
927ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
928363c3c85SNick Thomas }
929363c3c85SNick Thomas 
93028256d82SPaolo Bonzini static int coroutine_fn curl_co_preadv(BlockDriverState *bs,
93128256d82SPaolo Bonzini         uint64_t offset, uint64_t bytes, QEMUIOVector *qiov, int flags)
932363c3c85SNick Thomas {
93328256d82SPaolo Bonzini     CURLAIOCB acb = {
93428256d82SPaolo Bonzini         .co = qemu_coroutine_self(),
93528256d82SPaolo Bonzini         .ret = -EINPROGRESS,
93628256d82SPaolo Bonzini         .qiov = qiov,
93728256d82SPaolo Bonzini         .offset = offset,
93828256d82SPaolo Bonzini         .bytes = bytes
93928256d82SPaolo Bonzini     };
940363c3c85SNick Thomas 
94128256d82SPaolo Bonzini     curl_setup_preadv(bs, &acb);
94228256d82SPaolo Bonzini     while (acb.ret == -EINPROGRESS) {
94328256d82SPaolo Bonzini         qemu_coroutine_yield();
94428256d82SPaolo Bonzini     }
94528256d82SPaolo Bonzini     return acb.ret;
946769ce76dSAlexander Graf }
947769ce76dSAlexander Graf 
948769ce76dSAlexander Graf static void curl_close(BlockDriverState *bs)
949769ce76dSAlexander Graf {
950769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
951769ce76dSAlexander Graf 
952d0f2c4c6Smalc     DPRINTF("CURL: Close\n");
95363f0f45fSStefan Hajnoczi     curl_detach_aio_context(bs);
954ba3186c4SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
955031fd1beSPeter Maydell 
956a94f83d9SRichard W.M. Jones     g_free(s->cookie);
95745724d6dSStefan Weil     g_free(s->url);
958769ce76dSAlexander Graf }
959769ce76dSAlexander Graf 
960769ce76dSAlexander Graf static int64_t curl_getlength(BlockDriverState *bs)
961769ce76dSAlexander Graf {
962769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
963769ce76dSAlexander Graf     return s->len;
964769ce76dSAlexander Graf }
965769ce76dSAlexander Graf 
966769ce76dSAlexander Graf static BlockDriver bdrv_http = {
967769ce76dSAlexander Graf     .format_name                = "http",
968769ce76dSAlexander Graf     .protocol_name              = "http",
969769ce76dSAlexander Graf 
970769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9718e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
97266f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
973769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
974769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
975769ce76dSAlexander Graf 
97628256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
97763f0f45fSStefan Hajnoczi 
97863f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
97963f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
980769ce76dSAlexander Graf };
981769ce76dSAlexander Graf 
982769ce76dSAlexander Graf static BlockDriver bdrv_https = {
983769ce76dSAlexander Graf     .format_name                = "https",
984769ce76dSAlexander Graf     .protocol_name              = "https",
985769ce76dSAlexander Graf 
986769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9878e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
98866f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
989769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
990769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
991769ce76dSAlexander Graf 
99228256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
99363f0f45fSStefan Hajnoczi 
99463f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
99563f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
996769ce76dSAlexander Graf };
997769ce76dSAlexander Graf 
998769ce76dSAlexander Graf static BlockDriver bdrv_ftp = {
999769ce76dSAlexander Graf     .format_name                = "ftp",
1000769ce76dSAlexander Graf     .protocol_name              = "ftp",
1001769ce76dSAlexander Graf 
1002769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
10038e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
100466f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
1005769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
1006769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
1007769ce76dSAlexander Graf 
100828256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
100963f0f45fSStefan Hajnoczi 
101063f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
101163f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
1012769ce76dSAlexander Graf };
1013769ce76dSAlexander Graf 
1014769ce76dSAlexander Graf static BlockDriver bdrv_ftps = {
1015769ce76dSAlexander Graf     .format_name                = "ftps",
1016769ce76dSAlexander Graf     .protocol_name              = "ftps",
1017769ce76dSAlexander Graf 
1018769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
10198e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
102066f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
1021769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
1022769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
1023769ce76dSAlexander Graf 
102428256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
102563f0f45fSStefan Hajnoczi 
102663f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
102763f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
1028769ce76dSAlexander Graf };
1029769ce76dSAlexander Graf 
1030769ce76dSAlexander Graf static void curl_block_init(void)
1031769ce76dSAlexander Graf {
1032769ce76dSAlexander Graf     bdrv_register(&bdrv_http);
1033769ce76dSAlexander Graf     bdrv_register(&bdrv_https);
1034769ce76dSAlexander Graf     bdrv_register(&bdrv_ftp);
1035769ce76dSAlexander Graf     bdrv_register(&bdrv_ftps);
1036769ce76dSAlexander Graf }
1037769ce76dSAlexander Graf 
1038769ce76dSAlexander Graf block_init(curl_block_init);
1039