0dfa9414479fb2def55897f2ceda0e412b83bbd4
[oweals/gnunet.git] / 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_log_skip (1, GNUNET_NO);
188   GNUNET_assert (GNUNET_SYSERR ==
189                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
190                                             &metaDataR));
191   GNUNET_log_skip (0, GNUNET_YES);
192   msg = NULL;
193   GNUNET_BIO_read_close (fileR, &msg);
194   GNUNET_free (msg);
195   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
196   GNUNET_assert (NULL == metaDataR);
197   GNUNET_CONTAINER_meta_data_destroy (metaDataW);
198   GNUNET_free (fileName);
199   return 0;
200 }
201
202 static int
203 test_directory_r ()
204 {
205 #if LINUX || DARWIN
206   char *msg;
207   char readResult[200];
208   struct GNUNET_BIO_ReadHandle *fileR;
209
210   fileR = GNUNET_BIO_read_open ("/dev");
211   GNUNET_assert (NULL != fileR);
212   GNUNET_assert (GNUNET_SYSERR ==
213                  GNUNET_BIO_read (fileR, "Read error", readResult, sizeof(readResult)));
214   msg = NULL;
215   GNUNET_BIO_read_close (fileR, &msg);
216   GNUNET_free (msg);
217 #endif
218   return 0;
219 }
220
221 static int
222 test_nullfile_rw ()
223 {
224   char *msg;
225   int64_t testNum;
226   char *readResultString;
227   static char fileNameNO[102401];
228   char readResult[200];
229   struct GNUNET_BIO_WriteHandle *fileW, *fileWNO;
230   struct GNUNET_BIO_ReadHandle *fileR, *fileRNO;
231   struct GNUNET_CONTAINER_MetaData *metaDataW;
232   struct GNUNET_CONTAINER_MetaData *metaDataR;
233
234   memset (fileNameNO, 'a', sizeof (fileNameNO));
235   fileNameNO[sizeof (fileNameNO) - 1] = '\0';
236   metaDataW = GNUNET_CONTAINER_meta_data_create ();
237   metaDataR = GNUNET_CONTAINER_meta_data_create ();
238   GNUNET_CONTAINER_meta_data_add_publication_date (metaDataW);
239   
240   GNUNET_log_skip (1, GNUNET_NO);
241   fileWNO = GNUNET_BIO_write_open (fileNameNO);
242   GNUNET_log_skip (0, GNUNET_YES);
243   GNUNET_assert (NULL == fileWNO);
244
245   GNUNET_log_skip (1, GNUNET_NO);
246   fileRNO = GNUNET_BIO_read_open (fileNameNO);
247   GNUNET_log_skip (0, GNUNET_YES);
248   GNUNET_assert (NULL == fileRNO);
249
250 #ifdef LINUX 
251   /* /dev/full only seems to exist on Linux */
252   fileW = GNUNET_BIO_write_open ("/dev/full");
253   GNUNET_assert (NULL != fileW);
254   GNUNET_assert (GNUNET_SYSERR ==
255                  GNUNET_BIO_write (fileW, fileNameNO, sizeof (fileNameNO)));
256   GNUNET_assert (GNUNET_SYSERR ==
257                  GNUNET_BIO_write_string (fileW, TESTSTRING));
258   GNUNET_assert (GNUNET_SYSERR ==
259                  GNUNET_BIO_write_meta_data (fileW, metaDataW));
260   GNUNET_assert (GNUNET_SYSERR == GNUNET_BIO_write_close (fileW));
261
262   fileW = GNUNET_BIO_write_open ("/dev/full");
263   GNUNET_assert (NULL != fileW);
264   GNUNET_assert (GNUNET_SYSERR == GNUNET_BIO_write_close (fileW));
265
266   fileR = GNUNET_BIO_read_open ("/dev/null");
267   GNUNET_assert (NULL != fileR);
268   GNUNET_assert (GNUNET_SYSERR ==
269                  GNUNET_BIO_read (fileR, "Read error", readResult, sizeof(readResult)));
270   readResultString = NULL;
271   GNUNET_assert (GNUNET_SYSERR ==
272                  GNUNET_BIO_read_string (fileR, "Read string error",
273                                          &readResultString, 200));
274   GNUNET_assert (NULL == readResultString);
275   GNUNET_assert (GNUNET_SYSERR == GNUNET_BIO_read_int64 (fileR, &testNum));
276   GNUNET_assert (GNUNET_SYSERR ==
277                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
278                                             &metaDataR));
279   msg = NULL;
280   GNUNET_BIO_read_close (fileR, &msg);
281   GNUNET_free (msg);
282   GNUNET_CONTAINER_meta_data_destroy (metaDataR);
283   GNUNET_CONTAINER_meta_data_destroy (metaDataW);
284 #endif
285   return 0;
286 }
287
288 static int
289 test_fakestring_rw ()
290 {
291   char *msg;
292   int32_t tmpInt = 2;
293   char *readResult;
294   struct GNUNET_BIO_WriteHandle *fileW;
295   struct GNUNET_BIO_ReadHandle *fileR;
296   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
297
298   fileW = GNUNET_BIO_write_open (fileName);
299   GNUNET_assert (NULL != fileW);
300   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int32 (fileW, tmpInt));
301   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
302
303   fileR = GNUNET_BIO_read_open (fileName);
304   GNUNET_assert (NULL != fileR);
305   GNUNET_assert (GNUNET_SYSERR ==
306                  GNUNET_BIO_read_string (fileR, "Read string error",
307                                          &readResult, 200));
308   msg = NULL;
309   GNUNET_BIO_read_close (fileR, &msg);
310   GNUNET_free (msg);
311   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
312   GNUNET_free (fileName);
313   return 0;
314 }
315
316 static int
317 test_fakemeta_rw ()
318 {
319   char *msg;
320   int32_t tmpInt = 2;
321   struct GNUNET_BIO_WriteHandle *fileW;
322   struct GNUNET_BIO_ReadHandle *fileR;
323   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
324   struct GNUNET_CONTAINER_MetaData *metaDataR;
325
326   fileW = GNUNET_BIO_write_open (fileName);
327   GNUNET_assert (NULL != fileW);
328   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int32 (fileW, tmpInt));
329   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
330
331   fileR = GNUNET_BIO_read_open (fileName);
332   GNUNET_assert (NULL != fileR);
333   metaDataR = NULL;
334   GNUNET_assert (GNUNET_SYSERR ==
335                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
336                                             &metaDataR));
337   GNUNET_assert (NULL == metaDataR);
338   msg = NULL;
339   GNUNET_BIO_read_close (fileR, &msg);
340   GNUNET_free (msg);
341   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
342   GNUNET_free (fileName);
343   return 0;
344 }
345
346 static int
347 test_fakebigmeta_rw ()
348 {
349   char *msg;
350   int32_t tmpInt = 1024 * 1024 * 10;
351   struct GNUNET_BIO_WriteHandle *fileW;
352   struct GNUNET_BIO_ReadHandle *fileR;
353   char *fileName = GNUNET_DISK_mktemp ("gnunet_bio");
354   struct GNUNET_CONTAINER_MetaData *metaDataR;
355
356   fileW = GNUNET_BIO_write_open (fileName);
357   GNUNET_assert (NULL != fileW);
358   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_int32 (fileW, tmpInt));
359   GNUNET_assert (GNUNET_OK == GNUNET_BIO_write_close (fileW));
360
361   fileR = GNUNET_BIO_read_open (fileName);
362   GNUNET_assert (NULL != fileR);
363   metaDataR = NULL;
364   GNUNET_assert (GNUNET_SYSERR ==
365                  GNUNET_BIO_read_meta_data (fileR, "Read meta error",
366                                             &metaDataR));
367   msg = NULL;
368   GNUNET_BIO_read_close (fileR, &msg);
369   GNUNET_free (msg);
370   GNUNET_assert (GNUNET_OK == GNUNET_DISK_directory_remove (fileName));
371   GNUNET_assert (NULL == metaDataR);
372   GNUNET_free (fileName);
373   return 0;
374 }
375
376 static int
377 check_string_rw ()
378 {
379   GNUNET_assert (0 == test_nullstring_rw ());
380   GNUNET_assert (0 == test_emptystring_rw ());
381   GNUNET_assert (0 == test_bigstring_rw ());
382   GNUNET_assert (0 == test_fakestring_rw ());
383   return 0;
384 }
385
386 static int
387 check_metadata_rw ()
388 {
389   GNUNET_assert (0 == test_fakebigmeta_rw ());
390   GNUNET_assert (0 == test_fakemeta_rw ());
391   GNUNET_assert (0 == test_bigmeta_rw ());
392   return 0;
393 }
394
395 static int
396 check_file_rw ()
397 {
398   GNUNET_assert (0 == test_normal_rw ());
399   GNUNET_assert (0 == test_nullfile_rw ());
400   GNUNET_assert (0 == test_directory_r ());
401   return 0;
402 }
403
404 int
405 main (int argc, char *argv[])
406 {
407   GNUNET_log_setup ("test-bio", "WARNING", NULL);
408   GNUNET_assert (0 == check_file_rw ());
409   GNUNET_assert (0 == check_metadata_rw ());
410   GNUNET_assert (0 == check_string_rw ());
411   return 0;
412 }
413
414 /* end of test_bio.c */