Merge branch 'master' of ssh://gnunet.org/gnunet
[oweals/gnunet.git] / src / util / test_peer.c
1 /*
2      This file is part of GNUnet.
3      Copyright (C) 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  * @file util/test_peer.c
22  * @brief testcase for peer.c
23  * @author Safey Mohammed
24  */
25
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28 #include <gcrypt.h>
29
30 #define NUMBER_OF_PEERS 10
31
32 /**
33  * A list of Peer ID's to play with
34  */
35 static struct GNUNET_PeerIdentity pidArr[NUMBER_OF_PEERS];
36
37
38 static void
39 generatePeerIdList ()
40 {
41   int i;
42
43   for (i = 0; i < NUMBER_OF_PEERS; i++)
44   {
45     gcry_randomize (&pidArr[i],
46                     sizeof (struct GNUNET_PeerIdentity),
47                     GCRY_STRONG_RANDOM);
48     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
49                 "Peer %d: %s\n", i, GNUNET_i2s (&pidArr[i]));
50   }
51 }
52
53
54 static int
55 check ()
56 {
57   int i;
58   GNUNET_PEER_Id pid;
59   struct GNUNET_PeerIdentity res;
60   struct GNUNET_PeerIdentity zero;
61   GNUNET_PEER_Id ids[] = { 1, 2, 3 };
62
63   GNUNET_assert (0 == GNUNET_PEER_intern (NULL));
64   /* Insert Peers into PeerEntry table and hashmap */
65   for (i = 0; i < NUMBER_OF_PEERS; i++)
66   {
67     pid = GNUNET_PEER_intern (&pidArr[i]);
68     if (pid != (i + 1))
69     {
70       FPRINTF (stderr, "%s",  "Unexpected Peer ID returned by intern function\n");
71       return 1;
72     }
73   }
74
75   /* Referencing the first 3 peers once again */
76   for (i = 0; i < 3; i++)
77   {
78     pid = GNUNET_PEER_intern (&pidArr[i]);
79     if (pid != (i + 1))
80     {
81       FPRINTF (stderr, "%s",  "Unexpected Peer ID returned by intern function\n");
82       return 1;
83     }
84   }
85
86   /* Dereferencing the first 3 peers once [decrementing their reference count] */
87   GNUNET_PEER_decrement_rcs (ids, 3);
88
89   /* re-referencing the first 3 peers using the change_rc function */
90   for (i = 1; i <= 3; i++)
91     GNUNET_PEER_change_rc (i, 1);
92
93   /* Removing the second Peer from the PeerEntry hash map */
94   GNUNET_PEER_change_rc (2, -2);
95
96   /* convert the pid of the first PeerEntry into that of the third */
97   GNUNET_PEER_resolve (1, &res);
98   GNUNET_assert (0 == memcmp (&res, &pidArr[0], sizeof (res)));
99
100   /*
101    * Attempt to convert pid = 0 (which is reserved)
102    * into a peer identity object, the peer identity memory
103    * is expected to be set to zero
104    */
105   memset (&zero, 0, sizeof (struct GNUNET_PeerIdentity));
106   GNUNET_log_skip (1, GNUNET_YES);
107   GNUNET_PEER_resolve (0, &res);
108   GNUNET_assert (0 == memcmp (&res, &zero, sizeof (res)));
109
110   /* Removing peer entries 1 and 3 from table using the list decrement function */
111   /* If count = 0, nothing should be done whatsoever */
112   GNUNET_PEER_decrement_rcs (ids, 0);
113
114   ids[1] = 3;
115   GNUNET_PEER_decrement_rcs (ids, 2);
116   GNUNET_PEER_decrement_rcs (ids, 2);
117
118   return 0;
119 }
120
121
122 int
123 main ()
124 {
125   unsigned int i;
126
127   GNUNET_log_setup ("test-peer", "ERROR", NULL);
128   for (i = 0; i < 1; i++)
129   {
130     generatePeerIdList ();
131     if (0 != check ())
132       return 1;
133   }
134   return 0;
135 }
136
137 /* end of test_peer.c */