X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=gio%2Fgunixinputstream.c;h=24700ece815bb5a84e6755adfbfc90ef5282fc7a;hb=4482977238ae80f64c2fe318d1500f4662c73980;hp=2cd9aa48eb280207c5c178027503a44bd1699e31;hpb=5b946e0504952a472cea1ef95df0967e8a4a7b47;p=platform%2Fupstream%2Fglib.git diff --git a/gio/gunixinputstream.c b/gio/gunixinputstream.c index 2cd9aa4..24700ec 100644 --- a/gio/gunixinputstream.c +++ b/gio/gunixinputstream.c @@ -13,9 +13,7 @@ * 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, write to the - * Free Software Foundation, Inc., 59 Temple Place, Suite 330, - * Boston, MA 02111-1307, USA. + * Public License along with this library; if not, see . * * Author: Alexander Larsson */ @@ -31,11 +29,13 @@ #include #include +#include #include "gioerror.h" #include "gsimpleasyncresult.h" #include "gunixinputstream.h" #include "gcancellable.h" #include "gasynchelper.h" +#include "gfiledescriptorbased.h" #include "glibintl.h" @@ -45,13 +45,15 @@ * @include: gio/gunixinputstream.h * @see_also: #GInputStream * - * #GUnixInputStream implements #GInputStream for reading from a - * UNIX file descriptor, including asynchronous operations. The file - * descriptor must be selectable, so it doesn't work with opened files. + * #GUnixInputStream implements #GInputStream for reading from a UNIX + * file descriptor, including asynchronous operations. (If the file + * descriptor refers to a socket or pipe, this will use poll() to do + * asynchronous I/O. If it refers to a regular file, it will fall back + * to doing asynchronous I/O in another thread.) * - * Note that <gio/gunixinputstream.h> belongs - * to the UNIX-specific GIO interfaces, thus you have to use the - * gio-unix-2.0.pc pkg-config file when using it. + * Note that `` belongs to the UNIX-specific GIO + * interfaces, thus you have to use the `gio-unix-2.0.pc` pkg-config + * file when using it. */ enum { @@ -60,13 +62,23 @@ enum { PROP_CLOSE_FD }; -G_DEFINE_TYPE (GUnixInputStream, g_unix_input_stream, G_TYPE_INPUT_STREAM); - struct _GUnixInputStreamPrivate { int fd; - gboolean close_fd; + guint close_fd : 1; + guint is_pipe_or_socket : 1; }; +static void g_unix_input_stream_pollable_iface_init (GPollableInputStreamInterface *iface); +static void g_unix_input_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface); + +G_DEFINE_TYPE_WITH_CODE (GUnixInputStream, g_unix_input_stream, G_TYPE_INPUT_STREAM, + G_ADD_PRIVATE (GUnixInputStream) + G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_INPUT_STREAM, + g_unix_input_stream_pollable_iface_init) + G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED, + g_unix_input_stream_file_descriptor_based_iface_init) + ) + static void g_unix_input_stream_set_property (GObject *object, guint prop_id, const GValue *value, @@ -83,16 +95,6 @@ static gssize g_unix_input_stream_read (GInputStream *stream, static gboolean g_unix_input_stream_close (GInputStream *stream, GCancellable *cancellable, GError **error); -static void g_unix_input_stream_read_async (GInputStream *stream, - void *buffer, - gsize count, - int io_priority, - GCancellable *cancellable, - GAsyncReadyCallback callback, - gpointer data); -static gssize g_unix_input_stream_read_finish (GInputStream *stream, - GAsyncResult *result, - GError **error); static void g_unix_input_stream_skip_async (GInputStream *stream, gsize count, int io_priority, @@ -111,29 +113,22 @@ static gboolean g_unix_input_stream_close_finish (GInputStream *stream, GAsyncResult *result, GError **error); - -static void -g_unix_input_stream_finalize (GObject *object) -{ - G_OBJECT_CLASS (g_unix_input_stream_parent_class)->finalize (object); -} +static gboolean g_unix_input_stream_pollable_can_poll (GPollableInputStream *stream); +static gboolean g_unix_input_stream_pollable_is_readable (GPollableInputStream *stream); +static GSource *g_unix_input_stream_pollable_create_source (GPollableInputStream *stream, + GCancellable *cancellable); static void g_unix_input_stream_class_init (GUnixInputStreamClass *klass) { GObjectClass *gobject_class = G_OBJECT_CLASS (klass); GInputStreamClass *stream_class = G_INPUT_STREAM_CLASS (klass); - - g_type_class_add_private (klass, sizeof (GUnixInputStreamPrivate)); gobject_class->get_property = g_unix_input_stream_get_property; gobject_class->set_property = g_unix_input_stream_set_property; - gobject_class->finalize = g_unix_input_stream_finalize; stream_class->read_fn = g_unix_input_stream_read; stream_class->close_fn = g_unix_input_stream_close; - stream_class->read_async = g_unix_input_stream_read_async; - stream_class->read_finish = g_unix_input_stream_read_finish; if (0) { /* TODO: Implement instead of using fallbacks */ @@ -175,6 +170,20 @@ g_unix_input_stream_class_init (GUnixInputStreamClass *klass) } static void +g_unix_input_stream_pollable_iface_init (GPollableInputStreamInterface *iface) +{ + iface->can_poll = g_unix_input_stream_pollable_can_poll; + iface->is_readable = g_unix_input_stream_pollable_is_readable; + iface->create_source = g_unix_input_stream_pollable_create_source; +} + +static void +g_unix_input_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface) +{ + iface->get_fd = (int (*) (GFileDescriptorBased *))g_unix_input_stream_get_fd; +} + +static void g_unix_input_stream_set_property (GObject *object, guint prop_id, const GValue *value, @@ -188,6 +197,10 @@ g_unix_input_stream_set_property (GObject *object, { case PROP_FD: unix_stream->priv->fd = g_value_get_int (value); + if (lseek (unix_stream->priv->fd, 0, SEEK_CUR) == -1 && errno == ESPIPE) + unix_stream->priv->is_pipe_or_socket = TRUE; + else + unix_stream->priv->is_pipe_or_socket = FALSE; break; case PROP_CLOSE_FD: unix_stream->priv->close_fd = g_value_get_boolean (value); @@ -224,10 +237,7 @@ g_unix_input_stream_get_property (GObject *object, static void g_unix_input_stream_init (GUnixInputStream *unix_stream) { - unix_stream->priv = G_TYPE_INSTANCE_GET_PRIVATE (unix_stream, - G_TYPE_UNIX_INPUT_STREAM, - GUnixInputStreamPrivate); - + unix_stream->priv = g_unix_input_stream_get_instance_private (unix_stream); unix_stream->priv->fd = -1; unix_stream->priv->close_fd = TRUE; } @@ -291,7 +301,7 @@ g_unix_input_stream_set_close_fd (GUnixInputStream *stream, * Returns whether the file descriptor of @stream will be * closed when the stream is closed. * - * Return value: %TRUE if the file descriptor is closed when done + * Returns: %TRUE if the file descriptor is closed when done * * Since: 2.20 */ @@ -309,7 +319,7 @@ g_unix_input_stream_get_close_fd (GUnixInputStream *stream) * * Return the UNIX file descriptor that the stream reads from. * - * Return value: The file descriptor of @stream + * Returns: The file descriptor of @stream * * Since: 2.20 */ @@ -329,20 +339,27 @@ g_unix_input_stream_read (GInputStream *stream, GError **error) { GUnixInputStream *unix_stream; - gssize res; + gssize res = -1; GPollFD poll_fds[2]; + int nfds; int poll_ret; unix_stream = G_UNIX_INPUT_STREAM (stream); - if (g_cancellable_make_pollfd (cancellable, &poll_fds[1])) + poll_fds[0].fd = unix_stream->priv->fd; + poll_fds[0].events = G_IO_IN; + if (unix_stream->priv->is_pipe_or_socket && + g_cancellable_make_pollfd (cancellable, &poll_fds[1])) + nfds = 2; + else + nfds = 1; + + while (1) { - poll_fds[0].fd = unix_stream->priv->fd; - poll_fds[0].events = G_IO_IN; + poll_fds[0].revents = poll_fds[1].revents = 0; do - poll_ret = g_poll (poll_fds, 2, -1); + poll_ret = g_poll (poll_fds, nfds, -1); while (poll_ret == -1 && errno == EINTR); - g_cancellable_release_fd (cancellable); if (poll_ret == -1) { @@ -350,33 +367,36 @@ g_unix_input_stream_read (GInputStream *stream, g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errsv), - _("Error reading from unix: %s"), + _("Error reading from file descriptor: %s"), g_strerror (errsv)); - return -1; + break; } - } - while (1) - { if (g_cancellable_set_error_if_cancelled (cancellable, error)) - return -1; + break; + + if (!poll_fds[0].revents) + continue; + res = read (unix_stream->priv->fd, buffer, count); if (res == -1) { int errsv = errno; - if (errsv == EINTR) + if (errsv == EINTR || errsv == EAGAIN) continue; - + g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errsv), - _("Error reading from unix: %s"), + _("Error reading from file descriptor: %s"), g_strerror (errsv)); } - + break; } + if (nfds == 2) + g_cancellable_release_fd (cancellable); return res; } @@ -393,135 +413,21 @@ g_unix_input_stream_close (GInputStream *stream, if (!unix_stream->priv->close_fd) return TRUE; - while (1) + /* This might block during the close. Doesn't seem to be a way to avoid it though. */ + res = close (unix_stream->priv->fd); + if (res == -1) { - /* This might block during the close. Doesn't seem to be a way to avoid it though. */ - res = close (unix_stream->priv->fd); - if (res == -1) - { - int errsv = errno; + int errsv = errno; - g_set_error (error, G_IO_ERROR, - g_io_error_from_errno (errsv), - _("Error closing unix: %s"), - g_strerror (errsv)); - } - break; + g_set_error (error, G_IO_ERROR, + g_io_error_from_errno (errsv), + _("Error closing file descriptor: %s"), + g_strerror (errsv)); } return res != -1; } -typedef struct { - gsize count; - void *buffer; - GAsyncReadyCallback callback; - gpointer user_data; - GCancellable *cancellable; - GUnixInputStream *stream; -} ReadAsyncData; - -static gboolean -read_async_cb (ReadAsyncData *data, - GIOCondition condition, - int fd) -{ - GSimpleAsyncResult *simple; - GError *error = NULL; - gssize count_read; - - /* We know that we can read from fd once without blocking */ - while (1) - { - if (g_cancellable_set_error_if_cancelled (data->cancellable, &error)) - { - count_read = -1; - break; - } - count_read = read (data->stream->priv->fd, data->buffer, data->count); - if (count_read == -1) - { - int errsv = errno; - - if (errsv == EINTR) - continue; - - g_set_error (&error, G_IO_ERROR, - g_io_error_from_errno (errsv), - _("Error reading from unix: %s"), - g_strerror (errsv)); - } - break; - } - - simple = g_simple_async_result_new (G_OBJECT (data->stream), - data->callback, - data->user_data, - g_unix_input_stream_read_async); - - g_simple_async_result_set_op_res_gssize (simple, count_read); - - if (count_read == -1) - { - g_simple_async_result_set_from_error (simple, error); - g_error_free (error); - } - - /* Complete immediately, not in idle, since we're already in a mainloop callout */ - g_simple_async_result_complete (simple); - g_object_unref (simple); - - return FALSE; -} - -static void -g_unix_input_stream_read_async (GInputStream *stream, - void *buffer, - gsize count, - int io_priority, - GCancellable *cancellable, - GAsyncReadyCallback callback, - gpointer user_data) -{ - GSource *source; - GUnixInputStream *unix_stream; - ReadAsyncData *data; - - unix_stream = G_UNIX_INPUT_STREAM (stream); - - data = g_new0 (ReadAsyncData, 1); - data->count = count; - data->buffer = buffer; - data->callback = callback; - data->user_data = user_data; - data->cancellable = cancellable; - data->stream = unix_stream; - - source = _g_fd_source_new (unix_stream->priv->fd, - G_IO_IN, - cancellable); - - g_source_set_callback (source, (GSourceFunc)read_async_cb, data, g_free); - g_source_attach (source, g_main_context_get_thread_default ()); - - g_source_unref (source); -} - -static gssize -g_unix_input_stream_read_finish (GInputStream *stream, - GAsyncResult *result, - GError **error) -{ - GSimpleAsyncResult *simple; - gssize nread; - - simple = G_SIMPLE_ASYNC_RESULT (result); - g_warn_if_fail (g_simple_async_result_get_source_tag (simple) == g_unix_input_stream_read_async); - - nread = g_simple_async_result_get_op_res_gssize (simple); - return nread; -} - static void g_unix_input_stream_skip_async (GInputStream *stream, gsize count, @@ -544,101 +450,81 @@ g_unix_input_stream_skip_finish (GInputStream *stream, /* TODO: Not implemented */ } - -typedef struct { - GInputStream *stream; - GAsyncReadyCallback callback; - gpointer user_data; -} CloseAsyncData; - static void -close_async_data_free (gpointer _data) +g_unix_input_stream_close_async (GInputStream *stream, + int io_priority, + GCancellable *cancellable, + GAsyncReadyCallback callback, + gpointer user_data) { - CloseAsyncData *data = _data; + GTask *task; + GError *error = NULL; + + task = g_task_new (stream, cancellable, callback, user_data); + g_task_set_priority (task, io_priority); - g_free (data); + if (g_unix_input_stream_close (stream, cancellable, &error)) + g_task_return_boolean (task, TRUE); + else + g_task_return_error (task, error); + g_object_unref (task); } static gboolean -close_async_cb (CloseAsyncData *data) +g_unix_input_stream_close_finish (GInputStream *stream, + GAsyncResult *result, + GError **error) { - GUnixInputStream *unix_stream; - GSimpleAsyncResult *simple; - GError *error = NULL; - gboolean result; - int res; + g_return_val_if_fail (g_task_is_valid (result, stream), FALSE); - unix_stream = G_UNIX_INPUT_STREAM (data->stream); + return g_task_propagate_boolean (G_TASK (result), error); +} - if (!unix_stream->priv->close_fd) - { - result = TRUE; - goto out; - } - - while (1) - { - res = close (unix_stream->priv->fd); - if (res == -1) - { - int errsv = errno; +static gboolean +g_unix_input_stream_pollable_can_poll (GPollableInputStream *stream) +{ + return G_UNIX_INPUT_STREAM (stream)->priv->is_pipe_or_socket; +} - g_set_error (&error, G_IO_ERROR, - g_io_error_from_errno (errsv), - _("Error closing unix: %s"), - g_strerror (errsv)); - } - break; - } - - result = res != -1; +static gboolean +g_unix_input_stream_pollable_is_readable (GPollableInputStream *stream) +{ + GUnixInputStream *unix_stream = G_UNIX_INPUT_STREAM (stream); + GPollFD poll_fd; + gint result; - out: - simple = g_simple_async_result_new (G_OBJECT (data->stream), - data->callback, - data->user_data, - g_unix_input_stream_close_async); + poll_fd.fd = unix_stream->priv->fd; + poll_fd.events = G_IO_IN; + poll_fd.revents = 0; - if (!result) - { - g_simple_async_result_set_from_error (simple, error); - g_error_free (error); - } + do + result = g_poll (&poll_fd, 1, 0); + while (result == -1 && errno == EINTR); - /* Complete immediately, not in idle, since we're already in a mainloop callout */ - g_simple_async_result_complete (simple); - g_object_unref (simple); - - return FALSE; + return poll_fd.revents != 0; } -static void -g_unix_input_stream_close_async (GInputStream *stream, - int io_priority, - GCancellable *cancellable, - GAsyncReadyCallback callback, - gpointer user_data) +static GSource * +g_unix_input_stream_pollable_create_source (GPollableInputStream *stream, + GCancellable *cancellable) { - GSource *idle; - CloseAsyncData *data; + GUnixInputStream *unix_stream = G_UNIX_INPUT_STREAM (stream); + GSource *inner_source, *cancellable_source, *pollable_source; - data = g_new0 (CloseAsyncData, 1); + pollable_source = g_pollable_source_new (G_OBJECT (stream)); - data->stream = stream; - data->callback = callback; - data->user_data = user_data; - - idle = g_idle_source_new (); - g_source_set_callback (idle, (GSourceFunc)close_async_cb, data, close_async_data_free); - g_source_attach (idle, g_main_context_get_thread_default ()); - g_source_unref (idle); -} + inner_source = g_unix_fd_source_new (unix_stream->priv->fd, G_IO_IN); + g_source_set_dummy_callback (inner_source); + g_source_add_child_source (pollable_source, inner_source); + g_source_unref (inner_source); -static gboolean -g_unix_input_stream_close_finish (GInputStream *stream, - GAsyncResult *result, - GError **error) -{ - /* Failures handled in generic close_finish code */ - return TRUE; + if (cancellable) + { + cancellable_source = g_cancellable_source_new (cancellable); + g_source_set_dummy_callback (cancellable_source); + g_source_add_child_source (pollable_source, cancellable_source); + g_source_unref (cancellable_source); + } + + return pollable_source; }