xref: /openbmc/qemu/block/curl.c (revision 28256d8246f8905cc41cae3db50e5967059d4600)
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 
92769ce76dSAlexander Graf typedef struct CURLAIOCB {
93*28256d82SPaolo Bonzini     Coroutine *co;
94769ce76dSAlexander Graf     QEMUIOVector *qiov;
95363c3c85SNick Thomas 
962125e5eaSPaolo Bonzini     uint64_t offset;
972125e5eaSPaolo Bonzini     uint64_t bytes;
98*28256d82SPaolo Bonzini     int ret;
99363c3c85SNick Thomas 
100769ce76dSAlexander Graf     size_t start;
101769ce76dSAlexander Graf     size_t end;
102769ce76dSAlexander Graf } CURLAIOCB;
103769ce76dSAlexander Graf 
104ff5ca166SMax Reitz typedef struct CURLSocket {
105ff5ca166SMax Reitz     int fd;
106ff5ca166SMax Reitz     QLIST_ENTRY(CURLSocket) next;
107ff5ca166SMax Reitz } CURLSocket;
108ff5ca166SMax Reitz 
109769ce76dSAlexander Graf typedef struct CURLState
110769ce76dSAlexander Graf {
111769ce76dSAlexander Graf     struct BDRVCURLState *s;
112769ce76dSAlexander Graf     CURLAIOCB *acb[CURL_NUM_ACB];
113769ce76dSAlexander Graf     CURL *curl;
114ff5ca166SMax Reitz     QLIST_HEAD(, CURLSocket) sockets;
115769ce76dSAlexander Graf     char *orig_buf;
1162125e5eaSPaolo Bonzini     uint64_t buf_start;
117769ce76dSAlexander Graf     size_t buf_off;
118769ce76dSAlexander Graf     size_t buf_len;
119769ce76dSAlexander Graf     char range[128];
120769ce76dSAlexander Graf     char errmsg[CURL_ERROR_SIZE];
121769ce76dSAlexander Graf     char in_use;
122769ce76dSAlexander Graf } CURLState;
123769ce76dSAlexander Graf 
124769ce76dSAlexander Graf typedef struct BDRVCURLState {
125769ce76dSAlexander Graf     CURLM *multi;
126031fd1beSPeter Maydell     QEMUTimer timer;
1272125e5eaSPaolo Bonzini     uint64_t len;
128769ce76dSAlexander Graf     CURLState states[CURL_NUM_STATES];
129769ce76dSAlexander Graf     char *url;
130c76f4952SNolan     size_t readahead_size;
13197a3ea57SMatthew Booth     bool sslverify;
132f76faedaSRichard W.M. Jones     uint64_t timeout;
133a94f83d9SRichard W.M. Jones     char *cookie;
1343494d650SFam Zheng     bool accept_range;
13563f0f45fSStefan Hajnoczi     AioContext *aio_context;
136ba3186c4SPaolo Bonzini     QemuMutex mutex;
1371bff9606SDaniel P. Berrange     char *username;
1381bff9606SDaniel P. Berrange     char *password;
1391bff9606SDaniel P. Berrange     char *proxyusername;
1401bff9606SDaniel P. Berrange     char *proxypassword;
141769ce76dSAlexander Graf } BDRVCURLState;
142769ce76dSAlexander Graf 
143769ce76dSAlexander Graf static void curl_clean_state(CURLState *s);
144769ce76dSAlexander Graf static void curl_multi_do(void *arg);
145838ef602SMatthew Booth static void curl_multi_read(void *arg);
146769ce76dSAlexander Graf 
147031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
14834db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
149031fd1beSPeter Maydell static int curl_timer_cb(CURLM *multi, long timeout_ms, void *opaque)
150031fd1beSPeter Maydell {
151031fd1beSPeter Maydell     BDRVCURLState *s = opaque;
152031fd1beSPeter Maydell 
153031fd1beSPeter Maydell     DPRINTF("CURL: timer callback timeout_ms %ld\n", timeout_ms);
154031fd1beSPeter Maydell     if (timeout_ms == -1) {
155031fd1beSPeter Maydell         timer_del(&s->timer);
156031fd1beSPeter Maydell     } else {
157031fd1beSPeter Maydell         int64_t timeout_ns = (int64_t)timeout_ms * 1000 * 1000;
158031fd1beSPeter Maydell         timer_mod(&s->timer,
159031fd1beSPeter Maydell                   qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + timeout_ns);
160031fd1beSPeter Maydell     }
161031fd1beSPeter Maydell     return 0;
162031fd1beSPeter Maydell }
163031fd1beSPeter Maydell #endif
164031fd1beSPeter Maydell 
16534db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
166769ce76dSAlexander Graf static int curl_sock_cb(CURL *curl, curl_socket_t fd, int action,
16763f0f45fSStefan Hajnoczi                         void *userp, void *sp)
168769ce76dSAlexander Graf {
16963f0f45fSStefan Hajnoczi     BDRVCURLState *s;
170838ef602SMatthew Booth     CURLState *state = NULL;
171ff5ca166SMax Reitz     CURLSocket *socket;
172ff5ca166SMax Reitz 
173838ef602SMatthew Booth     curl_easy_getinfo(curl, CURLINFO_PRIVATE, (char **)&state);
17463f0f45fSStefan Hajnoczi     s = state->s;
175838ef602SMatthew Booth 
176ff5ca166SMax Reitz     QLIST_FOREACH(socket, &state->sockets, next) {
177ff5ca166SMax Reitz         if (socket->fd == fd) {
178ff5ca166SMax Reitz             if (action == CURL_POLL_REMOVE) {
179ff5ca166SMax Reitz                 QLIST_REMOVE(socket, next);
180ff5ca166SMax Reitz                 g_free(socket);
181ff5ca166SMax Reitz             }
182ff5ca166SMax Reitz             break;
183ff5ca166SMax Reitz         }
184ff5ca166SMax Reitz     }
185ff5ca166SMax Reitz     if (!socket) {
186ff5ca166SMax Reitz         socket = g_new0(CURLSocket, 1);
187ff5ca166SMax Reitz         socket->fd = fd;
188ff5ca166SMax Reitz         QLIST_INSERT_HEAD(&state->sockets, socket, next);
189ff5ca166SMax Reitz     }
190ff5ca166SMax Reitz     socket = NULL;
191ff5ca166SMax Reitz 
19292b6a160SFam Zheng     DPRINTF("CURL (AIO): Sock action %d on fd %d\n", action, (int)fd);
193769ce76dSAlexander Graf     switch (action) {
194769ce76dSAlexander Graf         case CURL_POLL_IN:
195dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
196f6a51c84SStefan Hajnoczi                                curl_multi_read, NULL, NULL, state);
197769ce76dSAlexander Graf             break;
198769ce76dSAlexander Graf         case CURL_POLL_OUT:
199dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
200f6a51c84SStefan Hajnoczi                                NULL, curl_multi_do, NULL, state);
201769ce76dSAlexander Graf             break;
202769ce76dSAlexander Graf         case CURL_POLL_INOUT:
203dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
204f6a51c84SStefan Hajnoczi                                curl_multi_read, curl_multi_do, NULL, state);
205769ce76dSAlexander Graf             break;
206769ce76dSAlexander Graf         case CURL_POLL_REMOVE:
207dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
208f6a51c84SStefan Hajnoczi                                NULL, NULL, NULL, NULL);
209769ce76dSAlexander Graf             break;
210769ce76dSAlexander Graf     }
211769ce76dSAlexander Graf 
212769ce76dSAlexander Graf     return 0;
213769ce76dSAlexander Graf }
214769ce76dSAlexander Graf 
21534db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
2163494d650SFam Zheng static size_t curl_header_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
217769ce76dSAlexander Graf {
2183494d650SFam Zheng     BDRVCURLState *s = opaque;
219769ce76dSAlexander Graf     size_t realsize = size * nmemb;
2203494d650SFam Zheng     const char *accept_line = "Accept-Ranges: bytes";
221769ce76dSAlexander Graf 
2223494d650SFam Zheng     if (realsize >= strlen(accept_line)
2233494d650SFam Zheng         && strncmp((char *)ptr, accept_line, strlen(accept_line)) == 0) {
2243494d650SFam Zheng         s->accept_range = true;
2250bfcd599SBlue Swirl     }
226769ce76dSAlexander Graf 
227769ce76dSAlexander Graf     return realsize;
228769ce76dSAlexander Graf }
229769ce76dSAlexander Graf 
23034db05e7SPaolo Bonzini /* Called from curl_multi_do_locked, with s->mutex held.  */
231769ce76dSAlexander Graf static size_t curl_read_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
232769ce76dSAlexander Graf {
233769ce76dSAlexander Graf     CURLState *s = ((CURLState*)opaque);
234769ce76dSAlexander Graf     size_t realsize = size * nmemb;
235769ce76dSAlexander Graf     int i;
236769ce76dSAlexander Graf 
2370bfcd599SBlue Swirl     DPRINTF("CURL: Just reading %zd bytes\n", realsize);
238769ce76dSAlexander Graf 
2394e767657SMax Reitz     if (!s || !s->orig_buf) {
2404e767657SMax Reitz         goto read_end;
2414e767657SMax Reitz     }
242769ce76dSAlexander Graf 
2436d4b9e55SFam Zheng     if (s->buf_off >= s->buf_len) {
2446d4b9e55SFam Zheng         /* buffer full, read nothing */
2454e767657SMax Reitz         goto read_end;
2466d4b9e55SFam Zheng     }
2476d4b9e55SFam Zheng     realsize = MIN(realsize, s->buf_len - s->buf_off);
248769ce76dSAlexander Graf     memcpy(s->orig_buf + s->buf_off, ptr, realsize);
249769ce76dSAlexander Graf     s->buf_off += realsize;
250769ce76dSAlexander Graf 
251769ce76dSAlexander Graf     for(i=0; i<CURL_NUM_ACB; i++) {
252769ce76dSAlexander Graf         CURLAIOCB *acb = s->acb[i];
253769ce76dSAlexander Graf 
254769ce76dSAlexander Graf         if (!acb)
255769ce76dSAlexander Graf             continue;
256769ce76dSAlexander Graf 
257769ce76dSAlexander Graf         if ((s->buf_off >= acb->end)) {
2582125e5eaSPaolo Bonzini             size_t request_length = acb->bytes;
2594e504535SMax Reitz 
26003396148SMichael Tokarev             qemu_iovec_from_buf(acb->qiov, 0, s->orig_buf + acb->start,
261769ce76dSAlexander Graf                                 acb->end - acb->start);
2624e504535SMax Reitz 
2634e504535SMax Reitz             if (acb->end - acb->start < request_length) {
2644e504535SMax Reitz                 size_t offset = acb->end - acb->start;
2654e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, offset, 0,
2664e504535SMax Reitz                                   request_length - offset);
2674e504535SMax Reitz             }
2684e504535SMax Reitz 
269*28256d82SPaolo Bonzini             acb->ret = 0;
270769ce76dSAlexander Graf             s->acb[i] = NULL;
271*28256d82SPaolo Bonzini             qemu_mutex_unlock(&s->s->mutex);
272*28256d82SPaolo Bonzini             aio_co_wake(acb->co);
273*28256d82SPaolo Bonzini             qemu_mutex_lock(&s->s->mutex);
274769ce76dSAlexander Graf         }
275769ce76dSAlexander Graf     }
276769ce76dSAlexander Graf 
2774e767657SMax Reitz read_end:
2784e767657SMax Reitz     /* curl will error out if we do not return this value */
2794e767657SMax Reitz     return size * nmemb;
280769ce76dSAlexander Graf }
281769ce76dSAlexander Graf 
282456af346SPaolo Bonzini /* Called with s->mutex held.  */
283*28256d82SPaolo Bonzini static bool curl_find_buf(BDRVCURLState *s, uint64_t start, uint64_t len,
284769ce76dSAlexander Graf                           CURLAIOCB *acb)
285769ce76dSAlexander Graf {
286769ce76dSAlexander Graf     int i;
2872125e5eaSPaolo Bonzini     uint64_t end = start + len;
2882125e5eaSPaolo Bonzini     uint64_t clamped_end = MIN(end, s->len);
2892125e5eaSPaolo Bonzini     uint64_t clamped_len = clamped_end - start;
290769ce76dSAlexander Graf 
291769ce76dSAlexander Graf     for (i=0; i<CURL_NUM_STATES; i++) {
292769ce76dSAlexander Graf         CURLState *state = &s->states[i];
2932125e5eaSPaolo Bonzini         uint64_t buf_end = (state->buf_start + state->buf_off);
2942125e5eaSPaolo Bonzini         uint64_t buf_fend = (state->buf_start + state->buf_len);
295769ce76dSAlexander Graf 
296769ce76dSAlexander Graf         if (!state->orig_buf)
297769ce76dSAlexander Graf             continue;
298769ce76dSAlexander Graf         if (!state->buf_off)
299769ce76dSAlexander Graf             continue;
300769ce76dSAlexander Graf 
301769ce76dSAlexander Graf         // Does the existing buffer cover our section?
302769ce76dSAlexander Graf         if ((start >= state->buf_start) &&
303769ce76dSAlexander Graf             (start <= buf_end) &&
3044e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3054e504535SMax Reitz             (clamped_end <= buf_end))
306769ce76dSAlexander Graf         {
307769ce76dSAlexander Graf             char *buf = state->orig_buf + (start - state->buf_start);
308769ce76dSAlexander Graf 
3094e504535SMax Reitz             qemu_iovec_from_buf(acb->qiov, 0, buf, clamped_len);
3104e504535SMax Reitz             if (clamped_len < len) {
3114e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, clamped_len, 0, len - clamped_len);
3124e504535SMax Reitz             }
313*28256d82SPaolo Bonzini             acb->ret = 0;
314*28256d82SPaolo Bonzini             return true;
315769ce76dSAlexander Graf         }
316769ce76dSAlexander Graf 
317769ce76dSAlexander Graf         // Wait for unfinished chunks
318b7079df4SMatthew Booth         if (state->in_use &&
319b7079df4SMatthew Booth             (start >= state->buf_start) &&
320769ce76dSAlexander Graf             (start <= buf_fend) &&
3214e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3224e504535SMax Reitz             (clamped_end <= buf_fend))
323769ce76dSAlexander Graf         {
324769ce76dSAlexander Graf             int j;
325769ce76dSAlexander Graf 
326769ce76dSAlexander Graf             acb->start = start - state->buf_start;
3274e504535SMax Reitz             acb->end = acb->start + clamped_len;
328769ce76dSAlexander Graf 
329769ce76dSAlexander Graf             for (j=0; j<CURL_NUM_ACB; j++) {
330769ce76dSAlexander Graf                 if (!state->acb[j]) {
331769ce76dSAlexander Graf                     state->acb[j] = acb;
332*28256d82SPaolo Bonzini                     return true;
333769ce76dSAlexander Graf                 }
334769ce76dSAlexander Graf             }
335769ce76dSAlexander Graf         }
336769ce76dSAlexander Graf     }
337769ce76dSAlexander Graf 
338*28256d82SPaolo Bonzini     return false;
339769ce76dSAlexander Graf }
340769ce76dSAlexander Graf 
341ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
342838ef602SMatthew Booth static void curl_multi_check_completion(BDRVCURLState *s)
343769ce76dSAlexander Graf {
344769ce76dSAlexander Graf     int msgs_in_queue;
345769ce76dSAlexander Graf 
346769ce76dSAlexander Graf     /* Try to find done transfers, so we can free the easy
347769ce76dSAlexander Graf      * handle again. */
3481f2cead3SMatthew Booth     for (;;) {
349769ce76dSAlexander Graf         CURLMsg *msg;
350769ce76dSAlexander Graf         msg = curl_multi_info_read(s->multi, &msgs_in_queue);
351769ce76dSAlexander Graf 
3521f2cead3SMatthew Booth         /* Quit when there are no more completions */
353769ce76dSAlexander Graf         if (!msg)
354769ce76dSAlexander Graf             break;
355769ce76dSAlexander Graf 
3561f2cead3SMatthew Booth         if (msg->msg == CURLMSG_DONE) {
357769ce76dSAlexander Graf             CURLState *state = NULL;
358f6246509SMatthew Booth             curl_easy_getinfo(msg->easy_handle, CURLINFO_PRIVATE,
359f6246509SMatthew Booth                               (char **)&state);
360f785a5aeSNicholas Thomas 
361f785a5aeSNicholas Thomas             /* ACBs for successful messages get completed in curl_read_cb */
362f785a5aeSNicholas Thomas             if (msg->data.result != CURLE_OK) {
363f785a5aeSNicholas Thomas                 int i;
364796a060bSRichard W.M. Jones                 static int errcount = 100;
365796a060bSRichard W.M. Jones 
366796a060bSRichard W.M. Jones                 /* Don't lose the original error message from curl, since
367796a060bSRichard W.M. Jones                  * it contains extra data.
368796a060bSRichard W.M. Jones                  */
369796a060bSRichard W.M. Jones                 if (errcount > 0) {
370796a060bSRichard W.M. Jones                     error_report("curl: %s", state->errmsg);
371796a060bSRichard W.M. Jones                     if (--errcount == 0) {
372796a060bSRichard W.M. Jones                         error_report("curl: further errors suppressed");
373796a060bSRichard W.M. Jones                     }
374796a060bSRichard W.M. Jones                 }
375796a060bSRichard W.M. Jones 
376f785a5aeSNicholas Thomas                 for (i = 0; i < CURL_NUM_ACB; i++) {
377f785a5aeSNicholas Thomas                     CURLAIOCB *acb = state->acb[i];
378f785a5aeSNicholas Thomas 
379f785a5aeSNicholas Thomas                     if (acb == NULL) {
380f785a5aeSNicholas Thomas                         continue;
381f785a5aeSNicholas Thomas                     }
382f785a5aeSNicholas Thomas 
383*28256d82SPaolo Bonzini                     acb->ret = -EIO;
384f785a5aeSNicholas Thomas                     state->acb[i] = NULL;
385*28256d82SPaolo Bonzini                     qemu_mutex_unlock(&s->mutex);
386*28256d82SPaolo Bonzini                     aio_co_wake(acb->co);
387*28256d82SPaolo Bonzini                     qemu_mutex_lock(&s->mutex);
388f785a5aeSNicholas Thomas                 }
389f785a5aeSNicholas Thomas             }
390f785a5aeSNicholas Thomas 
391769ce76dSAlexander Graf             curl_clean_state(state);
392769ce76dSAlexander Graf             break;
393769ce76dSAlexander Graf         }
394769ce76dSAlexander Graf     }
395769ce76dSAlexander Graf }
396769ce76dSAlexander Graf 
397ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
3989d456654SPaolo Bonzini static void curl_multi_do_locked(CURLState *s)
399031fd1beSPeter Maydell {
400ff5ca166SMax Reitz     CURLSocket *socket, *next_socket;
401031fd1beSPeter Maydell     int running;
402031fd1beSPeter Maydell     int r;
403031fd1beSPeter Maydell 
404838ef602SMatthew Booth     if (!s->s->multi) {
405031fd1beSPeter Maydell         return;
406031fd1beSPeter Maydell     }
407031fd1beSPeter Maydell 
408ff5ca166SMax Reitz     /* Need to use _SAFE because curl_multi_socket_action() may trigger
409ff5ca166SMax Reitz      * curl_sock_cb() which might modify this list */
410ff5ca166SMax Reitz     QLIST_FOREACH_SAFE(socket, &s->sockets, next, next_socket) {
411031fd1beSPeter Maydell         do {
412ff5ca166SMax Reitz             r = curl_multi_socket_action(s->s->multi, socket->fd, 0, &running);
413031fd1beSPeter Maydell         } while (r == CURLM_CALL_MULTI_PERFORM);
414ff5ca166SMax Reitz     }
415838ef602SMatthew Booth }
416838ef602SMatthew Booth 
4179d456654SPaolo Bonzini static void curl_multi_do(void *arg)
4189d456654SPaolo Bonzini {
4199d456654SPaolo Bonzini     CURLState *s = (CURLState *)arg;
4209d456654SPaolo Bonzini 
421ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->s->mutex);
4229d456654SPaolo Bonzini     curl_multi_do_locked(s);
423ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->s->mutex);
4249d456654SPaolo Bonzini }
4259d456654SPaolo Bonzini 
426838ef602SMatthew Booth static void curl_multi_read(void *arg)
427838ef602SMatthew Booth {
428838ef602SMatthew Booth     CURLState *s = (CURLState *)arg;
429838ef602SMatthew Booth 
430ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->s->mutex);
4319d456654SPaolo Bonzini     curl_multi_do_locked(s);
432838ef602SMatthew Booth     curl_multi_check_completion(s->s);
433ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->s->mutex);
434031fd1beSPeter Maydell }
435031fd1beSPeter Maydell 
436031fd1beSPeter Maydell static void curl_multi_timeout_do(void *arg)
437031fd1beSPeter Maydell {
438031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
439031fd1beSPeter Maydell     BDRVCURLState *s = (BDRVCURLState *)arg;
440031fd1beSPeter Maydell     int running;
441031fd1beSPeter Maydell 
442031fd1beSPeter Maydell     if (!s->multi) {
443031fd1beSPeter Maydell         return;
444031fd1beSPeter Maydell     }
445031fd1beSPeter Maydell 
446ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
447031fd1beSPeter Maydell     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
448031fd1beSPeter Maydell 
449838ef602SMatthew Booth     curl_multi_check_completion(s);
450ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
451031fd1beSPeter Maydell #else
452031fd1beSPeter Maydell     abort();
453031fd1beSPeter Maydell #endif
454031fd1beSPeter Maydell }
455031fd1beSPeter Maydell 
456456af346SPaolo Bonzini /* Called with s->mutex held.  */
4573ce6a729SPaolo Bonzini static CURLState *curl_find_state(BDRVCURLState *s)
458769ce76dSAlexander Graf {
459769ce76dSAlexander Graf     CURLState *state = NULL;
4603ce6a729SPaolo Bonzini     int i;
461769ce76dSAlexander Graf 
462769ce76dSAlexander Graf     for (i = 0; i < CURL_NUM_STATES; i++) {
4633ce6a729SPaolo Bonzini         if (!s->states[i].in_use) {
464769ce76dSAlexander Graf             state = &s->states[i];
465769ce76dSAlexander Graf             state->in_use = 1;
466769ce76dSAlexander Graf             break;
467769ce76dSAlexander Graf         }
468769ce76dSAlexander Graf     }
4693ce6a729SPaolo Bonzini     return state;
4703ce6a729SPaolo Bonzini }
471769ce76dSAlexander Graf 
4723ce6a729SPaolo Bonzini static int curl_init_state(BDRVCURLState *s, CURLState *state)
4733ce6a729SPaolo Bonzini {
4749e550b32SMatthew Booth     if (!state->curl) {
475769ce76dSAlexander Graf         state->curl = curl_easy_init();
4769e550b32SMatthew Booth         if (!state->curl) {
4773ce6a729SPaolo Bonzini             return -EIO;
4789e550b32SMatthew Booth         }
479769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_URL, s->url);
48097a3ea57SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYPEER,
48197a3ea57SMatthew Booth                          (long) s->sslverify);
482a94f83d9SRichard W.M. Jones         if (s->cookie) {
483a94f83d9SRichard W.M. Jones             curl_easy_setopt(state->curl, CURLOPT_COOKIE, s->cookie);
484a94f83d9SRichard W.M. Jones         }
485f76faedaSRichard W.M. Jones         curl_easy_setopt(state->curl, CURLOPT_TIMEOUT, (long)s->timeout);
4869e550b32SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_WRITEFUNCTION,
4879e550b32SMatthew Booth                          (void *)curl_read_cb);
488769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_WRITEDATA, (void *)state);
489769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_PRIVATE, (void *)state);
490769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_AUTOREFERER, 1);
491769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_FOLLOWLOCATION, 1);
492769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_NOSIGNAL, 1);
493769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_ERRORBUFFER, state->errmsg);
494f785a5aeSNicholas Thomas         curl_easy_setopt(state->curl, CURLOPT_FAILONERROR, 1);
495769ce76dSAlexander Graf 
4961bff9606SDaniel P. Berrange         if (s->username) {
4971bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_USERNAME, s->username);
4981bff9606SDaniel P. Berrange         }
4991bff9606SDaniel P. Berrange         if (s->password) {
5001bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_PASSWORD, s->password);
5011bff9606SDaniel P. Berrange         }
5021bff9606SDaniel P. Berrange         if (s->proxyusername) {
5031bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
5041bff9606SDaniel P. Berrange                              CURLOPT_PROXYUSERNAME, s->proxyusername);
5051bff9606SDaniel P. Berrange         }
5061bff9606SDaniel P. Berrange         if (s->proxypassword) {
5071bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
5081bff9606SDaniel P. Berrange                              CURLOPT_PROXYPASSWORD, s->proxypassword);
5091bff9606SDaniel P. Berrange         }
5101bff9606SDaniel P. Berrange 
511fb6d1bbdSStefan Hajnoczi         /* Restrict supported protocols to avoid security issues in the more
512fb6d1bbdSStefan Hajnoczi          * obscure protocols.  For example, do not allow POP3/SMTP/IMAP see
513fb6d1bbdSStefan Hajnoczi          * CVE-2013-0249.
5148a8f5840SStefan Hajnoczi          *
5158a8f5840SStefan Hajnoczi          * Restricting protocols is only supported from 7.19.4 upwards.
516fb6d1bbdSStefan Hajnoczi          */
5178a8f5840SStefan Hajnoczi #if LIBCURL_VERSION_NUM >= 0x071304
518fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_PROTOCOLS, PROTOCOLS);
519fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_REDIR_PROTOCOLS, PROTOCOLS);
5208a8f5840SStefan Hajnoczi #endif
521fb6d1bbdSStefan Hajnoczi 
522769ce76dSAlexander Graf #ifdef DEBUG_VERBOSE
523769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_VERBOSE, 1);
524769ce76dSAlexander Graf #endif
5259e550b32SMatthew Booth     }
526769ce76dSAlexander Graf 
527ff5ca166SMax Reitz     QLIST_INIT(&state->sockets);
528769ce76dSAlexander Graf     state->s = s;
529769ce76dSAlexander Graf 
5303ce6a729SPaolo Bonzini     return 0;
531769ce76dSAlexander Graf }
532769ce76dSAlexander Graf 
533456af346SPaolo Bonzini /* Called with s->mutex held.  */
534769ce76dSAlexander Graf static void curl_clean_state(CURLState *s)
535769ce76dSAlexander Graf {
536675a7756SPaolo Bonzini     int j;
537675a7756SPaolo Bonzini     for (j = 0; j < CURL_NUM_ACB; j++) {
538675a7756SPaolo Bonzini         assert(!s->acb[j]);
539675a7756SPaolo Bonzini     }
540675a7756SPaolo Bonzini 
541769ce76dSAlexander Graf     if (s->s->multi)
542769ce76dSAlexander Graf         curl_multi_remove_handle(s->s->multi, s->curl);
543ff5ca166SMax Reitz 
544ff5ca166SMax Reitz     while (!QLIST_EMPTY(&s->sockets)) {
545ff5ca166SMax Reitz         CURLSocket *socket = QLIST_FIRST(&s->sockets);
546ff5ca166SMax Reitz 
547ff5ca166SMax Reitz         QLIST_REMOVE(socket, next);
548ff5ca166SMax Reitz         g_free(socket);
549ff5ca166SMax Reitz     }
550ff5ca166SMax Reitz 
551769ce76dSAlexander Graf     s->in_use = 0;
552769ce76dSAlexander Graf }
553769ce76dSAlexander Graf 
5548e6d58cdSKevin Wolf static void curl_parse_filename(const char *filename, QDict *options,
5558e6d58cdSKevin Wolf                                 Error **errp)
556769ce76dSAlexander Graf {
55746f5ac20SEric Blake     qdict_put_str(options, CURL_BLOCK_OPT_URL, filename);
5588e6d58cdSKevin Wolf }
5598e6d58cdSKevin Wolf 
56063f0f45fSStefan Hajnoczi static void curl_detach_aio_context(BlockDriverState *bs)
56163f0f45fSStefan Hajnoczi {
56263f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
56363f0f45fSStefan Hajnoczi     int i;
56463f0f45fSStefan Hajnoczi 
565456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
56663f0f45fSStefan Hajnoczi     for (i = 0; i < CURL_NUM_STATES; i++) {
56763f0f45fSStefan Hajnoczi         if (s->states[i].in_use) {
56863f0f45fSStefan Hajnoczi             curl_clean_state(&s->states[i]);
56963f0f45fSStefan Hajnoczi         }
57063f0f45fSStefan Hajnoczi         if (s->states[i].curl) {
57163f0f45fSStefan Hajnoczi             curl_easy_cleanup(s->states[i].curl);
57263f0f45fSStefan Hajnoczi             s->states[i].curl = NULL;
57363f0f45fSStefan Hajnoczi         }
57463f0f45fSStefan Hajnoczi         g_free(s->states[i].orig_buf);
57563f0f45fSStefan Hajnoczi         s->states[i].orig_buf = NULL;
57663f0f45fSStefan Hajnoczi     }
57763f0f45fSStefan Hajnoczi     if (s->multi) {
57863f0f45fSStefan Hajnoczi         curl_multi_cleanup(s->multi);
57963f0f45fSStefan Hajnoczi         s->multi = NULL;
58063f0f45fSStefan Hajnoczi     }
581456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
58263f0f45fSStefan Hajnoczi 
58363f0f45fSStefan Hajnoczi     timer_del(&s->timer);
58463f0f45fSStefan Hajnoczi }
58563f0f45fSStefan Hajnoczi 
58663f0f45fSStefan Hajnoczi static void curl_attach_aio_context(BlockDriverState *bs,
58763f0f45fSStefan Hajnoczi                                     AioContext *new_context)
58863f0f45fSStefan Hajnoczi {
58963f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
59063f0f45fSStefan Hajnoczi 
59163f0f45fSStefan Hajnoczi     aio_timer_init(new_context, &s->timer,
59263f0f45fSStefan Hajnoczi                    QEMU_CLOCK_REALTIME, SCALE_NS,
59363f0f45fSStefan Hajnoczi                    curl_multi_timeout_do, s);
59463f0f45fSStefan Hajnoczi 
59563f0f45fSStefan Hajnoczi     assert(!s->multi);
59663f0f45fSStefan Hajnoczi     s->multi = curl_multi_init();
59763f0f45fSStefan Hajnoczi     s->aio_context = new_context;
59863f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_SOCKETFUNCTION, curl_sock_cb);
59963f0f45fSStefan Hajnoczi #ifdef NEED_CURL_TIMER_CALLBACK
60063f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERDATA, s);
60163f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERFUNCTION, curl_timer_cb);
60263f0f45fSStefan Hajnoczi #endif
60363f0f45fSStefan Hajnoczi }
60463f0f45fSStefan Hajnoczi 
6058e6d58cdSKevin Wolf static QemuOptsList runtime_opts = {
6068e6d58cdSKevin Wolf     .name = "curl",
6078e6d58cdSKevin Wolf     .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
6088e6d58cdSKevin Wolf     .desc = {
6098e6d58cdSKevin Wolf         {
610e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_URL,
6118e6d58cdSKevin Wolf             .type = QEMU_OPT_STRING,
6128e6d58cdSKevin Wolf             .help = "URL to open",
6138e6d58cdSKevin Wolf         },
6148e6d58cdSKevin Wolf         {
615e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_READAHEAD,
6168e6d58cdSKevin Wolf             .type = QEMU_OPT_SIZE,
6178e6d58cdSKevin Wolf             .help = "Readahead size",
6188e6d58cdSKevin Wolf         },
61997a3ea57SMatthew Booth         {
62097a3ea57SMatthew Booth             .name = CURL_BLOCK_OPT_SSLVERIFY,
62197a3ea57SMatthew Booth             .type = QEMU_OPT_BOOL,
62297a3ea57SMatthew Booth             .help = "Verify SSL certificate"
62397a3ea57SMatthew Booth         },
624212aefaaSDaniel Henrique Barboza         {
625212aefaaSDaniel Henrique Barboza             .name = CURL_BLOCK_OPT_TIMEOUT,
626212aefaaSDaniel Henrique Barboza             .type = QEMU_OPT_NUMBER,
627212aefaaSDaniel Henrique Barboza             .help = "Curl timeout"
628212aefaaSDaniel Henrique Barboza         },
629a94f83d9SRichard W.M. Jones         {
630a94f83d9SRichard W.M. Jones             .name = CURL_BLOCK_OPT_COOKIE,
631a94f83d9SRichard W.M. Jones             .type = QEMU_OPT_STRING,
632a94f83d9SRichard W.M. Jones             .help = "Pass the cookie or list of cookies with each request"
633a94f83d9SRichard W.M. Jones         },
6341bff9606SDaniel P. Berrange         {
635327c8ebdSPeter Krempa             .name = CURL_BLOCK_OPT_COOKIE_SECRET,
636327c8ebdSPeter Krempa             .type = QEMU_OPT_STRING,
637327c8ebdSPeter Krempa             .help = "ID of secret used as cookie passed with each request"
638327c8ebdSPeter Krempa         },
639327c8ebdSPeter Krempa         {
6401bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_USERNAME,
6411bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6421bff9606SDaniel P. Berrange             .help = "Username for HTTP auth"
6431bff9606SDaniel P. Berrange         },
6441bff9606SDaniel P. Berrange         {
6451bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PASSWORD_SECRET,
6461bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6471bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP auth",
6481bff9606SDaniel P. Berrange         },
6491bff9606SDaniel P. Berrange         {
6501bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_USERNAME,
6511bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6521bff9606SDaniel P. Berrange             .help = "Username for HTTP proxy auth"
6531bff9606SDaniel P. Berrange         },
6541bff9606SDaniel P. Berrange         {
6551bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET,
6561bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6571bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP proxy auth",
6581bff9606SDaniel P. Berrange         },
6598e6d58cdSKevin Wolf         { /* end of list */ }
6608e6d58cdSKevin Wolf     },
6618e6d58cdSKevin Wolf };
6628e6d58cdSKevin Wolf 
6631bff9606SDaniel P. Berrange 
664015a1036SMax Reitz static int curl_open(BlockDriverState *bs, QDict *options, int flags,
665015a1036SMax Reitz                      Error **errp)
6668e6d58cdSKevin Wolf {
6678e6d58cdSKevin Wolf     BDRVCURLState *s = bs->opaque;
6688e6d58cdSKevin Wolf     CURLState *state = NULL;
6698e6d58cdSKevin Wolf     QemuOpts *opts;
6708e6d58cdSKevin Wolf     Error *local_err = NULL;
6718e6d58cdSKevin Wolf     const char *file;
672a94f83d9SRichard W.M. Jones     const char *cookie;
673327c8ebdSPeter Krempa     const char *cookie_secret;
6748e6d58cdSKevin Wolf     double d;
6751bff9606SDaniel P. Berrange     const char *secretid;
67634634ca2SMax Reitz     const char *protocol_delimiter;
6778e6d58cdSKevin Wolf 
6788e6d58cdSKevin Wolf     static int inited = 0;
6798e6d58cdSKevin Wolf 
680a7cea2baSRichard W.M. Jones     if (flags & BDRV_O_RDWR) {
6812a94fee3SPaolo Bonzini         error_setg(errp, "curl block device does not support writes");
682a7cea2baSRichard W.M. Jones         return -EROFS;
683a7cea2baSRichard W.M. Jones     }
684a7cea2baSRichard W.M. Jones 
685456af346SPaolo Bonzini     qemu_mutex_init(&s->mutex);
68687ea75d5SPeter Crosthwaite     opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
6878e6d58cdSKevin Wolf     qemu_opts_absorb_qdict(opts, options, &local_err);
68884d18f06SMarkus Armbruster     if (local_err) {
6892a94fee3SPaolo Bonzini         error_propagate(errp, local_err);
6908e6d58cdSKevin Wolf         goto out_noclean;
6918e6d58cdSKevin Wolf     }
6928e6d58cdSKevin Wolf 
693e3542c67SMatthew Booth     s->readahead_size = qemu_opt_get_size(opts, CURL_BLOCK_OPT_READAHEAD,
694e3542c67SMatthew Booth                                           READ_AHEAD_DEFAULT);
695c76f4952SNolan     if ((s->readahead_size & 0x1ff) != 0) {
6962a94fee3SPaolo Bonzini         error_setg(errp, "HTTP_READAHEAD_SIZE %zd is not a multiple of 512",
697c76f4952SNolan                    s->readahead_size);
698c76f4952SNolan         goto out_noclean;
699c76f4952SNolan     }
700c76f4952SNolan 
701212aefaaSDaniel Henrique Barboza     s->timeout = qemu_opt_get_number(opts, CURL_BLOCK_OPT_TIMEOUT,
702212aefaaSDaniel Henrique Barboza                                      CURL_TIMEOUT_DEFAULT);
703f76faedaSRichard W.M. Jones     if (s->timeout > CURL_TIMEOUT_MAX) {
704f76faedaSRichard W.M. Jones         error_setg(errp, "timeout parameter is too large or negative");
705f76faedaSRichard W.M. Jones         goto out_noclean;
706f76faedaSRichard W.M. Jones     }
707212aefaaSDaniel Henrique Barboza 
70897a3ea57SMatthew Booth     s->sslverify = qemu_opt_get_bool(opts, CURL_BLOCK_OPT_SSLVERIFY, true);
70997a3ea57SMatthew Booth 
710a94f83d9SRichard W.M. Jones     cookie = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE);
711327c8ebdSPeter Krempa     cookie_secret = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE_SECRET);
712327c8ebdSPeter Krempa 
713327c8ebdSPeter Krempa     if (cookie && cookie_secret) {
714327c8ebdSPeter Krempa         error_setg(errp,
715327c8ebdSPeter Krempa                    "curl driver cannot handle both cookie and cookie secret");
716327c8ebdSPeter Krempa         goto out_noclean;
717327c8ebdSPeter Krempa     }
718327c8ebdSPeter Krempa 
719327c8ebdSPeter Krempa     if (cookie_secret) {
720327c8ebdSPeter Krempa         s->cookie = qcrypto_secret_lookup_as_utf8(cookie_secret, errp);
721327c8ebdSPeter Krempa         if (!s->cookie) {
722327c8ebdSPeter Krempa             goto out_noclean;
723327c8ebdSPeter Krempa         }
724327c8ebdSPeter Krempa     } else {
725a94f83d9SRichard W.M. Jones         s->cookie = g_strdup(cookie);
726327c8ebdSPeter Krempa     }
727a94f83d9SRichard W.M. Jones 
728e3542c67SMatthew Booth     file = qemu_opt_get(opts, CURL_BLOCK_OPT_URL);
7298e6d58cdSKevin Wolf     if (file == NULL) {
7302a94fee3SPaolo Bonzini         error_setg(errp, "curl block driver requires an 'url' option");
7318e6d58cdSKevin Wolf         goto out_noclean;
7328e6d58cdSKevin Wolf     }
7338e6d58cdSKevin Wolf 
73434634ca2SMax Reitz     if (!strstart(file, bs->drv->protocol_name, &protocol_delimiter) ||
73534634ca2SMax Reitz         !strstart(protocol_delimiter, "://", NULL))
73634634ca2SMax Reitz     {
73734634ca2SMax Reitz         error_setg(errp, "%s curl driver cannot handle the URL '%s' (does not "
73834634ca2SMax Reitz                    "start with '%s://')", bs->drv->protocol_name, file,
73934634ca2SMax Reitz                    bs->drv->protocol_name);
74034634ca2SMax Reitz         goto out_noclean;
74134634ca2SMax Reitz     }
74234634ca2SMax Reitz 
7431bff9606SDaniel P. Berrange     s->username = g_strdup(qemu_opt_get(opts, CURL_BLOCK_OPT_USERNAME));
7441bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PASSWORD_SECRET);
7451bff9606SDaniel P. Berrange 
7461bff9606SDaniel P. Berrange     if (secretid) {
7471bff9606SDaniel P. Berrange         s->password = qcrypto_secret_lookup_as_utf8(secretid, errp);
7481bff9606SDaniel P. Berrange         if (!s->password) {
7491bff9606SDaniel P. Berrange             goto out_noclean;
7501bff9606SDaniel P. Berrange         }
7511bff9606SDaniel P. Berrange     }
7521bff9606SDaniel P. Berrange 
7531bff9606SDaniel P. Berrange     s->proxyusername = g_strdup(
7541bff9606SDaniel P. Berrange         qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_USERNAME));
7551bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET);
7561bff9606SDaniel P. Berrange     if (secretid) {
7571bff9606SDaniel P. Berrange         s->proxypassword = qcrypto_secret_lookup_as_utf8(secretid, errp);
7581bff9606SDaniel P. Berrange         if (!s->proxypassword) {
7591bff9606SDaniel P. Berrange             goto out_noclean;
7601bff9606SDaniel P. Berrange         }
7611bff9606SDaniel P. Berrange     }
7621bff9606SDaniel P. Berrange 
763769ce76dSAlexander Graf     if (!inited) {
764769ce76dSAlexander Graf         curl_global_init(CURL_GLOBAL_ALL);
765769ce76dSAlexander Graf         inited = 1;
766769ce76dSAlexander Graf     }
767769ce76dSAlexander Graf 
768d0f2c4c6Smalc     DPRINTF("CURL: Opening %s\n", file);
76963f0f45fSStefan Hajnoczi     s->aio_context = bdrv_get_aio_context(bs);
7708e6d58cdSKevin Wolf     s->url = g_strdup(file);
771456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
7723ce6a729SPaolo Bonzini     state = curl_find_state(s);
773456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
7743ce6a729SPaolo Bonzini     if (!state) {
775769ce76dSAlexander Graf         goto out_noclean;
7763ce6a729SPaolo Bonzini     }
777769ce76dSAlexander Graf 
778769ce76dSAlexander Graf     // Get file size
779769ce76dSAlexander Graf 
7803ce6a729SPaolo Bonzini     if (curl_init_state(s, state) < 0) {
7813ce6a729SPaolo Bonzini         goto out;
7823ce6a729SPaolo Bonzini     }
7833ce6a729SPaolo Bonzini 
7843494d650SFam Zheng     s->accept_range = false;
785769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_NOBODY, 1);
7863494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERFUNCTION,
7873494d650SFam Zheng                      curl_header_cb);
7883494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERDATA, s);
789769ce76dSAlexander Graf     if (curl_easy_perform(state->curl))
790769ce76dSAlexander Graf         goto out;
791a41c4578STomáš Golembiovský     if (curl_easy_getinfo(state->curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d)) {
792769ce76dSAlexander Graf         goto out;
793a41c4578STomáš Golembiovský     }
794a41c4578STomáš Golembiovský     /* Prior CURL 7.19.4 return value of 0 could mean that the file size is not
795a41c4578STomáš Golembiovský      * know or the size is zero. From 7.19.4 CURL returns -1 if size is not
796a41c4578STomáš Golembiovský      * known and zero if it is realy zero-length file. */
797a41c4578STomáš Golembiovský #if LIBCURL_VERSION_NUM >= 0x071304
798a41c4578STomáš Golembiovský     if (d < 0) {
799a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
800a41c4578STomáš Golembiovský                 "Server didn't report file size.");
801a41c4578STomáš Golembiovský         goto out;
802a41c4578STomáš Golembiovský     }
803a41c4578STomáš Golembiovský #else
804a41c4578STomáš Golembiovský     if (d <= 0) {
805a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
806a41c4578STomáš Golembiovský                 "Unknown file size or zero-length file.");
807a41c4578STomáš Golembiovský         goto out;
808a41c4578STomáš Golembiovský     }
809a41c4578STomáš Golembiovský #endif
810a41c4578STomáš Golembiovský 
8112125e5eaSPaolo Bonzini     s->len = d;
812a41c4578STomáš Golembiovský 
8133494d650SFam Zheng     if ((!strncasecmp(s->url, "http://", strlen("http://"))
8143494d650SFam Zheng         || !strncasecmp(s->url, "https://", strlen("https://")))
8153494d650SFam Zheng         && !s->accept_range) {
8163494d650SFam Zheng         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
8173494d650SFam Zheng                 "Server does not support 'range' (byte ranges).");
8183494d650SFam Zheng         goto out;
8193494d650SFam Zheng     }
8202125e5eaSPaolo Bonzini     DPRINTF("CURL: Size = %" PRIu64 "\n", s->len);
821769ce76dSAlexander Graf 
822456af346SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
823769ce76dSAlexander Graf     curl_clean_state(state);
824456af346SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
825769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
826769ce76dSAlexander Graf     state->curl = NULL;
827769ce76dSAlexander Graf 
82863f0f45fSStefan Hajnoczi     curl_attach_aio_context(bs, bdrv_get_aio_context(bs));
829769ce76dSAlexander Graf 
8308e6d58cdSKevin Wolf     qemu_opts_del(opts);
831769ce76dSAlexander Graf     return 0;
832769ce76dSAlexander Graf 
833769ce76dSAlexander Graf out:
834acd7fdc6SMaria Kustova     error_setg(errp, "CURL: Error opening file: %s", state->errmsg);
835769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
836769ce76dSAlexander Graf     state->curl = NULL;
837769ce76dSAlexander Graf out_noclean:
838456af346SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
839a94f83d9SRichard W.M. Jones     g_free(s->cookie);
8408e6d58cdSKevin Wolf     g_free(s->url);
8418e6d58cdSKevin Wolf     qemu_opts_del(opts);
842769ce76dSAlexander Graf     return -EINVAL;
843769ce76dSAlexander Graf }
844769ce76dSAlexander Graf 
845*28256d82SPaolo Bonzini static void curl_setup_preadv(BlockDriverState *bs, CURLAIOCB *acb)
846769ce76dSAlexander Graf {
847769ce76dSAlexander Graf     CURLState *state;
848b69cdef8SMatthew Booth     int running;
849769ce76dSAlexander Graf 
8501919631eSPaolo Bonzini     BDRVCURLState *s = bs->opaque;
851769ce76dSAlexander Graf 
8522125e5eaSPaolo Bonzini     uint64_t start = acb->offset;
8532125e5eaSPaolo Bonzini     uint64_t end;
854769ce76dSAlexander Graf 
855ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
8561919631eSPaolo Bonzini 
857769ce76dSAlexander Graf     // In case we have the requested data already (e.g. read-ahead),
858769ce76dSAlexander Graf     // we can just call the callback and be done.
859*28256d82SPaolo Bonzini     if (curl_find_buf(s, start, acb->bytes, acb)) {
86034db05e7SPaolo Bonzini         goto out;
861769ce76dSAlexander Graf     }
862769ce76dSAlexander Graf 
863769ce76dSAlexander Graf     // No cache found, so let's start a new request
8643ce6a729SPaolo Bonzini     for (;;) {
8653ce6a729SPaolo Bonzini         state = curl_find_state(s);
8663ce6a729SPaolo Bonzini         if (state) {
8673ce6a729SPaolo Bonzini             break;
8683ce6a729SPaolo Bonzini         }
8693ce6a729SPaolo Bonzini         qemu_mutex_unlock(&s->mutex);
8703ce6a729SPaolo Bonzini         aio_poll(bdrv_get_aio_context(bs), true);
8713ce6a729SPaolo Bonzini         qemu_mutex_lock(&s->mutex);
8723ce6a729SPaolo Bonzini     }
8733ce6a729SPaolo Bonzini 
8743ce6a729SPaolo Bonzini     if (curl_init_state(s, state) < 0) {
8753ce6a729SPaolo Bonzini         curl_clean_state(state);
876*28256d82SPaolo Bonzini         acb->ret = -EIO;
8771919631eSPaolo Bonzini         goto out;
878363c3c85SNick Thomas     }
879769ce76dSAlexander Graf 
880769ce76dSAlexander Graf     acb->start = 0;
8812125e5eaSPaolo Bonzini     acb->end = MIN(acb->bytes, s->len - start);
882769ce76dSAlexander Graf 
883769ce76dSAlexander Graf     state->buf_off = 0;
8847267c094SAnthony Liguori     g_free(state->orig_buf);
885769ce76dSAlexander Graf     state->buf_start = start;
8864e504535SMax Reitz     state->buf_len = MIN(acb->end + s->readahead_size, s->len - start);
8874e504535SMax Reitz     end = start + state->buf_len - 1;
8888dc7a772SKevin Wolf     state->orig_buf = g_try_malloc(state->buf_len);
8898dc7a772SKevin Wolf     if (state->buf_len && state->orig_buf == NULL) {
8908dc7a772SKevin Wolf         curl_clean_state(state);
891*28256d82SPaolo Bonzini         acb->ret = -ENOMEM;
8921919631eSPaolo Bonzini         goto out;
8938dc7a772SKevin Wolf     }
894769ce76dSAlexander Graf     state->acb[0] = acb;
895769ce76dSAlexander Graf 
8962125e5eaSPaolo Bonzini     snprintf(state->range, 127, "%" PRIu64 "-%" PRIu64, start, end);
8972125e5eaSPaolo Bonzini     DPRINTF("CURL (AIO): Reading %" PRIu64 " at %" PRIu64 " (%s)\n",
8982125e5eaSPaolo Bonzini             acb->bytes, start, state->range);
899769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_RANGE, state->range);
900769ce76dSAlexander Graf 
901769ce76dSAlexander Graf     curl_multi_add_handle(s->multi, state->curl);
902769ce76dSAlexander Graf 
903b69cdef8SMatthew Booth     /* Tell curl it needs to kick things off */
904b69cdef8SMatthew Booth     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
9051919631eSPaolo Bonzini 
9061919631eSPaolo Bonzini out:
907ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
908363c3c85SNick Thomas }
909363c3c85SNick Thomas 
910*28256d82SPaolo Bonzini static int coroutine_fn curl_co_preadv(BlockDriverState *bs,
911*28256d82SPaolo Bonzini         uint64_t offset, uint64_t bytes, QEMUIOVector *qiov, int flags)
912363c3c85SNick Thomas {
913*28256d82SPaolo Bonzini     CURLAIOCB acb = {
914*28256d82SPaolo Bonzini         .co = qemu_coroutine_self(),
915*28256d82SPaolo Bonzini         .ret = -EINPROGRESS,
916*28256d82SPaolo Bonzini         .qiov = qiov,
917*28256d82SPaolo Bonzini         .offset = offset,
918*28256d82SPaolo Bonzini         .bytes = bytes
919*28256d82SPaolo Bonzini     };
920363c3c85SNick Thomas 
921*28256d82SPaolo Bonzini     curl_setup_preadv(bs, &acb);
922*28256d82SPaolo Bonzini     while (acb.ret == -EINPROGRESS) {
923*28256d82SPaolo Bonzini         qemu_coroutine_yield();
924*28256d82SPaolo Bonzini     }
925*28256d82SPaolo Bonzini     return acb.ret;
926769ce76dSAlexander Graf }
927769ce76dSAlexander Graf 
928769ce76dSAlexander Graf static void curl_close(BlockDriverState *bs)
929769ce76dSAlexander Graf {
930769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
931769ce76dSAlexander Graf 
932d0f2c4c6Smalc     DPRINTF("CURL: Close\n");
93363f0f45fSStefan Hajnoczi     curl_detach_aio_context(bs);
934ba3186c4SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
935031fd1beSPeter Maydell 
936a94f83d9SRichard W.M. Jones     g_free(s->cookie);
93745724d6dSStefan Weil     g_free(s->url);
938769ce76dSAlexander Graf }
939769ce76dSAlexander Graf 
940769ce76dSAlexander Graf static int64_t curl_getlength(BlockDriverState *bs)
941769ce76dSAlexander Graf {
942769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
943769ce76dSAlexander Graf     return s->len;
944769ce76dSAlexander Graf }
945769ce76dSAlexander Graf 
946769ce76dSAlexander Graf static BlockDriver bdrv_http = {
947769ce76dSAlexander Graf     .format_name                = "http",
948769ce76dSAlexander Graf     .protocol_name              = "http",
949769ce76dSAlexander Graf 
950769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9518e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
95266f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
953769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
954769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
955769ce76dSAlexander Graf 
956*28256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
95763f0f45fSStefan Hajnoczi 
95863f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
95963f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
960769ce76dSAlexander Graf };
961769ce76dSAlexander Graf 
962769ce76dSAlexander Graf static BlockDriver bdrv_https = {
963769ce76dSAlexander Graf     .format_name                = "https",
964769ce76dSAlexander Graf     .protocol_name              = "https",
965769ce76dSAlexander Graf 
966769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9678e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
96866f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
969769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
970769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
971769ce76dSAlexander Graf 
972*28256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
97363f0f45fSStefan Hajnoczi 
97463f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
97563f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
976769ce76dSAlexander Graf };
977769ce76dSAlexander Graf 
978769ce76dSAlexander Graf static BlockDriver bdrv_ftp = {
979769ce76dSAlexander Graf     .format_name                = "ftp",
980769ce76dSAlexander Graf     .protocol_name              = "ftp",
981769ce76dSAlexander Graf 
982769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9838e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
98466f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
985769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
986769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
987769ce76dSAlexander Graf 
988*28256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
98963f0f45fSStefan Hajnoczi 
99063f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
99163f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
992769ce76dSAlexander Graf };
993769ce76dSAlexander Graf 
994769ce76dSAlexander Graf static BlockDriver bdrv_ftps = {
995769ce76dSAlexander Graf     .format_name                = "ftps",
996769ce76dSAlexander Graf     .protocol_name              = "ftps",
997769ce76dSAlexander Graf 
998769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9998e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
100066f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
1001769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
1002769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
1003769ce76dSAlexander Graf 
1004*28256d82SPaolo Bonzini     .bdrv_co_preadv             = curl_co_preadv,
100563f0f45fSStefan Hajnoczi 
100663f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
100763f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
1008769ce76dSAlexander Graf };
1009769ce76dSAlexander Graf 
1010769ce76dSAlexander Graf static void curl_block_init(void)
1011769ce76dSAlexander Graf {
1012769ce76dSAlexander Graf     bdrv_register(&bdrv_http);
1013769ce76dSAlexander Graf     bdrv_register(&bdrv_https);
1014769ce76dSAlexander Graf     bdrv_register(&bdrv_ftp);
1015769ce76dSAlexander Graf     bdrv_register(&bdrv_ftps);
1016769ce76dSAlexander Graf }
1017769ce76dSAlexander Graf 
1018769ce76dSAlexander Graf block_init(curl_block_init);
1019