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, const 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, const 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   struct GNUNET_SET_OperationHandle *oh;
  95   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "listen cb called\n");
  96   GNUNET_SET_listen_cancel (listen_handle);
  97
  98   oh = GNUNET_SET_accept (request, GNUNET_SET_RESULT_ADDED, result_cb_set2, NULL);
  99   GNUNET_SET_conclude (oh, set2);
 100 }
 101
 102
 103 /**
 104  * Start the set operation.
 105  *
 106  * @param cls closure, unused
 107  */
 108 static void
 109 start (void *cls)
 110 {
 111   struct GNUNET_SET_OperationHandle *oh;
 112
 113   listen_handle = GNUNET_SET_listen (config, GNUNET_SET_OPERATION_UNION,
 114                                      &app_id, listen_cb, NULL);
 115   oh = GNUNET_SET_evaluate (&local_id, &app_id, NULL, 42,
 116                             GNUNET_SET_RESULT_ADDED,
 117                             result_cb_set1, NULL);
 118   GNUNET_SET_conclude (oh, set1);
 119 }
 120
 121
 122 /**
 123  * Initialize the second set, continue
 124  *
 125  * @param cls closure, unused
 126  */
 127 static void
 128 init_set2 (void *cls)
 129 {
 130   struct GNUNET_SET_Element element;
 131
 132
 133   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "initializing set 2\n");
 134
 135   element.data = "hello";
 136   element.size = strlen(element.data);
 137   GNUNET_SET_add_element (set2, &element, NULL, NULL);
 138   element.data = "quux";
 139   element.size = strlen(element.data);
 140   GNUNET_SET_add_element (set2, &element, start, NULL);
 141 }
 142
 143
 144 /**
 145  * Initialize the first set, continue.
 146  */
 147 static void
 148 init_set1 (void)
 149 {
 150   struct GNUNET_SET_Element element;
 151
 152   element.data = "hello";
 153   element.size = strlen(element.data);
 154   GNUNET_SET_add_element (set1, &element, NULL, NULL);
 155   element.data = "bar";
 156   element.size = strlen(element.data);
 157   GNUNET_SET_add_element (set1, &element, init_set2, NULL);
 158
 159   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "initialized set 1\n");
 160 }
 161
 162
 163 /**
 164  * Main function that will be run.
 165  *
 166  * @param cls closure
 167  * @param args remaining command-line arguments
 168  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 169  * @param cfg configuration
 170  */
 171 static void
 172 run (void *cls, char *const *args,
 173      const char *cfgfile,
 174      const struct GNUNET_CONFIGURATION_Handle *cfg)
 175 {
 176   static const char* app_str = "gnunet-set";
 177
 178   config = cfg;
 179
 180   GNUNET_CRYPTO_hash (app_str, strlen (app_str), &app_id);
 181
 182   GNUNET_CRYPTO_get_host_identity (cfg, &local_id);
 183
 184   set1 = GNUNET_SET_create (cfg, GNUNET_SET_OPERATION_UNION);
 185   set2 = GNUNET_SET_create (cfg, GNUNET_SET_OPERATION_UNION);
 186
 187   init_set1 ();
 188 }
 189
 190
 191
 192 int
 193 main (int argc, char **argv)
 194 {
 195    static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 196       GNUNET_GETOPT_OPTION_END
 197   };
 198   GNUNET_PROGRAM_run2 (argc, argv, "gnunet-set",
 199                       "help",
 200                       options, &run, NULL, GNUNET_NO);
 201   return 0;
 202 }
 203