src/gns/gnunet-gns-proxy.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2012-2014 GNUnet e.V.
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, USA.
  19 */
  20 /**
  21  * @author Martin Schanzenbach
  22  * @author Christian Grothoff
  23  * @file src/gns/gnunet-gns-proxy.c
  24  * @brief HTTP(S) proxy that rewrites URIs and fakes certificats to make GNS work
  25  *        with legacy browsers
  26  *
  27  * TODO:
  28  * - double-check queueing logic
  29  */
  30 #include "platform.h"
  31 #include <microhttpd.h>
  32 #if HAVE_CURL_CURL_H
  33 #include <curl/curl.h>
  34 #elif HAVE_GNURL_CURL_H
  35 #include <gnurl/curl.h>
  36 #endif
  37 #include <gnutls/gnutls.h>
  38 #include <gnutls/x509.h>
  39 #include <gnutls/abstract.h>
  40 #include <gnutls/crypto.h>
  41 #if HAVE_GNUTLS_DANE
  42 #include <gnutls/dane.h>
  43 #endif
  44 #include <regex.h>
  45 #include "gnunet_util_lib.h"
  46 #include "gnunet_gns_service.h"
  47 #include "gnunet_identity_service.h"
  48 #include "gns.h"
  49
  50
  51 /**
  52  * Default Socks5 listen port.
  53  */
  54 #define GNUNET_GNS_PROXY_PORT 7777
  55
  56 /**
  57  * Maximum supported length for a URI.
  58  * Should die. @deprecated
  59  */
  60 #define MAX_HTTP_URI_LENGTH 2048
  61
  62 /**
  63  * Size of the buffer for the data upload / download.  Must be
  64  * enough for curl, thus CURL_MAX_WRITE_SIZE is needed here (16k).
  65  */
  66 #define IO_BUFFERSIZE CURL_MAX_WRITE_SIZE
  67
  68 /**
  69  * Size of the read/write buffers for Socks.   Uses
  70  * 256 bytes for the hostname (at most), plus a few
  71  * bytes overhead for the messages.
  72  */
  73 #define SOCKS_BUFFERSIZE (256 + 32)
  74
  75 /**
  76  * Port for plaintext HTTP.
  77  */
  78 #define HTTP_PORT 80
  79
  80 /**
  81  * Port for HTTPS.
  82  */
  83 #define HTTPS_PORT 443
  84
  85 /**
  86  * Largest allowed size for a PEM certificate.
  87  */
  88 #define MAX_PEM_SIZE (10 * 1024)
  89
  90 /**
  91  * After how long do we clean up unused MHD SSL/TLS instances?
  92  */
  93 #define MHD_CACHE_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 5)
  94
  95 /**
  96  * After how long do we clean up Socks5 handles that failed to show any activity
  97  * with their respective MHD instance?
  98  */
  99 #define HTTP_HANDSHAKE_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 15)
 100
 101
 102 /**
 103  * Log curl error.
 104  *
 105  * @param level log level
 106  * @param fun name of curl_easy-function that gave the error
 107  * @param rc return code from curl
 108  */
 109 #define LOG_CURL_EASY(level,fun,rc) GNUNET_log(level, _("%s failed at %s:%d: `%s'\n"), fun, __FILE__, __LINE__, curl_easy_strerror (rc))
 110
 111
 112 /* *************** Socks protocol definitions (move to TUN?) ****************** */
 113
 114 /**
 115  * Which SOCKS version do we speak?
 116  */
 117 #define SOCKS_VERSION_5 0x05
 118
 119 /**
 120  * Flag to set for 'no authentication'.
 121  */
 122 #define SOCKS_AUTH_NONE 0
 123
 124
 125 /**
 126  * Commands in Socks5.
 127  */
 128 enum Socks5Commands
 129 {
 130   /**
 131    * Establish TCP/IP stream.
 132    */
 133   SOCKS5_CMD_TCP_STREAM = 1,
 134
 135   /**
 136    * Establish TCP port binding.
 137    */
 138   SOCKS5_CMD_TCP_PORT = 2,
 139
 140   /**
 141    * Establish UDP port binding.
 142    */
 143   SOCKS5_CMD_UDP_PORT = 3
 144 };
 145
 146
 147 /**
 148  * Address types in Socks5.
 149  */
 150 enum Socks5AddressType
 151 {
 152   /**
 153    * IPv4 address.
 154    */
 155   SOCKS5_AT_IPV4 = 1,
 156
 157   /**
 158    * IPv4 address.
 159    */
 160   SOCKS5_AT_DOMAINNAME = 3,
 161
 162   /**
 163    * IPv6 address.
 164    */
 165   SOCKS5_AT_IPV6 = 4
 166
 167 };
 168
 169
 170 /**
 171  * Status codes in Socks5 response.
 172  */
 173 enum Socks5StatusCode
 174 {
 175   SOCKS5_STATUS_REQUEST_GRANTED = 0,
 176   SOCKS5_STATUS_GENERAL_FAILURE = 1,
 177   SOCKS5_STATUS_CONNECTION_NOT_ALLOWED_BY_RULE = 2,
 178   SOCKS5_STATUS_NETWORK_UNREACHABLE = 3,
 179   SOCKS5_STATUS_HOST_UNREACHABLE = 4,
 180   SOCKS5_STATUS_CONNECTION_REFUSED_BY_HOST = 5,
 181   SOCKS5_STATUS_TTL_EXPIRED = 6,
 182   SOCKS5_STATUS_COMMAND_NOT_SUPPORTED = 7,
 183   SOCKS5_STATUS_ADDRESS_TYPE_NOT_SUPPORTED = 8
 184 };
 185
 186
 187 /**
 188  * Client hello in Socks5 protocol.
 189  */
 190 struct Socks5ClientHelloMessage
 191 {
 192   /**
 193    * Should be #SOCKS_VERSION_5.
 194    */
 195   uint8_t version;
 196
 197   /**
 198    * How many authentication methods does the client support.
 199    */
 200   uint8_t num_auth_methods;
 201
 202   /* followed by supported authentication methods, 1 byte per method */
 203
 204 };
 205
 206
 207 /**
 208  * Server hello in Socks5 protocol.
 209  */
 210 struct Socks5ServerHelloMessage
 211 {
 212   /**
 213    * Should be #SOCKS_VERSION_5.
 214    */
 215   uint8_t version;
 216
 217   /**
 218    * Chosen authentication method, for us always #SOCKS_AUTH_NONE,
 219    * which skips the authentication step.
 220    */
 221   uint8_t auth_method;
 222 };
 223
 224
 225 /**
 226  * Client socks request in Socks5 protocol.
 227  */
 228 struct Socks5ClientRequestMessage
 229 {
 230   /**
 231    * Should be #SOCKS_VERSION_5.
 232    */
 233   uint8_t version;
 234
 235   /**
 236    * Command code, we only uspport #SOCKS5_CMD_TCP_STREAM.
 237    */
 238   uint8_t command;
 239
 240   /**
 241    * Reserved, always zero.
 242    */
 243   uint8_t resvd;
 244
 245   /**
 246    * Address type, an `enum Socks5AddressType`.
 247    */
 248   uint8_t addr_type;
 249
 250   /*
 251    * Followed by either an ip4/ipv6 address or a domain name with a
 252    * length field (uint8_t) in front (depending on @e addr_type).
 253    * followed by port number in network byte order (uint16_t).
 254    */
 255 };
 256
 257
 258 /**
 259  * Server response to client requests in Socks5 protocol.
 260  */
 261 struct Socks5ServerResponseMessage
 262 {
 263   /**
 264    * Should be #SOCKS_VERSION_5.
 265    */
 266   uint8_t version;
 267
 268   /**
 269    * Status code, an `enum Socks5StatusCode`
 270    */
 271   uint8_t reply;
 272
 273   /**
 274    * Always zero.
 275    */
 276   uint8_t reserved;
 277
 278   /**
 279    * Address type, an `enum Socks5AddressType`.
 280    */
 281   uint8_t addr_type;
 282
 283   /*
 284    * Followed by either an ip4/ipv6 address or a domain name with a
 285    * length field (uint8_t) in front (depending on @e addr_type).
 286    * followed by port number in network byte order (uint16_t).
 287    */
 288
 289 };
 290
 291
 292
 293 /* *********************** Datastructures for HTTP handling ****************** */
 294
 295 /**
 296  * A structure for CA cert/key
 297  */
 298 struct ProxyCA
 299 {
 300   /**
 301    * The certificate
 302    */
 303   gnutls_x509_crt_t cert;
 304
 305   /**
 306    * The private key
 307    */
 308   gnutls_x509_privkey_t key;
 309 };
 310
 311
 312 /**
 313  * Structure for GNS certificates
 314  */
 315 struct ProxyGNSCertificate
 316 {
 317   /**
 318    * The certificate as PEM
 319    */
 320   char cert[MAX_PEM_SIZE];
 321
 322   /**
 323    * The private key as PEM
 324    */
 325   char key[MAX_PEM_SIZE];
 326 };
 327
 328
 329
 330 /**
 331  * A structure for all running Httpds
 332  */
 333 struct MhdHttpList
 334 {
 335   /**
 336    * DLL for httpds
 337    */
 338   struct MhdHttpList *prev;
 339
 340   /**
 341    * DLL for httpds
 342    */
 343   struct MhdHttpList *next;
 344
 345   /**
 346    * the domain name to server (only important for SSL)
 347    */
 348   char *domain;
 349
 350   /**
 351    * The daemon handle
 352    */
 353   struct MHD_Daemon *daemon;
 354
 355   /**
 356    * Optional proxy certificate used
 357    */
 358   struct ProxyGNSCertificate *proxy_cert;
 359
 360   /**
 361    * The task ID
 362    */
 363   struct GNUNET_SCHEDULER_Task *httpd_task;
 364
 365   /**
 366    * is this an ssl daemon?
 367    */
 368   int is_ssl;
 369
 370 };
 371
 372
 373 /* ***************** Datastructures for Socks handling **************** */
 374
 375
 376 /**
 377  * The socks phases.
 378  */
 379 enum SocksPhase
 380 {
 381   /**
 382    * We're waiting to get the client hello.
 383    */
 384   SOCKS5_INIT,
 385
 386   /**
 387    * We're waiting to get the initial request.
 388    */
 389   SOCKS5_REQUEST,
 390
 391   /**
 392    * We are currently resolving the destination.
 393    */
 394   SOCKS5_RESOLVING,
 395
 396   /**
 397    * We're in transfer mode.
 398    */
 399   SOCKS5_DATA_TRANSFER,
 400
 401   /**
 402    * Finish writing the write buffer, then clean up.
 403    */
 404   SOCKS5_WRITE_THEN_CLEANUP,
 405
 406   /**
 407    * Socket has been passed to MHD, do not close it anymore.
 408    */
 409   SOCKS5_SOCKET_WITH_MHD,
 410
 411   /**
 412    * We've started receiving upload data from MHD.
 413    */
 414   SOCKS5_SOCKET_UPLOAD_STARTED,
 415
 416   /**
 417    * We've finished receiving upload data from MHD.
 418    */
 419   SOCKS5_SOCKET_UPLOAD_DONE,
 420
 421   /**
 422    * We've finished uploading data via CURL and can now download.
 423    */
 424   SOCKS5_SOCKET_DOWNLOAD_STARTED,
 425
 426   /**
 427    * We've finished receiving download data from cURL.
 428    */
 429   SOCKS5_SOCKET_DOWNLOAD_DONE
 430 };
 431
 432
 433
 434 /**
 435  * A structure for socks requests
 436  */
 437 struct Socks5Request
 438 {
 439
 440   /**
 441    * DLL.
 442    */
 443   struct Socks5Request *next;
 444
 445   /**
 446    * DLL.
 447    */
 448   struct Socks5Request *prev;
 449
 450   /**
 451    * The client socket
 452    */
 453   struct GNUNET_NETWORK_Handle *sock;
 454
 455   /**
 456    * Handle to GNS lookup, during #SOCKS5_RESOLVING phase.
 457    */
 458   struct GNUNET_GNS_LookupRequest *gns_lookup;
 459
 460   /**
 461    * Client socket read task
 462    */
 463   struct GNUNET_SCHEDULER_Task * rtask;
 464
 465   /**
 466    * Client socket write task
 467    */
 468   struct GNUNET_SCHEDULER_Task * wtask;
 469
 470   /**
 471    * Timeout task
 472    */
 473   struct GNUNET_SCHEDULER_Task * timeout_task;
 474
 475   /**
 476    * Read buffer
 477    */
 478   char rbuf[SOCKS_BUFFERSIZE];
 479
 480   /**
 481    * Write buffer
 482    */
 483   char wbuf[SOCKS_BUFFERSIZE];
 484
 485   /**
 486    * Buffer we use for moving data between MHD and curl (in both directions).
 487    */
 488   char io_buf[IO_BUFFERSIZE];
 489
 490   /**
 491    * MHD HTTP instance handling this request, NULL for none.
 492    */
 493   struct MhdHttpList *hd;
 494
 495   /**
 496    * MHD response object for this request.
 497    */
 498   struct MHD_Response *response;
 499
 500   /**
 501    * the domain name to server (only important for SSL)
 502    */
 503   char *domain;
 504
 505   /**
 506    * DNS Legacy Host Name as given by GNS, NULL if not given.
 507    */
 508   char *leho;
 509
 510   /**
 511    * Payload of the (last) DANE record encountered.
 512    */
 513   char *dane_data;
 514
 515   /**
 516    * The URL to fetch
 517    */
 518   char *url;
 519
 520   /**
 521    * Handle to cURL
 522    */
 523   CURL *curl;
 524
 525   /**
 526    * HTTP request headers for the curl request.
 527    */
 528   struct curl_slist *headers;
 529
 530   /**
 531    * DNS->IP mappings resolved through GNS
 532    */
 533   struct curl_slist *hosts;
 534
 535   /**
 536    * HTTP response code to give to MHD for the response.
 537    */
 538   unsigned int response_code;
 539
 540   /**
 541    * Number of bytes in @e dane_data.
 542    */
 543   size_t dane_data_len;
 544
 545   /**
 546    * Number of bytes already in read buffer
 547    */
 548   size_t rbuf_len;
 549
 550   /**
 551    * Number of bytes already in write buffer
 552    */
 553   size_t wbuf_len;
 554
 555   /**
 556    * Number of bytes already in the IO buffer.
 557    */
 558   size_t io_len;
 559
 560   /**
 561    * Once known, what's the target address for the connection?
 562    */
 563   struct sockaddr_storage destination_address;
 564
 565   /**
 566    * The socks state
 567    */
 568   enum SocksPhase state;
 569
 570   /**
 571    * Desired destination port.
 572    */
 573   uint16_t port;
 574
 575 };
 576
 577
 578
 579 /* *********************** Globals **************************** */
 580
 581
 582 /**
 583  * The port the proxy is running on (default 7777)
 584  */
 585 static unsigned long port = GNUNET_GNS_PROXY_PORT;
 586
 587 /**
 588  * The CA file (pem) to use for the proxy CA
 589  */
 590 static char *cafile_opt;
 591
 592 /**
 593  * The listen socket of the proxy for IPv4
 594  */
 595 static struct GNUNET_NETWORK_Handle *lsock4;
 596
 597 /**
 598  * The listen socket of the proxy for IPv6
 599  */
 600 static struct GNUNET_NETWORK_Handle *lsock6;
 601
 602 /**
 603  * The listen task ID for IPv4
 604  */
 605 static struct GNUNET_SCHEDULER_Task * ltask4;
 606
 607 /**
 608  * The listen task ID for IPv6
 609  */
 610 static struct GNUNET_SCHEDULER_Task * ltask6;
 611
 612 /**
 613  * The cURL download task (curl multi API).
 614  */
 615 static struct GNUNET_SCHEDULER_Task * curl_download_task;
 616
 617 /**
 618  * The cURL multi handle
 619  */
 620 static CURLM *curl_multi;
 621
 622 /**
 623  * Handle to the GNS service
 624  */
 625 static struct GNUNET_GNS_Handle *gns_handle;
 626
 627 /**
 628  * DLL for http/https daemons
 629  */
 630 static struct MhdHttpList *mhd_httpd_head;
 631
 632 /**
 633  * DLL for http/https daemons
 634  */
 635 static struct MhdHttpList *mhd_httpd_tail;
 636
 637 /**
 638  * Daemon for HTTP (we have one per SSL certificate, and then one for
 639  * all HTTP connections; this is the one for HTTP, not HTTPS).
 640  */
 641 static struct MhdHttpList *httpd;
 642
 643 /**
 644  * DLL of active socks requests.
 645  */
 646 static struct Socks5Request *s5r_head;
 647
 648 /**
 649  * DLL of active socks requests.
 650  */
 651 static struct Socks5Request *s5r_tail;
 652
 653 /**
 654  * The users local GNS master zone
 655  */
 656 static struct GNUNET_CRYPTO_EcdsaPublicKey local_gns_zone;
 657
 658 /**
 659  * The users local shorten zone
 660  */
 661 static struct GNUNET_CRYPTO_EcdsaPrivateKey local_shorten_zone;
 662
 663 /**
 664  * Is shortening enabled?
 665  */
 666 static int do_shorten;
 667
 668 /**
 669  * The CA for SSL certificate generation
 670  */
 671 static struct ProxyCA proxy_ca;
 672
 673 /**
 674  * Response we return on cURL failures.
 675  */
 676 static struct MHD_Response *curl_failure_response;
 677
 678 /**
 679  * Connection to identity service.
 680  */
 681 static struct GNUNET_IDENTITY_Handle *identity;
 682
 683 /**
 684  * Request for our ego.
 685  */
 686 static struct GNUNET_IDENTITY_Operation *id_op;
 687
 688 /**
 689  * Our configuration.
 690  */
 691 static const struct GNUNET_CONFIGURATION_Handle *cfg;
 692
 693
 694 /* ************************* Global helpers ********************* */
 695
 696
 697 /**
 698  * Run MHD now, we have extra data ready for the callback.
 699  *
 700  * @param hd the daemon to run now.
 701  */
 702 static void
 703 run_mhd_now (struct MhdHttpList *hd);
 704
 705
 706 /**
 707  * Clean up s5r handles.
 708  *
 709  * @param s5r the handle to destroy
 710  */
 711 static void
 712 cleanup_s5r (struct Socks5Request *s5r)
 713 {
 714   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 715               "Cleaning up socks request\n");
 716   if (NULL != s5r->curl)
 717   {
 718     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 719                 "Cleaning up cURL handle\n");
 720     curl_multi_remove_handle (curl_multi, s5r->curl);
 721     curl_easy_cleanup (s5r->curl);
 722     s5r->curl = NULL;
 723   }
 724   curl_slist_free_all (s5r->headers);
 725   if (NULL != s5r->hosts)
 726   {
 727     curl_slist_free_all (s5r->hosts);
 728   }
 729   if ( (NULL != s5r->response) &&
 730        (curl_failure_response != s5r->response) )
 731     MHD_destroy_response (s5r->response);
 732   if (NULL != s5r->rtask)
 733     GNUNET_SCHEDULER_cancel (s5r->rtask);
 734   if (NULL != s5r->timeout_task)
 735     GNUNET_SCHEDULER_cancel (s5r->timeout_task);
 736   if (NULL != s5r->wtask)
 737     GNUNET_SCHEDULER_cancel (s5r->wtask);
 738   if (NULL != s5r->gns_lookup)
 739     GNUNET_GNS_lookup_cancel (s5r->gns_lookup);
 740   if (NULL != s5r->sock)
 741   {
 742     if (SOCKS5_SOCKET_WITH_MHD <= s5r->state)
 743       GNUNET_NETWORK_socket_free_memory_only_ (s5r->sock);
 744     else
 745       GNUNET_NETWORK_socket_close (s5r->sock);
 746   }
 747   GNUNET_CONTAINER_DLL_remove (s5r_head,
 748                                s5r_tail,
 749                                s5r);
 750   GNUNET_free_non_null (s5r->domain);
 751   GNUNET_free_non_null (s5r->leho);
 752   GNUNET_free_non_null (s5r->url);
 753   GNUNET_free_non_null (s5r->dane_data);
 754   GNUNET_free (s5r);
 755 }
 756
 757
 758 /* ************************* HTTP handling with cURL *********************** */
 759
 760
 761 /**
 762  * Callback for MHD response generation.  This function is called from
 763  * MHD whenever MHD expects to get data back.  Copies data from the
 764  * io_buf, if available.
 765  *
 766  * @param cls closure with our `struct Socks5Request`
 767  * @param pos in buffer
 768  * @param buf where to copy data
 769  * @param max available space in @a buf
 770  * @return number of bytes written to @a buf
 771  */
 772 static ssize_t
 773 mhd_content_cb (void *cls,
 774                 uint64_t pos,
 775                 char* buf,
 776                 size_t max)
 777 {
 778   struct Socks5Request *s5r = cls;
 779   size_t bytes_to_copy;
 780
 781   if ( (SOCKS5_SOCKET_UPLOAD_STARTED == s5r->state) ||
 782        (SOCKS5_SOCKET_UPLOAD_DONE == s5r->state) )
 783   {
 784     /* we're still not done with the upload, do not yet
 785        start the download, the IO buffer is still full
 786        with upload data. */
 787     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 788                 "Pausing MHD download, not yet ready for download\n");
 789     return 0; /* not yet ready for data download */
 790   }
 791   bytes_to_copy = GNUNET_MIN (max,
 792                               s5r->io_len);
 793   if ( (0 == bytes_to_copy) &&
 794        (SOCKS5_SOCKET_DOWNLOAD_DONE != s5r->state) )
 795   {
 796     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 797                 "Pausing MHD download, no data available\n");
 798     return 0; /* more data later */
 799   }
 800   if ( (0 == bytes_to_copy) &&
 801        (SOCKS5_SOCKET_DOWNLOAD_DONE == s5r->state) )
 802   {
 803     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 804                 "Completed MHD download\n");
 805     s5r->state = SOCKS5_SOCKET_WITH_MHD;
 806     return MHD_CONTENT_READER_END_OF_STREAM;
 807   }
 808   GNUNET_memcpy (buf, s5r->io_buf, bytes_to_copy);
 809   memmove (s5r->io_buf,
 810            &s5r->io_buf[bytes_to_copy],
 811            s5r->io_len - bytes_to_copy);
 812   s5r->io_len -= bytes_to_copy;
 813   if (NULL != s5r->curl)
 814     curl_easy_pause (s5r->curl, CURLPAUSE_CONT);
 815   return bytes_to_copy;
 816 }
 817
 818
 819 /**
 820  * Check that the website has presented us with a valid SSL certificate.
 821  * The certificate must either match the domain name or the LEHO name
 822  * (or, if available, the TLSA record).
 823  *
 824  * @param s5r request to check for.
 825  * @return #GNUNET_OK if the certificate is valid
 826  */
 827 static int
 828 check_ssl_certificate (struct Socks5Request *s5r)
 829 {
 830   unsigned int cert_list_size;
 831   const gnutls_datum_t *chainp;
 832   const struct curl_tlssessioninfo *tlsinfo;
 833   char certdn[GNUNET_DNSPARSER_MAX_NAME_LENGTH + 3];
 834   size_t size;
 835   gnutls_x509_crt_t x509_cert;
 836   int rc;
 837   const char *name;
 838
 839   if (CURLE_OK !=
 840       curl_easy_getinfo (s5r->curl,
 841                          CURLINFO_TLS_SESSION,
 842                          (struct curl_slist **) &tlsinfo))
 843     return GNUNET_SYSERR;
 844   if (CURLSSLBACKEND_GNUTLS != tlsinfo->backend)
 845   {
 846     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 847                 _("Unsupported CURL SSL backend %d\n"),
 848                 tlsinfo->backend);
 849     return GNUNET_SYSERR;
 850   }
 851   chainp = gnutls_certificate_get_peers (tlsinfo->internals, &cert_list_size);
 852   if ( (! chainp) || (0 == cert_list_size) )
 853     return GNUNET_SYSERR;
 854
 855   size = sizeof (certdn);
 856   /* initialize an X.509 certificate structure. */
 857   gnutls_x509_crt_init (&x509_cert);
 858   gnutls_x509_crt_import (x509_cert,
 859                           chainp,
 860                           GNUTLS_X509_FMT_DER);
 861
 862   if (0 != (rc = gnutls_x509_crt_get_dn_by_oid (x509_cert,
 863                                                 GNUTLS_OID_X520_COMMON_NAME,
 864                                                 0, /* the first and only one */
 865                                                 0 /* no DER encoding */,
 866                                                 certdn,
 867                                                 &size)))
 868   {
 869     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 870                 _("Failed to fetch CN from cert: %s\n"),
 871                 gnutls_strerror(rc));
 872     gnutls_x509_crt_deinit (x509_cert);
 873     return GNUNET_SYSERR;
 874   }
 875   /* check for TLSA/DANE records */
 876 #if HAVE_GNUTLS_DANE
 877   if (NULL != s5r->dane_data)
 878   {
 879     char *dd[] = { s5r->dane_data, NULL };
 880     int dlen[] = { s5r->dane_data_len, 0};
 881     dane_state_t dane_state;
 882     dane_query_t dane_query;
 883     unsigned int verify;
 884
 885     /* FIXME: add flags to gnutls to NOT read UNBOUND_ROOT_KEY_FILE here! */
 886     if (0 != (rc = dane_state_init (&dane_state,
 887 #ifdef DANE_F_IGNORE_DNSSEC
 888                                     DANE_F_IGNORE_DNSSEC |
 889 #endif
 890                                     DANE_F_IGNORE_LOCAL_RESOLVER)))
 891     {
 892       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 893                   _("Failed to initialize DANE: %s\n"),
 894                   dane_strerror(rc));
 895       gnutls_x509_crt_deinit (x509_cert);
 896       return GNUNET_SYSERR;
 897     }
 898     if (0 != (rc = dane_raw_tlsa (dane_state,
 899                                   &dane_query,
 900                                   dd,
 901                                   dlen,
 902                                   GNUNET_YES,
 903                                   GNUNET_NO)))
 904     {
 905       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 906                   _("Failed to parse DANE record: %s\n"),
 907                   dane_strerror(rc));
 908       dane_state_deinit (dane_state);
 909       gnutls_x509_crt_deinit (x509_cert);
 910       return GNUNET_SYSERR;
 911     }
 912     if (0 != (rc = dane_verify_crt_raw (dane_state,
 913                                         chainp,
 914                                         cert_list_size,
 915                                         gnutls_certificate_type_get (tlsinfo->internals),
 916                                         dane_query,
 917                                         0, 0,
 918                                         &verify)))
 919     {
 920       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 921                   _("Failed to verify TLS connection using DANE: %s\n"),
 922                   dane_strerror(rc));
 923       dane_query_deinit (dane_query);
 924       dane_state_deinit (dane_state);
 925       gnutls_x509_crt_deinit (x509_cert);
 926       return GNUNET_SYSERR;
 927     }
 928     if (0 != verify)
 929     {
 930       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 931                   _("Failed DANE verification failed with GnuTLS verify status code: %u\n"),
 932                   verify);
 933       dane_query_deinit (dane_query);
 934       dane_state_deinit (dane_state);
 935       gnutls_x509_crt_deinit (x509_cert);
 936       return GNUNET_SYSERR;
 937     }
 938     dane_query_deinit (dane_query);
 939     dane_state_deinit (dane_state);
 940     /* success! */
 941   }
 942   else
 943 #endif
 944   {
 945     /* try LEHO or ordinary domain name X509 verification */
 946     name = s5r->domain;
 947     if (NULL != s5r->leho)
 948       name = s5r->leho;
 949     if (NULL != name)
 950     {
 951       if (0 == (rc = gnutls_x509_crt_check_hostname (x509_cert,
 952                                                      name)))
 953       {
 954         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 955                     _("SSL certificate subject name (%s) does not match `%s'\n"),
 956                     certdn,
 957                     name);
 958         gnutls_x509_crt_deinit (x509_cert);
 959         return GNUNET_SYSERR;
 960       }
 961     }
 962     else
 963     {
 964       /* we did not even have the domain name!? */
 965       GNUNET_break (0);
 966       return GNUNET_SYSERR;
 967     }
 968   }
 969   gnutls_x509_crt_deinit (x509_cert);
 970   return GNUNET_OK;
 971 }
 972
 973
 974 /**
 975  * We're getting an HTTP response header from cURL.  Convert it to the
 976  * MHD response headers.  Mostly copies the headers, but makes special
 977  * adjustments to "Set-Cookie" and "Location" headers as those may need
 978  * to be changed from the LEHO to the domain the browser expects.
 979  *
 980  * @param buffer curl buffer with a single line of header data; not 0-terminated!
 981  * @param size curl blocksize
 982  * @param nmemb curl blocknumber
 983  * @param cls our `struct Socks5Request *`
 984  * @return size of processed bytes
 985  */
 986 static size_t
 987 curl_check_hdr (void *buffer, size_t size, size_t nmemb, void *cls)
 988 {
 989   struct Socks5Request *s5r = cls;
 990   size_t bytes = size * nmemb;
 991   char *ndup;
 992   const char *hdr_type;
 993   const char *cookie_domain;
 994   char *hdr_val;
 995   long resp_code;
 996   char *new_cookie_hdr;
 997   char *new_location;
 998   size_t offset;
 999   size_t delta_cdomain;
1000   int domain_matched;
1001   char *tok;
1002
1003   if (NULL == s5r->response)
1004   {
1005     /* first, check SSL certificate */
1006     if ( (HTTPS_PORT == s5r->port) &&
1007          (GNUNET_OK != check_ssl_certificate (s5r)) )
1008       return 0;
1009
1010     GNUNET_break (CURLE_OK ==
1011                   curl_easy_getinfo (s5r->curl,
1012                                      CURLINFO_RESPONSE_CODE,
1013                                      &resp_code));
1014     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1015                 "Creating MHD response with code %d\n",
1016                 (int) resp_code);
1017     s5r->response_code = resp_code;
1018     s5r->response = MHD_create_response_from_callback (MHD_SIZE_UNKNOWN,
1019                                                        IO_BUFFERSIZE,
1020                                                        &mhd_content_cb,
1021                                                        s5r,
1022                                                        NULL);
1023     if (NULL != s5r->leho)
1024     {
1025       char *cors_hdr;
1026
1027       GNUNET_asprintf (&cors_hdr,
1028                        (HTTPS_PORT == s5r->port)
1029                        ? "https://%s"
1030                        : "http://%s",
1031                        s5r->leho);
1032
1033       GNUNET_break (MHD_YES ==
1034                     MHD_add_response_header (s5r->response,
1035                                              MHD_HTTP_HEADER_ACCESS_CONTROL_ALLOW_ORIGIN,
1036                                              cors_hdr));
1037       GNUNET_free (cors_hdr);
1038     }
1039     /* force connection to be closed after each request, as we
1040        do not support HTTP pipelining (yet, FIXME!) */
1041     /*GNUNET_break (MHD_YES ==
1042                   MHD_add_response_header (s5r->response,
1043                                            MHD_HTTP_HEADER_CONNECTION,
1044                                            "close"));*/
1045   }
1046
1047   ndup = GNUNET_strndup (buffer, bytes);
1048   hdr_type = strtok (ndup, ":");
1049   if (NULL == hdr_type)
1050   {
1051     GNUNET_free (ndup);
1052     return bytes;
1053   }
1054   hdr_val = strtok (NULL, "");
1055   if (NULL == hdr_val)
1056   {
1057     GNUNET_free (ndup);
1058     return bytes;
1059   }
1060   if (' ' == *hdr_val)
1061     hdr_val++;
1062
1063   /* custom logic for certain header types */
1064   new_cookie_hdr = NULL;
1065   if ( (NULL != s5r->leho) &&
1066        (0 == strcasecmp (hdr_type,
1067                          MHD_HTTP_HEADER_SET_COOKIE)) )
1068
1069   {
1070     new_cookie_hdr = GNUNET_malloc (strlen (hdr_val) +
1071                                     strlen (s5r->domain) + 1);
1072     offset = 0;
1073     domain_matched = GNUNET_NO; /* make sure we match domain at most once */
1074     for (tok = strtok (hdr_val, ";"); NULL != tok; tok = strtok (NULL, ";"))
1075     {
1076       if ( (0 == strncasecmp (tok, " domain", strlen (" domain"))) &&
1077            (GNUNET_NO == domain_matched) )
1078       {
1079         domain_matched = GNUNET_YES;
1080         cookie_domain = tok + strlen (" domain") + 1;
1081         if (strlen (cookie_domain) < strlen (s5r->leho))
1082         {
1083           delta_cdomain = strlen (s5r->leho) - strlen (cookie_domain);
1084           if (0 == strcasecmp (cookie_domain, s5r->leho + delta_cdomain))
1085           {
1086             offset += sprintf (new_cookie_hdr + offset,
1087                                " domain=%s;",
1088                                s5r->domain);
1089             continue;
1090           }
1091         }
1092         else if (0 == strcmp (cookie_domain, s5r->leho))
1093         {
1094           offset += sprintf (new_cookie_hdr + offset,
1095                              " domain=%s;",
1096                              s5r->domain);
1097           continue;
1098         }
1099         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1100                     _("Cookie domain `%s' supplied by server is invalid\n"),
1101                     tok);
1102       }
1103       GNUNET_memcpy (new_cookie_hdr + offset, tok, strlen (tok));
1104       offset += strlen (tok);
1105       new_cookie_hdr[offset++] = ';';
1106     }
1107     hdr_val = new_cookie_hdr;
1108   }
1109
1110   new_location = NULL;
1111   if (0 == strcasecmp (MHD_HTTP_HEADER_LOCATION, hdr_type))
1112   {
1113     char *leho_host;
1114
1115     GNUNET_asprintf (&leho_host,
1116                      (HTTPS_PORT != s5r->port)
1117                      ? "http://%s"
1118                      : "https://%s",
1119                      s5r->leho);
1120     if (0 == strncmp (leho_host,
1121                       hdr_val,
1122                       strlen (leho_host)))
1123     {
1124       GNUNET_asprintf (&new_location,
1125                        "%s%s%s",
1126                        (HTTPS_PORT != s5r->port)
1127                        ? "http://"
1128                        : "https://",
1129                        s5r->domain,
1130                        hdr_val + strlen (leho_host));
1131       hdr_val = new_location;
1132     }
1133     GNUNET_free (leho_host);
1134   }
1135   /* MHD does not allow certain characters in values, remove those */
1136   if (NULL != (tok = strchr (hdr_val, '\n')))
1137     *tok = '\0';
1138   if (NULL != (tok = strchr (hdr_val, '\r')))
1139     *tok = '\0';
1140   if (NULL != (tok = strchr (hdr_val, '\t')))
1141     *tok = '\0';
1142   if ( (0 != strlen (hdr_val) ) &&
1143        /* We do chunked transfer encoding */
1144        (0 != strcasecmp (MHD_HTTP_HEADER_CONTENT_LENGTH, hdr_type)))
1145   {
1146     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1147                 "Adding header %s: %s to MHD response\n",
1148                 hdr_type,
1149                 hdr_val);
1150     GNUNET_break (MHD_YES ==
1151                   MHD_add_response_header (s5r->response,
1152                                            hdr_type,
1153                                            hdr_val));
1154   }
1155   GNUNET_free (ndup);
1156   GNUNET_free_non_null (new_cookie_hdr);
1157   GNUNET_free_non_null (new_location);
1158   return bytes;
1159 }
1160
1161
1162 /**
1163  * Handle response payload data from cURL.  Copies it into our `io_buf` to make
1164  * it available to MHD.
1165  *
1166  * @param ptr pointer to the data
1167  * @param size number of blocks of data
1168  * @param nmemb blocksize
1169  * @param ctx our `struct Socks5Request *`
1170  * @return number of bytes handled
1171  */
1172 static size_t
1173 curl_download_cb (void *ptr, size_t size, size_t nmemb, void* ctx)
1174 {
1175   struct Socks5Request *s5r = ctx;
1176   size_t total = size * nmemb;
1177
1178   if ( (SOCKS5_SOCKET_UPLOAD_STARTED == s5r->state) ||
1179        (SOCKS5_SOCKET_UPLOAD_DONE == s5r->state) )
1180   {
1181     /* we're still not done with the upload, do not yet
1182        start the download, the IO buffer is still full
1183        with upload data. */
1184     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1185                 "Pausing CURL download, waiting for UPLOAD to finish\n");
1186     return CURL_WRITEFUNC_PAUSE; /* not yet ready for data download */
1187   }
1188   if (sizeof (s5r->io_buf) - s5r->io_len < total)
1189   {
1190     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1191                 "Pausing CURL download, not enough space\n");
1192     return CURL_WRITEFUNC_PAUSE; /* not enough space */
1193   }
1194   GNUNET_memcpy (&s5r->io_buf[s5r->io_len],
1195           ptr,
1196           total);
1197   s5r->io_len += total;
1198   if (s5r->io_len == total)
1199     run_mhd_now (s5r->hd);
1200   return total;
1201 }
1202
1203
1204 /**
1205  * cURL callback for uploaded (PUT/POST) data.  Copies it into our `io_buf`
1206  * to make it available to MHD.
1207  *
1208  * @param buf where to write the data
1209  * @param size number of bytes per member
1210  * @param nmemb number of members available in @a buf
1211  * @param cls our `struct Socks5Request` that generated the data
1212  * @return number of bytes copied to @a buf
1213  */
1214 static size_t
1215 curl_upload_cb (void *buf, size_t size, size_t nmemb, void *cls)
1216 {
1217   struct Socks5Request *s5r = cls;
1218   size_t len = size * nmemb;
1219   size_t to_copy;
1220
1221   if ( (0 == s5r->io_len) &&
1222        (SOCKS5_SOCKET_UPLOAD_DONE != s5r->state) )
1223   {
1224     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1225                 "Pausing CURL UPLOAD, need more data\n");
1226     return CURL_READFUNC_PAUSE;
1227   }
1228   if ( (0 == s5r->io_len) &&
1229        (SOCKS5_SOCKET_UPLOAD_DONE == s5r->state) )
1230   {
1231     s5r->state = SOCKS5_SOCKET_DOWNLOAD_STARTED;
1232     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1233                 "Completed CURL UPLOAD\n");
1234     return 0; /* upload finished, can now download */
1235   }
1236   if ( (SOCKS5_SOCKET_UPLOAD_STARTED != s5r->state) &&
1237        (SOCKS5_SOCKET_UPLOAD_DONE != s5r->state) )
1238   {
1239     GNUNET_break (0);
1240     return CURL_READFUNC_ABORT;
1241   }
1242   to_copy = GNUNET_MIN (s5r->io_len,
1243                         len);
1244   GNUNET_memcpy (buf, s5r->io_buf, to_copy);
1245   memmove (s5r->io_buf,
1246            &s5r->io_buf[to_copy],
1247            s5r->io_len - to_copy);
1248   s5r->io_len -= to_copy;
1249   if (s5r->io_len + to_copy == sizeof (s5r->io_buf))
1250     run_mhd_now (s5r->hd); /* got more space for upload now */
1251   return to_copy;
1252 }
1253
1254
1255 /* ************************** main loop of cURL interaction ****************** */
1256
1257
1258 /**
1259  * Task that is run when we are ready to receive more data
1260  * from curl
1261  *
1262  * @param cls closure
1263  */
1264 static void
1265 curl_task_download (void *cls);
1266
1267
1268 /**
1269  * Ask cURL for the select() sets and schedule cURL operations.
1270  */
1271 static void
1272 curl_download_prepare ()
1273 {
1274   CURLMcode mret;
1275   fd_set rs;
1276   fd_set ws;
1277   fd_set es;
1278   int max;
1279   struct GNUNET_NETWORK_FDSet *grs;
1280   struct GNUNET_NETWORK_FDSet *gws;
1281   long to;
1282   struct GNUNET_TIME_Relative rtime;
1283
1284   if (NULL != curl_download_task)
1285   {
1286     GNUNET_SCHEDULER_cancel (curl_download_task);
1287     curl_download_task = NULL;
1288   }
1289   max = -1;
1290   FD_ZERO (&rs);
1291   FD_ZERO (&ws);
1292   FD_ZERO (&es);
1293   if (CURLM_OK != (mret = curl_multi_fdset (curl_multi, &rs, &ws, &es, &max)))
1294   {
1295     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1296                 "%s failed at %s:%d: `%s'\n",
1297                 "curl_multi_fdset", __FILE__, __LINE__,
1298                 curl_multi_strerror (mret));
1299     return;
1300   }
1301   to = -1;
1302   GNUNET_break (CURLM_OK == curl_multi_timeout (curl_multi, &to));
1303   if (-1 == to)
1304     rtime = GNUNET_TIME_UNIT_FOREVER_REL;
1305   else
1306     rtime = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, to);
1307   if (-1 != max)
1308   {
1309     grs = GNUNET_NETWORK_fdset_create ();
1310     gws = GNUNET_NETWORK_fdset_create ();
1311     GNUNET_NETWORK_fdset_copy_native (grs, &rs, max + 1);
1312     GNUNET_NETWORK_fdset_copy_native (gws, &ws, max + 1);
1313     curl_download_task = GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1314                                                       rtime,
1315                                                       grs, gws,
1316                                                       &curl_task_download, curl_multi);
1317     GNUNET_NETWORK_fdset_destroy (gws);
1318     GNUNET_NETWORK_fdset_destroy (grs);
1319   }
1320   else
1321   {
1322     curl_download_task = GNUNET_SCHEDULER_add_delayed (rtime,
1323                                                        &curl_task_download,
1324                                                        curl_multi);
1325   }
1326 }
1327
1328
1329 /**
1330  * Task that is run when we are ready to receive more data from curl.
1331  *
1332  * @param cls closure, NULL
1333  */
1334 static void
1335 curl_task_download (void *cls)
1336 {
1337   int running;
1338   int msgnum;
1339   struct CURLMsg *msg;
1340   CURLMcode mret;
1341   struct Socks5Request *s5r;
1342
1343   curl_download_task = NULL;
1344   do
1345   {
1346     running = 0;
1347     mret = curl_multi_perform (curl_multi, &running);
1348     while (NULL != (msg = curl_multi_info_read (curl_multi, &msgnum)))
1349     {
1350       GNUNET_break (CURLE_OK ==
1351                     curl_easy_getinfo (msg->easy_handle,
1352                                        CURLINFO_PRIVATE,
1353                                        (char **) &s5r ));
1354       if (NULL == s5r)
1355       {
1356         GNUNET_break (0);
1357         continue;
1358       }
1359       switch (msg->msg)
1360       {
1361         case CURLMSG_NONE:
1362           /* documentation says this is not used */
1363           GNUNET_break (0);
1364           break;
1365         case CURLMSG_DONE:
1366           switch (msg->data.result)
1367           {
1368             case CURLE_OK:
1369             case CURLE_GOT_NOTHING:
1370               GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1371                           "CURL download completed.\n");
1372               s5r->state = SOCKS5_SOCKET_DOWNLOAD_DONE;
1373               run_mhd_now (s5r->hd);
1374               break;
1375             default:
1376               GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1377                           "Download curl failed: %s\n",
1378                           curl_easy_strerror (msg->data.result));
1379               /* FIXME: indicate error somehow? close MHD connection badly as well? */
1380               s5r->state = SOCKS5_SOCKET_DOWNLOAD_DONE;
1381               run_mhd_now (s5r->hd);
1382               break;
1383           }
1384           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1385                       "Cleaning up cURL handle\n");
1386           curl_multi_remove_handle (curl_multi, s5r->curl);
1387           curl_easy_cleanup (s5r->curl);
1388           s5r->curl = NULL;
1389           if (NULL == s5r->response)
1390             s5r->response = curl_failure_response;
1391           break;
1392         case CURLMSG_LAST:
1393           /* documentation says this is not used */
1394           GNUNET_break (0);
1395           break;
1396         default:
1397           /* unexpected status code */
1398           GNUNET_break (0);
1399           break;
1400       }
1401     };
1402   } while (mret == CURLM_CALL_MULTI_PERFORM);
1403   if (CURLM_OK != mret)
1404     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1405                 "%s failed at %s:%d: `%s'\n",
1406                 "curl_multi_perform", __FILE__, __LINE__,
1407                 curl_multi_strerror (mret));
1408   if (0 == running)
1409   {
1410     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1411                 "Suspending cURL multi loop, no more events pending\n");
1412     return; /* nothing more in progress */
1413   }
1414   curl_download_prepare ();
1415 }
1416
1417
1418 /* ********************************* MHD response generation ******************* */
1419
1420
1421 /**
1422  * Read HTTP request header field from the request.  Copies the fields
1423  * over to the 'headers' that will be given to curl.  However, 'Host'
1424  * is substituted with the LEHO if present.  We also change the
1425  * 'Connection' header value to "close" as the proxy does not support
1426  * pipelining.
1427  *
1428  * @param cls our `struct Socks5Request`
1429  * @param kind value kind
1430  * @param key field key
1431  * @param value field value
1432  * @return MHD_YES to continue to iterate
1433  */
1434 static int
1435 con_val_iter (void *cls,
1436               enum MHD_ValueKind kind,
1437               const char *key,
1438               const char *value)
1439 {
1440   struct Socks5Request *s5r = cls;
1441   char *hdr;
1442
1443   if ( (0 == strcasecmp (MHD_HTTP_HEADER_HOST, key)) &&
1444        (NULL != s5r->leho) )
1445     value = s5r->leho;
1446   /*if (0 == strcasecmp (MHD_HTTP_HEADER_CONNECTION, key))
1447     value = "Close";*/
1448   GNUNET_asprintf (&hdr,
1449                    "%s: %s",
1450                    key,
1451                    value);
1452   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1453               "Adding HEADER `%s' to HTTP request\n",
1454               hdr);
1455   s5r->headers = curl_slist_append (s5r->headers,
1456                                     hdr);
1457   GNUNET_free (hdr);
1458   return MHD_YES;
1459 }
1460
1461
1462 /**
1463  * Main MHD callback for handling requests.
1464  *
1465  * @param cls unused
1466  * @param con MHD connection handle
1467  * @param url the url in the request
1468  * @param meth the HTTP method used ("GET", "PUT", etc.)
1469  * @param ver the HTTP version string (i.e. "HTTP/1.1")
1470  * @param upload_data the data being uploaded (excluding HEADERS,
1471  *        for a POST that fits into memory and that is encoded
1472  *        with a supported encoding, the POST data will NOT be
1473  *        given in upload_data and is instead available as
1474  *        part of MHD_get_connection_values; very large POST
1475  *        data *will* be made available incrementally in
1476  *        upload_data)
1477  * @param upload_data_size set initially to the size of the
1478  *        @a upload_data provided; the method must update this
1479  *        value to the number of bytes NOT processed;
1480  * @param con_cls pointer to location where we store the 'struct Request'
1481  * @return MHD_YES if the connection was handled successfully,
1482  *         MHD_NO if the socket must be closed due to a serious
1483  *         error while handling the request
1484  */
1485 static int
1486 create_response (void *cls,
1487                  struct MHD_Connection *con,
1488                  const char *url,
1489                  const char *meth,
1490                  const char *ver,
1491                  const char *upload_data,
1492                  size_t *upload_data_size,
1493                  void **con_cls)
1494 {
1495   struct Socks5Request *s5r = *con_cls;
1496   char *curlurl;
1497   char *curl_hosts;
1498   char ipstring[INET6_ADDRSTRLEN];
1499   char ipaddr[INET6_ADDRSTRLEN + 2];
1500   const struct sockaddr *sa;
1501   const struct sockaddr_in *s4;
1502   const struct sockaddr_in6 *s6;
1503   uint16_t port;
1504   size_t left;
1505
1506   if (NULL == s5r)
1507   {
1508     GNUNET_break (0);
1509     return MHD_NO;
1510   }
1511   //Fresh connection. Maybe move to notify callback??
1512   if (SOCKS5_SOCKET_WITH_MHD == s5r->state)
1513   {
1514     /* first time here, initialize curl handle */
1515     sa = (const struct sockaddr *) &s5r->destination_address;
1516     switch (sa->sa_family)
1517     {
1518       case AF_INET:
1519         s4 = (const struct sockaddr_in *) &s5r->destination_address;
1520         if (NULL == inet_ntop (AF_INET,
1521                                &s4->sin_addr,
1522                                ipstring,
1523                                sizeof (ipstring)))
1524         {
1525           GNUNET_break (0);
1526           return MHD_NO;
1527         }
1528         GNUNET_snprintf (ipaddr,
1529                          sizeof (ipaddr),
1530                          "%s",
1531                          ipstring);
1532         port = ntohs (s4->sin_port);
1533         break;
1534       case AF_INET6:
1535         s6 = (const struct sockaddr_in6 *) &s5r->destination_address;
1536         if (NULL == inet_ntop (AF_INET6,
1537                                &s6->sin6_addr,
1538                                ipstring,
1539                                sizeof (ipstring)))
1540         {
1541           GNUNET_break (0);
1542           return MHD_NO;
1543         }
1544         GNUNET_snprintf (ipaddr,
1545                          sizeof (ipaddr),
1546                          "[%s]",
1547                          ipstring);
1548         port = ntohs (s6->sin6_port);
1549         break;
1550       default:
1551         GNUNET_break (0);
1552         return MHD_NO;
1553     }
1554     if (NULL == s5r->curl)
1555       s5r->curl = curl_easy_init ();
1556     if (NULL == s5r->curl)
1557       return MHD_queue_response (con,
1558                                  MHD_HTTP_INTERNAL_SERVER_ERROR,
1559                                  curl_failure_response);
1560     curl_easy_setopt (s5r->curl, CURLOPT_HEADERFUNCTION, &curl_check_hdr);
1561     curl_easy_setopt (s5r->curl, CURLOPT_HEADERDATA, s5r);
1562     curl_easy_setopt (s5r->curl, CURLOPT_FOLLOWLOCATION, 0);
1563     curl_easy_setopt (s5r->curl, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4);
1564     curl_easy_setopt (s5r->curl, CURLOPT_CONNECTTIMEOUT, 600L);
1565     curl_easy_setopt (s5r->curl, CURLOPT_TIMEOUT, 600L);
1566     curl_easy_setopt (s5r->curl, CURLOPT_NOSIGNAL, 1L);
1567     curl_easy_setopt (s5r->curl, CURLOPT_HTTP_CONTENT_DECODING, 0);
1568     curl_easy_setopt (s5r->curl, CURLOPT_HTTP_TRANSFER_DECODING, 0);
1569     curl_easy_setopt (s5r->curl, CURLOPT_NOSIGNAL, 1L);
1570     curl_easy_setopt (s5r->curl, CURLOPT_PRIVATE, s5r);
1571     curl_easy_setopt (s5r->curl, CURLOPT_VERBOSE, 0);
1572     /**
1573      * Pre-populate cache to resolve Hostname.
1574      * This is necessary as the DNS name in the CURLOPT_URL is used
1575      * for SNI http://de.wikipedia.org/wiki/Server_Name_Indication
1576      */
1577     if (NULL != s5r->leho)
1578     {
1579       GNUNET_asprintf (&curl_hosts,
1580                        "%s:%d:%s",
1581                        s5r->leho,
1582                        port,
1583                        ipaddr);
1584       s5r->hosts = curl_slist_append(NULL, curl_hosts);
1585       curl_easy_setopt(s5r->curl, CURLOPT_RESOLVE, s5r->hosts);
1586       GNUNET_free (curl_hosts);
1587     }
1588     GNUNET_asprintf (&curlurl,
1589                      (HTTPS_PORT != s5r->port)
1590                      ? "http://%s:%d%s"
1591                      : "https://%s:%d%s",
1592                      (NULL != s5r->leho)
1593                      ? s5r->leho
1594                      : ipaddr,
1595                      port,
1596                      s5r->url);
1597     curl_easy_setopt (s5r->curl,
1598                       CURLOPT_URL,
1599                       curlurl);
1600     GNUNET_free (curlurl);
1601     if (0 == strcasecmp (meth, MHD_HTTP_METHOD_PUT))
1602     {
1603       s5r->state = SOCKS5_SOCKET_UPLOAD_STARTED;
1604       curl_easy_setopt (s5r->curl, CURLOPT_UPLOAD, 1);
1605       curl_easy_setopt (s5r->curl, CURLOPT_WRITEFUNCTION, &curl_download_cb);
1606       curl_easy_setopt (s5r->curl, CURLOPT_WRITEDATA, s5r);
1607       curl_easy_setopt (s5r->curl, CURLOPT_READFUNCTION, &curl_upload_cb);
1608       curl_easy_setopt (s5r->curl, CURLOPT_READDATA, s5r);
1609     }
1610     else if (0 == strcasecmp (meth, MHD_HTTP_METHOD_POST))
1611     {
1612       s5r->state = SOCKS5_SOCKET_UPLOAD_STARTED;
1613       curl_easy_setopt (s5r->curl, CURLOPT_POST, 1);
1614       curl_easy_setopt (s5r->curl, CURLOPT_WRITEFUNCTION, &curl_download_cb);
1615       curl_easy_setopt (s5r->curl, CURLOPT_WRITEDATA, s5r);
1616       curl_easy_setopt (s5r->curl, CURLOPT_READFUNCTION, &curl_upload_cb);
1617       curl_easy_setopt (s5r->curl, CURLOPT_READDATA, s5r);
1618     }
1619     else if (0 == strcasecmp (meth, MHD_HTTP_METHOD_HEAD))
1620     {
1621       s5r->state = SOCKS5_SOCKET_DOWNLOAD_STARTED;
1622       curl_easy_setopt (s5r->curl, CURLOPT_NOBODY, 1);
1623     }
1624     else if (0 == strcasecmp (meth, MHD_HTTP_METHOD_OPTIONS))
1625     {
1626       s5r->state = SOCKS5_SOCKET_DOWNLOAD_STARTED;
1627       curl_easy_setopt (s5r->curl, CURLOPT_CUSTOMREQUEST, "OPTIONS");
1628     }
1629     else if (0 == strcasecmp (meth, MHD_HTTP_METHOD_GET))
1630     {
1631       s5r->state = SOCKS5_SOCKET_DOWNLOAD_STARTED;
1632       curl_easy_setopt (s5r->curl, CURLOPT_HTTPGET, 1);
1633       curl_easy_setopt (s5r->curl, CURLOPT_WRITEFUNCTION, &curl_download_cb);
1634       curl_easy_setopt (s5r->curl, CURLOPT_WRITEDATA, s5r);
1635     }
1636     else
1637     {
1638       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1639                   _("Unsupported HTTP method `%s'\n"),
1640                   meth);
1641       curl_easy_cleanup (s5r->curl);
1642       s5r->curl = NULL;
1643       return MHD_NO;
1644     }
1645
1646     if (0 == strcasecmp (ver, MHD_HTTP_VERSION_1_0))
1647     {
1648       curl_easy_setopt (s5r->curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
1649     }
1650     else if (0 == strcasecmp (ver, MHD_HTTP_VERSION_1_1))
1651     {
1652       curl_easy_setopt (s5r->curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
1653     }
1654     else
1655     {
1656       curl_easy_setopt (s5r->curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_NONE);
1657     }
1658
1659     if (HTTPS_PORT == s5r->port)
1660     {
1661       curl_easy_setopt (s5r->curl, CURLOPT_USE_SSL, CURLUSESSL_ALL);
1662       curl_easy_setopt (s5r->curl, CURLOPT_SSL_VERIFYPEER, 1L);
1663       /* Disable cURL checking the hostname, as we will check ourselves
1664          as only we have the domain name or the LEHO or the DANE record */
1665       curl_easy_setopt (s5r->curl, CURLOPT_SSL_VERIFYHOST, 0L);
1666     }
1667     else
1668     {
1669       curl_easy_setopt (s5r->curl, CURLOPT_USE_SSL, CURLUSESSL_NONE);
1670     }
1671
1672     if (CURLM_OK != curl_multi_add_handle (curl_multi, s5r->curl))
1673     {
1674       GNUNET_break (0);
1675       curl_easy_cleanup (s5r->curl);
1676       s5r->curl = NULL;
1677       return MHD_NO;
1678     }
1679     MHD_get_connection_values (con,
1680                                MHD_HEADER_KIND,
1681                                &con_val_iter, s5r);
1682     //TODO is this sane?
1683     s5r->headers = curl_slist_append (s5r->headers,
1684                                       "Expect:");
1685     curl_easy_setopt (s5r->curl, CURLOPT_HTTPHEADER, s5r->headers);
1686     curl_download_prepare ();
1687     return MHD_YES;
1688   }
1689
1690   /* continuing to process request */
1691   if (0 != *upload_data_size)
1692   {
1693     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1694                 "Processing %lu bytes UPLOAD\n", *upload_data_size);
1695     left = GNUNET_MIN (*upload_data_size,
1696                        sizeof (s5r->io_buf) - s5r->io_len);
1697     GNUNET_memcpy (&s5r->io_buf[s5r->io_len],
1698                    upload_data,
1699                    left);
1700     s5r->io_len += left;
1701     *upload_data_size -= left;
1702     GNUNET_assert (NULL != s5r->curl);
1703     curl_easy_pause (s5r->curl, CURLPAUSE_CONT);
1704     curl_download_prepare ();
1705     return MHD_YES;
1706   }
1707   if (SOCKS5_SOCKET_UPLOAD_STARTED == s5r->state)
1708   {
1709     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1710                 "Finished processing UPLOAD\n");
1711     s5r->state = SOCKS5_SOCKET_UPLOAD_DONE;
1712   }
1713   if (NULL == s5r->response)
1714     return MHD_YES; /* too early to queue response, did not yet get headers from cURL */
1715   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1716               "Queueing response with MHD\n");
1717   run_mhd_now (s5r->hd);
1718   return MHD_queue_response (con,
1719                              s5r->response_code,
1720                              s5r->response);
1721 }
1722
1723
1724 /* ******************** MHD HTTP setup and event loop ******************** */
1725
1726
1727 /**
1728  * Function called when MHD decides that we are done with a request.
1729  *
1730  * @param cls NULL
1731  * @param connection connection handle
1732  * @param con_cls value as set by the last call to
1733  *        the MHD_AccessHandlerCallback, should be our `struct Socks5Request *`
1734  * @param toe reason for request termination (ignored)
1735  */
1736 static void
1737 mhd_completed_cb (void *cls,
1738                   struct MHD_Connection *connection,
1739                   void **con_cls,
1740                   enum MHD_RequestTerminationCode toe)
1741 {
1742   struct Socks5Request *s5r = *con_cls;
1743
1744   if (NULL == s5r)
1745     return;
1746   if (MHD_REQUEST_TERMINATED_COMPLETED_OK != toe)
1747     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1748                 "MHD encountered error handling request: %d\n",
1749                 toe);
1750   if (NULL != s5r->curl)
1751   {
1752     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1753                 "Cleaning up cURL handle\n");
1754     curl_multi_remove_handle (curl_multi, s5r->curl);
1755     curl_easy_cleanup (s5r->curl);
1756     s5r->curl = NULL;
1757   }
1758   if ( (NULL != s5r->response) &&
1759        (curl_failure_response != s5r->response) )
1760     MHD_destroy_response (s5r->response);
1761   s5r->response = NULL;
1762   curl_download_prepare();
1763   *con_cls = NULL;
1764 }
1765
1766
1767 /**
1768  * Function called when MHD connection is opened or closed.
1769  *
1770  * @param cls NULL
1771  * @param connection connection handle
1772  * @param con_cls value as set by the last call to
1773  *        the MHD_AccessHandlerCallback, should be our `struct Socks5Request *`
1774  * @param toe connection notification type
1775  */
1776 static void
1777 mhd_connection_cb (void *cls,
1778                    struct MHD_Connection *connection,
1779                    void **con_cls,
1780                    enum MHD_ConnectionNotificationCode cnc)
1781 {
1782   struct Socks5Request *s5r;
1783   const union MHD_ConnectionInfo *ci;
1784   int sock;
1785
1786   if (MHD_CONNECTION_NOTIFY_STARTED == cnc)
1787     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connection started...\n");
1788
1789   if (MHD_CONNECTION_NOTIFY_CLOSED != cnc)
1790     return; //Ignore
1791   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connection closed... cleaning up\n");
1792
1793   ci = MHD_get_connection_info (connection,
1794                                 MHD_CONNECTION_INFO_CONNECTION_FD);
1795   if (NULL == ci)
1796   {
1797     GNUNET_break (0);
1798     return;
1799   }
1800
1801   sock = ci->connect_fd;
1802   for (s5r = s5r_head; NULL != s5r; s5r = s5r->next)
1803     if (GNUNET_NETWORK_get_fd (s5r->sock) == sock)
1804       break;
1805
1806   if (NULL == s5r)
1807   {
1808     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connection stale!\n");
1809     return;
1810   }
1811   cleanup_s5r (s5r);
1812   *con_cls = NULL;
1813 }
1814
1815 /**
1816  * Function called when MHD first processes an incoming connection.
1817  * Gives us the respective URI information.
1818  *
1819  * We use this to associate the `struct MHD_Connection` with our
1820  * internal `struct Socks5Request` data structure (by checking
1821  * for matching sockets).
1822  *
1823  * @param cls the HTTP server handle (a `struct MhdHttpList`)
1824  * @param url the URL that is being requested
1825  * @param connection MHD connection object for the request
1826  * @return the `struct Socks5Request` that this @a connection is for
1827  */
1828 static void *
1829 mhd_log_callback (void *cls,
1830                   const char *url,
1831                   struct MHD_Connection *connection)
1832 {
1833   struct Socks5Request *s5r;
1834   const union MHD_ConnectionInfo *ci;
1835   int sock;
1836
1837   ci = MHD_get_connection_info (connection,
1838                                 MHD_CONNECTION_INFO_CONNECTION_FD);
1839   if (NULL == ci)
1840   {
1841     GNUNET_break (0);
1842     return NULL;
1843   }
1844   sock = ci->connect_fd;
1845   for (s5r = s5r_head; NULL != s5r; s5r = s5r->next)
1846   {
1847     if (GNUNET_NETWORK_get_fd (s5r->sock) == sock)
1848     {
1849       if (NULL != s5r->url)
1850       {
1851         GNUNET_break (0);
1852         return NULL;
1853       }
1854       s5r->url = GNUNET_strdup (url);
1855       GNUNET_SCHEDULER_cancel (s5r->timeout_task);
1856       s5r->timeout_task = NULL;
1857       return s5r;
1858     }
1859   }
1860   GNUNET_break (0);
1861   return NULL;
1862 }
1863
1864
1865 /**
1866  * Kill the given MHD daemon.
1867  *
1868  * @param hd daemon to stop
1869  */
1870 static void
1871 kill_httpd (struct MhdHttpList *hd)
1872 {
1873   GNUNET_CONTAINER_DLL_remove (mhd_httpd_head,
1874                                mhd_httpd_tail,
1875                                hd);
1876   GNUNET_free_non_null (hd->domain);
1877   MHD_stop_daemon (hd->daemon);
1878   if (NULL != hd->httpd_task)
1879   {
1880     GNUNET_SCHEDULER_cancel (hd->httpd_task);
1881     hd->httpd_task = NULL;
1882   }
1883   GNUNET_free_non_null (hd->proxy_cert);
1884   if (hd == httpd)
1885     httpd = NULL;
1886   GNUNET_free (hd);
1887 }
1888
1889
1890 /**
1891  * Task run whenever HTTP server is idle for too long. Kill it.
1892  *
1893  * @param cls the `struct MhdHttpList *`
1894  */
1895 static void
1896 kill_httpd_task (void *cls)
1897 {
1898   struct MhdHttpList *hd = cls;
1899
1900   hd->httpd_task = NULL;
1901   kill_httpd (hd);
1902 }
1903
1904
1905 /**
1906  * Task run whenever HTTP server operations are pending.
1907  *
1908  * @param cls the `struct MhdHttpList *` of the daemon that is being run
1909  */
1910 static void
1911 do_httpd (void *cls);
1912
1913
1914 /**
1915  * Schedule MHD.  This function should be called initially when an
1916  * MHD is first getting its client socket, and will then automatically
1917  * always be called later whenever there is work to be done.
1918  *
1919  * @param hd the daemon to schedule
1920  */
1921 static void
1922 schedule_httpd (struct MhdHttpList *hd)
1923 {
1924   fd_set rs;
1925   fd_set ws;
1926   fd_set es;
1927   struct GNUNET_NETWORK_FDSet *wrs;
1928   struct GNUNET_NETWORK_FDSet *wws;
1929   int max;
1930   int haveto;
1931   MHD_UNSIGNED_LONG_LONG timeout;
1932   struct GNUNET_TIME_Relative tv;
1933
1934   FD_ZERO (&rs);
1935   FD_ZERO (&ws);
1936   FD_ZERO (&es);
1937   max = -1;
1938   if (MHD_YES != MHD_get_fdset (hd->daemon, &rs, &ws, &es, &max))
1939   {
1940     kill_httpd (hd);
1941     return;
1942   }
1943   haveto = MHD_get_timeout (hd->daemon, &timeout);
1944   if (MHD_YES == haveto)
1945     tv.rel_value_us = (uint64_t) timeout * 1000LL;
1946   else
1947     tv = GNUNET_TIME_UNIT_FOREVER_REL;
1948   if (-1 != max)
1949   {
1950     wrs = GNUNET_NETWORK_fdset_create ();
1951     wws = GNUNET_NETWORK_fdset_create ();
1952     GNUNET_NETWORK_fdset_copy_native (wrs, &rs, max + 1);
1953     GNUNET_NETWORK_fdset_copy_native (wws, &ws, max + 1);
1954   }
1955   else
1956   {
1957     wrs = NULL;
1958     wws = NULL;
1959   }
1960   if (NULL != hd->httpd_task)
1961     GNUNET_SCHEDULER_cancel (hd->httpd_task);
1962   if ( (MHD_YES != haveto) &&
1963        (-1 == max) &&
1964        (hd != httpd) )
1965   {
1966     /* daemon is idle, kill after timeout */
1967     hd->httpd_task = GNUNET_SCHEDULER_add_delayed (MHD_CACHE_TIMEOUT,
1968                                                    &kill_httpd_task,
1969                                                    hd);
1970   }
1971   else
1972   {
1973     hd->httpd_task =
1974       GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1975                                    tv, wrs, wws,
1976                                    &do_httpd, hd);
1977   }
1978   if (NULL != wrs)
1979     GNUNET_NETWORK_fdset_destroy (wrs);
1980   if (NULL != wws)
1981     GNUNET_NETWORK_fdset_destroy (wws);
1982 }
1983
1984
1985 /**
1986  * Task run whenever HTTP server operations are pending.
1987  *
1988  * @param cls the `struct MhdHttpList` of the daemon that is being run
1989  */
1990 static void
1991 do_httpd (void *cls)
1992 {
1993   struct MhdHttpList *hd = cls;
1994
1995   hd->httpd_task = NULL;
1996   MHD_run (hd->daemon);
1997   schedule_httpd (hd);
1998 }
1999
2000
2001 /**
2002  * Run MHD now, we have extra data ready for the callback.
2003  *
2004  * @param hd the daemon to run now.
2005  */
2006 static void
2007 run_mhd_now (struct MhdHttpList *hd)
2008 {
2009   if (NULL !=
2010       hd->httpd_task)
2011     GNUNET_SCHEDULER_cancel (hd->httpd_task);
2012   hd->httpd_task = GNUNET_SCHEDULER_add_now (&do_httpd,
2013                                              hd);
2014 }
2015
2016
2017 /**
2018  * Read file in filename
2019  *
2020  * @param filename file to read
2021  * @param size pointer where filesize is stored
2022  * @return NULL on error
2023  */
2024 static void*
2025 load_file (const char* filename,
2026            unsigned int* size)
2027 {
2028   void *buffer;
2029   uint64_t fsize;
2030
2031   if (GNUNET_OK !=
2032       GNUNET_DISK_file_size (filename, &fsize,
2033                              GNUNET_YES, GNUNET_YES))
2034     return NULL;
2035   if (fsize > MAX_PEM_SIZE)
2036     return NULL;
2037   *size = (unsigned int) fsize;
2038   buffer = GNUNET_malloc (*size);
2039   if (fsize != GNUNET_DISK_fn_read (filename, buffer, (size_t) fsize))
2040   {
2041     GNUNET_free (buffer);
2042     return NULL;
2043   }
2044   return buffer;
2045 }
2046
2047
2048 /**
2049  * Load PEM key from file
2050  *
2051  * @param key where to store the data
2052  * @param keyfile path to the PEM file
2053  * @return #GNUNET_OK on success
2054  */
2055 static int
2056 load_key_from_file (gnutls_x509_privkey_t key,
2057                     const char* keyfile)
2058 {
2059   gnutls_datum_t key_data;
2060   int ret;
2061
2062   key_data.data = load_file (keyfile, &key_data.size);
2063   if (NULL == key_data.data)
2064     return GNUNET_SYSERR;
2065   ret = gnutls_x509_privkey_import (key, &key_data,
2066                                     GNUTLS_X509_FMT_PEM);
2067   if (GNUTLS_E_SUCCESS != ret)
2068   {
2069     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2070                 _("Unable to import private key from file `%s'\n"),
2071                 keyfile);
2072   }
2073   GNUNET_free_non_null (key_data.data);
2074   return (GNUTLS_E_SUCCESS != ret) ? GNUNET_SYSERR : GNUNET_OK;
2075 }
2076
2077
2078 /**
2079  * Load cert from file
2080  *
2081  * @param crt struct to store data in
2082  * @param certfile path to pem file
2083  * @return #GNUNET_OK on success
2084  */
2085 static int
2086 load_cert_from_file (gnutls_x509_crt_t crt,
2087                      const char* certfile)
2088 {
2089   gnutls_datum_t cert_data;
2090   int ret;
2091
2092   cert_data.data = load_file (certfile, &cert_data.size);
2093   if (NULL == cert_data.data)
2094     return GNUNET_SYSERR;
2095   ret = gnutls_x509_crt_import (crt, &cert_data,
2096                                 GNUTLS_X509_FMT_PEM);
2097   if (GNUTLS_E_SUCCESS != ret)
2098   {
2099     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2100                 _("Unable to import certificate %s\n"), certfile);
2101   }
2102   GNUNET_free_non_null (cert_data.data);
2103   return (GNUTLS_E_SUCCESS != ret) ? GNUNET_SYSERR : GNUNET_OK;
2104 }
2105
2106
2107 /**
2108  * Generate new certificate for specific name
2109  *
2110  * @param name the subject name to generate a cert for
2111  * @return a struct holding the PEM data, NULL on error
2112  */
2113 static struct ProxyGNSCertificate *
2114 generate_gns_certificate (const char *name)
2115 {
2116   unsigned int serial;
2117   size_t key_buf_size;
2118   size_t cert_buf_size;
2119   gnutls_x509_crt_t request;
2120   time_t etime;
2121   struct tm *tm_data;
2122   struct ProxyGNSCertificate *pgc;
2123
2124   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2125               "Generating TLS/SSL certificate for `%s'\n",
2126               name);
2127   GNUNET_break (GNUTLS_E_SUCCESS == gnutls_x509_crt_init (&request));
2128   GNUNET_break (GNUTLS_E_SUCCESS == gnutls_x509_crt_set_key (request, proxy_ca.key));
2129   pgc = GNUNET_new (struct ProxyGNSCertificate);
2130   gnutls_x509_crt_set_dn_by_oid (request, GNUTLS_OID_X520_COUNTRY_NAME,
2131                                  0, "ZZ", 2);
2132   gnutls_x509_crt_set_dn_by_oid (request, GNUTLS_OID_X520_ORGANIZATION_NAME,
2133                                  0, "GNU Name System", 4);
2134   gnutls_x509_crt_set_dn_by_oid (request, GNUTLS_OID_X520_COMMON_NAME,
2135                                  0, name, strlen (name));
2136   GNUNET_break (GNUTLS_E_SUCCESS == gnutls_x509_crt_set_version (request, 3));
2137   gnutls_rnd (GNUTLS_RND_NONCE, &serial, sizeof (serial));
2138   gnutls_x509_crt_set_serial (request,
2139                               &serial,
2140                               sizeof (serial));
2141   etime = time (NULL);
2142   tm_data = localtime (&etime);
2143   gnutls_x509_crt_set_activation_time (request,
2144                                        etime);
2145   tm_data->tm_year++;
2146   etime = mktime (tm_data);
2147   gnutls_x509_crt_set_expiration_time (request,
2148                                        etime);
2149   gnutls_x509_crt_sign (request,
2150                         proxy_ca.cert,
2151                         proxy_ca.key);
2152   key_buf_size = sizeof (pgc->key);
2153   cert_buf_size = sizeof (pgc->cert);
2154   gnutls_x509_crt_export (request, GNUTLS_X509_FMT_PEM,
2155                           pgc->cert, &cert_buf_size);
2156   gnutls_x509_privkey_export (proxy_ca.key, GNUTLS_X509_FMT_PEM,
2157                               pgc->key, &key_buf_size);
2158   gnutls_x509_crt_deinit (request);
2159   return pgc;
2160 }
2161
2162
2163 /**
2164  * Function called by MHD with errors, suppresses them all.
2165  *
2166  * @param cls closure
2167  * @param fm format string (`printf()`-style)
2168  * @param ap arguments to @a fm
2169  */
2170 static void
2171 mhd_error_log_callback (void *cls,
2172                         const char *fm,
2173                         va_list ap)
2174 {
2175   /* do nothing */
2176 }
2177
2178
2179 /**
2180  * Lookup (or create) an SSL MHD instance for a particular domain.
2181  *
2182  * @param domain the domain the SSL daemon has to serve
2183  * @return NULL on error
2184  */
2185 static struct MhdHttpList *
2186 lookup_ssl_httpd (const char* domain)
2187 {
2188   struct MhdHttpList *hd;
2189   struct ProxyGNSCertificate *pgc;
2190
2191   if (NULL == domain)
2192   {
2193     GNUNET_break (0);
2194     return NULL;
2195   }
2196   for (hd = mhd_httpd_head; NULL != hd; hd = hd->next)
2197     if ( (NULL != hd->domain) &&
2198          (0 == strcmp (hd->domain, domain)) )
2199       return hd;
2200   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2201               "Starting fresh MHD HTTPS instance for domain `%s'\n",
2202               domain);
2203   pgc = generate_gns_certificate (domain);
2204   hd = GNUNET_new (struct MhdHttpList);
2205   hd->is_ssl = GNUNET_YES;
2206   hd->domain = GNUNET_strdup (domain);
2207   hd->proxy_cert = pgc;
2208   hd->daemon = MHD_start_daemon (MHD_USE_DEBUG | MHD_USE_SSL | MHD_USE_NO_LISTEN_SOCKET,
2209                                  0,
2210                                  NULL, NULL,
2211                                  &create_response, hd,
2212                                  MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
2213                                  MHD_OPTION_NOTIFY_COMPLETED, &mhd_completed_cb, NULL,
2214                                  MHD_OPTION_NOTIFY_CONNECTION, &mhd_connection_cb, NULL,
2215                                  MHD_OPTION_URI_LOG_CALLBACK, &mhd_log_callback, NULL,
2216                                  MHD_OPTION_EXTERNAL_LOGGER, &mhd_error_log_callback, NULL,
2217                                  MHD_OPTION_HTTPS_MEM_KEY, pgc->key,
2218                                  MHD_OPTION_HTTPS_MEM_CERT, pgc->cert,
2219                                  MHD_OPTION_END);
2220   if (NULL == hd->daemon)
2221   {
2222     GNUNET_free (pgc);
2223     GNUNET_free (hd);
2224     return NULL;
2225   }
2226   GNUNET_CONTAINER_DLL_insert (mhd_httpd_head,
2227                                mhd_httpd_tail,
2228                                hd);
2229   return hd;
2230 }
2231
2232
2233 /**
2234  * Task run when a Socks5Request somehow fails to be associated with
2235  * an MHD connection (i.e. because the client never speaks HTTP after
2236  * the SOCKS5 handshake).  Clean up.
2237  *
2238  * @param cls the `struct Socks5Request *`
2239  */
2240 static void
2241 timeout_s5r_handshake (void *cls)
2242 {
2243   struct Socks5Request *s5r = cls;
2244
2245   s5r->timeout_task = NULL;
2246   cleanup_s5r (s5r);
2247 }
2248
2249
2250 /**
2251  * We're done with the Socks5 protocol, now we need to pass the
2252  * connection data through to the final destination, either
2253  * direct (if the protocol might not be HTTP), or via MHD
2254  * (if the port looks like it should be HTTP).
2255  *
2256  * @param s5r socks request that has reached the final stage
2257  */
2258 static void
2259 setup_data_transfer (struct Socks5Request *s5r)
2260 {
2261   struct MhdHttpList *hd;
2262   int fd;
2263   const struct sockaddr *addr;
2264   socklen_t len;
2265
2266   switch (s5r->port)
2267   {
2268     case HTTPS_PORT:
2269       hd = lookup_ssl_httpd (s5r->domain);
2270       if (NULL == hd)
2271       {
2272         GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2273                     _("Failed to start HTTPS server for `%s'\n"),
2274                     s5r->domain);
2275         cleanup_s5r (s5r);
2276         return;
2277       }
2278       break;
2279     case HTTP_PORT:
2280     default:
2281       GNUNET_assert (NULL != httpd);
2282       hd = httpd;
2283       break;
2284   }
2285   fd = GNUNET_NETWORK_get_fd (s5r->sock);
2286   addr = GNUNET_NETWORK_get_addr (s5r->sock);
2287   len = GNUNET_NETWORK_get_addrlen (s5r->sock);
2288   s5r->state = SOCKS5_SOCKET_WITH_MHD;
2289   if (MHD_YES != MHD_add_connection (hd->daemon, fd, addr, len))
2290   {
2291     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
2292                 _("Failed to pass client to MHD\n"));
2293     cleanup_s5r (s5r);
2294     return;
2295   }
2296   s5r->hd = hd;
2297   schedule_httpd (hd);
2298   s5r->timeout_task = GNUNET_SCHEDULER_add_delayed (HTTP_HANDSHAKE_TIMEOUT,
2299                                                     &timeout_s5r_handshake,
2300                                                     s5r);
2301 }
2302
2303
2304 /* ********************* SOCKS handling ************************* */
2305
2306
2307 /**
2308  * Write data from buffer to socks5 client, then continue with state machine.
2309  *
2310  * @param cls the closure with the `struct Socks5Request`
2311  */
2312 static void
2313 do_write (void *cls)
2314 {
2315   struct Socks5Request *s5r = cls;
2316   ssize_t len;
2317
2318   s5r->wtask = NULL;
2319   len = GNUNET_NETWORK_socket_send (s5r->sock,
2320                                     s5r->wbuf,
2321                                     s5r->wbuf_len);
2322   if (len <= 0)
2323   {
2324     /* write error: connection closed, shutdown, etc.; just clean up */
2325     cleanup_s5r (s5r);
2326     return;
2327   }
2328   memmove (s5r->wbuf,
2329            &s5r->wbuf[len],
2330            s5r->wbuf_len - len);
2331   s5r->wbuf_len -= len;
2332   if (s5r->wbuf_len > 0)
2333   {
2334     /* not done writing */
2335     s5r->wtask =
2336       GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2337                                       s5r->sock,
2338                                       &do_write, s5r);
2339     return;
2340   }
2341
2342   /* we're done writing, continue with state machine! */
2343
2344   switch (s5r->state)
2345   {
2346     case SOCKS5_INIT:
2347       GNUNET_assert (0);
2348       break;
2349     case SOCKS5_REQUEST:
2350       GNUNET_assert (NULL != s5r->rtask);
2351       break;
2352     case SOCKS5_DATA_TRANSFER:
2353       setup_data_transfer (s5r);
2354       return;
2355     case SOCKS5_WRITE_THEN_CLEANUP:
2356       cleanup_s5r (s5r);
2357       return;
2358     default:
2359       GNUNET_break (0);
2360       break;
2361   }
2362 }
2363
2364
2365 /**
2366  * Return a server response message indicating a failure to the client.
2367  *
2368  * @param s5r request to return failure code for
2369  * @param sc status code to return
2370  */
2371 static void
2372 signal_socks_failure (struct Socks5Request *s5r,
2373                       enum Socks5StatusCode sc)
2374 {
2375   struct Socks5ServerResponseMessage *s_resp;
2376
2377   s_resp = (struct Socks5ServerResponseMessage *) &s5r->wbuf[s5r->wbuf_len];
2378   memset (s_resp, 0, sizeof (struct Socks5ServerResponseMessage));
2379   s_resp->version = SOCKS_VERSION_5;
2380   s_resp->reply = sc;
2381   s5r->state = SOCKS5_WRITE_THEN_CLEANUP;
2382   if (NULL != s5r->wtask)
2383     s5r->wtask =
2384       GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2385                                       s5r->sock,
2386                                       &do_write, s5r);
2387 }
2388
2389
2390 /**
2391  * Return a server response message indicating success.
2392  *
2393  * @param s5r request to return success status message for
2394  */
2395 static void
2396 signal_socks_success (struct Socks5Request *s5r)
2397 {
2398   struct Socks5ServerResponseMessage *s_resp;
2399
2400   s_resp = (struct Socks5ServerResponseMessage *) &s5r->wbuf[s5r->wbuf_len];
2401   s_resp->version = SOCKS_VERSION_5;
2402   s_resp->reply = SOCKS5_STATUS_REQUEST_GRANTED;
2403   s_resp->reserved = 0;
2404   s_resp->addr_type = SOCKS5_AT_IPV4;
2405   /* zero out IPv4 address and port */
2406   memset (&s_resp[1],
2407           0,
2408           sizeof (struct in_addr) + sizeof (uint16_t));
2409   s5r->wbuf_len += sizeof (struct Socks5ServerResponseMessage) +
2410     sizeof (struct in_addr) + sizeof (uint16_t);
2411   if (NULL == s5r->wtask)
2412     s5r->wtask =
2413       GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2414                                       s5r->sock,
2415                                       &do_write, s5r);
2416 }
2417
2418
2419 /**
2420  * Process GNS results for target domain.
2421  *
2422  * @param cls the `struct Socks5Request *`
2423  * @param rd_count number of records returned
2424  * @param rd record data
2425  */
2426 static void
2427 handle_gns_result (void *cls,
2428                    uint32_t rd_count,
2429                    const struct GNUNET_GNSRECORD_Data *rd)
2430 {
2431   struct Socks5Request *s5r = cls;
2432   uint32_t i;
2433   const struct GNUNET_GNSRECORD_Data *r;
2434   int got_ip;
2435
2436   s5r->gns_lookup = NULL;
2437   got_ip = GNUNET_NO;
2438   for (i=0;i<rd_count;i++)
2439   {
2440     r = &rd[i];
2441     switch (r->record_type)
2442     {
2443       case GNUNET_DNSPARSER_TYPE_A:
2444         {
2445           struct sockaddr_in *in;
2446
2447           if (sizeof (struct in_addr) != r->data_size)
2448           {
2449             GNUNET_break_op (0);
2450             break;
2451           }
2452           if (GNUNET_YES == got_ip)
2453             break;
2454           if (GNUNET_OK !=
2455               GNUNET_NETWORK_test_pf (PF_INET))
2456             break;
2457           got_ip = GNUNET_YES;
2458           in = (struct sockaddr_in *) &s5r->destination_address;
2459           in->sin_family = AF_INET;
2460           GNUNET_memcpy (&in->sin_addr,
2461                          r->data,
2462                          r->data_size);
2463           in->sin_port = htons (s5r->port);
2464 #if HAVE_SOCKADDR_IN_SIN_LEN
2465           in->sin_len = sizeof (*in);
2466 #endif
2467         }
2468         break;
2469       case GNUNET_DNSPARSER_TYPE_AAAA:
2470         {
2471           struct sockaddr_in6 *in;
2472
2473           if (sizeof (struct in6_addr) != r->data_size)
2474           {
2475             GNUNET_break_op (0);
2476             break;
2477           }
2478           if (GNUNET_YES == got_ip)
2479             break;
2480           if (GNUNET_OK !=
2481               GNUNET_NETWORK_test_pf (PF_INET))
2482             break;
2483           /* FIXME: allow user to disable IPv6 per configuration option... */
2484           got_ip = GNUNET_YES;
2485           in = (struct sockaddr_in6 *) &s5r->destination_address;
2486           in->sin6_family = AF_INET6;
2487           GNUNET_memcpy (&in->sin6_addr,
2488                          r->data,
2489                          r->data_size);
2490           in->sin6_port = htons (s5r->port);
2491 #if HAVE_SOCKADDR_IN_SIN_LEN
2492           in->sin6_len = sizeof (*in);
2493 #endif
2494         }
2495         break;
2496       case GNUNET_GNSRECORD_TYPE_VPN:
2497         GNUNET_break (0); /* should have been translated within GNS */
2498         break;
2499       case GNUNET_GNSRECORD_TYPE_LEHO:
2500         GNUNET_free_non_null (s5r->leho);
2501         s5r->leho = GNUNET_strndup (r->data,
2502                                     r->data_size);
2503         break;
2504       case GNUNET_GNSRECORD_TYPE_BOX:
2505         {
2506           const struct GNUNET_GNSRECORD_BoxRecord *box;
2507
2508           if (r->data_size < sizeof (struct GNUNET_GNSRECORD_BoxRecord))
2509           {
2510             GNUNET_break_op (0);
2511             break;
2512           }
2513           box = r->data;
2514           if ( (ntohl (box->record_type) != GNUNET_DNSPARSER_TYPE_TLSA) ||
2515                (ntohs (box->protocol) != IPPROTO_TCP) ||
2516                (ntohs (box->service) != s5r->port) )
2517             break; /* BOX record does not apply */
2518           GNUNET_free_non_null (s5r->dane_data);
2519           s5r->dane_data_len = r->data_size - sizeof (struct GNUNET_GNSRECORD_BoxRecord);
2520           s5r->dane_data = GNUNET_malloc (s5r->dane_data_len);
2521           GNUNET_memcpy (s5r->dane_data,
2522                          &box[1],
2523                          s5r->dane_data_len);
2524           break;
2525         }
2526       default:
2527         /* don't care */
2528         break;
2529     }
2530   }
2531   if (GNUNET_YES != got_ip)
2532   {
2533     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2534                 "Name resolution failed to yield useful IP address.\n");
2535     signal_socks_failure (s5r,
2536                           SOCKS5_STATUS_GENERAL_FAILURE);
2537     return;
2538   }
2539   s5r->state = SOCKS5_DATA_TRANSFER;
2540   signal_socks_success (s5r);
2541 }
2542
2543
2544 /**
2545  * Remove the first @a len bytes from the beginning of the read buffer.
2546  *
2547  * @param s5r the handle clear the read buffer for
2548  * @param len number of bytes in read buffer to advance
2549  */
2550 static void
2551 clear_from_s5r_rbuf (struct Socks5Request *s5r,
2552                      size_t len)
2553 {
2554   GNUNET_assert (len <= s5r->rbuf_len);
2555   memmove (s5r->rbuf,
2556            &s5r->rbuf[len],
2557            s5r->rbuf_len - len);
2558   s5r->rbuf_len -= len;
2559 }
2560
2561
2562 /**
2563  * Read data from incoming Socks5 connection
2564  *
2565  * @param cls the closure with the `struct Socks5Request`
2566  */
2567 static void
2568 do_s5r_read (void *cls)
2569 {
2570   struct Socks5Request *s5r = cls;
2571   const struct Socks5ClientHelloMessage *c_hello;
2572   struct Socks5ServerHelloMessage *s_hello;
2573   const struct Socks5ClientRequestMessage *c_req;
2574   ssize_t rlen;
2575   size_t alen;
2576   const struct GNUNET_SCHEDULER_TaskContext *tc;
2577
2578   s5r->rtask = NULL;
2579   tc = GNUNET_SCHEDULER_get_task_context ();
2580   if ( (NULL != tc->read_ready) &&
2581        (GNUNET_NETWORK_fdset_isset (tc->read_ready, s5r->sock)) )
2582   {
2583     rlen = GNUNET_NETWORK_socket_recv (s5r->sock,
2584                                        &s5r->rbuf[s5r->rbuf_len],
2585                                        sizeof (s5r->rbuf) - s5r->rbuf_len);
2586     if (rlen <= 0)
2587     {
2588       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2589                   "socks5 client disconnected.\n");
2590       cleanup_s5r (s5r);
2591       return;
2592     }
2593     s5r->rbuf_len += rlen;
2594   }
2595   s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2596                                               s5r->sock,
2597                                               &do_s5r_read, s5r);
2598   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2599               "Processing %zu bytes of socks data in state %d\n",
2600               s5r->rbuf_len,
2601               s5r->state);
2602   switch (s5r->state)
2603   {
2604     case SOCKS5_INIT:
2605       c_hello = (const struct Socks5ClientHelloMessage*) &s5r->rbuf;
2606       if ( (s5r->rbuf_len < sizeof (struct Socks5ClientHelloMessage)) ||
2607            (s5r->rbuf_len < sizeof (struct Socks5ClientHelloMessage) + c_hello->num_auth_methods) )
2608         return; /* need more data */
2609       if (SOCKS_VERSION_5 != c_hello->version)
2610       {
2611         GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2612                     _("Unsupported socks version %d\n"),
2613                     (int) c_hello->version);
2614         cleanup_s5r (s5r);
2615         return;
2616       }
2617       clear_from_s5r_rbuf (s5r,
2618                            sizeof (struct Socks5ClientHelloMessage) + c_hello->num_auth_methods);
2619       GNUNET_assert (0 == s5r->wbuf_len);
2620       s_hello = (struct Socks5ServerHelloMessage *) &s5r->wbuf;
2621       s5r->wbuf_len = sizeof (struct Socks5ServerHelloMessage);
2622       s_hello->version = SOCKS_VERSION_5;
2623       s_hello->auth_method = SOCKS_AUTH_NONE;
2624       GNUNET_assert (NULL == s5r->wtask);
2625       s5r->wtask = GNUNET_SCHEDULER_add_write_net (GNUNET_TIME_UNIT_FOREVER_REL,
2626                                                    s5r->sock,
2627                                                    &do_write, s5r);
2628       s5r->state = SOCKS5_REQUEST;
2629       return;
2630     case SOCKS5_REQUEST:
2631       c_req = (const struct Socks5ClientRequestMessage *) &s5r->rbuf;
2632       if (s5r->rbuf_len < sizeof (struct Socks5ClientRequestMessage))
2633         return;
2634       switch (c_req->command)
2635       {
2636         case SOCKS5_CMD_TCP_STREAM:
2637           /* handled below */
2638           break;
2639         default:
2640           GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2641                       _("Unsupported socks command %d\n"),
2642                       (int) c_req->command);
2643           signal_socks_failure (s5r,
2644                                 SOCKS5_STATUS_COMMAND_NOT_SUPPORTED);
2645           return;
2646       }
2647       switch (c_req->addr_type)
2648       {
2649         case SOCKS5_AT_IPV4:
2650           {
2651             const struct in_addr *v4 = (const struct in_addr *) &c_req[1];
2652             const uint16_t *port = (const uint16_t *) &v4[1];
2653             struct sockaddr_in *in;
2654
2655             s5r->port = ntohs (*port);
2656             if (HTTPS_PORT == s5r->port)
2657             {
2658               GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2659                           _("SSL connection to plain IPv4 address requested\n"));
2660               signal_socks_failure (s5r,
2661                                     SOCKS5_STATUS_CONNECTION_NOT_ALLOWED_BY_RULE);
2662               return;
2663             }
2664             alen = sizeof (struct in_addr);
2665             if (s5r->rbuf_len < sizeof (struct Socks5ClientRequestMessage) +
2666                 alen + sizeof (uint16_t))
2667               return; /* need more data */
2668             in = (struct sockaddr_in *) &s5r->destination_address;
2669             in->sin_family = AF_INET;
2670             in->sin_addr = *v4;
2671             in->sin_port = *port;
2672 #if HAVE_SOCKADDR_IN_SIN_LEN
2673             in->sin_len = sizeof (*in);
2674 #endif
2675             s5r->state = SOCKS5_DATA_TRANSFER;
2676           }
2677           break;
2678         case SOCKS5_AT_IPV6:
2679           {
2680             const struct in6_addr *v6 = (const struct in6_addr *) &c_req[1];
2681             const uint16_t *port = (const uint16_t *) &v6[1];
2682             struct sockaddr_in6 *in;
2683
2684             s5r->port = ntohs (*port);
2685             if (HTTPS_PORT == s5r->port)
2686             {
2687               GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2688                           _("SSL connection to plain IPv4 address requested\n"));
2689               signal_socks_failure (s5r,
2690                                     SOCKS5_STATUS_CONNECTION_NOT_ALLOWED_BY_RULE);
2691               return;
2692             }
2693             alen = sizeof (struct in6_addr);
2694             if (s5r->rbuf_len < sizeof (struct Socks5ClientRequestMessage) +
2695                 alen + sizeof (uint16_t))
2696               return; /* need more data */
2697             in = (struct sockaddr_in6 *) &s5r->destination_address;
2698             in->sin6_family = AF_INET6;
2699             in->sin6_addr = *v6;
2700             in->sin6_port = *port;
2701 #if HAVE_SOCKADDR_IN_SIN_LEN
2702             in->sin6_len = sizeof (*in);
2703 #endif
2704             s5r->state = SOCKS5_DATA_TRANSFER;
2705           }
2706           break;
2707         case SOCKS5_AT_DOMAINNAME:
2708           {
2709             const uint8_t *dom_len;
2710             const char *dom_name;
2711             const uint16_t *port;
2712
2713             dom_len = (const uint8_t *) &c_req[1];
2714             alen = *dom_len + 1;
2715             if (s5r->rbuf_len < sizeof (struct Socks5ClientRequestMessage) +
2716                 alen + sizeof (uint16_t))
2717               return; /* need more data */
2718             dom_name = (const char *) &dom_len[1];
2719             port = (const uint16_t*) &dom_name[*dom_len];
2720             s5r->domain = GNUNET_strndup (dom_name, *dom_len);
2721             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2722                         "Requested connection is to %s:%d\n",
2723                         s5r->domain,
2724                         ntohs (*port));
2725             s5r->state = SOCKS5_RESOLVING;
2726             s5r->port = ntohs (*port);
2727             s5r->gns_lookup = GNUNET_GNS_lookup (gns_handle,
2728                                                  s5r->domain,
2729                                                  &local_gns_zone,
2730                                                  GNUNET_DNSPARSER_TYPE_A,
2731                                                  GNUNET_NO /* only cached */,
2732                                                  (GNUNET_YES == do_shorten) ? &local_shorten_zone : NULL,
2733                                                  &handle_gns_result,
2734                                                  s5r);
2735             break;
2736           }
2737         default:
2738           GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
2739                       _("Unsupported socks address type %d\n"),
2740                       (int) c_req->addr_type);
2741           signal_socks_failure (s5r,
2742                                 SOCKS5_STATUS_ADDRESS_TYPE_NOT_SUPPORTED);
2743           return;
2744       }
2745       clear_from_s5r_rbuf (s5r,
2746                            sizeof (struct Socks5ClientRequestMessage) +
2747                            alen + sizeof (uint16_t));
2748       if (0 != s5r->rbuf_len)
2749       {
2750         /* read more bytes than healthy, why did the client send more!? */
2751         GNUNET_break_op (0);
2752         signal_socks_failure (s5r,
2753                               SOCKS5_STATUS_GENERAL_FAILURE);
2754         return;
2755       }
2756       if (SOCKS5_DATA_TRANSFER == s5r->state)
2757       {
2758         /* if we are not waiting for GNS resolution, signal success */
2759         signal_socks_success (s5r);
2760       }
2761       /* We are done reading right now */
2762       GNUNET_SCHEDULER_cancel (s5r->rtask);
2763       s5r->rtask = NULL;
2764       return;
2765     case SOCKS5_RESOLVING:
2766       GNUNET_assert (0);
2767       return;
2768     case SOCKS5_DATA_TRANSFER:
2769       GNUNET_assert (0);
2770       return;
2771     default:
2772       GNUNET_assert (0);
2773       return;
2774   }
2775 }
2776
2777
2778 /**
2779  * Accept new incoming connections
2780  *
2781  * @param cls the closure with the lsock4 or lsock6
2782  * @param tc the scheduler context
2783  */
2784 static void
2785 do_accept (void *cls)
2786 {
2787   struct GNUNET_NETWORK_Handle *lsock = cls;
2788   struct GNUNET_NETWORK_Handle *s;
2789   struct Socks5Request *s5r;
2790
2791   if (lsock == lsock4)
2792     ltask4 = NULL;
2793   else
2794     ltask6 = NULL;
2795   if (lsock == lsock4)
2796     ltask4 = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2797                                             lsock,
2798                                             &do_accept, lsock);
2799   else
2800     ltask6 = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2801                                             lsock,
2802                                             &do_accept, lsock);
2803   s = GNUNET_NETWORK_socket_accept (lsock, NULL, NULL);
2804   if (NULL == s)
2805   {
2806     GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "accept");
2807     return;
2808   }
2809   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
2810               "Got an inbound connection, waiting for data\n");
2811   s5r = GNUNET_new (struct Socks5Request);
2812   GNUNET_CONTAINER_DLL_insert (s5r_head,
2813                                s5r_tail,
2814                                s5r);
2815   s5r->sock = s;
2816   s5r->state = SOCKS5_INIT;
2817   s5r->rtask = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2818                                               s5r->sock,
2819                                               &do_s5r_read, s5r);
2820 }
2821
2822
2823 /* ******************* General / main code ********************* */
2824
2825
2826 /**
2827  * Task run on shutdown
2828  *
2829  * @param cls closure
2830  */
2831 static void
2832 do_shutdown (void *cls)
2833 {
2834   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
2835               "Shutting down...\n");
2836   while (NULL != mhd_httpd_head)
2837     kill_httpd (mhd_httpd_head);
2838   while (NULL != s5r_head)
2839     cleanup_s5r (s5r_head);
2840   if (NULL != lsock4)
2841   {
2842     GNUNET_NETWORK_socket_close (lsock4);
2843     lsock4 = NULL;
2844   }
2845   if (NULL != lsock6)
2846   {
2847     GNUNET_NETWORK_socket_close (lsock6);
2848     lsock6 = NULL;
2849   }
2850   if (NULL != id_op)
2851   {
2852     GNUNET_IDENTITY_cancel (id_op);
2853     id_op = NULL;
2854   }
2855   if (NULL != identity)
2856   {
2857     GNUNET_IDENTITY_disconnect (identity);
2858     identity = NULL;
2859   }
2860   if (NULL != curl_multi)
2861   {
2862     curl_multi_cleanup (curl_multi);
2863     curl_multi = NULL;
2864   }
2865   if (NULL != gns_handle)
2866   {
2867     GNUNET_GNS_disconnect (gns_handle);
2868     gns_handle = NULL;
2869   }
2870   if (NULL != curl_download_task)
2871   {
2872     GNUNET_SCHEDULER_cancel (curl_download_task);
2873     curl_download_task = NULL;
2874   }
2875   if (NULL != ltask4)
2876   {
2877     GNUNET_SCHEDULER_cancel (ltask4);
2878     ltask4 = NULL;
2879   }
2880   if (NULL != ltask6)
2881   {
2882     GNUNET_SCHEDULER_cancel (ltask6);
2883     ltask6 = NULL;
2884   }
2885   gnutls_x509_crt_deinit (proxy_ca.cert);
2886   gnutls_x509_privkey_deinit (proxy_ca.key);
2887   gnutls_global_deinit ();
2888 }
2889
2890
2891 /**
2892  * Create an IPv4 listen socket bound to our port.
2893  *
2894  * @return NULL on error
2895  */
2896 static struct GNUNET_NETWORK_Handle *
2897 bind_v4 ()
2898 {
2899   struct GNUNET_NETWORK_Handle *ls;
2900   struct sockaddr_in sa4;
2901   int eno;
2902
2903   memset (&sa4, 0, sizeof (sa4));
2904   sa4.sin_family = AF_INET;
2905   sa4.sin_port = htons (port);
2906 #if HAVE_SOCKADDR_IN_SIN_LEN
2907   sa4.sin_len = sizeof (sa4);
2908 #endif
2909   ls = GNUNET_NETWORK_socket_create (AF_INET,
2910                                      SOCK_STREAM,
2911                                      0);
2912   if (NULL == ls)
2913     return NULL;
2914   if (GNUNET_OK !=
2915       GNUNET_NETWORK_socket_bind (ls, (const struct sockaddr *) &sa4,
2916                                   sizeof (sa4)))
2917   {
2918     eno = errno;
2919     GNUNET_NETWORK_socket_close (ls);
2920     errno = eno;
2921     return NULL;
2922   }
2923   return ls;
2924 }
2925
2926
2927 /**
2928  * Create an IPv6 listen socket bound to our port.
2929  *
2930  * @return NULL on error
2931  */
2932 static struct GNUNET_NETWORK_Handle *
2933 bind_v6 ()
2934 {
2935   struct GNUNET_NETWORK_Handle *ls;
2936   struct sockaddr_in6 sa6;
2937   int eno;
2938
2939   memset (&sa6, 0, sizeof (sa6));
2940   sa6.sin6_family = AF_INET6;
2941   sa6.sin6_port = htons (port);
2942 #if HAVE_SOCKADDR_IN_SIN_LEN
2943   sa6.sin6_len = sizeof (sa6);
2944 #endif
2945   ls = GNUNET_NETWORK_socket_create (AF_INET6,
2946                                      SOCK_STREAM,
2947                                      0);
2948   if (NULL == ls)
2949     return NULL;
2950   if (GNUNET_OK !=
2951       GNUNET_NETWORK_socket_bind (ls, (const struct sockaddr *) &sa6,
2952                                   sizeof (sa6)))
2953   {
2954     eno = errno;
2955     GNUNET_NETWORK_socket_close (ls);
2956     errno = eno;
2957     return NULL;
2958   }
2959   return ls;
2960 }
2961
2962
2963 /**
2964  * Continue initialization after we have our zone information.
2965  */
2966 static void
2967 run_cont ()
2968 {
2969   struct MhdHttpList *hd;
2970
2971   /* Open listen socket for socks proxy */
2972   lsock6 = bind_v6 ();
2973   if (NULL == lsock6)
2974     GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "bind");
2975   else
2976   {
2977     if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock6, 5))
2978     {
2979       GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "listen");
2980       GNUNET_NETWORK_socket_close (lsock6);
2981       lsock6 = NULL;
2982     }
2983     else
2984     {
2985       ltask6 = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
2986                                               lsock6, &do_accept, lsock6);
2987     }
2988   }
2989   lsock4 = bind_v4 ();
2990   if (NULL == lsock4)
2991     GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "bind");
2992   else
2993   {
2994     if (GNUNET_OK != GNUNET_NETWORK_socket_listen (lsock4, 5))
2995     {
2996       GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "listen");
2997       GNUNET_NETWORK_socket_close (lsock4);
2998       lsock4 = NULL;
2999     }
3000     else
3001     {
3002       ltask4 = GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
3003                                               lsock4, &do_accept, lsock4);
3004     }
3005   }
3006   if ( (NULL == lsock4) &&
3007        (NULL == lsock6) )
3008   {
3009     GNUNET_SCHEDULER_shutdown ();
3010     return;
3011   }
3012   if (0 != curl_global_init (CURL_GLOBAL_WIN32))
3013   {
3014     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3015                 "cURL global init failed!\n");
3016     GNUNET_SCHEDULER_shutdown ();
3017     return;
3018   }
3019   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3020               "Proxy listens on port %lu\n",
3021               port);
3022
3023   /* start MHD daemon for HTTP */
3024   hd = GNUNET_new (struct MhdHttpList);
3025   hd->daemon = MHD_start_daemon (MHD_USE_DEBUG | MHD_USE_NO_LISTEN_SOCKET,
3026                                  0,
3027                                  NULL, NULL,
3028                                  &create_response, hd,
3029                                  MHD_OPTION_CONNECTION_TIMEOUT, (unsigned int) 16,
3030                                  MHD_OPTION_NOTIFY_COMPLETED, &mhd_completed_cb, NULL,
3031                                  MHD_OPTION_NOTIFY_CONNECTION, &mhd_connection_cb, NULL,
3032                                  MHD_OPTION_URI_LOG_CALLBACK, &mhd_log_callback, NULL,
3033                                  MHD_OPTION_END);
3034   if (NULL == hd->daemon)
3035   {
3036     GNUNET_free (hd);
3037     GNUNET_SCHEDULER_shutdown ();
3038     return;
3039   }
3040   httpd = hd;
3041   GNUNET_CONTAINER_DLL_insert (mhd_httpd_head, mhd_httpd_tail, hd);
3042 }
3043
3044
3045 /**
3046  * Method called to inform about the egos of the shorten zone of this peer.
3047  *
3048  * When used with #GNUNET_IDENTITY_create or #GNUNET_IDENTITY_get,
3049  * this function is only called ONCE, and 'NULL' being passed in
3050  * @a ego does indicate an error (i.e. name is taken or no default
3051  * value is known).  If @a ego is non-NULL and if '*ctx'
3052  * is set in those callbacks, the value WILL be passed to a subsequent
3053  * call to the identity callback of #GNUNET_IDENTITY_connect (if
3054  * that one was not NULL).
3055  *
3056  * @param cls closure, NULL
3057  * @param ego ego handle
3058  * @param ctx context for application to store data for this ego
3059  *                 (during the lifetime of this process, initially NULL)
3060  * @param name name assigned by the user for this ego,
3061  *                   NULL if the user just deleted the ego and it
3062  *                   must thus no longer be used
3063  */
3064 static void
3065 identity_shorten_cb (void *cls,
3066                      struct GNUNET_IDENTITY_Ego *ego,
3067                      void **ctx,
3068                      const char *name)
3069 {
3070   id_op = NULL;
3071   if (NULL == ego)
3072   {
3073     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
3074                 _("No ego configured for `shorten-zone`\n"));
3075   }
3076   else
3077   {
3078     local_shorten_zone = *GNUNET_IDENTITY_ego_get_private_key (ego);
3079     do_shorten = GNUNET_YES;
3080   }
3081   run_cont ();
3082 }
3083
3084
3085 /**
3086  * Method called to inform about the egos of the master zone of this peer.
3087  *
3088  * When used with #GNUNET_IDENTITY_create or #GNUNET_IDENTITY_get,
3089  * this function is only called ONCE, and 'NULL' being passed in
3090  * @a ego does indicate an error (i.e. name is taken or no default
3091  * value is known).  If @a ego is non-NULL and if '*ctx'
3092  * is set in those callbacks, the value WILL be passed to a subsequent
3093  * call to the identity callback of #GNUNET_IDENTITY_connect (if
3094  * that one was not NULL).
3095  *
3096  * @param cls closure, NULL
3097  * @param ego ego handle
3098  * @param ctx context for application to store data for this ego
3099  *                 (during the lifetime of this process, initially NULL)
3100  * @param name name assigned by the user for this ego,
3101  *                   NULL if the user just deleted the ego and it
3102  *                   must thus no longer be used
3103  */
3104 static void
3105 identity_master_cb (void *cls,
3106                     struct GNUNET_IDENTITY_Ego *ego,
3107                     void **ctx,
3108                     const char *name)
3109 {
3110   id_op = NULL;
3111   if (NULL == ego)
3112   {
3113     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3114                 _("No ego configured for `%s`\n"),
3115                 "gns-proxy");
3116     GNUNET_SCHEDULER_shutdown ();
3117     return;
3118   }
3119   GNUNET_IDENTITY_ego_get_public_key (ego,
3120                                       &local_gns_zone);
3121   id_op = GNUNET_IDENTITY_get (identity,
3122                                "gns-short",
3123                                &identity_shorten_cb,
3124                                NULL);
3125 }
3126
3127
3128 /**
3129  * Main function that will be run
3130  *
3131  * @param cls closure
3132  * @param args remaining command-line arguments
3133  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
3134  * @param c configuration
3135  */
3136 static void
3137 run (void *cls, char *const *args, const char *cfgfile,
3138      const struct GNUNET_CONFIGURATION_Handle *c)
3139 {
3140   char* cafile_cfg = NULL;
3141   char* cafile;
3142
3143   cfg = c;
3144
3145   if (NULL == (curl_multi = curl_multi_init ()))
3146   {
3147     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3148                 "Failed to create cURL multi handle!\n");
3149     return;
3150   }
3151   cafile = cafile_opt;
3152   if (NULL == cafile)
3153   {
3154     if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns-proxy",
3155                                                               "PROXY_CACERT",
3156                                                               &cafile_cfg))
3157     {
3158       GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR,
3159                                  "gns-proxy",
3160                                  "PROXY_CACERT");
3161       return;
3162     }
3163     cafile = cafile_cfg;
3164   }
3165   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
3166               "Using %s as CA\n", cafile);
3167
3168   gnutls_global_init ();
3169   gnutls_x509_crt_init (&proxy_ca.cert);
3170   gnutls_x509_privkey_init (&proxy_ca.key);
3171
3172   if ( (GNUNET_OK != load_cert_from_file (proxy_ca.cert, cafile)) ||
3173        (GNUNET_OK != load_key_from_file (proxy_ca.key, cafile)) )
3174   {
3175     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3176                 _("Failed to load SSL/TLS key and certificate from `%s'\n"),
3177                 cafile);
3178     gnutls_x509_crt_deinit (proxy_ca.cert);
3179     gnutls_x509_privkey_deinit (proxy_ca.key);
3180     gnutls_global_deinit ();
3181     GNUNET_free_non_null (cafile_cfg);
3182     return;
3183   }
3184   GNUNET_free_non_null (cafile_cfg);
3185   if (NULL == (gns_handle = GNUNET_GNS_connect (cfg)))
3186   {
3187     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
3188                 "Unable to connect to GNS!\n");
3189     gnutls_x509_crt_deinit (proxy_ca.cert);
3190     gnutls_x509_privkey_deinit (proxy_ca.key);
3191     gnutls_global_deinit ();
3192     return;
3193   }
3194   identity = GNUNET_IDENTITY_connect (cfg,
3195                                       NULL, NULL);
3196   id_op = GNUNET_IDENTITY_get (identity,
3197                                "gns-proxy",
3198                                &identity_master_cb,
3199                                NULL);
3200   GNUNET_SCHEDULER_add_shutdown (&do_shutdown, NULL);
3201 }
3202
3203
3204 /**
3205  * The main function for gnunet-gns-proxy.
3206  *
3207  * @param argc number of arguments from the command line
3208  * @param argv command line arguments
3209  * @return 0 ok, 1 on error
3210  */
3211 int
3212 main (int argc, char *const *argv)
3213 {
3214   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
3215     {'p', "port", NULL,
3216       gettext_noop ("listen on specified port (default: 7777)"), 1,
3217       &GNUNET_GETOPT_set_ulong, &port},
3218     {'a', "authority", NULL,
3219       gettext_noop ("pem file to use as CA"), 1,
3220       &GNUNET_GETOPT_set_string, &cafile_opt},
3221     GNUNET_GETOPT_OPTION_END
3222   };
3223   static const char* page =
3224     "<html><head><title>gnunet-gns-proxy</title>"
3225     "</head><body>cURL fail</body></html>";
3226   int ret;
3227
3228   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
3229     return 2;
3230   GNUNET_log_setup ("gnunet-gns-proxy", "WARNING", NULL);
3231   curl_failure_response = MHD_create_response_from_buffer (strlen (page),
3232                                                            (void*)page,
3233                                                            MHD_RESPMEM_PERSISTENT);
3234
3235   ret =
3236     (GNUNET_OK ==
3237      GNUNET_PROGRAM_run (argc, argv, "gnunet-gns-proxy",
3238                          _("GNUnet GNS proxy"),
3239                          options,
3240                          &run, NULL)) ? 0 : 1;
3241   MHD_destroy_response (curl_failure_response);
3242   GNUNET_free_non_null ((char *) argv);
3243   GNUNET_CRYPTO_ecdsa_key_clear (&local_shorten_zone);
3244   return ret;
3245 }
3246
3247 /* end of gnunet-gns-proxy.c */