From 230745a350fea524f899c6319a7c529fd82bd377 Mon Sep 17 00:00:00 2001 From: Alexander Larsson Date: Wed, 18 Nov 2009 16:07:16 +0100 Subject: [PATCH] Add filter-cat test for GConverter streams --- gio/tests/Makefile.am | 4 + gio/tests/filter-cat.c | 227 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 231 insertions(+) create mode 100644 gio/tests/filter-cat.c diff --git a/gio/tests/Makefile.am b/gio/tests/Makefile.am index 89fcde05f..62b4b0dc9 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 000000000..0c1cc1364 --- /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 + */ + +#include + +#include +#include +#include +#include + +#include +#include + +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; +}