mirror of
				https://gitlab.gnome.org/GNOME/glib.git
				synced 2025-10-31 08:22:16 +01:00 
			
		
		
		
	Specifically, GIOStream and the TLS connection streams. Includes wording adapted from suggestions by Dan Winship <danw@gnome.org>. https://bugzilla.gnome.org/show_bug.cgi?id=735754
		
			
				
	
	
		
			918 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			918 lines
		
	
	
		
			26 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* GIO - GLib Input, Output and Streaming Library
 | ||
|  *
 | ||
|  * Copyright © 2008 codethink
 | ||
|  * Copyright © 2009 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/>.
 | ||
|  *
 | ||
|  * Authors: Ryan Lortie <desrt@desrt.ca>
 | ||
|  *          Alexander Larsson <alexl@redhat.com>
 | ||
|  */
 | ||
| 
 | ||
| #include "config.h"
 | ||
| #include <glib.h>
 | ||
| #include "glibintl.h"
 | ||
| 
 | ||
| #include "giostream.h"
 | ||
| #include "gasyncresult.h"
 | ||
| #include "gioprivate.h"
 | ||
| #include "gtask.h"
 | ||
| 
 | ||
| /**
 | ||
|  * SECTION:giostream
 | ||
|  * @short_description: Base class for implementing read/write streams
 | ||
|  * @include: gio/gio.h
 | ||
|  * @see_also: #GInputStream, #GOutputStream
 | ||
|  *
 | ||
|  * GIOStream represents an object that has both read and write streams.
 | ||
|  * Generally the two streams act as separate input and output streams,
 | ||
|  * but they share some common resources and state. For instance, for
 | ||
|  * seekable streams, both streams may use the same position.
 | ||
|  *
 | ||
|  * Examples of #GIOStream objects are #GSocketConnection, which represents
 | ||
|  * a two-way network connection; and #GFileIOStream, which represents a
 | ||
|  * file handle opened in read-write mode.
 | ||
|  *
 | ||
|  * To do the actual reading and writing you need to get the substreams
 | ||
|  * with g_io_stream_get_input_stream() and g_io_stream_get_output_stream().
 | ||
|  *
 | ||
|  * The #GIOStream object owns the input and the output streams, not the other
 | ||
|  * way around, so keeping the substreams alive will not keep the #GIOStream
 | ||
|  * object alive. If the #GIOStream object is freed it will be closed, thus
 | ||
|  * closing the substreams, so even if the substreams stay alive they will
 | ||
|  * always return %G_IO_ERROR_CLOSED for all operations.
 | ||
|  *
 | ||
|  * To close a stream use g_io_stream_close() which will close the common
 | ||
|  * stream object and also the individual substreams. You can also close
 | ||
|  * the substreams themselves. In most cases this only marks the
 | ||
|  * substream as closed, so further I/O on it fails but common state in the
 | ||
|  * #GIOStream may still be open. However, some streams may support
 | ||
|  * "half-closed" states where one direction of the stream is actually shut down.
 | ||
|  *
 | ||
|  * Operations on #GIOStreams cannot be started while another operation on the
 | ||
|  * #GIOStream or its substreams is in progress. Specifically, an application can
 | ||
|  * read from the #GInputStream and write to the #GOutputStream simultaneously
 | ||
|  * (either in separate threads, or as asynchronous operations in the same
 | ||
|  * thread), but an application cannot start any #GIOStream operation while there
 | ||
|  * is a #GIOStream, #GInputStream or #GOutputStream operation in progress, and
 | ||
|  * an application can’t start any #GInputStream or #GOutputStream operation
 | ||
|  * while there is a #GIOStream operation in progress.
 | ||
|  *
 | ||
|  * This is a product of individual stream operations being associated with a
 | ||
|  * given #GMainContext (the thread-default context at the time the operation was
 | ||
|  * started), rather than entire streams being associated with a single
 | ||
|  * #GMainContext.
 | ||
|  *
 | ||
|  * GIO may run operations on #GIOStreams from other (worker) threads, and this
 | ||
|  * may be exposed to application code in the behaviour of wrapper streams, such
 | ||
|  * as #GBufferedInputStream or #GTlsConnection. With such wrapper APIs,
 | ||
|  * application code may only run operations on the base (wrapped) stream when
 | ||
|  * the wrapper stream is idle. Note that the semantics of such operations may
 | ||
|  * not be well-defined due to the state the wrapper stream leaves the base
 | ||
|  * stream in (though they are guaranteed not to crash).
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| 
 | ||
| enum
 | ||
| {
 | ||
|   PROP_0,
 | ||
|   PROP_INPUT_STREAM,
 | ||
|   PROP_OUTPUT_STREAM,
 | ||
|   PROP_CLOSED
 | ||
| };
 | ||
| 
 | ||
| struct _GIOStreamPrivate {
 | ||
|   guint closed : 1;
 | ||
|   guint pending : 1;
 | ||
| };
 | ||
| 
 | ||
| static gboolean g_io_stream_real_close        (GIOStream            *stream,
 | ||
| 					       GCancellable         *cancellable,
 | ||
| 					       GError              **error);
 | ||
| static void     g_io_stream_real_close_async  (GIOStream            *stream,
 | ||
| 					       int                   io_priority,
 | ||
| 					       GCancellable         *cancellable,
 | ||
| 					       GAsyncReadyCallback   callback,
 | ||
| 					       gpointer              user_data);
 | ||
| static gboolean g_io_stream_real_close_finish (GIOStream            *stream,
 | ||
| 					       GAsyncResult         *result,
 | ||
| 					       GError              **error);
 | ||
| 
 | ||
| G_DEFINE_ABSTRACT_TYPE_WITH_PRIVATE (GIOStream, g_io_stream, G_TYPE_OBJECT)
 | ||
| 
 | ||
| static void
 | ||
| g_io_stream_dispose (GObject *object)
 | ||
| {
 | ||
|   GIOStream *stream;
 | ||
| 
 | ||
|   stream = G_IO_STREAM (object);
 | ||
| 
 | ||
|   if (!stream->priv->closed)
 | ||
|     g_io_stream_close (stream, NULL, NULL);
 | ||
| 
 | ||
|   G_OBJECT_CLASS (g_io_stream_parent_class)->dispose (object);
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| g_io_stream_init (GIOStream *stream)
 | ||
| {
 | ||
|   stream->priv = g_io_stream_get_instance_private (stream);
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| g_io_stream_get_property (GObject    *object,
 | ||
| 			  guint       prop_id,
 | ||
| 			  GValue     *value,
 | ||
| 			  GParamSpec *pspec)
 | ||
| {
 | ||
|   GIOStream *stream = G_IO_STREAM (object);
 | ||
| 
 | ||
|   switch (prop_id)
 | ||
|     {
 | ||
|       case PROP_CLOSED:
 | ||
|         g_value_set_boolean (value, stream->priv->closed);
 | ||
|         break;
 | ||
| 
 | ||
|       case PROP_INPUT_STREAM:
 | ||
|         g_value_set_object (value, g_io_stream_get_input_stream (stream));
 | ||
|         break;
 | ||
| 
 | ||
|       case PROP_OUTPUT_STREAM:
 | ||
|         g_value_set_object (value, g_io_stream_get_output_stream (stream));
 | ||
|         break;
 | ||
| 
 | ||
|       default:
 | ||
|         G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 | ||
|     }
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| g_io_stream_class_init (GIOStreamClass *klass)
 | ||
| {
 | ||
|   GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
 | ||
| 
 | ||
|   gobject_class->dispose = g_io_stream_dispose;
 | ||
|   gobject_class->get_property = g_io_stream_get_property;
 | ||
| 
 | ||
|   klass->close_fn = g_io_stream_real_close;
 | ||
|   klass->close_async = g_io_stream_real_close_async;
 | ||
|   klass->close_finish = g_io_stream_real_close_finish;
 | ||
| 
 | ||
|   g_object_class_install_property (gobject_class, PROP_CLOSED,
 | ||
|                                    g_param_spec_boolean ("closed",
 | ||
|                                                          P_("Closed"),
 | ||
|                                                          P_("Is the stream closed"),
 | ||
|                                                          FALSE,
 | ||
|                                                          G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 | ||
| 
 | ||
|   g_object_class_install_property (gobject_class, PROP_INPUT_STREAM,
 | ||
| 				   g_param_spec_object ("input-stream",
 | ||
| 							P_("Input stream"),
 | ||
| 							P_("The GInputStream to read from"),
 | ||
| 							G_TYPE_INPUT_STREAM,
 | ||
| 							G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 | ||
|   g_object_class_install_property (gobject_class, PROP_OUTPUT_STREAM,
 | ||
| 				   g_param_spec_object ("output-stream",
 | ||
| 							P_("Output stream"),
 | ||
| 							P_("The GOutputStream to write to"),
 | ||
| 							G_TYPE_OUTPUT_STREAM,
 | ||
| 							G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_is_closed:
 | ||
|  * @stream: a #GIOStream
 | ||
|  *
 | ||
|  * Checks if a stream is closed.
 | ||
|  *
 | ||
|  * Returns: %TRUE if the stream is closed.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| gboolean
 | ||
| g_io_stream_is_closed (GIOStream *stream)
 | ||
| {
 | ||
|   g_return_val_if_fail (G_IS_IO_STREAM (stream), TRUE);
 | ||
| 
 | ||
|   return stream->priv->closed;
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_get_input_stream:
 | ||
|  * @stream: a #GIOStream
 | ||
|  *
 | ||
|  * Gets the input stream for this object. This is used
 | ||
|  * for reading.
 | ||
|  *
 | ||
|  * Returns: (transfer none): a #GInputStream, owned by the #GIOStream.
 | ||
|  * Do not free.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| GInputStream *
 | ||
| g_io_stream_get_input_stream (GIOStream *stream)
 | ||
| {
 | ||
|   GIOStreamClass *klass;
 | ||
| 
 | ||
|   klass = G_IO_STREAM_GET_CLASS (stream);
 | ||
| 
 | ||
|   g_assert (klass->get_input_stream != NULL);
 | ||
| 
 | ||
|   return klass->get_input_stream (stream);
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_get_output_stream:
 | ||
|  * @stream: a #GIOStream
 | ||
|  *
 | ||
|  * Gets the output stream for this object. This is used for
 | ||
|  * writing.
 | ||
|  *
 | ||
|  * Returns: (transfer none): a #GOutputStream, owned by the #GIOStream.
 | ||
|  * Do not free.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| GOutputStream *
 | ||
| g_io_stream_get_output_stream (GIOStream *stream)
 | ||
| {
 | ||
|   GIOStreamClass *klass;
 | ||
| 
 | ||
|   klass = G_IO_STREAM_GET_CLASS (stream);
 | ||
| 
 | ||
|   g_assert (klass->get_output_stream != NULL);
 | ||
|   return klass->get_output_stream (stream);
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_has_pending:
 | ||
|  * @stream: a #GIOStream
 | ||
|  *
 | ||
|  * Checks if a stream has pending actions.
 | ||
|  *
 | ||
|  * Returns: %TRUE if @stream has pending actions.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  **/
 | ||
| gboolean
 | ||
| g_io_stream_has_pending (GIOStream *stream)
 | ||
| {
 | ||
|   g_return_val_if_fail (G_IS_IO_STREAM (stream), FALSE);
 | ||
| 
 | ||
|   return stream->priv->pending;
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_set_pending:
 | ||
|  * @stream: a #GIOStream
 | ||
|  * @error: a #GError location to store the error occurring, or %NULL to
 | ||
|  *     ignore
 | ||
|  *
 | ||
|  * Sets @stream to have actions pending. If the pending flag is
 | ||
|  * already set or @stream is closed, it will return %FALSE and set
 | ||
|  * @error.
 | ||
|  *
 | ||
|  * Returns: %TRUE if pending was previously unset and is now set.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| gboolean
 | ||
| g_io_stream_set_pending (GIOStream  *stream,
 | ||
| 			 GError    **error)
 | ||
| {
 | ||
|   g_return_val_if_fail (G_IS_IO_STREAM (stream), FALSE);
 | ||
| 
 | ||
|   if (stream->priv->closed)
 | ||
|     {
 | ||
|       g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_CLOSED,
 | ||
|                            _("Stream is already closed"));
 | ||
|       return FALSE;
 | ||
|     }
 | ||
| 
 | ||
|   if (stream->priv->pending)
 | ||
|     {
 | ||
|       g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PENDING,
 | ||
|                            /* Translators: This is an error you get if there is
 | ||
|                             * already an operation running against this stream when
 | ||
|                             * you try to start one */
 | ||
|                            _("Stream has outstanding operation"));
 | ||
|       return FALSE;
 | ||
|     }
 | ||
| 
 | ||
|   stream->priv->pending = TRUE;
 | ||
|   return TRUE;
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_clear_pending:
 | ||
|  * @stream: a #GIOStream
 | ||
|  *
 | ||
|  * Clears the pending flag on @stream.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| void
 | ||
| g_io_stream_clear_pending (GIOStream *stream)
 | ||
| {
 | ||
|   g_return_if_fail (G_IS_IO_STREAM (stream));
 | ||
| 
 | ||
|   stream->priv->pending = FALSE;
 | ||
| }
 | ||
| 
 | ||
| static gboolean
 | ||
| g_io_stream_real_close (GIOStream     *stream,
 | ||
| 			GCancellable  *cancellable,
 | ||
| 			GError       **error)
 | ||
| {
 | ||
|   gboolean res;
 | ||
| 
 | ||
|   res = g_output_stream_close (g_io_stream_get_output_stream (stream),
 | ||
| 			       cancellable, error);
 | ||
| 
 | ||
|   /* If this errored out, unset error so that we don't report
 | ||
|      further errors, but still do the following ops */
 | ||
|   if (error != NULL && *error != NULL)
 | ||
|     error = NULL;
 | ||
| 
 | ||
|   res &= g_input_stream_close (g_io_stream_get_input_stream (stream),
 | ||
| 			       cancellable, error);
 | ||
| 
 | ||
|   return res;
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_close:
 | ||
|  * @stream: a #GIOStream
 | ||
|  * @cancellable: (allow-none): optional #GCancellable object, %NULL to ignore
 | ||
|  * @error: location to store the error occurring, or %NULL to ignore
 | ||
|  *
 | ||
|  * Closes the stream, releasing resources related to it. This will also
 | ||
|  * close the individual input and output streams, if they are not already
 | ||
|  * closed.
 | ||
|  *
 | ||
|  * Once the stream is closed, all other operations will return
 | ||
|  * %G_IO_ERROR_CLOSED. Closing a stream multiple times will not
 | ||
|  * return an error.
 | ||
|  *
 | ||
|  * Closing a stream will automatically flush any outstanding buffers
 | ||
|  * in the stream.
 | ||
|  *
 | ||
|  * Streams will be automatically closed when the last reference
 | ||
|  * is dropped, but you might want to call this function to make sure
 | ||
|  * resources are released as early as possible.
 | ||
|  *
 | ||
|  * Some streams might keep the backing store of the stream (e.g. a file
 | ||
|  * descriptor) open after the stream is closed. See the documentation for
 | ||
|  * the individual stream for details.
 | ||
|  *
 | ||
|  * On failure the first error that happened will be reported, but the
 | ||
|  * close operation will finish as much as possible. A stream that failed
 | ||
|  * to close will still return %G_IO_ERROR_CLOSED for all operations.
 | ||
|  * Still, it is important to check and report the error to the user,
 | ||
|  * otherwise there might be a loss of data as all data might not be written.
 | ||
|  *
 | ||
|  * If @cancellable is not NULL, then the operation can be cancelled by
 | ||
|  * triggering the cancellable object from another thread. If the operation
 | ||
|  * was cancelled, the error %G_IO_ERROR_CANCELLED will be returned.
 | ||
|  * Cancelling a close will still leave the stream closed, but some streams
 | ||
|  * can use a faster close that doesn't block to e.g. check errors.
 | ||
|  *
 | ||
|  * The default implementation of this method just calls close on the
 | ||
|  * individual input/output streams.
 | ||
|  *
 | ||
|  * Returns: %TRUE on success, %FALSE on failure
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| gboolean
 | ||
| g_io_stream_close (GIOStream     *stream,
 | ||
| 		   GCancellable  *cancellable,
 | ||
| 		   GError       **error)
 | ||
| {
 | ||
|   GIOStreamClass *class;
 | ||
|   gboolean res;
 | ||
| 
 | ||
|   g_return_val_if_fail (G_IS_IO_STREAM (stream), FALSE);
 | ||
| 
 | ||
|   class = G_IO_STREAM_GET_CLASS (stream);
 | ||
| 
 | ||
|   if (stream->priv->closed)
 | ||
|     return TRUE;
 | ||
| 
 | ||
|   if (!g_io_stream_set_pending (stream, error))
 | ||
|     return FALSE;
 | ||
| 
 | ||
|   if (cancellable)
 | ||
|     g_cancellable_push_current (cancellable);
 | ||
| 
 | ||
|   res = TRUE;
 | ||
|   if (class->close_fn)
 | ||
|     res = class->close_fn (stream, cancellable, error);
 | ||
| 
 | ||
|   if (cancellable)
 | ||
|     g_cancellable_pop_current (cancellable);
 | ||
| 
 | ||
|   stream->priv->closed = TRUE;
 | ||
|   g_io_stream_clear_pending (stream);
 | ||
| 
 | ||
|   return res;
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| async_ready_close_callback_wrapper (GObject      *source_object,
 | ||
|                                     GAsyncResult *res,
 | ||
|                                     gpointer      user_data)
 | ||
| {
 | ||
|   GIOStream *stream = G_IO_STREAM (source_object);
 | ||
|   GIOStreamClass *klass = G_IO_STREAM_GET_CLASS (stream);
 | ||
|   GTask *task = user_data;
 | ||
|   GError *error = NULL;
 | ||
|   gboolean success;
 | ||
| 
 | ||
|   stream->priv->closed = TRUE;
 | ||
|   g_io_stream_clear_pending (stream);
 | ||
| 
 | ||
|   if (g_async_result_legacy_propagate_error (res, &error))
 | ||
|     success = FALSE;
 | ||
|   else
 | ||
|     success = klass->close_finish (stream, res, &error);
 | ||
| 
 | ||
|   if (error)
 | ||
|     g_task_return_error (task, error);
 | ||
|   else
 | ||
|     g_task_return_boolean (task, success);
 | ||
| 
 | ||
|   g_object_unref (task);
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_close_async:
 | ||
|  * @stream: a #GIOStream
 | ||
|  * @io_priority: the io priority of the request
 | ||
|  * @cancellable: (allow-none): optional cancellable object
 | ||
|  * @callback: (scope async): callback to call when the request is satisfied
 | ||
|  * @user_data: (closure): the data to pass to callback function
 | ||
|  *
 | ||
|  * Requests an asynchronous close of the stream, releasing resources
 | ||
|  * related to it. When the operation is finished @callback will be
 | ||
|  * called. You can then call g_io_stream_close_finish() to get
 | ||
|  * the result of the operation.
 | ||
|  *
 | ||
|  * For behaviour details see g_io_stream_close().
 | ||
|  *
 | ||
|  * The asynchronous methods have a default fallback that uses threads
 | ||
|  * to implement asynchronicity, so they are optional for inheriting
 | ||
|  * classes. However, if you override one you must override all.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| void
 | ||
| g_io_stream_close_async (GIOStream           *stream,
 | ||
| 			 int                  io_priority,
 | ||
| 			 GCancellable        *cancellable,
 | ||
| 			 GAsyncReadyCallback  callback,
 | ||
| 			 gpointer             user_data)
 | ||
| {
 | ||
|   GIOStreamClass *class;
 | ||
|   GError *error = NULL;
 | ||
|   GTask *task;
 | ||
| 
 | ||
|   g_return_if_fail (G_IS_IO_STREAM (stream));
 | ||
| 
 | ||
|   task = g_task_new (stream, cancellable, callback, user_data);
 | ||
| 
 | ||
|   if (stream->priv->closed)
 | ||
|     {
 | ||
|       g_task_return_boolean (task, TRUE);
 | ||
|       g_object_unref (task);
 | ||
|       return;
 | ||
|     }
 | ||
| 
 | ||
|   if (!g_io_stream_set_pending (stream, &error))
 | ||
|     {
 | ||
|       g_task_return_error (task, error);
 | ||
|       g_object_unref (task);
 | ||
|       return;
 | ||
|     }
 | ||
| 
 | ||
|   class = G_IO_STREAM_GET_CLASS (stream);
 | ||
| 
 | ||
|   class->close_async (stream, io_priority, cancellable,
 | ||
| 		      async_ready_close_callback_wrapper, task);
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_close_finish:
 | ||
|  * @stream: a #GIOStream
 | ||
|  * @result: a #GAsyncResult
 | ||
|  * @error: a #GError location to store the error occurring, or %NULL to
 | ||
|  *    ignore
 | ||
|  *
 | ||
|  * Closes a stream.
 | ||
|  *
 | ||
|  * Returns: %TRUE if stream was successfully closed, %FALSE otherwise.
 | ||
|  *
 | ||
|  * Since: 2.22
 | ||
|  */
 | ||
| gboolean
 | ||
| g_io_stream_close_finish (GIOStream     *stream,
 | ||
| 			  GAsyncResult  *result,
 | ||
| 			  GError       **error)
 | ||
| {
 | ||
|   g_return_val_if_fail (G_IS_IO_STREAM (stream), FALSE);
 | ||
|   g_return_val_if_fail (g_task_is_valid (result, stream), FALSE);
 | ||
| 
 | ||
|   return g_task_propagate_boolean (G_TASK (result), error);
 | ||
| }
 | ||
| 
 | ||
| 
 | ||
| static void
 | ||
| close_async_thread (GTask        *task,
 | ||
|                     gpointer      source_object,
 | ||
|                     gpointer      task_data,
 | ||
|                     GCancellable *cancellable)
 | ||
| {
 | ||
|   GIOStream *stream = source_object;
 | ||
|   GIOStreamClass *class;
 | ||
|   GError *error = NULL;
 | ||
|   gboolean result;
 | ||
| 
 | ||
|   class = G_IO_STREAM_GET_CLASS (stream);
 | ||
|   if (class->close_fn)
 | ||
|     {
 | ||
|       result = class->close_fn (stream,
 | ||
|                                 g_task_get_cancellable (task),
 | ||
|                                 &error);
 | ||
|       if (!result)
 | ||
|         {
 | ||
|           g_task_return_error (task, error);
 | ||
|           return;
 | ||
|         }
 | ||
|     }
 | ||
| 
 | ||
|   g_task_return_boolean (task, TRUE);
 | ||
| }
 | ||
| 
 | ||
| typedef struct
 | ||
| {
 | ||
|   GError *error;
 | ||
|   gint pending;
 | ||
| } CloseAsyncData;
 | ||
| 
 | ||
| static void
 | ||
| stream_close_complete (GObject      *source,
 | ||
|                        GAsyncResult *result,
 | ||
|                        gpointer      user_data)
 | ||
| {
 | ||
|   GTask *task = user_data;
 | ||
|   CloseAsyncData *data;
 | ||
| 
 | ||
|   data = g_task_get_task_data (task);
 | ||
|   data->pending--;
 | ||
| 
 | ||
|   if (G_IS_OUTPUT_STREAM (source))
 | ||
|     {
 | ||
|       GError *error = NULL;
 | ||
| 
 | ||
|       /* Match behaviour with the sync route and give precedent to the
 | ||
|        * error returned from closing the output stream.
 | ||
|        */
 | ||
|       g_output_stream_close_finish (G_OUTPUT_STREAM (source), result, &error);
 | ||
|       if (error)
 | ||
|         {
 | ||
|           if (data->error)
 | ||
|             g_error_free (data->error);
 | ||
|           data->error = error;
 | ||
|         }
 | ||
|     }
 | ||
|   else
 | ||
|     g_input_stream_close_finish (G_INPUT_STREAM (source), result, data->error ? NULL : &data->error);
 | ||
| 
 | ||
|   if (data->pending == 0)
 | ||
|     {
 | ||
|       if (data->error)
 | ||
|         g_task_return_error (task, data->error);
 | ||
|       else
 | ||
|         g_task_return_boolean (task, TRUE);
 | ||
| 
 | ||
|       g_slice_free (CloseAsyncData, data);
 | ||
|       g_object_unref (task);
 | ||
|     }
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| g_io_stream_real_close_async (GIOStream           *stream,
 | ||
| 			      int                  io_priority,
 | ||
| 			      GCancellable        *cancellable,
 | ||
| 			      GAsyncReadyCallback  callback,
 | ||
| 			      gpointer             user_data)
 | ||
| {
 | ||
|   GInputStream *input;
 | ||
|   GOutputStream *output;
 | ||
|   GTask *task;
 | ||
| 
 | ||
|   task = g_task_new (stream, cancellable, callback, user_data);
 | ||
|   g_task_set_check_cancellable (task, FALSE);
 | ||
|   g_task_set_priority (task, io_priority);
 | ||
| 
 | ||
|   input = g_io_stream_get_input_stream (stream);
 | ||
|   output = g_io_stream_get_output_stream (stream);
 | ||
| 
 | ||
|   if (g_input_stream_async_close_is_via_threads (input) && g_output_stream_async_close_is_via_threads (output))
 | ||
|     {
 | ||
|       /* No sense in dispatching to the thread twice -- just do it all
 | ||
|        * in one go.
 | ||
|        */
 | ||
|       g_task_run_in_thread (task, close_async_thread);
 | ||
|       g_object_unref (task);
 | ||
|     }
 | ||
|   else
 | ||
|     {
 | ||
|       CloseAsyncData *data;
 | ||
| 
 | ||
|       /* We should avoid dispatching to another thread in case either
 | ||
|        * object that would not do it for itself because it may not be
 | ||
|        * threadsafe.
 | ||
|        */
 | ||
|       data = g_slice_new (CloseAsyncData);
 | ||
|       data->error = NULL;
 | ||
|       data->pending = 2;
 | ||
| 
 | ||
|       g_task_set_task_data (task, data, NULL);
 | ||
|       g_input_stream_close_async (input, io_priority, cancellable, stream_close_complete, task);
 | ||
|       g_output_stream_close_async (output, io_priority, cancellable, stream_close_complete, task);
 | ||
|     }
 | ||
| }
 | ||
| 
 | ||
| static gboolean
 | ||
| g_io_stream_real_close_finish (GIOStream     *stream,
 | ||
| 			       GAsyncResult  *result,
 | ||
| 			       GError       **error)
 | ||
| {
 | ||
|   g_return_val_if_fail (g_task_is_valid (result, stream), FALSE);
 | ||
| 
 | ||
|   return g_task_propagate_boolean (G_TASK (result), error);
 | ||
| }
 | ||
| 
 | ||
| typedef struct
 | ||
| {
 | ||
|   GIOStream *stream1;
 | ||
|   GIOStream *stream2;
 | ||
|   GIOStreamSpliceFlags flags;
 | ||
|   gint io_priority;
 | ||
|   GCancellable *cancellable;
 | ||
|   gulong cancelled_id;
 | ||
|   GCancellable *op1_cancellable;
 | ||
|   GCancellable *op2_cancellable;
 | ||
|   guint completed;
 | ||
|   GError *error;
 | ||
| } SpliceContext;
 | ||
| 
 | ||
| static void
 | ||
| splice_context_free (SpliceContext *ctx)
 | ||
| {
 | ||
|   g_object_unref (ctx->stream1);
 | ||
|   g_object_unref (ctx->stream2);
 | ||
|   if (ctx->cancellable != NULL)
 | ||
|     g_object_unref (ctx->cancellable);
 | ||
|   g_object_unref (ctx->op1_cancellable);
 | ||
|   g_object_unref (ctx->op2_cancellable);
 | ||
|   g_clear_error (&ctx->error);
 | ||
|   g_slice_free (SpliceContext, ctx);
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| splice_complete (GTask         *task,
 | ||
|                  SpliceContext *ctx)
 | ||
| {
 | ||
|   if (ctx->cancelled_id != 0)
 | ||
|     g_cancellable_disconnect (ctx->cancellable, ctx->cancelled_id);
 | ||
|   ctx->cancelled_id = 0;
 | ||
| 
 | ||
|   if (ctx->error != NULL)
 | ||
|     {
 | ||
|       g_task_return_error (task, ctx->error);
 | ||
|       ctx->error = NULL;
 | ||
|     }
 | ||
|   else
 | ||
|     g_task_return_boolean (task, TRUE);
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| splice_close_cb (GObject      *iostream,
 | ||
|                  GAsyncResult *res,
 | ||
|                  gpointer      user_data)
 | ||
| {
 | ||
|   GTask *task = user_data;
 | ||
|   SpliceContext *ctx = g_task_get_task_data (task);
 | ||
|   GError *error = NULL;
 | ||
| 
 | ||
|   g_io_stream_close_finish (G_IO_STREAM (iostream), res, &error);
 | ||
| 
 | ||
|   ctx->completed++;
 | ||
| 
 | ||
|   /* Keep the first error that occurred */
 | ||
|   if (error != NULL && ctx->error == NULL)
 | ||
|     ctx->error = error;
 | ||
|   else
 | ||
|     g_clear_error (&error);
 | ||
| 
 | ||
|   /* If all operations are done, complete now */
 | ||
|   if (ctx->completed == 4)
 | ||
|     splice_complete (task, ctx);
 | ||
| 
 | ||
|   g_object_unref (task);
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| splice_cb (GObject      *ostream,
 | ||
|            GAsyncResult *res,
 | ||
|            gpointer      user_data)
 | ||
| {
 | ||
|   GTask *task = user_data;
 | ||
|   SpliceContext *ctx = g_task_get_task_data (task);
 | ||
|   GError *error = NULL;
 | ||
| 
 | ||
|   g_output_stream_splice_finish (G_OUTPUT_STREAM (ostream), res, &error);
 | ||
| 
 | ||
|   ctx->completed++;
 | ||
| 
 | ||
|   /* ignore cancellation error if it was not requested by the user */
 | ||
|   if (error != NULL &&
 | ||
|       g_error_matches (error, G_IO_ERROR, G_IO_ERROR_CANCELLED) &&
 | ||
|       (ctx->cancellable == NULL ||
 | ||
|        !g_cancellable_is_cancelled (ctx->cancellable)))
 | ||
|     g_clear_error (&error);
 | ||
| 
 | ||
|   /* Keep the first error that occurred */
 | ||
|   if (error != NULL && ctx->error == NULL)
 | ||
|     ctx->error = error;
 | ||
|   else
 | ||
|     g_clear_error (&error);
 | ||
| 
 | ||
|    if (ctx->completed == 1 &&
 | ||
|        (ctx->flags & G_IO_STREAM_SPLICE_WAIT_FOR_BOTH) == 0)
 | ||
|     {
 | ||
|       /* We don't want to wait for the 2nd operation to finish, cancel it */
 | ||
|       g_cancellable_cancel (ctx->op1_cancellable);
 | ||
|       g_cancellable_cancel (ctx->op2_cancellable);
 | ||
|     }
 | ||
|   else if (ctx->completed == 2)
 | ||
|     {
 | ||
|       if (ctx->cancellable == NULL ||
 | ||
|           !g_cancellable_is_cancelled (ctx->cancellable))
 | ||
|         {
 | ||
|           g_cancellable_reset (ctx->op1_cancellable);
 | ||
|           g_cancellable_reset (ctx->op2_cancellable);
 | ||
|         }
 | ||
| 
 | ||
|       /* Close the IO streams if needed */
 | ||
|       if ((ctx->flags & G_IO_STREAM_SPLICE_CLOSE_STREAM1) != 0)
 | ||
| 	{
 | ||
| 	  g_io_stream_close_async (ctx->stream1,
 | ||
|                                    g_task_get_priority (task),
 | ||
|                                    ctx->op1_cancellable,
 | ||
|                                    splice_close_cb, g_object_ref (task));
 | ||
| 	}
 | ||
|       else
 | ||
|         ctx->completed++;
 | ||
| 
 | ||
|       if ((ctx->flags & G_IO_STREAM_SPLICE_CLOSE_STREAM2) != 0)
 | ||
| 	{
 | ||
| 	  g_io_stream_close_async (ctx->stream2,
 | ||
|                                    g_task_get_priority (task),
 | ||
|                                    ctx->op2_cancellable,
 | ||
|                                    splice_close_cb, g_object_ref (task));
 | ||
| 	}
 | ||
|       else
 | ||
|         ctx->completed++;
 | ||
| 
 | ||
|       /* If all operations are done, complete now */
 | ||
|       if (ctx->completed == 4)
 | ||
|         splice_complete (task, ctx);
 | ||
|     }
 | ||
| 
 | ||
|   g_object_unref (task);
 | ||
| }
 | ||
| 
 | ||
| static void
 | ||
| splice_cancelled_cb (GCancellable *cancellable,
 | ||
|                      GTask        *task)
 | ||
| {
 | ||
|   SpliceContext *ctx;
 | ||
| 
 | ||
|   ctx = g_task_get_task_data (task);
 | ||
|   g_cancellable_cancel (ctx->op1_cancellable);
 | ||
|   g_cancellable_cancel (ctx->op2_cancellable);
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_splice_async:
 | ||
|  * @stream1: a #GIOStream.
 | ||
|  * @stream2: a #GIOStream.
 | ||
|  * @flags: a set of #GIOStreamSpliceFlags.
 | ||
|  * @io_priority: the io priority of the request.
 | ||
|  * @cancellable: (allow-none): optional #GCancellable object, %NULL to ignore.
 | ||
|  * @callback: (scope async): a #GAsyncReadyCallback.
 | ||
|  * @user_data: (closure): user data passed to @callback.
 | ||
|  *
 | ||
|  * Asyncronously splice the output stream of @stream1 to the input stream of
 | ||
|  * @stream2, and splice the output stream of @stream2 to the input stream of
 | ||
|  * @stream1.
 | ||
|  *
 | ||
|  * When the operation is finished @callback will be called.
 | ||
|  * You can then call g_io_stream_splice_finish() to get the
 | ||
|  * result of the operation.
 | ||
|  *
 | ||
|  * Since: 2.28
 | ||
|  **/
 | ||
| void
 | ||
| g_io_stream_splice_async (GIOStream            *stream1,
 | ||
|                           GIOStream            *stream2,
 | ||
|                           GIOStreamSpliceFlags  flags,
 | ||
|                           gint                  io_priority,
 | ||
|                           GCancellable         *cancellable,
 | ||
|                           GAsyncReadyCallback   callback,
 | ||
|                           gpointer              user_data)
 | ||
| {
 | ||
|   GTask *task;
 | ||
|   SpliceContext *ctx;
 | ||
|   GInputStream *istream;
 | ||
|   GOutputStream *ostream;
 | ||
| 
 | ||
|   if (cancellable != NULL && g_cancellable_is_cancelled (cancellable))
 | ||
|     {
 | ||
|       g_task_report_new_error (NULL, callback, user_data,
 | ||
|                                g_io_stream_splice_async,
 | ||
|                                G_IO_ERROR, G_IO_ERROR_CANCELLED,
 | ||
|                                "Operation has been cancelled");
 | ||
|       return;
 | ||
|     }
 | ||
| 
 | ||
|   ctx = g_slice_new0 (SpliceContext);
 | ||
|   ctx->stream1 = g_object_ref (stream1);
 | ||
|   ctx->stream2 = g_object_ref (stream2);
 | ||
|   ctx->flags = flags;
 | ||
|   ctx->op1_cancellable = g_cancellable_new ();
 | ||
|   ctx->op2_cancellable = g_cancellable_new ();
 | ||
|   ctx->completed = 0;
 | ||
| 
 | ||
|   task = g_task_new (NULL, cancellable, callback, user_data);
 | ||
|   g_task_set_task_data (task, ctx, (GDestroyNotify) splice_context_free);
 | ||
| 
 | ||
|   if (cancellable != NULL)
 | ||
|     {
 | ||
|       ctx->cancellable = g_object_ref (cancellable);
 | ||
|       ctx->cancelled_id = g_cancellable_connect (cancellable,
 | ||
|           G_CALLBACK (splice_cancelled_cb), g_object_ref (task),
 | ||
|           g_object_unref);
 | ||
|     }
 | ||
| 
 | ||
|   istream = g_io_stream_get_input_stream (stream1);
 | ||
|   ostream = g_io_stream_get_output_stream (stream2);
 | ||
|   g_output_stream_splice_async (ostream, istream, G_OUTPUT_STREAM_SPLICE_NONE,
 | ||
|       io_priority, ctx->op1_cancellable, splice_cb,
 | ||
|       g_object_ref (task));
 | ||
| 
 | ||
|   istream = g_io_stream_get_input_stream (stream2);
 | ||
|   ostream = g_io_stream_get_output_stream (stream1);
 | ||
|   g_output_stream_splice_async (ostream, istream, G_OUTPUT_STREAM_SPLICE_NONE,
 | ||
|       io_priority, ctx->op2_cancellable, splice_cb,
 | ||
|       g_object_ref (task));
 | ||
| 
 | ||
|   g_object_unref (task);
 | ||
| }
 | ||
| 
 | ||
| /**
 | ||
|  * g_io_stream_splice_finish:
 | ||
|  * @result: a #GAsyncResult.
 | ||
|  * @error: a #GError location to store the error occurring, or %NULL to
 | ||
|  * ignore.
 | ||
|  *
 | ||
|  * Finishes an asynchronous io stream splice operation.
 | ||
|  *
 | ||
|  * Returns: %TRUE on success, %FALSE otherwise.
 | ||
|  *
 | ||
|  * Since: 2.28
 | ||
|  **/
 | ||
| gboolean
 | ||
| g_io_stream_splice_finish (GAsyncResult  *result,
 | ||
|                            GError       **error)
 | ||
| {
 | ||
|   g_return_val_if_fail (g_task_is_valid (result, NULL), FALSE);
 | ||
| 
 | ||
|   return g_task_propagate_boolean (G_TASK (result), error);
 | ||
| }
 |