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 <gnunet_gns_service.h>
24 #include <microhttpd.h>
25 #include <curl/curl.h>
27 #include "gns_proxy_proto.h"
31 #include <gnutls/gnutls.h>
32 #include <gnutls/x509.h>
33 #include <gnutls/abstract.h>
34 #include <gnutls/crypto.h>
37 #define GNUNET_GNS_PROXY_PORT 7777
38 #define MHD_MAX_CONNECTIONS 300
40 /* MHD/cURL defines */
41 #define BUF_WAIT_FOR_CURL 0
42 #define BUF_WAIT_FOR_MHD 1
43 #define BUF_WAIT_FOR_PP 2
44 #define HTML_HDR_CONTENT "Content-Type: text/html"
46 /* buffer padding for proper RE matching */
47 #define CURL_BUF_PADDING 1000
50 //#define RE_DOTPLUS "<a href=\"http://(([A-Za-z]+[.])+)([+])"
51 #define RE_A_HREF "href=\"https?://(([A-Za-z0-9]+[.])+)([+]|[a-z]+)"
52 #define RE_N_MATCHES 4
54 /* The usual suspects */
56 #define HTTPS_PORT 443
60 * A structure for CA cert/key
65 gnutls_x509_crt_t cert;
68 gnutls_x509_privkey_t key;
71 #define MAX_PEM_SIZE (10 * 1024)
74 * Structure for GNS certificates
76 struct ProxyGNSCertificate
78 /* The certificate as PEM */
79 char cert[MAX_PEM_SIZE];
81 /* The private key as PEM */
82 char key[MAX_PEM_SIZE];
87 * A structure for socks requests
91 /* The client socket */
92 struct GNUNET_NETWORK_Handle *sock;
94 /* The server socket */
95 struct GNUNET_NETWORK_Handle *remote_sock;
100 /* Client socket read task */
101 GNUNET_SCHEDULER_TaskIdentifier rtask;
103 /* Server socket read task */
104 GNUNET_SCHEDULER_TaskIdentifier fwdrtask;
106 /* Client socket write task */
107 GNUNET_SCHEDULER_TaskIdentifier wtask;
109 /* Server socket write task */
110 GNUNET_SCHEDULER_TaskIdentifier fwdwtask;
118 /* Length of data in read buffer */
119 unsigned int rbuf_len;
121 /* Length of data in write buffer */
122 unsigned int wbuf_len;
124 /* This handle is scheduled for cleanup? */
127 /* Shall we close the client socket on cleanup? */
132 * DLL for Network Handles
134 struct NetworkHandleList
137 struct NetworkHandleList *next;
140 struct NetworkHandleList *prev;
143 struct GNUNET_NETWORK_Handle *h;
147 * A structure for all running Httpds
152 struct MhdHttpList *prev;
155 struct MhdHttpList *next;
157 /* is this an ssl daemon? */
160 /* the domain name to server (only important for SSL) */
163 /* The daemon handle */
164 struct MHD_Daemon *daemon;
166 /* Optional proxy certificate used */
167 struct ProxyGNSCertificate *proxy_cert;
170 GNUNET_SCHEDULER_TaskIdentifier httpd_task;
172 /* Handles associated with this daemon */
173 struct NetworkHandleList *socket_handles_head;
175 /* Handles associated with this daemon */
176 struct NetworkHandleList *socket_handles_tail;
180 * A structure for MHD<->cURL streams
185 struct ProxyCurlTask *prev;
188 struct ProxyCurlTask *next;
193 /* Optional header replacements for curl (LEHO) */
194 struct curl_slist *headers;
196 /* Optional resolver replacements for curl (LEHO) */
197 struct curl_slist *resolver;
199 /* curl response code */
200 long curl_response_code;
202 /* The URL to fetch */
205 /* The cURL write buffer / MHD read buffer */
206 char buffer[CURL_MAX_WRITE_SIZE + CURL_BUF_PADDING];
208 /* Read pos of the data in the buffer */
209 char *buffer_read_ptr;
211 /* Write pos in the buffer */
212 char *buffer_write_ptr;
214 /* The buffer status (BUF_WAIT_FOR_CURL or BUF_WAIT_FOR_MHD) */
217 /* Number of bytes in buffer */
218 unsigned int bytes_in_buffer;
220 /* Indicates wheather the download is in progress */
221 int download_in_progress;
223 /* Indicates wheather the download was successful */
224 int download_is_finished;
226 /* Indicates wheather the download failed */
229 /* Indicates wheather we need to parse HTML */
232 /* Indicates wheather we are postprocessing the HTML right now */
233 int is_postprocessing;
235 /* Indicates wheather postprocessing has finished */
239 GNUNET_SCHEDULER_TaskIdentifier pp_task;
242 struct ProxyREMatch *pp_match_head;
245 struct ProxyREMatch *pp_match_tail;
247 /* The authority of the corresponding host (site of origin) */
250 /* The hostname (Host header field) */
253 /* The LEgacy HOstname (can be empty) */
256 /* The associated daemon list entry */
257 struct MhdHttpList *mhd;
259 /* The associated response */
260 struct MHD_Response *response;
263 struct ProxySetCookieHeader *set_cookies_head;
266 struct ProxySetCookieHeader *set_cookies_tail;
268 /* connection status */
275 struct MHD_Connection *connection;
278 size_t put_read_offset;
279 size_t put_read_size;
284 * Struct for RE matches in postprocessing of HTML
289 struct ProxyREMatch *next;
292 struct ProxyREMatch *prev;
301 struct GNUNET_GNS_ShortenRequest *shorten_task;
306 /* start of match in buffer */
309 /* end of match in buffer */
312 /* associated proxycurltask */
313 struct ProxyCurlTask *ctask;
317 * Struct for set-cookies
319 struct ProxySetCookieHeader
322 struct ProxySetCookieHeader *next;
325 struct ProxySetCookieHeader *prev;
331 /* The port the proxy is running on (default 7777) */
332 static unsigned long port = GNUNET_GNS_PROXY_PORT;
334 /* The CA file (pem) to use for the proxy CA */
335 static char* cafile_opt;
337 /* The listen socket of the proxy */
338 static struct GNUNET_NETWORK_Handle *lsock;
340 /* The listen task ID */
341 GNUNET_SCHEDULER_TaskIdentifier ltask;
343 /* The cURL download task */
344 GNUNET_SCHEDULER_TaskIdentifier curl_download_task;
346 /* The non SSL httpd daemon handle */
347 static struct MHD_Daemon *httpd;
349 /* Number of current mhd connections */
350 static unsigned int total_mhd_connections;
352 /* The cURL multi handle */
353 static CURLM *curl_multi;
355 /* Handle to the GNS service */
356 static struct GNUNET_GNS_Handle *gns_handle;
358 /* DLL for ProxyCurlTasks */
359 static struct ProxyCurlTask *ctasks_head;
361 /* DLL for ProxyCurlTasks */
362 static struct ProxyCurlTask *ctasks_tail;
364 /* DLL for http daemons */
365 static struct MhdHttpList *mhd_httpd_head;
367 /* DLL for http daemons */
368 static struct MhdHttpList *mhd_httpd_tail;
370 /* Handle to the regex for dotplus (.+) replacement in HTML */
371 static regex_t re_dotplus;
373 /* The users local GNS zone hash */
374 static struct GNUNET_CRYPTO_ShortHashCode local_gns_zone;
376 /* The users local private zone */
377 static struct GNUNET_CRYPTO_ShortHashCode local_private_zone;
379 /* The users local shorten zone */
380 static struct GNUNET_CRYPTO_ShortHashCode local_shorten_zone;
382 /* The CA for SSL certificate generation */
383 static struct ProxyCA proxy_ca;
385 /* UNIX domain socket for mhd */
386 struct GNUNET_NETWORK_Handle *mhd_unix_socket;
388 /* Shorten zone private key */
389 struct GNUNET_CRYPTO_RsaPrivateKey *shorten_zonekey;
392 * Checks if name is in tld
394 * @param name the name to check
395 * @param tld the TLD to check for
396 * @return GNUNET_YES or GNUNET_NO
399 is_tld(const char* name, const char* tld)
403 if (strlen(name) <= strlen(tld))
406 offset = strlen(name) - strlen(tld);
407 if (0 != strcmp (name+offset, tld))
409 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
410 "%s is not in .%s TLD\n", name, tld);
418 get_uri_val_iter (void *cls,
419 enum MHD_ValueKind kind,
425 sprintf (buf+strlen (buf), "?%s=%s", key, value);
431 * Read HTTP request header field 'Host'
433 * @param cls buffer to write to
434 * @param kind value kind
435 * @param key field key
436 * @param value field value
437 * @return MHD_NO when Host found
440 con_val_iter (void *cls,
441 enum MHD_ValueKind kind,
445 struct ProxyCurlTask *ctask = cls;
446 char* buf = ctask->host;
450 if (0 == strcmp ("Host", key))
456 if (0 == strcmp ("Accept-Encoding", key))
461 cstr = GNUNET_malloc (strlen (key) + strlen (hdr_val) + 3);
462 GNUNET_snprintf (cstr, strlen (key) + strlen (hdr_val) + 3,
463 "%s: %s", key, hdr_val);
465 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
466 "Client Header: %s\n", cstr);
468 ctask->headers = curl_slist_append (ctask->headers, cstr);
476 * Callback for MHD response
479 * @param pos in buffer
481 * @param max space in buffer
482 * @return number of bytes written
485 mhd_content_cb (void *cls,
491 * Check HTTP response header for mime
493 * @param buffer curl buffer
494 * @param size curl blocksize
495 * @param nmemb curl blocknumber
497 * @return size of read bytes
500 curl_check_hdr (void *buffer, size_t size, size_t nmemb, void *cls)
502 size_t bytes = size * nmemb;
503 struct ProxyCurlTask *ctask = cls;
504 int html_mime_len = strlen (HTML_HDR_CONTENT);
505 int cookie_hdr_len = strlen (MHD_HTTP_HEADER_SET_COOKIE);
506 char hdr_mime[html_mime_len+1];
507 char hdr_generic[bytes+1];
508 char new_cookie_hdr[bytes+strlen (ctask->leho)+1];
517 if (NULL == ctask->response)
519 /* FIXME: get total size from curl (if available) */
520 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
521 "Creating response for %s\n", ctask->url);
522 ctask->response = MHD_create_response_from_callback (MHD_SIZE_UNKNOWN,
523 sizeof (ctask->buffer),
527 ctask->ready_to_queue = GNUNET_YES;
531 if (html_mime_len <= bytes)
533 memcpy (hdr_mime, buffer, html_mime_len);
534 hdr_mime[html_mime_len] = '\0';
536 if (0 == strcmp (hdr_mime, HTML_HDR_CONTENT))
538 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
539 "Got HTML HTTP response header\n");
540 ctask->parse_content = GNUNET_YES;
544 if (cookie_hdr_len > bytes)
547 memcpy (hdr_generic, buffer, bytes);
548 hdr_generic[bytes] = '\0';
550 if ('\n' == hdr_generic[bytes-1])
551 hdr_generic[bytes-1] = '\0';
553 if (hdr_generic[bytes-2] == '\r')
554 hdr_generic[bytes-2] = '\0';
556 if (0 == memcmp (hdr_generic,
557 MHD_HTTP_HEADER_SET_COOKIE,
560 ndup = GNUNET_strdup (hdr_generic+cookie_hdr_len+1);
561 memset (new_cookie_hdr, 0, sizeof (new_cookie_hdr));
562 tok = strtok (ndup, ";");
564 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
565 "Looking for cookie in : %s\n", hdr_generic);
567 for (; tok != NULL; tok = strtok (NULL, ";"))
569 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
570 "Got Cookie token: %s\n", tok);
571 //memcpy (new_cookie_hdr+offset, tok, strlen (tok));
572 if (0 == memcmp (tok, " domain", strlen (" domain")))
574 cookie_domain = tok + strlen (" domain") + 1;
576 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
577 "Got Set-Cookie Domain: %s\n", cookie_domain);
579 if (strlen (cookie_domain) < strlen (ctask->leho))
581 delta_cdomain = strlen (ctask->leho) - strlen (cookie_domain);
582 if (0 == strcmp (cookie_domain, ctask->leho + (delta_cdomain)))
584 GNUNET_snprintf (new_cookie_hdr+offset,
585 sizeof (new_cookie_hdr),
586 " domain=%s", ctask->authority);
587 offset += strlen (" domain=") + strlen (ctask->authority);
588 new_cookie_hdr[offset] = ';';
593 else if (strlen (cookie_domain) == strlen (ctask->leho))
595 if (0 == strcmp (cookie_domain, ctask->leho))
597 GNUNET_snprintf (new_cookie_hdr+offset,
598 sizeof (new_cookie_hdr),
599 " domain=%s", ctask->host);
600 offset += strlen (" domain=") + strlen (ctask->host);
601 new_cookie_hdr[offset] = ';';
606 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
607 "Cookie domain invalid\n");
611 memcpy (new_cookie_hdr+offset, tok, strlen (tok));
612 offset += strlen (tok);
613 new_cookie_hdr[offset] = ';';
619 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
620 "Got Set-Cookie HTTP header %s\n", new_cookie_hdr);
622 if (GNUNET_NO == MHD_add_response_header (ctask->response,
623 MHD_HTTP_HEADER_SET_COOKIE,
626 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
627 "MHD: Error adding set-cookie header field %s\n",
628 hdr_generic+cookie_hdr_len+1);
633 ndup = GNUNET_strdup (hdr_generic);
634 hdr_type = strtok (ndup, ":");
636 if (NULL == hdr_type)
642 hdr_val = strtok (NULL, "");
650 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
651 "Trying to set %s: %s\n",
654 if (GNUNET_NO == MHD_add_response_header (ctask->response,
658 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
659 "MHD: Error adding %s header field %s\n",
670 * @param hd a http daemon list entry
673 run_httpd (struct MhdHttpList *hd);
684 * Task run whenever HTTP server operations are pending.
687 * @param tc sched context
691 const struct GNUNET_SCHEDULER_TaskContext *tc);
694 run_mhd_now (struct MhdHttpList *hd)
696 if (GNUNET_SCHEDULER_NO_TASK != hd->httpd_task)
698 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
699 "MHD: killing old task\n");
700 GNUNET_SCHEDULER_cancel (hd->httpd_task);
702 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
703 "MHD: Scheduling MHD now\n");
704 hd->httpd_task = GNUNET_SCHEDULER_add_now (&do_httpd, hd);
708 * Ask cURL for the select sets and schedule download
711 curl_download_prepare ();
714 * Callback to free content
716 * @param cls content to free
717 * @param tc task context
720 mhd_content_free (void *cls,
721 const struct GNUNET_SCHEDULER_TaskContext *tc)
723 struct ProxyCurlTask *ctask = cls;
724 GNUNET_assert (NULL == ctask->pp_match_head);
726 if (NULL != ctask->headers)
727 curl_slist_free_all (ctask->headers);
729 if (NULL != ctask->headers)
730 curl_slist_free_all (ctask->resolver);
732 if (NULL != ctask->response)
733 MHD_destroy_response (ctask->response);
741 * Callback for MHD response
744 * @param pos in buffer
746 * @param max space in buffer
747 * @return number of bytes written
750 mhd_content_cb (void *cls,
755 struct ProxyCurlTask *ctask = cls;
756 struct ProxyREMatch *re_match = ctask->pp_match_head;
758 long long int bytes_to_copy = ctask->buffer_write_ptr - ctask->buffer_read_ptr;
760 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
761 "MHD: content cb for %s. To copy: %lld\n",
762 ctask->url, bytes_to_copy);
763 GNUNET_assert (bytes_to_copy >= 0);
765 if ((GNUNET_YES == ctask->download_is_finished) &&
766 (GNUNET_NO == ctask->download_error) &&
767 (0 == bytes_to_copy) &&
768 (BUF_WAIT_FOR_CURL == ctask->buf_status))
770 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
771 "MHD: sending response for %s\n", ctask->url);
772 ctask->download_in_progress = GNUNET_NO;
773 run_mhd_now (ctask->mhd);
774 GNUNET_SCHEDULER_add_now (&mhd_content_free, ctask);
775 total_mhd_connections--;
776 return MHD_CONTENT_READER_END_OF_STREAM;
779 if ((GNUNET_YES == ctask->download_error) &&
780 (GNUNET_YES == ctask->download_is_finished) &&
781 (0 == bytes_to_copy) &&
782 (BUF_WAIT_FOR_CURL == ctask->buf_status))
784 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
785 "MHD: sending error response\n");
786 ctask->download_in_progress = GNUNET_NO;
787 run_mhd_now (ctask->mhd);
788 GNUNET_SCHEDULER_add_now (&mhd_content_free, ctask);
789 total_mhd_connections--;
790 return MHD_CONTENT_READER_END_WITH_ERROR;
793 if ( ctask->buf_status == BUF_WAIT_FOR_CURL )
797 for (; NULL != re_match; re_match = ctask->pp_match_head)
799 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
800 "MHD: Processing PP %s\n",
802 bytes_to_copy = re_match->start - ctask->buffer_read_ptr;
803 GNUNET_assert (bytes_to_copy >= 0);
805 if (bytes_to_copy+copied > max)
807 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
808 "MHD: buffer in response too small for %d. Using available space (%d). (%s)\n",
812 memcpy (buf+copied, ctask->buffer_read_ptr, max-copied);
813 ctask->buffer_read_ptr += max-copied;
815 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
816 "MHD: copied %d bytes\n", copied);
820 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
821 "MHD: copying %d bytes to mhd response at offset %d\n",
822 bytes_to_copy, ctask->buffer_read_ptr);
823 memcpy (buf+copied, ctask->buffer_read_ptr, bytes_to_copy);
824 copied += bytes_to_copy;
826 if (GNUNET_NO == re_match->done)
828 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
829 "MHD: Waiting for PP of %s\n", re_match->hostname);
830 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
831 "MHD: copied %d bytes\n", copied);
832 ctask->buffer_read_ptr += bytes_to_copy;
836 if (strlen (re_match->result) > (max - copied))
838 //FIXME partially copy domain here
839 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
840 "MHD: buffer in response too small for %s! (%s)\n",
843 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
844 "MHD: copied %d bytes\n", copied);
845 ctask->buffer_read_ptr += bytes_to_copy;
849 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
850 "MHD: Adding PP result %s to buffer\n",
852 memcpy (buf+copied, re_match->result, strlen (re_match->result));
853 copied += strlen (re_match->result);
854 ctask->buffer_read_ptr = re_match->end;
855 GNUNET_CONTAINER_DLL_remove (ctask->pp_match_head,
856 ctask->pp_match_tail,
858 GNUNET_free (re_match);
861 bytes_to_copy = ctask->buffer_write_ptr - ctask->buffer_read_ptr;
863 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
864 "MHD: copied: %d left: %d, space left in buf: %d\n",
866 bytes_to_copy, max-copied);
868 GNUNET_assert (0 <= bytes_to_copy);
870 if (GNUNET_NO == ctask->download_is_finished)
872 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
873 "MHD: Purging buffer\n");
874 memmove (ctask->buffer, ctask->buffer_read_ptr, bytes_to_copy);
875 ctask->buffer_read_ptr = ctask->buffer;
876 ctask->buffer_write_ptr = ctask->buffer + bytes_to_copy;
877 ctask->buffer[bytes_to_copy] = '\0';
880 if (bytes_to_copy+copied > max)
881 bytes_to_copy = max-copied;
883 if (0 > bytes_to_copy)
886 memcpy (buf+copied, ctask->buffer_read_ptr, bytes_to_copy);
887 ctask->buffer_read_ptr += bytes_to_copy;
888 copied += bytes_to_copy;
889 ctask->buf_status = BUF_WAIT_FOR_CURL;
891 if (NULL != ctask->curl)
892 curl_easy_pause (ctask->curl, CURLPAUSE_CONT);
894 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
895 "MHD: copied %d bytes\n", copied);
896 run_mhd_now (ctask->mhd);
901 * Shorten result callback
903 * @param cls the proxycurltask
904 * @param short_name the shortened name (NULL on error)
907 process_shorten (void* cls, const char* short_name)
909 struct ProxyREMatch *re_match = cls;
910 char result[sizeof (re_match->result)];
912 if (NULL == short_name)
914 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
915 "PP: Unable to shorten %s\n",
917 GNUNET_CONTAINER_DLL_remove (re_match->ctask->pp_match_head,
918 re_match->ctask->pp_match_tail,
920 GNUNET_free (re_match);
924 if (0 == strcmp (short_name, re_match->ctask->leho))
925 strcpy (result, re_match->ctask->host);
927 strcpy (result, short_name);
929 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
930 "PP: Shorten %s -> %s\n",
934 if (re_match->ctask->mhd->is_ssl)
935 sprintf (re_match->result, "href=\"https://%s", result);
937 sprintf (re_match->result, "href=\"http://%s", result);
939 re_match->done = GNUNET_YES;
940 run_mhd_now (re_match->ctask->mhd);
945 * Postprocess data in buffer. From read ptr to write ptr
947 * @param cls the curlproxytask
948 * @param tc task context
951 postprocess_buffer (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
953 struct ProxyCurlTask *ctask = cls;
954 struct ProxyREMatch *re_match;
955 char* re_ptr = ctask->buffer_read_ptr;
956 char re_hostname[255];
957 regmatch_t m[RE_N_MATCHES];
959 ctask->pp_task = GNUNET_SCHEDULER_NO_TASK;
961 if (GNUNET_YES != ctask->parse_content)
963 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
964 "PP: Not parsing content\n");
965 ctask->buf_status = BUF_WAIT_FOR_MHD;
966 run_mhd_now (ctask->mhd);
970 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
971 "PP: We need to parse the HTML\n");
973 /* 0 means match found */
974 while (0 == regexec (&re_dotplus, re_ptr, RE_N_MATCHES, m, 0))
976 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
977 "PP: regex match\n");
979 GNUNET_assert (m[1].rm_so != -1);
981 memset (re_hostname, 0, sizeof (re_hostname));
982 memcpy (re_hostname, re_ptr+m[1].rm_so, (m[3].rm_eo-m[1].rm_so));
984 re_match = GNUNET_malloc (sizeof (struct ProxyREMatch));
985 re_match->start = re_ptr + m[0].rm_so;
986 re_match->end = re_ptr + m[3].rm_eo;
987 re_match->done = GNUNET_NO;
988 re_match->ctask = ctask;
989 strcpy (re_match->hostname, re_hostname);
990 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
991 "PP: Got hostname %s\n", re_hostname);
992 re_ptr += m[3].rm_eo;
994 if (GNUNET_YES == is_tld (re_match->hostname, GNUNET_GNS_TLD_PLUS))
996 re_match->hostname[strlen(re_match->hostname)-1] = '\0';
997 strcpy (re_match->hostname+strlen(re_match->hostname),
1001 re_match->shorten_task = GNUNET_GNS_shorten_zone (gns_handle,
1003 &local_private_zone,
1004 &local_shorten_zone,
1007 re_match); //FIXME cancel appropriately
1009 GNUNET_CONTAINER_DLL_insert_tail (ctask->pp_match_head,
1010 ctask->pp_match_tail,
1014 ctask->buf_status = BUF_WAIT_FOR_MHD;
1015 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1016 "PP: No more matches\n");
1017 run_mhd_now (ctask->mhd);
1021 * Handle data from cURL
1023 * @param ptr pointer to the data
1024 * @param size number of blocks of data
1025 * @param nmemb blocksize
1026 * @param ctx the curlproxytask
1027 * @return number of bytes handled
1030 curl_download_cb (void *ptr, size_t size, size_t nmemb, void* ctx)
1032 const char *cbuf = ptr;
1033 size_t total = size * nmemb;
1034 struct ProxyCurlTask *ctask = ctx;
1035 size_t buf_space = sizeof (ctask->buffer) -
1036 (ctask->buffer_write_ptr-ctask->buffer);
1038 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1039 "CURL: Got %d. %d free in buffer\n",
1042 if (total > (buf_space - CURL_BUF_PADDING))
1044 if (ctask->buf_status == BUF_WAIT_FOR_CURL)
1046 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1047 "CURL: Buffer full starting postprocessing\n");
1048 ctask->buf_status = BUF_WAIT_FOR_PP;
1049 ctask->pp_task = GNUNET_SCHEDULER_add_now (&postprocess_buffer,
1051 return CURL_WRITEFUNC_PAUSE;
1054 /* we should not get called in that case */
1055 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1056 "CURL: called out of context and no space in buffer!\n");
1057 return CURL_WRITEFUNC_PAUSE;
1060 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1061 "CURL: Copying %d bytes to buffer (%s)\n", total, ctask->url);
1062 memcpy (ctask->buffer_write_ptr, cbuf, total);
1063 ctask->bytes_in_buffer += total;
1064 ctask->buffer_write_ptr += total;
1065 ctask->buffer_write_ptr[0] = '\0';
1071 * Task that is run when we are ready to receive more data
1074 * @param cls closure
1075 * @param tc task context
1078 curl_task_download (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1081 * Ask cURL for the select sets and schedule download
1084 curl_download_prepare ()
1091 struct GNUNET_NETWORK_FDSet *grs;
1092 struct GNUNET_NETWORK_FDSet *gws;
1094 struct GNUNET_TIME_Relative rtime;
1100 mret = curl_multi_fdset (curl_multi, &rs, &ws, &es, &max);
1102 if (mret != CURLM_OK)
1104 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1105 "%s failed at %s:%d: `%s'\n",
1106 "curl_multi_fdset", __FILE__, __LINE__,
1107 curl_multi_strerror (mret));
1108 //TODO cleanup here?
1112 mret = curl_multi_timeout (curl_multi, &to);
1113 rtime = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, to);
1115 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1116 "cURL multi fds: max=%d timeout=%llu\n", max, to);
1118 grs = GNUNET_NETWORK_fdset_create ();
1119 gws = GNUNET_NETWORK_fdset_create ();
1120 GNUNET_NETWORK_fdset_copy_native (grs, &rs, max + 1);
1121 GNUNET_NETWORK_fdset_copy_native (gws, &ws, max + 1);
1122 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1123 "Scheduling task cURL\n");
1125 if (curl_download_task != GNUNET_SCHEDULER_NO_TASK)
1126 GNUNET_SCHEDULER_cancel (curl_download_task);
1130 curl_download_task =
1131 GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1134 &curl_task_download, curl_multi);
1136 else if (NULL != ctasks_head)
1138 /* as specified in curl docs */
1139 curl_download_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
1140 &curl_task_download,
1143 GNUNET_NETWORK_fdset_destroy (gws);
1144 GNUNET_NETWORK_fdset_destroy (grs);
1150 * Task that is run when we are ready to receive more data
1153 * @param cls closure
1154 * @param tc task context
1157 curl_task_download (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1161 struct CURLMsg *msg;
1163 struct ProxyCurlTask *ctask;
1167 struct ProxyCurlTask *clean_head = NULL;
1168 struct ProxyCurlTask *clean_tail = NULL;
1170 curl_download_task = GNUNET_SCHEDULER_NO_TASK;
1172 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1174 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1175 "Shutdown requested while trying to download\n");
1179 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1187 mret = curl_multi_perform (curl_multi, &running);
1189 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1190 "Running curl tasks: %d\n", running);
1192 ctask = ctasks_head;
1193 for (; ctask != NULL; ctask = ctask->next)
1195 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1196 "CTask: %s\n", ctask->url);
1200 if (num_ctasks != running)
1202 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1203 "%d ctasks, %d curl running\n", num_ctasks, running);
1208 ctask = ctasks_head;
1209 msg = curl_multi_info_read (curl_multi, &msgnum);
1210 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1211 "Messages left: %d\n", msgnum);
1218 if ((msg->data.result != CURLE_OK) &&
1219 (msg->data.result != CURLE_GOT_NOTHING))
1221 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1222 "Download curl failed");
1224 for (; ctask != NULL; ctask = ctask->next)
1226 if (NULL == ctask->curl)
1229 if (memcmp (msg->easy_handle, ctask->curl, sizeof (CURL)) != 0)
1232 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1233 "CURL: Download failed for task %s: %s.\n",
1235 curl_easy_strerror (msg->data.result));
1236 ctask->download_is_finished = GNUNET_YES;
1237 ctask->download_error = GNUNET_YES;
1238 if (CURLE_OK == curl_easy_getinfo (ctask->curl,
1239 CURLINFO_RESPONSE_CODE,
1241 ctask->curl_response_code = resp_code;
1242 ctask->ready_to_queue = MHD_YES;
1243 ctask->buf_status = BUF_WAIT_FOR_MHD;
1244 run_mhd_now (ctask->mhd);
1246 GNUNET_CONTAINER_DLL_remove (ctasks_head, ctasks_tail,
1248 GNUNET_CONTAINER_DLL_insert (clean_head, clean_tail, ctask);
1251 GNUNET_assert (ctask != NULL);
1255 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1256 "CURL: download completed.\n");
1258 for (; ctask != NULL; ctask = ctask->next)
1260 if (NULL == ctask->curl)
1263 if (memcmp (msg->easy_handle, ctask->curl, sizeof (CURL)) != 0)
1266 if (ctask->buf_status != BUF_WAIT_FOR_CURL)
1269 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1270 "CURL: completed task %s found.\n", ctask->url);
1271 if (CURLE_OK == curl_easy_getinfo (ctask->curl,
1272 CURLINFO_RESPONSE_CODE,
1274 ctask->curl_response_code = resp_code;
1277 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1278 "CURL: Completed ctask!\n");
1279 if (GNUNET_SCHEDULER_NO_TASK == ctask->pp_task)
1281 ctask->buf_status = BUF_WAIT_FOR_PP;
1282 ctask->pp_task = GNUNET_SCHEDULER_add_now (&postprocess_buffer,
1286 ctask->ready_to_queue = MHD_YES;
1287 ctask->download_is_finished = GNUNET_YES;
1289 //GNUNET_SCHEDULER_add_now (&run_mhd, ctask->mhd);
1291 /* We MUST not modify the multi handle else we loose messages */
1292 GNUNET_CONTAINER_DLL_remove (ctasks_head, ctasks_tail,
1294 GNUNET_CONTAINER_DLL_insert (clean_head, clean_tail, ctask);
1298 GNUNET_assert (ctask != NULL);
1300 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1301 "CURL: %s\n", curl_easy_strerror(msg->data.result));
1307 } while (msgnum > 0);
1309 for (ctask=clean_head; ctask != NULL; ctask = ctask->next)
1311 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1312 "CURL: Removing task %s.\n", ctask->url);
1313 curl_multi_remove_handle (curl_multi, ctask->curl);
1314 curl_easy_cleanup (ctask->curl);
1319 for (ctask=ctasks_head; ctask != NULL; ctask = ctask->next)
1324 if (num_ctasks != running)
1326 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1327 "CURL: %d tasks, %d running\n", num_ctasks, running);
1330 GNUNET_assert ( num_ctasks == running );
1333 } while (mret == CURLM_CALL_MULTI_PERFORM);
1335 if (mret != CURLM_OK)
1337 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "CURL: %s failed at %s:%d: `%s'\n",
1338 "curl_multi_perform", __FILE__, __LINE__,
1339 curl_multi_strerror (mret));
1341 curl_download_prepare();
1345 * Process LEHO lookup
1347 * @param cls the ctask
1348 * @param rd_count number of records returned
1349 * @param rd record data
1352 process_leho_lookup (void *cls,
1354 const struct GNUNET_NAMESTORE_RecordData *rd)
1356 struct ProxyCurlTask *ctask = cls;
1357 char hosthdr[262]; //256 + "Host: "
1361 struct hostent *phost;
1363 char resolvename[512];
1366 strcpy (ctask->leho, "");
1369 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1370 "No LEHO present!\n");
1372 for (i=0; i<rd_count; i++)
1374 if (rd[i].record_type != GNUNET_GNS_RECORD_LEHO)
1377 memcpy (ctask->leho, rd[i].data, rd[i].data_size);
1379 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1380 "Found LEHO %s for %s\n", ctask->leho, ctask->url);
1383 if (0 != strcmp (ctask->leho, ""))
1385 sprintf (hosthdr, "%s%s", "Host: ", ctask->leho);
1386 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1387 "New HTTP header value: %s\n", hosthdr);
1388 ctask->headers = curl_slist_append (ctask->headers, hosthdr);
1389 GNUNET_assert (NULL != ctask->headers);
1390 ret = curl_easy_setopt (ctask->curl, CURLOPT_HTTPHEADER, ctask->headers);
1391 if (CURLE_OK != ret)
1393 GNUNET_log(GNUNET_ERROR_TYPE_WARNING, "%s failed at %s:%d: `%s'\n",
1394 "curl_easy_setopt", __FILE__, __LINE__, curl_easy_strerror(ret));
1399 if (ctask->mhd->is_ssl)
1401 phost = (struct hostent*)gethostbyname (ctask->host);
1405 ssl_ip = inet_ntoa(*((struct in_addr*)(phost->h_addr)));
1406 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1407 "SSL target server: %s\n", ssl_ip);
1408 sprintf (resolvename, "%s:%d:%s", ctask->leho, HTTPS_PORT, ssl_ip);
1409 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1410 "Curl resolve: %s\n", resolvename);
1411 ctask->resolver = curl_slist_append ( ctask->resolver, resolvename);
1412 curl_easy_setopt (ctask->curl, CURLOPT_RESOLVE, ctask->resolver);
1413 sprintf (curlurl, "https://%s%s", ctask->leho, ctask->url);
1414 curl_easy_setopt (ctask->curl, CURLOPT_URL, curlurl);
1418 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1419 "gethostbyname failed for %s!\n", ctask->host);
1420 ctask->download_is_finished = GNUNET_YES;
1421 ctask->download_error = GNUNET_YES;
1426 if (CURLM_OK != (mret=curl_multi_add_handle (curl_multi, ctask->curl)))
1428 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1429 "%s failed at %s:%d: `%s'\n",
1430 "curl_multi_add_handle", __FILE__, __LINE__,
1431 curl_multi_strerror (mret));
1432 ctask->download_is_finished = GNUNET_YES;
1433 ctask->download_error = GNUNET_YES;
1436 GNUNET_CONTAINER_DLL_insert (ctasks_head, ctasks_tail, ctask);
1438 curl_download_prepare ();
1443 * Initialize download and trigger curl
1445 * @param cls the proxycurltask
1446 * @param auth_name the name of the authority (site of origin) of ctask->host
1450 process_get_authority (void *cls,
1451 const char* auth_name)
1453 struct ProxyCurlTask *ctask = cls;
1455 if (NULL == auth_name)
1457 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1458 "Get authority failed!\n");
1459 strcpy (ctask->authority, "");
1463 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1464 "Get authority yielded %s\n", auth_name);
1465 strcpy (ctask->authority, auth_name);
1468 GNUNET_GNS_lookup_zone (gns_handle,
1471 GNUNET_GNS_RECORD_LEHO,
1472 GNUNET_YES, //Only cached for performance
1474 &process_leho_lookup,
1480 * Main MHD callback for handling requests.
1483 * @param con MHD connection handle
1484 * @param url the url in the request
1485 * @param meth the HTTP method used ("GET", "PUT", etc.)
1486 * @param ver the HTTP version string (i.e. "HTTP/1.1")
1487 * @param upload_data the data being uploaded (excluding HEADERS,
1488 * for a POST that fits into memory and that is encoded
1489 * with a supported encoding, the POST data will NOT be
1490 * given in upload_data and is instead available as
1491 * part of MHD_get_connection_values; very large POST
1492 * data *will* be made available incrementally in
1494 * @param upload_data_size set initially to the size of the
1495 * upload_data provided; the method must update this
1496 * value to the number of bytes NOT processed;
1497 * @param con_cls pointer to location where we store the 'struct Request'
1498 * @return MHD_YES if the connection was handled successfully,
1499 * MHD_NO if the socket must be closed due to a serious
1500 * error while handling the request
1503 create_response (void *cls,
1504 struct MHD_Connection *con,
1508 const char *upload_data,
1509 size_t *upload_data_size,
1512 struct MhdHttpList* hd = cls;
1513 const char* page = "<html><head><title>gnoxy</title>"\
1514 "</head><body>cURL fail</body></html>";
1516 char curlurl[512]; // buffer overflow!
1519 struct ProxyCurlTask *ctask;
1522 if ((0 != strcasecmp (meth, MHD_HTTP_METHOD_GET)) &&
1523 (0 != strcasecmp (meth, MHD_HTTP_METHOD_PUT)) &&
1524 (0 != strcasecmp (meth, MHD_HTTP_METHOD_POST)))
1526 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1527 "MHD: %s NOT IMPLEMENTED!\n", meth);
1532 if (NULL == *con_cls)
1535 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1536 "Got %s request for %s\n", meth, url);
1537 ctask = GNUNET_malloc (sizeof (struct ProxyCurlTask));
1541 // FIXME: probably not best here, also never free'ed...
1542 if (NULL == curl_multi)
1543 curl_multi = curl_multi_init ();
1545 ctask->curl = curl_easy_init();
1547 if ((NULL == ctask->curl) || (NULL == curl_multi)) // ugh, curl_multi init failure check MUCH earlier
1549 ctask->response = MHD_create_response_from_buffer (strlen (page),
1551 MHD_RESPMEM_PERSISTENT);
1552 ret = MHD_queue_response (con,
1555 MHD_destroy_response (ctask->response);
1556 GNUNET_free (ctask);
1560 ctask->download_in_progress = GNUNET_YES;
1561 ctask->buf_status = BUF_WAIT_FOR_CURL;
1562 ctask->connection = con;
1563 ctask->curl_response_code = MHD_HTTP_OK;
1564 ctask->buffer_read_ptr = ctask->buffer;
1565 ctask->buffer_write_ptr = ctask->buffer;
1566 ctask->pp_task = GNUNET_SCHEDULER_NO_TASK;
1568 MHD_get_connection_values (con,
1570 &con_val_iter, ctask);
1572 if (0 == strcasecmp (meth, MHD_HTTP_METHOD_PUT))
1574 if (0 == *upload_data_size)
1576 curl_easy_cleanup (ctask->curl);
1577 GNUNET_free (ctask);
1580 ctask->put_read_offset = 0;
1581 ctask->put_read_size = *upload_data_size;
1582 curl_easy_setopt (ctask->curl, CURLOPT_UPLOAD, 1);
1583 curl_easy_setopt (ctask->curl, CURLOPT_READDATA, upload_data);
1584 //curl_easy_setopt (ctask->curl, CURLOPT_READFUNCTION, &curl_read_cb);
1586 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1587 "Got PUT data: %s\n", upload_data);
1588 curl_easy_cleanup (ctask->curl);
1589 GNUNET_free (ctask);
1593 if (0 == strcasecmp (meth, MHD_HTTP_METHOD_POST))
1595 if (0 == *upload_data_size)
1597 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1598 "NO data for post!\n");
1599 curl_easy_cleanup (ctask->curl);
1600 GNUNET_free (ctask);
1603 curl_easy_setopt (ctask->curl, CURLOPT_POST, 1);
1604 curl_easy_setopt (ctask->curl, CURLOPT_POSTFIELDSIZE, *upload_data_size);
1605 curl_easy_setopt (ctask->curl, CURLOPT_COPYPOSTFIELDS, upload_data);
1607 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1608 "Got POST data: %s\n", upload_data);
1609 curl_easy_cleanup (ctask->curl);
1610 GNUNET_free (ctask);
1614 curl_easy_setopt (ctask->curl, CURLOPT_HEADERFUNCTION, &curl_check_hdr);
1615 curl_easy_setopt (ctask->curl, CURLOPT_HEADERDATA, ctask);
1616 curl_easy_setopt (ctask->curl, CURLOPT_WRITEFUNCTION, &curl_download_cb);
1617 curl_easy_setopt (ctask->curl, CURLOPT_WRITEDATA, ctask);
1618 curl_easy_setopt (ctask->curl, CURLOPT_FOLLOWLOCATION, 0);
1619 curl_easy_setopt (ctask->curl, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4);
1620 //curl_easy_setopt (ctask->curl, CURLOPT_MAXREDIRS, 4);
1621 /* no need to abort if the above failed */
1622 if (GNUNET_NO == ctask->mhd->is_ssl)
1624 sprintf (curlurl, "http://%s%s", ctask->host, url);
1625 MHD_get_connection_values (con,
1626 MHD_GET_ARGUMENT_KIND,
1627 &get_uri_val_iter, curlurl);
1628 curl_easy_setopt (ctask->curl, CURLOPT_URL, curlurl);
1630 strcpy (ctask->url, url);
1631 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1632 "MHD: Adding new curl task for %s%s\n", ctask->host, url);
1633 MHD_get_connection_values (con,
1634 MHD_GET_ARGUMENT_KIND,
1635 &get_uri_val_iter, ctask->url);
1637 //curl_easy_setopt (ctask->curl, CURLOPT_URL, curlurl);
1638 curl_easy_setopt (ctask->curl, CURLOPT_FAILONERROR, 1);
1639 curl_easy_setopt (ctask->curl, CURLOPT_CONNECTTIMEOUT, 600L);
1640 curl_easy_setopt (ctask->curl, CURLOPT_TIMEOUT, 600L);
1642 GNUNET_GNS_get_authority (gns_handle,
1644 &process_get_authority,
1646 ctask->ready_to_queue = GNUNET_NO;
1647 ctask->fin = GNUNET_NO;
1651 ctask = (struct ProxyCurlTask *) *con_cls;
1653 if (GNUNET_YES != ctask->ready_to_queue)
1654 return MHD_YES; /* wait longer */
1656 if (GNUNET_YES == ctask->fin)
1659 ctask->fin = GNUNET_YES;
1660 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1661 "MHD: Queueing response for %s\n", ctask->url);
1662 ret = MHD_queue_response (con, ctask->curl_response_code, ctask->response);
1663 run_mhd_now (ctask->mhd);
1674 struct MhdHttpList *hd;
1676 for (hd=mhd_httpd_head; hd != NULL; hd = hd->next)
1684 * @param hd the daemon to run
1687 run_httpd (struct MhdHttpList *hd)
1692 struct GNUNET_NETWORK_FDSet *wrs;
1693 struct GNUNET_NETWORK_FDSet *wws;
1694 struct GNUNET_NETWORK_FDSet *wes;
1697 unsigned MHD_LONG_LONG timeout;
1698 struct GNUNET_TIME_Relative tv;
1703 wrs = GNUNET_NETWORK_fdset_create ();
1704 wes = GNUNET_NETWORK_fdset_create ();
1705 wws = GNUNET_NETWORK_fdset_create ();
1707 GNUNET_assert (MHD_YES == MHD_get_fdset (hd->daemon, &rs, &ws, &es, &max));
1710 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1711 "MHD fds: max=%d\n", max);
1713 haveto = MHD_get_timeout (hd->daemon, &timeout);
1715 if (haveto == MHD_YES)
1716 tv.rel_value = (uint64_t) timeout;
1718 tv = GNUNET_TIME_UNIT_FOREVER_REL;
1719 GNUNET_NETWORK_fdset_copy_native (wrs, &rs, max + 1);
1720 GNUNET_NETWORK_fdset_copy_native (wws, &ws, max + 1);
1721 GNUNET_NETWORK_fdset_copy_native (wes, &es, max + 1);
1723 if (hd->httpd_task != GNUNET_SCHEDULER_NO_TASK)
1724 GNUNET_SCHEDULER_cancel (hd->httpd_task);
1726 GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_HIGH,
1729 GNUNET_NETWORK_fdset_destroy (wrs);
1730 GNUNET_NETWORK_fdset_destroy (wws);
1731 GNUNET_NETWORK_fdset_destroy (wes);
1736 * Task run whenever HTTP server operations are pending.
1739 * @param tc sched context
1742 do_httpd (void *cls,
1743 const struct GNUNET_SCHEDULER_TaskContext *tc)
1745 struct MhdHttpList *hd = cls;
1747 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1748 "MHD: Main loop\n");
1749 hd->httpd_task = GNUNET_SCHEDULER_NO_TASK;
1750 MHD_run (hd->daemon);
1757 * Read data from socket
1759 * @param cls the closure
1760 * @param tc scheduler context
1763 do_read (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1766 * Read from remote end
1768 * @param cls closure
1769 * @param tc scheduler context
1772 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
1775 * Write data to remote socket
1777 * @param cls the closure
1778 * @param tc scheduler context
1781 do_write_remote (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1783 struct Socks5Request *s5r = cls;
1786 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
1788 if ((NULL != tc->read_ready) &&
1789 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->remote_sock)) &&
1790 ((len = GNUNET_NETWORK_socket_send (s5r->remote_sock, s5r->rbuf,
1793 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1794 "Successfully sent %d bytes to remote socket\n",
1799 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write remote");
1801 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
1802 GNUNET_SCHEDULER_cancel (s5r->rtask);
1803 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
1804 GNUNET_SCHEDULER_cancel (s5r->wtask);
1805 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
1806 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
1807 GNUNET_NETWORK_socket_close (s5r->remote_sock);
1808 GNUNET_NETWORK_socket_close (s5r->sock);
1814 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1821 * Clean up s5r handles
1823 * @param s5r the handle to destroy
1826 cleanup_s5r (struct Socks5Request *s5r)
1828 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
1829 GNUNET_SCHEDULER_cancel (s5r->rtask);
1830 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
1831 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
1832 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
1833 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
1835 if (NULL != s5r->remote_sock)
1836 GNUNET_NETWORK_socket_close (s5r->remote_sock);
1837 if ((NULL != s5r->sock) && (s5r->cleanup_sock == GNUNET_YES))
1838 GNUNET_NETWORK_socket_close (s5r->sock);
1844 * Write data to socket
1846 * @param cls the closure
1847 * @param tc scheduler context
1850 do_write (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1852 struct Socks5Request *s5r = cls;
1855 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
1857 if ((NULL != tc->read_ready) &&
1858 (GNUNET_NETWORK_fdset_isset (tc->write_ready, s5r->sock)) &&
1859 ((len = GNUNET_NETWORK_socket_send (s5r->sock, s5r->wbuf,
1862 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1863 "Successfully sent %d bytes to socket\n",
1869 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "write");
1870 s5r->cleanup = GNUNET_YES;
1871 s5r->cleanup_sock = GNUNET_YES;
1877 if (GNUNET_YES == s5r->cleanup)
1883 if ((s5r->state == SOCKS5_DATA_TRANSFER) &&
1884 (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK))
1886 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
1888 &do_read_remote, s5r);
1892 * Read from remote end
1894 * @param cls closure
1895 * @param tc scheduler context
1898 do_read_remote (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1900 struct Socks5Request *s5r = cls;
1902 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
1905 if ((NULL != tc->write_ready) &&
1906 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->remote_sock)) &&
1907 (s5r->wbuf_len = GNUNET_NETWORK_socket_recv (s5r->remote_sock, s5r->wbuf,
1908 sizeof (s5r->wbuf))))
1910 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1911 "Successfully read %d bytes from remote socket\n",
1916 if (s5r->wbuf_len == 0)
1917 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1918 "0 bytes received from remote... graceful shutdown!\n");
1919 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
1920 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
1921 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
1922 GNUNET_SCHEDULER_cancel (s5r->rtask);
1924 GNUNET_NETWORK_socket_close (s5r->remote_sock);
1925 s5r->remote_sock = NULL;
1926 GNUNET_NETWORK_socket_close (s5r->sock);
1932 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
1940 * Adds a socket to MHD
1942 * @param h the handle to the socket to add
1943 * @param daemon the daemon to add the fd to
1944 * @return whatever MHD_add_connection returns
1947 add_handle_to_mhd (struct GNUNET_NETWORK_Handle *h, struct MHD_Daemon *daemon)
1950 struct sockaddr *addr;
1953 fd = dup (GNUNET_NETWORK_get_fd (h));
1954 addr = GNUNET_NETWORK_get_addr (h);
1955 len = GNUNET_NETWORK_get_addrlen (h);
1957 return MHD_add_connection (daemon, fd, addr, len);
1961 * Read file in filename
1963 * @param filename file to read
1964 * @param size pointer where filesize is stored
1968 load_file (const char* filename,
1975 GNUNET_DISK_file_size (filename, &fsize,
1976 GNUNET_YES, GNUNET_YES))
1978 if (fsize > MAX_PEM_SIZE)
1980 *size = (unsigned int) fsize;
1981 buffer = GNUNET_malloc (*size);
1982 if (fsize != GNUNET_DISK_fn_read (filename, buffer, (size_t) fsize))
1984 GNUNET_free (buffer);
1992 * Load PEM key from file
1994 * @param key where to store the data
1995 * @param keyfile path to the PEM file
1996 * @return GNUNET_OK on success
1999 load_key_from_file (gnutls_x509_privkey_t key, const char* keyfile)
2001 gnutls_datum_t key_data;
2004 key_data.data = (unsigned char*) load_file (keyfile, &key_data.size);
2005 ret = gnutls_x509_privkey_import (key, &key_data,
2006 GNUTLS_X509_FMT_PEM);
2007 if (GNUTLS_E_SUCCESS != ret)
2009 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2010 _("Unable to import private key from file `%s'\n"),
2014 GNUNET_free (key_data.data);
2015 return (GNUTLS_E_SUCCESS != ret) ? GNUNET_SYSERR : GNUNET_OK;
2020 * Load cert from file
2022 * @param crt struct to store data in
2023 * @param certfile path to pem file
2024 * @return GNUNET_OK on success
2027 load_cert_from_file (gnutls_x509_crt_t crt, char* certfile)
2029 gnutls_datum_t cert_data;
2030 cert_data.data = NULL;
2033 cert_data.data = (unsigned char*) load_file (certfile, &cert_data.size);
2034 ret = gnutls_x509_crt_import (crt, &cert_data,
2035 GNUTLS_X509_FMT_PEM);
2036 if (GNUTLS_E_SUCCESS != ret)
2038 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2039 _("Unable to import certificate %s\n"), certfile);
2042 GNUNET_free (cert_data.data);
2043 return (GNUTLS_E_SUCCESS != ret) ? GNUNET_SYSERR : GNUNET_OK;
2048 * Generate new certificate for specific name
2050 * @param name the subject name to generate a cert for
2051 * @return a struct holding the PEM data
2053 static struct ProxyGNSCertificate *
2054 generate_gns_certificate (const char *name)
2058 unsigned int serial;
2059 size_t key_buf_size;
2060 size_t cert_buf_size;
2061 gnutls_x509_crt_t request;
2065 ret = gnutls_x509_crt_init (&request);
2067 if (GNUTLS_E_SUCCESS != ret)
2072 ret = gnutls_x509_crt_set_key (request, proxy_ca.key);
2074 if (GNUTLS_E_SUCCESS != ret)
2079 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Generating cert\n");
2081 struct ProxyGNSCertificate *pgc =
2082 GNUNET_malloc (sizeof (struct ProxyGNSCertificate));
2084 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Adding DNs\n");
2086 gnutls_x509_crt_set_dn_by_oid (request, GNUTLS_OID_X520_COUNTRY_NAME,
2089 gnutls_x509_crt_set_dn_by_oid (request, GNUTLS_OID_X520_ORGANIZATION_NAME,
2092 gnutls_x509_crt_set_dn_by_oid (request, GNUTLS_OID_X520_COMMON_NAME,
2093 0, name, strlen (name));
2095 ret = gnutls_x509_crt_set_version (request, 3);
2097 ret = gnutls_rnd (GNUTLS_RND_NONCE, &serial, sizeof (serial));
2099 etime = time (NULL);
2100 tm_data = localtime (&etime);
2103 ret = gnutls_x509_crt_set_serial (request,
2107 ret = gnutls_x509_crt_set_activation_time (request,
2110 etime = mktime (tm_data);
2117 ret = gnutls_x509_crt_set_expiration_time (request,
2119 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Signing...\n");
2121 ret = gnutls_x509_crt_sign (request, proxy_ca.cert, proxy_ca.key);
2123 key_buf_size = sizeof (pgc->key);
2124 cert_buf_size = sizeof (pgc->cert);
2126 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Exporting certificate...\n");
2128 gnutls_x509_crt_export (request, GNUTLS_X509_FMT_PEM,
2129 pgc->cert, &cert_buf_size);
2131 gnutls_x509_privkey_export (proxy_ca.key, GNUTLS_X509_FMT_PEM,
2132 pgc->key, &key_buf_size);
2135 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Cleaning up\n");
2136 gnutls_x509_crt_deinit (request);
2144 * Accept policy for mhdaemons
2147 * @param addr the sockaddr
2148 * @param addrlen the sockaddr length
2149 * @return MHD_NO if sockaddr is wrong or #conns too high
2152 accept_cb (void* cls, const struct sockaddr *addr, socklen_t addrlen)
2154 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2155 "In MHD accept policy cb\n");
2159 if (addr->sa_family == AF_UNIX)
2163 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2164 "Connection accepted\n");
2171 * Adds a socket to an SSL MHD instance
2172 * It is important the the domain name is
2173 * correct. In most cases we need to start a new daemon
2175 * @param h the handle to add to a daemon
2176 * @param domain the domain the ssl daemon has to serve
2177 * @return MHD_YES on success
2180 add_handle_to_ssl_mhd (struct GNUNET_NETWORK_Handle *h, const char* domain)
2182 struct MhdHttpList *hd = NULL;
2183 struct ProxyGNSCertificate *pgc;
2184 struct NetworkHandleList *nh;
2186 for (hd = mhd_httpd_head; NULL != hd; hd = hd->next)
2187 if (0 == strcmp (hd->domain, domain))
2192 pgc = generate_gns_certificate (domain);
2194 hd = GNUNET_malloc (sizeof (struct MhdHttpList));
2195 hd->is_ssl = GNUNET_YES;
2196 strcpy (hd->domain, domain);
2197 hd->proxy_cert = pgc;
2200 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2201 "No previous SSL instance found... starting new one for %s\n",
2204 hd->daemon = MHD_start_daemon (MHD_USE_DEBUG | MHD_USE_SSL, 4444,
2206 &create_response, hd,
2207 MHD_OPTION_LISTEN_SOCKET, GNUNET_NETWORK_get_fd (mhd_unix_socket),
2208 MHD_OPTION_CONNECTION_LIMIT, MHD_MAX_CONNECTIONS,
2209 MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
2210 MHD_OPTION_NOTIFY_COMPLETED,
2212 MHD_OPTION_HTTPS_MEM_KEY, pgc->key,
2213 MHD_OPTION_HTTPS_MEM_CERT, pgc->cert,
2216 GNUNET_assert (hd->daemon != NULL);
2217 hd->httpd_task = GNUNET_SCHEDULER_NO_TASK;
2219 GNUNET_CONTAINER_DLL_insert (mhd_httpd_head, mhd_httpd_tail, hd);
2222 nh = GNUNET_malloc (sizeof (struct NetworkHandleList));
2225 GNUNET_CONTAINER_DLL_insert (hd->socket_handles_head,
2226 hd->socket_handles_tail,
2229 return add_handle_to_mhd (h, hd->daemon);
2235 * Read data from incoming connection
2237 * @param cls the closure
2238 * @param tc the scheduler context
2241 do_read (void* cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2243 struct Socks5Request *s5r = cls;
2244 struct socks5_client_hello *c_hello;
2245 struct socks5_server_hello *s_hello;
2246 struct socks5_client_request *c_req;
2247 struct socks5_server_response *s_resp;
2253 struct hostent *phost;
2255 struct sockaddr_in remote_addr;
2256 struct in_addr *r_sin_addr;
2258 struct NetworkHandleList *nh;
2260 s5r->rtask = GNUNET_SCHEDULER_NO_TASK;
2262 if ((NULL != tc->write_ready) &&
2263 (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->sock)) &&
2264 (s5r->rbuf_len = GNUNET_NETWORK_socket_recv (s5r->sock, s5r->rbuf,
2265 sizeof (s5r->rbuf))))
2267 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2268 "Successfully read %d bytes from socket\n",
2273 if (s5r->rbuf_len != 0)
2274 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "read");
2276 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "client disco!\n");
2278 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
2279 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
2280 if (s5r->wtask != GNUNET_SCHEDULER_NO_TASK)
2281 GNUNET_SCHEDULER_cancel (s5r->wtask);
2282 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
2283 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
2284 GNUNET_NETWORK_socket_close (s5r->remote_sock);
2285 GNUNET_NETWORK_socket_close (s5r->sock);
2290 if (s5r->state == SOCKS5_INIT)
2292 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2294 c_hello = (struct socks5_client_hello*)&s5r->rbuf;
2296 GNUNET_assert (c_hello->version == SOCKS_VERSION_5);
2298 s_hello = (struct socks5_server_hello*)&s5r->wbuf;
2299 s5r->wbuf_len = sizeof( struct socks5_server_hello );
2301 s_hello->version = c_hello->version;
2302 s_hello->auth_method = SOCKS_AUTH_NONE;
2304 /* Write response to client */
2305 s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2309 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2313 s5r->state = SOCKS5_REQUEST;
2317 if (s5r->state == SOCKS5_REQUEST)
2319 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2320 "Processing SOCKS5 request\n");
2321 c_req = (struct socks5_client_request*)&s5r->rbuf;
2322 s_resp = (struct socks5_server_response*)&s5r->wbuf;
2323 //Only 10byte for ipv4 response!
2324 s5r->wbuf_len = 10;//sizeof (struct socks5_server_response);
2326 GNUNET_assert (c_req->addr_type == 3);
2328 dom_len = *((uint8_t*)(&(c_req->addr_type) + 1));
2329 memset(domain, 0, sizeof(domain));
2330 strncpy(domain, (char*)(&(c_req->addr_type) + 2), dom_len);
2331 req_port = *((uint16_t*)(&(c_req->addr_type) + 2 + dom_len));
2333 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2334 "Requested connection is %s:%d\n",
2338 if (is_tld (domain, GNUNET_GNS_TLD) ||
2339 is_tld (domain, GNUNET_GNS_TLD_ZKEY))
2341 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2342 "Requested connection is gnunet tld\n",
2346 if (ntohs(req_port) == HTTPS_PORT)
2348 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2349 "Requested connection is HTTPS\n");
2350 ret = add_handle_to_ssl_mhd ( s5r->sock, domain );
2352 else if (NULL != httpd)
2354 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2355 "Requested connection is HTTP\n");
2356 nh = GNUNET_malloc (sizeof (struct NetworkHandleList));
2359 GNUNET_CONTAINER_DLL_insert (mhd_httpd_head->socket_handles_head,
2360 mhd_httpd_head->socket_handles_tail,
2363 ret = add_handle_to_mhd ( s5r->sock, httpd );
2368 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2369 _("Failed to start HTTP server\n"));
2370 s_resp->version = 0x05;
2371 s_resp->reply = 0x01;
2372 s5r->cleanup = GNUNET_YES;
2373 s5r->cleanup_sock = GNUNET_YES;
2375 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2381 /* Signal success */
2382 s_resp->version = 0x05;
2383 s_resp->reply = 0x00;
2384 s_resp->reserved = 0x00;
2385 s_resp->addr_type = 0x01;
2387 s5r->cleanup = GNUNET_YES;
2388 s5r->cleanup_sock = GNUNET_NO;
2390 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2398 phost = (struct hostent*)gethostbyname (domain);
2401 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2402 "Resolve %s error!\n", domain );
2403 s_resp->version = 0x05;
2404 s_resp->reply = 0x01;
2405 s5r->cleanup = GNUNET_YES;
2406 s5r->cleanup_sock = GNUNET_YES;
2408 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2414 s5r->remote_sock = GNUNET_NETWORK_socket_create (AF_INET,
2417 r_sin_addr = (struct in_addr*)(phost->h_addr);
2418 remote_ip = r_sin_addr->s_addr;
2419 memset(&remote_addr, 0, sizeof(remote_addr));
2420 remote_addr.sin_family = AF_INET;
2421 #if HAVE_SOCKADDR_IN_SIN_LEN
2422 remote_addr.sin_len = sizeof (remote_addr);
2424 remote_addr.sin_addr.s_addr = remote_ip;
2425 remote_addr.sin_port = req_port;
2427 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2428 "target server: %s:%u\n", inet_ntoa(remote_addr.sin_addr),
2432 GNUNET_NETWORK_socket_connect ( s5r->remote_sock,
2433 (const struct sockaddr*)&remote_addr,
2434 sizeof (remote_addr)))
2435 && (errno != EINPROGRESS))
2437 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "connect");
2438 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2439 "socket request error...\n");
2440 s_resp->version = 0x05;
2441 s_resp->reply = 0x01;
2443 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2450 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2451 "new remote connection\n");
2453 s_resp->version = 0x05;
2454 s_resp->reply = 0x00;
2455 s_resp->reserved = 0x00;
2456 s_resp->addr_type = 0x01;
2458 s5r->state = SOCKS5_DATA_TRANSFER;
2461 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2465 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2473 if (s5r->state == SOCKS5_DATA_TRANSFER)
2475 if ((s5r->remote_sock == NULL) || (s5r->rbuf_len == 0))
2477 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2478 "Closing connection to client\n");
2479 if (s5r->rtask != GNUNET_SCHEDULER_NO_TASK)
2480 GNUNET_SCHEDULER_cancel (s5r->rtask);
2481 if (s5r->fwdwtask != GNUNET_SCHEDULER_NO_TASK)
2482 GNUNET_SCHEDULER_cancel (s5r->fwdwtask);
2483 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
2484 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
2485 if (s5r->fwdrtask != GNUNET_SCHEDULER_NO_TASK)
2486 GNUNET_SCHEDULER_cancel (s5r->fwdrtask);
2488 if (s5r->remote_sock != NULL)
2489 GNUNET_NETWORK_socket_close (s5r->remote_sock);
2490 GNUNET_NETWORK_socket_close (s5r->sock);
2495 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2496 "forwarding %d bytes from client\n", s5r->rbuf_len);
2499 GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2501 &do_write_remote, s5r);
2503 if (s5r->fwdrtask == GNUNET_SCHEDULER_NO_TASK)
2506 GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2508 &do_read_remote, s5r);
2514 //GNUNET_CONTAINER_DLL_remove (s5conns.head, s5conns.tail, s5r);
2520 * Accept new incoming connections
2522 * @param cls the closure
2523 * @param tc the scheduler context
2526 do_accept (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2528 struct GNUNET_NETWORK_Handle *s;
2529 struct Socks5Request *s5r;
2531 ltask = GNUNET_SCHEDULER_NO_TASK;
2532 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
2535 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2539 s = GNUNET_NETWORK_socket_accept (lsock, NULL, NULL);
2543 GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "accept");
2547 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2548 "Got an inbound connection, waiting for data\n");
2550 s5r = GNUNET_malloc (sizeof (struct Socks5Request));
2552 s5r->state = SOCKS5_INIT;
2553 s5r->wtask = GNUNET_SCHEDULER_NO_TASK;
2554 s5r->fwdwtask = GNUNET_SCHEDULER_NO_TASK;
2555 s5r->fwdrtask = GNUNET_SCHEDULER_NO_TASK;
2556 s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2559 //GNUNET_CONTAINER_DLL_insert (s5conns.head, s5conns.tail, s5r);
2564 * Task run on shutdown
2566 * @param cls closure
2567 * @param tc task context
2570 do_shutdown (void *cls,
2571 const struct GNUNET_SCHEDULER_TaskContext *tc)
2574 struct MhdHttpList *hd;
2575 struct MhdHttpList *tmp_hd;
2576 struct NetworkHandleList *nh;
2577 struct NetworkHandleList *tmp_nh;
2578 struct ProxyCurlTask *ctask;
2579 struct ProxyCurlTask *ctask_tmp;
2581 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
2582 "Shutting down...\n");
2584 gnutls_global_deinit ();
2586 if (GNUNET_SCHEDULER_NO_TASK != curl_download_task)
2588 GNUNET_SCHEDULER_cancel (curl_download_task);
2589 curl_download_task = GNUNET_SCHEDULER_NO_TASK;
2592 for (hd = mhd_httpd_head; hd != NULL; hd = tmp_hd)
2596 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2597 "Stopping daemon\n");
2599 if (GNUNET_SCHEDULER_NO_TASK != hd->httpd_task)
2601 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2602 "Stopping select task %d\n",
2604 GNUNET_SCHEDULER_cancel (hd->httpd_task);
2605 hd->httpd_task = GNUNET_SCHEDULER_NO_TASK;
2608 if (NULL != hd->daemon)
2610 MHD_stop_daemon (hd->daemon);
2614 for (nh = hd->socket_handles_head; nh != NULL; nh = tmp_nh)
2618 GNUNET_NETWORK_socket_close (nh->h);
2623 if (NULL != hd->proxy_cert)
2625 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2626 "Free certificate\n");
2627 GNUNET_free (hd->proxy_cert);
2633 for (ctask=ctasks_head; ctask != NULL; ctask=ctask_tmp)
2635 ctask_tmp = ctask->next;
2637 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2638 "Cleaning up cURL task\n");
2640 if (ctask->curl != NULL)
2641 curl_easy_cleanup (ctask->curl);
2643 if (NULL != ctask->headers)
2644 curl_slist_free_all (ctask->headers);
2645 if (NULL != ctask->resolver)
2646 curl_slist_free_all (ctask->resolver);
2648 if (NULL != ctask->response)
2649 MHD_destroy_response (ctask->response);
2652 GNUNET_free (ctask);
2655 GNUNET_GNS_disconnect (gns_handle);
2660 * Compiles a regex for us
2662 * @param re ptr to re struct
2663 * @param rt the expression to compile
2664 * @return 0 on success
2667 compile_regex (regex_t *re, const char* rt)
2672 status = regcomp (re, rt, REG_EXTENDED|REG_NEWLINE);
2675 regerror (status, re, err, 1024);
2676 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2677 "Regex error compiling '%s': %s\n", rt, err);
2685 * Loads the users local zone key
2687 * @return GNUNET_YES on success
2690 load_local_zone_key (const struct GNUNET_CONFIGURATION_Handle *cfg)
2693 struct GNUNET_CRYPTO_RsaPrivateKey *key = NULL;
2694 struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pkey;
2695 struct GNUNET_CRYPTO_ShortHashCode *zone = NULL;
2696 struct GNUNET_CRYPTO_ShortHashAsciiEncoded zonename;
2698 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns",
2699 "ZONEKEY", &keyfile))
2701 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2702 "Unable to load zone key config value!\n");
2706 if (GNUNET_NO == GNUNET_DISK_file_test (keyfile))
2708 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2709 "Unable to load zone key %s!\n", keyfile);
2710 GNUNET_free(keyfile);
2714 key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
2715 GNUNET_CRYPTO_rsa_key_get_public (key, &pkey);
2716 GNUNET_CRYPTO_short_hash(&pkey,
2717 sizeof(struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
2719 zone = &local_gns_zone;
2720 GNUNET_CRYPTO_short_hash_to_enc (zone, &zonename);
2721 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2722 "Using zone: %s!\n", &zonename);
2723 GNUNET_CRYPTO_rsa_key_free(key);
2724 GNUNET_free(keyfile);
2726 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns",
2727 "PRIVATE_ZONEKEY", &keyfile))
2729 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2730 "Unable to load private zone key config value!\n");
2734 if (GNUNET_NO == GNUNET_DISK_file_test (keyfile))
2736 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2737 "Unable to load private zone key %s!\n", keyfile);
2738 GNUNET_free(keyfile);
2742 key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
2743 GNUNET_CRYPTO_rsa_key_get_public (key, &pkey);
2744 GNUNET_CRYPTO_short_hash(&pkey,
2745 sizeof(struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
2746 &local_private_zone);
2747 GNUNET_CRYPTO_short_hash_to_enc (zone, &zonename);
2748 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2749 "Using private zone: %s!\n", &zonename);
2750 GNUNET_CRYPTO_rsa_key_free(key);
2751 GNUNET_free(keyfile);
2753 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns",
2754 "SHORTEN_ZONEKEY", &keyfile))
2756 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2757 "Unable to load shorten zone key config value!\n");
2761 if (GNUNET_NO == GNUNET_DISK_file_test (keyfile))
2763 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2764 "Unable to load shorten zone key %s!\n", keyfile);
2765 GNUNET_free(keyfile);
2769 key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
2770 GNUNET_CRYPTO_rsa_key_get_public (key, &pkey);
2771 GNUNET_CRYPTO_short_hash(&pkey,
2772 sizeof(struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
2773 &local_shorten_zone);
2774 GNUNET_CRYPTO_short_hash_to_enc (zone, &zonename);
2775 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2776 "Using shorten zone: %s!\n", &zonename);
2777 GNUNET_CRYPTO_rsa_key_free(key);
2778 GNUNET_free(keyfile);
2784 * Main function that will be run
2786 * @param cls closure
2787 * @param args remaining command-line arguments
2788 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
2789 * @param cfg configuration
2792 run (void *cls, char *const *args, const char *cfgfile,
2793 const struct GNUNET_CONFIGURATION_Handle *cfg)
2795 struct sockaddr_in sa;
2796 struct MhdHttpList *hd;
2797 struct sockaddr_un mhd_unix_sock_addr;
2799 char* proxy_sockfile;
2800 char* cafile_cfg = NULL;
2805 GNUNET_log (GNUNET_ERROR_TYPE_INFO,
2808 cafile = cafile_opt;
2812 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns-proxy",
2816 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2817 "Unable to load proxy CA config value!\n");
2818 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2819 "No proxy CA provided!\n");
2822 cafile = cafile_cfg;
2825 gnutls_global_init ();
2827 gnutls_x509_crt_init (&proxy_ca.cert);
2828 gnutls_x509_privkey_init (&proxy_ca.key);
2830 if ( (GNUNET_OK != load_cert_from_file (proxy_ca.cert, cafile)) ||
2831 (GNUNET_OK != load_key_from_file (proxy_ca.key, cafile)) )
2833 // FIXME: release resources...
2837 GNUNET_free_non_null (cafile_cfg);
2839 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2840 "Loading Template\n");
2842 compile_regex (&re_dotplus, (char*) RE_A_HREF);
2844 gns_handle = GNUNET_GNS_connect (cfg);
2846 if (GNUNET_NO == load_local_zone_key (cfg))
2848 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2849 "Unable to load zone!\n");
2853 if (NULL == gns_handle)
2855 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2856 "Unable to connect to GNS!\n");
2860 memset (&sa, 0, sizeof (sa));
2861 sa.sin_family = AF_INET;
2862 sa.sin_port = htons (port);
2863 #if HAVE_SOCKADDR_IN_SIN_LEN
2864 sa.sin_len = sizeof (sa);
2867 lsock = GNUNET_NETWORK_socket_create (AF_INET,
2871 if ((NULL == lsock) ||
2873 GNUNET_NETWORK_socket_bind (lsock, (const struct sockaddr *) &sa,
2876 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2877 "Failed to create listen socket bound to `%s'",
2878 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
2880 GNUNET_NETWORK_socket_close (lsock);
2884 if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock, 5))
2886 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2887 "Failed to listen on socket bound to `%s'",
2888 GNUNET_a2s ((const struct sockaddr *) &sa, sizeof (sa)));
2892 ltask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2893 lsock, &do_accept, NULL);
2898 if (0 != curl_global_init (CURL_GLOBAL_WIN32))
2900 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2901 "cURL global init failed!\n");
2905 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2906 "Proxy listens on port %u\n",
2909 mhd_httpd_head = NULL;
2910 mhd_httpd_tail = NULL;
2911 total_mhd_connections = 0;
2913 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns-proxy",
2917 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2918 "Specify PROXY_UNIXPATH in gns-proxy config section!\n");
2922 mhd_unix_socket = GNUNET_NETWORK_socket_create (AF_UNIX,
2926 if (NULL == mhd_unix_socket)
2928 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2929 "Unable to create unix domain socket!\n");
2933 mhd_unix_sock_addr.sun_family = AF_UNIX;
2934 strcpy (mhd_unix_sock_addr.sun_path, proxy_sockfile);
2937 mhd_unix_sock_addr.sun_path[0] = '\0';
2939 #if HAVE_SOCKADDR_IN_SIN_LEN
2940 mhd_unix_sock_addr.sun_len = (u_char) sizeof (struct sockaddr_un);
2943 len = strlen (proxy_sockfile) + sizeof(AF_UNIX);
2945 GNUNET_free (proxy_sockfile);
2947 if (GNUNET_OK != GNUNET_NETWORK_socket_bind (mhd_unix_socket,
2948 (struct sockaddr*)&mhd_unix_sock_addr,
2951 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2952 "Unable to bind unix domain socket!\n");
2956 if (GNUNET_OK != GNUNET_NETWORK_socket_listen (mhd_unix_socket,
2959 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2960 "Unable to listen on unix domain socket!\n");
2964 hd = GNUNET_malloc (sizeof (struct MhdHttpList));
2965 hd->is_ssl = GNUNET_NO;
2966 strcpy (hd->domain, "");
2967 httpd = MHD_start_daemon (MHD_USE_DEBUG, 4444, //Dummy port
2969 &create_response, hd,
2970 MHD_OPTION_LISTEN_SOCKET, GNUNET_NETWORK_get_fd (mhd_unix_socket),
2971 MHD_OPTION_CONNECTION_LIMIT, MHD_MAX_CONNECTIONS,
2972 MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
2973 MHD_OPTION_NOTIFY_COMPLETED,
2977 GNUNET_assert (httpd != NULL);
2979 hd->httpd_task = GNUNET_SCHEDULER_NO_TASK;
2981 GNUNET_CONTAINER_DLL_insert (mhd_httpd_head, mhd_httpd_tail, hd);
2985 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
2986 &do_shutdown, NULL);
2992 * The main function for gnunet-gns-proxy.
2994 * @param argc number of arguments from the command line
2995 * @param argv command line arguments
2996 * @return 0 ok, 1 on error
2999 main (int argc, char *const *argv)
3001 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
3003 gettext_noop ("listen on specified port"), 1,
3004 &GNUNET_GETOPT_set_string, &port},
3005 {'a', "authority", NULL,
3006 gettext_noop ("pem file to use as CA"), 1,
3007 &GNUNET_GETOPT_set_string, &cafile_opt},
3008 GNUNET_GETOPT_OPTION_END
3013 if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
3017 GNUNET_log_setup ("gnunet-gns-proxy", "WARNING", NULL);
3020 GNUNET_PROGRAM_run (argc, argv, "gnunet-gns-proxy",
3021 _("GNUnet GNS proxy"),
3023 &run, NULL)) ? 0 : 1;
3024 GNUNET_free_non_null ((char*)argv);