1 /* GIO - GLib Input, Output and Streaming Library
3 * Copyright (C) 2006-2007 Red Hat, Inc.
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.
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.
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, write to the
17 * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
18 * Boston, MA 02111-1307, USA.
20 * Author: Alexander Larsson <alexl@redhat.com>
25 #include <sys/types.h>
33 #include <glib/gstdio.h>
35 #include "gunixoutputstream.h"
36 #include "gcancellable.h"
37 #include "gsimpleasyncresult.h"
38 #include "gasynchelper.h"
39 #include "gfiledescriptorbased.h"
44 * SECTION:gunixoutputstream
45 * @short_description: Streaming output operations for UNIX file descriptors
46 * @include: gio/gunixoutputstream.h
47 * @see_also: #GOutputStream
49 * #GUnixOutputStream implements #GOutputStream for writing to a UNIX
50 * file descriptor, including asynchronous operations. (If the file
51 * descriptor refers to a socket or pipe, this will use poll() to do
52 * asynchronous I/O. If it refers to a regular file, it will fall back
53 * to doing asynchronous I/O in another thread.)
55 * Note that <filename><gio/gunixoutputstream.h></filename> belongs
56 * to the UNIX-specific GIO interfaces, thus you have to use the
57 * <filename>gio-unix-2.0.pc</filename> pkg-config file when using it.
66 static void g_unix_output_stream_pollable_iface_init (GPollableOutputStreamInterface
*iface
);
67 static void g_unix_output_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface
*iface
);
69 G_DEFINE_TYPE_WITH_CODE (GUnixOutputStream
, g_unix_output_stream
, G_TYPE_OUTPUT_STREAM
,
70 G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_OUTPUT_STREAM
,
71 g_unix_output_stream_pollable_iface_init
)
72 G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED
,
73 g_unix_output_stream_file_descriptor_based_iface_init
)
76 struct _GUnixOutputStreamPrivate
{
79 guint is_pipe_or_socket
: 1;
82 static void g_unix_output_stream_set_property (GObject
*object
,
86 static void g_unix_output_stream_get_property (GObject
*object
,
90 static gssize
g_unix_output_stream_write (GOutputStream
*stream
,
93 GCancellable
*cancellable
,
95 static gboolean
g_unix_output_stream_close (GOutputStream
*stream
,
96 GCancellable
*cancellable
,
98 static void g_unix_output_stream_close_async (GOutputStream
*stream
,
100 GCancellable
*cancellable
,
101 GAsyncReadyCallback callback
,
103 static gboolean
g_unix_output_stream_close_finish (GOutputStream
*stream
,
104 GAsyncResult
*result
,
107 static gboolean
g_unix_output_stream_pollable_can_poll (GPollableOutputStream
*stream
);
108 static gboolean
g_unix_output_stream_pollable_is_writable (GPollableOutputStream
*stream
);
109 static GSource
*g_unix_output_stream_pollable_create_source (GPollableOutputStream
*stream
,
110 GCancellable
*cancellable
);
113 g_unix_output_stream_finalize (GObject
*object
)
115 G_OBJECT_CLASS (g_unix_output_stream_parent_class
)->finalize (object
);
119 g_unix_output_stream_class_init (GUnixOutputStreamClass
*klass
)
121 GObjectClass
*gobject_class
= G_OBJECT_CLASS (klass
);
122 GOutputStreamClass
*stream_class
= G_OUTPUT_STREAM_CLASS (klass
);
124 g_type_class_add_private (klass
, sizeof (GUnixOutputStreamPrivate
));
126 gobject_class
->get_property
= g_unix_output_stream_get_property
;
127 gobject_class
->set_property
= g_unix_output_stream_set_property
;
128 gobject_class
->finalize
= g_unix_output_stream_finalize
;
130 stream_class
->write_fn
= g_unix_output_stream_write
;
131 stream_class
->close_fn
= g_unix_output_stream_close
;
132 stream_class
->close_async
= g_unix_output_stream_close_async
;
133 stream_class
->close_finish
= g_unix_output_stream_close_finish
;
136 * GUnixOutputStream:fd:
138 * The file descriptor that the stream writes to.
142 g_object_class_install_property (gobject_class
,
144 g_param_spec_int ("fd",
145 P_("File descriptor"),
146 P_("The file descriptor to write to"),
147 G_MININT
, G_MAXINT
, -1,
148 G_PARAM_READABLE
| G_PARAM_WRITABLE
| G_PARAM_CONSTRUCT_ONLY
| G_PARAM_STATIC_NAME
| G_PARAM_STATIC_NICK
| G_PARAM_STATIC_BLURB
));
151 * GUnixOutputStream:close-fd:
153 * Whether to close the file descriptor when the stream is closed.
157 g_object_class_install_property (gobject_class
,
159 g_param_spec_boolean ("close-fd",
160 P_("Close file descriptor"),
161 P_("Whether to close the file descriptor when the stream is closed"),
163 G_PARAM_READABLE
| G_PARAM_WRITABLE
| G_PARAM_STATIC_NAME
| G_PARAM_STATIC_NICK
| G_PARAM_STATIC_BLURB
));
167 g_unix_output_stream_pollable_iface_init (GPollableOutputStreamInterface
*iface
)
169 iface
->can_poll
= g_unix_output_stream_pollable_can_poll
;
170 iface
->is_writable
= g_unix_output_stream_pollable_is_writable
;
171 iface
->create_source
= g_unix_output_stream_pollable_create_source
;
175 g_unix_output_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface
*iface
)
177 iface
->get_fd
= (int (*) (GFileDescriptorBased
*))g_unix_output_stream_get_fd
;
181 g_unix_output_stream_set_property (GObject
*object
,
186 GUnixOutputStream
*unix_stream
;
188 unix_stream
= G_UNIX_OUTPUT_STREAM (object
);
193 unix_stream
->priv
->fd
= g_value_get_int (value
);
194 if (lseek (unix_stream
->priv
->fd
, 0, SEEK_CUR
) == -1 && errno
== ESPIPE
)
195 unix_stream
->priv
->is_pipe_or_socket
= TRUE
;
197 unix_stream
->priv
->is_pipe_or_socket
= FALSE
;
200 unix_stream
->priv
->close_fd
= g_value_get_boolean (value
);
203 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
209 g_unix_output_stream_get_property (GObject
*object
,
214 GUnixOutputStream
*unix_stream
;
216 unix_stream
= G_UNIX_OUTPUT_STREAM (object
);
221 g_value_set_int (value
, unix_stream
->priv
->fd
);
224 g_value_set_boolean (value
, unix_stream
->priv
->close_fd
);
227 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
232 g_unix_output_stream_init (GUnixOutputStream
*unix_stream
)
234 unix_stream
->priv
= G_TYPE_INSTANCE_GET_PRIVATE (unix_stream
,
235 G_TYPE_UNIX_OUTPUT_STREAM
,
236 GUnixOutputStreamPrivate
);
238 unix_stream
->priv
->fd
= -1;
239 unix_stream
->priv
->close_fd
= TRUE
;
243 * g_unix_output_stream_new:
244 * @fd: a UNIX file descriptor
245 * @close_fd: %TRUE to close the file descriptor when done
247 * Creates a new #GUnixOutputStream for the given @fd.
249 * If @close_fd, is %TRUE, the file descriptor will be closed when
250 * the output stream is destroyed.
252 * Returns: a new #GOutputStream
255 g_unix_output_stream_new (gint fd
,
258 GUnixOutputStream
*stream
;
260 g_return_val_if_fail (fd
!= -1, NULL
);
262 stream
= g_object_new (G_TYPE_UNIX_OUTPUT_STREAM
,
264 "close-fd", close_fd
,
267 return G_OUTPUT_STREAM (stream
);
271 * g_unix_output_stream_set_close_fd:
272 * @stream: a #GUnixOutputStream
273 * @close_fd: %TRUE to close the file descriptor when done
275 * Sets whether the file descriptor of @stream shall be closed
276 * when the stream is closed.
281 g_unix_output_stream_set_close_fd (GUnixOutputStream
*stream
,
284 g_return_if_fail (G_IS_UNIX_OUTPUT_STREAM (stream
));
286 close_fd
= close_fd
!= FALSE
;
287 if (stream
->priv
->close_fd
!= close_fd
)
289 stream
->priv
->close_fd
= close_fd
;
290 g_object_notify (G_OBJECT (stream
), "close-fd");
295 * g_unix_output_stream_get_close_fd:
296 * @stream: a #GUnixOutputStream
298 * Returns whether the file descriptor of @stream will be
299 * closed when the stream is closed.
301 * Return value: %TRUE if the file descriptor is closed when done
306 g_unix_output_stream_get_close_fd (GUnixOutputStream
*stream
)
308 g_return_val_if_fail (G_IS_UNIX_OUTPUT_STREAM (stream
), FALSE
);
310 return stream
->priv
->close_fd
;
314 * g_unix_output_stream_get_fd:
315 * @stream: a #GUnixOutputStream
317 * Return the UNIX file descriptor that the stream writes to.
319 * Return value: The file descriptor of @stream
324 g_unix_output_stream_get_fd (GUnixOutputStream
*stream
)
326 g_return_val_if_fail (G_IS_UNIX_OUTPUT_STREAM (stream
), -1);
328 return stream
->priv
->fd
;
332 g_unix_output_stream_write (GOutputStream
*stream
,
335 GCancellable
*cancellable
,
338 GUnixOutputStream
*unix_stream
;
344 unix_stream
= G_UNIX_OUTPUT_STREAM (stream
);
346 poll_fds
[0].fd
= unix_stream
->priv
->fd
;
347 poll_fds
[0].events
= G_IO_OUT
;
349 if (unix_stream
->priv
->is_pipe_or_socket
&&
350 g_cancellable_make_pollfd (cancellable
, &poll_fds
[1]))
357 poll_fds
[0].revents
= poll_fds
[1].revents
= 0;
359 poll_ret
= g_poll (poll_fds
, nfds
, -1);
360 while (poll_ret
== -1 && errno
== EINTR
);
366 g_set_error (error
, G_IO_ERROR
,
367 g_io_error_from_errno (errsv
),
368 _("Error writing to file descriptor: %s"),
373 if (g_cancellable_set_error_if_cancelled (cancellable
, error
))
376 if (!poll_fds
[0].revents
)
379 res
= write (unix_stream
->priv
->fd
, buffer
, count
);
384 if (errsv
== EINTR
|| errsv
== EAGAIN
)
387 g_set_error (error
, G_IO_ERROR
,
388 g_io_error_from_errno (errsv
),
389 _("Error writing to file descriptor: %s"),
397 g_cancellable_release_fd (cancellable
);
402 g_unix_output_stream_close (GOutputStream
*stream
,
403 GCancellable
*cancellable
,
406 GUnixOutputStream
*unix_stream
;
409 unix_stream
= G_UNIX_OUTPUT_STREAM (stream
);
411 if (!unix_stream
->priv
->close_fd
)
414 /* This might block during the close. Doesn't seem to be a way to avoid it though. */
415 res
= close (unix_stream
->priv
->fd
);
420 g_set_error (error
, G_IO_ERROR
,
421 g_io_error_from_errno (errsv
),
422 _("Error closing file descriptor: %s"),
430 g_unix_output_stream_close_async (GOutputStream
*stream
,
432 GCancellable
*cancellable
,
433 GAsyncReadyCallback callback
,
437 GError
*error
= NULL
;
439 task
= g_task_new (stream
, cancellable
, callback
, user_data
);
440 g_task_set_priority (task
, io_priority
);
442 if (g_unix_output_stream_close (stream
, cancellable
, &error
))
443 g_task_return_boolean (task
, TRUE
);
445 g_task_return_error (task
, error
);
446 g_object_unref (task
);
450 g_unix_output_stream_close_finish (GOutputStream
*stream
,
451 GAsyncResult
*result
,
454 g_return_val_if_fail (g_task_is_valid (result
, stream
), FALSE
);
456 return g_task_propagate_boolean (G_TASK (result
), error
);
460 g_unix_output_stream_pollable_can_poll (GPollableOutputStream
*stream
)
462 return G_UNIX_OUTPUT_STREAM (stream
)->priv
->is_pipe_or_socket
;
466 g_unix_output_stream_pollable_is_writable (GPollableOutputStream
*stream
)
468 GUnixOutputStream
*unix_stream
= G_UNIX_OUTPUT_STREAM (stream
);
472 poll_fd
.fd
= unix_stream
->priv
->fd
;
473 poll_fd
.events
= G_IO_OUT
;
477 result
= g_poll (&poll_fd
, 1, 0);
478 while (result
== -1 && errno
== EINTR
);
480 return poll_fd
.revents
!= 0;
484 g_unix_output_stream_pollable_create_source (GPollableOutputStream
*stream
,
485 GCancellable
*cancellable
)
487 GUnixOutputStream
*unix_stream
= G_UNIX_OUTPUT_STREAM (stream
);
488 GSource
*inner_source
, *pollable_source
;
490 pollable_source
= g_pollable_source_new (G_OBJECT (stream
));
492 inner_source
= _g_fd_source_new (unix_stream
->priv
->fd
, G_IO_OUT
, cancellable
);
493 g_source_set_dummy_callback (inner_source
);
494 g_source_add_child_source (pollable_source
, inner_source
);
495 g_source_unref (inner_source
);
497 return pollable_source
;