src/gns/gnunet-gns-proxy.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2012 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 #include "platform.h"
  22 #include <gnunet_util_lib.h>
  23 #include "gns_proxy_proto.h"
  24 #include "gns.h"
  25
  26 #define GNUNET_GNS_PROXY_PORT 7777
  27
  28 //TODO maybe make this an api call
  29 /**
  30  * Checks if name is in tld
  31  *
  32  * @param name the name to check
  33  * @param tld the TLD to check for
  34  * @return GNUNET_YES or GNUNET_NO
  35  */
  36 int
  37 is_tld(const char* name, const char* tld)
  38 {
  39   int offset = 0;
  40
  41   if (strlen(name) <= strlen(tld))
  42   {
  43     return GNUNET_NO;
  44   }
  45
  46   offset = strlen(name)-strlen(tld);
  47   if (strcmp (name+offset, tld) != 0)
  48   {
  49     GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
  50                "%s is not in .%s TLD\n", name, tld);
  51     return GNUNET_NO;
  52   }
  53
  54   return GNUNET_YES;
  55 }
  56
  57 struct Socks5Request
  58 {
  59   struct Socks5Request *prev;
  60   struct Socks5Request *next;
  61
  62   struct GNUNET_NETWORK_Handle *sock;
  63   struct GNUNET_NETWORK_Handle *remote_sock;
  64
  65   int state;
  66
  67   GNUNET_SCHEDULER_TaskIdentifier rtask;
  68   GNUNET_SCHEDULER_TaskIdentifier wtask;
  69
  70   char rbuf[512];
  71   char wbuf[512];
  72   unsigned int rbuf_len;
  73   unsigned int wbuf_len;
  74 };
  75
  76 struct Socks5Connections
  77 {
  78   struct Socks5Request *head;
  79   struct Socks5Request *tail;
  80 };
  81
  82
  83 unsigned long port = GNUNET_GNS_PROXY_PORT;
  84 static struct GNUNET_NETWORK_Handle *lsock;
  85 GNUNET_SCHEDULER_TaskIdentifier ltask;
  86 static struct Socks5Connections s5conns;
  87
  88 /**
  89  * Write data to socket
  90  *
  91  * @param cls the closure
  92  * @param tc scheduler context
  93  */
  94 static void
  95 do_write (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  96 {
  97   struct Socks5Request *s5r = cls;
  98   unsigned int len = s5r->wbuf_len;
  99
 100   s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
 101
 102   if ((NULL != tc->read_ready) &&
 103       (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->sock)) &&
 104       (len = GNUNET_NETWORK_socket_send (s5r->sock, &s5r->wbuf,
 105                                          len)))
 106   {
 107     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 108                 "Successfully sent %d bytes to socket\n",
 109                 len);
 110   }
 111   else
 112   {
 113     GNUNET_NETWORK_socket_close (s5r->sock);
 114     GNUNET_free(s5r);
 115     return;
 116   }
 117 }
 118
 119 /**
 120  * Read data from incoming connection
 121  *
 122  * @param cls the closure
 123  * @param tc the scheduler context
 124  */
 125 static void
 126 do_read (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 127 {
 128   struct Socks5Request *s5r = cls;
 129   unsigned int len;
 130   struct socks5_client_hello *c_hello;
 131   struct socks5_server_hello *s_hello;
 132   struct socks5_client_request *c_req;
 133   struct socks5_server_response *s_resp;
 134
 135   char domain[256];
 136   uint8_t dom_len;
 137   uint16_t req_port;
 138   struct hostent *phost;
 139   uint32_t remote_ip;
 140   struct sockaddr_in remote_addr;
 141   struct in_addr *r_sin_addr;
 142
 143   s5r->rtask = GNUNET_SCHEDULER_NO_TASK;
 144
 145   if ((NULL != tc->write_ready) &&
 146       (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->sock)) &&
 147       (len = GNUNET_NETWORK_socket_recv (s5r->sock, &s5r->rbuf,
 148                                          sizeof (s5r->rbuf))))
 149   {
 150     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 151                 "Successfully read %d bytes from socket\n",
 152                 len);
 153   }
 154   else
 155   {
 156     //ERROR!
 157     GNUNET_NETWORK_socket_close (s5r->sock);
 158     GNUNET_free(s5r);
 159     return;
 160   }
 161
 162   if (s5r->state == SOCKS5_INIT)
 163   {
 164     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 165                 "SOCKS5 init\n");
 166     c_hello = (struct socks5_client_hello*)&s5r->rbuf;
 167
 168     GNUNET_assert (c_hello->version == SOCKS_VERSION_5);
 169
 170     s_hello = (struct socks5_server_hello*)&s5r->wbuf;
 171     s5r->wbuf_len = sizeof( struct socks5_server_hello );
 172
 173     s_hello->version = c_hello->version;
 174     s_hello->auth_method = SOCKS_AUTH_NONE;
 175
 176     /* Write response to client */
 177     s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
 178                                                 s5r->sock,
 179                                                 &do_write, s5r);
 180
 181     s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 182                                                 s5r->sock,
 183                                                 &do_read, s5r);
 184
 185     s5r->state = SOCKS5_REQUEST;
 186     return;
 187   }
 188
 189   if (s5r->state == SOCKS5_REQUEST)
 190   {
 191     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 192                 "Processing SOCKS5 request\n");
 193     c_req = (struct socks5_client_request*)&s5r->rbuf;
 194     s_resp = (struct socks5_server_response*)&s5r->wbuf;
 195     s5r->wbuf_len = sizeof (struct socks5_server_response);
 196
 197     GNUNET_assert (c_req->addr_type == 3);
 198
 199     dom_len = *((uint8_t*)(&(c_req->addr_type) + 1));
 200     memset(domain, 0, sizeof(domain));
 201     strncpy(domain, (char*)(&(c_req->addr_type) + 2), dom_len);
 202     req_port = *((uint16_t*)(&(c_req->addr_type) + 2 + dom_len));
 203
 204     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 205                 "Requested connection is %s:%d\n",
 206                 domain,
 207                 ntohs(req_port));
 208
 209     if (is_tld (domain, GNUNET_GNS_TLD) ||
 210         is_tld (domain, GNUNET_GNS_TLD_ZKEY))
 211     {
 212       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 213                   "Requested connection is gnunet tld\n",
 214                   domain);
 215     }
 216     else
 217     {
 218       phost = (struct hostent*)gethostbyname (domain);
 219       if (phost == NULL)
 220       {
 221         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 222                     "Resolve %s error!\n", domain );
 223         s_resp->version = 0x05;
 224         s_resp->reply = 0x01;
 225         s5r->wtask =
 226           GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
 227                                           s5r->sock,
 228                                           &do_write, s5r);
 229         //ERROR!
 230         //TODO! close socket after the write! schedule task
 231         //GNUNET_NETWORK_socket_close (s5r->sock);
 232         //GNUNET_free(s5r);
 233         return;
 234       }
 235
 236       s5r->remote_sock = GNUNET_NETWORK_socket_create (AF_INET,
 237                                                        SOCK_STREAM,
 238                                                        0);
 239       r_sin_addr = (struct in_addr*)(phost->h_addr);
 240       remote_ip = r_sin_addr->s_addr;
 241       memset(&remote_addr, 0, sizeof(remote_addr));
 242       remote_addr.sin_family = AF_INET;
 243       remote_addr.sin_addr.s_addr = remote_ip;
 244       remote_addr.sin_port = req_port;
 245
 246       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 247                   "target server: %s:%u\n", inet_ntoa(remote_addr.sin_addr),
 248                   ntohs(req_port));
 249
 250       GNUNET_assert ( s5r->remote_sock != NULL );
 251
 252       if (GNUNET_OK !=
 253           GNUNET_NETWORK_socket_connect ( s5r->remote_sock,
 254                                           (struct sockaddr*)&remote_addr,
 255                                           sizeof (struct sockaddr)))
 256       {
 257         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 258                     "socket request error...\n");
 259         s_resp->version = 0x05;
 260         s_resp->reply = 0x01;
 261         s5r->wtask =
 262           GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
 263                                           s5r->sock,
 264                                           &do_write, s5r);
 265         //TODO see above
 266         return;
 267       }
 268
 269       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 270                   "new remote connection\n");
 271
 272       s_resp->version = 0x05;
 273       s_resp->reply = 0x00;
 274       s_resp->reserved = 0x00;
 275       s_resp->addr_type = 0x01;
 276
 277       s5r->state = SOCKS5_DATA_TRANSFER;
 278
 279       s5r->wtask =
 280         GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
 281                                         s5r->sock,
 282                                         &do_write, s5r);
 283
 284     }
 285   }
 286
 287   //GNUNET_CONTAINER_DLL_remove (s5conns.head, s5conns.tail, s5r);
 288
 289 }
 290
 291 /**
 292  * Accept new incoming connections
 293  *
 294  * @param cls the closure
 295  * @param tc the scheduler context
 296  */
 297 static void
 298 do_accept (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 299 {
 300   struct GNUNET_NETWORK_Handle *s;
 301   struct Socks5Request *s5r;
 302
 303   ltask = GNUNET_SCHEDULER_NO_TASK;
 304   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 305     return;
 306
 307   ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 308                                          lsock,
 309                                          &do_accept, NULL);
 310
 311   s = GNUNET_NETWORK_socket_accept (lsock, NULL, NULL);
 312
 313   if (NULL == s)
 314   {
 315     GNUNET_log_strerror (GNUNET_ERROR_TYPE_INFO, "accept");
 316     return;
 317   }
 318
 319   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 320               "Got an inbound connection, waiting for data\n");
 321
 322   s5r = GNUNET_malloc (sizeof (struct Socks5Request));
 323   s5r->sock = s;
 324   s5r->state = SOCKS5_INIT;
 325   s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 326                                               s5r->sock,
 327                                               &do_read, s5r);
 328   GNUNET_CONTAINER_DLL_insert (s5conns.head, s5conns.tail, s5r);
 329 }
 330
 331 /**
 332  * Main function that will be run
 333  *
 334  * @param cls closure
 335  * @param args remaining command-line arguments
 336  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 337  * @param cfg configuration
 338  */
 339 static void
 340 run (void *cls, char *const *args, const char *cfgfile,
 341      const struct GNUNET_CONFIGURATION_Handle *cfg)
 342 {
 343   struct sockaddr_in sa;
 344
 345   memset (&sa, 0, sizeof (sa));
 346   sa.sin_family = AF_INET;
 347   sa.sin_port = htons (port);
 348 #if HAVE_SOCKADDR_IN_SIN_LEN
 349   sa.sin_len = sizeof (sa);
 350 #endif
 351
 352   lsock = GNUNET_NETWORK_socket_create (AF_INET,
 353                                         SOCK_STREAM,
 354                                         0);
 355
 356   if ((NULL == lsock) ||
 357       (GNUNET_OK !=
 358        GNUNET_NETWORK_socket_bind (lsock, (const struct sockaddr *) &sa,
 359                                    sizeof (sa))))
 360   {
 361     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 362                 "Failed to create listen socket bound to `%s'",
 363                 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
 364     if (NULL != lsock)
 365       GNUNET_NETWORK_socket_close (lsock);
 366     return;
 367   }
 368
 369   if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock, 5))
 370   {
 371     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 372                 "Failed to listen on socket bound to `%s'",
 373                 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
 374     return;
 375   }
 376
 377   ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
 378                                          lsock, &do_accept, NULL);
 379
 380   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 381               "Proxy listens on port %u\n",
 382               port);
 383
 384 }
 385
 386 /**
 387  * The main function for gnunet-gns-proxy.
 388  *
 389  * @param argc number of arguments from the command line
 390  * @param argv command line arguments
 391  * @return 0 ok, 1 on error
 392  */
 393 int
 394 main (int argc, char *const *argv)
 395 {
 396   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 397     {'p', "port", NULL,
 398      gettext_noop ("listen on specified port"), 1,
 399      &GNUNET_GETOPT_set_string, &port},
 400     GNUNET_GETOPT_OPTION_END
 401   };
 402
 403   int ret;
 404
 405   GNUNET_log_setup ("gnunet-gns-proxy", "WARNING", NULL);
 406   ret =
 407       (GNUNET_OK ==
 408        GNUNET_PROGRAM_run (argc, argv, "gnunet-gns-proxy",
 409                            _("GNUnet GNS proxy"),
 410                            options,
 411                            &run, NULL)) ? 0 : 1;
 412   return ret;
 413 }