src/arm/test_gnunet_service_arm.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2009, 2014 GNUnet e.V.
   4
   5      GNUnet is free software: you can redistribute it and/or modify it
   6      under the terms of the GNU Affero General Public License as published
   7      by the Free Software Foundation, either version 3 of the License,
   8      or (at your 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      Affero General Public License for more details.
  14
  15      You should have received a copy of the GNU Affero General Public License
  16      along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18 /**
  19  * @file arm/test_gnunet_service_arm.c
  20  * @brief testcase for gnunet-service-arm.c; tests ARM by making it start the resolver
  21  * @author Safey
  22  * @author Christian Grothoff
  23  */
  24 #include "platform.h"
  25 #include "gnunet_arm_service.h"
  26 #include "gnunet_resolver_service.h"
  27 #include "gnunet_os_lib.h"
  28 #include "gnunet_program_lib.h"
  29
  30 /**
  31  * Timeout for starting services, very short because of the strange way start works
  32  * (by checking if running before starting, so really this time is always waited on
  33  * startup (annoying)).
  34  */
  35 #define START_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, 50)
  36
  37 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10)
  38
  39
  40 static int ret = 1;
  41
  42 static int resolved_ok;
  43
  44 static int asked_for_a_list;
  45
  46 static struct GNUNET_ARM_Handle *arm;
  47
  48
  49 static void
  50 trigger_disconnect (void *cls)
  51 {
  52   GNUNET_ARM_disconnect (arm);
  53   arm = NULL;
  54 }
  55
  56
  57 static void
  58 arm_stop_cb (void *cls,
  59              enum GNUNET_ARM_RequestStatus status,
  60              enum GNUNET_ARM_Result result)
  61 {
  62   GNUNET_break (status == GNUNET_ARM_REQUEST_SENT_OK);
  63   GNUNET_break (result == GNUNET_ARM_RESULT_STOPPED);
  64   if (result != GNUNET_ARM_RESULT_STOPPED)
  65   {
  66     GNUNET_break (0);
  67     ret = 4;
  68   }
  69   GNUNET_SCHEDULER_add_now (&trigger_disconnect, NULL);
  70 }
  71
  72
  73 static void
  74 service_list (void *cls,
  75               enum GNUNET_ARM_RequestStatus rs,
  76               unsigned int count,
  77               const char *const*list)
  78 {
  79   unsigned int i;
  80
  81   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
  82               "%u services are are currently running\n",
  83               count);
  84   if (GNUNET_ARM_REQUEST_SENT_OK != rs)
  85     goto stop_arm;
  86   for (i=0;i<count;i++)
  87   {
  88     if (0 == strcasecmp (list[i],
  89                          "resolver (gnunet-service-resolver)"))
  90     {
  91       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
  92                   "Got service list, now stopping arm\n");
  93       ret = 0;
  94     }
  95   }
  96
  97  stop_arm:
  98   GNUNET_ARM_request_service_stop (arm,
  99                                    "arm",
 100                                    &arm_stop_cb,
 101                                    NULL);
 102 }
 103
 104
 105 static void
 106 hostname_resolve_cb (void *cls,
 107                      const struct sockaddr *addr,
 108                      socklen_t addrlen)
 109 {
 110   if ((0 == ret) || (4 == ret) || (1 == resolved_ok))
 111     return;
 112   if (NULL == addr)
 113   {
 114     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 115                 "Name not resolved!\n");
 116     GNUNET_break (0);
 117     ret = 3;
 118     GNUNET_ARM_request_service_stop (arm,
 119                                      "arm",
 120                                      &arm_stop_cb,
 121                                      NULL);
 122     return;
 123   }
 124   if (0 == asked_for_a_list)
 125   {
 126     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 127                 "Resolved hostname, now checking the service list\n");
 128     GNUNET_ARM_request_service_list (arm,
 129                                      &service_list,
 130                                      NULL);
 131     asked_for_a_list = 1;
 132     resolved_ok = 1;
 133   }
 134 }
 135
 136
 137 static void
 138 arm_start_cb (void *cls,
 139               enum GNUNET_ARM_RequestStatus status,
 140               enum GNUNET_ARM_Result result)
 141 {
 142   GNUNET_break (status == GNUNET_ARM_REQUEST_SENT_OK);
 143   GNUNET_break (result == GNUNET_ARM_RESULT_STARTING);
 144   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 145               "Trying to resolve our own hostname!\n");
 146   /* connect to the resolver service */
 147   if (NULL ==
 148       GNUNET_RESOLVER_hostname_resolve (AF_UNSPEC,
 149                                         TIMEOUT,
 150                                         &hostname_resolve_cb,
 151                                         NULL))
 152   {
 153     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 154                 "Unable initiate connection to resolver service\n");
 155     GNUNET_break (0);
 156     ret = 2;
 157     GNUNET_ARM_request_service_stop (arm,
 158                                      "arm",
 159                                      &arm_stop_cb,
 160                                      NULL);
 161   }
 162 }
 163
 164
 165 static void
 166 run (void *cls,
 167      char *const *args,
 168      const char *cfgfile,
 169      const struct GNUNET_CONFIGURATION_Handle *c)
 170 {
 171   arm = GNUNET_ARM_connect (c,
 172                             NULL,
 173                             NULL);
 174   GNUNET_ARM_request_service_start (arm,
 175                                     "arm",
 176                                     GNUNET_OS_INHERIT_STD_OUT_AND_ERR,
 177                                     &arm_start_cb,
 178                                     NULL);
 179 }
 180
 181
 182 int
 183 main (int argc, char *av[])
 184 {
 185   static char *const argv[] = {
 186     "test-gnunet-service-arm",
 187     "-c",
 188     "test_arm_api_data.conf",
 189     NULL
 190   };
 191   static struct GNUNET_GETOPT_CommandLineOption options[] = {
 192     GNUNET_GETOPT_OPTION_END
 193   };
 194   char hostname[GNUNET_OS_get_hostname_max_length () + 1];
 195
 196   if (0 != gethostname (hostname, sizeof (hostname) - 1))
 197   {
 198     GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK,
 199                          "gethostname");
 200     FPRINTF (stderr,
 201              "%s",
 202              "Failed to determine my own hostname, testcase not run.\n");
 203     return 0;
 204   }
 205   if ( (0 == strcmp (hostname,
 206                      "localhost")) ||
 207        (0 == strcmp (hostname,
 208                      "ipv6-localnet")) )
 209   {
 210     /* we cannot use 'localhost' as this would not trigger the
 211        resolver service (see resolver_api.c); so in this case,
 212        we fall back to (ab)using gnu.org. */
 213     strcpy (hostname,
 214             "www.gnu.org");
 215   }
 216   /* trigger DNS lookup */
 217 #if HAVE_GETADDRINFO
 218   {
 219     struct addrinfo *ai;
 220     int ret;
 221
 222     if (0 != (ret = getaddrinfo (hostname, NULL, NULL, &ai)))
 223     {
 224       FPRINTF (stderr,
 225                "Failed to resolve my hostname `%s', testcase not run.\n",
 226                hostname);
 227       return 0;
 228     }
 229     freeaddrinfo (ai);
 230   }
 231 #elif HAVE_GETHOSTBYNAME2
 232   {
 233     struct hostent *host;
 234
 235     host = gethostbyname2 (hostname, AF_INET);
 236     if (NULL == host)
 237       host = gethostbyname2 (hostname, AF_INET6);
 238     if (NULL == host)
 239       {
 240         FPRINTF (stderr,
 241                  "Failed to resolve my hostname `%s', testcase not run.\n",
 242                  hostname);
 243         return 0;
 244       }
 245   }
 246 #elif HAVE_GETHOSTBYNAME
 247   {
 248     struct hostent *host;
 249
 250     host = gethostbyname (hostname);
 251     if (NULL == host)
 252       {
 253         FPRINTF (stderr,
 254                  "Failed to resolve my hostname `%s', testcase not run.\n",
 255                  hostname);
 256         return 0;
 257       }
 258   }
 259 #else
 260   FPRINTF (stderr,
 261            "libc fails to have resolver function, testcase not run.\n");
 262   return 0;
 263 #endif
 264   GNUNET_log_setup ("test-gnunet-service-arm",
 265                     "WARNING",
 266                     NULL);
 267   GNUNET_break (GNUNET_OK ==
 268                 GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1,
 269                                     argv, "test-gnunet-service-arm",
 270                                     "nohelp", options,
 271                                     &run, NULL));
 272   if (0 != ret)
 273   {
 274     fprintf (stderr,
 275              "Test failed with error code %d\n",
 276              ret);
 277   }
 278   return ret;
 279 }
 280
 281 /* end of test_gnunet_service_arm.c */