]> git.proxmox.com Git - mirror_qemu.git/blame - block/nfs.c
Include qemu/module.h where needed, drop it from qemu-common.h
[mirror_qemu.git] / block / nfs.c
CommitLineData
6542aa9c
PL
1/*
2 * QEMU Block driver for native access to files on NFS shares
3 *
f1a7ff77 4 * Copyright (c) 2014-2017 Peter Lieven <pl@kamp.de>
6542aa9c
PL
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to deal
8 * in the Software without restriction, including without limitation the rights
9 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 * copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
22 * THE SOFTWARE.
23 */
24
80c71a24 25#include "qemu/osdep.h"
6542aa9c
PL
26
27#include <poll.h>
6542aa9c
PL
28#include "qemu/config-file.h"
29#include "qemu/error-report.h"
d165b8cb 30#include "qapi/error.h"
6542aa9c 31#include "block/block_int.h"
609f45ea 32#include "block/qdict.h"
6542aa9c
PL
33#include "trace.h"
34#include "qemu/iov.h"
0b8fa32f 35#include "qemu/module.h"
922a01a0 36#include "qemu/option.h"
6542aa9c 37#include "qemu/uri.h"
0d94b746 38#include "qemu/cutils.h"
6542aa9c 39#include "sysemu/sysemu.h"
9af23989 40#include "qapi/qapi-visit-block-core.h"
94d6a7a7 41#include "qapi/qmp/qdict.h"
94d6a7a7 42#include "qapi/qmp/qstring.h"
94d6a7a7
AA
43#include "qapi/qobject-input-visitor.h"
44#include "qapi/qobject-output-visitor.h"
6542aa9c
PL
45#include <nfsc/libnfs.h>
46
94d6a7a7 47
29c838cd 48#define QEMU_NFS_MAX_READAHEAD_SIZE 1048576
d99b26c4 49#define QEMU_NFS_MAX_PAGECACHE_SIZE (8388608 / NFS_BLKSIZE)
7725b8bf 50#define QEMU_NFS_MAX_DEBUG_LEVEL 2
29c838cd 51
6542aa9c
PL
52typedef struct NFSClient {
53 struct nfs_context *context;
54 struct nfsfh *fh;
55 int events;
56 bool has_zero_init;
471799d1 57 AioContext *aio_context;
37d1e4d9 58 QemuMutex mutex;
18a8056e 59 blkcnt_t st_blocks;
38f8d5e0 60 bool cache_used;
94d6a7a7
AA
61 NFSServer *server;
62 char *path;
63 int64_t uid, gid, tcp_syncnt, readahead, pagecache, debug;
6542aa9c
PL
64} NFSClient;
65
66typedef struct NFSRPC {
d746427a 67 BlockDriverState *bs;
6542aa9c
PL
68 int ret;
69 int complete;
70 QEMUIOVector *iov;
71 struct stat *st;
72 Coroutine *co;
471799d1 73 NFSClient *client;
6542aa9c
PL
74} NFSRPC;
75
94d6a7a7
AA
76static int nfs_parse_uri(const char *filename, QDict *options, Error **errp)
77{
78 URI *uri = NULL;
79 QueryParams *qp = NULL;
80 int ret = -EINVAL, i;
81
82 uri = uri_parse(filename);
83 if (!uri) {
84 error_setg(errp, "Invalid URI specified");
85 goto out;
86 }
f69165a8 87 if (g_strcmp0(uri->scheme, "nfs") != 0) {
94d6a7a7
AA
88 error_setg(errp, "URI scheme must be 'nfs'");
89 goto out;
90 }
91
92 if (!uri->server) {
93 error_setg(errp, "missing hostname in URI");
94 goto out;
95 }
96
97 if (!uri->path) {
98 error_setg(errp, "missing file path in URI");
99 goto out;
100 }
101
102 qp = query_params_parse(uri->query);
103 if (!qp) {
104 error_setg(errp, "could not parse query parameters");
105 goto out;
106 }
107
46f5ac20
EB
108 qdict_put_str(options, "server.host", uri->server);
109 qdict_put_str(options, "server.type", "inet");
110 qdict_put_str(options, "path", uri->path);
94d6a7a7
AA
111
112 for (i = 0; i < qp->n; i++) {
8d20abe8 113 unsigned long long val;
94d6a7a7
AA
114 if (!qp->p[i].value) {
115 error_setg(errp, "Value for NFS parameter expected: %s",
116 qp->p[i].name);
117 goto out;
118 }
8d20abe8 119 if (parse_uint_full(qp->p[i].value, &val, 0)) {
94d6a7a7
AA
120 error_setg(errp, "Illegal value for NFS parameter: %s",
121 qp->p[i].name);
122 goto out;
123 }
124 if (!strcmp(qp->p[i].name, "uid")) {
46f5ac20 125 qdict_put_str(options, "user", qp->p[i].value);
94d6a7a7 126 } else if (!strcmp(qp->p[i].name, "gid")) {
46f5ac20 127 qdict_put_str(options, "group", qp->p[i].value);
94d6a7a7 128 } else if (!strcmp(qp->p[i].name, "tcp-syncnt")) {
46f5ac20 129 qdict_put_str(options, "tcp-syn-count", qp->p[i].value);
94d6a7a7 130 } else if (!strcmp(qp->p[i].name, "readahead")) {
46f5ac20 131 qdict_put_str(options, "readahead-size", qp->p[i].value);
94d6a7a7 132 } else if (!strcmp(qp->p[i].name, "pagecache")) {
46f5ac20 133 qdict_put_str(options, "page-cache-size", qp->p[i].value);
94d6a7a7 134 } else if (!strcmp(qp->p[i].name, "debug")) {
46f5ac20 135 qdict_put_str(options, "debug", qp->p[i].value);
94d6a7a7
AA
136 } else {
137 error_setg(errp, "Unknown NFS parameter name: %s",
138 qp->p[i].name);
139 goto out;
140 }
141 }
142 ret = 0;
143out:
144 if (qp) {
145 query_params_free(qp);
146 }
147 if (uri) {
148 uri_free(uri);
149 }
150 return ret;
151}
152
153static bool nfs_has_filename_options_conflict(QDict *options, Error **errp)
154{
155 const QDictEntry *qe;
156
157 for (qe = qdict_first(options); qe; qe = qdict_next(options, qe)) {
158 if (!strcmp(qe->key, "host") ||
159 !strcmp(qe->key, "path") ||
160 !strcmp(qe->key, "user") ||
161 !strcmp(qe->key, "group") ||
162 !strcmp(qe->key, "tcp-syn-count") ||
163 !strcmp(qe->key, "readahead-size") ||
164 !strcmp(qe->key, "page-cache-size") ||
7103d916 165 !strcmp(qe->key, "debug") ||
94d6a7a7
AA
166 strstart(qe->key, "server.", NULL))
167 {
168 error_setg(errp, "Option %s cannot be used with a filename",
169 qe->key);
170 return true;
171 }
172 }
173
174 return false;
175}
176
177static void nfs_parse_filename(const char *filename, QDict *options,
178 Error **errp)
179{
180 if (nfs_has_filename_options_conflict(options, errp)) {
181 return;
182 }
183
184 nfs_parse_uri(filename, options, errp);
185}
186
6542aa9c
PL
187static void nfs_process_read(void *arg);
188static void nfs_process_write(void *arg);
189
37d1e4d9 190/* Called with QemuMutex held. */
6542aa9c
PL
191static void nfs_set_events(NFSClient *client)
192{
193 int ev = nfs_which_events(client->context);
194 if (ev != client->events) {
dca21ef2
FZ
195 aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
196 false,
471799d1 197 (ev & POLLIN) ? nfs_process_read : NULL,
f6a51c84
SH
198 (ev & POLLOUT) ? nfs_process_write : NULL,
199 NULL, client);
6542aa9c
PL
200
201 }
202 client->events = ev;
203}
204
205static void nfs_process_read(void *arg)
206{
207 NFSClient *client = arg;
9d456654 208
37d1e4d9 209 qemu_mutex_lock(&client->mutex);
6542aa9c
PL
210 nfs_service(client->context, POLLIN);
211 nfs_set_events(client);
37d1e4d9 212 qemu_mutex_unlock(&client->mutex);
6542aa9c
PL
213}
214
215static void nfs_process_write(void *arg)
216{
217 NFSClient *client = arg;
9d456654 218
37d1e4d9 219 qemu_mutex_lock(&client->mutex);
6542aa9c
PL
220 nfs_service(client->context, POLLOUT);
221 nfs_set_events(client);
37d1e4d9 222 qemu_mutex_unlock(&client->mutex);
6542aa9c
PL
223}
224
d746427a 225static void nfs_co_init_task(BlockDriverState *bs, NFSRPC *task)
6542aa9c
PL
226{
227 *task = (NFSRPC) {
471799d1 228 .co = qemu_coroutine_self(),
d746427a
PB
229 .bs = bs,
230 .client = bs->opaque,
6542aa9c
PL
231 };
232}
233
234static void nfs_co_generic_bh_cb(void *opaque)
235{
236 NFSRPC *task = opaque;
1919631e 237
a2c0fe2f 238 task->complete = 1;
1919631e 239 aio_co_wake(task->co);
6542aa9c
PL
240}
241
37d1e4d9 242/* Called (via nfs_service) with QemuMutex held. */
6542aa9c
PL
243static void
244nfs_co_generic_cb(int ret, struct nfs_context *nfs, void *data,
245 void *private_data)
246{
247 NFSRPC *task = private_data;
6542aa9c 248 task->ret = ret;
d746427a 249 assert(!task->st);
6542aa9c
PL
250 if (task->ret > 0 && task->iov) {
251 if (task->ret <= task->iov->size) {
252 qemu_iovec_from_buf(task->iov, 0, data, task->ret);
253 } else {
254 task->ret = -EIO;
255 }
256 }
20fccb18
PL
257 if (task->ret < 0) {
258 error_report("NFS Error: %s", nfs_get_error(nfs));
259 }
d746427a
PB
260 aio_bh_schedule_oneshot(task->client->aio_context,
261 nfs_co_generic_bh_cb, task);
6542aa9c
PL
262}
263
69785a22
PL
264static int coroutine_fn nfs_co_preadv(BlockDriverState *bs, uint64_t offset,
265 uint64_t bytes, QEMUIOVector *iov,
266 int flags)
6542aa9c
PL
267{
268 NFSClient *client = bs->opaque;
269 NFSRPC task;
270
d746427a 271 nfs_co_init_task(bs, &task);
6542aa9c
PL
272 task.iov = iov;
273
37d1e4d9 274 qemu_mutex_lock(&client->mutex);
6542aa9c 275 if (nfs_pread_async(client->context, client->fh,
69785a22 276 offset, bytes, nfs_co_generic_cb, &task) != 0) {
37d1e4d9 277 qemu_mutex_unlock(&client->mutex);
6542aa9c
PL
278 return -ENOMEM;
279 }
280
aa92d6c4 281 nfs_set_events(client);
37d1e4d9 282 qemu_mutex_unlock(&client->mutex);
6542aa9c 283 while (!task.complete) {
6542aa9c
PL
284 qemu_coroutine_yield();
285 }
286
287 if (task.ret < 0) {
288 return task.ret;
289 }
290
291 /* zero pad short reads */
292 if (task.ret < iov->size) {
293 qemu_iovec_memset(iov, task.ret, 0, iov->size - task.ret);
294 }
295
296 return 0;
297}
298
69785a22
PL
299static int coroutine_fn nfs_co_pwritev(BlockDriverState *bs, uint64_t offset,
300 uint64_t bytes, QEMUIOVector *iov,
301 int flags)
6542aa9c
PL
302{
303 NFSClient *client = bs->opaque;
304 NFSRPC task;
305 char *buf = NULL;
ef503a84 306 bool my_buffer = false;
6542aa9c 307
d746427a 308 nfs_co_init_task(bs, &task);
6542aa9c 309
ef503a84
PL
310 if (iov->niov != 1) {
311 buf = g_try_malloc(bytes);
312 if (bytes && buf == NULL) {
313 return -ENOMEM;
314 }
315 qemu_iovec_to_buf(iov, 0, buf, bytes);
316 my_buffer = true;
317 } else {
318 buf = iov->iov[0].iov_base;
2347dd7b
KW
319 }
320
37d1e4d9 321 qemu_mutex_lock(&client->mutex);
6542aa9c 322 if (nfs_pwrite_async(client->context, client->fh,
69785a22
PL
323 offset, bytes, buf,
324 nfs_co_generic_cb, &task) != 0) {
37d1e4d9 325 qemu_mutex_unlock(&client->mutex);
ef503a84
PL
326 if (my_buffer) {
327 g_free(buf);
328 }
6542aa9c
PL
329 return -ENOMEM;
330 }
331
aa92d6c4 332 nfs_set_events(client);
37d1e4d9 333 qemu_mutex_unlock(&client->mutex);
6542aa9c 334 while (!task.complete) {
6542aa9c
PL
335 qemu_coroutine_yield();
336 }
337
ef503a84
PL
338 if (my_buffer) {
339 g_free(buf);
340 }
6542aa9c 341
69785a22 342 if (task.ret != bytes) {
6542aa9c
PL
343 return task.ret < 0 ? task.ret : -EIO;
344 }
345
346 return 0;
347}
348
349static int coroutine_fn nfs_co_flush(BlockDriverState *bs)
350{
351 NFSClient *client = bs->opaque;
352 NFSRPC task;
353
d746427a 354 nfs_co_init_task(bs, &task);
6542aa9c 355
37d1e4d9 356 qemu_mutex_lock(&client->mutex);
6542aa9c
PL
357 if (nfs_fsync_async(client->context, client->fh, nfs_co_generic_cb,
358 &task) != 0) {
37d1e4d9 359 qemu_mutex_unlock(&client->mutex);
6542aa9c
PL
360 return -ENOMEM;
361 }
362
aa92d6c4 363 nfs_set_events(client);
37d1e4d9 364 qemu_mutex_unlock(&client->mutex);
6542aa9c 365 while (!task.complete) {
6542aa9c
PL
366 qemu_coroutine_yield();
367 }
368
369 return task.ret;
370}
371
471799d1
SH
372static void nfs_detach_aio_context(BlockDriverState *bs)
373{
374 NFSClient *client = bs->opaque;
375
dca21ef2 376 aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
f6a51c84 377 false, NULL, NULL, NULL, NULL);
471799d1
SH
378 client->events = 0;
379}
380
381static void nfs_attach_aio_context(BlockDriverState *bs,
382 AioContext *new_context)
383{
384 NFSClient *client = bs->opaque;
385
386 client->aio_context = new_context;
387 nfs_set_events(client);
388}
389
6542aa9c
PL
390static void nfs_client_close(NFSClient *client)
391{
392 if (client->context) {
393 if (client->fh) {
394 nfs_close(client->context, client->fh);
113fe792 395 client->fh = NULL;
6542aa9c 396 }
dca21ef2 397 aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context),
f6a51c84 398 false, NULL, NULL, NULL, NULL);
6542aa9c 399 nfs_destroy_context(client->context);
113fe792 400 client->context = NULL;
6542aa9c 401 }
113fe792
JC
402 g_free(client->path);
403 qemu_mutex_destroy(&client->mutex);
404 qapi_free_NFSServer(client->server);
405 client->server = NULL;
6542aa9c
PL
406}
407
408static void nfs_file_close(BlockDriverState *bs)
409{
410 NFSClient *client = bs->opaque;
411 nfs_client_close(client);
412}
413
c22a0345 414static int64_t nfs_client_open(NFSClient *client, BlockdevOptionsNfs *opts,
cb8d4bf6 415 int flags, int open_flags, Error **errp)
6542aa9c 416{
f1a7ff77 417 int64_t ret = -EINVAL;
6542aa9c 418 struct stat st;
6542aa9c
PL
419 char *file = NULL, *strp = NULL;
420
113fe792 421 qemu_mutex_init(&client->mutex);
94d6a7a7 422
c22a0345 423 client->path = g_strdup(opts->path);
94d6a7a7
AA
424
425 strp = strrchr(client->path, '/');
6542aa9c
PL
426 if (strp == NULL) {
427 error_setg(errp, "Invalid URL specified");
428 goto fail;
429 }
430 file = g_strdup(strp);
431 *strp = 0;
432
c22a0345
KW
433 /* Steal the NFSServer object from opts; set the original pointer to NULL
434 * to avoid use after free and double free. */
435 client->server = opts->server;
436 opts->server = NULL;
94d6a7a7 437
6542aa9c
PL
438 client->context = nfs_init_context();
439 if (client->context == NULL) {
440 error_setg(errp, "Failed to init NFS context");
441 goto fail;
442 }
443
c22a0345
KW
444 if (opts->has_user) {
445 client->uid = opts->user;
94d6a7a7
AA
446 nfs_set_uid(client->context, client->uid);
447 }
448
c22a0345
KW
449 if (opts->has_group) {
450 client->gid = opts->group;
94d6a7a7
AA
451 nfs_set_gid(client->context, client->gid);
452 }
453
c22a0345
KW
454 if (opts->has_tcp_syn_count) {
455 client->tcp_syncnt = opts->tcp_syn_count;
94d6a7a7
AA
456 nfs_set_tcp_syncnt(client->context, client->tcp_syncnt);
457 }
458
459#ifdef LIBNFS_FEATURE_READAHEAD
c22a0345 460 if (opts->has_readahead_size) {
94d6a7a7
AA
461 if (open_flags & BDRV_O_NOCACHE) {
462 error_setg(errp, "Cannot enable NFS readahead "
463 "if cache.direct = on");
6542aa9c
PL
464 goto fail;
465 }
c22a0345 466 client->readahead = opts->readahead_size;
94d6a7a7 467 if (client->readahead > QEMU_NFS_MAX_READAHEAD_SIZE) {
3dc6f869
AF
468 warn_report("Truncating NFS readahead size to %d",
469 QEMU_NFS_MAX_READAHEAD_SIZE);
94d6a7a7 470 client->readahead = QEMU_NFS_MAX_READAHEAD_SIZE;
7c24384b 471 }
94d6a7a7 472 nfs_set_readahead(client->context, client->readahead);
d99b26c4 473#ifdef LIBNFS_FEATURE_PAGECACHE
94d6a7a7 474 nfs_set_pagecache_ttl(client->context, 0);
d99b26c4 475#endif
94d6a7a7
AA
476 client->cache_used = true;
477 }
d99b26c4 478#endif
94d6a7a7 479
d99b26c4 480#ifdef LIBNFS_FEATURE_PAGECACHE
c22a0345 481 if (opts->has_page_cache_size) {
94d6a7a7
AA
482 if (open_flags & BDRV_O_NOCACHE) {
483 error_setg(errp, "Cannot enable NFS pagecache "
484 "if cache.direct = on");
485 goto fail;
486 }
c22a0345 487 client->pagecache = opts->page_cache_size;
94d6a7a7 488 if (client->pagecache > QEMU_NFS_MAX_PAGECACHE_SIZE) {
3dc6f869
AF
489 warn_report("Truncating NFS pagecache size to %d pages",
490 QEMU_NFS_MAX_PAGECACHE_SIZE);
94d6a7a7
AA
491 client->pagecache = QEMU_NFS_MAX_PAGECACHE_SIZE;
492 }
493 nfs_set_pagecache(client->context, client->pagecache);
494 nfs_set_pagecache_ttl(client->context, 0);
495 client->cache_used = true;
496 }
7725b8bf 497#endif
94d6a7a7 498
7725b8bf 499#ifdef LIBNFS_FEATURE_DEBUG
c22a0345
KW
500 if (opts->has_debug) {
501 client->debug = opts->debug;
94d6a7a7
AA
502 /* limit the maximum debug level to avoid potential flooding
503 * of our log files. */
504 if (client->debug > QEMU_NFS_MAX_DEBUG_LEVEL) {
3dc6f869
AF
505 warn_report("Limiting NFS debug level to %d",
506 QEMU_NFS_MAX_DEBUG_LEVEL);
94d6a7a7 507 client->debug = QEMU_NFS_MAX_DEBUG_LEVEL;
6542aa9c 508 }
94d6a7a7 509 nfs_set_debug(client->context, client->debug);
6542aa9c 510 }
94d6a7a7 511#endif
6542aa9c 512
94d6a7a7 513 ret = nfs_mount(client->context, client->server->host, client->path);
6542aa9c
PL
514 if (ret < 0) {
515 error_setg(errp, "Failed to mount nfs share: %s",
516 nfs_get_error(client->context));
517 goto fail;
518 }
519
520 if (flags & O_CREAT) {
521 ret = nfs_creat(client->context, file, 0600, &client->fh);
522 if (ret < 0) {
523 error_setg(errp, "Failed to create file: %s",
524 nfs_get_error(client->context));
525 goto fail;
526 }
527 } else {
528 ret = nfs_open(client->context, file, flags, &client->fh);
529 if (ret < 0) {
530 error_setg(errp, "Failed to open file : %s",
531 nfs_get_error(client->context));
532 goto fail;
533 }
534 }
535
536 ret = nfs_fstat(client->context, client->fh, &st);
537 if (ret < 0) {
538 error_setg(errp, "Failed to fstat file: %s",
539 nfs_get_error(client->context));
540 goto fail;
541 }
542
543 ret = DIV_ROUND_UP(st.st_size, BDRV_SECTOR_SIZE);
18a8056e 544 client->st_blocks = st.st_blocks;
6542aa9c 545 client->has_zero_init = S_ISREG(st.st_mode);
94d6a7a7 546 *strp = '/';
6542aa9c 547 goto out;
94d6a7a7 548
6542aa9c
PL
549fail:
550 nfs_client_close(client);
551out:
6542aa9c
PL
552 g_free(file);
553 return ret;
554}
555
a1a42af4
KW
556static BlockdevOptionsNfs *nfs_options_qdict_to_qapi(QDict *options,
557 Error **errp)
c22a0345
KW
558{
559 BlockdevOptionsNfs *opts = NULL;
c22a0345 560 Visitor *v;
c82be42c 561 const QDictEntry *e;
c22a0345 562 Error *local_err = NULL;
c22a0345 563
af91062e
MA
564 v = qobject_input_visitor_new_flat_confused(options, errp);
565 if (!v) {
a1a42af4 566 return NULL;
c22a0345
KW
567 }
568
c22a0345
KW
569 visit_type_BlockdevOptionsNfs(v, NULL, &opts, &local_err);
570 visit_free(v);
571
572 if (local_err) {
54b7af43 573 error_propagate(errp, local_err);
a1a42af4
KW
574 return NULL;
575 }
576
c82be42c
KW
577 /* Remove the processed options from the QDict (the visitor processes
578 * _all_ options in the QDict) */
579 while ((e = qdict_first(options))) {
580 qdict_del(options, e->key);
581 }
582
a1a42af4
KW
583 return opts;
584}
585
586static int64_t nfs_client_open_qdict(NFSClient *client, QDict *options,
587 int flags, int open_flags, Error **errp)
588{
589 BlockdevOptionsNfs *opts;
590 int ret;
591
592 opts = nfs_options_qdict_to_qapi(options, errp);
593 if (opts == NULL) {
c22a0345
KW
594 ret = -EINVAL;
595 goto fail;
596 }
597
598 ret = nfs_client_open(client, opts, flags, open_flags, errp);
599fail:
c22a0345
KW
600 qapi_free_BlockdevOptionsNfs(opts);
601 return ret;
602}
603
6542aa9c
PL
604static int nfs_file_open(BlockDriverState *bs, QDict *options, int flags,
605 Error **errp) {
606 NFSClient *client = bs->opaque;
607 int64_t ret;
6542aa9c 608
471799d1
SH
609 client->aio_context = bdrv_get_aio_context(bs);
610
c22a0345
KW
611 ret = nfs_client_open_qdict(client, options,
612 (flags & BDRV_O_RDWR) ? O_RDWR : O_RDONLY,
613 bs->open_flags, errp);
6542aa9c 614 if (ret < 0) {
94d6a7a7 615 return ret;
6542aa9c 616 }
113fe792 617
6542aa9c 618 bs->total_sectors = ret;
810f4f86 619 ret = 0;
810f4f86 620 return ret;
6542aa9c
PL
621}
622
fd752801
HR
623static QemuOptsList nfs_create_opts = {
624 .name = "nfs-create-opts",
625 .head = QTAILQ_HEAD_INITIALIZER(nfs_create_opts.head),
626 .desc = {
627 {
628 .name = BLOCK_OPT_SIZE,
629 .type = QEMU_OPT_SIZE,
630 .help = "Virtual disk size"
631 },
632 { /* end of list */ }
633 }
634};
635
a1a42af4 636static int nfs_file_co_create(BlockdevCreateOptions *options, Error **errp)
6542aa9c 637{
a1a42af4 638 BlockdevCreateOptionsNfs *opts = &options->u.nfs;
5839e53b 639 NFSClient *client = g_new0(NFSClient, 1);
a1a42af4
KW
640 int ret;
641
642 assert(options->driver == BLOCKDEV_DRIVER_NFS);
6542aa9c 643
471799d1
SH
644 client->aio_context = qemu_get_aio_context();
645
a1a42af4
KW
646 ret = nfs_client_open(client, opts->location, O_CREAT, 0, errp);
647 if (ret < 0) {
648 goto out;
649 }
650 ret = nfs_ftruncate(client->context, client->fh, opts->size);
651 nfs_client_close(client);
652
653out:
654 g_free(client);
655 return ret;
656}
657
658static int coroutine_fn nfs_file_co_create_opts(const char *url, QemuOpts *opts,
659 Error **errp)
660{
661 BlockdevCreateOptions *create_options;
662 BlockdevCreateOptionsNfs *nfs_opts;
663 QDict *options;
664 int ret;
665
666 create_options = g_new0(BlockdevCreateOptions, 1);
667 create_options->driver = BLOCKDEV_DRIVER_NFS;
668 nfs_opts = &create_options->u.nfs;
669
6542aa9c 670 /* Read out options */
a1a42af4
KW
671 nfs_opts->size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0),
672 BDRV_SECTOR_SIZE);
6542aa9c 673
94d6a7a7
AA
674 options = qdict_new();
675 ret = nfs_parse_uri(url, options, errp);
676 if (ret < 0) {
677 goto out;
678 }
679
a1a42af4
KW
680 nfs_opts->location = nfs_options_qdict_to_qapi(options, errp);
681 if (nfs_opts->location == NULL) {
682 ret = -EINVAL;
683 goto out;
684 }
685
686 ret = nfs_file_co_create(create_options, errp);
6542aa9c
PL
687 if (ret < 0) {
688 goto out;
689 }
a1a42af4
KW
690
691 ret = 0;
6542aa9c 692out:
cb3e7f08 693 qobject_unref(options);
a1a42af4 694 qapi_free_BlockdevCreateOptions(create_options);
6542aa9c
PL
695 return ret;
696}
697
698static int nfs_has_zero_init(BlockDriverState *bs)
699{
700 NFSClient *client = bs->opaque;
701 return client->has_zero_init;
702}
703
37d1e4d9 704/* Called (via nfs_service) with QemuMutex held. */
d746427a
PB
705static void
706nfs_get_allocated_file_size_cb(int ret, struct nfs_context *nfs, void *data,
707 void *private_data)
708{
709 NFSRPC *task = private_data;
710 task->ret = ret;
711 if (task->ret == 0) {
712 memcpy(task->st, data, sizeof(struct stat));
713 }
714 if (task->ret < 0) {
715 error_report("NFS Error: %s", nfs_get_error(nfs));
716 }
e2a6ae7f
PB
717
718 /* Set task->complete before reading bs->wakeup. */
719 atomic_mb_set(&task->complete, 1);
c9d1a561 720 bdrv_wakeup(task->bs);
d746427a
PB
721}
722
6542aa9c
PL
723static int64_t nfs_get_allocated_file_size(BlockDriverState *bs)
724{
725 NFSClient *client = bs->opaque;
726 NFSRPC task = {0};
727 struct stat st;
728
18a8056e
PL
729 if (bdrv_is_read_only(bs) &&
730 !(bs->open_flags & BDRV_O_NOCACHE)) {
731 return client->st_blocks * 512;
732 }
733
d746427a 734 task.bs = bs;
6542aa9c 735 task.st = &st;
d746427a 736 if (nfs_fstat_async(client->context, client->fh, nfs_get_allocated_file_size_cb,
6542aa9c
PL
737 &task) != 0) {
738 return -ENOMEM;
739 }
740
aa92d6c4 741 nfs_set_events(client);
d746427a 742 BDRV_POLL_WHILE(bs, !task.complete);
6542aa9c 743
055c6f91 744 return (task.ret < 0 ? task.ret : st.st_blocks * 512);
6542aa9c
PL
745}
746
061ca8a3
KW
747static int coroutine_fn
748nfs_file_co_truncate(BlockDriverState *bs, int64_t offset,
749 PreallocMode prealloc, Error **errp)
6542aa9c
PL
750{
751 NFSClient *client = bs->opaque;
f59adb32
HR
752 int ret;
753
8243ccb7
HR
754 if (prealloc != PREALLOC_MODE_OFF) {
755 error_setg(errp, "Unsupported preallocation mode '%s'",
977c736f 756 PreallocMode_str(prealloc));
8243ccb7
HR
757 return -ENOTSUP;
758 }
759
f59adb32
HR
760 ret = nfs_ftruncate(client->context, client->fh, offset);
761 if (ret < 0) {
762 error_setg_errno(errp, -ret, "Failed to truncate file");
763 return ret;
764 }
765
766 return 0;
6542aa9c
PL
767}
768
18a8056e
PL
769/* Note that this will not re-establish a connection with the NFS server
770 * - it is effectively a NOP. */
771static int nfs_reopen_prepare(BDRVReopenState *state,
772 BlockReopenQueue *queue, Error **errp)
773{
774 NFSClient *client = state->bs->opaque;
775 struct stat st;
776 int ret = 0;
777
778 if (state->flags & BDRV_O_RDWR && bdrv_is_read_only(state->bs)) {
779 error_setg(errp, "Cannot open a read-only mount as read-write");
780 return -EACCES;
781 }
782
38f8d5e0 783 if ((state->flags & BDRV_O_NOCACHE) && client->cache_used) {
d99b26c4
PL
784 error_setg(errp, "Cannot disable cache if libnfs readahead or"
785 " pagecache is enabled");
38f8d5e0
PL
786 return -EINVAL;
787 }
788
18a8056e
PL
789 /* Update cache for read-only reopens */
790 if (!(state->flags & BDRV_O_RDWR)) {
791 ret = nfs_fstat(client->context, client->fh, &st);
792 if (ret < 0) {
793 error_setg(errp, "Failed to fstat file: %s",
794 nfs_get_error(client->context));
795 return ret;
796 }
797 client->st_blocks = st.st_blocks;
798 }
799
800 return 0;
801}
802
998b3a1e 803static void nfs_refresh_filename(BlockDriverState *bs)
94d6a7a7
AA
804{
805 NFSClient *client = bs->opaque;
94d6a7a7
AA
806
807 if (client->uid && !client->gid) {
808 snprintf(bs->exact_filename, sizeof(bs->exact_filename),
809 "nfs://%s%s?uid=%" PRId64, client->server->host, client->path,
810 client->uid);
811 } else if (!client->uid && client->gid) {
812 snprintf(bs->exact_filename, sizeof(bs->exact_filename),
813 "nfs://%s%s?gid=%" PRId64, client->server->host, client->path,
814 client->gid);
815 } else if (client->uid && client->gid) {
816 snprintf(bs->exact_filename, sizeof(bs->exact_filename),
817 "nfs://%s%s?uid=%" PRId64 "&gid=%" PRId64,
818 client->server->host, client->path, client->uid, client->gid);
819 } else {
820 snprintf(bs->exact_filename, sizeof(bs->exact_filename),
821 "nfs://%s%s", client->server->host, client->path);
822 }
94d6a7a7
AA
823}
824
0dcbc54a
HR
825static char *nfs_dirname(BlockDriverState *bs, Error **errp)
826{
827 NFSClient *client = bs->opaque;
828
829 if (client->uid || client->gid) {
830 bdrv_refresh_filename(bs);
831 error_setg(errp, "Cannot generate a base directory for NFS node '%s'",
832 bs->filename);
833 return NULL;
834 }
835
836 return g_strdup_printf("nfs://%s%s/", client->server->host, client->path);
837}
838
d99b26c4 839#ifdef LIBNFS_FEATURE_PAGECACHE
2b148f39
PB
840static void coroutine_fn nfs_co_invalidate_cache(BlockDriverState *bs,
841 Error **errp)
d99b26c4
PL
842{
843 NFSClient *client = bs->opaque;
844 nfs_pagecache_invalidate(client->context, client->fh);
845}
846#endif
847
2654267c
HR
848static const char *nfs_strong_runtime_opts[] = {
849 "path",
850 "user",
851 "group",
852 "server.",
853
854 NULL
855};
856
6542aa9c 857static BlockDriver bdrv_nfs = {
471799d1
SH
858 .format_name = "nfs",
859 .protocol_name = "nfs",
860
861 .instance_size = sizeof(NFSClient),
94d6a7a7 862 .bdrv_parse_filename = nfs_parse_filename,
fd752801
HR
863 .create_opts = &nfs_create_opts,
864
471799d1
SH
865 .bdrv_has_zero_init = nfs_has_zero_init,
866 .bdrv_get_allocated_file_size = nfs_get_allocated_file_size,
061ca8a3 867 .bdrv_co_truncate = nfs_file_co_truncate,
471799d1
SH
868
869 .bdrv_file_open = nfs_file_open,
870 .bdrv_close = nfs_file_close,
a1a42af4 871 .bdrv_co_create = nfs_file_co_create,
efc75e2a 872 .bdrv_co_create_opts = nfs_file_co_create_opts,
18a8056e 873 .bdrv_reopen_prepare = nfs_reopen_prepare,
471799d1 874
69785a22
PL
875 .bdrv_co_preadv = nfs_co_preadv,
876 .bdrv_co_pwritev = nfs_co_pwritev,
471799d1
SH
877 .bdrv_co_flush_to_disk = nfs_co_flush,
878
879 .bdrv_detach_aio_context = nfs_detach_aio_context,
880 .bdrv_attach_aio_context = nfs_attach_aio_context,
94d6a7a7 881 .bdrv_refresh_filename = nfs_refresh_filename,
0dcbc54a 882 .bdrv_dirname = nfs_dirname,
d99b26c4 883
2654267c
HR
884 .strong_runtime_opts = nfs_strong_runtime_opts,
885
d99b26c4 886#ifdef LIBNFS_FEATURE_PAGECACHE
2b148f39 887 .bdrv_co_invalidate_cache = nfs_co_invalidate_cache,
d99b26c4 888#endif
6542aa9c
PL
889};
890
891static void nfs_block_init(void)
892{
893 bdrv_register(&bdrv_nfs);
894}
895
896block_init(nfs_block_init);