double-free
[oweals/gnunet.git] / src / vpn / gnunet-service-dns.c
1 /*
2      This file is part of GNUnet.
3      (C) 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 vpn/gnunet-service-dns.c
23  * @author Philipp Toelke
24  */
25 #include "platform.h"
26 #include "gnunet_getopt_lib.h"
27 #include "gnunet_service_lib.h"
28 #include <gnunet_constants.h>
29 #include "gnunet_network_lib.h"
30 #include "gnunet_os_lib.h"
31 #include "gnunet-service-dns-p.h"
32 #include "gnunet_protocols.h"
33 #include "gnunet_applications.h"
34 #include "gnunet-vpn-packet.h"
35 #include "gnunet_container_lib.h"
36 #include "gnunet-dns-parser.h"
37 #include "gnunet_dht_service.h"
38 #include "gnunet_block_lib.h"
39 #include "block_dns.h"
40 #include "gnunet_crypto_lib.h"
41 #include "gnunet_mesh_service.h"
42 #include "gnunet_signatures.h"
43
44 struct GNUNET_MESH_Handle *mesh_handle;
45
46 /**
47  * The UDP-Socket through which DNS-Resolves will be sent if they are not to be
48  * sent through gnunet. The port of this socket will not be hijacked.
49  */
50 static struct GNUNET_NETWORK_Handle *dnsout;
51
52 /**
53  * The port bound to the socket dnsout
54  */
55 static unsigned short dnsoutport;
56
57 /**
58  * A handle to the DHT-Service
59  */
60 static struct GNUNET_DHT_Handle *dht;
61
62 /**
63  * The configuration to use
64  */
65 static const struct GNUNET_CONFIGURATION_Handle *cfg;
66
67 /**
68  * A list of DNS-Responses that have to be sent to the requesting client
69  */
70 static struct answer_packet_list *head;
71
72 /**
73  * The tail of the list of DNS-responses
74  */
75 static struct answer_packet_list *tail;
76
77 /**
78  * A structure containing a mapping from network-byte-ordered DNS-id (16 bit) to
79  * some information needed to handle this query
80  *
81  * It currently allocates at least
82  * (1 + machine-width + machine-width + 32 + 32 + 16 + machine-width + 8) * 65536 bit
83  * = 17 MiB on 64 bit.
84  * = 11 MiB on 32 bit.
85  */
86 static struct
87 {
88   unsigned valid:1;
89   struct GNUNET_SERVER_Client *client;
90   struct GNUNET_MESH_Tunnel *tunnel;
91   uint32_t local_ip;
92   uint32_t remote_ip;
93   uint16_t local_port;
94   char *name;
95   uint8_t namelen;
96 } query_states[UINT16_MAX];
97
98 /**
99  * A struct used to give more than one value as
100  * closure to receive_dht
101  */
102 struct receive_dht_cls
103 {
104   uint16_t id;
105   struct GNUNET_DHT_GetHandle *handle;
106 };
107
108 struct tunnel_notify_queue
109 {
110   struct tunnel_notify_queue *next;
111   struct tunnel_notify_queue *prev;
112   void *cls;
113   size_t len;
114   GNUNET_CONNECTION_TransmitReadyNotify cb;
115 };
116
117 /**
118  * Hijack all outgoing DNS-Traffic but for traffic leaving "our" port.
119  */
120 static void
121 hijack (void *cls
122         __attribute__ ((unused)), const struct GNUNET_SCHEDULER_TaskContext *tc)
123 {
124   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
125     return;
126
127   if (0 == dnsoutport)
128   {
129     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
130                 "Delaying the hijacking, port is still %d!\n", dnsoutport);
131     GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, hijack, NULL);
132     return;
133   }
134
135   char port_s[6];
136   char *virt_dns;
137   struct GNUNET_OS_Process *proc;
138
139   if (GNUNET_SYSERR ==
140       GNUNET_CONFIGURATION_get_value_string (cfg, "vpn", "VIRTDNS", &virt_dns))
141   {
142     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
143                 "No entry 'VIRTDNS' in configuration!\n");
144     exit (1);
145   }
146
147   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Hijacking, port is %d\n", dnsoutport);
148   snprintf (port_s, 6, "%d", dnsoutport);
149   if (NULL !=
150       (proc =
151        GNUNET_OS_start_process (NULL, NULL, "gnunet-helper-hijack-dns",
152                                 "gnunet-hijack-dns", port_s, virt_dns, NULL)))
153     GNUNET_OS_process_close (proc);
154   GNUNET_free (virt_dns);
155 }
156
157 /**
158  * Delete the hijacking-routes
159  */
160 static void
161 unhijack (unsigned short port)
162 {
163   char port_s[6];
164   char *virt_dns;
165   struct GNUNET_OS_Process *proc;
166
167   if (GNUNET_SYSERR ==
168       GNUNET_CONFIGURATION_get_value_string (cfg, "vpn", "VIRTDNS", &virt_dns))
169   {
170     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
171                 "No entry 'VIRTDNS' in configuration!\n");
172     exit (1);
173   }
174
175   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "unHijacking, port is %d\n", port);
176   snprintf (port_s, 6, "%d", port);
177   if (NULL !=
178       (proc =
179        GNUNET_OS_start_process (NULL, NULL, "gnunet-helper-hijack-dns",
180                                 "gnunet-hijack-dns", "-d", port_s, virt_dns,
181                                 NULL)))
182     GNUNET_OS_process_close (proc);
183   GNUNET_free (virt_dns);
184 }
185
186 /**
187  * Send the DNS-Response to the client. Gets called via the notify_transmit_ready-
188  * system.
189  */
190 static size_t
191 send_answer (void *cls, size_t size, void *buf)
192 {
193   struct answer_packet_list *query = head;
194   size_t len = ntohs (query->pkt.hdr.size);
195
196   GNUNET_assert (len <= size);
197
198   memcpy (buf, &query->pkt.hdr, len);
199
200   GNUNET_CONTAINER_DLL_remove (head, tail, query);
201
202   GNUNET_free (query);
203
204   /* When more data is to be sent, reschedule */
205   if (head != NULL)
206     GNUNET_SERVER_notify_transmit_ready (cls, ntohs (head->pkt.hdr.size),
207                                          GNUNET_TIME_UNIT_FOREVER_REL,
208                                          &send_answer, cls);
209
210   return len;
211 }
212
213 struct tunnel_cls
214 {
215   struct GNUNET_MESH_Tunnel *tunnel GNUNET_PACKED;
216   struct GNUNET_MessageHeader hdr;
217   struct dns_pkt dns;
218 };
219
220 struct tunnel_cls *remote_pending[UINT16_MAX];
221
222 static size_t
223 mesh_send_response (void *cls, size_t size, void *buf)
224 {
225   GNUNET_assert (size >= sizeof (struct GNUNET_MessageHeader));
226   struct GNUNET_MessageHeader *hdr = buf;
227   uint32_t *sz = cls;
228   struct GNUNET_MESH_Tunnel **tunnel = (struct GNUNET_MESH_Tunnel **) (sz + 1);
229   struct dns_pkt *dns = (struct dns_pkt *) (tunnel + 1);
230
231   hdr->type = htons (GNUNET_MESSAGE_TYPE_VPN_REMOTE_ANSWER_DNS);
232   hdr->size = htons (*sz + sizeof (struct GNUNET_MessageHeader));
233
234   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
235               "Sending response, size=%d, sz=%d, sz+hdr=%d\n", size, *sz,
236               *sz + sizeof (struct GNUNET_MessageHeader));
237
238   GNUNET_assert (size >= (*sz + sizeof (struct GNUNET_MessageHeader)));
239
240   memcpy (hdr + 1, dns, *sz);
241
242   if (NULL != GNUNET_MESH_tunnel_get_head (*tunnel))
243   {
244     struct tunnel_notify_queue *element = GNUNET_MESH_tunnel_get_head (*tunnel);
245     struct tunnel_notify_queue *head = GNUNET_MESH_tunnel_get_head (*tunnel);
246     struct tunnel_notify_queue *tail = GNUNET_MESH_tunnel_get_tail (*tunnel);
247
248     GNUNET_CONTAINER_DLL_remove (head, tail, element);
249
250     GNUNET_MESH_tunnel_set_head (*tunnel, head);
251     GNUNET_MESH_tunnel_set_tail (*tunnel, tail);
252     struct GNUNET_MESH_TransmitHandle *th =
253         GNUNET_MESH_notify_transmit_ready (*tunnel,
254                                            GNUNET_NO,
255                                            42,
256                                            GNUNET_TIME_relative_divide
257                                            (GNUNET_CONSTANTS_MAX_CORK_DELAY, 2),
258                                            (const struct GNUNET_PeerIdentity *)
259                                            NULL, element->len,
260                                            element->cb, element->cls);
261
262     /* save the handle */
263     GNUNET_MESH_tunnel_set_data (*tunnel, th);
264   }
265
266   GNUNET_free (cls);
267
268   return ntohs (hdr->size);
269 }
270
271 static size_t
272 mesh_send (void *cls, size_t size, void *buf)
273 {
274   struct tunnel_cls *cls_ = (struct tunnel_cls *) cls;
275
276   GNUNET_MESH_tunnel_set_data (cls_->tunnel, NULL);
277
278   GNUNET_assert (cls_->hdr.size <= size);
279
280   size = cls_->hdr.size;
281   cls_->hdr.size = htons (cls_->hdr.size);
282
283   memcpy (buf, &cls_->hdr, size);
284
285   if (NULL != GNUNET_MESH_tunnel_get_head (cls_->tunnel))
286   {
287     struct tunnel_notify_queue *element =
288         GNUNET_MESH_tunnel_get_head (cls_->tunnel);
289     struct tunnel_notify_queue *head =
290         GNUNET_MESH_tunnel_get_head (cls_->tunnel);
291     struct tunnel_notify_queue *tail =
292         GNUNET_MESH_tunnel_get_tail (cls_->tunnel);
293
294     GNUNET_CONTAINER_DLL_remove (head, tail, element);
295
296     GNUNET_MESH_tunnel_set_head (cls_->tunnel, head);
297     GNUNET_MESH_tunnel_set_tail (cls_->tunnel, tail);
298     struct GNUNET_MESH_TransmitHandle *th =
299         GNUNET_MESH_notify_transmit_ready (cls_->tunnel,
300                                            GNUNET_NO,
301                                            42,
302                                            GNUNET_TIME_relative_divide
303                                            (GNUNET_CONSTANTS_MAX_CORK_DELAY, 2),
304                                            (const struct GNUNET_PeerIdentity *)
305                                            NULL, element->len,
306                                            element->cb, element->cls);
307
308     /* save the handle */
309     GNUNET_MESH_tunnel_set_data (cls_->tunnel, th);
310     GNUNET_free (element);
311   }
312
313   return size;
314 }
315
316
317 void
318 mesh_connect (void *cls, const struct GNUNET_PeerIdentity *peer,
319               const struct GNUNET_TRANSPORT_ATS_Information *atsi
320               __attribute__ ((unused)))
321 {
322   if (NULL == peer)
323     return;
324   struct tunnel_cls *cls_ = (struct tunnel_cls *) cls;
325
326   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
327               "Connected to peer %s, %x, sending query with id %d\n",
328               GNUNET_i2s (peer), peer, ntohs (cls_->dns.s.id));
329
330   if (NULL == GNUNET_MESH_tunnel_get_data (cls_->tunnel))
331   {
332     struct GNUNET_MESH_TransmitHandle *th =
333         GNUNET_MESH_notify_transmit_ready (cls_->tunnel,
334                                            GNUNET_YES,
335                                            42,
336                                            GNUNET_TIME_UNIT_MINUTES,
337                                            NULL,
338                                            cls_->hdr.size,
339                                            mesh_send,
340                                            cls);
341
342     GNUNET_MESH_tunnel_set_data (cls_->tunnel, th);
343   }
344   else
345   {
346     struct tunnel_notify_queue *head =
347         GNUNET_MESH_tunnel_get_head (cls_->tunnel);
348     struct tunnel_notify_queue *tail =
349         GNUNET_MESH_tunnel_get_tail (cls_->tunnel);
350
351     struct tunnel_notify_queue *element =
352         GNUNET_malloc (sizeof (struct tunnel_notify_queue));
353     element->cls = cls;
354     element->len = cls_->hdr.size;
355     element->cb = mesh_send;
356
357     GNUNET_CONTAINER_DLL_insert_tail (head, tail, element);
358     GNUNET_MESH_tunnel_set_head (cls_->tunnel, head);
359     GNUNET_MESH_tunnel_set_tail (cls_->tunnel, tail);
360   }
361 }
362
363
364 static void
365 send_mesh_query (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
366 {
367   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
368     return;
369
370   struct tunnel_cls *cls_ = (struct tunnel_cls *) cls;
371
372   cls_->tunnel =
373       GNUNET_MESH_peer_request_connect_by_type (mesh_handle,
374                                                 GNUNET_TIME_UNIT_HOURS,
375                                                 GNUNET_APPLICATION_TYPE_INTERNET_RESOLVER,
376                                                 mesh_connect, NULL, cls_);
377
378   remote_pending[cls_->dns.s.id] = cls_;
379 }
380
381 static int
382 receive_mesh_query (void *cls
383                     __attribute__ ((unused)), struct GNUNET_MESH_Tunnel *tunnel,
384                     void **ctx
385                     __attribute__ ((unused)),
386                     const struct GNUNET_PeerIdentity *sender
387                     __attribute__ ((unused)),
388                     const struct GNUNET_MessageHeader *message,
389                     const struct GNUNET_TRANSPORT_ATS_Information *atsi
390                     __attribute__ ((unused)))
391 {
392   struct dns_pkt *dns = (struct dns_pkt *) (message + 1);
393
394   struct sockaddr_in dest;
395
396   memset (&dest, 0, sizeof dest);
397   dest.sin_port = htons (53);
398   /* TODO: read from config */
399   inet_pton (AF_INET, "8.8.8.8", &dest.sin_addr);
400
401   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Querying for remote, id=%d\n",
402               ntohs (dns->s.id));
403   query_states[dns->s.id].tunnel = tunnel;
404   query_states[dns->s.id].valid = GNUNET_YES;
405
406   GNUNET_NETWORK_socket_sendto (dnsout, dns,
407                                 ntohs (message->size) -
408                                 sizeof (struct GNUNET_MessageHeader),
409                                 (struct sockaddr *) &dest, sizeof dest);
410
411   return GNUNET_SYSERR;
412 }
413
414 static int
415 receive_mesh_answer (void *cls
416                      __attribute__ ((unused)),
417                      struct GNUNET_MESH_Tunnel *tunnel, void **ctx
418                      __attribute__ ((unused)),
419                      const struct GNUNET_PeerIdentity *sender,
420                      const struct GNUNET_MessageHeader *message,
421                      const struct GNUNET_TRANSPORT_ATS_Information *atsi
422                      __attribute__ ((unused)))
423 {
424   /* TODo: size check */
425   struct dns_pkt *dns = (struct dns_pkt *) (message + 1);
426
427   /* They sent us a packet we were not waiting for */
428   if (remote_pending[dns->s.id] == NULL ||
429       remote_pending[dns->s.id]->tunnel != tunnel)
430     return GNUNET_OK;
431
432   GNUNET_free (remote_pending[dns->s.id]);
433   remote_pending[dns->s.id] = NULL;
434
435   if (query_states[dns->s.id].valid != GNUNET_YES)
436     return GNUNET_SYSERR;
437   query_states[dns->s.id].valid = GNUNET_NO;
438
439   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
440               "Received answer from peer %s, dns-id %d\n", GNUNET_i2s (sender),
441               ntohs (dns->s.id));
442
443   size_t len = sizeof (struct answer_packet) - 1 + sizeof (struct dns_static) + query_states[dns->s.id].namelen + sizeof (struct dns_query_line) + 2    /* To hold the pointer (as defined in RFC1035) to the name */
444       + sizeof (struct dns_record_line) - 1 + 16;       /* To hold the IPv6-Address */
445
446   struct answer_packet_list *answer =
447       GNUNET_malloc (len + 2 * sizeof (struct answer_packet_list *));
448   memset (answer, 0, len + 2 * sizeof (struct answer_packet_list *));
449
450   answer->pkt.hdr.type = htons (GNUNET_MESSAGE_TYPE_VPN_DNS_LOCAL_RESPONSE_DNS);
451   answer->pkt.hdr.size = htons (len);
452
453   struct dns_pkt_parsed *pdns = parse_dns_packet (dns);
454
455   if (ntohs (pdns->s.ancount) < 1)
456   {
457     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Answer only contains %d answers.\n",
458                 ntohs (pdns->s.ancount));
459     free_parsed_dns_packet (pdns);
460     return GNUNET_OK;
461   }
462
463   answer->pkt.addrsize = ntohs (pdns->answers[0]->data_len);
464   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "The first answer has the addrlen %d\n",
465               answer->pkt.addrsize);
466   memcpy (answer->pkt.addr, pdns->answers[0]->data,
467           ntohs (pdns->answers[0]->data_len));
468
469   answer->pkt.from = query_states[dns->s.id].remote_ip;
470
471   answer->pkt.to = query_states[dns->s.id].local_ip;
472   answer->pkt.dst_port = query_states[dns->s.id].local_port;
473
474   struct dns_pkt *dpkt = (struct dns_pkt *) answer->pkt.data;
475
476   dpkt->s.id = dns->s.id;
477   dpkt->s.aa = 1;
478   dpkt->s.qr = 1;
479   dpkt->s.ra = 1;
480   dpkt->s.qdcount = htons (1);
481   dpkt->s.ancount = htons (1);
482
483   memcpy (dpkt->data, query_states[dns->s.id].name,
484           query_states[dns->s.id].namelen);
485   GNUNET_free (query_states[dns->s.id].name);
486   query_states[dns->s.id].name = NULL;
487
488   struct dns_query_line *dque =
489       (struct dns_query_line *) (dpkt->data +
490                                  (query_states[dns->s.id].namelen));
491
492   struct dns_record_line *drec_data =
493       (struct dns_record_line *) (dpkt->data +
494                                   (query_states[dns->s.id].namelen) +
495                                   sizeof (struct dns_query_line) + 2);
496   if (16 == answer->pkt.addrsize)
497   {
498     answer->pkt.subtype = GNUNET_DNS_ANSWER_TYPE_REMOTE_AAAA;
499     dque->type = htons (28);    /* AAAA */
500     drec_data->type = htons (28);       /* AAAA */
501     drec_data->data_len = htons (16);
502   }
503   else
504   {
505     answer->pkt.subtype = GNUNET_DNS_ANSWER_TYPE_REMOTE_A;
506     dque->type = htons (1);     /* A */
507     drec_data->type = htons (1);        /* A */
508     drec_data->data_len = htons (4);
509   }
510   dque->class = htons (1);      /* IN */
511
512   char *anname =
513       (char *) (dpkt->data + (query_states[dns->s.id].namelen) +
514                 sizeof (struct dns_query_line));
515   memcpy (anname, "\xc0\x0c", 2);
516   drec_data->class = htons (1); /* IN */
517
518   drec_data->ttl = pdns->answers[0]->ttl;
519
520   /* Calculate at which offset in the packet the IPv6-Address belongs, it is
521    * filled in by the daemon-vpn */
522   answer->pkt.addroffset =
523       htons ((unsigned short) ((unsigned long) (&drec_data->data) -
524                                (unsigned long) (&answer->pkt)));
525
526   GNUNET_CONTAINER_DLL_insert_after (head, tail, tail, answer);
527
528   GNUNET_SERVER_notify_transmit_ready (query_states[dns->s.id].client, len,
529                                        GNUNET_TIME_UNIT_FOREVER_REL,
530                                        &send_answer,
531                                        query_states[dns->s.id].client);
532
533   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
534               "Sent answer of length %d on to client, addroffset = %d\n", len,
535               answer->pkt.addroffset);
536
537   free_parsed_dns_packet (pdns);
538   return GNUNET_OK;
539 }
540
541
542 static void
543 send_rev_query (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
544 {
545   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
546     return;
547
548   struct dns_pkt_parsed *pdns = (struct dns_pkt_parsed *) cls;
549
550   unsigned short id = pdns->s.id;
551
552   free_parsed_dns_packet (pdns);
553
554   if (query_states[id].valid != GNUNET_YES)
555     return;
556   query_states[id].valid = GNUNET_NO;
557
558   GNUNET_assert (query_states[id].namelen == 74);
559
560   size_t len = sizeof (struct answer_packet) - 1 + sizeof (struct dns_static) + 74      /* this is the length of a reverse ipv6-lookup */
561       + sizeof (struct dns_query_line) + 2      /* To hold the pointer (as defined in RFC1035) to the name */
562       + sizeof (struct dns_record_line) - 1 -
563       2 /* We do not know the lenght of the answer yet */ ;
564
565   struct answer_packet_list *answer =
566       GNUNET_malloc (len + 2 * sizeof (struct answer_packet_list *));
567   memset (answer, 0, len + 2 * sizeof (struct answer_packet_list *));
568
569   answer->pkt.hdr.type = htons (GNUNET_MESSAGE_TYPE_VPN_DNS_LOCAL_RESPONSE_DNS);
570   answer->pkt.hdr.size = htons (len);
571   answer->pkt.subtype = GNUNET_DNS_ANSWER_TYPE_REV;
572
573   answer->pkt.from = query_states[id].remote_ip;
574
575   answer->pkt.to = query_states[id].local_ip;
576   answer->pkt.dst_port = query_states[id].local_port;
577
578   struct dns_pkt *dpkt = (struct dns_pkt *) answer->pkt.data;
579
580   dpkt->s.id = id;
581   dpkt->s.aa = 1;
582   dpkt->s.qr = 1;
583   dpkt->s.ra = 1;
584   dpkt->s.qdcount = htons (1);
585   dpkt->s.ancount = htons (1);
586
587   memcpy (dpkt->data, query_states[id].name, query_states[id].namelen);
588   GNUNET_free (query_states[id].name);
589   query_states[id].name = NULL;
590
591   struct dns_query_line *dque =
592       (struct dns_query_line *) (dpkt->data + (query_states[id].namelen));
593   dque->type = htons (12);      /* PTR */
594   dque->class = htons (1);      /* IN */
595
596   char *anname =
597       (char *) (dpkt->data + (query_states[id].namelen) +
598                 sizeof (struct dns_query_line));
599   memcpy (anname, "\xc0\x0c", 2);
600
601   struct dns_record_line *drec_data =
602       (struct dns_record_line *) (dpkt->data + (query_states[id].namelen) +
603                                   sizeof (struct dns_query_line) + 2);
604   drec_data->type = htons (12); /* AAAA */
605   drec_data->class = htons (1); /* IN */
606   /* FIXME: read the TTL from block:
607    * GNUNET_TIME_absolute_get_remaining(rec->expiration_time)
608    *
609    * But how to get the seconds out of this?
610    */
611   drec_data->ttl = htonl (3600);
612
613   /* Calculate at which offset in the packet the length of the name and the
614    * name, it is filled in by the daemon-vpn */
615   answer->pkt.addroffset =
616       htons ((unsigned short) ((unsigned long) (&drec_data->data_len) -
617                                (unsigned long) (&answer->pkt)));
618
619   GNUNET_CONTAINER_DLL_insert_after (head, tail, tail, answer);
620
621   GNUNET_SERVER_notify_transmit_ready (query_states[id].client, len,
622                                        GNUNET_TIME_UNIT_FOREVER_REL,
623                                        &send_answer, query_states[id].client);
624 }
625
626 /**
627  * Receive a block from the dht.
628  */
629 static void
630 receive_dht (void *cls, struct GNUNET_TIME_Absolute exp
631              __attribute__ ((unused)), const GNUNET_HashCode * key
632              __attribute__ ((unused)),
633              const struct GNUNET_PeerIdentity *const *get_path
634              __attribute__ ((unused)),
635              const struct GNUNET_PeerIdentity *const *put_path
636              __attribute__ ((unused)), enum GNUNET_BLOCK_Type type, size_t size,
637              const void *data)
638 {
639
640   unsigned short id = ((struct receive_dht_cls *) cls)->id;
641   struct GNUNET_DHT_GetHandle *handle =
642       ((struct receive_dht_cls *) cls)->handle;
643   GNUNET_free (cls);
644
645   GNUNET_assert (type == GNUNET_BLOCK_TYPE_DNS);
646
647   /* If no query with this id is pending, ignore the block */
648   if (query_states[id].valid != GNUNET_YES)
649     return;
650   query_states[id].valid = GNUNET_NO;
651
652   const struct GNUNET_DNS_Record *rec = data;
653
654   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
655               "Got block of size %d, peer: %08x, desc: %08x\n", size,
656               *((unsigned int *) &rec->peer),
657               *((unsigned int *) &rec->service_descriptor));
658
659   size_t len = sizeof (struct answer_packet) - 1 + sizeof (struct dns_static) + query_states[id].namelen + sizeof (struct dns_query_line) + 2   /* To hold the pointer (as defined in RFC1035) to the name */
660       + sizeof (struct dns_record_line) - 1 + 16;       /* To hold the IPv6-Address */
661
662   struct answer_packet_list *answer =
663       GNUNET_malloc (len + 2 * sizeof (struct answer_packet_list *));
664   memset (answer, 0, len + 2 * sizeof (struct answer_packet_list *));
665
666   answer->pkt.hdr.type = htons (GNUNET_MESSAGE_TYPE_VPN_DNS_LOCAL_RESPONSE_DNS);
667   answer->pkt.hdr.size = htons (len);
668   answer->pkt.subtype = GNUNET_DNS_ANSWER_TYPE_SERVICE;
669
670   GNUNET_CRYPTO_hash (&rec->peer,
671                       sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
672                       &answer->pkt.service_descr.peer);
673
674   memcpy (&answer->pkt.service_descr.service_descriptor,
675           &rec->service_descriptor, sizeof (GNUNET_HashCode));
676   memcpy (&answer->pkt.service_descr.service_type, &rec->service_type,
677           sizeof (answer->pkt.service_descr.service_type));
678   memcpy (&answer->pkt.service_descr.ports, &rec->ports,
679           sizeof (answer->pkt.service_descr.ports));
680
681   answer->pkt.from = query_states[id].remote_ip;
682
683   answer->pkt.to = query_states[id].local_ip;
684   answer->pkt.dst_port = query_states[id].local_port;
685
686   struct dns_pkt *dpkt = (struct dns_pkt *) answer->pkt.data;
687
688   dpkt->s.id = id;
689   dpkt->s.aa = 1;
690   dpkt->s.qr = 1;
691   dpkt->s.ra = 1;
692   dpkt->s.qdcount = htons (1);
693   dpkt->s.ancount = htons (1);
694
695   memcpy (dpkt->data, query_states[id].name, query_states[id].namelen);
696   GNUNET_free (query_states[id].name);
697
698   struct dns_query_line *dque =
699       (struct dns_query_line *) (dpkt->data + (query_states[id].namelen));
700   dque->type = htons (28);      /* AAAA */
701   dque->class = htons (1);      /* IN */
702
703   char *anname =
704       (char *) (dpkt->data + (query_states[id].namelen) +
705                 sizeof (struct dns_query_line));
706   memcpy (anname, "\xc0\x0c", 2);
707
708   struct dns_record_line *drec_data =
709       (struct dns_record_line *) (dpkt->data + (query_states[id].namelen) +
710                                   sizeof (struct dns_query_line) + 2);
711   drec_data->type = htons (28); /* AAAA */
712   drec_data->class = htons (1); /* IN */
713
714   /* FIXME: read the TTL from block:
715    * GNUNET_TIME_absolute_get_remaining(rec->expiration_time)
716    *
717    * But how to get the seconds out of this?
718    */
719   drec_data->ttl = htonl (3600);
720   drec_data->data_len = htons (16);
721
722   /* Calculate at which offset in the packet the IPv6-Address belongs, it is
723    * filled in by the daemon-vpn */
724   answer->pkt.addroffset =
725       htons ((unsigned short) ((unsigned long) (&drec_data->data) -
726                                (unsigned long) (&answer->pkt)));
727
728   GNUNET_CONTAINER_DLL_insert_after (head, tail, tail, answer);
729
730   GNUNET_SERVER_notify_transmit_ready (query_states[id].client, len,
731                                        GNUNET_TIME_UNIT_FOREVER_REL,
732                                        &send_answer, query_states[id].client);
733
734   GNUNET_DHT_get_stop (handle);
735 }
736
737 /**
738  * This receives a GNUNET_MESSAGE_TYPE_REHIJACK and rehijacks the DNS
739  */
740 static void
741 rehijack (void *cls
742           __attribute__ ((unused)), struct GNUNET_SERVER_Client *client,
743           const struct GNUNET_MessageHeader *message __attribute__ ((unused)))
744 {
745   unhijack (dnsoutport);
746   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, hijack, NULL);
747
748   GNUNET_SERVER_receive_done (client, GNUNET_OK);
749 }
750
751 /**
752  * This receives the dns-payload from the daemon-vpn and sends it on over the udp-socket
753  */
754 static void
755 receive_query (void *cls
756                __attribute__ ((unused)), struct GNUNET_SERVER_Client *client,
757                const struct GNUNET_MessageHeader *message)
758 {
759   struct query_packet *pkt = (struct query_packet *) message;
760   struct dns_pkt *dns = (struct dns_pkt *) pkt->data;
761   struct dns_pkt_parsed *pdns = parse_dns_packet (dns);
762
763   query_states[dns->s.id].valid = GNUNET_YES;
764   query_states[dns->s.id].client = client;
765   query_states[dns->s.id].local_ip = pkt->orig_from;
766   query_states[dns->s.id].local_port = pkt->src_port;
767   query_states[dns->s.id].remote_ip = pkt->orig_to;
768   query_states[dns->s.id].namelen = strlen ((char *) dns->data) + 1;
769   if (query_states[dns->s.id].name != NULL)
770     GNUNET_free (query_states[dns->s.id].name);
771   query_states[dns->s.id].name =
772       GNUNET_malloc (query_states[dns->s.id].namelen);
773   memcpy (query_states[dns->s.id].name, dns->data,
774           query_states[dns->s.id].namelen);
775
776   /* The query is for a .gnunet-address */
777   if (pdns->queries[0]->namelen > 9 &&
778       0 == strncmp (pdns->queries[0]->name + (pdns->queries[0]->namelen - 9),
779                     ".gnunet.", 9))
780   {
781     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Query for .gnunet!\n");
782     GNUNET_HashCode key;
783
784     GNUNET_CRYPTO_hash (pdns->queries[0]->name, pdns->queries[0]->namelen,
785                         &key);
786
787     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Getting with key %08x, len is %d\n",
788                 *((unsigned int *) &key), pdns->queries[0]->namelen);
789
790     struct receive_dht_cls *cls =
791         GNUNET_malloc (sizeof (struct receive_dht_cls));
792     cls->id = dns->s.id;
793
794     cls->handle =
795         GNUNET_DHT_get_start (dht, GNUNET_TIME_UNIT_MINUTES,
796                               GNUNET_BLOCK_TYPE_DNS, &key,
797                               DEFAULT_GET_REPLICATION, GNUNET_DHT_RO_NONE, NULL,
798                               0, NULL, 0, receive_dht, cls);
799
800     goto outfree;
801   }
802
803   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Query for '%s'; namelen=%d\n",
804               pdns->queries[0]->name, pdns->queries[0]->namelen);
805
806   /* This is a PTR-Query. Check if it is for "our" network */
807   if (htons (pdns->queries[0]->qtype) == 12 && 74 == pdns->queries[0]->namelen)
808   {
809     char *ipv6addr;
810     char ipv6[16];
811     char ipv6rev[74] =
812         "X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.X.ip6.arpa.";
813     unsigned int i;
814     unsigned long long ipv6prefix;
815     unsigned int comparelen;
816
817     GNUNET_assert (GNUNET_OK ==
818                    GNUNET_CONFIGURATION_get_value_string (cfg, "vpn",
819                                                           "IPV6ADDR",
820                                                           &ipv6addr));
821     inet_pton (AF_INET6, ipv6addr, ipv6);
822     GNUNET_free (ipv6addr);
823
824     GNUNET_assert (GNUNET_OK ==
825                    GNUNET_CONFIGURATION_get_value_number (cfg, "vpn",
826                                                           "IPV6PREFIX",
827                                                           &ipv6prefix));
828     GNUNET_assert (ipv6prefix < 127);
829     ipv6prefix = (ipv6prefix + 7) / 8;
830
831     for (i = ipv6prefix; i < 16; i++)
832       ipv6[i] = 0;
833
834     for (i = 0; i < 16; i++)
835     {
836       unsigned char c1 = ipv6[i] >> 4;
837       unsigned char c2 = ipv6[i] & 0xf;
838
839       if (c1 <= 9)
840         ipv6rev[62 - (4 * i)] = c1 + '0';
841       else
842         ipv6rev[62 - (4 * i)] = c1 + 87;        /* 87 is the difference between 'a' and 10 */
843
844       if (c2 <= 9)
845         ipv6rev[62 - ((4 * i) + 2)] = c2 + '0';
846       else
847         ipv6rev[62 - ((4 * i) + 2)] = c2 + 87;
848     }
849     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "My network is %s'.\n", ipv6rev);
850     comparelen = 10 + 4 * ipv6prefix;
851     if (0 ==
852         strncmp (pdns->queries[0]->name +
853                  (pdns->queries[0]->namelen - comparelen),
854                  ipv6rev + (74 - comparelen), comparelen))
855     {
856       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Reverse-Query for .gnunet!\n");
857
858       GNUNET_SCHEDULER_add_now (send_rev_query, pdns);
859
860       goto out;
861     }
862   }
863
864   char *virt_dns;
865   unsigned int virt_dns_bytes;
866
867   if (GNUNET_SYSERR ==
868       GNUNET_CONFIGURATION_get_value_string (cfg, "vpn", "VIRTDNS", &virt_dns))
869   {
870     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
871                 "No entry 'VIRTDNS' in configuration!\n");
872     exit (1);
873   }
874
875   if (1 != inet_pton (AF_INET, virt_dns, &virt_dns_bytes))
876   {
877     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Error parsing 'VIRTDNS': %s; %m!\n",
878                 virt_dns);
879     exit (1);
880   }
881
882   GNUNET_free (virt_dns);
883
884   if (virt_dns_bytes == pkt->orig_to)
885   {
886     /* This is a packet that was sent directly to the virtual dns-server
887      *
888      * This means we have to send this query over gnunet
889      */
890
891     size_t size =
892         sizeof (struct GNUNET_MESH_Tunnel *) +
893         sizeof (struct GNUNET_MessageHeader) + (ntohs (message->size) -
894                                                 sizeof (struct query_packet) +
895                                                 1);
896     struct tunnel_cls *cls_ = GNUNET_malloc (size);
897
898     cls_->hdr.size = size - sizeof (struct GNUNET_MESH_Tunnel *);
899
900     cls_->hdr.type = ntohs (GNUNET_MESSAGE_TYPE_VPN_REMOTE_QUERY_DNS);
901     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "size: %d\n", size);
902
903     memcpy (&cls_->dns, dns,
904             cls_->hdr.size - sizeof (struct GNUNET_MessageHeader));
905     GNUNET_SCHEDULER_add_now (send_mesh_query, cls_);
906
907     goto outfree;
908   }
909
910
911   /* The query should be sent to the network */
912
913   struct sockaddr_in dest;
914
915   memset (&dest, 0, sizeof dest);
916   dest.sin_port = htons (53);
917   dest.sin_addr.s_addr = pkt->orig_to;
918
919   GNUNET_NETWORK_socket_sendto (dnsout, dns,
920                                 ntohs (pkt->hdr.size) -
921                                 sizeof (struct query_packet) + 1,
922                                 (struct sockaddr *) &dest, sizeof dest);
923
924 outfree:
925   free_parsed_dns_packet (pdns);
926   pdns = NULL;
927 out:
928   GNUNET_SERVER_receive_done (client, GNUNET_OK);
929 }
930
931 static void
932 read_response (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
933
934 static int
935 open_port ()
936 {
937   struct sockaddr_in addr;
938
939   dnsout = GNUNET_NETWORK_socket_create (AF_INET, SOCK_DGRAM, 0);
940   if (dnsout == NULL)
941     return GNUNET_SYSERR;
942   memset (&addr, 0, sizeof (struct sockaddr_in));
943
944   addr.sin_family = AF_INET;
945   int err = GNUNET_NETWORK_socket_bind (dnsout,
946                                         (struct sockaddr *) &addr,
947                                         sizeof (struct sockaddr_in));
948
949   if (err != GNUNET_OK)
950   {
951     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Could not bind a port: %m\n");
952     return GNUNET_SYSERR;
953   }
954
955   /* Read the port we bound to */
956   socklen_t addrlen = sizeof (struct sockaddr_in);
957
958   err =
959       getsockname (GNUNET_NETWORK_get_fd (dnsout), (struct sockaddr *) &addr,
960                    &addrlen);
961
962   dnsoutport = htons (addr.sin_port);
963
964   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Bound to port %d.\n", dnsoutport);
965
966   GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL, dnsout,
967                                  &read_response, NULL);
968
969   return GNUNET_YES;
970 }
971
972 /**
973  * Read a response-packet of the UDP-Socket
974  */
975 static void
976 read_response (void *cls
977                __attribute__ ((unused)),
978                const struct GNUNET_SCHEDULER_TaskContext *tc)
979 {
980   struct sockaddr_in addr;
981   socklen_t addrlen = sizeof (addr);
982   int r;
983   int len;
984
985   if (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN)
986     return;
987
988   memset (&addr, 0, sizeof addr);
989
990 #ifndef MINGW
991   if (0 != ioctl (GNUNET_NETWORK_get_fd (dnsout), FIONREAD, &len))
992   {
993     unhijack (dnsoutport);
994     if (GNUNET_YES == open_port ())
995       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, hijack, NULL);
996     return;
997   }
998 #else
999   /* port the code above? */
1000   len = 65536;
1001 #endif
1002   {
1003     unsigned char buf[len];
1004     struct dns_pkt *dns = (struct dns_pkt *) buf;
1005
1006     r = GNUNET_NETWORK_socket_recvfrom (dnsout, buf, sizeof (buf),
1007                                         (struct sockaddr *) &addr, &addrlen);
1008
1009     if (r < 0)
1010     {
1011       unhijack (dnsoutport);
1012       if (GNUNET_YES == open_port ())
1013         GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, hijack, NULL);
1014       return;
1015     }
1016
1017     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Answer to query %d\n",
1018                 ntohs (dns->s.id));
1019
1020     if (query_states[dns->s.id].valid == GNUNET_YES)
1021     {
1022       if (query_states[dns->s.id].tunnel != NULL)
1023       {
1024         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1025                     "Answer to query %d for a remote peer!\n",
1026                     ntohs (dns->s.id));
1027         /* This response should go through a tunnel */
1028         uint32_t *c =
1029             GNUNET_malloc (4 + sizeof (struct GNUNET_MESH_Tunnel *) + r);
1030         *c = r;
1031         struct GNUNET_MESH_Tunnel **t = (struct GNUNET_MESH_Tunnel **) (c + 1);
1032
1033         *t = query_states[dns->s.id].tunnel;
1034         memcpy (t + 1, dns, r);
1035         if (NULL ==
1036             GNUNET_MESH_tunnel_get_data (query_states[dns->s.id].tunnel))
1037         {
1038           struct GNUNET_MESH_TransmitHandle *th =
1039               GNUNET_MESH_notify_transmit_ready (query_states[dns->s.id].tunnel,
1040                                                  GNUNET_YES,
1041                                                  32,
1042                                                  GNUNET_TIME_UNIT_MINUTES,
1043                                                  NULL,
1044                                                  r +
1045                                                  sizeof (struct
1046                                                          GNUNET_MessageHeader),
1047                                                  mesh_send_response, c);
1048
1049           GNUNET_MESH_tunnel_set_data (query_states[dns->s.id].tunnel, th);
1050         }
1051         else
1052         {
1053           struct tunnel_notify_queue *head =
1054               GNUNET_MESH_tunnel_get_head (query_states[dns->s.id].tunnel);
1055           struct tunnel_notify_queue *tail =
1056               GNUNET_MESH_tunnel_get_tail (query_states[dns->s.id].tunnel);
1057
1058           struct tunnel_notify_queue *element =
1059               GNUNET_malloc (sizeof (struct tunnel_notify_queue));
1060           element->cls = c;
1061           element->len = r + sizeof (struct GNUNET_MessageHeader);
1062           element->cb = mesh_send_response;
1063
1064           GNUNET_CONTAINER_DLL_insert_tail (head, tail, element);
1065           GNUNET_MESH_tunnel_set_head (query_states[dns->s.id].tunnel, head);
1066           GNUNET_MESH_tunnel_set_tail (query_states[dns->s.id].tunnel, tail);
1067         }
1068       }
1069       else
1070       {
1071         query_states[dns->s.id].valid = GNUNET_NO;
1072
1073         size_t len = sizeof (struct answer_packet) + r - 1;     /* 1 for the unsigned char data[1]; */
1074         struct answer_packet_list *answer =
1075             GNUNET_malloc (len + 2 * sizeof (struct answer_packet_list *));
1076         answer->pkt.hdr.type =
1077             htons (GNUNET_MESSAGE_TYPE_VPN_DNS_LOCAL_RESPONSE_DNS);
1078         answer->pkt.hdr.size = htons (len);
1079         answer->pkt.subtype = GNUNET_DNS_ANSWER_TYPE_IP;
1080         answer->pkt.from = addr.sin_addr.s_addr;
1081         answer->pkt.to = query_states[dns->s.id].local_ip;
1082         answer->pkt.dst_port = query_states[dns->s.id].local_port;
1083         memcpy (answer->pkt.data, buf, r);
1084
1085         GNUNET_CONTAINER_DLL_insert_after (head, tail, tail, answer);
1086
1087         GNUNET_SERVER_notify_transmit_ready (query_states[dns->s.id].client,
1088                                              len, GNUNET_TIME_UNIT_FOREVER_REL,
1089                                              &send_answer,
1090                                              query_states[dns->s.id].client);
1091       }
1092     }
1093   }
1094   GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL, dnsout,
1095                                  &read_response, NULL);
1096 }
1097
1098
1099 /**
1100  * Task run during shutdown.
1101  *
1102  * @param cls unused
1103  * @param tc unused
1104  */
1105 static void
1106 cleanup_task (void *cls
1107               __attribute__ ((unused)),
1108               const struct GNUNET_SCHEDULER_TaskContext *tc)
1109 {
1110   GNUNET_assert (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN));
1111
1112   unhijack (dnsoutport);
1113   GNUNET_DHT_disconnect (dht);
1114   GNUNET_MESH_disconnect (mesh_handle);
1115 }
1116
1117 /**
1118  * @brief Create a port-map from udp and tcp redirects
1119  *
1120  * @param udp_redirects
1121  * @param tcp_redirects
1122  *
1123  * @return
1124  */
1125 uint64_t
1126 get_port_from_redirects (const char *udp_redirects, const char *tcp_redirects)
1127 {
1128   uint64_t ret = 0;
1129   char *cpy, *hostname, *redirect;
1130   int local_port, count = 0;
1131
1132   if (NULL != udp_redirects)
1133   {
1134     cpy = GNUNET_strdup (udp_redirects);
1135     for (redirect = strtok (cpy, " "); redirect != NULL;
1136          redirect = strtok (NULL, " "))
1137     {
1138       if (NULL == (hostname = strstr (redirect, ":")))
1139       {
1140         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1141                     "Warning: option %s is not formatted correctly!\n",
1142                     redirect);
1143         continue;
1144       }
1145       hostname[0] = '\0';
1146       local_port = atoi (redirect);
1147       if (!((local_port > 0) && (local_port < 65536)))
1148         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1149                     "Warning: %s is not a correct port.", redirect);
1150
1151       ret |= (0xFFFF & htons (local_port));
1152       ret <<= 16;
1153       count++;
1154
1155       if (count > 4)
1156       {
1157         ret = 0;
1158         goto out;
1159       }
1160     }
1161     GNUNET_free (cpy);
1162     cpy = NULL;
1163   }
1164
1165   if (NULL != tcp_redirects)
1166   {
1167     cpy = GNUNET_strdup (tcp_redirects);
1168     for (redirect = strtok (cpy, " "); redirect != NULL;
1169          redirect = strtok (NULL, " "))
1170     {
1171       if (NULL == (hostname = strstr (redirect, ":")))
1172       {
1173         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1174                     "Warning: option %s is not formatted correctly!\n",
1175                     redirect);
1176         continue;
1177       }
1178       hostname[0] = '\0';
1179       local_port = atoi (redirect);
1180       if (!((local_port > 0) && (local_port < 65536)))
1181         GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1182                     "Warning: %s is not a correct port.", redirect);
1183
1184       ret |= (0xFFFF & htons (local_port));
1185       ret <<= 16;
1186       count++;
1187
1188       if (count > 4)
1189       {
1190         ret = 0;
1191         goto out;
1192       }
1193     }
1194     GNUNET_free (cpy);
1195     cpy = NULL;
1196   }
1197
1198 out:
1199   if (NULL != cpy)
1200     GNUNET_free (cpy);
1201   return ret;
1202 }
1203
1204 void
1205 publish_name (const char *name, uint64_t ports, uint32_t service_type,
1206               struct GNUNET_CRYPTO_RsaPrivateKey *my_private_key)
1207 {
1208   size_t size = sizeof (struct GNUNET_DNS_Record);
1209   struct GNUNET_DNS_Record data;
1210
1211   memset (&data, 0, size);
1212
1213   data.purpose.size = htonl (size - sizeof (struct GNUNET_CRYPTO_RsaSignature));
1214   data.purpose.purpose = GNUNET_SIGNATURE_PURPOSE_DNS_RECORD;
1215
1216   GNUNET_CRYPTO_hash (name, strlen (name) + 1, &data.service_descriptor);
1217   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Store with key1 %x\n",
1218               *((unsigned long long *) &data.service_descriptor));
1219
1220   data.service_type = service_type;
1221   data.ports = ports;
1222
1223   GNUNET_CRYPTO_rsa_key_get_public (my_private_key, &data.peer);
1224
1225   data.expiration_time =
1226       GNUNET_TIME_relative_to_absolute (GNUNET_TIME_relative_multiply
1227                                         (GNUNET_TIME_UNIT_HOURS, 2));
1228
1229   /* Sign the block */
1230   if (GNUNET_OK !=
1231       GNUNET_CRYPTO_rsa_sign (my_private_key, &data.purpose, &data.signature))
1232   {
1233     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "could not sign DNS_Record\n");
1234     return;
1235   }
1236
1237   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Putting with key %08x, size = %d\n",
1238               *((unsigned int *) &data.service_descriptor), size);
1239
1240   GNUNET_DHT_put (dht, &data.service_descriptor, DEFAULT_PUT_REPLICATION,
1241                   GNUNET_DHT_RO_NONE, GNUNET_BLOCK_TYPE_DNS, size,
1242                   (char *) &data,
1243                   GNUNET_TIME_relative_to_absolute (GNUNET_TIME_UNIT_HOURS),
1244                   GNUNET_TIME_UNIT_MINUTES, NULL, NULL);
1245 }
1246
1247 /**
1248  * @brief Publishes the record defined by the section section
1249  *
1250  * @param cls closure
1251  * @param section the current section
1252  */
1253 void
1254 publish_iterate (void *cls __attribute__ ((unused)), const char *section)
1255 {
1256   if ((strlen (section) < 8) ||
1257       (0 != strcmp (".gnunet.", section + (strlen (section) - 8))))
1258     return;
1259
1260   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Parsing dns-name %s\n", section);
1261
1262   char *udp_redirects, *tcp_redirects, *alternative_names, *alternative_name,
1263       *keyfile;
1264
1265   GNUNET_CONFIGURATION_get_value_string (cfg, section, "UDP_REDIRECTS",
1266                                          &udp_redirects);
1267   GNUNET_CONFIGURATION_get_value_string (cfg, section, "TCP_REDIRECTS",
1268                                          &tcp_redirects);
1269
1270   if (GNUNET_OK !=
1271       GNUNET_CONFIGURATION_get_value_filename (cfg, "GNUNETD", "HOSTKEY",
1272                                                &keyfile))
1273   {
1274     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "could not read keyfile-value\n");
1275     if (keyfile != NULL)
1276       GNUNET_free (keyfile);
1277     return;
1278   }
1279
1280   struct GNUNET_CRYPTO_RsaPrivateKey *my_private_key =
1281       GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
1282   GNUNET_free (keyfile);
1283   GNUNET_assert (my_private_key != NULL);
1284
1285   uint64_t ports = get_port_from_redirects (udp_redirects, tcp_redirects);
1286   uint32_t service_type = 0;
1287
1288   if (NULL != udp_redirects)
1289     service_type = GNUNET_DNS_SERVICE_TYPE_UDP;
1290
1291   if (NULL != tcp_redirects)
1292     service_type |= GNUNET_DNS_SERVICE_TYPE_TCP;
1293
1294   service_type = htonl (service_type);
1295
1296
1297   publish_name (section, ports, service_type, my_private_key);
1298
1299   GNUNET_CONFIGURATION_get_value_string (cfg, section, "ALTERNATIVE_NAMES",
1300                                          &alternative_names);
1301   for (alternative_name = strtok (alternative_names, " ");
1302        alternative_name != NULL; alternative_name = strtok (NULL, " "))
1303   {
1304     char *altname =
1305         alloca (strlen (alternative_name) + strlen (section) + 1 + 1);
1306     strcpy (altname, alternative_name);
1307     strcpy (altname + strlen (alternative_name) + 1, section);
1308     altname[strlen (alternative_name)] = '.';
1309
1310     publish_name (altname, ports, service_type, my_private_key);
1311   }
1312
1313   GNUNET_free_non_null (alternative_names);
1314   GNUNET_CRYPTO_rsa_key_free (my_private_key);
1315   GNUNET_free_non_null (udp_redirects);
1316   GNUNET_free_non_null (tcp_redirects);
1317 }
1318
1319 /**
1320  * Publish a DNS-record in the DHT.
1321  */
1322 static void
1323 publish_names (void *cls
1324                __attribute__ ((unused)),
1325                const struct GNUNET_SCHEDULER_TaskContext *tc)
1326 {
1327   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
1328     return;
1329
1330   GNUNET_CONFIGURATION_iterate_sections (cfg, publish_iterate, NULL);
1331
1332   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_HOURS, publish_names, NULL);
1333 }
1334
1335 /**
1336  * @param cls closure
1337  * @param server the initialized server
1338  * @param cfg_ configuration to use
1339  */
1340 static void
1341 run (void *cls, struct GNUNET_SERVER_Handle *server,
1342      const struct GNUNET_CONFIGURATION_Handle *cfg_)
1343 {
1344   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
1345     /* callback, cls, type, size */
1346     {&receive_query, NULL, GNUNET_MESSAGE_TYPE_VPN_DNS_LOCAL_QUERY_DNS, 0},
1347     {&rehijack, NULL, GNUNET_MESSAGE_TYPE_REHIJACK,
1348      sizeof (struct GNUNET_MessageHeader)},
1349     {NULL, NULL, 0, 0}
1350   };
1351
1352   static const struct GNUNET_MESH_MessageHandler mesh_handlers[] = {
1353     {receive_mesh_query, GNUNET_MESSAGE_TYPE_VPN_REMOTE_QUERY_DNS, 0},
1354     {receive_mesh_answer, GNUNET_MESSAGE_TYPE_VPN_REMOTE_ANSWER_DNS, 0},
1355     {NULL, 0, 0}
1356   };
1357
1358   static GNUNET_MESH_ApplicationType apptypes[] = {
1359     GNUNET_APPLICATION_TYPE_END,
1360     GNUNET_APPLICATION_TYPE_END
1361   };
1362
1363
1364   if (GNUNET_YES != open_port ())
1365   {
1366     GNUNET_SCHEDULER_shutdown ();
1367     return;
1368   }
1369
1370   if (GNUNET_YES ==
1371       GNUNET_CONFIGURATION_get_value_yesno (cfg_, "dns", "PROVIDE_EXIT"))
1372     apptypes[0] = GNUNET_APPLICATION_TYPE_INTERNET_RESOLVER;
1373   mesh_handle = GNUNET_MESH_connect (cfg_, NULL, NULL, mesh_handlers, apptypes);
1374
1375   cfg = cfg_;
1376
1377   unsigned int i;
1378
1379   for (i = 0; i < 65536; i++)
1380   {
1381     query_states[i].valid = GNUNET_NO;
1382   }
1383
1384   dht = GNUNET_DHT_connect (cfg, 1024);
1385
1386   GNUNET_SCHEDULER_add_now (publish_names, NULL);
1387
1388   GNUNET_SERVER_add_handlers (server, handlers);
1389   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &cleanup_task,
1390                                 cls);
1391 }
1392
1393 /**
1394  * The main function for the dns service.
1395  *
1396  * @param argc number of arguments from the command line
1397  * @param argv command line arguments
1398  * @return 0 ok, 1 on error
1399  */
1400 int
1401 main (int argc, char *const *argv)
1402 {
1403   return (GNUNET_OK ==
1404           GNUNET_SERVICE_run (argc, argv, "dns", GNUNET_SERVICE_OPTION_NONE,
1405                               &run, NULL)) ? 0 : 1;
1406 }