src/vpn/gnunet-daemon-exit.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2010 Christian Grothoff
   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 vpn/gnunet-daemon-exit.c
  23  * @brief
  24  * @author Philipp Toelke
  25  */
  26 #include <platform.h>
  27 #include <gnunet_common.h>
  28 #include <gnunet_program_lib.h>
  29 #include <gnunet_protocols.h>
  30 #include <gnunet_applications.h>
  31 #include <gnunet_mesh_service.h>
  32 #include <gnunet_constants.h>
  33 #include <string.h>
  34
  35 #include "gnunet-vpn-packet.h"
  36 #include "gnunet-helper-vpn-api.h"
  37 #include "gnunet-vpn-checksum.h"
  38
  39 GNUNET_SCHEDULER_TaskIdentifier shs_task;
  40
  41 /**
  42  * The handle to the configuration used throughout the process
  43  */
  44 static const struct GNUNET_CONFIGURATION_Handle *cfg;
  45
  46 /**
  47  * The handle to the helper
  48  */
  49 struct GNUNET_VPN_HELPER_Handle *helper_handle;
  50
  51 /**
  52  * Final status code.
  53  */
  54 static int ret;
  55
  56 /**
  57  * The handle to mesh
  58  */
  59 static struct GNUNET_MESH_Handle *mesh_handle;
  60
  61 /**
  62  * This hashmaps contains the mapping from peer, service-descriptor,
  63  * source-port and destination-port to a struct redirect_state
  64  */
  65 static struct GNUNET_CONTAINER_MultiHashMap *udp_connections;
  66 static struct GNUNET_CONTAINER_Heap *udp_connections_heap;
  67 static struct GNUNET_CONTAINER_MultiHashMap *tcp_connections;
  68 static struct GNUNET_CONTAINER_Heap *tcp_connections_heap;
  69
  70 /**
  71  * If there are at least this many udp-Connections, old ones will be removed
  72  */
  73 static long long unsigned int max_udp_connections = 200;
  74
  75 /**
  76  * If there are at least this many tcp-Connections, old ones will be removed
  77  */
  78 static long long unsigned int max_tcp_connections = 200;
  79
  80 struct remote_addr
  81 {
  82   char addrlen;
  83   unsigned char addr[16];
  84   char proto;
  85 };
  86
  87 /**
  88  * This struct is saved into the services-hashmap
  89  */
  90 struct redirect_service
  91 {
  92   /**
  93    * One of 4 or 6
  94    */
  95   unsigned int version;
  96   uint16_t my_port;
  97   uint16_t remote_port;
  98
  99   union
 100   {
 101     struct
 102     {
 103       char ip4address[4];
 104     } v4;
 105     struct
 106     {
 107       char ip6address[16];
 108     } v6;
 109   };
 110 };
 111
 112 struct redirect_info
 113 {
 114     /**
 115      * The source-address of this connection. When a packet to this address is
 116      * received, this tunnel is used to forward it.  ipv4-addresses will be put
 117      * here left-aligned */
 118   char addr[16];
 119     /**
 120      * The source-port of this connection
 121      */
 122   uint16_t pt;
 123 };
 124
 125 /**
 126  * This struct is saved into {tcp,udp}_connections;
 127  */
 128 struct redirect_state
 129 {
 130   struct GNUNET_MESH_Tunnel *tunnel;
 131   GNUNET_HashCode desc;
 132   struct redirect_service *serv;
 133   struct remote_addr remote;
 134
 135   struct GNUNET_CONTAINER_HeapNode *heap_node;
 136   struct GNUNET_CONTAINER_MultiHashMap *hashmap;
 137   GNUNET_HashCode hash;
 138
 139   enum
 140   { SERVICE, REMOTE } type;
 141
 142   /**
 143    * The source-address and -port of this connection
 144    */
 145   struct redirect_info redirect_info;
 146 };
 147
 148 /**
 149  * This hashmaps saves interesting things about the configured services
 150  */
 151 static struct GNUNET_CONTAINER_MultiHashMap *udp_services;
 152 static struct GNUNET_CONTAINER_MultiHashMap *tcp_services;
 153
 154 struct tunnel_notify_queue
 155 {
 156   struct tunnel_notify_queue *next;
 157   struct tunnel_notify_queue *prev;
 158   void *cls;
 159   size_t len;
 160 };
 161
 162 /**
 163  * Function that frees everything from a hashmap
 164  */
 165 static int
 166 free_iterate (void *cls __attribute__ ((unused)), const GNUNET_HashCode * hash
 167               __attribute__ ((unused)), void *value)
 168 {
 169   GNUNET_free (value);
 170   return GNUNET_YES;
 171 }
 172
 173 /**
 174  * Function scheduled as very last function, cleans up after us
 175  */
 176 static void
 177 cleanup (void *cls
 178          __attribute__ ((unused)),
 179          const struct GNUNET_SCHEDULER_TaskContext *tskctx)
 180 {
 181   GNUNET_assert (0 != (tskctx->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN));
 182
 183   GNUNET_CONTAINER_multihashmap_iterate (udp_connections, free_iterate, NULL);
 184
 185   GNUNET_CONTAINER_multihashmap_iterate (tcp_connections, free_iterate, NULL);
 186
 187   if (mesh_handle != NULL)
 188   {
 189     GNUNET_MESH_disconnect (mesh_handle);
 190     mesh_handle = NULL;
 191   }
 192 }
 193
 194 static void
 195 collect_connections (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 196 {
 197   if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
 198     return;
 199
 200
 201   struct GNUNET_CONTAINER_Heap *heap = cls;
 202
 203   struct redirect_state *state = GNUNET_CONTAINER_heap_remove_root (heap);
 204
 205   /* This is free()ed memory! */
 206   state->heap_node = NULL;
 207
 208   /* FIXME! GNUNET_MESH_close_tunnel(state->tunnel); */
 209
 210   GNUNET_assert (GNUNET_OK ==
 211                  GNUNET_CONTAINER_multihashmap_remove (state->hashmap, &state->hash, state));
 212
 213   GNUNET_free (state);
 214 }
 215
 216 static void
 217 hash_redirect_info (GNUNET_HashCode * hash, struct redirect_info *u_i,
 218                     size_t addrlen)
 219 {
 220
 221   /* the gnunet hashmap only uses the first sizeof(unsigned int) of the hash
 222    *
 223    * build the hash out of the last bytes of the address and the 2 bytes of
 224    * the port
 225    */
 226   memcpy (hash, &u_i->pt, sizeof (u_i->pt));
 227   memcpy (((unsigned char *) hash) + 2,
 228           u_i->addr + (addrlen - (sizeof (unsigned int) - 2)),
 229           (sizeof (unsigned int) - 2));
 230   memset (((unsigned char *) hash) + sizeof (unsigned int), 0,
 231           sizeof (GNUNET_HashCode) - sizeof (unsigned int));
 232 }
 233
 234 /**
 235  * cls is the pointer to a GNUNET_MessageHeader that is
 236  * followed by the service-descriptor and the udp-packet that should be sent;
 237  */
 238 static size_t
 239 send_udp_to_peer_notify_callback (void *cls, size_t size, void *buf)
 240 {
 241   struct GNUNET_MESH_Tunnel **tunnel = cls;
 242
 243   GNUNET_MESH_tunnel_set_data (*tunnel, NULL);
 244   struct GNUNET_MessageHeader *hdr =
 245       (struct GNUNET_MessageHeader *) (tunnel + 1);
 246   GNUNET_assert (size >= ntohs (hdr->size));
 247   memcpy (buf, hdr, ntohs (hdr->size));
 248   size = ntohs (hdr->size);
 249
 250   if (NULL != GNUNET_MESH_tunnel_get_head (*tunnel))
 251   {
 252     struct tunnel_notify_queue *element = GNUNET_MESH_tunnel_get_head (*tunnel);
 253     struct tunnel_notify_queue *head = GNUNET_MESH_tunnel_get_head (*tunnel);
 254     struct tunnel_notify_queue *tail = GNUNET_MESH_tunnel_get_tail (*tunnel);
 255
 256     GNUNET_CONTAINER_DLL_remove (head, tail, element);
 257
 258     GNUNET_MESH_tunnel_set_head (*tunnel, head);
 259     GNUNET_MESH_tunnel_set_tail (*tunnel, tail);
 260
 261     struct GNUNET_MESH_TransmitHandle *th =
 262         GNUNET_MESH_notify_transmit_ready (*tunnel,
 263                                            GNUNET_NO,
 264                                            42,
 265                                            GNUNET_TIME_relative_divide
 266                                            (GNUNET_CONSTANTS_MAX_CORK_DELAY, 2),
 267                                            (const struct GNUNET_PeerIdentity *)
 268                                            NULL, element->len,
 269                                            send_udp_to_peer_notify_callback,
 270                                            element->cls);
 271
 272     /* save the handle */
 273     GNUNET_MESH_tunnel_set_data (*tunnel, th);
 274     GNUNET_free (element);
 275   }
 276
 277   GNUNET_free (cls);
 278
 279   return size;
 280 }
 281
 282 /**
 283  * @brief Handles an UDP-Packet received from the helper.
 284  *
 285  * @param udp A pointer to the Packet
 286  * @param dadr The IP-Destination-address
 287  * @param addrlen The length of the address
 288  * @param version 4 or 6
 289  */
 290 static void
 291 udp_from_helper (struct udp_pkt *udp, unsigned char *dadr, size_t addrlen)
 292 {
 293   struct redirect_info u_i;
 294   struct GNUNET_MESH_Tunnel *tunnel;
 295   uint32_t len;
 296   struct GNUNET_MessageHeader *msg;
 297
 298   memset (&u_i, 0, sizeof (struct redirect_info));
 299
 300   memcpy (&u_i.addr, dadr, addrlen);
 301
 302   u_i.pt = udp->dpt;
 303
 304   /* get tunnel and service-descriptor from this */
 305   GNUNET_HashCode hash;
 306
 307   hash_redirect_info (&hash, &u_i, addrlen);
 308
 309   struct redirect_state *state =
 310       GNUNET_CONTAINER_multihashmap_get (udp_connections, &hash);
 311
 312   /* Mark this connection as freshly used */
 313   GNUNET_CONTAINER_heap_update_cost (udp_connections_heap, state->heap_node,
 314                                      GNUNET_TIME_absolute_get ().abs_value);
 315
 316   tunnel = state->tunnel;
 317
 318   if (state->type == SERVICE)
 319   {
 320     /* check if spt == serv.remote if yes: set spt = serv.myport ("nat") */
 321     if (ntohs (udp->spt) == state->serv->remote_port)
 322     {
 323       udp->spt = htons (state->serv->my_port);
 324     }
 325     else
 326     {
 327       /* otherwise the answer came from a different port (tftp does this)
 328        * add this new port to the list of all services, so that the packets
 329        * coming back from the client to this new port will be routed correctly
 330        */
 331       struct redirect_service *serv =
 332           GNUNET_malloc (sizeof (struct redirect_service));
 333       memcpy (serv, state->serv, sizeof (struct redirect_service));
 334       serv->my_port = ntohs (udp->spt);
 335       serv->remote_port = ntohs (udp->spt);
 336       uint16_t *desc = alloca (sizeof (GNUNET_HashCode) + 2);
 337
 338       memcpy ((GNUNET_HashCode *) (desc + 1), &state->desc,
 339               sizeof (GNUNET_HashCode));
 340       *desc = ntohs (udp->spt);
 341       GNUNET_assert (GNUNET_OK ==
 342                      GNUNET_CONTAINER_multihashmap_put (udp_services,
 343                                                         (GNUNET_HashCode *)
 344                                                         desc, serv,
 345                                                         GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 346
 347       state->serv = serv;
 348     }
 349   }
 350
 351   /* send udp-packet back */
 352   len =
 353       sizeof (struct GNUNET_MessageHeader) + sizeof (GNUNET_HashCode) +
 354       ntohs (udp->len);
 355   struct GNUNET_MESH_Tunnel **ctunnel =
 356       GNUNET_malloc (sizeof (struct GNUNET_MESH_TUNNEL *) + len);
 357   *ctunnel = tunnel;
 358   msg = (struct GNUNET_MessageHeader *) (ctunnel + 1);
 359   msg->size = htons (len);
 360   msg->type =
 361       htons (state->type ==
 362              SERVICE ? GNUNET_MESSAGE_TYPE_VPN_SERVICE_UDP_BACK :
 363              GNUNET_MESSAGE_TYPE_VPN_REMOTE_UDP_BACK);
 364   GNUNET_HashCode *desc = (GNUNET_HashCode *) (msg + 1);
 365
 366   if (state->type == SERVICE)
 367     memcpy (desc, &state->desc, sizeof (GNUNET_HashCode));
 368   else
 369     memcpy (desc, &state->remote, sizeof (struct remote_addr));
 370   void *_udp = desc + 1;
 371
 372   memcpy (_udp, udp, ntohs (udp->len));
 373
 374   if (NULL == GNUNET_MESH_tunnel_get_data (tunnel))
 375   {
 376     /* No notify is pending */
 377     struct GNUNET_MESH_TransmitHandle *th =
 378         GNUNET_MESH_notify_transmit_ready (tunnel,
 379                                            GNUNET_NO,
 380                                            42,
 381                                            GNUNET_TIME_relative_divide
 382                                            (GNUNET_CONSTANTS_MAX_CORK_DELAY, 2),
 383                                            (const struct GNUNET_PeerIdentity *)
 384                                            NULL, len,
 385                                            send_udp_to_peer_notify_callback,
 386                                            ctunnel);
 387
 388     /* save the handle */
 389     GNUNET_MESH_tunnel_set_data (tunnel, th);
 390   }
 391   else
 392   {
 393     struct tunnel_notify_queue *head = GNUNET_MESH_tunnel_get_head (tunnel);
 394     struct tunnel_notify_queue *tail = GNUNET_MESH_tunnel_get_tail (tunnel);
 395
 396     struct tunnel_notify_queue *element =
 397         GNUNET_malloc (sizeof (struct tunnel_notify_queue));
 398     element->cls = ctunnel;
 399     element->len = len;
 400
 401     GNUNET_CONTAINER_DLL_insert_tail (head, tail, element);
 402     GNUNET_MESH_tunnel_set_head (tunnel, head);
 403     GNUNET_MESH_tunnel_set_tail (tunnel, tail);
 404   }
 405 }
 406
 407 /**
 408  * @brief Handles a TCP-Packet received from the helper.
 409  *
 410  * @param tcp A pointer to the Packet
 411  * @param dadr The IP-Destination-address
 412  * @param addrlen The length of the address
 413  * @param version 4 or 6
 414  * @param pktlen the length of the packet, including its header
 415  */
 416 static void
 417 tcp_from_helper (struct tcp_pkt *tcp, unsigned char *dadr, size_t addrlen,
 418                  size_t pktlen)
 419 {
 420   struct redirect_info u_i;
 421   struct GNUNET_MESH_Tunnel *tunnel;
 422   uint32_t len;
 423   struct GNUNET_MessageHeader *msg;
 424
 425   memset (&u_i, 0, sizeof (struct redirect_info));
 426
 427   memcpy (&u_i.addr, dadr, addrlen);
 428   u_i.pt = tcp->dpt;
 429
 430   /* get tunnel and service-descriptor from this */
 431   GNUNET_HashCode hash;
 432
 433   hash_redirect_info (&hash, &u_i, addrlen);
 434
 435   struct redirect_state *state =
 436       GNUNET_CONTAINER_multihashmap_get (tcp_connections, &hash);
 437
 438   if (state == NULL)
 439   {
 440     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 441                 "No mapping for this connection; hash is %x\n",
 442                 *((uint32_t *) & hash));
 443     return;
 444   }
 445
 446   /* Mark this connection as freshly used */
 447   GNUNET_CONTAINER_heap_update_cost (tcp_connections_heap, state->heap_node,
 448                                      GNUNET_TIME_absolute_get ().abs_value);
 449
 450   tunnel = state->tunnel;
 451
 452   if (state->type == SERVICE)
 453   {
 454     /* check if spt == serv.remote if yes: set spt = serv.myport ("nat") */
 455     if (ntohs (tcp->spt) == state->serv->remote_port)
 456     {
 457       tcp->spt = htons (state->serv->my_port);
 458     }
 459     else
 460     {
 461       // This is an illegal packet.
 462       return;
 463     }
 464   }
 465
 466   /* send tcp-packet back */
 467   len =
 468       sizeof (struct GNUNET_MessageHeader) + sizeof (GNUNET_HashCode) + pktlen;
 469   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "len: %d\n", pktlen);
 470   struct GNUNET_MESH_Tunnel **ctunnel =
 471       GNUNET_malloc (sizeof (struct GNUNET_MESH_TUNNEL *) + len);
 472   *ctunnel = tunnel;
 473   msg = (struct GNUNET_MessageHeader *) (ctunnel + 1);
 474   msg->size = htons (len);
 475   msg->type =
 476       htons (state->type ==
 477              SERVICE ? GNUNET_MESSAGE_TYPE_VPN_SERVICE_TCP_BACK :
 478              GNUNET_MESSAGE_TYPE_VPN_REMOTE_TCP_BACK);
 479   GNUNET_HashCode *desc = (GNUNET_HashCode *) (msg + 1);
 480
 481   if (state->type == SERVICE)
 482     memcpy (desc, &state->desc, sizeof (GNUNET_HashCode));
 483   else
 484     memcpy (desc, &state->remote, sizeof (struct remote_addr));
 485   void *_tcp = desc + 1;
 486
 487   memcpy (_tcp, tcp, pktlen);
 488
 489   if (NULL == GNUNET_MESH_tunnel_get_data (tunnel))
 490   {
 491     /* No notify is pending */
 492     struct GNUNET_MESH_TransmitHandle *th =
 493         GNUNET_MESH_notify_transmit_ready (tunnel,
 494                                            GNUNET_NO,
 495                                            42,
 496                                            GNUNET_TIME_relative_divide
 497                                            (GNUNET_CONSTANTS_MAX_CORK_DELAY, 2),
 498                                            (const struct GNUNET_PeerIdentity *)
 499                                            NULL,
 500                                            len,
 501                                            send_udp_to_peer_notify_callback,
 502                                            ctunnel);
 503
 504     /* save the handle */
 505     GNUNET_MESH_tunnel_set_data (tunnel, th);
 506   }
 507   else
 508   {
 509     struct tunnel_notify_queue *head = GNUNET_MESH_tunnel_get_head (tunnel);
 510     struct tunnel_notify_queue *tail = GNUNET_MESH_tunnel_get_tail (tunnel);
 511
 512     struct tunnel_notify_queue *element =
 513         GNUNET_malloc (sizeof (struct tunnel_notify_queue));
 514     element->cls = ctunnel;
 515     element->len = len;
 516
 517     GNUNET_CONTAINER_DLL_insert_tail (head, tail, element);
 518   }
 519 }
 520
 521
 522 /**
 523  * Receive packets from the helper-process
 524  */
 525 static void
 526 message_token (void *cls __attribute__ ((unused)), void *client
 527                __attribute__ ((unused)),
 528                const struct GNUNET_MessageHeader *message)
 529 {
 530   GNUNET_assert (ntohs (message->type) == GNUNET_MESSAGE_TYPE_VPN_HELPER);
 531
 532   struct tun_pkt *pkt_tun = (struct tun_pkt *) message;
 533
 534   /* ethertype is ipv6 */
 535   if (ntohs (pkt_tun->tun.type) == 0x86dd)
 536   {
 537     struct ip6_pkt *pkt6 = (struct ip6_pkt *) pkt_tun;
 538
 539     if (0x11 == pkt6->ip6_hdr.nxthdr)
 540       udp_from_helper (&((struct ip6_udp *) pkt6)->udp_hdr,
 541                        (unsigned char *) &pkt6->ip6_hdr.dadr, 16);
 542     else if (0x06 == pkt6->ip6_hdr.nxthdr)
 543       tcp_from_helper (&((struct ip6_tcp *) pkt6)->tcp_hdr,
 544                        (unsigned char *) &pkt6->ip6_hdr.dadr, 16,
 545                        ntohs (pkt6->ip6_hdr.paylgth));
 546   }
 547   else if (ntohs (pkt_tun->tun.type) == 0x0800)
 548   {
 549     struct ip_pkt *pkt4 = (struct ip_pkt *) pkt_tun;
 550     uint32_t tmp = pkt4->ip_hdr.dadr;
 551
 552     if (0x11 == pkt4->ip_hdr.proto)
 553       udp_from_helper (&((struct ip_udp *) pkt4)->udp_hdr,
 554                        (unsigned char *) &tmp, 4);
 555     else if (0x06 == pkt4->ip_hdr.proto)
 556     {
 557       size_t pktlen = ntohs (pkt4->ip_hdr.tot_lngth);
 558
 559       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "tot: %d\n", pktlen);
 560       pktlen -= 4 * pkt4->ip_hdr.hdr_lngth;
 561       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "-hdr: %d\n", pktlen);
 562       tcp_from_helper (&((struct ip_tcp *) pkt4)->tcp_hdr,
 563                        (unsigned char *) &tmp, 4, pktlen);
 564     }
 565   }
 566   else
 567   {
 568     return;
 569   }
 570 }
 571
 572 /**
 573  * Reads the configuration servicecfg and populates udp_services
 574  *
 575  * @param cls unused
 576  * @param section name of section in config, equal to hostname
 577  */
 578 static void
 579 read_service_conf (void *cls __attribute__ ((unused)), const char *section)
 580 {
 581   if ((strlen (section) < 8) ||
 582       (0 != strcmp (".gnunet.", section + (strlen (section) - 8))))
 583     return;
 584
 585   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Parsing dns-name %d %s %s\n",
 586               strlen (section), section, section + (strlen (section) - 8));
 587
 588   char *cpy;
 589   char *redirect;
 590   char *hostname;
 591   char *hostport;
 592   uint16_t *desc = alloca (sizeof (GNUNET_HashCode) + 2);
 593
 594   GNUNET_CRYPTO_hash (section, strlen (section) + 1,
 595                       (GNUNET_HashCode *) (desc + 1));
 596
 597 #define TCP 2
 598 #define UDP 1
 599
 600   int proto = UDP;
 601
 602   do
 603   {
 604     if (proto == UDP &&
 605         (GNUNET_OK !=
 606          GNUNET_CONFIGURATION_get_value_string (cfg, section, "UDP_REDIRECTS",
 607                                                 &cpy)))
 608       goto next;
 609     else if (proto == TCP &&
 610              (GNUNET_OK !=
 611               GNUNET_CONFIGURATION_get_value_string (cfg, section,
 612                                                      "TCP_REDIRECTS", &cpy)))
 613       goto next;
 614
 615     for (redirect = strtok (cpy, " "); redirect != NULL;
 616          redirect = strtok (NULL, " "))
 617     {
 618       if (NULL == (hostname = strstr (redirect, ":")))
 619       {
 620         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 621                     "Warning: option %s is not formatted correctly!\n",
 622                     redirect);
 623         continue;
 624       }
 625       hostname[0] = '\0';
 626       hostname++;
 627       if (NULL == (hostport = strstr (hostname, ":")))
 628       {
 629         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 630                     "Warning: option %s is not formatted correctly!\n",
 631                     redirect);
 632         continue;
 633       }
 634       hostport[0] = '\0';
 635       hostport++;
 636
 637       int local_port = atoi (redirect);
 638
 639       if (!((local_port > 0) && (local_port < 65536)))
 640         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 641                     "Warning: %s is not a correct port.", redirect);
 642
 643       *desc = local_port;
 644
 645       struct redirect_service *serv =
 646           GNUNET_malloc (sizeof (struct redirect_service));
 647       memset (serv, 0, sizeof (struct redirect_service));
 648       serv->my_port = local_port;
 649
 650       if (0 == strcmp ("localhost4", hostname))
 651       {
 652         serv->version = 4;
 653
 654         char *ip4addr;
 655
 656         GNUNET_assert (GNUNET_OK ==
 657                        GNUNET_CONFIGURATION_get_value_string (cfg, "exit",
 658                                                               "IPV4ADDR",
 659                                                               &ip4addr));
 660         GNUNET_assert (1 == inet_pton (AF_INET, ip4addr, serv->v4.ip4address));
 661         GNUNET_free (ip4addr);
 662       }
 663       else if (0 == strcmp ("localhost6", hostname))
 664       {
 665         serv->version = 6;
 666
 667         char *ip6addr;
 668
 669         GNUNET_assert (GNUNET_OK ==
 670                        GNUNET_CONFIGURATION_get_value_string (cfg, "exit",
 671                                                               "IPV6ADDR",
 672                                                               &ip6addr));
 673         GNUNET_assert (1 == inet_pton (AF_INET6, ip6addr, serv->v6.ip6address));
 674         GNUNET_free (ip6addr);
 675       }
 676       else
 677       {
 678         // TODO Lookup, yadayadayada
 679         GNUNET_assert (0);
 680       }
 681       serv->remote_port = atoi (hostport);
 682       if (UDP == proto)
 683         GNUNET_assert (GNUNET_OK ==
 684                        GNUNET_CONTAINER_multihashmap_put (udp_services,
 685                                                           (GNUNET_HashCode *)
 686                                                           desc, serv,
 687                                                           GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 688       else
 689         GNUNET_assert (GNUNET_OK ==
 690                        GNUNET_CONTAINER_multihashmap_put (tcp_services,
 691                                                           (GNUNET_HashCode *)
 692                                                           desc, serv,
 693                                                           GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
 694
 695     }
 696     GNUNET_free (cpy);
 697 next:
 698     proto = (proto == UDP) ? TCP : UDP;
 699   }
 700   while (proto != UDP);
 701 }
 702
 703 /**
 704  * Start the helper-process
 705  *
 706  * If cls != NULL it is assumed that this function is called as a result of a dying
 707  * helper. cls is then taken as handle to the old helper and is cleaned up.
 708  */
 709 static void
 710 start_helper_and_schedule (void *cls,
 711                            const struct GNUNET_SCHEDULER_TaskContext *tc)
 712 {
 713   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 714     return;
 715
 716   if (cls != NULL)
 717     cleanup_helper (cls);
 718   cls = NULL;
 719
 720   char *ifname;
 721   char *ipv6addr;
 722   char *ipv6prefix;
 723   char *ipv4addr;
 724   char *ipv4mask;
 725
 726   if (GNUNET_SYSERR ==
 727       GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IFNAME", &ifname))
 728   {
 729     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 730                 "No entry 'IFNAME' in configuration!\n");
 731     exit (1);
 732   }
 733
 734   if (GNUNET_SYSERR ==
 735       GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV6ADDR",
 736                                              &ipv6addr))
 737   {
 738     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 739                 "No entry 'IPV6ADDR' in configuration!\n");
 740     exit (1);
 741   }
 742
 743   if (GNUNET_SYSERR ==
 744       GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV6PREFIX",
 745                                              &ipv6prefix))
 746   {
 747     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 748                 "No entry 'IPV6PREFIX' in configuration!\n");
 749     exit (1);
 750   }
 751
 752   if (GNUNET_SYSERR ==
 753       GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV4ADDR",
 754                                              &ipv4addr))
 755   {
 756     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 757                 "No entry 'IPV4ADDR' in configuration!\n");
 758     exit (1);
 759   }
 760
 761   if (GNUNET_SYSERR ==
 762       GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV4MASK",
 763                                              &ipv4mask))
 764   {
 765     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 766                 "No entry 'IPV4MASK' in configuration!\n");
 767     exit (1);
 768   }
 769
 770   /* Start the helper
 771    * Messages get passed to the function message_token
 772    * When the helper dies, this function will be called again with the
 773    * helper_handle as cls.
 774    */
 775   helper_handle =
 776       start_helper (ifname, ipv6addr, ipv6prefix, ipv4addr, ipv4mask,
 777                     "exit-gnunet", start_helper_and_schedule, message_token,
 778                     NULL);
 779
 780   GNUNET_free (ipv6addr);
 781   GNUNET_free (ipv6prefix);
 782   GNUNET_free (ipv4addr);
 783   GNUNET_free (ipv4mask);
 784   GNUNET_free (ifname);
 785 }
 786
 787 static void
 788 prepare_ipv4_packet (ssize_t len, ssize_t pktlen, void *payload,
 789                      uint16_t protocol, void *ipaddress, void *tunnel,
 790                      struct redirect_state *state, struct ip_pkt *pkt4)
 791 {
 792   uint32_t tmp, tmp2;
 793
 794   pkt4->shdr.type = htons (GNUNET_MESSAGE_TYPE_VPN_HELPER);
 795   pkt4->shdr.size = htons (len);
 796   pkt4->tun.flags = 0;
 797   pkt4->tun.type = htons (0x0800);
 798
 799   memcpy (&pkt4->data, payload, pktlen);
 800
 801   pkt4->ip_hdr.version = 4;
 802   pkt4->ip_hdr.hdr_lngth = 5;
 803   pkt4->ip_hdr.diff_serv = 0;
 804   pkt4->ip_hdr.tot_lngth = htons (20 + pktlen);
 805   pkt4->ip_hdr.ident = 0;
 806   pkt4->ip_hdr.flags = 0;
 807   pkt4->ip_hdr.frag_off = 0;
 808   pkt4->ip_hdr.ttl = 255;
 809   pkt4->ip_hdr.proto = protocol;
 810   pkt4->ip_hdr.chks = 0;        /* Will be calculated later */
 811
 812   memcpy (&tmp, ipaddress, 4);
 813   pkt4->ip_hdr.dadr = tmp;
 814
 815   /* Generate a new src-address */
 816   char *ipv4addr;
 817   char *ipv4mask;
 818
 819   GNUNET_assert (GNUNET_OK ==
 820                  GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV4ADDR",
 821                                                         &ipv4addr));
 822   GNUNET_assert (GNUNET_OK ==
 823                  GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV4MASK",
 824                                                         &ipv4mask));
 825   inet_pton (AF_INET, ipv4addr, &tmp);
 826   inet_pton (AF_INET, ipv4mask, &tmp2);
 827   GNUNET_free (ipv4addr);
 828   GNUNET_free (ipv4mask);
 829
 830   /* This should be a noop */
 831   tmp = tmp & tmp2;
 832
 833   tmp |= ntohl (*((uint32_t *) tunnel)) & (~tmp2);
 834
 835   pkt4->ip_hdr.sadr = tmp;
 836
 837   memcpy (&state->redirect_info.addr, &tmp, 4);
 838   if (0x11 == protocol)
 839   {
 840     struct ip_udp *pkt4_udp = (struct ip_udp *) pkt4;
 841
 842     state->redirect_info.pt = pkt4_udp->udp_hdr.spt;
 843
 844     pkt4_udp->udp_hdr.crc = 0;  /* Optional for IPv4 */
 845   }
 846   else if (0x06 == protocol)
 847   {
 848     struct ip_tcp *pkt4_tcp = (struct ip_tcp *) pkt4;
 849
 850     state->redirect_info.pt = pkt4_tcp->tcp_hdr.spt;
 851
 852     pkt4_tcp->tcp_hdr.crc = 0;
 853     uint32_t sum = 0;
 854
 855     tmp = pkt4->ip_hdr.sadr;
 856     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 857     tmp = pkt4->ip_hdr.dadr;
 858     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 859
 860     tmp = (protocol << 16) | (0xffff & pktlen);
 861
 862     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "line: %08x, %x \n", tmp,
 863                 (0xffff & pktlen));
 864
 865     tmp = htonl (tmp);
 866
 867     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 868
 869     sum =
 870         calculate_checksum_update (sum, (uint16_t *) & pkt4_tcp->tcp_hdr,
 871                                    pktlen);
 872     pkt4_tcp->tcp_hdr.crc = calculate_checksum_end (sum);
 873   }
 874
 875   pkt4->ip_hdr.chks =
 876       calculate_ip_checksum ((uint16_t *) & pkt4->ip_hdr, 5 * 4);
 877 }
 878
 879 static void
 880 prepare_ipv6_packet (ssize_t len, ssize_t pktlen, void *payload,
 881                      uint16_t protocol, void *ipaddress, void *tunnel,
 882                      struct redirect_state *state, struct ip6_pkt *pkt6)
 883 {
 884   uint32_t tmp;
 885
 886   pkt6->shdr.type = htons (GNUNET_MESSAGE_TYPE_VPN_HELPER);
 887   pkt6->shdr.size = htons (len);
 888   pkt6->tun.flags = 0;
 889
 890   pkt6->tun.type = htons (0x86dd);
 891
 892   memcpy (&pkt6->data, payload, pktlen);
 893
 894   pkt6->ip6_hdr.version = 6;
 895   pkt6->ip6_hdr.nxthdr = protocol;
 896   pkt6->ip6_hdr.paylgth = htons (pktlen);
 897   pkt6->ip6_hdr.hoplmt = 64;
 898
 899   memcpy (pkt6->ip6_hdr.dadr, ipaddress, 16);
 900
 901   /* Generate a new src-address
 902    * This takes as much from the address of the tunnel as fits into
 903    * the host-mask*/
 904   char *ipv6addr;
 905   unsigned long long ipv6prefix;
 906
 907   GNUNET_assert (GNUNET_OK ==
 908                  GNUNET_CONFIGURATION_get_value_string (cfg, "exit", "IPV6ADDR",
 909                                                         &ipv6addr));
 910   GNUNET_assert (GNUNET_OK ==
 911                  GNUNET_CONFIGURATION_get_value_number (cfg, "exit",
 912                                                         "IPV6PREFIX",
 913                                                         &ipv6prefix));
 914   GNUNET_assert (ipv6prefix < 127);
 915   ipv6prefix = (ipv6prefix + 7) / 8;
 916
 917   inet_pton (AF_INET6, ipv6addr, &pkt6->ip6_hdr.sadr);
 918   GNUNET_free (ipv6addr);
 919
 920   if (ipv6prefix < (16 - sizeof (void *)))
 921     ipv6prefix = 16 - sizeof (void *);
 922
 923   unsigned int offset = ipv6prefix - (16 - sizeof (void *));
 924
 925   memcpy ((((char *) &pkt6->ip6_hdr.sadr)) + ipv6prefix,
 926           ((char *) &tunnel) + offset, 16 - ipv6prefix);
 927
 928   /* copy the needed information into the state */
 929   memcpy (&state->redirect_info.addr, &pkt6->ip6_hdr.sadr, 16);
 930
 931   if (0x11 == protocol)
 932   {
 933     struct ip6_udp *pkt6_udp = (struct ip6_udp *) pkt6;
 934
 935     state->redirect_info.pt = pkt6_udp->udp_hdr.spt;
 936
 937     pkt6_udp->udp_hdr.crc = 0;
 938     uint32_t sum = 0;
 939
 940     sum =
 941         calculate_checksum_update (sum, (uint16_t *) & pkt6_udp->ip6_hdr.sadr,
 942                                    16);
 943     sum =
 944         calculate_checksum_update (sum, (uint16_t *) & pkt6_udp->ip6_hdr.dadr,
 945                                    16);
 946     tmp = (htons (pktlen) & 0xffff);
 947     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 948     tmp = htons (((pkt6_udp->ip6_hdr.nxthdr & 0x00ff)));
 949     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 950
 951     sum =
 952         calculate_checksum_update (sum, (uint16_t *) & pkt6_udp->udp_hdr,
 953                                    ntohs (pkt6_udp->udp_hdr.len));
 954     pkt6_udp->udp_hdr.crc = calculate_checksum_end (sum);
 955   }
 956   else if (0x06 == protocol)
 957   {
 958     struct ip6_tcp *pkt6_tcp = (struct ip6_tcp *) pkt6;
 959
 960     state->redirect_info.pt = pkt6_tcp->tcp_hdr.spt;
 961
 962     pkt6_tcp->tcp_hdr.crc = 0;
 963     uint32_t sum = 0;
 964
 965     sum =
 966         calculate_checksum_update (sum, (uint16_t *) & pkt6->ip6_hdr.sadr, 16);
 967     sum =
 968         calculate_checksum_update (sum, (uint16_t *) & pkt6->ip6_hdr.dadr, 16);
 969     tmp = htonl (pktlen);
 970     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 971     tmp = htonl (((pkt6->ip6_hdr.nxthdr & 0x000000ff)));
 972     sum = calculate_checksum_update (sum, (uint16_t *) & tmp, 4);
 973
 974     sum =
 975         calculate_checksum_update (sum, (uint16_t *) & pkt6_tcp->tcp_hdr,
 976                                    ntohs (pkt6->ip6_hdr.paylgth));
 977     pkt6_tcp->tcp_hdr.crc = calculate_checksum_end (sum);
 978   }
 979 }
 980
 981 /**
 982  * The messages are one GNUNET_HashCode for the service followed by a struct tcp_pkt
 983  */
 984 static int
 985 receive_tcp_service (void *cls
 986                      __attribute__ ((unused)),
 987                      struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx
 988                      __attribute__ ((unused)),
 989                      const struct GNUNET_PeerIdentity *sender
 990                      __attribute__ ((unused)),
 991                      const struct GNUNET_MessageHeader *message,
 992                      const struct GNUNET_TRANSPORT_ATS_Information *atsi
 993                      __attribute__ ((unused)))
 994 {
 995   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received TCP-Packet\n");
 996   GNUNET_HashCode *desc = (GNUNET_HashCode *) (message + 1);
 997   struct tcp_pkt *pkt = (struct tcp_pkt *) (desc + 1);
 998   unsigned int pkt_len =
 999       ntohs (message->size) - sizeof (struct GNUNET_MessageHeader) -
1000       sizeof (GNUNET_HashCode);
1001
1002   /** Get the configuration from the services-hashmap.
1003    *
1004    * Which service is needed only depends on the service-descriptor and the
1005    * destination-port
1006    */
1007   uint16_t *tcp_desc = alloca (sizeof (GNUNET_HashCode) + 2);
1008
1009   memcpy (tcp_desc + 1, desc, sizeof (GNUNET_HashCode));
1010   *tcp_desc = ntohs (pkt->dpt);
1011   struct redirect_service *serv =
1012       GNUNET_CONTAINER_multihashmap_get (tcp_services,
1013                                          (GNUNET_HashCode *) tcp_desc);
1014
1015   if (NULL == serv)
1016   {
1017     GNUNET_log (GNUNET_ERROR_TYPE_INFO, "No service found for TCP dpt %d!\n",
1018                 *tcp_desc);
1019     return GNUNET_YES;
1020   }
1021
1022   pkt->dpt = htons (serv->remote_port);
1023
1024   /*
1025    * At this point it would be possible to check against some kind of ACL.
1026    */
1027
1028   char *buf;
1029   size_t len;
1030
1031   /* Prepare the state.
1032    * This will be saved in the hashmap, so that the receiving procedure knows
1033    * through which tunnel this connection has to be routed.
1034    */
1035   struct redirect_state *state = GNUNET_malloc (sizeof (struct redirect_state));
1036
1037   memset (state, 0, sizeof (struct redirect_state));
1038   state->tunnel = tunnel;
1039   state->serv = serv;
1040   state->type = SERVICE;
1041   state->hashmap = tcp_connections;
1042   memcpy (&state->desc, desc, sizeof (GNUNET_HashCode));
1043
1044   len =
1045       sizeof (struct GNUNET_MessageHeader) + sizeof (struct pkt_tun) +
1046       sizeof (struct ip6_hdr) + pkt_len;
1047   buf = alloca (len);
1048
1049   memset (buf, 0, len);
1050
1051   switch (serv->version)
1052   {
1053   case 4:
1054     prepare_ipv4_packet (len, pkt_len, pkt, 0x06,       /* TCP */
1055                          &serv->v4.ip4address, tunnel, state,
1056                          (struct ip_pkt *) buf);
1057     break;
1058   case 6:
1059     prepare_ipv6_packet (len, pkt_len, pkt, 0x06,       /* TCP */
1060                          &serv->v6.ip6address, tunnel, state,
1061                          (struct ip6_pkt *) buf);
1062
1063     break;
1064   default:
1065     GNUNET_assert (0);
1066     break;
1067   }
1068
1069   hash_redirect_info (&state->hash, &state->redirect_info,
1070                       serv->version == 4 ? 4 : 16);
1071
1072   if (GNUNET_NO ==
1073       GNUNET_CONTAINER_multihashmap_contains (tcp_connections, &state->hash))
1074   {
1075     GNUNET_CONTAINER_multihashmap_put (tcp_connections, &state->hash, state,
1076                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
1077
1078     state->heap_node =
1079         GNUNET_CONTAINER_heap_insert (tcp_connections_heap, state,
1080                                       GNUNET_TIME_absolute_get ().abs_value);
1081
1082     if (GNUNET_CONTAINER_heap_get_size (tcp_connections_heap) >
1083         max_tcp_connections)
1084       GNUNET_SCHEDULER_add_now (collect_connections, tcp_connections_heap);
1085   }
1086   else
1087     GNUNET_free (state);
1088
1089   (void) GNUNET_DISK_file_write (helper_handle->fh_to_helper, buf, len);
1090   return GNUNET_YES;
1091 }
1092
1093 static int
1094 receive_tcp_remote (void *cls
1095                     __attribute__ ((unused)), struct GNUNET_MESH_Tunnel *tunnel,
1096                     void **tunnel_ctx
1097                     __attribute__ ((unused)),
1098                     const struct GNUNET_PeerIdentity *sender
1099                     __attribute__ ((unused)),
1100                     const struct GNUNET_MessageHeader *message,
1101                     const struct GNUNET_TRANSPORT_ATS_Information *atsi
1102                     __attribute__ ((unused)))
1103 {
1104   GNUNET_HashCode *desc = (GNUNET_HashCode *) (message + 1);
1105   struct tcp_pkt *pkt = (struct tcp_pkt *) (desc + 1);
1106   struct remote_addr *s = (struct remote_addr *) desc;
1107   char *buf;
1108   size_t len;
1109   unsigned int pkt_len =
1110       ntohs (message->size) - sizeof (struct GNUNET_MessageHeader) -
1111       sizeof (GNUNET_HashCode);
1112
1113   struct redirect_state *state = GNUNET_malloc (sizeof (struct redirect_state));
1114
1115   memset (state, 0, sizeof (struct redirect_state));
1116   state->tunnel = tunnel;
1117   state->type = REMOTE;
1118   state->hashmap = tcp_connections;
1119   memcpy (&state->remote, s, sizeof (struct remote_addr));
1120
1121   len =
1122       sizeof (struct GNUNET_MessageHeader) + sizeof (struct pkt_tun) +
1123       sizeof (struct ip6_hdr) + pkt_len;
1124   buf = alloca (len);
1125
1126   memset (buf, 0, len);
1127
1128   switch (s->addrlen)
1129   {
1130   case 4:
1131     prepare_ipv4_packet (len, pkt_len, pkt, 0x06,       /* TCP */
1132                          &s->addr, tunnel, state, (struct ip_pkt *) buf);
1133     break;
1134   case 16:
1135     prepare_ipv6_packet (len, pkt_len, pkt, 0x06,       /* TCP */
1136                          &s->addr, tunnel, state, (struct ip6_pkt *) buf);
1137     break;
1138   default:
1139     return GNUNET_SYSERR;
1140     break;
1141   }
1142
1143   hash_redirect_info (&state->hash, &state->redirect_info, s->addrlen);
1144
1145   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Packet from remote; hash is %x\n",
1146               *((uint32_t *) & state->hash));
1147
1148   if (GNUNET_NO ==
1149       GNUNET_CONTAINER_multihashmap_contains (tcp_connections, &state->hash))
1150   {
1151     GNUNET_CONTAINER_multihashmap_put (tcp_connections, &state->hash, state,
1152                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
1153
1154     state->heap_node =
1155         GNUNET_CONTAINER_heap_insert (tcp_connections_heap, state,
1156                                       GNUNET_TIME_absolute_get ().abs_value);
1157
1158     if (GNUNET_CONTAINER_heap_get_size (tcp_connections_heap) >
1159         max_tcp_connections)
1160       GNUNET_SCHEDULER_add_now (collect_connections, tcp_connections_heap);
1161   }
1162   else
1163     GNUNET_free (state);
1164
1165   (void) GNUNET_DISK_file_write (helper_handle->fh_to_helper, buf, len);
1166   return GNUNET_YES;
1167
1168 }
1169
1170 static int
1171 receive_udp_remote (void *cls
1172                     __attribute__ ((unused)), struct GNUNET_MESH_Tunnel *tunnel,
1173                     void **tunnel_ctx
1174                     __attribute__ ((unused)),
1175                     const struct GNUNET_PeerIdentity *sender
1176                     __attribute__ ((unused)),
1177                     const struct GNUNET_MessageHeader *message,
1178                     const struct GNUNET_TRANSPORT_ATS_Information *atsi
1179                     __attribute__ ((unused)))
1180 {
1181   GNUNET_HashCode *desc = (GNUNET_HashCode *) (message + 1);
1182   struct udp_pkt *pkt = (struct udp_pkt *) (desc + 1);
1183   struct remote_addr *s = (struct remote_addr *) desc;
1184   char *buf;
1185   size_t len;
1186
1187   GNUNET_assert (ntohs (pkt->len) ==
1188                  ntohs (message->size) - sizeof (struct GNUNET_MessageHeader) -
1189                  sizeof (GNUNET_HashCode));
1190
1191   /* Prepare the state.
1192    * This will be saved in the hashmap, so that the receiving procedure knows
1193    * through which tunnel this connection has to be routed.
1194    */
1195   struct redirect_state *state = GNUNET_malloc (sizeof (struct redirect_state));
1196
1197   memset (state, 0, sizeof (struct redirect_state));
1198   state->tunnel = tunnel;
1199   state->hashmap = udp_connections;
1200   state->type = REMOTE;
1201   memcpy (&state->remote, s, sizeof (struct remote_addr));
1202
1203   len =
1204       sizeof (struct GNUNET_MessageHeader) + sizeof (struct pkt_tun) +
1205       sizeof (struct ip6_hdr) + ntohs (pkt->len);
1206   buf = alloca (len);
1207
1208   memset (buf, 0, len);
1209
1210   switch (s->addrlen)
1211   {
1212   case 4:
1213     prepare_ipv4_packet (len, ntohs (pkt->len), pkt, 0x11,      /* UDP */
1214                          &s->addr, tunnel, state, (struct ip_pkt *) buf);
1215     break;
1216   case 16:
1217     prepare_ipv6_packet (len, ntohs (pkt->len), pkt, 0x11,      /* UDP */
1218                          &s->addr, tunnel, state, (struct ip6_pkt *) buf);
1219     break;
1220   default:
1221     GNUNET_assert (0);
1222     break;
1223   }
1224
1225   hash_redirect_info (&state->hash, &state->redirect_info, s->addrlen);
1226
1227   if (GNUNET_NO ==
1228       GNUNET_CONTAINER_multihashmap_contains (udp_connections, &state->hash))
1229   {
1230     GNUNET_CONTAINER_multihashmap_put (udp_connections, &state->hash, state,
1231                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
1232
1233     state->heap_node =
1234         GNUNET_CONTAINER_heap_insert (udp_connections_heap, state,
1235                                       GNUNET_TIME_absolute_get ().abs_value);
1236
1237     if (GNUNET_CONTAINER_heap_get_size (udp_connections_heap) >
1238         max_udp_connections)
1239       GNUNET_SCHEDULER_add_now (collect_connections, udp_connections_heap);
1240   }
1241   else
1242     GNUNET_free (state);
1243
1244   (void) GNUNET_DISK_file_write (helper_handle->fh_to_helper, buf, len);
1245   return GNUNET_YES;
1246 }
1247
1248 /**
1249  * The messages are one GNUNET_HashCode for the service, followed by a struct udp_pkt
1250  */
1251 static int
1252 receive_udp_service (void *cls
1253                      __attribute__ ((unused)),
1254                      struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx
1255                      __attribute__ ((unused)),
1256                      const struct GNUNET_PeerIdentity *sender
1257                      __attribute__ ((unused)),
1258                      const struct GNUNET_MessageHeader *message,
1259                      const struct GNUNET_TRANSPORT_ATS_Information *atsi
1260                      __attribute__ ((unused)))
1261 {
1262   GNUNET_HashCode *desc = (GNUNET_HashCode *) (message + 1);
1263   struct udp_pkt *pkt = (struct udp_pkt *) (desc + 1);
1264
1265   GNUNET_assert (ntohs (pkt->len) ==
1266                  ntohs (message->size) - sizeof (struct GNUNET_MessageHeader) -
1267                  sizeof (GNUNET_HashCode));
1268
1269   /* Get the configuration from the hashmap */
1270   uint16_t *udp_desc = alloca (sizeof (GNUNET_HashCode) + 2);
1271
1272   memcpy (udp_desc + 1, desc, sizeof (GNUNET_HashCode));
1273   *udp_desc = ntohs (pkt->dpt);
1274   struct redirect_service *serv =
1275       GNUNET_CONTAINER_multihashmap_get (udp_services,
1276                                          (GNUNET_HashCode *) udp_desc);
1277
1278   if (NULL == serv)
1279   {
1280     GNUNET_log (GNUNET_ERROR_TYPE_INFO, "No service found for UDP dpt %d!\n",
1281                 *udp_desc);
1282     return GNUNET_YES;
1283   }
1284
1285   pkt->dpt = htons (serv->remote_port);
1286
1287   /*
1288    * At this point it would be possible to check against some kind of ACL.
1289    */
1290
1291   char *buf;
1292   size_t len;
1293
1294   /* Prepare the state.
1295    * This will be saved in the hashmap, so that the receiving procedure knows
1296    * through which tunnel this connection has to be routed.
1297    */
1298   struct redirect_state *state = GNUNET_malloc (sizeof (struct redirect_state));
1299
1300   memset (state, 0, sizeof (struct redirect_state));
1301   state->tunnel = tunnel;
1302   state->serv = serv;
1303   state->type = SERVICE;
1304   state->hashmap = udp_connections;
1305   memcpy (&state->desc, desc, sizeof (GNUNET_HashCode));
1306
1307   len =
1308       sizeof (struct GNUNET_MessageHeader) + sizeof (struct pkt_tun) +
1309       sizeof (struct ip6_hdr) + ntohs (pkt->len);
1310   buf = alloca (len);
1311
1312   memset (buf, 0, len);
1313
1314   switch (serv->version)
1315   {
1316   case 4:
1317     prepare_ipv4_packet (len, ntohs (pkt->len), pkt, 0x11,      /* UDP */
1318                          &serv->v4.ip4address, tunnel, state,
1319                          (struct ip_pkt *) buf);
1320     break;
1321   case 6:
1322     prepare_ipv6_packet (len, ntohs (pkt->len), pkt, 0x11,      /* UDP */
1323                          &serv->v6.ip6address, tunnel, state,
1324                          (struct ip6_pkt *) buf);
1325
1326     break;
1327   default:
1328     GNUNET_assert (0);
1329     break;
1330   }
1331
1332   hash_redirect_info (&state->hash, &state->redirect_info,
1333                       serv->version == 4 ? 4 : 16);
1334
1335   if (GNUNET_NO ==
1336       GNUNET_CONTAINER_multihashmap_contains (udp_connections, &state->hash))
1337   {
1338     GNUNET_CONTAINER_multihashmap_put (udp_connections, &state->hash, state,
1339                                        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
1340
1341     state->heap_node =
1342         GNUNET_CONTAINER_heap_insert (udp_connections_heap, state,
1343                                       GNUNET_TIME_absolute_get ().abs_value);
1344
1345     if (GNUNET_CONTAINER_heap_get_size (udp_connections_heap) >
1346         max_udp_connections)
1347       GNUNET_SCHEDULER_add_now (collect_connections, udp_connections_heap);
1348   }
1349   else
1350     GNUNET_free (state);
1351
1352   (void) GNUNET_DISK_file_write (helper_handle->fh_to_helper, buf, len);
1353   return GNUNET_YES;
1354 }
1355
1356 static void
1357 connect_to_mesh ()
1358 {
1359   int udp, tcp;
1360   int handler_idx, app_idx;
1361
1362   udp = GNUNET_CONFIGURATION_get_value_yesno (cfg, "exit", "ENABLE_UDP");
1363   tcp = GNUNET_CONFIGURATION_get_value_yesno (cfg, "exit", "ENABLE_TCP");
1364
1365   static struct GNUNET_MESH_MessageHandler handlers[] = {
1366     {receive_udp_service, GNUNET_MESSAGE_TYPE_VPN_SERVICE_UDP, 0},
1367     {receive_tcp_service, GNUNET_MESSAGE_TYPE_VPN_SERVICE_TCP, 0},
1368     {NULL, 0, 0},
1369     {NULL, 0, 0},
1370     {NULL, 0, 0}
1371   };
1372
1373   static GNUNET_MESH_ApplicationType apptypes[] = {
1374     GNUNET_APPLICATION_TYPE_END,
1375     GNUNET_APPLICATION_TYPE_END,
1376     GNUNET_APPLICATION_TYPE_END
1377   };
1378
1379   app_idx = 0;
1380   handler_idx = 2;
1381
1382   if (GNUNET_YES == udp)
1383   {
1384     handlers[handler_idx].callback = receive_udp_remote;
1385     handlers[handler_idx].expected_size = 0;
1386     handlers[handler_idx].type = GNUNET_MESSAGE_TYPE_VPN_REMOTE_UDP;
1387     apptypes[app_idx] = GNUNET_APPLICATION_TYPE_INTERNET_UDP_GATEWAY;
1388     handler_idx++;
1389     app_idx++;
1390   }
1391
1392   if (GNUNET_YES == tcp)
1393   {
1394     handlers[handler_idx].callback = receive_tcp_remote;
1395     handlers[handler_idx].expected_size = 0;
1396     handlers[handler_idx].type = GNUNET_MESSAGE_TYPE_VPN_REMOTE_TCP;
1397     apptypes[app_idx] = GNUNET_APPLICATION_TYPE_INTERNET_TCP_GATEWAY;
1398     handler_idx++;
1399     app_idx++;
1400   }
1401
1402   mesh_handle = GNUNET_MESH_connect (cfg, NULL, NULL, handlers, apptypes);
1403 }
1404
1405 /**
1406  * @brief Main function that will be run by the scheduler.
1407  *
1408  * @param cls closure
1409  * @param args remaining command-line arguments
1410  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
1411  * @param cfg_ configuration
1412  */
1413 static void
1414 run (void *cls, char *const *args __attribute__ ((unused)), const char *cfgfile
1415      __attribute__ ((unused)), const struct GNUNET_CONFIGURATION_Handle *cfg_)
1416 {
1417   cfg = cfg_;
1418
1419   connect_to_mesh ();
1420
1421   udp_connections = GNUNET_CONTAINER_multihashmap_create (65536);
1422   udp_connections_heap =
1423       GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN);
1424   tcp_connections = GNUNET_CONTAINER_multihashmap_create (65536);
1425   tcp_connections_heap =
1426       GNUNET_CONTAINER_heap_create (GNUNET_CONTAINER_HEAP_ORDER_MIN);
1427   udp_services = GNUNET_CONTAINER_multihashmap_create (65536);
1428   tcp_services = GNUNET_CONTAINER_multihashmap_create (65536);
1429
1430   GNUNET_CONFIGURATION_get_value_number (cfg, "exit", "MAX_UDP_CONNECTIONS",
1431                                          &max_udp_connections);
1432   GNUNET_CONFIGURATION_get_value_number (cfg, "exit", "MAX_TCP_CONNECTIONS",
1433                                          &max_tcp_connections);
1434
1435   GNUNET_CONFIGURATION_iterate_sections (cfg, read_service_conf, NULL);
1436
1437   GNUNET_SCHEDULER_add_now (start_helper_and_schedule, NULL);
1438   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &cleanup, cls);
1439 }
1440
1441 /**
1442  * The main function to obtain template from gnunetd.
1443  *
1444  * @param argc number of arguments from the command line
1445  * @param argv command line arguments
1446  * @return 0 ok, 1 on error
1447  */
1448 int
1449 main (int argc, char *const *argv)
1450 {
1451   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
1452     GNUNET_GETOPT_OPTION_END
1453   };
1454
1455   return (GNUNET_OK ==
1456           GNUNET_PROGRAM_run (argc, argv, "exit", gettext_noop ("help text"),
1457                               options, &run, NULL)) ? ret : 1;
1458 }
1459
1460 /* end of gnunet-daemon-exit.c */