xref: /openbmc/qemu/block/curl.c (revision 327c8ebd7035e4d3d94b08dd741906f1d8bb8a53)
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 
79769ce76dSAlexander Graf #define FIND_RET_NONE   0
80769ce76dSAlexander Graf #define FIND_RET_OK     1
81769ce76dSAlexander Graf #define FIND_RET_WAIT   2
82769ce76dSAlexander Graf 
83e3542c67SMatthew Booth #define CURL_BLOCK_OPT_URL       "url"
84e3542c67SMatthew Booth #define CURL_BLOCK_OPT_READAHEAD "readahead"
8597a3ea57SMatthew Booth #define CURL_BLOCK_OPT_SSLVERIFY "sslverify"
86212aefaaSDaniel Henrique Barboza #define CURL_BLOCK_OPT_TIMEOUT "timeout"
87a94f83d9SRichard W.M. Jones #define CURL_BLOCK_OPT_COOKIE    "cookie"
88*327c8ebdSPeter Krempa #define CURL_BLOCK_OPT_COOKIE_SECRET "cookie-secret"
891bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_USERNAME "username"
901bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PASSWORD_SECRET "password-secret"
911bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_USERNAME "proxy-username"
921bff9606SDaniel P. Berrange #define CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET "proxy-password-secret"
93e3542c67SMatthew Booth 
94769ce76dSAlexander Graf struct BDRVCURLState;
95769ce76dSAlexander Graf 
96769ce76dSAlexander Graf typedef struct CURLAIOCB {
977c84b1b8SMarkus Armbruster     BlockAIOCB common;
98769ce76dSAlexander Graf     QEMUIOVector *qiov;
99363c3c85SNick Thomas 
100363c3c85SNick Thomas     int64_t sector_num;
101363c3c85SNick Thomas     int nb_sectors;
102363c3c85SNick Thomas 
103769ce76dSAlexander Graf     size_t start;
104769ce76dSAlexander Graf     size_t end;
105769ce76dSAlexander Graf } CURLAIOCB;
106769ce76dSAlexander Graf 
107ff5ca166SMax Reitz typedef struct CURLSocket {
108ff5ca166SMax Reitz     int fd;
109ff5ca166SMax Reitz     QLIST_ENTRY(CURLSocket) next;
110ff5ca166SMax Reitz } CURLSocket;
111ff5ca166SMax Reitz 
112769ce76dSAlexander Graf typedef struct CURLState
113769ce76dSAlexander Graf {
114769ce76dSAlexander Graf     struct BDRVCURLState *s;
115769ce76dSAlexander Graf     CURLAIOCB *acb[CURL_NUM_ACB];
116769ce76dSAlexander Graf     CURL *curl;
117ff5ca166SMax Reitz     QLIST_HEAD(, CURLSocket) sockets;
118769ce76dSAlexander Graf     char *orig_buf;
119769ce76dSAlexander Graf     size_t buf_start;
120769ce76dSAlexander Graf     size_t buf_off;
121769ce76dSAlexander Graf     size_t buf_len;
122769ce76dSAlexander Graf     char range[128];
123769ce76dSAlexander Graf     char errmsg[CURL_ERROR_SIZE];
124769ce76dSAlexander Graf     char in_use;
125769ce76dSAlexander Graf } CURLState;
126769ce76dSAlexander Graf 
127769ce76dSAlexander Graf typedef struct BDRVCURLState {
128769ce76dSAlexander Graf     CURLM *multi;
129031fd1beSPeter Maydell     QEMUTimer timer;
130769ce76dSAlexander Graf     size_t len;
131769ce76dSAlexander Graf     CURLState states[CURL_NUM_STATES];
132769ce76dSAlexander Graf     char *url;
133c76f4952SNolan     size_t readahead_size;
13497a3ea57SMatthew Booth     bool sslverify;
135f76faedaSRichard W.M. Jones     uint64_t timeout;
136a94f83d9SRichard W.M. Jones     char *cookie;
1373494d650SFam Zheng     bool accept_range;
13863f0f45fSStefan Hajnoczi     AioContext *aio_context;
139ba3186c4SPaolo Bonzini     QemuMutex mutex;
1401bff9606SDaniel P. Berrange     char *username;
1411bff9606SDaniel P. Berrange     char *password;
1421bff9606SDaniel P. Berrange     char *proxyusername;
1431bff9606SDaniel P. Berrange     char *proxypassword;
144769ce76dSAlexander Graf } BDRVCURLState;
145769ce76dSAlexander Graf 
146769ce76dSAlexander Graf static void curl_clean_state(CURLState *s);
147769ce76dSAlexander Graf static void curl_multi_do(void *arg);
148838ef602SMatthew Booth static void curl_multi_read(void *arg);
149769ce76dSAlexander Graf 
150031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
151031fd1beSPeter Maydell static int curl_timer_cb(CURLM *multi, long timeout_ms, void *opaque)
152031fd1beSPeter Maydell {
153031fd1beSPeter Maydell     BDRVCURLState *s = opaque;
154031fd1beSPeter Maydell 
155031fd1beSPeter Maydell     DPRINTF("CURL: timer callback timeout_ms %ld\n", timeout_ms);
156031fd1beSPeter Maydell     if (timeout_ms == -1) {
157031fd1beSPeter Maydell         timer_del(&s->timer);
158031fd1beSPeter Maydell     } else {
159031fd1beSPeter Maydell         int64_t timeout_ns = (int64_t)timeout_ms * 1000 * 1000;
160031fd1beSPeter Maydell         timer_mod(&s->timer,
161031fd1beSPeter Maydell                   qemu_clock_get_ns(QEMU_CLOCK_REALTIME) + timeout_ns);
162031fd1beSPeter Maydell     }
163031fd1beSPeter Maydell     return 0;
164031fd1beSPeter Maydell }
165031fd1beSPeter Maydell #endif
166031fd1beSPeter Maydell 
167769ce76dSAlexander Graf static int curl_sock_cb(CURL *curl, curl_socket_t fd, int action,
16863f0f45fSStefan Hajnoczi                         void *userp, void *sp)
169769ce76dSAlexander Graf {
17063f0f45fSStefan Hajnoczi     BDRVCURLState *s;
171838ef602SMatthew Booth     CURLState *state = NULL;
172ff5ca166SMax Reitz     CURLSocket *socket;
173ff5ca166SMax Reitz 
174838ef602SMatthew Booth     curl_easy_getinfo(curl, CURLINFO_PRIVATE, (char **)&state);
17563f0f45fSStefan Hajnoczi     s = state->s;
176838ef602SMatthew Booth 
177ff5ca166SMax Reitz     QLIST_FOREACH(socket, &state->sockets, next) {
178ff5ca166SMax Reitz         if (socket->fd == fd) {
179ff5ca166SMax Reitz             if (action == CURL_POLL_REMOVE) {
180ff5ca166SMax Reitz                 QLIST_REMOVE(socket, next);
181ff5ca166SMax Reitz                 g_free(socket);
182ff5ca166SMax Reitz             }
183ff5ca166SMax Reitz             break;
184ff5ca166SMax Reitz         }
185ff5ca166SMax Reitz     }
186ff5ca166SMax Reitz     if (!socket) {
187ff5ca166SMax Reitz         socket = g_new0(CURLSocket, 1);
188ff5ca166SMax Reitz         socket->fd = fd;
189ff5ca166SMax Reitz         QLIST_INSERT_HEAD(&state->sockets, socket, next);
190ff5ca166SMax Reitz     }
191ff5ca166SMax Reitz     socket = NULL;
192ff5ca166SMax Reitz 
19392b6a160SFam Zheng     DPRINTF("CURL (AIO): Sock action %d on fd %d\n", action, (int)fd);
194769ce76dSAlexander Graf     switch (action) {
195769ce76dSAlexander Graf         case CURL_POLL_IN:
196dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
197f6a51c84SStefan Hajnoczi                                curl_multi_read, NULL, NULL, state);
198769ce76dSAlexander Graf             break;
199769ce76dSAlexander Graf         case CURL_POLL_OUT:
200dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
201f6a51c84SStefan Hajnoczi                                NULL, curl_multi_do, NULL, state);
202769ce76dSAlexander Graf             break;
203769ce76dSAlexander Graf         case CURL_POLL_INOUT:
204dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
205f6a51c84SStefan Hajnoczi                                curl_multi_read, curl_multi_do, NULL, state);
206769ce76dSAlexander Graf             break;
207769ce76dSAlexander Graf         case CURL_POLL_REMOVE:
208dca21ef2SFam Zheng             aio_set_fd_handler(s->aio_context, fd, false,
209f6a51c84SStefan Hajnoczi                                NULL, NULL, NULL, NULL);
210769ce76dSAlexander Graf             break;
211769ce76dSAlexander Graf     }
212769ce76dSAlexander Graf 
213769ce76dSAlexander Graf     return 0;
214769ce76dSAlexander Graf }
215769ce76dSAlexander Graf 
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 
230769ce76dSAlexander Graf static size_t curl_read_cb(void *ptr, size_t size, size_t nmemb, void *opaque)
231769ce76dSAlexander Graf {
232769ce76dSAlexander Graf     CURLState *s = ((CURLState*)opaque);
233769ce76dSAlexander Graf     size_t realsize = size * nmemb;
234769ce76dSAlexander Graf     int i;
235769ce76dSAlexander Graf 
2360bfcd599SBlue Swirl     DPRINTF("CURL: Just reading %zd bytes\n", realsize);
237769ce76dSAlexander Graf 
2384e767657SMax Reitz     if (!s || !s->orig_buf) {
2394e767657SMax Reitz         goto read_end;
2404e767657SMax Reitz     }
241769ce76dSAlexander Graf 
2426d4b9e55SFam Zheng     if (s->buf_off >= s->buf_len) {
2436d4b9e55SFam Zheng         /* buffer full, read nothing */
2444e767657SMax Reitz         goto read_end;
2456d4b9e55SFam Zheng     }
2466d4b9e55SFam Zheng     realsize = MIN(realsize, s->buf_len - s->buf_off);
247769ce76dSAlexander Graf     memcpy(s->orig_buf + s->buf_off, ptr, realsize);
248769ce76dSAlexander Graf     s->buf_off += realsize;
249769ce76dSAlexander Graf 
250769ce76dSAlexander Graf     for(i=0; i<CURL_NUM_ACB; i++) {
251769ce76dSAlexander Graf         CURLAIOCB *acb = s->acb[i];
252769ce76dSAlexander Graf 
253769ce76dSAlexander Graf         if (!acb)
254769ce76dSAlexander Graf             continue;
255769ce76dSAlexander Graf 
256769ce76dSAlexander Graf         if ((s->buf_off >= acb->end)) {
2574e504535SMax Reitz             size_t request_length = acb->nb_sectors * BDRV_SECTOR_SIZE;
2584e504535SMax Reitz 
25903396148SMichael Tokarev             qemu_iovec_from_buf(acb->qiov, 0, s->orig_buf + acb->start,
260769ce76dSAlexander Graf                                 acb->end - acb->start);
2614e504535SMax Reitz 
2624e504535SMax Reitz             if (acb->end - acb->start < request_length) {
2634e504535SMax Reitz                 size_t offset = acb->end - acb->start;
2644e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, offset, 0,
2654e504535SMax Reitz                                   request_length - offset);
2664e504535SMax Reitz             }
2674e504535SMax Reitz 
268769ce76dSAlexander Graf             acb->common.cb(acb->common.opaque, 0);
2698007429aSFam Zheng             qemu_aio_unref(acb);
270769ce76dSAlexander Graf             s->acb[i] = NULL;
271769ce76dSAlexander Graf         }
272769ce76dSAlexander Graf     }
273769ce76dSAlexander Graf 
2744e767657SMax Reitz read_end:
2754e767657SMax Reitz     /* curl will error out if we do not return this value */
2764e767657SMax Reitz     return size * nmemb;
277769ce76dSAlexander Graf }
278769ce76dSAlexander Graf 
279769ce76dSAlexander Graf static int curl_find_buf(BDRVCURLState *s, size_t start, size_t len,
280769ce76dSAlexander Graf                          CURLAIOCB *acb)
281769ce76dSAlexander Graf {
282769ce76dSAlexander Graf     int i;
283769ce76dSAlexander Graf     size_t end = start + len;
2844e504535SMax Reitz     size_t clamped_end = MIN(end, s->len);
2854e504535SMax Reitz     size_t clamped_len = clamped_end - start;
286769ce76dSAlexander Graf 
287769ce76dSAlexander Graf     for (i=0; i<CURL_NUM_STATES; i++) {
288769ce76dSAlexander Graf         CURLState *state = &s->states[i];
289769ce76dSAlexander Graf         size_t buf_end = (state->buf_start + state->buf_off);
290769ce76dSAlexander Graf         size_t buf_fend = (state->buf_start + state->buf_len);
291769ce76dSAlexander Graf 
292769ce76dSAlexander Graf         if (!state->orig_buf)
293769ce76dSAlexander Graf             continue;
294769ce76dSAlexander Graf         if (!state->buf_off)
295769ce76dSAlexander Graf             continue;
296769ce76dSAlexander Graf 
297769ce76dSAlexander Graf         // Does the existing buffer cover our section?
298769ce76dSAlexander Graf         if ((start >= state->buf_start) &&
299769ce76dSAlexander Graf             (start <= buf_end) &&
3004e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3014e504535SMax Reitz             (clamped_end <= buf_end))
302769ce76dSAlexander Graf         {
303769ce76dSAlexander Graf             char *buf = state->orig_buf + (start - state->buf_start);
304769ce76dSAlexander Graf 
3054e504535SMax Reitz             qemu_iovec_from_buf(acb->qiov, 0, buf, clamped_len);
3064e504535SMax Reitz             if (clamped_len < len) {
3074e504535SMax Reitz                 qemu_iovec_memset(acb->qiov, clamped_len, 0, len - clamped_len);
3084e504535SMax Reitz             }
309769ce76dSAlexander Graf             acb->common.cb(acb->common.opaque, 0);
310769ce76dSAlexander Graf 
311769ce76dSAlexander Graf             return FIND_RET_OK;
312769ce76dSAlexander Graf         }
313769ce76dSAlexander Graf 
314769ce76dSAlexander Graf         // Wait for unfinished chunks
315b7079df4SMatthew Booth         if (state->in_use &&
316b7079df4SMatthew Booth             (start >= state->buf_start) &&
317769ce76dSAlexander Graf             (start <= buf_fend) &&
3184e504535SMax Reitz             (clamped_end >= state->buf_start) &&
3194e504535SMax Reitz             (clamped_end <= buf_fend))
320769ce76dSAlexander Graf         {
321769ce76dSAlexander Graf             int j;
322769ce76dSAlexander Graf 
323769ce76dSAlexander Graf             acb->start = start - state->buf_start;
3244e504535SMax Reitz             acb->end = acb->start + clamped_len;
325769ce76dSAlexander Graf 
326769ce76dSAlexander Graf             for (j=0; j<CURL_NUM_ACB; j++) {
327769ce76dSAlexander Graf                 if (!state->acb[j]) {
328769ce76dSAlexander Graf                     state->acb[j] = acb;
329769ce76dSAlexander Graf                     return FIND_RET_WAIT;
330769ce76dSAlexander Graf                 }
331769ce76dSAlexander Graf             }
332769ce76dSAlexander Graf         }
333769ce76dSAlexander Graf     }
334769ce76dSAlexander Graf 
335769ce76dSAlexander Graf     return FIND_RET_NONE;
336769ce76dSAlexander Graf }
337769ce76dSAlexander Graf 
338ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
339838ef602SMatthew Booth static void curl_multi_check_completion(BDRVCURLState *s)
340769ce76dSAlexander Graf {
341769ce76dSAlexander Graf     int msgs_in_queue;
342769ce76dSAlexander Graf 
343769ce76dSAlexander Graf     /* Try to find done transfers, so we can free the easy
344769ce76dSAlexander Graf      * handle again. */
3451f2cead3SMatthew Booth     for (;;) {
346769ce76dSAlexander Graf         CURLMsg *msg;
347769ce76dSAlexander Graf         msg = curl_multi_info_read(s->multi, &msgs_in_queue);
348769ce76dSAlexander Graf 
3491f2cead3SMatthew Booth         /* Quit when there are no more completions */
350769ce76dSAlexander Graf         if (!msg)
351769ce76dSAlexander Graf             break;
352769ce76dSAlexander Graf 
3531f2cead3SMatthew Booth         if (msg->msg == CURLMSG_DONE) {
354769ce76dSAlexander Graf             CURLState *state = NULL;
355f6246509SMatthew Booth             curl_easy_getinfo(msg->easy_handle, CURLINFO_PRIVATE,
356f6246509SMatthew Booth                               (char **)&state);
357f785a5aeSNicholas Thomas 
358f785a5aeSNicholas Thomas             /* ACBs for successful messages get completed in curl_read_cb */
359f785a5aeSNicholas Thomas             if (msg->data.result != CURLE_OK) {
360f785a5aeSNicholas Thomas                 int i;
361796a060bSRichard W.M. Jones                 static int errcount = 100;
362796a060bSRichard W.M. Jones 
363796a060bSRichard W.M. Jones                 /* Don't lose the original error message from curl, since
364796a060bSRichard W.M. Jones                  * it contains extra data.
365796a060bSRichard W.M. Jones                  */
366796a060bSRichard W.M. Jones                 if (errcount > 0) {
367796a060bSRichard W.M. Jones                     error_report("curl: %s", state->errmsg);
368796a060bSRichard W.M. Jones                     if (--errcount == 0) {
369796a060bSRichard W.M. Jones                         error_report("curl: further errors suppressed");
370796a060bSRichard W.M. Jones                     }
371796a060bSRichard W.M. Jones                 }
372796a060bSRichard W.M. Jones 
373f785a5aeSNicholas Thomas                 for (i = 0; i < CURL_NUM_ACB; i++) {
374f785a5aeSNicholas Thomas                     CURLAIOCB *acb = state->acb[i];
375f785a5aeSNicholas Thomas 
376f785a5aeSNicholas Thomas                     if (acb == NULL) {
377f785a5aeSNicholas Thomas                         continue;
378f785a5aeSNicholas Thomas                     }
379f785a5aeSNicholas Thomas 
380ba3186c4SPaolo Bonzini                     qemu_mutex_unlock(&s->mutex);
381eb048026SPaolo Bonzini                     acb->common.cb(acb->common.opaque, -EIO);
382ba3186c4SPaolo Bonzini                     qemu_mutex_lock(&s->mutex);
3838007429aSFam Zheng                     qemu_aio_unref(acb);
384f785a5aeSNicholas Thomas                     state->acb[i] = NULL;
385f785a5aeSNicholas Thomas                 }
386f785a5aeSNicholas Thomas             }
387f785a5aeSNicholas Thomas 
388769ce76dSAlexander Graf             curl_clean_state(state);
389769ce76dSAlexander Graf             break;
390769ce76dSAlexander Graf         }
391769ce76dSAlexander Graf     }
392769ce76dSAlexander Graf }
393769ce76dSAlexander Graf 
394ba3186c4SPaolo Bonzini /* Called with s->mutex held.  */
3959d456654SPaolo Bonzini static void curl_multi_do_locked(CURLState *s)
396031fd1beSPeter Maydell {
397ff5ca166SMax Reitz     CURLSocket *socket, *next_socket;
398031fd1beSPeter Maydell     int running;
399031fd1beSPeter Maydell     int r;
400031fd1beSPeter Maydell 
401838ef602SMatthew Booth     if (!s->s->multi) {
402031fd1beSPeter Maydell         return;
403031fd1beSPeter Maydell     }
404031fd1beSPeter Maydell 
405ff5ca166SMax Reitz     /* Need to use _SAFE because curl_multi_socket_action() may trigger
406ff5ca166SMax Reitz      * curl_sock_cb() which might modify this list */
407ff5ca166SMax Reitz     QLIST_FOREACH_SAFE(socket, &s->sockets, next, next_socket) {
408031fd1beSPeter Maydell         do {
409ff5ca166SMax Reitz             r = curl_multi_socket_action(s->s->multi, socket->fd, 0, &running);
410031fd1beSPeter Maydell         } while (r == CURLM_CALL_MULTI_PERFORM);
411ff5ca166SMax Reitz     }
412838ef602SMatthew Booth }
413838ef602SMatthew Booth 
4149d456654SPaolo Bonzini static void curl_multi_do(void *arg)
4159d456654SPaolo Bonzini {
4169d456654SPaolo Bonzini     CURLState *s = (CURLState *)arg;
4179d456654SPaolo Bonzini 
418ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->s->mutex);
4199d456654SPaolo Bonzini     curl_multi_do_locked(s);
420ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->s->mutex);
4219d456654SPaolo Bonzini }
4229d456654SPaolo Bonzini 
423838ef602SMatthew Booth static void curl_multi_read(void *arg)
424838ef602SMatthew Booth {
425838ef602SMatthew Booth     CURLState *s = (CURLState *)arg;
426838ef602SMatthew Booth 
427ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->s->mutex);
4289d456654SPaolo Bonzini     curl_multi_do_locked(s);
429838ef602SMatthew Booth     curl_multi_check_completion(s->s);
430ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->s->mutex);
431031fd1beSPeter Maydell }
432031fd1beSPeter Maydell 
433031fd1beSPeter Maydell static void curl_multi_timeout_do(void *arg)
434031fd1beSPeter Maydell {
435031fd1beSPeter Maydell #ifdef NEED_CURL_TIMER_CALLBACK
436031fd1beSPeter Maydell     BDRVCURLState *s = (BDRVCURLState *)arg;
437031fd1beSPeter Maydell     int running;
438031fd1beSPeter Maydell 
439031fd1beSPeter Maydell     if (!s->multi) {
440031fd1beSPeter Maydell         return;
441031fd1beSPeter Maydell     }
442031fd1beSPeter Maydell 
443ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
444031fd1beSPeter Maydell     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
445031fd1beSPeter Maydell 
446838ef602SMatthew Booth     curl_multi_check_completion(s);
447ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
448031fd1beSPeter Maydell #else
449031fd1beSPeter Maydell     abort();
450031fd1beSPeter Maydell #endif
451031fd1beSPeter Maydell }
452031fd1beSPeter Maydell 
453a2f468e4SRichard W.M. Jones static CURLState *curl_init_state(BlockDriverState *bs, BDRVCURLState *s)
454769ce76dSAlexander Graf {
455769ce76dSAlexander Graf     CURLState *state = NULL;
456769ce76dSAlexander Graf     int i, j;
457769ce76dSAlexander Graf 
458769ce76dSAlexander Graf     do {
459769ce76dSAlexander Graf         for (i=0; i<CURL_NUM_STATES; i++) {
460769ce76dSAlexander Graf             for (j=0; j<CURL_NUM_ACB; j++)
461769ce76dSAlexander Graf                 if (s->states[i].acb[j])
462769ce76dSAlexander Graf                     continue;
463769ce76dSAlexander Graf             if (s->states[i].in_use)
464769ce76dSAlexander Graf                 continue;
465769ce76dSAlexander Graf 
466769ce76dSAlexander Graf             state = &s->states[i];
467769ce76dSAlexander Graf             state->in_use = 1;
468769ce76dSAlexander Graf             break;
469769ce76dSAlexander Graf         }
470769ce76dSAlexander Graf         if (!state) {
471a2f468e4SRichard W.M. Jones             aio_poll(bdrv_get_aio_context(bs), true);
472769ce76dSAlexander Graf         }
473769ce76dSAlexander Graf     } while(!state);
474769ce76dSAlexander Graf 
4759e550b32SMatthew Booth     if (!state->curl) {
476769ce76dSAlexander Graf         state->curl = curl_easy_init();
4779e550b32SMatthew Booth         if (!state->curl) {
478769ce76dSAlexander Graf             return NULL;
4799e550b32SMatthew Booth         }
480769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_URL, s->url);
48197a3ea57SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_SSL_VERIFYPEER,
48297a3ea57SMatthew Booth                          (long) s->sslverify);
483a94f83d9SRichard W.M. Jones         if (s->cookie) {
484a94f83d9SRichard W.M. Jones             curl_easy_setopt(state->curl, CURLOPT_COOKIE, s->cookie);
485a94f83d9SRichard W.M. Jones         }
486f76faedaSRichard W.M. Jones         curl_easy_setopt(state->curl, CURLOPT_TIMEOUT, (long)s->timeout);
4879e550b32SMatthew Booth         curl_easy_setopt(state->curl, CURLOPT_WRITEFUNCTION,
4889e550b32SMatthew Booth                          (void *)curl_read_cb);
489769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_WRITEDATA, (void *)state);
490769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_PRIVATE, (void *)state);
491769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_AUTOREFERER, 1);
492769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_FOLLOWLOCATION, 1);
493769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_NOSIGNAL, 1);
494769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_ERRORBUFFER, state->errmsg);
495f785a5aeSNicholas Thomas         curl_easy_setopt(state->curl, CURLOPT_FAILONERROR, 1);
496769ce76dSAlexander Graf 
4971bff9606SDaniel P. Berrange         if (s->username) {
4981bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_USERNAME, s->username);
4991bff9606SDaniel P. Berrange         }
5001bff9606SDaniel P. Berrange         if (s->password) {
5011bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl, CURLOPT_PASSWORD, s->password);
5021bff9606SDaniel P. Berrange         }
5031bff9606SDaniel P. Berrange         if (s->proxyusername) {
5041bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
5051bff9606SDaniel P. Berrange                              CURLOPT_PROXYUSERNAME, s->proxyusername);
5061bff9606SDaniel P. Berrange         }
5071bff9606SDaniel P. Berrange         if (s->proxypassword) {
5081bff9606SDaniel P. Berrange             curl_easy_setopt(state->curl,
5091bff9606SDaniel P. Berrange                              CURLOPT_PROXYPASSWORD, s->proxypassword);
5101bff9606SDaniel P. Berrange         }
5111bff9606SDaniel P. Berrange 
512fb6d1bbdSStefan Hajnoczi         /* Restrict supported protocols to avoid security issues in the more
513fb6d1bbdSStefan Hajnoczi          * obscure protocols.  For example, do not allow POP3/SMTP/IMAP see
514fb6d1bbdSStefan Hajnoczi          * CVE-2013-0249.
5158a8f5840SStefan Hajnoczi          *
5168a8f5840SStefan Hajnoczi          * Restricting protocols is only supported from 7.19.4 upwards.
517fb6d1bbdSStefan Hajnoczi          */
5188a8f5840SStefan Hajnoczi #if LIBCURL_VERSION_NUM >= 0x071304
519fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_PROTOCOLS, PROTOCOLS);
520fb6d1bbdSStefan Hajnoczi         curl_easy_setopt(state->curl, CURLOPT_REDIR_PROTOCOLS, PROTOCOLS);
5218a8f5840SStefan Hajnoczi #endif
522fb6d1bbdSStefan Hajnoczi 
523769ce76dSAlexander Graf #ifdef DEBUG_VERBOSE
524769ce76dSAlexander Graf         curl_easy_setopt(state->curl, CURLOPT_VERBOSE, 1);
525769ce76dSAlexander Graf #endif
5269e550b32SMatthew Booth     }
527769ce76dSAlexander Graf 
528ff5ca166SMax Reitz     QLIST_INIT(&state->sockets);
529769ce76dSAlexander Graf     state->s = s;
530769ce76dSAlexander Graf 
531769ce76dSAlexander Graf     return state;
532769ce76dSAlexander Graf }
533769ce76dSAlexander Graf 
534769ce76dSAlexander Graf static void curl_clean_state(CURLState *s)
535769ce76dSAlexander Graf {
536769ce76dSAlexander Graf     if (s->s->multi)
537769ce76dSAlexander Graf         curl_multi_remove_handle(s->s->multi, s->curl);
538ff5ca166SMax Reitz 
539ff5ca166SMax Reitz     while (!QLIST_EMPTY(&s->sockets)) {
540ff5ca166SMax Reitz         CURLSocket *socket = QLIST_FIRST(&s->sockets);
541ff5ca166SMax Reitz 
542ff5ca166SMax Reitz         QLIST_REMOVE(socket, next);
543ff5ca166SMax Reitz         g_free(socket);
544ff5ca166SMax Reitz     }
545ff5ca166SMax Reitz 
546769ce76dSAlexander Graf     s->in_use = 0;
547769ce76dSAlexander Graf }
548769ce76dSAlexander Graf 
5498e6d58cdSKevin Wolf static void curl_parse_filename(const char *filename, QDict *options,
5508e6d58cdSKevin Wolf                                 Error **errp)
551769ce76dSAlexander Graf {
55246f5ac20SEric Blake     qdict_put_str(options, CURL_BLOCK_OPT_URL, filename);
5538e6d58cdSKevin Wolf }
5548e6d58cdSKevin Wolf 
55563f0f45fSStefan Hajnoczi static void curl_detach_aio_context(BlockDriverState *bs)
55663f0f45fSStefan Hajnoczi {
55763f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
55863f0f45fSStefan Hajnoczi     int i;
55963f0f45fSStefan Hajnoczi 
56063f0f45fSStefan Hajnoczi     for (i = 0; i < CURL_NUM_STATES; i++) {
56163f0f45fSStefan Hajnoczi         if (s->states[i].in_use) {
56263f0f45fSStefan Hajnoczi             curl_clean_state(&s->states[i]);
56363f0f45fSStefan Hajnoczi         }
56463f0f45fSStefan Hajnoczi         if (s->states[i].curl) {
56563f0f45fSStefan Hajnoczi             curl_easy_cleanup(s->states[i].curl);
56663f0f45fSStefan Hajnoczi             s->states[i].curl = NULL;
56763f0f45fSStefan Hajnoczi         }
56863f0f45fSStefan Hajnoczi         g_free(s->states[i].orig_buf);
56963f0f45fSStefan Hajnoczi         s->states[i].orig_buf = NULL;
57063f0f45fSStefan Hajnoczi     }
57163f0f45fSStefan Hajnoczi     if (s->multi) {
57263f0f45fSStefan Hajnoczi         curl_multi_cleanup(s->multi);
57363f0f45fSStefan Hajnoczi         s->multi = NULL;
57463f0f45fSStefan Hajnoczi     }
57563f0f45fSStefan Hajnoczi 
57663f0f45fSStefan Hajnoczi     timer_del(&s->timer);
57763f0f45fSStefan Hajnoczi }
57863f0f45fSStefan Hajnoczi 
57963f0f45fSStefan Hajnoczi static void curl_attach_aio_context(BlockDriverState *bs,
58063f0f45fSStefan Hajnoczi                                     AioContext *new_context)
58163f0f45fSStefan Hajnoczi {
58263f0f45fSStefan Hajnoczi     BDRVCURLState *s = bs->opaque;
58363f0f45fSStefan Hajnoczi 
58463f0f45fSStefan Hajnoczi     aio_timer_init(new_context, &s->timer,
58563f0f45fSStefan Hajnoczi                    QEMU_CLOCK_REALTIME, SCALE_NS,
58663f0f45fSStefan Hajnoczi                    curl_multi_timeout_do, s);
58763f0f45fSStefan Hajnoczi 
58863f0f45fSStefan Hajnoczi     assert(!s->multi);
58963f0f45fSStefan Hajnoczi     s->multi = curl_multi_init();
59063f0f45fSStefan Hajnoczi     s->aio_context = new_context;
59163f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_SOCKETFUNCTION, curl_sock_cb);
59263f0f45fSStefan Hajnoczi #ifdef NEED_CURL_TIMER_CALLBACK
59363f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERDATA, s);
59463f0f45fSStefan Hajnoczi     curl_multi_setopt(s->multi, CURLMOPT_TIMERFUNCTION, curl_timer_cb);
59563f0f45fSStefan Hajnoczi #endif
59663f0f45fSStefan Hajnoczi }
59763f0f45fSStefan Hajnoczi 
5988e6d58cdSKevin Wolf static QemuOptsList runtime_opts = {
5998e6d58cdSKevin Wolf     .name = "curl",
6008e6d58cdSKevin Wolf     .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
6018e6d58cdSKevin Wolf     .desc = {
6028e6d58cdSKevin Wolf         {
603e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_URL,
6048e6d58cdSKevin Wolf             .type = QEMU_OPT_STRING,
6058e6d58cdSKevin Wolf             .help = "URL to open",
6068e6d58cdSKevin Wolf         },
6078e6d58cdSKevin Wolf         {
608e3542c67SMatthew Booth             .name = CURL_BLOCK_OPT_READAHEAD,
6098e6d58cdSKevin Wolf             .type = QEMU_OPT_SIZE,
6108e6d58cdSKevin Wolf             .help = "Readahead size",
6118e6d58cdSKevin Wolf         },
61297a3ea57SMatthew Booth         {
61397a3ea57SMatthew Booth             .name = CURL_BLOCK_OPT_SSLVERIFY,
61497a3ea57SMatthew Booth             .type = QEMU_OPT_BOOL,
61597a3ea57SMatthew Booth             .help = "Verify SSL certificate"
61697a3ea57SMatthew Booth         },
617212aefaaSDaniel Henrique Barboza         {
618212aefaaSDaniel Henrique Barboza             .name = CURL_BLOCK_OPT_TIMEOUT,
619212aefaaSDaniel Henrique Barboza             .type = QEMU_OPT_NUMBER,
620212aefaaSDaniel Henrique Barboza             .help = "Curl timeout"
621212aefaaSDaniel Henrique Barboza         },
622a94f83d9SRichard W.M. Jones         {
623a94f83d9SRichard W.M. Jones             .name = CURL_BLOCK_OPT_COOKIE,
624a94f83d9SRichard W.M. Jones             .type = QEMU_OPT_STRING,
625a94f83d9SRichard W.M. Jones             .help = "Pass the cookie or list of cookies with each request"
626a94f83d9SRichard W.M. Jones         },
6271bff9606SDaniel P. Berrange         {
628*327c8ebdSPeter Krempa             .name = CURL_BLOCK_OPT_COOKIE_SECRET,
629*327c8ebdSPeter Krempa             .type = QEMU_OPT_STRING,
630*327c8ebdSPeter Krempa             .help = "ID of secret used as cookie passed with each request"
631*327c8ebdSPeter Krempa         },
632*327c8ebdSPeter Krempa         {
6331bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_USERNAME,
6341bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6351bff9606SDaniel P. Berrange             .help = "Username for HTTP auth"
6361bff9606SDaniel P. Berrange         },
6371bff9606SDaniel P. Berrange         {
6381bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PASSWORD_SECRET,
6391bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6401bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP auth",
6411bff9606SDaniel P. Berrange         },
6421bff9606SDaniel P. Berrange         {
6431bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_USERNAME,
6441bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6451bff9606SDaniel P. Berrange             .help = "Username for HTTP proxy auth"
6461bff9606SDaniel P. Berrange         },
6471bff9606SDaniel P. Berrange         {
6481bff9606SDaniel P. Berrange             .name = CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET,
6491bff9606SDaniel P. Berrange             .type = QEMU_OPT_STRING,
6501bff9606SDaniel P. Berrange             .help = "ID of secret used as password for HTTP proxy auth",
6511bff9606SDaniel P. Berrange         },
6528e6d58cdSKevin Wolf         { /* end of list */ }
6538e6d58cdSKevin Wolf     },
6548e6d58cdSKevin Wolf };
6558e6d58cdSKevin Wolf 
6561bff9606SDaniel P. Berrange 
657015a1036SMax Reitz static int curl_open(BlockDriverState *bs, QDict *options, int flags,
658015a1036SMax Reitz                      Error **errp)
6598e6d58cdSKevin Wolf {
6608e6d58cdSKevin Wolf     BDRVCURLState *s = bs->opaque;
6618e6d58cdSKevin Wolf     CURLState *state = NULL;
6628e6d58cdSKevin Wolf     QemuOpts *opts;
6638e6d58cdSKevin Wolf     Error *local_err = NULL;
6648e6d58cdSKevin Wolf     const char *file;
665a94f83d9SRichard W.M. Jones     const char *cookie;
666*327c8ebdSPeter Krempa     const char *cookie_secret;
6678e6d58cdSKevin Wolf     double d;
6681bff9606SDaniel P. Berrange     const char *secretid;
66934634ca2SMax Reitz     const char *protocol_delimiter;
6708e6d58cdSKevin Wolf 
6718e6d58cdSKevin Wolf     static int inited = 0;
6728e6d58cdSKevin Wolf 
673a7cea2baSRichard W.M. Jones     if (flags & BDRV_O_RDWR) {
6742a94fee3SPaolo Bonzini         error_setg(errp, "curl block device does not support writes");
675a7cea2baSRichard W.M. Jones         return -EROFS;
676a7cea2baSRichard W.M. Jones     }
677a7cea2baSRichard W.M. Jones 
67887ea75d5SPeter Crosthwaite     opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
6798e6d58cdSKevin Wolf     qemu_opts_absorb_qdict(opts, options, &local_err);
68084d18f06SMarkus Armbruster     if (local_err) {
6812a94fee3SPaolo Bonzini         error_propagate(errp, local_err);
6828e6d58cdSKevin Wolf         goto out_noclean;
6838e6d58cdSKevin Wolf     }
6848e6d58cdSKevin Wolf 
685e3542c67SMatthew Booth     s->readahead_size = qemu_opt_get_size(opts, CURL_BLOCK_OPT_READAHEAD,
686e3542c67SMatthew Booth                                           READ_AHEAD_DEFAULT);
687c76f4952SNolan     if ((s->readahead_size & 0x1ff) != 0) {
6882a94fee3SPaolo Bonzini         error_setg(errp, "HTTP_READAHEAD_SIZE %zd is not a multiple of 512",
689c76f4952SNolan                    s->readahead_size);
690c76f4952SNolan         goto out_noclean;
691c76f4952SNolan     }
692c76f4952SNolan 
693212aefaaSDaniel Henrique Barboza     s->timeout = qemu_opt_get_number(opts, CURL_BLOCK_OPT_TIMEOUT,
694212aefaaSDaniel Henrique Barboza                                      CURL_TIMEOUT_DEFAULT);
695f76faedaSRichard W.M. Jones     if (s->timeout > CURL_TIMEOUT_MAX) {
696f76faedaSRichard W.M. Jones         error_setg(errp, "timeout parameter is too large or negative");
697f76faedaSRichard W.M. Jones         goto out_noclean;
698f76faedaSRichard W.M. Jones     }
699212aefaaSDaniel Henrique Barboza 
70097a3ea57SMatthew Booth     s->sslverify = qemu_opt_get_bool(opts, CURL_BLOCK_OPT_SSLVERIFY, true);
70197a3ea57SMatthew Booth 
702a94f83d9SRichard W.M. Jones     cookie = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE);
703*327c8ebdSPeter Krempa     cookie_secret = qemu_opt_get(opts, CURL_BLOCK_OPT_COOKIE_SECRET);
704*327c8ebdSPeter Krempa 
705*327c8ebdSPeter Krempa     if (cookie && cookie_secret) {
706*327c8ebdSPeter Krempa         error_setg(errp,
707*327c8ebdSPeter Krempa                    "curl driver cannot handle both cookie and cookie secret");
708*327c8ebdSPeter Krempa         goto out_noclean;
709*327c8ebdSPeter Krempa     }
710*327c8ebdSPeter Krempa 
711*327c8ebdSPeter Krempa     if (cookie_secret) {
712*327c8ebdSPeter Krempa         s->cookie = qcrypto_secret_lookup_as_utf8(cookie_secret, errp);
713*327c8ebdSPeter Krempa         if (!s->cookie) {
714*327c8ebdSPeter Krempa             goto out_noclean;
715*327c8ebdSPeter Krempa         }
716*327c8ebdSPeter Krempa     } else {
717a94f83d9SRichard W.M. Jones         s->cookie = g_strdup(cookie);
718*327c8ebdSPeter Krempa     }
719a94f83d9SRichard W.M. Jones 
720e3542c67SMatthew Booth     file = qemu_opt_get(opts, CURL_BLOCK_OPT_URL);
7218e6d58cdSKevin Wolf     if (file == NULL) {
7222a94fee3SPaolo Bonzini         error_setg(errp, "curl block driver requires an 'url' option");
7238e6d58cdSKevin Wolf         goto out_noclean;
7248e6d58cdSKevin Wolf     }
7258e6d58cdSKevin Wolf 
72634634ca2SMax Reitz     if (!strstart(file, bs->drv->protocol_name, &protocol_delimiter) ||
72734634ca2SMax Reitz         !strstart(protocol_delimiter, "://", NULL))
72834634ca2SMax Reitz     {
72934634ca2SMax Reitz         error_setg(errp, "%s curl driver cannot handle the URL '%s' (does not "
73034634ca2SMax Reitz                    "start with '%s://')", bs->drv->protocol_name, file,
73134634ca2SMax Reitz                    bs->drv->protocol_name);
73234634ca2SMax Reitz         goto out_noclean;
73334634ca2SMax Reitz     }
73434634ca2SMax Reitz 
7351bff9606SDaniel P. Berrange     s->username = g_strdup(qemu_opt_get(opts, CURL_BLOCK_OPT_USERNAME));
7361bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PASSWORD_SECRET);
7371bff9606SDaniel P. Berrange 
7381bff9606SDaniel P. Berrange     if (secretid) {
7391bff9606SDaniel P. Berrange         s->password = qcrypto_secret_lookup_as_utf8(secretid, errp);
7401bff9606SDaniel P. Berrange         if (!s->password) {
7411bff9606SDaniel P. Berrange             goto out_noclean;
7421bff9606SDaniel P. Berrange         }
7431bff9606SDaniel P. Berrange     }
7441bff9606SDaniel P. Berrange 
7451bff9606SDaniel P. Berrange     s->proxyusername = g_strdup(
7461bff9606SDaniel P. Berrange         qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_USERNAME));
7471bff9606SDaniel P. Berrange     secretid = qemu_opt_get(opts, CURL_BLOCK_OPT_PROXY_PASSWORD_SECRET);
7481bff9606SDaniel P. Berrange     if (secretid) {
7491bff9606SDaniel P. Berrange         s->proxypassword = qcrypto_secret_lookup_as_utf8(secretid, errp);
7501bff9606SDaniel P. Berrange         if (!s->proxypassword) {
7511bff9606SDaniel P. Berrange             goto out_noclean;
7521bff9606SDaniel P. Berrange         }
7531bff9606SDaniel P. Berrange     }
7541bff9606SDaniel P. Berrange 
755769ce76dSAlexander Graf     if (!inited) {
756769ce76dSAlexander Graf         curl_global_init(CURL_GLOBAL_ALL);
757769ce76dSAlexander Graf         inited = 1;
758769ce76dSAlexander Graf     }
759769ce76dSAlexander Graf 
760d0f2c4c6Smalc     DPRINTF("CURL: Opening %s\n", file);
76163f0f45fSStefan Hajnoczi     s->aio_context = bdrv_get_aio_context(bs);
7628e6d58cdSKevin Wolf     s->url = g_strdup(file);
763a2f468e4SRichard W.M. Jones     state = curl_init_state(bs, s);
764769ce76dSAlexander Graf     if (!state)
765769ce76dSAlexander Graf         goto out_noclean;
766769ce76dSAlexander Graf 
767769ce76dSAlexander Graf     // Get file size
768769ce76dSAlexander Graf 
7693494d650SFam Zheng     s->accept_range = false;
770769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_NOBODY, 1);
7713494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERFUNCTION,
7723494d650SFam Zheng                      curl_header_cb);
7733494d650SFam Zheng     curl_easy_setopt(state->curl, CURLOPT_HEADERDATA, s);
774769ce76dSAlexander Graf     if (curl_easy_perform(state->curl))
775769ce76dSAlexander Graf         goto out;
776a41c4578STomáš Golembiovský     if (curl_easy_getinfo(state->curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &d)) {
777769ce76dSAlexander Graf         goto out;
778a41c4578STomáš Golembiovský     }
779a41c4578STomáš Golembiovský     /* Prior CURL 7.19.4 return value of 0 could mean that the file size is not
780a41c4578STomáš Golembiovský      * know or the size is zero. From 7.19.4 CURL returns -1 if size is not
781a41c4578STomáš Golembiovský      * known and zero if it is realy zero-length file. */
782a41c4578STomáš Golembiovský #if LIBCURL_VERSION_NUM >= 0x071304
783a41c4578STomáš Golembiovský     if (d < 0) {
784a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
785a41c4578STomáš Golembiovský                 "Server didn't report file size.");
786a41c4578STomáš Golembiovský         goto out;
787a41c4578STomáš Golembiovský     }
788a41c4578STomáš Golembiovský #else
789a41c4578STomáš Golembiovský     if (d <= 0) {
790a41c4578STomáš Golembiovský         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
791a41c4578STomáš Golembiovský                 "Unknown file size or zero-length file.");
792a41c4578STomáš Golembiovský         goto out;
793a41c4578STomáš Golembiovský     }
794a41c4578STomáš Golembiovský #endif
795a41c4578STomáš Golembiovský 
796a41c4578STomáš Golembiovský     s->len = (size_t)d;
797a41c4578STomáš Golembiovský 
7983494d650SFam Zheng     if ((!strncasecmp(s->url, "http://", strlen("http://"))
7993494d650SFam Zheng         || !strncasecmp(s->url, "https://", strlen("https://")))
8003494d650SFam Zheng         && !s->accept_range) {
8013494d650SFam Zheng         pstrcpy(state->errmsg, CURL_ERROR_SIZE,
8023494d650SFam Zheng                 "Server does not support 'range' (byte ranges).");
8033494d650SFam Zheng         goto out;
8043494d650SFam Zheng     }
8050bfcd599SBlue Swirl     DPRINTF("CURL: Size = %zd\n", s->len);
806769ce76dSAlexander Graf 
807769ce76dSAlexander Graf     curl_clean_state(state);
808769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
809769ce76dSAlexander Graf     state->curl = NULL;
810769ce76dSAlexander Graf 
811ba3186c4SPaolo Bonzini     qemu_mutex_init(&s->mutex);
81263f0f45fSStefan Hajnoczi     curl_attach_aio_context(bs, bdrv_get_aio_context(bs));
813769ce76dSAlexander Graf 
8148e6d58cdSKevin Wolf     qemu_opts_del(opts);
815769ce76dSAlexander Graf     return 0;
816769ce76dSAlexander Graf 
817769ce76dSAlexander Graf out:
818acd7fdc6SMaria Kustova     error_setg(errp, "CURL: Error opening file: %s", state->errmsg);
819769ce76dSAlexander Graf     curl_easy_cleanup(state->curl);
820769ce76dSAlexander Graf     state->curl = NULL;
821769ce76dSAlexander Graf out_noclean:
822a94f83d9SRichard W.M. Jones     g_free(s->cookie);
8238e6d58cdSKevin Wolf     g_free(s->url);
8248e6d58cdSKevin Wolf     qemu_opts_del(opts);
825769ce76dSAlexander Graf     return -EINVAL;
826769ce76dSAlexander Graf }
827769ce76dSAlexander Graf 
828d7331bedSStefan Hajnoczi static const AIOCBInfo curl_aiocb_info = {
829c16b5a2cSChristoph Hellwig     .aiocb_size         = sizeof(CURLAIOCB),
830c16b5a2cSChristoph Hellwig };
831c16b5a2cSChristoph Hellwig 
832363c3c85SNick Thomas 
833363c3c85SNick Thomas static void curl_readv_bh_cb(void *p)
834769ce76dSAlexander Graf {
835769ce76dSAlexander Graf     CURLState *state;
836b69cdef8SMatthew Booth     int running;
8371919631eSPaolo Bonzini     int ret = -EINPROGRESS;
838769ce76dSAlexander Graf 
839363c3c85SNick Thomas     CURLAIOCB *acb = p;
8401919631eSPaolo Bonzini     BlockDriverState *bs = acb->common.bs;
8411919631eSPaolo Bonzini     BDRVCURLState *s = bs->opaque;
842769ce76dSAlexander Graf 
8439054d9f6SMax Reitz     size_t start = acb->sector_num * BDRV_SECTOR_SIZE;
844363c3c85SNick Thomas     size_t end;
845769ce76dSAlexander Graf 
846ba3186c4SPaolo Bonzini     qemu_mutex_lock(&s->mutex);
8471919631eSPaolo Bonzini 
848769ce76dSAlexander Graf     // In case we have the requested data already (e.g. read-ahead),
849769ce76dSAlexander Graf     // we can just call the callback and be done.
8509054d9f6SMax Reitz     switch (curl_find_buf(s, start, acb->nb_sectors * BDRV_SECTOR_SIZE, acb)) {
851769ce76dSAlexander Graf         case FIND_RET_OK:
8528007429aSFam Zheng             qemu_aio_unref(acb);
853769ce76dSAlexander Graf             // fall through
854769ce76dSAlexander Graf         case FIND_RET_WAIT:
8551919631eSPaolo Bonzini             goto out;
856769ce76dSAlexander Graf         default:
857769ce76dSAlexander Graf             break;
858769ce76dSAlexander Graf     }
859769ce76dSAlexander Graf 
860769ce76dSAlexander Graf     // No cache found, so let's start a new request
861a2f468e4SRichard W.M. Jones     state = curl_init_state(acb->common.bs, s);
862363c3c85SNick Thomas     if (!state) {
8631919631eSPaolo Bonzini         ret = -EIO;
8641919631eSPaolo Bonzini         goto out;
865363c3c85SNick Thomas     }
866769ce76dSAlexander Graf 
867769ce76dSAlexander Graf     acb->start = 0;
8684e504535SMax Reitz     acb->end = MIN(acb->nb_sectors * BDRV_SECTOR_SIZE, s->len - start);
869769ce76dSAlexander Graf 
870769ce76dSAlexander Graf     state->buf_off = 0;
8717267c094SAnthony Liguori     g_free(state->orig_buf);
872769ce76dSAlexander Graf     state->buf_start = start;
8734e504535SMax Reitz     state->buf_len = MIN(acb->end + s->readahead_size, s->len - start);
8744e504535SMax Reitz     end = start + state->buf_len - 1;
8758dc7a772SKevin Wolf     state->orig_buf = g_try_malloc(state->buf_len);
8768dc7a772SKevin Wolf     if (state->buf_len && state->orig_buf == NULL) {
8778dc7a772SKevin Wolf         curl_clean_state(state);
8781919631eSPaolo Bonzini         ret = -ENOMEM;
8791919631eSPaolo Bonzini         goto out;
8808dc7a772SKevin Wolf     }
881769ce76dSAlexander Graf     state->acb[0] = acb;
882769ce76dSAlexander Graf 
8830bfcd599SBlue Swirl     snprintf(state->range, 127, "%zd-%zd", start, end);
8849054d9f6SMax Reitz     DPRINTF("CURL (AIO): Reading %llu at %zd (%s)\n",
8859054d9f6SMax Reitz             (acb->nb_sectors * BDRV_SECTOR_SIZE), start, state->range);
886769ce76dSAlexander Graf     curl_easy_setopt(state->curl, CURLOPT_RANGE, state->range);
887769ce76dSAlexander Graf 
888769ce76dSAlexander Graf     curl_multi_add_handle(s->multi, state->curl);
889769ce76dSAlexander Graf 
890b69cdef8SMatthew Booth     /* Tell curl it needs to kick things off */
891b69cdef8SMatthew Booth     curl_multi_socket_action(s->multi, CURL_SOCKET_TIMEOUT, 0, &running);
8921919631eSPaolo Bonzini 
8931919631eSPaolo Bonzini out:
894ba3186c4SPaolo Bonzini     qemu_mutex_unlock(&s->mutex);
8951919631eSPaolo Bonzini     if (ret != -EINPROGRESS) {
8961919631eSPaolo Bonzini         acb->common.cb(acb->common.opaque, ret);
8971919631eSPaolo Bonzini         qemu_aio_unref(acb);
8981919631eSPaolo Bonzini     }
899363c3c85SNick Thomas }
900363c3c85SNick Thomas 
9017c84b1b8SMarkus Armbruster static BlockAIOCB *curl_aio_readv(BlockDriverState *bs,
902363c3c85SNick Thomas         int64_t sector_num, QEMUIOVector *qiov, int nb_sectors,
903097310b5SMarkus Armbruster         BlockCompletionFunc *cb, void *opaque)
904363c3c85SNick Thomas {
905363c3c85SNick Thomas     CURLAIOCB *acb;
906363c3c85SNick Thomas 
907d7331bedSStefan Hajnoczi     acb = qemu_aio_get(&curl_aiocb_info, bs, cb, opaque);
908363c3c85SNick Thomas 
909363c3c85SNick Thomas     acb->qiov = qiov;
910363c3c85SNick Thomas     acb->sector_num = sector_num;
911363c3c85SNick Thomas     acb->nb_sectors = nb_sectors;
912363c3c85SNick Thomas 
913fffb6e12SPaolo Bonzini     aio_bh_schedule_oneshot(bdrv_get_aio_context(bs), curl_readv_bh_cb, acb);
914769ce76dSAlexander Graf     return &acb->common;
915769ce76dSAlexander Graf }
916769ce76dSAlexander Graf 
917769ce76dSAlexander Graf static void curl_close(BlockDriverState *bs)
918769ce76dSAlexander Graf {
919769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
920769ce76dSAlexander Graf 
921d0f2c4c6Smalc     DPRINTF("CURL: Close\n");
92263f0f45fSStefan Hajnoczi     curl_detach_aio_context(bs);
923ba3186c4SPaolo Bonzini     qemu_mutex_destroy(&s->mutex);
924031fd1beSPeter Maydell 
925a94f83d9SRichard W.M. Jones     g_free(s->cookie);
92645724d6dSStefan Weil     g_free(s->url);
927769ce76dSAlexander Graf }
928769ce76dSAlexander Graf 
929769ce76dSAlexander Graf static int64_t curl_getlength(BlockDriverState *bs)
930769ce76dSAlexander Graf {
931769ce76dSAlexander Graf     BDRVCURLState *s = bs->opaque;
932769ce76dSAlexander Graf     return s->len;
933769ce76dSAlexander Graf }
934769ce76dSAlexander Graf 
935769ce76dSAlexander Graf static BlockDriver bdrv_http = {
936769ce76dSAlexander Graf     .format_name                = "http",
937769ce76dSAlexander Graf     .protocol_name              = "http",
938769ce76dSAlexander Graf 
939769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9408e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
94166f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
942769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
943769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
944769ce76dSAlexander Graf 
945769ce76dSAlexander Graf     .bdrv_aio_readv             = curl_aio_readv,
94663f0f45fSStefan Hajnoczi 
94763f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
94863f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
949769ce76dSAlexander Graf };
950769ce76dSAlexander Graf 
951769ce76dSAlexander Graf static BlockDriver bdrv_https = {
952769ce76dSAlexander Graf     .format_name                = "https",
953769ce76dSAlexander Graf     .protocol_name              = "https",
954769ce76dSAlexander Graf 
955769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9568e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
95766f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
958769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
959769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
960769ce76dSAlexander Graf 
961769ce76dSAlexander Graf     .bdrv_aio_readv             = curl_aio_readv,
96263f0f45fSStefan Hajnoczi 
96363f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
96463f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
965769ce76dSAlexander Graf };
966769ce76dSAlexander Graf 
967769ce76dSAlexander Graf static BlockDriver bdrv_ftp = {
968769ce76dSAlexander Graf     .format_name                = "ftp",
969769ce76dSAlexander Graf     .protocol_name              = "ftp",
970769ce76dSAlexander Graf 
971769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9728e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
97366f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
974769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
975769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
976769ce76dSAlexander Graf 
977769ce76dSAlexander Graf     .bdrv_aio_readv             = curl_aio_readv,
97863f0f45fSStefan Hajnoczi 
97963f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
98063f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
981769ce76dSAlexander Graf };
982769ce76dSAlexander Graf 
983769ce76dSAlexander Graf static BlockDriver bdrv_ftps = {
984769ce76dSAlexander Graf     .format_name                = "ftps",
985769ce76dSAlexander Graf     .protocol_name              = "ftps",
986769ce76dSAlexander Graf 
987769ce76dSAlexander Graf     .instance_size              = sizeof(BDRVCURLState),
9888e6d58cdSKevin Wolf     .bdrv_parse_filename        = curl_parse_filename,
98966f82ceeSKevin Wolf     .bdrv_file_open             = curl_open,
990769ce76dSAlexander Graf     .bdrv_close                 = curl_close,
991769ce76dSAlexander Graf     .bdrv_getlength             = curl_getlength,
992769ce76dSAlexander Graf 
993769ce76dSAlexander Graf     .bdrv_aio_readv             = curl_aio_readv,
99463f0f45fSStefan Hajnoczi 
99563f0f45fSStefan Hajnoczi     .bdrv_detach_aio_context    = curl_detach_aio_context,
99663f0f45fSStefan Hajnoczi     .bdrv_attach_aio_context    = curl_attach_aio_context,
997769ce76dSAlexander Graf };
998769ce76dSAlexander Graf 
999769ce76dSAlexander Graf static void curl_block_init(void)
1000769ce76dSAlexander Graf {
1001769ce76dSAlexander Graf     bdrv_register(&bdrv_http);
1002769ce76dSAlexander Graf     bdrv_register(&bdrv_https);
1003769ce76dSAlexander Graf     bdrv_register(&bdrv_ftp);
1004769ce76dSAlexander Graf     bdrv_register(&bdrv_ftps);
1005769ce76dSAlexander Graf }
1006769ce76dSAlexander Graf 
1007769ce76dSAlexander Graf block_init(curl_block_init);
1008