1019d6b8fSAnthony Liguori /* 2019d6b8fSAnthony Liguori * QEMU Block driver for NBD 3019d6b8fSAnthony Liguori * 4019d6b8fSAnthony Liguori * Copyright (C) 2008 Bull S.A.S. 5019d6b8fSAnthony Liguori * Author: Laurent Vivier <Laurent.Vivier@bull.net> 6019d6b8fSAnthony Liguori * 7019d6b8fSAnthony Liguori * Some parts: 8019d6b8fSAnthony Liguori * Copyright (C) 2007 Anthony Liguori <anthony@codemonkey.ws> 9019d6b8fSAnthony Liguori * 10019d6b8fSAnthony Liguori * Permission is hereby granted, free of charge, to any person obtaining a copy 11019d6b8fSAnthony Liguori * of this software and associated documentation files (the "Software"), to deal 12019d6b8fSAnthony Liguori * in the Software without restriction, including without limitation the rights 13019d6b8fSAnthony Liguori * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 14019d6b8fSAnthony Liguori * copies of the Software, and to permit persons to whom the Software is 15019d6b8fSAnthony Liguori * furnished to do so, subject to the following conditions: 16019d6b8fSAnthony Liguori * 17019d6b8fSAnthony Liguori * The above copyright notice and this permission notice shall be included in 18019d6b8fSAnthony Liguori * all copies or substantial portions of the Software. 19019d6b8fSAnthony Liguori * 20019d6b8fSAnthony Liguori * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 21019d6b8fSAnthony Liguori * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 22019d6b8fSAnthony Liguori * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 23019d6b8fSAnthony Liguori * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 24019d6b8fSAnthony Liguori * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 25019d6b8fSAnthony Liguori * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 26019d6b8fSAnthony Liguori * THE SOFTWARE. 27019d6b8fSAnthony Liguori */ 28019d6b8fSAnthony Liguori 2980c71a24SPeter Maydell #include "qemu/osdep.h" 302302c1caSMarc-André Lureau #include "block/nbd-client.h" 311de7afc9SPaolo Bonzini #include "qemu/uri.h" 32737e150eSPaolo Bonzini #include "block/block_int.h" 331de7afc9SPaolo Bonzini #include "qemu/module.h" 342019d68bSMax Reitz #include "qapi/qmp/qdict.h" 35f53a1febSKevin Wolf #include "qapi/qmp/qjson.h" 36f53a1febSKevin Wolf #include "qapi/qmp/qint.h" 372019d68bSMax Reitz #include "qapi/qmp/qstring.h" 38019d6b8fSAnthony Liguori 39019d6b8fSAnthony Liguori 401d45f8b5SLaurent Vivier #define EN_OPTSTR ":exportname=" 411d45f8b5SLaurent Vivier 42019d6b8fSAnthony Liguori typedef struct BDRVNBDState { 432302c1caSMarc-André Lureau NbdClientSession client; 44019d6b8fSAnthony Liguori } BDRVNBDState; 45019d6b8fSAnthony Liguori 46f53a1febSKevin Wolf static int nbd_parse_uri(const char *filename, QDict *options) 471d7d2a9dSPaolo Bonzini { 481d7d2a9dSPaolo Bonzini URI *uri; 491d7d2a9dSPaolo Bonzini const char *p; 501d7d2a9dSPaolo Bonzini QueryParams *qp = NULL; 511d7d2a9dSPaolo Bonzini int ret = 0; 52f53a1febSKevin Wolf bool is_unix; 531d7d2a9dSPaolo Bonzini 541d7d2a9dSPaolo Bonzini uri = uri_parse(filename); 551d7d2a9dSPaolo Bonzini if (!uri) { 561d7d2a9dSPaolo Bonzini return -EINVAL; 571d7d2a9dSPaolo Bonzini } 581d7d2a9dSPaolo Bonzini 591d7d2a9dSPaolo Bonzini /* transport */ 601d7d2a9dSPaolo Bonzini if (!strcmp(uri->scheme, "nbd")) { 61f53a1febSKevin Wolf is_unix = false; 621d7d2a9dSPaolo Bonzini } else if (!strcmp(uri->scheme, "nbd+tcp")) { 63f53a1febSKevin Wolf is_unix = false; 641d7d2a9dSPaolo Bonzini } else if (!strcmp(uri->scheme, "nbd+unix")) { 65f53a1febSKevin Wolf is_unix = true; 661d7d2a9dSPaolo Bonzini } else { 671d7d2a9dSPaolo Bonzini ret = -EINVAL; 681d7d2a9dSPaolo Bonzini goto out; 691d7d2a9dSPaolo Bonzini } 701d7d2a9dSPaolo Bonzini 711d7d2a9dSPaolo Bonzini p = uri->path ? uri->path : "/"; 721d7d2a9dSPaolo Bonzini p += strspn(p, "/"); 731d7d2a9dSPaolo Bonzini if (p[0]) { 74f53a1febSKevin Wolf qdict_put(options, "export", qstring_from_str(p)); 751d7d2a9dSPaolo Bonzini } 761d7d2a9dSPaolo Bonzini 771d7d2a9dSPaolo Bonzini qp = query_params_parse(uri->query); 78f53a1febSKevin Wolf if (qp->n > 1 || (is_unix && !qp->n) || (!is_unix && qp->n)) { 791d7d2a9dSPaolo Bonzini ret = -EINVAL; 801d7d2a9dSPaolo Bonzini goto out; 811d7d2a9dSPaolo Bonzini } 821d7d2a9dSPaolo Bonzini 83f53a1febSKevin Wolf if (is_unix) { 841d7d2a9dSPaolo Bonzini /* nbd+unix:///export?socket=path */ 851d7d2a9dSPaolo Bonzini if (uri->server || uri->port || strcmp(qp->p[0].name, "socket")) { 861d7d2a9dSPaolo Bonzini ret = -EINVAL; 871d7d2a9dSPaolo Bonzini goto out; 881d7d2a9dSPaolo Bonzini } 89f53a1febSKevin Wolf qdict_put(options, "path", qstring_from_str(qp->p[0].value)); 901d7d2a9dSPaolo Bonzini } else { 9123307908SJán Tomko QString *host; 92bebbf7faSKevin Wolf /* nbd[+tcp]://host[:port]/export */ 931d7d2a9dSPaolo Bonzini if (!uri->server) { 941d7d2a9dSPaolo Bonzini ret = -EINVAL; 951d7d2a9dSPaolo Bonzini goto out; 961d7d2a9dSPaolo Bonzini } 97f17c90beSKevin Wolf 9823307908SJán Tomko /* strip braces from literal IPv6 address */ 9923307908SJán Tomko if (uri->server[0] == '[') { 10023307908SJán Tomko host = qstring_from_substr(uri->server, 1, 10123307908SJán Tomko strlen(uri->server) - 2); 10223307908SJán Tomko } else { 10323307908SJán Tomko host = qstring_from_str(uri->server); 10423307908SJán Tomko } 10523307908SJán Tomko 10623307908SJán Tomko qdict_put(options, "host", host); 107bebbf7faSKevin Wolf if (uri->port) { 108bebbf7faSKevin Wolf char* port_str = g_strdup_printf("%d", uri->port); 109f53a1febSKevin Wolf qdict_put(options, "port", qstring_from_str(port_str)); 110f53a1febSKevin Wolf g_free(port_str); 1111d7d2a9dSPaolo Bonzini } 112bebbf7faSKevin Wolf } 1131d7d2a9dSPaolo Bonzini 1141d7d2a9dSPaolo Bonzini out: 1151d7d2a9dSPaolo Bonzini if (qp) { 1161d7d2a9dSPaolo Bonzini query_params_free(qp); 1171d7d2a9dSPaolo Bonzini } 1181d7d2a9dSPaolo Bonzini uri_free(uri); 1191d7d2a9dSPaolo Bonzini return ret; 1201d7d2a9dSPaolo Bonzini } 1211d7d2a9dSPaolo Bonzini 1226963a30dSKevin Wolf static void nbd_parse_filename(const char *filename, QDict *options, 1236963a30dSKevin Wolf Error **errp) 124019d6b8fSAnthony Liguori { 1251d45f8b5SLaurent Vivier char *file; 12633897dc7SNick Thomas char *export_name; 12733897dc7SNick Thomas const char *host_spec; 128019d6b8fSAnthony Liguori const char *unixpath; 129019d6b8fSAnthony Liguori 130681e7ad0SKevin Wolf if (qdict_haskey(options, "host") 131681e7ad0SKevin Wolf || qdict_haskey(options, "port") 132681e7ad0SKevin Wolf || qdict_haskey(options, "path")) 133681e7ad0SKevin Wolf { 134681e7ad0SKevin Wolf error_setg(errp, "host/port/path and a file name may not be specified " 135681e7ad0SKevin Wolf "at the same time"); 136681e7ad0SKevin Wolf return; 137681e7ad0SKevin Wolf } 138681e7ad0SKevin Wolf 1391d7d2a9dSPaolo Bonzini if (strstr(filename, "://")) { 1406963a30dSKevin Wolf int ret = nbd_parse_uri(filename, options); 1416963a30dSKevin Wolf if (ret < 0) { 1426963a30dSKevin Wolf error_setg(errp, "No valid URL specified"); 1436963a30dSKevin Wolf } 1446963a30dSKevin Wolf return; 1451d7d2a9dSPaolo Bonzini } 1461d7d2a9dSPaolo Bonzini 1477267c094SAnthony Liguori file = g_strdup(filename); 1481d45f8b5SLaurent Vivier 14933897dc7SNick Thomas export_name = strstr(file, EN_OPTSTR); 15033897dc7SNick Thomas if (export_name) { 15133897dc7SNick Thomas if (export_name[strlen(EN_OPTSTR)] == 0) { 1521d45f8b5SLaurent Vivier goto out; 1531d45f8b5SLaurent Vivier } 15433897dc7SNick Thomas export_name[0] = 0; /* truncate 'file' */ 15533897dc7SNick Thomas export_name += strlen(EN_OPTSTR); 156f53a1febSKevin Wolf 157f53a1febSKevin Wolf qdict_put(options, "export", qstring_from_str(export_name)); 1581d45f8b5SLaurent Vivier } 1591d45f8b5SLaurent Vivier 16033897dc7SNick Thomas /* extract the host_spec - fail if it's not nbd:... */ 16133897dc7SNick Thomas if (!strstart(file, "nbd:", &host_spec)) { 1626963a30dSKevin Wolf error_setg(errp, "File name string for NBD must start with 'nbd:'"); 1631d45f8b5SLaurent Vivier goto out; 1641d45f8b5SLaurent Vivier } 165019d6b8fSAnthony Liguori 166f53a1febSKevin Wolf if (!*host_spec) { 167f53a1febSKevin Wolf goto out; 168f53a1febSKevin Wolf } 169f53a1febSKevin Wolf 17033897dc7SNick Thomas /* are we a UNIX or TCP socket? */ 17133897dc7SNick Thomas if (strstart(host_spec, "unix:", &unixpath)) { 172f53a1febSKevin Wolf qdict_put(options, "path", qstring_from_str(unixpath)); 173019d6b8fSAnthony Liguori } else { 174f53a1febSKevin Wolf InetSocketAddress *addr = NULL; 175f53a1febSKevin Wolf 1766963a30dSKevin Wolf addr = inet_parse(host_spec, errp); 17792de9012SMarkus Armbruster if (!addr) { 178f17c90beSKevin Wolf goto out; 179f17c90beSKevin Wolf } 180f53a1febSKevin Wolf 181f53a1febSKevin Wolf qdict_put(options, "host", qstring_from_str(addr->host)); 182f53a1febSKevin Wolf qdict_put(options, "port", qstring_from_str(addr->port)); 183f53a1febSKevin Wolf qapi_free_InetSocketAddress(addr); 1841d45f8b5SLaurent Vivier } 1851d45f8b5SLaurent Vivier 1861d45f8b5SLaurent Vivier out: 1877267c094SAnthony Liguori g_free(file); 18833897dc7SNick Thomas } 189f53a1febSKevin Wolf 1907a5ed437SDaniel P. Berrange static SocketAddress *nbd_config(BDRVNBDState *s, QDict *options, char **export, 19177e8b9caSPaolo Bonzini Error **errp) 192f53a1febSKevin Wolf { 1937a5ed437SDaniel P. Berrange SocketAddress *saddr; 194f53a1febSKevin Wolf 195a69d9af4SPaolo Bonzini if (qdict_haskey(options, "path") == qdict_haskey(options, "host")) { 196f53a1febSKevin Wolf if (qdict_haskey(options, "path")) { 19777e8b9caSPaolo Bonzini error_setg(errp, "path and host may not be used at the same time."); 198f53a1febSKevin Wolf } else { 19977e8b9caSPaolo Bonzini error_setg(errp, "one of path and host must be specified."); 200a69d9af4SPaolo Bonzini } 2017a5ed437SDaniel P. Berrange return NULL; 202019d6b8fSAnthony Liguori } 203019d6b8fSAnthony Liguori 2047a5ed437SDaniel P. Berrange saddr = g_new0(SocketAddress, 1); 205f53a1febSKevin Wolf 2067a5ed437SDaniel P. Berrange if (qdict_haskey(options, "path")) { 2070399293eSEric Blake UnixSocketAddress *q_unix; 2086a8f9661SEric Blake saddr->type = SOCKET_ADDRESS_KIND_UNIX; 209*32bafa8fSEric Blake q_unix = saddr->u.q_unix.data = g_new0(UnixSocketAddress, 1); 2100399293eSEric Blake q_unix->path = g_strdup(qdict_get_str(options, "path")); 2117a5ed437SDaniel P. Berrange qdict_del(options, "path"); 2127a5ed437SDaniel P. Berrange } else { 2130399293eSEric Blake InetSocketAddress *inet; 2146a8f9661SEric Blake saddr->type = SOCKET_ADDRESS_KIND_INET; 215*32bafa8fSEric Blake inet = saddr->u.inet.data = g_new0(InetSocketAddress, 1); 2160399293eSEric Blake inet->host = g_strdup(qdict_get_str(options, "host")); 2177a5ed437SDaniel P. Berrange if (!qdict_get_try_str(options, "port")) { 2180399293eSEric Blake inet->port = g_strdup_printf("%d", NBD_DEFAULT_PORT); 2197a5ed437SDaniel P. Berrange } else { 2200399293eSEric Blake inet->port = g_strdup(qdict_get_str(options, "port")); 2217a5ed437SDaniel P. Berrange } 2227a5ed437SDaniel P. Berrange qdict_del(options, "host"); 2237a5ed437SDaniel P. Berrange qdict_del(options, "port"); 224f53a1febSKevin Wolf } 225f53a1febSKevin Wolf 2266a8f9661SEric Blake s->client.is_unix = saddr->type == SOCKET_ADDRESS_KIND_UNIX; 227bebbf7faSKevin Wolf 228e2bc625fSMarc-André Lureau *export = g_strdup(qdict_get_try_str(options, "export")); 229e2bc625fSMarc-André Lureau if (*export) { 230f53a1febSKevin Wolf qdict_del(options, "export"); 231f53a1febSKevin Wolf } 2327a5ed437SDaniel P. Berrange 2337a5ed437SDaniel P. Berrange return saddr; 234f53a1febSKevin Wolf } 235f53a1febSKevin Wolf 236f53a829bSMax Reitz NbdClientSession *nbd_get_client_session(BlockDriverState *bs) 237f53a829bSMax Reitz { 238f53a829bSMax Reitz BDRVNBDState *s = bs->opaque; 239f53a829bSMax Reitz return &s->client; 240f53a829bSMax Reitz } 241f53a829bSMax Reitz 242064097d9SDaniel P. Berrange static QIOChannelSocket *nbd_establish_connection(SocketAddress *saddr, 2437a5ed437SDaniel P. Berrange Error **errp) 24433897dc7SNick Thomas { 245064097d9SDaniel P. Berrange QIOChannelSocket *sioc; 246064097d9SDaniel P. Berrange Error *local_err = NULL; 24733897dc7SNick Thomas 248064097d9SDaniel P. Berrange sioc = qio_channel_socket_new(); 24933897dc7SNick Thomas 250064097d9SDaniel P. Berrange qio_channel_socket_connect_sync(sioc, 251064097d9SDaniel P. Berrange saddr, 252064097d9SDaniel P. Berrange &local_err); 253064097d9SDaniel P. Berrange if (local_err) { 254064097d9SDaniel P. Berrange error_propagate(errp, local_err); 255064097d9SDaniel P. Berrange return NULL; 25633897dc7SNick Thomas } 25733897dc7SNick Thomas 258064097d9SDaniel P. Berrange qio_channel_set_delay(QIO_CHANNEL(sioc), false); 2597a5ed437SDaniel P. Berrange 260064097d9SDaniel P. Berrange return sioc; 26133897dc7SNick Thomas } 26233897dc7SNick Thomas 26375822a12SDaniel P. Berrange 26475822a12SDaniel P. Berrange static QCryptoTLSCreds *nbd_get_tls_creds(const char *id, Error **errp) 26575822a12SDaniel P. Berrange { 26675822a12SDaniel P. Berrange Object *obj; 26775822a12SDaniel P. Berrange QCryptoTLSCreds *creds; 26875822a12SDaniel P. Berrange 26975822a12SDaniel P. Berrange obj = object_resolve_path_component( 27075822a12SDaniel P. Berrange object_get_objects_root(), id); 27175822a12SDaniel P. Berrange if (!obj) { 27275822a12SDaniel P. Berrange error_setg(errp, "No TLS credentials with id '%s'", 27375822a12SDaniel P. Berrange id); 27475822a12SDaniel P. Berrange return NULL; 27575822a12SDaniel P. Berrange } 27675822a12SDaniel P. Berrange creds = (QCryptoTLSCreds *) 27775822a12SDaniel P. Berrange object_dynamic_cast(obj, TYPE_QCRYPTO_TLS_CREDS); 27875822a12SDaniel P. Berrange if (!creds) { 27975822a12SDaniel P. Berrange error_setg(errp, "Object with id '%s' is not TLS credentials", 28075822a12SDaniel P. Berrange id); 28175822a12SDaniel P. Berrange return NULL; 28275822a12SDaniel P. Berrange } 28375822a12SDaniel P. Berrange 28475822a12SDaniel P. Berrange if (creds->endpoint != QCRYPTO_TLS_CREDS_ENDPOINT_CLIENT) { 28575822a12SDaniel P. Berrange error_setg(errp, 28675822a12SDaniel P. Berrange "Expecting TLS credentials with a client endpoint"); 28775822a12SDaniel P. Berrange return NULL; 28875822a12SDaniel P. Berrange } 28975822a12SDaniel P. Berrange object_ref(obj); 29075822a12SDaniel P. Berrange return creds; 29175822a12SDaniel P. Berrange } 29275822a12SDaniel P. Berrange 29375822a12SDaniel P. Berrange 294015a1036SMax Reitz static int nbd_open(BlockDriverState *bs, QDict *options, int flags, 295015a1036SMax Reitz Error **errp) 29633897dc7SNick Thomas { 29733897dc7SNick Thomas BDRVNBDState *s = bs->opaque; 298e2bc625fSMarc-André Lureau char *export = NULL; 29975822a12SDaniel P. Berrange QIOChannelSocket *sioc = NULL; 3007a5ed437SDaniel P. Berrange SocketAddress *saddr; 30175822a12SDaniel P. Berrange const char *tlscredsid; 30275822a12SDaniel P. Berrange QCryptoTLSCreds *tlscreds = NULL; 30375822a12SDaniel P. Berrange const char *hostname = NULL; 30475822a12SDaniel P. Berrange int ret = -EINVAL; 305ae255e52SPaolo Bonzini 30633897dc7SNick Thomas /* Pop the config into our state object. Exit if invalid. */ 3077a5ed437SDaniel P. Berrange saddr = nbd_config(s, options, &export, errp); 3087a5ed437SDaniel P. Berrange if (!saddr) { 30975822a12SDaniel P. Berrange goto error; 31075822a12SDaniel P. Berrange } 31175822a12SDaniel P. Berrange 31275822a12SDaniel P. Berrange tlscredsid = g_strdup(qdict_get_try_str(options, "tls-creds")); 31375822a12SDaniel P. Berrange if (tlscredsid) { 31475822a12SDaniel P. Berrange qdict_del(options, "tls-creds"); 31575822a12SDaniel P. Berrange tlscreds = nbd_get_tls_creds(tlscredsid, errp); 31675822a12SDaniel P. Berrange if (!tlscreds) { 31775822a12SDaniel P. Berrange goto error; 31875822a12SDaniel P. Berrange } 31975822a12SDaniel P. Berrange 32075822a12SDaniel P. Berrange if (saddr->type != SOCKET_ADDRESS_KIND_INET) { 32175822a12SDaniel P. Berrange error_setg(errp, "TLS only supported over IP sockets"); 32275822a12SDaniel P. Berrange goto error; 32375822a12SDaniel P. Berrange } 324*32bafa8fSEric Blake hostname = saddr->u.inet.data->host; 32533897dc7SNick Thomas } 32633897dc7SNick Thomas 32733897dc7SNick Thomas /* establish TCP connection, return error if it fails 32833897dc7SNick Thomas * TODO: Configurable retry-until-timeout behaviour. 32933897dc7SNick Thomas */ 330064097d9SDaniel P. Berrange sioc = nbd_establish_connection(saddr, errp); 331064097d9SDaniel P. Berrange if (!sioc) { 33275822a12SDaniel P. Berrange ret = -ECONNREFUSED; 33375822a12SDaniel P. Berrange goto error; 33433897dc7SNick Thomas } 33533897dc7SNick Thomas 3362302c1caSMarc-André Lureau /* NBD handshake */ 33775822a12SDaniel P. Berrange ret = nbd_client_init(bs, sioc, export, 33875822a12SDaniel P. Berrange tlscreds, hostname, errp); 33975822a12SDaniel P. Berrange error: 34075822a12SDaniel P. Berrange if (sioc) { 341064097d9SDaniel P. Berrange object_unref(OBJECT(sioc)); 34275822a12SDaniel P. Berrange } 34375822a12SDaniel P. Berrange if (tlscreds) { 34475822a12SDaniel P. Berrange object_unref(OBJECT(tlscreds)); 34575822a12SDaniel P. Berrange } 34675822a12SDaniel P. Berrange qapi_free_SocketAddress(saddr); 347e2bc625fSMarc-André Lureau g_free(export); 34875822a12SDaniel P. Berrange return ret; 349ae255e52SPaolo Bonzini } 350d9b09f13SPaolo Bonzini 351d9b09f13SPaolo Bonzini static int nbd_co_readv(BlockDriverState *bs, int64_t sector_num, 352d9b09f13SPaolo Bonzini int nb_sectors, QEMUIOVector *qiov) 353d9b09f13SPaolo Bonzini { 354f53a829bSMax Reitz return nbd_client_co_readv(bs, sector_num, nb_sectors, qiov); 355d9b09f13SPaolo Bonzini } 356d9b09f13SPaolo Bonzini 357d9b09f13SPaolo Bonzini static int nbd_co_writev(BlockDriverState *bs, int64_t sector_num, 358d9b09f13SPaolo Bonzini int nb_sectors, QEMUIOVector *qiov) 359d9b09f13SPaolo Bonzini { 360f53a829bSMax Reitz return nbd_client_co_writev(bs, sector_num, nb_sectors, qiov); 361d9b09f13SPaolo Bonzini } 362d9b09f13SPaolo Bonzini 3631486d04aSPaolo Bonzini static int nbd_co_flush(BlockDriverState *bs) 3641486d04aSPaolo Bonzini { 365f53a829bSMax Reitz return nbd_client_co_flush(bs); 3661486d04aSPaolo Bonzini } 3671486d04aSPaolo Bonzini 368fa21e6faSDenis V. Lunev static void nbd_refresh_limits(BlockDriverState *bs, Error **errp) 369fa21e6faSDenis V. Lunev { 370fa21e6faSDenis V. Lunev bs->bl.max_discard = UINT32_MAX >> BDRV_SECTOR_BITS; 371fa21e6faSDenis V. Lunev bs->bl.max_transfer_length = UINT32_MAX >> BDRV_SECTOR_BITS; 372fa21e6faSDenis V. Lunev } 373fa21e6faSDenis V. Lunev 3747a706633SPaolo Bonzini static int nbd_co_discard(BlockDriverState *bs, int64_t sector_num, 3757a706633SPaolo Bonzini int nb_sectors) 3767a706633SPaolo Bonzini { 377f53a829bSMax Reitz return nbd_client_co_discard(bs, sector_num, nb_sectors); 3787a706633SPaolo Bonzini } 3797a706633SPaolo Bonzini 380019d6b8fSAnthony Liguori static void nbd_close(BlockDriverState *bs) 381019d6b8fSAnthony Liguori { 382f53a829bSMax Reitz nbd_client_close(bs); 383019d6b8fSAnthony Liguori } 384019d6b8fSAnthony Liguori 385019d6b8fSAnthony Liguori static int64_t nbd_getlength(BlockDriverState *bs) 386019d6b8fSAnthony Liguori { 387019d6b8fSAnthony Liguori BDRVNBDState *s = bs->opaque; 388019d6b8fSAnthony Liguori 3892302c1caSMarc-André Lureau return s->client.size; 390019d6b8fSAnthony Liguori } 391019d6b8fSAnthony Liguori 39269447cd8SStefan Hajnoczi static void nbd_detach_aio_context(BlockDriverState *bs) 39369447cd8SStefan Hajnoczi { 394f53a829bSMax Reitz nbd_client_detach_aio_context(bs); 39569447cd8SStefan Hajnoczi } 39669447cd8SStefan Hajnoczi 39769447cd8SStefan Hajnoczi static void nbd_attach_aio_context(BlockDriverState *bs, 39869447cd8SStefan Hajnoczi AioContext *new_context) 39969447cd8SStefan Hajnoczi { 400f53a829bSMax Reitz nbd_client_attach_aio_context(bs, new_context); 40169447cd8SStefan Hajnoczi } 40269447cd8SStefan Hajnoczi 4034cdd01d3SKevin Wolf static void nbd_refresh_filename(BlockDriverState *bs, QDict *options) 4042019d68bSMax Reitz { 4052019d68bSMax Reitz QDict *opts = qdict_new(); 4064cdd01d3SKevin Wolf const char *path = qdict_get_try_str(options, "path"); 4074cdd01d3SKevin Wolf const char *host = qdict_get_try_str(options, "host"); 4084cdd01d3SKevin Wolf const char *port = qdict_get_try_str(options, "port"); 4094cdd01d3SKevin Wolf const char *export = qdict_get_try_str(options, "export"); 41075822a12SDaniel P. Berrange const char *tlscreds = qdict_get_try_str(options, "tls-creds"); 4112019d68bSMax Reitz 4122019d68bSMax Reitz qdict_put_obj(opts, "driver", QOBJECT(qstring_from_str("nbd"))); 4132019d68bSMax Reitz 414ec0de768SMax Reitz if (path && export) { 415ec0de768SMax Reitz snprintf(bs->exact_filename, sizeof(bs->exact_filename), 416ec0de768SMax Reitz "nbd+unix:///%s?socket=%s", export, path); 417ec0de768SMax Reitz } else if (path && !export) { 418ec0de768SMax Reitz snprintf(bs->exact_filename, sizeof(bs->exact_filename), 419ec0de768SMax Reitz "nbd+unix://?socket=%s", path); 420ec0de768SMax Reitz } else if (!path && export && port) { 421ec0de768SMax Reitz snprintf(bs->exact_filename, sizeof(bs->exact_filename), 422ec0de768SMax Reitz "nbd://%s:%s/%s", host, port, export); 423ec0de768SMax Reitz } else if (!path && export && !port) { 424ec0de768SMax Reitz snprintf(bs->exact_filename, sizeof(bs->exact_filename), 425ec0de768SMax Reitz "nbd://%s/%s", host, export); 426ec0de768SMax Reitz } else if (!path && !export && port) { 427ec0de768SMax Reitz snprintf(bs->exact_filename, sizeof(bs->exact_filename), 428ec0de768SMax Reitz "nbd://%s:%s", host, port); 429ec0de768SMax Reitz } else if (!path && !export && !port) { 430ec0de768SMax Reitz snprintf(bs->exact_filename, sizeof(bs->exact_filename), 431ec0de768SMax Reitz "nbd://%s", host); 432ec0de768SMax Reitz } 433ec0de768SMax Reitz 4342019d68bSMax Reitz if (path) { 4352019d68bSMax Reitz qdict_put_obj(opts, "path", QOBJECT(qstring_from_str(path))); 436ec0de768SMax Reitz } else if (port) { 4372019d68bSMax Reitz qdict_put_obj(opts, "host", QOBJECT(qstring_from_str(host))); 4382019d68bSMax Reitz qdict_put_obj(opts, "port", QOBJECT(qstring_from_str(port))); 4392019d68bSMax Reitz } else { 4402019d68bSMax Reitz qdict_put_obj(opts, "host", QOBJECT(qstring_from_str(host))); 441ec0de768SMax Reitz } 442ec0de768SMax Reitz if (export) { 443ec0de768SMax Reitz qdict_put_obj(opts, "export", QOBJECT(qstring_from_str(export))); 4442019d68bSMax Reitz } 44575822a12SDaniel P. Berrange if (tlscreds) { 44675822a12SDaniel P. Berrange qdict_put_obj(opts, "tls-creds", QOBJECT(qstring_from_str(tlscreds))); 44775822a12SDaniel P. Berrange } 4482019d68bSMax Reitz 4492019d68bSMax Reitz bs->full_open_options = opts; 4502019d68bSMax Reitz } 4512019d68bSMax Reitz 452019d6b8fSAnthony Liguori static BlockDriver bdrv_nbd = { 453019d6b8fSAnthony Liguori .format_name = "nbd", 4541d7d2a9dSPaolo Bonzini .protocol_name = "nbd", 455019d6b8fSAnthony Liguori .instance_size = sizeof(BDRVNBDState), 4566963a30dSKevin Wolf .bdrv_parse_filename = nbd_parse_filename, 45766f82ceeSKevin Wolf .bdrv_file_open = nbd_open, 458ae255e52SPaolo Bonzini .bdrv_co_readv = nbd_co_readv, 459ae255e52SPaolo Bonzini .bdrv_co_writev = nbd_co_writev, 460019d6b8fSAnthony Liguori .bdrv_close = nbd_close, 4611486d04aSPaolo Bonzini .bdrv_co_flush_to_os = nbd_co_flush, 4627a706633SPaolo Bonzini .bdrv_co_discard = nbd_co_discard, 463fa21e6faSDenis V. Lunev .bdrv_refresh_limits = nbd_refresh_limits, 464019d6b8fSAnthony Liguori .bdrv_getlength = nbd_getlength, 46569447cd8SStefan Hajnoczi .bdrv_detach_aio_context = nbd_detach_aio_context, 46669447cd8SStefan Hajnoczi .bdrv_attach_aio_context = nbd_attach_aio_context, 4672019d68bSMax Reitz .bdrv_refresh_filename = nbd_refresh_filename, 4681d7d2a9dSPaolo Bonzini }; 4691d7d2a9dSPaolo Bonzini 4701d7d2a9dSPaolo Bonzini static BlockDriver bdrv_nbd_tcp = { 4711d7d2a9dSPaolo Bonzini .format_name = "nbd", 4721d7d2a9dSPaolo Bonzini .protocol_name = "nbd+tcp", 4731d7d2a9dSPaolo Bonzini .instance_size = sizeof(BDRVNBDState), 4746963a30dSKevin Wolf .bdrv_parse_filename = nbd_parse_filename, 4751d7d2a9dSPaolo Bonzini .bdrv_file_open = nbd_open, 4761d7d2a9dSPaolo Bonzini .bdrv_co_readv = nbd_co_readv, 4771d7d2a9dSPaolo Bonzini .bdrv_co_writev = nbd_co_writev, 4781d7d2a9dSPaolo Bonzini .bdrv_close = nbd_close, 4791d7d2a9dSPaolo Bonzini .bdrv_co_flush_to_os = nbd_co_flush, 4801d7d2a9dSPaolo Bonzini .bdrv_co_discard = nbd_co_discard, 481fa21e6faSDenis V. Lunev .bdrv_refresh_limits = nbd_refresh_limits, 4821d7d2a9dSPaolo Bonzini .bdrv_getlength = nbd_getlength, 48369447cd8SStefan Hajnoczi .bdrv_detach_aio_context = nbd_detach_aio_context, 48469447cd8SStefan Hajnoczi .bdrv_attach_aio_context = nbd_attach_aio_context, 4852019d68bSMax Reitz .bdrv_refresh_filename = nbd_refresh_filename, 4861d7d2a9dSPaolo Bonzini }; 4871d7d2a9dSPaolo Bonzini 4881d7d2a9dSPaolo Bonzini static BlockDriver bdrv_nbd_unix = { 4891d7d2a9dSPaolo Bonzini .format_name = "nbd", 4901d7d2a9dSPaolo Bonzini .protocol_name = "nbd+unix", 4911d7d2a9dSPaolo Bonzini .instance_size = sizeof(BDRVNBDState), 4926963a30dSKevin Wolf .bdrv_parse_filename = nbd_parse_filename, 4931d7d2a9dSPaolo Bonzini .bdrv_file_open = nbd_open, 4941d7d2a9dSPaolo Bonzini .bdrv_co_readv = nbd_co_readv, 4951d7d2a9dSPaolo Bonzini .bdrv_co_writev = nbd_co_writev, 4961d7d2a9dSPaolo Bonzini .bdrv_close = nbd_close, 4971d7d2a9dSPaolo Bonzini .bdrv_co_flush_to_os = nbd_co_flush, 4981d7d2a9dSPaolo Bonzini .bdrv_co_discard = nbd_co_discard, 499fa21e6faSDenis V. Lunev .bdrv_refresh_limits = nbd_refresh_limits, 5001d7d2a9dSPaolo Bonzini .bdrv_getlength = nbd_getlength, 50169447cd8SStefan Hajnoczi .bdrv_detach_aio_context = nbd_detach_aio_context, 50269447cd8SStefan Hajnoczi .bdrv_attach_aio_context = nbd_attach_aio_context, 5032019d68bSMax Reitz .bdrv_refresh_filename = nbd_refresh_filename, 504019d6b8fSAnthony Liguori }; 505019d6b8fSAnthony Liguori 506019d6b8fSAnthony Liguori static void bdrv_nbd_init(void) 507019d6b8fSAnthony Liguori { 508019d6b8fSAnthony Liguori bdrv_register(&bdrv_nbd); 5091d7d2a9dSPaolo Bonzini bdrv_register(&bdrv_nbd_tcp); 5101d7d2a9dSPaolo Bonzini bdrv_register(&bdrv_nbd_unix); 511019d6b8fSAnthony Liguori } 512019d6b8fSAnthony Liguori 513019d6b8fSAnthony Liguori block_init(bdrv_nbd_init); 514