io/channel-watch.c

   1 /*
   2  * QEMU I/O channels watch helper APIs
   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.1 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
  21 #include "qemu/osdep.h"
  22 #include "io/channel-watch.h"
  23
  24 typedef struct QIOChannelFDSource QIOChannelFDSource;
  25 struct QIOChannelFDSource {
  26     GSource parent;
  27     GPollFD fd;
  28     QIOChannel *ioc;
  29     GIOCondition condition;
  30 };
  31
  32
  33 #ifdef CONFIG_WIN32
  34 typedef struct QIOChannelSocketSource QIOChannelSocketSource;
  35 struct QIOChannelSocketSource {
  36     GSource parent;
  37     GPollFD fd;
  38     QIOChannel *ioc;
  39     SOCKET socket;
  40     int revents;
  41     GIOCondition condition;
  42 };
  43
  44 #endif
  45
  46
  47 typedef struct QIOChannelFDPairSource QIOChannelFDPairSource;
  48 struct QIOChannelFDPairSource {
  49     GSource parent;
  50     GPollFD fdread;
  51     GPollFD fdwrite;
  52     QIOChannel *ioc;
  53     GIOCondition condition;
  54 };
  55
  56
  57 static gboolean
  58 qio_channel_fd_source_prepare(GSource *source G_GNUC_UNUSED,
  59                               gint *timeout)
  60 {
  61     *timeout = -1;
  62
  63     return FALSE;
  64 }
  65
  66
  67 static gboolean
  68 qio_channel_fd_source_check(GSource *source)
  69 {
  70     QIOChannelFDSource *ssource = (QIOChannelFDSource *)source;
  71
  72     return ssource->fd.revents & ssource->condition;
  73 }
  74
  75
  76 static gboolean
  77 qio_channel_fd_source_dispatch(GSource *source,
  78                                GSourceFunc callback,
  79                                gpointer user_data)
  80 {
  81     QIOChannelFunc func = (QIOChannelFunc)callback;
  82     QIOChannelFDSource *ssource = (QIOChannelFDSource *)source;
  83
  84     return (*func)(ssource->ioc,
  85                    ssource->fd.revents & ssource->condition,
  86                    user_data);
  87 }
  88
  89
  90 static void
  91 qio_channel_fd_source_finalize(GSource *source)
  92 {
  93     QIOChannelFDSource *ssource = (QIOChannelFDSource *)source;
  94
  95     object_unref(OBJECT(ssource->ioc));
  96 }
  97
  98
  99 #ifdef CONFIG_WIN32
 100 static gboolean
 101 qio_channel_socket_source_prepare(GSource *source G_GNUC_UNUSED,
 102                                   gint *timeout)
 103 {
 104     *timeout = -1;
 105
 106     return FALSE;
 107 }
 108
 109
 110 /*
 111  * NB, this impl only works when the socket is in non-blocking
 112  * mode on Win32
 113  */
 114 static gboolean
 115 qio_channel_socket_source_check(GSource *source)
 116 {
 117     static struct timeval tv0;
 118     QIOChannelSocketSource *ssource = (QIOChannelSocketSource *)source;
 119     fd_set rfds, wfds, xfds;
 120
 121     if (!ssource->condition) {
 122         return 0;
 123     }
 124
 125     FD_ZERO(&rfds);
 126     FD_ZERO(&wfds);
 127     FD_ZERO(&xfds);
 128     if (ssource->condition & G_IO_IN) {
 129         FD_SET(ssource->socket, &rfds);
 130     }
 131     if (ssource->condition & G_IO_OUT) {
 132         FD_SET(ssource->socket, &wfds);
 133     }
 134     if (ssource->condition & G_IO_PRI) {
 135         FD_SET(ssource->socket, &xfds);
 136     }
 137     ssource->revents = 0;
 138     if (select(0, &rfds, &wfds, &xfds, &tv0) == 0) {
 139         return 0;
 140     }
 141
 142     if (FD_ISSET(ssource->socket, &rfds)) {
 143         ssource->revents |= G_IO_IN;
 144     }
 145     if (FD_ISSET(ssource->socket, &wfds)) {
 146         ssource->revents |= G_IO_OUT;
 147     }
 148     if (FD_ISSET(ssource->socket, &xfds)) {
 149         ssource->revents |= G_IO_PRI;
 150     }
 151
 152     return ssource->revents;
 153 }
 154
 155
 156 static gboolean
 157 qio_channel_socket_source_dispatch(GSource *source,
 158                                    GSourceFunc callback,
 159                                    gpointer user_data)
 160 {
 161     QIOChannelFunc func = (QIOChannelFunc)callback;
 162     QIOChannelSocketSource *ssource = (QIOChannelSocketSource *)source;
 163
 164     return (*func)(ssource->ioc, ssource->revents, user_data);
 165 }
 166
 167
 168 static void
 169 qio_channel_socket_source_finalize(GSource *source)
 170 {
 171     QIOChannelSocketSource *ssource = (QIOChannelSocketSource *)source;
 172
 173     object_unref(OBJECT(ssource->ioc));
 174 }
 175
 176
 177 GSourceFuncs qio_channel_socket_source_funcs = {
 178     qio_channel_socket_source_prepare,
 179     qio_channel_socket_source_check,
 180     qio_channel_socket_source_dispatch,
 181     qio_channel_socket_source_finalize
 182 };
 183 #endif
 184
 185
 186 static gboolean
 187 qio_channel_fd_pair_source_prepare(GSource *source G_GNUC_UNUSED,
 188                                    gint *timeout)
 189 {
 190     *timeout = -1;
 191
 192     return FALSE;
 193 }
 194
 195
 196 static gboolean
 197 qio_channel_fd_pair_source_check(GSource *source)
 198 {
 199     QIOChannelFDPairSource *ssource = (QIOChannelFDPairSource *)source;
 200     GIOCondition poll_condition = ssource->fdread.revents |
 201         ssource->fdwrite.revents;
 202
 203     return poll_condition & ssource->condition;
 204 }
 205
 206
 207 static gboolean
 208 qio_channel_fd_pair_source_dispatch(GSource *source,
 209                                     GSourceFunc callback,
 210                                     gpointer user_data)
 211 {
 212     QIOChannelFunc func = (QIOChannelFunc)callback;
 213     QIOChannelFDPairSource *ssource = (QIOChannelFDPairSource *)source;
 214     GIOCondition poll_condition = ssource->fdread.revents |
 215         ssource->fdwrite.revents;
 216
 217     return (*func)(ssource->ioc,
 218                    poll_condition & ssource->condition,
 219                    user_data);
 220 }
 221
 222
 223 static void
 224 qio_channel_fd_pair_source_finalize(GSource *source)
 225 {
 226     QIOChannelFDPairSource *ssource = (QIOChannelFDPairSource *)source;
 227
 228     object_unref(OBJECT(ssource->ioc));
 229 }
 230
 231
 232 GSourceFuncs qio_channel_fd_source_funcs = {
 233     qio_channel_fd_source_prepare,
 234     qio_channel_fd_source_check,
 235     qio_channel_fd_source_dispatch,
 236     qio_channel_fd_source_finalize
 237 };
 238
 239
 240 GSourceFuncs qio_channel_fd_pair_source_funcs = {
 241     qio_channel_fd_pair_source_prepare,
 242     qio_channel_fd_pair_source_check,
 243     qio_channel_fd_pair_source_dispatch,
 244     qio_channel_fd_pair_source_finalize
 245 };
 246
 247
 248 GSource *qio_channel_create_fd_watch(QIOChannel *ioc,
 249                                      int fd,
 250                                      GIOCondition condition)
 251 {
 252     GSource *source;
 253     QIOChannelFDSource *ssource;
 254
 255     source = g_source_new(&qio_channel_fd_source_funcs,
 256                           sizeof(QIOChannelFDSource));
 257     ssource = (QIOChannelFDSource *)source;
 258
 259     ssource->ioc = ioc;
 260     object_ref(OBJECT(ioc));
 261
 262     ssource->condition = condition;
 263
 264 #ifdef CONFIG_WIN32
 265     ssource->fd.fd = (gint64)_get_osfhandle(fd);
 266 #else
 267     ssource->fd.fd = fd;
 268 #endif
 269     ssource->fd.events = condition;
 270
 271     g_source_add_poll(source, &ssource->fd);
 272
 273     return source;
 274 }
 275
 276 #ifdef CONFIG_WIN32
 277 GSource *qio_channel_create_socket_watch(QIOChannel *ioc,
 278                                          int socket,
 279                                          GIOCondition condition)
 280 {
 281     GSource *source;
 282     QIOChannelSocketSource *ssource;
 283
 284     WSAEventSelect(socket, ioc->event,
 285                    FD_READ | FD_ACCEPT | FD_CLOSE |
 286                    FD_CONNECT | FD_WRITE | FD_OOB);
 287
 288     source = g_source_new(&qio_channel_socket_source_funcs,
 289                           sizeof(QIOChannelSocketSource));
 290     ssource = (QIOChannelSocketSource *)source;
 291
 292     ssource->ioc = ioc;
 293     object_ref(OBJECT(ioc));
 294
 295     ssource->condition = condition;
 296     ssource->socket = socket;
 297     ssource->revents = 0;
 298
 299     ssource->fd.fd = (gintptr)ioc->event;
 300     ssource->fd.events = G_IO_IN;
 301
 302     g_source_add_poll(source, &ssource->fd);
 303
 304     return source;
 305 }
 306 #else
 307 GSource *qio_channel_create_socket_watch(QIOChannel *ioc,
 308                                          int socket,
 309                                          GIOCondition condition)
 310 {
 311     return qio_channel_create_fd_watch(ioc, socket, condition);
 312 }
 313 #endif
 314
 315 GSource *qio_channel_create_fd_pair_watch(QIOChannel *ioc,
 316                                           int fdread,
 317                                           int fdwrite,
 318                                           GIOCondition condition)
 319 {
 320     GSource *source;
 321     QIOChannelFDPairSource *ssource;
 322
 323     source = g_source_new(&qio_channel_fd_pair_source_funcs,
 324                           sizeof(QIOChannelFDPairSource));
 325     ssource = (QIOChannelFDPairSource *)source;
 326
 327     ssource->ioc = ioc;
 328     object_ref(OBJECT(ioc));
 329
 330     ssource->condition = condition;
 331
 332 #ifdef CONFIG_WIN32
 333     ssource->fdread.fd = (gint64)_get_osfhandle(fdread);
 334     ssource->fdwrite.fd = (gint64)_get_osfhandle(fdwrite);
 335 #else
 336     ssource->fdread.fd = fdread;
 337     ssource->fdwrite.fd = fdwrite;
 338 #endif
 339
 340     ssource->fdread.events = condition & G_IO_IN;
 341     ssource->fdwrite.events = condition & G_IO_OUT;
 342
 343     g_source_add_poll(source, &ssource->fdread);
 344     g_source_add_poll(source, &ssource->fdwrite);
 345
 346     return source;
 347 }