[libyelp] Adding GIO decompressors for bz2, lzma, and magic

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;
+}