src/arm/test_gnunet_service_arm.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009 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 arm/test_gnunet_service_arm.c
  22  * @brief testcase for gnunet-service-arm.c; tests ARM by making it start the resolver
  23  * @author Safey
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_arm_service.h"
  28 #include "gnunet_resolver_service.h"
  29 #include "gnunet_os_lib.h"
  30 #include "gnunet_program_lib.h"
  31
  32 /**
  33  * Timeout for starting services, very short because of the strange way start works
  34  * (by checking if running before starting, so really this time is always waited on
  35  * startup (annoying)).
  36  */
  37 #define START_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, 50)
  38
  39 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10)
  40
  41 static int ret = 1;
  42
  43 static int resolved_ok = 0;
  44
  45 static int asked_for_a_list = 0;
  46
  47 static struct GNUNET_ARM_Handle *arm;
  48
  49 static void
  50 trigger_disconnect (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  51 {
  52   GNUNET_ARM_disconnect_and_free (arm);
  53   arm = NULL;
  54 }
  55
  56
  57 static void
  58 arm_stop_cb (void *cls,
  59              enum GNUNET_ARM_RequestStatus status,
  60              const char *servicename,
  61              enum GNUNET_ARM_Result result)
  62 {
  63   GNUNET_break (status == GNUNET_ARM_REQUEST_SENT_OK);
  64   GNUNET_break (result == GNUNET_ARM_RESULT_STOPPED);
  65   if (result != GNUNET_ARM_RESULT_STOPPED)
  66     ret = 4;
  67   GNUNET_SCHEDULER_add_now (trigger_disconnect, NULL);
  68 }
  69
  70
  71 static void
  72 service_list (void *cls,
  73               enum GNUNET_ARM_RequestStatus rs,
  74               unsigned int count, const char *const*list)
  75 {
  76   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
  77               "%u services are are currently running\n",
  78               count);
  79   if (GNUNET_ARM_REQUEST_SENT_OK != rs)
  80     goto stop_arm;
  81   if (1 == count)
  82   {
  83     GNUNET_break (0 == strcasecmp (list[0], "resolver (gnunet-service-resolver)"));
  84     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got service list, now stopping arm\n");
  85     ret = 0;
  86   }
  87
  88  stop_arm:
  89   GNUNET_ARM_request_service_stop (arm, "arm", TIMEOUT, arm_stop_cb, NULL);
  90 }
  91
  92
  93 static void
  94 hostNameResolveCB (void *cls, const struct sockaddr *addr, socklen_t addrlen)
  95 {
  96   if ((ret == 0) || (ret == 4) || (resolved_ok == 1))
  97     return;
  98   if (NULL == addr)
  99   {
 100     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Name not resolved!\n");
 101     ret = 3;
 102     GNUNET_ARM_request_service_stop (arm, "arm", TIMEOUT, arm_stop_cb, NULL);
 103   }
 104   else if (asked_for_a_list == 0)
 105   {
 106     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 107         "Resolved hostname, now checking the service list\n");
 108     GNUNET_ARM_request_service_list (arm, TIMEOUT, service_list, NULL);
 109     asked_for_a_list = 1;
 110     resolved_ok = 1;
 111   }
 112 }
 113
 114
 115 static void
 116 arm_start_cb (void *cls,
 117               enum GNUNET_ARM_RequestStatus status,
 118               const char *servicename,
 119               enum GNUNET_ARM_Result result)
 120 {
 121   GNUNET_break (status == GNUNET_ARM_REQUEST_SENT_OK);
 122   GNUNET_break (result == GNUNET_ARM_RESULT_STARTING);
 123   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 124       "Trying to resolve our own hostname!\n");
 125   /* connect to the resolver service */
 126   if (NULL == GNUNET_RESOLVER_hostname_resolve (
 127       AF_UNSPEC, TIMEOUT, &hostNameResolveCB, NULL))
 128   {
 129     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 130         "Unable initiate connection to resolver service\n");
 131     ret = 2;
 132     GNUNET_ARM_request_service_stop (arm, "arm", TIMEOUT, arm_stop_cb, NULL);
 133   }
 134 }
 135
 136
 137 static void
 138 run (void *cls, char *const *args, const char *cfgfile,
 139      const struct GNUNET_CONFIGURATION_Handle *c)
 140 {
 141   char *armconfig;
 142
 143   if (NULL != cfgfile)
 144   {
 145     if (GNUNET_OK !=
 146         GNUNET_CONFIGURATION_get_value_filename (c, "arm", "CONFIG",
 147                                                &armconfig))
 148     {
 149       GNUNET_CONFIGURATION_set_value_string ((struct GNUNET_CONFIGURATION_Handle
 150                                               *) c, "arm", "CONFIG",
 151                                              cfgfile);
 152     }
 153     else
 154       GNUNET_free (armconfig);
 155   }
 156   arm = GNUNET_ARM_connect (c, NULL, NULL);
 157   GNUNET_ARM_request_service_start (arm, "arm",
 158       GNUNET_OS_INHERIT_STD_OUT_AND_ERR, START_TIMEOUT, arm_start_cb, NULL);
 159 }
 160
 161
 162 int
 163 main (int argc, char *av[])
 164 {
 165   static char *const argv[] = {
 166     "test-gnunet-service-arm",
 167     "-c", "test_arm_api_data.conf",
 168     NULL
 169   };
 170   static struct GNUNET_GETOPT_CommandLineOption options[] = {
 171     GNUNET_GETOPT_OPTION_END
 172   };
 173   char hostname[GNUNET_OS_get_hostname_max_length () + 1];
 174
 175   if (0 != gethostname (hostname, sizeof (hostname) - 1))
 176     {
 177       GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK,
 178                            "gethostname");
 179       FPRINTF (stderr,
 180                "%s", "Failed to determine my own hostname, testcase not run.\n");
 181       return 0;
 182     }
 183   if (NULL == gethostbyname (hostname))
 184     {
 185       FPRINTF (stderr,
 186                "Failed to resolve my hostname `%s', testcase not run.\n",
 187                hostname);
 188       return 0;
 189     }
 190   GNUNET_log_setup ("test-gnunet-service-arm",
 191                     "WARNING",
 192                     NULL);
 193   GNUNET_break (GNUNET_OK ==
 194                 GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1,
 195                                     argv, "test-gnunet-service-arm",
 196                                     "nohelp", options, &run, NULL));
 197   return ret;
 198 }
 199
 200 /* end of test_gnunet_service_arm.c */