src/set/gnunet-set.c

   1 /*
   2       This file is part of GNUnet
   3       (C) 2012 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 2, 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 set/gnunet-set.c
  23  * @brief profiling tool for the set service
  24  * @author Florian Dold
  25  */
  26 #include "platform.h"
  27 #include "gnunet_common.h"
  28 #include "gnunet_util_lib.h"
  29 #include "gnunet_testbed_service.h"
  30 #include "gnunet_set_service.h"
  31
  32
  33 static struct GNUNET_PeerIdentity local_id;
  34
  35 static struct GNUNET_HashCode app_id;
  36 static struct GNUNET_SET_Handle *set1;
  37 static struct GNUNET_SET_Handle *set2;
  38 static struct GNUNET_SET_ListenHandle *listen_handle;
  39 const static struct GNUNET_CONFIGURATION_Handle *config;
  40
  41 int num_done;
  42
  43
  44 static void
  45 result_cb_set1 (void *cls, struct GNUNET_SET_Element *element,
  46            enum GNUNET_SET_Status status)
  47 {
  48   switch (status)
  49   {
  50     case GNUNET_SET_STATUS_OK:
  51       printf ("set 1: got element\n");
  52       break;
  53     case GNUNET_SET_STATUS_FAILURE:
  54       printf ("set 1: failure\n");
  55       break;
  56     case GNUNET_SET_STATUS_DONE:
  57       printf ("set 1: done\n");
  58       GNUNET_SET_destroy (set1);
  59       break;
  60     default:
  61       GNUNET_assert (0);
  62   }
  63 }
  64
  65
  66 static void
  67 result_cb_set2 (void *cls, struct GNUNET_SET_Element *element,
  68            enum GNUNET_SET_Status status)
  69 {
  70   switch (status)
  71   {
  72     case GNUNET_SET_STATUS_OK:
  73       printf ("set 2: got element\n");
  74       break;
  75     case GNUNET_SET_STATUS_FAILURE:
  76       printf ("set 2: failure\n");
  77       break;
  78     case GNUNET_SET_STATUS_DONE:
  79       printf ("set 2: done\n");
  80       GNUNET_SET_destroy (set2);
  81       break;
  82     default:
  83       GNUNET_assert (0);
  84   }
  85 }
  86
  87
  88 static void
  89 listen_cb (void *cls,
  90            const struct GNUNET_PeerIdentity *other_peer,
  91            const struct GNUNET_MessageHeader *context_msg,
  92            struct GNUNET_SET_Request *request)
  93 {
  94   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "listen cb called\n");
  95   GNUNET_SET_listen_cancel (listen_handle);
  96
  97   GNUNET_SET_accept (request, set2, GNUNET_TIME_UNIT_FOREVER_REL,
  98                      GNUNET_SET_RESULT_ADDED, result_cb_set2, NULL);
  99 }
 100
 101
 102 /**
 103  * Start the set operation.
 104  *
 105  * @param cls closure, unused
 106  */
 107 static void
 108 start (void *cls)
 109 {
 110   listen_handle = GNUNET_SET_listen (config, GNUNET_SET_OPERATION_UNION,
 111                                      &app_id, listen_cb, NULL);
 112   GNUNET_SET_evaluate (set1, &local_id, &app_id, NULL, 42,
 113                        GNUNET_TIME_UNIT_FOREVER_REL, GNUNET_SET_RESULT_ADDED,
 114                        result_cb_set1, NULL);
 115 }
 116
 117
 118 /**
 119  * Initialize the second set, continue
 120  *
 121  * @param cls closure, unused
 122  */
 123 static void
 124 init_set2 (void *cls)
 125 {
 126   struct GNUNET_SET_Element element;
 127
 128
 129   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "initializing set 2\n");
 130
 131   element.data = "hello";
 132   element.size = strlen(element.data);
 133   GNUNET_SET_add_element (set2, &element, NULL, NULL);
 134   element.data = "quux";
 135   element.size = strlen(element.data);
 136   GNUNET_SET_add_element (set2, &element, start, NULL);
 137 }
 138
 139
 140 /**
 141  * Initialize the first set, continue.
 142  */
 143 static void
 144 init_set1 (void)
 145 {
 146   struct GNUNET_SET_Element element;
 147
 148   element.data = "hello";
 149   element.size = strlen(element.data);
 150   GNUNET_SET_add_element (set1, &element, NULL, NULL);
 151   element.data = "bar";
 152   element.size = strlen(element.data);
 153   GNUNET_SET_add_element (set1, &element, init_set2, NULL);
 154
 155   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "initialized set 1\n");
 156 }
 157
 158
 159 /**
 160  * Main function that will be run.
 161  *
 162  * @param cls closure
 163  * @param args remaining command-line arguments
 164  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 165  * @param cfg configuration
 166  */
 167 static void
 168 run (void *cls, char *const *args,
 169      const char *cfgfile,
 170      const struct GNUNET_CONFIGURATION_Handle *cfg)
 171 {
 172   static const char* app_str = "gnunet-set";
 173
 174   config = cfg;
 175
 176   GNUNET_CRYPTO_hash (app_str, strlen (app_str), &app_id);
 177
 178   GNUNET_CRYPTO_get_host_identity (cfg, &local_id);
 179
 180   set1 = GNUNET_SET_create (cfg, GNUNET_SET_OPERATION_UNION);
 181   set2 = GNUNET_SET_create (cfg, GNUNET_SET_OPERATION_UNION);
 182
 183   init_set1 ();
 184 }
 185
 186
 187
 188 int
 189 main (int argc, char **argv)
 190 {
 191    static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 192       GNUNET_GETOPT_OPTION_END
 193   };
 194   GNUNET_PROGRAM_run2 (argc, argv, "gnunet-set",
 195                       "help",
 196                       options, &run, NULL, GNUNET_NO);
 197   return 0;
 198 }
 199