nexmon – Blame information for rev 1
?pathlinks?
Rev | Author | Line No. | Line |
---|---|---|---|
1 | office | 1 | /* GIO - GLib Input, Output and Streaming Library |
2 | * |
||
3 | * Copyright (C) 2009 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 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 "gzlibdecompressor.h" |
||
24 | |||
25 | #include <errno.h> |
||
26 | #include <zlib.h> |
||
27 | #include <string.h> |
||
28 | |||
29 | #include "gfileinfo.h" |
||
30 | #include "gioerror.h" |
||
31 | #include "gioenums.h" |
||
32 | #include "gioenumtypes.h" |
||
33 | #include "glibintl.h" |
||
34 | |||
35 | |||
36 | enum { |
||
37 | PROP_0, |
||
38 | PROP_FORMAT, |
||
39 | PROP_FILE_INFO |
||
40 | }; |
||
41 | |||
42 | /** |
||
43 | * SECTION:gzdecompressor |
||
44 | * @short_description: Zlib decompressor |
||
45 | * @include: gio/gio.h |
||
46 | * |
||
47 | * #GZlibDecompressor is an implementation of #GConverter that |
||
48 | * decompresses data compressed with zlib. |
||
49 | */ |
||
50 | |||
51 | static void g_zlib_decompressor_iface_init (GConverterIface *iface); |
||
52 | |||
53 | typedef struct { |
||
54 | gz_header gzheader; |
||
55 | char filename[257]; |
||
56 | GFileInfo *file_info; |
||
57 | } HeaderData; |
||
58 | |||
59 | /** |
||
60 | * GZlibDecompressor: |
||
61 | * |
||
62 | * Zlib decompression |
||
63 | */ |
||
64 | struct _GZlibDecompressor |
||
65 | { |
||
66 | GObject parent_instance; |
||
67 | |||
68 | GZlibCompressorFormat format; |
||
69 | z_stream zstream; |
||
70 | HeaderData *header_data; |
||
71 | }; |
||
72 | |||
73 | static void |
||
74 | g_zlib_decompressor_set_gzheader (GZlibDecompressor *decompressor) |
||
75 | { |
||
76 | /* On win32, these functions were not exported before 1.2.4 */ |
||
77 | #if !defined (G_OS_WIN32) || ZLIB_VERNUM >= 0x1240 |
||
78 | if (decompressor->format != G_ZLIB_COMPRESSOR_FORMAT_GZIP) |
||
79 | return; |
||
80 | |||
81 | if (decompressor->header_data != NULL) |
||
82 | { |
||
83 | if (decompressor->header_data->file_info) |
||
84 | g_object_unref (decompressor->header_data->file_info); |
||
85 | |||
86 | memset (decompressor->header_data, 0, sizeof (HeaderData)); |
||
87 | } |
||
88 | else |
||
89 | { |
||
90 | decompressor->header_data = g_new0 (HeaderData, 1); |
||
91 | } |
||
92 | |||
93 | decompressor->header_data->gzheader.name = (Bytef*) &decompressor->header_data->filename; |
||
94 | /* We keep one byte to guarantee the string is 0-terminated */ |
||
95 | decompressor->header_data->gzheader.name_max = 256; |
||
96 | |||
97 | if (inflateGetHeader (&decompressor->zstream, &decompressor->header_data->gzheader) != Z_OK) |
||
98 | g_warning ("unexpected zlib error: %s\n", decompressor->zstream.msg); |
||
99 | #endif /* !G_OS_WIN32 || ZLIB >= 1.2.4 */ |
||
100 | } |
||
101 | |||
102 | G_DEFINE_TYPE_WITH_CODE (GZlibDecompressor, g_zlib_decompressor, G_TYPE_OBJECT, |
||
103 | G_IMPLEMENT_INTERFACE (G_TYPE_CONVERTER, |
||
104 | g_zlib_decompressor_iface_init)) |
||
105 | |||
106 | static void |
||
107 | g_zlib_decompressor_finalize (GObject *object) |
||
108 | { |
||
109 | GZlibDecompressor *decompressor; |
||
110 | |||
111 | decompressor = G_ZLIB_DECOMPRESSOR (object); |
||
112 | |||
113 | inflateEnd (&decompressor->zstream); |
||
114 | |||
115 | if (decompressor->header_data != NULL) |
||
116 | { |
||
117 | if (decompressor->header_data->file_info) |
||
118 | g_object_unref (decompressor->header_data->file_info); |
||
119 | g_free (decompressor->header_data); |
||
120 | } |
||
121 | |||
122 | G_OBJECT_CLASS (g_zlib_decompressor_parent_class)->finalize (object); |
||
123 | } |
||
124 | |||
125 | |||
126 | static void |
||
127 | g_zlib_decompressor_set_property (GObject *object, |
||
128 | guint prop_id, |
||
129 | const GValue *value, |
||
130 | GParamSpec *pspec) |
||
131 | { |
||
132 | GZlibDecompressor *decompressor; |
||
133 | |||
134 | decompressor = G_ZLIB_DECOMPRESSOR (object); |
||
135 | |||
136 | switch (prop_id) |
||
137 | { |
||
138 | case PROP_FORMAT: |
||
139 | decompressor->format = g_value_get_enum (value); |
||
140 | break; |
||
141 | |||
142 | default: |
||
143 | G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); |
||
144 | break; |
||
145 | } |
||
146 | |||
147 | } |
||
148 | |||
149 | static void |
||
150 | g_zlib_decompressor_get_property (GObject *object, |
||
151 | guint prop_id, |
||
152 | GValue *value, |
||
153 | GParamSpec *pspec) |
||
154 | { |
||
155 | GZlibDecompressor *decompressor; |
||
156 | |||
157 | decompressor = G_ZLIB_DECOMPRESSOR (object); |
||
158 | |||
159 | switch (prop_id) |
||
160 | { |
||
161 | case PROP_FORMAT: |
||
162 | g_value_set_enum (value, decompressor->format); |
||
163 | break; |
||
164 | |||
165 | case PROP_FILE_INFO: |
||
166 | if (decompressor->header_data) |
||
167 | g_value_set_object (value, decompressor->header_data->file_info); |
||
168 | else |
||
169 | g_value_set_object (value, NULL); |
||
170 | break; |
||
171 | |||
172 | default: |
||
173 | G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); |
||
174 | break; |
||
175 | } |
||
176 | } |
||
177 | |||
178 | static void |
||
179 | g_zlib_decompressor_init (GZlibDecompressor *decompressor) |
||
180 | { |
||
181 | } |
||
182 | |||
183 | static void |
||
184 | g_zlib_decompressor_constructed (GObject *object) |
||
185 | { |
||
186 | GZlibDecompressor *decompressor; |
||
187 | int res; |
||
188 | |||
189 | decompressor = G_ZLIB_DECOMPRESSOR (object); |
||
190 | |||
191 | if (decompressor->format == G_ZLIB_COMPRESSOR_FORMAT_GZIP) |
||
192 | { |
||
193 | /* + 16 for gzip */ |
||
194 | res = inflateInit2 (&decompressor->zstream, MAX_WBITS + 16); |
||
195 | } |
||
196 | else if (decompressor->format == G_ZLIB_COMPRESSOR_FORMAT_RAW) |
||
197 | { |
||
198 | /* Negative for raw */ |
||
199 | res = inflateInit2 (&decompressor->zstream, -MAX_WBITS); |
||
200 | } |
||
201 | else /* ZLIB */ |
||
202 | res = inflateInit (&decompressor->zstream); |
||
203 | |||
204 | if (res == Z_MEM_ERROR ) |
||
205 | g_error ("GZlibDecompressor: Not enough memory for zlib use"); |
||
206 | |||
207 | if (res != Z_OK) |
||
208 | g_warning ("unexpected zlib error: %s\n", decompressor->zstream.msg); |
||
209 | |||
210 | g_zlib_decompressor_set_gzheader (decompressor); |
||
211 | } |
||
212 | |||
213 | static void |
||
214 | g_zlib_decompressor_class_init (GZlibDecompressorClass *klass) |
||
215 | { |
||
216 | GObjectClass *gobject_class = G_OBJECT_CLASS (klass); |
||
217 | |||
218 | gobject_class->finalize = g_zlib_decompressor_finalize; |
||
219 | gobject_class->constructed = g_zlib_decompressor_constructed; |
||
220 | gobject_class->get_property = g_zlib_decompressor_get_property; |
||
221 | gobject_class->set_property = g_zlib_decompressor_set_property; |
||
222 | |||
223 | g_object_class_install_property (gobject_class, |
||
224 | PROP_FORMAT, |
||
225 | g_param_spec_enum ("format", |
||
226 | P_("compression format"), |
||
227 | P_("The format of the compressed data"), |
||
228 | G_TYPE_ZLIB_COMPRESSOR_FORMAT, |
||
229 | G_ZLIB_COMPRESSOR_FORMAT_ZLIB, |
||
230 | G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | |
||
231 | G_PARAM_STATIC_STRINGS)); |
||
232 | |||
233 | /** |
||
234 | * GZlibDecompressor:file-info: |
||
235 | * |
||
236 | * A #GFileInfo containing the information found in the GZIP header |
||
237 | * of the data stream processed, or %NULL if the header was not yet |
||
238 | * fully processed, is not present at all, or the compressor's |
||
239 | * #GZlibDecompressor:format property is not %G_ZLIB_COMPRESSOR_FORMAT_GZIP. |
||
240 | * |
||
241 | * Since: 2.26 |
||
242 | */ |
||
243 | g_object_class_install_property (gobject_class, |
||
244 | PROP_FILE_INFO, |
||
245 | g_param_spec_object ("file-info", |
||
246 | P_("file info"), |
||
247 | P_("File info"), |
||
248 | G_TYPE_FILE_INFO, |
||
249 | G_PARAM_READABLE | |
||
250 | G_PARAM_STATIC_STRINGS)); |
||
251 | } |
||
252 | |||
253 | /** |
||
254 | * g_zlib_decompressor_new: |
||
255 | * @format: The format to use for the compressed data |
||
256 | * |
||
257 | * Creates a new #GZlibDecompressor. |
||
258 | * |
||
259 | * Returns: a new #GZlibDecompressor |
||
260 | * |
||
261 | * Since: 2.24 |
||
262 | **/ |
||
263 | GZlibDecompressor * |
||
264 | g_zlib_decompressor_new (GZlibCompressorFormat format) |
||
265 | { |
||
266 | GZlibDecompressor *decompressor; |
||
267 | |||
268 | decompressor = g_object_new (G_TYPE_ZLIB_DECOMPRESSOR, |
||
269 | "format", format, |
||
270 | NULL); |
||
271 | |||
272 | return decompressor; |
||
273 | } |
||
274 | |||
275 | /** |
||
276 | * g_zlib_decompressor_get_file_info: |
||
277 | * @decompressor: a #GZlibDecompressor |
||
278 | * |
||
279 | * Retrieves the #GFileInfo constructed from the GZIP header data |
||
280 | * of compressed data processed by @compressor, or %NULL if @decompressor's |
||
281 | * #GZlibDecompressor:format property is not %G_ZLIB_COMPRESSOR_FORMAT_GZIP, |
||
282 | * or the header data was not fully processed yet, or it not present in the |
||
283 | * data stream at all. |
||
284 | * |
||
285 | * Returns: (transfer none): a #GFileInfo, or %NULL |
||
286 | * |
||
287 | * Since: 2.26 |
||
288 | */ |
||
289 | GFileInfo * |
||
290 | g_zlib_decompressor_get_file_info (GZlibDecompressor *decompressor) |
||
291 | { |
||
292 | g_return_val_if_fail (G_IS_ZLIB_DECOMPRESSOR (decompressor), NULL); |
||
293 | |||
294 | if (decompressor->header_data) |
||
295 | return decompressor->header_data->file_info; |
||
296 | |||
297 | return NULL; |
||
298 | } |
||
299 | |||
300 | static void |
||
301 | g_zlib_decompressor_reset (GConverter *converter) |
||
302 | { |
||
303 | GZlibDecompressor *decompressor = G_ZLIB_DECOMPRESSOR (converter); |
||
304 | int res; |
||
305 | |||
306 | res = inflateReset (&decompressor->zstream); |
||
307 | if (res != Z_OK) |
||
308 | g_warning ("unexpected zlib error: %s\n", decompressor->zstream.msg); |
||
309 | |||
310 | g_zlib_decompressor_set_gzheader (decompressor); |
||
311 | } |
||
312 | |||
313 | static GConverterResult |
||
314 | g_zlib_decompressor_convert (GConverter *converter, |
||
315 | const void *inbuf, |
||
316 | gsize inbuf_size, |
||
317 | void *outbuf, |
||
318 | gsize outbuf_size, |
||
319 | GConverterFlags flags, |
||
320 | gsize *bytes_read, |
||
321 | gsize *bytes_written, |
||
322 | GError **error) |
||
323 | { |
||
324 | GZlibDecompressor *decompressor; |
||
325 | int res; |
||
326 | |||
327 | decompressor = G_ZLIB_DECOMPRESSOR (converter); |
||
328 | |||
329 | decompressor->zstream.next_in = (void *)inbuf; |
||
330 | decompressor->zstream.avail_in = inbuf_size; |
||
331 | |||
332 | decompressor->zstream.next_out = outbuf; |
||
333 | decompressor->zstream.avail_out = outbuf_size; |
||
334 | |||
335 | res = inflate (&decompressor->zstream, Z_NO_FLUSH); |
||
336 | |||
337 | if (res == Z_DATA_ERROR || res == Z_NEED_DICT) |
||
338 | { |
||
339 | g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_INVALID_DATA, |
||
340 | _("Invalid compressed data")); |
||
341 | return G_CONVERTER_ERROR; |
||
342 | } |
||
343 | |||
344 | if (res == Z_MEM_ERROR) |
||
345 | { |
||
346 | g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED, |
||
347 | _("Not enough memory")); |
||
348 | return G_CONVERTER_ERROR; |
||
349 | } |
||
350 | |||
351 | if (res == Z_STREAM_ERROR) |
||
352 | { |
||
353 | g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED, |
||
354 | _("Internal error: %s"), decompressor->zstream.msg); |
||
355 | return G_CONVERTER_ERROR; |
||
356 | } |
||
357 | |||
358 | if (res == Z_BUF_ERROR) |
||
359 | { |
||
360 | if (flags & G_CONVERTER_FLUSH) |
||
361 | return G_CONVERTER_FLUSHED; |
||
362 | |||
363 | /* Z_FINISH not set, so this means no progress could be made */ |
||
364 | /* We do have output space, so this should only happen if we |
||
365 | have no input but need some */ |
||
366 | |||
367 | g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_PARTIAL_INPUT, |
||
368 | _("Need more input")); |
||
369 | return G_CONVERTER_ERROR; |
||
370 | } |
||
371 | |||
372 | g_assert (res == Z_OK || res == Z_STREAM_END); |
||
373 | |||
374 | *bytes_read = inbuf_size - decompressor->zstream.avail_in; |
||
375 | *bytes_written = outbuf_size - decompressor->zstream.avail_out; |
||
376 | |||
377 | #if !defined (G_OS_WIN32) || ZLIB_VERNUM >= 0x1240 |
||
378 | if (decompressor->header_data != NULL && |
||
379 | decompressor->header_data->gzheader.done == 1) |
||
380 | { |
||
381 | HeaderData *data = decompressor->header_data; |
||
382 | |||
383 | /* So we don't notify again */ |
||
384 | data->gzheader.done = 2; |
||
385 | |||
386 | data->file_info = g_file_info_new (); |
||
387 | g_file_info_set_attribute_uint64 (data->file_info, |
||
388 | G_FILE_ATTRIBUTE_TIME_MODIFIED, |
||
389 | data->gzheader.time); |
||
390 | g_file_info_set_attribute_uint32 (data->file_info, |
||
391 | G_FILE_ATTRIBUTE_TIME_MODIFIED_USEC, |
||
392 | 0); |
||
393 | |||
394 | if (data->filename[0] != '\0') |
||
395 | g_file_info_set_attribute_byte_string (data->file_info, |
||
396 | G_FILE_ATTRIBUTE_STANDARD_NAME, |
||
397 | data->filename); |
||
398 | |||
399 | g_object_notify (G_OBJECT (decompressor), "file-info"); |
||
400 | } |
||
401 | #endif /* !G_OS_WIN32 || ZLIB >= 1.2.4 */ |
||
402 | |||
403 | if (res == Z_STREAM_END) |
||
404 | return G_CONVERTER_FINISHED; |
||
405 | return G_CONVERTER_CONVERTED; |
||
406 | } |
||
407 | |||
408 | static void |
||
409 | g_zlib_decompressor_iface_init (GConverterIface *iface) |
||
410 | { |
||
411 | iface->convert = g_zlib_decompressor_convert; |
||
412 | iface->reset = g_zlib_decompressor_reset; |
||
413 | } |