arg
[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 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., 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, 
93                                      "<test>",
94                                      EXTRACTOR_METATYPE_TITLE,
95                                      EXTRACTOR_METAFORMAT_UTF8,
96                                      "text/plain",
97                                      "A title",
98                                      strlen("A title")+1);
99   GNUNET_CONTAINER_meta_data_insert (meta, 
100                                      "<test>",
101                                      EXTRACTOR_METATYPE_AUTHOR_NAME, 
102                                      EXTRACTOR_METAFORMAT_UTF8,
103                                      "text/plain",
104                                      "An author",
105                                      strlen ("An author")+1);
106   for (p = 0; p < i; p++)
107     {
108       mds[p] = GNUNET_CONTAINER_meta_data_create ();
109       for (q = 0; q <= p; q++)
110         {
111           GNUNET_snprintf (txt, sizeof(txt), "%u -- %u\n", p, q);
112           GNUNET_CONTAINER_meta_data_insert (mds[p],
113                                              "<test>",
114                                              q % EXTRACTOR_metatype_get_max (),
115                                              EXTRACTOR_METAFORMAT_UTF8,
116                                              "text/plain",
117                                              txt,
118                                              strlen(txt)+1);
119         }
120       GNUNET_snprintf (uri,
121                        sizeof(uri),
122                        "gnunet://fs/chk/C282GG70GKK41O4551011DO413KFBVTVMQG1OG30I0K4045N0G41HAPB82G680A02JRVVFO8URVRU2F159011DO41000000022RG820.RNVVVVOOLCLK065B5D04HTNVNSIB2AI022RG8200HSLK1CO1000ATQ98824DMA2032LIMG50CG0K057NVUVG200000H000004400000.%u",
123                        p);
124       emsg = NULL;
125       uris[p] = GNUNET_FS_uri_parse (uri, &emsg);
126       if (uris[p] == NULL)
127         {
128           GNUNET_CONTAINER_meta_data_destroy (mds[p]);
129           while (--p > 0)
130             {
131               GNUNET_CONTAINER_meta_data_destroy (mds[p]);
132               GNUNET_FS_uri_destroy (uris[p]);
133             }
134           GNUNET_free (mds);
135           GNUNET_free (uris);
136           GNUNET_free (emsg);
137           GNUNET_CONTAINER_meta_data_destroy (meta);
138           ABORT ();             /* error in testcase */
139         }
140       GNUNET_assert (emsg == NULL);
141     }
142   start = GNUNET_TIME_absolute_get ();
143   db = GNUNET_FS_directory_builder_create (meta);
144   for (p = 0; p < i; p++)
145     GNUNET_FS_directory_builder_add (db, uris[p], mds[p], NULL);
146   GNUNET_FS_directory_builder_finish (db,
147                                       &dlen,
148                                       (void**) &data);
149   s = GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_duration (start));
150   fprintf (stdout,
151            "Creating directory with %u entires and total size %llu took %s\n",
152            i,
153            (unsigned long long) dlen,
154            s);
155   GNUNET_free (s);
156   if (i < 100)
157     {
158       cls.pos = 0;
159       cls.uri = uris;
160       cls.md = mds;
161       GNUNET_FS_directory_list_contents (dlen, data, 0, 
162                                          &processor, &cls);
163       GNUNET_assert (cls.pos == i);
164     }
165   GNUNET_free (data);
166   GNUNET_CONTAINER_meta_data_destroy (meta);
167   for (p = 0; p < i; p++)
168     {
169       GNUNET_CONTAINER_meta_data_destroy (mds[p]);
170       GNUNET_FS_uri_destroy (uris[p]);
171     }
172   GNUNET_free (uris);
173   GNUNET_free (mds);
174   return ret;
175 }
176
177
178 int
179 main (int argc, char *argv[])
180 {
181   int failureCount = 0;
182   int i;
183
184   GNUNET_log_setup ("test_fs_directory", 
185 #if VERBOSE
186                     "DEBUG",
187 #else
188                     "WARNING",
189 #endif
190                     NULL);
191   for (i = 17; i < 1000; i *= 2)
192     failureCount += testDirectory (i);    
193   if (failureCount != 0)
194     return 1;
195   return 0;
196 }
197
198 /* end of test_fs_directory.c */