src/nat-auto/nat_auto_api_test.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2011, 2016 GNUnet e.V.
   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  * @file nat/nat_auto_api_test.c
  22  * @brief functions to test if the NAT configuration is successful at achieving NAT traversal (with the help of a gnunet-nat-server)
  23  * @author Christian Grothoff
  24  */
  25 #include "platform.h"
  26 #include "gnunet_util_lib.h"
  27 #include "gnunet_nat_service.h"
  28 #include "gnunet_nat_auto_service.h"
  29 #include "nat-auto.h"
  30
  31 #define LOG(kind,...) GNUNET_log_from (kind, "nat-auto", __VA_ARGS__)
  32
  33 #define NAT_SERVER_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  34
  35 /**
  36  * Entry we keep for each incoming connection.
  37  */
  38 struct NatActivity
  39 {
  40   /**
  41    * This is a doubly-linked list.
  42    */
  43   struct NatActivity *next;
  44
  45   /**
  46    * This is a doubly-linked list.
  47    */
  48   struct NatActivity *prev;
  49
  50   /**
  51    * Socket of the incoming connection.
  52    */
  53   struct GNUNET_NETWORK_Handle *sock;
  54
  55   /**
  56    * Handle of the master context.
  57    */
  58   struct GNUNET_NAT_AUTO_Test *h;
  59
  60   /**
  61    * Task reading from the incoming connection.
  62    */
  63   struct GNUNET_SCHEDULER_Task *rtask;
  64 };
  65
  66
  67 /**
  68  * Entry we keep for each connection to the gnunet-nat-service.
  69  */
  70 struct ClientActivity
  71 {
  72   /**
  73    * This is a doubly-linked list.
  74    */
  75   struct ClientActivity *next;
  76
  77   /**
  78    * This is a doubly-linked list.
  79    */
  80   struct ClientActivity *prev;
  81
  82   /**
  83    * Socket of the incoming connection.
  84    */
  85   struct GNUNET_MQ_Handle *mq;
  86
  87   /**
  88    * Handle to overall NAT test.
  89    */
  90   struct GNUNET_NAT_AUTO_Test *h;
  91
  92 };
  93
  94
  95 /**
  96  * Handle to a NAT test.
  97  */
  98 struct GNUNET_NAT_AUTO_Test
  99 {
 100
 101   /**
 102    * Configuration used
 103    */
 104   const struct GNUNET_CONFIGURATION_Handle *cfg;
 105
 106   /**
 107    * Function to call with success report
 108    */
 109   GNUNET_NAT_TestCallback report;
 110
 111   /**
 112    * Closure for @e report.
 113    */
 114   void *report_cls;
 115
 116   /**
 117    * Handle to NAT traversal in use
 118    */
 119   struct GNUNET_NAT_Handle *nat;
 120
 121   /**
 122    * Handle to listen socket, or NULL
 123    */
 124   struct GNUNET_NETWORK_Handle *lsock;
 125
 126   /**
 127    * Head of list of nat activities.
 128    */
 129   struct NatActivity *na_head;
 130
 131   /**
 132    * Tail of list of nat activities.
 133    */
 134   struct NatActivity *na_tail;
 135
 136   /**
 137    * Head of list of client activities.
 138    */
 139   struct ClientActivity *ca_head;
 140
 141   /**
 142    * Tail of list of client activities.
 143    */
 144   struct ClientActivity *ca_tail;
 145
 146   /**
 147    * Identity of task for the listen socket (if any)
 148    */
 149   struct GNUNET_SCHEDULER_Task *ltask;
 150
 151   /**
 152    * Task identifier for the timeout (if any)
 153    */
 154   struct GNUNET_SCHEDULER_Task *ttask;
 155
 156   /**
 157    * Section name of plugin to test.
 158    */
 159   char *section_name;
 160
 161   /**
 162    * IPPROTO_TCP or IPPROTO_UDP.
 163    */
 164   int proto;
 165
 166   /**
 167    * Data that should be transmitted or source-port.
 168    */
 169   uint16_t data;
 170
 171   /**
 172    * Status code to be reported to the timeout/status call
 173    */
 174   enum GNUNET_NAT_StatusCode status;
 175 };
 176
 177
 178 /**
 179  * Function called from #GNUNET_NAT_register whenever someone asks us
 180  * to do connection reversal.
 181  *
 182  * @param cls closure, our `struct GNUNET_NAT_Handle`
 183  * @param addr public IP address of the other peer
 184  * @param addrlen actual lenght of the @a addr
 185  */
 186 static void
 187 reversal_cb (void *cls,
 188              const struct sockaddr *addr,
 189              socklen_t addrlen)
 190 {
 191   struct GNUNET_NAT_AUTO_Test *h = cls;
 192   const struct sockaddr_in *sa;
 193
 194   if (sizeof (struct sockaddr_in) != addrlen)
 195     return;
 196   sa = (const struct sockaddr_in *) addr;
 197   if (h->data != sa->sin_port)
 198   {
 199     LOG (GNUNET_ERROR_TYPE_DEBUG,
 200          "Received connection reversal request for wrong port\n");
 201     return;                     /* wrong port */
 202   }
 203   /* report success */
 204   h->report (h->report_cls,
 205              GNUNET_NAT_ERROR_SUCCESS);
 206 }
 207
 208
 209 /**
 210  * Activity on our incoming socket.  Read data from the
 211  * incoming connection.
 212  *
 213  * @param cls the `struct GNUNET_NAT_AUTO_Test`
 214  */
 215 static void
 216 do_udp_read (void *cls)
 217 {
 218   struct GNUNET_NAT_AUTO_Test *tst = cls;
 219   uint16_t data;
 220   const struct GNUNET_SCHEDULER_TaskContext *tc;
 221
 222   tc = GNUNET_SCHEDULER_get_task_context ();
 223   tst->ltask =
 224       GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 225                                      tst->lsock,
 226                                      &do_udp_read,
 227                                      tst);
 228   if ((NULL != tc->write_ready) &&
 229       (GNUNET_NETWORK_fdset_isset (tc->read_ready,
 230                                    tst->lsock)) &&
 231       (sizeof (data) ==
 232        GNUNET_NETWORK_socket_recv (tst->lsock,
 233                                    &data,
 234                                    sizeof (data))))
 235   {
 236     if (data == tst->data)
 237       tst->report (tst->report_cls,
 238                    GNUNET_NAT_ERROR_SUCCESS);
 239     else
 240       LOG (GNUNET_ERROR_TYPE_DEBUG,
 241            "Received data mismatches expected value\n");
 242   }
 243   else
 244     LOG (GNUNET_ERROR_TYPE_DEBUG,
 245          "Failed to receive data from inbound connection\n");
 246 }
 247
 248
 249 /**
 250  * Activity on our incoming socket.  Read data from the
 251  * incoming connection.
 252  *
 253  * @param cls the `struct NatActivity`
 254  */
 255 static void
 256 do_read (void *cls)
 257 {
 258   struct NatActivity *na = cls;
 259   struct GNUNET_NAT_AUTO_Test *tst;
 260   uint16_t data;
 261   const struct GNUNET_SCHEDULER_TaskContext *tc;
 262
 263   tc = GNUNET_SCHEDULER_get_task_context ();
 264   na->rtask = NULL;
 265   tst = na->h;
 266   GNUNET_CONTAINER_DLL_remove (tst->na_head,
 267                                tst->na_tail,
 268                                na);
 269   if ((NULL != tc->write_ready) &&
 270       (GNUNET_NETWORK_fdset_isset (tc->read_ready,
 271                                    na->sock)) &&
 272       (sizeof (data) ==
 273        GNUNET_NETWORK_socket_recv (na->sock,
 274                                    &data,
 275                                    sizeof (data))))
 276   {
 277     if (data == tst->data)
 278       tst->report (tst->report_cls,
 279                    GNUNET_NAT_ERROR_SUCCESS);
 280     else
 281       LOG (GNUNET_ERROR_TYPE_DEBUG,
 282            "Received data does not match expected value\n");
 283   }
 284   else
 285     LOG (GNUNET_ERROR_TYPE_DEBUG,
 286          "Failed to receive data from inbound connection\n");
 287   GNUNET_NETWORK_socket_close (na->sock);
 288   GNUNET_free (na);
 289 }
 290
 291
 292 /**
 293  * Activity on our listen socket. Accept the
 294  * incoming connection.
 295  *
 296  * @param cls the `struct GNUNET_NAT_AUTO_Test`
 297  */
 298 static void
 299 do_accept (void *cls)
 300 {
 301   struct GNUNET_NAT_AUTO_Test *tst = cls;
 302   struct GNUNET_NETWORK_Handle *s;
 303   struct NatActivity *wl;
 304
 305   tst->ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 306                                               tst->lsock,
 307                                               &do_accept,
 308                                               tst);
 309   s = GNUNET_NETWORK_socket_accept (tst->lsock,
 310                                     NULL,
 311                                     NULL);
 312   if (NULL == s)
 313   {
 314     GNUNET_log_strerror (GNUNET_ERROR_TYPE_INFO,
 315                          "accept");
 316     return;                     /* odd error */
 317   }
 318   LOG (GNUNET_ERROR_TYPE_DEBUG,
 319        "Got an inbound connection, waiting for data\n");
 320   wl = GNUNET_new (struct NatActivity);
 321   wl->sock = s;
 322   wl->h = tst;
 323   wl->rtask =
 324     GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 325                                    wl->sock,
 326                                    &do_read,
 327                                    wl);
 328   GNUNET_CONTAINER_DLL_insert (tst->na_head,
 329                                tst->na_tail,
 330                                wl);
 331 }
 332
 333
 334 /**
 335  * We got disconnected from the NAT server.  Stop
 336  * waiting for a reply.
 337  *
 338  * @param cls the `struct ClientActivity`
 339  * @param error error code
 340  */
 341 static void
 342 mq_error_handler (void *cls,
 343                   enum GNUNET_MQ_Error error)
 344 {
 345   struct ClientActivity *ca = cls;
 346   struct GNUNET_NAT_AUTO_Test *tst = ca->h;
 347
 348   GNUNET_CONTAINER_DLL_remove (tst->ca_head,
 349                                tst->ca_tail,
 350                                ca);
 351   GNUNET_MQ_destroy (ca->mq);
 352   GNUNET_free (ca);
 353 }
 354
 355
 356 /**
 357  * Address-callback, used to send message to gnunet-nat-server.
 358  *
 359  * @param cls closure
 360  * @param add_remove #GNUNET_YES to mean the new public IP address, #GNUNET_NO to mean
 361  *     the previous (now invalid) one
 362  * @param ac address class the address belongs to
 363  * @param addr either the previous or the new public IP address
 364  * @param addrlen actual length of the @a addr
 365  */
 366 static void
 367 addr_cb (void *cls,
 368          int add_remove,
 369          enum GNUNET_NAT_AddressClass ac,
 370          const struct sockaddr *addr,
 371          socklen_t addrlen)
 372 {
 373   struct GNUNET_NAT_AUTO_Test *h = cls;
 374   struct ClientActivity *ca;
 375   struct GNUNET_MQ_Envelope *env;
 376   struct GNUNET_NAT_AUTO_TestMessage *msg;
 377   const struct sockaddr_in *sa;
 378
 379   if (GNUNET_YES != add_remove)
 380     return;
 381   if (addrlen != sizeof (struct sockaddr_in))
 382   {
 383     LOG (GNUNET_ERROR_TYPE_DEBUG,
 384          "NAT test ignores IPv6 address `%s' returned from NAT library\n",
 385          GNUNET_a2s (addr,
 386                      addrlen));
 387     return;                     /* ignore IPv6 here */
 388   }
 389   LOG (GNUNET_ERROR_TYPE_INFO,
 390        "Asking gnunet-nat-server to connect to `%s'\n",
 391        GNUNET_a2s (addr,
 392                    addrlen));
 393
 394   ca = GNUNET_new (struct ClientActivity);
 395   ca->h = h;
 396   ca->mq = GNUNET_CLIENT_connect (h->cfg,
 397                                   "gnunet-nat-server",
 398                                   NULL,
 399                                   &mq_error_handler,
 400                                   ca);
 401   if (NULL == ca->mq)
 402   {
 403     GNUNET_free (ca);
 404     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 405                 _("Failed to connect to `gnunet-nat-server'\n"));
 406     return;
 407   }
 408   GNUNET_CONTAINER_DLL_insert (h->ca_head,
 409                                h->ca_tail,
 410                                ca);
 411   sa = (const struct sockaddr_in *) addr;
 412   env = GNUNET_MQ_msg (msg,
 413                        GNUNET_MESSAGE_TYPE_NAT_TEST);
 414   msg->dst_ipv4 = sa->sin_addr.s_addr;
 415   msg->dport = sa->sin_port;
 416   msg->data = h->data;
 417   msg->is_tcp = htonl ((uint32_t) (h->proto == IPPROTO_TCP));
 418   GNUNET_MQ_send (ca->mq,
 419                   env);
 420 }
 421
 422
 423 /**
 424  * Calls the report-callback reporting failure.
 425  *
 426  * Destroys the nat handle after the callback has been processed.
 427  *
 428  * @param cls handle to the timed out NAT test
 429  */
 430 static void
 431 do_fail (void *cls)
 432 {
 433   struct GNUNET_NAT_AUTO_Test *nh = cls;
 434
 435   nh->ttask = NULL;
 436   nh->report (nh->report_cls,
 437               nh->status);
 438 }
 439
 440
 441 /**
 442  * Start testing if NAT traversal works using the given configuration.
 443  *  The transport adapters should be down while using this function.
 444  *
 445  * @param cfg configuration for the NAT traversal
 446  * @param proto protocol to test, i.e. IPPROTO_TCP or IPPROTO_UDP
 447  * @param section_name configuration section to use for configuration
 448  * @param report function to call with the result of the test
 449  * @param report_cls closure for @a report
 450  * @return handle to cancel NAT test
 451  */
 452 struct GNUNET_NAT_AUTO_Test *
 453 GNUNET_NAT_AUTO_test_start (const struct GNUNET_CONFIGURATION_Handle *cfg,
 454                             uint8_t proto,
 455                             const char *section_name,
 456                             GNUNET_NAT_TestCallback report,
 457                             void *report_cls)
 458 {
 459   struct GNUNET_NAT_AUTO_Test *nh;
 460   unsigned long long bnd_port;
 461   struct sockaddr_in sa;
 462   const struct sockaddr *addrs[] = {
 463     (const struct sockaddr *) &sa
 464   };
 465   const socklen_t addrlens[] = {
 466     sizeof (sa)
 467   };
 468
 469   if ( (GNUNET_OK !=
 470         GNUNET_CONFIGURATION_get_value_number (cfg,
 471                                                section_name,
 472                                                "PORT",
 473                                                &bnd_port)) ||
 474        (bnd_port > 65535) )
 475   {
 476     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 477                 _("Failed to find valid PORT in section `%s'\n"),
 478                 section_name);
 479     return NULL;
 480   }
 481
 482   memset (&sa, 0, sizeof (sa));
 483   sa.sin_family = AF_INET;
 484   sa.sin_port = htons ((uint16_t) bnd_port);
 485 #if HAVE_SOCKADDR_IN_SIN_LEN
 486   sa.sin_len = sizeof (sa);
 487 #endif
 488
 489   nh = GNUNET_new (struct GNUNET_NAT_AUTO_Test);
 490   nh->cfg = cfg;
 491   nh->proto = proto;
 492   nh->section_name = GNUNET_strdup (section_name);
 493   nh->report = report;
 494   nh->report_cls = report_cls;
 495   nh->status = GNUNET_NAT_ERROR_SUCCESS;
 496   if (0 == bnd_port)
 497   {
 498     nh->nat
 499       = GNUNET_NAT_register (cfg,
 500                              section_name,
 501                              proto,
 502                              0, NULL, NULL,
 503                              &addr_cb,
 504                              &reversal_cb,
 505                              nh);
 506   }
 507   else
 508   {
 509     nh->lsock
 510       = GNUNET_NETWORK_socket_create (AF_INET,
 511                                       (IPPROTO_UDP == proto)
 512                                       ? SOCK_DGRAM
 513                                       : SOCK_STREAM,
 514                                       proto);
 515     if ( (NULL == nh->lsock) ||
 516          (GNUNET_OK !=
 517           GNUNET_NETWORK_socket_bind (nh->lsock,
 518                                       (const struct sockaddr *) &sa,
 519                                       sizeof (sa))))
 520     {
 521       LOG (GNUNET_ERROR_TYPE_ERROR,
 522            _("Failed to create socket bound to `%s' for NAT test: %s\n"),
 523            GNUNET_a2s ((const struct sockaddr *) &sa,
 524                        sizeof (sa)),
 525            STRERROR (errno));
 526       if (NULL != nh->lsock)
 527       {
 528         GNUNET_NETWORK_socket_close (nh->lsock);
 529         nh->lsock = NULL;
 530       }
 531       nh->status = GNUNET_NAT_ERROR_INTERNAL_NETWORK_ERROR;
 532       nh->ttask = GNUNET_SCHEDULER_add_now (&do_fail,
 533                                             nh);
 534       return nh;
 535     }
 536     if (IPPROTO_TCP == proto)
 537     {
 538       GNUNET_break (GNUNET_OK ==
 539                     GNUNET_NETWORK_socket_listen (nh->lsock,
 540                                                   5));
 541       nh->ltask =
 542           GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 543                                          nh->lsock,
 544                                          &do_accept,
 545                                          nh);
 546     }
 547     else
 548     {
 549       nh->ltask =
 550           GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 551                                          nh->lsock,
 552                                          &do_udp_read,
 553                                          nh);
 554     }
 555     LOG (GNUNET_ERROR_TYPE_INFO,
 556          "NAT test listens on port %llu (%s)\n",
 557          bnd_port,
 558          (IPPROTO_TCP == proto) ? "tcp" : "udp");
 559     nh->nat = GNUNET_NAT_register (cfg,
 560                                    section_name,
 561                                    proto,
 562                                    1,
 563                                    addrs,
 564                                    addrlens,
 565                                    &addr_cb,
 566                                    NULL,
 567                                    nh);
 568     if (NULL == nh->nat)
 569     {
 570       LOG (GNUNET_ERROR_TYPE_INFO,
 571           _("NAT test failed to start NAT library\n"));
 572       if (NULL != nh->ltask)
 573       {
 574         GNUNET_SCHEDULER_cancel (nh->ltask);
 575         nh->ltask = NULL;
 576       }
 577       if (NULL != nh->lsock)
 578       {
 579         GNUNET_NETWORK_socket_close (nh->lsock);
 580         nh->lsock = NULL;
 581       }
 582       nh->status = GNUNET_NAT_ERROR_NAT_REGISTER_FAILED;
 583       nh->ttask = GNUNET_SCHEDULER_add_now (&do_fail,
 584                                             nh);
 585       return nh;
 586     }
 587   }
 588   return nh;
 589 }
 590
 591
 592 /**
 593  * Stop an active NAT test.
 594  *
 595  * @param tst test to stop.
 596  */
 597 void
 598 GNUNET_NAT_AUTO_test_stop (struct GNUNET_NAT_AUTO_Test *tst)
 599 {
 600   struct NatActivity *pos;
 601   struct ClientActivity *cpos;
 602
 603   LOG (GNUNET_ERROR_TYPE_DEBUG,
 604        "Stopping NAT test\n");
 605   while (NULL != (cpos = tst->ca_head))
 606   {
 607     GNUNET_CONTAINER_DLL_remove (tst->ca_head,
 608                                  tst->ca_tail,
 609                                  cpos);
 610     GNUNET_MQ_destroy (cpos->mq);
 611     GNUNET_free (cpos);
 612   }
 613   while (NULL != (pos = tst->na_head))
 614   {
 615     GNUNET_CONTAINER_DLL_remove (tst->na_head,
 616                                  tst->na_tail,
 617                                  pos);
 618     GNUNET_SCHEDULER_cancel (pos->rtask);
 619     GNUNET_NETWORK_socket_close (pos->sock);
 620     GNUNET_free (pos);
 621   }
 622   if (NULL != tst->ttask)
 623   {
 624     GNUNET_SCHEDULER_cancel (tst->ttask);
 625     tst->ttask = NULL;
 626   }
 627   if (NULL != tst->ltask)
 628   {
 629     GNUNET_SCHEDULER_cancel (tst->ltask);
 630     tst->ltask = NULL;
 631   }
 632   if (NULL != tst->lsock)
 633   {
 634     GNUNET_NETWORK_socket_close (tst->lsock);
 635     tst->lsock = NULL;
 636   }
 637   if (NULL != tst->nat)
 638   {
 639     GNUNET_NAT_unregister (tst->nat);
 640     tst->nat = NULL;
 641   }
 642   GNUNET_free (tst->section_name);
 643   GNUNET_free (tst);
 644 }
 645
 646 /* end of nat_auto_api_test.c */