diff options
Diffstat (limited to 'libyelp/yelp-lzma-decompressor.c')
| -rw-r--r-- | libyelp/yelp-lzma-decompressor.c | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/libyelp/yelp-lzma-decompressor.c b/libyelp/yelp-lzma-decompressor.c new file mode 100644 index 00000000..1bd38ef6 --- /dev/null +++ b/libyelp/yelp-lzma-decompressor.c @@ -0,0 +1,174 @@ +/* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ +/* + * Copyright (C) 2009 Red Hat, Inc. + * Copyright (C) 2009 Shaun McCance <shaunm@gnome.org> + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General + * Public License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place, Suite 330, + * Boston, MA 02111-1307, USA. + * + * Author: Shaun McCance <shaunm@gnome.org> + */ + +#include "config.h" + +#include <errno.h> +#include <string.h> + +#include <glib/gi18n.h> + +#include "yelp-lzma-decompressor.h" + +static void yelp_lzma_decompressor_iface_init (GConverterIface *iface); + +struct _YelpLzmaDecompressor +{ + GObject parent_instance; + + lzmadec_stream lzmastream; +}; + +G_DEFINE_TYPE_WITH_CODE (YelpLzmaDecompressor, yelp_lzma_decompressor, G_TYPE_OBJECT, + G_IMPLEMENT_INTERFACE (G_TYPE_CONVERTER, + yelp_lzma_decompressor_iface_init)) + +static void +yelp_lzma_decompressor_finalize (GObject *object) +{ + YelpLzmaDecompressor *decompressor; + + decompressor = YELP_LZMA_DECOMPRESSOR (object); + + lzmadec_end (&decompressor->lzmastream); + + G_OBJECT_CLASS (yelp_lzma_decompressor_parent_class)->finalize (object); +} + +static void +yelp_lzma_decompressor_init (YelpLzmaDecompressor *decompressor) +{ +} + +static void +yelp_lzma_decompressor_constructed (GObject *object) +{ + YelpLzmaDecompressor *decompressor; + int res; + + decompressor = YELP_LZMA_DECOMPRESSOR (object); + + res = lzmadec_init (&decompressor->lzmastream); + + if (res == LZMADEC_MEM_ERROR ) + g_error ("YelpLzmaDecompressor: Not enough memory for lzma use"); + + if (res != LZMADEC_OK) + g_error ("YelpLzmaDecompressor: Unexpected lzma error"); +} + +static void +yelp_lzma_decompressor_class_init (YelpLzmaDecompressorClass *klass) +{ + GObjectClass *gobject_class = G_OBJECT_CLASS (klass); + + gobject_class->finalize = yelp_lzma_decompressor_finalize; + gobject_class->constructed = yelp_lzma_decompressor_constructed; +} + +YelpLzmaDecompressor * +yelp_lzma_decompressor_new (void) +{ + YelpLzmaDecompressor *decompressor; + + decompressor = g_object_new (YELP_TYPE_LZMA_DECOMPRESSOR, NULL); + + return decompressor; +} + +static void +yelp_lzma_decompressor_reset (GConverter *converter) +{ + YelpLzmaDecompressor *decompressor = YELP_LZMA_DECOMPRESSOR (converter); + int res; + + /* lzmadec doesn't have a reset function. Ending and reiniting + * might do the trick. But this is untested. If reset matters + * to you, test this. + */ + lzmadec_end (&decompressor->lzmastream); + res = lzmadec_init (&decompressor->lzmastream); + + if (res == LZMADEC_MEM_ERROR ) + g_error ("YelpLzmaDecompressor: Not enough memory for lzma use"); + + if (res != LZMADEC_OK) + g_error ("YelpLzmaDecompressor: Unexpected lzma error"); +} + +static GConverterResult +yelp_lzma_decompressor_convert (GConverter *converter, + const void *inbuf, + gsize inbuf_size, + void *outbuf, + gsize outbuf_size, + GConverterFlags flags, + gsize *bytes_read, + gsize *bytes_written, + GError **error) +{ + YelpLzmaDecompressor *decompressor; + gsize header_size; + int res; + + decompressor = YELP_LZMA_DECOMPRESSOR (converter); + + decompressor->lzmastream.next_in = (void *)inbuf; + decompressor->lzmastream.avail_in = inbuf_size; + + decompressor->lzmastream.next_out = outbuf; + decompressor->lzmastream.avail_out = outbuf_size; + + res = lzmadec_decode (&decompressor->lzmastream, + flags & G_CONVERTER_INPUT_AT_END); + + if (res == LZMADEC_DATA_ERROR || res == LZMADEC_HEADER_ERROR) { + g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_INVALID_DATA, + _("Invalid compressed data")); + return G_CONVERTER_ERROR; + } + + if (res == LZMADEC_MEM_ERROR) { + g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED, + _("Not enough memory")); + return G_CONVERTER_ERROR; + } + + if (res == LZMADEC_OK || res == LZMADEC_STREAM_END) { + *bytes_read = inbuf_size - decompressor->lzmastream.avail_in; + *bytes_written = outbuf_size - decompressor->lzmastream.avail_out; + + if (res == LZMADEC_STREAM_END) + return G_CONVERTER_FINISHED; + return G_CONVERTER_CONVERTED; + } + + g_assert_not_reached (); +} + +static void +yelp_lzma_decompressor_iface_init (GConverterIface *iface) +{ + iface->convert = yelp_lzma_decompressor_convert; + iface->reset = yelp_lzma_decompressor_reset; +} |