| /* |
| * QEMU I/O channels watch helper APIs |
| * |
| * 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 "io/channel-watch.h" |
| |
| typedef struct QIOChannelFDSource QIOChannelFDSource; |
| struct QIOChannelFDSource { |
| GSource parent; |
| GPollFD fd; |
| QIOChannel *ioc; |
| GIOCondition condition; |
| }; |
| |
| |
| #ifdef CONFIG_WIN32 |
| typedef struct QIOChannelSocketSource QIOChannelSocketSource; |
| struct QIOChannelSocketSource { |
| GSource parent; |
| GPollFD fd; |
| QIOChannel *ioc; |
| SOCKET socket; |
| int revents; |
| GIOCondition condition; |
| }; |
| |
| #endif |
| |
| |
| typedef struct QIOChannelFDPairSource QIOChannelFDPairSource; |
| struct QIOChannelFDPairSource { |
| GSource parent; |
| GPollFD fdread; |
| GPollFD fdwrite; |
| QIOChannel *ioc; |
| GIOCondition condition; |
| }; |
| |
| |
| static gboolean |
| qio_channel_fd_source_prepare(GSource *source G_GNUC_UNUSED, |
| gint *timeout) |
| { |
| *timeout = -1; |
| |
| return FALSE; |
| } |
| |
| |
| static gboolean |
| qio_channel_fd_source_check(GSource *source) |
| { |
| QIOChannelFDSource *ssource = (QIOChannelFDSource *)source; |
| |
| return ssource->fd.revents & ssource->condition; |
| } |
| |
| |
| static gboolean |
| qio_channel_fd_source_dispatch(GSource *source, |
| GSourceFunc callback, |
| gpointer user_data) |
| { |
| QIOChannelFunc func = (QIOChannelFunc)callback; |
| QIOChannelFDSource *ssource = (QIOChannelFDSource *)source; |
| |
| return (*func)(ssource->ioc, |
| ssource->fd.revents & ssource->condition, |
| user_data); |
| } |
| |
| |
| static void |
| qio_channel_fd_source_finalize(GSource *source) |
| { |
| QIOChannelFDSource *ssource = (QIOChannelFDSource *)source; |
| |
| object_unref(OBJECT(ssource->ioc)); |
| } |
| |
| |
| #ifdef CONFIG_WIN32 |
| static gboolean |
| qio_channel_socket_source_prepare(GSource *source G_GNUC_UNUSED, |
| gint *timeout) |
| { |
| *timeout = -1; |
| |
| return FALSE; |
| } |
| |
| |
| /* |
| * NB, this impl only works when the socket is in non-blocking |
| * mode on Win32 |
| */ |
| static gboolean |
| qio_channel_socket_source_check(GSource *source) |
| { |
| static struct timeval tv0; |
| |
| QIOChannelSocketSource *ssource = (QIOChannelSocketSource *)source; |
| WSANETWORKEVENTS ev; |
| fd_set rfds, wfds, xfds; |
| |
| if (!ssource->condition) { |
| return 0; |
| } |
| |
| WSAEnumNetworkEvents(ssource->socket, ssource->ioc->event, &ev); |
| |
| FD_ZERO(&rfds); |
| FD_ZERO(&wfds); |
| FD_ZERO(&xfds); |
| if (ssource->condition & G_IO_IN) { |
| FD_SET((SOCKET)ssource->socket, &rfds); |
| } |
| if (ssource->condition & G_IO_OUT) { |
| FD_SET((SOCKET)ssource->socket, &wfds); |
| } |
| if (ssource->condition & G_IO_PRI) { |
| FD_SET((SOCKET)ssource->socket, &xfds); |
| } |
| ssource->revents = 0; |
| if (select(0, &rfds, &wfds, &xfds, &tv0) == 0) { |
| return 0; |
| } |
| |
| if (FD_ISSET(ssource->socket, &rfds)) { |
| ssource->revents |= G_IO_IN; |
| } |
| if (FD_ISSET(ssource->socket, &wfds)) { |
| ssource->revents |= G_IO_OUT; |
| } |
| if (FD_ISSET(ssource->socket, &xfds)) { |
| ssource->revents |= G_IO_PRI; |
| } |
| |
| return ssource->revents; |
| } |
| |
| |
| static gboolean |
| qio_channel_socket_source_dispatch(GSource *source, |
| GSourceFunc callback, |
| gpointer user_data) |
| { |
| QIOChannelFunc func = (QIOChannelFunc)callback; |
| QIOChannelSocketSource *ssource = (QIOChannelSocketSource *)source; |
| |
| return (*func)(ssource->ioc, ssource->revents, user_data); |
| } |
| |
| |
| static void |
| qio_channel_socket_source_finalize(GSource *source) |
| { |
| QIOChannelSocketSource *ssource = (QIOChannelSocketSource *)source; |
| |
| object_unref(OBJECT(ssource->ioc)); |
| } |
| |
| |
| GSourceFuncs qio_channel_socket_source_funcs = { |
| qio_channel_socket_source_prepare, |
| qio_channel_socket_source_check, |
| qio_channel_socket_source_dispatch, |
| qio_channel_socket_source_finalize |
| }; |
| #endif |
| |
| |
| static gboolean |
| qio_channel_fd_pair_source_prepare(GSource *source G_GNUC_UNUSED, |
| gint *timeout) |
| { |
| *timeout = -1; |
| |
| return FALSE; |
| } |
| |
| |
| static gboolean |
| qio_channel_fd_pair_source_check(GSource *source) |
| { |
| QIOChannelFDPairSource *ssource = (QIOChannelFDPairSource *)source; |
| GIOCondition poll_condition = ssource->fdread.revents | |
| ssource->fdwrite.revents; |
| |
| return poll_condition & ssource->condition; |
| } |
| |
| |
| static gboolean |
| qio_channel_fd_pair_source_dispatch(GSource *source, |
| GSourceFunc callback, |
| gpointer user_data) |
| { |
| QIOChannelFunc func = (QIOChannelFunc)callback; |
| QIOChannelFDPairSource *ssource = (QIOChannelFDPairSource *)source; |
| GIOCondition poll_condition = ssource->fdread.revents | |
| ssource->fdwrite.revents; |
| |
| return (*func)(ssource->ioc, |
| poll_condition & ssource->condition, |
| user_data); |
| } |
| |
| |
| static void |
| qio_channel_fd_pair_source_finalize(GSource *source) |
| { |
| QIOChannelFDPairSource *ssource = (QIOChannelFDPairSource *)source; |
| |
| object_unref(OBJECT(ssource->ioc)); |
| } |
| |
| |
| GSourceFuncs qio_channel_fd_source_funcs = { |
| qio_channel_fd_source_prepare, |
| qio_channel_fd_source_check, |
| qio_channel_fd_source_dispatch, |
| qio_channel_fd_source_finalize |
| }; |
| |
| |
| GSourceFuncs qio_channel_fd_pair_source_funcs = { |
| qio_channel_fd_pair_source_prepare, |
| qio_channel_fd_pair_source_check, |
| qio_channel_fd_pair_source_dispatch, |
| qio_channel_fd_pair_source_finalize |
| }; |
| |
| |
| GSource *qio_channel_create_fd_watch(QIOChannel *ioc, |
| int fd, |
| GIOCondition condition) |
| { |
| GSource *source; |
| QIOChannelFDSource *ssource; |
| |
| source = g_source_new(&qio_channel_fd_source_funcs, |
| sizeof(QIOChannelFDSource)); |
| ssource = (QIOChannelFDSource *)source; |
| |
| ssource->ioc = ioc; |
| object_ref(OBJECT(ioc)); |
| |
| ssource->condition = condition; |
| |
| #ifdef CONFIG_WIN32 |
| ssource->fd.fd = (gint64)_get_osfhandle(fd); |
| #else |
| ssource->fd.fd = fd; |
| #endif |
| ssource->fd.events = condition; |
| |
| g_source_add_poll(source, &ssource->fd); |
| |
| return source; |
| } |
| |
| #ifdef CONFIG_WIN32 |
| GSource *qio_channel_create_socket_watch(QIOChannel *ioc, |
| int socket, |
| GIOCondition condition) |
| { |
| GSource *source; |
| QIOChannelSocketSource *ssource; |
| |
| source = g_source_new(&qio_channel_socket_source_funcs, |
| sizeof(QIOChannelSocketSource)); |
| ssource = (QIOChannelSocketSource *)source; |
| |
| ssource->ioc = ioc; |
| object_ref(OBJECT(ioc)); |
| |
| ssource->condition = condition; |
| ssource->socket = socket; |
| ssource->revents = 0; |
| |
| ssource->fd.fd = (gintptr)ioc->event; |
| ssource->fd.events = G_IO_IN; |
| |
| g_source_add_poll(source, &ssource->fd); |
| |
| return source; |
| } |
| #else |
| GSource *qio_channel_create_socket_watch(QIOChannel *ioc, |
| int socket, |
| GIOCondition condition) |
| { |
| return qio_channel_create_fd_watch(ioc, socket, condition); |
| } |
| #endif |
| |
| GSource *qio_channel_create_fd_pair_watch(QIOChannel *ioc, |
| int fdread, |
| int fdwrite, |
| GIOCondition condition) |
| { |
| GSource *source; |
| QIOChannelFDPairSource *ssource; |
| |
| source = g_source_new(&qio_channel_fd_pair_source_funcs, |
| sizeof(QIOChannelFDPairSource)); |
| ssource = (QIOChannelFDPairSource *)source; |
| |
| ssource->ioc = ioc; |
| object_ref(OBJECT(ioc)); |
| |
| ssource->condition = condition; |
| |
| #ifdef CONFIG_WIN32 |
| ssource->fdread.fd = (gint64)_get_osfhandle(fdread); |
| ssource->fdwrite.fd = (gint64)_get_osfhandle(fdwrite); |
| #else |
| ssource->fdread.fd = fdread; |
| ssource->fdwrite.fd = fdwrite; |
| #endif |
| |
| ssource->fdread.events = condition & G_IO_IN; |
| ssource->fdwrite.events = condition & G_IO_OUT; |
| |
| g_source_add_poll(source, &ssource->fdread); |
| g_source_add_poll(source, &ssource->fdwrite); |
| |
| return source; |
| } |