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 <microhttpd.h>
24 #include <curl/curl.h>
25 #include "gns_proxy_proto.h"
28 #define GNUNET_GNS_PROXY_PORT 7777
30 //TODO maybe make this an api call
32 * Checks if name is in tld
34 * @param name the name to check
35 * @param tld the TLD to check for
36 * @return GNUNET_YES or GNUNET_NO
39 is_tld(const char* name, const char* tld)
43 if (strlen(name) <= strlen(tld))
48 offset = strlen(name)-strlen(tld);
49 if (strcmp (name+offset, tld) != 0)
51 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
52 "%s is not in .%s TLD\n", name, tld);
61 struct GNUNET_NETWORK_Handle *sock;
62 struct GNUNET_NETWORK_Handle *remote_sock;
66 GNUNET_SCHEDULER_TaskIdentifier rtask;
67 GNUNET_SCHEDULER_TaskIdentifier fwdrtask;
68 GNUNET_SCHEDULER_TaskIdentifier wtask;
69 GNUNET_SCHEDULER_TaskIdentifier fwdwtask;
73 unsigned int rbuf_len;
74 unsigned int wbuf_len;
78 #define BUF_WAIT_FOR_CURL 0
79 #define BUF_WAIT_FOR_MHD 1
84 struct ProxyCurlTask *prev;
85 struct ProxyCurlTask *next;
89 char buffer[CURL_MAX_WRITE_SIZE];
92 unsigned int bytes_downloaded;
93 unsigned int bytes_in_buffer;
94 int download_in_progress;
95 int download_successful;
97 struct MHD_Connection *connection;
101 unsigned long port = GNUNET_GNS_PROXY_PORT;
102 static struct GNUNET_NETWORK_Handle *lsock;
103 GNUNET_SCHEDULER_TaskIdentifier ltask;
104 GNUNET_SCHEDULER_TaskIdentifier curl_download_task;
105 static struct MHD_Daemon *httpd;
106 static GNUNET_SCHEDULER_TaskIdentifier httpd_task;
109 struct ProxyCurlTask *ctasks_head;
110 struct ProxyCurlTask *ctasks_tail;
113 con_val_iter (void *cls,
114 enum MHD_ValueKind kind,
118 char* buf = (char*)cls;
120 if (0 == strcmp ("Host", key))
137 * Process cURL download bits
139 * @param ptr buffer with data
140 * @param size size of a record
141 * @param nmemb number of records downloaded
143 * @return number of processed bytes
146 callback_download (void *ptr, size_t size, size_t nmemb, void *ctx)
148 const char *cbuf = ptr;
150 struct ProxyCurlTask *ctask = ctx;
155 ctask->bytes_downloaded += total;
162 if (total > sizeof (ctask->buffer))
164 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
165 "CURL gave us too much data to handle (%d)!\n",
170 if (ctask->buf_status == BUF_WAIT_FOR_MHD)
172 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
173 "CURL: Waiting for MHD (%s)\n", ctask->url);
174 return CURL_WRITEFUNC_PAUSE;
178 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
179 "CURL: Copying to MHD (%s, %d)\n", ctask->url, total);
180 memcpy (ctask->buffer, cbuf, total);
181 ctask->bytes_in_buffer = total;
182 ctask->buffer_ptr = ctask->buffer;
184 ctask->buf_status = BUF_WAIT_FOR_MHD;
186 //GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
187 // "cURL chunk:\n%s\n", (char*)ctask->buffer);
195 * Callback to free content
197 * @param cls content to free
200 mhd_content_free (void *cls)
202 struct ProxyCurlTask *ctask = cls;
204 if (ctask->curl != NULL)
205 curl_easy_cleanup (ctask->curl);
212 * Callback for MHD response
215 * @param pos in buffer
217 * @param max space in buffer
220 mhd_content_cb (void *cls,
225 struct ProxyCurlTask *ctask = cls;
228 if (ctask->download_successful &&
229 (ctask->buf_status == BUF_WAIT_FOR_CURL))
231 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
232 "MHD: sending response for %s\n", ctask->url);
233 ctask->download_in_progress = GNUNET_NO;
234 curl_multi_remove_handle (curl_multi, ctask->curl);
235 curl_easy_cleanup (ctask->curl);
236 return MHD_CONTENT_READER_END_OF_STREAM;
239 if (ctask->download_error &&
240 (ctask->buf_status == BUF_WAIT_FOR_CURL))
242 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
243 "MHD: sending error response\n");
244 ctask->download_in_progress = GNUNET_NO;
245 curl_multi_remove_handle (curl_multi, ctask->curl);
246 curl_easy_cleanup (ctask->curl);
247 return MHD_CONTENT_READER_END_WITH_ERROR;
250 if ( ctask->buf_status == BUF_WAIT_FOR_CURL )
255 if ( ctask->bytes_in_buffer > max )
257 GNUNET_log ( GNUNET_ERROR_TYPE_DEBUG,
258 "MHD: buffer in response too small! (%s)\n",
260 memcpy ( buf, ctask->buffer_ptr, max);
261 ctask->bytes_in_buffer -= max;
262 ctask->buffer_ptr += max;
267 GNUNET_log ( GNUNET_ERROR_TYPE_DEBUG,
268 "MHD: copying %d bytes to mhd response at offset %d\n",
269 ctask->bytes_in_buffer, pos);
271 memcpy ( buf, ctask->buffer_ptr, ctask->bytes_in_buffer );
272 copied = ctask->bytes_in_buffer;
273 ctask->buf_status = BUF_WAIT_FOR_CURL;
274 ctask->buffer_ptr = ctask->buffer;
275 curl_easy_pause (ctask->curl, CURLPAUSE_CONT);
284 * Task that is run when we are ready to receive more data
288 * @param tc task context
291 curl_task_download (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
294 * Ask cURL for the select sets and schedule download
297 curl_download_prepare ()
304 struct GNUNET_NETWORK_FDSet *grs;
305 struct GNUNET_NETWORK_FDSet *gws;
307 struct GNUNET_TIME_Relative rtime;
313 mret = curl_multi_fdset (curl_multi, &rs, &ws, &es, &max);
315 if (mret != CURLM_OK)
317 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
318 "%s failed at %s:%d: `%s'\n",
319 "curl_multi_fdset", __FILE__, __LINE__,
320 curl_multi_strerror (mret));
325 mret = curl_multi_timeout (curl_multi, &to);
326 rtime = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, to);
328 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
329 "cURL multi fds: max=%d timeout=%llu\n", max, to);
331 grs = GNUNET_NETWORK_fdset_create ();
332 gws = GNUNET_NETWORK_fdset_create ();
333 GNUNET_NETWORK_fdset_copy_native (grs, &rs, max + 1);
334 GNUNET_NETWORK_fdset_copy_native (gws, &ws, max + 1);
335 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
336 "Scheduling task cURL\n");
338 if (curl_download_task != GNUNET_SCHEDULER_NO_TASK)
339 GNUNET_SCHEDULER_cancel (curl_download_task);
342 GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
345 &curl_task_download, curl_multi);
346 GNUNET_NETWORK_fdset_destroy (gws);
347 GNUNET_NETWORK_fdset_destroy (grs);
353 * Task that is run when we are ready to receive more data
357 * @param tc task context
360 curl_task_download (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
366 struct ProxyCurlTask *ctask;
369 curl_download_task = GNUNET_SCHEDULER_NO_TASK;
371 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
373 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
374 "Shutdown requested while trying to download\n");
378 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
386 mret = curl_multi_perform (curl_multi, &running);
388 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
389 "Running curl tasks: %d\n", running);
392 for (; ctask != NULL; ctask = ctask->next)
394 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
395 "CTask: %s\n", ctask->url);
399 if (num_ctasks != running)
401 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
402 "%d ctasks, %d curl running\n", num_ctasks, running);
408 msg = curl_multi_info_read (curl_multi, &msgnum);
409 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
410 "Messages left: %d\n", msgnum);
417 if ((msg->data.result != CURLE_OK) &&
418 (msg->data.result != CURLE_GOT_NOTHING))
420 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
421 "Download curl failed");
423 for (; ctask != NULL; ctask = ctask->next)
425 if (memcmp (msg->easy_handle, ctask->curl, sizeof (CURL)) != 0)
428 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
429 "Download curl failed for task %s: %s.\n",
431 curl_easy_strerror (msg->data.result));
432 ctask->download_successful = GNUNET_NO;
433 ctask->download_error = GNUNET_YES;
434 //curl_multi_remove_handle (curl_multi, ctask->curl);
435 //curl_easy_cleanup (ctask->curl);
436 GNUNET_CONTAINER_DLL_remove (ctasks_head, ctasks_tail,
440 GNUNET_assert (ctask != NULL);
444 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
445 "cURL download completed.\n");
447 for (; ctask != NULL; ctask = ctask->next)
449 if (memcmp (msg->easy_handle, ctask->curl, sizeof (CURL)) != 0)
452 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
453 "cURL task %s found.\n", ctask->url);
454 ctask->download_successful = GNUNET_YES;
455 //curl_multi_remove_handle (curl_multi, ctask->curl);
456 //curl_easy_cleanup (ctask->curl);
457 GNUNET_CONTAINER_DLL_remove (ctasks_head, ctasks_tail,
461 GNUNET_assert (ctask != NULL);
463 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
464 "curl end %s\n", curl_easy_strerror(msg->data.result));
470 } while (msgnum > 0);
473 for (ctask=ctasks_head; ctask != NULL; ctask = ctask->next)
478 if (num_ctasks != running)
480 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
481 "%d ctasks, %d curl running\n", num_ctasks, running);
484 GNUNET_assert ( num_ctasks == running );
488 } while (mret == CURLM_CALL_MULTI_PERFORM);
491 if (mret != CURLM_OK)
493 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "%s failed at %s:%d: `%s'\n",
494 "curl_multi_perform", __FILE__, __LINE__,
495 curl_multi_strerror (mret));
497 curl_download_prepare();
502 * Main MHD callback for handling requests.
505 * @param con MHD connection handle
506 * @param meth the HTTP method used ("GET", "PUT", etc.)
507 * @param ver the HTTP version string (i.e. "HTTP/1.1")
508 * @param upload_data the data being uploaded (excluding HEADERS,
509 * for a POST that fits into memory and that is encoded
510 * with a supported encoding, the POST data will NOT be
511 * given in upload_data and is instead available as
512 * part of MHD_get_connection_values; very large POST
513 * data *will* be made available incrementally in
515 * @param upload_data_size set initially to the size of the
516 * upload_data provided; the method must update this
517 * value to the number of bytes NOT processed;
518 * @param con_cls pointer to location where we store the 'struct Request'
519 * @return MHD_YES if the connection was handled successfully,
520 * MHD_NO if the socket must be closed due to a serious
521 * error while handling the request
524 create_response (void *cls,
525 struct MHD_Connection *con,
529 const char *upload_data,
530 size_t *upload_data_size,
534 const char* page = "<html><head><title>gnoxy</title>"\
535 "</head><body>cURL fail</body></html>";
536 struct MHD_Response *response;
542 struct ProxyCurlTask *ctask;
544 if (0 != strcmp (meth, "GET"))
546 if (&dummy != *con_cls)
552 if (0 != *upload_data_size)
557 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
560 MHD_get_connection_values (con,
562 &con_val_iter, host);
566 ctask = GNUNET_malloc (sizeof (struct ProxyCurlTask));
567 ctask->curl = curl_easy_init();
569 if (curl_multi == NULL)
570 curl_multi = curl_multi_init ();
572 if ((ctask->curl == NULL) || (curl_multi == NULL))
574 response = MHD_create_response_from_buffer (strlen (page),
576 MHD_RESPMEM_PERSISTENT);
577 ret = MHD_queue_response (con,
580 MHD_destroy_response (response);
585 ctask->download_in_progress = GNUNET_YES;
586 ctask->download_successful = GNUNET_NO;
587 ctask->bytes_downloaded = 0;
588 ctask->connection = con;
589 ctask->buf_status = BUF_WAIT_FOR_CURL;
590 ctask->bytes_in_buffer = 0;
592 curl_easy_setopt (ctask->curl, CURLOPT_WRITEFUNCTION, &callback_download);
593 curl_easy_setopt (ctask->curl, CURLOPT_WRITEDATA, ctask);
594 curl_easy_setopt (ctask->curl, CURLOPT_FOLLOWLOCATION, 1);
595 curl_easy_setopt (ctask->curl, CURLOPT_MAXREDIRS, 4);
596 /* no need to abort if the above failed */
597 sprintf (curlurl, "http://%s%s", host, url);
598 strcpy (ctask->url, curlurl);
599 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
600 "Adding new curl task for %s\n", curlurl);
602 curl_easy_setopt (ctask->curl, CURLOPT_URL, curlurl);
603 curl_easy_setopt (ctask->curl, CURLOPT_FAILONERROR, 1);
604 curl_easy_setopt (ctask->curl, CURLOPT_CONNECTTIMEOUT, 600L);
605 curl_easy_setopt (ctask->curl, CURLOPT_TIMEOUT, 600L);
607 mret = curl_multi_add_handle (curl_multi, ctask->curl);
609 if (mret != CURLM_OK)
611 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
612 "%s failed at %s:%d: `%s'\n",
613 "curl_multi_add_handle", __FILE__, __LINE__,
614 curl_multi_strerror (mret));
615 response = MHD_create_response_from_buffer (strlen (page),
617 MHD_RESPMEM_PERSISTENT);
618 ret = MHD_queue_response (con,
621 MHD_destroy_response (response);
623 curl_easy_cleanup (ctask->curl);
628 GNUNET_CONTAINER_DLL_insert (ctasks_head, ctasks_tail, ctask);
630 curl_download_prepare ();
632 response = MHD_create_response_from_callback (-1, -1,
637 ret = MHD_queue_response (con, MHD_HTTP_OK, response);
639 //MHD_destroy_response (response);
645 * Task run whenever HTTP server operations are pending.
648 * @param tc sched context
652 const struct GNUNET_SCHEDULER_TaskContext *tc);
664 struct GNUNET_NETWORK_FDSet *wrs;
665 struct GNUNET_NETWORK_FDSet *wws;
666 struct GNUNET_NETWORK_FDSet *wes;
669 unsigned MHD_LONG_LONG timeout;
670 struct GNUNET_TIME_Relative tv;
675 wrs = GNUNET_NETWORK_fdset_create ();
676 wes = GNUNET_NETWORK_fdset_create ();
677 wws = GNUNET_NETWORK_fdset_create ();
679 GNUNET_assert (MHD_YES == MHD_get_fdset (httpd, &rs, &ws, &es, &max));
682 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
683 "MHD fds: max=%d\n", max);
685 haveto = MHD_get_timeout (httpd, &timeout);
687 if (haveto == MHD_YES)
688 tv.rel_value = (uint64_t) timeout;
690 tv = GNUNET_TIME_UNIT_FOREVER_REL;
691 GNUNET_NETWORK_fdset_copy_native (wrs, &rs, max + 1);
692 GNUNET_NETWORK_fdset_copy_native (wws, &ws, max + 1);
693 GNUNET_NETWORK_fdset_copy_native (wes, &es, max + 1);
695 if (httpd_task != GNUNET_SCHEDULER_NO_TASK)
696 GNUNET_SCHEDULER_cancel (httpd_task);
698 GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_HIGH,
701 GNUNET_NETWORK_fdset_destroy (wrs);
702 GNUNET_NETWORK_fdset_destroy (wws);
703 GNUNET_NETWORK_fdset_destroy (wes);
707 * Task run whenever HTTP server operations are pending.
710 * @param tc sched context
714 const struct GNUNET_SCHEDULER_TaskContext *tc)
716 httpd_task = GNUNET_SCHEDULER_NO_TASK;
717 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
724 * Read data from socket
726 * @param cls the closure
727 * @param tc scheduler context
730 do_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
733 * Read from remote end
736 * @param tc scheduler context
739 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
742 * Write data to remote socket
744 * @param cls the closure
745 * @param tc scheduler context
748 do_write_remote (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
750 struct Socks5Request *s5r = cls;
753 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
755 if ((NULL != tc->read_ready) &&
756 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->remote_sock)) &&
757 ((len = GNUNET_NETWORK_socket_send (s5r->remote_sock, s5r->rbuf,
760 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
761 "Successfully sent %d bytes to remote socket\n",
766 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write remote");
768 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
769 GNUNET_SCHEDULER_cancel (s5r->rtask);
770 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
771 GNUNET_SCHEDULER_cancel (s5r->wtask);
772 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
773 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
774 GNUNET_NETWORK_socket_close (s5r->remote_sock);
775 GNUNET_NETWORK_socket_close (s5r->sock);
781 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
788 * Write data to socket
790 * @param cls the closure
791 * @param tc scheduler context
794 do_write (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
796 struct Socks5Request *s5r = cls;
799 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
801 if ((NULL != tc->read_ready) &&
802 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->sock)) &&
803 ((len = GNUNET_NETWORK_socket_send (s5r->sock, s5r->wbuf,
806 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
807 "Successfully sent %d bytes to socket\n",
813 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write");
815 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
816 GNUNET_SCHEDULER_cancel (s5r->rtask);
817 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
818 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
819 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
820 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
821 GNUNET_NETWORK_socket_close (s5r->remote_sock);
822 GNUNET_NETWORK_socket_close (s5r->sock);
827 if ((s5r->state == SOCKS5_DATA_TRANSFER) &&
828 (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK))
830 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
832 &do_read_remote, s5r);
836 * Read from remote end
839 * @param tc scheduler context
842 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
844 struct Socks5Request *s5r = cls;
846 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
849 if ((NULL != tc->write_ready) &&
850 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->remote_sock)) &&
851 (s5r->wbuf_len = GNUNET_NETWORK_socket_recv (s5r->remote_sock, s5r->wbuf,
852 sizeof (s5r->wbuf))))
854 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
855 "Successfully read %d bytes from remote socket\n",
860 if (s5r->wbuf_len == 0)
861 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
862 "0 bytes received from remote... graceful shutdown!\n");
863 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
864 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
865 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
866 GNUNET_SCHEDULER_cancel (s5r->rtask);
868 GNUNET_NETWORK_socket_close (s5r->remote_sock);
869 s5r->remote_sock = NULL;
870 GNUNET_NETWORK_socket_close (s5r->sock);
876 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
884 add_handle_to_mhd (struct GNUNET_NETWORK_Handle *h)
887 struct sockaddr *addr;
890 fd = GNUNET_NETWORK_get_fd (h);
891 addr = GNUNET_NETWORK_get_addr (h);
892 len = GNUNET_NETWORK_get_addrlen (h);
894 return MHD_add_connection (httpd, fd, addr, len);
898 * Read data from incoming connection
900 * @param cls the closure
901 * @param tc the scheduler context
904 do_read (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
906 struct Socks5Request *s5r = cls;
907 struct socks5_client_hello *c_hello;
908 struct socks5_server_hello *s_hello;
909 struct socks5_client_request *c_req;
910 struct socks5_server_response *s_resp;
915 struct hostent *phost;
917 struct sockaddr_in remote_addr;
918 struct in_addr *r_sin_addr;
920 s5r->rtask = GNUNET_SCHEDULER_NO_TASK;
922 if ((NULL != tc->write_ready) &&
923 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->sock)) &&
924 (s5r->rbuf_len = GNUNET_NETWORK_socket_recv (s5r->sock, s5r->rbuf,
925 sizeof (s5r->rbuf))))
927 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
928 "Successfully read %d bytes from socket\n",
933 if (s5r->rbuf_len != 0)
934 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "read");
936 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disco!\n");
938 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
939 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
940 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
941 GNUNET_SCHEDULER_cancel (s5r->wtask);
942 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
943 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
944 GNUNET_NETWORK_socket_close (s5r->remote_sock);
945 GNUNET_NETWORK_socket_close (s5r->sock);
950 if (s5r->state == SOCKS5_INIT)
952 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
954 c_hello = (struct socks5_client_hello*)&s5r->rbuf;
956 GNUNET_assert (c_hello->version == SOCKS_VERSION_5);
958 s_hello = (struct socks5_server_hello*)&s5r->wbuf;
959 s5r->wbuf_len = sizeof( struct socks5_server_hello );
961 s_hello->version = c_hello->version;
962 s_hello->auth_method = SOCKS_AUTH_NONE;
964 /* Write response to client */
965 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
969 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
973 s5r->state = SOCKS5_REQUEST;
977 if (s5r->state == SOCKS5_REQUEST)
979 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
980 "Processing SOCKS5 request\n");
981 c_req = (struct socks5_client_request*)&s5r->rbuf;
982 s_resp = (struct socks5_server_response*)&s5r->wbuf;
983 //Only 10byte for ipv4 response!
984 s5r->wbuf_len = 10;//sizeof (struct socks5_server_response);
986 GNUNET_assert (c_req->addr_type == 3);
988 dom_len = *((uint8_t*)(&(c_req->addr_type) + 1));
989 memset(domain, 0, sizeof(domain));
990 strncpy(domain, (char*)(&(c_req->addr_type) + 2), dom_len);
991 req_port = *((uint16_t*)(&(c_req->addr_type) + 2 + dom_len));
993 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
994 "Requested connection is %s:%d\n",
998 if (is_tld (domain, GNUNET_GNS_TLD) ||
999 is_tld (domain, GNUNET_GNS_TLD_ZKEY))
1001 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1002 "Requested connection is gnunet tld\n",
1007 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1008 _("Failed to start HTTP server\n"));
1009 s_resp->version = 0x05;
1010 s_resp->reply = 0x01;
1012 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1016 //TODO! close socket after the write! schedule task
1017 //GNUNET_NETWORK_socket_close (s5r->sock);
1022 if (MHD_YES == add_handle_to_mhd ( s5r->sock ))
1023 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1024 "Sucessfully added client to MHD!\n");
1025 s_resp->version = 0x05;
1026 s_resp->reply = 0x00;
1027 s_resp->reserved = 0x00;
1028 s_resp->addr_type = 0x01;
1031 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1035 //GNUNET_free ( s5r );
1036 //FIXME complete socks resp!
1041 phost = (struct hostent*)gethostbyname (domain);
1044 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1045 "Resolve %s error!\n", domain );
1046 s_resp->version = 0x05;
1047 s_resp->reply = 0x01;
1049 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1053 //TODO! close socket after the write! schedule task
1054 //GNUNET_NETWORK_socket_close (s5r->sock);
1059 s5r->remote_sock = GNUNET_NETWORK_socket_create (AF_INET,
1062 r_sin_addr = (struct in_addr*)(phost->h_addr);
1063 remote_ip = r_sin_addr->s_addr;
1064 memset(&remote_addr, 0, sizeof(remote_addr));
1065 remote_addr.sin_family = AF_INET;
1066 #if HAVE_SOCKADDR_IN_SIN_LEN
1067 remote_addr.sin_len = sizeof (remote_addr);
1069 remote_addr.sin_addr.s_addr = remote_ip;
1070 remote_addr.sin_port = req_port;
1072 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1073 "target server: %s:%u\n", inet_ntoa(remote_addr.sin_addr),
1077 GNUNET_NETWORK_socket_connect ( s5r->remote_sock,
1078 (const struct sockaddr*)&remote_addr,
1079 sizeof (remote_addr)))
1080 && (errno != EINPROGRESS))
1082 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "connect");
1083 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1084 "socket request error...\n");
1085 s_resp->version = 0x05;
1086 s_resp->reply = 0x01;
1088 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1095 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1096 "new remote connection\n");
1098 s_resp->version = 0x05;
1099 s_resp->reply = 0x00;
1100 s_resp->reserved = 0x00;
1101 s_resp->addr_type = 0x01;
1103 s5r->state = SOCKS5_DATA_TRANSFER;
1106 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1110 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1118 if (s5r->state == SOCKS5_DATA_TRANSFER)
1120 if ((s5r->remote_sock == NULL) || (s5r->rbuf_len == 0))
1122 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1123 "Closing connection to client\n");
1124 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
1125 GNUNET_SCHEDULER_cancel (s5r->rtask);
1126 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
1127 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
1128 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
1129 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
1130 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
1131 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
1133 if (s5r->remote_sock != NULL)
1134 GNUNET_NETWORK_socket_close (s5r->remote_sock);
1135 GNUNET_NETWORK_socket_close (s5r->sock);
1140 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1141 "forwarding %d bytes from client\n", s5r->rbuf_len);
1144 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1146 &do_write_remote, s5r);
1148 if (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK)
1151 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1153 &do_read_remote, s5r);
1159 //GNUNET_CONTAINER_DLL_remove (s5conns.head, s5conns.tail, s5r);
1164 * Accept new incoming connections
1166 * @param cls the closure
1167 * @param tc the scheduler context
1170 do_accept (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1172 struct GNUNET_NETWORK_Handle *s;
1173 struct Socks5Request *s5r;
1175 ltask = GNUNET_SCHEDULER_NO_TASK;
1176 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1179 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1183 s = GNUNET_NETWORK_socket_accept (lsock, NULL, NULL);
1187 GNUNET_log_strerror (GNUNET_ERROR_TYPE_INFO, "accept");
1191 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1192 "Got an inbound connection, waiting for data\n");
1194 s5r = GNUNET_malloc (sizeof (struct Socks5Request));
1196 s5r->state = SOCKS5_INIT;
1197 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
1198 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
1199 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
1200 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1203 //GNUNET_CONTAINER_DLL_insert (s5conns.head, s5conns.tail, s5r);
1207 * Task run on shutdown
1209 * @param cls closure
1210 * @param tc task context
1213 do_shutdown (void *cls,
1214 const struct GNUNET_SCHEDULER_TaskContext *tc)
1216 if (GNUNET_SCHEDULER_NO_TASK != httpd_task)
1218 GNUNET_SCHEDULER_cancel (httpd_task);
1219 httpd_task = GNUNET_SCHEDULER_NO_TASK;
1222 if (GNUNET_SCHEDULER_NO_TASK != curl_download_task)
1224 GNUNET_SCHEDULER_cancel (curl_download_task);
1225 curl_download_task = GNUNET_SCHEDULER_NO_TASK;
1230 MHD_stop_daemon (httpd);
1236 * Main function that will be run
1238 * @param cls closure
1239 * @param args remaining command-line arguments
1240 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
1241 * @param cfg configuration
1244 run (void *cls, char *const *args, const char *cfgfile,
1245 const struct GNUNET_CONFIGURATION_Handle *cfg)
1247 struct sockaddr_in sa;
1249 memset (&sa, 0, sizeof (sa));
1250 sa.sin_family = AF_INET;
1251 sa.sin_port = htons (port);
1252 #if HAVE_SOCKADDR_IN_SIN_LEN
1253 sa.sin_len = sizeof (sa);
1256 lsock = GNUNET_NETWORK_socket_create (AF_INET,
1260 if ((NULL == lsock) ||
1262 GNUNET_NETWORK_socket_bind (lsock, (const struct sockaddr *) &sa,
1265 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1266 "Failed to create listen socket bound to `%s'",
1267 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
1269 GNUNET_NETWORK_socket_close (lsock);
1273 if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock, 5))
1275 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1276 "Failed to listen on socket bound to `%s'",
1277 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
1281 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1282 lsock, &do_accept, NULL);
1287 if (0 != curl_global_init (CURL_GLOBAL_WIN32))
1289 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1290 "cURL global init failed!\n");
1294 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1295 "Proxy listens on port %u\n",
1298 httpd = MHD_start_daemon (MHD_USE_DEBUG, 4444,
1300 &create_response, NULL,
1301 MHD_OPTION_CONNECTION_LIMIT, (unsigned int) 128,
1302 MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
1303 MHD_OPTION_NOTIFY_COMPLETED,
1308 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
1309 &do_shutdown, NULL);
1314 * The main function for gnunet-gns-proxy.
1316 * @param argc number of arguments from the command line
1317 * @param argv command line arguments
1318 * @return 0 ok, 1 on error
1321 main (int argc, char *const *argv)
1323 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
1325 gettext_noop ("listen on specified port"), 1,
1326 &GNUNET_GETOPT_set_string, &port},
1327 GNUNET_GETOPT_OPTION_END
1332 GNUNET_log_setup ("gnunet-gns-proxy", "WARNING", NULL);
1335 GNUNET_PROGRAM_run (argc, argv, "gnunet-gns-proxy",
1336 _("GNUnet GNS proxy"),
1338 &run, NULL)) ? 0 : 1;