Commit | Line | Data |
---|---|---|
cd831bd7 | 1 | /* |
7a5ca864 FB |
2 | * Copyright (C) 2005 Anthony Liguori <anthony@codemonkey.ws> |
3 | * | |
4 | * Network Block Device | |
5 | * | |
6 | * This program is free software; you can redistribute it and/or modify | |
7 | * it under the terms of the GNU General Public License as published by | |
8 | * the Free Software Foundation; under version 2 of the License. | |
9 | * | |
10 | * This program is distributed in the hope that it will be useful, | |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
13 | * GNU General Public License for more details. | |
14 | * | |
15 | * You should have received a copy of the GNU General Public License | |
8167ee88 | 16 | * along with this program; if not, see <http://www.gnu.org/licenses/>. |
7a5ca864 FB |
17 | */ |
18 | ||
5a61cb60 | 19 | #include "qemu-common.h" |
26f54e9a | 20 | #include "sysemu/block-backend.h" |
b3838a40 | 21 | #include "block/block_int.h" |
737e150e | 22 | #include "block/nbd.h" |
6a1751b7 | 23 | #include "qemu/main-loop.h" |
537b41f5 PB |
24 | #include "qemu/sockets.h" |
25 | #include "qemu/error-report.h" | |
8c116b0e | 26 | #include "block/snapshot.h" |
b3838a40 | 27 | #include "qapi/util.h" |
d49b6836 | 28 | #include "qapi/qmp/qstring.h" |
7a5ca864 | 29 | |
7a5ca864 FB |
30 | #include <stdarg.h> |
31 | #include <stdio.h> | |
32 | #include <getopt.h> | |
cd831bd7 | 33 | #include <sys/types.h> |
7a5ca864 FB |
34 | #include <sys/socket.h> |
35 | #include <netinet/in.h> | |
36 | #include <netinet/tcp.h> | |
37 | #include <arpa/inet.h> | |
cd831bd7 | 38 | #include <signal.h> |
2bff4b6f | 39 | #include <libgen.h> |
a517e88b | 40 | #include <pthread.h> |
cd831bd7 | 41 | |
713cc671 PL |
42 | #define SOCKET_PATH "/var/lock/qemu-nbd-%s" |
43 | #define QEMU_NBD_OPT_CACHE 1 | |
44 | #define QEMU_NBD_OPT_AIO 2 | |
45 | #define QEMU_NBD_OPT_DISCARD 3 | |
b3838a40 | 46 | #define QEMU_NBD_OPT_DETECT_ZEROES 4 |
7a5ca864 | 47 | |
af49bbbe | 48 | static NBDExport *exp; |
b1d8e52e | 49 | static int verbose; |
a517e88b | 50 | static char *srcpath; |
48bec07e | 51 | static SocketAddress *saddr; |
7860a380 PB |
52 | static int persistent = 0; |
53 | static enum { RUNNING, TERMINATE, TERMINATING, TERMINATED } state; | |
a61c6782 PB |
54 | static int shared = 1; |
55 | static int nb_fds; | |
e4afbf4f | 56 | static int server_fd; |
7a5ca864 FB |
57 | |
58 | static void usage(const char *name) | |
59 | { | |
b033cd86 | 60 | (printf) ( |
7a5ca864 FB |
61 | "Usage: %s [OPTIONS] FILE\n" |
62 | "QEMU Disk Network Block Device Server\n" | |
63 | "\n" | |
713cc671 PL |
64 | " -h, --help display this help and exit\n" |
65 | " -V, --version output version information and exit\n" | |
b033cd86 PB |
66 | "\n" |
67 | "Connection properties:\n" | |
713cc671 PL |
68 | " -p, --port=PORT port to listen on (default `%d')\n" |
69 | " -b, --bind=IFACE interface to bind to (default `0.0.0.0')\n" | |
70 | " -k, --socket=PATH path to the unix socket\n" | |
71 | " (default '"SOCKET_PATH"')\n" | |
72 | " -e, --shared=NUM device can be shared by NUM clients (default '1')\n" | |
73 | " -t, --persistent don't exit on the last connection\n" | |
74 | " -v, --verbose display extra debugging information\n" | |
7a5ca864 | 75 | "\n" |
b033cd86 | 76 | "Exposing part of the image:\n" |
713cc671 PL |
77 | " -o, --offset=OFFSET offset into the image\n" |
78 | " -P, --partition=NUM only expose partition NUM\n" | |
b033cd86 PB |
79 | "\n" |
80 | #ifdef __linux__ | |
81 | "Kernel NBD client support:\n" | |
713cc671 PL |
82 | " -c, --connect=DEV connect FILE to the local NBD device DEV\n" |
83 | " -d, --disconnect disconnect the specified device\n" | |
b033cd86 PB |
84 | "\n" |
85 | #endif | |
86 | "\n" | |
87 | "Block device options:\n" | |
713cc671 PL |
88 | " -f, --format=FORMAT set image format (raw, qcow2, ...)\n" |
89 | " -r, --read-only export read-only\n" | |
90 | " -s, --snapshot use FILE as an external snapshot, create a temporary\n" | |
91 | " file with backing_file=FILE, redirect the write to\n" | |
92 | " the temporary one\n" | |
8c116b0e | 93 | " -l, --load-snapshot=SNAPSHOT_PARAM\n" |
713cc671 PL |
94 | " load an internal snapshot inside FILE and export it\n" |
95 | " as an read-only device, SNAPSHOT_PARAM format is\n" | |
96 | " 'snapshot.id=[ID],snapshot.name=[NAME]', or\n" | |
97 | " '[ID_OR_NAME]'\n" | |
98 | " -n, --nocache disable host cache\n" | |
99 | " --cache=MODE set cache mode (none, writeback, ...)\n" | |
713cc671 | 100 | " --aio=MODE set AIO mode (native or threads)\n" |
b3838a40 | 101 | " --discard=MODE set discard mode (ignore, unmap)\n" |
6883de6c | 102 | " --detect-zeroes=MODE set detect-zeroes mode (off, on, unmap)\n" |
b033cd86 PB |
103 | "\n" |
104 | "Report bugs to <qemu-devel@nongnu.org>\n" | |
c2e2872b | 105 | , name, NBD_DEFAULT_PORT, "DEVICE"); |
7a5ca864 FB |
106 | } |
107 | ||
108 | static void version(const char *name) | |
109 | { | |
110 | printf( | |
315bc7aa | 111 | "%s version 0.0.1\n" |
7a5ca864 FB |
112 | "Written by Anthony Liguori.\n" |
113 | "\n" | |
114 | "Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>.\n" | |
115 | "This is free software; see the source for copying conditions. There is NO\n" | |
116 | "warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n" | |
315bc7aa | 117 | , name); |
7a5ca864 FB |
118 | } |
119 | ||
120 | struct partition_record | |
121 | { | |
122 | uint8_t bootable; | |
123 | uint8_t start_head; | |
124 | uint32_t start_cylinder; | |
125 | uint8_t start_sector; | |
126 | uint8_t system; | |
127 | uint8_t end_head; | |
128 | uint8_t end_cylinder; | |
129 | uint8_t end_sector; | |
130 | uint32_t start_sector_abs; | |
131 | uint32_t nb_sectors_abs; | |
132 | }; | |
133 | ||
134 | static void read_partition(uint8_t *p, struct partition_record *r) | |
135 | { | |
136 | r->bootable = p[0]; | |
137 | r->start_head = p[1]; | |
138 | r->start_cylinder = p[3] | ((p[2] << 2) & 0x0300); | |
139 | r->start_sector = p[2] & 0x3f; | |
140 | r->system = p[4]; | |
141 | r->end_head = p[5]; | |
142 | r->end_cylinder = p[7] | ((p[6] << 2) & 0x300); | |
143 | r->end_sector = p[6] & 0x3f; | |
ac97393d HR |
144 | |
145 | r->start_sector_abs = le32_to_cpup((uint32_t *)(p + 8)); | |
146 | r->nb_sectors_abs = le32_to_cpup((uint32_t *)(p + 12)); | |
7a5ca864 FB |
147 | } |
148 | ||
4c58e80a | 149 | static int find_partition(BlockBackend *blk, int partition, |
7a5ca864 FB |
150 | off_t *offset, off_t *size) |
151 | { | |
152 | struct partition_record mbr[4]; | |
153 | uint8_t data[512]; | |
154 | int i; | |
155 | int ext_partnum = 4; | |
cb7cf0e3 | 156 | int ret; |
7a5ca864 | 157 | |
4c58e80a | 158 | if ((ret = blk_read(blk, 0, data, 1)) < 0) { |
cb7cf0e3 | 159 | errno = -ret; |
85b01e09 MA |
160 | error_report("error while reading: %s", strerror(errno)); |
161 | exit(EXIT_FAILURE); | |
cb7cf0e3 | 162 | } |
7a5ca864 FB |
163 | |
164 | if (data[510] != 0x55 || data[511] != 0xaa) { | |
185b4338 | 165 | return -EINVAL; |
7a5ca864 FB |
166 | } |
167 | ||
168 | for (i = 0; i < 4; i++) { | |
169 | read_partition(&data[446 + 16 * i], &mbr[i]); | |
170 | ||
453b07b1 | 171 | if (!mbr[i].system || !mbr[i].nb_sectors_abs) { |
7a5ca864 | 172 | continue; |
453b07b1 | 173 | } |
7a5ca864 FB |
174 | |
175 | if (mbr[i].system == 0xF || mbr[i].system == 0x5) { | |
176 | struct partition_record ext[4]; | |
177 | uint8_t data1[512]; | |
178 | int j; | |
179 | ||
4c58e80a | 180 | if ((ret = blk_read(blk, mbr[i].start_sector_abs, data1, 1)) < 0) { |
cb7cf0e3 | 181 | errno = -ret; |
85b01e09 MA |
182 | error_report("error while reading: %s", strerror(errno)); |
183 | exit(EXIT_FAILURE); | |
cb7cf0e3 | 184 | } |
7a5ca864 FB |
185 | |
186 | for (j = 0; j < 4; j++) { | |
187 | read_partition(&data1[446 + 16 * j], &ext[j]); | |
453b07b1 | 188 | if (!ext[j].system || !ext[j].nb_sectors_abs) { |
7a5ca864 | 189 | continue; |
453b07b1 | 190 | } |
7a5ca864 FB |
191 | |
192 | if ((ext_partnum + j + 1) == partition) { | |
193 | *offset = (uint64_t)ext[j].start_sector_abs << 9; | |
194 | *size = (uint64_t)ext[j].nb_sectors_abs << 9; | |
195 | return 0; | |
196 | } | |
197 | } | |
198 | ext_partnum += 4; | |
199 | } else if ((i + 1) == partition) { | |
200 | *offset = (uint64_t)mbr[i].start_sector_abs << 9; | |
201 | *size = (uint64_t)mbr[i].nb_sectors_abs << 9; | |
202 | return 0; | |
203 | } | |
204 | } | |
205 | ||
185b4338 | 206 | return -ENOENT; |
7a5ca864 FB |
207 | } |
208 | ||
bb345110 PB |
209 | static void termsig_handler(int signum) |
210 | { | |
7860a380 | 211 | state = TERMINATE; |
a61c6782 | 212 | qemu_notify_event(); |
bb345110 PB |
213 | } |
214 | ||
537b41f5 | 215 | |
a517e88b | 216 | static void *show_parts(void *arg) |
cd831bd7 | 217 | { |
a6ac2313 | 218 | char *device = arg; |
a517e88b PB |
219 | int nbd; |
220 | ||
221 | /* linux just needs an open() to trigger | |
222 | * the partition table update | |
223 | * but remember to load the module with max_part != 0 : | |
224 | * modprobe nbd max_part=63 | |
225 | */ | |
226 | nbd = open(device, O_RDWR); | |
fc19f8a0 | 227 | if (nbd >= 0) { |
a517e88b PB |
228 | close(nbd); |
229 | } | |
230 | return NULL; | |
231 | } | |
cd831bd7 | 232 | |
a517e88b PB |
233 | static void *nbd_client_thread(void *arg) |
234 | { | |
a6ac2313 | 235 | char *device = arg; |
a517e88b | 236 | off_t size; |
a517e88b | 237 | uint32_t nbdflags; |
a6ac2313 | 238 | int fd, sock; |
a517e88b PB |
239 | int ret; |
240 | pthread_t show_parts_thread; | |
1ce52846 | 241 | Error *local_error = NULL; |
a517e88b | 242 | |
48bec07e DB |
243 | |
244 | sock = socket_connect(saddr, &local_error, NULL, NULL); | |
fc19f8a0 | 245 | if (sock < 0) { |
48bec07e | 246 | error_report_err(local_error); |
dc10e8b3 SH |
247 | goto out; |
248 | } | |
a517e88b PB |
249 | |
250 | ret = nbd_receive_negotiate(sock, NULL, &nbdflags, | |
3f472659 | 251 | &size, &local_error); |
fc19f8a0 | 252 | if (ret < 0) { |
1ce52846 HR |
253 | if (local_error) { |
254 | fprintf(stderr, "%s\n", error_get_pretty(local_error)); | |
255 | error_free(local_error); | |
256 | } | |
0c544d73 | 257 | goto out_socket; |
a517e88b PB |
258 | } |
259 | ||
a6ac2313 | 260 | fd = open(device, O_RDWR); |
fc19f8a0 | 261 | if (fd < 0) { |
a6ac2313 | 262 | /* Linux-only, we can use %m in printf. */ |
5672ee54 | 263 | fprintf(stderr, "Failed to open %s: %m\n", device); |
0c544d73 | 264 | goto out_socket; |
a6ac2313 PB |
265 | } |
266 | ||
3f472659 | 267 | ret = nbd_init(fd, sock, nbdflags, size); |
fc19f8a0 | 268 | if (ret < 0) { |
0c544d73 | 269 | goto out_fd; |
a517e88b PB |
270 | } |
271 | ||
272 | /* update partition table */ | |
a6ac2313 | 273 | pthread_create(&show_parts_thread, NULL, show_parts, device); |
a517e88b | 274 | |
c1f8fdc3 PB |
275 | if (verbose) { |
276 | fprintf(stderr, "NBD device %s is now connected to %s\n", | |
277 | device, srcpath); | |
278 | } else { | |
279 | /* Close stderr so that the qemu-nbd process exits. */ | |
280 | dup2(STDOUT_FILENO, STDERR_FILENO); | |
281 | } | |
a517e88b PB |
282 | |
283 | ret = nbd_client(fd); | |
284 | if (ret) { | |
0c544d73 | 285 | goto out_fd; |
cd831bd7 | 286 | } |
a517e88b PB |
287 | close(fd); |
288 | kill(getpid(), SIGTERM); | |
289 | return (void *) EXIT_SUCCESS; | |
290 | ||
0c544d73 PB |
291 | out_fd: |
292 | close(fd); | |
293 | out_socket: | |
294 | closesocket(sock); | |
a517e88b PB |
295 | out: |
296 | kill(getpid(), SIGTERM); | |
297 | return (void *) EXIT_FAILURE; | |
cd831bd7 TS |
298 | } |
299 | ||
e4afbf4f | 300 | static int nbd_can_accept(void) |
a61c6782 PB |
301 | { |
302 | return nb_fds < shared; | |
303 | } | |
304 | ||
7860a380 PB |
305 | static void nbd_export_closed(NBDExport *exp) |
306 | { | |
307 | assert(state == TERMINATING); | |
308 | state = TERMINATED; | |
309 | } | |
310 | ||
e4afbf4f FZ |
311 | static void nbd_update_server_fd_handler(int fd); |
312 | ||
1743b515 | 313 | static void nbd_client_closed(NBDClient *client) |
a61c6782 | 314 | { |
1743b515 | 315 | nb_fds--; |
7860a380 PB |
316 | if (nb_fds == 0 && !persistent && state == RUNNING) { |
317 | state = TERMINATE; | |
318 | } | |
e4afbf4f | 319 | nbd_update_server_fd_handler(server_fd); |
7860a380 | 320 | nbd_client_put(client); |
a61c6782 PB |
321 | } |
322 | ||
323 | static void nbd_accept(void *opaque) | |
324 | { | |
a61c6782 PB |
325 | struct sockaddr_in addr; |
326 | socklen_t addr_len = sizeof(addr); | |
327 | ||
328 | int fd = accept(server_fd, (struct sockaddr *)&addr, &addr_len); | |
0c544d73 PB |
329 | if (fd < 0) { |
330 | perror("accept"); | |
331 | return; | |
332 | } | |
333 | ||
7860a380 PB |
334 | if (state >= TERMINATE) { |
335 | close(fd); | |
336 | return; | |
337 | } | |
338 | ||
36af5994 | 339 | if (nbd_client_new(exp, fd, nbd_client_closed)) { |
a61c6782 | 340 | nb_fds++; |
e4afbf4f | 341 | nbd_update_server_fd_handler(server_fd); |
36af5994 | 342 | } else { |
27e5eae4 | 343 | shutdown(fd, 2); |
36af5994 | 344 | close(fd); |
a61c6782 PB |
345 | } |
346 | } | |
347 | ||
e4afbf4f FZ |
348 | static void nbd_update_server_fd_handler(int fd) |
349 | { | |
350 | if (nbd_can_accept()) { | |
351 | qemu_set_fd_handler(fd, nbd_accept, NULL, (void *)(uintptr_t)fd); | |
352 | } else { | |
353 | qemu_set_fd_handler(fd, NULL, NULL, NULL); | |
354 | } | |
355 | } | |
356 | ||
48bec07e DB |
357 | |
358 | static SocketAddress *nbd_build_socket_address(const char *sockpath, | |
359 | const char *bindto, | |
360 | const char *port) | |
361 | { | |
362 | SocketAddress *saddr; | |
363 | ||
364 | saddr = g_new0(SocketAddress, 1); | |
365 | if (sockpath) { | |
2d32adda EB |
366 | saddr->type = SOCKET_ADDRESS_KIND_UNIX; |
367 | saddr->u.q_unix = g_new0(UnixSocketAddress, 1); | |
368 | saddr->u.q_unix->path = g_strdup(sockpath); | |
48bec07e | 369 | } else { |
2d32adda EB |
370 | saddr->type = SOCKET_ADDRESS_KIND_INET; |
371 | saddr->u.inet = g_new0(InetSocketAddress, 1); | |
372 | saddr->u.inet->host = g_strdup(bindto); | |
48bec07e | 373 | if (port) { |
2d32adda | 374 | saddr->u.inet->port = g_strdup(port); |
48bec07e | 375 | } else { |
2d32adda | 376 | saddr->u.inet->port = g_strdup_printf("%d", NBD_DEFAULT_PORT); |
48bec07e DB |
377 | } |
378 | } | |
379 | ||
380 | return saddr; | |
381 | } | |
382 | ||
383 | ||
7a5ca864 FB |
384 | int main(int argc, char **argv) |
385 | { | |
26f54e9a | 386 | BlockBackend *blk; |
7a5ca864 FB |
387 | BlockDriverState *bs; |
388 | off_t dev_offset = 0; | |
b90fb4b8 | 389 | uint32_t nbdflags = 0; |
cd831bd7 | 390 | bool disconnect = false; |
7a5ca864 | 391 | const char *bindto = "0.0.0.0"; |
48bec07e DB |
392 | const char *port = NULL; |
393 | char *sockpath = NULL; | |
a6ac2313 | 394 | char *device = NULL; |
7a5ca864 | 395 | off_t fd_size; |
8c116b0e WX |
396 | QemuOpts *sn_opts = NULL; |
397 | const char *sn_id_or_name = NULL; | |
398 | const char *sopt = "hVb:o:p:rsnP:c:dvk:e:f:tl:"; | |
7a5ca864 | 399 | struct option lopt[] = { |
660f11be BS |
400 | { "help", 0, NULL, 'h' }, |
401 | { "version", 0, NULL, 'V' }, | |
402 | { "bind", 1, NULL, 'b' }, | |
403 | { "port", 1, NULL, 'p' }, | |
404 | { "socket", 1, NULL, 'k' }, | |
405 | { "offset", 1, NULL, 'o' }, | |
406 | { "read-only", 0, NULL, 'r' }, | |
407 | { "partition", 1, NULL, 'P' }, | |
408 | { "connect", 1, NULL, 'c' }, | |
409 | { "disconnect", 0, NULL, 'd' }, | |
410 | { "snapshot", 0, NULL, 's' }, | |
8c116b0e | 411 | { "load-snapshot", 1, NULL, 'l' }, |
660f11be | 412 | { "nocache", 0, NULL, 'n' }, |
39a5235c | 413 | { "cache", 1, NULL, QEMU_NBD_OPT_CACHE }, |
39a5235c | 414 | { "aio", 1, NULL, QEMU_NBD_OPT_AIO }, |
ded9d2d5 | 415 | { "discard", 1, NULL, QEMU_NBD_OPT_DISCARD }, |
b3838a40 | 416 | { "detect-zeroes", 1, NULL, QEMU_NBD_OPT_DETECT_ZEROES }, |
660f11be | 417 | { "shared", 1, NULL, 'e' }, |
e6b63677 | 418 | { "format", 1, NULL, 'f' }, |
660f11be BS |
419 | { "persistent", 0, NULL, 't' }, |
420 | { "verbose", 0, NULL, 'v' }, | |
421 | { NULL, 0, NULL, 0 } | |
7a5ca864 FB |
422 | }; |
423 | int ch; | |
424 | int opt_ind = 0; | |
7a5ca864 | 425 | char *end; |
f5edb014 | 426 | int flags = BDRV_O_RDWR; |
7a5ca864 | 427 | int partition = -1; |
4fbec260 | 428 | int ret = 0; |
3b05a8e9 | 429 | int fd; |
39a5235c | 430 | bool seen_cache = false; |
ded9d2d5 | 431 | bool seen_discard = false; |
39a5235c | 432 | bool seen_aio = false; |
a517e88b | 433 | pthread_t client_thread; |
e6b63677 | 434 | const char *fmt = NULL; |
34b5d2c6 | 435 | Error *local_err = NULL; |
b3838a40 | 436 | BlockdevDetectZeroesOptions detect_zeroes = BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF; |
4fbec260 | 437 | QDict *options = NULL; |
7a5ca864 | 438 | |
a517e88b PB |
439 | /* The client thread uses SIGTERM to interrupt the server. A signal |
440 | * handler ensures that "qemu-nbd -v -c" exits with a nice status code. | |
441 | */ | |
bb345110 | 442 | struct sigaction sa_sigterm; |
bb345110 PB |
443 | memset(&sa_sigterm, 0, sizeof(sa_sigterm)); |
444 | sa_sigterm.sa_handler = termsig_handler; | |
445 | sigaction(SIGTERM, &sa_sigterm, NULL); | |
10f5bff6 | 446 | qemu_init_exec_dir(argv[0]); |
bb345110 | 447 | |
7a5ca864 FB |
448 | while ((ch = getopt_long(argc, argv, sopt, lopt, &opt_ind)) != -1) { |
449 | switch (ch) { | |
450 | case 's': | |
2f726488 TS |
451 | flags |= BDRV_O_SNAPSHOT; |
452 | break; | |
453 | case 'n': | |
39a5235c PB |
454 | optarg = (char *) "none"; |
455 | /* fallthrough */ | |
456 | case QEMU_NBD_OPT_CACHE: | |
457 | if (seen_cache) { | |
85b01e09 MA |
458 | error_report("-n and --cache can only be specified once"); |
459 | exit(EXIT_FAILURE); | |
39a5235c PB |
460 | } |
461 | seen_cache = true; | |
462 | if (bdrv_parse_cache_flags(optarg, &flags) == -1) { | |
85b01e09 MA |
463 | error_report("Invalid cache mode `%s'", optarg); |
464 | exit(EXIT_FAILURE); | |
39a5235c | 465 | } |
7a5ca864 | 466 | break; |
39a5235c PB |
467 | case QEMU_NBD_OPT_AIO: |
468 | if (seen_aio) { | |
85b01e09 MA |
469 | error_report("--aio can only be specified once"); |
470 | exit(EXIT_FAILURE); | |
39a5235c PB |
471 | } |
472 | seen_aio = true; | |
473 | if (!strcmp(optarg, "native")) { | |
474 | flags |= BDRV_O_NATIVE_AIO; | |
475 | } else if (!strcmp(optarg, "threads")) { | |
476 | /* this is the default */ | |
477 | } else { | |
85b01e09 MA |
478 | error_report("invalid aio mode `%s'", optarg); |
479 | exit(EXIT_FAILURE); | |
39a5235c PB |
480 | } |
481 | break; | |
ded9d2d5 PB |
482 | case QEMU_NBD_OPT_DISCARD: |
483 | if (seen_discard) { | |
85b01e09 MA |
484 | error_report("--discard can only be specified once"); |
485 | exit(EXIT_FAILURE); | |
ded9d2d5 PB |
486 | } |
487 | seen_discard = true; | |
488 | if (bdrv_parse_discard_flags(optarg, &flags) == -1) { | |
85b01e09 MA |
489 | error_report("Invalid discard mode `%s'", optarg); |
490 | exit(EXIT_FAILURE); | |
ded9d2d5 PB |
491 | } |
492 | break; | |
b3838a40 PL |
493 | case QEMU_NBD_OPT_DETECT_ZEROES: |
494 | detect_zeroes = | |
495 | qapi_enum_parse(BlockdevDetectZeroesOptions_lookup, | |
496 | optarg, | |
7fb1cf16 | 497 | BLOCKDEV_DETECT_ZEROES_OPTIONS__MAX, |
b3838a40 PL |
498 | BLOCKDEV_DETECT_ZEROES_OPTIONS_OFF, |
499 | &local_err); | |
500 | if (local_err) { | |
85b01e09 MA |
501 | error_report("Failed to parse detect_zeroes mode: %s", |
502 | error_get_pretty(local_err)); | |
503 | exit(EXIT_FAILURE); | |
b3838a40 PL |
504 | } |
505 | if (detect_zeroes == BLOCKDEV_DETECT_ZEROES_OPTIONS_UNMAP && | |
506 | !(flags & BDRV_O_UNMAP)) { | |
85b01e09 MA |
507 | error_report("setting detect-zeroes to unmap is not allowed " |
508 | "without setting discard operation to unmap"); | |
509 | exit(EXIT_FAILURE); | |
b3838a40 PL |
510 | } |
511 | break; | |
7a5ca864 FB |
512 | case 'b': |
513 | bindto = optarg; | |
514 | break; | |
515 | case 'p': | |
48bec07e | 516 | port = optarg; |
7a5ca864 FB |
517 | break; |
518 | case 'o': | |
519 | dev_offset = strtoll (optarg, &end, 0); | |
520 | if (*end) { | |
85b01e09 MA |
521 | error_report("Invalid offset `%s'", optarg); |
522 | exit(EXIT_FAILURE); | |
7a5ca864 FB |
523 | } |
524 | if (dev_offset < 0) { | |
85b01e09 MA |
525 | error_report("Offset must be positive `%s'", optarg); |
526 | exit(EXIT_FAILURE); | |
7a5ca864 FB |
527 | } |
528 | break; | |
8c116b0e WX |
529 | case 'l': |
530 | if (strstart(optarg, SNAPSHOT_OPT_BASE, NULL)) { | |
70b94331 MA |
531 | sn_opts = qemu_opts_parse_noisily(&internal_snapshot_opts, |
532 | optarg, false); | |
8c116b0e | 533 | if (!sn_opts) { |
85b01e09 MA |
534 | error_report("Failed in parsing snapshot param `%s'", |
535 | optarg); | |
536 | exit(EXIT_FAILURE); | |
8c116b0e WX |
537 | } |
538 | } else { | |
539 | sn_id_or_name = optarg; | |
540 | } | |
541 | /* fall through */ | |
7a5ca864 | 542 | case 'r': |
b90fb4b8 | 543 | nbdflags |= NBD_FLAG_READ_ONLY; |
07108b29 | 544 | flags &= ~BDRV_O_RDWR; |
7a5ca864 FB |
545 | break; |
546 | case 'P': | |
547 | partition = strtol(optarg, &end, 0); | |
713cc671 | 548 | if (*end) { |
85b01e09 MA |
549 | error_report("Invalid partition `%s'", optarg); |
550 | exit(EXIT_FAILURE); | |
713cc671 PL |
551 | } |
552 | if (partition < 1 || partition > 8) { | |
85b01e09 MA |
553 | error_report("Invalid partition %d", partition); |
554 | exit(EXIT_FAILURE); | |
713cc671 | 555 | } |
7a5ca864 | 556 | break; |
cd831bd7 | 557 | case 'k': |
b32f6c28 | 558 | sockpath = optarg; |
713cc671 | 559 | if (sockpath[0] != '/') { |
85b01e09 MA |
560 | error_report("socket path must be absolute\n"); |
561 | exit(EXIT_FAILURE); | |
713cc671 | 562 | } |
cd831bd7 TS |
563 | break; |
564 | case 'd': | |
565 | disconnect = true; | |
566 | break; | |
567 | case 'c': | |
568 | device = optarg; | |
569 | break; | |
3b05a8e9 TS |
570 | case 'e': |
571 | shared = strtol(optarg, &end, 0); | |
572 | if (*end) { | |
85b01e09 MA |
573 | error_report("Invalid shared device number '%s'", optarg); |
574 | exit(EXIT_FAILURE); | |
3b05a8e9 TS |
575 | } |
576 | if (shared < 1) { | |
85b01e09 MA |
577 | error_report("Shared device number must be greater than 0\n"); |
578 | exit(EXIT_FAILURE); | |
3b05a8e9 TS |
579 | } |
580 | break; | |
e6b63677 DB |
581 | case 'f': |
582 | fmt = optarg; | |
583 | break; | |
713cc671 PL |
584 | case 't': |
585 | persistent = 1; | |
586 | break; | |
7a5ca864 FB |
587 | case 'v': |
588 | verbose = 1; | |
589 | break; | |
590 | case 'V': | |
591 | version(argv[0]); | |
592 | exit(0); | |
593 | break; | |
594 | case 'h': | |
595 | usage(argv[0]); | |
596 | exit(0); | |
597 | break; | |
598 | case '?': | |
85b01e09 MA |
599 | error_report("Try `%s --help' for more information.", argv[0]); |
600 | exit(EXIT_FAILURE); | |
7a5ca864 FB |
601 | } |
602 | } | |
603 | ||
604 | if ((argc - optind) != 1) { | |
85b01e09 MA |
605 | error_report("Invalid number of argument.\n" |
606 | "Try `%s --help' for more information.", | |
607 | argv[0]); | |
608 | exit(EXIT_FAILURE); | |
7a5ca864 FB |
609 | } |
610 | ||
cd831bd7 TS |
611 | if (disconnect) { |
612 | fd = open(argv[optind], O_RDWR); | |
fc19f8a0 | 613 | if (fd < 0) { |
85b01e09 MA |
614 | error_report("Cannot open %s: %s", argv[optind], |
615 | strerror(errno)); | |
616 | exit(EXIT_FAILURE); | |
fc19f8a0 | 617 | } |
cd831bd7 TS |
618 | nbd_disconnect(fd); |
619 | ||
620 | close(fd); | |
621 | ||
622 | printf("%s disconnected\n", argv[optind]); | |
623 | ||
713cc671 | 624 | return 0; |
cd831bd7 TS |
625 | } |
626 | ||
c1f8fdc3 PB |
627 | if (device && !verbose) { |
628 | int stderr_fd[2]; | |
629 | pid_t pid; | |
630 | int ret; | |
631 | ||
fc19f8a0 | 632 | if (qemu_pipe(stderr_fd) < 0) { |
85b01e09 MA |
633 | error_report("Error setting up communication pipe: %s", |
634 | strerror(errno)); | |
635 | exit(EXIT_FAILURE); | |
c1f8fdc3 PB |
636 | } |
637 | ||
638 | /* Now daemonize, but keep a communication channel open to | |
639 | * print errors and exit with the proper status code. | |
640 | */ | |
641 | pid = fork(); | |
70d4739e | 642 | if (pid < 0) { |
85b01e09 MA |
643 | error_report("Failed to fork: %s", strerror(errno)); |
644 | exit(EXIT_FAILURE); | |
70d4739e | 645 | } else if (pid == 0) { |
c1f8fdc3 | 646 | close(stderr_fd[0]); |
9faf31b6 | 647 | ret = qemu_daemon(1, 0); |
c1f8fdc3 PB |
648 | |
649 | /* Temporarily redirect stderr to the parent's pipe... */ | |
650 | dup2(stderr_fd[1], STDERR_FILENO); | |
fc19f8a0 | 651 | if (ret < 0) { |
85b01e09 MA |
652 | error_report("Failed to daemonize: %s", strerror(errno)); |
653 | exit(EXIT_FAILURE); | |
c1f8fdc3 PB |
654 | } |
655 | ||
656 | /* ... close the descriptor we inherited and go on. */ | |
657 | close(stderr_fd[1]); | |
658 | } else { | |
659 | bool errors = false; | |
660 | char *buf; | |
661 | ||
662 | /* In the parent. Print error messages from the child until | |
663 | * it closes the pipe. | |
664 | */ | |
665 | close(stderr_fd[1]); | |
666 | buf = g_malloc(1024); | |
667 | while ((ret = read(stderr_fd[0], buf, 1024)) > 0) { | |
668 | errors = true; | |
669 | ret = qemu_write_full(STDERR_FILENO, buf, ret); | |
fc19f8a0 | 670 | if (ret < 0) { |
c1f8fdc3 PB |
671 | exit(EXIT_FAILURE); |
672 | } | |
673 | } | |
fc19f8a0 | 674 | if (ret < 0) { |
85b01e09 MA |
675 | error_report("Cannot read from daemon: %s", |
676 | strerror(errno)); | |
677 | exit(EXIT_FAILURE); | |
c1f8fdc3 PB |
678 | } |
679 | ||
680 | /* Usually the daemon should not print any message. | |
681 | * Exit with zero status in that case. | |
682 | */ | |
683 | exit(errors); | |
684 | } | |
685 | } | |
686 | ||
a6ac2313 PB |
687 | if (device != NULL && sockpath == NULL) { |
688 | sockpath = g_malloc(128); | |
689 | snprintf(sockpath, 128, SOCKET_PATH, basename(device)); | |
cd831bd7 TS |
690 | } |
691 | ||
48bec07e DB |
692 | saddr = nbd_build_socket_address(sockpath, bindto, port); |
693 | ||
2f78e491 | 694 | if (qemu_init_main_loop(&local_err)) { |
565f65d2 | 695 | error_report_err(local_err); |
2f78e491 CN |
696 | exit(EXIT_FAILURE); |
697 | } | |
802ddc37 PB |
698 | bdrv_init(); |
699 | atexit(bdrv_close_all); | |
700 | ||
e6b63677 | 701 | if (fmt) { |
4fbec260 HR |
702 | options = qdict_new(); |
703 | qdict_put(options, "driver", qstring_from_str(fmt)); | |
e6b63677 DB |
704 | } |
705 | ||
802ddc37 | 706 | srcpath = argv[optind]; |
4fbec260 HR |
707 | blk = blk_new_open("hda", srcpath, NULL, options, flags, &local_err); |
708 | if (!blk) { | |
85b01e09 MA |
709 | error_report("Failed to blk_new_open '%s': %s", argv[optind], |
710 | error_get_pretty(local_err)); | |
711 | exit(EXIT_FAILURE); | |
802ddc37 | 712 | } |
4fbec260 | 713 | bs = blk_bs(blk); |
802ddc37 | 714 | |
8c116b0e WX |
715 | if (sn_opts) { |
716 | ret = bdrv_snapshot_load_tmp(bs, | |
717 | qemu_opt_get(sn_opts, SNAPSHOT_OPT_ID), | |
718 | qemu_opt_get(sn_opts, SNAPSHOT_OPT_NAME), | |
719 | &local_err); | |
720 | } else if (sn_id_or_name) { | |
721 | ret = bdrv_snapshot_load_tmp_by_id_or_name(bs, sn_id_or_name, | |
722 | &local_err); | |
723 | } | |
724 | if (ret < 0) { | |
725 | errno = -ret; | |
85b01e09 MA |
726 | error_report("Failed to load snapshot: %s: %s", |
727 | error_get_pretty(local_err), strerror(errno)); | |
728 | exit(EXIT_FAILURE); | |
8c116b0e WX |
729 | } |
730 | ||
b3838a40 | 731 | bs->detect_zeroes = detect_zeroes; |
4c58e80a | 732 | fd_size = blk_getlength(blk); |
98f44bbe | 733 | if (fd_size < 0) { |
85b01e09 MA |
734 | error_report("Failed to determine the image length: %s", |
735 | strerror(-fd_size)); | |
736 | exit(EXIT_FAILURE); | |
98f44bbe | 737 | } |
802ddc37 | 738 | |
185b4338 | 739 | if (partition != -1) { |
4c58e80a | 740 | ret = find_partition(blk, partition, &dev_offset, &fd_size); |
185b4338 PB |
741 | if (ret < 0) { |
742 | errno = -ret; | |
85b01e09 MA |
743 | error_report("Could not find partition %d: %s", partition, |
744 | strerror(errno)); | |
745 | exit(EXIT_FAILURE); | |
185b4338 | 746 | } |
802ddc37 PB |
747 | } |
748 | ||
98f44bbe HR |
749 | exp = nbd_export_new(blk, dev_offset, fd_size, nbdflags, nbd_export_closed, |
750 | &local_err); | |
751 | if (!exp) { | |
4fffeb5e | 752 | error_report_err(local_err); |
85b01e09 | 753 | exit(EXIT_FAILURE); |
98f44bbe | 754 | } |
3b05a8e9 | 755 | |
48bec07e | 756 | fd = socket_listen(saddr, &local_err); |
fc19f8a0 | 757 | if (fd < 0) { |
48bec07e | 758 | error_report_err(local_err); |
7a5ca864 | 759 | return 1; |
a61c6782 | 760 | } |
f1ef5555 PB |
761 | |
762 | if (device) { | |
763 | int ret; | |
764 | ||
a6ac2313 | 765 | ret = pthread_create(&client_thread, NULL, nbd_client_thread, device); |
f1ef5555 | 766 | if (ret != 0) { |
85b01e09 MA |
767 | error_report("Failed to create client thread: %s", strerror(ret)); |
768 | exit(EXIT_FAILURE); | |
f1ef5555 PB |
769 | } |
770 | } else { | |
771 | /* Shut up GCC warnings. */ | |
772 | memset(&client_thread, 0, sizeof(client_thread)); | |
773 | } | |
774 | ||
e4afbf4f FZ |
775 | server_fd = fd; |
776 | nbd_update_server_fd_handler(fd); | |
7a5ca864 | 777 | |
9faf31b6 MT |
778 | /* now when the initialization is (almost) complete, chdir("/") |
779 | * to free any busy filesystems */ | |
780 | if (chdir("/") < 0) { | |
85b01e09 MA |
781 | error_report("Could not chdir to root directory: %s", |
782 | strerror(errno)); | |
783 | exit(EXIT_FAILURE); | |
9faf31b6 MT |
784 | } |
785 | ||
7860a380 | 786 | state = RUNNING; |
3b05a8e9 | 787 | do { |
a61c6782 | 788 | main_loop_wait(false); |
7860a380 PB |
789 | if (state == TERMINATE) { |
790 | state = TERMINATING; | |
791 | nbd_export_close(exp); | |
792 | nbd_export_put(exp); | |
793 | exp = NULL; | |
794 | } | |
795 | } while (state != TERMINATED); | |
7a5ca864 | 796 | |
26f54e9a | 797 | blk_unref(blk); |
b32f6c28 PB |
798 | if (sockpath) { |
799 | unlink(sockpath); | |
800 | } | |
7a5ca864 | 801 | |
fbf28a43 | 802 | qemu_opts_del(sn_opts); |
8c116b0e | 803 | |
a517e88b PB |
804 | if (device) { |
805 | void *ret; | |
806 | pthread_join(client_thread, &ret); | |
807 | exit(ret != NULL); | |
808 | } else { | |
809 | exit(EXIT_SUCCESS); | |
810 | } | |
7a5ca864 | 811 | } |