src/nat/nat_mini.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2011 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 /**
  22  * @file nat/nat_mini.c
  23  * @brief functions for interaction with miniupnp
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28 #include "gnunet_nat_lib.h"
  29 #include "nat.h"
  30
  31 /**
  32  * How long do we give upnpc to create a mapping?
  33  */
  34 #define MAP_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 15)
  35
  36
  37 /**
  38  * How long do we give upnpc to remove a mapping?
  39  */
  40 #define UNMAP_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1)
  41
  42 /**
  43  * How often do we check for changes in the mapping?
  44  */
  45 #define MAP_REFRESH_FREQ GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 15)
  46
  47
  48 /**
  49  * Try to get the external IPv4 address of this peer.
  50  * Note: calling this function may block this process
  51  * for a few seconds (!).
  52  *
  53  * @param addr address to set
  54  * @return GNUNET_OK on success,
  55  *         GNUNET_NO if the result is questionable,
  56  *         GNUNET_SYSERR on error
  57  */
  58 int
  59 GNUNET_NAT_mini_get_external_ipv4 (struct in_addr *addr)
  60 {
  61   struct GNUNET_OS_Process *eip;
  62   struct GNUNET_DISK_PipeHandle *opipe;
  63   const struct GNUNET_DISK_FileHandle *r;
  64   size_t off;
  65   char buf[17];
  66   ssize_t ret;
  67   int iret;
  68
  69   opipe = GNUNET_DISK_pipe (GNUNET_YES,
  70                             GNUNET_NO,
  71                             GNUNET_YES);
  72   if (NULL == opipe)
  73     return GNUNET_SYSERR;
  74   eip = GNUNET_OS_start_process (NULL,
  75                                  opipe,
  76                                  "external-ip",
  77                                  "external-ip", NULL);
  78   if (NULL == eip)
  79     {
  80       GNUNET_DISK_pipe_close (opipe);
  81       return GNUNET_SYSERR;
  82     }
  83   GNUNET_DISK_pipe_close_end (opipe, GNUNET_DISK_PIPE_END_WRITE);
  84   iret = GNUNET_SYSERR;
  85   r = GNUNET_DISK_pipe_handle (opipe,
  86                                GNUNET_DISK_PIPE_END_READ);
  87   off = 0;
  88   while (0 < (ret = GNUNET_DISK_file_read (r, &buf[off], sizeof (buf)-off)))
  89     off += ret;
  90   if ( (off > 7) &&
  91        (buf[off-1] == '\n') )
  92     {
  93       buf[off-1] = '\0';
  94       if (1 == inet_pton (AF_INET, buf, addr))
  95         {
  96           if (addr->s_addr == 0)
  97             iret = GNUNET_NO; /* got 0.0.0.0 */
  98           iret = GNUNET_OK;
  99         }
 100     }
 101   (void) GNUNET_OS_process_kill (eip, SIGKILL);
 102   GNUNET_OS_process_close (eip);
 103   GNUNET_DISK_pipe_close (opipe);
 104   return iret;
 105 }
 106
 107
 108 /**
 109  * Handle to a mapping created with upnpc.
 110  */
 111 struct GNUNET_NAT_MiniHandle
 112 {
 113
 114   /**
 115    * Function to call on mapping changes.
 116    */
 117   GNUNET_NAT_AddressCallback ac;
 118
 119   /**
 120    * Closure for 'ac'.
 121    */
 122   void *ac_cls;
 123
 124   /**
 125    * Command used to install the map.
 126    */
 127   struct GNUNET_OS_CommandHandle *map_cmd;
 128
 129   /**
 130    * Command used to refresh our map information.
 131    */
 132   struct GNUNET_OS_CommandHandle *refresh_cmd;
 133
 134   /**
 135    * Command used to remove the mapping.
 136    */
 137   struct GNUNET_OS_CommandHandle *unmap_cmd;
 138
 139   /**
 140    * Our current external mapping (if we have one).
 141    */
 142   struct sockaddr_in current_addr;
 143
 144   /**
 145    * We check the mapping periodically to see if it
 146    * still works.  This task triggers the check.
 147    */
 148   GNUNET_SCHEDULER_TaskIdentifier refresh_task;
 149
 150   /**
 151    * Are we mapping TCP or UDP?
 152    */
 153   int is_tcp;
 154
 155   /**
 156    * Did we succeed with creating a mapping?
 157    */
 158   int did_map;
 159
 160   /**
 161    * Which port are we mapping?
 162    */
 163   uint16_t port;
 164
 165 };
 166
 167
 168 /**
 169  * Run upnpc -l to find out if our mapping changed.
 170  *
 171  * @param cls the 'struct GNUNET_NAT_MiniHandle'
 172  * @param tc scheduler context
 173  */
 174 static void
 175 do_refresh (void *cls,
 176             const struct GNUNET_SCHEDULER_TaskContext *tc);
 177
 178
 179 /**
 180  * Process the output from 'upnpc -l' to see if our
 181  * external mapping changed.  If so, do the notifications.
 182  *
 183  * @param cls the 'struct GNUNET_NAT_MiniHandle'
 184  * @param line line of output, NULL at the end
 185  */
 186 static void
 187 process_refresh_output (void *cls,
 188                         const char *line)
 189 {
 190   struct GNUNET_NAT_MiniHandle *mini = cls;
 191   enum GNUNET_OS_ProcessStatusType type;
 192   unsigned long code;
 193
 194   if (NULL == line)
 195     {
 196       GNUNET_OS_command_stop (mini->refresh_cmd,
 197                               &type, &code);
 198       mini->refresh_cmd = NULL;
 199       mini->refresh_task = GNUNET_SCHEDULER_add_delayed (MAP_REFRESH_FREQ,
 200                                                          &do_refresh,
 201                                                          mini);
 202       return;
 203     }
 204   /* FIXME: parse 'line' */
 205   fprintf (stderr,
 206            "Refresh output: `%s'\n",
 207            line);
 208 }
 209
 210
 211 /**
 212  * Run upnpc -l to find out if our mapping changed.
 213  *
 214  * @param cls the 'struct GNUNET_NAT_MiniHandle'
 215  * @param tc scheduler context
 216  */
 217 static void
 218 do_refresh (void *cls,
 219             const struct GNUNET_SCHEDULER_TaskContext *tc)
 220 {
 221   struct GNUNET_NAT_MiniHandle *mini = cls;
 222
 223   mini->refresh_task = GNUNET_SCHEDULER_NO_TASK;
 224   mini->refresh_cmd = GNUNET_OS_command_run (&process_refresh_output,
 225                                              mini,
 226                                              MAP_TIMEOUT,
 227                                              "upnpc",
 228                                              "upnpc",
 229                                              "-l",
 230                                              NULL);
 231 }
 232
 233
 234 /**
 235  * Process the output from the 'upnpc -r' command.
 236  *
 237  * @param cls the 'struct GNUNET_NAT_MiniHandle'
 238  * @param line line of output, NULL at the end
 239  */
 240 static void
 241 process_map_output (void *cls,
 242                     const char *line)
 243 {
 244   struct GNUNET_NAT_MiniHandle *mini = cls;
 245   enum GNUNET_OS_ProcessStatusType type;
 246   unsigned long code;
 247   const char *ipaddr;
 248   char *ipa;
 249   const char *pstr;
 250   unsigned int port;
 251
 252   if (NULL == line)
 253     {
 254       GNUNET_OS_command_stop (mini->map_cmd,
 255                               &type, &code);
 256       mini->map_cmd = NULL;
 257       if (mini->did_map == GNUNET_YES)
 258         mini->refresh_task = GNUNET_SCHEDULER_add_delayed (MAP_REFRESH_FREQ,
 259                                                            &do_refresh,
 260                                                            mini);
 261       return;
 262     }
 263   /*
 264     The upnpc output we're after looks like this:
 265
 266      "external 87.123.42.204:3000 TCP is redirected to internal 192.168.2.150:3000"
 267   */
 268   if ( (NULL == (ipaddr = strstr (line, " "))) ||
 269        (NULL == (pstr = strstr (ipaddr, ":"))) ||
 270        (1 != sscanf (pstr + 1, "%u", &port)) )
 271     {
 272       fprintf (stderr,
 273                "Skipping output `%s'\n",
 274                line);
 275       return; /* skip line */
 276     }
 277   ipa = GNUNET_strdup (ipaddr + 1);
 278   strstr (ipa, ":")[0] = '\0';
 279   if (1 != inet_pton (AF_INET,
 280                       ipa,
 281                       &mini->current_addr.sin_addr))
 282     {
 283       GNUNET_free (ipa);
 284       fprintf (stderr,
 285                "Skipping output `%s'\n",
 286                line);
 287       return; /* skip line */
 288     }
 289   GNUNET_free (ipa);
 290
 291   mini->current_addr.sin_port = htons (port);
 292   mini->current_addr.sin_family = AF_INET;
 293 #if HAVE_SOCKADDR_IN_SIN_LEN
 294   mini->current_addr.sin_len = sizeof (struct sockaddr_in);
 295 #endif
 296   mini->did_map = GNUNET_YES;
 297   mini->ac (mini->ac_cls, GNUNET_YES,
 298             (const struct sockaddr*) &mini->current_addr,
 299             sizeof (mini->current_addr));
 300 }
 301
 302
 303 /**
 304  * Start mapping the given port using (mini)upnpc.  This function
 305  * should typically not be used directly (it is used within the
 306  * general-purpose 'GNUNET_NAT_register' code).  However, it can be
 307  * used if specifically UPnP-based NAT traversal is to be used or
 308  * tested.
 309  *
 310  * @param port port to map
 311  * @param is_tcp GNUNET_YES to map TCP, GNUNET_NO for UDP
 312  * @param ac function to call with mapping result
 313  * @param ac_cls closure for 'ac'
 314  * @return NULL on error
 315  */
 316 struct GNUNET_NAT_MiniHandle *
 317 GNUNET_NAT_mini_map_start (uint16_t port,
 318                            int is_tcp,
 319                            GNUNET_NAT_AddressCallback ac,
 320                            void *ac_cls)
 321 {
 322   struct GNUNET_NAT_MiniHandle *ret;
 323   char pstr[6];
 324
 325   ret = GNUNET_malloc (sizeof (struct GNUNET_NAT_MiniHandle));
 326   ret->ac = ac;
 327   ret->ac_cls = ac_cls;
 328   ret->is_tcp = is_tcp;
 329   ret->port = port;
 330   GNUNET_snprintf (pstr, sizeof (pstr),
 331                    "%u",
 332                    (unsigned int) port);
 333   ret->map_cmd = GNUNET_OS_command_run (&process_map_output,
 334                                         ret,
 335                                         MAP_TIMEOUT,
 336                                         "upnpc",
 337                                         "upnpc",
 338                                         "-r", pstr,
 339                                         is_tcp ? "tcp" : "udp",
 340                                         NULL);
 341
 342   return ret;
 343 }
 344
 345
 346 /**
 347  * Process output from our 'unmap' command.
 348  *
 349  * @param cls the 'struct GNUNET_NAT_MiniHandle'
 350  * @param line line of output, NULL at the end
 351  */
 352 static void
 353 process_unmap_output (void *cls,
 354                       const char *line)
 355 {
 356   struct GNUNET_NAT_MiniHandle *mini = cls;
 357   enum GNUNET_OS_ProcessStatusType type;
 358   unsigned long code;
 359
 360   if (NULL == line)
 361     {
 362       GNUNET_OS_command_stop (mini->unmap_cmd,
 363                               &type, &code);
 364       mini->unmap_cmd = NULL;
 365       GNUNET_free (mini);
 366       return;
 367     }
 368   /* we don't really care about the output... */
 369 }
 370
 371
 372 /**
 373  * Remove a mapping created with (mini)upnpc.  Calling
 374  * this function will give 'upnpc' 1s to remove tha mapping,
 375  * so while this function is non-blocking, a task will be
 376  * left with the scheduler for up to 1s past this call.
 377  *
 378  * @param mini the handle
 379  */
 380 void
 381 GNUNET_NAT_mini_map_stop (struct GNUNET_NAT_MiniHandle *mini)
 382 {
 383   char pstr[6];
 384   enum GNUNET_OS_ProcessStatusType type;
 385   unsigned long code;
 386
 387   if (! mini->did_map)
 388     {
 389       if (mini->map_cmd != NULL)
 390         {
 391           GNUNET_OS_command_stop (mini->map_cmd,
 392                                   &type, &code);
 393           mini->map_cmd = NULL;
 394         }
 395       GNUNET_free (mini);
 396       return;
 397     }
 398   if (GNUNET_SCHEDULER_NO_TASK != mini->refresh_task)
 399     {
 400       GNUNET_SCHEDULER_cancel (mini->refresh_task);
 401       mini->refresh_task = GNUNET_SCHEDULER_NO_TASK;
 402     }
 403   if (mini->refresh_cmd != NULL)
 404     {
 405       GNUNET_OS_command_stop (mini->refresh_cmd,
 406                               &type, &code);
 407       mini->refresh_cmd = NULL;
 408     }
 409   mini->ac (mini->ac_cls, GNUNET_NO,
 410             (const struct sockaddr*) &mini->current_addr,
 411             sizeof (mini->current_addr));
 412   GNUNET_snprintf (pstr, sizeof (pstr),
 413                    "%u",
 414                    (unsigned int) mini->port);
 415   mini->unmap_cmd = GNUNET_OS_command_run (&process_unmap_output,
 416                                            mini,
 417                                            UNMAP_TIMEOUT,
 418                                            "upnpc",
 419                                            "upnpc",
 420                                            "-d", pstr,
 421                                            mini->is_tcp ? "tcp" : "udp",
 422                                            NULL);
 423 }
 424
 425
 426 /* end of nat_mini.c */