gio/tests/readwrite.c

   1 #include <glib/glib.h>
   2 #include <glib/gstdio.h>
   3 #include <gio/gio.h>
   4 #include <string.h>
   5
   6 #ifdef G_OS_UNIX
   7 #include <unistd.h>
   8 #endif
   9 #ifdef G_OS_WIN32
  10 #include <io.h> /* for close() */
  11 #endif
  12
  13 static const char *original_data = "This is some test data that we can put in a file...";
  14 static const char *new_data = "new data..";
  15
  16 static void
  17 verify_pos (GIOStream *iostream, goffset expected_pos)
  18 {
  19   goffset pos;
  20
  21   pos = g_seekable_tell (G_SEEKABLE (iostream));
  22   g_assert_cmpint (pos, ==, expected_pos);
  23
  24   pos = g_seekable_tell (G_SEEKABLE (g_io_stream_get_input_stream (iostream)));
  25   g_assert_cmpint (pos, ==, expected_pos);
  26
  27   pos = g_seekable_tell (G_SEEKABLE (g_io_stream_get_output_stream (iostream)));
  28   g_assert_cmpint (pos, ==, expected_pos);
  29 }
  30
  31 static void
  32 verify_iostream (GFileIOStream *file_iostream)
  33 {
  34   gboolean res;
  35   gssize skipped;
  36   GIOStream *iostream;
  37   GError *error;
  38   GInputStream *in;
  39   GOutputStream *out;
  40   char buffer[1024];
  41   gsize n_bytes;
  42   char *modified_data;
  43
  44   iostream = G_IO_STREAM (file_iostream);
  45
  46   verify_pos (iostream, 0);
  47
  48   in = g_io_stream_get_input_stream (iostream);
  49   out = g_io_stream_get_output_stream (iostream);
  50
  51   res = g_input_stream_read_all (in, buffer, 20, &n_bytes, NULL, NULL);
  52   g_assert (res);
  53   g_assert_cmpmem (buffer, n_bytes, original_data, 20);
  54
  55   verify_pos (iostream, 20);
  56
  57   res = g_seekable_seek (G_SEEKABLE (iostream),
  58                          -10, G_SEEK_END,
  59                          NULL, NULL);
  60   g_assert (res);
  61   verify_pos (iostream, strlen (original_data) - 10);
  62
  63   res = g_input_stream_read_all (in, buffer, 20, &n_bytes, NULL, NULL);
  64   g_assert (res);
  65   g_assert_cmpmem (buffer, n_bytes, original_data + strlen (original_data) - 10, 10);
  66
  67   verify_pos (iostream, strlen (original_data));
  68
  69   res = g_seekable_seek (G_SEEKABLE (iostream),
  70                          10, G_SEEK_SET,
  71                          NULL, NULL);
  72
  73   res = g_input_stream_skip (in, 5, NULL, NULL);
  74   g_assert (res == 5);
  75   verify_pos (iostream, 15);
  76
  77   skipped = g_input_stream_skip (in, 10000, NULL, NULL);
  78   g_assert_cmpint (skipped, >=, 0);
  79   g_assert ((gsize) skipped == strlen (original_data) - 15);
  80   verify_pos (iostream, strlen (original_data));
  81
  82   res = g_seekable_seek (G_SEEKABLE (iostream),
  83                          10, G_SEEK_SET,
  84                          NULL, NULL);
  85
  86   verify_pos (iostream, 10);
  87
  88   res = g_output_stream_write_all (out, new_data, strlen (new_data),
  89                                    &n_bytes, NULL, NULL);
  90   g_assert (res);
  91   g_assert_cmpint (n_bytes, ==, strlen (new_data));
  92
  93   verify_pos (iostream, 10 + strlen (new_data));
  94
  95   res = g_seekable_seek (G_SEEKABLE (iostream),
  96                          0, G_SEEK_SET,
  97                          NULL, NULL);
  98   g_assert (res);
  99   verify_pos (iostream, 0);
 100
 101   res = g_input_stream_read_all (in, buffer, strlen (original_data), &n_bytes, NULL, NULL);
 102   g_assert (res);
 103   g_assert_cmpint ((int)n_bytes, ==, strlen (original_data));
 104   buffer[n_bytes] = 0;
 105
 106   modified_data = g_strdup (original_data);
 107   memcpy (modified_data + 10, new_data, strlen (new_data));
 108   g_assert_cmpstr (buffer, ==, modified_data);
 109
 110   verify_pos (iostream, strlen (original_data));
 111
 112   res = g_seekable_seek (G_SEEKABLE (iostream),
 113                          0, G_SEEK_SET,
 114                          NULL, NULL);
 115   g_assert (res);
 116   verify_pos (iostream, 0);
 117
 118   res = g_output_stream_close (out, NULL, NULL);
 119   g_assert (res);
 120
 121   res = g_input_stream_read_all (in, buffer, 15, &n_bytes, NULL, NULL);
 122   g_assert (res);
 123   g_assert_cmpmem (buffer, n_bytes, modified_data, 15);
 124
 125   error = NULL;
 126   res = g_output_stream_write_all (out, new_data, strlen (new_data),
 127                                    &n_bytes, NULL, &error);
 128   g_assert (!res);
 129   g_assert_error (error, G_IO_ERROR, G_IO_ERROR_CLOSED);
 130   g_error_free (error);
 131
 132   error = NULL;
 133   res = g_io_stream_close (iostream, NULL, &error);
 134   g_assert (res);
 135   g_assert_no_error (error);
 136
 137   g_free (modified_data);
 138 }
 139
 140 static void
 141 test_g_file_open_readwrite (void)
 142 {
 143   char *tmp_file;
 144   int fd;
 145   gboolean res;
 146   GFileIOStream *file_iostream;
 147   char *path;
 148   GFile *file;
 149   GError *error;
 150
 151   fd = g_file_open_tmp ("readwrite_XXXXXX",
 152                         &tmp_file, NULL);
 153   g_assert (fd != -1);
 154   close (fd);
 155
 156   res = g_file_set_contents (tmp_file,
 157                              original_data, -1, NULL);
 158   g_assert (res);
 159
 160   path = g_build_filename (g_get_tmp_dir (), "g-a-nonexisting-file", NULL);
 161   file = g_file_new_for_path (path);
 162   g_free (path);
 163   error = NULL;
 164   file_iostream = g_file_open_readwrite (file, NULL, &error);
 165   g_assert (file_iostream == NULL);
 166   g_assert (g_error_matches (error, G_IO_ERROR, G_IO_ERROR_NOT_FOUND));
 167   g_error_free (error);
 168   g_object_unref (file);
 169
 170   file = g_file_new_for_path (tmp_file);
 171   error = NULL;
 172   file_iostream = g_file_open_readwrite (file, NULL, &error);
 173   g_assert (file_iostream != NULL);
 174   g_object_unref (file);
 175
 176   verify_iostream (file_iostream);
 177
 178   g_object_unref (file_iostream);
 179
 180   g_unlink (tmp_file);
 181   g_free (tmp_file);
 182 }
 183
 184 static void
 185 test_g_file_create_readwrite (void)
 186 {
 187   char *tmp_file;
 188   int fd;
 189   gboolean res;
 190   GFileIOStream *file_iostream;
 191   GOutputStream *out;
 192   GFile *file;
 193   GError *error;
 194   gsize n_bytes;
 195
 196   fd = g_file_open_tmp ("readwrite_XXXXXX",
 197                         &tmp_file, NULL);
 198   g_assert (fd != -1);
 199   close (fd);
 200
 201   file = g_file_new_for_path (tmp_file);
 202   error = NULL;
 203   file_iostream = g_file_create_readwrite (file, 0, NULL, &error);
 204   g_assert (file_iostream == NULL);
 205   g_assert_error (error, G_IO_ERROR, G_IO_ERROR_EXISTS);
 206   g_error_free (error);
 207
 208   g_unlink (tmp_file);
 209   file_iostream = g_file_create_readwrite (file, 0, NULL, &error);
 210   g_assert (file_iostream != NULL);
 211
 212   out = g_io_stream_get_output_stream (G_IO_STREAM (file_iostream));
 213   res = g_output_stream_write_all (out, original_data, strlen (original_data),
 214                                    &n_bytes, NULL, NULL);
 215   g_assert (res);
 216   g_assert_cmpint (n_bytes, ==, strlen (original_data));
 217
 218   res = g_seekable_seek (G_SEEKABLE (file_iostream),
 219                          0, G_SEEK_SET,
 220                          NULL, NULL);
 221   g_assert (res);
 222
 223   verify_iostream (file_iostream);
 224
 225   g_object_unref (file_iostream);
 226   g_object_unref (file);
 227
 228   g_unlink (tmp_file);
 229   g_free (tmp_file);
 230 }
 231
 232 static void
 233 test_g_file_replace_readwrite (void)
 234 {
 235   char *tmp_file, *backup, *data;
 236   int fd;
 237   gboolean res;
 238   GFileIOStream *file_iostream;
 239   GInputStream *in;
 240   GOutputStream *out;
 241   GFile *file;
 242   GError *error;
 243   char buffer[1024];
 244   gsize n_bytes;
 245
 246   fd = g_file_open_tmp ("readwrite_XXXXXX",
 247                         &tmp_file, NULL);
 248   g_assert (fd != -1);
 249   close (fd);
 250
 251   res = g_file_set_contents (tmp_file,
 252                              new_data, -1, NULL);
 253   g_assert (res);
 254
 255   file = g_file_new_for_path (tmp_file);
 256   error = NULL;
 257   file_iostream = g_file_replace_readwrite (file, NULL,
 258                                             TRUE, 0, NULL, &error);
 259   g_assert (file_iostream != NULL);
 260
 261   in = g_io_stream_get_input_stream (G_IO_STREAM (file_iostream));
 262
 263   /* Ensure its empty */
 264   res = g_input_stream_read_all (in, buffer, sizeof buffer, &n_bytes, NULL, NULL);
 265   g_assert (res);
 266   g_assert_cmpint ((int)n_bytes, ==, 0);
 267
 268   out = g_io_stream_get_output_stream (G_IO_STREAM (file_iostream));
 269   res = g_output_stream_write_all (out, original_data, strlen (original_data),
 270                                    &n_bytes, NULL, NULL);
 271   g_assert (res);
 272   g_assert_cmpint (n_bytes, ==, strlen (original_data));
 273
 274   res = g_seekable_seek (G_SEEKABLE (file_iostream),
 275                          0, G_SEEK_SET,
 276                          NULL, NULL);
 277   g_assert (res);
 278
 279   verify_iostream (file_iostream);
 280
 281   g_object_unref (file_iostream);
 282   g_object_unref (file);
 283
 284   backup = g_strconcat (tmp_file, "~", NULL);
 285   res = g_file_get_contents (backup,
 286                              &data,
 287                              NULL, NULL);
 288   g_assert (res);
 289   g_assert_cmpstr (data, ==, new_data);
 290   g_free (data);
 291   g_unlink (backup);
 292   g_free (backup);
 293
 294   g_unlink (tmp_file);
 295   g_free (tmp_file);
 296 }
 297
 298
 299 int
 300 main (int   argc,
 301       char *argv[])
 302 {
 303   g_test_init (&argc, &argv, NULL);
 304
 305   g_test_add_func ("/readwrite/test_g_file_open_readwrite",
 306                    test_g_file_open_readwrite);
 307   g_test_add_func ("/readwrite/test_g_file_create_readwrite",
 308                    test_g_file_create_readwrite);
 309   g_test_add_func ("/readwrite/test_g_file_replace_readwrite",
 310                    test_g_file_replace_readwrite);
 311
 312   return g_test_run();
 313 }