gio/gpollableinputstream.c

   1 /* GIO - GLib Input, Output and Streaming Library
   2  *
   3  * Copyright (C) 2010 Red Hat, Inc.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General
  16  * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include "config.h"
  20
  21 #include <errno.h>
  22
  23 #include "gpollableinputstream.h"
  24 #include "gasynchelper.h"
  25 #include "glibintl.h"
  26
  27 /**
  28  * SECTION:gpollableinputstream
  29  * @short_description: Interface for pollable input streams
  30  * @include: gio/gio.h
  31  * @see_also: #GInputStream, #GPollableOutputStream, #GFileDescriptorBased
  32  *
  33  * #GPollableInputStream is implemented by #GInputStreams that
  34  * can be polled for readiness to read. This can be used when
  35  * interfacing with a non-GIO API that expects
  36  * UNIX-file-descriptor-style asynchronous I/O rather than GIO-style.
  37  *
  38  * Since: 2.28
  39  */
  40
  41 G_DEFINE_INTERFACE (GPollableInputStream, g_pollable_input_stream, G_TYPE_INPUT_STREAM)
  42
  43 static gboolean g_pollable_input_stream_default_can_poll         (GPollableInputStream *stream);
  44 static gssize   g_pollable_input_stream_default_read_nonblocking (GPollableInputStream  *stream,
  45                                                                   void                  *buffer,
  46                                                                   gsize                  count,
  47                                                                   GError               **error);
  48
  49 static void
  50 g_pollable_input_stream_default_init (GPollableInputStreamInterface *iface)
  51 {
  52   iface->can_poll         = g_pollable_input_stream_default_can_poll;
  53   iface->read_nonblocking = g_pollable_input_stream_default_read_nonblocking;
  54 }
  55
  56 static gboolean
  57 g_pollable_input_stream_default_can_poll (GPollableInputStream *stream)
  58 {
  59   return TRUE;
  60 }
  61
  62 /**
  63  * g_pollable_input_stream_can_poll:
  64  * @stream: a #GPollableInputStream.
  65  *
  66  * Checks if @stream is actually pollable. Some classes may implement
  67  * #GPollableInputStream but have only certain instances of that class
  68  * be pollable. If this method returns %FALSE, then the behavior of
  69  * other #GPollableInputStream methods is undefined.
  70  *
  71  * For any given stream, the value returned by this method is constant;
  72  * a stream cannot switch from pollable to non-pollable or vice versa.
  73  *
  74  * Returns: %TRUE if @stream is pollable, %FALSE if not.
  75  *
  76  * Since: 2.28
  77  */
  78 gboolean
  79 g_pollable_input_stream_can_poll (GPollableInputStream *stream)
  80 {
  81   g_return_val_if_fail (G_IS_POLLABLE_INPUT_STREAM (stream), FALSE);
  82
  83   return G_POLLABLE_INPUT_STREAM_GET_INTERFACE (stream)->can_poll (stream);
  84 }
  85
  86 /**
  87  * g_pollable_input_stream_is_readable:
  88  * @stream: a #GPollableInputStream.
  89  *
  90  * Checks if @stream can be read.
  91  *
  92  * Note that some stream types may not be able to implement this 100%
  93  * reliably, and it is possible that a call to g_input_stream_read()
  94  * after this returns %TRUE would still block. To guarantee
  95  * non-blocking behavior, you should always use
  96  * g_pollable_input_stream_read_nonblocking(), which will return a
  97  * %G_IO_ERROR_WOULD_BLOCK error rather than blocking.
  98  *
  99  * Returns: %TRUE if @stream is readable, %FALSE if not. If an error
 100  *   has occurred on @stream, this will result in
 101  *   g_pollable_input_stream_is_readable() returning %TRUE, and the
 102  *   next attempt to read will return the error.
 103  *
 104  * Since: 2.28
 105  */
 106 gboolean
 107 g_pollable_input_stream_is_readable (GPollableInputStream *stream)
 108 {
 109   g_return_val_if_fail (G_IS_POLLABLE_INPUT_STREAM (stream), FALSE);
 110
 111   return G_POLLABLE_INPUT_STREAM_GET_INTERFACE (stream)->is_readable (stream);
 112 }
 113
 114 /**
 115  * g_pollable_input_stream_create_source:
 116  * @stream: a #GPollableInputStream.
 117  * @cancellable: (allow-none): a #GCancellable, or %NULL
 118  *
 119  * Creates a #GSource that triggers when @stream can be read, or
 120  * @cancellable is triggered or an error occurs. The callback on the
 121  * source is of the #GPollableSourceFunc type.
 122  *
 123  * As with g_pollable_input_stream_is_readable(), it is possible that
 124  * the stream may not actually be readable even after the source
 125  * triggers, so you should use g_pollable_input_stream_read_nonblocking()
 126  * rather than g_input_stream_read() from the callback.
 127  *
 128  * Returns: (transfer full): a new #GSource
 129  *
 130  * Since: 2.28
 131  */
 132 GSource *
 133 g_pollable_input_stream_create_source (GPollableInputStream *stream,
 134                                        GCancellable         *cancellable)
 135 {
 136   g_return_val_if_fail (G_IS_POLLABLE_INPUT_STREAM (stream), NULL);
 137
 138   return G_POLLABLE_INPUT_STREAM_GET_INTERFACE (stream)->
 139           create_source (stream, cancellable);
 140 }
 141
 142 static gssize
 143 g_pollable_input_stream_default_read_nonblocking (GPollableInputStream  *stream,
 144                                                   void                  *buffer,
 145                                                   gsize                  count,
 146                                                   GError               **error)
 147 {
 148   if (!g_pollable_input_stream_is_readable (stream))
 149     {
 150       g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK,
 151                            g_strerror (EAGAIN));
 152       return -1;
 153     }
 154
 155   return G_INPUT_STREAM_GET_CLASS (stream)->
 156     read_fn (G_INPUT_STREAM (stream), buffer, count, NULL, error);
 157 }
 158
 159 /**
 160  * g_pollable_input_stream_read_nonblocking:
 161  * @stream: a #GPollableInputStream
 162  * @buffer: (array length=count) (element-type guint8): a buffer to
 163  *     read data into (which should be at least @count bytes long).
 164  * @count: the number of bytes you want to read
 165  * @cancellable: (allow-none): a #GCancellable, or %NULL
 166  * @error: #GError for error reporting, or %NULL to ignore.
 167  *
 168  * Attempts to read up to @count bytes from @stream into @buffer, as
 169  * with g_input_stream_read(). If @stream is not currently readable,
 170  * this will immediately return %G_IO_ERROR_WOULD_BLOCK, and you can
 171  * use g_pollable_input_stream_create_source() to create a #GSource
 172  * that will be triggered when @stream is readable.
 173  *
 174  * Note that since this method never blocks, you cannot actually
 175  * use @cancellable to cancel it. However, it will return an error
 176  * if @cancellable has already been cancelled when you call, which
 177  * may happen if you call this method after a source triggers due
 178  * to having been cancelled.
 179  *
 180  * Virtual: read_nonblocking
 181  * Returns: the number of bytes read, or -1 on error (including
 182  *   %G_IO_ERROR_WOULD_BLOCK).
 183  */
 184 gssize
 185 g_pollable_input_stream_read_nonblocking (GPollableInputStream  *stream,
 186                                           void                  *buffer,
 187                                           gsize                  count,
 188                                           GCancellable          *cancellable,
 189                                           GError               **error)
 190 {
 191   gssize res;
 192
 193   g_return_val_if_fail (G_IS_POLLABLE_INPUT_STREAM (stream), -1);
 194   g_return_val_if_fail (buffer != NULL, 0);
 195
 196   if (g_cancellable_set_error_if_cancelled (cancellable, error))
 197     return -1;
 198
 199   if (count == 0)
 200     return 0;
 201
 202   if (((gssize) count) < 0)
 203     {
 204       g_set_error (error, G_IO_ERROR, G_IO_ERROR_INVALID_ARGUMENT,
 205                    _("Too large count value passed to %s"), G_STRFUNC);
 206       return -1;
 207     }
 208
 209   if (cancellable)
 210     g_cancellable_push_current (cancellable);
 211
 212   res = G_POLLABLE_INPUT_STREAM_GET_INTERFACE (stream)->
 213     read_nonblocking (stream, buffer, count, error);
 214
 215   if (cancellable)
 216     g_cancellable_pop_current (cancellable);
 217
 218   return res;
 219 }