xref: /qemu/block/nfs.c (revision 048abb7b)
16542aa9cSPeter Lieven /*
26542aa9cSPeter Lieven  * QEMU Block driver for native access to files on NFS shares
36542aa9cSPeter Lieven  *
438f8d5e0SPeter Lieven  * Copyright (c) 2014-2016 Peter Lieven <pl@kamp.de>
56542aa9cSPeter Lieven  *
66542aa9cSPeter Lieven  * Permission is hereby granted, free of charge, to any person obtaining a copy
76542aa9cSPeter Lieven  * of this software and associated documentation files (the "Software"), to deal
86542aa9cSPeter Lieven  * in the Software without restriction, including without limitation the rights
96542aa9cSPeter Lieven  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
106542aa9cSPeter Lieven  * copies of the Software, and to permit persons to whom the Software is
116542aa9cSPeter Lieven  * furnished to do so, subject to the following conditions:
126542aa9cSPeter Lieven  *
136542aa9cSPeter Lieven  * The above copyright notice and this permission notice shall be included in
146542aa9cSPeter Lieven  * all copies or substantial portions of the Software.
156542aa9cSPeter Lieven  *
166542aa9cSPeter Lieven  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
176542aa9cSPeter Lieven  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
186542aa9cSPeter Lieven  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
196542aa9cSPeter Lieven  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
206542aa9cSPeter Lieven  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
216542aa9cSPeter Lieven  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
226542aa9cSPeter Lieven  * THE SOFTWARE.
236542aa9cSPeter Lieven  */
246542aa9cSPeter Lieven 
2580c71a24SPeter Maydell #include "qemu/osdep.h"
266542aa9cSPeter Lieven 
276542aa9cSPeter Lieven #include <poll.h>
286542aa9cSPeter Lieven #include "qemu-common.h"
296542aa9cSPeter Lieven #include "qemu/config-file.h"
306542aa9cSPeter Lieven #include "qemu/error-report.h"
31d165b8cbSStefan Hajnoczi #include "qapi/error.h"
326542aa9cSPeter Lieven #include "block/block_int.h"
336542aa9cSPeter Lieven #include "trace.h"
346542aa9cSPeter Lieven #include "qemu/iov.h"
356542aa9cSPeter Lieven #include "qemu/uri.h"
360d94b746SStefan Hajnoczi #include "qemu/cutils.h"
376542aa9cSPeter Lieven #include "sysemu/sysemu.h"
3894d6a7a7SAshijeet Acharya #include "qapi/qmp/qdict.h"
3994d6a7a7SAshijeet Acharya #include "qapi/qmp/qint.h"
4094d6a7a7SAshijeet Acharya #include "qapi/qmp/qstring.h"
4194d6a7a7SAshijeet Acharya #include "qapi-visit.h"
4294d6a7a7SAshijeet Acharya #include "qapi/qobject-input-visitor.h"
4394d6a7a7SAshijeet Acharya #include "qapi/qobject-output-visitor.h"
446542aa9cSPeter Lieven #include <nfsc/libnfs.h>
456542aa9cSPeter Lieven 
4694d6a7a7SAshijeet Acharya 
4729c838cdSPeter Lieven #define QEMU_NFS_MAX_READAHEAD_SIZE 1048576
48d99b26c4SPeter Lieven #define QEMU_NFS_MAX_PAGECACHE_SIZE (8388608 / NFS_BLKSIZE)
497725b8bfSPeter Lieven #define QEMU_NFS_MAX_DEBUG_LEVEL 2
5029c838cdSPeter Lieven 
516542aa9cSPeter Lieven typedef struct NFSClient {
526542aa9cSPeter Lieven     struct nfs_context *context;
536542aa9cSPeter Lieven     struct nfsfh *fh;
546542aa9cSPeter Lieven     int events;
556542aa9cSPeter Lieven     bool has_zero_init;
56471799d1SStefan Hajnoczi     AioContext *aio_context;
5737d1e4d9SPaolo Bonzini     QemuMutex mutex;
5818a8056eSPeter Lieven     blkcnt_t st_blocks;
5938f8d5e0SPeter Lieven     bool cache_used;
6094d6a7a7SAshijeet Acharya     NFSServer *server;
6194d6a7a7SAshijeet Acharya     char *path;
6294d6a7a7SAshijeet Acharya     int64_t uid, gid, tcp_syncnt, readahead, pagecache, debug;
636542aa9cSPeter Lieven } NFSClient;
646542aa9cSPeter Lieven 
656542aa9cSPeter Lieven typedef struct NFSRPC {
66d746427aSPaolo Bonzini     BlockDriverState *bs;
676542aa9cSPeter Lieven     int ret;
686542aa9cSPeter Lieven     int complete;
696542aa9cSPeter Lieven     QEMUIOVector *iov;
706542aa9cSPeter Lieven     struct stat *st;
716542aa9cSPeter Lieven     Coroutine *co;
72471799d1SStefan Hajnoczi     NFSClient *client;
736542aa9cSPeter Lieven } NFSRPC;
746542aa9cSPeter Lieven 
7594d6a7a7SAshijeet Acharya static int nfs_parse_uri(const char *filename, QDict *options, Error **errp)
7694d6a7a7SAshijeet Acharya {
7794d6a7a7SAshijeet Acharya     URI *uri = NULL;
7894d6a7a7SAshijeet Acharya     QueryParams *qp = NULL;
7994d6a7a7SAshijeet Acharya     int ret = -EINVAL, i;
8094d6a7a7SAshijeet Acharya 
8194d6a7a7SAshijeet Acharya     uri = uri_parse(filename);
8294d6a7a7SAshijeet Acharya     if (!uri) {
8394d6a7a7SAshijeet Acharya         error_setg(errp, "Invalid URI specified");
8494d6a7a7SAshijeet Acharya         goto out;
8594d6a7a7SAshijeet Acharya     }
8694d6a7a7SAshijeet Acharya     if (strcmp(uri->scheme, "nfs") != 0) {
8794d6a7a7SAshijeet Acharya         error_setg(errp, "URI scheme must be 'nfs'");
8894d6a7a7SAshijeet Acharya         goto out;
8994d6a7a7SAshijeet Acharya     }
9094d6a7a7SAshijeet Acharya 
9194d6a7a7SAshijeet Acharya     if (!uri->server) {
9294d6a7a7SAshijeet Acharya         error_setg(errp, "missing hostname in URI");
9394d6a7a7SAshijeet Acharya         goto out;
9494d6a7a7SAshijeet Acharya     }
9594d6a7a7SAshijeet Acharya 
9694d6a7a7SAshijeet Acharya     if (!uri->path) {
9794d6a7a7SAshijeet Acharya         error_setg(errp, "missing file path in URI");
9894d6a7a7SAshijeet Acharya         goto out;
9994d6a7a7SAshijeet Acharya     }
10094d6a7a7SAshijeet Acharya 
10194d6a7a7SAshijeet Acharya     qp = query_params_parse(uri->query);
10294d6a7a7SAshijeet Acharya     if (!qp) {
10394d6a7a7SAshijeet Acharya         error_setg(errp, "could not parse query parameters");
10494d6a7a7SAshijeet Acharya         goto out;
10594d6a7a7SAshijeet Acharya     }
10694d6a7a7SAshijeet Acharya 
10794d6a7a7SAshijeet Acharya     qdict_put(options, "server.host", qstring_from_str(uri->server));
10894d6a7a7SAshijeet Acharya     qdict_put(options, "server.type", qstring_from_str("inet"));
10994d6a7a7SAshijeet Acharya     qdict_put(options, "path", qstring_from_str(uri->path));
11094d6a7a7SAshijeet Acharya 
11194d6a7a7SAshijeet Acharya     for (i = 0; i < qp->n; i++) {
1128d20abe8SPeter Lieven         unsigned long long val;
11394d6a7a7SAshijeet Acharya         if (!qp->p[i].value) {
11494d6a7a7SAshijeet Acharya             error_setg(errp, "Value for NFS parameter expected: %s",
11594d6a7a7SAshijeet Acharya                        qp->p[i].name);
11694d6a7a7SAshijeet Acharya             goto out;
11794d6a7a7SAshijeet Acharya         }
1188d20abe8SPeter Lieven         if (parse_uint_full(qp->p[i].value, &val, 0)) {
11994d6a7a7SAshijeet Acharya             error_setg(errp, "Illegal value for NFS parameter: %s",
12094d6a7a7SAshijeet Acharya                        qp->p[i].name);
12194d6a7a7SAshijeet Acharya             goto out;
12294d6a7a7SAshijeet Acharya         }
12394d6a7a7SAshijeet Acharya         if (!strcmp(qp->p[i].name, "uid")) {
12494d6a7a7SAshijeet Acharya             qdict_put(options, "user",
12594d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
12694d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "gid")) {
12794d6a7a7SAshijeet Acharya             qdict_put(options, "group",
12894d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
12994d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "tcp-syncnt")) {
13094d6a7a7SAshijeet Acharya             qdict_put(options, "tcp-syn-count",
13194d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
13294d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "readahead")) {
13394d6a7a7SAshijeet Acharya             qdict_put(options, "readahead-size",
13494d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
13594d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "pagecache")) {
13694d6a7a7SAshijeet Acharya             qdict_put(options, "page-cache-size",
13794d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
13894d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "debug")) {
1397103d916SPrasanna Kumar Kalever             qdict_put(options, "debug",
14094d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
14194d6a7a7SAshijeet Acharya         } else {
14294d6a7a7SAshijeet Acharya             error_setg(errp, "Unknown NFS parameter name: %s",
14394d6a7a7SAshijeet Acharya                        qp->p[i].name);
14494d6a7a7SAshijeet Acharya             goto out;
14594d6a7a7SAshijeet Acharya         }
14694d6a7a7SAshijeet Acharya     }
14794d6a7a7SAshijeet Acharya     ret = 0;
14894d6a7a7SAshijeet Acharya out:
14994d6a7a7SAshijeet Acharya     if (qp) {
15094d6a7a7SAshijeet Acharya         query_params_free(qp);
15194d6a7a7SAshijeet Acharya     }
15294d6a7a7SAshijeet Acharya     if (uri) {
15394d6a7a7SAshijeet Acharya         uri_free(uri);
15494d6a7a7SAshijeet Acharya     }
15594d6a7a7SAshijeet Acharya     return ret;
15694d6a7a7SAshijeet Acharya }
15794d6a7a7SAshijeet Acharya 
15894d6a7a7SAshijeet Acharya static bool nfs_has_filename_options_conflict(QDict *options, Error **errp)
15994d6a7a7SAshijeet Acharya {
16094d6a7a7SAshijeet Acharya     const QDictEntry *qe;
16194d6a7a7SAshijeet Acharya 
16294d6a7a7SAshijeet Acharya     for (qe = qdict_first(options); qe; qe = qdict_next(options, qe)) {
16394d6a7a7SAshijeet Acharya         if (!strcmp(qe->key, "host") ||
16494d6a7a7SAshijeet Acharya             !strcmp(qe->key, "path") ||
16594d6a7a7SAshijeet Acharya             !strcmp(qe->key, "user") ||
16694d6a7a7SAshijeet Acharya             !strcmp(qe->key, "group") ||
16794d6a7a7SAshijeet Acharya             !strcmp(qe->key, "tcp-syn-count") ||
16894d6a7a7SAshijeet Acharya             !strcmp(qe->key, "readahead-size") ||
16994d6a7a7SAshijeet Acharya             !strcmp(qe->key, "page-cache-size") ||
1707103d916SPrasanna Kumar Kalever             !strcmp(qe->key, "debug") ||
17194d6a7a7SAshijeet Acharya             strstart(qe->key, "server.", NULL))
17294d6a7a7SAshijeet Acharya         {
17394d6a7a7SAshijeet Acharya             error_setg(errp, "Option %s cannot be used with a filename",
17494d6a7a7SAshijeet Acharya                        qe->key);
17594d6a7a7SAshijeet Acharya             return true;
17694d6a7a7SAshijeet Acharya         }
17794d6a7a7SAshijeet Acharya     }
17894d6a7a7SAshijeet Acharya 
17994d6a7a7SAshijeet Acharya     return false;
18094d6a7a7SAshijeet Acharya }
18194d6a7a7SAshijeet Acharya 
18294d6a7a7SAshijeet Acharya static void nfs_parse_filename(const char *filename, QDict *options,
18394d6a7a7SAshijeet Acharya                                Error **errp)
18494d6a7a7SAshijeet Acharya {
18594d6a7a7SAshijeet Acharya     if (nfs_has_filename_options_conflict(options, errp)) {
18694d6a7a7SAshijeet Acharya         return;
18794d6a7a7SAshijeet Acharya     }
18894d6a7a7SAshijeet Acharya 
18994d6a7a7SAshijeet Acharya     nfs_parse_uri(filename, options, errp);
19094d6a7a7SAshijeet Acharya }
19194d6a7a7SAshijeet Acharya 
1926542aa9cSPeter Lieven static void nfs_process_read(void *arg);
1936542aa9cSPeter Lieven static void nfs_process_write(void *arg);
1946542aa9cSPeter Lieven 
19537d1e4d9SPaolo Bonzini /* Called with QemuMutex held.  */
1966542aa9cSPeter Lieven static void nfs_set_events(NFSClient *client)
1976542aa9cSPeter Lieven {
1986542aa9cSPeter Lieven     int ev = nfs_which_events(client->context);
1996542aa9cSPeter Lieven     if (ev != client->events) {
200dca21ef2SFam Zheng         aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
201dca21ef2SFam Zheng                            false,
2026542aa9cSPeter Lieven                            (ev & POLLIN) ? nfs_process_read : NULL,
203f6a51c84SStefan Hajnoczi                            (ev & POLLOUT) ? nfs_process_write : NULL,
204f6a51c84SStefan Hajnoczi                            NULL, client);
2056542aa9cSPeter Lieven 
2066542aa9cSPeter Lieven     }
2076542aa9cSPeter Lieven     client->events = ev;
2086542aa9cSPeter Lieven }
2096542aa9cSPeter Lieven 
2106542aa9cSPeter Lieven static void nfs_process_read(void *arg)
2116542aa9cSPeter Lieven {
2126542aa9cSPeter Lieven     NFSClient *client = arg;
2139d456654SPaolo Bonzini 
21437d1e4d9SPaolo Bonzini     qemu_mutex_lock(&client->mutex);
2156542aa9cSPeter Lieven     nfs_service(client->context, POLLIN);
2166542aa9cSPeter Lieven     nfs_set_events(client);
21737d1e4d9SPaolo Bonzini     qemu_mutex_unlock(&client->mutex);
2186542aa9cSPeter Lieven }
2196542aa9cSPeter Lieven 
2206542aa9cSPeter Lieven static void nfs_process_write(void *arg)
2216542aa9cSPeter Lieven {
2226542aa9cSPeter Lieven     NFSClient *client = arg;
2239d456654SPaolo Bonzini 
22437d1e4d9SPaolo Bonzini     qemu_mutex_lock(&client->mutex);
2256542aa9cSPeter Lieven     nfs_service(client->context, POLLOUT);
2266542aa9cSPeter Lieven     nfs_set_events(client);
22737d1e4d9SPaolo Bonzini     qemu_mutex_unlock(&client->mutex);
2286542aa9cSPeter Lieven }
2296542aa9cSPeter Lieven 
230d746427aSPaolo Bonzini static void nfs_co_init_task(BlockDriverState *bs, NFSRPC *task)
2316542aa9cSPeter Lieven {
2326542aa9cSPeter Lieven     *task = (NFSRPC) {
2336542aa9cSPeter Lieven         .co             = qemu_coroutine_self(),
234d746427aSPaolo Bonzini         .bs             = bs,
235d746427aSPaolo Bonzini         .client         = bs->opaque,
2366542aa9cSPeter Lieven     };
2376542aa9cSPeter Lieven }
2386542aa9cSPeter Lieven 
2396542aa9cSPeter Lieven static void nfs_co_generic_bh_cb(void *opaque)
2406542aa9cSPeter Lieven {
2416542aa9cSPeter Lieven     NFSRPC *task = opaque;
2421919631eSPaolo Bonzini 
243a2c0fe2fSPeter Lieven     task->complete = 1;
2441919631eSPaolo Bonzini     aio_co_wake(task->co);
2456542aa9cSPeter Lieven }
2466542aa9cSPeter Lieven 
24737d1e4d9SPaolo Bonzini /* Called (via nfs_service) with QemuMutex held.  */
2486542aa9cSPeter Lieven static void
2496542aa9cSPeter Lieven nfs_co_generic_cb(int ret, struct nfs_context *nfs, void *data,
2506542aa9cSPeter Lieven                   void *private_data)
2516542aa9cSPeter Lieven {
2526542aa9cSPeter Lieven     NFSRPC *task = private_data;
2536542aa9cSPeter Lieven     task->ret = ret;
254d746427aSPaolo Bonzini     assert(!task->st);
2556542aa9cSPeter Lieven     if (task->ret > 0 && task->iov) {
2566542aa9cSPeter Lieven         if (task->ret <= task->iov->size) {
2576542aa9cSPeter Lieven             qemu_iovec_from_buf(task->iov, 0, data, task->ret);
2586542aa9cSPeter Lieven         } else {
2596542aa9cSPeter Lieven             task->ret = -EIO;
2606542aa9cSPeter Lieven         }
2616542aa9cSPeter Lieven     }
26220fccb18SPeter Lieven     if (task->ret < 0) {
26320fccb18SPeter Lieven         error_report("NFS Error: %s", nfs_get_error(nfs));
26420fccb18SPeter Lieven     }
265fffb6e12SPaolo Bonzini     aio_bh_schedule_oneshot(task->client->aio_context,
266471799d1SStefan Hajnoczi                             nfs_co_generic_bh_cb, task);
2676542aa9cSPeter Lieven }
2686542aa9cSPeter Lieven 
26969785a22SPeter Lieven static int coroutine_fn nfs_co_preadv(BlockDriverState *bs, uint64_t offset,
27069785a22SPeter Lieven                                       uint64_t bytes, QEMUIOVector *iov,
27169785a22SPeter Lieven                                       int flags)
2726542aa9cSPeter Lieven {
2736542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
2746542aa9cSPeter Lieven     NFSRPC task;
2756542aa9cSPeter Lieven 
276d746427aSPaolo Bonzini     nfs_co_init_task(bs, &task);
2776542aa9cSPeter Lieven     task.iov = iov;
2786542aa9cSPeter Lieven 
27937d1e4d9SPaolo Bonzini     qemu_mutex_lock(&client->mutex);
2806542aa9cSPeter Lieven     if (nfs_pread_async(client->context, client->fh,
28169785a22SPeter Lieven                         offset, bytes, nfs_co_generic_cb, &task) != 0) {
28237d1e4d9SPaolo Bonzini         qemu_mutex_unlock(&client->mutex);
2836542aa9cSPeter Lieven         return -ENOMEM;
2846542aa9cSPeter Lieven     }
2856542aa9cSPeter Lieven 
2866542aa9cSPeter Lieven     nfs_set_events(client);
28737d1e4d9SPaolo Bonzini     qemu_mutex_unlock(&client->mutex);
288aa92d6c4SPaolo Bonzini     while (!task.complete) {
2896542aa9cSPeter Lieven         qemu_coroutine_yield();
2906542aa9cSPeter Lieven     }
2916542aa9cSPeter Lieven 
2926542aa9cSPeter Lieven     if (task.ret < 0) {
2936542aa9cSPeter Lieven         return task.ret;
2946542aa9cSPeter Lieven     }
2956542aa9cSPeter Lieven 
2966542aa9cSPeter Lieven     /* zero pad short reads */
2976542aa9cSPeter Lieven     if (task.ret < iov->size) {
2986542aa9cSPeter Lieven         qemu_iovec_memset(iov, task.ret, 0, iov->size - task.ret);
2996542aa9cSPeter Lieven     }
3006542aa9cSPeter Lieven 
3016542aa9cSPeter Lieven     return 0;
3026542aa9cSPeter Lieven }
3036542aa9cSPeter Lieven 
30469785a22SPeter Lieven static int coroutine_fn nfs_co_pwritev(BlockDriverState *bs, uint64_t offset,
30569785a22SPeter Lieven                                        uint64_t bytes, QEMUIOVector *iov,
30669785a22SPeter Lieven                                        int flags)
3076542aa9cSPeter Lieven {
3086542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
3096542aa9cSPeter Lieven     NFSRPC task;
3106542aa9cSPeter Lieven     char *buf = NULL;
311ef503a84SPeter Lieven     bool my_buffer = false;
3126542aa9cSPeter Lieven 
313d746427aSPaolo Bonzini     nfs_co_init_task(bs, &task);
3146542aa9cSPeter Lieven 
315ef503a84SPeter Lieven     if (iov->niov != 1) {
31669785a22SPeter Lieven         buf = g_try_malloc(bytes);
31769785a22SPeter Lieven         if (bytes && buf == NULL) {
3182347dd7bSKevin Wolf             return -ENOMEM;
3192347dd7bSKevin Wolf         }
32069785a22SPeter Lieven         qemu_iovec_to_buf(iov, 0, buf, bytes);
321ef503a84SPeter Lieven         my_buffer = true;
322ef503a84SPeter Lieven     } else {
323ef503a84SPeter Lieven         buf = iov->iov[0].iov_base;
324ef503a84SPeter Lieven     }
3256542aa9cSPeter Lieven 
32637d1e4d9SPaolo Bonzini     qemu_mutex_lock(&client->mutex);
3276542aa9cSPeter Lieven     if (nfs_pwrite_async(client->context, client->fh,
32869785a22SPeter Lieven                          offset, bytes, buf,
32969785a22SPeter Lieven                          nfs_co_generic_cb, &task) != 0) {
33037d1e4d9SPaolo Bonzini         qemu_mutex_unlock(&client->mutex);
331ef503a84SPeter Lieven         if (my_buffer) {
3326542aa9cSPeter Lieven             g_free(buf);
333ef503a84SPeter Lieven         }
3346542aa9cSPeter Lieven         return -ENOMEM;
3356542aa9cSPeter Lieven     }
3366542aa9cSPeter Lieven 
3376542aa9cSPeter Lieven     nfs_set_events(client);
33837d1e4d9SPaolo Bonzini     qemu_mutex_unlock(&client->mutex);
339aa92d6c4SPaolo Bonzini     while (!task.complete) {
3406542aa9cSPeter Lieven         qemu_coroutine_yield();
3416542aa9cSPeter Lieven     }
3426542aa9cSPeter Lieven 
343ef503a84SPeter Lieven     if (my_buffer) {
3446542aa9cSPeter Lieven         g_free(buf);
345ef503a84SPeter Lieven     }
3466542aa9cSPeter Lieven 
34769785a22SPeter Lieven     if (task.ret != bytes) {
3486542aa9cSPeter Lieven         return task.ret < 0 ? task.ret : -EIO;
3496542aa9cSPeter Lieven     }
3506542aa9cSPeter Lieven 
3516542aa9cSPeter Lieven     return 0;
3526542aa9cSPeter Lieven }
3536542aa9cSPeter Lieven 
3546542aa9cSPeter Lieven static int coroutine_fn nfs_co_flush(BlockDriverState *bs)
3556542aa9cSPeter Lieven {
3566542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
3576542aa9cSPeter Lieven     NFSRPC task;
3586542aa9cSPeter Lieven 
359d746427aSPaolo Bonzini     nfs_co_init_task(bs, &task);
3606542aa9cSPeter Lieven 
36137d1e4d9SPaolo Bonzini     qemu_mutex_lock(&client->mutex);
3626542aa9cSPeter Lieven     if (nfs_fsync_async(client->context, client->fh, nfs_co_generic_cb,
3636542aa9cSPeter Lieven                         &task) != 0) {
36437d1e4d9SPaolo Bonzini         qemu_mutex_unlock(&client->mutex);
3656542aa9cSPeter Lieven         return -ENOMEM;
3666542aa9cSPeter Lieven     }
3676542aa9cSPeter Lieven 
3686542aa9cSPeter Lieven     nfs_set_events(client);
36937d1e4d9SPaolo Bonzini     qemu_mutex_unlock(&client->mutex);
370aa92d6c4SPaolo Bonzini     while (!task.complete) {
3716542aa9cSPeter Lieven         qemu_coroutine_yield();
3726542aa9cSPeter Lieven     }
3736542aa9cSPeter Lieven 
3746542aa9cSPeter Lieven     return task.ret;
3756542aa9cSPeter Lieven }
3766542aa9cSPeter Lieven 
3776542aa9cSPeter Lieven static QemuOptsList runtime_opts = {
3786542aa9cSPeter Lieven     .name = "nfs",
3796542aa9cSPeter Lieven     .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
3806542aa9cSPeter Lieven     .desc = {
3816542aa9cSPeter Lieven         {
38294d6a7a7SAshijeet Acharya             .name = "path",
3836542aa9cSPeter Lieven             .type = QEMU_OPT_STRING,
38494d6a7a7SAshijeet Acharya             .help = "Path of the image on the host",
38594d6a7a7SAshijeet Acharya         },
38694d6a7a7SAshijeet Acharya         {
387f67409a5SPeter Lieven             .name = "user",
38894d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
38994d6a7a7SAshijeet Acharya             .help = "UID value to use when talking to the server",
39094d6a7a7SAshijeet Acharya         },
39194d6a7a7SAshijeet Acharya         {
392f67409a5SPeter Lieven             .name = "group",
39394d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
39494d6a7a7SAshijeet Acharya             .help = "GID value to use when talking to the server",
39594d6a7a7SAshijeet Acharya         },
39694d6a7a7SAshijeet Acharya         {
397f67409a5SPeter Lieven             .name = "tcp-syn-count",
39894d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
39994d6a7a7SAshijeet Acharya             .help = "Number of SYNs to send during the session establish",
40094d6a7a7SAshijeet Acharya         },
40194d6a7a7SAshijeet Acharya         {
402f67409a5SPeter Lieven             .name = "readahead-size",
40394d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
40494d6a7a7SAshijeet Acharya             .help = "Set the readahead size in bytes",
40594d6a7a7SAshijeet Acharya         },
40694d6a7a7SAshijeet Acharya         {
407f67409a5SPeter Lieven             .name = "page-cache-size",
40894d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
40994d6a7a7SAshijeet Acharya             .help = "Set the pagecache size in bytes",
41094d6a7a7SAshijeet Acharya         },
41194d6a7a7SAshijeet Acharya         {
41294d6a7a7SAshijeet Acharya             .name = "debug",
41394d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
41494d6a7a7SAshijeet Acharya             .help = "Set the NFS debug level (max 2)",
4156542aa9cSPeter Lieven         },
4166542aa9cSPeter Lieven         { /* end of list */ }
4176542aa9cSPeter Lieven     },
4186542aa9cSPeter Lieven };
4196542aa9cSPeter Lieven 
420471799d1SStefan Hajnoczi static void nfs_detach_aio_context(BlockDriverState *bs)
421471799d1SStefan Hajnoczi {
422471799d1SStefan Hajnoczi     NFSClient *client = bs->opaque;
423471799d1SStefan Hajnoczi 
424dca21ef2SFam Zheng     aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
425f6a51c84SStefan Hajnoczi                        false, NULL, NULL, NULL, NULL);
426471799d1SStefan Hajnoczi     client->events = 0;
427471799d1SStefan Hajnoczi }
428471799d1SStefan Hajnoczi 
429471799d1SStefan Hajnoczi static void nfs_attach_aio_context(BlockDriverState *bs,
430471799d1SStefan Hajnoczi                                    AioContext *new_context)
431471799d1SStefan Hajnoczi {
432471799d1SStefan Hajnoczi     NFSClient *client = bs->opaque;
433471799d1SStefan Hajnoczi 
434471799d1SStefan Hajnoczi     client->aio_context = new_context;
435471799d1SStefan Hajnoczi     nfs_set_events(client);
436471799d1SStefan Hajnoczi }
437471799d1SStefan Hajnoczi 
4386542aa9cSPeter Lieven static void nfs_client_close(NFSClient *client)
4396542aa9cSPeter Lieven {
4406542aa9cSPeter Lieven     if (client->context) {
4416542aa9cSPeter Lieven         if (client->fh) {
4426542aa9cSPeter Lieven             nfs_close(client->context, client->fh);
4436542aa9cSPeter Lieven         }
444dca21ef2SFam Zheng         aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
445f6a51c84SStefan Hajnoczi                            false, NULL, NULL, NULL, NULL);
4466542aa9cSPeter Lieven         nfs_destroy_context(client->context);
4476542aa9cSPeter Lieven     }
4486542aa9cSPeter Lieven     memset(client, 0, sizeof(NFSClient));
4496542aa9cSPeter Lieven }
4506542aa9cSPeter Lieven 
4516542aa9cSPeter Lieven static void nfs_file_close(BlockDriverState *bs)
4526542aa9cSPeter Lieven {
4536542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
4546542aa9cSPeter Lieven     nfs_client_close(client);
45537d1e4d9SPaolo Bonzini     qemu_mutex_destroy(&client->mutex);
4566542aa9cSPeter Lieven }
4576542aa9cSPeter Lieven 
45894d6a7a7SAshijeet Acharya static NFSServer *nfs_config(QDict *options, Error **errp)
45994d6a7a7SAshijeet Acharya {
46094d6a7a7SAshijeet Acharya     NFSServer *server = NULL;
46194d6a7a7SAshijeet Acharya     QDict *addr = NULL;
46294d6a7a7SAshijeet Acharya     QObject *crumpled_addr = NULL;
46394d6a7a7SAshijeet Acharya     Visitor *iv = NULL;
46494d6a7a7SAshijeet Acharya     Error *local_error = NULL;
46594d6a7a7SAshijeet Acharya 
46694d6a7a7SAshijeet Acharya     qdict_extract_subqdict(options, &addr, "server.");
46794d6a7a7SAshijeet Acharya     if (!qdict_size(addr)) {
46894d6a7a7SAshijeet Acharya         error_setg(errp, "NFS server address missing");
46994d6a7a7SAshijeet Acharya         goto out;
47094d6a7a7SAshijeet Acharya     }
47194d6a7a7SAshijeet Acharya 
47294d6a7a7SAshijeet Acharya     crumpled_addr = qdict_crumple(addr, errp);
47394d6a7a7SAshijeet Acharya     if (!crumpled_addr) {
47494d6a7a7SAshijeet Acharya         goto out;
47594d6a7a7SAshijeet Acharya     }
47694d6a7a7SAshijeet Acharya 
477*048abb7bSMarkus Armbruster     iv = qobject_input_visitor_new(crumpled_addr);
47894d6a7a7SAshijeet Acharya     visit_type_NFSServer(iv, NULL, &server, &local_error);
47994d6a7a7SAshijeet Acharya     if (local_error) {
48094d6a7a7SAshijeet Acharya         error_propagate(errp, local_error);
48194d6a7a7SAshijeet Acharya         goto out;
48294d6a7a7SAshijeet Acharya     }
48394d6a7a7SAshijeet Acharya 
48494d6a7a7SAshijeet Acharya out:
48594d6a7a7SAshijeet Acharya     QDECREF(addr);
48694d6a7a7SAshijeet Acharya     qobject_decref(crumpled_addr);
48794d6a7a7SAshijeet Acharya     visit_free(iv);
48894d6a7a7SAshijeet Acharya     return server;
48994d6a7a7SAshijeet Acharya }
49094d6a7a7SAshijeet Acharya 
49194d6a7a7SAshijeet Acharya 
49294d6a7a7SAshijeet Acharya static int64_t nfs_client_open(NFSClient *client, QDict *options,
49338f8d5e0SPeter Lieven                                int flags, Error **errp, int open_flags)
4946542aa9cSPeter Lieven {
49594d6a7a7SAshijeet Acharya     int ret = -EINVAL;
49694d6a7a7SAshijeet Acharya     QemuOpts *opts = NULL;
49794d6a7a7SAshijeet Acharya     Error *local_err = NULL;
4986542aa9cSPeter Lieven     struct stat st;
4996542aa9cSPeter Lieven     char *file = NULL, *strp = NULL;
5006542aa9cSPeter Lieven 
50194d6a7a7SAshijeet Acharya     opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
50294d6a7a7SAshijeet Acharya     qemu_opts_absorb_qdict(opts, options, &local_err);
50394d6a7a7SAshijeet Acharya     if (local_err) {
50494d6a7a7SAshijeet Acharya         error_propagate(errp, local_err);
50594d6a7a7SAshijeet Acharya         ret = -EINVAL;
5066542aa9cSPeter Lieven         goto fail;
5076542aa9cSPeter Lieven     }
50894d6a7a7SAshijeet Acharya 
50994d6a7a7SAshijeet Acharya     client->path = g_strdup(qemu_opt_get(opts, "path"));
51094d6a7a7SAshijeet Acharya     if (!client->path) {
51194d6a7a7SAshijeet Acharya         ret = -EINVAL;
51294d6a7a7SAshijeet Acharya         error_setg(errp, "No path was specified");
5135f4d5e1aSMax Reitz         goto fail;
5145f4d5e1aSMax Reitz     }
51594d6a7a7SAshijeet Acharya 
51694d6a7a7SAshijeet Acharya     strp = strrchr(client->path, '/');
5176542aa9cSPeter Lieven     if (strp == NULL) {
5186542aa9cSPeter Lieven         error_setg(errp, "Invalid URL specified");
5196542aa9cSPeter Lieven         goto fail;
5206542aa9cSPeter Lieven     }
5216542aa9cSPeter Lieven     file = g_strdup(strp);
5226542aa9cSPeter Lieven     *strp = 0;
5236542aa9cSPeter Lieven 
52494d6a7a7SAshijeet Acharya     /* Pop the config into our state object, Exit if invalid */
52594d6a7a7SAshijeet Acharya     client->server = nfs_config(options, errp);
52694d6a7a7SAshijeet Acharya     if (!client->server) {
52794d6a7a7SAshijeet Acharya         ret = -EINVAL;
52894d6a7a7SAshijeet Acharya         goto fail;
52994d6a7a7SAshijeet Acharya     }
53094d6a7a7SAshijeet Acharya 
5316542aa9cSPeter Lieven     client->context = nfs_init_context();
5326542aa9cSPeter Lieven     if (client->context == NULL) {
5336542aa9cSPeter Lieven         error_setg(errp, "Failed to init NFS context");
5346542aa9cSPeter Lieven         goto fail;
5356542aa9cSPeter Lieven     }
5366542aa9cSPeter Lieven 
537f67409a5SPeter Lieven     if (qemu_opt_get(opts, "user")) {
538f67409a5SPeter Lieven         client->uid = qemu_opt_get_number(opts, "user", 0);
53994d6a7a7SAshijeet Acharya         nfs_set_uid(client->context, client->uid);
5406542aa9cSPeter Lieven     }
54194d6a7a7SAshijeet Acharya 
542f67409a5SPeter Lieven     if (qemu_opt_get(opts, "group")) {
543f67409a5SPeter Lieven         client->gid = qemu_opt_get_number(opts, "group", 0);
54494d6a7a7SAshijeet Acharya         nfs_set_gid(client->context, client->gid);
5457c24384bSPeter Lieven     }
54694d6a7a7SAshijeet Acharya 
547f67409a5SPeter Lieven     if (qemu_opt_get(opts, "tcp-syn-count")) {
548f67409a5SPeter Lieven         client->tcp_syncnt = qemu_opt_get_number(opts, "tcp-syn-count", 0);
54994d6a7a7SAshijeet Acharya         nfs_set_tcp_syncnt(client->context, client->tcp_syncnt);
55094d6a7a7SAshijeet Acharya     }
55194d6a7a7SAshijeet Acharya 
552f42ca3caSPeter Lieven #ifdef LIBNFS_FEATURE_READAHEAD
553f67409a5SPeter Lieven     if (qemu_opt_get(opts, "readahead-size")) {
55438f8d5e0SPeter Lieven         if (open_flags & BDRV_O_NOCACHE) {
55538f8d5e0SPeter Lieven             error_setg(errp, "Cannot enable NFS readahead "
55638f8d5e0SPeter Lieven                              "if cache.direct = on");
55738f8d5e0SPeter Lieven             goto fail;
55838f8d5e0SPeter Lieven         }
559f67409a5SPeter Lieven         client->readahead = qemu_opt_get_number(opts, "readahead-size", 0);
56094d6a7a7SAshijeet Acharya         if (client->readahead > QEMU_NFS_MAX_READAHEAD_SIZE) {
56129c838cdSPeter Lieven             error_report("NFS Warning: Truncating NFS readahead "
56229c838cdSPeter Lieven                          "size to %d", QEMU_NFS_MAX_READAHEAD_SIZE);
56394d6a7a7SAshijeet Acharya             client->readahead = QEMU_NFS_MAX_READAHEAD_SIZE;
56429c838cdSPeter Lieven         }
56594d6a7a7SAshijeet Acharya         nfs_set_readahead(client->context, client->readahead);
566d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
567d99b26c4SPeter Lieven         nfs_set_pagecache_ttl(client->context, 0);
568d99b26c4SPeter Lieven #endif
569d99b26c4SPeter Lieven         client->cache_used = true;
57094d6a7a7SAshijeet Acharya     }
571d99b26c4SPeter Lieven #endif
57294d6a7a7SAshijeet Acharya 
573d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
574f67409a5SPeter Lieven     if (qemu_opt_get(opts, "page-cache-size")) {
575d99b26c4SPeter Lieven         if (open_flags & BDRV_O_NOCACHE) {
576d99b26c4SPeter Lieven             error_setg(errp, "Cannot enable NFS pagecache "
577d99b26c4SPeter Lieven                              "if cache.direct = on");
578d99b26c4SPeter Lieven             goto fail;
579d99b26c4SPeter Lieven         }
580f67409a5SPeter Lieven         client->pagecache = qemu_opt_get_number(opts, "page-cache-size", 0);
58194d6a7a7SAshijeet Acharya         if (client->pagecache > QEMU_NFS_MAX_PAGECACHE_SIZE) {
582d99b26c4SPeter Lieven             error_report("NFS Warning: Truncating NFS pagecache "
583d99b26c4SPeter Lieven                          "size to %d pages", QEMU_NFS_MAX_PAGECACHE_SIZE);
58494d6a7a7SAshijeet Acharya             client->pagecache = QEMU_NFS_MAX_PAGECACHE_SIZE;
585d99b26c4SPeter Lieven         }
58694d6a7a7SAshijeet Acharya         nfs_set_pagecache(client->context, client->pagecache);
587d99b26c4SPeter Lieven         nfs_set_pagecache_ttl(client->context, 0);
58838f8d5e0SPeter Lieven         client->cache_used = true;
58994d6a7a7SAshijeet Acharya     }
590f42ca3caSPeter Lieven #endif
59194d6a7a7SAshijeet Acharya 
5927725b8bfSPeter Lieven #ifdef LIBNFS_FEATURE_DEBUG
59394d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "debug")) {
59494d6a7a7SAshijeet Acharya         client->debug = qemu_opt_get_number(opts, "debug", 0);
5957725b8bfSPeter Lieven         /* limit the maximum debug level to avoid potential flooding
5967725b8bfSPeter Lieven          * of our log files. */
59794d6a7a7SAshijeet Acharya         if (client->debug > QEMU_NFS_MAX_DEBUG_LEVEL) {
5987725b8bfSPeter Lieven             error_report("NFS Warning: Limiting NFS debug level "
5997725b8bfSPeter Lieven                          "to %d", QEMU_NFS_MAX_DEBUG_LEVEL);
60094d6a7a7SAshijeet Acharya             client->debug = QEMU_NFS_MAX_DEBUG_LEVEL;
6017725b8bfSPeter Lieven         }
60294d6a7a7SAshijeet Acharya         nfs_set_debug(client->context, client->debug);
60394d6a7a7SAshijeet Acharya     }
6047725b8bfSPeter Lieven #endif
6056542aa9cSPeter Lieven 
60694d6a7a7SAshijeet Acharya     ret = nfs_mount(client->context, client->server->host, client->path);
6076542aa9cSPeter Lieven     if (ret < 0) {
6086542aa9cSPeter Lieven         error_setg(errp, "Failed to mount nfs share: %s",
6096542aa9cSPeter Lieven                    nfs_get_error(client->context));
6106542aa9cSPeter Lieven         goto fail;
6116542aa9cSPeter Lieven     }
6126542aa9cSPeter Lieven 
6136542aa9cSPeter Lieven     if (flags & O_CREAT) {
6146542aa9cSPeter Lieven         ret = nfs_creat(client->context, file, 0600, &client->fh);
6156542aa9cSPeter Lieven         if (ret < 0) {
6166542aa9cSPeter Lieven             error_setg(errp, "Failed to create file: %s",
6176542aa9cSPeter Lieven                        nfs_get_error(client->context));
6186542aa9cSPeter Lieven             goto fail;
6196542aa9cSPeter Lieven         }
6206542aa9cSPeter Lieven     } else {
6216542aa9cSPeter Lieven         ret = nfs_open(client->context, file, flags, &client->fh);
6226542aa9cSPeter Lieven         if (ret < 0) {
6236542aa9cSPeter Lieven             error_setg(errp, "Failed to open file : %s",
6246542aa9cSPeter Lieven                        nfs_get_error(client->context));
6256542aa9cSPeter Lieven             goto fail;
6266542aa9cSPeter Lieven         }
6276542aa9cSPeter Lieven     }
6286542aa9cSPeter Lieven 
6296542aa9cSPeter Lieven     ret = nfs_fstat(client->context, client->fh, &st);
6306542aa9cSPeter Lieven     if (ret < 0) {
6316542aa9cSPeter Lieven         error_setg(errp, "Failed to fstat file: %s",
6326542aa9cSPeter Lieven                    nfs_get_error(client->context));
6336542aa9cSPeter Lieven         goto fail;
6346542aa9cSPeter Lieven     }
6356542aa9cSPeter Lieven 
6366542aa9cSPeter Lieven     ret = DIV_ROUND_UP(st.st_size, BDRV_SECTOR_SIZE);
63718a8056eSPeter Lieven     client->st_blocks = st.st_blocks;
6386542aa9cSPeter Lieven     client->has_zero_init = S_ISREG(st.st_mode);
63994d6a7a7SAshijeet Acharya     *strp = '/';
6406542aa9cSPeter Lieven     goto out;
64194d6a7a7SAshijeet Acharya 
6426542aa9cSPeter Lieven fail:
6436542aa9cSPeter Lieven     nfs_client_close(client);
6446542aa9cSPeter Lieven out:
64594d6a7a7SAshijeet Acharya     qemu_opts_del(opts);
6466542aa9cSPeter Lieven     g_free(file);
6476542aa9cSPeter Lieven     return ret;
6486542aa9cSPeter Lieven }
6496542aa9cSPeter Lieven 
6506542aa9cSPeter Lieven static int nfs_file_open(BlockDriverState *bs, QDict *options, int flags,
6516542aa9cSPeter Lieven                          Error **errp) {
6526542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
6536542aa9cSPeter Lieven     int64_t ret;
6546542aa9cSPeter Lieven 
655471799d1SStefan Hajnoczi     client->aio_context = bdrv_get_aio_context(bs);
656471799d1SStefan Hajnoczi 
65794d6a7a7SAshijeet Acharya     ret = nfs_client_open(client, options,
6586542aa9cSPeter Lieven                           (flags & BDRV_O_RDWR) ? O_RDWR : O_RDONLY,
65938f8d5e0SPeter Lieven                           errp, bs->open_flags);
6606542aa9cSPeter Lieven     if (ret < 0) {
66194d6a7a7SAshijeet Acharya         return ret;
6626542aa9cSPeter Lieven     }
66337d1e4d9SPaolo Bonzini     qemu_mutex_init(&client->mutex);
6646542aa9cSPeter Lieven     bs->total_sectors = ret;
665810f4f86SFam Zheng     ret = 0;
666810f4f86SFam Zheng     return ret;
6676542aa9cSPeter Lieven }
6686542aa9cSPeter Lieven 
669fd752801SMax Reitz static QemuOptsList nfs_create_opts = {
670fd752801SMax Reitz     .name = "nfs-create-opts",
671fd752801SMax Reitz     .head = QTAILQ_HEAD_INITIALIZER(nfs_create_opts.head),
672fd752801SMax Reitz     .desc = {
673fd752801SMax Reitz         {
674fd752801SMax Reitz             .name = BLOCK_OPT_SIZE,
675fd752801SMax Reitz             .type = QEMU_OPT_SIZE,
676fd752801SMax Reitz             .help = "Virtual disk size"
677fd752801SMax Reitz         },
678fd752801SMax Reitz         { /* end of list */ }
679fd752801SMax Reitz     }
680fd752801SMax Reitz };
681fd752801SMax Reitz 
68298c10b81SChunyan Liu static int nfs_file_create(const char *url, QemuOpts *opts, Error **errp)
6836542aa9cSPeter Lieven {
6846542aa9cSPeter Lieven     int ret = 0;
6856542aa9cSPeter Lieven     int64_t total_size = 0;
6865839e53bSMarkus Armbruster     NFSClient *client = g_new0(NFSClient, 1);
68794d6a7a7SAshijeet Acharya     QDict *options = NULL;
6886542aa9cSPeter Lieven 
689471799d1SStefan Hajnoczi     client->aio_context = qemu_get_aio_context();
690471799d1SStefan Hajnoczi 
6916542aa9cSPeter Lieven     /* Read out options */
692c2eb918eSHu Tao     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
693c2eb918eSHu Tao                           BDRV_SECTOR_SIZE);
6946542aa9cSPeter Lieven 
69594d6a7a7SAshijeet Acharya     options = qdict_new();
69694d6a7a7SAshijeet Acharya     ret = nfs_parse_uri(url, options, errp);
69794d6a7a7SAshijeet Acharya     if (ret < 0) {
69894d6a7a7SAshijeet Acharya         goto out;
69994d6a7a7SAshijeet Acharya     }
70094d6a7a7SAshijeet Acharya 
70194d6a7a7SAshijeet Acharya     ret = nfs_client_open(client, options, O_CREAT, errp, 0);
7026542aa9cSPeter Lieven     if (ret < 0) {
7036542aa9cSPeter Lieven         goto out;
7046542aa9cSPeter Lieven     }
7056542aa9cSPeter Lieven     ret = nfs_ftruncate(client->context, client->fh, total_size);
7066542aa9cSPeter Lieven     nfs_client_close(client);
7076542aa9cSPeter Lieven out:
70807555ba6SKevin Wolf     QDECREF(options);
7096542aa9cSPeter Lieven     g_free(client);
7106542aa9cSPeter Lieven     return ret;
7116542aa9cSPeter Lieven }
7126542aa9cSPeter Lieven 
7136542aa9cSPeter Lieven static int nfs_has_zero_init(BlockDriverState *bs)
7146542aa9cSPeter Lieven {
7156542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
7166542aa9cSPeter Lieven     return client->has_zero_init;
7176542aa9cSPeter Lieven }
7186542aa9cSPeter Lieven 
71937d1e4d9SPaolo Bonzini /* Called (via nfs_service) with QemuMutex held.  */
720d746427aSPaolo Bonzini static void
721d746427aSPaolo Bonzini nfs_get_allocated_file_size_cb(int ret, struct nfs_context *nfs, void *data,
722d746427aSPaolo Bonzini                                void *private_data)
723d746427aSPaolo Bonzini {
724d746427aSPaolo Bonzini     NFSRPC *task = private_data;
725d746427aSPaolo Bonzini     task->ret = ret;
726d746427aSPaolo Bonzini     if (task->ret == 0) {
727d746427aSPaolo Bonzini         memcpy(task->st, data, sizeof(struct stat));
728d746427aSPaolo Bonzini     }
729d746427aSPaolo Bonzini     if (task->ret < 0) {
730d746427aSPaolo Bonzini         error_report("NFS Error: %s", nfs_get_error(nfs));
731d746427aSPaolo Bonzini     }
732d746427aSPaolo Bonzini     task->complete = 1;
733c9d1a561SPaolo Bonzini     bdrv_wakeup(task->bs);
734d746427aSPaolo Bonzini }
735d746427aSPaolo Bonzini 
7366542aa9cSPeter Lieven static int64_t nfs_get_allocated_file_size(BlockDriverState *bs)
7376542aa9cSPeter Lieven {
7386542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
7396542aa9cSPeter Lieven     NFSRPC task = {0};
7406542aa9cSPeter Lieven     struct stat st;
7416542aa9cSPeter Lieven 
74218a8056eSPeter Lieven     if (bdrv_is_read_only(bs) &&
74318a8056eSPeter Lieven         !(bs->open_flags & BDRV_O_NOCACHE)) {
74418a8056eSPeter Lieven         return client->st_blocks * 512;
74518a8056eSPeter Lieven     }
74618a8056eSPeter Lieven 
747d746427aSPaolo Bonzini     task.bs = bs;
7486542aa9cSPeter Lieven     task.st = &st;
749d746427aSPaolo Bonzini     if (nfs_fstat_async(client->context, client->fh, nfs_get_allocated_file_size_cb,
7506542aa9cSPeter Lieven                         &task) != 0) {
7516542aa9cSPeter Lieven         return -ENOMEM;
7526542aa9cSPeter Lieven     }
7536542aa9cSPeter Lieven 
7546542aa9cSPeter Lieven     nfs_set_events(client);
755d746427aSPaolo Bonzini     BDRV_POLL_WHILE(bs, !task.complete);
7566542aa9cSPeter Lieven 
757055c6f91SPeter Lieven     return (task.ret < 0 ? task.ret : st.st_blocks * 512);
7586542aa9cSPeter Lieven }
7596542aa9cSPeter Lieven 
7606542aa9cSPeter Lieven static int nfs_file_truncate(BlockDriverState *bs, int64_t offset)
7616542aa9cSPeter Lieven {
7626542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
7636542aa9cSPeter Lieven     return nfs_ftruncate(client->context, client->fh, offset);
7646542aa9cSPeter Lieven }
7656542aa9cSPeter Lieven 
76618a8056eSPeter Lieven /* Note that this will not re-establish a connection with the NFS server
76718a8056eSPeter Lieven  * - it is effectively a NOP.  */
76818a8056eSPeter Lieven static int nfs_reopen_prepare(BDRVReopenState *state,
76918a8056eSPeter Lieven                               BlockReopenQueue *queue, Error **errp)
77018a8056eSPeter Lieven {
77118a8056eSPeter Lieven     NFSClient *client = state->bs->opaque;
77218a8056eSPeter Lieven     struct stat st;
77318a8056eSPeter Lieven     int ret = 0;
77418a8056eSPeter Lieven 
77518a8056eSPeter Lieven     if (state->flags & BDRV_O_RDWR && bdrv_is_read_only(state->bs)) {
77618a8056eSPeter Lieven         error_setg(errp, "Cannot open a read-only mount as read-write");
77718a8056eSPeter Lieven         return -EACCES;
77818a8056eSPeter Lieven     }
77918a8056eSPeter Lieven 
78038f8d5e0SPeter Lieven     if ((state->flags & BDRV_O_NOCACHE) && client->cache_used) {
781d99b26c4SPeter Lieven         error_setg(errp, "Cannot disable cache if libnfs readahead or"
782d99b26c4SPeter Lieven                          " pagecache is enabled");
78338f8d5e0SPeter Lieven         return -EINVAL;
78438f8d5e0SPeter Lieven     }
78538f8d5e0SPeter Lieven 
78618a8056eSPeter Lieven     /* Update cache for read-only reopens */
78718a8056eSPeter Lieven     if (!(state->flags & BDRV_O_RDWR)) {
78818a8056eSPeter Lieven         ret = nfs_fstat(client->context, client->fh, &st);
78918a8056eSPeter Lieven         if (ret < 0) {
79018a8056eSPeter Lieven             error_setg(errp, "Failed to fstat file: %s",
79118a8056eSPeter Lieven                        nfs_get_error(client->context));
79218a8056eSPeter Lieven             return ret;
79318a8056eSPeter Lieven         }
79418a8056eSPeter Lieven         client->st_blocks = st.st_blocks;
79518a8056eSPeter Lieven     }
79618a8056eSPeter Lieven 
79718a8056eSPeter Lieven     return 0;
79818a8056eSPeter Lieven }
79918a8056eSPeter Lieven 
80094d6a7a7SAshijeet Acharya static void nfs_refresh_filename(BlockDriverState *bs, QDict *options)
80194d6a7a7SAshijeet Acharya {
80294d6a7a7SAshijeet Acharya     NFSClient *client = bs->opaque;
80394d6a7a7SAshijeet Acharya     QDict *opts = qdict_new();
80494d6a7a7SAshijeet Acharya     QObject *server_qdict;
80594d6a7a7SAshijeet Acharya     Visitor *ov;
80694d6a7a7SAshijeet Acharya 
80794d6a7a7SAshijeet Acharya     qdict_put(opts, "driver", qstring_from_str("nfs"));
80894d6a7a7SAshijeet Acharya 
80994d6a7a7SAshijeet Acharya     if (client->uid && !client->gid) {
81094d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
81194d6a7a7SAshijeet Acharya                  "nfs://%s%s?uid=%" PRId64, client->server->host, client->path,
81294d6a7a7SAshijeet Acharya                  client->uid);
81394d6a7a7SAshijeet Acharya     } else if (!client->uid && client->gid) {
81494d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
81594d6a7a7SAshijeet Acharya                  "nfs://%s%s?gid=%" PRId64, client->server->host, client->path,
81694d6a7a7SAshijeet Acharya                  client->gid);
81794d6a7a7SAshijeet Acharya     } else if (client->uid && client->gid) {
81894d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
81994d6a7a7SAshijeet Acharya                  "nfs://%s%s?uid=%" PRId64 "&gid=%" PRId64,
82094d6a7a7SAshijeet Acharya                  client->server->host, client->path, client->uid, client->gid);
82194d6a7a7SAshijeet Acharya     } else {
82294d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
82394d6a7a7SAshijeet Acharya                  "nfs://%s%s", client->server->host, client->path);
82494d6a7a7SAshijeet Acharya     }
82594d6a7a7SAshijeet Acharya 
82694d6a7a7SAshijeet Acharya     ov = qobject_output_visitor_new(&server_qdict);
82794d6a7a7SAshijeet Acharya     visit_type_NFSServer(ov, NULL, &client->server, &error_abort);
82894d6a7a7SAshijeet Acharya     visit_complete(ov, &server_qdict);
82994d6a7a7SAshijeet Acharya     qdict_put_obj(opts, "server", server_qdict);
83094d6a7a7SAshijeet Acharya     qdict_put(opts, "path", qstring_from_str(client->path));
83194d6a7a7SAshijeet Acharya 
83294d6a7a7SAshijeet Acharya     if (client->uid) {
833f67409a5SPeter Lieven         qdict_put(opts, "user", qint_from_int(client->uid));
83494d6a7a7SAshijeet Acharya     }
83594d6a7a7SAshijeet Acharya     if (client->gid) {
836f67409a5SPeter Lieven         qdict_put(opts, "group", qint_from_int(client->gid));
83794d6a7a7SAshijeet Acharya     }
83894d6a7a7SAshijeet Acharya     if (client->tcp_syncnt) {
839f67409a5SPeter Lieven         qdict_put(opts, "tcp-syn-cnt",
84094d6a7a7SAshijeet Acharya                   qint_from_int(client->tcp_syncnt));
84194d6a7a7SAshijeet Acharya     }
84294d6a7a7SAshijeet Acharya     if (client->readahead) {
843f67409a5SPeter Lieven         qdict_put(opts, "readahead-size",
84494d6a7a7SAshijeet Acharya                   qint_from_int(client->readahead));
84594d6a7a7SAshijeet Acharya     }
84694d6a7a7SAshijeet Acharya     if (client->pagecache) {
847f67409a5SPeter Lieven         qdict_put(opts, "page-cache-size",
84894d6a7a7SAshijeet Acharya                   qint_from_int(client->pagecache));
84994d6a7a7SAshijeet Acharya     }
85094d6a7a7SAshijeet Acharya     if (client->debug) {
85194d6a7a7SAshijeet Acharya         qdict_put(opts, "debug", qint_from_int(client->debug));
85294d6a7a7SAshijeet Acharya     }
85394d6a7a7SAshijeet Acharya 
85494d6a7a7SAshijeet Acharya     visit_free(ov);
85594d6a7a7SAshijeet Acharya     qdict_flatten(opts);
85694d6a7a7SAshijeet Acharya     bs->full_open_options = opts;
85794d6a7a7SAshijeet Acharya }
85894d6a7a7SAshijeet Acharya 
859d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
860d99b26c4SPeter Lieven static void nfs_invalidate_cache(BlockDriverState *bs,
861d99b26c4SPeter Lieven                                  Error **errp)
862d99b26c4SPeter Lieven {
863d99b26c4SPeter Lieven     NFSClient *client = bs->opaque;
864d99b26c4SPeter Lieven     nfs_pagecache_invalidate(client->context, client->fh);
865d99b26c4SPeter Lieven }
866d99b26c4SPeter Lieven #endif
867d99b26c4SPeter Lieven 
8686542aa9cSPeter Lieven static BlockDriver bdrv_nfs = {
8696542aa9cSPeter Lieven     .format_name                    = "nfs",
8706542aa9cSPeter Lieven     .protocol_name                  = "nfs",
8716542aa9cSPeter Lieven 
8726542aa9cSPeter Lieven     .instance_size                  = sizeof(NFSClient),
87394d6a7a7SAshijeet Acharya     .bdrv_parse_filename            = nfs_parse_filename,
874fd752801SMax Reitz     .create_opts                    = &nfs_create_opts,
875fd752801SMax Reitz 
8766542aa9cSPeter Lieven     .bdrv_has_zero_init             = nfs_has_zero_init,
8776542aa9cSPeter Lieven     .bdrv_get_allocated_file_size   = nfs_get_allocated_file_size,
8786542aa9cSPeter Lieven     .bdrv_truncate                  = nfs_file_truncate,
8796542aa9cSPeter Lieven 
8806542aa9cSPeter Lieven     .bdrv_file_open                 = nfs_file_open,
8816542aa9cSPeter Lieven     .bdrv_close                     = nfs_file_close,
882c282e1fdSChunyan Liu     .bdrv_create                    = nfs_file_create,
88318a8056eSPeter Lieven     .bdrv_reopen_prepare            = nfs_reopen_prepare,
8846542aa9cSPeter Lieven 
88569785a22SPeter Lieven     .bdrv_co_preadv                 = nfs_co_preadv,
88669785a22SPeter Lieven     .bdrv_co_pwritev                = nfs_co_pwritev,
8876542aa9cSPeter Lieven     .bdrv_co_flush_to_disk          = nfs_co_flush,
888471799d1SStefan Hajnoczi 
889471799d1SStefan Hajnoczi     .bdrv_detach_aio_context        = nfs_detach_aio_context,
890471799d1SStefan Hajnoczi     .bdrv_attach_aio_context        = nfs_attach_aio_context,
89194d6a7a7SAshijeet Acharya     .bdrv_refresh_filename          = nfs_refresh_filename,
892d99b26c4SPeter Lieven 
893d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
894d99b26c4SPeter Lieven     .bdrv_invalidate_cache          = nfs_invalidate_cache,
895d99b26c4SPeter Lieven #endif
8966542aa9cSPeter Lieven };
8976542aa9cSPeter Lieven 
8986542aa9cSPeter Lieven static void nfs_block_init(void)
8996542aa9cSPeter Lieven {
9006542aa9cSPeter Lieven     bdrv_register(&bdrv_nfs);
9016542aa9cSPeter Lieven }
9026542aa9cSPeter Lieven 
9036542aa9cSPeter Lieven block_init(nfs_block_init);
904