1/* GIO - GLib Input, Output and Streaming Library
2 *
3 * Copyright (C) 2006-2007 Red Hat, Inc.
4 *
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.1 of the License, or (at your option) any later version.
9 *
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.
14 *
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
17 *
18 * Author: Alexander Larsson <alexl@redhat.com>
19 */
20
21#include "config.h"
22
23#include <sys/types.h>
24#include <sys/stat.h>
25#include <fcntl.h>
26#include <errno.h>
27
28#include <glib.h>
29#include <glib/gstdio.h>
30#include "gcancellable.h"
31#include "gioerror.h"
32#include "glocalfileinputstream.h"
33#include "glocalfileinfo.h"
34#include "glibintl.h"
35
36#ifdef G_OS_UNIX
37#include <unistd.h>
38#include "glib-unix.h"
39#include "gfiledescriptorbased.h"
40#endif
41
42#ifdef G_OS_WIN32
43#include <io.h>
44#endif
45
46struct _GLocalFileInputStreamPrivate {
47 int fd;
48 guint do_close : 1;
49};
50
51#ifdef G_OS_UNIX
52static void g_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface);
53#endif
54
55#define g_local_file_input_stream_get_type _g_local_file_input_stream_get_type
56#ifdef G_OS_UNIX
57G_DEFINE_TYPE_WITH_CODE (GLocalFileInputStream, g_local_file_input_stream, G_TYPE_FILE_INPUT_STREAM,
58 G_ADD_PRIVATE (GLocalFileInputStream)
59 G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED,
60 g_file_descriptor_based_iface_init))
61#else
62G_DEFINE_TYPE_WITH_CODE (GLocalFileInputStream, g_local_file_input_stream, G_TYPE_FILE_INPUT_STREAM,
63 G_ADD_PRIVATE (GLocalFileInputStream))
64#endif
65
66static gssize g_local_file_input_stream_read (GInputStream *stream,
67 void *buffer,
68 gsize count,
69 GCancellable *cancellable,
70 GError **error);
71static gboolean g_local_file_input_stream_close (GInputStream *stream,
72 GCancellable *cancellable,
73 GError **error);
74static goffset g_local_file_input_stream_tell (GFileInputStream *stream);
75static gboolean g_local_file_input_stream_can_seek (GFileInputStream *stream);
76static gboolean g_local_file_input_stream_seek (GFileInputStream *stream,
77 goffset offset,
78 GSeekType type,
79 GCancellable *cancellable,
80 GError **error);
81static GFileInfo *g_local_file_input_stream_query_info (GFileInputStream *stream,
82 const char *attributes,
83 GCancellable *cancellable,
84 GError **error);
85#ifdef G_OS_UNIX
86static int g_local_file_input_stream_get_fd (GFileDescriptorBased *stream);
87#endif
88
89void
90_g_local_file_input_stream_set_do_close (GLocalFileInputStream *in,
91 gboolean do_close)
92{
93 in->priv->do_close = do_close;
94}
95
96static void
97g_local_file_input_stream_class_init (GLocalFileInputStreamClass *klass)
98{
99 GInputStreamClass *stream_class = G_INPUT_STREAM_CLASS (klass);
100 GFileInputStreamClass *file_stream_class = G_FILE_INPUT_STREAM_CLASS (klass);
101
102 stream_class->read_fn = g_local_file_input_stream_read;
103 stream_class->close_fn = g_local_file_input_stream_close;
104 file_stream_class->tell = g_local_file_input_stream_tell;
105 file_stream_class->can_seek = g_local_file_input_stream_can_seek;
106 file_stream_class->seek = g_local_file_input_stream_seek;
107 file_stream_class->query_info = g_local_file_input_stream_query_info;
108}
109
110#ifdef G_OS_UNIX
111static void
112g_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface)
113{
114 iface->get_fd = g_local_file_input_stream_get_fd;
115}
116#endif
117
118static void
119g_local_file_input_stream_init (GLocalFileInputStream *info)
120{
121 info->priv = g_local_file_input_stream_get_instance_private (self: info);
122 info->priv->do_close = TRUE;
123}
124
125GFileInputStream *
126_g_local_file_input_stream_new (int fd)
127{
128 GLocalFileInputStream *stream;
129
130 stream = g_object_new (G_TYPE_LOCAL_FILE_INPUT_STREAM, NULL);
131 stream->priv->fd = fd;
132
133 return G_FILE_INPUT_STREAM (stream);
134}
135
136static gssize
137g_local_file_input_stream_read (GInputStream *stream,
138 void *buffer,
139 gsize count,
140 GCancellable *cancellable,
141 GError **error)
142{
143 GLocalFileInputStream *file;
144 gssize res;
145
146 file = G_LOCAL_FILE_INPUT_STREAM (stream);
147
148 res = -1;
149 while (1)
150 {
151 if (g_cancellable_set_error_if_cancelled (cancellable, error))
152 break;
153 res = read (fd: file->priv->fd, buf: buffer, nbytes: count);
154 if (res == -1)
155 {
156 int errsv = errno;
157
158 if (errsv == EINTR)
159 continue;
160
161 g_set_error (err: error, G_IO_ERROR,
162 code: g_io_error_from_errno (err_no: errsv),
163 _("Error reading from file: %s"),
164 g_strerror (errnum: errsv));
165 }
166
167 break;
168 }
169
170 return res;
171}
172
173static gboolean
174g_local_file_input_stream_close (GInputStream *stream,
175 GCancellable *cancellable,
176 GError **error)
177{
178 GLocalFileInputStream *file;
179
180 file = G_LOCAL_FILE_INPUT_STREAM (stream);
181
182 if (!file->priv->do_close)
183 return TRUE;
184
185 if (file->priv->fd == -1)
186 return TRUE;
187
188 if (!g_close (fd: file->priv->fd, NULL))
189 {
190 int errsv = errno;
191
192 g_set_error (err: error, G_IO_ERROR,
193 code: g_io_error_from_errno (err_no: errsv),
194 _("Error closing file: %s"),
195 g_strerror (errnum: errsv));
196 return FALSE;
197 }
198
199 return TRUE;
200}
201
202
203static goffset
204g_local_file_input_stream_tell (GFileInputStream *stream)
205{
206 GLocalFileInputStream *file;
207 off_t pos;
208
209 file = G_LOCAL_FILE_INPUT_STREAM (stream);
210
211 pos = lseek (fd: file->priv->fd, offset: 0, SEEK_CUR);
212
213 if (pos == (off_t)-1)
214 return 0;
215
216 return pos;
217}
218
219static gboolean
220g_local_file_input_stream_can_seek (GFileInputStream *stream)
221{
222 GLocalFileInputStream *file;
223 off_t pos;
224
225 file = G_LOCAL_FILE_INPUT_STREAM (stream);
226
227 pos = lseek (fd: file->priv->fd, offset: 0, SEEK_CUR);
228
229 if (pos == (off_t)-1 && errno == ESPIPE)
230 return FALSE;
231
232 return TRUE;
233}
234
235static int
236seek_type_to_lseek (GSeekType type)
237{
238 switch (type)
239 {
240 default:
241 case G_SEEK_CUR:
242 return SEEK_CUR;
243
244 case G_SEEK_SET:
245 return SEEK_SET;
246
247 case G_SEEK_END:
248 return SEEK_END;
249 }
250}
251
252static gboolean
253g_local_file_input_stream_seek (GFileInputStream *stream,
254 goffset offset,
255 GSeekType type,
256 GCancellable *cancellable,
257 GError **error)
258{
259 GLocalFileInputStream *file;
260 off_t pos;
261
262 file = G_LOCAL_FILE_INPUT_STREAM (stream);
263
264 pos = lseek (fd: file->priv->fd, offset: offset, whence: seek_type_to_lseek (type));
265
266 if (pos == (off_t)-1)
267 {
268 int errsv = errno;
269
270 g_set_error (err: error, G_IO_ERROR,
271 code: g_io_error_from_errno (err_no: errsv),
272 _("Error seeking in file: %s"),
273 g_strerror (errnum: errsv));
274 return FALSE;
275 }
276
277 return TRUE;
278}
279
280static GFileInfo *
281g_local_file_input_stream_query_info (GFileInputStream *stream,
282 const char *attributes,
283 GCancellable *cancellable,
284 GError **error)
285{
286 GLocalFileInputStream *file;
287
288 file = G_LOCAL_FILE_INPUT_STREAM (stream);
289
290 if (g_cancellable_set_error_if_cancelled (cancellable, error))
291 return NULL;
292
293 return _g_local_file_info_get_from_fd (fd: file->priv->fd,
294 attributes,
295 error);
296}
297
298#ifdef G_OS_UNIX
299static int
300g_local_file_input_stream_get_fd (GFileDescriptorBased *fd_based)
301{
302 GLocalFileInputStream *stream = G_LOCAL_FILE_INPUT_STREAM (fd_based);
303 return stream->priv->fd;
304}
305#endif
306

source code of gtk/subprojects/glib/gio/glocalfileinputstream.c