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      SPDX-License-Identifier: AGPL3.0-or-later
  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   for (unsigned int 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 %u: %s\n",
  48                  i,
  49                  GNUNET_i2s(&pidArr[i]));
  50     }
  51 }
  52
  53
  54 static int
  55 check()
  56 {
  57   GNUNET_PEER_Id pid;
  58   struct GNUNET_PeerIdentity res;
  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 (unsigned int 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 (unsigned int 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 (unsigned int 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,
  96                       &res);
  97   GNUNET_assert(0 ==
  98                 GNUNET_memcmp(&res,
  99                               &pidArr[0]));
 100
 101   /*
 102    * Attempt to convert pid = 0 (which is reserved)
 103    * into a peer identity object, the peer identity memory
 104    * is expected to be set to zero
 105    */
 106   GNUNET_log_skip(1, GNUNET_YES);
 107   GNUNET_PEER_resolve(0, &res);
 108   GNUNET_assert(0 ==
 109                 GNUNET_is_zero(&res));
 110
 111   /* Removing peer entries 1 and 3 from table using the list decrement function */
 112   /* If count = 0, nothing should be done whatsoever */
 113   GNUNET_PEER_decrement_rcs(ids, 0);
 114
 115   ids[1] = 3;
 116   GNUNET_PEER_decrement_rcs(ids, 2);
 117   GNUNET_PEER_decrement_rcs(ids, 2);
 118
 119   return 0;
 120 }
 121
 122
 123 int
 124 main()
 125 {
 126   GNUNET_log_setup("test-peer",
 127                    "ERROR",
 128                    NULL);
 129   for (unsigned int i = 0; i < 1; i++)
 130     {
 131       generatePeerIdList();
 132       if (0 != check())
 133         return 1;
 134     }
 135   return 0;
 136 }
 137
 138 /* end of test_peer.c */