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];
91 unsigned int bytes_downloaded;
92 unsigned int bytes_in_buffer;
93 int download_in_progress;
94 int download_successful;
96 struct MHD_Connection *connection;
100 unsigned long port = GNUNET_GNS_PROXY_PORT;
101 static struct GNUNET_NETWORK_Handle *lsock;
102 GNUNET_SCHEDULER_TaskIdentifier ltask;
103 GNUNET_SCHEDULER_TaskIdentifier curl_download_task;
104 static struct MHD_Daemon *httpd;
105 static GNUNET_SCHEDULER_TaskIdentifier httpd_task;
108 struct ProxyCurlTask *ctasks_head;
109 struct ProxyCurlTask *ctasks_tail;
112 con_val_iter (void *cls,
113 enum MHD_ValueKind kind,
117 char* buf = (char*)cls;
119 if (0 == strcmp ("Host", key))
136 * Process cURL download bits
138 * @param ptr buffer with data
139 * @param size size of a record
140 * @param nmemb number of records downloaded
142 * @return number of processed bytes
145 callback_download (void *ptr, size_t size, size_t nmemb, void *ctx)
147 const char *cbuf = ptr;
149 struct ProxyCurlTask *ctask = ctx;
152 ctask->bytes_downloaded += total;
159 if (total > sizeof (ctask->buffer))
161 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
162 "cURL gave us too much data to handle (%d)!\n",
167 if (ctask->buf_status == BUF_WAIT_FOR_MHD)
169 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
170 "Waiting for MHD\n");
171 return CURL_WRITEFUNC_PAUSE;
174 memcpy (ctask->buffer, cbuf, total);
175 ctask->bytes_in_buffer = total;
177 ctask->buf_status = BUF_WAIT_FOR_MHD;
179 //GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
180 // "cURL chunk:\n%s\n", (char*)ctask->buffer);
188 * Callback to free content
190 * @param cls content to free
193 mhd_content_free (void *cls)
195 struct ProxyCurlTask *ctask = cls;
197 if (ctask->curl != NULL)
198 curl_easy_cleanup (ctask->curl);
205 * Callback for MHD response
208 * @param pos in buffer
210 * @param max space in buffer
213 mhd_content_cb (void *cls,
218 struct ProxyCurlTask *ctask = cls;
220 if (ctask->download_successful &&
221 (ctask->buf_status == BUF_WAIT_FOR_CURL))
223 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
224 "MHD: sending response\n");
225 ctask->download_in_progress = GNUNET_NO;
226 curl_multi_remove_handle (curl_multi, ctask->curl);
227 curl_easy_cleanup (ctask->curl);
228 return MHD_CONTENT_READER_END_OF_STREAM;
231 if (ctask->download_error &&
232 (ctask->buf_status == BUF_WAIT_FOR_CURL))
234 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
235 "MHD: sending error response\n");
236 ctask->download_in_progress = GNUNET_NO;
237 curl_multi_remove_handle (curl_multi, ctask->curl);
238 curl_easy_cleanup (ctask->curl);
239 return MHD_CONTENT_READER_END_WITH_ERROR;
242 if ( ctask->buf_status == BUF_WAIT_FOR_CURL )
247 if ( ctask->bytes_in_buffer > max )
249 GNUNET_log ( GNUNET_ERROR_TYPE_ERROR,
250 "MHD: buffer in response too small!\n");
251 return MHD_CONTENT_READER_END_WITH_ERROR;
254 if ( 0 != ctask->bytes_in_buffer )
256 GNUNET_log ( GNUNET_ERROR_TYPE_DEBUG,
257 "MHD: copying %d bytes to mhd response at offset %d\n",
258 ctask->bytes_in_buffer, pos);
259 memcpy ( buf, ctask->buffer, ctask->bytes_in_buffer );
262 ctask->buf_status = BUF_WAIT_FOR_CURL;
263 curl_easy_pause (ctask->curl, CURLPAUSE_CONT);
265 return ctask->bytes_in_buffer;
271 * Task that is run when we are ready to receive more data
275 * @param tc task context
278 curl_task_download (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
281 * Ask cURL for the select sets and schedule download
284 curl_download_prepare ()
291 struct GNUNET_NETWORK_FDSet *grs;
292 struct GNUNET_NETWORK_FDSet *gws;
294 struct GNUNET_TIME_Relative rtime;
300 mret = curl_multi_fdset (curl_multi, &rs, &ws, &es, &max);
302 if (mret != CURLM_OK)
304 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
305 "%s failed at %s:%d: `%s'\n",
306 "curl_multi_fdset", __FILE__, __LINE__,
307 curl_multi_strerror (mret));
312 mret = curl_multi_timeout (curl_multi, &to);
313 rtime = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, to);
315 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
316 "cURL multi fds: max=%d\n", max);
318 grs = GNUNET_NETWORK_fdset_create ();
319 gws = GNUNET_NETWORK_fdset_create ();
320 GNUNET_NETWORK_fdset_copy_native (grs, &rs, max + 1);
321 GNUNET_NETWORK_fdset_copy_native (gws, &ws, max + 1);
322 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
323 "Scheduling task cURL\n");
325 if (curl_download_task != GNUNET_SCHEDULER_NO_TASK)
326 GNUNET_SCHEDULER_cancel (curl_download_task);
329 GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
332 &curl_task_download, curl_multi);
333 GNUNET_NETWORK_fdset_destroy (gws);
334 GNUNET_NETWORK_fdset_destroy (grs);
340 * Task that is run when we are ready to receive more data
344 * @param tc task context
347 curl_task_download (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
353 struct ProxyCurlTask *ctask;
356 curl_download_task = GNUNET_SCHEDULER_NO_TASK;
358 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
360 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
361 "Shutdown requested while trying to download\n");
365 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
373 mret = curl_multi_perform (curl_multi, &running);
375 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
376 "Running curl tasks: %d\n", running);
379 for (; ctask != NULL; ctask = ctask->next)
381 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
382 "CTask: %s\n", ctask->url);
386 if (num_ctasks != running)
388 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
389 "%d ctasks, %d curl running\n", num_ctasks, running);
395 msg = curl_multi_info_read (curl_multi, &msgnum);
396 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
397 "Messages left: %d\n", msgnum);
404 if ((msg->data.result != CURLE_OK) &&
405 (msg->data.result != CURLE_GOT_NOTHING))
407 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
408 "Download curl failed");
410 for (; ctask != NULL; ctask = ctask->next)
412 if (memcmp (msg->easy_handle, ctask->curl, sizeof (CURL)) != 0)
415 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
416 "Download curl failed for task %s: %s.\n",
418 curl_easy_strerror (msg->data.result));
419 ctask->download_successful = GNUNET_NO;
420 ctask->download_error = GNUNET_YES;
421 //curl_multi_remove_handle (curl_multi, ctask->curl);
422 //curl_easy_cleanup (ctask->curl);
423 GNUNET_CONTAINER_DLL_remove (ctasks_head, ctasks_tail,
427 GNUNET_assert (ctask != NULL);
431 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
432 "cURL download completed.\n");
434 for (; ctask != NULL; ctask = ctask->next)
436 if (memcmp (msg->easy_handle, ctask->curl, sizeof (CURL)) != 0)
439 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
440 "cURL task %s found.\n", ctask->url);
441 ctask->download_successful = GNUNET_YES;
442 //curl_multi_remove_handle (curl_multi, ctask->curl);
443 //curl_easy_cleanup (ctask->curl);
444 GNUNET_CONTAINER_DLL_remove (ctasks_head, ctasks_tail,
448 GNUNET_assert (ctask != NULL);
450 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
451 "curl end %s\n", curl_easy_strerror(msg->data.result));
458 } while (msgnum > 0);
461 for (ctask=ctasks_head; ctask != NULL; ctask = ctask->next)
466 if (num_ctasks != running)
468 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
469 "%d ctasks, %d curl running\n", num_ctasks, running);
472 GNUNET_assert ( num_ctasks == running );
476 } while (mret == CURLM_CALL_MULTI_PERFORM);
479 if (mret != CURLM_OK)
481 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "%s failed at %s:%d: `%s'\n",
482 "curl_multi_perform", __FILE__, __LINE__,
483 curl_multi_strerror (mret));
485 curl_download_prepare();
490 * Main MHD callback for handling requests.
493 * @param con MHD connection handle
494 * @param meth the HTTP method used ("GET", "PUT", etc.)
495 * @param ver the HTTP version string (i.e. "HTTP/1.1")
496 * @param upload_data the data being uploaded (excluding HEADERS,
497 * for a POST that fits into memory and that is encoded
498 * with a supported encoding, the POST data will NOT be
499 * given in upload_data and is instead available as
500 * part of MHD_get_connection_values; very large POST
501 * data *will* be made available incrementally in
503 * @param upload_data_size set initially to the size of the
504 * upload_data provided; the method must update this
505 * value to the number of bytes NOT processed;
506 * @param con_cls pointer to location where we store the 'struct Request'
507 * @return MHD_YES if the connection was handled successfully,
508 * MHD_NO if the socket must be closed due to a serious
509 * error while handling the request
512 create_response (void *cls,
513 struct MHD_Connection *con,
517 const char *upload_data,
518 size_t *upload_data_size,
522 const char* page = "<html><head><title>gnoxy</title>"\
523 "</head><body>cURL fail</body></html>";
524 struct MHD_Response *response;
530 struct ProxyCurlTask *ctask;
532 if (0 != strcmp (meth, "GET"))
534 if (&dummy != *con_cls)
540 if (0 != *upload_data_size)
545 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
548 MHD_get_connection_values (con,
550 &con_val_iter, host);
554 ctask = GNUNET_malloc (sizeof (struct ProxyCurlTask));
555 ctask->curl = curl_easy_init();
557 if (curl_multi == NULL)
558 curl_multi = curl_multi_init ();
560 if ((ctask->curl == NULL) || (curl_multi == NULL))
562 response = MHD_create_response_from_buffer (strlen (page),
564 MHD_RESPMEM_PERSISTENT);
565 ret = MHD_queue_response (con,
568 MHD_destroy_response (response);
573 ctask->download_in_progress = GNUNET_YES;
574 ctask->download_successful = GNUNET_NO;
575 ctask->bytes_downloaded = 0;
576 ctask->connection = con;
577 ctask->buf_status = BUF_WAIT_FOR_CURL;
578 ctask->bytes_in_buffer = 0;
580 curl_easy_setopt (ctask->curl, CURLOPT_WRITEFUNCTION, &callback_download);
581 curl_easy_setopt (ctask->curl, CURLOPT_WRITEDATA, ctask);
582 curl_easy_setopt (ctask->curl, CURLOPT_FOLLOWLOCATION, 1);
583 curl_easy_setopt (ctask->curl, CURLOPT_MAXREDIRS, 4);
584 /* no need to abort if the above failed */
585 sprintf (curlurl, "http://%s%s", host, url);
586 strcpy (ctask->url, curlurl);
587 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
588 "Adding new curl task for %s\n", curlurl);
590 curl_easy_setopt (ctask->curl, CURLOPT_URL, curlurl);
591 curl_easy_setopt (ctask->curl, CURLOPT_FAILONERROR, 1);
592 curl_easy_setopt (ctask->curl, CURLOPT_CONNECTTIMEOUT, 600L);
593 curl_easy_setopt (ctask->curl, CURLOPT_TIMEOUT, 600L);
595 mret = curl_multi_add_handle (curl_multi, ctask->curl);
597 if (mret != CURLM_OK)
599 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
600 "%s failed at %s:%d: `%s'\n",
601 "curl_multi_add_handle", __FILE__, __LINE__,
602 curl_multi_strerror (mret));
603 response = MHD_create_response_from_buffer (strlen (page),
605 MHD_RESPMEM_PERSISTENT);
606 ret = MHD_queue_response (con,
609 MHD_destroy_response (response);
611 curl_easy_cleanup (ctask->curl);
616 GNUNET_CONTAINER_DLL_insert (ctasks_head, ctasks_tail, ctask);
618 curl_download_prepare ();
620 response = MHD_create_response_from_callback (-1, -1,
625 ret = MHD_queue_response (con, MHD_HTTP_OK, response);
627 //MHD_destroy_response (response);
633 * Task run whenever HTTP server operations are pending.
636 * @param tc sched context
640 const struct GNUNET_SCHEDULER_TaskContext *tc);
652 struct GNUNET_NETWORK_FDSet *wrs;
653 struct GNUNET_NETWORK_FDSet *wws;
654 struct GNUNET_NETWORK_FDSet *wes;
657 unsigned MHD_LONG_LONG timeout;
658 struct GNUNET_TIME_Relative tv;
663 wrs = GNUNET_NETWORK_fdset_create ();
664 wes = GNUNET_NETWORK_fdset_create ();
665 wws = GNUNET_NETWORK_fdset_create ();
667 GNUNET_assert (MHD_YES == MHD_get_fdset (httpd, &rs, &ws, &es, &max));
670 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
671 "MHD fds: max=%d\n", max);
673 haveto = MHD_get_timeout (httpd, &timeout);
675 if (haveto == MHD_YES)
676 tv.rel_value = (uint64_t) timeout;
678 tv = GNUNET_TIME_UNIT_FOREVER_REL;
679 GNUNET_NETWORK_fdset_copy_native (wrs, &rs, max + 1);
680 GNUNET_NETWORK_fdset_copy_native (wws, &ws, max + 1);
681 GNUNET_NETWORK_fdset_copy_native (wes, &es, max + 1);
683 if (httpd_task != GNUNET_SCHEDULER_NO_TASK)
684 GNUNET_SCHEDULER_cancel (httpd_task);
686 GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_HIGH,
689 GNUNET_NETWORK_fdset_destroy (wrs);
690 GNUNET_NETWORK_fdset_destroy (wws);
691 GNUNET_NETWORK_fdset_destroy (wes);
695 * Task run whenever HTTP server operations are pending.
698 * @param tc sched context
702 const struct GNUNET_SCHEDULER_TaskContext *tc)
704 httpd_task = GNUNET_SCHEDULER_NO_TASK;
705 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
712 * Read data from socket
714 * @param cls the closure
715 * @param tc scheduler context
718 do_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
721 * Read from remote end
724 * @param tc scheduler context
727 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
730 * Write data to remote socket
732 * @param cls the closure
733 * @param tc scheduler context
736 do_write_remote (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
738 struct Socks5Request *s5r = cls;
741 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
743 if ((NULL != tc->read_ready) &&
744 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->remote_sock)) &&
745 ((len = GNUNET_NETWORK_socket_send (s5r->remote_sock, s5r->rbuf,
748 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
749 "Successfully sent %d bytes to remote socket\n",
754 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write remote");
756 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
757 GNUNET_SCHEDULER_cancel (s5r->rtask);
758 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
759 GNUNET_SCHEDULER_cancel (s5r->wtask);
760 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
761 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
762 GNUNET_NETWORK_socket_close (s5r->remote_sock);
763 GNUNET_NETWORK_socket_close (s5r->sock);
769 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
776 * Write data to socket
778 * @param cls the closure
779 * @param tc scheduler context
782 do_write (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
784 struct Socks5Request *s5r = cls;
787 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
789 if ((NULL != tc->read_ready) &&
790 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->sock)) &&
791 ((len = GNUNET_NETWORK_socket_send (s5r->sock, s5r->wbuf,
794 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
795 "Successfully sent %d bytes to socket\n",
801 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write");
803 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
804 GNUNET_SCHEDULER_cancel (s5r->rtask);
805 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
806 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
807 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
808 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
809 GNUNET_NETWORK_socket_close (s5r->remote_sock);
810 GNUNET_NETWORK_socket_close (s5r->sock);
815 if ((s5r->state == SOCKS5_DATA_TRANSFER) &&
816 (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK))
818 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
820 &do_read_remote, s5r);
824 * Read from remote end
827 * @param tc scheduler context
830 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
832 struct Socks5Request *s5r = cls;
834 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
837 if ((NULL != tc->write_ready) &&
838 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->remote_sock)) &&
839 (s5r->wbuf_len = GNUNET_NETWORK_socket_recv (s5r->remote_sock, s5r->wbuf,
840 sizeof (s5r->wbuf))))
842 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
843 "Successfully read %d bytes from remote socket\n",
848 if (s5r->wbuf_len == 0)
849 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
850 "0 bytes received from remote... graceful shutdown!\n");
851 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
852 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
853 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
854 GNUNET_SCHEDULER_cancel (s5r->rtask);
856 GNUNET_NETWORK_socket_close (s5r->remote_sock);
857 s5r->remote_sock = NULL;
858 GNUNET_NETWORK_socket_close (s5r->sock);
864 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
872 add_handle_to_mhd (struct GNUNET_NETWORK_Handle *h)
875 struct sockaddr *addr;
878 fd = GNUNET_NETWORK_get_fd (h);
879 addr = GNUNET_NETWORK_get_addr (h);
880 len = GNUNET_NETWORK_get_addrlen (h);
882 return MHD_add_connection (httpd, fd, addr, len);
886 * Read data from incoming connection
888 * @param cls the closure
889 * @param tc the scheduler context
892 do_read (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
894 struct Socks5Request *s5r = cls;
895 struct socks5_client_hello *c_hello;
896 struct socks5_server_hello *s_hello;
897 struct socks5_client_request *c_req;
898 struct socks5_server_response *s_resp;
903 struct hostent *phost;
905 struct sockaddr_in remote_addr;
906 struct in_addr *r_sin_addr;
908 s5r->rtask = GNUNET_SCHEDULER_NO_TASK;
910 if ((NULL != tc->write_ready) &&
911 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->sock)) &&
912 (s5r->rbuf_len = GNUNET_NETWORK_socket_recv (s5r->sock, s5r->rbuf,
913 sizeof (s5r->rbuf))))
915 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
916 "Successfully read %d bytes from socket\n",
921 if (s5r->rbuf_len != 0)
922 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "read");
924 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disco!\n");
926 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
927 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
928 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
929 GNUNET_SCHEDULER_cancel (s5r->wtask);
930 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
931 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
932 GNUNET_NETWORK_socket_close (s5r->remote_sock);
933 GNUNET_NETWORK_socket_close (s5r->sock);
938 if (s5r->state == SOCKS5_INIT)
940 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
942 c_hello = (struct socks5_client_hello*)&s5r->rbuf;
944 GNUNET_assert (c_hello->version == SOCKS_VERSION_5);
946 s_hello = (struct socks5_server_hello*)&s5r->wbuf;
947 s5r->wbuf_len = sizeof( struct socks5_server_hello );
949 s_hello->version = c_hello->version;
950 s_hello->auth_method = SOCKS_AUTH_NONE;
952 /* Write response to client */
953 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
957 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
961 s5r->state = SOCKS5_REQUEST;
965 if (s5r->state == SOCKS5_REQUEST)
967 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
968 "Processing SOCKS5 request\n");
969 c_req = (struct socks5_client_request*)&s5r->rbuf;
970 s_resp = (struct socks5_server_response*)&s5r->wbuf;
971 //Only 10byte for ipv4 response!
972 s5r->wbuf_len = 10;//sizeof (struct socks5_server_response);
974 GNUNET_assert (c_req->addr_type == 3);
976 dom_len = *((uint8_t*)(&(c_req->addr_type) + 1));
977 memset(domain, 0, sizeof(domain));
978 strncpy(domain, (char*)(&(c_req->addr_type) + 2), dom_len);
979 req_port = *((uint16_t*)(&(c_req->addr_type) + 2 + dom_len));
981 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
982 "Requested connection is %s:%d\n",
986 if (is_tld (domain, GNUNET_GNS_TLD) ||
987 is_tld (domain, GNUNET_GNS_TLD_ZKEY))
989 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
990 "Requested connection is gnunet tld\n",
995 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
996 _("Failed to start HTTP server\n"));
997 s_resp->version = 0x05;
998 s_resp->reply = 0x01;
1000 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1004 //TODO! close socket after the write! schedule task
1005 //GNUNET_NETWORK_socket_close (s5r->sock);
1010 if (MHD_YES == add_handle_to_mhd ( s5r->sock ))
1011 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1012 "Sucessfully added client to MHD!\n");
1013 s_resp->version = 0x05;
1014 s_resp->reply = 0x00;
1015 s_resp->reserved = 0x00;
1016 s_resp->addr_type = 0x01;
1019 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1023 //GNUNET_free ( s5r );
1024 //FIXME complete socks resp!
1029 phost = (struct hostent*)gethostbyname (domain);
1032 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1033 "Resolve %s error!\n", domain );
1034 s_resp->version = 0x05;
1035 s_resp->reply = 0x01;
1037 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1041 //TODO! close socket after the write! schedule task
1042 //GNUNET_NETWORK_socket_close (s5r->sock);
1047 s5r->remote_sock = GNUNET_NETWORK_socket_create (AF_INET,
1050 r_sin_addr = (struct in_addr*)(phost->h_addr);
1051 remote_ip = r_sin_addr->s_addr;
1052 memset(&remote_addr, 0, sizeof(remote_addr));
1053 remote_addr.sin_family = AF_INET;
1054 #if HAVE_SOCKADDR_IN_SIN_LEN
1055 remote_addr.sin_len = sizeof (remote_addr);
1057 remote_addr.sin_addr.s_addr = remote_ip;
1058 remote_addr.sin_port = req_port;
1060 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1061 "target server: %s:%u\n", inet_ntoa(remote_addr.sin_addr),
1065 GNUNET_NETWORK_socket_connect ( s5r->remote_sock,
1066 (const struct sockaddr*)&remote_addr,
1067 sizeof (remote_addr)))
1068 && (errno != EINPROGRESS))
1070 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "connect");
1071 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1072 "socket request error...\n");
1073 s_resp->version = 0x05;
1074 s_resp->reply = 0x01;
1076 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1083 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1084 "new remote connection\n");
1086 s_resp->version = 0x05;
1087 s_resp->reply = 0x00;
1088 s_resp->reserved = 0x00;
1089 s_resp->addr_type = 0x01;
1091 s5r->state = SOCKS5_DATA_TRANSFER;
1094 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1098 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1106 if (s5r->state == SOCKS5_DATA_TRANSFER)
1108 if ((s5r->remote_sock == NULL) || (s5r->rbuf_len == 0))
1110 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1111 "Closing connection to client\n");
1112 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
1113 GNUNET_SCHEDULER_cancel (s5r->rtask);
1114 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
1115 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
1116 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
1117 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
1118 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
1119 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
1121 if (s5r->remote_sock != NULL)
1122 GNUNET_NETWORK_socket_close (s5r->remote_sock);
1123 GNUNET_NETWORK_socket_close (s5r->sock);
1128 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1129 "forwarding %d bytes from client\n", s5r->rbuf_len);
1132 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1134 &do_write_remote, s5r);
1136 if (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK)
1139 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1141 &do_read_remote, s5r);
1147 //GNUNET_CONTAINER_DLL_remove (s5conns.head, s5conns.tail, s5r);
1152 * Accept new incoming connections
1154 * @param cls the closure
1155 * @param tc the scheduler context
1158 do_accept (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1160 struct GNUNET_NETWORK_Handle *s;
1161 struct Socks5Request *s5r;
1163 ltask = GNUNET_SCHEDULER_NO_TASK;
1164 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1167 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1171 s = GNUNET_NETWORK_socket_accept (lsock, NULL, NULL);
1175 GNUNET_log_strerror (GNUNET_ERROR_TYPE_INFO, "accept");
1179 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1180 "Got an inbound connection, waiting for data\n");
1182 s5r = GNUNET_malloc (sizeof (struct Socks5Request));
1184 s5r->state = SOCKS5_INIT;
1185 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
1186 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
1187 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
1188 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1191 //GNUNET_CONTAINER_DLL_insert (s5conns.head, s5conns.tail, s5r);
1195 * Task run on shutdown
1197 * @param cls closure
1198 * @param tc task context
1201 do_shutdown (void *cls,
1202 const struct GNUNET_SCHEDULER_TaskContext *tc)
1204 if (GNUNET_SCHEDULER_NO_TASK != httpd_task)
1206 GNUNET_SCHEDULER_cancel (httpd_task);
1207 httpd_task = GNUNET_SCHEDULER_NO_TASK;
1210 if (GNUNET_SCHEDULER_NO_TASK != curl_download_task)
1212 GNUNET_SCHEDULER_cancel (curl_download_task);
1213 curl_download_task = GNUNET_SCHEDULER_NO_TASK;
1218 MHD_stop_daemon (httpd);
1224 * Main function that will be run
1226 * @param cls closure
1227 * @param args remaining command-line arguments
1228 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
1229 * @param cfg configuration
1232 run (void *cls, char *const *args, const char *cfgfile,
1233 const struct GNUNET_CONFIGURATION_Handle *cfg)
1235 struct sockaddr_in sa;
1237 memset (&sa, 0, sizeof (sa));
1238 sa.sin_family = AF_INET;
1239 sa.sin_port = htons (port);
1240 #if HAVE_SOCKADDR_IN_SIN_LEN
1241 sa.sin_len = sizeof (sa);
1244 lsock = GNUNET_NETWORK_socket_create (AF_INET,
1248 if ((NULL == lsock) ||
1250 GNUNET_NETWORK_socket_bind (lsock, (const struct sockaddr *) &sa,
1253 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1254 "Failed to create listen socket bound to `%s'",
1255 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
1257 GNUNET_NETWORK_socket_close (lsock);
1261 if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock, 5))
1263 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1264 "Failed to listen on socket bound to `%s'",
1265 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
1269 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1270 lsock, &do_accept, NULL);
1275 if (0 != curl_global_init (CURL_GLOBAL_WIN32))
1277 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1278 "cURL global init failed!\n");
1282 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1283 "Proxy listens on port %u\n",
1286 httpd = MHD_start_daemon (MHD_USE_DEBUG, 4444,
1288 &create_response, NULL,
1289 MHD_OPTION_CONNECTION_LIMIT, (unsigned int) 128,
1290 MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
1291 MHD_OPTION_NOTIFY_COMPLETED,
1296 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
1297 &do_shutdown, NULL);
1302 * The main function for gnunet-gns-proxy.
1304 * @param argc number of arguments from the command line
1305 * @param argv command line arguments
1306 * @return 0 ok, 1 on error
1309 main (int argc, char *const *argv)
1311 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
1313 gettext_noop ("listen on specified port"), 1,
1314 &GNUNET_GETOPT_set_string, &port},
1315 GNUNET_GETOPT_OPTION_END
1320 GNUNET_log_setup ("gnunet-gns-proxy", "WARNING", NULL);
1323 GNUNET_PROGRAM_run (argc, argv, "gnunet-gns-proxy",
1324 _("GNUnet GNS proxy"),
1326 &run, NULL)) ? 0 : 1;