xref: /qemu/qemu-nbd.c (revision d794f7f3)
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"
28b3838a40SPeter Lieven #include "block/block_int.h"
29737e150eSPaolo Bonzini #include "block/nbd.h"
306a1751b7SAlex Bligh #include "qemu/main-loop.h"
310b8fa32fSMarkus Armbruster #include "qemu/module.h"
32922a01a0SMarkus Armbruster #include "qemu/option.h"
33537b41f5SPaolo Bonzini #include "qemu/error-report.h"
340ab3b337SDaniel P. Berrange #include "qemu/config-file.h"
3558369e22SPaolo Bonzini #include "qemu/bswap.h"
3639ca463eSDenis V. Lunev #include "qemu/log.h"
3753fabd4bSPaolo Bonzini #include "qemu/systemd.h"
388c116b0eSWenchao Xia #include "block/snapshot.h"
39452fcdbcSMarkus Armbruster #include "qapi/qmp/qdict.h"
40d49b6836SMarkus Armbruster #include "qapi/qmp/qstring.h"
410ab3b337SDaniel P. Berrange #include "qom/object_interfaces.h"
42d0d6ff58SDaniel P. Berrange #include "io/channel-socket.h"
43e4849c1dSDaniel P. Berrange #include "io/net-listener.h"
44c2297088SDaniel P. Berrange #include "crypto/init.h"
4539ca463eSDenis V. Lunev #include "trace/control.h"
46be377133SEric Blake #include "qemu-version.h"
477a5ca864Sbellard 
483c1fa35dSEric Blake #ifdef __linux__
493c1fa35dSEric Blake #define HAVE_NBD_DEVICE 1
503c1fa35dSEric Blake #else
513c1fa35dSEric Blake #define HAVE_NBD_DEVICE 0
523c1fa35dSEric Blake #endif
533c1fa35dSEric Blake 
54cd831bd7Sths #define SOCKET_PATH                "/var/lock/qemu-nbd-%s"
55fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_CACHE         256
56fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_AIO           257
57fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_DISCARD       258
58fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_DETECT_ZEROES 259
59fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_OBJECT        260
60fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_TLSCREDS      261
61fa8b7ce2SDaniel P. Berrange #define QEMU_NBD_OPT_IMAGE_OPTS    262
62ffb31e1dSMax Reitz #define QEMU_NBD_OPT_FORK          263
63b25e12daSDaniel P. Berrange #define QEMU_NBD_OPT_TLSAUTHZ      264
64637bc5a5SMax Reitz #define QEMU_NBD_OPT_PID_FILE      265
657a5ca864Sbellard 
66bd31c214SEric Blake #define MBR_SIZE 512
67bd31c214SEric Blake 
689d976580SPhilippe Mathieu-Daudé static NBDExport *export;
69b1d8e52eSblueswir1 static int verbose;
70a517e88bSPaolo Bonzini static char *srcpath;
71bd269ebcSMarkus Armbruster static SocketAddress *saddr;
727860a380SPaolo Bonzini static int persistent = 0;
73*d794f7f3SKevin 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"
103636192c4SEric Blake "  -B, --bitmap=NAME         expose a persistent dirty bitmap\n"
104b033cd86SPaolo Bonzini "\n"
1050ab3b337SDaniel P. Berrange "General purpose options:\n"
10668b96f15SEric Blake "  -L, --list                list exports available from another NBD server\n"
1070ab3b337SDaniel P. Berrange "  --object type,id=ID,...   define an object such as 'secret' for providing\n"
1080ab3b337SDaniel P. Berrange "                            passwords and/or encryption keys\n"
109f7812df7SEric Blake "  --tls-creds=ID            use id of an earlier --object to provide TLS\n"
110b25e12daSDaniel P. Berrange "  --tls-authz=ID            use id of an earlier --object to provide\n"
111b25e12daSDaniel P. Berrange "                            authorization\n"
11239ca463eSDenis V. Lunev "  -T, --trace [[enable=]<pattern>][,events=<file>][,file=<file>]\n"
11339ca463eSDenis V. Lunev "                            specify tracing options\n"
114ffb31e1dSMax Reitz "  --fork                    fork off the server process and exit the parent\n"
115ffb31e1dSMax Reitz "                            once the server is running\n"
116637bc5a5SMax Reitz "  --pid-file=PATH           store the server's process ID in the given file\n"
1173c1fa35dSEric Blake #if HAVE_NBD_DEVICE
1183c1fa35dSEric Blake "\n"
119b033cd86SPaolo Bonzini "Kernel NBD client support:\n"
120b033cd86SPaolo Bonzini "  -c, --connect=DEV         connect FILE to the local NBD device DEV\n"
121b033cd86SPaolo Bonzini "  -d, --disconnect          disconnect the specified device\n"
122b033cd86SPaolo Bonzini #endif
123b033cd86SPaolo Bonzini "\n"
124b033cd86SPaolo Bonzini "Block device options:\n"
1254323fdcfSWenchao Xia "  -f, --format=FORMAT       set image format (raw, qcow2, ...)\n"
126b033cd86SPaolo Bonzini "  -r, --read-only           export read-only\n"
1278c116b0eSWenchao Xia "  -s, --snapshot            use FILE as an external snapshot, create a temporary\n"
1288c116b0eSWenchao Xia "                            file with backing_file=FILE, redirect the write to\n"
1298c116b0eSWenchao Xia "                            the temporary one\n"
1308c116b0eSWenchao Xia "  -l, --load-snapshot=SNAPSHOT_PARAM\n"
1318c116b0eSWenchao Xia "                            load an internal snapshot inside FILE and export it\n"
1328c116b0eSWenchao Xia "                            as an read-only device, SNAPSHOT_PARAM format is\n"
1338c116b0eSWenchao Xia "                            'snapshot.id=[ID],snapshot.name=[NAME]', or\n"
1348c116b0eSWenchao Xia "                            '[ID_OR_NAME]'\n"
135b033cd86SPaolo Bonzini "  -n, --nocache             disable host cache\n"
13639a5235cSPaolo Bonzini "      --cache=MODE          set cache mode (none, writeback, ...)\n"
1377680274dSAarushi Mehta "      --aio=MODE            set AIO mode (native, io_uring or threads)\n"
138b3838a40SPeter Lieven "      --discard=MODE        set discard mode (ignore, unmap)\n"
1396883de6cSAndrey Korolyov "      --detect-zeroes=MODE  set detect-zeroes mode (off, on, unmap)\n"
14077c9aaefSDaniel P. Berrange "      --image-opts          treat FILE as a full set of image options\n"
141b033cd86SPaolo Bonzini "\n"
142f5048cb7SEric Blake QEMU_HELP_BOTTOM "\n"
14368b96f15SEric Blake     , name, name, NBD_DEFAULT_PORT, "DEVICE");
1447a5ca864Sbellard }
1457a5ca864Sbellard 
1467a5ca864Sbellard static void version(const char *name)
1477a5ca864Sbellard {
1487a5ca864Sbellard     printf(
1497e563bfbSThomas Huth "%s " QEMU_FULL_VERSION "\n"
1507a5ca864Sbellard "Written by Anthony Liguori.\n"
1517a5ca864Sbellard "\n"
152be377133SEric Blake QEMU_COPYRIGHT "\n"
1537a5ca864Sbellard "This is free software; see the source for copying conditions.  There is NO\n"
1547a5ca864Sbellard "warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"
155315bc7aaSths     , name);
1567a5ca864Sbellard }
1577a5ca864Sbellard 
1586e64dd57SDaniel P. Berrangé #if HAVE_NBD_DEVICE
159bb345110SPaolo Bonzini static void termsig_handler(int signum)
160bb345110SPaolo Bonzini {
161d73415a3SStefan Hajnoczi     qatomic_cmpxchg(&state, RUNNING, TERMINATE);
162a61c6782SPaolo Bonzini     qemu_notify_event();
163bb345110SPaolo Bonzini }
1646e64dd57SDaniel P. Berrangé #endif /* HAVE_NBD_DEVICE */
165537b41f5SPaolo Bonzini 
16668b96f15SEric Blake static int qemu_nbd_client_list(SocketAddress *saddr, QCryptoTLSCreds *tls,
16768b96f15SEric Blake                                 const char *hostname)
16868b96f15SEric Blake {
16968b96f15SEric Blake     int ret = EXIT_FAILURE;
17068b96f15SEric Blake     int rc;
17168b96f15SEric Blake     Error *err = NULL;
17268b96f15SEric Blake     QIOChannelSocket *sioc;
17368b96f15SEric Blake     NBDExportInfo *list;
17468b96f15SEric Blake     int i, j;
17568b96f15SEric Blake 
17668b96f15SEric Blake     sioc = qio_channel_socket_new();
17768b96f15SEric Blake     if (qio_channel_socket_connect_sync(sioc, saddr, &err) < 0) {
17868b96f15SEric Blake         error_report_err(err);
17968b96f15SEric Blake         return EXIT_FAILURE;
18068b96f15SEric Blake     }
18168b96f15SEric Blake     rc = nbd_receive_export_list(QIO_CHANNEL(sioc), tls, hostname, &list,
18268b96f15SEric Blake                                  &err);
18368b96f15SEric Blake     if (rc < 0) {
18468b96f15SEric Blake         if (err) {
18568b96f15SEric Blake             error_report_err(err);
18668b96f15SEric Blake         }
18768b96f15SEric Blake         goto out;
18868b96f15SEric Blake     }
18968b96f15SEric Blake     printf("exports available: %d\n", rc);
19068b96f15SEric Blake     for (i = 0; i < rc; i++) {
19168b96f15SEric Blake         printf(" export: '%s'\n", list[i].name);
19268b96f15SEric Blake         if (list[i].description && *list[i].description) {
19368b96f15SEric Blake             printf("  description: %s\n", list[i].description);
19468b96f15SEric Blake         }
19568b96f15SEric Blake         if (list[i].flags & NBD_FLAG_HAS_FLAGS) {
196c4e2aff8SMax Reitz             static const char *const flag_names[] = {
197c4e2aff8SMax Reitz                 [NBD_FLAG_READ_ONLY_BIT]            = "readonly",
198c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_FLUSH_BIT]           = "flush",
199c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_FUA_BIT]             = "fua",
200c4e2aff8SMax Reitz                 [NBD_FLAG_ROTATIONAL_BIT]           = "rotational",
201c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_TRIM_BIT]            = "trim",
202c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_WRITE_ZEROES_BIT]    = "zeroes",
203c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_DF_BIT]              = "df",
204c4e2aff8SMax Reitz                 [NBD_FLAG_CAN_MULTI_CONN_BIT]       = "multi",
205c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_RESIZE_BIT]          = "resize",
206c4e2aff8SMax Reitz                 [NBD_FLAG_SEND_CACHE_BIT]           = "cache",
2070a479545SEric Blake                 [NBD_FLAG_SEND_FAST_ZERO_BIT]       = "fast-zero",
208c4e2aff8SMax Reitz             };
209c4e2aff8SMax Reitz 
21068b96f15SEric Blake             printf("  size:  %" PRIu64 "\n", list[i].size);
21168b96f15SEric Blake             printf("  flags: 0x%x (", list[i].flags);
212c4e2aff8SMax Reitz             for (size_t bit = 0; bit < ARRAY_SIZE(flag_names); bit++) {
213c4e2aff8SMax Reitz                 if (flag_names[bit] && (list[i].flags & (1 << bit))) {
214c4e2aff8SMax Reitz                     printf(" %s", flag_names[bit]);
21568b96f15SEric Blake                 }
21668b96f15SEric Blake             }
21768b96f15SEric Blake             printf(" )\n");
21868b96f15SEric Blake         }
21968b96f15SEric Blake         if (list[i].min_block) {
22068b96f15SEric Blake             printf("  min block: %u\n", list[i].min_block);
22168b96f15SEric Blake             printf("  opt block: %u\n", list[i].opt_block);
22268b96f15SEric Blake             printf("  max block: %u\n", list[i].max_block);
22368b96f15SEric Blake         }
22468b96f15SEric Blake         if (list[i].n_contexts) {
22568b96f15SEric Blake             printf("  available meta contexts: %d\n", list[i].n_contexts);
22668b96f15SEric Blake             for (j = 0; j < list[i].n_contexts; j++) {
22768b96f15SEric Blake                 printf("   %s\n", list[i].contexts[j]);
22868b96f15SEric Blake             }
22968b96f15SEric Blake         }
23068b96f15SEric Blake     }
23168b96f15SEric Blake     nbd_free_export_list(list, rc);
23268b96f15SEric Blake 
23368b96f15SEric Blake     ret = EXIT_SUCCESS;
23468b96f15SEric Blake  out:
23568b96f15SEric Blake     object_unref(OBJECT(sioc));
23668b96f15SEric Blake     return ret;
23768b96f15SEric Blake }
23868b96f15SEric Blake 
23968b96f15SEric Blake 
2403c1fa35dSEric Blake #if HAVE_NBD_DEVICE
241a517e88bSPaolo Bonzini static void *show_parts(void *arg)
242cd831bd7Sths {
243a6ac2313SPaolo Bonzini     char *device = arg;
244cd831bd7Sths     int nbd;
245cd831bd7Sths 
246cd831bd7Sths     /* linux just needs an open() to trigger
247cd831bd7Sths      * the partition table update
248cd831bd7Sths      * but remember to load the module with max_part != 0 :
249cd831bd7Sths      *     modprobe nbd max_part=63
250cd831bd7Sths      */
251cd831bd7Sths     nbd = open(device, O_RDWR);
252fc19f8a0SPaolo Bonzini     if (nbd >= 0) {
253cd831bd7Sths         close(nbd);
254cd831bd7Sths     }
255a517e88bSPaolo Bonzini     return NULL;
256a517e88bSPaolo Bonzini }
257a517e88bSPaolo Bonzini 
258a517e88bSPaolo Bonzini static void *nbd_client_thread(void *arg)
259a517e88bSPaolo Bonzini {
260a6ac2313SPaolo Bonzini     char *device = arg;
2616dc1667dSEric Blake     NBDExportInfo info = { .request_sizes = false, .name = g_strdup("") };
262d0d6ff58SDaniel P. Berrange     QIOChannelSocket *sioc;
263d0d6ff58SDaniel P. Berrange     int fd;
264a517e88bSPaolo Bonzini     int ret;
265a517e88bSPaolo Bonzini     pthread_t show_parts_thread;
2661ce52846SMax Reitz     Error *local_error = NULL;
267a517e88bSPaolo Bonzini 
268d0d6ff58SDaniel P. Berrange     sioc = qio_channel_socket_new();
269d0d6ff58SDaniel P. Berrange     if (qio_channel_socket_connect_sync(sioc,
270d0d6ff58SDaniel P. Berrange                                         saddr,
271d0d6ff58SDaniel P. Berrange                                         &local_error) < 0) {
27248bec07eSDaniel P. Berrange         error_report_err(local_error);
273a517e88bSPaolo Bonzini         goto out;
274a517e88bSPaolo Bonzini     }
275a517e88bSPaolo Bonzini 
276a8e2bb6aSVladimir Sementsov-Ogievskiy     ret = nbd_receive_negotiate(NULL, QIO_CHANNEL(sioc),
277004a89fcSEric Blake                                 NULL, NULL, NULL, &info, &local_error);
278fc19f8a0SPaolo Bonzini     if (ret < 0) {
2791ce52846SMax Reitz         if (local_error) {
28078288671SMarkus Armbruster             error_report_err(local_error);
2811ce52846SMax Reitz         }
2820c544d73SPaolo Bonzini         goto out_socket;
283a517e88bSPaolo Bonzini     }
284a517e88bSPaolo Bonzini 
285a6ac2313SPaolo Bonzini     fd = open(device, O_RDWR);
286fc19f8a0SPaolo Bonzini     if (fd < 0) {
287a6ac2313SPaolo Bonzini         /* Linux-only, we can use %m in printf.  */
288b9884681SMarkus Armbruster         error_report("Failed to open %s: %m", device);
2890c544d73SPaolo Bonzini         goto out_socket;
290a6ac2313SPaolo Bonzini     }
291a6ac2313SPaolo Bonzini 
292004a89fcSEric Blake     ret = nbd_init(fd, sioc, &info, &local_error);
293fc19f8a0SPaolo Bonzini     if (ret < 0) {
294be41c100SVladimir Sementsov-Ogievskiy         error_report_err(local_error);
2950c544d73SPaolo Bonzini         goto out_fd;
296a517e88bSPaolo Bonzini     }
297a517e88bSPaolo Bonzini 
298a517e88bSPaolo Bonzini     /* update partition table */
299a6ac2313SPaolo Bonzini     pthread_create(&show_parts_thread, NULL, show_parts, device);
300a517e88bSPaolo Bonzini 
301c1f8fdc3SPaolo Bonzini     if (verbose) {
302a517e88bSPaolo Bonzini         fprintf(stderr, "NBD device %s is now connected to %s\n",
303a517e88bSPaolo Bonzini                 device, srcpath);
304c1f8fdc3SPaolo Bonzini     } else {
305c1f8fdc3SPaolo Bonzini         /* Close stderr so that the qemu-nbd process exits.  */
306c1f8fdc3SPaolo Bonzini         dup2(STDOUT_FILENO, STDERR_FILENO);
307c1f8fdc3SPaolo Bonzini     }
308a517e88bSPaolo Bonzini 
309a517e88bSPaolo Bonzini     ret = nbd_client(fd);
310a517e88bSPaolo Bonzini     if (ret) {
3110c544d73SPaolo Bonzini         goto out_fd;
312a517e88bSPaolo Bonzini     }
313a517e88bSPaolo Bonzini     close(fd);
314d0d6ff58SDaniel P. Berrange     object_unref(OBJECT(sioc));
3156dc1667dSEric Blake     g_free(info.name);
316a517e88bSPaolo Bonzini     kill(getpid(), SIGTERM);
317a517e88bSPaolo Bonzini     return (void *) EXIT_SUCCESS;
318a517e88bSPaolo Bonzini 
3190c544d73SPaolo Bonzini out_fd:
3200c544d73SPaolo Bonzini     close(fd);
3210c544d73SPaolo Bonzini out_socket:
322d0d6ff58SDaniel P. Berrange     object_unref(OBJECT(sioc));
323a517e88bSPaolo Bonzini out:
3246dc1667dSEric Blake     g_free(info.name);
325a517e88bSPaolo Bonzini     kill(getpid(), SIGTERM);
326a517e88bSPaolo Bonzini     return (void *) EXIT_FAILURE;
327cd831bd7Sths }
3283c1fa35dSEric Blake #endif /* HAVE_NBD_DEVICE */
329cd831bd7Sths 
330e4afbf4fSFam Zheng static int nbd_can_accept(void)
331a61c6782SPaolo Bonzini {
332df8ad9f1SEric Blake     return state == RUNNING && nb_fds < shared;
333a61c6782SPaolo Bonzini }
334a61c6782SPaolo Bonzini 
335d0d6ff58SDaniel P. Berrange static void nbd_update_server_watch(void);
336e4afbf4fSFam Zheng 
3370c9390d9SEric Blake static void nbd_client_closed(NBDClient *client, bool negotiated)
338a61c6782SPaolo Bonzini {
339a61c6782SPaolo Bonzini     nb_fds--;
3400c9390d9SEric Blake     if (negotiated && nb_fds == 0 && !persistent && state == RUNNING) {
3417860a380SPaolo Bonzini         state = TERMINATE;
3427860a380SPaolo Bonzini     }
343d0d6ff58SDaniel P. Berrange     nbd_update_server_watch();
3447860a380SPaolo Bonzini     nbd_client_put(client);
345a61c6782SPaolo Bonzini }
346a61c6782SPaolo Bonzini 
347e4849c1dSDaniel P. Berrange static void nbd_accept(QIONetListener *listener, QIOChannelSocket *cioc,
348e4849c1dSDaniel P. Berrange                        gpointer opaque)
349a61c6782SPaolo Bonzini {
3507860a380SPaolo Bonzini     if (state >= TERMINATE) {
351e4849c1dSDaniel P. Berrange         return;
3527860a380SPaolo Bonzini     }
3537860a380SPaolo Bonzini 
354a61c6782SPaolo Bonzini     nb_fds++;
355d0d6ff58SDaniel P. Berrange     nbd_update_server_watch();
356b25e12daSDaniel P. Berrange     nbd_client_new(cioc, tlscreds, tlsauthz, nbd_client_closed);
357d0d6ff58SDaniel P. Berrange }
358d0d6ff58SDaniel P. Berrange 
359d0d6ff58SDaniel P. Berrange static void nbd_update_server_watch(void)
360e4afbf4fSFam Zheng {
361e4afbf4fSFam Zheng     if (nbd_can_accept()) {
362e4849c1dSDaniel P. Berrange         qio_net_listener_set_client_func(server, nbd_accept, NULL, NULL);
363e4afbf4fSFam Zheng     } else {
364e4849c1dSDaniel P. Berrange         qio_net_listener_set_client_func(server, NULL, NULL, NULL);
365e4afbf4fSFam Zheng     }
366e4afbf4fSFam Zheng }
367e4afbf4fSFam Zheng 
36848bec07eSDaniel P. Berrange 
369bd269ebcSMarkus Armbruster static SocketAddress *nbd_build_socket_address(const char *sockpath,
37048bec07eSDaniel P. Berrange                                                const char *bindto,
37148bec07eSDaniel P. Berrange                                                const char *port)
37248bec07eSDaniel P. Berrange {
373bd269ebcSMarkus Armbruster     SocketAddress *saddr;
37448bec07eSDaniel P. Berrange 
375bd269ebcSMarkus Armbruster     saddr = g_new0(SocketAddress, 1);
37648bec07eSDaniel P. Berrange     if (sockpath) {
377bd269ebcSMarkus Armbruster         saddr->type = SOCKET_ADDRESS_TYPE_UNIX;
378bd269ebcSMarkus Armbruster         saddr->u.q_unix.path = g_strdup(sockpath);
37948bec07eSDaniel P. Berrange     } else {
3800399293eSEric Blake         InetSocketAddress *inet;
381bd269ebcSMarkus Armbruster         saddr->type = SOCKET_ADDRESS_TYPE_INET;
382bd269ebcSMarkus Armbruster         inet = &saddr->u.inet;
3830399293eSEric Blake         inet->host = g_strdup(bindto);
38448bec07eSDaniel P. Berrange         if (port) {
3850399293eSEric Blake             inet->port = g_strdup(port);
38648bec07eSDaniel P. Berrange         } else  {
3870399293eSEric Blake             inet->port = g_strdup_printf("%d", NBD_DEFAULT_PORT);
38848bec07eSDaniel P. Berrange         }
38948bec07eSDaniel P. Berrange     }
39048bec07eSDaniel P. Berrange 
39148bec07eSDaniel P. Berrange     return saddr;
39248bec07eSDaniel P. Berrange }
39348bec07eSDaniel P. Berrange 
39448bec07eSDaniel P. Berrange 
39577c9aaefSDaniel P. Berrange static QemuOptsList file_opts = {
39677c9aaefSDaniel P. Berrange     .name = "file",
39777c9aaefSDaniel P. Berrange     .implied_opt_name = "file",
39877c9aaefSDaniel P. Berrange     .head = QTAILQ_HEAD_INITIALIZER(file_opts.head),
39977c9aaefSDaniel P. Berrange     .desc = {
40077c9aaefSDaniel P. Berrange         /* no elements => accept any params */
40177c9aaefSDaniel P. Berrange         { /* end of list */ }
40277c9aaefSDaniel P. Berrange     },
40377c9aaefSDaniel P. Berrange };
40477c9aaefSDaniel P. Berrange 
4050ab3b337SDaniel P. Berrange static QemuOptsList qemu_object_opts = {
4060ab3b337SDaniel P. Berrange     .name = "object",
4070ab3b337SDaniel P. Berrange     .implied_opt_name = "qom-type",
4080ab3b337SDaniel P. Berrange     .head = QTAILQ_HEAD_INITIALIZER(qemu_object_opts.head),
4090ab3b337SDaniel P. Berrange     .desc = {
4100ab3b337SDaniel P. Berrange         { }
4110ab3b337SDaniel P. Berrange     },
4120ab3b337SDaniel P. Berrange };
4130ab3b337SDaniel P. Berrange 
414495bf893SKevin Wolf static bool qemu_nbd_object_print_help(const char *type, QemuOpts *opts)
415495bf893SKevin Wolf {
416495bf893SKevin Wolf     if (user_creatable_print_help(type, opts)) {
417495bf893SKevin Wolf         exit(0);
418495bf893SKevin Wolf     }
419495bf893SKevin Wolf     return true;
420495bf893SKevin Wolf }
4210ab3b337SDaniel P. Berrange 
422145614a1SDaniel P. Berrange 
42368b96f15SEric Blake static QCryptoTLSCreds *nbd_get_tls_creds(const char *id, bool list,
42468b96f15SEric Blake                                           Error **errp)
425145614a1SDaniel P. Berrange {
426145614a1SDaniel P. Berrange     Object *obj;
427145614a1SDaniel P. Berrange     QCryptoTLSCreds *creds;
428145614a1SDaniel P. Berrange 
429145614a1SDaniel P. Berrange     obj = object_resolve_path_component(
430145614a1SDaniel P. Berrange         object_get_objects_root(), id);
431145614a1SDaniel P. Berrange     if (!obj) {
432145614a1SDaniel P. Berrange         error_setg(errp, "No TLS credentials with id '%s'",
433145614a1SDaniel P. Berrange                    id);
434145614a1SDaniel P. Berrange         return NULL;
435145614a1SDaniel P. Berrange     }
436145614a1SDaniel P. Berrange     creds = (QCryptoTLSCreds *)
437145614a1SDaniel P. Berrange         object_dynamic_cast(obj, TYPE_QCRYPTO_TLS_CREDS);
438145614a1SDaniel P. Berrange     if (!creds) {
439145614a1SDaniel P. Berrange         error_setg(errp, "Object with id '%s' is not TLS credentials",
440145614a1SDaniel P. Berrange                    id);
441145614a1SDaniel P. Berrange         return NULL;
442145614a1SDaniel P. Berrange     }
443145614a1SDaniel P. Berrange 
44468b96f15SEric Blake     if (list) {
44568b96f15SEric Blake         if (creds->endpoint != QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT) {
44668b96f15SEric Blake             error_setg(errp,
44768b96f15SEric Blake                        "Expecting TLS credentials with a client endpoint");
44868b96f15SEric Blake             return NULL;
44968b96f15SEric Blake         }
45068b96f15SEric Blake     } else {
451145614a1SDaniel P. Berrange         if (creds->endpoint != QCRYPTO_TLS_CREDS_ENDPOINT_SERVER) {
452145614a1SDaniel P. Berrange             error_setg(errp,
453145614a1SDaniel P. Berrange                        "Expecting TLS credentials with a server endpoint");
454145614a1SDaniel P. Berrange             return NULL;
455145614a1SDaniel P. Berrange         }
45668b96f15SEric Blake     }
457145614a1SDaniel P. Berrange     object_ref(obj);
458145614a1SDaniel P. Berrange     return creds;
459145614a1SDaniel P. Berrange }
460145614a1SDaniel P. Berrange 
461a721f53bSRichard W.M. Jones static void setup_address_and_port(const char **address, const char **port)
462a721f53bSRichard W.M. Jones {
463a721f53bSRichard W.M. Jones     if (*address == NULL) {
464a721f53bSRichard W.M. Jones         *address = "0.0.0.0";
465a721f53bSRichard W.M. Jones     }
466a721f53bSRichard W.M. Jones 
467a721f53bSRichard W.M. Jones     if (*port == NULL) {
468a721f53bSRichard W.M. Jones         *port = stringify(NBD_DEFAULT_PORT);
469a721f53bSRichard W.M. Jones     }
470a721f53bSRichard W.M. Jones }
471a721f53bSRichard W.M. Jones 
472a721f53bSRichard W.M. Jones /*
473a721f53bSRichard W.M. Jones  * Check socket parameters compatibility when socket activation is used.
474a721f53bSRichard W.M. Jones  */
475a721f53bSRichard W.M. Jones static const char *socket_activation_validate_opts(const char *device,
476a721f53bSRichard W.M. Jones                                                    const char *sockpath,
477a721f53bSRichard W.M. Jones                                                    const char *address,
47868b96f15SEric Blake                                                    const char *port,
47968b96f15SEric Blake                                                    bool list)
480a721f53bSRichard W.M. Jones {
481a721f53bSRichard W.M. Jones     if (device != NULL) {
482a721f53bSRichard W.M. Jones         return "NBD device can't be set when using socket activation";
483a721f53bSRichard W.M. Jones     }
484a721f53bSRichard W.M. Jones 
485a721f53bSRichard W.M. Jones     if (sockpath != NULL) {
486a721f53bSRichard W.M. Jones         return "Unix socket can't be set when using socket activation";
487a721f53bSRichard W.M. Jones     }
488a721f53bSRichard W.M. Jones 
489a721f53bSRichard W.M. Jones     if (address != NULL) {
490a721f53bSRichard W.M. Jones         return "The interface can't be set when using socket activation";
491a721f53bSRichard W.M. Jones     }
492a721f53bSRichard W.M. Jones 
493a721f53bSRichard W.M. Jones     if (port != NULL) {
494a721f53bSRichard W.M. Jones         return "TCP port number can't be set when using socket activation";
495a721f53bSRichard W.M. Jones     }
496a721f53bSRichard W.M. Jones 
49768b96f15SEric Blake     if (list) {
49868b96f15SEric Blake         return "List mode is incompatible with socket activation";
49968b96f15SEric Blake     }
50068b96f15SEric Blake 
501a721f53bSRichard W.M. Jones     return NULL;
502a721f53bSRichard W.M. Jones }
503145614a1SDaniel P. Berrange 
504b3b5299dSKevin Wolf static void qemu_nbd_shutdown(void)
505b3b5299dSKevin Wolf {
506b3b5299dSKevin Wolf     job_cancel_sync_all();
507b3b5299dSKevin Wolf     bdrv_close_all();
508b3b5299dSKevin Wolf }
509b3b5299dSKevin Wolf 
5107a5ca864Sbellard int main(int argc, char **argv)
5117a5ca864Sbellard {
51226f54e9aSMarkus Armbruster     BlockBackend *blk;
5137a5ca864Sbellard     BlockDriverState *bs;
5149d26dfcbSEric Blake     uint64_t dev_offset = 0;
515dbb38caaSEric Blake     bool readonly = false;
516cd831bd7Sths     bool disconnect = false;
517a721f53bSRichard W.M. Jones     const char *bindto = NULL;
51848bec07eSDaniel P. Berrange     const char *port = NULL;
51948bec07eSDaniel P. Berrange     char *sockpath = NULL;
520a6ac2313SPaolo Bonzini     char *device = NULL;
5218c116b0eSWenchao Xia     QemuOpts *sn_opts = NULL;
5228c116b0eSWenchao Xia     const char *sn_id_or_name = NULL;
5230bc16997SEric Blake     const char *sopt = "hVb:o:p:rsnc:dvk:e:f:tl:x:T:D:B:L";
5247a5ca864Sbellard     struct option lopt[] = {
525aa6e546cSDaniel P. Berrange         { "help", no_argument, NULL, 'h' },
526aa6e546cSDaniel P. Berrange         { "version", no_argument, NULL, 'V' },
527aa6e546cSDaniel P. Berrange         { "bind", required_argument, NULL, 'b' },
528aa6e546cSDaniel P. Berrange         { "port", required_argument, NULL, 'p' },
529aa6e546cSDaniel P. Berrange         { "socket", required_argument, NULL, 'k' },
530aa6e546cSDaniel P. Berrange         { "offset", required_argument, NULL, 'o' },
531aa6e546cSDaniel P. Berrange         { "read-only", no_argument, NULL, 'r' },
532636192c4SEric Blake         { "bitmap", required_argument, NULL, 'B' },
533aa6e546cSDaniel P. Berrange         { "connect", required_argument, NULL, 'c' },
534aa6e546cSDaniel P. Berrange         { "disconnect", no_argument, NULL, 'd' },
53568b96f15SEric Blake         { "list", no_argument, NULL, 'L' },
536aa6e546cSDaniel P. Berrange         { "snapshot", no_argument, NULL, 's' },
537aa6e546cSDaniel P. Berrange         { "load-snapshot", required_argument, NULL, 'l' },
538aa6e546cSDaniel P. Berrange         { "nocache", no_argument, NULL, 'n' },
539aa6e546cSDaniel P. Berrange         { "cache", required_argument, NULL, QEMU_NBD_OPT_CACHE },
540aa6e546cSDaniel P. Berrange         { "aio", required_argument, NULL, QEMU_NBD_OPT_AIO },
541aa6e546cSDaniel P. Berrange         { "discard", required_argument, NULL, QEMU_NBD_OPT_DISCARD },
542aa6e546cSDaniel P. Berrange         { "detect-zeroes", required_argument, NULL,
543aa6e546cSDaniel P. Berrange           QEMU_NBD_OPT_DETECT_ZEROES },
544aa6e546cSDaniel P. Berrange         { "shared", required_argument, NULL, 'e' },
545aa6e546cSDaniel P. Berrange         { "format", required_argument, NULL, 'f' },
546aa6e546cSDaniel P. Berrange         { "persistent", no_argument, NULL, 't' },
547aa6e546cSDaniel P. Berrange         { "verbose", no_argument, NULL, 'v' },
548aa6e546cSDaniel P. Berrange         { "object", required_argument, NULL, QEMU_NBD_OPT_OBJECT },
549aa6e546cSDaniel P. Berrange         { "export-name", required_argument, NULL, 'x' },
550b1a75b33SEric Blake         { "description", required_argument, NULL, 'D' },
551aa6e546cSDaniel P. Berrange         { "tls-creds", required_argument, NULL, QEMU_NBD_OPT_TLSCREDS },
552b25e12daSDaniel P. Berrange         { "tls-authz", required_argument, NULL, QEMU_NBD_OPT_TLSAUTHZ },
553aa6e546cSDaniel P. Berrange         { "image-opts", no_argument, NULL, QEMU_NBD_OPT_IMAGE_OPTS },
55439ca463eSDenis V. Lunev         { "trace", required_argument, NULL, 'T' },
555ffb31e1dSMax Reitz         { "fork", no_argument, NULL, QEMU_NBD_OPT_FORK },
556637bc5a5SMax Reitz         { "pid-file", required_argument, NULL, QEMU_NBD_OPT_PID_FILE },
557660f11beSBlue Swirl         { NULL, 0, NULL, 0 }
5587a5ca864Sbellard     };
5597a5ca864Sbellard     int ch;
5607a5ca864Sbellard     int opt_ind = 0;
561f5edb014SNaphtali Sprei     int flags = BDRV_O_RDWR;
5624fbec260SMax Reitz     int ret = 0;
56339a5235cSPaolo Bonzini     bool seen_cache = false;
564ded9d2d5SPaolo Bonzini     bool seen_discard = false;
56539a5235cSPaolo Bonzini     bool seen_aio = false;
566a517e88bSPaolo Bonzini     pthread_t client_thread;
567e6b63677SDaniel P. Berrange     const char *fmt = NULL;
56834b5d2c6SMax Reitz     Error *local_err = NULL;
569b3838a40SPeter Lieven     BlockdevDetectZeroesOptions detect_zeroes = BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF;
5704fbec260SMax Reitz     QDict *options = NULL;
57168b96f15SEric Blake     const char *export_name = NULL; /* defaults to "" later for server mode */
572b1a75b33SEric Blake     const char *export_description = NULL;
573636192c4SEric Blake     const char *bitmap = NULL;
574145614a1SDaniel P. Berrange     const char *tlscredsid = NULL;
57577c9aaefSDaniel P. Berrange     bool imageOpts = false;
5766effd5bfSKevin Wolf     bool writethrough = true;
57739ca463eSDenis V. Lunev     char *trace_file = NULL;
578ffb31e1dSMax Reitz     bool fork_process = false;
57968b96f15SEric Blake     bool list = false;
580ffb31e1dSMax Reitz     int old_stderr = -1;
581a721f53bSRichard W.M. Jones     unsigned socket_activation;
582637bc5a5SMax Reitz     const char *pid_file_name = NULL;
5837a5ca864Sbellard 
5846e64dd57SDaniel P. Berrangé #if HAVE_NBD_DEVICE
585a517e88bSPaolo Bonzini     /* The client thread uses SIGTERM to interrupt the server.  A signal
586a517e88bSPaolo Bonzini      * handler ensures that "qemu-nbd -v -c" exits with a nice status code.
587a517e88bSPaolo Bonzini      */
588bb345110SPaolo Bonzini     struct sigaction sa_sigterm;
589bb345110SPaolo Bonzini     memset(&sa_sigterm, 0, sizeof(sa_sigterm));
590bb345110SPaolo Bonzini     sa_sigterm.sa_handler = termsig_handler;
591bb345110SPaolo Bonzini     sigaction(SIGTERM, &sa_sigterm, NULL);
5926e64dd57SDaniel P. Berrangé #endif /* HAVE_NBD_DEVICE */
593c2297088SDaniel P. Berrange 
594041e32b8SMax Reitz #ifdef CONFIG_POSIX
595041e32b8SMax Reitz     signal(SIGPIPE, SIG_IGN);
596041e32b8SMax Reitz #endif
597041e32b8SMax Reitz 
59898c5d2e7SDaniel P. Berrangé     socket_init();
599f5852efaSChristophe Fergeau     error_init(argv[0]);
600fe4db84dSDaniel P. Berrange     module_call_init(MODULE_INIT_TRACE);
601e8f2d272SEduardo Habkost     qcrypto_init(&error_fatal);
602c2297088SDaniel P. Berrange 
6030ab3b337SDaniel P. Berrange     module_call_init(MODULE_INIT_QOM);
6040ab3b337SDaniel P. Berrange     qemu_add_opts(&qemu_object_opts);
60539ca463eSDenis V. Lunev     qemu_add_opts(&qemu_trace_opts);
60610f5bff6SFam Zheng     qemu_init_exec_dir(argv[0]);
607bb345110SPaolo Bonzini 
6087a5ca864Sbellard     while ((ch = getopt_long(argc, argv, sopt, lopt, &opt_ind)) != -1) {
6097a5ca864Sbellard         switch (ch) {
6107a5ca864Sbellard         case 's':
6112f726488Sths             flags |= BDRV_O_SNAPSHOT;
6122f726488Sths             break;
6132f726488Sths         case 'n':
61439a5235cSPaolo Bonzini             optarg = (char *) "none";
61539a5235cSPaolo Bonzini             /* fallthrough */
61639a5235cSPaolo Bonzini         case QEMU_NBD_OPT_CACHE:
61739a5235cSPaolo Bonzini             if (seen_cache) {
61885b01e09SMarkus Armbruster                 error_report("-n and --cache can only be specified once");
61985b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
62039a5235cSPaolo Bonzini             }
62139a5235cSPaolo Bonzini             seen_cache = true;
6226effd5bfSKevin Wolf             if (bdrv_parse_cache_mode(optarg, &flags, &writethrough) == -1) {
62385b01e09SMarkus Armbruster                 error_report("Invalid cache mode `%s'", optarg);
62485b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
62539a5235cSPaolo Bonzini             }
6267a5ca864Sbellard             break;
62739a5235cSPaolo Bonzini         case QEMU_NBD_OPT_AIO:
62839a5235cSPaolo Bonzini             if (seen_aio) {
62985b01e09SMarkus Armbruster                 error_report("--aio can only be specified once");
63085b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
63139a5235cSPaolo Bonzini             }
63239a5235cSPaolo Bonzini             seen_aio = true;
6337680274dSAarushi Mehta             if (bdrv_parse_aio(optarg, &flags) < 0) {
6347680274dSAarushi Mehta                 error_report("Invalid aio mode '%s'", optarg);
63585b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
63639a5235cSPaolo Bonzini             }
63739a5235cSPaolo Bonzini             break;
638ded9d2d5SPaolo Bonzini         case QEMU_NBD_OPT_DISCARD:
639ded9d2d5SPaolo Bonzini             if (seen_discard) {
64085b01e09SMarkus Armbruster                 error_report("--discard can only be specified once");
64185b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
642ded9d2d5SPaolo Bonzini             }
643ded9d2d5SPaolo Bonzini             seen_discard = true;
644ded9d2d5SPaolo Bonzini             if (bdrv_parse_discard_flags(optarg, &flags) == -1) {
64585b01e09SMarkus Armbruster                 error_report("Invalid discard mode `%s'", optarg);
64685b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
647ded9d2d5SPaolo Bonzini             }
648ded9d2d5SPaolo Bonzini             break;
649b3838a40SPeter Lieven         case QEMU_NBD_OPT_DETECT_ZEROES:
650b3838a40SPeter Lieven             detect_zeroes =
651f7abe0ecSMarc-André Lureau                 qapi_enum_parse(&BlockdevDetectZeroesOptions_lookup,
652b3838a40SPeter Lieven                                 optarg,
653b3838a40SPeter Lieven                                 BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF,
654b3838a40SPeter Lieven                                 &local_err);
655b3838a40SPeter Lieven             if (local_err) {
656c29b77f9SMarkus Armbruster                 error_reportf_err(local_err,
657c29b77f9SMarkus Armbruster                                   "Failed to parse detect_zeroes mode: ");
65885b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
659b3838a40SPeter Lieven             }
660b3838a40SPeter Lieven             if (detect_zeroes == BLOCKDEV_DETECT_ZEROES_OPTIONS_UNMAP &&
661b3838a40SPeter Lieven                 !(flags & BDRV_O_UNMAP)) {
66285b01e09SMarkus Armbruster                 error_report("setting detect-zeroes to unmap is not allowed "
663b3838a40SPeter Lieven                              "without setting discard operation to unmap");
66485b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
665b3838a40SPeter Lieven             }
666b3838a40SPeter Lieven             break;
6677a5ca864Sbellard         case 'b':
6687a5ca864Sbellard             bindto = optarg;
6697a5ca864Sbellard             break;
6707a5ca864Sbellard         case 'p':
67148bec07eSDaniel P. Berrange             port = optarg;
6727a5ca864Sbellard             break;
6737a5ca864Sbellard         case 'o':
67443b51011SEric Blake             if (qemu_strtou64(optarg, NULL, 0, &dev_offset) < 0) {
67543b51011SEric Blake                 error_report("Invalid offset '%s'", optarg);
67685b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
6777a5ca864Sbellard             }
6787a5ca864Sbellard             break;
6798c116b0eSWenchao Xia         case 'l':
6808c116b0eSWenchao Xia             if (strstart(optarg, SNAPSHOT_OPT_BASE, NULL)) {
68170b94331SMarkus Armbruster                 sn_opts = qemu_opts_parse_noisily(&internal_snapshot_opts,
68270b94331SMarkus Armbruster                                                   optarg, false);
6838c116b0eSWenchao Xia                 if (!sn_opts) {
68485b01e09SMarkus Armbruster                     error_report("Failed in parsing snapshot param `%s'",
6858c116b0eSWenchao Xia                                  optarg);
68685b01e09SMarkus Armbruster                     exit(EXIT_FAILURE);
6878c116b0eSWenchao Xia                 }
6888c116b0eSWenchao Xia             } else {
6898c116b0eSWenchao Xia                 sn_id_or_name = optarg;
6908c116b0eSWenchao Xia             }
6918c116b0eSWenchao Xia             /* fall through */
6927a5ca864Sbellard         case 'r':
693dbb38caaSEric Blake             readonly = true;
69407108b29SNaphtali Sprei             flags &= ~BDRV_O_RDWR;
6957a5ca864Sbellard             break;
696636192c4SEric Blake         case 'B':
697636192c4SEric Blake             bitmap = optarg;
698636192c4SEric Blake             break;
699cd831bd7Sths         case 'k':
700b32f6c28SPaolo Bonzini             sockpath = optarg;
701713cc671SPeter Lieven             if (sockpath[0] != '/') {
7029af9e0feSMarkus Armbruster                 error_report("socket path must be absolute");
70385b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
704713cc671SPeter Lieven             }
705cd831bd7Sths             break;
706cd831bd7Sths         case 'd':
707cd831bd7Sths             disconnect = true;
708cd831bd7Sths             break;
709cd831bd7Sths         case 'c':
710cd831bd7Sths             device = optarg;
711cd831bd7Sths             break;
7123b05a8e9Sths         case 'e':
71343b51011SEric Blake             if (qemu_strtoi(optarg, NULL, 0, &shared) < 0 ||
71443b51011SEric Blake                 shared < 1) {
71585b01e09SMarkus Armbruster                 error_report("Invalid shared device number '%s'", optarg);
71685b01e09SMarkus Armbruster                 exit(EXIT_FAILURE);
7173b05a8e9Sths             }
7183b05a8e9Sths             break;
719e6b63677SDaniel P. Berrange         case 'f':
720e6b63677SDaniel P. Berrange             fmt = optarg;
721e6b63677SDaniel P. Berrange             break;
72275818250Sths         case 't':
72375818250Sths             persistent = 1;
72475818250Sths             break;
7253d4b2f9cSDaniel P. Berrange         case 'x':
7263d4b2f9cSDaniel P. Berrange             export_name = optarg;
72793676c88SEric Blake             if (strlen(export_name) > NBD_MAX_STRING_SIZE) {
72893676c88SEric Blake                 error_report("export name '%s' too long", export_name);
72993676c88SEric Blake                 exit(EXIT_FAILURE);
73093676c88SEric Blake             }
7313d4b2f9cSDaniel P. Berrange             break;
732b1a75b33SEric Blake         case 'D':
733b1a75b33SEric Blake             export_description = optarg;
73493676c88SEric Blake             if (strlen(export_description) > NBD_MAX_STRING_SIZE) {
73593676c88SEric Blake                 error_report("export description '%s' too long",
73693676c88SEric Blake                              export_description);
73793676c88SEric Blake                 exit(EXIT_FAILURE);
73893676c88SEric Blake             }
739b1a75b33SEric Blake             break;
7407a5ca864Sbellard         case 'v':
7417a5ca864Sbellard             verbose = 1;
7427a5ca864Sbellard             break;
7437a5ca864Sbellard         case 'V':
7447a5ca864Sbellard             version(argv[0]);
7457a5ca864Sbellard             exit(0);
7467a5ca864Sbellard             break;
7477a5ca864Sbellard         case 'h':
7487a5ca864Sbellard             usage(argv[0]);
7497a5ca864Sbellard             exit(0);
7507a5ca864Sbellard             break;
7517a5ca864Sbellard         case '?':
75285b01e09SMarkus Armbruster             error_report("Try `%s --help' for more information.", argv[0]);
75385b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
7540ab3b337SDaniel P. Berrange         case QEMU_NBD_OPT_OBJECT: {
7550ab3b337SDaniel P. Berrange             QemuOpts *opts;
7560ab3b337SDaniel P. Berrange             opts = qemu_opts_parse_noisily(&qemu_object_opts,
7570ab3b337SDaniel P. Berrange                                            optarg, true);
7580ab3b337SDaniel P. Berrange             if (!opts) {
7590ab3b337SDaniel P. Berrange                 exit(EXIT_FAILURE);
7600ab3b337SDaniel P. Berrange             }
7610ab3b337SDaniel P. Berrange         }   break;
762145614a1SDaniel P. Berrange         case QEMU_NBD_OPT_TLSCREDS:
763145614a1SDaniel P. Berrange             tlscredsid = optarg;
764145614a1SDaniel P. Berrange             break;
76577c9aaefSDaniel P. Berrange         case QEMU_NBD_OPT_IMAGE_OPTS:
76677c9aaefSDaniel P. Berrange             imageOpts = true;
76777c9aaefSDaniel P. Berrange             break;
76839ca463eSDenis V. Lunev         case 'T':
76939ca463eSDenis V. Lunev             g_free(trace_file);
77039ca463eSDenis V. Lunev             trace_file = 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 ||
79368b96f15SEric Blake             device || disconnect || fmt || sn_id_or_name || bitmap ||
79468b96f15SEric Blake             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     }
81739ca463eSDenis V. Lunev     trace_init_file(trace_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 
1062b57e4de0SKevin Wolf     export = nbd_export_new(bs, export_name,
1063dbb38caaSEric Blake                             export_description, bitmap, readonly, shared > 1,
1064*d794f7f3SKevin Wolf                             writethrough, &error_fatal);
10653b05a8e9Sths 
1066f1ef5555SPaolo Bonzini     if (device) {
10673c1fa35dSEric Blake #if HAVE_NBD_DEVICE
1068f1ef5555SPaolo Bonzini         int ret;
1069f1ef5555SPaolo Bonzini 
1070a6ac2313SPaolo Bonzini         ret = pthread_create(&client_thread, NULL, nbd_client_thread, device);
1071f1ef5555SPaolo Bonzini         if (ret != 0) {
107285b01e09SMarkus Armbruster             error_report("Failed to create client thread: %s", strerror(ret));
107385b01e09SMarkus Armbruster             exit(EXIT_FAILURE);
1074f1ef5555SPaolo Bonzini         }
10753c1fa35dSEric Blake #endif
1076f1ef5555SPaolo Bonzini     } else {
1077f1ef5555SPaolo Bonzini         /* Shut up GCC warnings.  */
1078f1ef5555SPaolo Bonzini         memset(&client_thread, 0, sizeof(client_thread));
1079f1ef5555SPaolo Bonzini     }
1080f1ef5555SPaolo Bonzini 
1081d0d6ff58SDaniel P. Berrange     nbd_update_server_watch();
10827a5ca864Sbellard 
1083637bc5a5SMax Reitz     if (pid_file_name) {
1084637bc5a5SMax Reitz         qemu_write_pidfile(pid_file_name, &error_fatal);
1085637bc5a5SMax Reitz     }
1086637bc5a5SMax Reitz 
10879faf31b6SMichael Tokarev     /* now when the initialization is (almost) complete, chdir("/")
10889faf31b6SMichael Tokarev      * to free any busy filesystems */
10899faf31b6SMichael Tokarev     if (chdir("/") < 0) {
109085b01e09SMarkus Armbruster         error_report("Could not chdir to root directory: %s",
109185b01e09SMarkus Armbruster                      strerror(errno));
109285b01e09SMarkus Armbruster         exit(EXIT_FAILURE);
10939faf31b6SMichael Tokarev     }
10949faf31b6SMichael Tokarev 
1095ffb31e1dSMax Reitz     if (fork_process) {
1096ffb31e1dSMax Reitz         dup2(old_stderr, STDERR_FILENO);
1097ffb31e1dSMax Reitz         close(old_stderr);
1098ffb31e1dSMax Reitz     }
1099ffb31e1dSMax Reitz 
11007860a380SPaolo Bonzini     state = RUNNING;
11013b05a8e9Sths     do {
1102a61c6782SPaolo Bonzini         main_loop_wait(false);
11037860a380SPaolo Bonzini         if (state == TERMINATE) {
11049d976580SPhilippe Mathieu-Daudé             nbd_export_put(export);
1105*d794f7f3SKevin Wolf             nbd_export_close_all();
11069d976580SPhilippe Mathieu-Daudé             export = NULL;
1107*d794f7f3SKevin Wolf             state = TERMINATED;
11087860a380SPaolo Bonzini         }
11097860a380SPaolo Bonzini     } while (state != TERMINATED);
11102c8d9f06SPaolo Bonzini 
111126f54e9aSMarkus Armbruster     blk_unref(blk);
1112b32f6c28SPaolo Bonzini     if (sockpath) {
1113b32f6c28SPaolo Bonzini         unlink(sockpath);
1114b32f6c28SPaolo Bonzini     }
11157a5ca864Sbellard 
11168c116b0eSWenchao Xia     qemu_opts_del(sn_opts);
11178c116b0eSWenchao Xia 
1118a517e88bSPaolo Bonzini     if (device) {
1119a517e88bSPaolo Bonzini         void *ret;
1120a517e88bSPaolo Bonzini         pthread_join(client_thread, &ret);
1121a517e88bSPaolo Bonzini         exit(ret != NULL);
1122a517e88bSPaolo Bonzini     } else {
1123a517e88bSPaolo Bonzini         exit(EXIT_SUCCESS);
1124a517e88bSPaolo Bonzini     }
11257a5ca864Sbellard }
1126