Revision 7a5ca864

b/Makefile
17 17
LDFLAGS += -static
18 18
endif
19 19
ifdef BUILD_DOCS
20
DOCS=qemu-doc.html qemu-tech.html qemu.1 qemu-img.1
20
DOCS=qemu-doc.html qemu-tech.html qemu.1 qemu-img.1 qemu-nbd.8
21 21
else
22 22
DOCS=
23 23
endif
......
159 159
%.o: %.c
160 160
	$(CC) $(CFLAGS) $(CPPFLAGS) -c -o $@ $<
161 161

  
162
qemu-nbd$(EXESUF):  qemu-nbd.o nbd.o qemu-img-block.o \
163
		    $(QEMU_IMG_BLOCK_OBJS)
164
	$(CC) $(LDFLAGS) -o $@ $^ -lz $(LIBS)
165

  
162 166
# dyngen host tool
163 167
dyngen$(EXESUF): dyngen.c
164 168
	$(HOST_CC) $(CFLAGS) $(CPPFLAGS) -o $@ $^
......
191 195
ifndef CONFIG_WIN32
192 196
	mkdir -p "$(DESTDIR)$(mandir)/man1"
193 197
	$(INSTALL) qemu.1 qemu-img.1 "$(DESTDIR)$(mandir)/man1"
198
	mkdir -p "$(DESTDIR)$(mandir)/man8"
199
	$(INSTALL) qemu-nbd.8 "$(DESTDIR)$(mandir)/man8"
194 200
endif
195 201

  
196 202
install: all $(if $(BUILD_DOCS),install-doc)
......
244 250
	$(SRC_PATH)/texi2pod.pl $< qemu-img.pod
245 251
	pod2man --section=1 --center=" " --release=" " qemu-img.pod > $@
246 252

  
253
qemu-nbd.8: qemu-nbd.texi
254
	$(SRC_PATH)/texi2pod.pl $< qemu-nbd.pod
255
	pod2man --section=8 --center=" " --release=" " qemu-nbd.pod > $@
256

  
247 257
info: qemu-doc.info qemu-tech.info
248 258

  
249 259
dvi: qemu-doc.dvi qemu-tech.dvi
......
296 306
        $(bindir)/qemu-sh4eb \
297 307
        $(bindir)/qemu-cris \
298 308
        $(bindir)/qemu-img \
309
        $(bindir)/qemu-nbd \
299 310
	$(datadir)/bios.bin \
300 311
	$(datadir)/vgabios.bin \
301 312
	$(datadir)/vgabios-cirrus.bin \
......
309 320
	$(docdir)/qemu-doc.html \
310 321
	$(docdir)/qemu-tech.html \
311 322
	$(mandir)/man1/qemu.1 $(mandir)/man1/qemu-img.1
323
	$(mandir)/man8/qemu-nbd.8
312 324

  
313 325
# Include automatically generated dependency files
314 326
-include $(wildcard *.d audio/*.d slirp/*.d)
b/configure
1105 1105
tools=
1106 1106
if test `expr "$target_list" : ".*softmmu.*"` != 0 ; then
1107 1107
  tools="qemu-img\$(EXESUF) $tools"
1108
  if [ "$linux" = "yes" ] ; then
1109
      tools="qemu-nbd\$(EXESUF) $tools"
1110
  fi
1108 1111
fi
1109 1112
echo "TOOLS=$tools" >> $config_mak
1110 1113

  
b/exec-all.h
126 126
#define CODE_GEN_BUFFER_SIZE     (6 * 1024 * 1024)
127 127
#else
128 128
/* XXX: make it dynamic on x86 */
129
#define CODE_GEN_BUFFER_SIZE     (16 * 1024 * 1024)
129
#define CODE_GEN_BUFFER_SIZE     (64 * 1024 * 1024)
130 130
#endif
131 131

  
132 132
//#define CODE_GEN_BUFFER_SIZE     (128 * 1024)
b/nbd.c
1
/*\
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
16
 *  along with this program; if not, write to the Free Software
17
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
\*/
19

  
20
#include "nbd.h"
21

  
22
#include <errno.h>
23
#include <string.h>
24
#include <sys/ioctl.h>
25
#include <ctype.h>
26
#include <inttypes.h>
27
#include <sys/socket.h>
28
#include <netinet/in.h>
29
#include <netinet/tcp.h>
30
#include <arpa/inet.h>
31
#include <netdb.h>
32

  
33
extern int verbose;
34

  
35
#define LOG(msg, ...) do { \
36
    fprintf(stderr, "%s:%s():L%d: " msg "\n", \
37
            __FILE__, __FUNCTION__, __LINE__, ## __VA_ARGS__); \
38
} while(0)
39

  
40
#define TRACE(msg, ...) do { \
41
    if (verbose) LOG(msg, ## __VA_ARGS__); \
42
} while(0)
43

  
44
/* This is all part of the "official" NBD API */
45

  
46
#define NBD_REQUEST_MAGIC       0x25609513
47
#define NBD_REPLY_MAGIC         0x67446698
48

  
49
#define NBD_SET_SOCK            _IO(0xab, 0)
50
#define NBD_SET_BLKSIZE         _IO(0xab, 1)
51
#define NBD_SET_SIZE            _IO(0xab, 2)
52
#define NBD_DO_IT               _IO(0xab, 3)
53
#define NBD_CLEAR_SOCK          _IO(0xab, 4)
54
#define NBD_CLEAR_QUE           _IO(0xab, 5)
55
#define NBD_PRINT_DEBUG	        _IO(0xab, 6)
56
#define NBD_SET_SIZE_BLOCKS	_IO(0xab, 7)
57
#define NBD_DISCONNECT          _IO(0xab, 8)
58

  
59
/* That's all folks */
60

  
61
#define read_sync(fd, buffer, size) wr_sync(fd, buffer, size, true)
62
#define write_sync(fd, buffer, size) wr_sync(fd, buffer, size, false)
63

  
64
static size_t wr_sync(int fd, void *buffer, size_t size, bool do_read)
65
{
66
    size_t offset = 0;
67

  
68
    while (offset < size) {
69
        ssize_t len;
70

  
71
        if (do_read) {
72
            len = read(fd, buffer + offset, size - offset);
73
        } else {
74
            len = write(fd, buffer + offset, size - offset);
75
        }
76

  
77
        /* recoverable error */
78
        if (len == -1 && errno == EAGAIN) {
79
            continue;
80
        }
81

  
82
        /* eof */
83
        if (len == 0) {
84
            break;
85
        }
86

  
87
        /* unrecoverable error */
88
        if (len == -1) {
89
            return 0;
90
        }
91

  
92
        offset += len;
93
    }
94

  
95
    return offset;
96
}
97

  
98
static int tcp_socket_outgoing(const char *address, uint16_t port)
99
{
100
    int s;
101
    struct in_addr in;
102
    struct sockaddr_in addr;
103
    int serrno;
104

  
105
    s = socket(PF_INET, SOCK_STREAM, 0);
106
    if (s == -1) {
107
        return -1;
108
    }
109

  
110
    if (inet_aton(address, &in) == 0) {
111
        struct hostent *ent;
112

  
113
        ent = gethostbyname(address);
114
        if (ent == NULL) {
115
            goto error;
116
        }
117

  
118
        memcpy(&in, ent->h_addr, sizeof(in));
119
    }
120

  
121
    addr.sin_family = AF_INET;
122
    addr.sin_port = htons(port);
123
    memcpy(&addr.sin_addr.s_addr, &in, sizeof(in));
124

  
125
    if (connect(s, (struct sockaddr *)&addr, sizeof(addr)) == -1) {
126
        goto error;
127
    }
128

  
129
    return s;
130
error:
131
    serrno = errno;
132
    close(s);
133
    errno = serrno;
134
    return -1;
135
}
136

  
137
int tcp_socket_incoming(const char *address, uint16_t port)
138
{
139
    int s;
140
    struct in_addr in;
141
    struct sockaddr_in addr;
142
    int serrno;
143
    int opt;
144

  
145
    s = socket(PF_INET, SOCK_STREAM, 0);
146
    if (s == -1) {
147
        return -1;
148
    }
149

  
150
    if (inet_aton(address, &in) == 0) {
151
        struct hostent *ent;
152

  
153
        ent = gethostbyname(address);
154
        if (ent == NULL) {
155
            goto error;
156
        }
157

  
158
        memcpy(&in, ent->h_addr, sizeof(in));
159
    }
160

  
161
    addr.sin_family = AF_INET;
162
    addr.sin_port = htons(port);
163
    memcpy(&addr.sin_addr.s_addr, &in, sizeof(in));
164

  
165
    opt = 1;
166
    if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) == -1) {
167
        goto error;
168
    }
169

  
170
    if (bind(s, (struct sockaddr *)&addr, sizeof(addr)) == -1) {
171
        goto error;
172
    }
173

  
174
    if (listen(s, 128) == -1) {
175
        goto error;
176
    }
177

  
178
    return s;
179
error:
180
    serrno = errno;
181
    close(s);
182
    errno = serrno;
183
    return -1;
184
}
185

  
186
/* Basic flow
187

  
188
   Server         Client
189

  
190
   Negotiate
191
                  Request
192
   Response
193
                  Request
194
   Response
195
                  ...
196
   ...
197
                  Request (type == 2)
198
*/
199

  
200
int nbd_negotiate(BlockDriverState *bs, int csock, off_t size)
201
{
202
	char buf[8 + 8 + 8 + 128];
203

  
204
	/* Negotiate
205
	   [ 0 ..   7]   passwd   ("NBDMAGIC")
206
	   [ 8 ..  15]   magic    (0x00420281861253)
207
	   [16 ..  23]   size
208
	   [24 .. 151]   reserved (0)
209
	 */
210

  
211
	TRACE("Beginning negotiation.");
212
	memcpy(buf, "NBDMAGIC", 8);
213
	cpu_to_be64w((uint64_t*)(buf + 8), 0x00420281861253LL);
214
	cpu_to_be64w((uint64_t*)(buf + 16), size);
215
	memset(buf + 24, 0, 128);
216

  
217
	if (write_sync(csock, buf, sizeof(buf)) != sizeof(buf)) {
218
		LOG("write failed");
219
		errno = EINVAL;
220
		return -1;
221
	}
222

  
223
	TRACE("Negotation succeeded.");
224

  
225
	return 0;
226
}
227

  
228
int nbd_receive_negotiate(int fd, int csock)
229
{
230
	char buf[8 + 8 + 8 + 128];
231
	uint64_t magic;
232
	off_t size;
233
	size_t blocksize;
234

  
235
	TRACE("Receiving negotation.");
236

  
237
	if (read_sync(csock, buf, sizeof(buf)) != sizeof(buf)) {
238
		LOG("read failed");
239
		errno = EINVAL;
240
		return -1;
241
	}
242

  
243
	magic = be64_to_cpup((uint64_t*)(buf + 8));
244
	size = be64_to_cpup((uint64_t*)(buf + 16));
245
	blocksize = 1024;
246

  
247
	TRACE("Magic is %c%c%c%c%c%c%c%c",
248
	      isprint(buf[0]) ? buf[0] : '.',
249
	      isprint(buf[1]) ? buf[1] : '.',
250
	      isprint(buf[2]) ? buf[2] : '.',
251
	      isprint(buf[3]) ? buf[3] : '.',
252
	      isprint(buf[4]) ? buf[4] : '.',
253
	      isprint(buf[5]) ? buf[5] : '.',
254
	      isprint(buf[6]) ? buf[6] : '.',
255
	      isprint(buf[7]) ? buf[7] : '.');
256
	TRACE("Magic is 0x%" PRIx64, magic);
257
	TRACE("Size is %" PRIu64, size);
258

  
259
	if (memcmp(buf, "NBDMAGIC", 8) != 0) {
260
		LOG("Invalid magic received");
261
		errno = EINVAL;
262
		return -1;
263
	}
264

  
265
	TRACE("Checking magic");
266

  
267
	if (magic != 0x00420281861253LL) {
268
		LOG("Bad magic received");
269
		errno = EINVAL;
270
		return -1;
271
	}
272

  
273
	TRACE("Setting block size to %lu", (unsigned long)blocksize);
274

  
275
	if (ioctl(fd, NBD_SET_BLKSIZE, blocksize) == -1) {
276
		int serrno = errno;
277
		LOG("Failed setting NBD block size");
278
		errno = serrno;
279
		return -1;
280
	}
281

  
282
	TRACE("Setting size to %llu block(s)",
283
	      (unsigned long long)(size / blocksize));
284

  
285
	if (ioctl(fd, NBD_SET_SIZE_BLOCKS, size / blocksize) == -1) {
286
		int serrno = errno;
287
		LOG("Failed setting size (in blocks)");
288
		errno = serrno;
289
		return -1;
290
	}
291

  
292
	TRACE("Clearing NBD socket");
293

  
294
	if (ioctl(fd, NBD_CLEAR_SOCK) == -1) {
295
		int serrno = errno;
296
		LOG("Failed clearing NBD socket");
297
		errno = serrno;
298
		return -1;
299
	}
300

  
301
	TRACE("Setting NBD socket");
302

  
303
	if (ioctl(fd, NBD_SET_SOCK, csock) == -1) {
304
		int serrno = errno;
305
		LOG("Failed to set NBD socket");
306
		errno = serrno;
307
		return -1;
308
	}
309

  
310
	TRACE("Negotiation ended");
311

  
312
	return 0;
313
}
314

  
315
int nbd_disconnect(int fd)
316
{
317
	ioctl(fd, NBD_CLEAR_QUE);
318
	ioctl(fd, NBD_DISCONNECT);
319
	ioctl(fd, NBD_CLEAR_SOCK);
320
	return 0;
321
}
322

  
323
int nbd_client(int fd, int csock)
324
{
325
	int ret;
326
	int serrno;
327

  
328
	TRACE("Doing NBD loop");
329

  
330
	ret = ioctl(fd, NBD_DO_IT);
331
	serrno = errno;
332

  
333
	TRACE("NBD loop returned %d: %s", ret, strerror(serrno));
334

  
335
	TRACE("Clearing NBD queue");
336
	ioctl(fd, NBD_CLEAR_QUE);
337

  
338
	TRACE("Clearing NBD socket");
339
	ioctl(fd, NBD_CLEAR_SOCK);
340

  
341
	errno = serrno;
342
	return ret;
343
}
344

  
345
int nbd_trip(BlockDriverState *bs, int csock, off_t size, uint64_t dev_offset, off_t *offset, bool readonly)
346
{
347
#ifndef _REENTRANT
348
	static uint8_t data[1024 * 1024]; // keep this off of the stack
349
#else
350
	uint8_t data[1024 * 1024];
351
#endif
352
	uint8_t buf[4 + 4 + 8 + 8 + 4];
353
	uint32_t magic;
354
	uint32_t type;
355
	uint64_t from;
356
	uint32_t len;
357

  
358
	TRACE("Reading request.");
359

  
360
	if (read_sync(csock, buf, sizeof(buf)) != sizeof(buf)) {
361
		LOG("read failed");
362
		errno = EINVAL;
363
		return -1;
364
	}
365

  
366
	/* Request
367
	  [ 0 ..  3]   magic   (NBD_REQUEST_MAGIC)
368
	  [ 4 ..  7]   type    (0 == READ, 1 == WRITE)
369
	  [ 8 .. 15]   handle
370
	  [16 .. 23]   from
371
	  [24 .. 27]   len
372
	 */
373

  
374
	magic = be32_to_cpup((uint32_t*)buf);
375
	type  = be32_to_cpup((uint32_t*)(buf + 4));
376
	from  = be64_to_cpup((uint64_t*)(buf + 16));
377
	len   = be32_to_cpup((uint32_t*)(buf + 24));
378

  
379
	TRACE("Got request: "
380
	      "{ magic = 0x%x, .type = %d, from = %" PRIu64" , len = %u }",
381
	      magic, type, from, len);
382

  
383

  
384
	if (magic != NBD_REQUEST_MAGIC) {
385
		LOG("invalid magic (got 0x%x)", magic);
386
		errno = EINVAL;
387
		return -1;
388
	}
389

  
390
	if (len > sizeof(data)) {
391
		LOG("len (%u) is larger than max len (%lu)",
392
		    len, sizeof(data));
393
		errno = EINVAL;
394
		return -1;
395
	}
396

  
397
	if ((from + len) < from) {
398
		LOG("integer overflow detected! "
399
		    "you're probably being attacked");
400
		errno = EINVAL;
401
		return -1;
402
	}
403

  
404
	if ((from + len) > size) {
405
	        LOG("From: %" PRIu64 ", Len: %u, Size: %" PRIu64
406
		    ", Offset: %" PRIu64 "\n",
407
		     from, len, size, dev_offset);
408
		LOG("requested operation past EOF--bad client?");
409
		errno = EINVAL;
410
		return -1;
411
	}
412

  
413
	/* Reply
414
	 [ 0 ..  3]    magic   (NBD_REPLY_MAGIC)
415
	 [ 4 ..  7]    error   (0 == no error)
416
         [ 7 .. 15]    handle
417
	 */
418
	cpu_to_be32w((uint32_t*)buf, NBD_REPLY_MAGIC);
419
	cpu_to_be32w((uint32_t*)(buf + 4), 0);
420

  
421
	TRACE("Decoding type");
422

  
423
	switch (type) {
424
	case 0:
425
		TRACE("Request type is READ");
426

  
427
		if (bdrv_read(bs, (from + dev_offset) / 512, data, len / 512) == -1) {
428
			LOG("reading from file failed");
429
			errno = EINVAL;
430
			return -1;
431
		}
432
		*offset += len;
433

  
434
		TRACE("Read %u byte(s)", len);
435

  
436
		TRACE("Sending OK response");
437

  
438
		if (write_sync(csock, buf, 16) != 16) {
439
			LOG("writing to socket failed");
440
			errno = EINVAL;
441
			return -1;
442
		}
443

  
444
		TRACE("Sending data to client");
445

  
446
		if (write_sync(csock, data, len) != len) {
447
			LOG("writing to socket failed");
448
			errno = EINVAL;
449
			return -1;
450
		}
451
		break;
452
	case 1:
453
		TRACE("Request type is WRITE");
454

  
455
		TRACE("Reading %u byte(s)", len);
456

  
457
		if (read_sync(csock, data, len) != len) {
458
			LOG("reading from socket failed");
459
			errno = EINVAL;
460
			return -1;
461
		}
462

  
463
		if (readonly) {
464
			TRACE("Server is read-only, return error");
465

  
466
			cpu_to_be32w((uint32_t*)(buf + 4), 1);
467
		} else {
468
			TRACE("Writing to device");
469

  
470
			if (bdrv_write(bs, (from + dev_offset) / 512, data, len / 512) == -1) {
471
				LOG("writing to file failed");
472
				errno = EINVAL;
473
				return -1;
474
			}
475

  
476
			*offset += len;
477
		}
478

  
479
		TRACE("Sending response to client");
480

  
481
		if (write_sync(csock, buf, 16) != 16) {
482
			LOG("writing to socket failed");
483
			errno = EINVAL;
484
			return -1;
485
		}
486
		break;
487
	case 2:
488
		TRACE("Request type is DISCONNECT");
489
		errno = 0;
490
		return 1;
491
	default:
492
		LOG("invalid request type (%u) received", type);
493
		errno = EINVAL;
494
		return -1;
495
	}
496

  
497
	TRACE("Request/Reply complete");
498

  
499
	return 0;
500
}
b/nbd.h
1
/*\
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
16
 *  along with this program; if not, write to the Free Software
17
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
\*/
19

  
20
#ifndef NBD_H
21
#define NBD_H
22

  
23
#include <sys/types.h>
24
#include <stdbool.h>
25

  
26
#include <qemu-common.h>
27
#include "block_int.h"
28

  
29
int tcp_socket_incoming(const char *address, uint16_t port);
30

  
31
int nbd_negotiate(BlockDriverState *bs, int csock, off_t size);
32
int nbd_receive_negotiate(int fd, int csock);
33
int nbd_trip(BlockDriverState *bs, int csock, off_t size, uint64_t dev_offset, off_t *offset, bool readonly);
34
int nbd_client(int fd, int csock);
35
int nbd_disconnect(int fd);
36

  
37
#endif
b/qemu-nbd.c
1
/*\
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
16
 *  along with this program; if not, write to the Free Software
17
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18
 */
19

  
20
#include <qemu-common.h>
21
#include "block_int.h"
22
#include "nbd.h"
23

  
24
#include <malloc.h>
25
#include <stdarg.h>
26
#include <stdio.h>
27
#include <getopt.h>
28
#include <err.h>
29
#include <sys/socket.h>
30
#include <netinet/in.h>
31
#include <netinet/tcp.h>
32
#include <arpa/inet.h>
33

  
34
int verbose;
35

  
36
static void usage(const char *name)
37
{
38
    printf(
39
"Usage: %s [OPTIONS] FILE\n"
40
"QEMU Disk Network Block Device Server\n"
41
"\n"
42
"  -p, --port=PORT      port to listen on (default `1024')\n"
43
"  -o, --offset=OFFSET  offset into the image\n"
44
"  -b, --bind=IFACE     interface to bind to (default `0.0.0.0')\n"
45
"  -r, --read-only      export read-only\n"
46
"  -P, --partition=NUM  only expose partition NUM\n"
47
"  -v, --verbose        display extra debugging information\n"
48
"  -h, --help           display this help and exit\n"
49
"  -V, --version        output version information and exit\n"
50
"\n"
51
"Report bugs to <anthony@codemonkey.ws>\n"
52
    , name);
53
}
54

  
55
static void version(const char *name)
56
{
57
    printf(
58
"qemu-nbd version 0.0.1\n"
59
"Written by Anthony Liguori.\n"
60
"\n"
61
"Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>.\n"
62
"This is free software; see the source for copying conditions.  There is NO\n"
63
"warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"
64
    );
65
}
66

  
67
struct partition_record
68
{
69
    uint8_t bootable;
70
    uint8_t start_head;
71
    uint32_t start_cylinder;
72
    uint8_t start_sector;
73
    uint8_t system;
74
    uint8_t end_head;
75
    uint8_t end_cylinder;
76
    uint8_t end_sector;
77
    uint32_t start_sector_abs;
78
    uint32_t nb_sectors_abs;
79
};
80

  
81
static void read_partition(uint8_t *p, struct partition_record *r)
82
{
83
    r->bootable = p[0];
84
    r->start_head = p[1];
85
    r->start_cylinder = p[3] | ((p[2] << 2) & 0x0300);
86
    r->start_sector = p[2] & 0x3f;
87
    r->system = p[4];
88
    r->end_head = p[5];
89
    r->end_cylinder = p[7] | ((p[6] << 2) & 0x300);
90
    r->end_sector = p[6] & 0x3f;
91
    r->start_sector_abs = p[8] | p[9] << 8 | p[10] << 16 | p[11] << 24;
92
    r->nb_sectors_abs = p[12] | p[13] << 8 | p[14] << 16 | p[15] << 24;
93
}
94

  
95
static int find_partition(BlockDriverState *bs, int partition,
96
                          off_t *offset, off_t *size)
97
{
98
    struct partition_record mbr[4];
99
    uint8_t data[512];
100
    int i;
101
    int ext_partnum = 4;
102

  
103
    if (bdrv_read(bs, 0, data, 1))
104
        errx(EINVAL, "error while reading");
105

  
106
    if (data[510] != 0x55 || data[511] != 0xaa) {
107
        errno = -EINVAL;
108
        return -1;
109
    }
110

  
111
    for (i = 0; i < 4; i++) {
112
        read_partition(&data[446 + 16 * i], &mbr[i]);
113

  
114
        if (!mbr[i].nb_sectors_abs)
115
            continue;
116

  
117
        if (mbr[i].system == 0xF || mbr[i].system == 0x5) {
118
            struct partition_record ext[4];
119
            uint8_t data1[512];
120
            int j;
121

  
122
            if (bdrv_read(bs, mbr[i].start_sector_abs, data1, 1))
123
                errx(EINVAL, "error while reading");
124

  
125
            for (j = 0; j < 4; j++) {
126
                read_partition(&data1[446 + 16 * j], &ext[j]);
127
                if (!ext[j].nb_sectors_abs)
128
                    continue;
129

  
130
                if ((ext_partnum + j + 1) == partition) {
131
                    *offset = (uint64_t)ext[j].start_sector_abs << 9;
132
                    *size = (uint64_t)ext[j].nb_sectors_abs << 9;
133
                    return 0;
134
                }
135
            }
136
            ext_partnum += 4;
137
        } else if ((i + 1) == partition) {
138
            *offset = (uint64_t)mbr[i].start_sector_abs << 9;
139
            *size = (uint64_t)mbr[i].nb_sectors_abs << 9;
140
            return 0;
141
        }
142
    }
143

  
144
    errno = -ENOENT;
145
    return -1;
146
}
147

  
148
int main(int argc, char **argv)
149
{
150
    BlockDriverState *bs;
151
    off_t dev_offset = 0;
152
    off_t offset = 0;
153
    bool readonly = false;
154
    const char *bindto = "0.0.0.0";
155
    int port = 1024;
156
    int sock, csock;
157
    struct sockaddr_in addr;
158
    socklen_t addr_len = sizeof(addr);
159
    off_t fd_size;
160
    const char *sopt = "hVbo:p:rsP:v";
161
    struct option lopt[] = {
162
        { "help", 0, 0, 'h' },
163
        { "version", 0, 0, 'V' },
164
        { "bind", 1, 0, 'b' },
165
        { "port", 1, 0, 'p' },
166
        { "offset", 1, 0, 'o' },
167
        { "read-only", 0, 0, 'r' },
168
        { "partition", 1, 0, 'P' },
169
        { "snapshot", 0, 0, 's' },
170
        { "verbose", 0, 0, 'v' },
171
    };
172
    int ch;
173
    int opt_ind = 0;
174
    int li;
175
    char *end;
176
    bool snapshot = false;
177
    int partition = -1;
178

  
179
    while ((ch = getopt_long(argc, argv, sopt, lopt, &opt_ind)) != -1) {
180
        switch (ch) {
181
        case 's':
182
            snapshot = true;
183
            break;
184
        case 'b':
185
            bindto = optarg;
186
            break;
187
        case 'p':
188
            li = strtol(optarg, &end, 0);
189
            if (*end) {
190
                errx(EINVAL, "Invalid port `%s'", optarg);
191
            }
192
            if (li < 1 || li > 65535) {
193
                errx(EINVAL, "Port out of range `%s'", optarg);
194
            }
195
            port = (uint16_t)li;
196
            break;
197
        case 'o':
198
                dev_offset = strtoll (optarg, &end, 0);
199
            if (*end) {
200
                errx(EINVAL, "Invalid offset `%s'", optarg);
201
            }
202
            if (dev_offset < 0) {
203
                errx(EINVAL, "Offset must be positive `%s'", optarg);
204
            }
205
            break;
206
        case 'r':
207
            readonly = true;
208
            break;
209
        case 'P':
210
            partition = strtol(optarg, &end, 0);
211
            if (*end)
212
                errx(EINVAL, "Invalid partition `%s'", optarg);
213
            if (partition < 1 || partition > 8)
214
                errx(EINVAL, "Invalid partition %d", partition);
215
            break;
216
        case 'v':
217
            verbose = 1;
218
            break;
219
        case 'V':
220
            version(argv[0]);
221
            exit(0);
222
            break;
223
        case 'h':
224
            usage(argv[0]);
225
            exit(0);
226
            break;
227
        case '?':
228
            errx(EINVAL, "Try `%s --help' for more information.",
229
                 argv[0]);
230
        }
231
    }
232

  
233
    if ((argc - optind) != 1) {
234
        errx(EINVAL, "Invalid number of argument.\n"
235
             "Try `%s --help' for more information.",
236
             argv[0]);
237
    }
238

  
239
    bdrv_init();
240

  
241
    bs = bdrv_new("hda");
242
    if (bs == NULL)
243
        return 1;
244

  
245
    if (bdrv_open(bs, argv[optind], snapshot) == -1)
246
        return 1;
247

  
248
    fd_size = bs->total_sectors * 512;
249

  
250
    if (partition != -1 &&
251
        find_partition(bs, partition, &dev_offset, &fd_size))
252
        errx(errno, "Could not find partition %d", partition);
253

  
254
    sock = tcp_socket_incoming(bindto, port);
255
    if (sock == -1)
256
        return 1;
257

  
258
    csock = accept(sock,
259
               (struct sockaddr *)&addr,
260
               &addr_len);
261
    if (csock == -1)
262
        return 1;
263

  
264
    /* new fd_size is calculated by find_partition */
265
    if (nbd_negotiate(bs, csock, fd_size) == -1)
266
        return 1;
267

  
268
    while (nbd_trip(bs, csock, fd_size, dev_offset, &offset, readonly) == 0);
269

  
270
    close(csock);
271
    close(sock);
272
    bdrv_close(bs);
273

  
274
    return 0;
275
}
b/qemu-nbd.texi
1
@example
2
@c man begin SYNOPSIS
3
usage: qemu-nbd [OPTION]...  @var{filename}
4
@c man end
5
@end example
6

  
7
@c man begin DESCRIPTION
8

  
9
Export Qemu disk image using NBD protocol.
10

  
11
@c man end
12

  
13
@c man begin OPTIONS
14
@table @var
15
@item filename
16
 is a disk image filename
17
@item -p, --port=PORT
18
  port to listen on (default `1024')
19
@item -o, --offset=OFFSET
20
  offset into the image
21
@item -b, --bind=IFACE
22
  interface to bind to (default `0.0.0.0')
23
@item -r, --read-only
24
  export read-only
25
@item -P, --partition=NUM
26
  only expose partition NUM
27
@item -v, --verbose
28
  display extra debugging information
29
@item -h, --help
30
  display this help and exit
31
@item -V, --version
32
  output version information and exit
33
@end table
34

  
35
@c man end
36

  
37
@ignore
38

  
39
@setfilename qemu-nbd
40
@settitle QEMU Disk Network Block Device Server
41

  
42
@c man begin AUTHOR
43
Copyright (C) 2006 Anthony Liguori <anthony@codemonkey.ws>.
44
This is free software; see the source for copying conditions.  There is NO
45
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
46
@c man end
47

  
48
@c man begin SEEALSO
49
qemu-img(1)
50
@c man end
51

  
52
@end ignore

Also available in: Unified diff