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
test_small_writes(void)15 test_small_writes (void)
16 {
17 GIOChannel *io;
18 GIOStatus status = G_IO_STATUS_ERROR;
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_clear_error (&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
main(gint argc,gchar * argv[])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 gint rlength = 0;
62 glong wlength = 0;
63 gsize length_out;
64 const gchar encoding[] = "EUC-JP";
65 GIOStatus status;
66
67 g_test_init (&argc, &argv, NULL);
68
69 filename = g_test_build_filename (G_TEST_DIST, "iochannel-test-infile", NULL);
70
71 setbuf (stdout, NULL); /* For debugging */
72
73 gio_r = g_io_channel_new_file (filename, "r", &gerr);
74 if (gerr)
75 {
76 g_warning ("Unable to open file %s: %s", filename, gerr->message);
77 g_clear_error (&gerr);
78 return 1;
79 }
80 gio_w = g_io_channel_new_file ("iochannel-test-outfile", "w", &gerr);
81 if (gerr)
82 {
83 g_warning ("Unable to open file %s: %s", "iochannel-test-outfile", gerr->message);
84 g_clear_error (&gerr);
85 return 1;
86 }
87
88 g_io_channel_set_encoding (gio_r, encoding, &gerr);
89 if (gerr)
90 {
91 g_warning ("%s", gerr->message);
92 /* Keep going if this is just a case of iconv not supporting EUC-JP, see bug 428048 */
93 if (gerr->code != G_CONVERT_ERROR_NO_CONVERSION)
94 return 1;
95 g_clear_error (&gerr);
96 }
97
98 g_io_channel_set_buffer_size (gio_r, BUFFER_SIZE);
99
100 status = g_io_channel_set_flags (gio_r, G_IO_FLAG_NONBLOCK, &gerr);
101 if (status == G_IO_STATUS_ERROR)
102 {
103 g_warning ("%s", gerr->message);
104 g_clear_error (&gerr);
105 }
106 buffer = g_string_sized_new (BUFFER_SIZE);
107
108 while (TRUE)
109 {
110 do
111 status = g_io_channel_read_line_string (gio_r, buffer, NULL, &gerr);
112 while (status == G_IO_STATUS_AGAIN);
113 if (status != G_IO_STATUS_NORMAL)
114 break;
115
116 rlength += buffer->len;
117
118 do
119 status = g_io_channel_write_chars (gio_w, buffer->str, buffer->len,
120 &length_out, &gerr);
121 while (status == G_IO_STATUS_AGAIN);
122 if (status != G_IO_STATUS_NORMAL)
123 break;
124
125 wlength += length_out;
126
127 if (length_out < buffer->len)
128 g_warning ("Only wrote part of the line.");
129
130 #ifdef VERBOSE
131 g_print ("%s", buffer->str);
132 #endif
133 g_string_truncate (buffer, 0);
134 }
135
136 switch (status)
137 {
138 case G_IO_STATUS_EOF:
139 break;
140 case G_IO_STATUS_ERROR:
141 g_warning ("%s", gerr->message);
142 g_clear_error (&gerr);
143 break;
144 default:
145 g_warning ("Abnormal exit from write loop.");
146 break;
147 }
148
149 do
150 status = g_io_channel_flush (gio_w, &gerr);
151 while (status == G_IO_STATUS_AGAIN);
152
153 if (status == G_IO_STATUS_ERROR)
154 {
155 g_warning ("%s", gerr->message);
156 g_clear_error (&gerr);
157 }
158
159 #ifdef VERBOSE
160 g_print ("read %d bytes, wrote %ld bytes\n", rlength, wlength);
161 #endif
162
163 g_io_channel_unref(gio_r);
164 g_io_channel_unref(gio_w);
165
166 test_small_writes ();
167
168 g_free (filename);
169 g_string_free (buffer, TRUE);
170
171 return 0;
172 }
173