src/sensor/gnunet-sensor.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C)
   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., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, USA.
  19 */
  20
  21 /**
  22  * @file sensor/gnunet-sensor.c
  23  * @brief sensor tool
  24  * @author Omar Tarabai
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28 #include "gnunet_sensor_service.h"
  29
  30 static int ret;
  31
  32 /**
  33  * option '-a'
  34  */
  35 static int get_all;
  36
  37 /**
  38  * option '-g'
  39  */
  40 static char *get_sensor;
  41
  42 /**
  43  * option '-f'
  44  */
  45 static char *force_anomaly;
  46
  47 /*
  48  * Handle to sensor service
  49  */
  50 static struct GNUNET_SENSOR_Handle *sensor_handle;
  51
  52 /**
  53  * Run on shutdown
  54  *
  55  * @param cls unused
  56  * @param tc scheduler context
  57  */
  58 static void
  59 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  60 {
  61   if (NULL != sensor_handle)
  62   {
  63     GNUNET_SENSOR_disconnect (sensor_handle);
  64     sensor_handle = NULL;
  65   }
  66 }
  67
  68
  69 /**
  70  * Callback for getting sensor info from service
  71  *
  72  * @param cls not used
  73  * @param sensor brief information about sensor (NULL means end of transmission)
  74  * @param err_msg contains error string if any
  75  */
  76 void
  77 print_sensor_info (void *cls, const struct SensorInfoShort *sensor,
  78                    const char *err_msg)
  79 {
  80   if (NULL != err_msg)
  81   {
  82     printf ("Error: %s\n", err_msg);
  83     GNUNET_SCHEDULER_shutdown ();
  84     return;
  85   }
  86   if (NULL == sensor)           /* no more sensors from service */
  87   {
  88     GNUNET_SCHEDULER_shutdown ();
  89     return;
  90   }
  91   printf ("Name: %s\nVersion: %d.%d\n", sensor->name, sensor->version_major,
  92           sensor->version_minor);
  93   if (NULL != sensor->description)
  94     printf ("Description: %s\n", sensor->description);
  95   printf ("\n");
  96 }
  97
  98
  99 /**
 100  * Continuation called after a force anomaly request is sent.
 101  *
 102  * @param cls Closure (unused)
 103  * @param emsg Error message, NULL of no error
 104  */
 105 void
 106 force_anomaly_cont (void *cls, const char *emsg)
 107 {
 108   if (NULL != emsg)
 109     printf ("Error: %s\n", emsg);
 110   GNUNET_SCHEDULER_shutdown ();
 111 }
 112
 113
 114 /**
 115  * Main function that will be run by the scheduler.
 116  *
 117  * @param cls closure
 118  * @param args remaining command-line arguments
 119  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 120  * @param cfg configuration
 121  */
 122 static void
 123 run (void *cls, char *const *args, const char *cfgfile,
 124      const struct GNUNET_CONFIGURATION_Handle *cfg)
 125 {
 126   sensor_handle = NULL;
 127   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
 128                                 NULL);
 129   sensor_handle = GNUNET_SENSOR_connect (cfg);
 130   GNUNET_assert (NULL != sensor_handle);
 131   if (GNUNET_YES == get_all)
 132   {
 133     GNUNET_SENSOR_iterate (sensor_handle, GNUNET_TIME_UNIT_FOREVER_REL, NULL,
 134                            &print_sensor_info, NULL);
 135   }
 136   else if (NULL != get_sensor)
 137   {
 138     GNUNET_SENSOR_iterate (sensor_handle, GNUNET_TIME_UNIT_FOREVER_REL,
 139                            get_sensor, &print_sensor_info, NULL);
 140   }
 141   else if (NULL != force_anomaly)
 142   {
 143     GNUNET_SENSOR_force_anomaly (sensor_handle, "nse", GNUNET_YES,
 144                                  &force_anomaly_cont, NULL);
 145   }
 146   ret = 0;
 147 }
 148
 149
 150 /**
 151  * The main function to sensor.
 152  *
 153  * @param argc number of arguments from the command line
 154  * @param argv command line arguments
 155  * @return 0 ok, 1 on error
 156  */
 157 int
 158 main (int argc, char *const *argv)
 159 {
 160   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 161     {'a', "all", NULL,
 162      gettext_noop ("Retrieve information about all defined sensors"),
 163      0, &GNUNET_GETOPT_set_one, &get_all},
 164     {'g', "get-sensor", NULL,
 165      gettext_noop ("Retrieve information about a single sensor"),
 166      1, &GNUNET_GETOPT_set_string, &get_sensor},
 167     {'f', "force-anomaly", NULL,
 168      gettext_noop ("Force an anomaly on a sensor, use only for testing"),
 169      1, &GNUNET_GETOPT_set_string, &force_anomaly},
 170     GNUNET_GETOPT_OPTION_END
 171   };
 172
 173   return (GNUNET_OK ==
 174           GNUNET_PROGRAM_run (argc, argv, "gnunet-sensor [options [value]]",
 175                               gettext_noop ("sensor"), options, &run,
 176                               NULL)) ? ret : 1;
 177 }
 178
 179 /* end of gnunet-sensor.c */