~jderose/ubuntu/raring/qemu/vde-again

« back to all changes in this revision

Viewing changes to block-nbd.c

  • Committer: Bazaar Package Importer
  • Author(s): Aurelien Jarno, Aurelien Jarno
  • Date: 2009-03-22 10:13:17 UTC
  • mfrom: (1.2.1 upstream) (6.1.1 sid)
  • Revision ID: james.westby@ubuntu.com-20090322101317-iigjtnu5qil35dtb
Tags: 0.10.1-1
[ Aurelien Jarno ]
* New upstream stable release:
  - patches/80_stable-branch.patch: remove.
* debian/control: 
  - Remove depends on proll.
  - Move depends on device-tree-compiler to build-depends.
  - Bump Standards-Version to 3.8.1 (no changes).
* patches/82_qemu-img_decimal.patch: new patch from upstream to make
  qemu-img accept sizes with decimal values (closes: bug#501400).

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/*
 
2
 * QEMU Block driver for  NBD
 
3
 *
 
4
 * Copyright (C) 2008 Bull S.A.S.
 
5
 *     Author: Laurent Vivier <Laurent.Vivier@bull.net>
 
6
 *
 
7
 * Some parts:
 
8
 *    Copyright (C) 2007 Anthony Liguori <anthony@codemonkey.ws>
 
9
 *
 
10
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 
11
 * of this software and associated documentation files (the "Software"), to deal
 
12
 * in the Software without restriction, including without limitation the rights
 
13
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 
14
 * copies of the Software, and to permit persons to whom the Software is
 
15
 * furnished to do so, subject to the following conditions:
 
16
 *
 
17
 * The above copyright notice and this permission notice shall be included in
 
18
 * all copies or substantial portions of the Software.
 
19
 *
 
20
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 
21
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 
22
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 
23
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 
24
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 
25
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 
26
 * THE SOFTWARE.
 
27
 */
 
28
 
 
29
#include "qemu-common.h"
 
30
#include "nbd.h"
 
31
 
 
32
#include <sys/types.h>
 
33
#include <unistd.h>
 
34
 
 
35
typedef struct BDRVNBDState {
 
36
    int sock;
 
37
    off_t size;
 
38
    size_t blocksize;
 
39
} BDRVNBDState;
 
40
 
 
41
static int nbd_open(BlockDriverState *bs, const char* filename, int flags)
 
42
{
 
43
    BDRVNBDState *s = bs->opaque;
 
44
    const char *host;
 
45
    const char *unixpath;
 
46
    int sock;
 
47
    off_t size;
 
48
    size_t blocksize;
 
49
    int ret;
 
50
 
 
51
    if ((flags & BDRV_O_CREAT))
 
52
        return -EINVAL;
 
53
 
 
54
    if (!strstart(filename, "nbd:", &host))
 
55
        return -EINVAL;
 
56
 
 
57
    if (strstart(host, "unix:", &unixpath)) {
 
58
 
 
59
        if (unixpath[0] != '/')
 
60
            return -EINVAL;
 
61
 
 
62
        sock = unix_socket_outgoing(unixpath);
 
63
 
 
64
    } else {
 
65
        uint16_t port;
 
66
        char *p, *r;
 
67
        char hostname[128];
 
68
 
 
69
        pstrcpy(hostname, 128, host);
 
70
 
 
71
        p = strchr(hostname, ':');
 
72
        if (p == NULL)
 
73
            return -EINVAL;
 
74
 
 
75
        *p = '\0';
 
76
        p++;
 
77
 
 
78
        port = strtol(p, &r, 0);
 
79
        if (r == p)
 
80
            return -EINVAL;
 
81
        sock = tcp_socket_outgoing(hostname, port);
 
82
    }
 
83
 
 
84
    if (sock == -1)
 
85
        return -errno;
 
86
 
 
87
    ret = nbd_receive_negotiate(sock, &size, &blocksize);
 
88
    if (ret == -1)
 
89
        return -errno;
 
90
 
 
91
    s->sock = sock;
 
92
    s->size = size;
 
93
    s->blocksize = blocksize;
 
94
 
 
95
    return 0;
 
96
}
 
97
 
 
98
static int nbd_read(BlockDriverState *bs, int64_t sector_num,
 
99
                    uint8_t *buf, int nb_sectors)
 
100
{
 
101
    BDRVNBDState *s = bs->opaque;
 
102
    struct nbd_request request;
 
103
    struct nbd_reply reply;
 
104
 
 
105
    request.type = NBD_CMD_READ;
 
106
    request.handle = (uint64_t)(intptr_t)bs;
 
107
    request.from = sector_num * 512;;
 
108
    request.len = nb_sectors * 512;
 
109
 
 
110
    if (nbd_send_request(s->sock, &request) == -1)
 
111
        return -errno;
 
112
 
 
113
    if (nbd_receive_reply(s->sock, &reply) == -1)
 
114
        return -errno;
 
115
 
 
116
    if (reply.error !=0)
 
117
        return -reply.error;
 
118
 
 
119
    if (reply.handle != request.handle)
 
120
        return -EIO;
 
121
 
 
122
    if (nbd_wr_sync(s->sock, buf, request.len, 1) != request.len)
 
123
        return -EIO;
 
124
 
 
125
    return 0;
 
126
}
 
127
 
 
128
static int nbd_write(BlockDriverState *bs, int64_t sector_num,
 
129
                     const uint8_t *buf, int nb_sectors)
 
130
{
 
131
    BDRVNBDState *s = bs->opaque;
 
132
    struct nbd_request request;
 
133
    struct nbd_reply reply;
 
134
 
 
135
    request.type = NBD_CMD_WRITE;
 
136
    request.handle = (uint64_t)(intptr_t)bs;
 
137
    request.from = sector_num * 512;;
 
138
    request.len = nb_sectors * 512;
 
139
 
 
140
    if (nbd_send_request(s->sock, &request) == -1)
 
141
        return -errno;
 
142
 
 
143
    if (nbd_wr_sync(s->sock, (uint8_t*)buf, request.len, 0) != request.len)
 
144
        return -EIO;
 
145
 
 
146
    if (nbd_receive_reply(s->sock, &reply) == -1)
 
147
        return -errno;
 
148
 
 
149
    if (reply.error !=0)
 
150
        return -reply.error;
 
151
 
 
152
    if (reply.handle != request.handle)
 
153
        return -EIO;
 
154
 
 
155
    return 0;
 
156
}
 
157
 
 
158
static void nbd_close(BlockDriverState *bs)
 
159
{
 
160
    BDRVNBDState *s = bs->opaque;
 
161
    struct nbd_request request;
 
162
 
 
163
    request.type = NBD_CMD_DISC;
 
164
    request.handle = (uint64_t)(intptr_t)bs;
 
165
    request.from = 0;
 
166
    request.len = 0;
 
167
    nbd_send_request(s->sock, &request);
 
168
 
 
169
    close(s->sock);
 
170
}
 
171
 
 
172
static int64_t nbd_getlength(BlockDriverState *bs)
 
173
{
 
174
    BDRVNBDState *s = bs->opaque;
 
175
 
 
176
    return s->size;
 
177
}
 
178
 
 
179
BlockDriver bdrv_nbd = {
 
180
    "nbd",
 
181
    sizeof(BDRVNBDState),
 
182
    NULL, /* no probe for protocols */
 
183
    nbd_open,
 
184
    nbd_read,
 
185
    nbd_write,
 
186
    nbd_close,
 
187
    .bdrv_getlength = nbd_getlength,
 
188
    .protocol_name = "nbd",
 
189
};