src/transport/plugin_transport_udp.c

   1 /*
   2      This file is part of GNUnet
   3      (C) 2010 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 2, 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 transport/plugin_transport_udp.c
  23  * @brief Implementation of the UDP NAT punching
  24  *        transport service
  25  * @author Christian Grothoff
  26  * @author Nathan Evans
  27  *
  28  * The idea with this transport is to connect gnunet peers to each other
  29  * when ONE is behind a NAT.  This is based on pwnat (http://samy.pl/pwnat)
  30  * created by Samy Kamkar.  When configured with the PWNAT options, this
  31  * transport will start a server daemon which sends dummy ICMP and UDP
  32  * messages out to a predefined address (typically 1.2.3.4).
  33  *
  34  * When a non-NAT'd peer (the client) learns of the NAT'd peer (the server)
  35  * address, it will send ICMP RESPONSES to the NAT'd peers external address.
  36  * The NAT box should forward these faked responses to the server, which
  37  * can then connect directly to the non-NAT'd peer.
  38  */
  39
  40 #include "platform.h"
  41 #include "gnunet_hello_lib.h"
  42 #include "gnunet_connection_lib.h"
  43 #include "gnunet_os_lib.h"
  44 #include "gnunet_peerinfo_service.h"
  45 #include "gnunet_protocols.h"
  46 #include "gnunet_resolver_service.h"
  47 #include "gnunet_server_lib.h"
  48 #include "gnunet_service_lib.h"
  49 #include "gnunet_signatures.h"
  50 #include "gnunet_statistics_service.h"
  51 #include "gnunet_transport_service.h"
  52 #include "plugin_transport.h"
  53 #include "transport.h"
  54
  55 #define DEBUG_UDP GNUNET_YES
  56
  57 #define MAX_PROBES 20
  58
  59 /*
  60  * Transport cost to peer, always 1 for UDP (direct connection)
  61  */
  62 #define UDP_DIRECT_DISTANCE 1
  63
  64 /**
  65  * Handle for request of hostname resolution, non-NULL if pending.
  66  */
  67 static struct GNUNET_RESOLVER_RequestHandle *hostname_dns;
  68
  69 /**
  70  * How long until we give up on transmitting the welcome message?
  71  */
  72 #define HOSTNAME_RESOLVE_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
  73
  74 /**
  75  * Starting port for listening and sending, eventually a config value
  76  */
  77 #define UDP_NAT_DEFAULT_PORT 22086
  78
  79 /**
  80  * UDP Message-Packet header.
  81  */
  82 struct UDPMessage
  83 {
  84   /**
  85    * Message header.
  86    */
  87   struct GNUNET_MessageHeader header;
  88
  89   /**
  90    * What is the identity of the sender (GNUNET_hash of public key)
  91    */
  92   struct GNUNET_PeerIdentity sender;
  93
  94 };
  95
  96
  97 /* Forward definition */
  98 struct Plugin;
  99
 100 struct PrettyPrinterContext
 101 {
 102   GNUNET_TRANSPORT_AddressStringCallback asc;
 103   void *asc_cls;
 104   uint16_t port;
 105 };
 106
 107 struct MessageQueue
 108 {
 109   /**
 110    * Linked List
 111    */
 112   struct MessageQueue *next;
 113
 114   /**
 115    * Session this message belongs to
 116    */
 117   struct PeerSession *session;
 118
 119   /**
 120    * Actual message to be sent
 121    */
 122   char *msgbuf;
 123
 124   /**
 125    * Size of message buffer to be sent
 126    */
 127   size_t msgbuf_size;
 128
 129   /**
 130    * When to discard this message
 131    */
 132   struct GNUNET_TIME_Absolute timeout;
 133
 134   /**
 135    * Continuation to call when this message goes out
 136    */
 137   GNUNET_TRANSPORT_TransmitContinuation cont;
 138
 139   /**
 140    * closure for continuation
 141    */
 142   void *cont_cls;
 143
 144 };
 145
 146 /**
 147  * UDP NAT Probe message definition
 148  */
 149 struct UDP_NAT_ProbeMessage
 150 {
 151   /**
 152    * Message header
 153    */
 154   struct GNUNET_MessageHeader header;
 155
 156 };
 157
 158 /**
 159  * UDP NAT Probe message reply definition
 160  */
 161 struct UDP_NAT_ProbeMessageReply
 162 {
 163   /**
 164    * Message header
 165    */
 166   struct GNUNET_MessageHeader header;
 167
 168 };
 169
 170
 171 /**
 172  * UDP NAT Probe message confirm definition
 173  */
 174 struct UDP_NAT_ProbeMessageConfirmation
 175 {
 176   /**
 177    * Message header
 178    */
 179   struct GNUNET_MessageHeader header;
 180
 181 };
 182
 183
 184
 185 /**
 186  * UDP NAT "Session"
 187  */
 188 struct PeerSession
 189 {
 190
 191   /**
 192    * Stored in a linked list.
 193    */
 194   struct PeerSession *next;
 195
 196   /**
 197    * Pointer to the global plugin struct.
 198    */
 199   struct Plugin *plugin;
 200
 201   /**
 202    * To whom are we talking to (set to our identity
 203    * if we are still waiting for the welcome message)
 204    */
 205   struct GNUNET_PeerIdentity target;
 206
 207   /**
 208    * Address of the other peer (either based on our 'connect'
 209    * call or on our 'accept' call).
 210    */
 211   void *connect_addr;
 212
 213   /**
 214    * Length of connect_addr.
 215    */
 216   size_t connect_alen;
 217
 218   /**
 219    * Are we still expecting the welcome message? (GNUNET_YES/GNUNET_NO)
 220    */
 221   int expecting_welcome;
 222
 223   /**
 224    * From which socket do we need to send to this peer?
 225    */
 226   struct GNUNET_NETWORK_Handle *sock;
 227
 228   /*
 229    * Queue of messages for this peer, in the case that
 230    * we have to await a connection...
 231    */
 232   struct MessageQueue *messages;
 233
 234 };
 235
 236 struct UDP_NAT_Probes
 237 {
 238
 239   /**
 240    * Linked list
 241    */
 242   struct UDP_NAT_Probes *next;
 243
 244   /**
 245    * Address string that the server process returned to us
 246    */
 247   char *address_string;
 248
 249   /**
 250    * Timeout for this set of probes
 251    */
 252   struct GNUNET_TIME_Absolute timeout;
 253
 254   /**
 255    * Count of how many probes we've attempted
 256    */
 257   int count;
 258
 259   /**
 260    * The plugin this probe belongs to
 261    */
 262   struct Plugin *plugin;
 263
 264   /**
 265    * The task used to send these probes
 266    */
 267   GNUNET_SCHEDULER_TaskIdentifier task;
 268
 269   /**
 270    * Network address (always ipv4)
 271    */
 272   struct sockaddr_in sock_addr;
 273
 274   /**
 275    * The port to send this probe to, 0 to choose randomly
 276    */
 277   int port;
 278
 279 };
 280
 281
 282 /**
 283  * Encapsulation of all of the state of the plugin.
 284  */
 285 struct Plugin
 286 {
 287   /**
 288    * Our environment.
 289    */
 290   struct GNUNET_TRANSPORT_PluginEnvironment *env;
 291
 292   /**
 293    * Handle to the network service.
 294    */
 295   struct GNUNET_SERVICE_Context *service;
 296
 297   /*
 298    * Session of peers with whom we are currently connected
 299    */
 300   struct PeerSession *sessions;
 301
 302   /**
 303    * Handle for request of hostname resolution, non-NULL if pending.
 304    */
 305   struct GNUNET_RESOLVER_RequestHandle *hostname_dns;
 306
 307   /**
 308    * ID of task used to update our addresses when one expires.
 309    */
 310   GNUNET_SCHEDULER_TaskIdentifier address_update_task;
 311
 312   /**
 313    * ID of select task
 314    */
 315   GNUNET_SCHEDULER_TaskIdentifier select_task;
 316
 317   /**
 318    * Port to listen on.
 319    */
 320   uint16_t port;
 321
 322   /**
 323    * The external address given to us by the user.  Must be actual
 324    * outside visible address for NAT punching to work.
 325    */
 326   char *external_address;
 327
 328   /**
 329    * The internal address given to us by the user (or discovered).
 330    */
 331   char *internal_address;
 332
 333   /*
 334    * FD Read set
 335    */
 336   struct GNUNET_NETWORK_FDSet *rs;
 337
 338   /*
 339    * stdout pipe handle for the gnunet-nat-server process
 340    */
 341   struct GNUNET_DISK_PipeHandle *server_stdout;
 342
 343   /*
 344    * stdout file handle (for reading) for the gnunet-nat-server process
 345    */
 346   const struct GNUNET_DISK_FileHandle *server_stdout_handle;
 347
 348   /**
 349    * ID of select gnunet-nat-server stdout read task
 350    */
 351   GNUNET_SCHEDULER_TaskIdentifier server_read_task;
 352
 353   /**
 354    * Is this transport configured to be behind a NAT?
 355    */
 356   int behind_nat;
 357
 358   /**
 359    * Is this transport configured to allow connections to NAT'd peers?
 360    */
 361   int allow_nat;
 362
 363   /**
 364    * The process id of the server process (if behind NAT)
 365    */
 366   pid_t server_pid;
 367
 368   /**
 369    * Probes in flight
 370    */
 371   struct UDP_NAT_Probes *probes;
 372
 373 };
 374
 375
 376 struct UDP_Sock_Info
 377 {
 378   /* The network handle */
 379   struct GNUNET_NETWORK_Handle *desc;
 380
 381   /* The port we bound to */
 382   int port;
 383 };
 384
 385 /* *********** globals ************* */
 386
 387 /**
 388  * the socket that we transmit all data with
 389  */
 390 static struct UDP_Sock_Info udp_sock;
 391
 392
 393 /**
 394  * Forward declaration.
 395  */
 396 void
 397 udp_probe_continuation (void *cls, const struct GNUNET_PeerIdentity *target, int result);
 398
 399
 400 /**
 401  * Disconnect from a remote node.  Clean up session if we have one for this peer
 402  *
 403  * @param cls closure for this call (should be handle to Plugin)
 404  * @param target the peeridentity of the peer to disconnect
 405  * @return GNUNET_OK on success, GNUNET_SYSERR if the operation failed
 406  */
 407 void
 408 udp_disconnect (void *cls, const struct GNUNET_PeerIdentity *target)
 409 {
 410   /** TODO: Implement! */
 411   return;
 412 }
 413
 414 /**
 415  * Shutdown the server process (stop receiving inbound traffic). Maybe
 416  * restarted later!
 417  *
 418  * @param cls Handle to the plugin for this transport
 419  *
 420  * @return returns the number of sockets successfully closed,
 421  *         should equal the number of sockets successfully opened
 422  */
 423 static int
 424 udp_transport_server_stop (void *cls)
 425 {
 426   struct Plugin *plugin = cls;
 427   int ret;
 428   int ok;
 429
 430   ret = 0;
 431   if (plugin->select_task != GNUNET_SCHEDULER_NO_TASK)
 432     {
 433       GNUNET_SCHEDULER_cancel (plugin->env->sched, plugin->select_task);
 434       plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
 435     }
 436
 437   ok = GNUNET_NETWORK_socket_close (udp_sock.desc);
 438   if (ok == GNUNET_OK)
 439     udp_sock.desc = NULL;
 440   ret += ok;
 441
 442   if (plugin->behind_nat == GNUNET_YES)
 443     {
 444       if (0 != PLIBC_KILL (plugin->server_pid, SIGTERM))
 445         {
 446           GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
 447         }
 448       GNUNET_OS_process_wait (plugin->server_pid);
 449     }
 450
 451   if (ret != GNUNET_OK)
 452     return GNUNET_SYSERR;
 453   return ret;
 454 }
 455
 456
 457 struct PeerSession *
 458 find_session (struct Plugin *plugin, const struct GNUNET_PeerIdentity *peer)
 459 {
 460   struct PeerSession *pos;
 461
 462   pos = plugin->sessions;
 463   while (pos != NULL)
 464     {
 465       if (memcmp(&pos->target, peer, sizeof(struct GNUNET_PeerIdentity)) == 0)
 466         return pos;
 467       pos = pos->next;
 468     }
 469
 470   return pos;
 471 }
 472
 473
 474 /**
 475  * Actually send out the message, assume we've got the address and
 476  * send_handle squared away!
 477  *
 478  * @param cls closure
 479  * @param send_handle which handle to send message on
 480  * @param target who should receive this message (ignored by UDP)
 481  * @param msgbuf one or more GNUNET_MessageHeader(s) strung together
 482  * @param msgbuf_size the size of the msgbuf to send
 483  * @param priority how important is the message (ignored by UDP)
 484  * @param timeout when should we time out (give up) if we can not transmit?
 485  * @param addr the addr to send the message to, needs to be a sockaddr for us
 486  * @param addrlen the len of addr
 487  * @param cont continuation to call once the message has
 488  *        been transmitted (or if the transport is ready
 489  *        for the next transmission call; or if the
 490  *        peer disconnected...)
 491  * @param cont_cls closure for cont
 492  * @return the number of bytes written
 493  */
 494 static ssize_t
 495 udp_real_send (void *cls,
 496                    struct GNUNET_NETWORK_Handle *send_handle,
 497                    const struct GNUNET_PeerIdentity *target,
 498                    const char *msgbuf,
 499                    size_t msgbuf_size,
 500                    unsigned int priority,
 501                    struct GNUNET_TIME_Relative timeout,
 502                    const void *addr,
 503                    size_t addrlen,
 504                    GNUNET_TRANSPORT_TransmitContinuation cont,
 505                    void *cont_cls)
 506 {
 507   struct Plugin *plugin = cls;
 508   struct UDPMessage *message;
 509   int ssize;
 510   ssize_t sent;
 511
 512   if ((addr == NULL) || (addrlen == 0))
 513     {
 514 #if DEBUG_UDP_NAT
 515   GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp", _
 516                    ("udp_plugin_send called without address, returning!\n"));
 517 #endif
 518       cont (cont_cls, target, GNUNET_SYSERR);
 519       return 0; /* Can never send if we don't have an address!! */
 520     }
 521
 522   /* Build the message to be sent */
 523   message = GNUNET_malloc (sizeof (struct UDPMessage) + msgbuf_size);
 524   ssize = sizeof (struct UDPMessage) + msgbuf_size;
 525
 526   message->header.size = htons (ssize);
 527   message->header.type = htons (0);
 528   memcpy (&message->sender, plugin->env->my_identity,
 529           sizeof (struct GNUNET_PeerIdentity));
 530   memcpy (&message[1], msgbuf, msgbuf_size);
 531
 532   /* Actually send the message */
 533   sent =
 534     GNUNET_NETWORK_socket_sendto (send_handle, message, ssize,
 535                                   addr,
 536                                   addrlen);
 537
 538   if (cont != NULL)
 539     {
 540       if (sent == GNUNET_SYSERR)
 541         cont (cont_cls, target, GNUNET_SYSERR);
 542       else
 543         {
 544           cont (cont_cls, target, GNUNET_OK);
 545         }
 546     }
 547
 548   GNUNET_free (message);
 549   return sent;
 550 }
 551
 552 /**
 553  * We learned about a peer (possibly behind NAT) so run the
 554  * gnunet-nat-client to send dummy ICMP responses
 555  *
 556  * @param plugin the plugin for this transport
 557  * @param addr the address of the peer
 558  * @param addrlen the length of the address
 559  */
 560 void
 561 run_gnunet_nat_client (struct Plugin *plugin, const char *addr, size_t addrlen)
 562 {
 563   char inet4[INET_ADDRSTRLEN];
 564   char *address_as_string;
 565   char *port_as_string;
 566   pid_t pid;
 567   const struct sockaddr *sa = (const struct sockaddr *)addr;
 568
 569   if (addrlen < sizeof (struct sockaddr))
 570     return;
 571   switch (sa->sa_family)
 572     {
 573     case AF_INET:
 574       if (addrlen != sizeof (struct sockaddr_in))
 575         return;
 576       inet_ntop (AF_INET,
 577                  &((struct sockaddr_in *) sa)->sin_addr,
 578                  inet4, INET_ADDRSTRLEN);
 579       address_as_string = GNUNET_strdup (inet4);
 580       break;
 581     case AF_INET6:
 582     default:
 583       return;
 584     }
 585
 586   GNUNET_asprintf(&port_as_string, "%d", plugin->port);
 587 #if DEBUG_UDP_NAT
 588   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 589                   _("Running gnunet-nat-client with arguments: %s %s %d\n"), plugin->external_address, address_as_string, plugin->port);
 590 #endif
 591
 592   /* Start the server process */
 593   pid = GNUNET_OS_start_process(NULL, NULL, "gnunet-nat-client", "gnunet-nat-client", plugin->external_address, address_as_string, port_as_string, NULL);
 594   GNUNET_free(address_as_string);
 595   GNUNET_free(port_as_string);
 596   GNUNET_OS_process_wait (pid);
 597 }
 598
 599 /**
 600  * Function that can be used by the transport service to transmit
 601  * a message using the plugin.
 602  *
 603  * @param cls closure
 604  * @param target who should receive this message (ignored by UDP)
 605  * @param msgbuf one or more GNUNET_MessageHeader(s) strung together
 606  * @param msgbuf_size the size of the msgbuf to send
 607  * @param priority how important is the message (ignored by UDP)
 608  * @param timeout when should we time out (give up) if we can not transmit?
 609  * @param session identifier used for this session (can be NULL)
 610  * @param addr the addr to send the message to, needs to be a sockaddr for us
 611  * @param addrlen the len of addr
 612  * @param force_address not used, we had better have an address to send to
 613  *        because we are stateless!!
 614  * @param cont continuation to call once the message has
 615  *        been transmitted (or if the transport is ready
 616  *        for the next transmission call; or if the
 617  *        peer disconnected...)
 618  * @param cont_cls closure for cont
 619  *
 620  * @return the number of bytes written (may return 0 and the message can
 621  *         still be transmitted later!)
 622  */
 623 static ssize_t
 624 udp_plugin_send (void *cls,
 625                      const struct GNUNET_PeerIdentity *target,
 626                      const char *msgbuf,
 627                      size_t msgbuf_size,
 628                      unsigned int priority,
 629                      struct GNUNET_TIME_Relative timeout,
 630                      struct Session *session,
 631                      const void *addr,
 632                      size_t addrlen,
 633                      int force_address,
 634                      GNUNET_TRANSPORT_TransmitContinuation cont, void *cont_cls)
 635 {
 636   struct Plugin *plugin = cls;
 637   ssize_t sent;
 638   struct MessageQueue *temp_message;
 639   struct PeerSession *peer_session;
 640   struct sockaddr_in *sockaddr = (struct sockaddr_in *)addr;
 641   int other_peer_natd;
 642
 643   GNUNET_assert (NULL == session);
 644   other_peer_natd = GNUNET_NO;
 645   if ((sockaddr->sin_family == AF_INET) && (ntohs(sockaddr->sin_port) == 0))
 646     {
 647       other_peer_natd = GNUNET_YES;
 648     }
 649
 650   sent = 0;
 651
 652   if ((other_peer_natd == GNUNET_YES) && (plugin->allow_nat == GNUNET_YES))
 653     {
 654       peer_session = find_session(plugin, target);
 655       if (peer_session == NULL) /* We have a new peer to add */
 656         {
 657           /*
 658            * The first time, we can assume we have no knowledge of a
 659            * working port for this peer, call the ICMP/UDP message sender
 660            * and wait...
 661            */
 662           peer_session = GNUNET_malloc(sizeof(struct PeerSession));
 663           peer_session->connect_addr = GNUNET_malloc(addrlen);
 664           memcpy(peer_session->connect_addr, addr, addrlen);
 665           peer_session->connect_alen = addrlen;
 666           peer_session->plugin = plugin;
 667           peer_session->sock = NULL;
 668           memcpy(&peer_session->target, target, sizeof(struct GNUNET_PeerIdentity));
 669           peer_session->expecting_welcome = GNUNET_YES;
 670
 671           peer_session->next = plugin->sessions;
 672           plugin->sessions = peer_session;
 673
 674           peer_session->messages = GNUNET_malloc(sizeof(struct MessageQueue));
 675           peer_session->messages->msgbuf = GNUNET_malloc(msgbuf_size);
 676           memcpy(peer_session->messages->msgbuf, msgbuf, msgbuf_size);
 677           peer_session->messages->msgbuf_size = msgbuf_size;
 678           peer_session->messages->next = NULL;
 679           peer_session->messages->timeout = GNUNET_TIME_relative_to_absolute(timeout);
 680           peer_session->messages->cont = cont;
 681           peer_session->messages->cont_cls = cont_cls;
 682 #if DEBUG_UDP_NAT
 683           GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 684                           _("Other peer is NAT'd, set up peer session for peer %s\n"), GNUNET_i2s(target));
 685 #endif
 686           run_gnunet_nat_client(plugin, addr, addrlen);
 687         }
 688       else
 689         {
 690           if (peer_session->expecting_welcome == GNUNET_NO) /* We are "connected" */
 691             {
 692               sent = udp_real_send(cls, peer_session->sock, target, msgbuf, msgbuf_size, priority, timeout, peer_session->connect_addr, peer_session->connect_alen, cont, cont_cls);
 693             }
 694           else /* Haven't gotten a response from this peer, queue message */
 695             {
 696               temp_message = GNUNET_malloc(sizeof(struct MessageQueue));
 697               temp_message->msgbuf = GNUNET_malloc(msgbuf_size);
 698               memcpy(temp_message->msgbuf, msgbuf, msgbuf_size);
 699               temp_message->msgbuf_size = msgbuf_size;
 700               temp_message->timeout = GNUNET_TIME_relative_to_absolute(timeout);
 701               temp_message->cont = cont;
 702               temp_message->cont_cls = cont_cls;
 703               temp_message->next = peer_session->messages;
 704               peer_session->messages = temp_message;
 705             }
 706         }
 707     }
 708   else if (other_peer_natd == GNUNET_NO) /* Other peer not behind a NAT, so we can just send the message as is */
 709     {
 710       sent = udp_real_send(cls, udp_sock.desc, target, msgbuf, msgbuf_size, priority, timeout, addr, addrlen, cont, cont_cls);
 711     }
 712   else /* Other peer is NAT'd, but we don't want to play with them (or can't!) */
 713     return GNUNET_SYSERR;
 714
 715   /* When GNUNET_SYSERR is returned from udp_real_send, we will still call
 716    * the callback so must not return GNUNET_SYSERR!
 717    * If we do, then transport context get freed twice. */
 718   if (sent == GNUNET_SYSERR)
 719     return 0;
 720
 721   return sent;
 722 }
 723
 724
 725 /**
 726  * Add the IP of our network interface to the list of
 727  * our external IP addresses.
 728  */
 729 static int
 730 process_interfaces (void *cls,
 731                     const char *name,
 732                     int isDefault,
 733                     const struct sockaddr *addr, socklen_t addrlen)
 734 {
 735   struct Plugin *plugin = cls;
 736   int af;
 737   struct sockaddr_in *v4;
 738   struct sockaddr_in6 *v6;
 739
 740   af = addr->sa_family;
 741   if (af == AF_INET)
 742     {
 743       v4 = (struct sockaddr_in *) addr;
 744       if (plugin->behind_nat == GNUNET_YES)
 745         {
 746           GNUNET_assert(inet_pton(AF_INET, plugin->external_address, &v4->sin_addr) == GNUNET_OK);
 747           v4->sin_port = htons (0); /* Indicates to receiver we are behind NAT */
 748         }
 749       else
 750         v4->sin_port = htons (plugin->port);
 751     }
 752   else
 753     {
 754       GNUNET_assert (af == AF_INET6);
 755       v6 = (struct sockaddr_in6 *) addr;
 756       if (plugin->behind_nat == GNUNET_YES)
 757         v6->sin6_port = htons (0);
 758       else
 759         v6->sin6_port = htons (plugin->port);
 760     }
 761
 762     GNUNET_log_from (GNUNET_ERROR_TYPE_INFO |
 763                       GNUNET_ERROR_TYPE_BULK,
 764                       "udp", _("Found address `%s' (%s)\n"),
 765                       GNUNET_a2s (addr, addrlen), name);
 766     plugin->env->notify_address (plugin->env->cls,
 767                                 "udp",
 768                                 addr, addrlen, GNUNET_TIME_UNIT_FOREVER_REL);
 769
 770   return GNUNET_OK;
 771 }
 772
 773
 774 /**
 775  * Function called by the resolver for each address obtained from DNS
 776  * for our own hostname.  Add the addresses to the list of our
 777  * external IP addresses.
 778  *
 779  * @param cls closure
 780  * @param addr one of the addresses of the host, NULL for the last address
 781  * @param addrlen length of the address
 782  */
 783 static void
 784 process_hostname_ips (void *cls,
 785                       const struct sockaddr *addr, socklen_t addrlen)
 786 {
 787   struct Plugin *plugin = cls;
 788
 789   if (addr == NULL)
 790     {
 791       plugin->hostname_dns = NULL;
 792       return;
 793     }
 794   process_interfaces (plugin, "<hostname>", GNUNET_YES, addr, addrlen);
 795 }
 796
 797
 798 /**
 799  * Send UDP probe messages or UDP keepalive messages, depending on the
 800  * state of the connection.
 801  *
 802  * @param cls closure for this call (should be the main Plugin)
 803  * @param tc task context for running this
 804  */
 805 static void
 806 send_udp_probe_message (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 807 {
 808   struct UDP_NAT_Probes *probe = cls;
 809   struct UDP_NAT_ProbeMessage *message;
 810   struct Plugin *plugin = probe->plugin;
 811
 812   message = GNUNET_malloc(sizeof(struct UDP_NAT_ProbeMessage));
 813   message->header.size = htons(sizeof(struct UDP_NAT_ProbeMessage));
 814   message->header.type = htons(GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE);
 815   /* If they gave us a port, use that.  If not, try our port. */
 816   if (probe->port != 0)
 817     probe->sock_addr.sin_port = htons(probe->port);
 818   else
 819     probe->sock_addr.sin_port = htons(plugin->port);
 820
 821 #if DEBUG_UDP_NAT
 822       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 823                       _("Sending a probe to port %d\n"), ntohs(probe->sock_addr.sin_port));
 824 #endif
 825
 826   probe->count++;
 827
 828   udp_real_send(plugin, udp_sock.desc, NULL,
 829                     (char *)message, ntohs(message->header.size), 0,
 830                     GNUNET_TIME_relative_get_unit(),
 831                     &probe->sock_addr, sizeof(probe->sock_addr),
 832                     &udp_probe_continuation, probe);
 833
 834   GNUNET_free(message);
 835 }
 836
 837
 838 /**
 839  * Continuation for probe sends.  If the last probe was sent
 840  * "successfully", schedule sending of another one.  If not,
 841  *
 842  */
 843 void
 844 udp_probe_continuation (void *cls, const struct GNUNET_PeerIdentity *target, int result)
 845 {
 846   struct UDP_NAT_Probes *probe = cls;
 847   struct Plugin *plugin = probe->plugin;
 848
 849   if ((result == GNUNET_OK) && (probe->count < MAX_PROBES))
 850     {
 851 #if DEBUG_UDP_NAT
 852       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 853                        _("Scheduling next probe for 10000 milliseconds\n"));
 854 #endif
 855       probe->task = GNUNET_SCHEDULER_add_delayed(plugin->env->sched, GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MILLISECONDS, 10000), &send_udp_probe_message, probe);
 856     }
 857   else /* Destroy the probe context. */
 858     {
 859 #if DEBUG_UDP_NAT
 860       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 861                       _("Sending probe didn't go well...\n"));
 862 #endif
 863     }
 864 }
 865
 866 /**
 867  * Find probe message by address
 868  *
 869  * @param plugin the plugin for this transport
 870  * @param address_string the ip address as a string
 871  */
 872 struct UDP_NAT_Probes *
 873 find_probe(struct Plugin *plugin, char * address_string)
 874 {
 875   struct UDP_NAT_Probes *pos;
 876
 877   pos = plugin->probes;
 878   while (pos != NULL)
 879     if (strcmp(pos->address_string, address_string) == 0)
 880       return pos;
 881
 882   return pos;
 883 }
 884
 885
 886 /*
 887  * @param cls the plugin handle
 888  * @param tc the scheduling context (for rescheduling this function again)
 889  *
 890  * We have been notified that gnunet-nat-server has written something to stdout.
 891  * Handle the output, then reschedule this function to be called again once
 892  * more is available.
 893  *
 894  */
 895 static void
 896 udp_plugin_server_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 897 {
 898   struct Plugin *plugin = cls;
 899   char mybuf[40];
 900   ssize_t bytes;
 901   memset(&mybuf, 0, sizeof(mybuf));
 902   int i;
 903   struct UDP_NAT_Probes *temp_probe;
 904   int port;
 905   char *port_start;
 906   struct sockaddr_in in_addr;
 907
 908   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
 909     return;
 910
 911   bytes = GNUNET_DISK_file_read(plugin->server_stdout_handle, &mybuf, sizeof(mybuf));
 912
 913   if (bytes < 1)
 914     {
 915 #if DEBUG_UDP_NAT
 916       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 917                       _("Finished reading from server stdout with code: %d\n"), bytes);
 918 #endif
 919       return;
 920     }
 921
 922   port = 0;
 923   port_start = NULL;
 924   for (i = 0; i < sizeof(mybuf); i++)
 925     {
 926       if (mybuf[i] == '\n')
 927         mybuf[i] = '\0';
 928
 929       if ((mybuf[i] == ':') && (i + 1 < sizeof(mybuf)))
 930         {
 931           mybuf[i] = '\0';
 932           port_start = &mybuf[i + 1];
 933         }
 934     }
 935
 936   if (port_start != NULL)
 937     port = atoi(port_start);
 938   else
 939     {
 940       plugin->server_read_task =
 941            GNUNET_SCHEDULER_add_read_file (plugin->env->sched,
 942                                            GNUNET_TIME_UNIT_FOREVER_REL,
 943                                            plugin->server_stdout_handle, &udp_plugin_server_read, plugin);
 944       return;
 945     }
 946
 947 #if DEBUG_UDP_NAT
 948   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
 949                   _("nat-server-read read: %s port %d\n"), &mybuf, port);
 950 #endif
 951
 952   /**
 953    * We have received an ICMP response, ostensibly from a non-NAT'd peer
 954    *  that wants to connect to us! Send a message to establish a connection.
 955    */
 956   if (inet_pton(AF_INET, &mybuf[0], &in_addr.sin_addr) != 1)
 957     {
 958
 959       GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING, "udp",
 960                   _("nat-server-read malformed address\n"), &mybuf, port);
 961
 962       plugin->server_read_task =
 963           GNUNET_SCHEDULER_add_read_file (plugin->env->sched,
 964                                           GNUNET_TIME_UNIT_FOREVER_REL,
 965                                           plugin->server_stdout_handle, &udp_plugin_server_read, plugin);
 966       return;
 967     }
 968
 969   temp_probe = find_probe(plugin, &mybuf[0]);
 970
 971   if (temp_probe == NULL)
 972     {
 973       temp_probe = GNUNET_malloc(sizeof(struct UDP_NAT_Probes));
 974       temp_probe->address_string = strdup(&mybuf[0]);
 975       temp_probe->sock_addr.sin_family = AF_INET;
 976       GNUNET_assert(inet_pton(AF_INET, &mybuf[0], &temp_probe->sock_addr.sin_addr) == 1);
 977       temp_probe->port = port;
 978       temp_probe->next = plugin->probes;
 979       temp_probe->plugin = plugin;
 980       temp_probe->task = GNUNET_SCHEDULER_add_delayed(plugin->env->sched, GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MILLISECONDS, 500), &send_udp_probe_message, temp_probe);
 981       plugin->probes = temp_probe;
 982     }
 983
 984   plugin->server_read_task =
 985        GNUNET_SCHEDULER_add_read_file (plugin->env->sched,
 986                                        GNUNET_TIME_UNIT_FOREVER_REL,
 987                                        plugin->server_stdout_handle, &udp_plugin_server_read, plugin);
 988
 989 }
 990
 991
 992 /**
 993  * Demultiplexer for UDP NAT messages
 994  *
 995  * @param plugin the main plugin for this transport
 996  * @param sender from which peer the message was received
 997  * @param currhdr pointer to the header of the message
 998  * @param sender_addr the address from which the message was received
 999  * @param fromlen the length of the address
1000  * @param sockinfo which socket did we receive the message on
1001  */
1002 static void
1003 udp_demultiplexer(struct Plugin *plugin, struct GNUNET_PeerIdentity *sender, const struct GNUNET_MessageHeader *currhdr, struct sockaddr_storage *sender_addr, socklen_t fromlen, struct UDP_Sock_Info *sockinfo)
1004 {
1005   struct UDP_NAT_ProbeMessageReply *outgoing_probe_reply;
1006   struct UDP_NAT_ProbeMessageConfirmation *outgoing_probe_confirmation;
1007
1008   char addr_buf[INET_ADDRSTRLEN];
1009   struct UDP_NAT_Probes *outgoing_probe;
1010   struct PeerSession *peer_session;
1011   struct MessageQueue *pending_message;
1012   struct MessageQueue *pending_message_temp;
1013
1014   if (memcmp(sender, plugin->env->my_identity, sizeof(struct GNUNET_PeerIdentity)) == 0)
1015     {
1016 #if DEBUG_UDP_NAT
1017       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1018                       _("Received a message from myself, dropping!!!\n"));
1019 #endif
1020       return;
1021     }
1022
1023   switch (ntohs(currhdr->type))
1024   {
1025     case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE:
1026       /* Send probe reply */
1027       outgoing_probe_reply = GNUNET_malloc(sizeof(struct UDP_NAT_ProbeMessageReply));
1028       outgoing_probe_reply->header.size = htons(sizeof(struct UDP_NAT_ProbeMessageReply));
1029       outgoing_probe_reply->header.type = htons(GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE_REPLY);
1030
1031 #if DEBUG_UDP_NAT
1032       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1033                       _("Received a probe on listen port %d, sent_from port %d\n"), sockinfo->port, ntohs(((struct sockaddr_in *)sender_addr)->sin_port));
1034 #endif
1035
1036       udp_real_send(plugin, sockinfo->desc, NULL,
1037                         (char *)outgoing_probe_reply,
1038                         ntohs(outgoing_probe_reply->header.size), 0,
1039                         GNUNET_TIME_relative_get_unit(),
1040                         sender_addr, fromlen,
1041                         NULL, NULL);
1042
1043 #if DEBUG_UDP_NAT
1044       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1045                       _("Sent PROBE REPLY to port %d on outgoing port %d\n"), ntohs(((struct sockaddr_in *)sender_addr)->sin_port), sockinfo->port);
1046 #endif
1047       GNUNET_free(outgoing_probe_reply);
1048       break;
1049     case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE_REPLY:
1050       /* Check for existing probe, check ports returned, send confirmation if all is well */
1051 #if DEBUG_UDP_NAT
1052       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1053                       _("Received PROBE REPLY from port %d on incoming port %d\n"), ntohs(((struct sockaddr_in *)sender_addr)->sin_port), sockinfo->port);
1054 #endif
1055       /* FIXME: use nonce, then IPv6 replies could work I think... */
1056       if (sender_addr->ss_family == AF_INET)
1057         {
1058           memset(&addr_buf, 0, sizeof(addr_buf));
1059           inet_ntop(AF_INET, &((struct sockaddr_in *) sender_addr)->sin_addr, addr_buf, INET_ADDRSTRLEN);
1060           outgoing_probe = find_probe(plugin, &addr_buf[0]);
1061           if (outgoing_probe != NULL)
1062             {
1063 #if DEBUG_UDP_NAT
1064               GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1065                               _("Sending confirmation that we were reached!\n"));
1066 #endif
1067               outgoing_probe_confirmation = GNUNET_malloc(sizeof(struct UDP_NAT_ProbeMessageConfirmation));
1068               outgoing_probe_confirmation->header.size = htons(sizeof(struct UDP_NAT_ProbeMessageConfirmation));
1069               outgoing_probe_confirmation->header.type = htons(GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE_CONFIRM);
1070
1071               udp_real_send(plugin, sockinfo->desc, NULL, (char *)outgoing_probe_confirmation, ntohs(outgoing_probe_confirmation->header.size), 0, GNUNET_TIME_relative_get_unit(), sender_addr, fromlen, NULL, NULL);
1072
1073               if (outgoing_probe->task != GNUNET_SCHEDULER_NO_TASK)
1074                 {
1075                   GNUNET_SCHEDULER_cancel(plugin->env->sched, outgoing_probe->task);
1076                   outgoing_probe->task = GNUNET_SCHEDULER_NO_TASK;
1077                   /* Schedule task to timeout and remove probe if confirmation not received */
1078                 }
1079               GNUNET_free(outgoing_probe_confirmation);
1080             }
1081           else
1082             {
1083 #if DEBUG_UDP_NAT
1084               GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp",
1085                               _("Received a probe reply, but have no record of a sent probe!\n"));
1086 #endif
1087             }
1088         }
1089       break;
1090     case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE_CONFIRM:
1091       peer_session = find_session(plugin, sender);
1092 #if DEBUG_UDP_NAT
1093           GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1094                           _("Looking up peer session for peer %s\n"), GNUNET_i2s(sender));
1095 #endif
1096       if (peer_session == NULL) /* Shouldn't this NOT happen? */
1097         {
1098 #if DEBUG_UDP_NAT
1099           GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp",
1100                           _("Peer not in list, adding (THIS MAY BE A MISTAKE) %s\n"), GNUNET_i2s(sender));
1101 #endif
1102           peer_session = GNUNET_malloc(sizeof(struct PeerSession));
1103           peer_session->connect_addr = GNUNET_malloc(fromlen);
1104           memcpy(peer_session->connect_addr, sender_addr, fromlen);
1105           peer_session->connect_alen = fromlen;
1106           peer_session->plugin = plugin;
1107           peer_session->sock = sockinfo->desc;
1108           memcpy(&peer_session->target, sender, sizeof(struct GNUNET_PeerIdentity));
1109           peer_session->expecting_welcome = GNUNET_NO;
1110
1111           peer_session->next = plugin->sessions;
1112           plugin->sessions = peer_session;
1113
1114           peer_session->messages = NULL;
1115         }
1116       else if (peer_session->expecting_welcome == GNUNET_YES)
1117         {
1118           peer_session->expecting_welcome = GNUNET_NO;
1119           /* FIXME: There is no way to find this based on receiving port at the moment! */
1120           peer_session->sock = sockinfo->desc; /* This may matter, not sure right now... */
1121           ((struct sockaddr_in *)peer_session->connect_addr)->sin_port = ((struct sockaddr_in *) sender_addr)->sin_port;
1122 #if DEBUG_UDP_NAT
1123               GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp",
1124                               _("Received a probe confirmation, will send to peer on port %d\n"), ntohs(((struct sockaddr_in *)peer_session->connect_addr)->sin_port));
1125 #endif
1126           if (peer_session->messages != NULL)
1127             {
1128 #if DEBUG_UDP_NAT
1129               GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp",
1130                               _("Received a probe confirmation, sending queued messages.\n"));
1131 #endif
1132               pending_message = peer_session->messages;
1133               int count = 0;
1134               while (pending_message != NULL)
1135                 {
1136 #if DEBUG_UDP_NAT
1137                   GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp",
1138                                   _("sending queued message %d\n"), count);
1139 #endif
1140                   udp_real_send(plugin, peer_session->sock, &peer_session->target, pending_message->msgbuf, pending_message->msgbuf_size, 0, GNUNET_TIME_relative_get_unit(), peer_session->connect_addr, peer_session->connect_alen, pending_message->cont, pending_message->cont_cls);
1141                   pending_message_temp = pending_message;
1142                   pending_message = pending_message->next;
1143                   GNUNET_free(pending_message_temp->msgbuf);
1144                   GNUNET_free(pending_message_temp);
1145 #if DEBUG_UDP_NAT
1146                   GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp",
1147                                   _("finished sending queued message %d\n"), count);
1148 #endif
1149                   count++;
1150                 }
1151             }
1152
1153         }
1154       else
1155         {
1156 #if DEBUG_UDP_NAT
1157           GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp",
1158                           _("Received probe confirmation for already confirmed peer!\n"));
1159 #endif
1160         }
1161       /* Received confirmation, add peer with address/port specified */
1162       break;
1163     case GNUNET_MESSAGE_TYPE_TRANSPORT_UDP_NAT_PROBE_KEEPALIVE:
1164       /* Once we've sent NAT_PROBE_CONFIRM change to sending keepalives */
1165       /* If we receive these just ignore! */
1166       break;
1167     default:
1168       plugin->env->receive (plugin->env->cls, sender, currhdr, UDP_DIRECT_DISTANCE,
1169                             NULL, (char *)sender_addr, fromlen);
1170   }
1171
1172 }
1173
1174
1175 /*
1176  * @param cls the plugin handle
1177  * @param tc the scheduling context (for rescheduling this function again)
1178  *
1179  * We have been notified that our writeset has something to read.  We don't
1180  * know which socket needs to be read, so we have to check each one
1181  * Then reschedule this function to be called again once more is available.
1182  *
1183  */
1184 static void
1185 udp_plugin_select (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1186 {
1187   struct Plugin *plugin = cls;
1188   char *buf;
1189   struct UDPMessage *msg;
1190   struct GNUNET_PeerIdentity *sender;
1191   unsigned int buflen;
1192   socklen_t fromlen;
1193   struct sockaddr_storage addr;
1194   ssize_t ret;
1195   int offset;
1196   int count;
1197   int tsize;
1198   char *msgbuf;
1199   const struct GNUNET_MessageHeader *currhdr;
1200
1201   plugin->select_task = GNUNET_SCHEDULER_NO_TASK;
1202
1203   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
1204     return;
1205
1206   buf = NULL;
1207   sender = NULL;
1208
1209   buflen = GNUNET_NETWORK_socket_recvfrom_amount (udp_sock.desc);
1210
1211   if (buflen == GNUNET_NO)
1212     return;
1213
1214   buf = GNUNET_malloc (buflen);
1215   fromlen = sizeof (addr);
1216   memset (&addr, 0, fromlen);
1217   ret =
1218     GNUNET_NETWORK_socket_recvfrom (udp_sock.desc, buf, buflen,
1219                                     (struct sockaddr *) &addr, &fromlen);
1220
1221 #if DEBUG_UDP_NAT
1222   GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp", _
1223                    ("socket_recv returned %u, src_addr_len is %u\n"), ret,
1224                    fromlen);
1225 #endif
1226
1227   if (ret <= 0)
1228     {
1229       GNUNET_free (buf);
1230       return;
1231     }
1232   msg = (struct UDPMessage *) buf;
1233
1234 #if DEBUG_UDP_NAT
1235   GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp", _
1236                   ("header reports message size of %d, type %d\n"),
1237                   ntohs (msg->header.size), ntohs (msg->header.type));
1238 #endif
1239   if (ntohs (msg->header.size) < sizeof (struct UDPMessage))
1240     {
1241       GNUNET_free (buf);
1242       return;
1243     }
1244
1245   msgbuf = (char *)&msg[1];
1246   sender = GNUNET_malloc (sizeof (struct GNUNET_PeerIdentity));
1247   memcpy (sender, &msg->sender, sizeof (struct GNUNET_PeerIdentity));
1248
1249   offset = 0;
1250   count = 0;
1251   tsize = ntohs (msg->header.size) - sizeof(struct UDPMessage);
1252
1253   while (offset < tsize)
1254     {
1255       currhdr = (struct GNUNET_MessageHeader *)&msgbuf[offset];
1256 #if DEBUG_UDP_NAT
1257       GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp", _
1258                        ("processing msg %d: type %d, size %d at offset %d\n"),
1259                        count, ntohs(currhdr->type), ntohs(currhdr->size), offset);
1260 #endif
1261       udp_demultiplexer(plugin, sender, currhdr, &addr, fromlen, &udp_sock);
1262 #if DEBUG_UDP_NAT
1263       GNUNET_log_from (GNUNET_ERROR_TYPE_INFO, "udp", _
1264                        ("processing done msg %d: type %d, size %d at offset %d\n"),
1265                        count, ntohs(currhdr->type), ntohs(currhdr->size), offset);
1266 #endif
1267       offset += ntohs(currhdr->size);
1268       count++;
1269     }
1270   GNUNET_free_non_null (buf);
1271   GNUNET_free_non_null (sender);
1272
1273
1274   plugin->select_task =
1275     GNUNET_SCHEDULER_add_select (plugin->env->sched,
1276                                  GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1277                                  GNUNET_SCHEDULER_NO_TASK,
1278                                  GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
1279                                  NULL, &udp_plugin_select, plugin);
1280
1281 }
1282
1283 /**
1284  * Create a slew of UDP sockets.  If possible, use IPv6, otherwise
1285  * try IPv4.
1286  *
1287  * @param cls closure for server start, should be a struct Plugin *
1288  *
1289  * @return number of sockets created or GNUNET_SYSERR on error
1290  */
1291 static int
1292 udp_transport_server_start (void *cls)
1293 {
1294   struct Plugin *plugin = cls;
1295   struct sockaddr_in serverAddrv4;
1296   struct sockaddr_in6 serverAddrv6;
1297   struct sockaddr *serverAddr;
1298   socklen_t addrlen;
1299   int sockets_created;
1300
1301   /* Pipe to read from started processes stdout (on read end) */
1302   plugin->server_stdout = GNUNET_DISK_pipe(GNUNET_YES);
1303
1304   sockets_created = 0;
1305   if (plugin->server_stdout == NULL)
1306     return sockets_created;
1307
1308   if (plugin->behind_nat == GNUNET_YES)
1309     {
1310 #if DEBUG_UDP_NAT
1311   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG,
1312                    "udp",
1313                    "Starting gnunet-nat-server process cmd: %s %s\n", "gnunet-nat-server", plugin->internal_address);
1314 #endif
1315       /* Start the server process */
1316       plugin->server_pid = GNUNET_OS_start_process(NULL, plugin->server_stdout, "gnunet-nat-server", "gnunet-nat-server", plugin->internal_address, NULL);
1317       if (plugin->server_pid == GNUNET_SYSERR)
1318         {
1319 #if DEBUG_UDP_NAT
1320           GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG,
1321                            "udp",
1322                            "Failed to start gnunet-nat-server process\n");
1323 #endif
1324           return GNUNET_SYSERR;
1325         }
1326       /* Close the write end of the read pipe */
1327       GNUNET_DISK_pipe_close_end(plugin->server_stdout, GNUNET_DISK_PIPE_END_WRITE);
1328
1329       plugin->server_stdout_handle = GNUNET_DISK_pipe_handle(plugin->server_stdout, GNUNET_DISK_PIPE_END_READ);
1330       plugin->server_read_task =
1331           GNUNET_SCHEDULER_add_read_file (plugin->env->sched,
1332                                           GNUNET_TIME_UNIT_FOREVER_REL,
1333                                           plugin->server_stdout_handle, &udp_plugin_server_read, plugin);
1334     }
1335
1336     udp_sock.desc = NULL;
1337
1338
1339     udp_sock.desc = GNUNET_NETWORK_socket_create (PF_INET, SOCK_DGRAM, 17);
1340     if (NULL == udp_sock.desc)
1341       {
1342         GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, "udp", "socket");
1343         return sockets_created;
1344       }
1345     else
1346       {
1347         memset (&serverAddrv4, 0, sizeof (serverAddrv4));
1348 #if HAVE_SOCKADDR_IN_SIN_LEN
1349         serverAddrv4.sin_len = sizeof (serverAddrv4);
1350 #endif
1351         serverAddrv4.sin_family = AF_INET;
1352         serverAddrv4.sin_addr.s_addr = INADDR_ANY;
1353         serverAddrv4.sin_port = htons (plugin->port);
1354         addrlen = sizeof (serverAddrv4);
1355         serverAddr = (struct sockaddr *) &serverAddrv4;
1356 #if DEBUG_UDP_NAT
1357         GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG,
1358                          "udp",
1359                          "Binding to port %d\n", ntohs(serverAddrv4.sin_port));
1360 #endif
1361         while (GNUNET_NETWORK_socket_bind (udp_sock.desc, serverAddr, addrlen) !=
1362                        GNUNET_OK)
1363           {
1364             serverAddrv4.sin_port = htons (GNUNET_CRYPTO_random_u32(GNUNET_CRYPTO_QUALITY_STRONG, 33537) + 32000); /* Find a good, non-root port */
1365 #if DEBUG_UDP_NAT
1366         GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG,
1367                         "udp",
1368                         "Binding failed, trying new port %d\n", ntohs(serverAddrv4.sin_port));
1369 #endif
1370           }
1371         udp_sock.port = ntohs(serverAddrv4.sin_port);
1372         sockets_created++;
1373       }
1374
1375
1376   if ((udp_sock.desc == NULL) && (GNUNET_YES !=
1377       GNUNET_CONFIGURATION_get_value_yesno (plugin->env->cfg, "GNUNETD",
1378                                             "DISABLE-IPV6")))
1379     {
1380       udp_sock.desc = GNUNET_NETWORK_socket_create (PF_INET6, SOCK_DGRAM, 17);
1381       if (udp_sock.desc != NULL)
1382         {
1383           memset (&serverAddrv6, 0, sizeof (serverAddrv6));
1384 #if HAVE_SOCKADDR_IN_SIN_LEN
1385           serverAddrv6.sin6_len = sizeof (serverAddrv6);
1386 #endif
1387           serverAddrv6.sin6_family = AF_INET6;
1388           serverAddrv6.sin6_addr = in6addr_any;
1389           serverAddrv6.sin6_port = htons (plugin->port);
1390           addrlen = sizeof (serverAddrv6);
1391           serverAddr = (struct sockaddr *) &serverAddrv6;
1392           sockets_created++;
1393         }
1394     }
1395
1396   plugin->rs = GNUNET_NETWORK_fdset_create ();
1397
1398   GNUNET_NETWORK_fdset_zero (plugin->rs);
1399
1400
1401   GNUNET_NETWORK_fdset_set (plugin->rs, udp_sock.desc);
1402
1403   plugin->select_task =
1404     GNUNET_SCHEDULER_add_select (plugin->env->sched,
1405                                  GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1406                                  GNUNET_SCHEDULER_NO_TASK,
1407                                  GNUNET_TIME_UNIT_FOREVER_REL, plugin->rs,
1408                                  NULL, &udp_plugin_select, plugin);
1409
1410   return sockets_created;
1411 }
1412
1413
1414 /**
1415  * Check if the given port is plausible (must be either
1416  * our listen port or our advertised port).  If it is
1417  * neither, we return one of these two ports at random.
1418  *
1419  * @return either in_port or a more plausible port
1420  */
1421 static uint16_t
1422 check_port (struct Plugin *plugin, uint16_t in_port)
1423 {
1424
1425   /* FIXME: remember what ports we are using to better respond to this */
1426   return in_port;
1427   /*
1428   for (i = plugin->starting_port; i < plugin->num_ports + plugin->starting_port; i++)
1429     {
1430       if (in_port == i)
1431         return in_port;
1432     }
1433
1434   return GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
1435                                     plugin->num_ports) + plugin->starting_port;
1436   */
1437 }
1438
1439
1440 /**
1441  * Another peer has suggested an address for this peer and transport
1442  * plugin.  Check that this could be a valid address.  This function
1443  * is not expected to 'validate' the address in the sense of trying to
1444  * connect to it but simply to see if the binary format is technically
1445  * legal for establishing a connection.
1446  *
1447  * @param cls closure, should be our handle to the Plugin
1448  * @param addr pointer to the address, may be modified (slightly)
1449  * @param addrlen length of addr
1450  * @return GNUNET_OK if this is a plausible address for this peer
1451  *         and transport, GNUNET_SYSERR if not
1452  *
1453  */
1454 static int
1455 udp_check_address (void *cls, void *addr, size_t addrlen)
1456 {
1457   struct Plugin *plugin = cls;
1458   char buf[sizeof (struct sockaddr_in6)];
1459
1460   struct sockaddr_in *v4;
1461   struct sockaddr_in6 *v6;
1462
1463   if ((addrlen != sizeof (struct sockaddr_in)) &&
1464       (addrlen != sizeof (struct sockaddr_in6)))
1465     {
1466       GNUNET_break_op (0);
1467       return GNUNET_SYSERR;
1468     }
1469   memcpy (buf, addr, sizeof (struct sockaddr_in6));
1470   if (addrlen == sizeof (struct sockaddr_in))
1471     {
1472       v4 = (struct sockaddr_in *) buf;
1473       v4->sin_port = htons (check_port (plugin, ntohs (v4->sin_port)));
1474     }
1475   else
1476     {
1477       v6 = (struct sockaddr_in6 *) buf;
1478       v6->sin6_port = htons (check_port (plugin, ntohs (v6->sin6_port)));
1479     }
1480
1481 #if DEBUG_UDP_NAT
1482   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG,
1483                    "udp",
1484                    "Informing transport service about my address `%s'.\n",
1485                    GNUNET_a2s (addr, addrlen));
1486 #endif
1487   return GNUNET_OK;
1488 }
1489
1490
1491 /**
1492  * Append our port and forward the result.
1493  */
1494 static void
1495 append_port (void *cls, const char *hostname)
1496 {
1497   struct PrettyPrinterContext *ppc = cls;
1498   char *ret;
1499
1500   if (hostname == NULL)
1501     {
1502       ppc->asc (ppc->asc_cls, NULL);
1503       GNUNET_free (ppc);
1504       return;
1505     }
1506   GNUNET_asprintf (&ret, "%s:%d", hostname, ppc->port);
1507   ppc->asc (ppc->asc_cls, ret);
1508   GNUNET_free (ret);
1509 }
1510
1511
1512 /**
1513  * Convert the transports address to a nice, human-readable
1514  * format.
1515  *
1516  * @param cls closure
1517  * @param type name of the transport that generated the address
1518  * @param addr one of the addresses of the host, NULL for the last address
1519  *        the specific address format depends on the transport
1520  * @param addrlen length of the address
1521  * @param numeric should (IP) addresses be displayed in numeric form?
1522  * @param timeout after how long should we give up?
1523  * @param asc function to call on each string
1524  * @param asc_cls closure for asc
1525  */
1526 static void
1527 udp_plugin_address_pretty_printer (void *cls,
1528                                    const char *type,
1529                                    const void *addr,
1530                                    size_t addrlen,
1531                                    int numeric,
1532                                    struct GNUNET_TIME_Relative timeout,
1533                                    GNUNET_TRANSPORT_AddressStringCallback asc,
1534                                    void *asc_cls)
1535 {
1536   struct Plugin *plugin = cls;
1537   const struct sockaddr_in *v4;
1538   const struct sockaddr_in6 *v6;
1539   struct PrettyPrinterContext *ppc;
1540
1541   if ((addrlen != sizeof (struct sockaddr_in)) &&
1542       (addrlen != sizeof (struct sockaddr_in6)))
1543     {
1544       /* invalid address */
1545       GNUNET_break_op (0);
1546       asc (asc_cls, NULL);
1547       return;
1548     }
1549   ppc = GNUNET_malloc (sizeof (struct PrettyPrinterContext));
1550   ppc->asc = asc;
1551   ppc->asc_cls = asc_cls;
1552   if (addrlen == sizeof (struct sockaddr_in))
1553     {
1554       v4 = (const struct sockaddr_in *) addr;
1555       ppc->port = ntohs (v4->sin_port);
1556     }
1557   else
1558     {
1559       v6 = (const struct sockaddr_in6 *) addr;
1560       ppc->port = ntohs (v6->sin6_port);
1561
1562     }
1563   GNUNET_RESOLVER_hostname_get (plugin->env->sched,
1564                                 plugin->env->cfg,
1565                                 addr,
1566                                 addrlen,
1567                                 !numeric, timeout, &append_port, ppc);
1568 }
1569
1570 /**
1571  * Return the actual path to a file found in the current
1572  * PATH environment variable.
1573  *
1574  * @param binary the name of the file to find
1575  */
1576 static char *
1577 get_path_from_PATH (char *binary)
1578 {
1579   char *path;
1580   char *pos;
1581   char *end;
1582   char *buf;
1583   const char *p;
1584
1585   p = getenv ("PATH");
1586   if (p == NULL)
1587     return NULL;
1588   path = GNUNET_strdup (p);     /* because we write on it */
1589   buf = GNUNET_malloc (strlen (path) + 20);
1590   pos = path;
1591
1592   while (NULL != (end = strchr (pos, ':')))
1593     {
1594       *end = '\0';
1595       sprintf (buf, "%s/%s", pos, binary);
1596       if (GNUNET_DISK_file_test (buf) == GNUNET_YES)
1597         {
1598           GNUNET_free (path);
1599           return buf;
1600         }
1601       pos = end + 1;
1602     }
1603   sprintf (buf, "%s/%s", pos, binary);
1604   if (GNUNET_DISK_file_test (buf) == GNUNET_YES)
1605     {
1606       GNUNET_free (path);
1607       return buf;
1608     }
1609   GNUNET_free (buf);
1610   GNUNET_free (path);
1611   return NULL;
1612 }
1613
1614 /**
1615  * Check whether the suid bit is set on a file.
1616  * Attempts to find the file using the current
1617  * PATH environment variable as a search path.
1618  *
1619  * @param binary the name of the file to check
1620  */
1621 static int
1622 check_gnunet_nat_binary(char *binary)
1623 {
1624   struct stat statbuf;
1625   char *p;
1626
1627   p = get_path_from_PATH (binary);
1628   if (p == NULL)
1629     return GNUNET_NO;
1630   if (0 != STAT (p, &statbuf))
1631     {
1632       GNUNET_free (p);
1633       return GNUNET_SYSERR;
1634     }
1635   GNUNET_free (p);
1636   if ( (0 != (statbuf.st_mode & S_ISUID)) &&
1637        (statbuf.st_uid == 0) )
1638     return GNUNET_YES;
1639   return GNUNET_NO;
1640 }
1641
1642 /**
1643  * The exported method. Makes the core api available via a global and
1644  * returns the udp transport API.
1645  */
1646 void *
1647 libgnunet_plugin_transport_udp_init (void *cls)
1648 {
1649   struct GNUNET_TRANSPORT_PluginEnvironment *env = cls;
1650   unsigned long long mtu;
1651   unsigned long long port;
1652   struct GNUNET_TRANSPORT_PluginFunctions *api;
1653   struct Plugin *plugin;
1654   struct GNUNET_SERVICE_Context *service;
1655   int sockets_created;
1656   int behind_nat;
1657   int allow_nat;
1658   char *internal_address;
1659   char *external_address;
1660
1661   service = GNUNET_SERVICE_start ("transport-udp", env->sched, env->cfg);
1662   if (service == NULL)
1663     {
1664       GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING, "udp", _
1665                        ("Failed to start service for `%s' transport plugin.\n"),
1666                        "udp");
1667       return NULL;
1668     }
1669
1670   if (GNUNET_YES == GNUNET_CONFIGURATION_get_value_yesno (env->cfg,
1671                                                          "transport-udp",
1672                                                          "BEHIND_NAT"))
1673     {
1674       /* We are behind nat (according to the user) */
1675       if (check_gnunet_nat_binary("gnunet-nat-server") == GNUNET_YES)
1676         behind_nat = GNUNET_YES;
1677       else
1678         {
1679           behind_nat = GNUNET_NO;
1680           GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING, "udp", "Configuration specified you are behind a NAT, but gnunet-nat-server is not installed properly (suid bit not set)!\n");
1681         }
1682     }
1683   else
1684     behind_nat = GNUNET_NO; /* We are not behind nat! */
1685
1686   if (GNUNET_YES == GNUNET_CONFIGURATION_get_value_yesno (env->cfg,
1687                                                          "transport-udp",
1688                                                          "ALLOW_NAT"))
1689     {
1690       if (check_gnunet_nat_binary("gnunet-nat-client") == GNUNET_YES)
1691         allow_nat = GNUNET_YES; /* We will try to connect to NAT'd peers */
1692       else
1693       {
1694         allow_nat = GNUNET_NO;
1695         GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING, "udp", "Configuration specified you want to connect to NAT'd peers, but gnunet-nat-client is not installed properly (suid bit not set)!\n");
1696       }
1697
1698     }
1699   else
1700     allow_nat = GNUNET_NO; /* We don't want to try to help NAT'd peers */
1701
1702   external_address = NULL;
1703   if (((GNUNET_YES == behind_nat) || (GNUNET_YES == allow_nat)) && (GNUNET_OK !=
1704          GNUNET_CONFIGURATION_get_value_string (env->cfg,
1705                                                 "transport-udp",
1706                                                 "EXTERNAL_ADDRESS",
1707                                                 &external_address)))
1708     {
1709       GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
1710                        "udp",
1711                        _
1712                        ("Require EXTERNAL_ADDRESS for service `%s' in configuration (either BEHIND_NAT or ALLOW_NAT set to YES)!\n"),
1713                        "transport-udp");
1714       GNUNET_SERVICE_stop (service);
1715       return NULL;
1716     }
1717
1718   internal_address = NULL;
1719   if ((GNUNET_YES == behind_nat) && (GNUNET_OK !=
1720          GNUNET_CONFIGURATION_get_value_string (env->cfg,
1721                                                 "transport-udp",
1722                                                 "INTERNAL_ADDRESS",
1723                                                 &internal_address)))
1724     {
1725       GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
1726                        "udp",
1727                        _
1728                        ("Require INTERNAL_ADDRESS for service `%s' in configuration!\n"),
1729                        "transport-udp");
1730       GNUNET_SERVICE_stop (service);
1731       GNUNET_free_non_null(external_address);
1732       return NULL;
1733     }
1734
1735   if (GNUNET_OK !=
1736       GNUNET_CONFIGURATION_get_value_number (env->cfg,
1737                                              "transport-udp",
1738                                              "PORT",
1739                                              &port))
1740     port = UDP_NAT_DEFAULT_PORT;
1741   else if (port > 65535)
1742     {
1743       GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING,
1744                        "udp",
1745                        _("Given `%s' option is out of range: %llu > %u\n"),
1746                        "PORT",
1747                        port,
1748                        65535);
1749       GNUNET_SERVICE_stop (service);
1750       GNUNET_free_non_null(external_address);
1751       GNUNET_free_non_null(internal_address);
1752       return NULL;
1753     }
1754
1755   mtu = 1240;
1756   if (mtu < 1200)
1757     GNUNET_log_from (GNUNET_ERROR_TYPE_INFO,
1758                      "udp",
1759                      _("MTU %llu for `%s' is probably too low!\n"), mtu,
1760                      "UDP");
1761
1762   plugin = GNUNET_malloc (sizeof (struct Plugin));
1763   plugin->external_address = external_address;
1764   plugin->internal_address = internal_address;
1765   plugin->port = port;
1766   plugin->behind_nat = behind_nat;
1767   plugin->allow_nat = allow_nat;
1768   plugin->env = env;
1769
1770   api = GNUNET_malloc (sizeof (struct GNUNET_TRANSPORT_PluginFunctions));
1771   api->cls = plugin;
1772
1773   api->send = &udp_plugin_send;
1774   api->disconnect = &udp_disconnect;
1775   api->address_pretty_printer = &udp_plugin_address_pretty_printer;
1776   api->check_address = &udp_check_address;
1777
1778   plugin->service = service;
1779
1780   GNUNET_OS_network_interfaces_list (&process_interfaces, plugin);
1781   plugin->hostname_dns = GNUNET_RESOLVER_hostname_resolve (env->sched,
1782                                                            env->cfg,
1783                                                            AF_UNSPEC,
1784                                                            HOSTNAME_RESOLVE_TIMEOUT,
1785                                                            &process_hostname_ips,
1786                                                            plugin);
1787
1788   sockets_created = udp_transport_server_start (plugin);
1789
1790   GNUNET_assert (sockets_created == 1);
1791
1792   return api;
1793 }
1794
1795
1796 void *
1797 libgnunet_plugin_transport_udp_done (void *cls)
1798 {
1799   struct GNUNET_TRANSPORT_PluginFunctions *api = cls;
1800   struct Plugin *plugin = api->cls;
1801
1802   udp_transport_server_stop (plugin);
1803   if (NULL != hostname_dns)
1804     {
1805       GNUNET_RESOLVER_request_cancel (hostname_dns);
1806       hostname_dns = NULL;
1807     }
1808
1809   GNUNET_SERVICE_stop (plugin->service);
1810
1811   GNUNET_NETWORK_fdset_destroy (plugin->rs);
1812   GNUNET_free (plugin);
1813   GNUNET_free (api);
1814   return NULL;
1815 }
1816
1817 /* end of plugin_transport_udp.c */