tests/iochannel-test.c

   1 #undef G_DISABLE_ASSERT
   2 #undef G_LOG_DOMAIN
   3
   4 #include <sys/types.h>
   5 #include <sys/stat.h>
   6 #include <fcntl.h>
   7 #include <glib.h>
   8 #include <string.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11
  12 #define BUFFER_SIZE 1024
  13
  14 static void
  15 test_small_writes (void)
  16 {
  17   GIOChannel *io;
  18   GIOStatus status;
  19   guint cnt;
  20   gchar tmp;
  21   GError *error = NULL;
  22
  23   io = g_io_channel_new_file ("iochannel-test-outfile", "w", &error);
  24   if (error)
  25     {
  26       g_warning ("Unable to open file %s: %s",
  27                  "iochannel-test-outfile",
  28                  error->message);
  29       g_error_free (error);
  30
  31       exit (1);
  32     }
  33
  34   g_io_channel_set_encoding (io, NULL, NULL);
  35   g_io_channel_set_buffer_size (io, 1022);
  36
  37   cnt = 2 * g_io_channel_get_buffer_size (io);
  38   tmp = 0;
  39
  40   while (cnt)
  41     {
  42       status = g_io_channel_write_chars (io, &tmp, 1, NULL, NULL);
  43       if (status == G_IO_STATUS_ERROR)
  44         break;
  45       if (status == G_IO_STATUS_NORMAL)
  46         cnt--;
  47     }
  48
  49   g_assert (status == G_IO_STATUS_NORMAL);
  50
  51   g_io_channel_unref (io);
  52 }
  53
  54
  55 gint main (gint argc, gchar * argv[])
  56 {
  57     GIOChannel *gio_r, *gio_w ;
  58     GError *gerr = NULL;
  59     GString *buffer;
  60     char *filename;
  61     char *srcdir = getenv ("srcdir");
  62     gint rlength = 0;
  63     glong wlength = 0;
  64     gsize length_out;
  65     const gchar encoding[] = "EUC-JP";
  66     GIOStatus status;
  67     GIOFlags flags;
  68
  69     if (!srcdir)
  70       srcdir = ".";
  71     filename = g_strconcat (srcdir, G_DIR_SEPARATOR_S, "iochannel-test-infile", NULL);
  72
  73     setbuf (stdout, NULL); /* For debugging */
  74
  75     gio_r = g_io_channel_new_file (filename, "r", &gerr);
  76     if (gerr)
  77       {
  78         g_warning ("Unable to open file %s: %s", filename, gerr->message);
  79         g_error_free (gerr);
  80         return 1;
  81       }
  82     gio_w = g_io_channel_new_file ("iochannel-test-outfile", "w", &gerr);
  83     if (gerr)
  84       {
  85         g_warning ("Unable to open file %s: %s", "iochannel-test-outfile", gerr->message);
  86         g_error_free (gerr);
  87         return 1;
  88       }
  89
  90     g_io_channel_set_encoding (gio_r, encoding, &gerr);
  91     if (gerr)
  92       {
  93         g_warning (gerr->message);
  94         /* Keep going if this is just a case of iconv not supporting EUC-JP, see bug 428048 */
  95         if (gerr->code != G_CONVERT_ERROR_NO_CONVERSION)
  96           return 1;
  97         g_error_free (gerr);
  98         gerr = NULL;
  99       }
 100
 101     g_io_channel_set_buffer_size (gio_r, BUFFER_SIZE);
 102
 103     status = g_io_channel_set_flags (gio_r, G_IO_FLAG_NONBLOCK, &gerr);
 104     if (status == G_IO_STATUS_ERROR)
 105       {
 106         g_warning (gerr->message);
 107         g_error_free (gerr);
 108         gerr = NULL;
 109       }
 110     flags = g_io_channel_get_flags (gio_r);
 111     buffer = g_string_sized_new (BUFFER_SIZE);
 112
 113     while (TRUE)
 114     {
 115         do
 116           status = g_io_channel_read_line_string (gio_r, buffer, NULL, &gerr);
 117         while (status == G_IO_STATUS_AGAIN);
 118         if (status != G_IO_STATUS_NORMAL)
 119           break;
 120
 121         rlength += buffer->len;
 122
 123         do
 124           status = g_io_channel_write_chars (gio_w, buffer->str, buffer->len,
 125             &length_out, &gerr);
 126         while (status == G_IO_STATUS_AGAIN);
 127         if (status != G_IO_STATUS_NORMAL)
 128           break;
 129
 130         wlength += length_out;
 131
 132         if (length_out < buffer->len)
 133           g_warning ("Only wrote part of the line.");
 134
 135 #ifdef VERBOSE
 136         g_print ("%s", buffer->str);
 137 #endif
 138         g_string_truncate (buffer, 0);
 139     }
 140
 141     switch (status)
 142       {
 143         case G_IO_STATUS_EOF:
 144           break;
 145         case G_IO_STATUS_ERROR:
 146           g_warning (gerr->message);
 147           g_error_free (gerr);
 148           gerr = NULL;
 149           break;
 150         default:
 151           g_warning ("Abnormal exit from write loop.");
 152           break;
 153       }
 154
 155     do
 156       status = g_io_channel_flush (gio_w, &gerr);
 157     while (status == G_IO_STATUS_AGAIN);
 158
 159     if (status == G_IO_STATUS_ERROR)
 160       {
 161         g_warning (gerr->message);
 162         g_error_free (gerr);
 163         gerr = NULL;
 164       }
 165
 166 #ifdef VERBOSE
 167     g_print ("read %d bytes, wrote %ld bytes\n", rlength, wlength);
 168 #endif
 169
 170     g_io_channel_unref(gio_r);
 171     g_io_channel_unref(gio_w);
 172
 173     test_small_writes ();
 174
 175     return 0;
 176 }