xref: /qemu/qemu-nbd.c (revision 992809bf)
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 
24a8d25326SMarkus Armbruster #include "qemu-common.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"
4639ca463eSDenis V. Lunev #include "trace/control.h"
47be377133SEric Blake #include "qemu-version.h"
487a5ca864Sbellard 
493c1fa35dSEric Blake #ifdef __linux__
503c1fa35dSEric Blake #define HAVE_NBD_DEVICE 1
513c1fa35dSEric Blake #else
523c1fa35dSEric Blake #define HAVE_NBD_DEVICE 0
533c1fa35dSEric Blake #endif
543c1fa35dSEric Blake 
55cd831bd7Sths #define SOCKET_PATH                "/var/lock/qemu-nbd-%s"
56fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_CACHE         256
57fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_AIO           257
58fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_DISCARD       258
59fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_DETECT_ZEROES 259
60fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_OBJECT        260
61fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_TLSCREDS      261
62fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_IMAGE_OPTS    262
63ffb31e1dSMax Reitz #define QEMU_NBD_OPT_FORK          263
64b25e12daSDaniel P. Berrange #define QEMU_NBD_OPT_TLSAUTHZ      264
65637bc5a5SMax Reitz #define QEMU_NBD_OPT_PID_FILE      265
667a5ca864Sbellard 
67bd31c214SEric Blake #define MBR_SIZE 512
68bd31c214SEric Blake 
69b1d8e52eSblueswir1 static int verbose;
70a517e88bSPaolo Bonzini static char *srcpath;
71bd269ebcSMarkus Armbruster static SocketAddress *saddr;
727860a380SPaolo Bonzini static int persistent = 0;
73d794f7f3SKevin Wolf static enum { RUNNING, TERMINATE, TERMINATED } state;
74a61c6782SPaolo Bonzini static int shared = 1;
75a61c6782SPaolo Bonzini static int nb_fds;
76e4849c1dSDaniel P. Berrange static QIONetListener *server;
77145614a1SDaniel P. Berrange static QCryptoTLSCreds *tlscreds;
78b25e12daSDaniel P. Berrange static const char *tlsauthz;
797a5ca864Sbellard 
807a5ca864Sbellard static void usage(const char *name)
817a5ca864Sbellard {
82b033cd86SPaolo Bonzini     (printf) (
837a5ca864Sbellard "Usage: %s [OPTIONS] FILE\n"
8468b96f15SEric Blake "  or:  %s -L [OPTIONS]\n"
8568b96f15SEric Blake "QEMU Disk Network Block Device Utility\n"
867a5ca864Sbellard "\n"
877a5ca864Sbellard "  -h, --help                display this help and exit\n"
887a5ca864Sbellard "  -V, --version             output version information and exit\n"
897a5ca864Sbellard "\n"
90b033cd86SPaolo Bonzini "Connection properties:\n"
91b033cd86SPaolo Bonzini "  -p, --port=PORT           port to listen on (default `%d')\n"
92b033cd86SPaolo Bonzini "  -b, --bind=IFACE          interface to bind to (default `0.0.0.0')\n"
93b033cd86SPaolo Bonzini "  -k, --socket=PATH         path to the unix socket\n"
94b033cd86SPaolo Bonzini "                            (default '"SOCKET_PATH"')\n"
95b033cd86SPaolo Bonzini "  -e, --shared=NUM          device can be shared by NUM clients (default '1')\n"
96b033cd86SPaolo Bonzini "  -t, --persistent          don't exit on the last connection\n"
97b033cd86SPaolo Bonzini "  -v, --verbose             display extra debugging information\n"
98f5cd0bb5SVladimir Sementsov-Ogievskiy "  -x, --export-name=NAME    expose export by name (default is empty string)\n"
99f5cd0bb5SVladimir Sementsov-Ogievskiy "  -D, --description=TEXT    export a human-readable description\n"
100b033cd86SPaolo Bonzini "\n"
101b033cd86SPaolo Bonzini "Exposing part of the image:\n"
102b033cd86SPaolo Bonzini "  -o, --offset=OFFSET       offset into the image\n"
103dbc7b014SEric Blake "  -A, --allocation-depth    expose the allocation depth\n"
104636192c4SEric Blake "  -B, --bitmap=NAME         expose a persistent dirty bitmap\n"
105b033cd86SPaolo Bonzini "\n"
1060ab3b337SDaniel P. Berrange "General purpose options:\n"
10768b96f15SEric Blake "  -L, --list                list exports available from another NBD server\n"
1080ab3b337SDaniel P. Berrange "  --object type,id=ID,...   define an object such as 'secret' for providing\n"
1090ab3b337SDaniel P. Berrange "                            passwords and/or encryption keys\n"
110f7812df7SEric Blake "  --tls-creds=ID            use id of an earlier --object to provide TLS\n"
111b25e12daSDaniel P. Berrange "  --tls-authz=ID            use id of an earlier --object to provide\n"
112b25e12daSDaniel P. Berrange "                            authorization\n"
11339ca463eSDenis V. Lunev "  -T, --trace [[enable=]<pattern>][,events=<file>][,file=<file>]\n"
11439ca463eSDenis V. Lunev "                            specify tracing options\n"
115ffb31e1dSMax Reitz "  --fork                    fork off the server process and exit the parent\n"
116ffb31e1dSMax Reitz "                            once the server is running\n"
117637bc5a5SMax Reitz "  --pid-file=PATH           store the server's process ID in the given file\n"
1183c1fa35dSEric Blake #if HAVE_NBD_DEVICE
1193c1fa35dSEric Blake "\n"
120b033cd86SPaolo Bonzini "Kernel NBD client support:\n"
121b033cd86SPaolo Bonzini "  -c, --connect=DEV         connect FILE to the local NBD device DEV\n"
122b033cd86SPaolo Bonzini "  -d, --disconnect          disconnect the specified device\n"
123b033cd86SPaolo Bonzini #endif
124b033cd86SPaolo Bonzini "\n"
125b033cd86SPaolo Bonzini "Block device options:\n"
1264323fdcfSWenchao Xia "  -f, --format=FORMAT       set image format (raw, qcow2, ...)\n"
127b033cd86SPaolo Bonzini "  -r, --read-only           export read-only\n"
1288c116b0eSWenchao Xia "  -s, --snapshot            use FILE as an external snapshot, create a temporary\n"
1298c116b0eSWenchao Xia "                            file with backing_file=FILE, redirect the write to\n"
1308c116b0eSWenchao Xia "                            the temporary one\n"
1318c116b0eSWenchao Xia "  -l, --load-snapshot=SNAPSHOT_PARAM\n"
1328c116b0eSWenchao Xia "                            load an internal snapshot inside FILE and export it\n"
1338c116b0eSWenchao Xia "                            as an read-only device, SNAPSHOT_PARAM format is\n"
1348c116b0eSWenchao Xia "                            'snapshot.id=[ID],snapshot.name=[NAME]', or\n"
1358c116b0eSWenchao Xia "                            '[ID_OR_NAME]'\n"
136b033cd86SPaolo Bonzini "  -n, --nocache             disable host cache\n"
13739a5235cSPaolo Bonzini "      --cache=MODE          set cache mode (none, writeback, ...)\n"
1387680274dSAarushi Mehta "      --aio=MODE            set AIO mode (native, io_uring or threads)\n"
139b3838a40SPeter Lieven "      --discard=MODE        set discard mode (ignore, unmap)\n"
1406883de6cSAndrey Korolyov "      --detect-zeroes=MODE  set detect-zeroes mode (off, on, unmap)\n"
14177c9aaefSDaniel P. Berrange "      --image-opts          treat FILE as a full set of image options\n"
142b033cd86SPaolo Bonzini "\n"
143f5048cb7SEric Blake QEMU_HELP_BOTTOM "\n"
14468b96f15SEric Blake     , name, name, NBD_DEFAULT_PORT, "DEVICE");
1457a5ca864Sbellard }
1467a5ca864Sbellard 
1477a5ca864Sbellard static void version(const char *name)
1487a5ca864Sbellard {
1497a5ca864Sbellard     printf(
1507e563bfbSThomas Huth "%s " QEMU_FULL_VERSION "\n"
1517a5ca864Sbellard "Written by Anthony Liguori.\n"
1527a5ca864Sbellard "\n"
153be377133SEric Blake QEMU_COPYRIGHT "\n"
1547a5ca864Sbellard "This is free software; see the source for copying conditions.  There is NO\n"
1557a5ca864Sbellard "warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"
156315bc7aaSths     , name);
1577a5ca864Sbellard }
1587a5ca864Sbellard 
159029a88c9SEric Blake #ifdef CONFIG_POSIX
160cbc20bfbSStefan Hajnoczi /*
161cbc20bfbSStefan Hajnoczi  * The client thread uses SIGTERM to interrupt the server.  A signal
162cbc20bfbSStefan Hajnoczi  * handler ensures that "qemu-nbd -v -c" exits with a nice status code.
163cbc20bfbSStefan Hajnoczi  */
164cbc20bfbSStefan Hajnoczi void qemu_system_killed(int signum, pid_t pid)
165bb345110SPaolo Bonzini {
166d73415a3SStefan Hajnoczi     qatomic_cmpxchg(&state, RUNNING, TERMINATE);
167a61c6782SPaolo Bonzini     qemu_notify_event();
168bb345110SPaolo Bonzini }
169029a88c9SEric Blake #endif /* CONFIG_POSIX */
170537b41f5SPaolo Bonzini 
17168b96f15SEric Blake static int qemu_nbd_client_list(SocketAddress *saddr, QCryptoTLSCreds *tls,
17268b96f15SEric Blake                                 const char *hostname)
17368b96f15SEric Blake {
17468b96f15SEric Blake     int ret = EXIT_FAILURE;
17568b96f15SEric Blake     int rc;
17668b96f15SEric Blake     Error *err = NULL;
17768b96f15SEric Blake     QIOChannelSocket *sioc;
17868b96f15SEric Blake     NBDExportInfo *list;
17968b96f15SEric Blake     int i, j;
18068b96f15SEric Blake 
18168b96f15SEric Blake     sioc = qio_channel_socket_new();
18268b96f15SEric Blake     if (qio_channel_socket_connect_sync(sioc, saddr, &err) < 0) {
18368b96f15SEric Blake         error_report_err(err);
184*992809bfSAlex Chen         goto out;
18568b96f15SEric Blake     }
18668b96f15SEric Blake     rc = nbd_receive_export_list(QIO_CHANNEL(sioc), tls, hostname, &list,
18768b96f15SEric Blake                                  &err);
18868b96f15SEric Blake     if (rc < 0) {
18968b96f15SEric Blake         if (err) {
19068b96f15SEric Blake             error_report_err(err);
19168b96f15SEric Blake         }
19268b96f15SEric Blake         goto out;
19368b96f15SEric Blake     }
19468b96f15SEric Blake     printf("exports available: %d\n", rc);
19568b96f15SEric Blake     for (i = 0; i < rc; i++) {
19668b96f15SEric Blake         printf(" export: '%s'\n", list[i].name);
19768b96f15SEric Blake         if (list[i].description && *list[i].description) {
19868b96f15SEric Blake             printf("  description: %s\n", list[i].description);
19968b96f15SEric Blake         }
20068b96f15SEric Blake         if (list[i].flags & NBD_FLAG_HAS_FLAGS) {
201c4e2aff8SMax Reitz             static const char *const flag_names[] = {
202c4e2aff8SMax Reitz                 [NBD_FLAG_READ_ONLY_BIT]            = "readonly",
203c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_FLUSH_BIT]           = "flush",
204c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_FUA_BIT]             = "fua",
205c4e2aff8SMax Reitz                 [NBD_FLAG_ROTATIONAL_BIT]           = "rotational",
206c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_TRIM_BIT]            = "trim",
207c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_WRITE_ZEROES_BIT]    = "zeroes",
208c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_DF_BIT]              = "df",
209c4e2aff8SMax Reitz                 [NBD_FLAG_CAN_MULTI_CONN_BIT]       = "multi",
210c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_RESIZE_BIT]          = "resize",
211c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_CACHE_BIT]           = "cache",
2120a479545SEric Blake                 [NBD_FLAG_SEND_FAST_ZERO_BIT]       = "fast-zero",
213c4e2aff8SMax Reitz             };
214c4e2aff8SMax Reitz 
21568b96f15SEric Blake             printf("  size:  %" PRIu64 "\n", list[i].size);
21668b96f15SEric Blake             printf("  flags: 0x%x (", list[i].flags);
217c4e2aff8SMax Reitz             for (size_t bit = 0; bit < ARRAY_SIZE(flag_names); bit++) {
218c4e2aff8SMax Reitz                 if (flag_names[bit] && (list[i].flags & (1 << bit))) {
219c4e2aff8SMax Reitz                     printf(" %s", flag_names[bit]);
22068b96f15SEric Blake                 }
22168b96f15SEric Blake             }
22268b96f15SEric Blake             printf(" )\n");
22368b96f15SEric Blake         }
22468b96f15SEric Blake         if (list[i].min_block) {
22568b96f15SEric Blake             printf("  min block: %u\n", list[i].min_block);
22668b96f15SEric Blake             printf("  opt block: %u\n", list[i].opt_block);
22768b96f15SEric Blake             printf("  max block: %u\n", list[i].max_block);
22868b96f15SEric Blake         }
22968b96f15SEric Blake         if (list[i].n_contexts) {
23068b96f15SEric Blake             printf("  available meta contexts: %d\n", list[i].n_contexts);
23168b96f15SEric Blake             for (j = 0; j < list[i].n_contexts; j++) {
23268b96f15SEric Blake                 printf("   %s\n", list[i].contexts[j]);
23368b96f15SEric Blake             }
23468b96f15SEric Blake         }
23568b96f15SEric Blake     }
23668b96f15SEric Blake     nbd_free_export_list(list, rc);
23768b96f15SEric Blake 
23868b96f15SEric Blake     ret = EXIT_SUCCESS;
23968b96f15SEric Blake  out:
24068b96f15SEric Blake     object_unref(OBJECT(sioc));
24168b96f15SEric Blake     return ret;
24268b96f15SEric Blake }
24368b96f15SEric Blake 
24468b96f15SEric Blake 
2453c1fa35dSEric Blake #if HAVE_NBD_DEVICE
246a517e88bSPaolo Bonzini static void *show_parts(void *arg)
247cd831bd7Sths {
248a6ac2313SPaolo Bonzini     char *device = arg;
249cd831bd7Sths     int nbd;
250cd831bd7Sths 
251cd831bd7Sths     /* linux just needs an open() to trigger
252cd831bd7Sths      * the partition table update
253cd831bd7Sths      * but remember to load the module with max_part != 0 :
254cd831bd7Sths      *     modprobe nbd max_part=63
255cd831bd7Sths      */
256cd831bd7Sths     nbd = open(device, O_RDWR);
257fc19f8a0SPaolo Bonzini     if (nbd >= 0) {
258cd831bd7Sths         close(nbd);
259cd831bd7Sths     }
260a517e88bSPaolo Bonzini     return NULL;
261a517e88bSPaolo Bonzini }
262a517e88bSPaolo Bonzini 
263a517e88bSPaolo Bonzini static void *nbd_client_thread(void *arg)
264a517e88bSPaolo Bonzini {
265a6ac2313SPaolo Bonzini     char *device = arg;
2666dc1667dSEric Blake     NBDExportInfo info = { .request_sizes = false, .name = g_strdup("") };
267d0d6ff58SDaniel P. Berrange     QIOChannelSocket *sioc;
268d0d6ff58SDaniel P. Berrange     int fd;
269a517e88bSPaolo Bonzini     int ret;
270a517e88bSPaolo Bonzini     pthread_t show_parts_thread;
2711ce52846SMax Reitz     Error *local_error = NULL;
272a517e88bSPaolo Bonzini 
273d0d6ff58SDaniel P. Berrange     sioc = qio_channel_socket_new();
274d0d6ff58SDaniel P. Berrange     if (qio_channel_socket_connect_sync(sioc,
275d0d6ff58SDaniel P. Berrange                                         saddr,
276d0d6ff58SDaniel P. Berrange                                         &local_error) < 0) {
27748bec07eSDaniel P. Berrange         error_report_err(local_error);
278a517e88bSPaolo Bonzini         goto out;
279a517e88bSPaolo Bonzini     }
280a517e88bSPaolo Bonzini 
281a8e2bb6aSVladimir Sementsov-Ogievskiy     ret = nbd_receive_negotiate(NULL, QIO_CHANNEL(sioc),
282004a89fcSEric Blake                                 NULL, NULL, NULL, &info, &local_error);
283fc19f8a0SPaolo Bonzini     if (ret < 0) {
2841ce52846SMax Reitz         if (local_error) {
28578288671SMarkus Armbruster             error_report_err(local_error);
2861ce52846SMax Reitz         }
2870c544d73SPaolo Bonzini         goto out_socket;
288a517e88bSPaolo Bonzini     }
289a517e88bSPaolo Bonzini 
290a6ac2313SPaolo Bonzini     fd = open(device, O_RDWR);
291fc19f8a0SPaolo Bonzini     if (fd < 0) {
292a6ac2313SPaolo Bonzini         /* Linux-only, we can use %m in printf.  */
293b9884681SMarkus Armbruster         error_report("Failed to open %s: %m", device);
2940c544d73SPaolo Bonzini         goto out_socket;
295a6ac2313SPaolo Bonzini     }
296a6ac2313SPaolo Bonzini 
297004a89fcSEric Blake     ret = nbd_init(fd, sioc, &info, &local_error);
298fc19f8a0SPaolo Bonzini     if (ret < 0) {
299be41c100SVladimir Sementsov-Ogievskiy         error_report_err(local_error);
3000c544d73SPaolo Bonzini         goto out_fd;
301a517e88bSPaolo Bonzini     }
302a517e88bSPaolo Bonzini 
303a517e88bSPaolo Bonzini     /* update partition table */
304a6ac2313SPaolo Bonzini     pthread_create(&show_parts_thread, NULL, show_parts, device);
305a517e88bSPaolo Bonzini 
306c1f8fdc3SPaolo Bonzini     if (verbose) {
307a517e88bSPaolo Bonzini         fprintf(stderr, "NBD device %s is now connected to %s\n",
308a517e88bSPaolo Bonzini                 device, srcpath);
309c1f8fdc3SPaolo Bonzini     } else {
310c1f8fdc3SPaolo Bonzini         /* Close stderr so that the qemu-nbd process exits.  */
311c1f8fdc3SPaolo Bonzini         dup2(STDOUT_FILENO, STDERR_FILENO);
312c1f8fdc3SPaolo Bonzini     }
313a517e88bSPaolo Bonzini 
314a517e88bSPaolo Bonzini     ret = nbd_client(fd);
315a517e88bSPaolo Bonzini     if (ret) {
3160c544d73SPaolo Bonzini         goto out_fd;
317a517e88bSPaolo Bonzini     }
318a517e88bSPaolo Bonzini     close(fd);
319d0d6ff58SDaniel P. Berrange     object_unref(OBJECT(sioc));
3206dc1667dSEric Blake     g_free(info.name);
321a517e88bSPaolo Bonzini     kill(getpid(), SIGTERM);
322a517e88bSPaolo Bonzini     return (void *) EXIT_SUCCESS;
323a517e88bSPaolo Bonzini 
3240c544d73SPaolo Bonzini out_fd:
3250c544d73SPaolo Bonzini     close(fd);
3260c544d73SPaolo Bonzini out_socket:
327d0d6ff58SDaniel P. Berrange     object_unref(OBJECT(sioc));
328a517e88bSPaolo Bonzini out:
3296dc1667dSEric Blake     g_free(info.name);
330a517e88bSPaolo Bonzini     kill(getpid(), SIGTERM);
331a517e88bSPaolo Bonzini     return (void *) EXIT_FAILURE;
332cd831bd7Sths }
3333c1fa35dSEric Blake #endif /* HAVE_NBD_DEVICE */
334cd831bd7Sths 
335e4afbf4fSFam Zheng static int nbd_can_accept(void)
336a61c6782SPaolo Bonzini {
337df8ad9f1SEric Blake     return state == RUNNING && nb_fds < shared;
338a61c6782SPaolo Bonzini }
339a61c6782SPaolo Bonzini 
340d0d6ff58SDaniel P. Berrange static void nbd_update_server_watch(void);
341e4afbf4fSFam Zheng 
3420c9390d9SEric Blake static void nbd_client_closed(NBDClient *client, bool negotiated)
343a61c6782SPaolo Bonzini {
344a61c6782SPaolo Bonzini     nb_fds--;
3450c9390d9SEric Blake     if (negotiated && nb_fds == 0 && !persistent && state == RUNNING) {
3467860a380SPaolo Bonzini         state = TERMINATE;
3477860a380SPaolo Bonzini     }
348d0d6ff58SDaniel P. Berrange     nbd_update_server_watch();
3497860a380SPaolo Bonzini     nbd_client_put(client);
350a61c6782SPaolo Bonzini }
351a61c6782SPaolo Bonzini 
352e4849c1dSDaniel P. Berrange static void nbd_accept(QIONetListener *listener, QIOChannelSocket *cioc,
353e4849c1dSDaniel P. Berrange                        gpointer opaque)
354a61c6782SPaolo Bonzini {
3557860a380SPaolo Bonzini     if (state >= TERMINATE) {
356e4849c1dSDaniel P. Berrange         return;
3577860a380SPaolo Bonzini     }
3587860a380SPaolo Bonzini 
359a61c6782SPaolo Bonzini     nb_fds++;
360d0d6ff58SDaniel P. Berrange     nbd_update_server_watch();
361b25e12daSDaniel P. Berrange     nbd_client_new(cioc, tlscreds, tlsauthz, nbd_client_closed);
362d0d6ff58SDaniel P. Berrange }
363d0d6ff58SDaniel P. Berrange 
364d0d6ff58SDaniel P. Berrange static void nbd_update_server_watch(void)
365e4afbf4fSFam Zheng {
366e4afbf4fSFam Zheng     if (nbd_can_accept()) {
367e4849c1dSDaniel P. Berrange         qio_net_listener_set_client_func(server, nbd_accept, NULL, NULL);
368e4afbf4fSFam Zheng     } else {
369e4849c1dSDaniel P. Berrange         qio_net_listener_set_client_func(server, NULL, NULL, NULL);
370e4afbf4fSFam Zheng     }
371e4afbf4fSFam Zheng }
372e4afbf4fSFam Zheng 
37348bec07eSDaniel P. Berrange 
374bd269ebcSMarkus Armbruster static SocketAddress *nbd_build_socket_address(const char *sockpath,
37548bec07eSDaniel P. Berrange                                                const char *bindto,
37648bec07eSDaniel P. Berrange                                                const char *port)
37748bec07eSDaniel P. Berrange {
378bd269ebcSMarkus Armbruster     SocketAddress *saddr;
37948bec07eSDaniel P. Berrange 
380bd269ebcSMarkus Armbruster     saddr = g_new0(SocketAddress, 1);
38148bec07eSDaniel P. Berrange     if (sockpath) {
382bd269ebcSMarkus Armbruster         saddr->type = SOCKET_ADDRESS_TYPE_UNIX;
383bd269ebcSMarkus Armbruster         saddr->u.q_unix.path = g_strdup(sockpath);
38448bec07eSDaniel P. Berrange     } else {
3850399293eSEric Blake         InetSocketAddress *inet;
386bd269ebcSMarkus Armbruster         saddr->type = SOCKET_ADDRESS_TYPE_INET;
387bd269ebcSMarkus Armbruster         inet = &saddr->u.inet;
3880399293eSEric Blake         inet->host = g_strdup(bindto);
38948bec07eSDaniel P. Berrange         if (port) {
3900399293eSEric Blake             inet->port = g_strdup(port);
39148bec07eSDaniel P. Berrange         } else  {
3920399293eSEric Blake             inet->port = g_strdup_printf("%d", NBD_DEFAULT_PORT);
39348bec07eSDaniel P. Berrange         }
39448bec07eSDaniel P. Berrange     }
39548bec07eSDaniel P. Berrange 
39648bec07eSDaniel P. Berrange     return saddr;
39748bec07eSDaniel P. Berrange }
39848bec07eSDaniel P. Berrange 
39948bec07eSDaniel P. Berrange 
40077c9aaefSDaniel P. Berrange static QemuOptsList file_opts = {
40177c9aaefSDaniel P. Berrange     .name = "file",
40277c9aaefSDaniel P. Berrange     .implied_opt_name = "file",
40377c9aaefSDaniel P. Berrange     .head = QTAILQ_HEAD_INITIALIZER(file_opts.head),
40477c9aaefSDaniel P. Berrange     .desc = {
40577c9aaefSDaniel P. Berrange         /* no elements => accept any params */
40677c9aaefSDaniel P. Berrange         { /* end of list */ }
40777c9aaefSDaniel P. Berrange     },
40877c9aaefSDaniel P. Berrange };
40977c9aaefSDaniel P. Berrange 
4100ab3b337SDaniel P. Berrange static QemuOptsList qemu_object_opts = {
4110ab3b337SDaniel P. Berrange     .name = "object",
4120ab3b337SDaniel P. Berrange     .implied_opt_name = "qom-type",
4130ab3b337SDaniel P. Berrange     .head = QTAILQ_HEAD_INITIALIZER(qemu_object_opts.head),
4140ab3b337SDaniel P. Berrange     .desc = {
4150ab3b337SDaniel P. Berrange         { }
4160ab3b337SDaniel P. Berrange     },
4170ab3b337SDaniel P. Berrange };
4180ab3b337SDaniel P. Berrange 
419495bf893SKevin Wolf static bool qemu_nbd_object_print_help(const char *type, QemuOpts *opts)
420495bf893SKevin Wolf {
421495bf893SKevin Wolf     if (user_creatable_print_help(type, opts)) {
422495bf893SKevin Wolf         exit(0);
423495bf893SKevin Wolf     }
424495bf893SKevin Wolf     return true;
425495bf893SKevin Wolf }
4260ab3b337SDaniel P. Berrange 
427145614a1SDaniel P. Berrange 
42868b96f15SEric Blake static QCryptoTLSCreds *nbd_get_tls_creds(const char *id, bool list,
42968b96f15SEric Blake                                           Error **errp)
430145614a1SDaniel P. Berrange {
431145614a1SDaniel P. Berrange     Object *obj;
432145614a1SDaniel P. Berrange     QCryptoTLSCreds *creds;
433145614a1SDaniel P. Berrange 
434145614a1SDaniel P. Berrange     obj = object_resolve_path_component(
435145614a1SDaniel P. Berrange         object_get_objects_root(), id);
436145614a1SDaniel P. Berrange     if (!obj) {
437145614a1SDaniel P. Berrange         error_setg(errp, "No TLS credentials with id '%s'",
438145614a1SDaniel P. Berrange                    id);
439145614a1SDaniel P. Berrange         return NULL;
440145614a1SDaniel P. Berrange     }
441145614a1SDaniel P. Berrange     creds = (QCryptoTLSCreds *)
442145614a1SDaniel P. Berrange         object_dynamic_cast(obj, TYPE_QCRYPTO_TLS_CREDS);
443145614a1SDaniel P. Berrange     if (!creds) {
444145614a1SDaniel P. Berrange         error_setg(errp, "Object with id '%s' is not TLS credentials",
445145614a1SDaniel P. Berrange                    id);
446145614a1SDaniel P. Berrange         return NULL;
447145614a1SDaniel P. Berrange     }
448145614a1SDaniel P. Berrange 
44968b96f15SEric Blake     if (list) {
45068b96f15SEric Blake         if (creds->endpoint != QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT) {
45168b96f15SEric Blake             error_setg(errp,
45268b96f15SEric Blake                        "Expecting TLS credentials with a client endpoint");
45368b96f15SEric Blake             return NULL;
45468b96f15SEric Blake         }
45568b96f15SEric Blake     } else {
456145614a1SDaniel P. Berrange         if (creds->endpoint != QCRYPTO_TLS_CREDS_ENDPOINT_SERVER) {
457145614a1SDaniel P. Berrange             error_setg(errp,
458145614a1SDaniel P. Berrange                        "Expecting TLS credentials with a server endpoint");
459145614a1SDaniel P. Berrange             return NULL;
460145614a1SDaniel P. Berrange         }
46168b96f15SEric Blake     }
462145614a1SDaniel P. Berrange     object_ref(obj);
463145614a1SDaniel P. Berrange     return creds;
464145614a1SDaniel P. Berrange }
465145614a1SDaniel P. Berrange 
466a721f53bSRichard W.M. Jones static void setup_address_and_port(const char **address, const char **port)
467a721f53bSRichard W.M. Jones {
468a721f53bSRichard W.M. Jones     if (*address == NULL) {
469a721f53bSRichard W.M. Jones         *address = "0.0.0.0";
470a721f53bSRichard W.M. Jones     }
471a721f53bSRichard W.M. Jones 
472a721f53bSRichard W.M. Jones     if (*port == NULL) {
473a721f53bSRichard W.M. Jones         *port = stringify(NBD_DEFAULT_PORT);
474a721f53bSRichard W.M. Jones     }
475a721f53bSRichard W.M. Jones }
476a721f53bSRichard W.M. Jones 
477a721f53bSRichard W.M. Jones /*
478a721f53bSRichard W.M. Jones  * Check socket parameters compatibility when socket activation is used.
479a721f53bSRichard W.M. Jones  */
480a721f53bSRichard W.M. Jones static const char *socket_activation_validate_opts(const char *device,
481a721f53bSRichard W.M. Jones                                                    const char *sockpath,
482a721f53bSRichard W.M. Jones                                                    const char *address,
48368b96f15SEric Blake                                                    const char *port,
48468b96f15SEric Blake                                                    bool list)
485a721f53bSRichard W.M. Jones {
486a721f53bSRichard W.M. Jones     if (device != NULL) {
487a721f53bSRichard W.M. Jones         return "NBD device can't be set when using socket activation";
488a721f53bSRichard W.M. Jones     }
489a721f53bSRichard W.M. Jones 
490a721f53bSRichard W.M. Jones     if (sockpath != NULL) {
491a721f53bSRichard W.M. Jones         return "Unix socket can't be set when using socket activation";
492a721f53bSRichard W.M. Jones     }
493a721f53bSRichard W.M. Jones 
494a721f53bSRichard W.M. Jones     if (address != NULL) {
495a721f53bSRichard W.M. Jones         return "The interface can't be set when using socket activation";
496a721f53bSRichard W.M. Jones     }
497a721f53bSRichard W.M. Jones 
498a721f53bSRichard W.M. Jones     if (port != NULL) {
499a721f53bSRichard W.M. Jones         return "TCP port number can't be set when using socket activation";
500a721f53bSRichard W.M. Jones     }
501a721f53bSRichard W.M. Jones 
50268b96f15SEric Blake     if (list) {
50368b96f15SEric Blake         return "List mode is incompatible with socket activation";
50468b96f15SEric Blake     }
50568b96f15SEric Blake 
506a721f53bSRichard W.M. Jones     return NULL;
507a721f53bSRichard W.M. Jones }
508145614a1SDaniel P. Berrange 
509b3b5299dSKevin Wolf static void qemu_nbd_shutdown(void)
510b3b5299dSKevin Wolf {
511b3b5299dSKevin Wolf     job_cancel_sync_all();
512b3b5299dSKevin Wolf     bdrv_close_all();
513b3b5299dSKevin Wolf }
514b3b5299dSKevin Wolf 
5157a5ca864Sbellard int main(int argc, char **argv)
5167a5ca864Sbellard {
51726f54e9aSMarkus Armbruster     BlockBackend *blk;
5187a5ca864Sbellard     BlockDriverState *bs;
5199d26dfcbSEric Blake     uint64_t dev_offset = 0;
520dbb38caaSEric Blake     bool readonly = false;
521cd831bd7Sths     bool disconnect = false;
522a721f53bSRichard W.M. Jones     const char *bindto = NULL;
52348bec07eSDaniel P. Berrange     const char *port = NULL;
52448bec07eSDaniel P. Berrange     char *sockpath = NULL;
525a6ac2313SPaolo Bonzini     char *device = NULL;
5268c116b0eSWenchao Xia     QemuOpts *sn_opts = NULL;
5278c116b0eSWenchao Xia     const char *sn_id_or_name = NULL;
528dbc7b014SEric Blake     const char *sopt = "hVb:o:p:rsnc:dvk:e:f:tl:x:T:D:AB:L";
5297a5ca864Sbellard     struct option lopt[] = {
530aa6e546cSDaniel P. Berrange         { "help", no_argument, NULL, 'h' },
531aa6e546cSDaniel P. Berrange         { "version", no_argument, NULL, 'V' },
532aa6e546cSDaniel P. Berrange         { "bind", required_argument, NULL, 'b' },
533aa6e546cSDaniel P. Berrange         { "port", required_argument, NULL, 'p' },
534aa6e546cSDaniel P. Berrange         { "socket", required_argument, NULL, 'k' },
535aa6e546cSDaniel P. Berrange         { "offset", required_argument, NULL, 'o' },
536aa6e546cSDaniel P. Berrange         { "read-only", no_argument, NULL, 'r' },
537dbc7b014SEric Blake         { "allocation-depth", no_argument, NULL, 'A' },
538636192c4SEric Blake         { "bitmap", required_argument, NULL, 'B' },
539aa6e546cSDaniel P. Berrange         { "connect", required_argument, NULL, 'c' },
540aa6e546cSDaniel P. Berrange         { "disconnect", no_argument, NULL, 'd' },
54168b96f15SEric Blake         { "list", no_argument, NULL, 'L' },
542aa6e546cSDaniel P. Berrange         { "snapshot", no_argument, NULL, 's' },
543aa6e546cSDaniel P. Berrange         { "load-snapshot", required_argument, NULL, 'l' },
544aa6e546cSDaniel P. Berrange         { "nocache", no_argument, NULL, 'n' },
545aa6e546cSDaniel P. Berrange         { "cache", required_argument, NULL, QEMU_NBD_OPT_CACHE },
546aa6e546cSDaniel P. Berrange         { "aio", required_argument, NULL, QEMU_NBD_OPT_AIO },
547aa6e546cSDaniel P. Berrange         { "discard", required_argument, NULL, QEMU_NBD_OPT_DISCARD },
548aa6e546cSDaniel P. Berrange         { "detect-zeroes", required_argument, NULL,
549aa6e546cSDaniel P. Berrange           QEMU_NBD_OPT_DETECT_ZEROES },
550aa6e546cSDaniel P. Berrange         { "shared", required_argument, NULL, 'e' },
551aa6e546cSDaniel P. Berrange         { "format", required_argument, NULL, 'f' },
552aa6e546cSDaniel P. Berrange         { "persistent", no_argument, NULL, 't' },
553aa6e546cSDaniel P. Berrange         { "verbose", no_argument, NULL, 'v' },
554aa6e546cSDaniel P. Berrange         { "object", required_argument, NULL, QEMU_NBD_OPT_OBJECT },
555aa6e546cSDaniel P. Berrange         { "export-name", required_argument, NULL, 'x' },
556b1a75b33SEric Blake         { "description", required_argument, NULL, 'D' },
557aa6e546cSDaniel P. Berrange         { "tls-creds", required_argument, NULL, QEMU_NBD_OPT_TLSCREDS },
558b25e12daSDaniel P. Berrange         { "tls-authz", required_argument, NULL, QEMU_NBD_OPT_TLSAUTHZ },
559aa6e546cSDaniel P. Berrange         { "image-opts", no_argument, NULL, QEMU_NBD_OPT_IMAGE_OPTS },
56039ca463eSDenis V. Lunev         { "trace", required_argument, NULL, 'T' },
561ffb31e1dSMax Reitz         { "fork", no_argument, NULL, QEMU_NBD_OPT_FORK },
562637bc5a5SMax Reitz         { "pid-file", required_argument, NULL, QEMU_NBD_OPT_PID_FILE },
563660f11beSBlue Swirl         { NULL, 0, NULL, 0 }
5647a5ca864Sbellard     };
5657a5ca864Sbellard     int ch;
5667a5ca864Sbellard     int opt_ind = 0;
567f5edb014SNaphtali Sprei     int flags = BDRV_O_RDWR;
5684fbec260SMax Reitz     int ret = 0;
56939a5235cSPaolo Bonzini     bool seen_cache = false;
570ded9d2d5SPaolo Bonzini     bool seen_discard = false;
57139a5235cSPaolo Bonzini     bool seen_aio = false;
572a517e88bSPaolo Bonzini     pthread_t client_thread;
573e6b63677SDaniel P. Berrange     const char *fmt = NULL;
57434b5d2c6SMax Reitz     Error *local_err = NULL;
575b3838a40SPeter Lieven     BlockdevDetectZeroesOptions detect_zeroes = BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF;
5764fbec260SMax Reitz     QDict *options = NULL;
57768b96f15SEric Blake     const char *export_name = NULL; /* defaults to "" later for server mode */
578b1a75b33SEric Blake     const char *export_description = NULL;
579cbad81ceSEric Blake     strList *bitmaps = NULL;
580dbc7b014SEric Blake     bool alloc_depth = false;
581145614a1SDaniel P. Berrange     const char *tlscredsid = NULL;
58277c9aaefSDaniel P. Berrange     bool imageOpts = false;
5836effd5bfSKevin Wolf     bool writethrough = true;
584ffb31e1dSMax Reitz     bool fork_process = false;
58568b96f15SEric Blake     bool list = false;
586ffb31e1dSMax Reitz     int old_stderr = -1;
587a721f53bSRichard W.M. Jones     unsigned socket_activation;
588637bc5a5SMax Reitz     const char *pid_file_name = NULL;
58900917172SKevin Wolf     BlockExportOptions *export_opts;
5907a5ca864Sbellard 
591029a88c9SEric Blake #ifdef CONFIG_POSIX
592cbc20bfbSStefan Hajnoczi     os_setup_early_signal_handling();
593cbc20bfbSStefan Hajnoczi     os_setup_signal_handling();
594041e32b8SMax Reitz #endif
595041e32b8SMax Reitz 
59698c5d2e7SDaniel P. Berrangé     socket_init();
597f5852efaSChristophe Fergeau     error_init(argv[0]);
598fe4db84dSDaniel P. Berrange     module_call_init(MODULE_INIT_TRACE);
599e8f2d272SEduardo Habkost     qcrypto_init(&error_fatal);
600c2297088SDaniel P. Berrange 
6010ab3b337SDaniel P. Berrange     module_call_init(MODULE_INIT_QOM);
6020ab3b337SDaniel P. Berrange     qemu_add_opts(&qemu_object_opts);
60339ca463eSDenis V. Lunev     qemu_add_opts(&qemu_trace_opts);
60410f5bff6SFam Zheng     qemu_init_exec_dir(argv[0]);
605bb345110SPaolo Bonzini 
6067a5ca864Sbellard     while ((ch = getopt_long(argc, argv, sopt, lopt, &opt_ind)) != -1) {
6077a5ca864Sbellard         switch (ch) {
6087a5ca864Sbellard         case 's':
6092f726488Sths             flags |= BDRV_O_SNAPSHOT;
6102f726488Sths             break;
6112f726488Sths         case 'n':
61239a5235cSPaolo Bonzini             optarg = (char *) "none";
61339a5235cSPaolo Bonzini             /* fallthrough */
61439a5235cSPaolo Bonzini         case QEMU_NBD_OPT_CACHE:
61539a5235cSPaolo Bonzini             if (seen_cache) {
61685b01e09SMarkus Armbruster                 error_report("-n and --cache can only be specified once");
61785b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
61839a5235cSPaolo Bonzini             }
61939a5235cSPaolo Bonzini             seen_cache = true;
6206effd5bfSKevin Wolf             if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) == -1) {
62185b01e09SMarkus Armbruster                 error_report("Invalid cache mode `%s'", optarg);
62285b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
62339a5235cSPaolo Bonzini             }
6247a5ca864Sbellard             break;
62539a5235cSPaolo Bonzini         case QEMU_NBD_OPT_AIO:
62639a5235cSPaolo Bonzini             if (seen_aio) {
62785b01e09SMarkus Armbruster                 error_report("--aio can only be specified once");
62885b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
62939a5235cSPaolo Bonzini             }
63039a5235cSPaolo Bonzini             seen_aio = true;
6317680274dSAarushi Mehta             if (bdrv_parse_aio(optarg, &flags) < 0) {
6327680274dSAarushi Mehta                 error_report("Invalid aio mode '%s'", optarg);
63385b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
63439a5235cSPaolo Bonzini             }
63539a5235cSPaolo Bonzini             break;
636ded9d2d5SPaolo Bonzini         case QEMU_NBD_OPT_DISCARD:
637ded9d2d5SPaolo Bonzini             if (seen_discard) {
63885b01e09SMarkus Armbruster                 error_report("--discard can only be specified once");
63985b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
640ded9d2d5SPaolo Bonzini             }
641ded9d2d5SPaolo Bonzini             seen_discard = true;
642ded9d2d5SPaolo Bonzini             if (bdrv_parse_discard_flags(optarg, &flags) == -1) {
64385b01e09SMarkus Armbruster                 error_report("Invalid discard mode `%s'", optarg);
64485b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
645ded9d2d5SPaolo Bonzini             }
646ded9d2d5SPaolo Bonzini             break;
647b3838a40SPeter Lieven         case QEMU_NBD_OPT_DETECT_ZEROES:
648b3838a40SPeter Lieven             detect_zeroes =
649f7abe0ecSMarc-André Lureau                 qapi_enum_parse(&BlockdevDetectZeroesOptions_lookup,
650b3838a40SPeter Lieven                                 optarg,
651b3838a40SPeter Lieven                                 BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF,
652b3838a40SPeter Lieven                                 &local_err);
653b3838a40SPeter Lieven             if (local_err) {
654c29b77f9SMarkus Armbruster                 error_reportf_err(local_err,
655c29b77f9SMarkus Armbruster                                   "Failed to parse detect_zeroes mode: ");
65685b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
657b3838a40SPeter Lieven             }
658b3838a40SPeter Lieven             if (detect_zeroes == BLOCKDEV_DETECT_ZEROES_OPTIONS_UNMAP &&
659b3838a40SPeter Lieven                 !(flags & BDRV_O_UNMAP)) {
66085b01e09SMarkus Armbruster                 error_report("setting detect-zeroes to unmap is not allowed "
661b3838a40SPeter Lieven                              "without setting discard operation to unmap");
66285b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
663b3838a40SPeter Lieven             }
664b3838a40SPeter Lieven             break;
6657a5ca864Sbellard         case 'b':
6667a5ca864Sbellard             bindto = optarg;
6677a5ca864Sbellard             break;
6687a5ca864Sbellard         case 'p':
66948bec07eSDaniel P. Berrange             port = optarg;
6707a5ca864Sbellard             break;
6717a5ca864Sbellard         case 'o':
67243b51011SEric Blake             if (qemu_strtou64(optarg, NULL, 0, &dev_offset) < 0) {
67343b51011SEric Blake                 error_report("Invalid offset '%s'", optarg);
67485b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
6757a5ca864Sbellard             }
6767a5ca864Sbellard             break;
6778c116b0eSWenchao Xia         case 'l':
6788c116b0eSWenchao Xia             if (strstart(optarg, SNAPSHOT_OPT_BASE, NULL)) {
67970b94331SMarkus Armbruster                 sn_opts = qemu_opts_parse_noisily(&internal_snapshot_opts,
68070b94331SMarkus Armbruster                                                   optarg, false);
6818c116b0eSWenchao Xia                 if (!sn_opts) {
68285b01e09SMarkus Armbruster                     error_report("Failed in parsing snapshot param `%s'",
6838c116b0eSWenchao Xia                                  optarg);
68485b01e09SMarkus Armbruster                     exit(EXIT_FAILURE);
6858c116b0eSWenchao Xia                 }
6868c116b0eSWenchao Xia             } else {
6878c116b0eSWenchao Xia                 sn_id_or_name = optarg;
6888c116b0eSWenchao Xia             }
6898c116b0eSWenchao Xia             /* fall through */
6907a5ca864Sbellard         case 'r':
691dbb38caaSEric Blake             readonly = true;
69207108b29SNaphtali Sprei             flags &= ~BDRV_O_RDWR;
6937a5ca864Sbellard             break;
694dbc7b014SEric Blake         case 'A':
695dbc7b014SEric Blake             alloc_depth = true;
696dbc7b014SEric Blake             break;
697636192c4SEric Blake         case 'B':
698cbad81ceSEric Blake             QAPI_LIST_PREPEND(bitmaps, g_strdup(optarg));
699636192c4SEric Blake             break;
700cd831bd7Sths         case 'k':
701b32f6c28SPaolo Bonzini             sockpath = optarg;
702713cc671SPeter Lieven             if (sockpath[0] != '/') {
7039af9e0feSMarkus Armbruster                 error_report("socket path must be absolute");
70485b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
705713cc671SPeter Lieven             }
706cd831bd7Sths             break;
707cd831bd7Sths         case 'd':
708cd831bd7Sths             disconnect = true;
709cd831bd7Sths             break;
710cd831bd7Sths         case 'c':
711cd831bd7Sths             device = optarg;
712cd831bd7Sths             break;
7133b05a8e9Sths         case 'e':
71443b51011SEric Blake             if (qemu_strtoi(optarg, NULL, 0, &shared) < 0 ||
71543b51011SEric Blake                 shared < 1) {
71685b01e09SMarkus Armbruster                 error_report("Invalid shared device number '%s'", optarg);
71785b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
7183b05a8e9Sths             }
7193b05a8e9Sths             break;
720e6b63677SDaniel P. Berrange         case 'f':
721e6b63677SDaniel P. Berrange             fmt = optarg;
722e6b63677SDaniel P. Berrange             break;
72375818250Sths         case 't':
72475818250Sths             persistent = 1;
72575818250Sths             break;
7263d4b2f9cSDaniel P. Berrange         case 'x':
7273d4b2f9cSDaniel P. Berrange             export_name = optarg;
72893676c88SEric Blake             if (strlen(export_name) > NBD_MAX_STRING_SIZE) {
72993676c88SEric Blake                 error_report("export name '%s' too long", export_name);
73093676c88SEric Blake                 exit(EXIT_FAILURE);
73193676c88SEric Blake             }
7323d4b2f9cSDaniel P. Berrange             break;
733b1a75b33SEric Blake         case 'D':
734b1a75b33SEric Blake             export_description = optarg;
73593676c88SEric Blake             if (strlen(export_description) > NBD_MAX_STRING_SIZE) {
73693676c88SEric Blake                 error_report("export description '%s' too long",
73793676c88SEric Blake                              export_description);
73893676c88SEric Blake                 exit(EXIT_FAILURE);
73993676c88SEric Blake             }
740b1a75b33SEric Blake             break;
7417a5ca864Sbellard         case 'v':
7427a5ca864Sbellard             verbose = 1;
7437a5ca864Sbellard             break;
7447a5ca864Sbellard         case 'V':
7457a5ca864Sbellard             version(argv[0]);
7467a5ca864Sbellard             exit(0);
7477a5ca864Sbellard             break;
7487a5ca864Sbellard         case 'h':
7497a5ca864Sbellard             usage(argv[0]);
7507a5ca864Sbellard             exit(0);
7517a5ca864Sbellard             break;
7527a5ca864Sbellard         case '?':
75385b01e09SMarkus Armbruster             error_report("Try `%s --help' for more information.", argv[0]);
75485b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
7550ab3b337SDaniel P. Berrange         case QEMU_NBD_OPT_OBJECT: {
7560ab3b337SDaniel P. Berrange             QemuOpts *opts;
7570ab3b337SDaniel P. Berrange             opts = qemu_opts_parse_noisily(&qemu_object_opts,
7580ab3b337SDaniel P. Berrange                                            optarg, true);
7590ab3b337SDaniel P. Berrange             if (!opts) {
7600ab3b337SDaniel P. Berrange                 exit(EXIT_FAILURE);
7610ab3b337SDaniel P. Berrange             }
7620ab3b337SDaniel P. Berrange         }   break;
763145614a1SDaniel P. Berrange         case QEMU_NBD_OPT_TLSCREDS:
764145614a1SDaniel P. Berrange             tlscredsid = optarg;
765145614a1SDaniel P. Berrange             break;
76677c9aaefSDaniel P. Berrange         case QEMU_NBD_OPT_IMAGE_OPTS:
76777c9aaefSDaniel P. Berrange             imageOpts = true;
76877c9aaefSDaniel P. Berrange             break;
76939ca463eSDenis V. Lunev         case 'T':
77092eecfffSPaolo Bonzini             trace_opt_parse(optarg);
77139ca463eSDenis V. Lunev             break;
772b25e12daSDaniel P. Berrange         case QEMU_NBD_OPT_TLSAUTHZ:
773b25e12daSDaniel P. Berrange             tlsauthz = optarg;
774b25e12daSDaniel P. Berrange             break;
775ffb31e1dSMax Reitz         case QEMU_NBD_OPT_FORK:
776ffb31e1dSMax Reitz             fork_process = true;
777ffb31e1dSMax Reitz             break;
77868b96f15SEric Blake         case 'L':
77968b96f15SEric Blake             list = true;
78068b96f15SEric Blake             break;
781637bc5a5SMax Reitz         case QEMU_NBD_OPT_PID_FILE:
782637bc5a5SMax Reitz             pid_file_name = optarg;
783637bc5a5SMax Reitz             break;
7847a5ca864Sbellard         }
7857a5ca864Sbellard     }
7867a5ca864Sbellard 
78768b96f15SEric Blake     if (list) {
78868b96f15SEric Blake         if (argc != optind) {
78968b96f15SEric Blake             error_report("List mode is incompatible with a file name");
79068b96f15SEric Blake             exit(EXIT_FAILURE);
79168b96f15SEric Blake         }
7920bc16997SEric Blake         if (export_name || export_description || dev_offset ||
793cbad81ceSEric Blake             device || disconnect || fmt || sn_id_or_name || bitmaps ||
794dbc7b014SEric Blake             alloc_depth || seen_aio || seen_discard || seen_cache) {
79568b96f15SEric Blake             error_report("List mode is incompatible with per-device settings");
79668b96f15SEric Blake             exit(EXIT_FAILURE);
79768b96f15SEric Blake         }
79868b96f15SEric Blake         if (fork_process) {
79968b96f15SEric Blake             error_report("List mode is incompatible with forking");
80068b96f15SEric Blake             exit(EXIT_FAILURE);
80168b96f15SEric Blake         }
80268b96f15SEric Blake     } else if ((argc - optind) != 1) {
803433672b0SMarkus Armbruster         error_report("Invalid number of arguments");
804433672b0SMarkus Armbruster         error_printf("Try `%s --help' for more information.\n", argv[0]);
80585b01e09SMarkus Armbruster         exit(EXIT_FAILURE);
80668b96f15SEric Blake     } else if (!export_name) {
80768b96f15SEric Blake         export_name = "";
8087a5ca864Sbellard     }
8097a5ca864Sbellard 
8107e1e0c11SMarkus Armbruster     qemu_opts_foreach(&qemu_object_opts,
8110ab3b337SDaniel P. Berrange                       user_creatable_add_opts_foreach,
812495bf893SKevin Wolf                       qemu_nbd_object_print_help, &error_fatal);
8130ab3b337SDaniel P. Berrange 
81439ca463eSDenis V. Lunev     if (!trace_init_backends()) {
81539ca463eSDenis V. Lunev         exit(1);
81639ca463eSDenis V. Lunev     }
81792eecfffSPaolo Bonzini     trace_init_file();
81839ca463eSDenis V. Lunev     qemu_set_log(LOG_TRACE);
81939ca463eSDenis V. Lunev 
820a721f53bSRichard W.M. Jones     socket_activation = check_socket_activation();
821a721f53bSRichard W.M. Jones     if (socket_activation == 0) {
822a721f53bSRichard W.M. Jones         setup_address_and_port(&bindto, &port);
823a721f53bSRichard W.M. Jones     } else {
824a721f53bSRichard W.M. Jones         /* Using socket activation - check user didn't use -p etc. */
825a721f53bSRichard W.M. Jones         const char *err_msg = socket_activation_validate_opts(device, sockpath,
82668b96f15SEric Blake                                                               bindto, port,
82768b96f15SEric Blake                                                               list);
828a721f53bSRichard W.M. Jones         if (err_msg != NULL) {
829a721f53bSRichard W.M. Jones             error_report("%s", err_msg);
830a721f53bSRichard W.M. Jones             exit(EXIT_FAILURE);
831a721f53bSRichard W.M. Jones         }
83253fabd4bSPaolo Bonzini 
83353fabd4bSPaolo Bonzini         /* qemu-nbd can only listen on a single socket.  */
83453fabd4bSPaolo Bonzini         if (socket_activation > 1) {
83553fabd4bSPaolo Bonzini             error_report("qemu-nbd does not support socket activation with %s > 1",
83653fabd4bSPaolo Bonzini                          "LISTEN_FDS");
83753fabd4bSPaolo Bonzini             exit(EXIT_FAILURE);
83853fabd4bSPaolo Bonzini         }
839a721f53bSRichard W.M. Jones     }
840a721f53bSRichard W.M. Jones 
841145614a1SDaniel P. Berrange     if (tlscredsid) {
842145614a1SDaniel P. Berrange         if (sockpath) {
843145614a1SDaniel P. Berrange             error_report("TLS is only supported with IPv4/IPv6");
844145614a1SDaniel P. Berrange             exit(EXIT_FAILURE);
845145614a1SDaniel P. Berrange         }
846145614a1SDaniel P. Berrange         if (device) {
847145614a1SDaniel P. Berrange             error_report("TLS is not supported with a host device");
848145614a1SDaniel P. Berrange             exit(EXIT_FAILURE);
849145614a1SDaniel P. Berrange         }
850b25e12daSDaniel P. Berrange         if (tlsauthz && list) {
851b25e12daSDaniel P. Berrange             error_report("TLS authorization is incompatible with export list");
852b25e12daSDaniel P. Berrange             exit(EXIT_FAILURE);
853b25e12daSDaniel P. Berrange         }
85468b96f15SEric Blake         tlscreds = nbd_get_tls_creds(tlscredsid, list, &local_err);
855145614a1SDaniel P. Berrange         if (local_err) {
8565217f188SMarkus Armbruster             error_reportf_err(local_err, "Failed to get TLS creds: ");
857145614a1SDaniel P. Berrange             exit(EXIT_FAILURE);
858145614a1SDaniel P. Berrange         }
859b25e12daSDaniel P. Berrange     } else {
860b25e12daSDaniel P. Berrange         if (tlsauthz) {
861b25e12daSDaniel P. Berrange             error_report("--tls-authz is not permitted without --tls-creds");
862b25e12daSDaniel P. Berrange             exit(EXIT_FAILURE);
863b25e12daSDaniel P. Berrange         }
864145614a1SDaniel P. Berrange     }
865145614a1SDaniel P. Berrange 
86668b96f15SEric Blake     if (list) {
86768b96f15SEric Blake         saddr = nbd_build_socket_address(sockpath, bindto, port);
86868b96f15SEric Blake         return qemu_nbd_client_list(saddr, tlscreds, bindto);
86968b96f15SEric Blake     }
87068b96f15SEric Blake 
8713c1fa35dSEric Blake #if !HAVE_NBD_DEVICE
8723c1fa35dSEric Blake     if (disconnect || device) {
8733c1fa35dSEric Blake         error_report("Kernel /dev/nbdN support not available");
8743c1fa35dSEric Blake         exit(EXIT_FAILURE);
8753c1fa35dSEric Blake     }
8763c1fa35dSEric Blake #else /* HAVE_NBD_DEVICE */
877cd831bd7Sths     if (disconnect) {
878d0d6ff58SDaniel P. Berrange         int nbdfd = open(argv[optind], O_RDWR);
879d0d6ff58SDaniel P. Berrange         if (nbdfd < 0) {
88085b01e09SMarkus Armbruster             error_report("Cannot open %s: %s", argv[optind],
88185b01e09SMarkus Armbruster                          strerror(errno));
88285b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
883fc19f8a0SPaolo Bonzini         }
884d0d6ff58SDaniel P. Berrange         nbd_disconnect(nbdfd);
885cd831bd7Sths 
886d0d6ff58SDaniel P. Berrange         close(nbdfd);
887cd831bd7Sths 
888cd831bd7Sths         printf("%s disconnected\n", argv[optind]);
889cd831bd7Sths 
890cd831bd7Sths         return 0;
891cd831bd7Sths     }
8923c1fa35dSEric Blake #endif
893cd831bd7Sths 
894ffb31e1dSMax Reitz     if ((device && !verbose) || fork_process) {
895eb705985SDaniel P. Berrangé #ifndef WIN32
896c1f8fdc3SPaolo Bonzini         int stderr_fd[2];
897c1f8fdc3SPaolo Bonzini         pid_t pid;
898c1f8fdc3SPaolo Bonzini         int ret;
899c1f8fdc3SPaolo Bonzini 
900fc19f8a0SPaolo Bonzini         if (qemu_pipe(stderr_fd) < 0) {
90185b01e09SMarkus Armbruster             error_report("Error setting up communication pipe: %s",
90285b01e09SMarkus Armbruster                          strerror(errno));
90385b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
904c1f8fdc3SPaolo Bonzini         }
905c1f8fdc3SPaolo Bonzini 
906c1f8fdc3SPaolo Bonzini         /* Now daemonize, but keep a communication channel open to
907c1f8fdc3SPaolo Bonzini          * print errors and exit with the proper status code.
908c1f8fdc3SPaolo Bonzini          */
909c1f8fdc3SPaolo Bonzini         pid = fork();
91070d4739eSMax Reitz         if (pid < 0) {
91185b01e09SMarkus Armbruster             error_report("Failed to fork: %s", strerror(errno));
91285b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
91370d4739eSMax Reitz         } else if (pid == 0) {
914c1f8fdc3SPaolo Bonzini             close(stderr_fd[0]);
915e6df58a5SMax Reitz 
9160eaf453eSRaphael Pour             /* Remember parent's stderr if we will be restoring it. */
9170eaf453eSRaphael Pour             if (fork_process) {
918e6df58a5SMax Reitz                 old_stderr = dup(STDERR_FILENO);
9190eaf453eSRaphael Pour             }
9200eaf453eSRaphael Pour 
9219faf31b6SMichael Tokarev             ret = qemu_daemon(1, 0);
922c1f8fdc3SPaolo Bonzini 
923c1f8fdc3SPaolo Bonzini             /* Temporarily redirect stderr to the parent's pipe...  */
924c1f8fdc3SPaolo Bonzini             dup2(stderr_fd[1], STDERR_FILENO);
925fc19f8a0SPaolo Bonzini             if (ret < 0) {
92685b01e09SMarkus Armbruster                 error_report("Failed to daemonize: %s", strerror(errno));
92785b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
928c1f8fdc3SPaolo Bonzini             }
929c1f8fdc3SPaolo Bonzini 
930c1f8fdc3SPaolo Bonzini             /* ... close the descriptor we inherited and go on.  */
931c1f8fdc3SPaolo Bonzini             close(stderr_fd[1]);
932c1f8fdc3SPaolo Bonzini         } else {
933c1f8fdc3SPaolo Bonzini             bool errors = false;
934c1f8fdc3SPaolo Bonzini             char *buf;
935c1f8fdc3SPaolo Bonzini 
936c1f8fdc3SPaolo Bonzini             /* In the parent.  Print error messages from the child until
937c1f8fdc3SPaolo Bonzini              * it closes the pipe.
938c1f8fdc3SPaolo Bonzini              */
939c1f8fdc3SPaolo Bonzini             close(stderr_fd[1]);
940c1f8fdc3SPaolo Bonzini             buf = g_malloc(1024);
941c1f8fdc3SPaolo Bonzini             while ((ret = read(stderr_fd[0], buf, 1024)) > 0) {
942c1f8fdc3SPaolo Bonzini                 errors = true;
943c1f8fdc3SPaolo Bonzini                 ret = qemu_write_full(STDERR_FILENO, buf, ret);
944fc19f8a0SPaolo Bonzini                 if (ret < 0) {
945c1f8fdc3SPaolo Bonzini                     exit(EXIT_FAILURE);
946c1f8fdc3SPaolo Bonzini                 }
947c1f8fdc3SPaolo Bonzini             }
948fc19f8a0SPaolo Bonzini             if (ret < 0) {
94985b01e09SMarkus Armbruster                 error_report("Cannot read from daemon: %s",
95085b01e09SMarkus Armbruster                              strerror(errno));
95185b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
952c1f8fdc3SPaolo Bonzini             }
953c1f8fdc3SPaolo Bonzini 
954c1f8fdc3SPaolo Bonzini             /* Usually the daemon should not print any message.
955c1f8fdc3SPaolo Bonzini              * Exit with zero status in that case.
956c1f8fdc3SPaolo Bonzini              */
957c1f8fdc3SPaolo Bonzini             exit(errors);
958c1f8fdc3SPaolo Bonzini         }
959eb705985SDaniel P. Berrangé #else /* WIN32 */
960eb705985SDaniel P. Berrangé         error_report("Unable to fork into background on Windows hosts");
961eb705985SDaniel P. Berrangé         exit(EXIT_FAILURE);
962eb705985SDaniel P. Berrangé #endif /* WIN32 */
963c1f8fdc3SPaolo Bonzini     }
964c1f8fdc3SPaolo Bonzini 
965a6ac2313SPaolo Bonzini     if (device != NULL && sockpath == NULL) {
966b32f6c28SPaolo Bonzini         sockpath = g_malloc(128);
967b32f6c28SPaolo Bonzini         snprintf(sockpath, 128, SOCKET_PATH, basename(device));
968cd831bd7Sths     }
969cd831bd7Sths 
970e4849c1dSDaniel P. Berrange     server = qio_net_listener_new();
971a721f53bSRichard W.M. Jones     if (socket_activation == 0) {
97248bec07eSDaniel P. Berrange         saddr = nbd_build_socket_address(sockpath, bindto, port);
973fc8135c6SJuan Quintela         if (qio_net_listener_open_sync(server, saddr, 1, &local_err) < 0) {
974e4849c1dSDaniel P. Berrange             object_unref(OBJECT(server));
975a721f53bSRichard W.M. Jones             error_report_err(local_err);
976e4849c1dSDaniel P. Berrange             exit(EXIT_FAILURE);
977a721f53bSRichard W.M. Jones         }
978a721f53bSRichard W.M. Jones     } else {
979e4849c1dSDaniel P. Berrange         size_t i;
980a721f53bSRichard W.M. Jones         /* See comment in check_socket_activation above. */
981e4849c1dSDaniel P. Berrange         for (i = 0; i < socket_activation; i++) {
982e4849c1dSDaniel P. Berrange             QIOChannelSocket *sioc;
983e4849c1dSDaniel P. Berrange             sioc = qio_channel_socket_new_fd(FIRST_SOCKET_ACTIVATION_FD + i,
984a721f53bSRichard W.M. Jones                                              &local_err);
985e4849c1dSDaniel P. Berrange             if (sioc == NULL) {
986e4849c1dSDaniel P. Berrange                 object_unref(OBJECT(server));
9875217f188SMarkus Armbruster                 error_reportf_err(local_err,
9885217f188SMarkus Armbruster                                   "Failed to use socket activation: ");
989a721f53bSRichard W.M. Jones                 exit(EXIT_FAILURE);
990a721f53bSRichard W.M. Jones             }
991e4849c1dSDaniel P. Berrange             qio_net_listener_add(server, sioc);
992e4849c1dSDaniel P. Berrange             object_unref(OBJECT(sioc));
993e4849c1dSDaniel P. Berrange         }
994a721f53bSRichard W.M. Jones     }
99548bec07eSDaniel P. Berrange 
9962f78e491SChrysostomos Nanakos     if (qemu_init_main_loop(&local_err)) {
997565f65d2SMarkus Armbruster         error_report_err(local_err);
9982f78e491SChrysostomos Nanakos         exit(EXIT_FAILURE);
9992f78e491SChrysostomos Nanakos     }
1000802ddc37SPaolo Bonzini     bdrv_init();
1001b3b5299dSKevin Wolf     atexit(qemu_nbd_shutdown);
1002802ddc37SPaolo Bonzini 
100377c9aaefSDaniel P. Berrange     srcpath = argv[optind];
100477c9aaefSDaniel P. Berrange     if (imageOpts) {
100577c9aaefSDaniel P. Berrange         QemuOpts *opts;
100677c9aaefSDaniel P. Berrange         if (fmt) {
100777c9aaefSDaniel P. Berrange             error_report("--image-opts and -f are mutually exclusive");
100877c9aaefSDaniel P. Berrange             exit(EXIT_FAILURE);
100977c9aaefSDaniel P. Berrange         }
101077c9aaefSDaniel P. Berrange         opts = qemu_opts_parse_noisily(&file_opts, srcpath, true);
101177c9aaefSDaniel P. Berrange         if (!opts) {
101277c9aaefSDaniel P. Berrange             qemu_opts_reset(&file_opts);
101377c9aaefSDaniel P. Berrange             exit(EXIT_FAILURE);
101477c9aaefSDaniel P. Berrange         }
101577c9aaefSDaniel P. Berrange         options = qemu_opts_to_qdict(opts, NULL);
101677c9aaefSDaniel P. Berrange         qemu_opts_reset(&file_opts);
1017efaa7c4eSMax Reitz         blk = blk_new_open(NULL, NULL, options, flags, &local_err);
101877c9aaefSDaniel P. Berrange     } else {
1019e6b63677SDaniel P. Berrange         if (fmt) {
10204fbec260SMax Reitz             options = qdict_new();
102146f5ac20SEric Blake             qdict_put_str(options, "driver", fmt);
1022e6b63677SDaniel P. Berrange         }
1023efaa7c4eSMax Reitz         blk = blk_new_open(srcpath, NULL, options, flags, &local_err);
102477c9aaefSDaniel P. Berrange     }
102577c9aaefSDaniel P. Berrange 
10264fbec260SMax Reitz     if (!blk) {
1027c29b77f9SMarkus Armbruster         error_reportf_err(local_err, "Failed to blk_new_open '%s': ",
1028c29b77f9SMarkus Armbruster                           argv[optind]);
102985b01e09SMarkus Armbruster         exit(EXIT_FAILURE);
1030802ddc37SPaolo Bonzini     }
10314fbec260SMax Reitz     bs = blk_bs(blk);
1032802ddc37SPaolo Bonzini 
1033b57e4de0SKevin Wolf     if (dev_offset) {
1034b57e4de0SKevin Wolf         QDict *raw_opts = qdict_new();
1035b57e4de0SKevin Wolf         qdict_put_str(raw_opts, "driver", "raw");
1036b57e4de0SKevin Wolf         qdict_put_str(raw_opts, "file", bs->node_name);
1037b57e4de0SKevin Wolf         qdict_put_int(raw_opts, "offset", dev_offset);
1038b57e4de0SKevin Wolf         bs = bdrv_open(NULL, NULL, raw_opts, flags, &error_fatal);
1039b57e4de0SKevin Wolf         blk_remove_bs(blk);
1040b57e4de0SKevin Wolf         blk_insert_bs(blk, bs, &error_fatal);
1041b57e4de0SKevin Wolf         bdrv_unref(bs);
1042b57e4de0SKevin Wolf     }
1043b57e4de0SKevin Wolf 
10446effd5bfSKevin Wolf     blk_set_enable_write_cache(blk, !writethrough);
10456effd5bfSKevin Wolf 
10468c116b0eSWenchao Xia     if (sn_opts) {
10478c116b0eSWenchao Xia         ret = bdrv_snapshot_load_tmp(bs,
10488c116b0eSWenchao Xia                                      qemu_opt_get(sn_opts, SNAPSHOT_OPT_ID),
10498c116b0eSWenchao Xia                                      qemu_opt_get(sn_opts, SNAPSHOT_OPT_NAME),
10508c116b0eSWenchao Xia                                      &local_err);
10518c116b0eSWenchao Xia     } else if (sn_id_or_name) {
10528c116b0eSWenchao Xia         ret = bdrv_snapshot_load_tmp_by_id_or_name(bs, sn_id_or_name,
10538c116b0eSWenchao Xia                                                    &local_err);
10548c116b0eSWenchao Xia     }
10558c116b0eSWenchao Xia     if (ret < 0) {
1056c29b77f9SMarkus Armbruster         error_reportf_err(local_err, "Failed to load snapshot: ");
105785b01e09SMarkus Armbruster         exit(EXIT_FAILURE);
10588c116b0eSWenchao Xia     }
10598c116b0eSWenchao Xia 
1060b3838a40SPeter Lieven     bs->detect_zeroes = detect_zeroes;
1061802ddc37SPaolo Bonzini 
106200917172SKevin Wolf     nbd_server_is_qemu_nbd(true);
106300917172SKevin Wolf 
106400917172SKevin Wolf     export_opts = g_new(BlockExportOptions, 1);
106500917172SKevin Wolf     *export_opts = (BlockExportOptions) {
106600917172SKevin Wolf         .type               = BLOCK_EXPORT_TYPE_NBD,
1067d53be9ceSKevin Wolf         .id                 = g_strdup("qemu-nbd-export"),
1068b6076afcSKevin Wolf         .node_name          = g_strdup(bdrv_get_node_name(bs)),
106900917172SKevin Wolf         .has_writethrough   = true,
107000917172SKevin Wolf         .writethrough       = writethrough,
107130dbc81dSKevin Wolf         .has_writable       = true,
107230dbc81dSKevin Wolf         .writable           = !readonly,
107300917172SKevin Wolf         .u.nbd = {
107400917172SKevin Wolf             .has_name             = true,
107500917172SKevin Wolf             .name                 = g_strdup(export_name),
107600917172SKevin Wolf             .has_description      = !!export_description,
107700917172SKevin Wolf             .description          = g_strdup(export_description),
1078cbad81ceSEric Blake             .has_bitmaps          = !!bitmaps,
1079cbad81ceSEric Blake             .bitmaps              = bitmaps,
1080dbc7b014SEric Blake             .has_allocation_depth = alloc_depth,
1081dbc7b014SEric Blake             .allocation_depth     = alloc_depth,
108200917172SKevin Wolf         },
108300917172SKevin Wolf     };
108400917172SKevin Wolf     blk_exp_add(export_opts, &error_fatal);
108500917172SKevin Wolf     qapi_free_BlockExportOptions(export_opts);
10863b05a8e9Sths 
1087f1ef5555SPaolo Bonzini     if (device) {
10883c1fa35dSEric Blake #if HAVE_NBD_DEVICE
1089f1ef5555SPaolo Bonzini         int ret;
1090f1ef5555SPaolo Bonzini 
1091a6ac2313SPaolo Bonzini         ret = pthread_create(&client_thread, NULL, nbd_client_thread, device);
1092f1ef5555SPaolo Bonzini         if (ret != 0) {
109385b01e09SMarkus Armbruster             error_report("Failed to create client thread: %s", strerror(ret));
109485b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
1095f1ef5555SPaolo Bonzini         }
10963c1fa35dSEric Blake #endif
1097f1ef5555SPaolo Bonzini     } else {
1098f1ef5555SPaolo Bonzini         /* Shut up GCC warnings.  */
1099f1ef5555SPaolo Bonzini         memset(&client_thread, 0, sizeof(client_thread));
1100f1ef5555SPaolo Bonzini     }
1101f1ef5555SPaolo Bonzini 
1102d0d6ff58SDaniel P. Berrange     nbd_update_server_watch();
11037a5ca864Sbellard 
1104637bc5a5SMax Reitz     if (pid_file_name) {
1105637bc5a5SMax Reitz         qemu_write_pidfile(pid_file_name, &error_fatal);
1106637bc5a5SMax Reitz     }
1107637bc5a5SMax Reitz 
11089faf31b6SMichael Tokarev     /* now when the initialization is (almost) complete, chdir("/")
11099faf31b6SMichael Tokarev      * to free any busy filesystems */
11109faf31b6SMichael Tokarev     if (chdir("/") < 0) {
111185b01e09SMarkus Armbruster         error_report("Could not chdir to root directory: %s",
111285b01e09SMarkus Armbruster                      strerror(errno));
111385b01e09SMarkus Armbruster         exit(EXIT_FAILURE);
11149faf31b6SMichael Tokarev     }
11159faf31b6SMichael Tokarev 
1116ffb31e1dSMax Reitz     if (fork_process) {
1117ffb31e1dSMax Reitz         dup2(old_stderr, STDERR_FILENO);
1118ffb31e1dSMax Reitz         close(old_stderr);
1119ffb31e1dSMax Reitz     }
1120ffb31e1dSMax Reitz 
11217860a380SPaolo Bonzini     state = RUNNING;
11223b05a8e9Sths     do {
1123a61c6782SPaolo Bonzini         main_loop_wait(false);
11247860a380SPaolo Bonzini         if (state == TERMINATE) {
1125bc4ee65bSKevin Wolf             blk_exp_close_all();
1126d794f7f3SKevin Wolf             state = TERMINATED;
11277860a380SPaolo Bonzini         }
11287860a380SPaolo Bonzini     } while (state != TERMINATED);
11292c8d9f06SPaolo Bonzini 
113026f54e9aSMarkus Armbruster     blk_unref(blk);
1131b32f6c28SPaolo Bonzini     if (sockpath) {
1132b32f6c28SPaolo Bonzini         unlink(sockpath);
1133b32f6c28SPaolo Bonzini     }
11347a5ca864Sbellard 
11358c116b0eSWenchao Xia     qemu_opts_del(sn_opts);
11368c116b0eSWenchao Xia 
1137a517e88bSPaolo Bonzini     if (device) {
1138a517e88bSPaolo Bonzini         void *ret;
1139a517e88bSPaolo Bonzini         pthread_join(client_thread, &ret);
1140a517e88bSPaolo Bonzini         exit(ret != NULL);
1141a517e88bSPaolo Bonzini     } else {
1142a517e88bSPaolo Bonzini         exit(EXIT_SUCCESS);
1143a517e88bSPaolo Bonzini     }
11447a5ca864Sbellard }
1145