xref: /qemu/block/nfs.c (revision 8d20abe8)
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;
5718a8056eSPeter Lieven     blkcnt_t st_blocks;
5838f8d5e0SPeter Lieven     bool cache_used;
5994d6a7a7SAshijeet Acharya     NFSServer *server;
6094d6a7a7SAshijeet Acharya     char *path;
6194d6a7a7SAshijeet Acharya     int64_t uid, gid, tcp_syncnt, readahead, pagecache, debug;
626542aa9cSPeter Lieven } NFSClient;
636542aa9cSPeter Lieven 
646542aa9cSPeter Lieven typedef struct NFSRPC {
65d746427aSPaolo Bonzini     BlockDriverState *bs;
666542aa9cSPeter Lieven     int ret;
676542aa9cSPeter Lieven     int complete;
686542aa9cSPeter Lieven     QEMUIOVector *iov;
696542aa9cSPeter Lieven     struct stat *st;
706542aa9cSPeter Lieven     Coroutine *co;
71471799d1SStefan Hajnoczi     NFSClient *client;
726542aa9cSPeter Lieven } NFSRPC;
736542aa9cSPeter Lieven 
7494d6a7a7SAshijeet Acharya static int nfs_parse_uri(const char *filename, QDict *options, Error **errp)
7594d6a7a7SAshijeet Acharya {
7694d6a7a7SAshijeet Acharya     URI *uri = NULL;
7794d6a7a7SAshijeet Acharya     QueryParams *qp = NULL;
7894d6a7a7SAshijeet Acharya     int ret = -EINVAL, i;
7994d6a7a7SAshijeet Acharya 
8094d6a7a7SAshijeet Acharya     uri = uri_parse(filename);
8194d6a7a7SAshijeet Acharya     if (!uri) {
8294d6a7a7SAshijeet Acharya         error_setg(errp, "Invalid URI specified");
8394d6a7a7SAshijeet Acharya         goto out;
8494d6a7a7SAshijeet Acharya     }
8594d6a7a7SAshijeet Acharya     if (strcmp(uri->scheme, "nfs") != 0) {
8694d6a7a7SAshijeet Acharya         error_setg(errp, "URI scheme must be 'nfs'");
8794d6a7a7SAshijeet Acharya         goto out;
8894d6a7a7SAshijeet Acharya     }
8994d6a7a7SAshijeet Acharya 
9094d6a7a7SAshijeet Acharya     if (!uri->server) {
9194d6a7a7SAshijeet Acharya         error_setg(errp, "missing hostname in URI");
9294d6a7a7SAshijeet Acharya         goto out;
9394d6a7a7SAshijeet Acharya     }
9494d6a7a7SAshijeet Acharya 
9594d6a7a7SAshijeet Acharya     if (!uri->path) {
9694d6a7a7SAshijeet Acharya         error_setg(errp, "missing file path in URI");
9794d6a7a7SAshijeet Acharya         goto out;
9894d6a7a7SAshijeet Acharya     }
9994d6a7a7SAshijeet Acharya 
10094d6a7a7SAshijeet Acharya     qp = query_params_parse(uri->query);
10194d6a7a7SAshijeet Acharya     if (!qp) {
10294d6a7a7SAshijeet Acharya         error_setg(errp, "could not parse query parameters");
10394d6a7a7SAshijeet Acharya         goto out;
10494d6a7a7SAshijeet Acharya     }
10594d6a7a7SAshijeet Acharya 
10694d6a7a7SAshijeet Acharya     qdict_put(options, "server.host", qstring_from_str(uri->server));
10794d6a7a7SAshijeet Acharya     qdict_put(options, "server.type", qstring_from_str("inet"));
10894d6a7a7SAshijeet Acharya     qdict_put(options, "path", qstring_from_str(uri->path));
10994d6a7a7SAshijeet Acharya 
11094d6a7a7SAshijeet Acharya     for (i = 0; i < qp->n; i++) {
111*8d20abe8SPeter Lieven         unsigned long long val;
11294d6a7a7SAshijeet Acharya         if (!qp->p[i].value) {
11394d6a7a7SAshijeet Acharya             error_setg(errp, "Value for NFS parameter expected: %s",
11494d6a7a7SAshijeet Acharya                        qp->p[i].name);
11594d6a7a7SAshijeet Acharya             goto out;
11694d6a7a7SAshijeet Acharya         }
117*8d20abe8SPeter Lieven         if (parse_uint_full(qp->p[i].value, &val, 0)) {
11894d6a7a7SAshijeet Acharya             error_setg(errp, "Illegal value for NFS parameter: %s",
11994d6a7a7SAshijeet Acharya                        qp->p[i].name);
12094d6a7a7SAshijeet Acharya             goto out;
12194d6a7a7SAshijeet Acharya         }
12294d6a7a7SAshijeet Acharya         if (!strcmp(qp->p[i].name, "uid")) {
12394d6a7a7SAshijeet Acharya             qdict_put(options, "user",
12494d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
12594d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "gid")) {
12694d6a7a7SAshijeet Acharya             qdict_put(options, "group",
12794d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
12894d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "tcp-syncnt")) {
12994d6a7a7SAshijeet Acharya             qdict_put(options, "tcp-syn-count",
13094d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
13194d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "readahead")) {
13294d6a7a7SAshijeet Acharya             qdict_put(options, "readahead-size",
13394d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
13494d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "pagecache")) {
13594d6a7a7SAshijeet Acharya             qdict_put(options, "page-cache-size",
13694d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
13794d6a7a7SAshijeet Acharya         } else if (!strcmp(qp->p[i].name, "debug")) {
1387103d916SPrasanna Kumar Kalever             qdict_put(options, "debug",
13994d6a7a7SAshijeet Acharya                       qstring_from_str(qp->p[i].value));
14094d6a7a7SAshijeet Acharya         } else {
14194d6a7a7SAshijeet Acharya             error_setg(errp, "Unknown NFS parameter name: %s",
14294d6a7a7SAshijeet Acharya                        qp->p[i].name);
14394d6a7a7SAshijeet Acharya             goto out;
14494d6a7a7SAshijeet Acharya         }
14594d6a7a7SAshijeet Acharya     }
14694d6a7a7SAshijeet Acharya     ret = 0;
14794d6a7a7SAshijeet Acharya out:
14894d6a7a7SAshijeet Acharya     if (qp) {
14994d6a7a7SAshijeet Acharya         query_params_free(qp);
15094d6a7a7SAshijeet Acharya     }
15194d6a7a7SAshijeet Acharya     if (uri) {
15294d6a7a7SAshijeet Acharya         uri_free(uri);
15394d6a7a7SAshijeet Acharya     }
15494d6a7a7SAshijeet Acharya     return ret;
15594d6a7a7SAshijeet Acharya }
15694d6a7a7SAshijeet Acharya 
15794d6a7a7SAshijeet Acharya static bool nfs_has_filename_options_conflict(QDict *options, Error **errp)
15894d6a7a7SAshijeet Acharya {
15994d6a7a7SAshijeet Acharya     const QDictEntry *qe;
16094d6a7a7SAshijeet Acharya 
16194d6a7a7SAshijeet Acharya     for (qe = qdict_first(options); qe; qe = qdict_next(options, qe)) {
16294d6a7a7SAshijeet Acharya         if (!strcmp(qe->key, "host") ||
16394d6a7a7SAshijeet Acharya             !strcmp(qe->key, "path") ||
16494d6a7a7SAshijeet Acharya             !strcmp(qe->key, "user") ||
16594d6a7a7SAshijeet Acharya             !strcmp(qe->key, "group") ||
16694d6a7a7SAshijeet Acharya             !strcmp(qe->key, "tcp-syn-count") ||
16794d6a7a7SAshijeet Acharya             !strcmp(qe->key, "readahead-size") ||
16894d6a7a7SAshijeet Acharya             !strcmp(qe->key, "page-cache-size") ||
1697103d916SPrasanna Kumar Kalever             !strcmp(qe->key, "debug") ||
17094d6a7a7SAshijeet Acharya             strstart(qe->key, "server.", NULL))
17194d6a7a7SAshijeet Acharya         {
17294d6a7a7SAshijeet Acharya             error_setg(errp, "Option %s cannot be used with a filename",
17394d6a7a7SAshijeet Acharya                        qe->key);
17494d6a7a7SAshijeet Acharya             return true;
17594d6a7a7SAshijeet Acharya         }
17694d6a7a7SAshijeet Acharya     }
17794d6a7a7SAshijeet Acharya 
17894d6a7a7SAshijeet Acharya     return false;
17994d6a7a7SAshijeet Acharya }
18094d6a7a7SAshijeet Acharya 
18194d6a7a7SAshijeet Acharya static void nfs_parse_filename(const char *filename, QDict *options,
18294d6a7a7SAshijeet Acharya                                Error **errp)
18394d6a7a7SAshijeet Acharya {
18494d6a7a7SAshijeet Acharya     if (nfs_has_filename_options_conflict(options, errp)) {
18594d6a7a7SAshijeet Acharya         return;
18694d6a7a7SAshijeet Acharya     }
18794d6a7a7SAshijeet Acharya 
18894d6a7a7SAshijeet Acharya     nfs_parse_uri(filename, options, errp);
18994d6a7a7SAshijeet Acharya }
19094d6a7a7SAshijeet Acharya 
1916542aa9cSPeter Lieven static void nfs_process_read(void *arg);
1926542aa9cSPeter Lieven static void nfs_process_write(void *arg);
1936542aa9cSPeter Lieven 
1946542aa9cSPeter Lieven static void nfs_set_events(NFSClient *client)
1956542aa9cSPeter Lieven {
1966542aa9cSPeter Lieven     int ev = nfs_which_events(client->context);
1976542aa9cSPeter Lieven     if (ev != client->events) {
198dca21ef2SFam Zheng         aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
199dca21ef2SFam Zheng                            false,
2006542aa9cSPeter Lieven                            (ev & POLLIN) ? nfs_process_read : NULL,
201f6a51c84SStefan Hajnoczi                            (ev & POLLOUT) ? nfs_process_write : NULL,
202f6a51c84SStefan Hajnoczi                            NULL, client);
2036542aa9cSPeter Lieven 
2046542aa9cSPeter Lieven     }
2056542aa9cSPeter Lieven     client->events = ev;
2066542aa9cSPeter Lieven }
2076542aa9cSPeter Lieven 
2086542aa9cSPeter Lieven static void nfs_process_read(void *arg)
2096542aa9cSPeter Lieven {
2106542aa9cSPeter Lieven     NFSClient *client = arg;
2116542aa9cSPeter Lieven     nfs_service(client->context, POLLIN);
2126542aa9cSPeter Lieven     nfs_set_events(client);
2136542aa9cSPeter Lieven }
2146542aa9cSPeter Lieven 
2156542aa9cSPeter Lieven static void nfs_process_write(void *arg)
2166542aa9cSPeter Lieven {
2176542aa9cSPeter Lieven     NFSClient *client = arg;
2186542aa9cSPeter Lieven     nfs_service(client->context, POLLOUT);
2196542aa9cSPeter Lieven     nfs_set_events(client);
2206542aa9cSPeter Lieven }
2216542aa9cSPeter Lieven 
222d746427aSPaolo Bonzini static void nfs_co_init_task(BlockDriverState *bs, NFSRPC *task)
2236542aa9cSPeter Lieven {
2246542aa9cSPeter Lieven     *task = (NFSRPC) {
2256542aa9cSPeter Lieven         .co             = qemu_coroutine_self(),
226d746427aSPaolo Bonzini         .bs             = bs,
227d746427aSPaolo Bonzini         .client         = bs->opaque,
2286542aa9cSPeter Lieven     };
2296542aa9cSPeter Lieven }
2306542aa9cSPeter Lieven 
2316542aa9cSPeter Lieven static void nfs_co_generic_bh_cb(void *opaque)
2326542aa9cSPeter Lieven {
2336542aa9cSPeter Lieven     NFSRPC *task = opaque;
234a2c0fe2fSPeter Lieven     task->complete = 1;
2350b8b8753SPaolo Bonzini     qemu_coroutine_enter(task->co);
2366542aa9cSPeter Lieven }
2376542aa9cSPeter Lieven 
2386542aa9cSPeter Lieven static void
2396542aa9cSPeter Lieven nfs_co_generic_cb(int ret, struct nfs_context *nfs, void *data,
2406542aa9cSPeter Lieven                   void *private_data)
2416542aa9cSPeter Lieven {
2426542aa9cSPeter Lieven     NFSRPC *task = private_data;
2436542aa9cSPeter Lieven     task->ret = ret;
244d746427aSPaolo Bonzini     assert(!task->st);
2456542aa9cSPeter Lieven     if (task->ret > 0 && task->iov) {
2466542aa9cSPeter Lieven         if (task->ret <= task->iov->size) {
2476542aa9cSPeter Lieven             qemu_iovec_from_buf(task->iov, 0, data, task->ret);
2486542aa9cSPeter Lieven         } else {
2496542aa9cSPeter Lieven             task->ret = -EIO;
2506542aa9cSPeter Lieven         }
2516542aa9cSPeter Lieven     }
25220fccb18SPeter Lieven     if (task->ret < 0) {
25320fccb18SPeter Lieven         error_report("NFS Error: %s", nfs_get_error(nfs));
25420fccb18SPeter Lieven     }
255fffb6e12SPaolo Bonzini     aio_bh_schedule_oneshot(task->client->aio_context,
256471799d1SStefan Hajnoczi                             nfs_co_generic_bh_cb, task);
2576542aa9cSPeter Lieven }
2586542aa9cSPeter Lieven 
2596542aa9cSPeter Lieven static int coroutine_fn nfs_co_readv(BlockDriverState *bs,
2606542aa9cSPeter Lieven                                      int64_t sector_num, int nb_sectors,
2616542aa9cSPeter Lieven                                      QEMUIOVector *iov)
2626542aa9cSPeter Lieven {
2636542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
2646542aa9cSPeter Lieven     NFSRPC task;
2656542aa9cSPeter Lieven 
266d746427aSPaolo Bonzini     nfs_co_init_task(bs, &task);
2676542aa9cSPeter Lieven     task.iov = iov;
2686542aa9cSPeter Lieven 
2696542aa9cSPeter Lieven     if (nfs_pread_async(client->context, client->fh,
2706542aa9cSPeter Lieven                         sector_num * BDRV_SECTOR_SIZE,
2716542aa9cSPeter Lieven                         nb_sectors * BDRV_SECTOR_SIZE,
2726542aa9cSPeter Lieven                         nfs_co_generic_cb, &task) != 0) {
2736542aa9cSPeter Lieven         return -ENOMEM;
2746542aa9cSPeter Lieven     }
2756542aa9cSPeter Lieven 
2766542aa9cSPeter Lieven     nfs_set_events(client);
277aa92d6c4SPaolo Bonzini     while (!task.complete) {
2786542aa9cSPeter Lieven         qemu_coroutine_yield();
2796542aa9cSPeter Lieven     }
2806542aa9cSPeter Lieven 
2816542aa9cSPeter Lieven     if (task.ret < 0) {
2826542aa9cSPeter Lieven         return task.ret;
2836542aa9cSPeter Lieven     }
2846542aa9cSPeter Lieven 
2856542aa9cSPeter Lieven     /* zero pad short reads */
2866542aa9cSPeter Lieven     if (task.ret < iov->size) {
2876542aa9cSPeter Lieven         qemu_iovec_memset(iov, task.ret, 0, iov->size - task.ret);
2886542aa9cSPeter Lieven     }
2896542aa9cSPeter Lieven 
2906542aa9cSPeter Lieven     return 0;
2916542aa9cSPeter Lieven }
2926542aa9cSPeter Lieven 
2936542aa9cSPeter Lieven static int coroutine_fn nfs_co_writev(BlockDriverState *bs,
2946542aa9cSPeter Lieven                                         int64_t sector_num, int nb_sectors,
2956542aa9cSPeter Lieven                                         QEMUIOVector *iov)
2966542aa9cSPeter Lieven {
2976542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
2986542aa9cSPeter Lieven     NFSRPC task;
2996542aa9cSPeter Lieven     char *buf = NULL;
3006542aa9cSPeter Lieven 
301d746427aSPaolo Bonzini     nfs_co_init_task(bs, &task);
3026542aa9cSPeter Lieven 
3032347dd7bSKevin Wolf     buf = g_try_malloc(nb_sectors * BDRV_SECTOR_SIZE);
3042347dd7bSKevin Wolf     if (nb_sectors && buf == NULL) {
3052347dd7bSKevin Wolf         return -ENOMEM;
3062347dd7bSKevin Wolf     }
3072347dd7bSKevin Wolf 
3086542aa9cSPeter Lieven     qemu_iovec_to_buf(iov, 0, buf, nb_sectors * BDRV_SECTOR_SIZE);
3096542aa9cSPeter Lieven 
3106542aa9cSPeter Lieven     if (nfs_pwrite_async(client->context, client->fh,
3116542aa9cSPeter Lieven                          sector_num * BDRV_SECTOR_SIZE,
3126542aa9cSPeter Lieven                          nb_sectors * BDRV_SECTOR_SIZE,
3136542aa9cSPeter Lieven                          buf, nfs_co_generic_cb, &task) != 0) {
3146542aa9cSPeter Lieven         g_free(buf);
3156542aa9cSPeter Lieven         return -ENOMEM;
3166542aa9cSPeter Lieven     }
3176542aa9cSPeter Lieven 
3186542aa9cSPeter Lieven     nfs_set_events(client);
319aa92d6c4SPaolo Bonzini     while (!task.complete) {
3206542aa9cSPeter Lieven         qemu_coroutine_yield();
3216542aa9cSPeter Lieven     }
3226542aa9cSPeter Lieven 
3236542aa9cSPeter Lieven     g_free(buf);
3246542aa9cSPeter Lieven 
3256542aa9cSPeter Lieven     if (task.ret != nb_sectors * BDRV_SECTOR_SIZE) {
3266542aa9cSPeter Lieven         return task.ret < 0 ? task.ret : -EIO;
3276542aa9cSPeter Lieven     }
3286542aa9cSPeter Lieven 
3296542aa9cSPeter Lieven     return 0;
3306542aa9cSPeter Lieven }
3316542aa9cSPeter Lieven 
3326542aa9cSPeter Lieven static int coroutine_fn nfs_co_flush(BlockDriverState *bs)
3336542aa9cSPeter Lieven {
3346542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
3356542aa9cSPeter Lieven     NFSRPC task;
3366542aa9cSPeter Lieven 
337d746427aSPaolo Bonzini     nfs_co_init_task(bs, &task);
3386542aa9cSPeter Lieven 
3396542aa9cSPeter Lieven     if (nfs_fsync_async(client->context, client->fh, nfs_co_generic_cb,
3406542aa9cSPeter Lieven                         &task) != 0) {
3416542aa9cSPeter Lieven         return -ENOMEM;
3426542aa9cSPeter Lieven     }
3436542aa9cSPeter Lieven 
3446542aa9cSPeter Lieven     nfs_set_events(client);
345aa92d6c4SPaolo Bonzini     while (!task.complete) {
3466542aa9cSPeter Lieven         qemu_coroutine_yield();
3476542aa9cSPeter Lieven     }
3486542aa9cSPeter Lieven 
3496542aa9cSPeter Lieven     return task.ret;
3506542aa9cSPeter Lieven }
3516542aa9cSPeter Lieven 
3526542aa9cSPeter Lieven static QemuOptsList runtime_opts = {
3536542aa9cSPeter Lieven     .name = "nfs",
3546542aa9cSPeter Lieven     .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head),
3556542aa9cSPeter Lieven     .desc = {
3566542aa9cSPeter Lieven         {
35794d6a7a7SAshijeet Acharya             .name = "path",
3586542aa9cSPeter Lieven             .type = QEMU_OPT_STRING,
35994d6a7a7SAshijeet Acharya             .help = "Path of the image on the host",
36094d6a7a7SAshijeet Acharya         },
36194d6a7a7SAshijeet Acharya         {
36294d6a7a7SAshijeet Acharya             .name = "uid",
36394d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
36494d6a7a7SAshijeet Acharya             .help = "UID value to use when talking to the server",
36594d6a7a7SAshijeet Acharya         },
36694d6a7a7SAshijeet Acharya         {
36794d6a7a7SAshijeet Acharya             .name = "gid",
36894d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
36994d6a7a7SAshijeet Acharya             .help = "GID value to use when talking to the server",
37094d6a7a7SAshijeet Acharya         },
37194d6a7a7SAshijeet Acharya         {
37294d6a7a7SAshijeet Acharya             .name = "tcp-syncnt",
37394d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
37494d6a7a7SAshijeet Acharya             .help = "Number of SYNs to send during the session establish",
37594d6a7a7SAshijeet Acharya         },
37694d6a7a7SAshijeet Acharya         {
37794d6a7a7SAshijeet Acharya             .name = "readahead",
37894d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
37994d6a7a7SAshijeet Acharya             .help = "Set the readahead size in bytes",
38094d6a7a7SAshijeet Acharya         },
38194d6a7a7SAshijeet Acharya         {
38294d6a7a7SAshijeet Acharya             .name = "pagecache",
38394d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
38494d6a7a7SAshijeet Acharya             .help = "Set the pagecache size in bytes",
38594d6a7a7SAshijeet Acharya         },
38694d6a7a7SAshijeet Acharya         {
38794d6a7a7SAshijeet Acharya             .name = "debug",
38894d6a7a7SAshijeet Acharya             .type = QEMU_OPT_NUMBER,
38994d6a7a7SAshijeet Acharya             .help = "Set the NFS debug level (max 2)",
3906542aa9cSPeter Lieven         },
3916542aa9cSPeter Lieven         { /* end of list */ }
3926542aa9cSPeter Lieven     },
3936542aa9cSPeter Lieven };
3946542aa9cSPeter Lieven 
395471799d1SStefan Hajnoczi static void nfs_detach_aio_context(BlockDriverState *bs)
396471799d1SStefan Hajnoczi {
397471799d1SStefan Hajnoczi     NFSClient *client = bs->opaque;
398471799d1SStefan Hajnoczi 
399dca21ef2SFam Zheng     aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
400f6a51c84SStefan Hajnoczi                        false, NULL, NULL, NULL, NULL);
401471799d1SStefan Hajnoczi     client->events = 0;
402471799d1SStefan Hajnoczi }
403471799d1SStefan Hajnoczi 
404471799d1SStefan Hajnoczi static void nfs_attach_aio_context(BlockDriverState *bs,
405471799d1SStefan Hajnoczi                                    AioContext *new_context)
406471799d1SStefan Hajnoczi {
407471799d1SStefan Hajnoczi     NFSClient *client = bs->opaque;
408471799d1SStefan Hajnoczi 
409471799d1SStefan Hajnoczi     client->aio_context = new_context;
410471799d1SStefan Hajnoczi     nfs_set_events(client);
411471799d1SStefan Hajnoczi }
412471799d1SStefan Hajnoczi 
4136542aa9cSPeter Lieven static void nfs_client_close(NFSClient *client)
4146542aa9cSPeter Lieven {
4156542aa9cSPeter Lieven     if (client->context) {
4166542aa9cSPeter Lieven         if (client->fh) {
4176542aa9cSPeter Lieven             nfs_close(client->context, client->fh);
4186542aa9cSPeter Lieven         }
419dca21ef2SFam Zheng         aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
420f6a51c84SStefan Hajnoczi                            false, NULL, NULL, NULL, NULL);
4216542aa9cSPeter Lieven         nfs_destroy_context(client->context);
4226542aa9cSPeter Lieven     }
4236542aa9cSPeter Lieven     memset(client, 0, sizeof(NFSClient));
4246542aa9cSPeter Lieven }
4256542aa9cSPeter Lieven 
4266542aa9cSPeter Lieven static void nfs_file_close(BlockDriverState *bs)
4276542aa9cSPeter Lieven {
4286542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
4296542aa9cSPeter Lieven     nfs_client_close(client);
4306542aa9cSPeter Lieven }
4316542aa9cSPeter Lieven 
43294d6a7a7SAshijeet Acharya static NFSServer *nfs_config(QDict *options, Error **errp)
43394d6a7a7SAshijeet Acharya {
43494d6a7a7SAshijeet Acharya     NFSServer *server = NULL;
43594d6a7a7SAshijeet Acharya     QDict *addr = NULL;
43694d6a7a7SAshijeet Acharya     QObject *crumpled_addr = NULL;
43794d6a7a7SAshijeet Acharya     Visitor *iv = NULL;
43894d6a7a7SAshijeet Acharya     Error *local_error = NULL;
43994d6a7a7SAshijeet Acharya 
44094d6a7a7SAshijeet Acharya     qdict_extract_subqdict(options, &addr, "server.");
44194d6a7a7SAshijeet Acharya     if (!qdict_size(addr)) {
44294d6a7a7SAshijeet Acharya         error_setg(errp, "NFS server address missing");
44394d6a7a7SAshijeet Acharya         goto out;
44494d6a7a7SAshijeet Acharya     }
44594d6a7a7SAshijeet Acharya 
44694d6a7a7SAshijeet Acharya     crumpled_addr = qdict_crumple(addr, errp);
44794d6a7a7SAshijeet Acharya     if (!crumpled_addr) {
44894d6a7a7SAshijeet Acharya         goto out;
44994d6a7a7SAshijeet Acharya     }
45094d6a7a7SAshijeet Acharya 
45194d6a7a7SAshijeet Acharya     iv = qobject_input_visitor_new(crumpled_addr, true);
45294d6a7a7SAshijeet Acharya     visit_type_NFSServer(iv, NULL, &server, &local_error);
45394d6a7a7SAshijeet Acharya     if (local_error) {
45494d6a7a7SAshijeet Acharya         error_propagate(errp, local_error);
45594d6a7a7SAshijeet Acharya         goto out;
45694d6a7a7SAshijeet Acharya     }
45794d6a7a7SAshijeet Acharya 
45894d6a7a7SAshijeet Acharya out:
45994d6a7a7SAshijeet Acharya     QDECREF(addr);
46094d6a7a7SAshijeet Acharya     qobject_decref(crumpled_addr);
46194d6a7a7SAshijeet Acharya     visit_free(iv);
46294d6a7a7SAshijeet Acharya     return server;
46394d6a7a7SAshijeet Acharya }
46494d6a7a7SAshijeet Acharya 
46594d6a7a7SAshijeet Acharya 
46694d6a7a7SAshijeet Acharya static int64_t nfs_client_open(NFSClient *client, QDict *options,
46738f8d5e0SPeter Lieven                                int flags, Error **errp, int open_flags)
4686542aa9cSPeter Lieven {
46994d6a7a7SAshijeet Acharya     int ret = -EINVAL;
47094d6a7a7SAshijeet Acharya     QemuOpts *opts = NULL;
47194d6a7a7SAshijeet Acharya     Error *local_err = NULL;
4726542aa9cSPeter Lieven     struct stat st;
4736542aa9cSPeter Lieven     char *file = NULL, *strp = NULL;
4746542aa9cSPeter Lieven 
47594d6a7a7SAshijeet Acharya     opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort);
47694d6a7a7SAshijeet Acharya     qemu_opts_absorb_qdict(opts, options, &local_err);
47794d6a7a7SAshijeet Acharya     if (local_err) {
47894d6a7a7SAshijeet Acharya         error_propagate(errp, local_err);
47994d6a7a7SAshijeet Acharya         ret = -EINVAL;
4806542aa9cSPeter Lieven         goto fail;
4816542aa9cSPeter Lieven     }
48294d6a7a7SAshijeet Acharya 
48394d6a7a7SAshijeet Acharya     client->path = g_strdup(qemu_opt_get(opts, "path"));
48494d6a7a7SAshijeet Acharya     if (!client->path) {
48594d6a7a7SAshijeet Acharya         ret = -EINVAL;
48694d6a7a7SAshijeet Acharya         error_setg(errp, "No path was specified");
4875f4d5e1aSMax Reitz         goto fail;
4885f4d5e1aSMax Reitz     }
48994d6a7a7SAshijeet Acharya 
49094d6a7a7SAshijeet Acharya     strp = strrchr(client->path, '/');
4916542aa9cSPeter Lieven     if (strp == NULL) {
4926542aa9cSPeter Lieven         error_setg(errp, "Invalid URL specified");
4936542aa9cSPeter Lieven         goto fail;
4946542aa9cSPeter Lieven     }
4956542aa9cSPeter Lieven     file = g_strdup(strp);
4966542aa9cSPeter Lieven     *strp = 0;
4976542aa9cSPeter Lieven 
49894d6a7a7SAshijeet Acharya     /* Pop the config into our state object, Exit if invalid */
49994d6a7a7SAshijeet Acharya     client->server = nfs_config(options, errp);
50094d6a7a7SAshijeet Acharya     if (!client->server) {
50194d6a7a7SAshijeet Acharya         ret = -EINVAL;
50294d6a7a7SAshijeet Acharya         goto fail;
50394d6a7a7SAshijeet Acharya     }
50494d6a7a7SAshijeet Acharya 
5056542aa9cSPeter Lieven     client->context = nfs_init_context();
5066542aa9cSPeter Lieven     if (client->context == NULL) {
5076542aa9cSPeter Lieven         error_setg(errp, "Failed to init NFS context");
5086542aa9cSPeter Lieven         goto fail;
5096542aa9cSPeter Lieven     }
5106542aa9cSPeter Lieven 
51194d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "uid")) {
51294d6a7a7SAshijeet Acharya         client->uid = qemu_opt_get_number(opts, "uid", 0);
51394d6a7a7SAshijeet Acharya         nfs_set_uid(client->context, client->uid);
5146542aa9cSPeter Lieven     }
51594d6a7a7SAshijeet Acharya 
51694d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "gid")) {
51794d6a7a7SAshijeet Acharya         client->gid = qemu_opt_get_number(opts, "gid", 0);
51894d6a7a7SAshijeet Acharya         nfs_set_gid(client->context, client->gid);
5197c24384bSPeter Lieven     }
52094d6a7a7SAshijeet Acharya 
52194d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "tcp-syncnt")) {
52294d6a7a7SAshijeet Acharya         client->tcp_syncnt = qemu_opt_get_number(opts, "tcp-syncnt", 0);
52394d6a7a7SAshijeet Acharya         nfs_set_tcp_syncnt(client->context, client->tcp_syncnt);
52494d6a7a7SAshijeet Acharya     }
52594d6a7a7SAshijeet Acharya 
526f42ca3caSPeter Lieven #ifdef LIBNFS_FEATURE_READAHEAD
52794d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "readahead")) {
52838f8d5e0SPeter Lieven         if (open_flags & BDRV_O_NOCACHE) {
52938f8d5e0SPeter Lieven             error_setg(errp, "Cannot enable NFS readahead "
53038f8d5e0SPeter Lieven                              "if cache.direct = on");
53138f8d5e0SPeter Lieven             goto fail;
53238f8d5e0SPeter Lieven         }
53394d6a7a7SAshijeet Acharya         client->readahead = qemu_opt_get_number(opts, "readahead", 0);
53494d6a7a7SAshijeet Acharya         if (client->readahead > QEMU_NFS_MAX_READAHEAD_SIZE) {
53529c838cdSPeter Lieven             error_report("NFS Warning: Truncating NFS readahead "
53629c838cdSPeter Lieven                          "size to %d", QEMU_NFS_MAX_READAHEAD_SIZE);
53794d6a7a7SAshijeet Acharya             client->readahead = QEMU_NFS_MAX_READAHEAD_SIZE;
53829c838cdSPeter Lieven         }
53994d6a7a7SAshijeet Acharya         nfs_set_readahead(client->context, client->readahead);
540d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
541d99b26c4SPeter Lieven         nfs_set_pagecache_ttl(client->context, 0);
542d99b26c4SPeter Lieven #endif
543d99b26c4SPeter Lieven         client->cache_used = true;
54494d6a7a7SAshijeet Acharya     }
545d99b26c4SPeter Lieven #endif
54694d6a7a7SAshijeet Acharya 
547d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
54894d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "pagecache")) {
549d99b26c4SPeter Lieven         if (open_flags & BDRV_O_NOCACHE) {
550d99b26c4SPeter Lieven             error_setg(errp, "Cannot enable NFS pagecache "
551d99b26c4SPeter Lieven                              "if cache.direct = on");
552d99b26c4SPeter Lieven             goto fail;
553d99b26c4SPeter Lieven         }
55494d6a7a7SAshijeet Acharya         client->pagecache = qemu_opt_get_number(opts, "pagecache", 0);
55594d6a7a7SAshijeet Acharya         if (client->pagecache > QEMU_NFS_MAX_PAGECACHE_SIZE) {
556d99b26c4SPeter Lieven             error_report("NFS Warning: Truncating NFS pagecache "
557d99b26c4SPeter Lieven                          "size to %d pages", QEMU_NFS_MAX_PAGECACHE_SIZE);
55894d6a7a7SAshijeet Acharya             client->pagecache = QEMU_NFS_MAX_PAGECACHE_SIZE;
559d99b26c4SPeter Lieven         }
56094d6a7a7SAshijeet Acharya         nfs_set_pagecache(client->context, client->pagecache);
561d99b26c4SPeter Lieven         nfs_set_pagecache_ttl(client->context, 0);
56238f8d5e0SPeter Lieven         client->cache_used = true;
56394d6a7a7SAshijeet Acharya     }
564f42ca3caSPeter Lieven #endif
56594d6a7a7SAshijeet Acharya 
5667725b8bfSPeter Lieven #ifdef LIBNFS_FEATURE_DEBUG
56794d6a7a7SAshijeet Acharya     if (qemu_opt_get(opts, "debug")) {
56894d6a7a7SAshijeet Acharya         client->debug = qemu_opt_get_number(opts, "debug", 0);
5697725b8bfSPeter Lieven         /* limit the maximum debug level to avoid potential flooding
5707725b8bfSPeter Lieven          * of our log files. */
57194d6a7a7SAshijeet Acharya         if (client->debug > QEMU_NFS_MAX_DEBUG_LEVEL) {
5727725b8bfSPeter Lieven             error_report("NFS Warning: Limiting NFS debug level "
5737725b8bfSPeter Lieven                          "to %d", QEMU_NFS_MAX_DEBUG_LEVEL);
57494d6a7a7SAshijeet Acharya             client->debug = QEMU_NFS_MAX_DEBUG_LEVEL;
5757725b8bfSPeter Lieven         }
57694d6a7a7SAshijeet Acharya         nfs_set_debug(client->context, client->debug);
57794d6a7a7SAshijeet Acharya     }
5787725b8bfSPeter Lieven #endif
5796542aa9cSPeter Lieven 
58094d6a7a7SAshijeet Acharya     ret = nfs_mount(client->context, client->server->host, client->path);
5816542aa9cSPeter Lieven     if (ret < 0) {
5826542aa9cSPeter Lieven         error_setg(errp, "Failed to mount nfs share: %s",
5836542aa9cSPeter Lieven                    nfs_get_error(client->context));
5846542aa9cSPeter Lieven         goto fail;
5856542aa9cSPeter Lieven     }
5866542aa9cSPeter Lieven 
5876542aa9cSPeter Lieven     if (flags & O_CREAT) {
5886542aa9cSPeter Lieven         ret = nfs_creat(client->context, file, 0600, &client->fh);
5896542aa9cSPeter Lieven         if (ret < 0) {
5906542aa9cSPeter Lieven             error_setg(errp, "Failed to create file: %s",
5916542aa9cSPeter Lieven                        nfs_get_error(client->context));
5926542aa9cSPeter Lieven             goto fail;
5936542aa9cSPeter Lieven         }
5946542aa9cSPeter Lieven     } else {
5956542aa9cSPeter Lieven         ret = nfs_open(client->context, file, flags, &client->fh);
5966542aa9cSPeter Lieven         if (ret < 0) {
5976542aa9cSPeter Lieven             error_setg(errp, "Failed to open file : %s",
5986542aa9cSPeter Lieven                        nfs_get_error(client->context));
5996542aa9cSPeter Lieven             goto fail;
6006542aa9cSPeter Lieven         }
6016542aa9cSPeter Lieven     }
6026542aa9cSPeter Lieven 
6036542aa9cSPeter Lieven     ret = nfs_fstat(client->context, client->fh, &st);
6046542aa9cSPeter Lieven     if (ret < 0) {
6056542aa9cSPeter Lieven         error_setg(errp, "Failed to fstat file: %s",
6066542aa9cSPeter Lieven                    nfs_get_error(client->context));
6076542aa9cSPeter Lieven         goto fail;
6086542aa9cSPeter Lieven     }
6096542aa9cSPeter Lieven 
6106542aa9cSPeter Lieven     ret = DIV_ROUND_UP(st.st_size, BDRV_SECTOR_SIZE);
61118a8056eSPeter Lieven     client->st_blocks = st.st_blocks;
6126542aa9cSPeter Lieven     client->has_zero_init = S_ISREG(st.st_mode);
61394d6a7a7SAshijeet Acharya     *strp = '/';
6146542aa9cSPeter Lieven     goto out;
61594d6a7a7SAshijeet Acharya 
6166542aa9cSPeter Lieven fail:
6176542aa9cSPeter Lieven     nfs_client_close(client);
6186542aa9cSPeter Lieven out:
61994d6a7a7SAshijeet Acharya     qemu_opts_del(opts);
6206542aa9cSPeter Lieven     g_free(file);
6216542aa9cSPeter Lieven     return ret;
6226542aa9cSPeter Lieven }
6236542aa9cSPeter Lieven 
6246542aa9cSPeter Lieven static int nfs_file_open(BlockDriverState *bs, QDict *options, int flags,
6256542aa9cSPeter Lieven                          Error **errp) {
6266542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
6276542aa9cSPeter Lieven     int64_t ret;
6286542aa9cSPeter Lieven 
629471799d1SStefan Hajnoczi     client->aio_context = bdrv_get_aio_context(bs);
630471799d1SStefan Hajnoczi 
63194d6a7a7SAshijeet Acharya     ret = nfs_client_open(client, options,
6326542aa9cSPeter Lieven                           (flags & BDRV_O_RDWR) ? O_RDWR : O_RDONLY,
63338f8d5e0SPeter Lieven                           errp, bs->open_flags);
6346542aa9cSPeter Lieven     if (ret < 0) {
63594d6a7a7SAshijeet Acharya         return ret;
6366542aa9cSPeter Lieven     }
6376542aa9cSPeter Lieven     bs->total_sectors = ret;
638810f4f86SFam Zheng     ret = 0;
639810f4f86SFam Zheng     return ret;
6406542aa9cSPeter Lieven }
6416542aa9cSPeter Lieven 
642fd752801SMax Reitz static QemuOptsList nfs_create_opts = {
643fd752801SMax Reitz     .name = "nfs-create-opts",
644fd752801SMax Reitz     .head = QTAILQ_HEAD_INITIALIZER(nfs_create_opts.head),
645fd752801SMax Reitz     .desc = {
646fd752801SMax Reitz         {
647fd752801SMax Reitz             .name = BLOCK_OPT_SIZE,
648fd752801SMax Reitz             .type = QEMU_OPT_SIZE,
649fd752801SMax Reitz             .help = "Virtual disk size"
650fd752801SMax Reitz         },
651fd752801SMax Reitz         { /* end of list */ }
652fd752801SMax Reitz     }
653fd752801SMax Reitz };
654fd752801SMax Reitz 
65598c10b81SChunyan Liu static int nfs_file_create(const char *url, QemuOpts *opts, Error **errp)
6566542aa9cSPeter Lieven {
6576542aa9cSPeter Lieven     int ret = 0;
6586542aa9cSPeter Lieven     int64_t total_size = 0;
6595839e53bSMarkus Armbruster     NFSClient *client = g_new0(NFSClient, 1);
66094d6a7a7SAshijeet Acharya     QDict *options = NULL;
6616542aa9cSPeter Lieven 
662471799d1SStefan Hajnoczi     client->aio_context = qemu_get_aio_context();
663471799d1SStefan Hajnoczi 
6646542aa9cSPeter Lieven     /* Read out options */
665c2eb918eSHu Tao     total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
666c2eb918eSHu Tao                           BDRV_SECTOR_SIZE);
6676542aa9cSPeter Lieven 
66894d6a7a7SAshijeet Acharya     options = qdict_new();
66994d6a7a7SAshijeet Acharya     ret = nfs_parse_uri(url, options, errp);
67094d6a7a7SAshijeet Acharya     if (ret < 0) {
67194d6a7a7SAshijeet Acharya         goto out;
67294d6a7a7SAshijeet Acharya     }
67394d6a7a7SAshijeet Acharya 
67494d6a7a7SAshijeet Acharya     ret = nfs_client_open(client, options, O_CREAT, errp, 0);
6756542aa9cSPeter Lieven     if (ret < 0) {
6766542aa9cSPeter Lieven         goto out;
6776542aa9cSPeter Lieven     }
6786542aa9cSPeter Lieven     ret = nfs_ftruncate(client->context, client->fh, total_size);
6796542aa9cSPeter Lieven     nfs_client_close(client);
6806542aa9cSPeter Lieven out:
68107555ba6SKevin Wolf     QDECREF(options);
6826542aa9cSPeter Lieven     g_free(client);
6836542aa9cSPeter Lieven     return ret;
6846542aa9cSPeter Lieven }
6856542aa9cSPeter Lieven 
6866542aa9cSPeter Lieven static int nfs_has_zero_init(BlockDriverState *bs)
6876542aa9cSPeter Lieven {
6886542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
6896542aa9cSPeter Lieven     return client->has_zero_init;
6906542aa9cSPeter Lieven }
6916542aa9cSPeter Lieven 
692d746427aSPaolo Bonzini static void
693d746427aSPaolo Bonzini nfs_get_allocated_file_size_cb(int ret, struct nfs_context *nfs, void *data,
694d746427aSPaolo Bonzini                                void *private_data)
695d746427aSPaolo Bonzini {
696d746427aSPaolo Bonzini     NFSRPC *task = private_data;
697d746427aSPaolo Bonzini     task->ret = ret;
698d746427aSPaolo Bonzini     if (task->ret == 0) {
699d746427aSPaolo Bonzini         memcpy(task->st, data, sizeof(struct stat));
700d746427aSPaolo Bonzini     }
701d746427aSPaolo Bonzini     if (task->ret < 0) {
702d746427aSPaolo Bonzini         error_report("NFS Error: %s", nfs_get_error(nfs));
703d746427aSPaolo Bonzini     }
704d746427aSPaolo Bonzini     task->complete = 1;
705c9d1a561SPaolo Bonzini     bdrv_wakeup(task->bs);
706d746427aSPaolo Bonzini }
707d746427aSPaolo Bonzini 
7086542aa9cSPeter Lieven static int64_t nfs_get_allocated_file_size(BlockDriverState *bs)
7096542aa9cSPeter Lieven {
7106542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
7116542aa9cSPeter Lieven     NFSRPC task = {0};
7126542aa9cSPeter Lieven     struct stat st;
7136542aa9cSPeter Lieven 
71418a8056eSPeter Lieven     if (bdrv_is_read_only(bs) &&
71518a8056eSPeter Lieven         !(bs->open_flags & BDRV_O_NOCACHE)) {
71618a8056eSPeter Lieven         return client->st_blocks * 512;
71718a8056eSPeter Lieven     }
71818a8056eSPeter Lieven 
719d746427aSPaolo Bonzini     task.bs = bs;
7206542aa9cSPeter Lieven     task.st = &st;
721d746427aSPaolo Bonzini     if (nfs_fstat_async(client->context, client->fh, nfs_get_allocated_file_size_cb,
7226542aa9cSPeter Lieven                         &task) != 0) {
7236542aa9cSPeter Lieven         return -ENOMEM;
7246542aa9cSPeter Lieven     }
7256542aa9cSPeter Lieven 
7266542aa9cSPeter Lieven     nfs_set_events(client);
727d746427aSPaolo Bonzini     BDRV_POLL_WHILE(bs, !task.complete);
7286542aa9cSPeter Lieven 
729055c6f91SPeter Lieven     return (task.ret < 0 ? task.ret : st.st_blocks * 512);
7306542aa9cSPeter Lieven }
7316542aa9cSPeter Lieven 
7326542aa9cSPeter Lieven static int nfs_file_truncate(BlockDriverState *bs, int64_t offset)
7336542aa9cSPeter Lieven {
7346542aa9cSPeter Lieven     NFSClient *client = bs->opaque;
7356542aa9cSPeter Lieven     return nfs_ftruncate(client->context, client->fh, offset);
7366542aa9cSPeter Lieven }
7376542aa9cSPeter Lieven 
73818a8056eSPeter Lieven /* Note that this will not re-establish a connection with the NFS server
73918a8056eSPeter Lieven  * - it is effectively a NOP.  */
74018a8056eSPeter Lieven static int nfs_reopen_prepare(BDRVReopenState *state,
74118a8056eSPeter Lieven                               BlockReopenQueue *queue, Error **errp)
74218a8056eSPeter Lieven {
74318a8056eSPeter Lieven     NFSClient *client = state->bs->opaque;
74418a8056eSPeter Lieven     struct stat st;
74518a8056eSPeter Lieven     int ret = 0;
74618a8056eSPeter Lieven 
74718a8056eSPeter Lieven     if (state->flags & BDRV_O_RDWR && bdrv_is_read_only(state->bs)) {
74818a8056eSPeter Lieven         error_setg(errp, "Cannot open a read-only mount as read-write");
74918a8056eSPeter Lieven         return -EACCES;
75018a8056eSPeter Lieven     }
75118a8056eSPeter Lieven 
75238f8d5e0SPeter Lieven     if ((state->flags & BDRV_O_NOCACHE) && client->cache_used) {
753d99b26c4SPeter Lieven         error_setg(errp, "Cannot disable cache if libnfs readahead or"
754d99b26c4SPeter Lieven                          " pagecache is enabled");
75538f8d5e0SPeter Lieven         return -EINVAL;
75638f8d5e0SPeter Lieven     }
75738f8d5e0SPeter Lieven 
75818a8056eSPeter Lieven     /* Update cache for read-only reopens */
75918a8056eSPeter Lieven     if (!(state->flags & BDRV_O_RDWR)) {
76018a8056eSPeter Lieven         ret = nfs_fstat(client->context, client->fh, &st);
76118a8056eSPeter Lieven         if (ret < 0) {
76218a8056eSPeter Lieven             error_setg(errp, "Failed to fstat file: %s",
76318a8056eSPeter Lieven                        nfs_get_error(client->context));
76418a8056eSPeter Lieven             return ret;
76518a8056eSPeter Lieven         }
76618a8056eSPeter Lieven         client->st_blocks = st.st_blocks;
76718a8056eSPeter Lieven     }
76818a8056eSPeter Lieven 
76918a8056eSPeter Lieven     return 0;
77018a8056eSPeter Lieven }
77118a8056eSPeter Lieven 
77294d6a7a7SAshijeet Acharya static void nfs_refresh_filename(BlockDriverState *bs, QDict *options)
77394d6a7a7SAshijeet Acharya {
77494d6a7a7SAshijeet Acharya     NFSClient *client = bs->opaque;
77594d6a7a7SAshijeet Acharya     QDict *opts = qdict_new();
77694d6a7a7SAshijeet Acharya     QObject *server_qdict;
77794d6a7a7SAshijeet Acharya     Visitor *ov;
77894d6a7a7SAshijeet Acharya 
77994d6a7a7SAshijeet Acharya     qdict_put(opts, "driver", qstring_from_str("nfs"));
78094d6a7a7SAshijeet Acharya 
78194d6a7a7SAshijeet Acharya     if (client->uid && !client->gid) {
78294d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
78394d6a7a7SAshijeet Acharya                  "nfs://%s%s?uid=%" PRId64, client->server->host, client->path,
78494d6a7a7SAshijeet Acharya                  client->uid);
78594d6a7a7SAshijeet Acharya     } else if (!client->uid && client->gid) {
78694d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
78794d6a7a7SAshijeet Acharya                  "nfs://%s%s?gid=%" PRId64, client->server->host, client->path,
78894d6a7a7SAshijeet Acharya                  client->gid);
78994d6a7a7SAshijeet Acharya     } else if (client->uid && client->gid) {
79094d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
79194d6a7a7SAshijeet Acharya                  "nfs://%s%s?uid=%" PRId64 "&gid=%" PRId64,
79294d6a7a7SAshijeet Acharya                  client->server->host, client->path, client->uid, client->gid);
79394d6a7a7SAshijeet Acharya     } else {
79494d6a7a7SAshijeet Acharya         snprintf(bs->exact_filename, sizeof(bs->exact_filename),
79594d6a7a7SAshijeet Acharya                  "nfs://%s%s", client->server->host, client->path);
79694d6a7a7SAshijeet Acharya     }
79794d6a7a7SAshijeet Acharya 
79894d6a7a7SAshijeet Acharya     ov = qobject_output_visitor_new(&server_qdict);
79994d6a7a7SAshijeet Acharya     visit_type_NFSServer(ov, NULL, &client->server, &error_abort);
80094d6a7a7SAshijeet Acharya     visit_complete(ov, &server_qdict);
80194d6a7a7SAshijeet Acharya     assert(qobject_type(server_qdict) == QTYPE_QDICT);
80294d6a7a7SAshijeet Acharya 
80394d6a7a7SAshijeet Acharya     qdict_put_obj(opts, "server", server_qdict);
80494d6a7a7SAshijeet Acharya     qdict_put(opts, "path", qstring_from_str(client->path));
80594d6a7a7SAshijeet Acharya 
80694d6a7a7SAshijeet Acharya     if (client->uid) {
80794d6a7a7SAshijeet Acharya         qdict_put(opts, "uid", qint_from_int(client->uid));
80894d6a7a7SAshijeet Acharya     }
80994d6a7a7SAshijeet Acharya     if (client->gid) {
81094d6a7a7SAshijeet Acharya         qdict_put(opts, "gid", qint_from_int(client->gid));
81194d6a7a7SAshijeet Acharya     }
81294d6a7a7SAshijeet Acharya     if (client->tcp_syncnt) {
81394d6a7a7SAshijeet Acharya         qdict_put(opts, "tcp-syncnt",
81494d6a7a7SAshijeet Acharya                       qint_from_int(client->tcp_syncnt));
81594d6a7a7SAshijeet Acharya     }
81694d6a7a7SAshijeet Acharya     if (client->readahead) {
81794d6a7a7SAshijeet Acharya         qdict_put(opts, "readahead",
81894d6a7a7SAshijeet Acharya                       qint_from_int(client->readahead));
81994d6a7a7SAshijeet Acharya     }
82094d6a7a7SAshijeet Acharya     if (client->pagecache) {
82194d6a7a7SAshijeet Acharya         qdict_put(opts, "pagecache",
82294d6a7a7SAshijeet Acharya                       qint_from_int(client->pagecache));
82394d6a7a7SAshijeet Acharya     }
82494d6a7a7SAshijeet Acharya     if (client->debug) {
82594d6a7a7SAshijeet Acharya         qdict_put(opts, "debug", qint_from_int(client->debug));
82694d6a7a7SAshijeet Acharya     }
82794d6a7a7SAshijeet Acharya 
82894d6a7a7SAshijeet Acharya     visit_free(ov);
82994d6a7a7SAshijeet Acharya     qdict_flatten(opts);
83094d6a7a7SAshijeet Acharya     bs->full_open_options = opts;
83194d6a7a7SAshijeet Acharya }
83294d6a7a7SAshijeet Acharya 
833d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
834d99b26c4SPeter Lieven static void nfs_invalidate_cache(BlockDriverState *bs,
835d99b26c4SPeter Lieven                                  Error **errp)
836d99b26c4SPeter Lieven {
837d99b26c4SPeter Lieven     NFSClient *client = bs->opaque;
838d99b26c4SPeter Lieven     nfs_pagecache_invalidate(client->context, client->fh);
839d99b26c4SPeter Lieven }
840d99b26c4SPeter Lieven #endif
841d99b26c4SPeter Lieven 
8426542aa9cSPeter Lieven static BlockDriver bdrv_nfs = {
8436542aa9cSPeter Lieven     .format_name                    = "nfs",
8446542aa9cSPeter Lieven     .protocol_name                  = "nfs",
8456542aa9cSPeter Lieven 
8466542aa9cSPeter Lieven     .instance_size                  = sizeof(NFSClient),
84794d6a7a7SAshijeet Acharya     .bdrv_parse_filename            = nfs_parse_filename,
848fd752801SMax Reitz     .create_opts                    = &nfs_create_opts,
849fd752801SMax Reitz 
8506542aa9cSPeter Lieven     .bdrv_has_zero_init             = nfs_has_zero_init,
8516542aa9cSPeter Lieven     .bdrv_get_allocated_file_size   = nfs_get_allocated_file_size,
8526542aa9cSPeter Lieven     .bdrv_truncate                  = nfs_file_truncate,
8536542aa9cSPeter Lieven 
8546542aa9cSPeter Lieven     .bdrv_file_open                 = nfs_file_open,
8556542aa9cSPeter Lieven     .bdrv_close                     = nfs_file_close,
856c282e1fdSChunyan Liu     .bdrv_create                    = nfs_file_create,
85718a8056eSPeter Lieven     .bdrv_reopen_prepare            = nfs_reopen_prepare,
8586542aa9cSPeter Lieven 
8596542aa9cSPeter Lieven     .bdrv_co_readv                  = nfs_co_readv,
8606542aa9cSPeter Lieven     .bdrv_co_writev                 = nfs_co_writev,
8616542aa9cSPeter Lieven     .bdrv_co_flush_to_disk          = nfs_co_flush,
862471799d1SStefan Hajnoczi 
863471799d1SStefan Hajnoczi     .bdrv_detach_aio_context        = nfs_detach_aio_context,
864471799d1SStefan Hajnoczi     .bdrv_attach_aio_context        = nfs_attach_aio_context,
86594d6a7a7SAshijeet Acharya     .bdrv_refresh_filename          = nfs_refresh_filename,
866d99b26c4SPeter Lieven 
867d99b26c4SPeter Lieven #ifdef LIBNFS_FEATURE_PAGECACHE
868d99b26c4SPeter Lieven     .bdrv_invalidate_cache          = nfs_invalidate_cache,
869d99b26c4SPeter Lieven #endif
8706542aa9cSPeter Lieven };
8716542aa9cSPeter Lieven 
8726542aa9cSPeter Lieven static void nfs_block_init(void)
8736542aa9cSPeter Lieven {
8746542aa9cSPeter Lieven     bdrv_register(&bdrv_nfs);
8756542aa9cSPeter Lieven }
8766542aa9cSPeter Lieven 
8776542aa9cSPeter Lieven block_init(nfs_block_init);
878