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 #include "module.h" 32 #include "qemu_socket.h" 33 34 #include <sys/types.h> 35 #include <unistd.h> 36 37 #define EN_OPTSTR ":exportname=" 38 39 /* #define DEBUG_NBD */ 40 41 #if defined(DEBUG_NBD) 42 #define logout(fmt, ...) \ 43 fprintf(stderr, "nbd\t%-24s" fmt, __func__, ##__VA_ARGS__) 44 #else 45 #define logout(fmt, ...) ((void)0) 46 #endif 47 48 typedef struct BDRVNBDState { 49 int sock; 50 off_t size; 51 size_t blocksize; 52 char *export_name; /* An NBD server may export several devices */ 53 54 /* If it begins with '/', this is a UNIX domain socket. Otherwise, 55 * it's a string of the form <hostname|ip4|\[ip6\]>:port 56 */ 57 char *host_spec; 58 } BDRVNBDState; 59 60 static int nbd_config(BDRVNBDState *s, const char *filename, int flags) 61 { 62 char *file; 63 char *export_name; 64 const char *host_spec; 65 const char *unixpath; 66 int err = -EINVAL; 67 68 file = g_strdup(filename); 69 70 export_name = strstr(file, EN_OPTSTR); 71 if (export_name) { 72 if (export_name[strlen(EN_OPTSTR)] == 0) { 73 goto out; 74 } 75 export_name[0] = 0; /* truncate 'file' */ 76 export_name += strlen(EN_OPTSTR); 77 s->export_name = g_strdup(export_name); 78 } 79 80 /* extract the host_spec - fail if it's not nbd:... */ 81 if (!strstart(file, "nbd:", &host_spec)) { 82 goto out; 83 } 84 85 /* are we a UNIX or TCP socket? */ 86 if (strstart(host_spec, "unix:", &unixpath)) { 87 if (unixpath[0] != '/') { /* We demand an absolute path*/ 88 goto out; 89 } 90 s->host_spec = g_strdup(unixpath); 91 } else { 92 s->host_spec = g_strdup(host_spec); 93 } 94 95 err = 0; 96 97 out: 98 g_free(file); 99 if (err != 0) { 100 g_free(s->export_name); 101 g_free(s->host_spec); 102 } 103 return err; 104 } 105 106 static int nbd_establish_connection(BlockDriverState *bs) 107 { 108 BDRVNBDState *s = bs->opaque; 109 int sock; 110 int ret; 111 off_t size; 112 size_t blocksize; 113 uint32_t nbdflags; 114 115 if (s->host_spec[0] == '/') { 116 sock = unix_socket_outgoing(s->host_spec); 117 } else { 118 sock = tcp_socket_outgoing_spec(s->host_spec); 119 } 120 121 /* Failed to establish connection */ 122 if (sock == -1) { 123 logout("Failed to establish connection to NBD server\n"); 124 return -errno; 125 } 126 127 /* NBD handshake */ 128 ret = nbd_receive_negotiate(sock, s->export_name, &nbdflags, &size, 129 &blocksize); 130 if (ret == -1) { 131 logout("Failed to negotiate with the NBD server\n"); 132 closesocket(sock); 133 return -errno; 134 } 135 136 /* Now that we're connected, set the socket to be non-blocking */ 137 socket_set_nonblock(sock); 138 139 s->sock = sock; 140 s->size = size; 141 s->blocksize = blocksize; 142 143 logout("Established connection with NBD server\n"); 144 return 0; 145 } 146 147 static void nbd_teardown_connection(BlockDriverState *bs) 148 { 149 BDRVNBDState *s = bs->opaque; 150 struct nbd_request request; 151 152 request.type = NBD_CMD_DISC; 153 request.handle = (uint64_t)(intptr_t)bs; 154 request.from = 0; 155 request.len = 0; 156 nbd_send_request(s->sock, &request); 157 158 closesocket(s->sock); 159 } 160 161 static int nbd_open(BlockDriverState *bs, const char* filename, int flags) 162 { 163 BDRVNBDState *s = bs->opaque; 164 int result; 165 166 /* Pop the config into our state object. Exit if invalid. */ 167 result = nbd_config(s, filename, flags); 168 if (result != 0) { 169 return result; 170 } 171 172 /* establish TCP connection, return error if it fails 173 * TODO: Configurable retry-until-timeout behaviour. 174 */ 175 result = nbd_establish_connection(bs); 176 177 return result; 178 } 179 180 static int nbd_read(BlockDriverState *bs, int64_t sector_num, 181 uint8_t *buf, int nb_sectors) 182 { 183 BDRVNBDState *s = bs->opaque; 184 struct nbd_request request; 185 struct nbd_reply reply; 186 187 request.type = NBD_CMD_READ; 188 request.handle = (uint64_t)(intptr_t)bs; 189 request.from = sector_num * 512;; 190 request.len = nb_sectors * 512; 191 192 if (nbd_send_request(s->sock, &request) == -1) 193 return -errno; 194 195 if (nbd_receive_reply(s->sock, &reply) == -1) 196 return -errno; 197 198 if (reply.error !=0) 199 return -reply.error; 200 201 if (reply.handle != request.handle) 202 return -EIO; 203 204 if (nbd_wr_sync(s->sock, buf, request.len, 1) != request.len) 205 return -EIO; 206 207 return 0; 208 } 209 210 static int nbd_write(BlockDriverState *bs, int64_t sector_num, 211 const uint8_t *buf, int nb_sectors) 212 { 213 BDRVNBDState *s = bs->opaque; 214 struct nbd_request request; 215 struct nbd_reply reply; 216 217 request.type = NBD_CMD_WRITE; 218 request.handle = (uint64_t)(intptr_t)bs; 219 request.from = sector_num * 512;; 220 request.len = nb_sectors * 512; 221 222 if (nbd_send_request(s->sock, &request) == -1) 223 return -errno; 224 225 if (nbd_wr_sync(s->sock, (uint8_t*)buf, request.len, 0) != request.len) 226 return -EIO; 227 228 if (nbd_receive_reply(s->sock, &reply) == -1) 229 return -errno; 230 231 if (reply.error !=0) 232 return -reply.error; 233 234 if (reply.handle != request.handle) 235 return -EIO; 236 237 return 0; 238 } 239 240 static void nbd_close(BlockDriverState *bs) 241 { 242 BDRVNBDState *s = bs->opaque; 243 g_free(s->export_name); 244 g_free(s->host_spec); 245 246 nbd_teardown_connection(bs); 247 } 248 249 static int64_t nbd_getlength(BlockDriverState *bs) 250 { 251 BDRVNBDState *s = bs->opaque; 252 253 return s->size; 254 } 255 256 static BlockDriver bdrv_nbd = { 257 .format_name = "nbd", 258 .instance_size = sizeof(BDRVNBDState), 259 .bdrv_file_open = nbd_open, 260 .bdrv_read = nbd_read, 261 .bdrv_write = nbd_write, 262 .bdrv_close = nbd_close, 263 .bdrv_getlength = nbd_getlength, 264 .protocol_name = "nbd", 265 }; 266 267 static void bdrv_nbd_init(void) 268 { 269 bdrv_register(&bdrv_nbd); 270 } 271 272 block_init(bdrv_nbd_init); 273