1 /* GIO - GLib Input, Output and Streaming Library
3 * Copyright © 2008 Christian Kellner, Samuel Cormier-Iijima
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 of the License, or (at your option) any later version.
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.
16 * You should have received a copy of the GNU Lesser General
17 * Public License along with this library; if not, write to the
18 * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
19 * Boston, MA 02111-1307, USA.
21 * Authors: Christian Kellner <gicmo@gnome.org>
22 * Samuel Cormier-Iijima <sciyoshi@gmail.com>
23 * Ryan Lortie <desrt@desrt.ca>
27 #include "goutputstream.h"
28 #include "gsocketoutputstream.h"
32 #include "gsimpleasyncresult.h"
33 #include "gcancellable.h"
34 #include "gpollableinputstream.h"
35 #include "gpollableoutputstream.h"
38 #include "gfiledescriptorbased.h"
40 static void g_socket_output_stream_pollable_iface_init (GPollableOutputStreamInterface
*iface
);
42 static void g_socket_output_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface
*iface
);
45 #define g_socket_output_stream_get_type _g_socket_output_stream_get_type
48 G_DEFINE_TYPE_WITH_CODE (GSocketOutputStream
, g_socket_output_stream
, G_TYPE_OUTPUT_STREAM
,
49 G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_OUTPUT_STREAM
, g_socket_output_stream_pollable_iface_init
)
50 G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED
, g_socket_output_stream_file_descriptor_based_iface_init
)
53 G_DEFINE_TYPE_WITH_CODE (GSocketOutputStream
, g_socket_output_stream
, G_TYPE_OUTPUT_STREAM
,
54 G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_OUTPUT_STREAM
, g_socket_output_stream_pollable_iface_init
)
64 struct _GSocketOutputStreamPrivate
68 /* pending operation metadata */
69 GSimpleAsyncResult
*result
;
70 GCancellable
*cancellable
;
76 g_socket_output_stream_get_property (GObject
*object
,
81 GSocketOutputStream
*stream
= G_SOCKET_OUTPUT_STREAM (object
);
86 g_value_set_object (value
, stream
->priv
->socket
);
90 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
95 g_socket_output_stream_set_property (GObject
*object
,
100 GSocketOutputStream
*stream
= G_SOCKET_OUTPUT_STREAM (object
);
105 stream
->priv
->socket
= g_value_dup_object (value
);
109 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
114 g_socket_output_stream_finalize (GObject
*object
)
116 GSocketOutputStream
*stream
= G_SOCKET_OUTPUT_STREAM (object
);
118 if (stream
->priv
->socket
)
119 g_object_unref (stream
->priv
->socket
);
121 if (G_OBJECT_CLASS (g_socket_output_stream_parent_class
)->finalize
)
122 (*G_OBJECT_CLASS (g_socket_output_stream_parent_class
)->finalize
) (object
);
126 g_socket_output_stream_write (GOutputStream
*stream
,
129 GCancellable
*cancellable
,
132 GSocketOutputStream
*onput_stream
= G_SOCKET_OUTPUT_STREAM (stream
);
134 return g_socket_send_with_blocking (onput_stream
->priv
->socket
,
140 g_socket_output_stream_write_ready (GSocket
*socket
,
141 GIOCondition condition
,
142 GSocketOutputStream
*stream
)
144 GSimpleAsyncResult
*simple
;
145 GError
*error
= NULL
;
148 result
= g_socket_send_with_blocking (stream
->priv
->socket
,
149 stream
->priv
->buffer
,
152 stream
->priv
->cancellable
,
155 if (g_error_matches (error
, G_IO_ERROR
, G_IO_ERROR_WOULD_BLOCK
))
158 simple
= stream
->priv
->result
;
159 stream
->priv
->result
= NULL
;
162 g_simple_async_result_set_op_res_gssize (simple
, result
);
165 g_simple_async_result_take_error (simple
, error
);
167 if (stream
->priv
->cancellable
)
168 g_object_unref (stream
->priv
->cancellable
);
170 g_simple_async_result_complete (simple
);
171 g_object_unref (simple
);
177 g_socket_output_stream_write_async (GOutputStream
*stream
,
181 GCancellable
*cancellable
,
182 GAsyncReadyCallback callback
,
185 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (stream
);
188 g_assert (output_stream
->priv
->result
== NULL
);
190 output_stream
->priv
->result
=
191 g_simple_async_result_new (G_OBJECT (stream
), callback
, user_data
,
192 g_socket_output_stream_write_async
);
194 g_object_ref (cancellable
);
195 output_stream
->priv
->cancellable
= cancellable
;
196 output_stream
->priv
->buffer
= buffer
;
197 output_stream
->priv
->count
= count
;
199 source
= g_socket_create_source (output_stream
->priv
->socket
,
200 G_IO_OUT
| G_IO_HUP
| G_IO_ERR
,
202 g_source_set_callback (source
,
203 (GSourceFunc
) g_socket_output_stream_write_ready
,
204 g_object_ref (output_stream
), g_object_unref
);
205 g_source_attach (source
, g_main_context_get_thread_default ());
206 g_source_unref (source
);
210 g_socket_output_stream_write_finish (GOutputStream
*stream
,
211 GAsyncResult
*result
,
214 GSimpleAsyncResult
*simple
;
217 g_return_val_if_fail (G_IS_SOCKET_OUTPUT_STREAM (stream
), -1);
219 simple
= G_SIMPLE_ASYNC_RESULT (result
);
221 g_warn_if_fail (g_simple_async_result_get_source_tag (simple
) == g_socket_output_stream_write_async
);
223 count
= g_simple_async_result_get_op_res_gssize (simple
);
229 g_socket_output_stream_pollable_is_writable (GPollableOutputStream
*pollable
)
231 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (pollable
);
233 return g_socket_condition_check (output_stream
->priv
->socket
, G_IO_OUT
);
237 g_socket_output_stream_pollable_create_source (GPollableOutputStream
*pollable
,
238 GCancellable
*cancellable
)
240 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (pollable
);
241 GSource
*socket_source
, *pollable_source
;
243 pollable_source
= g_pollable_source_new (G_OBJECT (output_stream
));
244 socket_source
= g_socket_create_source (output_stream
->priv
->socket
,
245 G_IO_OUT
, cancellable
);
246 g_source_set_dummy_callback (socket_source
);
247 g_source_add_child_source (pollable_source
, socket_source
);
248 g_source_unref (socket_source
);
250 return pollable_source
;
254 g_socket_output_stream_pollable_write_nonblocking (GPollableOutputStream
*pollable
,
259 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (pollable
);
261 return g_socket_send_with_blocking (output_stream
->priv
->socket
,
268 g_socket_output_stream_get_fd (GFileDescriptorBased
*fd_based
)
270 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (fd_based
);
272 return g_socket_get_fd (output_stream
->priv
->socket
);
277 g_socket_output_stream_class_init (GSocketOutputStreamClass
*klass
)
279 GObjectClass
*gobject_class
= G_OBJECT_CLASS (klass
);
280 GOutputStreamClass
*goutputstream_class
= G_OUTPUT_STREAM_CLASS (klass
);
282 g_type_class_add_private (klass
, sizeof (GSocketOutputStreamPrivate
));
284 gobject_class
->finalize
= g_socket_output_stream_finalize
;
285 gobject_class
->get_property
= g_socket_output_stream_get_property
;
286 gobject_class
->set_property
= g_socket_output_stream_set_property
;
288 goutputstream_class
->write_fn
= g_socket_output_stream_write
;
289 goutputstream_class
->write_async
= g_socket_output_stream_write_async
;
290 goutputstream_class
->write_finish
= g_socket_output_stream_write_finish
;
292 g_object_class_install_property (gobject_class
, PROP_SOCKET
,
293 g_param_spec_object ("socket",
295 P_("The socket that this stream wraps"),
296 G_TYPE_SOCKET
, G_PARAM_CONSTRUCT_ONLY
|
297 G_PARAM_READWRITE
| G_PARAM_STATIC_STRINGS
));
302 g_socket_output_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface
*iface
)
304 iface
->get_fd
= g_socket_output_stream_get_fd
;
309 g_socket_output_stream_pollable_iface_init (GPollableOutputStreamInterface
*iface
)
311 iface
->is_writable
= g_socket_output_stream_pollable_is_writable
;
312 iface
->create_source
= g_socket_output_stream_pollable_create_source
;
313 iface
->write_nonblocking
= g_socket_output_stream_pollable_write_nonblocking
;
317 g_socket_output_stream_init (GSocketOutputStream
*stream
)
319 stream
->priv
= G_TYPE_INSTANCE_GET_PRIVATE (stream
, G_TYPE_SOCKET_OUTPUT_STREAM
, GSocketOutputStreamPrivate
);
322 GSocketOutputStream
*
323 _g_socket_output_stream_new (GSocket
*socket
)
325 return G_SOCKET_OUTPUT_STREAM (g_object_new (G_TYPE_SOCKET_OUTPUT_STREAM
, "socket", socket
, NULL
));