src/scalarproduct/gnunet-scalarproduct.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2013, 2014 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 /**
  22  * @file scalarproduct/gnunet-scalarproduct.c
  23  * @brief scalarproduct client
  24  * @author Christian M. Fuchs
  25  */
  26 #define GCRYPT_NO_DEPRECATED
  27 #include <gcrypt.h>
  28 #include <inttypes.h>
  29
  30 #include "platform.h"
  31 #include "gnunet_util_lib.h"
  32 #include "gnunet_scalarproduct_service.h"
  33 #include "gnunet_protocols.h"
  34 #include "scalarproduct.h"
  35
  36 #define LOG(kind,...) GNUNET_log_from (kind, "gnunet-scalarproduct",__VA_ARGS__)
  37
  38
  39 /**
  40  * the session key identifying this computation
  41  */
  42 static struct GNUNET_HashCode session_key;
  43
  44 /**
  45  * PeerID we want to compute a scalar product with
  46  */
  47 static struct GNUNET_PeerIdentity peer_id;
  48
  49 /**
  50  * Option -p: destination peer identity for checking message-ids with
  51  */
  52 static char *input_peer_id;
  53
  54 /**
  55  * Option -p: destination peer identity for checking message-ids with
  56  */
  57 static char *input_session_key;
  58
  59 /**
  60  * Option -e: vector to calculate a scalarproduct with
  61  */
  62 static char *input_elements;
  63
  64 /**
  65  * Global return value
  66  */
  67 static int ret = -1;
  68
  69 /**
  70  * our Scalarproduct Computation handle
  71  */
  72 static struct GNUNET_SCALARPRODUCT_ComputationHandle *computation;
  73
  74
  75 /**
  76  * Callback called if we are initiating a new computation session
  77  *
  78  * @param cls unused
  79  * @param status if our job was successfully processed
  80  */
  81 static void
  82 responder_callback (void *cls,
  83                     enum GNUNET_SCALARPRODUCT_ResponseStatus status)
  84 {
  85   switch (status)
  86   {
  87   case GNUNET_SCALARPRODUCT_STATUS_SUCCESS:
  88     ret = 0;
  89     LOG (GNUNET_ERROR_TYPE_INFO,
  90          "Session %s concluded.\n",
  91          GNUNET_h2s (&session_key));
  92     break;
  93   case GNUNET_SCALARPRODUCT_STATUS_INVALID_RESPONSE:
  94     LOG (GNUNET_ERROR_TYPE_ERROR,
  95          "Session %s failed: invalid response\n",
  96          GNUNET_h2s (&session_key));
  97     break;
  98   case GNUNET_SCALARPRODUCT_STATUS_FAILURE:
  99     LOG (GNUNET_ERROR_TYPE_ERROR,
 100          "Session %s failed: service failure\n",
 101          GNUNET_h2s (&session_key));
 102     break;
 103   case GNUNET_SCALARPRODUCT_STATUS_DISCONNECTED:
 104     LOG (GNUNET_ERROR_TYPE_ERROR,
 105          "Session %s failed: service disconnect!\n",
 106          GNUNET_h2s (&session_key));
 107     break;
 108   default:
 109     LOG (GNUNET_ERROR_TYPE_ERROR,
 110          "Session %s failed: return code %d\n",
 111          GNUNET_h2s (&session_key),
 112          status);
 113   }
 114   computation = NULL;
 115   GNUNET_SCHEDULER_shutdown ();
 116 }
 117
 118
 119 /**
 120  * Callback called if we are initiating a new computation session
 121  *
 122  * @param cls unused
 123  * @param status if our job was successfully processed
 124  * @param result the result in gnu/gcry MPI format
 125  */
 126 static void
 127 requester_callback (void *cls,
 128                     enum GNUNET_SCALARPRODUCT_ResponseStatus status,
 129                     gcry_mpi_t result)
 130 {
 131   unsigned char *buf;
 132   gcry_error_t rc;
 133
 134   switch (status)
 135   {
 136   case GNUNET_SCALARPRODUCT_STATUS_SUCCESS:
 137     if (0 == (rc = gcry_mpi_aprint (GCRYMPI_FMT_HEX, &buf, NULL, result)))
 138     {
 139       ret = 0;
 140       fprintf (stdout,
 141                "%s%s\n",
 142                (0 > gcry_mpi_cmp_ui (result, 0)) ? "-" : "",
 143                buf);
 144       fflush (stdout);
 145     }
 146     else
 147       LOG_GCRY (GNUNET_ERROR_TYPE_ERROR,
 148                 "gcry_mpi_aprint",
 149                 rc);
 150     break;
 151   case GNUNET_SCALARPRODUCT_STATUS_INVALID_RESPONSE:
 152     LOG (GNUNET_ERROR_TYPE_ERROR,
 153          "Session %s with peer %s failed: invalid response received\n",
 154          GNUNET_h2s (&session_key),
 155          GNUNET_i2s (&peer_id));
 156     break;
 157   case GNUNET_SCALARPRODUCT_STATUS_FAILURE:
 158     LOG (GNUNET_ERROR_TYPE_ERROR,
 159          "Session %s with peer %s failed: API failure\n",
 160          GNUNET_h2s (&session_key),
 161          GNUNET_i2s (&peer_id));
 162     break;
 163   case GNUNET_SCALARPRODUCT_STATUS_DISCONNECTED:
 164     LOG (GNUNET_ERROR_TYPE_ERROR,
 165          "Session %s with peer %s was disconnected from service.\n",
 166          GNUNET_h2s (&session_key),
 167          GNUNET_i2s (&peer_id));
 168     break;
 169   default:
 170     LOG (GNUNET_ERROR_TYPE_ERROR,
 171          "Session %s with peer %s failed: return code %d\n",
 172          GNUNET_h2s (&session_key),
 173          GNUNET_i2s (&peer_id),
 174          status);
 175   }
 176   computation = NULL;
 177   GNUNET_SCHEDULER_shutdown ();
 178 }
 179
 180
 181 /**
 182  * Task run during shutdown.
 183  *
 184  * @param cls unused
 185  * @param tc unused
 186  */
 187 static void
 188 shutdown_task (void *cls,
 189                const struct GNUNET_SCHEDULER_TaskContext *tc)
 190 {
 191   if (NULL != computation)
 192   {
 193     GNUNET_SCALARPRODUCT_cancel (computation);
 194     ret = 1; /* aborted */
 195   }
 196 }
 197
 198
 199 /**
 200  * Main function that will be run by the scheduler.
 201  *
 202  * @param cls closure
 203  * @param args remaining command-line arguments
 204  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 205  * @param cfg configuration
 206  */
 207 static void
 208 run (void *cls,
 209      char *const *args,
 210      const char *cfgfile,
 211      const struct GNUNET_CONFIGURATION_Handle *cfg)
 212 {
 213   char *begin = input_elements;
 214   char *end;
 215   unsigned int i;
 216   struct GNUNET_SCALARPRODUCT_Element *elements;
 217   uint32_t element_count = 0;
 218
 219   if (NULL == input_elements)
 220   {
 221     LOG (GNUNET_ERROR_TYPE_ERROR,
 222          _("You must specify at least one message ID to check!\n"));
 223     return;
 224   }
 225   if ( (NULL == input_session_key) ||
 226        (0 == strlen (input_session_key)) )
 227   {
 228     LOG (GNUNET_ERROR_TYPE_ERROR,
 229          _("This program needs a session identifier for comparing vectors.\n"));
 230     return;
 231   }
 232   GNUNET_CRYPTO_hash (input_session_key,
 233                       strlen (input_session_key),
 234                       &session_key);
 235   if ( (NULL != input_peer_id) &&
 236        (GNUNET_OK !=
 237         GNUNET_CRYPTO_eddsa_public_key_from_string (input_peer_id,
 238                                                     strlen (input_peer_id),
 239                                                     &peer_id.public_key)) )
 240   {
 241     LOG (GNUNET_ERROR_TYPE_ERROR,
 242          _("Tried to set initiator mode, as peer ID was given. "
 243            "However, `%s' is not a valid peer identifier.\n"),
 244          input_peer_id);
 245     return;
 246   }
 247
 248   for (end = begin; 0 != *end; end++)
 249     if (*end == ';')
 250       element_count++;
 251   if (0 == element_count) {
 252     LOG (GNUNET_ERROR_TYPE_ERROR,
 253          _("Need elements to compute the vectorproduct, got none.\n"));
 254     return;
 255   }
 256
 257   elements = GNUNET_malloc (sizeof(struct GNUNET_SCALARPRODUCT_Element) * element_count);
 258
 259   for (i = 0; i < element_count;i++)
 260   {
 261     struct GNUNET_SCALARPRODUCT_Element element;
 262     char* separator = NULL;
 263
 264     /* get the length of the current key,value; tupel */
 265     for (end = begin; *end != ';'; end++)
 266       if (*end == ',')
 267         separator = end;
 268
 269     /* final element */
 270     if ( (NULL == separator) ||
 271          (begin == separator) ||
 272          (separator == end - 1) )
 273     {
 274       LOG (GNUNET_ERROR_TYPE_ERROR,
 275            _("Malformed input, could not parse `%s'\n"),
 276            begin);
 277       GNUNET_free (elements);
 278       return;
 279     }
 280
 281     /* read the element's key */
 282     *separator = 0;
 283     GNUNET_CRYPTO_hash (begin,
 284                         strlen (begin),
 285                         &element.key);
 286
 287     /* read the element's value */
 288     if (1 !=
 289         sscanf (separator + 1,
 290                 "%" SCNd64 ";",
 291                 &element.value) )
 292     {
 293       LOG (GNUNET_ERROR_TYPE_ERROR,
 294            _("Could not convert `%s' to int64_t.\n"),
 295            begin);
 296       GNUNET_free (elements);
 297       return;
 298     }
 299     element.value = GNUNET_htonll (element.value);
 300     elements[i] = element;
 301     begin = end + 1;
 302   }
 303
 304   if ( ( (NULL != input_peer_id) &&
 305          (NULL == (computation
 306                    = GNUNET_SCALARPRODUCT_start_computation (cfg,
 307                                                              &session_key,
 308                                                              &peer_id,
 309                                                              elements, element_count,
 310                                                              &requester_callback,
 311                                                              NULL))) ) ||
 312        ( (NULL == input_peer_id) &&
 313          (NULL == (computation
 314                    = GNUNET_SCALARPRODUCT_accept_computation (cfg,
 315                                                               &session_key,
 316                                                               elements, element_count,
 317                                                               &responder_callback,
 318                                                               NULL))) ) )
 319   {
 320     GNUNET_break (0);
 321     GNUNET_free (elements);
 322     return;
 323   }
 324   GNUNET_free (elements);
 325   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
 326                                 &shutdown_task,
 327                                 NULL);
 328   ret = 0;
 329 }
 330
 331
 332 /**
 333  * The main function to the scalarproduct client.
 334  *
 335  * @param argc number of arguments from the command line
 336  * @param argv command line arguments
 337  * @return 0 ok, 1 on error
 338  */
 339 int
 340 main (int argc, char *const *argv)
 341 {
 342   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 343     {'e', "elements", "\"key1,val1;key2,val2;...,keyn,valn;\"",
 344       gettext_noop ("A comma separated list of elements to compare as vector with our remote peer."),
 345       1, &GNUNET_GETOPT_set_string, &input_elements},
 346     {'p', "peer", "PEERID",
 347       gettext_noop ("[Optional] peer to calculate our scalarproduct with. If this parameter is not given, the service will wait for a remote peer to compute the request."),
 348       1, &GNUNET_GETOPT_set_string, &input_peer_id},
 349     {'k', "key", "TRANSACTION_ID",
 350       gettext_noop ("Transaction ID shared with peer."),
 351       1, &GNUNET_GETOPT_set_string, &input_session_key},
 352     GNUNET_GETOPT_OPTION_END
 353   };
 354
 355   return (GNUNET_OK ==
 356           GNUNET_PROGRAM_run (argc,
 357                               argv,
 358                               "gnunet-scalarproduct",
 359                               gettext_noop ("Calculate the Vectorproduct with a GNUnet peer."),
 360                               options, &run, NULL)) ? ret : 1;
 361 }
 362
 363 /* end of gnunet-scalarproduct.c */