2 This file is part of GNUnet.
3 (C) 2012 Christian Grothoff (and other contributing authors)
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.
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.
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.
22 #include <gnunet_util_lib.h>
23 #include "gns_proxy_proto.h"
26 #define GNUNET_GNS_PROXY_PORT 7777
28 //TODO maybe make this an api call
30 * Checks if name is in tld
32 * @param name the name to check
33 * @param tld the TLD to check for
34 * @return GNUNET_YES or GNUNET_NO
37 is_tld(const char* name, const char* tld)
41 if (strlen(name) <= strlen(tld))
46 offset = strlen(name)-strlen(tld);
47 if (strcmp (name+offset, tld) != 0)
49 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
50 "%s is not in .%s TLD\n", name, tld);
59 struct Socks5Request *prev;
60 struct Socks5Request *next;
62 struct GNUNET_NETWORK_Handle *sock;
63 struct GNUNET_NETWORK_Handle *remote_sock;
67 GNUNET_SCHEDULER_TaskIdentifier rtask;
68 GNUNET_SCHEDULER_TaskIdentifier fwdrtask;
69 GNUNET_SCHEDULER_TaskIdentifier wtask;
70 GNUNET_SCHEDULER_TaskIdentifier fwdwtask;
74 unsigned int rbuf_len;
75 unsigned int wbuf_len;
78 struct Socks5Connections
80 struct Socks5Request *head;
81 struct Socks5Request *tail;
85 unsigned long port = GNUNET_GNS_PROXY_PORT;
86 static struct GNUNET_NETWORK_Handle *lsock;
87 GNUNET_SCHEDULER_TaskIdentifier ltask;
88 static struct Socks5Connections s5conns;
91 * Read data from socket
93 * @param cls the closure
94 * @param tc scheduler context
97 do_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
100 * Read from remote end
103 * @param tc scheduler context
106 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
109 * Write data to remote socket
111 * @param cls the closure
112 * @param tc scheduler context
115 do_write_remote (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
117 struct Socks5Request *s5r = cls;
120 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
122 if ((NULL != tc->read_ready) &&
123 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->remote_sock)) &&
124 (len = GNUNET_NETWORK_socket_send (s5r->remote_sock, s5r->rbuf,
127 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
128 "Successfully sent %d bytes to remote socket\n",
133 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write remote");
135 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
136 GNUNET_SCHEDULER_cancel (s5r->rtask);
137 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
138 GNUNET_SCHEDULER_cancel (s5r->wtask);
139 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
140 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
141 GNUNET_NETWORK_socket_close (s5r->remote_sock);
142 GNUNET_NETWORK_socket_close (s5r->sock);
148 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
155 * Write data to socket
157 * @param cls the closure
158 * @param tc scheduler context
161 do_write (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
163 struct Socks5Request *s5r = cls;
166 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
168 if ((NULL != tc->read_ready) &&
169 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->sock)) &&
170 (len = GNUNET_NETWORK_socket_send (s5r->sock, s5r->wbuf,
173 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
174 "Successfully sent %d bytes to socket\n",
180 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write");
182 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
183 GNUNET_SCHEDULER_cancel (s5r->rtask);
184 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
185 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
186 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
187 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
188 GNUNET_NETWORK_socket_close (s5r->remote_sock);
189 GNUNET_NETWORK_socket_close (s5r->sock);
194 if ((s5r->state == SOCKS5_DATA_TRANSFER) &&
195 (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK))
197 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
199 &do_read_remote, s5r);
203 * Read from remote end
206 * @param tc scheduler context
209 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
211 struct Socks5Request *s5r = cls;
213 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
216 if ((NULL != tc->write_ready) &&
217 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->remote_sock)) &&
218 (s5r->wbuf_len = GNUNET_NETWORK_socket_recv (s5r->remote_sock, s5r->wbuf,
219 sizeof (s5r->wbuf))))
221 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
222 "Successfully read %d bytes from remote socket\n",
227 if (s5r->wbuf_len == 0)
228 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
229 "0 bytes received from remote... graceful shutdown!\n");
230 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
231 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
232 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
233 GNUNET_SCHEDULER_cancel (s5r->rtask);
235 GNUNET_NETWORK_socket_close (s5r->remote_sock);
236 s5r->remote_sock = NULL;
237 GNUNET_NETWORK_socket_close (s5r->sock);
243 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
252 * Read data from incoming connection
254 * @param cls the closure
255 * @param tc the scheduler context
258 do_read (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
260 struct Socks5Request *s5r = cls;
261 struct socks5_client_hello *c_hello;
262 struct socks5_server_hello *s_hello;
263 struct socks5_client_request *c_req;
264 struct socks5_server_response *s_resp;
269 struct hostent *phost;
271 struct sockaddr_in remote_addr;
272 struct in_addr *r_sin_addr;
274 s5r->rtask = GNUNET_SCHEDULER_NO_TASK;
276 if ((NULL != tc->write_ready) &&
277 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->sock)) &&
278 (s5r->rbuf_len = GNUNET_NETWORK_socket_recv (s5r->sock, s5r->rbuf,
279 sizeof (s5r->rbuf))))
281 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
282 "Successfully read %d bytes from socket\n",
287 if (s5r->rbuf_len != 0)
288 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "read");
290 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disco!\n");
292 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
293 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
294 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
295 GNUNET_SCHEDULER_cancel (s5r->wtask);
296 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
297 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
298 GNUNET_NETWORK_socket_close (s5r->remote_sock);
299 GNUNET_NETWORK_socket_close (s5r->sock);
304 if (s5r->state == SOCKS5_INIT)
306 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
308 c_hello = (struct socks5_client_hello*)&s5r->rbuf;
310 GNUNET_assert (c_hello->version == SOCKS_VERSION_5);
312 s_hello = (struct socks5_server_hello*)&s5r->wbuf;
313 s5r->wbuf_len = sizeof( struct socks5_server_hello );
315 s_hello->version = c_hello->version;
316 s_hello->auth_method = SOCKS_AUTH_NONE;
318 /* Write response to client */
319 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
323 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
327 s5r->state = SOCKS5_REQUEST;
331 if (s5r->state == SOCKS5_REQUEST)
333 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
334 "Processing SOCKS5 request\n");
335 c_req = (struct socks5_client_request*)&s5r->rbuf;
336 s_resp = (struct socks5_server_response*)&s5r->wbuf;
337 //Only 10byte for ipv4 response!
338 s5r->wbuf_len = 10;//sizeof (struct socks5_server_response);
340 GNUNET_assert (c_req->addr_type == 3);
342 dom_len = *((uint8_t*)(&(c_req->addr_type) + 1));
343 memset(domain, 0, sizeof(domain));
344 strncpy(domain, (char*)(&(c_req->addr_type) + 2), dom_len);
345 req_port = *((uint16_t*)(&(c_req->addr_type) + 2 + dom_len));
347 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
348 "Requested connection is %s:%d\n",
352 if (is_tld (domain, GNUNET_GNS_TLD) ||
353 is_tld (domain, GNUNET_GNS_TLD_ZKEY))
355 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
356 "Requested connection is gnunet tld\n",
361 phost = (struct hostent*)gethostbyname (domain);
364 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
365 "Resolve %s error!\n", domain );
366 s_resp->version = 0x05;
367 s_resp->reply = 0x01;
369 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
373 //TODO! close socket after the write! schedule task
374 //GNUNET_NETWORK_socket_close (s5r->sock);
379 s5r->remote_sock = GNUNET_NETWORK_socket_create (AF_INET,
382 r_sin_addr = (struct in_addr*)(phost->h_addr);
383 remote_ip = r_sin_addr->s_addr;
384 memset(&remote_addr, 0, sizeof(remote_addr));
385 remote_addr.sin_family = AF_INET;
386 #if HAVE_SOCKADDR_IN_SIN_LEN
387 remote_addr.sin_len = sizeof (remote_addr);
389 remote_addr.sin_addr.s_addr = remote_ip;
390 remote_addr.sin_port = req_port;
392 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
393 "target server: %s:%u\n", inet_ntoa(remote_addr.sin_addr),
397 GNUNET_NETWORK_socket_connect ( s5r->remote_sock,
398 (const struct sockaddr*)&remote_addr,
399 sizeof (remote_addr)))
400 && (errno != EINPROGRESS))
402 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "connect");
403 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
404 "socket request error...\n");
405 s_resp->version = 0x05;
406 s_resp->reply = 0x01;
408 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
415 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
416 "new remote connection\n");
418 s_resp->version = 0x05;
419 s_resp->reply = 0x00;
420 s_resp->reserved = 0x00;
421 s_resp->addr_type = 0x01;
423 s5r->state = SOCKS5_DATA_TRANSFER;
426 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
430 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
438 if (s5r->state == SOCKS5_DATA_TRANSFER)
440 if ((s5r->remote_sock == NULL) || (s5r->rbuf_len == 0))
442 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
443 "Closing connection to client\n");
444 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
445 GNUNET_SCHEDULER_cancel (s5r->rtask);
446 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
447 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
448 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
449 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
450 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
451 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
453 if (s5r->remote_sock != NULL)
454 GNUNET_NETWORK_socket_close (s5r->remote_sock);
455 GNUNET_NETWORK_socket_close (s5r->sock);
460 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
461 "forwarding %d bytes from client\n", s5r->rbuf_len);
464 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
466 &do_write_remote, s5r);
468 if (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK)
471 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
473 &do_read_remote, s5r);
479 //GNUNET_CONTAINER_DLL_remove (s5conns.head, s5conns.tail, s5r);
484 * Accept new incoming connections
486 * @param cls the closure
487 * @param tc the scheduler context
490 do_accept (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
492 struct GNUNET_NETWORK_Handle *s;
493 struct Socks5Request *s5r;
495 ltask = GNUNET_SCHEDULER_NO_TASK;
496 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
499 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
503 s = GNUNET_NETWORK_socket_accept (lsock, NULL, NULL);
507 GNUNET_log_strerror (GNUNET_ERROR_TYPE_INFO, "accept");
511 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
512 "Got an inbound connection, waiting for data\n");
514 s5r = GNUNET_malloc (sizeof (struct Socks5Request));
516 s5r->state = SOCKS5_INIT;
517 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
518 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
519 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
520 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
523 //GNUNET_CONTAINER_DLL_insert (s5conns.head, s5conns.tail, s5r);
527 * Main function that will be run
530 * @param args remaining command-line arguments
531 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
532 * @param cfg configuration
535 run (void *cls, char *const *args, const char *cfgfile,
536 const struct GNUNET_CONFIGURATION_Handle *cfg)
538 struct sockaddr_in sa;
540 memset (&sa, 0, sizeof (sa));
541 sa.sin_family = AF_INET;
542 sa.sin_port = htons (port);
543 #if HAVE_SOCKADDR_IN_SIN_LEN
544 sa.sin_len = sizeof (sa);
547 lsock = GNUNET_NETWORK_socket_create (AF_INET,
551 if ((NULL == lsock) ||
553 GNUNET_NETWORK_socket_bind (lsock, (const struct sockaddr *) &sa,
556 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
557 "Failed to create listen socket bound to `%s'",
558 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
560 GNUNET_NETWORK_socket_close (lsock);
564 if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock, 5))
566 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
567 "Failed to listen on socket bound to `%s'",
568 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
572 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
573 lsock, &do_accept, NULL);
575 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
576 "Proxy listens on port %u\n",
582 * The main function for gnunet-gns-proxy.
584 * @param argc number of arguments from the command line
585 * @param argv command line arguments
586 * @return 0 ok, 1 on error
589 main (int argc, char *const *argv)
591 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
593 gettext_noop ("listen on specified port"), 1,
594 &GNUNET_GETOPT_set_string, &port},
595 GNUNET_GETOPT_OPTION_END
600 GNUNET_log_setup ("gnunet-gns-proxy", "WARNING", NULL);
603 GNUNET_PROGRAM_run (argc, argv, "gnunet-gns-proxy",
604 _("GNUnet GNS proxy"),
606 &run, NULL)) ? 0 : 1;