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 <gio/gsimpleasyncresult.h>
31 #include <gio/gcancellable.h>
32 #include <gio/gioerror.h>
35 #define g_socket_input_stream_get_type _g_socket_input_stream_get_type
36 G_DEFINE_TYPE (GSocketInputStream
, g_socket_input_stream
, G_TYPE_INPUT_STREAM
);
44 struct _GSocketInputStreamPrivate
48 /* pending operation metadata */
49 GSimpleAsyncResult
*result
;
50 GCancellable
*cancellable
;
56 g_socket_input_stream_get_property (GObject
*object
,
61 GSocketInputStream
*stream
= G_SOCKET_INPUT_STREAM (object
);
66 g_value_set_object (value
, stream
->priv
->socket
);
70 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
75 g_socket_input_stream_set_property (GObject
*object
,
80 GSocketInputStream
*stream
= G_SOCKET_INPUT_STREAM (object
);
85 stream
->priv
->socket
= g_value_dup_object (value
);
89 G_OBJECT_WARN_INVALID_PROPERTY_ID (object
, prop_id
, pspec
);
94 g_socket_input_stream_finalize (GObject
*object
)
96 GSocketInputStream
*stream
= G_SOCKET_INPUT_STREAM (object
);
98 if (stream
->priv
->socket
)
99 g_object_unref (stream
->priv
->socket
);
101 if (G_OBJECT_CLASS (g_socket_input_stream_parent_class
)->finalize
)
102 (*G_OBJECT_CLASS (g_socket_input_stream_parent_class
)->finalize
) (object
);
106 g_socket_input_stream_read (GInputStream
*stream
,
109 GCancellable
*cancellable
,
112 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (stream
);
114 return g_socket_receive (input_stream
->priv
->socket
, buffer
, count
,
119 g_socket_input_stream_read_ready (GSocket
*socket
,
120 GIOCondition condition
,
121 GSocketInputStream
*stream
)
123 GSimpleAsyncResult
*simple
;
124 GError
*error
= NULL
;
127 result
= g_socket_receive (stream
->priv
->socket
,
128 stream
->priv
->buffer
,
130 stream
->priv
->cancellable
,
133 if (g_error_matches (error
, G_IO_ERROR
, G_IO_ERROR_WOULD_BLOCK
))
136 simple
= stream
->priv
->result
;
137 stream
->priv
->result
= NULL
;
140 g_simple_async_result_set_op_res_gssize (simple
, result
);
144 g_simple_async_result_set_from_error (simple
, error
);
145 g_error_free (error
);
148 if (stream
->priv
->cancellable
)
149 g_object_unref (stream
->priv
->cancellable
);
151 g_simple_async_result_complete (simple
);
152 g_object_unref (simple
);
158 g_socket_input_stream_read_async (GInputStream
*stream
,
162 GCancellable
*cancellable
,
163 GAsyncReadyCallback callback
,
166 GSocketInputStream
*input_stream
= G_SOCKET_INPUT_STREAM (stream
);
169 g_assert (input_stream
->priv
->result
== NULL
);
171 input_stream
->priv
->result
=
172 g_simple_async_result_new (G_OBJECT (stream
), callback
, user_data
,
173 g_socket_input_stream_read_async
);
175 g_object_ref (cancellable
);
176 input_stream
->priv
->cancellable
= cancellable
;
177 input_stream
->priv
->buffer
= buffer
;
178 input_stream
->priv
->count
= count
;
180 source
= g_socket_create_source (input_stream
->priv
->socket
,
181 G_IO_IN
| G_IO_HUP
| G_IO_ERR
,
183 g_source_set_callback (source
,
184 (GSourceFunc
) g_socket_input_stream_read_ready
,
185 g_object_ref (input_stream
), g_object_unref
);
186 g_source_attach (source
, g_main_context_get_thread_default ());
187 g_source_unref (source
);
191 g_socket_input_stream_read_finish (GInputStream
*stream
,
192 GAsyncResult
*result
,
195 GSimpleAsyncResult
*simple
;
198 g_return_val_if_fail (G_IS_SOCKET_INPUT_STREAM (stream
), -1);
200 simple
= G_SIMPLE_ASYNC_RESULT (result
);
202 g_warn_if_fail (g_simple_async_result_get_source_tag (simple
) == g_socket_input_stream_read_async
);
204 count
= g_simple_async_result_get_op_res_gssize (simple
);
210 g_socket_input_stream_class_init (GSocketInputStreamClass
*klass
)
212 GObjectClass
*gobject_class
= G_OBJECT_CLASS (klass
);
213 GInputStreamClass
*ginputstream_class
= G_INPUT_STREAM_CLASS (klass
);
215 g_type_class_add_private (klass
, sizeof (GSocketInputStreamPrivate
));
217 gobject_class
->finalize
= g_socket_input_stream_finalize
;
218 gobject_class
->get_property
= g_socket_input_stream_get_property
;
219 gobject_class
->set_property
= g_socket_input_stream_set_property
;
221 ginputstream_class
->read_fn
= g_socket_input_stream_read
;
222 ginputstream_class
->read_async
= g_socket_input_stream_read_async
;
223 ginputstream_class
->read_finish
= g_socket_input_stream_read_finish
;
225 g_object_class_install_property (gobject_class
, PROP_SOCKET
,
226 g_param_spec_object ("socket",
228 P_("The socket that this stream wraps"),
229 G_TYPE_SOCKET
, G_PARAM_CONSTRUCT_ONLY
|
230 G_PARAM_READWRITE
| G_PARAM_STATIC_STRINGS
));
234 g_socket_input_stream_init (GSocketInputStream
*stream
)
236 stream
->priv
= G_TYPE_INSTANCE_GET_PRIVATE (stream
, G_TYPE_SOCKET_INPUT_STREAM
, GSocketInputStreamPrivate
);
240 _g_socket_input_stream_new (GSocket
*socket
)
242 return G_SOCKET_INPUT_STREAM (g_object_new (G_TYPE_SOCKET_INPUT_STREAM
, "socket", socket
, NULL
));