d1ef263edb45eacc32137bc5ce624e89a9c20b88
[oweals/gnunet.git] / src / fs / test_fs_directory.c
1 /*
2      This file is part of GNUnet.
3      (C) 2005, 2006, 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 fs/test_fs_directory.c
23  * @brief Test for fs_directory.c
24  * @author Christian Grothoff
25  */
26
27 #include "platform.h"
28 #include <extractor.h>
29 #include "gnunet_util_lib.h"
30 #include "gnunet_fs_service.h"
31 #include "fs.h"
32
33 #define ABORT() { fprintf(stderr, "Error at %s:%d\n", __FILE__, __LINE__); return 1; }
34
35 struct PCLS
36 {
37   struct GNUNET_FS_Uri **uri;
38   struct GNUNET_CONTAINER_MetaData **md;
39   unsigned int pos;
40   unsigned int max;
41 };
42
43 static void
44 processor (void *cls,
45            const char *filename,
46            const struct GNUNET_FS_Uri *uri,
47            const struct GNUNET_CONTAINER_MetaData *md,
48            size_t length,
49            const void *data)
50 {
51   struct PCLS *p = cls;
52   int i;
53
54   if (NULL == uri)
55     return; /* ignore directory's meta data */
56   for (i = 0; i < p->max; i++)
57     {
58       if (GNUNET_CONTAINER_meta_data_test_equal (p->md[i],
59                                                  md) &&
60           GNUNET_FS_uri_test_equal (p->uri[i], uri))
61         {
62           p->pos++;
63           return;
64         }
65     }
66   fprintf (stderr, "Error at %s:%d\n", __FILE__, __LINE__);
67 }
68
69 static int
70 testDirectory (unsigned int i)
71 {
72   struct GNUNET_FS_DirectoryBuilder *db;
73   char *data;
74   size_t dlen;
75   struct GNUNET_FS_Uri **uris;
76   struct GNUNET_CONTAINER_MetaData **mds;
77   struct GNUNET_CONTAINER_MetaData *meta;
78   struct PCLS cls;
79   char *emsg;
80   int p;
81   int q;
82   char uri[512];
83   char txt[128];
84   int ret = 0;
85   struct GNUNET_TIME_Absolute start;
86   char *s;
87
88   cls.max = i;
89   uris = GNUNET_malloc (sizeof (struct GNUNET_FS_Uri*) * i);
90   mds = GNUNET_malloc (sizeof (struct GNUNET_CONTAINER_MetaData*) * i);
91   meta = GNUNET_CONTAINER_meta_data_create ();
92   GNUNET_CONTAINER_meta_data_insert (meta, EXTRACTOR_TITLE, "A title");
93   GNUNET_CONTAINER_meta_data_insert (meta, EXTRACTOR_AUTHOR, "An author");
94   for (p = 0; p < i; p++)
95     {
96       mds[p] = GNUNET_CONTAINER_meta_data_create ();
97       for (q = 0; q <= p; q++)
98         {
99           GNUNET_snprintf (txt, sizeof(txt), "%u -- %u\n", p, q);
100           GNUNET_CONTAINER_meta_data_insert (mds[p],
101                                    q %
102                                    EXTRACTOR_getHighestKeywordTypeNumber
103                                    (), txt);
104         }
105       GNUNET_snprintf (uri,
106                        sizeof(uri),
107                        "gnunet://fs/chk/C282GG70GKK41O4551011DO413KFBVTVMQG1OG30I0K4045N0G41HAPB82G680A02JRVVFO8URVRU2F159011DO41000000022RG820.RNVVVVOOLCLK065B5D04HTNVNSIB2AI022RG8200HSLK1CO1000ATQ98824DMA2032LIMG50CG0K057NVUVG200000H000004400000.%u",
108                        p);
109       emsg = NULL;
110       uris[p] = GNUNET_FS_uri_parse (uri, &emsg);
111       if (uris[p] == NULL)
112         {
113           GNUNET_CONTAINER_meta_data_destroy (mds[p]);
114           while (--p > 0)
115             {
116               GNUNET_CONTAINER_meta_data_destroy (mds[p]);
117               GNUNET_FS_uri_destroy (uris[p]);
118             }
119           GNUNET_free (mds);
120           GNUNET_free (uris);
121           GNUNET_free (emsg);
122           GNUNET_CONTAINER_meta_data_destroy (meta);
123           ABORT ();             /* error in testcase */
124         }
125       GNUNET_assert (emsg == NULL);
126     }
127   start = GNUNET_TIME_absolute_get ();
128   db = GNUNET_FS_directory_builder_create (meta);
129   for (p = 0; p < i; p++)
130     GNUNET_FS_directory_builder_add (db, uris[p], mds[p], NULL);
131   GNUNET_FS_directory_builder_finish (db,
132                                       &dlen,
133                                       (void**) &data);
134   s = GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_duration (start));
135   fprintf (stdout,
136            "Creating directory with %u entires and total size %llu took %s\n",
137            i,
138            (unsigned long long) dlen,
139            s);
140   GNUNET_free (s);
141   if (i < 100)
142     {
143       cls.pos = 0;
144       cls.uri = uris;
145       cls.md = mds;
146       GNUNET_FS_directory_list_contents (dlen, data, 0, 
147                                          &processor, &cls);
148       GNUNET_assert (cls.pos == i);
149     }
150   GNUNET_free (data);
151   GNUNET_CONTAINER_meta_data_destroy (meta);
152   for (p = 0; p < i; p++)
153     {
154       GNUNET_CONTAINER_meta_data_destroy (mds[p]);
155       GNUNET_FS_uri_destroy (uris[p]);
156     }
157   GNUNET_free (uris);
158   GNUNET_free (mds);
159   return ret;
160 }
161
162
163 int
164 main (int argc, char *argv[])
165 {
166   int failureCount = 0;
167   int i;
168
169   GNUNET_log_setup ("test_fs_directory", 
170 #if VERBOSE
171                     "DEBUG",
172 #else
173                     "WARNING",
174 #endif
175                     NULL);
176   for (i = 17; i < 1000; i *= 2)
177     failureCount += testDirectory (i);    
178   if (failureCount != 0)
179     return 1;
180   return 0;
181 }
182
183 /* end of test_fs_directory.c */