[mirror_qemu.git] / io / channel-socket.c

/*
 * QEMU I/O channels sockets driver
 *
 * Copyright (c) 2015 Red Hat, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "qemu/osdep.h"
#include "qapi/error.h"
#include "io/channel-socket.h"
#include "io/channel-watch.h"
#include "trace.h"
#include "qapi/clone-visitor.h"

#define SOCKET_MAX_FDS 16

SocketAddress *
qio_channel_socket_get_local_address(QIOChannelSocket *ioc,
                                     Error **errp)
{
    return socket_sockaddr_to_address(&ioc->localAddr,
                                      ioc->localAddrLen,
                                      errp);
}

SocketAddress *
qio_channel_socket_get_remote_address(QIOChannelSocket *ioc,
                                      Error **errp)
{
    return socket_sockaddr_to_address(&ioc->remoteAddr,
                                      ioc->remoteAddrLen,
                                      errp);
}

QIOChannelSocket *
qio_channel_socket_new(void)
{
    QIOChannelSocket *sioc;
    QIOChannel *ioc;

    sioc = QIO_CHANNEL_SOCKET(object_new(TYPE_QIO_CHANNEL_SOCKET));
    sioc->fd = -1;

    ioc = QIO_CHANNEL(sioc);
    ioc->features |= (1 << QIO_CHANNEL_FEATURE_SHUTDOWN);

#ifdef WIN32
    ioc->event = CreateEvent(NULL, FALSE, FALSE, NULL);
#endif

    trace_qio_channel_socket_new(sioc);

    return sioc;
}


static int
qio_channel_socket_set_fd(QIOChannelSocket *sioc,
                          int fd,
                          Error **errp)
{
    int val;
    socklen_t len = sizeof(val);

    if (sioc->fd != -1) {
        error_setg(errp, "Socket is already open");
        return -1;
    }

    sioc->fd = fd;
    sioc->remoteAddrLen = sizeof(sioc->remoteAddr);
    sioc->localAddrLen = sizeof(sioc->localAddr);


    if (getpeername(fd, (struct sockaddr *)&sioc->remoteAddr,
                    &sioc->remoteAddrLen) < 0) {
        if (errno == ENOTCONN) {
            memset(&sioc->remoteAddr, 0, sizeof(sioc->remoteAddr));
            sioc->remoteAddrLen = sizeof(sioc->remoteAddr);
        } else {
            error_setg_errno(errp, errno,
                             "Unable to query remote socket address");
            goto error;
        }
    }

    if (getsockname(fd, (struct sockaddr *)&sioc->localAddr,
                    &sioc->localAddrLen) < 0) {
        error_setg_errno(errp, errno,
                         "Unable to query local socket address");
        goto error;
    }

#ifndef WIN32
    if (sioc->localAddr.ss_family == AF_UNIX) {
        QIOChannel *ioc = QIO_CHANNEL(sioc);
        ioc->features |= (1 << QIO_CHANNEL_FEATURE_FD_PASS);
    }
#endif /* WIN32 */
    if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, &val, &len) == 0 && val) {
        QIOChannel *ioc = QIO_CHANNEL(sioc);
        ioc->features |= (1 << QIO_CHANNEL_FEATURE_LISTEN);
    }

    return 0;

 error:
    sioc->fd = -1; /* Let the caller close FD on failure */
    return -1;
}

QIOChannelSocket *
qio_channel_socket_new_fd(int fd,
                          Error **errp)
{
    QIOChannelSocket *ioc;

    ioc = qio_channel_socket_new();
    if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
        object_unref(OBJECT(ioc));
        return NULL;
    }

    trace_qio_channel_socket_new_fd(ioc, fd);

    return ioc;
}


int qio_channel_socket_connect_sync(QIOChannelSocket *ioc,
                                    SocketAddress *addr,
                                    Error **errp)
{
    int fd;

    trace_qio_channel_socket_connect_sync(ioc, addr);
    fd = socket_connect(addr, errp, NULL, NULL);
    if (fd < 0) {
        trace_qio_channel_socket_connect_fail(ioc);
        return -1;
    }

    trace_qio_channel_socket_connect_complete(ioc, fd);
    if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
        close(fd);
        return -1;
    }

    return 0;
}


static int qio_channel_socket_connect_worker(QIOTask *task,
                                             Error **errp,
                                             gpointer opaque)
{
    QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
    SocketAddress *addr = opaque;
    int ret;

    ret = qio_channel_socket_connect_sync(ioc,
                                          addr,
                                          errp);

    object_unref(OBJECT(ioc));
    return ret;
}


void qio_channel_socket_connect_async(QIOChannelSocket *ioc,
                                      SocketAddress *addr,
                                      QIOTaskFunc callback,
                                      gpointer opaque,
                                      GDestroyNotify destroy)
{
    QIOTask *task = qio_task_new(
        OBJECT(ioc), callback, opaque, destroy);
    SocketAddress *addrCopy;

    addrCopy = QAPI_CLONE(SocketAddress, addr);

    /* socket_connect() does a non-blocking connect(), but it
     * still blocks in DNS lookups, so we must use a thread */
    trace_qio_channel_socket_connect_async(ioc, addr);
    qio_task_run_in_thread(task,
                           qio_channel_socket_connect_worker,
                           addrCopy,
                           (GDestroyNotify)qapi_free_SocketAddress);
}


int qio_channel_socket_listen_sync(QIOChannelSocket *ioc,
                                   SocketAddress *addr,
                                   Error **errp)
{
    int fd;

    trace_qio_channel_socket_listen_sync(ioc, addr);
    fd = socket_listen(addr, errp);
    if (fd < 0) {
        trace_qio_channel_socket_listen_fail(ioc);
        return -1;
    }

    trace_qio_channel_socket_listen_complete(ioc, fd);
    if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
        close(fd);
        return -1;
    }

    return 0;
}


static int qio_channel_socket_listen_worker(QIOTask *task,
                                            Error **errp,
                                            gpointer opaque)
{
    QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
    SocketAddress *addr = opaque;
    int ret;

    ret = qio_channel_socket_listen_sync(ioc,
                                         addr,
                                         errp);

    object_unref(OBJECT(ioc));
    return ret;
}


void qio_channel_socket_listen_async(QIOChannelSocket *ioc,
                                     SocketAddress *addr,
                                     QIOTaskFunc callback,
                                     gpointer opaque,
                                     GDestroyNotify destroy)
{
    QIOTask *task = qio_task_new(
        OBJECT(ioc), callback, opaque, destroy);
    SocketAddress *addrCopy;

    addrCopy = QAPI_CLONE(SocketAddress, addr);

    /* socket_listen() blocks in DNS lookups, so we must use a thread */
    trace_qio_channel_socket_listen_async(ioc, addr);
    qio_task_run_in_thread(task,
                           qio_channel_socket_listen_worker,
                           addrCopy,
                           (GDestroyNotify)qapi_free_SocketAddress);
}


int qio_channel_socket_dgram_sync(QIOChannelSocket *ioc,
                                  SocketAddress *localAddr,
                                  SocketAddress *remoteAddr,
                                  Error **errp)
{
    int fd;

    trace_qio_channel_socket_dgram_sync(ioc, localAddr, remoteAddr);
    fd = socket_dgram(remoteAddr, localAddr, errp);
    if (fd < 0) {
        trace_qio_channel_socket_dgram_fail(ioc);
        return -1;
    }

    trace_qio_channel_socket_dgram_complete(ioc, fd);
    if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
        close(fd);
        return -1;
    }

    return 0;
}


struct QIOChannelSocketDGramWorkerData {
    SocketAddress *localAddr;
    SocketAddress *remoteAddr;
};


static void qio_channel_socket_dgram_worker_free(gpointer opaque)
{
    struct QIOChannelSocketDGramWorkerData *data = opaque;
    qapi_free_SocketAddress(data->localAddr);
    qapi_free_SocketAddress(data->remoteAddr);
    g_free(data);
}

static int qio_channel_socket_dgram_worker(QIOTask *task,
                                           Error **errp,
                                           gpointer opaque)
{
    QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
    struct QIOChannelSocketDGramWorkerData *data = opaque;
    int ret;

    /* socket_dgram() blocks in DNS lookups, so we must use a thread */
    ret = qio_channel_socket_dgram_sync(ioc,
                                        data->localAddr,
                                        data->remoteAddr,
                                        errp);

    object_unref(OBJECT(ioc));
    return ret;
}


void qio_channel_socket_dgram_async(QIOChannelSocket *ioc,
                                    SocketAddress *localAddr,
                                    SocketAddress *remoteAddr,
                                    QIOTaskFunc callback,
                                    gpointer opaque,
                                    GDestroyNotify destroy)
{
    QIOTask *task = qio_task_new(
        OBJECT(ioc), callback, opaque, destroy);
    struct QIOChannelSocketDGramWorkerData *data = g_new0(
        struct QIOChannelSocketDGramWorkerData, 1);

    data->localAddr = QAPI_CLONE(SocketAddress, localAddr);
    data->remoteAddr = QAPI_CLONE(SocketAddress, remoteAddr);

    trace_qio_channel_socket_dgram_async(ioc, localAddr, remoteAddr);
    qio_task_run_in_thread(task,
                           qio_channel_socket_dgram_worker,
                           data,
                           qio_channel_socket_dgram_worker_free);
}


QIOChannelSocket *
qio_channel_socket_accept(QIOChannelSocket *ioc,
                          Error **errp)
{
    QIOChannelSocket *cioc;

    cioc = QIO_CHANNEL_SOCKET(object_new(TYPE_QIO_CHANNEL_SOCKET));
    cioc->fd = -1;
    cioc->remoteAddrLen = sizeof(ioc->remoteAddr);
    cioc->localAddrLen = sizeof(ioc->localAddr);

#ifdef WIN32
    QIO_CHANNEL(cioc)->event = CreateEvent(NULL, FALSE, FALSE, NULL);
#endif


 retry:
    trace_qio_channel_socket_accept(ioc);
    cioc->fd = qemu_accept(ioc->fd, (struct sockaddr *)&cioc->remoteAddr,
                           &cioc->remoteAddrLen);
    if (cioc->fd < 0) {
        trace_qio_channel_socket_accept_fail(ioc);
        if (errno == EINTR) {
            goto retry;
        }
        goto error;
    }

    if (getsockname(cioc->fd, (struct sockaddr *)&cioc->localAddr,
                    &cioc->localAddrLen) < 0) {
        error_setg_errno(errp, errno,
                         "Unable to query local socket address");
        goto error;
    }

#ifndef WIN32
    if (cioc->localAddr.ss_family == AF_UNIX) {
        QIO_CHANNEL(cioc)->features |= (1 << QIO_CHANNEL_FEATURE_FD_PASS);
    }
#endif /* WIN32 */

    trace_qio_channel_socket_accept_complete(ioc, cioc, cioc->fd);
    return cioc;

 error:
    object_unref(OBJECT(cioc));
    return NULL;
}

static void qio_channel_socket_init(Object *obj)
{
    QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(obj);
    ioc->fd = -1;
}

static void qio_channel_socket_finalize(Object *obj)
{
    QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(obj);

    if (ioc->fd != -1) {
        if (QIO_CHANNEL(ioc)->features & QIO_CHANNEL_FEATURE_LISTEN) {
            Error *err = NULL;

            socket_listen_cleanup(ioc->fd, &err);
            if (err) {
                error_report_err(err);
                err = NULL;
            }
        }
#ifdef WIN32
        WSAEventSelect(ioc->fd, NULL, 0);
#endif
        closesocket(ioc->fd);
        ioc->fd = -1;
    }
}


#ifndef WIN32
static void qio_channel_socket_copy_fds(struct msghdr *msg,
                                        int **fds, size_t *nfds)
{
    struct cmsghdr *cmsg;

    *nfds = 0;
    *fds = NULL;

    for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg)) {
        int fd_size, i;
        int gotfds;

        if (cmsg->cmsg_len < CMSG_LEN(sizeof(int)) ||
            cmsg->cmsg_level != SOL_SOCKET ||
            cmsg->cmsg_type != SCM_RIGHTS) {
            continue;
        }

        fd_size = cmsg->cmsg_len - CMSG_LEN(0);

        if (!fd_size) {
            continue;
        }

        gotfds = fd_size / sizeof(int);
        *fds = g_renew(int, *fds, *nfds + gotfds);
        memcpy(*fds + *nfds, CMSG_DATA(cmsg), fd_size);

        for (i = 0; i < gotfds; i++) {
            int fd = (*fds)[*nfds + i];
            if (fd < 0) {
                continue;
            }

            /* O_NONBLOCK is preserved across SCM_RIGHTS so reset it */
            qemu_set_block(fd);

#ifndef MSG_CMSG_CLOEXEC
            qemu_set_cloexec(fd);
#endif
        }
        *nfds += gotfds;
    }
}


static ssize_t qio_channel_socket_readv(QIOChannel *ioc,
                                        const struct iovec *iov,
                                        size_t niov,
                                        int **fds,
                                        size_t *nfds,
                                        Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    ssize_t ret;
    struct msghdr msg = { NULL, };
    char control[CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS)];
    int sflags = 0;

    memset(control, 0, CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS));

#ifdef MSG_CMSG_CLOEXEC
    sflags |= MSG_CMSG_CLOEXEC;
#endif

    msg.msg_iov = (struct iovec *)iov;
    msg.msg_iovlen = niov;
    if (fds && nfds) {
        msg.msg_control = control;
        msg.msg_controllen = sizeof(control);
    }

 retry:
    ret = recvmsg(sioc->fd, &msg, sflags);
    if (ret < 0) {
        if (errno == EAGAIN) {
            return QIO_CHANNEL_ERR_BLOCK;
        }
        if (errno == EINTR) {
            goto retry;
        }

        error_setg_errno(errp, errno,
                         "Unable to read from socket");
        return -1;
    }

    if (fds && nfds) {
        qio_channel_socket_copy_fds(&msg, fds, nfds);
    }

    return ret;
}

static ssize_t qio_channel_socket_writev(QIOChannel *ioc,
                                         const struct iovec *iov,
                                         size_t niov,
                                         int *fds,
                                         size_t nfds,
                                         Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    ssize_t ret;
    struct msghdr msg = { NULL, };
    char control[CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS)];
    size_t fdsize = sizeof(int) * nfds;
    struct cmsghdr *cmsg;

    memset(control, 0, CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS));

    msg.msg_iov = (struct iovec *)iov;
    msg.msg_iovlen = niov;

    if (nfds) {
        if (nfds > SOCKET_MAX_FDS) {
            error_setg_errno(errp, EINVAL,
                             "Only %d FDs can be sent, got %zu",
                             SOCKET_MAX_FDS, nfds);
            return -1;
        }

        msg.msg_control = control;
        msg.msg_controllen = CMSG_SPACE(sizeof(int) * nfds);

        cmsg = CMSG_FIRSTHDR(&msg);
        cmsg->cmsg_len = CMSG_LEN(fdsize);
        cmsg->cmsg_level = SOL_SOCKET;
        cmsg->cmsg_type = SCM_RIGHTS;
        memcpy(CMSG_DATA(cmsg), fds, fdsize);
    }

 retry:
    ret = sendmsg(sioc->fd, &msg, 0);
    if (ret <= 0) {
        if (errno == EAGAIN) {
            return QIO_CHANNEL_ERR_BLOCK;
        }
        if (errno == EINTR) {
            goto retry;
        }
        error_setg_errno(errp, errno,
                         "Unable to write to socket");
        return -1;
    }
    return ret;
}
#else /* WIN32 */
static ssize_t qio_channel_socket_readv(QIOChannel *ioc,
                                        const struct iovec *iov,
                                        size_t niov,
                                        int **fds,
                                        size_t *nfds,
                                        Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    ssize_t done = 0;
    ssize_t i;

    for (i = 0; i < niov; i++) {
        ssize_t ret;
    retry:
        ret = recv(sioc->fd,
                   iov[i].iov_base,
                   iov[i].iov_len,
                   0);
        if (ret < 0) {
            if (errno == EAGAIN) {
                if (done) {
                    return done;
                } else {
                    return QIO_CHANNEL_ERR_BLOCK;
                }
            } else if (errno == EINTR) {
                goto retry;
            } else {
                error_setg_errno(errp, errno,
                                 "Unable to read from socket");
                return -1;
            }
        }
        done += ret;
        if (ret < iov[i].iov_len) {
            return done;
        }
    }

    return done;
}

static ssize_t qio_channel_socket_writev(QIOChannel *ioc,
                                         const struct iovec *iov,
                                         size_t niov,
                                         int *fds,
                                         size_t nfds,
                                         Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    ssize_t done = 0;
    ssize_t i;

    for (i = 0; i < niov; i++) {
        ssize_t ret;
    retry:
        ret = send(sioc->fd,
                   iov[i].iov_base,
                   iov[i].iov_len,
                   0);
        if (ret < 0) {
            if (errno == EAGAIN) {
                if (done) {
                    return done;
                } else {
                    return QIO_CHANNEL_ERR_BLOCK;
                }
            } else if (errno == EINTR) {
                goto retry;
            } else {
                error_setg_errno(errp, errno,
                                 "Unable to write to socket");
                return -1;
            }
        }
        done += ret;
        if (ret < iov[i].iov_len) {
            return done;
        }
    }

    return done;
}
#endif /* WIN32 */

static int
qio_channel_socket_set_blocking(QIOChannel *ioc,
                                bool enabled,
                                Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);

    if (enabled) {
        qemu_set_block(sioc->fd);
    } else {
        qemu_set_nonblock(sioc->fd);
#ifdef WIN32
        WSAEventSelect(sioc->fd, ioc->event,
                       FD_READ | FD_ACCEPT | FD_CLOSE |
                       FD_CONNECT | FD_WRITE | FD_OOB);
#endif
    }
    return 0;
}


static void
qio_channel_socket_set_delay(QIOChannel *ioc,
                             bool enabled)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    int v = enabled ? 0 : 1;

    qemu_setsockopt(sioc->fd,
                    IPPROTO_TCP, TCP_NODELAY,
                    &v, sizeof(v));
}


static void
qio_channel_socket_set_cork(QIOChannel *ioc,
                            bool enabled)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    int v = enabled ? 1 : 0;

    socket_set_cork(sioc->fd, v);
}


static int
qio_channel_socket_close(QIOChannel *ioc,
                         Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);

    if (sioc->fd != -1) {
#ifdef WIN32
        WSAEventSelect(sioc->fd, NULL, 0);
#endif
        if (closesocket(sioc->fd) < 0) {
            sioc->fd = -1;
            error_setg_errno(errp, errno,
                             "Unable to close socket");
            return -1;
        }
        sioc->fd = -1;
    }
    return 0;
}

static int
qio_channel_socket_shutdown(QIOChannel *ioc,
                            QIOChannelShutdown how,
                            Error **errp)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    int sockhow;

    switch (how) {
    case QIO_CHANNEL_SHUTDOWN_READ:
        sockhow = SHUT_RD;
        break;
    case QIO_CHANNEL_SHUTDOWN_WRITE:
        sockhow = SHUT_WR;
        break;
    case QIO_CHANNEL_SHUTDOWN_BOTH:
    default:
        sockhow = SHUT_RDWR;
        break;
    }

    if (shutdown(sioc->fd, sockhow) < 0) {
        error_setg_errno(errp, errno,
                         "Unable to shutdown socket");
        return -1;
    }
    return 0;
}

static GSource *qio_channel_socket_create_watch(QIOChannel *ioc,
                                                GIOCondition condition)
{
    QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
    return qio_channel_create_socket_watch(ioc,
                                           sioc->fd,
                                           condition);
}

static void qio_channel_socket_class_init(ObjectClass *klass,
                                          void *class_data G_GNUC_UNUSED)
{
    QIOChannelClass *ioc_klass = QIO_CHANNEL_CLASS(klass);

    ioc_klass->io_writev = qio_channel_socket_writev;
    ioc_klass->io_readv = qio_channel_socket_readv;
    ioc_klass->io_set_blocking = qio_channel_socket_set_blocking;
    ioc_klass->io_close = qio_channel_socket_close;
    ioc_klass->io_shutdown = qio_channel_socket_shutdown;
    ioc_klass->io_set_cork = qio_channel_socket_set_cork;
    ioc_klass->io_set_delay = qio_channel_socket_set_delay;
    ioc_klass->io_create_watch = qio_channel_socket_create_watch;
}

static const TypeInfo qio_channel_socket_info = {
    .parent = TYPE_QIO_CHANNEL,
    .name = TYPE_QIO_CHANNEL_SOCKET,
    .instance_size = sizeof(QIOChannelSocket),
    .instance_init = qio_channel_socket_init,
    .instance_finalize = qio_channel_socket_finalize,
    .class_init = qio_channel_socket_class_init,
};

static void qio_channel_socket_register_types(void)
{
    type_register_static(&qio_channel_socket_info);
}

type_init(qio_channel_socket_register_types);
Commit	Line	Data
559607ea DB	1	/*
	2	* QEMU I/O channels sockets driver
	3	*
	4	* Copyright (c) 2015 Red Hat, Inc.
	5	*
	6	* This library is free software; you can redistribute it and/or
	7	* modify it under the terms of the GNU Lesser General Public
	8	* License as published by the Free Software Foundation; either
	9	* version 2 of the License, or (at your option) any later version.
	10	*
	11	* This library is distributed in the hope that it will be useful,
	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	14	* Lesser General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU Lesser General Public
	17	* License along with this library; if not, see <http://www.gnu.org/licenses/>.
	18	*
	19	*/
	20
cae9fc56	21	#include "qemu/osdep.h"
da34e65c	22	#include "qapi/error.h"
559607ea DB	23	#include "io/channel-socket.h"
	24	#include "io/channel-watch.h"
	25	#include "trace.h"
37f9e0a2	26	#include "qapi/clone-visitor.h"
559607ea DB	27
	28	#define SOCKET_MAX_FDS 16
	29
	30	SocketAddress *
	31	qio_channel_socket_get_local_address(QIOChannelSocket *ioc,
	32	Error **errp)
	33	{
	34	return socket_sockaddr_to_address(&ioc->localAddr,
	35	ioc->localAddrLen,
	36	errp);
	37	}
	38
	39	SocketAddress *
	40	qio_channel_socket_get_remote_address(QIOChannelSocket *ioc,
	41	Error **errp)
	42	{
	43	return socket_sockaddr_to_address(&ioc->remoteAddr,
	44	ioc->remoteAddrLen,
	45	errp);
	46	}
	47
	48	QIOChannelSocket *
	49	qio_channel_socket_new(void)
	50	{
	51	QIOChannelSocket *sioc;
	52	QIOChannel *ioc;
	53
	54	sioc = QIO_CHANNEL_SOCKET(object_new(TYPE_QIO_CHANNEL_SOCKET));
	55	sioc->fd = -1;
	56
	57	ioc = QIO_CHANNEL(sioc);
	58	ioc->features \|= (1 << QIO_CHANNEL_FEATURE_SHUTDOWN);
	59
a5897205 PB	60	#ifdef WIN32
	61	ioc->event = CreateEvent(NULL, FALSE, FALSE, NULL);
	62	#endif
	63
559607ea DB	64	trace_qio_channel_socket_new(sioc);
	65
	66	return sioc;
	67	}
	68
	69
	70	static int
	71	qio_channel_socket_set_fd(QIOChannelSocket *sioc,
	72	int fd,
	73	Error **errp)
	74	{
3fa27a9a MAL	75	int val;
	76	socklen_t len = sizeof(val);
	77
559607ea DB	78	if (sioc->fd != -1) {
	79	error_setg(errp, "Socket is already open");
	80	return -1;
	81	}
	82
	83	sioc->fd = fd;
	84	sioc->remoteAddrLen = sizeof(sioc->remoteAddr);
	85	sioc->localAddrLen = sizeof(sioc->localAddr);
	86
	87
	88	if (getpeername(fd, (struct sockaddr *)&sioc->remoteAddr,
	89	&sioc->remoteAddrLen) < 0) {
b16a44e1	90	if (errno == ENOTCONN) {
559607ea DB	91	memset(&sioc->remoteAddr, 0, sizeof(sioc->remoteAddr));
	92	sioc->remoteAddrLen = sizeof(sioc->remoteAddr);
	93	} else {
b16a44e1	94	error_setg_errno(errp, errno,
559607ea DB	95	"Unable to query remote socket address");
	96	goto error;
	97	}
	98	}
	99
	100	if (getsockname(fd, (struct sockaddr *)&sioc->localAddr,
	101	&sioc->localAddrLen) < 0) {
b16a44e1	102	error_setg_errno(errp, errno,
559607ea DB	103	"Unable to query local socket address");
	104	goto error;
	105	}
	106
	107	#ifndef WIN32
	108	if (sioc->localAddr.ss_family == AF_UNIX) {
	109	QIOChannel *ioc = QIO_CHANNEL(sioc);
	110	ioc->features \|= (1 << QIO_CHANNEL_FEATURE_FD_PASS);
	111	}
	112	#endif /* WIN32 */
3fa27a9a MAL	113	if (getsockopt(fd, SOL_SOCKET, SO_ACCEPTCONN, &val, &len) == 0 && val) {
	114	QIOChannel *ioc = QIO_CHANNEL(sioc);
	115	ioc->features \|= (1 << QIO_CHANNEL_FEATURE_LISTEN);
	116	}
559607ea DB	117
	118	return 0;
	119
	120	error:
	121	sioc->fd = -1; /* Let the caller close FD on failure */
	122	return -1;
	123	}
	124
	125	QIOChannelSocket *
	126	qio_channel_socket_new_fd(int fd,
	127	Error **errp)
	128	{
	129	QIOChannelSocket *ioc;
	130
	131	ioc = qio_channel_socket_new();
	132	if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
	133	object_unref(OBJECT(ioc));
	134	return NULL;
	135	}
	136
	137	trace_qio_channel_socket_new_fd(ioc, fd);
	138
	139	return ioc;
	140	}
	141
	142
	143	int qio_channel_socket_connect_sync(QIOChannelSocket *ioc,
	144	SocketAddress *addr,
	145	Error **errp)
	146	{
	147	int fd;
	148
	149	trace_qio_channel_socket_connect_sync(ioc, addr);
	150	fd = socket_connect(addr, errp, NULL, NULL);
	151	if (fd < 0) {
	152	trace_qio_channel_socket_connect_fail(ioc);
	153	return -1;
	154	}
	155
	156	trace_qio_channel_socket_connect_complete(ioc, fd);
	157	if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
	158	close(fd);
	159	return -1;
	160	}
	161
	162	return 0;
	163	}
	164
	165
	166	static int qio_channel_socket_connect_worker(QIOTask *task,
	167	Error **errp,
	168	gpointer opaque)
	169	{
	170	QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
	171	SocketAddress *addr = opaque;
	172	int ret;
	173
	174	ret = qio_channel_socket_connect_sync(ioc,
	175	addr,
	176	errp);
	177
	178	object_unref(OBJECT(ioc));
	179	return ret;
	180	}
181
182
183	void qio_channel_socket_connect_async(QIOChannelSocket *ioc,
184	SocketAddress *addr,
185	QIOTaskFunc callback,
186	gpointer opaque,
187	GDestroyNotify destroy)
188	{
189	QIOTask *task = qio_task_new(
190	OBJECT(ioc), callback, opaque, destroy);
191	SocketAddress *addrCopy;
192
37f9e0a2	193	addrCopy = QAPI_CLONE(SocketAddress, addr);
559607ea DB	194
	195	/* socket_connect() does a non-blocking connect(), but it
	196	* still blocks in DNS lookups, so we must use a thread */
	197	trace_qio_channel_socket_connect_async(ioc, addr);
	198	qio_task_run_in_thread(task,
	199	qio_channel_socket_connect_worker,
	200	addrCopy,
	201	(GDestroyNotify)qapi_free_SocketAddress);
	202	}
	203
	204
	205	int qio_channel_socket_listen_sync(QIOChannelSocket *ioc,
	206	SocketAddress *addr,
	207	Error **errp)
	208	{
	209	int fd;
	210
	211	trace_qio_channel_socket_listen_sync(ioc, addr);
	212	fd = socket_listen(addr, errp);
	213	if (fd < 0) {
	214	trace_qio_channel_socket_listen_fail(ioc);
	215	return -1;
	216	}
	217
	218	trace_qio_channel_socket_listen_complete(ioc, fd);
	219	if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
	220	close(fd);
	221	return -1;
	222	}
	223
	224	return 0;
	225	}
	226
	227
	228	static int qio_channel_socket_listen_worker(QIOTask *task,
	229	Error **errp,
	230	gpointer opaque)
	231	{
	232	QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
	233	SocketAddress *addr = opaque;
	234	int ret;
	235
	236	ret = qio_channel_socket_listen_sync(ioc,
	237	addr,
	238	errp);
	239
	240	object_unref(OBJECT(ioc));
	241	return ret;
	242	}
	243
	244
	245	void qio_channel_socket_listen_async(QIOChannelSocket *ioc,
	246	SocketAddress *addr,
	247	QIOTaskFunc callback,
	248	gpointer opaque,
	249	GDestroyNotify destroy)
	250	{
	251	QIOTask *task = qio_task_new(
	252	OBJECT(ioc), callback, opaque, destroy);
	253	SocketAddress *addrCopy;
	254
37f9e0a2	255	addrCopy = QAPI_CLONE(SocketAddress, addr);
559607ea DB	256
	257	/* socket_listen() blocks in DNS lookups, so we must use a thread */
	258	trace_qio_channel_socket_listen_async(ioc, addr);
	259	qio_task_run_in_thread(task,
	260	qio_channel_socket_listen_worker,
	261	addrCopy,
	262	(GDestroyNotify)qapi_free_SocketAddress);
	263	}
	264
	265
	266	int qio_channel_socket_dgram_sync(QIOChannelSocket *ioc,
	267	SocketAddress *localAddr,
	268	SocketAddress *remoteAddr,
	269	Error **errp)
	270	{
	271	int fd;
	272
	273	trace_qio_channel_socket_dgram_sync(ioc, localAddr, remoteAddr);
150dcd1a	274	fd = socket_dgram(remoteAddr, localAddr, errp);
559607ea DB	275	if (fd < 0) {
	276	trace_qio_channel_socket_dgram_fail(ioc);
	277	return -1;
	278	}
	279
	280	trace_qio_channel_socket_dgram_complete(ioc, fd);
	281	if (qio_channel_socket_set_fd(ioc, fd, errp) < 0) {
	282	close(fd);
	283	return -1;
	284	}
	285
	286	return 0;
	287	}
	288
	289
	290	struct QIOChannelSocketDGramWorkerData {
	291	SocketAddress *localAddr;
	292	SocketAddress *remoteAddr;
	293	};
	294
	295
	296	static void qio_channel_socket_dgram_worker_free(gpointer opaque)
	297	{
	298	struct QIOChannelSocketDGramWorkerData *data = opaque;
	299	qapi_free_SocketAddress(data->localAddr);
	300	qapi_free_SocketAddress(data->remoteAddr);
	301	g_free(data);
	302	}
	303
	304	static int qio_channel_socket_dgram_worker(QIOTask *task,
	305	Error **errp,
	306	gpointer opaque)
	307	{
	308	QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(qio_task_get_source(task));
	309	struct QIOChannelSocketDGramWorkerData *data = opaque;
	310	int ret;
	311
	312	/* socket_dgram() blocks in DNS lookups, so we must use a thread */
	313	ret = qio_channel_socket_dgram_sync(ioc,
	314	data->localAddr,
	315	data->remoteAddr,
	316	errp);
	317
	318	object_unref(OBJECT(ioc));
	319	return ret;
	320	}
	321
	322
	323	void qio_channel_socket_dgram_async(QIOChannelSocket *ioc,
	324	SocketAddress *localAddr,
	325	SocketAddress *remoteAddr,
	326	QIOTaskFunc callback,
	327	gpointer opaque,
	328	GDestroyNotify destroy)
	329	{
	330	QIOTask *task = qio_task_new(
	331	OBJECT(ioc), callback, opaque, destroy);
	332	struct QIOChannelSocketDGramWorkerData *data = g_new0(
	333	struct QIOChannelSocketDGramWorkerData, 1);
	334
37f9e0a2 EB	335	data->localAddr = QAPI_CLONE(SocketAddress, localAddr);
37f9e0a2 EB	336	data->remoteAddr = QAPI_CLONE(SocketAddress, remoteAddr);
559607ea DB	337
	338	trace_qio_channel_socket_dgram_async(ioc, localAddr, remoteAddr);
	339	qio_task_run_in_thread(task,
	340	qio_channel_socket_dgram_worker,
	341	data,
	342	qio_channel_socket_dgram_worker_free);
	343	}
	344
	345
	346	QIOChannelSocket *
	347	qio_channel_socket_accept(QIOChannelSocket *ioc,
	348	Error **errp)
	349	{
	350	QIOChannelSocket *cioc;
	351
	352	cioc = QIO_CHANNEL_SOCKET(object_new(TYPE_QIO_CHANNEL_SOCKET));
	353	cioc->fd = -1;
	354	cioc->remoteAddrLen = sizeof(ioc->remoteAddr);
	355	cioc->localAddrLen = sizeof(ioc->localAddr);
	356
a5897205 PB	357	#ifdef WIN32
	358	QIO_CHANNEL(cioc)->event = CreateEvent(NULL, FALSE, FALSE, NULL);
	359	#endif
	360
	361
559607ea DB	362	retry:
559607ea DB	363	trace_qio_channel_socket_accept(ioc);
de7971ff DB	364	cioc->fd = qemu_accept(ioc->fd, (struct sockaddr *)&cioc->remoteAddr,
de7971ff DB	365	&cioc->remoteAddrLen);
559607ea DB	366	if (cioc->fd < 0) {
559607ea DB	367	trace_qio_channel_socket_accept_fail(ioc);
b16a44e1	368	if (errno == EINTR) {
559607ea DB	369	goto retry;
	370	}
	371	goto error;
	372	}
	373
bead5994 DB	374	if (getsockname(cioc->fd, (struct sockaddr *)&cioc->localAddr,
bead5994 DB	375	&cioc->localAddrLen) < 0) {
b16a44e1	376	error_setg_errno(errp, errno,
559607ea DB	377	"Unable to query local socket address");
	378	goto error;
	379	}
	380
bead5994 DB	381	#ifndef WIN32
	382	if (cioc->localAddr.ss_family == AF_UNIX) {
	383	QIO_CHANNEL(cioc)->features \|= (1 << QIO_CHANNEL_FEATURE_FD_PASS);
	384	}
	385	#endif /* WIN32 */
	386
559607ea DB	387	trace_qio_channel_socket_accept_complete(ioc, cioc, cioc->fd);
	388	return cioc;
	389
	390	error:
	391	object_unref(OBJECT(cioc));
	392	return NULL;
	393	}
	394
	395	static void qio_channel_socket_init(Object *obj)
	396	{
	397	QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(obj);
	398	ioc->fd = -1;
	399	}
	400
	401	static void qio_channel_socket_finalize(Object *obj)
	402	{
	403	QIOChannelSocket *ioc = QIO_CHANNEL_SOCKET(obj);
74b6ce43	404
559607ea	405	if (ioc->fd != -1) {
74b6ce43 MAL	406	if (QIO_CHANNEL(ioc)->features & QIO_CHANNEL_FEATURE_LISTEN) {
	407	Error *err = NULL;
	408
	409	socket_listen_cleanup(ioc->fd, &err);
	410	if (err) {
	411	error_report_err(err);
	412	err = NULL;
	413	}
	414	}
a5897205 PB	415	#ifdef WIN32
	416	WSAEventSelect(ioc->fd, NULL, 0);
	417	#endif
	418	closesocket(ioc->fd);
559607ea DB	419	ioc->fd = -1;
	420	}
	421	}
	422
	423
	424	#ifndef WIN32
	425	static void qio_channel_socket_copy_fds(struct msghdr *msg,
	426	int *fds, size_t nfds)
	427	{
	428	struct cmsghdr *cmsg;
	429
	430	*nfds = 0;
	431	*fds = NULL;
	432
	433	for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg)) {
	434	int fd_size, i;
	435	int gotfds;
	436
	437	if (cmsg->cmsg_len < CMSG_LEN(sizeof(int)) \|\|
	438	cmsg->cmsg_level != SOL_SOCKET \|\|
	439	cmsg->cmsg_type != SCM_RIGHTS) {
	440	continue;
	441	}
	442
	443	fd_size = cmsg->cmsg_len - CMSG_LEN(0);
	444
	445	if (!fd_size) {
	446	continue;
	447	}
	448
	449	gotfds = fd_size / sizeof(int);
	450	fds = g_renew(int, fds, *nfds + gotfds);
	451	memcpy(fds + nfds, CMSG_DATA(cmsg), fd_size);
	452
	453	for (i = 0; i < gotfds; i++) {
	454	int fd = (fds)[nfds + i];
	455	if (fd < 0) {
	456	continue;
	457	}
	458
	459	/* O_NONBLOCK is preserved across SCM_RIGHTS so reset it */
	460	qemu_set_block(fd);
	461
	462	#ifndef MSG_CMSG_CLOEXEC
	463	qemu_set_cloexec(fd);
	464	#endif
	465	}
	466	*nfds += gotfds;
	467	}
	468	}
	469
	470
	471	static ssize_t qio_channel_socket_readv(QIOChannel *ioc,
	472	const struct iovec *iov,
	473	size_t niov,
	474	int **fds,
	475	size_t *nfds,
	476	Error **errp)
	477	{
	478	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	479	ssize_t ret;
	480	struct msghdr msg = { NULL, };
	481	char control[CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS)];
	482	int sflags = 0;
483
ccf1e2dc DB	484	memset(control, 0, CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS));
ccf1e2dc DB	485
559607ea DB	486	#ifdef MSG_CMSG_CLOEXEC
	487	sflags \|= MSG_CMSG_CLOEXEC;
	488	#endif
	489
	490	msg.msg_iov = (struct iovec *)iov;
	491	msg.msg_iovlen = niov;
	492	if (fds && nfds) {
	493	msg.msg_control = control;
	494	msg.msg_controllen = sizeof(control);
	495	}
	496
	497	retry:
	498	ret = recvmsg(sioc->fd, &msg, sflags);
	499	if (ret < 0) {
b16a44e1	500	if (errno == EAGAIN) {
559607ea DB	501	return QIO_CHANNEL_ERR_BLOCK;
559607ea DB	502	}
b16a44e1	503	if (errno == EINTR) {
559607ea DB	504	goto retry;
	505	}
	506
b16a44e1	507	error_setg_errno(errp, errno,
559607ea DB	508	"Unable to read from socket");
	509	return -1;
	510	}
	511
	512	if (fds && nfds) {
	513	qio_channel_socket_copy_fds(&msg, fds, nfds);
	514	}
	515
	516	return ret;
	517	}
	518
	519	static ssize_t qio_channel_socket_writev(QIOChannel *ioc,
	520	const struct iovec *iov,
	521	size_t niov,
	522	int *fds,
	523	size_t nfds,
	524	Error **errp)
	525	{
	526	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	527	ssize_t ret;
	528	struct msghdr msg = { NULL, };
ccf1e2dc	529	char control[CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS)];
7b3c618a DB	530	size_t fdsize = sizeof(int) * nfds;
7b3c618a DB	531	struct cmsghdr *cmsg;
559607ea	532
ccf1e2dc DB	533	memset(control, 0, CMSG_SPACE(sizeof(int) * SOCKET_MAX_FDS));
ccf1e2dc DB	534
559607ea DB	535	msg.msg_iov = (struct iovec *)iov;
	536	msg.msg_iovlen = niov;
	537
	538	if (nfds) {
559607ea	539	if (nfds > SOCKET_MAX_FDS) {
cc75a50c	540	error_setg_errno(errp, EINVAL,
559607ea DB	541	"Only %d FDs can be sent, got %zu",
	542	SOCKET_MAX_FDS, nfds);
	543	return -1;
	544	}
	545
	546	msg.msg_control = control;
	547	msg.msg_controllen = CMSG_SPACE(sizeof(int) * nfds);
	548
	549	cmsg = CMSG_FIRSTHDR(&msg);
	550	cmsg->cmsg_len = CMSG_LEN(fdsize);
	551	cmsg->cmsg_level = SOL_SOCKET;
	552	cmsg->cmsg_type = SCM_RIGHTS;
	553	memcpy(CMSG_DATA(cmsg), fds, fdsize);
	554	}
	555
	556	retry:
	557	ret = sendmsg(sioc->fd, &msg, 0);
	558	if (ret <= 0) {
b16a44e1	559	if (errno == EAGAIN) {
559607ea DB	560	return QIO_CHANNEL_ERR_BLOCK;
559607ea DB	561	}
b16a44e1	562	if (errno == EINTR) {
559607ea DB	563	goto retry;
559607ea DB	564	}
b16a44e1	565	error_setg_errno(errp, errno,
559607ea DB	566	"Unable to write to socket");
	567	return -1;
	568	}
	569	return ret;
	570	}
	571	#else /* WIN32 */
	572	static ssize_t qio_channel_socket_readv(QIOChannel *ioc,
	573	const struct iovec *iov,
	574	size_t niov,
	575	int **fds,
	576	size_t *nfds,
	577	Error **errp)
	578	{
	579	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	580	ssize_t done = 0;
	581	ssize_t i;
	582
	583	for (i = 0; i < niov; i++) {
	584	ssize_t ret;
	585	retry:
	586	ret = recv(sioc->fd,
	587	iov[i].iov_base,
	588	iov[i].iov_len,
	589	0);
	590	if (ret < 0) {
b16a44e1	591	if (errno == EAGAIN) {
559607ea DB	592	if (done) {
	593	return done;
	594	} else {
	595	return QIO_CHANNEL_ERR_BLOCK;
	596	}
b16a44e1	597	} else if (errno == EINTR) {
559607ea DB	598	goto retry;
559607ea DB	599	} else {
b16a44e1	600	error_setg_errno(errp, errno,
5151d23e	601	"Unable to read from socket");
559607ea DB	602	return -1;
	603	}
	604	}
	605	done += ret;
	606	if (ret < iov[i].iov_len) {
	607	return done;
	608	}
	609	}
	610
	611	return done;
	612	}
	613
	614	static ssize_t qio_channel_socket_writev(QIOChannel *ioc,
	615	const struct iovec *iov,
	616	size_t niov,
	617	int *fds,
	618	size_t nfds,
	619	Error **errp)
	620	{
	621	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	622	ssize_t done = 0;
	623	ssize_t i;
	624
	625	for (i = 0; i < niov; i++) {
	626	ssize_t ret;
	627	retry:
	628	ret = send(sioc->fd,
	629	iov[i].iov_base,
	630	iov[i].iov_len,
	631	0);
	632	if (ret < 0) {
b16a44e1	633	if (errno == EAGAIN) {
559607ea DB	634	if (done) {
	635	return done;
	636	} else {
	637	return QIO_CHANNEL_ERR_BLOCK;
	638	}
b16a44e1	639	} else if (errno == EINTR) {
559607ea DB	640	goto retry;
559607ea DB	641	} else {
b16a44e1	642	error_setg_errno(errp, errno,
559607ea DB	643	"Unable to write to socket");
	644	return -1;
	645	}
	646	}
	647	done += ret;
	648	if (ret < iov[i].iov_len) {
	649	return done;
	650	}
	651	}
	652
	653	return done;
	654	}
	655	#endif /* WIN32 */
	656
	657	static int
	658	qio_channel_socket_set_blocking(QIOChannel *ioc,
	659	bool enabled,
	660	Error **errp)
	661	{
	662	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	663
	664	if (enabled) {
	665	qemu_set_block(sioc->fd);
	666	} else {
	667	qemu_set_nonblock(sioc->fd);
a5897205 PB	668	#ifdef WIN32
	669	WSAEventSelect(sioc->fd, ioc->event,
	670	FD_READ \| FD_ACCEPT \| FD_CLOSE \|
	671	FD_CONNECT \| FD_WRITE \| FD_OOB);
	672	#endif
559607ea DB	673	}
	674	return 0;
	675	}
	676
	677
	678	static void
	679	qio_channel_socket_set_delay(QIOChannel *ioc,
	680	bool enabled)
	681	{
	682	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	683	int v = enabled ? 0 : 1;
	684
	685	qemu_setsockopt(sioc->fd,
	686	IPPROTO_TCP, TCP_NODELAY,
	687	&v, sizeof(v));
	688	}
	689
	690
	691	static void
	692	qio_channel_socket_set_cork(QIOChannel *ioc,
	693	bool enabled)
	694	{
	695	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	696	int v = enabled ? 1 : 0;
	697
	698	socket_set_cork(sioc->fd, v);
	699	}
	700
	701
	702	static int
	703	qio_channel_socket_close(QIOChannel *ioc,
	704	Error **errp)
	705	{
	706	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	707
a5897205 PB	708	if (sioc->fd != -1) {
	709	#ifdef WIN32
	710	WSAEventSelect(sioc->fd, NULL, 0);
	711	#endif
	712	if (closesocket(sioc->fd) < 0) {
	713	sioc->fd = -1;
b16a44e1	714	error_setg_errno(errp, errno,
a5897205 PB	715	"Unable to close socket");
	716	return -1;
	717	}
559607ea	718	sioc->fd = -1;
559607ea	719	}
559607ea DB	720	return 0;
	721	}
	722
	723	static int
	724	qio_channel_socket_shutdown(QIOChannel *ioc,
	725	QIOChannelShutdown how,
	726	Error **errp)
	727	{
	728	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
	729	int sockhow;
	730
	731	switch (how) {
	732	case QIO_CHANNEL_SHUTDOWN_READ:
	733	sockhow = SHUT_RD;
	734	break;
	735	case QIO_CHANNEL_SHUTDOWN_WRITE:
	736	sockhow = SHUT_WR;
	737	break;
	738	case QIO_CHANNEL_SHUTDOWN_BOTH:
	739	default:
	740	sockhow = SHUT_RDWR;
	741	break;
	742	}
	743
	744	if (shutdown(sioc->fd, sockhow) < 0) {
b16a44e1	745	error_setg_errno(errp, errno,
559607ea DB	746	"Unable to shutdown socket");
	747	return -1;
	748	}
	749	return 0;
	750	}
	751
	752	static GSource qio_channel_socket_create_watch(QIOChannel ioc,
	753	GIOCondition condition)
	754	{
	755	QIOChannelSocket *sioc = QIO_CHANNEL_SOCKET(ioc);
b83b68a0 PB	756	return qio_channel_create_socket_watch(ioc,
	757	sioc->fd,
	758	condition);
559607ea DB	759	}
	760
	761	static void qio_channel_socket_class_init(ObjectClass *klass,
	762	void *class_data G_GNUC_UNUSED)
	763	{
	764	QIOChannelClass *ioc_klass = QIO_CHANNEL_CLASS(klass);
	765
	766	ioc_klass->io_writev = qio_channel_socket_writev;
	767	ioc_klass->io_readv = qio_channel_socket_readv;
	768	ioc_klass->io_set_blocking = qio_channel_socket_set_blocking;
	769	ioc_klass->io_close = qio_channel_socket_close;
	770	ioc_klass->io_shutdown = qio_channel_socket_shutdown;
	771	ioc_klass->io_set_cork = qio_channel_socket_set_cork;
	772	ioc_klass->io_set_delay = qio_channel_socket_set_delay;
	773	ioc_klass->io_create_watch = qio_channel_socket_create_watch;
	774	}
	775
	776	static const TypeInfo qio_channel_socket_info = {
	777	.parent = TYPE_QIO_CHANNEL,
	778	.name = TYPE_QIO_CHANNEL_SOCKET,
	779	.instance_size = sizeof(QIOChannelSocket),
	780	.instance_init = qio_channel_socket_init,
	781	.instance_finalize = qio_channel_socket_finalize,
	782	.class_init = qio_channel_socket_class_init,
	783	};
	784
	785	static void qio_channel_socket_register_types(void)
	786	{
	787	type_register_static(&qio_channel_socket_info);
	788	}
	789
	790	type_init(qio_channel_socket_register_types);