cd521921e629b6e2083e2ada2aecb5e827ed82c3
[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 Tölke
24  */
25 #include "platform.h"
26 #include "gnunet_getopt_lib.h"
27 #include "gnunet_service_lib.h"
28 #include "gnunet_network_lib.h"
29 #include "gnunet_os_lib.h"
30 #include "gnunet-service-dns-p.h"
31 #include "gnunet_protocols.h"
32 #include "gnunet-vpn-packet.h"
33 #include "gnunet-vpn-pretty-print.h"
34
35
36 struct dns_cls {
37         struct GNUNET_SCHEDULER_Handle *sched;
38
39         struct GNUNET_NETWORK_Handle *dnsout;
40
41         unsigned short dnsoutport;
42 };
43
44 static struct dns_cls mycls;
45
46 void hijack(unsigned short port) {
47         char port_s[6];
48
49         GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Hijacking, port is %d\n", port);
50         snprintf(port_s, 6, "%d", port);
51         GNUNET_OS_start_process(NULL, NULL, "gnunet-helper-hijack-dns", "gnunet-hijack-dns", port_s, NULL);
52 }
53
54 void unhijack(unsigned short port) {
55         char port_s[6];
56
57         GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "unHijacking, port is %d\n", port);
58         snprintf(port_s, 6, "%d", port);
59         GNUNET_OS_start_process(NULL, NULL, "gnunet-helper-hijack-dns", "gnunet-hijack-dns", "-d", port_s, NULL);
60 }
61
62 void receive_query(void *cls, struct GNUNET_SERVER_Client *client, const struct GNUNET_MessageHeader *message)
63 {
64         struct query_packet* pkt = (struct query_packet*)message;
65         GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Received query of length %d\n", ntohs(pkt->hdr.size));
66         struct dns_pkt* dns = (struct dns_pkt*)pkt->data;
67
68         struct sockaddr_in dest;
69         memset(&dest, 0, sizeof dest);
70         dest.sin_port = htons(53);
71         dest.sin_addr.s_addr = pkt->orig_to;
72         /* TODO:
73          * State merken, damit die Antwort korrekt zurückgeschickt werden kann
74          */
75
76         int r = GNUNET_NETWORK_socket_sendto(mycls.dnsout, dns, ntohs(pkt->hdr.size) - sizeof(struct query_packet) + 1, (struct sockaddr*) &dest, sizeof dest);
77         GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "send %d bytes to socket\n", r);
78
79         GNUNET_SERVER_receive_done(client, GNUNET_OK);
80 }
81
82 /**
83  * Task run during shutdown.
84  *
85  * @param cls unused
86  * @param tc unused
87  */
88 static void
89 cleanup_task (void *cls,
90               const struct GNUNET_SCHEDULER_TaskContext *tc)
91 {
92         unhijack(mycls.dnsoutport);
93 }
94
95 /**
96  * @param cls closure
97  * @param sched scheduler to use
98  * @param server the initialized server
99  * @param cfg configuration to use
100  */
101 static void
102 run (void *cls,
103      struct GNUNET_SCHEDULER_Handle *sched,
104      struct GNUNET_SERVER_Handle *server,
105      const struct GNUNET_CONFIGURATION_Handle *cfg)
106 {
107   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
108           /* callback, cls, type, size */
109     {&receive_query, NULL, GNUNET_MESSAGE_TYPE_LOCAL_QUERY_DNS, 0},
110     {NULL, NULL, 0, 0}
111   };
112   struct sockaddr_in addr;
113
114   mycls.sched = sched;
115   mycls.dnsout = GNUNET_NETWORK_socket_create (AF_INET, SOCK_DGRAM, 0);
116   if (mycls.dnsout == NULL) 
117     return;
118   memset(&addr, 0, sizeof(struct sockaddr_in));
119
120   int err = GNUNET_NETWORK_socket_bind (mycls.dnsout,
121                                         (struct sockaddr*)&addr, 
122                                         sizeof(struct sockaddr_in));
123
124   if (err != GNUNET_YES) {
125         GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Could not bind a port, exiting\n");
126         return;
127   }
128   socklen_t addrlen = sizeof(struct sockaddr_in);
129   err = getsockname(GNUNET_NETWORK_get_fd(mycls.dnsout),
130                     (struct sockaddr*) &addr, 
131                     &addrlen);
132
133   mycls.dnsoutport = htons(addr.sin_port);
134
135   hijack(htons(addr.sin_port));
136
137   GNUNET_SERVER_add_handlers (server, handlers);
138   GNUNET_SCHEDULER_add_delayed (sched,
139                   GNUNET_TIME_UNIT_FOREVER_REL,
140                   &cleanup_task,
141                   cls);
142 }
143
144 /**
145  * The main function for the dns service.
146  *
147  * @param argc number of arguments from the command line
148  * @param argv command line arguments
149  * @return 0 ok, 1 on error
150  */
151 int
152 main (int argc, char *const *argv)
153 {
154   return (GNUNET_OK ==
155           GNUNET_SERVICE_run (argc,
156                               argv,
157                               "dns",
158                               GNUNET_SERVICE_OPTION_NONE,
159                               &run, NULL)) ? 0 : 1;
160 }