1cd831bd7Sths /*
27a5ca864Sbellard * Copyright (C) 2005 Anthony Liguori <anthony@codemonkey.ws>
37a5ca864Sbellard *
47a5ca864Sbellard * Network Block Device
57a5ca864Sbellard *
67a5ca864Sbellard * This program is free software; you can redistribute it and/or modify
77a5ca864Sbellard * it under the terms of the GNU General Public License as published by
87a5ca864Sbellard * the Free Software Foundation; under version 2 of the License.
97a5ca864Sbellard *
107a5ca864Sbellard * This program is distributed in the hope that it will be useful,
117a5ca864Sbellard * but WITHOUT ANY WARRANTY; without even the implied warranty of
127a5ca864Sbellard * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
137a5ca864Sbellard * GNU General Public License for more details.
147a5ca864Sbellard *
157a5ca864Sbellard * You should have received a copy of the GNU General Public License
168167ee88SBlue Swirl * along with this program; if not, see <http://www.gnu.org/licenses/>.
177a5ca864Sbellard */
187a5ca864Sbellard
19d38ea87aSPeter Maydell #include "qemu/osdep.h"
20c2a3d7daSEric Blake #include <getopt.h>
21c2a3d7daSEric Blake #include <libgen.h>
22c2a3d7daSEric Blake #include <pthread.h>
23c2a3d7daSEric Blake
2449f95221SMarc-André Lureau #include "qemu/help-texts.h"
25da34e65cSMarkus Armbruster #include "qapi/error.h"
26f348b6d1SVeronia Bahaa #include "qemu/cutils.h"
2726f54e9aSMarkus Armbruster #include "sysemu/block-backend.h"
28cbc20bfbSStefan Hajnoczi #include "sysemu/runstate.h" /* for qemu_system_killed() prototype */
29b3838a40SPeter Lieven #include "block/block_int.h"
30737e150eSPaolo Bonzini #include "block/nbd.h"
316a1751b7SAlex Bligh #include "qemu/main-loop.h"
320b8fa32fSMarkus Armbruster #include "qemu/module.h"
33922a01a0SMarkus Armbruster #include "qemu/option.h"
34537b41f5SPaolo Bonzini #include "qemu/error-report.h"
350ab3b337SDaniel P. Berrange #include "qemu/config-file.h"
3658369e22SPaolo Bonzini #include "qemu/bswap.h"
3739ca463eSDenis V. Lunev #include "qemu/log.h"
3853fabd4bSPaolo Bonzini #include "qemu/systemd.h"
398c116b0eSWenchao Xia #include "block/snapshot.h"
40452fcdbcSMarkus Armbruster #include "qapi/qmp/qdict.h"
41d49b6836SMarkus Armbruster #include "qapi/qmp/qstring.h"
420ab3b337SDaniel P. Berrange #include "qom/object_interfaces.h"
43d0d6ff58SDaniel P. Berrange #include "io/channel-socket.h"
44e4849c1dSDaniel P. Berrange #include "io/net-listener.h"
45c2297088SDaniel P. Berrange #include "crypto/init.h"
460279cd95SPhilippe Mathieu-Daudé #include "crypto/tlscreds.h"
4739ca463eSDenis V. Lunev #include "trace/control.h"
48be377133SEric Blake #include "qemu-version.h"
497a5ca864Sbellard
503d212b41SRichard W.M. Jones #ifdef CONFIG_SELINUX
513d212b41SRichard W.M. Jones #include <selinux/selinux.h>
523d212b41SRichard W.M. Jones #endif
533d212b41SRichard W.M. Jones
543c1fa35dSEric Blake #ifdef __linux__
553c1fa35dSEric Blake #define HAVE_NBD_DEVICE 1
563c1fa35dSEric Blake #else
573c1fa35dSEric Blake #define HAVE_NBD_DEVICE 0
583c1fa35dSEric Blake #endif
593c1fa35dSEric Blake
60cd831bd7Sths #define SOCKET_PATH "/var/lock/qemu-nbd-%s"
61fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_CACHE 256
62fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_AIO 257
63fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_DISCARD 258
64fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_DETECT_ZEROES 259
65fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_OBJECT 260
66fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_TLSCREDS 261
67fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_IMAGE_OPTS 262
68ffb31e1dSMax Reitz #define QEMU_NBD_OPT_FORK 263
69b25e12daSDaniel P. Berrange #define QEMU_NBD_OPT_TLSAUTHZ 264
70637bc5a5SMax Reitz #define QEMU_NBD_OPT_PID_FILE 265
713d212b41SRichard W.M. Jones #define QEMU_NBD_OPT_SELINUX_LABEL 266
72003b2b25SDaniel P. Berrangé #define QEMU_NBD_OPT_TLSHOSTNAME 267
737a5ca864Sbellard
74bd31c214SEric Blake #define MBR_SIZE 512
75bd31c214SEric Blake
767860a380SPaolo Bonzini static int persistent = 0;
77d794f7f3SKevin Wolf static enum { RUNNING, TERMINATE, TERMINATED } state;
78a61c6782SPaolo Bonzini static int shared = 1;
79a61c6782SPaolo Bonzini static int nb_fds;
80e4849c1dSDaniel P. Berrange static QIONetListener *server;
81145614a1SDaniel P. Berrange static QCryptoTLSCreds *tlscreds;
82b25e12daSDaniel P. Berrange static const char *tlsauthz;
837a5ca864Sbellard
usage(const char * name)847a5ca864Sbellard static void usage(const char *name)
857a5ca864Sbellard {
86b033cd86SPaolo Bonzini (printf) (
877a5ca864Sbellard "Usage: %s [OPTIONS] FILE\n"
8868b96f15SEric Blake " or: %s -L [OPTIONS]\n"
8968b96f15SEric Blake "QEMU Disk Network Block Device Utility\n"
907a5ca864Sbellard "\n"
917a5ca864Sbellard " -h, --help display this help and exit\n"
927a5ca864Sbellard " -V, --version output version information and exit\n"
937a5ca864Sbellard "\n"
94b033cd86SPaolo Bonzini "Connection properties:\n"
95b033cd86SPaolo Bonzini " -p, --port=PORT port to listen on (default `%d')\n"
96b033cd86SPaolo Bonzini " -b, --bind=IFACE interface to bind to (default `0.0.0.0')\n"
97b033cd86SPaolo Bonzini " -k, --socket=PATH path to the unix socket\n"
98b033cd86SPaolo Bonzini " (default '"SOCKET_PATH"')\n"
99b033cd86SPaolo Bonzini " -e, --shared=NUM device can be shared by NUM clients (default '1')\n"
100b033cd86SPaolo Bonzini " -t, --persistent don't exit on the last connection\n"
101b033cd86SPaolo Bonzini " -v, --verbose display extra debugging information\n"
102f5cd0bb5SVladimir Sementsov-Ogievskiy " -x, --export-name=NAME expose export by name (default is empty string)\n"
103f5cd0bb5SVladimir Sementsov-Ogievskiy " -D, --description=TEXT export a human-readable description\n"
104b033cd86SPaolo Bonzini "\n"
105b033cd86SPaolo Bonzini "Exposing part of the image:\n"
106b033cd86SPaolo Bonzini " -o, --offset=OFFSET offset into the image\n"
107dbc7b014SEric Blake " -A, --allocation-depth expose the allocation depth\n"
108636192c4SEric Blake " -B, --bitmap=NAME expose a persistent dirty bitmap\n"
109b033cd86SPaolo Bonzini "\n"
1100ab3b337SDaniel P. Berrange "General purpose options:\n"
11168b96f15SEric Blake " -L, --list list exports available from another NBD server\n"
1120ab3b337SDaniel P. Berrange " --object type,id=ID,... define an object such as 'secret' for providing\n"
1130ab3b337SDaniel P. Berrange " passwords and/or encryption keys\n"
114f7812df7SEric Blake " --tls-creds=ID use id of an earlier --object to provide TLS\n"
115b25e12daSDaniel P. Berrange " --tls-authz=ID use id of an earlier --object to provide\n"
116b25e12daSDaniel P. Berrange " authorization\n"
117*d81b797cSMichael Tokarev " --tls-hostname=HOSTNAME override hostname used to check x509 certificate\n"
11839ca463eSDenis V. Lunev " -T, --trace [[enable=]<pattern>][,events=<file>][,file=<file>]\n"
11939ca463eSDenis V. Lunev " specify tracing options\n"
120ffb31e1dSMax Reitz " --fork fork off the server process and exit the parent\n"
121ffb31e1dSMax Reitz " once the server is running\n"
122637bc5a5SMax Reitz " --pid-file=PATH store the server's process ID in the given file\n"
1233d212b41SRichard W.M. Jones #ifdef CONFIG_SELINUX
1243d212b41SRichard W.M. Jones " --selinux-label=LABEL set SELinux process label on listening socket\n"
1253d212b41SRichard W.M. Jones #endif
1263c1fa35dSEric Blake #if HAVE_NBD_DEVICE
1273c1fa35dSEric Blake "\n"
128b033cd86SPaolo Bonzini "Kernel NBD client support:\n"
129b033cd86SPaolo Bonzini " -c, --connect=DEV connect FILE to the local NBD device DEV\n"
130b033cd86SPaolo Bonzini " -d, --disconnect disconnect the specified device\n"
131b033cd86SPaolo Bonzini #endif
132b033cd86SPaolo Bonzini "\n"
133b033cd86SPaolo Bonzini "Block device options:\n"
1344323fdcfSWenchao Xia " -f, --format=FORMAT set image format (raw, qcow2, ...)\n"
135b033cd86SPaolo Bonzini " -r, --read-only export read-only\n"
1368c116b0eSWenchao Xia " -s, --snapshot use FILE as an external snapshot, create a temporary\n"
1378c116b0eSWenchao Xia " file with backing_file=FILE, redirect the write to\n"
1388c116b0eSWenchao Xia " the temporary one\n"
1398c116b0eSWenchao Xia " -l, --load-snapshot=SNAPSHOT_PARAM\n"
1408c116b0eSWenchao Xia " load an internal snapshot inside FILE and export it\n"
1418c116b0eSWenchao Xia " as an read-only device, SNAPSHOT_PARAM format is\n"
1428c116b0eSWenchao Xia " 'snapshot.id=[ID],snapshot.name=[NAME]', or\n"
1438c116b0eSWenchao Xia " '[ID_OR_NAME]'\n"
144b033cd86SPaolo Bonzini " -n, --nocache disable host cache\n"
14509615257SNir Soffer " --cache=MODE set cache mode used to access the disk image, the\n"
14609615257SNir Soffer " valid options are: 'none', 'writeback' (default),\n"
14709615257SNir Soffer " 'writethrough', 'directsync' and 'unsafe'\n"
1487680274dSAarushi Mehta " --aio=MODE set AIO mode (native, io_uring or threads)\n"
149b3838a40SPeter Lieven " --discard=MODE set discard mode (ignore, unmap)\n"
1506883de6cSAndrey Korolyov " --detect-zeroes=MODE set detect-zeroes mode (off, on, unmap)\n"
15177c9aaefSDaniel P. Berrange " --image-opts treat FILE as a full set of image options\n"
152b033cd86SPaolo Bonzini "\n"
153f5048cb7SEric Blake QEMU_HELP_BOTTOM "\n"
15468b96f15SEric Blake , name, name, NBD_DEFAULT_PORT, "DEVICE");
1557a5ca864Sbellard }
1567a5ca864Sbellard
version(const char * name)1577a5ca864Sbellard static void version(const char *name)
1587a5ca864Sbellard {
1597a5ca864Sbellard printf(
1607e563bfbSThomas Huth "%s " QEMU_FULL_VERSION "\n"
1617a5ca864Sbellard "Written by Anthony Liguori.\n"
1627a5ca864Sbellard "\n"
163be377133SEric Blake QEMU_COPYRIGHT "\n"
1647a5ca864Sbellard "This is free software; see the source for copying conditions. There is NO\n"
1657a5ca864Sbellard "warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"
166315bc7aaSths , name);
1677a5ca864Sbellard }
1687a5ca864Sbellard
169029a88c9SEric Blake #ifdef CONFIG_POSIX
170cbc20bfbSStefan Hajnoczi /*
171cbc20bfbSStefan Hajnoczi * The client thread uses SIGTERM to interrupt the server. A signal
172cbc20bfbSStefan Hajnoczi * handler ensures that "qemu-nbd -v -c" exits with a nice status code.
173cbc20bfbSStefan Hajnoczi */
qemu_system_killed(int signum,pid_t pid)174cbc20bfbSStefan Hajnoczi void qemu_system_killed(int signum, pid_t pid)
175bb345110SPaolo Bonzini {
176d73415a3SStefan Hajnoczi qatomic_cmpxchg(&state, RUNNING, TERMINATE);
177a61c6782SPaolo Bonzini qemu_notify_event();
178bb345110SPaolo Bonzini }
179029a88c9SEric Blake #endif /* CONFIG_POSIX */
180537b41f5SPaolo Bonzini
qemu_nbd_client_list(SocketAddress * saddr,QCryptoTLSCreds * tls,const char * hostname)18168b96f15SEric Blake static int qemu_nbd_client_list(SocketAddress *saddr, QCryptoTLSCreds *tls,
18268b96f15SEric Blake const char *hostname)
18368b96f15SEric Blake {
18468b96f15SEric Blake int ret = EXIT_FAILURE;
18568b96f15SEric Blake int rc;
18668b96f15SEric Blake Error *err = NULL;
18768b96f15SEric Blake QIOChannelSocket *sioc;
18868b96f15SEric Blake NBDExportInfo *list;
18968b96f15SEric Blake int i, j;
19068b96f15SEric Blake
19168b96f15SEric Blake sioc = qio_channel_socket_new();
19268b96f15SEric Blake if (qio_channel_socket_connect_sync(sioc, saddr, &err) < 0) {
19368b96f15SEric Blake error_report_err(err);
194992809bfSAlex Chen goto out;
19568b96f15SEric Blake }
19668b96f15SEric Blake rc = nbd_receive_export_list(QIO_CHANNEL(sioc), tls, hostname, &list,
19768b96f15SEric Blake &err);
19868b96f15SEric Blake if (rc < 0) {
19968b96f15SEric Blake if (err) {
20068b96f15SEric Blake error_report_err(err);
20168b96f15SEric Blake }
20268b96f15SEric Blake goto out;
20368b96f15SEric Blake }
20468b96f15SEric Blake printf("exports available: %d\n", rc);
20568b96f15SEric Blake for (i = 0; i < rc; i++) {
20668b96f15SEric Blake printf(" export: '%s'\n", list[i].name);
20768b96f15SEric Blake if (list[i].description && *list[i].description) {
20868b96f15SEric Blake printf(" description: %s\n", list[i].description);
20968b96f15SEric Blake }
21068b96f15SEric Blake if (list[i].flags & NBD_FLAG_HAS_FLAGS) {
211c4e2aff8SMax Reitz static const char *const flag_names[] = {
212c4e2aff8SMax Reitz [NBD_FLAG_READ_ONLY_BIT] = "readonly",
213c4e2aff8SMax Reitz [NBD_FLAG_SEND_FLUSH_BIT] = "flush",
214c4e2aff8SMax Reitz [NBD_FLAG_SEND_FUA_BIT] = "fua",
215c4e2aff8SMax Reitz [NBD_FLAG_ROTATIONAL_BIT] = "rotational",
216c4e2aff8SMax Reitz [NBD_FLAG_SEND_TRIM_BIT] = "trim",
217c4e2aff8SMax Reitz [NBD_FLAG_SEND_WRITE_ZEROES_BIT] = "zeroes",
218c4e2aff8SMax Reitz [NBD_FLAG_SEND_DF_BIT] = "df",
219c4e2aff8SMax Reitz [NBD_FLAG_CAN_MULTI_CONN_BIT] = "multi",
220c4e2aff8SMax Reitz [NBD_FLAG_SEND_RESIZE_BIT] = "resize",
221c4e2aff8SMax Reitz [NBD_FLAG_SEND_CACHE_BIT] = "cache",
2220a479545SEric Blake [NBD_FLAG_SEND_FAST_ZERO_BIT] = "fast-zero",
2232dcbb11bSEric Blake [NBD_FLAG_BLOCK_STAT_PAYLOAD_BIT] = "block-status-payload",
224c4e2aff8SMax Reitz };
225c4e2aff8SMax Reitz
22668b96f15SEric Blake printf(" size: %" PRIu64 "\n", list[i].size);
22768b96f15SEric Blake printf(" flags: 0x%x (", list[i].flags);
228c4e2aff8SMax Reitz for (size_t bit = 0; bit < ARRAY_SIZE(flag_names); bit++) {
229c4e2aff8SMax Reitz if (flag_names[bit] && (list[i].flags & (1 << bit))) {
230c4e2aff8SMax Reitz printf(" %s", flag_names[bit]);
23168b96f15SEric Blake }
23268b96f15SEric Blake }
23368b96f15SEric Blake printf(" )\n");
23468b96f15SEric Blake }
23568b96f15SEric Blake if (list[i].min_block) {
23668b96f15SEric Blake printf(" min block: %u\n", list[i].min_block);
23768b96f15SEric Blake printf(" opt block: %u\n", list[i].opt_block);
23868b96f15SEric Blake printf(" max block: %u\n", list[i].max_block);
23968b96f15SEric Blake }
24056cf9d04SEric Blake printf(" transaction size: %s\n",
24156cf9d04SEric Blake list[i].mode >= NBD_MODE_EXTENDED ?
24256cf9d04SEric Blake "64-bit" : "32-bit");
24368b96f15SEric Blake if (list[i].n_contexts) {
24468b96f15SEric Blake printf(" available meta contexts: %d\n", list[i].n_contexts);
24568b96f15SEric Blake for (j = 0; j < list[i].n_contexts; j++) {
24668b96f15SEric Blake printf(" %s\n", list[i].contexts[j]);
24768b96f15SEric Blake }
24868b96f15SEric Blake }
24968b96f15SEric Blake }
25068b96f15SEric Blake nbd_free_export_list(list, rc);
25168b96f15SEric Blake
25268b96f15SEric Blake ret = EXIT_SUCCESS;
25368b96f15SEric Blake out:
25468b96f15SEric Blake object_unref(OBJECT(sioc));
25568b96f15SEric Blake return ret;
25668b96f15SEric Blake }
25768b96f15SEric Blake
25868b96f15SEric Blake
2593484f676SDenis V. Lunev struct NbdClientOpts {
2603484f676SDenis V. Lunev char *device;
261b18d72d7SDenis V. Lunev char *srcpath;
262b2cecdfeSDenis V. Lunev SocketAddress *saddr;
263f6f4620cSDenis V. Lunev int old_stderr;
2643484f676SDenis V. Lunev bool fork_process;
2653484f676SDenis V. Lunev bool verbose;
2663484f676SDenis V. Lunev };
2673484f676SDenis V. Lunev
nbd_client_release_pipe(int old_stderr)268f6f4620cSDenis V. Lunev static void nbd_client_release_pipe(int old_stderr)
2692eb7c2abSDenis V. Lunev {
2702eb7c2abSDenis V. Lunev /* Close stderr so that the qemu-nbd process exits. */
271f6f4620cSDenis V. Lunev if (dup2(old_stderr, STDERR_FILENO) < 0) {
2722eb7c2abSDenis V. Lunev error_report("Could not release pipe to parent: %s",
2732eb7c2abSDenis V. Lunev strerror(errno));
2742eb7c2abSDenis V. Lunev exit(EXIT_FAILURE);
2752eb7c2abSDenis V. Lunev }
276f6f4620cSDenis V. Lunev if (old_stderr != STDOUT_FILENO && close(old_stderr) < 0) {
277f6f4620cSDenis V. Lunev error_report("Could not release qemu-nbd: %s", strerror(errno));
278f6f4620cSDenis V. Lunev exit(EXIT_FAILURE);
279f6f4620cSDenis V. Lunev }
2802eb7c2abSDenis V. Lunev }
2812eb7c2abSDenis V. Lunev
2823c1fa35dSEric Blake #if HAVE_NBD_DEVICE
show_parts(void * arg)283a517e88bSPaolo Bonzini static void *show_parts(void *arg)
284cd831bd7Sths {
285a6ac2313SPaolo Bonzini char *device = arg;
286cd831bd7Sths int nbd;
287cd831bd7Sths
288cd831bd7Sths /* linux just needs an open() to trigger
289cd831bd7Sths * the partition table update
290cd831bd7Sths * but remember to load the module with max_part != 0 :
291cd831bd7Sths * modprobe nbd max_part=63
292cd831bd7Sths */
293cd831bd7Sths nbd = open(device, O_RDWR);
294fc19f8a0SPaolo Bonzini if (nbd >= 0) {
295cd831bd7Sths close(nbd);
296cd831bd7Sths }
297a517e88bSPaolo Bonzini return NULL;
298a517e88bSPaolo Bonzini }
299a517e88bSPaolo Bonzini
nbd_client_thread(void * arg)300a517e88bSPaolo Bonzini static void *nbd_client_thread(void *arg)
301a517e88bSPaolo Bonzini {
30203b67621SDenis V. Lunev struct NbdClientOpts *opts = arg;
303ac132d05SEric Blake /* TODO: Revisit this if nbd.ko ever gains support for structured reply */
304ac132d05SEric Blake NBDExportInfo info = { .request_sizes = false, .name = g_strdup(""),
305ac132d05SEric Blake .mode = NBD_MODE_SIMPLE };
306d0d6ff58SDaniel P. Berrange QIOChannelSocket *sioc;
307af74b550SAlex Chen int fd = -1;
308af74b550SAlex Chen int ret = EXIT_FAILURE;
309a517e88bSPaolo Bonzini pthread_t show_parts_thread;
3101ce52846SMax Reitz Error *local_error = NULL;
311a517e88bSPaolo Bonzini
312d0d6ff58SDaniel P. Berrange sioc = qio_channel_socket_new();
313d0d6ff58SDaniel P. Berrange if (qio_channel_socket_connect_sync(sioc,
314b2cecdfeSDenis V. Lunev opts->saddr,
315d0d6ff58SDaniel P. Berrange &local_error) < 0) {
31648bec07eSDaniel P. Berrange error_report_err(local_error);
317a517e88bSPaolo Bonzini goto out;
318a517e88bSPaolo Bonzini }
319a517e88bSPaolo Bonzini
320b84ca91cSStefan Hajnoczi if (nbd_receive_negotiate(QIO_CHANNEL(sioc), NULL, NULL, NULL,
321b84ca91cSStefan Hajnoczi &info, &local_error) < 0) {
3221ce52846SMax Reitz if (local_error) {
32378288671SMarkus Armbruster error_report_err(local_error);
3241ce52846SMax Reitz }
325af74b550SAlex Chen goto out;
326a517e88bSPaolo Bonzini }
327a517e88bSPaolo Bonzini
32803b67621SDenis V. Lunev fd = open(opts->device, O_RDWR);
329fc19f8a0SPaolo Bonzini if (fd < 0) {
330a6ac2313SPaolo Bonzini /* Linux-only, we can use %m in printf. */
33103b67621SDenis V. Lunev error_report("Failed to open %s: %m", opts->device);
332af74b550SAlex Chen goto out;
333a6ac2313SPaolo Bonzini }
334a6ac2313SPaolo Bonzini
335af74b550SAlex Chen if (nbd_init(fd, sioc, &info, &local_error) < 0) {
336be41c100SVladimir Sementsov-Ogievskiy error_report_err(local_error);
337af74b550SAlex Chen goto out;
338a517e88bSPaolo Bonzini }
339a517e88bSPaolo Bonzini
340a517e88bSPaolo Bonzini /* update partition table */
34103b67621SDenis V. Lunev pthread_create(&show_parts_thread, NULL, show_parts, opts->device);
342a517e88bSPaolo Bonzini
343414c0cf0SDenis V. Lunev if (opts->verbose && !opts->fork_process) {
344a517e88bSPaolo Bonzini fprintf(stderr, "NBD device %s is now connected to %s\n",
345b18d72d7SDenis V. Lunev opts->device, opts->srcpath);
346c1f8fdc3SPaolo Bonzini } else {
347f6f4620cSDenis V. Lunev nbd_client_release_pipe(opts->old_stderr);
348c1f8fdc3SPaolo Bonzini }
349a517e88bSPaolo Bonzini
350af74b550SAlex Chen if (nbd_client(fd) < 0) {
351af74b550SAlex Chen goto out;
352a517e88bSPaolo Bonzini }
353a517e88bSPaolo Bonzini
354af74b550SAlex Chen ret = EXIT_SUCCESS;
355af74b550SAlex Chen
356a517e88bSPaolo Bonzini out:
357af74b550SAlex Chen if (fd >= 0) {
358af74b550SAlex Chen close(fd);
359af74b550SAlex Chen }
360af74b550SAlex Chen object_unref(OBJECT(sioc));
3616dc1667dSEric Blake g_free(info.name);
362a517e88bSPaolo Bonzini kill(getpid(), SIGTERM);
363af74b550SAlex Chen return (void *) (intptr_t) ret;
364cd831bd7Sths }
3653c1fa35dSEric Blake #endif /* HAVE_NBD_DEVICE */
366cd831bd7Sths
nbd_can_accept(void)367e4afbf4fSFam Zheng static int nbd_can_accept(void)
368a61c6782SPaolo Bonzini {
3693dcf56e6SEric Blake return state == RUNNING && (shared == 0 || nb_fds < shared);
370a61c6782SPaolo Bonzini }
371a61c6782SPaolo Bonzini
372d0d6ff58SDaniel P. Berrange static void nbd_update_server_watch(void);
373e4afbf4fSFam Zheng
nbd_client_closed(NBDClient * client,bool negotiated)3740c9390d9SEric Blake static void nbd_client_closed(NBDClient *client, bool negotiated)
375a61c6782SPaolo Bonzini {
376a61c6782SPaolo Bonzini nb_fds--;
3770c9390d9SEric Blake if (negotiated && nb_fds == 0 && !persistent && state == RUNNING) {
3787860a380SPaolo Bonzini state = TERMINATE;
3797860a380SPaolo Bonzini }
380d0d6ff58SDaniel P. Berrange nbd_update_server_watch();
3817860a380SPaolo Bonzini nbd_client_put(client);
382a61c6782SPaolo Bonzini }
383a61c6782SPaolo Bonzini
nbd_accept(QIONetListener * listener,QIOChannelSocket * cioc,gpointer opaque)384e4849c1dSDaniel P. Berrange static void nbd_accept(QIONetListener *listener, QIOChannelSocket *cioc,
385e4849c1dSDaniel P. Berrange gpointer opaque)
386a61c6782SPaolo Bonzini {
3877860a380SPaolo Bonzini if (state >= TERMINATE) {
388e4849c1dSDaniel P. Berrange return;
3897860a380SPaolo Bonzini }
3907860a380SPaolo Bonzini
391a61c6782SPaolo Bonzini nb_fds++;
392d0d6ff58SDaniel P. Berrange nbd_update_server_watch();
393b25e12daSDaniel P. Berrange nbd_client_new(cioc, tlscreds, tlsauthz, nbd_client_closed);
394d0d6ff58SDaniel P. Berrange }
395d0d6ff58SDaniel P. Berrange
nbd_update_server_watch(void)396d0d6ff58SDaniel P. Berrange static void nbd_update_server_watch(void)
397e4afbf4fSFam Zheng {
398e4afbf4fSFam Zheng if (nbd_can_accept()) {
399e4849c1dSDaniel P. Berrange qio_net_listener_set_client_func(server, nbd_accept, NULL, NULL);
400e4afbf4fSFam Zheng } else {
401e4849c1dSDaniel P. Berrange qio_net_listener_set_client_func(server, NULL, NULL, NULL);
402e4afbf4fSFam Zheng }
403e4afbf4fSFam Zheng }
404e4afbf4fSFam Zheng
40548bec07eSDaniel P. Berrange
nbd_build_socket_address(const char * sockpath,const char * bindto,const char * port)406bd269ebcSMarkus Armbruster static SocketAddress *nbd_build_socket_address(const char *sockpath,
40748bec07eSDaniel P. Berrange const char *bindto,
40848bec07eSDaniel P. Berrange const char *port)
40948bec07eSDaniel P. Berrange {
410bd269ebcSMarkus Armbruster SocketAddress *saddr;
41148bec07eSDaniel P. Berrange
412bd269ebcSMarkus Armbruster saddr = g_new0(SocketAddress, 1);
41348bec07eSDaniel P. Berrange if (sockpath) {
414bd269ebcSMarkus Armbruster saddr->type = SOCKET_ADDRESS_TYPE_UNIX;
415bd269ebcSMarkus Armbruster saddr->u.q_unix.path = g_strdup(sockpath);
41648bec07eSDaniel P. Berrange } else {
4170399293eSEric Blake InetSocketAddress *inet;
418bd269ebcSMarkus Armbruster saddr->type = SOCKET_ADDRESS_TYPE_INET;
419bd269ebcSMarkus Armbruster inet = &saddr->u.inet;
4200399293eSEric Blake inet->host = g_strdup(bindto);
42148bec07eSDaniel P. Berrange if (port) {
4220399293eSEric Blake inet->port = g_strdup(port);
42348bec07eSDaniel P. Berrange } else {
4240399293eSEric Blake inet->port = g_strdup_printf("%d", NBD_DEFAULT_PORT);
42548bec07eSDaniel P. Berrange }
42648bec07eSDaniel P. Berrange }
42748bec07eSDaniel P. Berrange
42848bec07eSDaniel P. Berrange return saddr;
42948bec07eSDaniel P. Berrange }
43048bec07eSDaniel P. Berrange
43148bec07eSDaniel P. Berrange
43277c9aaefSDaniel P. Berrange static QemuOptsList file_opts = {
43377c9aaefSDaniel P. Berrange .name = "file",
43477c9aaefSDaniel P. Berrange .implied_opt_name = "file",
43577c9aaefSDaniel P. Berrange .head = QTAILQ_HEAD_INITIALIZER(file_opts.head),
43677c9aaefSDaniel P. Berrange .desc = {
43777c9aaefSDaniel P. Berrange /* no elements => accept any params */
43877c9aaefSDaniel P. Berrange { /* end of list */ }
43977c9aaefSDaniel P. Berrange },
44077c9aaefSDaniel P. Berrange };
44177c9aaefSDaniel P. Berrange
nbd_get_tls_creds(const char * id,bool list,Error ** errp)44268b96f15SEric Blake static QCryptoTLSCreds *nbd_get_tls_creds(const char *id, bool list,
44368b96f15SEric Blake Error **errp)
444145614a1SDaniel P. Berrange {
445145614a1SDaniel P. Berrange Object *obj;
446145614a1SDaniel P. Berrange QCryptoTLSCreds *creds;
447145614a1SDaniel P. Berrange
448145614a1SDaniel P. Berrange obj = object_resolve_path_component(
449145614a1SDaniel P. Berrange object_get_objects_root(), id);
450145614a1SDaniel P. Berrange if (!obj) {
451145614a1SDaniel P. Berrange error_setg(errp, "No TLS credentials with id '%s'",
452145614a1SDaniel P. Berrange id);
453145614a1SDaniel P. Berrange return NULL;
454145614a1SDaniel P. Berrange }
455145614a1SDaniel P. Berrange creds = (QCryptoTLSCreds *)
456145614a1SDaniel P. Berrange object_dynamic_cast(obj, TYPE_QCRYPTO_TLS_CREDS);
457145614a1SDaniel P. Berrange if (!creds) {
458145614a1SDaniel P. Berrange error_setg(errp, "Object with id '%s' is not TLS credentials",
459145614a1SDaniel P. Berrange id);
460145614a1SDaniel P. Berrange return NULL;
461145614a1SDaniel P. Berrange }
462145614a1SDaniel P. Berrange
4630279cd95SPhilippe Mathieu-Daudé if (!qcrypto_tls_creds_check_endpoint(creds,
4640279cd95SPhilippe Mathieu-Daudé list
4650279cd95SPhilippe Mathieu-Daudé ? QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT
4660279cd95SPhilippe Mathieu-Daudé : QCRYPTO_TLS_CREDS_ENDPOINT_SERVER,
4670279cd95SPhilippe Mathieu-Daudé errp)) {
46868b96f15SEric Blake return NULL;
46968b96f15SEric Blake }
470145614a1SDaniel P. Berrange object_ref(obj);
471145614a1SDaniel P. Berrange return creds;
472145614a1SDaniel P. Berrange }
473145614a1SDaniel P. Berrange
setup_address_and_port(const char ** address,const char ** port)474a721f53bSRichard W.M. Jones static void setup_address_and_port(const char **address, const char **port)
475a721f53bSRichard W.M. Jones {
476a721f53bSRichard W.M. Jones if (*address == NULL) {
477a721f53bSRichard W.M. Jones *address = "0.0.0.0";
478a721f53bSRichard W.M. Jones }
479a721f53bSRichard W.M. Jones
480a721f53bSRichard W.M. Jones if (*port == NULL) {
481a721f53bSRichard W.M. Jones *port = stringify(NBD_DEFAULT_PORT);
482a721f53bSRichard W.M. Jones }
483a721f53bSRichard W.M. Jones }
484a721f53bSRichard W.M. Jones
485a721f53bSRichard W.M. Jones /*
486a721f53bSRichard W.M. Jones * Check socket parameters compatibility when socket activation is used.
487a721f53bSRichard W.M. Jones */
socket_activation_validate_opts(const char * device,const char * sockpath,const char * address,const char * port,const char * selinux,bool list)488a721f53bSRichard W.M. Jones static const char *socket_activation_validate_opts(const char *device,
489a721f53bSRichard W.M. Jones const char *sockpath,
490a721f53bSRichard W.M. Jones const char *address,
49168b96f15SEric Blake const char *port,
4923d212b41SRichard W.M. Jones const char *selinux,
49368b96f15SEric Blake bool list)
494a721f53bSRichard W.M. Jones {
495a721f53bSRichard W.M. Jones if (device != NULL) {
496a721f53bSRichard W.M. Jones return "NBD device can't be set when using socket activation";
497a721f53bSRichard W.M. Jones }
498a721f53bSRichard W.M. Jones
499a721f53bSRichard W.M. Jones if (sockpath != NULL) {
500a721f53bSRichard W.M. Jones return "Unix socket can't be set when using socket activation";
501a721f53bSRichard W.M. Jones }
502a721f53bSRichard W.M. Jones
503a721f53bSRichard W.M. Jones if (address != NULL) {
504a721f53bSRichard W.M. Jones return "The interface can't be set when using socket activation";
505a721f53bSRichard W.M. Jones }
506a721f53bSRichard W.M. Jones
507a721f53bSRichard W.M. Jones if (port != NULL) {
508a721f53bSRichard W.M. Jones return "TCP port number can't be set when using socket activation";
509a721f53bSRichard W.M. Jones }
510a721f53bSRichard W.M. Jones
5113d212b41SRichard W.M. Jones if (selinux != NULL) {
5123d212b41SRichard W.M. Jones return "SELinux label can't be set when using socket activation";
5133d212b41SRichard W.M. Jones }
5143d212b41SRichard W.M. Jones
51568b96f15SEric Blake if (list) {
51668b96f15SEric Blake return "List mode is incompatible with socket activation";
51768b96f15SEric Blake }
51868b96f15SEric Blake
519a721f53bSRichard W.M. Jones return NULL;
520a721f53bSRichard W.M. Jones }
521145614a1SDaniel P. Berrange
qemu_nbd_shutdown(void)522b3b5299dSKevin Wolf static void qemu_nbd_shutdown(void)
523b3b5299dSKevin Wolf {
524b3b5299dSKevin Wolf job_cancel_sync_all();
5251895b977SSergio Lopez blk_exp_close_all();
526b3b5299dSKevin Wolf bdrv_close_all();
527b3b5299dSKevin Wolf }
528b3b5299dSKevin Wolf
main(int argc,char ** argv)5297a5ca864Sbellard int main(int argc, char **argv)
5307a5ca864Sbellard {
53126f54e9aSMarkus Armbruster BlockBackend *blk;
5327a5ca864Sbellard BlockDriverState *bs;
5339d26dfcbSEric Blake uint64_t dev_offset = 0;
534dbb38caaSEric Blake bool readonly = false;
535cd831bd7Sths bool disconnect = false;
536a721f53bSRichard W.M. Jones const char *bindto = NULL;
53748bec07eSDaniel P. Berrange const char *port = NULL;
53848bec07eSDaniel P. Berrange char *sockpath = NULL;
5398c116b0eSWenchao Xia QemuOpts *sn_opts = NULL;
5408c116b0eSWenchao Xia const char *sn_id_or_name = NULL;
541dbc7b014SEric Blake const char *sopt = "hVb:o:p:rsnc:dvk:e:f:tl:x:T:D:AB:L";
5427a5ca864Sbellard struct option lopt[] = {
543aa6e546cSDaniel P. Berrange { "help", no_argument, NULL, 'h' },
544aa6e546cSDaniel P. Berrange { "version", no_argument, NULL, 'V' },
545aa6e546cSDaniel P. Berrange { "bind", required_argument, NULL, 'b' },
546aa6e546cSDaniel P. Berrange { "port", required_argument, NULL, 'p' },
547aa6e546cSDaniel P. Berrange { "socket", required_argument, NULL, 'k' },
548aa6e546cSDaniel P. Berrange { "offset", required_argument, NULL, 'o' },
549aa6e546cSDaniel P. Berrange { "read-only", no_argument, NULL, 'r' },
550dbc7b014SEric Blake { "allocation-depth", no_argument, NULL, 'A' },
551636192c4SEric Blake { "bitmap", required_argument, NULL, 'B' },
552aa6e546cSDaniel P. Berrange { "connect", required_argument, NULL, 'c' },
553aa6e546cSDaniel P. Berrange { "disconnect", no_argument, NULL, 'd' },
55468b96f15SEric Blake { "list", no_argument, NULL, 'L' },
555aa6e546cSDaniel P. Berrange { "snapshot", no_argument, NULL, 's' },
556aa6e546cSDaniel P. Berrange { "load-snapshot", required_argument, NULL, 'l' },
557aa6e546cSDaniel P. Berrange { "nocache", no_argument, NULL, 'n' },
558aa6e546cSDaniel P. Berrange { "cache", required_argument, NULL, QEMU_NBD_OPT_CACHE },
559aa6e546cSDaniel P. Berrange { "aio", required_argument, NULL, QEMU_NBD_OPT_AIO },
560aa6e546cSDaniel P. Berrange { "discard", required_argument, NULL, QEMU_NBD_OPT_DISCARD },
561aa6e546cSDaniel P. Berrange { "detect-zeroes", required_argument, NULL,
562aa6e546cSDaniel P. Berrange QEMU_NBD_OPT_DETECT_ZEROES },
563aa6e546cSDaniel P. Berrange { "shared", required_argument, NULL, 'e' },
564aa6e546cSDaniel P. Berrange { "format", required_argument, NULL, 'f' },
565aa6e546cSDaniel P. Berrange { "persistent", no_argument, NULL, 't' },
566aa6e546cSDaniel P. Berrange { "verbose", no_argument, NULL, 'v' },
567aa6e546cSDaniel P. Berrange { "object", required_argument, NULL, QEMU_NBD_OPT_OBJECT },
568aa6e546cSDaniel P. Berrange { "export-name", required_argument, NULL, 'x' },
569b1a75b33SEric Blake { "description", required_argument, NULL, 'D' },
570aa6e546cSDaniel P. Berrange { "tls-creds", required_argument, NULL, QEMU_NBD_OPT_TLSCREDS },
571003b2b25SDaniel P. Berrangé { "tls-hostname", required_argument, NULL, QEMU_NBD_OPT_TLSHOSTNAME },
572b25e12daSDaniel P. Berrange { "tls-authz", required_argument, NULL, QEMU_NBD_OPT_TLSAUTHZ },
573aa6e546cSDaniel P. Berrange { "image-opts", no_argument, NULL, QEMU_NBD_OPT_IMAGE_OPTS },
57439ca463eSDenis V. Lunev { "trace", required_argument, NULL, 'T' },
575ffb31e1dSMax Reitz { "fork", no_argument, NULL, QEMU_NBD_OPT_FORK },
576637bc5a5SMax Reitz { "pid-file", required_argument, NULL, QEMU_NBD_OPT_PID_FILE },
5773d212b41SRichard W.M. Jones { "selinux-label", required_argument, NULL,
5783d212b41SRichard W.M. Jones QEMU_NBD_OPT_SELINUX_LABEL },
579660f11beSBlue Swirl { NULL, 0, NULL, 0 }
5807a5ca864Sbellard };
5817a5ca864Sbellard int ch;
5827a5ca864Sbellard int opt_ind = 0;
583f5edb014SNaphtali Sprei int flags = BDRV_O_RDWR;
5844fbec260SMax Reitz int ret = 0;
58539a5235cSPaolo Bonzini bool seen_cache = false;
586ded9d2d5SPaolo Bonzini bool seen_discard = false;
58739a5235cSPaolo Bonzini bool seen_aio = false;
588a517e88bSPaolo Bonzini pthread_t client_thread;
589e6b63677SDaniel P. Berrange const char *fmt = NULL;
59034b5d2c6SMax Reitz Error *local_err = NULL;
591b3838a40SPeter Lieven BlockdevDetectZeroesOptions detect_zeroes = BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF;
5924fbec260SMax Reitz QDict *options = NULL;
59368b96f15SEric Blake const char *export_name = NULL; /* defaults to "" later for server mode */
594b1a75b33SEric Blake const char *export_description = NULL;
595e5fb29d5SVladimir Sementsov-Ogievskiy BlockDirtyBitmapOrStrList *bitmaps = NULL;
596dbc7b014SEric Blake bool alloc_depth = false;
597145614a1SDaniel P. Berrange const char *tlscredsid = NULL;
598003b2b25SDaniel P. Berrangé const char *tlshostname = NULL;
59977c9aaefSDaniel P. Berrange bool imageOpts = false;
60009615257SNir Soffer bool writethrough = false; /* Client will flush as needed. */
60168b96f15SEric Blake bool list = false;
602a721f53bSRichard W.M. Jones unsigned socket_activation;
603637bc5a5SMax Reitz const char *pid_file_name = NULL;
6043d212b41SRichard W.M. Jones const char *selinux_label = NULL;
60500917172SKevin Wolf BlockExportOptions *export_opts;
6063484f676SDenis V. Lunev struct NbdClientOpts opts = {
6073484f676SDenis V. Lunev .fork_process = false,
6083484f676SDenis V. Lunev .verbose = false,
6093484f676SDenis V. Lunev .device = NULL,
610b18d72d7SDenis V. Lunev .srcpath = NULL,
611b2cecdfeSDenis V. Lunev .saddr = NULL,
612f6f4620cSDenis V. Lunev .old_stderr = STDOUT_FILENO,
6133484f676SDenis V. Lunev };
6147a5ca864Sbellard
615029a88c9SEric Blake #ifdef CONFIG_POSIX
616cbc20bfbSStefan Hajnoczi os_setup_early_signal_handling();
617cbc20bfbSStefan Hajnoczi os_setup_signal_handling();
618041e32b8SMax Reitz #endif
619041e32b8SMax Reitz
62098c5d2e7SDaniel P. Berrangé socket_init();
621f5852efaSChristophe Fergeau error_init(argv[0]);
622fe4db84dSDaniel P. Berrange module_call_init(MODULE_INIT_TRACE);
623e8f2d272SEduardo Habkost qcrypto_init(&error_fatal);
624c2297088SDaniel P. Berrange
6250ab3b337SDaniel P. Berrange module_call_init(MODULE_INIT_QOM);
62639ca463eSDenis V. Lunev qemu_add_opts(&qemu_trace_opts);
62710f5bff6SFam Zheng qemu_init_exec_dir(argv[0]);
628bb345110SPaolo Bonzini
6297a5ca864Sbellard while ((ch = getopt_long(argc, argv, sopt, lopt, &opt_ind)) != -1) {
6307a5ca864Sbellard switch (ch) {
6317a5ca864Sbellard case 's':
6322f726488Sths flags |= BDRV_O_SNAPSHOT;
6332f726488Sths break;
6342f726488Sths case 'n':
63539a5235cSPaolo Bonzini optarg = (char *) "none";
63639a5235cSPaolo Bonzini /* fallthrough */
63739a5235cSPaolo Bonzini case QEMU_NBD_OPT_CACHE:
63839a5235cSPaolo Bonzini if (seen_cache) {
63985b01e09SMarkus Armbruster error_report("-n and --cache can only be specified once");
64085b01e09SMarkus Armbruster exit(EXIT_FAILURE);
64139a5235cSPaolo Bonzini }
64239a5235cSPaolo Bonzini seen_cache = true;
6436effd5bfSKevin Wolf if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) == -1) {
64485b01e09SMarkus Armbruster error_report("Invalid cache mode `%s'", optarg);
64585b01e09SMarkus Armbruster exit(EXIT_FAILURE);
64639a5235cSPaolo Bonzini }
6477a5ca864Sbellard break;
64839a5235cSPaolo Bonzini case QEMU_NBD_OPT_AIO:
64939a5235cSPaolo Bonzini if (seen_aio) {
65085b01e09SMarkus Armbruster error_report("--aio can only be specified once");
65185b01e09SMarkus Armbruster exit(EXIT_FAILURE);
65239a5235cSPaolo Bonzini }
65339a5235cSPaolo Bonzini seen_aio = true;
6547680274dSAarushi Mehta if (bdrv_parse_aio(optarg, &flags) < 0) {
6557680274dSAarushi Mehta error_report("Invalid aio mode '%s'", optarg);
65685b01e09SMarkus Armbruster exit(EXIT_FAILURE);
65739a5235cSPaolo Bonzini }
65839a5235cSPaolo Bonzini break;
659ded9d2d5SPaolo Bonzini case QEMU_NBD_OPT_DISCARD:
660ded9d2d5SPaolo Bonzini if (seen_discard) {
66185b01e09SMarkus Armbruster error_report("--discard can only be specified once");
66285b01e09SMarkus Armbruster exit(EXIT_FAILURE);
663ded9d2d5SPaolo Bonzini }
664ded9d2d5SPaolo Bonzini seen_discard = true;
665ded9d2d5SPaolo Bonzini if (bdrv_parse_discard_flags(optarg, &flags) == -1) {
66685b01e09SMarkus Armbruster error_report("Invalid discard mode `%s'", optarg);
66785b01e09SMarkus Armbruster exit(EXIT_FAILURE);
668ded9d2d5SPaolo Bonzini }
669ded9d2d5SPaolo Bonzini break;
670b3838a40SPeter Lieven case QEMU_NBD_OPT_DETECT_ZEROES:
671b3838a40SPeter Lieven detect_zeroes =
672f7abe0ecSMarc-André Lureau qapi_enum_parse(&BlockdevDetectZeroesOptions_lookup,
673b3838a40SPeter Lieven optarg,
674b3838a40SPeter Lieven BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF,
675b3838a40SPeter Lieven &local_err);
676b3838a40SPeter Lieven if (local_err) {
677c29b77f9SMarkus Armbruster error_reportf_err(local_err,
678c29b77f9SMarkus Armbruster "Failed to parse detect_zeroes mode: ");
67985b01e09SMarkus Armbruster exit(EXIT_FAILURE);
680b3838a40SPeter Lieven }
681b3838a40SPeter Lieven if (detect_zeroes == BLOCKDEV_DETECT_ZEROES_OPTIONS_UNMAP &&
682b3838a40SPeter Lieven !(flags & BDRV_O_UNMAP)) {
68385b01e09SMarkus Armbruster error_report("setting detect-zeroes to unmap is not allowed "
684b3838a40SPeter Lieven "without setting discard operation to unmap");
68585b01e09SMarkus Armbruster exit(EXIT_FAILURE);
686b3838a40SPeter Lieven }
687b3838a40SPeter Lieven break;
6887a5ca864Sbellard case 'b':
6897a5ca864Sbellard bindto = optarg;
6907a5ca864Sbellard break;
6917a5ca864Sbellard case 'p':
69248bec07eSDaniel P. Berrange port = optarg;
6937a5ca864Sbellard break;
6947a5ca864Sbellard case 'o':
69543b51011SEric Blake if (qemu_strtou64(optarg, NULL, 0, &dev_offset) < 0) {
69643b51011SEric Blake error_report("Invalid offset '%s'", optarg);
69785b01e09SMarkus Armbruster exit(EXIT_FAILURE);
6987a5ca864Sbellard }
6997a5ca864Sbellard break;
7008c116b0eSWenchao Xia case 'l':
7018c116b0eSWenchao Xia if (strstart(optarg, SNAPSHOT_OPT_BASE, NULL)) {
70270b94331SMarkus Armbruster sn_opts = qemu_opts_parse_noisily(&internal_snapshot_opts,
70370b94331SMarkus Armbruster optarg, false);
7048c116b0eSWenchao Xia if (!sn_opts) {
70585b01e09SMarkus Armbruster error_report("Failed in parsing snapshot param `%s'",
7068c116b0eSWenchao Xia optarg);
70785b01e09SMarkus Armbruster exit(EXIT_FAILURE);
7088c116b0eSWenchao Xia }
7098c116b0eSWenchao Xia } else {
7108c116b0eSWenchao Xia sn_id_or_name = optarg;
7118c116b0eSWenchao Xia }
7128c116b0eSWenchao Xia /* fall through */
7137a5ca864Sbellard case 'r':
714dbb38caaSEric Blake readonly = true;
71507108b29SNaphtali Sprei flags &= ~BDRV_O_RDWR;
7167a5ca864Sbellard break;
717dbc7b014SEric Blake case 'A':
718dbc7b014SEric Blake alloc_depth = true;
719dbc7b014SEric Blake break;
720636192c4SEric Blake case 'B':
721e5fb29d5SVladimir Sementsov-Ogievskiy {
722e5fb29d5SVladimir Sementsov-Ogievskiy BlockDirtyBitmapOrStr *el = g_new(BlockDirtyBitmapOrStr, 1);
723e5fb29d5SVladimir Sementsov-Ogievskiy *el = (BlockDirtyBitmapOrStr) {
724e5fb29d5SVladimir Sementsov-Ogievskiy .type = QTYPE_QSTRING,
725e5fb29d5SVladimir Sementsov-Ogievskiy .u.local = g_strdup(optarg),
726e5fb29d5SVladimir Sementsov-Ogievskiy };
727e5fb29d5SVladimir Sementsov-Ogievskiy QAPI_LIST_PREPEND(bitmaps, el);
728e5fb29d5SVladimir Sementsov-Ogievskiy }
729636192c4SEric Blake break;
730cd831bd7Sths case 'k':
731b32f6c28SPaolo Bonzini sockpath = optarg;
732713cc671SPeter Lieven if (sockpath[0] != '/') {
7339af9e0feSMarkus Armbruster error_report("socket path must be absolute");
73485b01e09SMarkus Armbruster exit(EXIT_FAILURE);
735713cc671SPeter Lieven }
736cd831bd7Sths break;
737cd831bd7Sths case 'd':
738cd831bd7Sths disconnect = true;
739cd831bd7Sths break;
740cd831bd7Sths case 'c':
7413484f676SDenis V. Lunev opts.device = optarg;
742cd831bd7Sths break;
7433b05a8e9Sths case 'e':
74443b51011SEric Blake if (qemu_strtoi(optarg, NULL, 0, &shared) < 0 ||
7453dcf56e6SEric Blake shared < 0) {
74685b01e09SMarkus Armbruster error_report("Invalid shared device number '%s'", optarg);
74785b01e09SMarkus Armbruster exit(EXIT_FAILURE);
7483b05a8e9Sths }
7493b05a8e9Sths break;
750e6b63677SDaniel P. Berrange case 'f':
751e6b63677SDaniel P. Berrange fmt = optarg;
752e6b63677SDaniel P. Berrange break;
75375818250Sths case 't':
75475818250Sths persistent = 1;
75575818250Sths break;
7563d4b2f9cSDaniel P. Berrange case 'x':
7573d4b2f9cSDaniel P. Berrange export_name = optarg;
75893676c88SEric Blake if (strlen(export_name) > NBD_MAX_STRING_SIZE) {
75993676c88SEric Blake error_report("export name '%s' too long", export_name);
76093676c88SEric Blake exit(EXIT_FAILURE);
76193676c88SEric Blake }
7623d4b2f9cSDaniel P. Berrange break;
763b1a75b33SEric Blake case 'D':
764b1a75b33SEric Blake export_description = optarg;
76593676c88SEric Blake if (strlen(export_description) > NBD_MAX_STRING_SIZE) {
76693676c88SEric Blake error_report("export description '%s' too long",
76793676c88SEric Blake export_description);
76893676c88SEric Blake exit(EXIT_FAILURE);
76993676c88SEric Blake }
770b1a75b33SEric Blake break;
7717a5ca864Sbellard case 'v':
7723484f676SDenis V. Lunev opts.verbose = true;
7737a5ca864Sbellard break;
7747a5ca864Sbellard case 'V':
7757a5ca864Sbellard version(argv[0]);
7767a5ca864Sbellard exit(0);
7777a5ca864Sbellard break;
7787a5ca864Sbellard case 'h':
7797a5ca864Sbellard usage(argv[0]);
7807a5ca864Sbellard exit(0);
7817a5ca864Sbellard break;
7827a5ca864Sbellard case '?':
78385b01e09SMarkus Armbruster error_report("Try `%s --help' for more information.", argv[0]);
78485b01e09SMarkus Armbruster exit(EXIT_FAILURE);
785fa40e43cSKevin Wolf case QEMU_NBD_OPT_OBJECT:
786fa40e43cSKevin Wolf user_creatable_process_cmdline(optarg);
787fa40e43cSKevin Wolf break;
788145614a1SDaniel P. Berrange case QEMU_NBD_OPT_TLSCREDS:
789145614a1SDaniel P. Berrange tlscredsid = optarg;
790145614a1SDaniel P. Berrange break;
791003b2b25SDaniel P. Berrangé case QEMU_NBD_OPT_TLSHOSTNAME:
792003b2b25SDaniel P. Berrangé tlshostname = optarg;
793003b2b25SDaniel P. Berrangé break;
79477c9aaefSDaniel P. Berrange case QEMU_NBD_OPT_IMAGE_OPTS:
79577c9aaefSDaniel P. Berrange imageOpts = true;
79677c9aaefSDaniel P. Berrange break;
79739ca463eSDenis V. Lunev case 'T':
79892eecfffSPaolo Bonzini trace_opt_parse(optarg);
79939ca463eSDenis V. Lunev break;
800b25e12daSDaniel P. Berrange case QEMU_NBD_OPT_TLSAUTHZ:
801b25e12daSDaniel P. Berrange tlsauthz = optarg;
802b25e12daSDaniel P. Berrange break;
803ffb31e1dSMax Reitz case QEMU_NBD_OPT_FORK:
8043484f676SDenis V. Lunev opts.fork_process = true;
805ffb31e1dSMax Reitz break;
80668b96f15SEric Blake case 'L':
80768b96f15SEric Blake list = true;
80868b96f15SEric Blake break;
809637bc5a5SMax Reitz case QEMU_NBD_OPT_PID_FILE:
810637bc5a5SMax Reitz pid_file_name = optarg;
811637bc5a5SMax Reitz break;
8123d212b41SRichard W.M. Jones case QEMU_NBD_OPT_SELINUX_LABEL:
8133d212b41SRichard W.M. Jones selinux_label = optarg;
8143d212b41SRichard W.M. Jones break;
8157a5ca864Sbellard }
8167a5ca864Sbellard }
8177a5ca864Sbellard
81868b96f15SEric Blake if (list) {
81968b96f15SEric Blake if (argc != optind) {
82068b96f15SEric Blake error_report("List mode is incompatible with a file name");
82168b96f15SEric Blake exit(EXIT_FAILURE);
82268b96f15SEric Blake }
8230bc16997SEric Blake if (export_name || export_description || dev_offset ||
8243484f676SDenis V. Lunev opts.device || disconnect || fmt || sn_id_or_name || bitmaps ||
825dbc7b014SEric Blake alloc_depth || seen_aio || seen_discard || seen_cache) {
82668b96f15SEric Blake error_report("List mode is incompatible with per-device settings");
82768b96f15SEric Blake exit(EXIT_FAILURE);
82868b96f15SEric Blake }
8293484f676SDenis V. Lunev if (opts.fork_process) {
83068b96f15SEric Blake error_report("List mode is incompatible with forking");
83168b96f15SEric Blake exit(EXIT_FAILURE);
83268b96f15SEric Blake }
83368b96f15SEric Blake } else if ((argc - optind) != 1) {
834433672b0SMarkus Armbruster error_report("Invalid number of arguments");
835433672b0SMarkus Armbruster error_printf("Try `%s --help' for more information.\n", argv[0]);
83685b01e09SMarkus Armbruster exit(EXIT_FAILURE);
83768b96f15SEric Blake } else if (!export_name) {
83868b96f15SEric Blake export_name = "";
8397a5ca864Sbellard }
8407a5ca864Sbellard
84139ca463eSDenis V. Lunev if (!trace_init_backends()) {
84239ca463eSDenis V. Lunev exit(1);
84339ca463eSDenis V. Lunev }
84492eecfffSPaolo Bonzini trace_init_file();
845c5955f4fSRichard Henderson qemu_set_log(LOG_TRACE, &error_fatal);
84639ca463eSDenis V. Lunev
847a721f53bSRichard W.M. Jones socket_activation = check_socket_activation();
848a721f53bSRichard W.M. Jones if (socket_activation == 0) {
849e8ae8b1aSDaniel P. Berrangé if (!sockpath) {
850a721f53bSRichard W.M. Jones setup_address_and_port(&bindto, &port);
851e8ae8b1aSDaniel P. Berrangé }
852a721f53bSRichard W.M. Jones } else {
853a721f53bSRichard W.M. Jones /* Using socket activation - check user didn't use -p etc. */
8543484f676SDenis V. Lunev const char *err_msg = socket_activation_validate_opts(opts.device,
8553484f676SDenis V. Lunev sockpath,
85668b96f15SEric Blake bindto, port,
8573d212b41SRichard W.M. Jones selinux_label,
85868b96f15SEric Blake list);
859a721f53bSRichard W.M. Jones if (err_msg != NULL) {
860a721f53bSRichard W.M. Jones error_report("%s", err_msg);
861a721f53bSRichard W.M. Jones exit(EXIT_FAILURE);
862a721f53bSRichard W.M. Jones }
86353fabd4bSPaolo Bonzini
86453fabd4bSPaolo Bonzini /* qemu-nbd can only listen on a single socket. */
86553fabd4bSPaolo Bonzini if (socket_activation > 1) {
86653fabd4bSPaolo Bonzini error_report("qemu-nbd does not support socket activation with %s > 1",
86753fabd4bSPaolo Bonzini "LISTEN_FDS");
86853fabd4bSPaolo Bonzini exit(EXIT_FAILURE);
86953fabd4bSPaolo Bonzini }
870a721f53bSRichard W.M. Jones }
871a721f53bSRichard W.M. Jones
872145614a1SDaniel P. Berrange if (tlscredsid) {
8733484f676SDenis V. Lunev if (opts.device) {
874145614a1SDaniel P. Berrange error_report("TLS is not supported with a host device");
875145614a1SDaniel P. Berrange exit(EXIT_FAILURE);
876145614a1SDaniel P. Berrange }
877b25e12daSDaniel P. Berrange if (tlsauthz && list) {
878b25e12daSDaniel P. Berrange error_report("TLS authorization is incompatible with export list");
879b25e12daSDaniel P. Berrange exit(EXIT_FAILURE);
880b25e12daSDaniel P. Berrange }
881003b2b25SDaniel P. Berrangé if (tlshostname && !list) {
882003b2b25SDaniel P. Berrangé error_report("TLS hostname is only supported with export list");
883003b2b25SDaniel P. Berrangé exit(EXIT_FAILURE);
884003b2b25SDaniel P. Berrangé }
88568b96f15SEric Blake tlscreds = nbd_get_tls_creds(tlscredsid, list, &local_err);
886145614a1SDaniel P. Berrange if (local_err) {
8875217f188SMarkus Armbruster error_reportf_err(local_err, "Failed to get TLS creds: ");
888145614a1SDaniel P. Berrange exit(EXIT_FAILURE);
889145614a1SDaniel P. Berrange }
890b25e12daSDaniel P. Berrange } else {
891b25e12daSDaniel P. Berrange if (tlsauthz) {
892b25e12daSDaniel P. Berrange error_report("--tls-authz is not permitted without --tls-creds");
893b25e12daSDaniel P. Berrange exit(EXIT_FAILURE);
894b25e12daSDaniel P. Berrange }
895003b2b25SDaniel P. Berrangé if (tlshostname) {
896003b2b25SDaniel P. Berrangé error_report("--tls-hostname is not permitted without --tls-creds");
897003b2b25SDaniel P. Berrangé exit(EXIT_FAILURE);
898003b2b25SDaniel P. Berrangé }
899145614a1SDaniel P. Berrange }
900145614a1SDaniel P. Berrange
9013d212b41SRichard W.M. Jones if (selinux_label) {
9023d212b41SRichard W.M. Jones #ifdef CONFIG_SELINUX
9033484f676SDenis V. Lunev if (sockpath == NULL && opts.device == NULL) {
9043d212b41SRichard W.M. Jones error_report("--selinux-label is not permitted without --socket");
9053d212b41SRichard W.M. Jones exit(EXIT_FAILURE);
9063d212b41SRichard W.M. Jones }
9073d212b41SRichard W.M. Jones #else
9083d212b41SRichard W.M. Jones error_report("SELinux support not enabled in this binary");
9093d212b41SRichard W.M. Jones exit(EXIT_FAILURE);
9103d212b41SRichard W.M. Jones #endif
9113d212b41SRichard W.M. Jones }
9123d212b41SRichard W.M. Jones
91368b96f15SEric Blake if (list) {
914b2cecdfeSDenis V. Lunev opts.saddr = nbd_build_socket_address(sockpath, bindto, port);
915b2cecdfeSDenis V. Lunev return qemu_nbd_client_list(opts.saddr, tlscreds,
916003b2b25SDaniel P. Berrangé tlshostname ? tlshostname : bindto);
91768b96f15SEric Blake }
91868b96f15SEric Blake
9193c1fa35dSEric Blake #if !HAVE_NBD_DEVICE
9203484f676SDenis V. Lunev if (disconnect || opts.device) {
9213c1fa35dSEric Blake error_report("Kernel /dev/nbdN support not available");
9223c1fa35dSEric Blake exit(EXIT_FAILURE);
9233c1fa35dSEric Blake }
9243c1fa35dSEric Blake #else /* HAVE_NBD_DEVICE */
925cd831bd7Sths if (disconnect) {
926d0d6ff58SDaniel P. Berrange int nbdfd = open(argv[optind], O_RDWR);
927d0d6ff58SDaniel P. Berrange if (nbdfd < 0) {
92885b01e09SMarkus Armbruster error_report("Cannot open %s: %s", argv[optind],
92985b01e09SMarkus Armbruster strerror(errno));
93085b01e09SMarkus Armbruster exit(EXIT_FAILURE);
931fc19f8a0SPaolo Bonzini }
932d0d6ff58SDaniel P. Berrange nbd_disconnect(nbdfd);
933cd831bd7Sths
934d0d6ff58SDaniel P. Berrange close(nbdfd);
935cd831bd7Sths
936cd831bd7Sths printf("%s disconnected\n", argv[optind]);
937cd831bd7Sths
938cd831bd7Sths return 0;
939cd831bd7Sths }
9403c1fa35dSEric Blake #endif
941cd831bd7Sths
9423484f676SDenis V. Lunev if ((opts.device && !opts.verbose) || opts.fork_process) {
943eb705985SDaniel P. Berrangé #ifndef WIN32
944a7241974SMarc-André Lureau g_autoptr(GError) err = NULL;
945c1f8fdc3SPaolo Bonzini int stderr_fd[2];
946c1f8fdc3SPaolo Bonzini pid_t pid;
947c1f8fdc3SPaolo Bonzini
948a7241974SMarc-André Lureau if (!g_unix_open_pipe(stderr_fd, FD_CLOEXEC, &err)) {
94985b01e09SMarkus Armbruster error_report("Error setting up communication pipe: %s",
950a7241974SMarc-André Lureau err->message);
95185b01e09SMarkus Armbruster exit(EXIT_FAILURE);
952c1f8fdc3SPaolo Bonzini }
953c1f8fdc3SPaolo Bonzini
954c1f8fdc3SPaolo Bonzini /* Now daemonize, but keep a communication channel open to
955c1f8fdc3SPaolo Bonzini * print errors and exit with the proper status code.
956c1f8fdc3SPaolo Bonzini */
957c1f8fdc3SPaolo Bonzini pid = fork();
95870d4739eSMax Reitz if (pid < 0) {
95985b01e09SMarkus Armbruster error_report("Failed to fork: %s", strerror(errno));
96085b01e09SMarkus Armbruster exit(EXIT_FAILURE);
96170d4739eSMax Reitz } else if (pid == 0) {
9621dc82151SDenis V. Lunev int saved_errno;
9631dc82151SDenis V. Lunev
964c1f8fdc3SPaolo Bonzini close(stderr_fd[0]);
965e6df58a5SMax Reitz
966f6f4620cSDenis V. Lunev /* Remember parent's stderr if we will be restoring it. */
967f6f4620cSDenis V. Lunev if (opts.verbose /* fork_process is set */) {
968f6f4620cSDenis V. Lunev opts.old_stderr = dup(STDERR_FILENO);
969f6f4620cSDenis V. Lunev if (opts.old_stderr < 0) {
970f6f4620cSDenis V. Lunev error_report("Could not dup original stderr: %s",
971f6f4620cSDenis V. Lunev strerror(errno));
972f6f4620cSDenis V. Lunev exit(EXIT_FAILURE);
973f6f4620cSDenis V. Lunev }
974f6f4620cSDenis V. Lunev }
975f6f4620cSDenis V. Lunev
9769faf31b6SMichael Tokarev ret = qemu_daemon(1, 0);
9771dc82151SDenis V. Lunev saved_errno = errno; /* dup2 will overwrite error below */
978c1f8fdc3SPaolo Bonzini
979c1f8fdc3SPaolo Bonzini /* Temporarily redirect stderr to the parent's pipe... */
980e0892cedSDenis V. Lunev if (dup2(stderr_fd[1], STDERR_FILENO) < 0) {
981e0892cedSDenis V. Lunev char str[256];
982e0892cedSDenis V. Lunev snprintf(str, sizeof(str),
983e0892cedSDenis V. Lunev "%s: Failed to link stderr to the pipe: %s\n",
984e0892cedSDenis V. Lunev g_get_prgname(), strerror(errno));
985e0892cedSDenis V. Lunev /*
986e0892cedSDenis V. Lunev * We are unable to use error_report() here as we need to get
987e0892cedSDenis V. Lunev * stderr pointed to the parent's pipe. Write to that pipe
988e0892cedSDenis V. Lunev * manually.
989e0892cedSDenis V. Lunev */
990e0892cedSDenis V. Lunev ret = write(stderr_fd[1], str, strlen(str));
991e0892cedSDenis V. Lunev exit(EXIT_FAILURE);
992e0892cedSDenis V. Lunev }
993e0892cedSDenis V. Lunev
994fc19f8a0SPaolo Bonzini if (ret < 0) {
9951dc82151SDenis V. Lunev error_report("Failed to daemonize: %s", strerror(saved_errno));
99685b01e09SMarkus Armbruster exit(EXIT_FAILURE);
997c1f8fdc3SPaolo Bonzini }
998c1f8fdc3SPaolo Bonzini
999c1f8fdc3SPaolo Bonzini /* ... close the descriptor we inherited and go on. */
1000c1f8fdc3SPaolo Bonzini close(stderr_fd[1]);
1001c1f8fdc3SPaolo Bonzini } else {
1002c1f8fdc3SPaolo Bonzini bool errors = false;
1003c1f8fdc3SPaolo Bonzini char *buf;
1004c1f8fdc3SPaolo Bonzini
1005c1f8fdc3SPaolo Bonzini /* In the parent. Print error messages from the child until
1006c1f8fdc3SPaolo Bonzini * it closes the pipe.
1007c1f8fdc3SPaolo Bonzini */
1008c1f8fdc3SPaolo Bonzini close(stderr_fd[1]);
1009c1f8fdc3SPaolo Bonzini buf = g_malloc(1024);
1010c1f8fdc3SPaolo Bonzini while ((ret = read(stderr_fd[0], buf, 1024)) > 0) {
1011c1f8fdc3SPaolo Bonzini errors = true;
1012c1f8fdc3SPaolo Bonzini ret = qemu_write_full(STDERR_FILENO, buf, ret);
1013fc19f8a0SPaolo Bonzini if (ret < 0) {
1014c1f8fdc3SPaolo Bonzini exit(EXIT_FAILURE);
1015c1f8fdc3SPaolo Bonzini }
1016c1f8fdc3SPaolo Bonzini }
1017fc19f8a0SPaolo Bonzini if (ret < 0) {
101885b01e09SMarkus Armbruster error_report("Cannot read from daemon: %s",
101985b01e09SMarkus Armbruster strerror(errno));
102085b01e09SMarkus Armbruster exit(EXIT_FAILURE);
1021c1f8fdc3SPaolo Bonzini }
1022c1f8fdc3SPaolo Bonzini
1023c1f8fdc3SPaolo Bonzini /* Usually the daemon should not print any message.
1024c1f8fdc3SPaolo Bonzini * Exit with zero status in that case.
1025c1f8fdc3SPaolo Bonzini */
1026c1f8fdc3SPaolo Bonzini exit(errors);
1027c1f8fdc3SPaolo Bonzini }
1028eb705985SDaniel P. Berrangé #else /* WIN32 */
1029eb705985SDaniel P. Berrangé error_report("Unable to fork into background on Windows hosts");
1030eb705985SDaniel P. Berrangé exit(EXIT_FAILURE);
1031eb705985SDaniel P. Berrangé #endif /* WIN32 */
1032c1f8fdc3SPaolo Bonzini }
1033c1f8fdc3SPaolo Bonzini
10343484f676SDenis V. Lunev if (opts.device != NULL && sockpath == NULL) {
1035b32f6c28SPaolo Bonzini sockpath = g_malloc(128);
10363484f676SDenis V. Lunev snprintf(sockpath, 128, SOCKET_PATH, basename(opts.device));
1037cd831bd7Sths }
1038cd831bd7Sths
1039e4849c1dSDaniel P. Berrange server = qio_net_listener_new();
1040a721f53bSRichard W.M. Jones if (socket_activation == 0) {
1041582d4210SEric Blake int backlog;
1042582d4210SEric Blake
10433dcf56e6SEric Blake if (persistent || shared == 0) {
1044582d4210SEric Blake backlog = SOMAXCONN;
1045582d4210SEric Blake } else {
1046582d4210SEric Blake backlog = MIN(shared, SOMAXCONN);
1047582d4210SEric Blake }
10483d212b41SRichard W.M. Jones #ifdef CONFIG_SELINUX
10493d212b41SRichard W.M. Jones if (selinux_label && setsockcreatecon_raw(selinux_label) == -1) {
10503d212b41SRichard W.M. Jones error_report("Cannot set SELinux socket create context to %s: %s",
10513d212b41SRichard W.M. Jones selinux_label, strerror(errno));
10523d212b41SRichard W.M. Jones exit(EXIT_FAILURE);
10533d212b41SRichard W.M. Jones }
10543d212b41SRichard W.M. Jones #endif
1055b2cecdfeSDenis V. Lunev opts.saddr = nbd_build_socket_address(sockpath, bindto, port);
1056b2cecdfeSDenis V. Lunev if (qio_net_listener_open_sync(server, opts.saddr, backlog,
1057582d4210SEric Blake &local_err) < 0) {
1058e4849c1dSDaniel P. Berrange object_unref(OBJECT(server));
1059a721f53bSRichard W.M. Jones error_report_err(local_err);
1060e4849c1dSDaniel P. Berrange exit(EXIT_FAILURE);
1061a721f53bSRichard W.M. Jones }
10623d212b41SRichard W.M. Jones #ifdef CONFIG_SELINUX
10633d212b41SRichard W.M. Jones if (selinux_label && setsockcreatecon_raw(NULL) == -1) {
10643d212b41SRichard W.M. Jones error_report("Cannot clear SELinux socket create context: %s",
10653d212b41SRichard W.M. Jones strerror(errno));
10663d212b41SRichard W.M. Jones exit(EXIT_FAILURE);
10673d212b41SRichard W.M. Jones }
10683d212b41SRichard W.M. Jones #endif
1069a721f53bSRichard W.M. Jones } else {
1070e4849c1dSDaniel P. Berrange size_t i;
1071a721f53bSRichard W.M. Jones /* See comment in check_socket_activation above. */
1072e4849c1dSDaniel P. Berrange for (i = 0; i < socket_activation; i++) {
1073e4849c1dSDaniel P. Berrange QIOChannelSocket *sioc;
1074e4849c1dSDaniel P. Berrange sioc = qio_channel_socket_new_fd(FIRST_SOCKET_ACTIVATION_FD + i,
1075a721f53bSRichard W.M. Jones &local_err);
1076e4849c1dSDaniel P. Berrange if (sioc == NULL) {
1077e4849c1dSDaniel P. Berrange object_unref(OBJECT(server));
10785217f188SMarkus Armbruster error_reportf_err(local_err,
10795217f188SMarkus Armbruster "Failed to use socket activation: ");
1080a721f53bSRichard W.M. Jones exit(EXIT_FAILURE);
1081a721f53bSRichard W.M. Jones }
1082e4849c1dSDaniel P. Berrange qio_net_listener_add(server, sioc);
1083e4849c1dSDaniel P. Berrange object_unref(OBJECT(sioc));
1084e4849c1dSDaniel P. Berrange }
1085a721f53bSRichard W.M. Jones }
108648bec07eSDaniel P. Berrange
1087f9734d5dSMarkus Armbruster qemu_init_main_loop(&error_fatal);
1088802ddc37SPaolo Bonzini bdrv_init();
1089b3b5299dSKevin Wolf atexit(qemu_nbd_shutdown);
1090802ddc37SPaolo Bonzini
1091b18d72d7SDenis V. Lunev opts.srcpath = argv[optind];
109277c9aaefSDaniel P. Berrange if (imageOpts) {
1093b18d72d7SDenis V. Lunev QemuOpts *o;
109477c9aaefSDaniel P. Berrange if (fmt) {
109577c9aaefSDaniel P. Berrange error_report("--image-opts and -f are mutually exclusive");
109677c9aaefSDaniel P. Berrange exit(EXIT_FAILURE);
109777c9aaefSDaniel P. Berrange }
1098b18d72d7SDenis V. Lunev o = qemu_opts_parse_noisily(&file_opts, opts.srcpath, true);
1099b18d72d7SDenis V. Lunev if (!o) {
110077c9aaefSDaniel P. Berrange qemu_opts_reset(&file_opts);
110177c9aaefSDaniel P. Berrange exit(EXIT_FAILURE);
110277c9aaefSDaniel P. Berrange }
1103b18d72d7SDenis V. Lunev options = qemu_opts_to_qdict(o, NULL);
110477c9aaefSDaniel P. Berrange qemu_opts_reset(&file_opts);
1105efaa7c4eSMax Reitz blk = blk_new_open(NULL, NULL, options, flags, &local_err);
110677c9aaefSDaniel P. Berrange } else {
1107e6b63677SDaniel P. Berrange if (fmt) {
11084fbec260SMax Reitz options = qdict_new();
110946f5ac20SEric Blake qdict_put_str(options, "driver", fmt);
1110e6b63677SDaniel P. Berrange }
1111b18d72d7SDenis V. Lunev blk = blk_new_open(opts.srcpath, NULL, options, flags, &local_err);
111277c9aaefSDaniel P. Berrange }
111377c9aaefSDaniel P. Berrange
11144fbec260SMax Reitz if (!blk) {
1115c29b77f9SMarkus Armbruster error_reportf_err(local_err, "Failed to blk_new_open '%s': ",
1116c29b77f9SMarkus Armbruster argv[optind]);
111785b01e09SMarkus Armbruster exit(EXIT_FAILURE);
1118802ddc37SPaolo Bonzini }
11194fbec260SMax Reitz bs = blk_bs(blk);
1120802ddc37SPaolo Bonzini
1121b57e4de0SKevin Wolf if (dev_offset) {
1122b57e4de0SKevin Wolf QDict *raw_opts = qdict_new();
1123b57e4de0SKevin Wolf qdict_put_str(raw_opts, "driver", "raw");
1124b57e4de0SKevin Wolf qdict_put_str(raw_opts, "file", bs->node_name);
1125b57e4de0SKevin Wolf qdict_put_int(raw_opts, "offset", dev_offset);
1126c6e0a6deSKevin Wolf
1127b57e4de0SKevin Wolf bs = bdrv_open(NULL, NULL, raw_opts, flags, &error_fatal);
1128c6e0a6deSKevin Wolf
1129b57e4de0SKevin Wolf blk_remove_bs(blk);
1130b57e4de0SKevin Wolf blk_insert_bs(blk, bs, &error_fatal);
1131b57e4de0SKevin Wolf bdrv_unref(bs);
1132b57e4de0SKevin Wolf }
1133b57e4de0SKevin Wolf
11346effd5bfSKevin Wolf blk_set_enable_write_cache(blk, !writethrough);
11356effd5bfSKevin Wolf
11368c116b0eSWenchao Xia if (sn_opts) {
11378c116b0eSWenchao Xia ret = bdrv_snapshot_load_tmp(bs,
11388c116b0eSWenchao Xia qemu_opt_get(sn_opts, SNAPSHOT_OPT_ID),
11398c116b0eSWenchao Xia qemu_opt_get(sn_opts, SNAPSHOT_OPT_NAME),
11408c116b0eSWenchao Xia &local_err);
11418c116b0eSWenchao Xia } else if (sn_id_or_name) {
11428c116b0eSWenchao Xia ret = bdrv_snapshot_load_tmp_by_id_or_name(bs, sn_id_or_name,
11438c116b0eSWenchao Xia &local_err);
11448c116b0eSWenchao Xia }
11458c116b0eSWenchao Xia if (ret < 0) {
1146c29b77f9SMarkus Armbruster error_reportf_err(local_err, "Failed to load snapshot: ");
114785b01e09SMarkus Armbruster exit(EXIT_FAILURE);
11488c116b0eSWenchao Xia }
11498c116b0eSWenchao Xia
1150b3838a40SPeter Lieven bs->detect_zeroes = detect_zeroes;
1151802ddc37SPaolo Bonzini
1152a5fced40SEric Blake nbd_server_is_qemu_nbd(shared);
115300917172SKevin Wolf
115400917172SKevin Wolf export_opts = g_new(BlockExportOptions, 1);
115500917172SKevin Wolf *export_opts = (BlockExportOptions) {
115600917172SKevin Wolf .type = BLOCK_EXPORT_TYPE_NBD,
1157d53be9ceSKevin Wolf .id = g_strdup("qemu-nbd-export"),
1158b6076afcSKevin Wolf .node_name = g_strdup(bdrv_get_node_name(bs)),
115900917172SKevin Wolf .has_writethrough = true,
116000917172SKevin Wolf .writethrough = writethrough,
116130dbc81dSKevin Wolf .has_writable = true,
116230dbc81dSKevin Wolf .writable = !readonly,
116300917172SKevin Wolf .u.nbd = {
116400917172SKevin Wolf .name = g_strdup(export_name),
116500917172SKevin Wolf .description = g_strdup(export_description),
1166cbad81ceSEric Blake .has_bitmaps = !!bitmaps,
1167cbad81ceSEric Blake .bitmaps = bitmaps,
1168dbc7b014SEric Blake .has_allocation_depth = alloc_depth,
1169dbc7b014SEric Blake .allocation_depth = alloc_depth,
117000917172SKevin Wolf },
117100917172SKevin Wolf };
117200917172SKevin Wolf blk_exp_add(export_opts, &error_fatal);
117300917172SKevin Wolf qapi_free_BlockExportOptions(export_opts);
11743b05a8e9Sths
11753484f676SDenis V. Lunev if (opts.device) {
11763c1fa35dSEric Blake #if HAVE_NBD_DEVICE
117703b67621SDenis V. Lunev ret = pthread_create(&client_thread, NULL, nbd_client_thread, &opts);
1178f1ef5555SPaolo Bonzini if (ret != 0) {
117985b01e09SMarkus Armbruster error_report("Failed to create client thread: %s", strerror(ret));
118085b01e09SMarkus Armbruster exit(EXIT_FAILURE);
1181f1ef5555SPaolo Bonzini }
11823c1fa35dSEric Blake #endif
1183f1ef5555SPaolo Bonzini } else {
1184f1ef5555SPaolo Bonzini /* Shut up GCC warnings. */
1185f1ef5555SPaolo Bonzini memset(&client_thread, 0, sizeof(client_thread));
1186f1ef5555SPaolo Bonzini }
1187f1ef5555SPaolo Bonzini
1188d0d6ff58SDaniel P. Berrange nbd_update_server_watch();
11897a5ca864Sbellard
1190637bc5a5SMax Reitz if (pid_file_name) {
1191637bc5a5SMax Reitz qemu_write_pidfile(pid_file_name, &error_fatal);
1192637bc5a5SMax Reitz }
1193637bc5a5SMax Reitz
11949faf31b6SMichael Tokarev /* now when the initialization is (almost) complete, chdir("/")
11959faf31b6SMichael Tokarev * to free any busy filesystems */
11969faf31b6SMichael Tokarev if (chdir("/") < 0) {
119785b01e09SMarkus Armbruster error_report("Could not chdir to root directory: %s",
119885b01e09SMarkus Armbruster strerror(errno));
119985b01e09SMarkus Armbruster exit(EXIT_FAILURE);
12009faf31b6SMichael Tokarev }
12019faf31b6SMichael Tokarev
12023484f676SDenis V. Lunev if (opts.fork_process) {
1203f6f4620cSDenis V. Lunev nbd_client_release_pipe(opts.old_stderr);
1204ffb31e1dSMax Reitz }
1205ffb31e1dSMax Reitz
12067860a380SPaolo Bonzini state = RUNNING;
12073b05a8e9Sths do {
1208a61c6782SPaolo Bonzini main_loop_wait(false);
12097860a380SPaolo Bonzini if (state == TERMINATE) {
1210bc4ee65bSKevin Wolf blk_exp_close_all();
1211d794f7f3SKevin Wolf state = TERMINATED;
12127860a380SPaolo Bonzini }
12137860a380SPaolo Bonzini } while (state != TERMINATED);
12142c8d9f06SPaolo Bonzini
121526f54e9aSMarkus Armbruster blk_unref(blk);
1216b32f6c28SPaolo Bonzini if (sockpath) {
1217b32f6c28SPaolo Bonzini unlink(sockpath);
1218b32f6c28SPaolo Bonzini }
12197a5ca864Sbellard
12208c116b0eSWenchao Xia qemu_opts_del(sn_opts);
12218c116b0eSWenchao Xia
12223484f676SDenis V. Lunev if (opts.device) {
1223e161785cSEric Blake void *result;
1224e161785cSEric Blake pthread_join(client_thread, &result);
1225e161785cSEric Blake ret = (intptr_t)result;
1226e161785cSEric Blake exit(ret);
1227a517e88bSPaolo Bonzini } else {
1228a517e88bSPaolo Bonzini exit(EXIT_SUCCESS);
1229a517e88bSPaolo Bonzini }
12307a5ca864Sbellard }
1231