[glib] Add filter-cat test for GConverter streams

[Alexander Larsson <alexl src gnome org>]

commit 230745a350fea524f899c6319a7c529fd82bd377
Author: Alexander Larsson <alexl redhat com>
Date:   Wed Nov 18 16:07:16 2009 +0100

    Add filter-cat test for GConverter streams

 gio/tests/Makefile.am  |    4 +
 gio/tests/filter-cat.c |  227 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 231 insertions(+), 0 deletions(-)
---
diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am
index 89fcde0..62b4b0d 100644
--- a/gio/tests/Makefile.am
+++ b/gio/tests/Makefile.am
@@ -26,6 +26,7 @@ TEST_PROGS +=	 		\
 	converter-stream	\
 	data-input-stream 	\
 	data-output-stream 	\
+	filter-cat		\
 	g-icon			\
 	buffered-input-stream	\
 	sleepy-stream		\
@@ -64,6 +65,9 @@ data_input_stream_LDADD		= $(progs_ldadd)
 data_output_stream_SOURCES	= data-output-stream.c
 data_output_stream_LDADD	= $(progs_ldadd)
 
+filter_cat_SOURCES	= filter-cat.c
+filter_cat_LDADD	= $(progs_ldadd)
+
 g_icon_SOURCES	= g-icon.c
 g_icon_LDADD	= $(progs_ldadd)
 
diff --git a/gio/tests/filter-cat.c b/gio/tests/filter-cat.c
new file mode 100644
index 0000000..0c1cc13
--- /dev/null
+++ b/gio/tests/filter-cat.c
@@ -0,0 +1,227 @@
+/* GIO - GLib Input, Output and Streaming Library
+ *
+ * Copyright (C) 2009 Red Hat, Inc.
+ *
+ * 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: Alexander Larsson <alexl redhat com>
+ */
+
+#include <config.h>
+
+#include <stdio.h>
+#include <unistd.h>
+#include <locale.h>
+#include <errno.h>
+
+#include <glib.h>
+#include <gio/gio.h>
+
+static gchar **locations = NULL;
+static char *from_charset = NULL;
+static char *to_charset = NULL;
+static gboolean decompress = FALSE;
+static gboolean compress = FALSE;
+static gboolean gzip = FALSE;
+static gboolean fallback = FALSE;
+
+static GOptionEntry entries[] = {
+  {"decompress", 0, 0, G_OPTION_ARG_NONE, &decompress, "decompress", NULL},
+  {"compress", 0, 0, G_OPTION_ARG_NONE, &compress, "compress", NULL},
+  {"gzip", 0, 0, G_OPTION_ARG_NONE, &gzip, "use gzip format", NULL},
+  {"from-charset", 0, 0, G_OPTION_ARG_STRING, &from_charset, "from charset", NULL},
+  {"to-charset", 0, 0, G_OPTION_ARG_STRING, &to_charset, "to charset", NULL},
+  {"fallback", 0, 0, G_OPTION_ARG_NONE, &fallback, "use fallback", NULL},
+  {G_OPTION_REMAINING, 0, 0, G_OPTION_ARG_FILENAME_ARRAY, &locations, "locations", NULL},
+  {NULL}
+};
+
+static void
+cat (GFile * file)
+{
+  GInputStream *in;
+  char buffer[1024 * 8 + 1];
+  char *p;
+  gssize res;
+  gboolean close_res;
+  GError *error;
+  GConverter *conv;
+  GCharsetConverter *cconv = NULL;
+
+  error = NULL;
+  in = (GInputStream *) g_file_read (file, NULL, &error);
+  if (in == NULL)
+    {
+      /* Translators: the first %s is the program name, the second one  */
+      /* is the URI of the file, the third is the error message.        */
+      g_printerr ("%s: %s: error opening file: %s\n",
+		  g_get_prgname (), g_file_get_uri (file), error->message);
+      g_error_free (error);
+      return;
+    }
+
+  if (decompress)
+    {
+      GInputStream *old;
+      conv = (GConverter *)g_zlib_decompressor_new (gzip?G_ZLIB_COMPRESSOR_FORMAT_GZIP:G_ZLIB_COMPRESSOR_FORMAT_RAW);
+      old = in;
+      in = (GInputStream *) g_converter_input_stream_new (in, conv);
+      g_object_unref (conv);
+      g_object_unref (old);
+    }
+
+  if (from_charset && to_charset)
+    {
+      cconv = g_charset_converter_new (to_charset, from_charset, &error);
+      conv = (GConverter *)cconv;
+      if (conv)
+	{
+	  GInputStream *old;
+
+	  g_charset_converter_set_use_fallback (cconv, fallback);
+
+	  old = in;
+	  in = (GInputStream *) g_converter_input_stream_new (in, conv);
+	  g_object_unref (conv);
+	  g_object_unref (old);
+	}
+      else
+	{
+	  g_printerr ("%s: Can't convert between charsets: %s\n",
+		      g_get_prgname (), error->message);
+	}
+    }
+
+  if (compress)
+    {
+      GInputStream *old;
+      conv = (GConverter *)g_zlib_compressor_new (gzip?G_ZLIB_COMPRESSOR_FORMAT_GZIP:G_ZLIB_COMPRESSOR_FORMAT_RAW, -1);
+      old = in;
+      in = (GInputStream *) g_converter_input_stream_new (in, conv);
+      g_object_unref (conv);
+      g_object_unref (old);
+    }
+
+  while (1)
+    {
+      res =
+	g_input_stream_read (in, buffer, sizeof (buffer) - 1, NULL, &error);
+      if (res > 0)
+	{
+	  ssize_t written;
+
+	  p = buffer;
+	  while (res > 0)
+	    {
+	      written = write (STDOUT_FILENO, p, res);
+
+	      if (written == -1 && errno != EINTR)
+		{
+		  /* Translators: the first %s is the program name, the */
+		  /* second one is the URI of the file.                 */
+		  g_printerr ("%s: %s, error writing to stdout",
+			      g_get_prgname (), g_file_get_uri (file));
+		  goto out;
+		}
+	      res -= written;
+	      p += written;
+	    }
+	}
+      else if (res < 0)
+	{
+	  g_printerr ("%s: %s: error reading: %s\n",
+		      g_get_prgname (), g_file_get_uri (file),
+		      error->message);
+	  g_error_free (error);
+	  error = NULL;
+	  break;
+	}
+      else if (res == 0)
+	break;
+    }
+
+ out:
+
+  close_res = g_input_stream_close (in, NULL, &error);
+  if (!close_res)
+    {
+      g_printerr ("%s: %s:error closing: %s\n",
+		  g_get_prgname (), g_file_get_uri (file), error->message);
+      g_error_free (error);
+    }
+
+  if (cconv != NULL && fallback)
+    {
+      guint num = g_charset_converter_get_num_fallbacks (cconv);
+      if (num > 0)
+	g_printerr ("Number of fallback errors: %u\n", num);
+    }
+}
+
+int
+main (int argc, char *argv[])
+{
+  GError *error = NULL;
+  GOptionContext *context = NULL;
+  GFile *file;
+  int i;
+
+  g_type_init ();
+
+  context =
+    g_option_context_new ("LOCATION... - concatenate LOCATIONS "
+			  "to standard output.");
+
+  g_option_context_set_summary (context, "filter files");
+
+  g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE);
+  g_option_context_parse (context, &argc, &argv, &error);
+
+  g_option_context_free (context);
+
+  if (error != NULL)
+    {
+      g_printerr ("Error parsing commandline options: %s\n", error->message);
+      g_printerr ("\n");
+      g_printerr ("Try \"%s --help\" for more information.",
+		  g_get_prgname ());
+      g_printerr ("\n");
+      g_error_free(error);
+      return 1;
+    }
+
+  if (!locations)
+    {
+      g_printerr ("%s: missing locations", g_get_prgname ());
+      g_printerr ("\n");
+      g_printerr ("Try \"%s --help\" for more information.",
+		  g_get_prgname ());
+      g_printerr ("\n");
+      return 1;
+    }
+
+  i = 0;
+
+  do
+    {
+      file = g_file_new_for_commandline_arg (locations[i]);
+      cat (file);
+      g_object_unref (file);
+    }
+  while (locations[++i] != NULL);
+
+  return 0;
+}