src/util/test_bio.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 2, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 /**
  22  * @file util/test_bio.c
  23  * @brief testcase for the buffered IO module
  24  * @author Ji Lu
  25  */
  26
  27
  28 #include "platform.h"
  29 #include "gnunet_util_lib.h"
  30 #define TESTSTRING "testString"
  31 #define TESTNUMBER64 ((int64_t)100000L)
  32
  33 static int
  34 test_normal_rw ()
  35 {
  36   char *msg;
  37   int64_t testNum;
  38   char *readResultString;
  39   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
  40   struct GNUNET_BIO_WriteHandle *fileW;
  41   struct GNUNET_BIO_ReadHandle *fileR;
  42   struct GNUNET_CONTAINER_MetaData *metaDataW;
  43   struct GNUNET_CONTAINER_MetaData *metaDataR;
  44   metaDataW = GNUNET_CONTAINER_meta_data_create ();
  45   metaDataR = NULL;
  46   GNUNET_CONTAINER_meta_data_add_publication_date (metaDataW);
  47
  48   fileW = GNUNET_BIO_write_open (fileName);
  49   GNUNET_assert (NULL != fileW);
  50   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_string (fileW, TESTSTRING));
  51   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_meta_data (fileW, metaDataW));
  52   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int64 (fileW, TESTNUMBER64));
  53   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
  54
  55   fileR = GNUNET_BIO_read_open (fileName);
  56   GNUNET_assert (NULL != fileR);
  57   readResultString = NULL;
  58   GNUNET_assert (GNUNET_OK ==
  59                  GNUNET_BIO_read_string (fileR, "Read string error",
  60                                          &readResultString, 200));
  61   GNUNET_assert (0 == strcmp (TESTSTRING, readResultString));
  62   GNUNET_free (readResultString);
  63   GNUNET_assert (GNUNET_OK ==
  64                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
  65                                             &metaDataR));
  66   GNUNET_assert (GNUNET_YES ==
  67                  GNUNET_CONTAINER_meta_data_test_equal (metaDataR,
  68                                                         metaDataW));
  69   GNUNET_assert (GNUNET_OK == GNUNET_BIO_read_int64 (fileR, &testNum));
  70   GNUNET_BIO_read_close (fileR, &msg);
  71   GNUNET_CONTAINER_meta_data_destroy (metaDataW);
  72   GNUNET_CONTAINER_meta_data_destroy (metaDataR);
  73   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
  74   GNUNET_free (fileName);
  75   return 0;
  76 }
  77
  78 static int
  79 test_nullstring_rw ()
  80 {
  81   char *msg;
  82   char *readResultString = (char *) "not null";
  83   struct GNUNET_BIO_WriteHandle *fileW;
  84   struct GNUNET_BIO_ReadHandle *fileR;
  85   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
  86
  87   fileW = GNUNET_BIO_write_open (fileName);
  88   GNUNET_assert (NULL != fileW);
  89   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_string (fileW, NULL));
  90   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
  91
  92   fileR = GNUNET_BIO_read_open (fileName);
  93   GNUNET_assert (NULL != fileR);
  94   GNUNET_assert (GNUNET_OK ==
  95                  GNUNET_BIO_read_string (fileR, "Read string error",
  96                                          &readResultString, 200));
  97   GNUNET_assert (NULL == readResultString);
  98   GNUNET_BIO_read_close (fileR, &msg);
  99   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 100   GNUNET_free (fileName);
 101
 102   return 0;
 103 }
 104
 105 static int
 106 test_emptystring_rw ()
 107 {
 108   char *msg;
 109   char *readResultString;
 110   struct GNUNET_BIO_WriteHandle *fileW;
 111   struct GNUNET_BIO_ReadHandle *fileR;
 112   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
 113
 114   fileW = GNUNET_BIO_write_open (fileName);
 115   GNUNET_assert (NULL != fileW);
 116   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_string (fileW, ""));
 117   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
 118
 119   fileR = GNUNET_BIO_read_open (fileName);
 120   GNUNET_assert (NULL != fileR);
 121   readResultString = NULL;
 122   GNUNET_assert (GNUNET_OK ==
 123                  GNUNET_BIO_read_string (fileR, "Read string error",
 124                                          &readResultString, 200));
 125   GNUNET_free (readResultString);
 126   GNUNET_BIO_read_close (fileR, &msg);
 127   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 128   GNUNET_free (fileName);
 129   return 0;
 130 }
 131
 132 static int
 133 test_bigstring_rw ()
 134 {
 135   char *msg;
 136   char *readResultString;
 137   struct GNUNET_BIO_WriteHandle *fileW;
 138   struct GNUNET_BIO_ReadHandle *fileR;
 139   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
 140
 141   fileW = GNUNET_BIO_write_open (fileName);
 142   GNUNET_assert (NULL != fileW);
 143   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_string (fileW, TESTSTRING));
 144   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
 145
 146   fileR = GNUNET_BIO_read_open (fileName);
 147   GNUNET_assert (NULL != fileR);
 148   readResultString = NULL;
 149   GNUNET_assert (GNUNET_SYSERR ==
 150                  GNUNET_BIO_read_string (fileR, "Read string error",
 151                                          &readResultString, 1));
 152   GNUNET_assert (NULL == readResultString);
 153   msg = NULL;
 154   GNUNET_BIO_read_close (fileR, &msg);
 155   GNUNET_free (msg);
 156   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 157   GNUNET_free (fileName);
 158   return 0;
 159 }
 160
 161 static int
 162 test_bigmeta_rw ()
 163 {
 164   char *msg;
 165   static char meta[1024 * 1024 * 10];
 166   memset (meta, 'b', sizeof (meta));
 167   meta[sizeof (meta) - 1] = '\0';
 168   struct GNUNET_BIO_WriteHandle *fileW;
 169   struct GNUNET_BIO_ReadHandle *fileR;
 170   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
 171   struct GNUNET_CONTAINER_MetaData *metaDataW;
 172   struct GNUNET_CONTAINER_MetaData *metaDataR;
 173   metaDataW = GNUNET_CONTAINER_meta_data_create ();
 174   GNUNET_CONTAINER_meta_data_add_publication_date (metaDataW);
 175   GNUNET_assert (GNUNET_OK ==
 176                  GNUNET_CONTAINER_meta_data_insert (metaDataW,
 177                                                     EXTRACTOR_COMMENT, meta));
 178
 179   fileW = GNUNET_BIO_write_open (fileName);
 180   GNUNET_assert (NULL != fileW);
 181   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_meta_data (fileW, metaDataW));
 182   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
 183
 184   fileR = GNUNET_BIO_read_open (fileName);
 185   GNUNET_assert (NULL != fileR);
 186   metaDataR = NULL;
 187   GNUNET_assert (GNUNET_SYSERR ==
 188                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
 189                                             &metaDataR));
 190   msg = NULL;
 191   GNUNET_BIO_read_close (fileR, &msg);
 192   GNUNET_free (msg);
 193   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 194   GNUNET_assert (NULL == metaDataR);
 195   GNUNET_CONTAINER_meta_data_destroy (metaDataW);
 196   GNUNET_free (fileName);
 197   return 0;
 198 }
 199
 200 static int
 201 test_directory_r ()
 202 {
 203   char *msg;
 204   char readResult[200];
 205   struct GNUNET_BIO_ReadHandle *fileR;
 206
 207   fileR = GNUNET_BIO_read_open ("/dev");
 208   GNUNET_assert (NULL != fileR);
 209   GNUNET_assert (GNUNET_SYSERR ==
 210                  GNUNET_BIO_read (fileR, "Read error", readResult, 65537));
 211   msg = NULL;
 212   GNUNET_BIO_read_close (fileR, &msg);
 213   GNUNET_free (msg);
 214   return 0;
 215 }
 216
 217 static int
 218 test_nullfile_rw ()
 219 {
 220   char *msg;
 221   int64_t testNum;
 222   char *readResultString;
 223   static char fileNameNO[102401];
 224   char readResult[200];
 225   memset (fileNameNO, 'a', sizeof (fileNameNO));
 226   fileNameNO[sizeof (fileNameNO) - 1] = '\0';
 227   struct GNUNET_BIO_WriteHandle *fileW, *fileWNO;
 228   struct GNUNET_BIO_ReadHandle *fileR, *fileRNO;
 229   struct GNUNET_CONTAINER_MetaData *metaDataW;
 230   struct GNUNET_CONTAINER_MetaData *metaDataR;
 231   metaDataW = GNUNET_CONTAINER_meta_data_create ();
 232   metaDataR = GNUNET_CONTAINER_meta_data_create ();
 233   GNUNET_CONTAINER_meta_data_add_publication_date (metaDataW);
 234
 235   fileWNO = GNUNET_BIO_write_open (fileNameNO);
 236   GNUNET_assert (NULL == fileWNO);
 237
 238   fileRNO = GNUNET_BIO_read_open (fileNameNO);
 239   GNUNET_assert (NULL == fileRNO);
 240
 241   fileW = GNUNET_BIO_write_open ("/dev/full");
 242   GNUNET_assert (NULL != fileW);
 243   GNUNET_assert (GNUNET_SYSERR ==
 244                  GNUNET_BIO_write (fileW, fileNameNO, sizeof (fileNameNO)));
 245   GNUNET_assert (GNUNET_SYSERR ==
 246                  GNUNET_BIO_write_string (fileW, TESTSTRING));
 247   GNUNET_assert (GNUNET_SYSERR ==
 248                  GNUNET_BIO_write_meta_data (fileW, metaDataW));
 249   GNUNET_assert (GNUNET_SYSERR == GNUNET_BIO_write_close (fileW));
 250
 251   fileW = GNUNET_BIO_write_open ("/dev/full");
 252   GNUNET_assert (NULL != fileW);
 253   GNUNET_assert (GNUNET_SYSERR == GNUNET_BIO_write_close (fileW));
 254
 255   fileR = GNUNET_BIO_read_open ("/dev/null");
 256   GNUNET_assert (NULL != fileR);
 257   GNUNET_assert (GNUNET_SYSERR ==
 258                  GNUNET_BIO_read (fileR, "Read error", readResult, 65537));
 259   readResultString = NULL;
 260   GNUNET_assert (GNUNET_SYSERR ==
 261                  GNUNET_BIO_read_string (fileR, "Read string error",
 262                                          &readResultString, 200));
 263   GNUNET_assert (NULL == readResultString);
 264   GNUNET_assert (GNUNET_SYSERR == GNUNET_BIO_read_int64 (fileR, &testNum));
 265   GNUNET_assert (GNUNET_SYSERR ==
 266                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
 267                                             &metaDataR));
 268   msg = NULL;
 269   GNUNET_BIO_read_close (fileR, &msg);
 270   GNUNET_free (msg);
 271   GNUNET_CONTAINER_meta_data_destroy (metaDataR);
 272   GNUNET_CONTAINER_meta_data_destroy (metaDataW);
 273   return 0;
 274 }
 275
 276 static int
 277 test_fakestring_rw ()
 278 {
 279   char *msg;
 280   int32_t tmpInt = 2;
 281   char *readResult;
 282   struct GNUNET_BIO_WriteHandle *fileW;
 283   struct GNUNET_BIO_ReadHandle *fileR;
 284   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
 285
 286   fileW = GNUNET_BIO_write_open (fileName);
 287   GNUNET_assert (NULL != fileW);
 288   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int32 (fileW, tmpInt));
 289   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
 290
 291   fileR = GNUNET_BIO_read_open (fileName);
 292   GNUNET_assert (NULL != fileR);
 293   GNUNET_assert (GNUNET_SYSERR ==
 294                  GNUNET_BIO_read_string (fileR, "Read string error",
 295                                          &readResult, 200));
 296   msg = NULL;
 297   GNUNET_BIO_read_close (fileR, &msg);
 298   GNUNET_free (msg);
 299   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 300   GNUNET_free (fileName);
 301   return 0;
 302 }
 303
 304 static int
 305 test_fakemeta_rw ()
 306 {
 307   char *msg;
 308   int32_t tmpInt = 2;
 309   struct GNUNET_BIO_WriteHandle *fileW;
 310   struct GNUNET_BIO_ReadHandle *fileR;
 311   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
 312   struct GNUNET_CONTAINER_MetaData *metaDataR;
 313
 314   fileW = GNUNET_BIO_write_open (fileName);
 315   GNUNET_assert (NULL != fileW);
 316   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int32 (fileW, tmpInt));
 317   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
 318
 319   fileR = GNUNET_BIO_read_open (fileName);
 320   GNUNET_assert (NULL != fileR);
 321   metaDataR = NULL;
 322   GNUNET_assert (GNUNET_SYSERR ==
 323                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
 324                                             &metaDataR));
 325   GNUNET_assert (NULL == metaDataR);
 326   msg = NULL;
 327   GNUNET_BIO_read_close (fileR, &msg);
 328   GNUNET_free (msg);
 329   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 330   GNUNET_free (fileName);
 331   return 0;
 332 }
 333
 334 static int
 335 test_fakebigmeta_rw ()
 336 {
 337   char *msg;
 338   int32_t tmpInt = 1024 * 1024 * 10;
 339   struct GNUNET_BIO_WriteHandle *fileW;
 340   struct GNUNET_BIO_ReadHandle *fileR;
 341   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
 342   struct GNUNET_CONTAINER_MetaData *metaDataR;
 343
 344   fileW = GNUNET_BIO_write_open (fileName);
 345   GNUNET_assert (NULL != fileW);
 346   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int32 (fileW, tmpInt));
 347   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
 348
 349   fileR = GNUNET_BIO_read_open (fileName);
 350   GNUNET_assert (NULL != fileR);
 351   metaDataR = NULL;
 352   GNUNET_assert (GNUNET_SYSERR ==
 353                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
 354                                             &metaDataR));
 355   msg = NULL;
 356   GNUNET_BIO_read_close (fileR, &msg);
 357   GNUNET_free (msg);
 358   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
 359   GNUNET_assert (NULL == metaDataR);
 360   GNUNET_free (fileName);
 361   return 0;
 362 }
 363
 364 static int
 365 check_string_rw ()
 366 {
 367   GNUNET_assert (0 == test_nullstring_rw ());
 368   GNUNET_assert (0 == test_emptystring_rw ());
 369   GNUNET_assert (0 == test_bigstring_rw ());
 370   GNUNET_assert (0 == test_fakestring_rw ());
 371   return 0;
 372 }
 373
 374 static int
 375 check_metadata_rw ()
 376 {
 377   GNUNET_assert (0 == test_fakebigmeta_rw ());
 378   GNUNET_assert (0 == test_fakemeta_rw ());
 379   GNUNET_assert (0 == test_bigmeta_rw ());
 380   return 0;
 381 }
 382
 383 static int
 384 check_file_rw ()
 385 {
 386   GNUNET_assert (0 == test_normal_rw ());
 387   GNUNET_assert (0 == test_nullfile_rw ());
 388   GNUNET_assert (0 == test_directory_r ());
 389   return 0;
 390 }
 391
 392 int
 393 main (int argc, char *argv[])
 394 {
 395   GNUNET_assert (0 == check_file_rw ());
 396   GNUNET_assert (0 == check_metadata_rw ());
 397   GNUNET_assert (0 == check_string_rw ());
 398   return 0;
 399 }                               /* end of main */