src/ats/perf_ats_mlp.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2010,2011 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  * @file ats/test_ats_mlp.c
  22  * @brief test for the MLP solver
  23  * @author Christian Grothoff
  24  * @author Matthias Wachs
  25
  26  */
  27 #include "platform.h"
  28 #include "gnunet_util_lib.h"
  29 #include "gnunet_statistics_service.h"
  30 #include "gnunet-service-ats_addresses_mlp.h"
  31
  32 #define VERBOSE GNUNET_YES
  33 #define VERBOSE_ARM GNUNET_NO
  34
  35 #define MLP_MAX_EXEC_DURATION   GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 3)
  36 #define MLP_MAX_ITERATIONS      INT_MAX
  37
  38 #define DEF_PEERS 10
  39 #define DEF_ADDRESSES_PER_PEER 5
  40
  41 static unsigned int peers;
  42 static unsigned int addresses;
  43
  44 static int ret;
  45
  46 struct GNUNET_STATISTICS_Handle * stats;
  47
  48 struct GNUNET_CONTAINER_MultiHashMap * amap;
  49
  50 struct GAS_MLP_Handle *mlp;
  51
  52 struct PeerContext
  53 {
  54   struct GNUNET_PeerIdentity id;
  55
  56   struct Address *addr;
  57 };
  58
  59 struct Address
  60 {
  61   char *plugin;
  62   size_t plugin_len;
  63
  64   void *addr;
  65   size_t addr_len;
  66
  67   struct GNUNET_ATS_Information *ats;
  68   int ats_count;
  69
  70   void *session;
  71 };
  72
  73 static void
  74 check (void *cls, char *const *args, const char *cfgfile,
  75        const struct GNUNET_CONFIGURATION_Handle *cfg)
  76 {
  77 #if !HAVE_LIBGLPK
  78   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "GLPK not installed!");
  79   ret = 1;
  80   return;
  81 #endif
  82   unsigned int c = 0;
  83   unsigned int c2 = 0;
  84   unsigned int ca = 0;
  85   //char * pid;
  86
  87   if (peers == 0)
  88     peers = DEF_PEERS;
  89   if (addresses == 0)
  90     addresses = DEF_ADDRESSES_PER_PEER;
  91
  92   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Setting up %u peers with %u addresses per peer\n", peers, addresses);
  93
  94   struct PeerContext p[peers];
  95   struct ATS_Address a[addresses * peers];
  96
  97   amap = GNUNET_CONTAINER_multihashmap_create(addresses * peers);
  98
  99   mlp = GAS_mlp_init (cfg, NULL, MLP_MAX_EXEC_DURATION, MLP_MAX_ITERATIONS);
 100   mlp->auto_solve = GNUNET_NO;
 101   for (c=0; c < peers; c++)
 102   {
 103     GNUNET_CRYPTO_hash_create_random(GNUNET_CRYPTO_QUALITY_WEAK, &p[c].id.hashPubKey);
 104     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "peer %s\n", GNUNET_h2s_full(&p[c].id.hashPubKey));
 105
 106     for (c2=0; c2 < addresses; c2++)
 107     {
 108       a[ca].peer = p[c].id;
 109       a[ca].plugin = strdup("test");
 110       a[ca].addr = GNUNET_HELLO_address_allocate(&a[ca].peer, a[ca].plugin, NULL, 0);
 111       a[ca].addr_len = GNUNET_HELLO_address_get_size(a[ca].addr);
 112       a[ca].ats = NULL;
 113       ca++;
 114       GAS_mlp_address_update(mlp, amap, &a[c2]);
 115     }
 116   }
 117
 118   GAS_mlp_solve_problem(mlp);
 119
 120
 121   GAS_mlp_done (mlp);
 122
 123   for (c2=0; c2 < (peers * addresses); c2++)
 124   {
 125     GNUNET_free (a[c2].plugin);
 126     GNUNET_free (a[c2].addr);
 127 //      GAS_mlp_address_update(mlp, amap, &a[c2]);
 128   }
 129
 130   ret = 0;
 131   return;
 132 }
 133
 134
 135 int
 136 main (int argc, char *argv[])
 137 {
 138
 139   static char *const argv2[] = { "test_ats_mlp",
 140     "-c",
 141     "test_ats_api.conf",
 142 #if VERBOSE
 143     "-L", "DEBUG",
 144 #else
 145     "-L", "WARNING",
 146 #endif
 147     NULL
 148   };
 149
 150   static struct GNUNET_GETOPT_CommandLineOption options[] = {
 151     {'a', "addresses", NULL,
 152      gettext_noop ("addresses per peer"), 1,
 153      &GNUNET_GETOPT_set_uint, &addresses},
 154     {'p', "peers", NULL,
 155      gettext_noop ("peers"), 1,
 156      &GNUNET_GETOPT_set_uint, &peers},
 157     GNUNET_GETOPT_OPTION_END
 158   };
 159
 160   GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1, argv2,
 161                       "test_ats_mlp", "nohelp", options,
 162                       &check, NULL);
 163
 164
 165   return ret;
 166 }
 167
 168 /* end of file test_ats_api_bandwidth_consumption.c */