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 "gsocketinputstream.h"
30 #include "gsimpleasyncresult.h"
31 #include "gcancellable.h"
32 #include "gpollableinputstream.h"
34 #include "gfiledescriptorbased.h"
36 static void g_socket_input_stream_pollable_iface_init (GPollableInputStreamInterface
*iface
);
38 static void g_socket_input_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface
*iface
);
41 #define g_socket_input_stream_get_type _g_socket_input_stream_get_type
44 G_DEFINE_TYPE_WITH_CODE (GSocketInputStream
, g_socket_input_stream
, G_TYPE_INPUT_STREAM
,
45 G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_INPUT_STREAM
, g_socket_input_stream_pollable_iface_init
)
46 G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED
, g_socket_input_stream_file_descriptor_based_iface_init
)
49 G_DEFINE_TYPE_WITH_CODE (GSocketInputStream
, g_socket_input_stream
, G_TYPE_INPUT_STREAM
,
50 G_IMPLEMENT_INTERFACE (G_TYPE_POLLABLE_INPUT_STREAM
, g_socket_input_stream_pollable_iface_init
)
60 struct _GSocketInputStreamPrivate
64 /* pending operation metadata */
65 GSimpleAsyncResult
*result
;
66 GCancellable
*cancellable
;
72 g_socket_input_stream_get_property (GObject
*object
,
77 GSocketInputStream
*stream
= G_SOCKET_INPUT_STREAM (object
);
82 g_value_set_object (value
, stream
->priv
->socket
);
86 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
91 g_socket_input_stream_set_property (GObject
*object
,
96 GSocketInputStream
*stream
= G_SOCKET_INPUT_STREAM (object
);
101 stream
->priv
->socket
= g_value_dup_object (value
);
105 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
110 g_socket_input_stream_finalize (GObject
*object
)
112 GSocketInputStream
*stream
= G_SOCKET_INPUT_STREAM (object
);
114 if (stream
->priv
->socket
)
115 g_object_unref (stream
->priv
->socket
);
117 if (G_OBJECT_CLASS (g_socket_input_stream_parent_class
)->finalize
)
118 (*G_OBJECT_CLASS (g_socket_input_stream_parent_class
)->finalize
) (object
);
122 g_socket_input_stream_read (GInputStream
*stream
,
125 GCancellable
*cancellable
,
128 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (stream
);
130 return g_socket_receive_with_blocking (input_stream
->priv
->socket
,
136 g_socket_input_stream_read_ready (GSocket
*socket
,
137 GIOCondition condition
,
138 GSocketInputStream
*stream
)
140 GSimpleAsyncResult
*simple
;
141 GError
*error
= NULL
;
144 result
= g_socket_receive_with_blocking (stream
->priv
->socket
,
145 stream
->priv
->buffer
,
148 stream
->priv
->cancellable
,
151 if (g_error_matches (error
, G_IO_ERROR
, G_IO_ERROR_WOULD_BLOCK
))
154 simple
= stream
->priv
->result
;
155 stream
->priv
->result
= NULL
;
158 g_simple_async_result_set_op_res_gssize (simple
, result
);
161 g_simple_async_result_take_error (simple
, error
);
163 if (stream
->priv
->cancellable
)
164 g_object_unref (stream
->priv
->cancellable
);
166 g_simple_async_result_complete (simple
);
167 g_object_unref (simple
);
173 g_socket_input_stream_read_async (GInputStream
*stream
,
177 GCancellable
*cancellable
,
178 GAsyncReadyCallback callback
,
181 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (stream
);
184 g_assert (input_stream
->priv
->result
== NULL
);
186 input_stream
->priv
->result
=
187 g_simple_async_result_new (G_OBJECT (stream
), callback
, user_data
,
188 g_socket_input_stream_read_async
);
190 g_object_ref (cancellable
);
191 input_stream
->priv
->cancellable
= cancellable
;
192 input_stream
->priv
->buffer
= buffer
;
193 input_stream
->priv
->count
= count
;
195 source
= g_socket_create_source (input_stream
->priv
->socket
,
196 G_IO_IN
| G_IO_HUP
| G_IO_ERR
,
198 g_source_set_callback (source
,
199 (GSourceFunc
) g_socket_input_stream_read_ready
,
200 g_object_ref (input_stream
), g_object_unref
);
201 g_source_attach (source
, g_main_context_get_thread_default ());
202 g_source_unref (source
);
206 g_socket_input_stream_read_finish (GInputStream
*stream
,
207 GAsyncResult
*result
,
210 GSimpleAsyncResult
*simple
;
213 g_return_val_if_fail (G_IS_SOCKET_INPUT_STREAM (stream
), -1);
215 simple
= G_SIMPLE_ASYNC_RESULT (result
);
217 g_warn_if_fail (g_simple_async_result_get_source_tag (simple
) == g_socket_input_stream_read_async
);
219 count
= g_simple_async_result_get_op_res_gssize (simple
);
225 g_socket_input_stream_pollable_is_readable (GPollableInputStream
*pollable
)
227 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (pollable
);
229 return g_socket_condition_check (input_stream
->priv
->socket
, G_IO_IN
);
233 g_socket_input_stream_pollable_create_source (GPollableInputStream
*pollable
,
234 GCancellable
*cancellable
)
236 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (pollable
);
237 GSource
*socket_source
, *pollable_source
;
239 pollable_source
= g_pollable_source_new (G_OBJECT (input_stream
));
240 socket_source
= g_socket_create_source (input_stream
->priv
->socket
,
241 G_IO_IN
, cancellable
);
242 g_source_set_dummy_callback (socket_source
);
243 g_source_add_child_source (pollable_source
, socket_source
);
244 g_source_unref (socket_source
);
246 return pollable_source
;
250 g_socket_input_stream_pollable_read_nonblocking (GPollableInputStream
*pollable
,
255 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (pollable
);
257 return g_socket_receive_with_blocking (input_stream
->priv
->socket
,
264 g_socket_input_stream_get_fd (GFileDescriptorBased
*fd_based
)
266 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (fd_based
);
268 return g_socket_get_fd (input_stream
->priv
->socket
);
273 g_socket_input_stream_class_init (GSocketInputStreamClass
*klass
)
275 GObjectClass
*gobject_class
= G_OBJECT_CLASS (klass
);
276 GInputStreamClass
*ginputstream_class
= G_INPUT_STREAM_CLASS (klass
);
278 g_type_class_add_private (klass
, sizeof (GSocketInputStreamPrivate
));
280 gobject_class
->finalize
= g_socket_input_stream_finalize
;
281 gobject_class
->get_property
= g_socket_input_stream_get_property
;
282 gobject_class
->set_property
= g_socket_input_stream_set_property
;
284 ginputstream_class
->read_fn
= g_socket_input_stream_read
;
285 ginputstream_class
->read_async
= g_socket_input_stream_read_async
;
286 ginputstream_class
->read_finish
= g_socket_input_stream_read_finish
;
288 g_object_class_install_property (gobject_class
, PROP_SOCKET
,
289 g_param_spec_object ("socket",
291 P_("The socket that this stream wraps"),
292 G_TYPE_SOCKET
, G_PARAM_CONSTRUCT_ONLY
|
293 G_PARAM_READWRITE
| G_PARAM_STATIC_STRINGS
));
298 g_socket_input_stream_file_descriptor_based_iface_init (GFileDescriptorBasedIface
*iface
)
300 iface
->get_fd
= g_socket_input_stream_get_fd
;
305 g_socket_input_stream_pollable_iface_init (GPollableInputStreamInterface
*iface
)
307 iface
->is_readable
= g_socket_input_stream_pollable_is_readable
;
308 iface
->create_source
= g_socket_input_stream_pollable_create_source
;
309 iface
->read_nonblocking
= g_socket_input_stream_pollable_read_nonblocking
;
313 g_socket_input_stream_init (GSocketInputStream
*stream
)
315 stream
->priv
= G_TYPE_INSTANCE_GET_PRIVATE (stream
, G_TYPE_SOCKET_INPUT_STREAM
, GSocketInputStreamPrivate
);
319 _g_socket_input_stream_new (GSocket
*socket
)
321 return G_SOCKET_INPUT_STREAM (g_object_new (G_TYPE_SOCKET_INPUT_STREAM
, "socket", socket
, NULL
));