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