nbd/server: use errp instead of LOG
Move to modern errp scheme from just LOGging errors. Signed-off-by: Vladimir Sementsov-Ogievskiy <vsementsov@virtuozzo.com> Message-Id: <20170707152918.23086-4-vsementsov@virtuozzo.com> Signed-off-by: Eric Blake <eblake@redhat.com>
This commit is contained in:
		
							parent
							
								
									76ff081d91
								
							
						
					
					
						commit
						2fd2c8407e
					
				
							
								
								
									
										266
									
								
								nbd/server.c
								
								
								
								
							
							
						
						
									
										266
									
								
								nbd/server.c
								
								
								
								
							| 
						 | 
				
			
			@ -134,7 +134,7 @@ static void nbd_client_receive_next_request(NBDClient *client);
 | 
			
		|||
/* Send a reply header, including length, but no payload.
 | 
			
		||||
 * Return -errno on error, 0 on success. */
 | 
			
		||||
static int nbd_negotiate_send_rep_len(QIOChannel *ioc, uint32_t type,
 | 
			
		||||
                                      uint32_t opt, uint32_t len)
 | 
			
		||||
                                      uint32_t opt, uint32_t len, Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    uint64_t magic;
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -142,23 +142,26 @@ static int nbd_negotiate_send_rep_len(QIOChannel *ioc, uint32_t type,
 | 
			
		|||
          type, opt, len);
 | 
			
		||||
 | 
			
		||||
    magic = cpu_to_be64(NBD_REP_MAGIC);
 | 
			
		||||
    if (nbd_write(ioc, &magic, sizeof(magic), NULL) < 0) {
 | 
			
		||||
        LOG("write failed (rep magic)");
 | 
			
		||||
    if (nbd_write(ioc, &magic, sizeof(magic), errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (rep magic): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    opt = cpu_to_be32(opt);
 | 
			
		||||
    if (nbd_write(ioc, &opt, sizeof(opt), NULL) < 0) {
 | 
			
		||||
        LOG("write failed (rep opt)");
 | 
			
		||||
    if (nbd_write(ioc, &opt, sizeof(opt), errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (rep opt): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    type = cpu_to_be32(type);
 | 
			
		||||
    if (nbd_write(ioc, &type, sizeof(type), NULL) < 0) {
 | 
			
		||||
        LOG("write failed (rep type)");
 | 
			
		||||
    if (nbd_write(ioc, &type, sizeof(type), errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (rep type): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    len = cpu_to_be32(len);
 | 
			
		||||
    if (nbd_write(ioc, &len, sizeof(len), NULL) < 0) {
 | 
			
		||||
        LOG("write failed (rep data length)");
 | 
			
		||||
    if (nbd_write(ioc, &len, sizeof(len), errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (rep data length): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    return 0;
 | 
			
		||||
| 
						 | 
				
			
			@ -166,16 +169,17 @@ static int nbd_negotiate_send_rep_len(QIOChannel *ioc, uint32_t type,
 | 
			
		|||
 | 
			
		||||
/* Send a reply header with default 0 length.
 | 
			
		||||
 * Return -errno on error, 0 on success. */
 | 
			
		||||
static int nbd_negotiate_send_rep(QIOChannel *ioc, uint32_t type, uint32_t opt)
 | 
			
		||||
static int nbd_negotiate_send_rep(QIOChannel *ioc, uint32_t type, uint32_t opt,
 | 
			
		||||
                                  Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    return nbd_negotiate_send_rep_len(ioc, type, opt, 0);
 | 
			
		||||
    return nbd_negotiate_send_rep_len(ioc, type, opt, 0, errp);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/* Send an error reply.
 | 
			
		||||
 * Return -errno on error, 0 on success. */
 | 
			
		||||
static int GCC_FMT_ATTR(4, 5)
 | 
			
		||||
static int GCC_FMT_ATTR(5, 6)
 | 
			
		||||
nbd_negotiate_send_rep_err(QIOChannel *ioc, uint32_t type,
 | 
			
		||||
                           uint32_t opt, const char *fmt, ...)
 | 
			
		||||
                           uint32_t opt, Error **errp, const char *fmt, ...)
 | 
			
		||||
{
 | 
			
		||||
    va_list va;
 | 
			
		||||
    char *msg;
 | 
			
		||||
| 
						 | 
				
			
			@ -188,16 +192,17 @@ nbd_negotiate_send_rep_err(QIOChannel *ioc, uint32_t type,
 | 
			
		|||
    len = strlen(msg);
 | 
			
		||||
    assert(len < 4096);
 | 
			
		||||
    TRACE("sending error message \"%s\"", msg);
 | 
			
		||||
    ret = nbd_negotiate_send_rep_len(ioc, type, opt, len);
 | 
			
		||||
    ret = nbd_negotiate_send_rep_len(ioc, type, opt, len, errp);
 | 
			
		||||
    if (ret < 0) {
 | 
			
		||||
        goto out;
 | 
			
		||||
    }
 | 
			
		||||
    if (nbd_write(ioc, msg, len, NULL) < 0) {
 | 
			
		||||
        LOG("write failed (error message)");
 | 
			
		||||
    if (nbd_write(ioc, msg, len, errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (error message): ");
 | 
			
		||||
        ret = -EIO;
 | 
			
		||||
    } else {
 | 
			
		||||
        ret = 0;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
out:
 | 
			
		||||
    g_free(msg);
 | 
			
		||||
    return ret;
 | 
			
		||||
| 
						 | 
				
			
			@ -205,7 +210,8 @@ out:
 | 
			
		|||
 | 
			
		||||
/* Send a single NBD_REP_SERVER reply to NBD_OPT_LIST, including payload.
 | 
			
		||||
 * Return -errno on error, 0 on success. */
 | 
			
		||||
static int nbd_negotiate_send_rep_list(QIOChannel *ioc, NBDExport *exp)
 | 
			
		||||
static int nbd_negotiate_send_rep_list(QIOChannel *ioc, NBDExport *exp,
 | 
			
		||||
                                       Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    size_t name_len, desc_len;
 | 
			
		||||
    uint32_t len;
 | 
			
		||||
| 
						 | 
				
			
			@ -217,53 +223,60 @@ static int nbd_negotiate_send_rep_list(QIOChannel *ioc, NBDExport *exp)
 | 
			
		|||
    name_len = strlen(name);
 | 
			
		||||
    desc_len = strlen(desc);
 | 
			
		||||
    len = name_len + desc_len + sizeof(len);
 | 
			
		||||
    ret = nbd_negotiate_send_rep_len(ioc, NBD_REP_SERVER, NBD_OPT_LIST, len);
 | 
			
		||||
    ret = nbd_negotiate_send_rep_len(ioc, NBD_REP_SERVER, NBD_OPT_LIST, len,
 | 
			
		||||
                                     errp);
 | 
			
		||||
    if (ret < 0) {
 | 
			
		||||
        return ret;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    len = cpu_to_be32(name_len);
 | 
			
		||||
    if (nbd_write(ioc, &len, sizeof(len), NULL) < 0) {
 | 
			
		||||
        LOG("write failed (name length)");
 | 
			
		||||
    if (nbd_write(ioc, &len, sizeof(len), errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (name length): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    if (nbd_write(ioc, name, name_len, NULL) < 0) {
 | 
			
		||||
        LOG("write failed (name buffer)");
 | 
			
		||||
 | 
			
		||||
    if (nbd_write(ioc, name, name_len, errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (name buffer): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    if (nbd_write(ioc, desc, desc_len, NULL) < 0) {
 | 
			
		||||
        LOG("write failed (description buffer)");
 | 
			
		||||
 | 
			
		||||
    if (nbd_write(ioc, desc, desc_len, errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "write failed (description buffer): ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/* Process the NBD_OPT_LIST command, with a potential series of replies.
 | 
			
		||||
 * Return -errno on error, 0 on success. */
 | 
			
		||||
static int nbd_negotiate_handle_list(NBDClient *client, uint32_t length)
 | 
			
		||||
static int nbd_negotiate_handle_list(NBDClient *client, uint32_t length,
 | 
			
		||||
                                     Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    NBDExport *exp;
 | 
			
		||||
 | 
			
		||||
    if (length) {
 | 
			
		||||
        if (nbd_drop(client->ioc, length, NULL) < 0) {
 | 
			
		||||
        if (nbd_drop(client->ioc, length, errp) < 0) {
 | 
			
		||||
            return -EIO;
 | 
			
		||||
        }
 | 
			
		||||
        return nbd_negotiate_send_rep_err(client->ioc,
 | 
			
		||||
                                          NBD_REP_ERR_INVALID, NBD_OPT_LIST,
 | 
			
		||||
                                          errp,
 | 
			
		||||
                                          "OPT_LIST should not have length");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /* For each export, send a NBD_REP_SERVER reply. */
 | 
			
		||||
    QTAILQ_FOREACH(exp, &exports, next) {
 | 
			
		||||
        if (nbd_negotiate_send_rep_list(client->ioc, exp)) {
 | 
			
		||||
        if (nbd_negotiate_send_rep_list(client->ioc, exp, errp)) {
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    /* Finish with a NBD_REP_ACK. */
 | 
			
		||||
    return nbd_negotiate_send_rep(client->ioc, NBD_REP_ACK, NBD_OPT_LIST);
 | 
			
		||||
    return nbd_negotiate_send_rep(client->ioc, NBD_REP_ACK, NBD_OPT_LIST, errp);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
static int nbd_negotiate_handle_export_name(NBDClient *client, uint32_t length)
 | 
			
		||||
static int nbd_negotiate_handle_export_name(NBDClient *client, uint32_t length,
 | 
			
		||||
                                            Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    char name[NBD_MAX_NAME_SIZE + 1];
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -272,11 +285,11 @@ static int nbd_negotiate_handle_export_name(NBDClient *client, uint32_t length)
 | 
			
		|||
     */
 | 
			
		||||
    TRACE("Checking length");
 | 
			
		||||
    if (length >= sizeof(name)) {
 | 
			
		||||
        LOG("Bad length received");
 | 
			
		||||
        error_setg(errp, "Bad length received");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    if (nbd_read(client->ioc, name, length, NULL) < 0) {
 | 
			
		||||
        LOG("read failed");
 | 
			
		||||
    if (nbd_read(client->ioc, name, length, errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "read failed: ");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    name[length] = '\0';
 | 
			
		||||
| 
						 | 
				
			
			@ -285,7 +298,7 @@ static int nbd_negotiate_handle_export_name(NBDClient *client, uint32_t length)
 | 
			
		|||
 | 
			
		||||
    client->exp = nbd_export_find(name);
 | 
			
		||||
    if (!client->exp) {
 | 
			
		||||
        LOG("export not found");
 | 
			
		||||
        error_setg(errp, "export not found");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -298,7 +311,8 @@ static int nbd_negotiate_handle_export_name(NBDClient *client, uint32_t length)
 | 
			
		|||
/* Handle NBD_OPT_STARTTLS. Return NULL to drop connection, or else the
 | 
			
		||||
 * new channel for all further (now-encrypted) communication. */
 | 
			
		||||
static QIOChannel *nbd_negotiate_handle_starttls(NBDClient *client,
 | 
			
		||||
                                                 uint32_t length)
 | 
			
		||||
                                                 uint32_t length,
 | 
			
		||||
                                                 Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    QIOChannel *ioc;
 | 
			
		||||
    QIOChannelTLS *tioc;
 | 
			
		||||
| 
						 | 
				
			
			@ -307,23 +321,24 @@ static QIOChannel *nbd_negotiate_handle_starttls(NBDClient *client,
 | 
			
		|||
    TRACE("Setting up TLS");
 | 
			
		||||
    ioc = client->ioc;
 | 
			
		||||
    if (length) {
 | 
			
		||||
        if (nbd_drop(ioc, length, NULL) < 0) {
 | 
			
		||||
        if (nbd_drop(ioc, length, errp) < 0) {
 | 
			
		||||
            return NULL;
 | 
			
		||||
        }
 | 
			
		||||
        nbd_negotiate_send_rep_err(ioc, NBD_REP_ERR_INVALID, NBD_OPT_STARTTLS,
 | 
			
		||||
                                   errp,
 | 
			
		||||
                                   "OPT_STARTTLS should not have length");
 | 
			
		||||
        return NULL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (nbd_negotiate_send_rep(client->ioc, NBD_REP_ACK,
 | 
			
		||||
                               NBD_OPT_STARTTLS) < 0) {
 | 
			
		||||
                               NBD_OPT_STARTTLS, errp) < 0) {
 | 
			
		||||
        return NULL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    tioc = qio_channel_tls_new_server(ioc,
 | 
			
		||||
                                      client->tlscreds,
 | 
			
		||||
                                      client->tlsaclname,
 | 
			
		||||
                                      NULL);
 | 
			
		||||
                                      errp);
 | 
			
		||||
    if (!tioc) {
 | 
			
		||||
        return NULL;
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -342,7 +357,7 @@ static QIOChannel *nbd_negotiate_handle_starttls(NBDClient *client,
 | 
			
		|||
    g_main_loop_unref(data.loop);
 | 
			
		||||
    if (data.error) {
 | 
			
		||||
        object_unref(OBJECT(tioc));
 | 
			
		||||
        error_free(data.error);
 | 
			
		||||
        error_propagate(errp, data.error);
 | 
			
		||||
        return NULL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -352,14 +367,16 @@ static QIOChannel *nbd_negotiate_handle_starttls(NBDClient *client,
 | 
			
		|||
/* nbd_negotiate_options
 | 
			
		||||
 * Process all NBD_OPT_* client option commands.
 | 
			
		||||
 * Return:
 | 
			
		||||
 * -errno  on error
 | 
			
		||||
 * 0       on successful negotiation
 | 
			
		||||
 * 1       if client sent NBD_OPT_ABORT, i.e. on valid disconnect
 | 
			
		||||
 * -errno  on error, errp is set
 | 
			
		||||
 * 0       on successful negotiation, errp is not set
 | 
			
		||||
 * 1       if client sent NBD_OPT_ABORT, i.e. on valid disconnect,
 | 
			
		||||
 *         errp is not set
 | 
			
		||||
 */
 | 
			
		||||
static int nbd_negotiate_options(NBDClient *client)
 | 
			
		||||
static int nbd_negotiate_options(NBDClient *client, Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    uint32_t flags;
 | 
			
		||||
    bool fixedNewstyle = false;
 | 
			
		||||
    Error *local_err = NULL;
 | 
			
		||||
 | 
			
		||||
    /* Client sends:
 | 
			
		||||
        [ 0 ..   3]   client flags
 | 
			
		||||
| 
						 | 
				
			
			@ -375,8 +392,8 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
        ...           Rest of request
 | 
			
		||||
    */
 | 
			
		||||
 | 
			
		||||
    if (nbd_read(client->ioc, &flags, sizeof(flags), NULL) < 0) {
 | 
			
		||||
        LOG("read failed");
 | 
			
		||||
    if (nbd_read(client->ioc, &flags, sizeof(flags), errp) < 0) {
 | 
			
		||||
        error_prepend(errp, "read failed: ");
 | 
			
		||||
        return -EIO;
 | 
			
		||||
    }
 | 
			
		||||
    TRACE("Checking client flags");
 | 
			
		||||
| 
						 | 
				
			
			@ -392,7 +409,7 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
        flags &= ~NBD_FLAG_C_NO_ZEROES;
 | 
			
		||||
    }
 | 
			
		||||
    if (flags != 0) {
 | 
			
		||||
        TRACE("Unknown client flags 0x%" PRIx32 " received", flags);
 | 
			
		||||
        error_setg(errp, "Unknown client flags 0x%" PRIx32 " received", flags);
 | 
			
		||||
        return -EIO;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -401,26 +418,25 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
        uint32_t clientflags, length;
 | 
			
		||||
        uint64_t magic;
 | 
			
		||||
 | 
			
		||||
        if (nbd_read(client->ioc, &magic, sizeof(magic), NULL) < 0) {
 | 
			
		||||
            LOG("read failed");
 | 
			
		||||
        if (nbd_read(client->ioc, &magic, sizeof(magic), errp) < 0) {
 | 
			
		||||
            error_prepend(errp, "read failed: ");
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
        TRACE("Checking opts magic");
 | 
			
		||||
        if (magic != be64_to_cpu(NBD_OPTS_MAGIC)) {
 | 
			
		||||
            LOG("Bad magic received");
 | 
			
		||||
            error_setg(errp, "Bad magic received");
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (nbd_read(client->ioc, &clientflags,
 | 
			
		||||
                      sizeof(clientflags), NULL) < 0)
 | 
			
		||||
        {
 | 
			
		||||
            LOG("read failed");
 | 
			
		||||
                     sizeof(clientflags), errp) < 0) {
 | 
			
		||||
            error_prepend(errp, "read failed: ");
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
        clientflags = be32_to_cpu(clientflags);
 | 
			
		||||
 | 
			
		||||
        if (nbd_read(client->ioc, &length, sizeof(length), NULL) < 0) {
 | 
			
		||||
            LOG("read failed");
 | 
			
		||||
        if (nbd_read(client->ioc, &length, sizeof(length), errp) < 0) {
 | 
			
		||||
            error_prepend(errp, "read failed: ");
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
        length = be32_to_cpu(length);
 | 
			
		||||
| 
						 | 
				
			
			@ -430,12 +446,12 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
            client->ioc == (QIOChannel *)client->sioc) {
 | 
			
		||||
            QIOChannel *tioc;
 | 
			
		||||
            if (!fixedNewstyle) {
 | 
			
		||||
                TRACE("Unsupported option 0x%" PRIx32, clientflags);
 | 
			
		||||
                error_setg(errp, "Unsupported option 0x%" PRIx32, clientflags);
 | 
			
		||||
                return -EINVAL;
 | 
			
		||||
            }
 | 
			
		||||
            switch (clientflags) {
 | 
			
		||||
            case NBD_OPT_STARTTLS:
 | 
			
		||||
                tioc = nbd_negotiate_handle_starttls(client, length);
 | 
			
		||||
                tioc = nbd_negotiate_handle_starttls(client, length, errp);
 | 
			
		||||
                if (!tioc) {
 | 
			
		||||
                    return -EIO;
 | 
			
		||||
                }
 | 
			
		||||
| 
						 | 
				
			
			@ -445,16 +461,17 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
 | 
			
		||||
            case NBD_OPT_EXPORT_NAME:
 | 
			
		||||
                /* No way to return an error to client, so drop connection */
 | 
			
		||||
                TRACE("Option 0x%x not permitted before TLS", clientflags);
 | 
			
		||||
                error_setg(errp, "Option 0x%x not permitted before TLS",
 | 
			
		||||
                           clientflags);
 | 
			
		||||
                return -EINVAL;
 | 
			
		||||
 | 
			
		||||
            default:
 | 
			
		||||
                if (nbd_drop(client->ioc, length, NULL) < 0) {
 | 
			
		||||
                if (nbd_drop(client->ioc, length, errp) < 0) {
 | 
			
		||||
                    return -EIO;
 | 
			
		||||
                }
 | 
			
		||||
                ret = nbd_negotiate_send_rep_err(client->ioc,
 | 
			
		||||
                                                 NBD_REP_ERR_TLS_REQD,
 | 
			
		||||
                                                 clientflags,
 | 
			
		||||
                                                 clientflags, errp,
 | 
			
		||||
                                                 "Option 0x%" PRIx32
 | 
			
		||||
                                                 "not permitted before TLS",
 | 
			
		||||
                                                 clientflags);
 | 
			
		||||
| 
						 | 
				
			
			@ -470,7 +487,7 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
        } else if (fixedNewstyle) {
 | 
			
		||||
            switch (clientflags) {
 | 
			
		||||
            case NBD_OPT_LIST:
 | 
			
		||||
                ret = nbd_negotiate_handle_list(client, length);
 | 
			
		||||
                ret = nbd_negotiate_handle_list(client, length, errp);
 | 
			
		||||
                if (ret < 0) {
 | 
			
		||||
                    return ret;
 | 
			
		||||
                }
 | 
			
		||||
| 
						 | 
				
			
			@ -480,25 +497,33 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
                /* NBD spec says we must try to reply before
 | 
			
		||||
                 * disconnecting, but that we must also tolerate
 | 
			
		||||
                 * guests that don't wait for our reply. */
 | 
			
		||||
                nbd_negotiate_send_rep(client->ioc, NBD_REP_ACK, clientflags);
 | 
			
		||||
                nbd_negotiate_send_rep(client->ioc, NBD_REP_ACK, clientflags,
 | 
			
		||||
                                       &local_err);
 | 
			
		||||
 | 
			
		||||
                if (local_err != NULL) {
 | 
			
		||||
                    TRACE("Reply to NBD_OPT_ABORT request failed: %s",
 | 
			
		||||
                          error_get_pretty(local_err));
 | 
			
		||||
                    error_free(local_err);
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                return 1;
 | 
			
		||||
 | 
			
		||||
            case NBD_OPT_EXPORT_NAME:
 | 
			
		||||
                return nbd_negotiate_handle_export_name(client, length);
 | 
			
		||||
                return nbd_negotiate_handle_export_name(client, length, errp);
 | 
			
		||||
 | 
			
		||||
            case NBD_OPT_STARTTLS:
 | 
			
		||||
                if (nbd_drop(client->ioc, length, NULL) < 0) {
 | 
			
		||||
                if (nbd_drop(client->ioc, length, errp) < 0) {
 | 
			
		||||
                    return -EIO;
 | 
			
		||||
                }
 | 
			
		||||
                if (client->tlscreds) {
 | 
			
		||||
                    ret = nbd_negotiate_send_rep_err(client->ioc,
 | 
			
		||||
                                                     NBD_REP_ERR_INVALID,
 | 
			
		||||
                                                     clientflags,
 | 
			
		||||
                                                     clientflags, errp,
 | 
			
		||||
                                                     "TLS already enabled");
 | 
			
		||||
                } else {
 | 
			
		||||
                    ret = nbd_negotiate_send_rep_err(client->ioc,
 | 
			
		||||
                                                     NBD_REP_ERR_POLICY,
 | 
			
		||||
                                                     clientflags,
 | 
			
		||||
                                                     clientflags, errp,
 | 
			
		||||
                                                     "TLS not configured");
 | 
			
		||||
                }
 | 
			
		||||
                if (ret < 0) {
 | 
			
		||||
| 
						 | 
				
			
			@ -506,12 +531,12 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
                }
 | 
			
		||||
                break;
 | 
			
		||||
            default:
 | 
			
		||||
                if (nbd_drop(client->ioc, length, NULL) < 0) {
 | 
			
		||||
                if (nbd_drop(client->ioc, length, errp) < 0) {
 | 
			
		||||
                    return -EIO;
 | 
			
		||||
                }
 | 
			
		||||
                ret = nbd_negotiate_send_rep_err(client->ioc,
 | 
			
		||||
                                                 NBD_REP_ERR_UNSUP,
 | 
			
		||||
                                                 clientflags,
 | 
			
		||||
                                                 clientflags, errp,
 | 
			
		||||
                                                 "Unsupported option 0x%"
 | 
			
		||||
                                                 PRIx32,
 | 
			
		||||
                                                 clientflags);
 | 
			
		||||
| 
						 | 
				
			
			@ -527,10 +552,10 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
             */
 | 
			
		||||
            switch (clientflags) {
 | 
			
		||||
            case NBD_OPT_EXPORT_NAME:
 | 
			
		||||
                return nbd_negotiate_handle_export_name(client, length);
 | 
			
		||||
                return nbd_negotiate_handle_export_name(client, length, errp);
 | 
			
		||||
 | 
			
		||||
            default:
 | 
			
		||||
                TRACE("Unsupported option 0x%" PRIx32, clientflags);
 | 
			
		||||
                error_setg(errp, "Unsupported option 0x%" PRIx32, clientflags);
 | 
			
		||||
                return -EINVAL;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
| 
						 | 
				
			
			@ -539,11 +564,12 @@ static int nbd_negotiate_options(NBDClient *client)
 | 
			
		|||
 | 
			
		||||
/* nbd_negotiate
 | 
			
		||||
 * Return:
 | 
			
		||||
 * -errno  on error
 | 
			
		||||
 * 0       on successful negotiation
 | 
			
		||||
 * 1       if client sent NBD_OPT_ABORT, i.e. on valid disconnect
 | 
			
		||||
 * -errno  on error, errp is set
 | 
			
		||||
 * 0       on successful negotiation, errp is not set
 | 
			
		||||
 * 1       if client sent NBD_OPT_ABORT, i.e. on valid disconnect,
 | 
			
		||||
 *         errp is not set
 | 
			
		||||
 */
 | 
			
		||||
static coroutine_fn int nbd_negotiate(NBDClient *client)
 | 
			
		||||
static coroutine_fn int nbd_negotiate(NBDClient *client, Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    char buf[8 + 8 + 8 + 128];
 | 
			
		||||
    int ret;
 | 
			
		||||
| 
						 | 
				
			
			@ -585,21 +611,23 @@ static coroutine_fn int nbd_negotiate(NBDClient *client)
 | 
			
		|||
        stq_be_p(buf + 16, client->exp->size);
 | 
			
		||||
        stw_be_p(buf + 26, client->exp->nbdflags | myflags);
 | 
			
		||||
 | 
			
		||||
        if (nbd_write(client->ioc, buf, sizeof(buf), NULL) < 0) {
 | 
			
		||||
            LOG("write failed");
 | 
			
		||||
        if (nbd_write(client->ioc, buf, sizeof(buf), errp) < 0) {
 | 
			
		||||
            error_prepend(errp, "write failed: ");
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
    } else {
 | 
			
		||||
        stq_be_p(buf + 8, NBD_OPTS_MAGIC);
 | 
			
		||||
        stw_be_p(buf + 16, NBD_FLAG_FIXED_NEWSTYLE | NBD_FLAG_NO_ZEROES);
 | 
			
		||||
 | 
			
		||||
        if (nbd_write(client->ioc, buf, 18, NULL) < 0) {
 | 
			
		||||
            LOG("write failed");
 | 
			
		||||
        if (nbd_write(client->ioc, buf, 18, errp) < 0) {
 | 
			
		||||
            error_prepend(errp, "write failed: ");
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
        ret = nbd_negotiate_options(client);
 | 
			
		||||
        ret = nbd_negotiate_options(client, errp);
 | 
			
		||||
        if (ret != 0) {
 | 
			
		||||
            LOG("option negotiation failed");
 | 
			
		||||
            if (ret < 0) {
 | 
			
		||||
                error_prepend(errp, "option negotiation failed: ");
 | 
			
		||||
            }
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -608,9 +636,9 @@ static coroutine_fn int nbd_negotiate(NBDClient *client)
 | 
			
		|||
        stq_be_p(buf + 18, client->exp->size);
 | 
			
		||||
        stw_be_p(buf + 26, client->exp->nbdflags | myflags);
 | 
			
		||||
        len = client->no_zeroes ? 10 : sizeof(buf) - 18;
 | 
			
		||||
        ret = nbd_write(client->ioc, buf + 18, len, NULL);
 | 
			
		||||
        ret = nbd_write(client->ioc, buf + 18, len, errp);
 | 
			
		||||
        if (ret < 0) {
 | 
			
		||||
            LOG("write failed");
 | 
			
		||||
            error_prepend(errp, "write failed: ");
 | 
			
		||||
            return ret;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -620,13 +648,14 @@ static coroutine_fn int nbd_negotiate(NBDClient *client)
 | 
			
		|||
    return 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
static int nbd_receive_request(QIOChannel *ioc, NBDRequest *request)
 | 
			
		||||
static int nbd_receive_request(QIOChannel *ioc, NBDRequest *request,
 | 
			
		||||
                               Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    uint8_t buf[NBD_REQUEST_SIZE];
 | 
			
		||||
    uint32_t magic;
 | 
			
		||||
    int ret;
 | 
			
		||||
 | 
			
		||||
    ret = nbd_read(ioc, buf, sizeof(buf), NULL);
 | 
			
		||||
    ret = nbd_read(ioc, buf, sizeof(buf), errp);
 | 
			
		||||
    if (ret < 0) {
 | 
			
		||||
        return ret;
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -652,7 +681,7 @@ static int nbd_receive_request(QIOChannel *ioc, NBDRequest *request)
 | 
			
		|||
          magic, request->flags, request->type, request->from, request->len);
 | 
			
		||||
 | 
			
		||||
    if (magic != NBD_REQUEST_MAGIC) {
 | 
			
		||||
        LOG("invalid magic (got 0x%" PRIx32 ")", magic);
 | 
			
		||||
        error_setg(errp, "invalid magic (got 0x%" PRIx32 ")", magic);
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    return 0;
 | 
			
		||||
| 
						 | 
				
			
			@ -998,13 +1027,14 @@ static int nbd_co_send_reply(NBDRequestData *req, NBDReply *reply, int len)
 | 
			
		|||
 * the client (although the caller may still need to disconnect after reporting
 | 
			
		||||
 * the error).
 | 
			
		||||
 */
 | 
			
		||||
static int nbd_co_receive_request(NBDRequestData *req, NBDRequest *request)
 | 
			
		||||
static int nbd_co_receive_request(NBDRequestData *req, NBDRequest *request,
 | 
			
		||||
                                  Error **errp)
 | 
			
		||||
{
 | 
			
		||||
    NBDClient *client = req->client;
 | 
			
		||||
 | 
			
		||||
    g_assert(qemu_in_coroutine());
 | 
			
		||||
    assert(client->recv_coroutine == qemu_coroutine_self());
 | 
			
		||||
    if (nbd_receive_request(client->ioc, request) < 0) {
 | 
			
		||||
    if (nbd_receive_request(client->ioc, request, errp) < 0) {
 | 
			
		||||
        return -EIO;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1026,27 +1056,29 @@ static int nbd_co_receive_request(NBDRequestData *req, NBDRequest *request)
 | 
			
		|||
     * checks as possible until after reading any NBD_CMD_WRITE
 | 
			
		||||
     * payload, so we can try and keep the connection alive.  */
 | 
			
		||||
    if ((request->from + request->len) < request->from) {
 | 
			
		||||
        LOG("integer overflow detected, you're probably being attacked");
 | 
			
		||||
        error_setg(errp,
 | 
			
		||||
                   "integer overflow detected, you're probably being attacked");
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (request->type == NBD_CMD_READ || request->type == NBD_CMD_WRITE) {
 | 
			
		||||
        if (request->len > NBD_MAX_BUFFER_SIZE) {
 | 
			
		||||
            LOG("len (%" PRIu32" ) is larger than max len (%u)",
 | 
			
		||||
                request->len, NBD_MAX_BUFFER_SIZE);
 | 
			
		||||
            error_setg(errp, "len (%" PRIu32" ) is larger than max len (%u)",
 | 
			
		||||
                       request->len, NBD_MAX_BUFFER_SIZE);
 | 
			
		||||
            return -EINVAL;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        req->data = blk_try_blockalign(client->exp->blk, request->len);
 | 
			
		||||
        if (req->data == NULL) {
 | 
			
		||||
            error_setg(errp, "No memory");
 | 
			
		||||
            return -ENOMEM;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    if (request->type == NBD_CMD_WRITE) {
 | 
			
		||||
        TRACE("Reading %" PRIu32 " byte(s)", request->len);
 | 
			
		||||
 | 
			
		||||
        if (nbd_read(client->ioc, req->data, request->len, NULL) < 0) {
 | 
			
		||||
            LOG("reading from socket failed");
 | 
			
		||||
        if (nbd_read(client->ioc, req->data, request->len, errp) < 0) {
 | 
			
		||||
            error_prepend(errp, "reading from socket failed: ");
 | 
			
		||||
            return -EIO;
 | 
			
		||||
        }
 | 
			
		||||
        req->complete = true;
 | 
			
		||||
| 
						 | 
				
			
			@ -1054,18 +1086,18 @@ static int nbd_co_receive_request(NBDRequestData *req, NBDRequest *request)
 | 
			
		|||
 | 
			
		||||
    /* Sanity checks, part 2. */
 | 
			
		||||
    if (request->from + request->len > client->exp->size) {
 | 
			
		||||
        LOG("operation past EOF; From: %" PRIu64 ", Len: %" PRIu32
 | 
			
		||||
            ", Size: %" PRIu64, request->from, request->len,
 | 
			
		||||
            (uint64_t)client->exp->size);
 | 
			
		||||
        error_setg(errp, "operation past EOF; From: %" PRIu64 ", Len: %" PRIu32
 | 
			
		||||
                   ", Size: %" PRIu64, request->from, request->len,
 | 
			
		||||
                   (uint64_t)client->exp->size);
 | 
			
		||||
        return request->type == NBD_CMD_WRITE ? -ENOSPC : -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    if (request->flags & ~(NBD_CMD_FLAG_FUA | NBD_CMD_FLAG_NO_HOLE)) {
 | 
			
		||||
        LOG("unsupported flags (got 0x%x)", request->flags);
 | 
			
		||||
        error_setg(errp, "unsupported flags (got 0x%x)", request->flags);
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
    if (request->type != NBD_CMD_WRITE_ZEROES &&
 | 
			
		||||
        (request->flags & NBD_CMD_FLAG_NO_HOLE)) {
 | 
			
		||||
        LOG("unexpected flags (got 0x%x)", request->flags);
 | 
			
		||||
        error_setg(errp, "unexpected flags (got 0x%x)", request->flags);
 | 
			
		||||
        return -EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1083,6 +1115,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
    int ret;
 | 
			
		||||
    int flags;
 | 
			
		||||
    int reply_data_len = 0;
 | 
			
		||||
    Error *local_err = NULL;
 | 
			
		||||
 | 
			
		||||
    TRACE("Reading request.");
 | 
			
		||||
    if (client->closing) {
 | 
			
		||||
| 
						 | 
				
			
			@ -1091,7 +1124,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
    }
 | 
			
		||||
 | 
			
		||||
    req = nbd_request_get(client);
 | 
			
		||||
    ret = nbd_co_receive_request(req, &request);
 | 
			
		||||
    ret = nbd_co_receive_request(req, &request, &local_err);
 | 
			
		||||
    client->recv_coroutine = NULL;
 | 
			
		||||
    nbd_client_receive_next_request(client);
 | 
			
		||||
    if (ret == -EIO) {
 | 
			
		||||
| 
						 | 
				
			
			@ -1122,7 +1155,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
        if (request.flags & NBD_CMD_FLAG_FUA) {
 | 
			
		||||
            ret = blk_co_flush(exp->blk);
 | 
			
		||||
            if (ret < 0) {
 | 
			
		||||
                LOG("flush failed");
 | 
			
		||||
                error_setg_errno(&local_err, -ret, "flush failed");
 | 
			
		||||
                reply.error = -ret;
 | 
			
		||||
                break;
 | 
			
		||||
            }
 | 
			
		||||
| 
						 | 
				
			
			@ -1131,7 +1164,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
        ret = blk_pread(exp->blk, request.from + exp->dev_offset,
 | 
			
		||||
                        req->data, request.len);
 | 
			
		||||
        if (ret < 0) {
 | 
			
		||||
            LOG("reading from file failed");
 | 
			
		||||
            error_setg_errno(&local_err, -ret, "reading from file failed");
 | 
			
		||||
            reply.error = -ret;
 | 
			
		||||
            break;
 | 
			
		||||
        }
 | 
			
		||||
| 
						 | 
				
			
			@ -1158,7 +1191,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
        ret = blk_pwrite(exp->blk, request.from + exp->dev_offset,
 | 
			
		||||
                         req->data, request.len, flags);
 | 
			
		||||
        if (ret < 0) {
 | 
			
		||||
            LOG("writing to file failed");
 | 
			
		||||
            error_setg_errno(&local_err, -ret, "writing to file failed");
 | 
			
		||||
            reply.error = -ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1167,7 +1200,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
        TRACE("Request type is WRITE_ZEROES");
 | 
			
		||||
 | 
			
		||||
        if (exp->nbdflags & NBD_FLAG_READ_ONLY) {
 | 
			
		||||
            TRACE("Server is read-only, return error");
 | 
			
		||||
            error_setg(&local_err, "Server is read-only, return error");
 | 
			
		||||
            reply.error = EROFS;
 | 
			
		||||
            break;
 | 
			
		||||
        }
 | 
			
		||||
| 
						 | 
				
			
			@ -1184,7 +1217,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
        ret = blk_pwrite_zeroes(exp->blk, request.from + exp->dev_offset,
 | 
			
		||||
                                request.len, flags);
 | 
			
		||||
        if (ret < 0) {
 | 
			
		||||
            LOG("writing to file failed");
 | 
			
		||||
            error_setg_errno(&local_err, -ret, "writing to file failed");
 | 
			
		||||
            reply.error = -ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1198,7 +1231,7 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
 | 
			
		||||
        ret = blk_co_flush(exp->blk);
 | 
			
		||||
        if (ret < 0) {
 | 
			
		||||
            LOG("flush failed");
 | 
			
		||||
            error_setg_errno(&local_err, -ret, "flush failed");
 | 
			
		||||
            reply.error = -ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1208,21 +1241,35 @@ static coroutine_fn void nbd_trip(void *opaque)
 | 
			
		|||
        ret = blk_co_pdiscard(exp->blk, request.from + exp->dev_offset,
 | 
			
		||||
                              request.len);
 | 
			
		||||
        if (ret < 0) {
 | 
			
		||||
            LOG("discard failed");
 | 
			
		||||
            error_setg_errno(&local_err, -ret, "discard failed");
 | 
			
		||||
            reply.error = -ret;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        break;
 | 
			
		||||
    default:
 | 
			
		||||
        LOG("invalid request type (%" PRIu32 ") received", request.type);
 | 
			
		||||
        error_setg(&local_err, "invalid request type (%" PRIu32 ") received",
 | 
			
		||||
                   request.type);
 | 
			
		||||
        reply.error = EINVAL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
reply:
 | 
			
		||||
    if (local_err) {
 | 
			
		||||
        /* If we are here local_err is not fatal error, already stored in
 | 
			
		||||
         * reply.error */
 | 
			
		||||
        error_report_err(local_err);
 | 
			
		||||
        local_err = NULL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (nbd_co_send_reply(req, &reply, reply_data_len) < 0) {
 | 
			
		||||
        error_setg(&local_err, "Failed to send reply");
 | 
			
		||||
        goto disconnect;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /* We must disconnect after NBD_CMD_WRITE if we did not
 | 
			
		||||
     * read the payload.
 | 
			
		||||
     */
 | 
			
		||||
    if (nbd_co_send_reply(req, &reply, reply_data_len) < 0 || !req->complete) {
 | 
			
		||||
    if (!req->complete) {
 | 
			
		||||
        error_setg(&local_err, "Request handling failed in intermediate state");
 | 
			
		||||
        goto disconnect;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1234,6 +1281,9 @@ done:
 | 
			
		|||
    return;
 | 
			
		||||
 | 
			
		||||
disconnect:
 | 
			
		||||
    if (local_err) {
 | 
			
		||||
        error_reportf_err(local_err, "Disconnect client, due to: ");
 | 
			
		||||
    }
 | 
			
		||||
    nbd_request_put(req);
 | 
			
		||||
    client_close(client, true);
 | 
			
		||||
    nbd_client_put(client);
 | 
			
		||||
| 
						 | 
				
			
			@ -1252,6 +1302,7 @@ static coroutine_fn void nbd_co_client_start(void *opaque)
 | 
			
		|||
{
 | 
			
		||||
    NBDClient *client = opaque;
 | 
			
		||||
    NBDExport *exp = client->exp;
 | 
			
		||||
    Error *local_err = NULL;
 | 
			
		||||
 | 
			
		||||
    if (exp) {
 | 
			
		||||
        nbd_export_get(exp);
 | 
			
		||||
| 
						 | 
				
			
			@ -1259,7 +1310,10 @@ static coroutine_fn void nbd_co_client_start(void *opaque)
 | 
			
		|||
    }
 | 
			
		||||
    qemu_co_mutex_init(&client->send_lock);
 | 
			
		||||
 | 
			
		||||
    if (nbd_negotiate(client)) {
 | 
			
		||||
    if (nbd_negotiate(client, &local_err)) {
 | 
			
		||||
        if (local_err) {
 | 
			
		||||
            error_report_err(local_err);
 | 
			
		||||
        }
 | 
			
		||||
        client_close(client, false);
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue