removing ats functions from plugins, instead provide callback function
[oweals/gnunet.git] / src / transport / plugin_transport_http_server.c
1 /*
2      This file is part of GNUnet
3      (C) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009 Christian Grothoff (and other contributing authors)
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., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20
21 /**
22  * @file transport/plugin_transport_http.c
23  * @brief http transport service plugin
24  * @author Matthias Wachs
25  */
26
27 #include "plugin_transport_http.h"
28
29 #define HTTP_ERROR_RESPONSE "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\"><HTML><HEAD><TITLE>404 Not Found</TITLE></HEAD><BODY><H1>Not Found</H1>The requested URL was not found on this server.<P><HR><ADDRESS></ADDRESS></BODY></HTML>"
30 #define _RECEIVE 0
31 #define _SEND 1
32
33 struct ServerConnection
34 {
35   /* _RECV or _SEND */
36   int direction;
37
38   /* should this connection get disconnected? GNUNET_YES/NO  */
39   int disconnect;
40
41   struct Session *session;
42   struct MHD_Connection *mhd_conn;
43 };
44
45 /**
46  * Function that queries MHD's select sets and
47  * starts the task waiting for them.
48  * @param plugin plugin
49  * @param daemon_handle the MHD daemon handle
50  * @param now schedule now or with MHD delay
51  * @return gnunet task identifier
52  */
53 static GNUNET_SCHEDULER_TaskIdentifier
54 server_schedule (struct Plugin *plugin,
55                  struct MHD_Daemon *daemon_handle,
56                  int now);
57
58 static void
59 server_log (void *arg, const char *fmt, va_list ap)
60 {
61   char text[1024];
62
63   vsnprintf (text, sizeof (text), fmt, ap);
64   va_end (ap);
65   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Server: %s\n", text);
66 }
67
68 /**
69  * Check if incoming connection is accepted.
70  * NOTE: Here every connection is accepted
71  * @param cls plugin as closure
72  * @param addr address of incoming connection
73  * @param addr_len address length of incoming connection
74  * @return MHD_YES if connection is accepted, MHD_NO if connection is rejected
75  *
76  */
77 static int
78 server_accept_cb (void *cls, const struct sockaddr *addr, socklen_t addr_len)
79 {
80   struct Plugin *plugin = cls;
81
82   if (plugin->cur_connections <= plugin->max_connections)
83     return MHD_YES;
84   else
85   {
86     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
87                 "Server: Cannot accept new connections\n");
88     return MHD_NO;
89   }
90 }
91
92
93 #if BUILD_HTTPS
94 static char *
95 server_load_file (const char *file)
96 {
97   struct GNUNET_DISK_FileHandle *gn_file;
98   struct stat fstat;
99   char *text = NULL;
100
101   if (0 != STAT (file, &fstat))
102     return NULL;
103   text = GNUNET_malloc (fstat.st_size + 1);
104   gn_file =
105       GNUNET_DISK_file_open (file, GNUNET_DISK_OPEN_READ,
106                              GNUNET_DISK_PERM_USER_READ);
107   if (gn_file == NULL)
108   {
109     GNUNET_free (text);
110     return NULL;
111   }
112   if (GNUNET_SYSERR == GNUNET_DISK_file_read (gn_file, text, fstat.st_size))
113   {
114     GNUNET_free (text);
115     GNUNET_DISK_file_close (gn_file);
116     return NULL;
117   }
118   text[fstat.st_size] = '\0';
119   GNUNET_DISK_file_close (gn_file);
120   return text;
121 }
122 #endif
123
124
125 #if BUILD_HTTPS
126
127 static int
128 server_load_certificate (struct Plugin *plugin)
129 {
130   int res = GNUNET_OK;
131
132   char *key_file;
133   char *cert_file;
134
135   /* Get crypto init string from config
136    * If not present just use default values */
137
138   GNUNET_assert (GNUNET_OK ==
139                  GNUNET_CONFIGURATION_get_value_string (plugin->env->cfg,
140                                                         plugin->name,
141                                                         "CRYPTO_INIT",
142                                                         &plugin->crypto_init));
143
144   if (GNUNET_OK !=
145       GNUNET_CONFIGURATION_get_value_filename (plugin->env->cfg, plugin->name,
146                                                "KEY_FILE", &key_file))
147   {
148     key_file = GNUNET_strdup ("https_key.key");
149   }
150
151   if (GNUNET_OK !=
152       GNUNET_CONFIGURATION_get_value_filename (plugin->env->cfg, plugin->name,
153                                                "CERT_FILE", &cert_file))
154   {
155     GNUNET_asprintf (&cert_file, "%s", "https_cert.crt");
156   }
157
158   /* read key & certificates from file */
159 #if VERBOSE_SERVER
160   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
161               "Loading TLS certificate from key-file `%s' cert-file`%s'\n",
162               key_file, cert_file);
163 #endif
164
165   plugin->key = server_load_file (key_file);
166   plugin->cert = server_load_file (cert_file);
167
168   if ((plugin->key == NULL) || (plugin->cert == NULL))
169   {
170     struct GNUNET_OS_Process *cert_creation;
171
172     GNUNET_free_non_null (plugin->key);
173     plugin->key = NULL;
174     GNUNET_free_non_null (plugin->cert);
175     plugin->cert = NULL;
176
177 #if VERBOSE_SERVER
178     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
179                 "No usable TLS certificate found, creating certificate\n");
180 #endif
181     errno = 0;
182     cert_creation =
183         GNUNET_OS_start_process (NULL, NULL,
184                                  "gnunet-transport-certificate-creation",
185                                  "gnunet-transport-certificate-creation",
186                                  key_file, cert_file, NULL);
187     if (cert_creation == NULL)
188     {
189       GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
190                        _
191                        ("Could not create a new TLS certificate, program `gnunet-transport-certificate-creation' could not be started!\n"));
192       GNUNET_free (key_file);
193       GNUNET_free (cert_file);
194
195       GNUNET_free_non_null (plugin->key);
196       plugin->key = NULL;
197       GNUNET_free_non_null (plugin->cert);
198       plugin->cert = NULL;
199       GNUNET_free_non_null (plugin->crypto_init);
200       plugin->crypto_init = NULL;
201
202       return GNUNET_SYSERR;
203     }
204     GNUNET_assert (GNUNET_OK == GNUNET_OS_process_wait (cert_creation));
205     GNUNET_OS_process_close (cert_creation);
206
207     plugin->key = server_load_file (key_file);
208     plugin->cert = server_load_file (cert_file);
209   }
210
211   if ((plugin->key == NULL) || (plugin->cert == NULL))
212   {
213     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
214                      _
215                      ("No usable TLS certificate found and creating one failed!\n"),
216                      "transport-https");
217     GNUNET_free (key_file);
218     GNUNET_free (cert_file);
219
220     GNUNET_free_non_null (plugin->key);
221     plugin->key = NULL;
222     GNUNET_free_non_null (plugin->cert);
223     plugin->cert = NULL;
224     GNUNET_free_non_null (plugin->crypto_init);
225     plugin->crypto_init = NULL;
226
227     return GNUNET_SYSERR;
228   }
229   GNUNET_free (key_file);
230   GNUNET_free (cert_file);
231 #if DEBUG_HTTP
232   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "TLS certificate loaded\n");
233 #endif
234
235   return res;
236 }
237 #endif
238
239
240 /**
241  * Reschedule the execution of both IPv4 and IPv6 server
242  * @param plugin the plugin
243  * @param server which server to schedule v4 or v6?
244  * @param now GNUNET_YES to schedule execution immediately, GNUNET_NO to wait
245  * until timeout
246  */
247
248 static void
249 server_reschedule (struct Plugin *plugin, struct MHD_Daemon *server, int now)
250 {
251   if ((server == plugin->server_v4) && (plugin->server_v4 != NULL))
252   {
253     if (plugin->server_v4_task != GNUNET_SCHEDULER_NO_TASK)
254     {
255       GNUNET_SCHEDULER_cancel (plugin->server_v4_task);
256       plugin->server_v4_task = GNUNET_SCHEDULER_NO_TASK;
257     }
258     plugin->server_v4_task = server_schedule (plugin, plugin->server_v4, now);
259   }
260
261   if ((server == plugin->server_v6) && (plugin->server_v6 != NULL))
262   {
263     if (plugin->server_v6_task != GNUNET_SCHEDULER_NO_TASK)
264     {
265       GNUNET_SCHEDULER_cancel (plugin->server_v6_task);
266       plugin->server_v6_task = GNUNET_SCHEDULER_NO_TASK;
267     }
268     plugin->server_v6_task = server_schedule (plugin, plugin->server_v6, now);
269   }
270 }
271
272 /**
273  * Callback called by MessageStreamTokenizer when a message has arrived
274  * @param cls current session as closure
275  * @param client clien
276  * @param message the message to be forwarded to transport service
277  */
278 static void
279 server_receive_mst_cb (void *cls, void *client,
280                        const struct GNUNET_MessageHeader *message)
281 {
282   struct Session *s = cls;
283
284 #if VERBOSE_SERVER
285   struct Plugin *plugin = s->plugin;
286 #endif
287   struct GNUNET_TIME_Relative delay;
288
289   delay = http_plugin_receive (s, &s->target, message, s, s->addr, s->addrlen);
290
291   s->next_receive =
292       GNUNET_TIME_absolute_add (GNUNET_TIME_absolute_get (), delay);
293
294   if (delay.rel_value > 0)
295   {
296 #if VERBOSE_SERVER
297     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
298                      "Server: peer `%s' address `%s' next read delayed for %llu ms\n",
299                      GNUNET_i2s (&s->target),
300                      http_plugin_address_to_string (NULL, s->addr, s->addrlen),
301                      delay);
302 #endif
303   }
304 }
305
306 /**
307  * Callback called by MHD when it needs data to send
308  * @param cls current session
309  * @param pos position in buffer
310  * @param buf the buffer to write data to
311  * @param max max number of bytes available in buffer
312  * @return bytes written to buffer
313  */
314 static ssize_t
315 server_send_callback (void *cls, uint64_t pos, char *buf, size_t max)
316 {
317   struct Session *s = cls;
318
319 #if VERBOSE_SERVER
320   GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, s->plugin->name,
321                    "Server: %X can sent maximum  %u \n", s, max);
322 #endif
323   struct HTTP_Message *msg;
324   int bytes_read = 0;
325
326   //static int c = 0;
327   msg = s->msg_head;
328   if (msg != NULL)
329   {
330     /* sending */
331     if ((msg->size - msg->pos) <= max)
332     {
333       memcpy (buf, &msg->buf[msg->pos], (msg->size - msg->pos));
334       bytes_read = msg->size - msg->pos;
335       msg->pos += (msg->size - msg->pos);
336     }
337     else
338     {
339       memcpy (buf, &msg->buf[msg->pos], max);
340       msg->pos += max;
341       bytes_read = max;
342     }
343
344     /* removing message */
345     if (msg->pos == msg->size)
346     {
347       if (NULL != msg->transmit_cont)
348         msg->transmit_cont (msg->transmit_cont_cls, &s->target, GNUNET_OK);
349       GNUNET_CONTAINER_DLL_remove (s->msg_head, s->msg_tail, msg);
350       GNUNET_free (msg);
351     }
352   }
353
354 #if VERBOSE_SERVER
355   struct Plugin *plugin = s->plugin;
356
357   GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
358                    "Server: %X: sent %u bytes\n", s, bytes_read);
359 #endif
360   return bytes_read;
361 }
362
363 static struct ServerConnection *
364 server_lookup_session (struct Plugin *plugin,
365                        struct MHD_Connection *mhd_connection, const char *url,
366                        const char *method)
367 {
368   struct Session *s = NULL;
369   struct Session *t;
370   struct ServerConnection *sc = NULL;
371   const union MHD_ConnectionInfo *conn_info;
372
373   struct IPv4HttpAddress a4;
374   struct IPv6HttpAddress a6;
375   struct sockaddr_in *s4;
376   struct sockaddr_in6 *s6;
377   void *a;
378   size_t a_len;
379   struct GNUNET_PeerIdentity target;
380   int check = GNUNET_NO;
381   uint32_t tag = 0;
382   int direction;
383
384   conn_info =
385       MHD_get_connection_info (mhd_connection,
386                                MHD_CONNECTION_INFO_CLIENT_ADDRESS);
387   if ((conn_info->client_addr->sa_family != AF_INET) &&
388       (conn_info->client_addr->sa_family != AF_INET6))
389     return MHD_NO;
390
391   if ((strlen (&url[1]) >= 105) && (url[104] == ';'))
392   {
393     char hash[104];
394     char *tagc = (char *) &url[105];
395
396     memcpy (&hash, &url[1], 103);
397     hash[103] = '\0';
398     if (GNUNET_OK ==
399         GNUNET_CRYPTO_hash_from_string ((const char *) &hash,
400                                         &(target.hashPubKey)))
401     {
402       tag = strtoul (tagc, NULL, 10);
403       if (tagc > 0)
404         check = GNUNET_YES;
405     }
406   }
407
408   if (0 == strcmp (MHD_HTTP_METHOD_PUT, method))
409     direction = _RECEIVE;
410   if (0 == strcmp (MHD_HTTP_METHOD_GET, method))
411     direction = _SEND;
412
413   if (check == GNUNET_NO)
414     goto error;
415
416   plugin->cur_connections++;
417
418 #if VERBOSE_SERVER
419   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
420                    "Server: New inbound connection from %s with tag %u\n",
421                    GNUNET_i2s (&target), tag);
422 #endif
423   /* find duplicate session */
424
425   t = plugin->head;
426
427   while (t != NULL)
428   {
429     if ((t->inbound) &&
430         (0 == memcmp (&t->target, &target, sizeof (struct GNUNET_PeerIdentity)))
431         &&
432         /* FIXME add source address comparison */
433         (t->tag == tag))
434       break;
435     t = t->next;
436   }
437   if (t != NULL)
438   {
439 #if VERBOSE_SERVER
440     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
441                      "Server: Duplicate session, dismissing new connection from peer `%s'\n",
442                      GNUNET_i2s (&target));
443 #endif
444     goto error;
445   }
446
447   /* find semi-session */
448   t = plugin->server_semi_head;
449
450   while (t != NULL)
451   {
452     /* FIXME add source address comparison */
453     if ((0 == memcmp (&t->target, &target, sizeof (struct GNUNET_PeerIdentity)))
454         && (t->tag == tag))
455     {
456       break;
457     }
458     t = t->next;
459   }
460
461   if (t == NULL)
462     goto create;
463
464 #if VERBOSE_SERVER
465   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
466                    "Server: Found existing semi-session for `%s'\n",
467                    GNUNET_i2s (&target));
468 #endif
469
470   if ((direction == _SEND) && (t->server_send != NULL))
471   {
472 #if VERBOSE_SERVER
473     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
474                      "Server: Duplicate GET session, dismissing new connection from peer `%s'\n",
475                      GNUNET_i2s (&target));
476 #endif
477     goto error;
478   }
479   else
480   {
481     s = t;
482     GNUNET_CONTAINER_DLL_remove (plugin->server_semi_head,
483                                  plugin->server_semi_tail, s);
484     GNUNET_CONTAINER_DLL_insert (plugin->head, plugin->tail, s);
485 #if VERBOSE_SERVER
486     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
487                      "Server: Found matching semi-session, merging session for peer `%s'\n",
488                      GNUNET_i2s (&target));
489 #endif
490
491     goto found;
492   }
493   if ((direction == _RECEIVE) && (t->server_recv != NULL))
494   {
495 #if VERBOSE_SERVER
496     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
497                      "Server: Duplicate PUT session, dismissing new connection from peer `%s'\n",
498                      GNUNET_i2s (&target));
499 #endif
500     goto error;
501   }
502   else
503   {
504     s = t;
505     GNUNET_CONTAINER_DLL_remove (plugin->server_semi_head,
506                                  plugin->server_semi_tail, s);
507     GNUNET_CONTAINER_DLL_insert (plugin->head, plugin->tail, s);
508 #if VERBOSE_SERVER
509     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
510                      "Server: Found matching semi-session, merging session for peer `%s'\n",
511                      GNUNET_i2s (&target));
512 #endif
513     goto found;
514   }
515
516 create:
517 /* create new session */
518 #if VERBOSE_SERVER
519   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
520                    "Server: Creating new session for peer `%s' \n",
521                    GNUNET_i2s (&target));
522 #endif
523
524   switch (conn_info->client_addr->sa_family)
525   {
526   case (AF_INET):
527     s4 = ((struct sockaddr_in *) conn_info->client_addr);
528     a4.u4_port = s4->sin_port;
529     memcpy (&a4.ipv4_addr, &s4->sin_addr, sizeof (struct in_addr));
530     a = &a4;
531     a_len = sizeof (struct IPv4HttpAddress);
532     break;
533   case (AF_INET6):
534     s6 = ((struct sockaddr_in6 *) conn_info->client_addr);
535     a6.u6_port = s6->sin6_port;
536     memcpy (&a6.ipv6_addr, &s6->sin6_addr, sizeof (struct in6_addr));
537     a = &a6;
538     a_len = sizeof (struct IPv6HttpAddress);
539     break;
540   default:
541     GNUNET_break (0);
542     goto error;
543   }
544   s = create_session (plugin, &target, a, a_len, NULL, NULL);
545
546   s->inbound = GNUNET_YES;
547   s->next_receive = GNUNET_TIME_absolute_get_zero ();
548   s->tag = tag;
549   if (0 == strcmp (MHD_HTTP_METHOD_PUT, method))
550     s->server_recv = s;
551   if (0 == strcmp (MHD_HTTP_METHOD_GET, method))
552     s->server_send = s;
553   GNUNET_CONTAINER_DLL_insert (plugin->server_semi_head,
554                                plugin->server_semi_tail, s);
555   goto found;
556
557 error:
558 #if VERBOSE_SERVER
559   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
560                    "Server: Invalid connection request\n");
561 #endif
562   return NULL;
563
564 found:
565   sc = GNUNET_malloc (sizeof (struct ServerConnection));
566   sc->mhd_conn = mhd_connection;
567   sc->direction = direction;
568   sc->session = s;
569   if (direction == _SEND)
570     s->server_send = sc;
571   if (direction == _RECEIVE)
572     s->server_recv = sc;
573
574 #if MHD_VERSION >= 0x00090E00
575   int to = (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value / 1000);
576
577 #if VERBOSE_SERVER
578   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
579                    "Server: Setting timeout for %X to %u sec.\n", sc, to);
580 #endif
581   MHD_set_connection_option (mhd_connection, MHD_CONNECTION_OPTION_TIMEOUT, to);
582
583   struct MHD_Daemon *d = NULL;
584
585   if (s->addrlen == sizeof (struct IPv6HttpAddress))
586     d = plugin->server_v6;
587   if (s->addrlen == sizeof (struct IPv4HttpAddress))
588     d = plugin->server_v4;
589
590   server_reschedule (plugin, d, GNUNET_NO);
591 #endif
592   return sc;
593 }
594
595 /**
596  * Process GET or PUT request received via MHD.  For
597  * GET, queue response that will send back our pending
598  * messages.  For PUT, process incoming data and send
599  * to GNUnet core.  In either case, check if a session
600  * already exists and create a new one if not.
601  */
602 static int
603 server_access_cb (void *cls, struct MHD_Connection *mhd_connection,
604                   const char *url, const char *method, const char *version,
605                   const char *upload_data, size_t * upload_data_size,
606                   void **httpSessionCache)
607 {
608
609   struct Plugin *plugin = cls;
610   struct ServerConnection *sc = *httpSessionCache;
611   struct Session *s = NULL;
612
613   int res = MHD_YES;
614   struct MHD_Response *response;
615
616   GNUNET_assert (cls != NULL);
617   /* new connection */
618   if (sc == NULL)
619   {
620     sc = server_lookup_session (plugin, mhd_connection, url, method);
621     if (sc != NULL)
622       (*httpSessionCache) = sc;
623     else
624     {
625       response =
626           MHD_create_response_from_data (strlen (HTTP_ERROR_RESPONSE),
627                                          HTTP_ERROR_RESPONSE, MHD_NO, MHD_NO);
628       res = MHD_queue_response (mhd_connection, MHD_HTTP_NOT_FOUND, response);
629       MHD_destroy_response (response);
630       return res;
631     }
632   }
633
634   /* existing connection */
635   sc = (*httpSessionCache);
636   s = sc->session;
637
638   /* connection is to be disconnected */
639   if (sc->disconnect == GNUNET_YES)
640   {
641     response =
642         MHD_create_response_from_data (strlen ("Thank you!"), "Thank you!",
643                                        MHD_NO, MHD_NO);
644     res = MHD_queue_response (mhd_connection, MHD_HTTP_OK, response);
645 #if VERBOSE_SERVER
646     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
647                 "Sent HTTP/1.1: 200 OK as PUT Response\n");
648 #endif
649     MHD_destroy_response (response);
650     return MHD_YES;
651   }
652
653   GNUNET_assert (s != NULL);
654   /* Check if both directions are connected */
655   if ((sc->session->server_recv == NULL) || (sc->session->server_send == NULL))
656   {
657 #if VERBOSE_SERVER
658     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
659                      "Server: Delayed read from `%s' `%s' since not both semi-connections are connected\n",
660                      GNUNET_i2s (&s->target),
661                      http_plugin_address_to_string (NULL, s->addr, s->addrlen));
662 #endif
663     return MHD_YES;
664   }
665
666   if (sc->direction == _SEND)
667   {
668     response =
669         MHD_create_response_from_callback (-1, 32 * 1024, &server_send_callback,
670                                            s, NULL);
671     res = MHD_queue_response (mhd_connection, MHD_HTTP_OK, response);
672     MHD_destroy_response (response);
673     return MHD_YES;
674   }
675   if (sc->direction == _RECEIVE)
676   {
677     if (*upload_data_size == 0)
678     {
679 #if VERBOSE_SERVER
680       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
681                        "Server: Peer `%s' PUT on address `%s' connected\n",
682                        GNUNET_i2s (&s->target),
683                        http_plugin_address_to_string (NULL, s->addr,
684                                                       s->addrlen));
685 #endif
686       return MHD_YES;
687     }
688
689     /* Receiving data */
690     if ((*upload_data_size > 0))
691     {
692 #if VERBOSE_SERVER
693       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
694                        "Server: peer `%s' PUT on address `%s' received %u bytes\n",
695                        GNUNET_i2s (&s->target),
696                        http_plugin_address_to_string (NULL, s->addr,
697                                                       s->addrlen),
698                        *upload_data_size);
699 #endif
700       struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get ();
701
702       if ((s->next_receive.abs_value <= now.abs_value))
703       {
704 #if VERBOSE_SERVER
705         GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
706                          "Server: %X: PUT with %u bytes forwarded to MST\n", s,
707                          *upload_data_size);
708 #endif
709         if (s->msg_tk == NULL)
710         {
711           s->msg_tk = GNUNET_SERVER_mst_create (&server_receive_mst_cb, s);
712         }
713         res =
714             GNUNET_SERVER_mst_receive (s->msg_tk, s, upload_data,
715                                        *upload_data_size, GNUNET_NO, GNUNET_NO);
716
717 #if MHD_VERSION >= 0x00090E00
718         int to = (GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value / 1000);
719         struct ServerConnection *t = NULL;
720
721 #if VERBOSE_SERVER
722         GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
723                          "Server: Received %u bytes\n", *upload_data_size);
724 #endif
725
726         /* Setting timeouts for other connections */
727         if (s->server_recv != NULL)
728         {
729           t = s->server_recv;
730 #if VERBOSE_SERVER
731           GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
732                            "Server: Setting timeout for %X to %u sec.\n", t,
733                            to);
734 #endif
735           MHD_set_connection_option (t->mhd_conn, MHD_CONNECTION_OPTION_TIMEOUT,
736                                      to);
737         }
738         if (s->server_send != NULL)
739         {
740           t = s->server_send;
741 #if VERBOSE_SERVER
742           GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
743                            "Server: Setting timeout for %X to %u sec.\n", t,
744                            to);
745 #endif
746           MHD_set_connection_option (t->mhd_conn, MHD_CONNECTION_OPTION_TIMEOUT,
747                                      to);
748         }
749         struct MHD_Daemon *d = NULL;
750
751         if (s->addrlen == sizeof (struct IPv6HttpAddress))
752           d = plugin->server_v6;
753         if (s->addrlen == sizeof (struct IPv4HttpAddress))
754           d = plugin->server_v4;
755         server_reschedule (plugin, d, GNUNET_NO);
756 #endif
757         (*upload_data_size) = 0;
758       }
759       else
760       {
761 #if DEBUG_HTTP
762         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
763                     "Server: %X no inbound bandwidth available! Next read was delayed by %llu ms\n",
764                     s, now.abs_value - s->next_receive.abs_value);
765 #endif
766       }
767       return MHD_YES;
768     }
769     else
770       return MHD_NO;
771   }
772   return res;
773 }
774
775 static void
776 server_disconnect_cb (void *cls, struct MHD_Connection *connection,
777                       void **httpSessionCache)
778 {
779   struct ServerConnection *sc = *httpSessionCache;
780   struct ServerConnection *tc = *httpSessionCache;
781   struct Session *s = NULL;
782   struct Session *t = NULL;
783   struct Plugin *plugin = NULL;
784
785   if (sc == NULL)
786     return;
787
788   s = sc->session;
789   plugin = s->plugin;
790   if (sc->direction == _SEND)
791   {
792 #if VERBOSE_SERVER
793     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
794                      "Server: %X peer `%s' GET on address `%s' disconnected\n",
795                      s->server_send, GNUNET_i2s (&s->target),
796                      http_plugin_address_to_string (NULL, s->addr, s->addrlen));
797 #endif
798     s->server_send = NULL;
799
800     if (s->server_recv != NULL)
801     {
802       tc = s->server_recv;
803       tc->disconnect = GNUNET_YES;
804 #if MHD_VERSION >= 0x00090E00
805       MHD_set_connection_option (sc->mhd_conn, MHD_CONNECTION_OPTION_TIMEOUT,
806                                  1);
807 #endif
808     }
809   }
810   if (sc->direction == _RECEIVE)
811   {
812 #if VERBOSE_SERVER
813     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
814                      "Server: %X peer `%s' PUT on address `%s' disconnected\n",
815                      s->server_recv, GNUNET_i2s (&s->target),
816                      http_plugin_address_to_string (NULL, s->addr, s->addrlen));
817 #endif
818     s->server_recv = NULL;
819     if (s->server_send != NULL)
820     {
821       tc = s->server_send;
822       tc->disconnect = GNUNET_YES;
823 #if MHD_VERSION >= 0x00090E00
824       MHD_set_connection_option (sc->mhd_conn, MHD_CONNECTION_OPTION_TIMEOUT,
825                                  1);
826 #endif
827     }
828     if (s->msg_tk != NULL)
829     {
830       GNUNET_SERVER_mst_destroy (s->msg_tk);
831       s->msg_tk = NULL;
832     }
833   }
834   GNUNET_free (sc);
835
836   t = plugin->server_semi_head;
837   while (t != NULL)
838   {
839     if (t == s)
840     {
841       GNUNET_CONTAINER_DLL_remove (plugin->server_semi_head,
842                                    plugin->server_semi_tail, s);
843       GNUNET_CONTAINER_DLL_insert (plugin->head, plugin->tail, s);
844       break;
845     }
846     t = t->next;
847   }
848   plugin->cur_connections--;
849
850   struct MHD_Daemon *d = NULL;
851
852   if (s->addrlen == sizeof (struct IPv6HttpAddress))
853     d = plugin->server_v6;
854   if (s->addrlen == sizeof (struct IPv4HttpAddress))
855     d = plugin->server_v4;
856   server_reschedule (plugin, d, GNUNET_NO);
857
858   if ((s->server_send == NULL) && (s->server_recv == NULL))
859   {
860 #if VERBOSE_SERVER
861     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
862                      "Server: peer `%s' on address `%s' disconnected\n",
863                      GNUNET_i2s (&s->target),
864                      http_plugin_address_to_string (NULL, s->addr, s->addrlen));
865 #endif
866     if (s->msg_tk != NULL)
867     {
868       GNUNET_SERVER_mst_destroy (s->msg_tk);
869       s->msg_tk = NULL;
870     }
871
872     notify_session_end (s->plugin, &s->target, s);
873   }
874 }
875
876 int
877 server_disconnect (struct Session *s)
878 {
879   struct Plugin *plugin = s->plugin;
880   struct Session *t = plugin->head;
881
882   while (t != NULL)
883   {
884     if (t->inbound == GNUNET_YES)
885     {
886       if (t->server_send != NULL)
887       {
888         ((struct ServerConnection *) t->server_send)->disconnect = GNUNET_YES;
889       }
890       if (t->server_send != NULL)
891       {
892         ((struct ServerConnection *) t->server_send)->disconnect = GNUNET_YES;
893       }
894     }
895     t = t->next;
896   }
897   return GNUNET_OK;
898 }
899
900 int
901 server_send (struct Session *s, struct HTTP_Message *msg)
902 {
903   GNUNET_CONTAINER_DLL_insert (s->msg_head, s->msg_tail, msg);
904
905   if (s->addrlen == sizeof (struct IPv4HttpAddress))
906   {
907     server_reschedule (s->plugin, s->plugin->server_v4, GNUNET_YES);
908   }
909   else if (s->addrlen == sizeof (struct IPv6HttpAddress))
910   {
911     server_reschedule (s->plugin, s->plugin->server_v6, GNUNET_YES);
912   }
913   else
914     return GNUNET_SYSERR;
915   return GNUNET_OK;
916 }
917
918
919
920 /**
921  * Call MHD IPv4 to process pending requests and then go back
922  * and schedule the next run.
923  * @param cls plugin as closure
924  * @param tc task context
925  */
926 static void
927 server_v4_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
928 {
929   struct Plugin *plugin = cls;
930
931   GNUNET_assert (cls != NULL);
932
933   plugin->server_v4_task = GNUNET_SCHEDULER_NO_TASK;
934
935   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
936     return;
937 #if VERBOSE_SERVER
938   GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
939                    "Running IPv4 server\n");
940 #endif
941   GNUNET_assert (MHD_YES == MHD_run (plugin->server_v4));
942   if (plugin->server_v4 != NULL)
943     plugin->server_v4_task =
944         server_schedule (plugin, plugin->server_v4, GNUNET_NO);
945 }
946
947
948 /**
949  * Call MHD IPv6 to process pending requests and then go back
950  * and schedule the next run.
951  * @param cls plugin as closure
952  * @param tc task context
953  */
954 static void
955 server_v6_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
956 {
957   struct Plugin *plugin = cls;
958
959   GNUNET_assert (cls != NULL);
960
961   plugin->server_v6_task = GNUNET_SCHEDULER_NO_TASK;
962
963   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
964     return;
965 #if VERBOSE_SERVER
966   GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
967                    "Running IPv6 server\n");
968 #endif
969   GNUNET_assert (MHD_YES == MHD_run (plugin->server_v6));
970   if (plugin->server_v6 != NULL)
971     plugin->server_v6_task =
972         server_schedule (plugin, plugin->server_v6, GNUNET_NO);
973 }
974
975 /**
976  * Function that queries MHD's select sets and
977  * starts the task waiting for them.
978  * @param plugin plugin
979  * @param daemon_handle the MHD daemon handle
980  * @return gnunet task identifier
981  */
982 static GNUNET_SCHEDULER_TaskIdentifier
983 server_schedule (struct Plugin *plugin, struct MHD_Daemon *daemon_handle,
984                  int now)
985 {
986   GNUNET_SCHEDULER_TaskIdentifier ret;
987   fd_set rs;
988   fd_set ws;
989   fd_set es;
990   struct GNUNET_NETWORK_FDSet *wrs;
991   struct GNUNET_NETWORK_FDSet *wws;
992   struct GNUNET_NETWORK_FDSet *wes;
993   int max;
994   unsigned long long timeout;
995   static unsigned long long last_timeout = 0;
996   int haveto;
997
998   struct GNUNET_TIME_Relative tv;
999
1000   ret = GNUNET_SCHEDULER_NO_TASK;
1001   FD_ZERO (&rs);
1002   FD_ZERO (&ws);
1003   FD_ZERO (&es);
1004   wrs = GNUNET_NETWORK_fdset_create ();
1005   wes = GNUNET_NETWORK_fdset_create ();
1006   wws = GNUNET_NETWORK_fdset_create ();
1007   max = -1;
1008   GNUNET_assert (MHD_YES == MHD_get_fdset (daemon_handle, &rs, &ws, &es, &max));
1009   haveto = MHD_get_timeout (daemon_handle, &timeout);
1010   if (haveto == MHD_YES)
1011   {
1012     if (timeout != last_timeout)
1013     {
1014 #if VERBOSE_SERVER
1015       GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
1016                        "SELECT Timeout changed from %llu to %llu\n",
1017                        last_timeout, timeout);
1018 #endif
1019       last_timeout = timeout;
1020     }
1021     tv.rel_value = (uint64_t) timeout;
1022   }
1023   else
1024     tv = GNUNET_TIME_UNIT_SECONDS;
1025   /* Force immediate run, since we have outbound data to send */
1026   if (now == GNUNET_YES)
1027     tv = GNUNET_TIME_UNIT_MILLISECONDS;
1028   GNUNET_NETWORK_fdset_copy_native (wrs, &rs, max + 1);
1029   GNUNET_NETWORK_fdset_copy_native (wws, &ws, max + 1);
1030   GNUNET_NETWORK_fdset_copy_native (wes, &es, max + 1);
1031
1032   if (daemon_handle == plugin->server_v4)
1033   {
1034     if (plugin->server_v4_task != GNUNET_SCHEDULER_NO_TASK)
1035     {
1036       GNUNET_SCHEDULER_cancel (plugin->server_v4_task);
1037       plugin->server_v4_task = GNUNET_SCHEDULER_NO_TASK;
1038     }
1039 #if VERBOSE_SERVER
1040     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
1041                      "Scheduling IPv4 server task in %llu ms\n", tv);
1042 #endif
1043     ret =
1044         GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1045                                      GNUNET_SCHEDULER_NO_TASK, tv, wrs, wws,
1046                                      &server_v4_run, plugin);
1047   }
1048   if (daemon_handle == plugin->server_v6)
1049   {
1050     if (plugin->server_v6_task != GNUNET_SCHEDULER_NO_TASK)
1051     {
1052       GNUNET_SCHEDULER_cancel (plugin->server_v6_task);
1053       plugin->server_v6_task = GNUNET_SCHEDULER_NO_TASK;
1054     }
1055 #if VERBOSE_SERVER
1056     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
1057                      "Scheduling IPv6 server task in %llu ms\n", tv);
1058 #endif
1059     ret =
1060         GNUNET_SCHEDULER_add_select (GNUNET_SCHEDULER_PRIORITY_DEFAULT,
1061                                      GNUNET_SCHEDULER_NO_TASK, tv, wrs, wws,
1062                                      &server_v6_run, plugin);
1063   }
1064   GNUNET_NETWORK_fdset_destroy (wrs);
1065   GNUNET_NETWORK_fdset_destroy (wws);
1066   GNUNET_NETWORK_fdset_destroy (wes);
1067   return ret;
1068 }
1069
1070 int
1071 server_start (struct Plugin *plugin)
1072 {
1073   int res = GNUNET_OK;
1074   unsigned int timeout;
1075
1076 #if BUILD_HTTPS
1077   res = server_load_certificate (plugin);
1078   if (res == GNUNET_SYSERR)
1079   {
1080     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
1081                      "Could not load or create server certificate! Loading plugin failed!\n");
1082     return res;
1083   }
1084 #endif
1085
1086
1087 #if MHD_VERSION >= 0x00090E00
1088   timeout = HTTP_NOT_VALIDATED_TIMEOUT.rel_value / 1000;
1089   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
1090                    "MHD can set timeout per connection! Default time out %u sec.\n",
1091                    timeout);
1092 #else
1093   timeout = GNUNET_CONSTANTS_IDLE_CONNECTION_TIMEOUT.rel_value / 1000;
1094   GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING, plugin->name,
1095                    "MHD cannot set timeout per connection! Default time out %u sec.\n",
1096                    timeout);
1097 #endif
1098   plugin->server_v4 = NULL;
1099   if (plugin->ipv4 == GNUNET_YES)
1100   {
1101     plugin->server_v4 = MHD_start_daemon (
1102 #if VERBOSE_SERVER
1103                                            MHD_USE_DEBUG |
1104 #endif
1105 #if BUILD_HTTPS
1106                                            MHD_USE_SSL |
1107 #endif
1108                                            MHD_NO_FLAG, plugin->port,
1109                                            &server_accept_cb, plugin,
1110                                            &server_access_cb, plugin,
1111                                            MHD_OPTION_SOCK_ADDR,
1112                                            (struct sockaddr_in *)
1113                                            plugin->server_addr_v4,
1114                                            MHD_OPTION_CONNECTION_LIMIT,
1115                                            (unsigned int)
1116                                            plugin->max_connections,
1117 #if BUILD_HTTPS
1118                                            MHD_OPTION_HTTPS_PRIORITIES,
1119                                            plugin->crypto_init,
1120                                            MHD_OPTION_HTTPS_MEM_KEY,
1121                                            plugin->key,
1122                                            MHD_OPTION_HTTPS_MEM_CERT,
1123                                            plugin->cert,
1124 #endif
1125                                            MHD_OPTION_CONNECTION_TIMEOUT,
1126                                            timeout,
1127                                            MHD_OPTION_CONNECTION_MEMORY_LIMIT,
1128                                            (size_t) (2 *
1129                                                      GNUNET_SERVER_MAX_MESSAGE_SIZE),
1130                                            MHD_OPTION_NOTIFY_COMPLETED,
1131                                            &server_disconnect_cb, plugin,
1132                                            MHD_OPTION_EXTERNAL_LOGGER,
1133                                            server_log, NULL, MHD_OPTION_END);
1134   }
1135   plugin->server_v6 = NULL;
1136   if (plugin->ipv6 == GNUNET_YES)
1137   {
1138     plugin->server_v6 = MHD_start_daemon (
1139 #if VERBOSE_SERVER
1140                                            MHD_USE_DEBUG |
1141 #endif
1142 #if BUILD_HTTPS
1143                                            MHD_USE_SSL |
1144 #endif
1145                                            MHD_USE_IPv6, plugin->port,
1146                                            &server_accept_cb, plugin,
1147                                            &server_access_cb, plugin,
1148                                            MHD_OPTION_SOCK_ADDR,
1149                                            (struct sockaddr_in6 *)
1150                                            plugin->server_addr_v6,
1151                                            MHD_OPTION_CONNECTION_LIMIT,
1152                                            (unsigned int)
1153                                            plugin->max_connections,
1154 #if BUILD_HTTPS
1155                                            MHD_OPTION_HTTPS_PRIORITIES,
1156                                            plugin->crypto_init,
1157                                            MHD_OPTION_HTTPS_MEM_KEY,
1158                                            plugin->key,
1159                                            MHD_OPTION_HTTPS_MEM_CERT,
1160                                            plugin->cert,
1161 #endif
1162                                            MHD_OPTION_CONNECTION_TIMEOUT,
1163                                            timeout,
1164                                            MHD_OPTION_CONNECTION_MEMORY_LIMIT,
1165                                            (size_t) (2 *
1166                                                      GNUNET_SERVER_MAX_MESSAGE_SIZE),
1167                                            MHD_OPTION_NOTIFY_COMPLETED,
1168                                            &server_disconnect_cb, plugin,
1169                                            MHD_OPTION_EXTERNAL_LOGGER,
1170                                            server_log, NULL, MHD_OPTION_END);
1171
1172   }
1173
1174   if ((plugin->ipv4 == GNUNET_YES) && (plugin->server_v4 == NULL))
1175   {
1176     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
1177                      "Failed to start %s IPv4 server component on port %u\n",
1178                      plugin->name, plugin->port);
1179     return GNUNET_SYSERR;
1180   }
1181   server_reschedule (plugin, plugin->server_v4, GNUNET_NO);
1182
1183   if ((plugin->ipv6 == GNUNET_YES) && (plugin->server_v6 == NULL))
1184   {
1185     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
1186                      "Failed to start %s IPv6 server component on port %u\n",
1187                      plugin->name, plugin->port);
1188     return GNUNET_SYSERR;
1189   }
1190   server_reschedule (plugin, plugin->server_v6, GNUNET_NO);
1191
1192
1193 #if DEBUG_HTTP
1194   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
1195                    "%s server component started on port %u\n", plugin->name,
1196                    plugin->port);
1197 #endif
1198   return res;
1199 }
1200
1201 void
1202 server_stop (struct Plugin *plugin)
1203 {
1204   struct Session *s = NULL;
1205   struct Session *t = NULL;
1206
1207 #if VERBOSE_SERVER
1208   GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name, "server_stop\n");
1209 #endif
1210
1211   struct MHD_Daemon *server_v4_tmp = plugin->server_v4;
1212
1213   plugin->server_v4 = NULL;
1214   struct MHD_Daemon *server_v6_tmp = plugin->server_v6;
1215
1216   plugin->server_v6 = NULL;
1217
1218   if (plugin->server_v4_task != GNUNET_SCHEDULER_NO_TASK)
1219   {
1220     GNUNET_SCHEDULER_cancel (plugin->server_v4_task);
1221     plugin->server_v4_task = GNUNET_SCHEDULER_NO_TASK;
1222   }
1223
1224   if (plugin->server_v6_task != GNUNET_SCHEDULER_NO_TASK)
1225   {
1226     GNUNET_SCHEDULER_cancel (plugin->server_v6_task);
1227     plugin->server_v6_task = GNUNET_SCHEDULER_NO_TASK;
1228   }
1229
1230   if (server_v6_tmp != NULL)
1231   {
1232     MHD_stop_daemon (server_v4_tmp);
1233   }
1234   if (server_v6_tmp != NULL)
1235   {
1236     MHD_stop_daemon (server_v6_tmp);
1237   }
1238
1239   /* cleaning up semi-sessions never propagated */
1240   s = plugin->server_semi_head;
1241   while (s != NULL)
1242   {
1243 #if VERBOSE_SERVER
1244     GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, plugin->name,
1245                      "Deleting semi-sessions %p\n", s);
1246 #endif
1247     t = s->next;
1248     struct HTTP_Message *msg = s->msg_head;
1249     struct HTTP_Message *tmp = s->msg_head;
1250
1251     while (msg != NULL)
1252     {
1253       tmp = msg->next;
1254
1255       GNUNET_CONTAINER_DLL_remove (s->msg_head, s->msg_tail, msg);
1256       if (msg->transmit_cont != NULL)
1257       {
1258         msg->transmit_cont (msg->transmit_cont_cls, &s->target, GNUNET_SYSERR);
1259       }
1260       GNUNET_free (msg);
1261       msg = tmp;
1262     }
1263
1264     delete_session (s);
1265     s = t;
1266   }
1267
1268 #if BUILD_HTTPS
1269   GNUNET_free_non_null (plugin->crypto_init);
1270   GNUNET_free_non_null (plugin->cert);
1271   GNUNET_free_non_null (plugin->key);
1272 #endif
1273
1274 #if DEBUG_HTTP
1275   GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG, plugin->name,
1276                    "%s server component stopped\n", plugin->name);
1277 #endif
1278 }
1279
1280
1281
1282 /* end of plugin_transport_http.c */