Commit 8ecc2f9e authored by Markus Armbruster's avatar Markus Armbruster Committed by Kevin Wolf
Browse files

sheepdog: Use SocketAddress and socket_connect() 



sd_parse_uri() builds a string from host and port parts for
inet_connect().  inet_connect() parses it into host, port and options.
Whether this gets exactly the same host, port and no options for all
inputs is not obvious.

Cut out the string middleman and build a SocketAddress for
socket_connect() instead.

Signed-off-by: default avatarMarkus Armbruster <armbru@redhat.com>
Reviewed-by: default avatarKevin Wolf <kwolf@redhat.com>
Signed-off-by: default avatarKevin Wolf <kwolf@redhat.com>
parent 36bcac16

block/sheepdog.c

+30 −23
Original line number Diff line number Diff line
@@ -374,8 +374,7 @@ struct BDRVSheepdogState { 
    uint32_t cache_flags;

    bool discard_supported;



    char *host_spec;

    bool is_unix;

    SocketAddress *addr;

    int fd;



    CoMutex lock;
@@ -532,18 +531,14 @@ static int connect_to_sdog(BDRVSheepdogState *s, Error **errp) 
{

    int fd;



    if (s->is_unix) {

        fd = unix_connect(s->host_spec, errp);

    } else {

        fd = inet_connect(s->host_spec, errp);

    fd = socket_connect(s->addr, errp, NULL, NULL);



        if (fd >= 0) {

    if (s->addr->type == SOCKET_ADDRESS_KIND_INET && fd >= 0) {

        int ret = socket_set_nodelay(fd);

        if (ret < 0) {

            error_report("%s", strerror(errno));

        }

    }

    }



    if (fd >= 0) {

        qemu_set_nonblock(fd);
@@ -820,8 +815,7 @@ static void coroutine_fn aio_read_response(void *opaque) 
    case AIOCB_DISCARD_OBJ:

        switch (rsp.result) {

        case SD_RES_INVALID_PARMS:

            error_report("sheep(%s) doesn't support discard command",

                         s->host_spec);

            error_report("server doesn't support discard command");

            rsp.result = SD_RES_SUCCESS;

            s->discard_supported = false;

            break;
@@ -962,8 +956,10 @@ static void sd_parse_uri(BDRVSheepdogState *s, const char *filename, 
                         Error **errp)

{

    Error *err = NULL;

    URI *uri;

    QueryParams *qp = NULL;

    SocketAddress *addr = NULL;

    bool is_unix;

    URI *uri;



    uri = uri_parse(filename);

    if (!uri) {
@@ -973,11 +969,11 @@ static void sd_parse_uri(BDRVSheepdogState *s, const char *filename, 


    /* transport */

    if (!strcmp(uri->scheme, "sheepdog")) {

        s->is_unix = false;

        is_unix = false;

    } else if (!strcmp(uri->scheme, "sheepdog+tcp")) {

        s->is_unix = false;

        is_unix = false;

    } else if (!strcmp(uri->scheme, "sheepdog+unix")) {

        s->is_unix = true;

        is_unix = true;

    } else {

        error_setg(&err, "URI scheme must be 'sheepdog', 'sheepdog+tcp',"

                   " or 'sheepdog+unix'");
@@ -994,8 +990,9 @@ static void sd_parse_uri(BDRVSheepdogState *s, const char *filename, 
    }



    qp = query_params_parse(uri->query);

    addr = g_new0(SocketAddress, 1);



    if (s->is_unix) {

    if (is_unix) {

        /* sheepdog+unix:///vdiname?socket=path */

        if (uri->server || uri->port) {

            error_setg(&err, "URI scheme %s doesn't accept a server address",
@@ -1012,14 +1009,19 @@ static void sd_parse_uri(BDRVSheepdogState *s, const char *filename, 
            error_setg(&err, "unexpected query parameters");

            goto out;

        }

        s->host_spec = g_strdup(qp->p[0].value);

        addr->type = SOCKET_ADDRESS_KIND_UNIX;

        addr->u.q_unix.data = g_new0(UnixSocketAddress, 1);

        addr->u.q_unix.data->path = g_strdup(qp->p[0].value);

    } else {

        /* sheepdog[+tcp]://[host:port]/vdiname */

        if (qp->n) {

            error_setg(&err, "unexpected query parameters");

            goto out;

        }

        s->host_spec = g_strdup_printf("%s:%d", uri->server ?: SD_DEFAULT_ADDR,

        addr->type = SOCKET_ADDRESS_KIND_INET;

        addr->u.inet.data = g_new0(InetSocketAddress, 1);

        addr->u.inet.data->host = g_strdup(uri->server ?: SD_DEFAULT_ADDR);

        addr->u.inet.data->port = g_strdup_printf("%d",

                                        uri->port ?: SD_DEFAULT_PORT);

    }
@@ -1035,7 +1037,12 @@ static void sd_parse_uri(BDRVSheepdogState *s, const char *filename, 
    }



out:

    if (err) {

        error_propagate(errp, err);

        qapi_free_SocketAddress(addr);

    } else {

        s->addr = addr;

    }

    if (qp) {

        query_params_free(qp);

    }
@@ -1998,7 +2005,7 @@ static void sd_close(BlockDriverState *bs) 
    aio_set_fd_handler(bdrv_get_aio_context(bs), s->fd,

                       false, NULL, NULL, NULL, NULL);

    closesocket(s->fd);

    g_free(s->host_spec);

    qapi_free_SocketAddress(s->addr);

}



static int64_t sd_getlength(BlockDriverState *bs)