src/util/test_disk.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2001, 2002, 2003, 2005, 2006, 2009 GNUnet e.V.
   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 3, 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., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, USA.
  19 */
  20
  21 /**
  22  * @file util/test_disk.c
  23  * @brief testcase for the storage module
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28
  29 #define TESTSTRING "Hello World\0"
  30
  31
  32 static int
  33 testReadWrite ()
  34 {
  35   char tmp[100 + 1];
  36   int ret;
  37
  38   if (strlen (TESTSTRING) !=
  39       GNUNET_DISK_fn_write (".testfile", TESTSTRING, strlen (TESTSTRING),
  40                             GNUNET_DISK_PERM_USER_READ |
  41                             GNUNET_DISK_PERM_USER_WRITE))
  42     return 1;
  43   if (GNUNET_OK != GNUNET_DISK_file_test (".testfile"))
  44     return 1;
  45   ret = GNUNET_DISK_fn_read (".testfile", tmp, sizeof (tmp) - 1);
  46   if (ret < 0)
  47   {
  48     FPRINTF (stderr, "Error reading file `%s' in testReadWrite\n", ".testfile");
  49     return 1;
  50   }
  51   tmp[ret] = '\0';
  52   if (0 != memcmp (tmp, TESTSTRING, strlen (TESTSTRING) + 1))
  53   {
  54     FPRINTF (stderr, "Error in testReadWrite: *%s* != *%s* for file %s\n", tmp,
  55              TESTSTRING, ".testfile");
  56     return 1;
  57   }
  58   GNUNET_DISK_file_copy (".testfile", ".testfile2");
  59   memset (tmp, 0, sizeof (tmp));
  60   ret = GNUNET_DISK_fn_read (".testfile2", tmp, sizeof (tmp) - 1);
  61   if (ret < 0)
  62   {
  63     FPRINTF (stderr, "Error reading file `%s' in testReadWrite\n",
  64              ".testfile2");
  65     return 1;
  66   }
  67   tmp[ret] = '\0';
  68   if (0 != memcmp (tmp, TESTSTRING, strlen (TESTSTRING) + 1))
  69   {
  70     FPRINTF (stderr, "Error in testReadWrite: *%s* != *%s* for file %s\n", tmp,
  71              TESTSTRING, ".testfile2");
  72     return 1;
  73   }
  74
  75   GNUNET_break (0 == UNLINK (".testfile"));
  76   GNUNET_break (0 == UNLINK (".testfile2"));
  77   if (GNUNET_NO != GNUNET_DISK_file_test (".testfile"))
  78     return 1;
  79
  80   return 0;
  81 }
  82
  83
  84 static int
  85 testOpenClose ()
  86 {
  87   struct GNUNET_DISK_FileHandle *fh;
  88   uint64_t size;
  89
  90   fh = GNUNET_DISK_file_open (".testfile",
  91                               GNUNET_DISK_OPEN_READWRITE |
  92                               GNUNET_DISK_OPEN_CREATE,
  93                               GNUNET_DISK_PERM_USER_READ |
  94                               GNUNET_DISK_PERM_USER_WRITE);
  95   GNUNET_assert (GNUNET_NO == GNUNET_DISK_handle_invalid (fh));
  96   GNUNET_break (5 == GNUNET_DISK_file_write (fh, "Hello", 5));
  97   GNUNET_DISK_file_close (fh);
  98   GNUNET_break (GNUNET_OK ==
  99                 GNUNET_DISK_file_size (".testfile", &size, GNUNET_NO, GNUNET_YES));
 100   if (size != 5)
 101     return 1;
 102   GNUNET_break (0 == UNLINK (".testfile"));
 103
 104   return 0;
 105 }
 106
 107 static int ok;
 108
 109
 110 static int
 111 scan_callback (void *want, const char *filename)
 112 {
 113   if (NULL != strstr (filename, want))
 114     ok++;
 115   return GNUNET_OK;
 116 }
 117
 118
 119 static int
 120 testDirScan ()
 121 {
 122   if (GNUNET_OK !=
 123       GNUNET_DISK_directory_create ("test" DIR_SEPARATOR_STR "entry"))
 124   {
 125     GNUNET_break (0);
 126     return 1;
 127   }
 128   if (GNUNET_OK !=
 129       GNUNET_DISK_directory_create ("test" DIR_SEPARATOR_STR "entry_more"))
 130   {
 131     GNUNET_break (0);
 132     return 1;
 133   }
 134   GNUNET_DISK_directory_scan ("test", &scan_callback,
 135                               "test" DIR_SEPARATOR_STR "entry");
 136   if (GNUNET_OK != GNUNET_DISK_directory_remove ("test"))
 137   {
 138     GNUNET_break (0);
 139     return 1;
 140   }
 141   if (ok < 2)
 142   {
 143     GNUNET_break (0);
 144     return 1;
 145   }
 146   return 0;
 147 }
 148
 149
 150 static int
 151 iter_callback (void *cls,
 152                const char *filename)
 153 {
 154   int *i = cls;
 155
 156   (*i)++;
 157   return GNUNET_OK;
 158 }
 159
 160
 161 static int
 162 testDirIter ()
 163 {
 164   int i;
 165
 166   i = 0;
 167   if (GNUNET_OK != GNUNET_DISK_directory_create ("test/entry"))
 168   {
 169     GNUNET_break (0);
 170     return 1;
 171   }
 172   if (GNUNET_OK != GNUNET_DISK_directory_create ("test/entry_many"))
 173   {
 174     GNUNET_break (0);
 175     return 1;
 176   }
 177   if (GNUNET_OK != GNUNET_DISK_directory_create ("test/entry_more"))
 178   {
 179     GNUNET_break (0);
 180     return 1;
 181   }
 182   GNUNET_DISK_directory_scan ("test",
 183                               &iter_callback,
 184                               &i);
 185   if (GNUNET_OK != GNUNET_DISK_directory_remove ("test"))
 186   {
 187     GNUNET_break (0);
 188     return 1;
 189   }
 190   if (i < 3)
 191   {
 192     GNUNET_break (0);
 193     return 1;
 194   }
 195   return 0;
 196 }
 197
 198
 199 static int
 200 testCanonicalize ()
 201 {
 202   char *fn = GNUNET_strdup ("ab?><|cd*ef:/g\"");
 203
 204   GNUNET_DISK_filename_canonicalize (fn);
 205   if (0 != strcmp (fn, "ab____cd_ef__g_"))
 206   {
 207     GNUNET_free (fn);
 208     return 1;
 209   }
 210   GNUNET_free (fn);
 211   return 0;
 212 }
 213
 214
 215 static int
 216 testChangeOwner ()
 217 {
 218 #ifndef WINDOWS
 219   GNUNET_log_skip (1, GNUNET_NO);
 220   if (GNUNET_OK == GNUNET_DISK_file_change_owner ("/dev/null", "unknownuser"))
 221     return 1;
 222 #endif
 223   return 0;
 224 }
 225
 226
 227 static int
 228 testDirMani ()
 229 {
 230   if (GNUNET_OK != GNUNET_DISK_directory_create_for_file ("test/ing"))
 231   {
 232     GNUNET_break (0);
 233     return 1;
 234   }
 235   if (GNUNET_NO != GNUNET_DISK_file_test ("test"))
 236   {
 237     GNUNET_break (0);
 238     return 1;
 239   }
 240   if (GNUNET_NO != GNUNET_DISK_file_test ("test/ing"))
 241   {
 242     GNUNET_break (0);
 243     return 1;
 244   }
 245   if (GNUNET_OK != GNUNET_DISK_directory_remove ("test"))
 246   {
 247     GNUNET_break (0);
 248     return 1;
 249   }
 250   if (GNUNET_OK != GNUNET_DISK_directory_create ("test"))
 251   {
 252     GNUNET_break (0);
 253     return 1;
 254   }
 255   if (GNUNET_YES != GNUNET_DISK_directory_test ("test", GNUNET_YES))
 256   {
 257     GNUNET_break (0);
 258     return 1;
 259   }
 260   if (GNUNET_OK != GNUNET_DISK_directory_remove ("test"))
 261   {
 262     GNUNET_break (0);
 263     return 1;
 264   }
 265   return 0;
 266 }
 267
 268
 269 int
 270 main (int argc, char *argv[])
 271 {
 272   unsigned int failureCount = 0;
 273
 274   GNUNET_log_setup ("test-disk", "WARNING", NULL);
 275   failureCount += testReadWrite ();
 276   failureCount += testOpenClose ();
 277   failureCount += testDirScan ();
 278   failureCount += testDirIter ();
 279   failureCount += testCanonicalize ();
 280   failureCount += testChangeOwner ();
 281   failureCount += testDirMani ();
 282   if (0 != failureCount)
 283   {
 284     FPRINTF (stderr,
 285              "\n%u TESTS FAILED!\n",
 286              failureCount);
 287     return -1;
 288   }
 289   return 0;
 290 }                               /* end of main */