]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | * QEMU Block driver for native access to files on NFS shares | |
3 | * | |
4 | * Copyright (c) 2014-2017 Peter Lieven <[email protected]> | |
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 | ||
25 | #include "qemu/osdep.h" | |
26 | ||
27 | #include <poll.h> | |
28 | #include "qemu-common.h" | |
29 | #include "qemu/config-file.h" | |
30 | #include "qemu/error-report.h" | |
31 | #include "qapi/error.h" | |
32 | #include "block/block_int.h" | |
33 | #include "trace.h" | |
34 | #include "qemu/iov.h" | |
35 | #include "qemu/uri.h" | |
36 | #include "qemu/cutils.h" | |
37 | #include "sysemu/sysemu.h" | |
38 | #include "qapi/qmp/qdict.h" | |
39 | #include "qapi/qmp/qstring.h" | |
40 | #include "qapi-visit.h" | |
41 | #include "qapi/qobject-input-visitor.h" | |
42 | #include "qapi/qobject-output-visitor.h" | |
43 | #include <nfsc/libnfs.h> | |
44 | ||
45 | ||
46 | #define QEMU_NFS_MAX_READAHEAD_SIZE 1048576 | |
47 | #define QEMU_NFS_MAX_PAGECACHE_SIZE (8388608 / NFS_BLKSIZE) | |
48 | #define QEMU_NFS_MAX_DEBUG_LEVEL 2 | |
49 | ||
50 | typedef struct NFSClient { | |
51 | struct nfs_context *context; | |
52 | struct nfsfh *fh; | |
53 | int events; | |
54 | bool has_zero_init; | |
55 | AioContext *aio_context; | |
56 | QemuMutex mutex; | |
57 | blkcnt_t st_blocks; | |
58 | bool cache_used; | |
59 | NFSServer *server; | |
60 | char *path; | |
61 | int64_t uid, gid, tcp_syncnt, readahead, pagecache, debug; | |
62 | } NFSClient; | |
63 | ||
64 | typedef struct NFSRPC { | |
65 | BlockDriverState *bs; | |
66 | int ret; | |
67 | int complete; | |
68 | QEMUIOVector *iov; | |
69 | struct stat *st; | |
70 | Coroutine *co; | |
71 | NFSClient *client; | |
72 | } NFSRPC; | |
73 | ||
74 | static int nfs_parse_uri(const char *filename, QDict *options, Error **errp) | |
75 | { | |
76 | URI *uri = NULL; | |
77 | QueryParams *qp = NULL; | |
78 | int ret = -EINVAL, i; | |
79 | ||
80 | uri = uri_parse(filename); | |
81 | if (!uri) { | |
82 | error_setg(errp, "Invalid URI specified"); | |
83 | goto out; | |
84 | } | |
85 | if (g_strcmp0(uri->scheme, "nfs") != 0) { | |
86 | error_setg(errp, "URI scheme must be 'nfs'"); | |
87 | goto out; | |
88 | } | |
89 | ||
90 | if (!uri->server) { | |
91 | error_setg(errp, "missing hostname in URI"); | |
92 | goto out; | |
93 | } | |
94 | ||
95 | if (!uri->path) { | |
96 | error_setg(errp, "missing file path in URI"); | |
97 | goto out; | |
98 | } | |
99 | ||
100 | qp = query_params_parse(uri->query); | |
101 | if (!qp) { | |
102 | error_setg(errp, "could not parse query parameters"); | |
103 | goto out; | |
104 | } | |
105 | ||
106 | qdict_put_str(options, "server.host", uri->server); | |
107 | qdict_put_str(options, "server.type", "inet"); | |
108 | qdict_put_str(options, "path", uri->path); | |
109 | ||
110 | for (i = 0; i < qp->n; i++) { | |
111 | unsigned long long val; | |
112 | if (!qp->p[i].value) { | |
113 | error_setg(errp, "Value for NFS parameter expected: %s", | |
114 | qp->p[i].name); | |
115 | goto out; | |
116 | } | |
117 | if (parse_uint_full(qp->p[i].value, &val, 0)) { | |
118 | error_setg(errp, "Illegal value for NFS parameter: %s", | |
119 | qp->p[i].name); | |
120 | goto out; | |
121 | } | |
122 | if (!strcmp(qp->p[i].name, "uid")) { | |
123 | qdict_put_str(options, "user", qp->p[i].value); | |
124 | } else if (!strcmp(qp->p[i].name, "gid")) { | |
125 | qdict_put_str(options, "group", qp->p[i].value); | |
126 | } else if (!strcmp(qp->p[i].name, "tcp-syncnt")) { | |
127 | qdict_put_str(options, "tcp-syn-count", qp->p[i].value); | |
128 | } else if (!strcmp(qp->p[i].name, "readahead")) { | |
129 | qdict_put_str(options, "readahead-size", qp->p[i].value); | |
130 | } else if (!strcmp(qp->p[i].name, "pagecache")) { | |
131 | qdict_put_str(options, "page-cache-size", qp->p[i].value); | |
132 | } else if (!strcmp(qp->p[i].name, "debug")) { | |
133 | qdict_put_str(options, "debug", qp->p[i].value); | |
134 | } else { | |
135 | error_setg(errp, "Unknown NFS parameter name: %s", | |
136 | qp->p[i].name); | |
137 | goto out; | |
138 | } | |
139 | } | |
140 | ret = 0; | |
141 | out: | |
142 | if (qp) { | |
143 | query_params_free(qp); | |
144 | } | |
145 | if (uri) { | |
146 | uri_free(uri); | |
147 | } | |
148 | return ret; | |
149 | } | |
150 | ||
151 | static bool nfs_has_filename_options_conflict(QDict *options, Error **errp) | |
152 | { | |
153 | const QDictEntry *qe; | |
154 | ||
155 | for (qe = qdict_first(options); qe; qe = qdict_next(options, qe)) { | |
156 | if (!strcmp(qe->key, "host") || | |
157 | !strcmp(qe->key, "path") || | |
158 | !strcmp(qe->key, "user") || | |
159 | !strcmp(qe->key, "group") || | |
160 | !strcmp(qe->key, "tcp-syn-count") || | |
161 | !strcmp(qe->key, "readahead-size") || | |
162 | !strcmp(qe->key, "page-cache-size") || | |
163 | !strcmp(qe->key, "debug") || | |
164 | strstart(qe->key, "server.", NULL)) | |
165 | { | |
166 | error_setg(errp, "Option %s cannot be used with a filename", | |
167 | qe->key); | |
168 | return true; | |
169 | } | |
170 | } | |
171 | ||
172 | return false; | |
173 | } | |
174 | ||
175 | static void nfs_parse_filename(const char *filename, QDict *options, | |
176 | Error **errp) | |
177 | { | |
178 | if (nfs_has_filename_options_conflict(options, errp)) { | |
179 | return; | |
180 | } | |
181 | ||
182 | nfs_parse_uri(filename, options, errp); | |
183 | } | |
184 | ||
185 | static void nfs_process_read(void *arg); | |
186 | static void nfs_process_write(void *arg); | |
187 | ||
188 | /* Called with QemuMutex held. */ | |
189 | static void nfs_set_events(NFSClient *client) | |
190 | { | |
191 | int ev = nfs_which_events(client->context); | |
192 | if (ev != client->events) { | |
193 | aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context), | |
194 | false, | |
195 | (ev & POLLIN) ? nfs_process_read : NULL, | |
196 | (ev & POLLOUT) ? nfs_process_write : NULL, | |
197 | NULL, client); | |
198 | ||
199 | } | |
200 | client->events = ev; | |
201 | } | |
202 | ||
203 | static void nfs_process_read(void *arg) | |
204 | { | |
205 | NFSClient *client = arg; | |
206 | ||
207 | qemu_mutex_lock(&client->mutex); | |
208 | nfs_service(client->context, POLLIN); | |
209 | nfs_set_events(client); | |
210 | qemu_mutex_unlock(&client->mutex); | |
211 | } | |
212 | ||
213 | static void nfs_process_write(void *arg) | |
214 | { | |
215 | NFSClient *client = arg; | |
216 | ||
217 | qemu_mutex_lock(&client->mutex); | |
218 | nfs_service(client->context, POLLOUT); | |
219 | nfs_set_events(client); | |
220 | qemu_mutex_unlock(&client->mutex); | |
221 | } | |
222 | ||
223 | static void nfs_co_init_task(BlockDriverState *bs, NFSRPC *task) | |
224 | { | |
225 | *task = (NFSRPC) { | |
226 | .co = qemu_coroutine_self(), | |
227 | .bs = bs, | |
228 | .client = bs->opaque, | |
229 | }; | |
230 | } | |
231 | ||
232 | static void nfs_co_generic_bh_cb(void *opaque) | |
233 | { | |
234 | NFSRPC *task = opaque; | |
235 | ||
236 | task->complete = 1; | |
237 | aio_co_wake(task->co); | |
238 | } | |
239 | ||
240 | /* Called (via nfs_service) with QemuMutex held. */ | |
241 | static void | |
242 | nfs_co_generic_cb(int ret, struct nfs_context *nfs, void *data, | |
243 | void *private_data) | |
244 | { | |
245 | NFSRPC *task = private_data; | |
246 | task->ret = ret; | |
247 | assert(!task->st); | |
248 | if (task->ret > 0 && task->iov) { | |
249 | if (task->ret <= task->iov->size) { | |
250 | qemu_iovec_from_buf(task->iov, 0, data, task->ret); | |
251 | } else { | |
252 | task->ret = -EIO; | |
253 | } | |
254 | } | |
255 | if (task->ret < 0) { | |
256 | error_report("NFS Error: %s", nfs_get_error(nfs)); | |
257 | } | |
258 | aio_bh_schedule_oneshot(task->client->aio_context, | |
259 | nfs_co_generic_bh_cb, task); | |
260 | } | |
261 | ||
262 | static int coroutine_fn nfs_co_preadv(BlockDriverState *bs, uint64_t offset, | |
263 | uint64_t bytes, QEMUIOVector *iov, | |
264 | int flags) | |
265 | { | |
266 | NFSClient *client = bs->opaque; | |
267 | NFSRPC task; | |
268 | ||
269 | nfs_co_init_task(bs, &task); | |
270 | task.iov = iov; | |
271 | ||
272 | qemu_mutex_lock(&client->mutex); | |
273 | if (nfs_pread_async(client->context, client->fh, | |
274 | offset, bytes, nfs_co_generic_cb, &task) != 0) { | |
275 | qemu_mutex_unlock(&client->mutex); | |
276 | return -ENOMEM; | |
277 | } | |
278 | ||
279 | nfs_set_events(client); | |
280 | qemu_mutex_unlock(&client->mutex); | |
281 | while (!task.complete) { | |
282 | qemu_coroutine_yield(); | |
283 | } | |
284 | ||
285 | if (task.ret < 0) { | |
286 | return task.ret; | |
287 | } | |
288 | ||
289 | /* zero pad short reads */ | |
290 | if (task.ret < iov->size) { | |
291 | qemu_iovec_memset(iov, task.ret, 0, iov->size - task.ret); | |
292 | } | |
293 | ||
294 | return 0; | |
295 | } | |
296 | ||
297 | static int coroutine_fn nfs_co_pwritev(BlockDriverState *bs, uint64_t offset, | |
298 | uint64_t bytes, QEMUIOVector *iov, | |
299 | int flags) | |
300 | { | |
301 | NFSClient *client = bs->opaque; | |
302 | NFSRPC task; | |
303 | char *buf = NULL; | |
304 | bool my_buffer = false; | |
305 | ||
306 | nfs_co_init_task(bs, &task); | |
307 | ||
308 | if (iov->niov != 1) { | |
309 | buf = g_try_malloc(bytes); | |
310 | if (bytes && buf == NULL) { | |
311 | return -ENOMEM; | |
312 | } | |
313 | qemu_iovec_to_buf(iov, 0, buf, bytes); | |
314 | my_buffer = true; | |
315 | } else { | |
316 | buf = iov->iov[0].iov_base; | |
317 | } | |
318 | ||
319 | qemu_mutex_lock(&client->mutex); | |
320 | if (nfs_pwrite_async(client->context, client->fh, | |
321 | offset, bytes, buf, | |
322 | nfs_co_generic_cb, &task) != 0) { | |
323 | qemu_mutex_unlock(&client->mutex); | |
324 | if (my_buffer) { | |
325 | g_free(buf); | |
326 | } | |
327 | return -ENOMEM; | |
328 | } | |
329 | ||
330 | nfs_set_events(client); | |
331 | qemu_mutex_unlock(&client->mutex); | |
332 | while (!task.complete) { | |
333 | qemu_coroutine_yield(); | |
334 | } | |
335 | ||
336 | if (my_buffer) { | |
337 | g_free(buf); | |
338 | } | |
339 | ||
340 | if (task.ret != bytes) { | |
341 | return task.ret < 0 ? task.ret : -EIO; | |
342 | } | |
343 | ||
344 | return 0; | |
345 | } | |
346 | ||
347 | static int coroutine_fn nfs_co_flush(BlockDriverState *bs) | |
348 | { | |
349 | NFSClient *client = bs->opaque; | |
350 | NFSRPC task; | |
351 | ||
352 | nfs_co_init_task(bs, &task); | |
353 | ||
354 | qemu_mutex_lock(&client->mutex); | |
355 | if (nfs_fsync_async(client->context, client->fh, nfs_co_generic_cb, | |
356 | &task) != 0) { | |
357 | qemu_mutex_unlock(&client->mutex); | |
358 | return -ENOMEM; | |
359 | } | |
360 | ||
361 | nfs_set_events(client); | |
362 | qemu_mutex_unlock(&client->mutex); | |
363 | while (!task.complete) { | |
364 | qemu_coroutine_yield(); | |
365 | } | |
366 | ||
367 | return task.ret; | |
368 | } | |
369 | ||
370 | static QemuOptsList runtime_opts = { | |
371 | .name = "nfs", | |
372 | .head = QTAILQ_HEAD_INITIALIZER(runtime_opts.head), | |
373 | .desc = { | |
374 | { | |
375 | .name = "path", | |
376 | .type = QEMU_OPT_STRING, | |
377 | .help = "Path of the image on the host", | |
378 | }, | |
379 | { | |
380 | .name = "user", | |
381 | .type = QEMU_OPT_NUMBER, | |
382 | .help = "UID value to use when talking to the server", | |
383 | }, | |
384 | { | |
385 | .name = "group", | |
386 | .type = QEMU_OPT_NUMBER, | |
387 | .help = "GID value to use when talking to the server", | |
388 | }, | |
389 | { | |
390 | .name = "tcp-syn-count", | |
391 | .type = QEMU_OPT_NUMBER, | |
392 | .help = "Number of SYNs to send during the session establish", | |
393 | }, | |
394 | { | |
395 | .name = "readahead-size", | |
396 | .type = QEMU_OPT_NUMBER, | |
397 | .help = "Set the readahead size in bytes", | |
398 | }, | |
399 | { | |
400 | .name = "page-cache-size", | |
401 | .type = QEMU_OPT_NUMBER, | |
402 | .help = "Set the pagecache size in bytes", | |
403 | }, | |
404 | { | |
405 | .name = "debug", | |
406 | .type = QEMU_OPT_NUMBER, | |
407 | .help = "Set the NFS debug level (max 2)", | |
408 | }, | |
409 | { /* end of list */ } | |
410 | }, | |
411 | }; | |
412 | ||
413 | static void nfs_detach_aio_context(BlockDriverState *bs) | |
414 | { | |
415 | NFSClient *client = bs->opaque; | |
416 | ||
417 | aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context), | |
418 | false, NULL, NULL, NULL, NULL); | |
419 | client->events = 0; | |
420 | } | |
421 | ||
422 | static void nfs_attach_aio_context(BlockDriverState *bs, | |
423 | AioContext *new_context) | |
424 | { | |
425 | NFSClient *client = bs->opaque; | |
426 | ||
427 | client->aio_context = new_context; | |
428 | nfs_set_events(client); | |
429 | } | |
430 | ||
431 | static void nfs_client_close(NFSClient *client) | |
432 | { | |
433 | if (client->context) { | |
434 | if (client->fh) { | |
435 | nfs_close(client->context, client->fh); | |
436 | client->fh = NULL; | |
437 | } | |
438 | aio_set_fd_handler(client->aio_context, nfs_get_fd(client->context), | |
439 | false, NULL, NULL, NULL, NULL); | |
440 | nfs_destroy_context(client->context); | |
441 | client->context = NULL; | |
442 | } | |
443 | g_free(client->path); | |
444 | qemu_mutex_destroy(&client->mutex); | |
445 | qapi_free_NFSServer(client->server); | |
446 | client->server = NULL; | |
447 | } | |
448 | ||
449 | static void nfs_file_close(BlockDriverState *bs) | |
450 | { | |
451 | NFSClient *client = bs->opaque; | |
452 | nfs_client_close(client); | |
453 | } | |
454 | ||
455 | static NFSServer *nfs_config(QDict *options, Error **errp) | |
456 | { | |
457 | NFSServer *server = NULL; | |
458 | QDict *addr = NULL; | |
459 | QObject *crumpled_addr = NULL; | |
460 | Visitor *iv = NULL; | |
461 | Error *local_error = NULL; | |
462 | ||
463 | qdict_extract_subqdict(options, &addr, "server."); | |
464 | if (!qdict_size(addr)) { | |
465 | error_setg(errp, "NFS server address missing"); | |
466 | goto out; | |
467 | } | |
468 | ||
469 | crumpled_addr = qdict_crumple(addr, errp); | |
470 | if (!crumpled_addr) { | |
471 | goto out; | |
472 | } | |
473 | ||
474 | /* | |
475 | * Caution: this works only because all scalar members of | |
476 | * NFSServer are QString in @crumpled_addr. The visitor expects | |
477 | * @crumpled_addr to be typed according to the QAPI schema. It | |
478 | * is when @options come from -blockdev or blockdev_add. But when | |
479 | * they come from -drive, they're all QString. | |
480 | */ | |
481 | iv = qobject_input_visitor_new(crumpled_addr); | |
482 | visit_type_NFSServer(iv, NULL, &server, &local_error); | |
483 | if (local_error) { | |
484 | error_propagate(errp, local_error); | |
485 | goto out; | |
486 | } | |
487 | ||
488 | out: | |
489 | QDECREF(addr); | |
490 | qobject_decref(crumpled_addr); | |
491 | visit_free(iv); | |
492 | return server; | |
493 | } | |
494 | ||
495 | ||
496 | static int64_t nfs_client_open(NFSClient *client, QDict *options, | |
497 | int flags, int open_flags, Error **errp) | |
498 | { | |
499 | int64_t ret = -EINVAL; | |
500 | QemuOpts *opts = NULL; | |
501 | Error *local_err = NULL; | |
502 | struct stat st; | |
503 | char *file = NULL, *strp = NULL; | |
504 | ||
505 | qemu_mutex_init(&client->mutex); | |
506 | opts = qemu_opts_create(&runtime_opts, NULL, 0, &error_abort); | |
507 | qemu_opts_absorb_qdict(opts, options, &local_err); | |
508 | if (local_err) { | |
509 | error_propagate(errp, local_err); | |
510 | ret = -EINVAL; | |
511 | goto fail; | |
512 | } | |
513 | ||
514 | client->path = g_strdup(qemu_opt_get(opts, "path")); | |
515 | if (!client->path) { | |
516 | ret = -EINVAL; | |
517 | error_setg(errp, "No path was specified"); | |
518 | goto fail; | |
519 | } | |
520 | ||
521 | strp = strrchr(client->path, '/'); | |
522 | if (strp == NULL) { | |
523 | error_setg(errp, "Invalid URL specified"); | |
524 | goto fail; | |
525 | } | |
526 | file = g_strdup(strp); | |
527 | *strp = 0; | |
528 | ||
529 | /* Pop the config into our state object, Exit if invalid */ | |
530 | client->server = nfs_config(options, errp); | |
531 | if (!client->server) { | |
532 | ret = -EINVAL; | |
533 | goto fail; | |
534 | } | |
535 | ||
536 | client->context = nfs_init_context(); | |
537 | if (client->context == NULL) { | |
538 | error_setg(errp, "Failed to init NFS context"); | |
539 | goto fail; | |
540 | } | |
541 | ||
542 | if (qemu_opt_get(opts, "user")) { | |
543 | client->uid = qemu_opt_get_number(opts, "user", 0); | |
544 | nfs_set_uid(client->context, client->uid); | |
545 | } | |
546 | ||
547 | if (qemu_opt_get(opts, "group")) { | |
548 | client->gid = qemu_opt_get_number(opts, "group", 0); | |
549 | nfs_set_gid(client->context, client->gid); | |
550 | } | |
551 | ||
552 | if (qemu_opt_get(opts, "tcp-syn-count")) { | |
553 | client->tcp_syncnt = qemu_opt_get_number(opts, "tcp-syn-count", 0); | |
554 | nfs_set_tcp_syncnt(client->context, client->tcp_syncnt); | |
555 | } | |
556 | ||
557 | #ifdef LIBNFS_FEATURE_READAHEAD | |
558 | if (qemu_opt_get(opts, "readahead-size")) { | |
559 | if (open_flags & BDRV_O_NOCACHE) { | |
560 | error_setg(errp, "Cannot enable NFS readahead " | |
561 | "if cache.direct = on"); | |
562 | goto fail; | |
563 | } | |
564 | client->readahead = qemu_opt_get_number(opts, "readahead-size", 0); | |
565 | if (client->readahead > QEMU_NFS_MAX_READAHEAD_SIZE) { | |
566 | warn_report("Truncating NFS readahead size to %d", | |
567 | QEMU_NFS_MAX_READAHEAD_SIZE); | |
568 | client->readahead = QEMU_NFS_MAX_READAHEAD_SIZE; | |
569 | } | |
570 | nfs_set_readahead(client->context, client->readahead); | |
571 | #ifdef LIBNFS_FEATURE_PAGECACHE | |
572 | nfs_set_pagecache_ttl(client->context, 0); | |
573 | #endif | |
574 | client->cache_used = true; | |
575 | } | |
576 | #endif | |
577 | ||
578 | #ifdef LIBNFS_FEATURE_PAGECACHE | |
579 | if (qemu_opt_get(opts, "page-cache-size")) { | |
580 | if (open_flags & BDRV_O_NOCACHE) { | |
581 | error_setg(errp, "Cannot enable NFS pagecache " | |
582 | "if cache.direct = on"); | |
583 | goto fail; | |
584 | } | |
585 | client->pagecache = qemu_opt_get_number(opts, "page-cache-size", 0); | |
586 | if (client->pagecache > QEMU_NFS_MAX_PAGECACHE_SIZE) { | |
587 | warn_report("Truncating NFS pagecache size to %d pages", | |
588 | QEMU_NFS_MAX_PAGECACHE_SIZE); | |
589 | client->pagecache = QEMU_NFS_MAX_PAGECACHE_SIZE; | |
590 | } | |
591 | nfs_set_pagecache(client->context, client->pagecache); | |
592 | nfs_set_pagecache_ttl(client->context, 0); | |
593 | client->cache_used = true; | |
594 | } | |
595 | #endif | |
596 | ||
597 | #ifdef LIBNFS_FEATURE_DEBUG | |
598 | if (qemu_opt_get(opts, "debug")) { | |
599 | client->debug = qemu_opt_get_number(opts, "debug", 0); | |
600 | /* limit the maximum debug level to avoid potential flooding | |
601 | * of our log files. */ | |
602 | if (client->debug > QEMU_NFS_MAX_DEBUG_LEVEL) { | |
603 | warn_report("Limiting NFS debug level to %d", | |
604 | QEMU_NFS_MAX_DEBUG_LEVEL); | |
605 | client->debug = QEMU_NFS_MAX_DEBUG_LEVEL; | |
606 | } | |
607 | nfs_set_debug(client->context, client->debug); | |
608 | } | |
609 | #endif | |
610 | ||
611 | ret = nfs_mount(client->context, client->server->host, client->path); | |
612 | if (ret < 0) { | |
613 | error_setg(errp, "Failed to mount nfs share: %s", | |
614 | nfs_get_error(client->context)); | |
615 | goto fail; | |
616 | } | |
617 | ||
618 | if (flags & O_CREAT) { | |
619 | ret = nfs_creat(client->context, file, 0600, &client->fh); | |
620 | if (ret < 0) { | |
621 | error_setg(errp, "Failed to create file: %s", | |
622 | nfs_get_error(client->context)); | |
623 | goto fail; | |
624 | } | |
625 | } else { | |
626 | ret = nfs_open(client->context, file, flags, &client->fh); | |
627 | if (ret < 0) { | |
628 | error_setg(errp, "Failed to open file : %s", | |
629 | nfs_get_error(client->context)); | |
630 | goto fail; | |
631 | } | |
632 | } | |
633 | ||
634 | ret = nfs_fstat(client->context, client->fh, &st); | |
635 | if (ret < 0) { | |
636 | error_setg(errp, "Failed to fstat file: %s", | |
637 | nfs_get_error(client->context)); | |
638 | goto fail; | |
639 | } | |
640 | ||
641 | ret = DIV_ROUND_UP(st.st_size, BDRV_SECTOR_SIZE); | |
642 | client->st_blocks = st.st_blocks; | |
643 | client->has_zero_init = S_ISREG(st.st_mode); | |
644 | *strp = '/'; | |
645 | goto out; | |
646 | ||
647 | fail: | |
648 | nfs_client_close(client); | |
649 | out: | |
650 | qemu_opts_del(opts); | |
651 | g_free(file); | |
652 | return ret; | |
653 | } | |
654 | ||
655 | static int nfs_file_open(BlockDriverState *bs, QDict *options, int flags, | |
656 | Error **errp) { | |
657 | NFSClient *client = bs->opaque; | |
658 | int64_t ret; | |
659 | ||
660 | client->aio_context = bdrv_get_aio_context(bs); | |
661 | ||
662 | ret = nfs_client_open(client, options, | |
663 | (flags & BDRV_O_RDWR) ? O_RDWR : O_RDONLY, | |
664 | bs->open_flags, errp); | |
665 | if (ret < 0) { | |
666 | return ret; | |
667 | } | |
668 | ||
669 | bs->total_sectors = ret; | |
670 | ret = 0; | |
671 | return ret; | |
672 | } | |
673 | ||
674 | static QemuOptsList nfs_create_opts = { | |
675 | .name = "nfs-create-opts", | |
676 | .head = QTAILQ_HEAD_INITIALIZER(nfs_create_opts.head), | |
677 | .desc = { | |
678 | { | |
679 | .name = BLOCK_OPT_SIZE, | |
680 | .type = QEMU_OPT_SIZE, | |
681 | .help = "Virtual disk size" | |
682 | }, | |
683 | { /* end of list */ } | |
684 | } | |
685 | }; | |
686 | ||
687 | static int nfs_file_create(const char *url, QemuOpts *opts, Error **errp) | |
688 | { | |
689 | int64_t ret, total_size; | |
690 | NFSClient *client = g_new0(NFSClient, 1); | |
691 | QDict *options = NULL; | |
692 | ||
693 | client->aio_context = qemu_get_aio_context(); | |
694 | ||
695 | /* Read out options */ | |
696 | total_size = ROUND_UP(qemu_opt_get_size_del(opts, BLOCK_OPT_SIZE, 0), | |
697 | BDRV_SECTOR_SIZE); | |
698 | ||
699 | options = qdict_new(); | |
700 | ret = nfs_parse_uri(url, options, errp); | |
701 | if (ret < 0) { | |
702 | goto out; | |
703 | } | |
704 | ||
705 | ret = nfs_client_open(client, options, O_CREAT, 0, errp); | |
706 | if (ret < 0) { | |
707 | goto out; | |
708 | } | |
709 | ret = nfs_ftruncate(client->context, client->fh, total_size); | |
710 | nfs_client_close(client); | |
711 | out: | |
712 | QDECREF(options); | |
713 | g_free(client); | |
714 | return ret; | |
715 | } | |
716 | ||
717 | static int nfs_has_zero_init(BlockDriverState *bs) | |
718 | { | |
719 | NFSClient *client = bs->opaque; | |
720 | return client->has_zero_init; | |
721 | } | |
722 | ||
723 | /* Called (via nfs_service) with QemuMutex held. */ | |
724 | static void | |
725 | nfs_get_allocated_file_size_cb(int ret, struct nfs_context *nfs, void *data, | |
726 | void *private_data) | |
727 | { | |
728 | NFSRPC *task = private_data; | |
729 | task->ret = ret; | |
730 | if (task->ret == 0) { | |
731 | memcpy(task->st, data, sizeof(struct stat)); | |
732 | } | |
733 | if (task->ret < 0) { | |
734 | error_report("NFS Error: %s", nfs_get_error(nfs)); | |
735 | } | |
736 | ||
737 | /* Set task->complete before reading bs->wakeup. */ | |
738 | atomic_mb_set(&task->complete, 1); | |
739 | bdrv_wakeup(task->bs); | |
740 | } | |
741 | ||
742 | static int64_t nfs_get_allocated_file_size(BlockDriverState *bs) | |
743 | { | |
744 | NFSClient *client = bs->opaque; | |
745 | NFSRPC task = {0}; | |
746 | struct stat st; | |
747 | ||
748 | if (bdrv_is_read_only(bs) && | |
749 | !(bs->open_flags & BDRV_O_NOCACHE)) { | |
750 | return client->st_blocks * 512; | |
751 | } | |
752 | ||
753 | task.bs = bs; | |
754 | task.st = &st; | |
755 | if (nfs_fstat_async(client->context, client->fh, nfs_get_allocated_file_size_cb, | |
756 | &task) != 0) { | |
757 | return -ENOMEM; | |
758 | } | |
759 | ||
760 | nfs_set_events(client); | |
761 | BDRV_POLL_WHILE(bs, !task.complete); | |
762 | ||
763 | return (task.ret < 0 ? task.ret : st.st_blocks * 512); | |
764 | } | |
765 | ||
766 | static int nfs_file_truncate(BlockDriverState *bs, int64_t offset, | |
767 | PreallocMode prealloc, Error **errp) | |
768 | { | |
769 | NFSClient *client = bs->opaque; | |
770 | int ret; | |
771 | ||
772 | if (prealloc != PREALLOC_MODE_OFF) { | |
773 | error_setg(errp, "Unsupported preallocation mode '%s'", | |
774 | PreallocMode_str(prealloc)); | |
775 | return -ENOTSUP; | |
776 | } | |
777 | ||
778 | ret = nfs_ftruncate(client->context, client->fh, offset); | |
779 | if (ret < 0) { | |
780 | error_setg_errno(errp, -ret, "Failed to truncate file"); | |
781 | return ret; | |
782 | } | |
783 | ||
784 | return 0; | |
785 | } | |
786 | ||
787 | /* Note that this will not re-establish a connection with the NFS server | |
788 | * - it is effectively a NOP. */ | |
789 | static int nfs_reopen_prepare(BDRVReopenState *state, | |
790 | BlockReopenQueue *queue, Error **errp) | |
791 | { | |
792 | NFSClient *client = state->bs->opaque; | |
793 | struct stat st; | |
794 | int ret = 0; | |
795 | ||
796 | if (state->flags & BDRV_O_RDWR && bdrv_is_read_only(state->bs)) { | |
797 | error_setg(errp, "Cannot open a read-only mount as read-write"); | |
798 | return -EACCES; | |
799 | } | |
800 | ||
801 | if ((state->flags & BDRV_O_NOCACHE) && client->cache_used) { | |
802 | error_setg(errp, "Cannot disable cache if libnfs readahead or" | |
803 | " pagecache is enabled"); | |
804 | return -EINVAL; | |
805 | } | |
806 | ||
807 | /* Update cache for read-only reopens */ | |
808 | if (!(state->flags & BDRV_O_RDWR)) { | |
809 | ret = nfs_fstat(client->context, client->fh, &st); | |
810 | if (ret < 0) { | |
811 | error_setg(errp, "Failed to fstat file: %s", | |
812 | nfs_get_error(client->context)); | |
813 | return ret; | |
814 | } | |
815 | client->st_blocks = st.st_blocks; | |
816 | } | |
817 | ||
818 | return 0; | |
819 | } | |
820 | ||
821 | static void nfs_refresh_filename(BlockDriverState *bs, QDict *options) | |
822 | { | |
823 | NFSClient *client = bs->opaque; | |
824 | QDict *opts = qdict_new(); | |
825 | QObject *server_qdict; | |
826 | Visitor *ov; | |
827 | ||
828 | qdict_put_str(opts, "driver", "nfs"); | |
829 | ||
830 | if (client->uid && !client->gid) { | |
831 | snprintf(bs->exact_filename, sizeof(bs->exact_filename), | |
832 | "nfs://%s%s?uid=%" PRId64, client->server->host, client->path, | |
833 | client->uid); | |
834 | } else if (!client->uid && client->gid) { | |
835 | snprintf(bs->exact_filename, sizeof(bs->exact_filename), | |
836 | "nfs://%s%s?gid=%" PRId64, client->server->host, client->path, | |
837 | client->gid); | |
838 | } else if (client->uid && client->gid) { | |
839 | snprintf(bs->exact_filename, sizeof(bs->exact_filename), | |
840 | "nfs://%s%s?uid=%" PRId64 "&gid=%" PRId64, | |
841 | client->server->host, client->path, client->uid, client->gid); | |
842 | } else { | |
843 | snprintf(bs->exact_filename, sizeof(bs->exact_filename), | |
844 | "nfs://%s%s", client->server->host, client->path); | |
845 | } | |
846 | ||
847 | ov = qobject_output_visitor_new(&server_qdict); | |
848 | visit_type_NFSServer(ov, NULL, &client->server, &error_abort); | |
849 | visit_complete(ov, &server_qdict); | |
850 | qdict_put_obj(opts, "server", server_qdict); | |
851 | qdict_put_str(opts, "path", client->path); | |
852 | ||
853 | if (client->uid) { | |
854 | qdict_put_int(opts, "user", client->uid); | |
855 | } | |
856 | if (client->gid) { | |
857 | qdict_put_int(opts, "group", client->gid); | |
858 | } | |
859 | if (client->tcp_syncnt) { | |
860 | qdict_put_int(opts, "tcp-syn-cnt", client->tcp_syncnt); | |
861 | } | |
862 | if (client->readahead) { | |
863 | qdict_put_int(opts, "readahead-size", client->readahead); | |
864 | } | |
865 | if (client->pagecache) { | |
866 | qdict_put_int(opts, "page-cache-size", client->pagecache); | |
867 | } | |
868 | if (client->debug) { | |
869 | qdict_put_int(opts, "debug", client->debug); | |
870 | } | |
871 | ||
872 | visit_free(ov); | |
873 | qdict_flatten(opts); | |
874 | bs->full_open_options = opts; | |
875 | } | |
876 | ||
877 | #ifdef LIBNFS_FEATURE_PAGECACHE | |
878 | static void nfs_invalidate_cache(BlockDriverState *bs, | |
879 | Error **errp) | |
880 | { | |
881 | NFSClient *client = bs->opaque; | |
882 | nfs_pagecache_invalidate(client->context, client->fh); | |
883 | } | |
884 | #endif | |
885 | ||
886 | static BlockDriver bdrv_nfs = { | |
887 | .format_name = "nfs", | |
888 | .protocol_name = "nfs", | |
889 | ||
890 | .instance_size = sizeof(NFSClient), | |
891 | .bdrv_parse_filename = nfs_parse_filename, | |
892 | .create_opts = &nfs_create_opts, | |
893 | ||
894 | .bdrv_has_zero_init = nfs_has_zero_init, | |
895 | .bdrv_get_allocated_file_size = nfs_get_allocated_file_size, | |
896 | .bdrv_truncate = nfs_file_truncate, | |
897 | ||
898 | .bdrv_file_open = nfs_file_open, | |
899 | .bdrv_close = nfs_file_close, | |
900 | .bdrv_create = nfs_file_create, | |
901 | .bdrv_reopen_prepare = nfs_reopen_prepare, | |
902 | ||
903 | .bdrv_co_preadv = nfs_co_preadv, | |
904 | .bdrv_co_pwritev = nfs_co_pwritev, | |
905 | .bdrv_co_flush_to_disk = nfs_co_flush, | |
906 | ||
907 | .bdrv_detach_aio_context = nfs_detach_aio_context, | |
908 | .bdrv_attach_aio_context = nfs_attach_aio_context, | |
909 | .bdrv_refresh_filename = nfs_refresh_filename, | |
910 | ||
911 | #ifdef LIBNFS_FEATURE_PAGECACHE | |
912 | .bdrv_invalidate_cache = nfs_invalidate_cache, | |
913 | #endif | |
914 | }; | |
915 | ||
916 | static void nfs_block_init(void) | |
917 | { | |
918 | bdrv_register(&bdrv_nfs); | |
919 | } | |
920 | ||
921 | block_init(nfs_block_init); |