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"
31 #include "gsimpleasyncresult.h"
32 #include "gcancellable.h"
33 #include "gpollableinputstream.h"
34 #include "gpollableoutputstream.h"
39 static void g_socket_output_stream_pollable_iface_init (GPollableOutputStreamInterface
*iface
);
41 #define g_socket_output_stream_get_type _g_socket_output_stream_get_type
42 G_DEFINE_TYPE_WITH_CODE (GSocketOutputStream
, g_socket_output_stream
, G_TYPE_OUTPUT_STREAM
,
43 G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_OUTPUT_STREAM
, g_socket_output_stream_pollable_iface_init
)
52 struct _GSocketOutputStreamPrivate
56 /* pending operation metadata */
57 GSimpleAsyncResult
*result
;
58 GCancellable
*cancellable
;
64 g_socket_output_stream_get_property (GObject
*object
,
69 GSocketOutputStream
*stream
= G_SOCKET_OUTPUT_STREAM (object
);
74 g_value_set_object (value
, stream
->priv
->socket
);
78 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
83 g_socket_output_stream_set_property (GObject
*object
,
88 GSocketOutputStream
*stream
= G_SOCKET_OUTPUT_STREAM (object
);
93 stream
->priv
->socket
= g_value_dup_object (value
);
97 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
102 g_socket_output_stream_finalize (GObject
*object
)
104 GSocketOutputStream
*stream
= G_SOCKET_OUTPUT_STREAM (object
);
106 if (stream
->priv
->socket
)
107 g_object_unref (stream
->priv
->socket
);
109 if (G_OBJECT_CLASS (g_socket_output_stream_parent_class
)->finalize
)
110 (*G_OBJECT_CLASS (g_socket_output_stream_parent_class
)->finalize
) (object
);
114 g_socket_output_stream_write (GOutputStream
*stream
,
117 GCancellable
*cancellable
,
120 GSocketOutputStream
*onput_stream
= G_SOCKET_OUTPUT_STREAM (stream
);
122 return g_socket_send_with_blocking (onput_stream
->priv
->socket
,
128 g_socket_output_stream_write_ready (GSocket
*socket
,
129 GIOCondition condition
,
130 GSocketOutputStream
*stream
)
132 GSimpleAsyncResult
*simple
;
133 GError
*error
= NULL
;
136 result
= g_socket_send_with_blocking (stream
->priv
->socket
,
137 stream
->priv
->buffer
,
140 stream
->priv
->cancellable
,
143 if (g_error_matches (error
, G_IO_ERROR
, G_IO_ERROR_WOULD_BLOCK
))
146 simple
= stream
->priv
->result
;
147 stream
->priv
->result
= NULL
;
150 g_simple_async_result_set_op_res_gssize (simple
, result
);
153 g_simple_async_result_take_error (simple
, error
);
155 if (stream
->priv
->cancellable
)
156 g_object_unref (stream
->priv
->cancellable
);
158 g_simple_async_result_complete (simple
);
159 g_object_unref (simple
);
165 g_socket_output_stream_write_async (GOutputStream
*stream
,
169 GCancellable
*cancellable
,
170 GAsyncReadyCallback callback
,
173 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (stream
);
176 g_assert (output_stream
->priv
->result
== NULL
);
178 output_stream
->priv
->result
=
179 g_simple_async_result_new (G_OBJECT (stream
), callback
, user_data
,
180 g_socket_output_stream_write_async
);
182 g_object_ref (cancellable
);
183 output_stream
->priv
->cancellable
= cancellable
;
184 output_stream
->priv
->buffer
= buffer
;
185 output_stream
->priv
->count
= count
;
187 source
= g_socket_create_source (output_stream
->priv
->socket
,
188 G_IO_OUT
| G_IO_HUP
| G_IO_ERR
,
190 g_source_set_callback (source
,
191 (GSourceFunc
) g_socket_output_stream_write_ready
,
192 g_object_ref (output_stream
), g_object_unref
);
193 g_source_attach (source
, g_main_context_get_thread_default ());
194 g_source_unref (source
);
198 g_socket_output_stream_write_finish (GOutputStream
*stream
,
199 GAsyncResult
*result
,
202 GSimpleAsyncResult
*simple
;
205 g_return_val_if_fail (G_IS_SOCKET_OUTPUT_STREAM (stream
), -1);
207 simple
= G_SIMPLE_ASYNC_RESULT (result
);
209 g_warn_if_fail (g_simple_async_result_get_source_tag (simple
) == g_socket_output_stream_write_async
);
211 count
= g_simple_async_result_get_op_res_gssize (simple
);
217 g_socket_output_stream_pollable_is_writable (GPollableOutputStream
*pollable
)
219 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (pollable
);
221 return g_socket_condition_check (output_stream
->priv
->socket
, G_IO_OUT
);
225 g_socket_output_stream_pollable_create_source (GPollableOutputStream
*pollable
,
226 GCancellable
*cancellable
)
228 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (pollable
);
229 GSource
*socket_source
, *pollable_source
;
231 pollable_source
= g_pollable_source_new (G_OBJECT (output_stream
));
232 socket_source
= g_socket_create_source (output_stream
->priv
->socket
,
233 G_IO_OUT
, cancellable
);
234 g_source_set_dummy_callback (socket_source
);
235 g_source_add_child_source (pollable_source
, socket_source
);
236 g_source_unref (socket_source
);
238 return pollable_source
;
242 g_socket_output_stream_pollable_write_nonblocking (GPollableOutputStream
*pollable
,
247 GSocketOutputStream
*output_stream
= G_SOCKET_OUTPUT_STREAM (pollable
);
249 return g_socket_send_with_blocking (output_stream
->priv
->socket
,
255 g_socket_output_stream_class_init (GSocketOutputStreamClass
*klass
)
257 GObjectClass
*gobject_class
= G_OBJECT_CLASS (klass
);
258 GOutputStreamClass
*goutputstream_class
= G_OUTPUT_STREAM_CLASS (klass
);
260 g_type_class_add_private (klass
, sizeof (GSocketOutputStreamPrivate
));
262 gobject_class
->finalize
= g_socket_output_stream_finalize
;
263 gobject_class
->get_property
= g_socket_output_stream_get_property
;
264 gobject_class
->set_property
= g_socket_output_stream_set_property
;
266 goutputstream_class
->write_fn
= g_socket_output_stream_write
;
267 goutputstream_class
->write_async
= g_socket_output_stream_write_async
;
268 goutputstream_class
->write_finish
= g_socket_output_stream_write_finish
;
270 g_object_class_install_property (gobject_class
, PROP_SOCKET
,
271 g_param_spec_object ("socket",
273 P_("The socket that this stream wraps"),
274 G_TYPE_SOCKET
, G_PARAM_CONSTRUCT_ONLY
|
275 G_PARAM_READWRITE
| G_PARAM_STATIC_STRINGS
));
279 g_socket_output_stream_pollable_iface_init (GPollableOutputStreamInterface
*iface
)
281 iface
->is_writable
= g_socket_output_stream_pollable_is_writable
;
282 iface
->create_source
= g_socket_output_stream_pollable_create_source
;
283 iface
->write_nonblocking
= g_socket_output_stream_pollable_write_nonblocking
;
287 g_socket_output_stream_init (GSocketOutputStream
*stream
)
289 stream
->priv
= G_TYPE_INSTANCE_GET_PRIVATE (stream
, G_TYPE_SOCKET_OUTPUT_STREAM
, GSocketOutputStreamPrivate
);
292 GSocketOutputStream
*
293 _g_socket_output_stream_new (GSocket
*socket
)
295 return G_SOCKET_OUTPUT_STREAM (g_object_new (G_TYPE_SOCKET_OUTPUT_STREAM
, "socket", socket
, NULL
));