-minor changes
[oweals/gnunet.git] / src / gns / gnunet-service-gns.c
1 /*
2      This file is part of GNUnet.
3      (C) 2009, 2010, 2011 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 gns/gnunet-service-gns.c
23  * @brief GNUnet GNS service
24  * @author Martin Schanzenbach
25  */
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28 #include "gnunet_transport_service.h"
29 #include "gnunet_dns_service.h"
30 #include "gnunet_dnsparser_lib.h"
31 #include "gnunet_namestore_service.h"
32 #include "gnunet_gns_service.h"
33 #include "gns.h"
34
35
36 /* TODO into gnunet_protocols */
37 #define GNUNET_MESSAGE_TYPE_GNS_CLIENT_LOOKUP 23
38 #define GNUNET_MESSAGE_TYPE_GNS_CLIENT_RESULT 24
39
40 /**
41  * A result list for namestore queries
42  */
43 struct GNUNET_GNS_PendingQuery
44 {
45   /* the answer packet */
46   struct GNUNET_DNSPARSER_Packet *answer;
47
48   /* records to put into answer packet */
49   struct GNUNET_CONTAINER_SList *records;
50
51   int num_records;
52   int num_authority_records; //FIXME are all of our replies auth?
53   
54   /* the dns request id */
55   int id; // FIXME can handle->request_id also be used here?
56
57   /* the request handle to reply to */
58   struct GNUNET_DNS_RequestHandle *request_handle;
59 };
60
61
62 /**
63  * Our handle to the DNS handler library
64  */
65 struct GNUNET_DNS_Handle *dns_handle;
66
67 /**
68  * Our handle to the namestore service
69  */
70 struct GNUNET_NAMESTORE_Handle *namestore_handle;
71
72 /**
73  * The configuration the GNS service is running with
74  */
75 const struct GNUNET_CONFIGURATION_Handle *GNS_cfg;
76
77 /**
78  * Our notification context.
79  */
80 static struct GNUNET_SERVER_NotificationContext *nc;
81
82 /**
83  * Our zone hash
84  */
85 const GNUNET_HashCode *my_zone;
86
87 /**
88  * Task run during shutdown.
89  *
90  * @param cls unused
91  * @param tc unused
92  */
93 static void
94 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
95 {
96   GNUNET_DNS_disconnect(dns_handle);
97   GNUNET_NAMESTORE_disconnect(namestore_handle, 0);
98 }
99
100 static void
101 process_ns_result(void* cls, const GNUNET_HashCode *zone,
102                   const char *name, uint32_t record_type,
103                   struct GNUNET_TIME_Absolute expiration,
104                   enum GNUNET_NAMESTORE_RecordFlags flags,
105                   const struct GNUNET_NAMESTORE_SignatureLocation *sig_loc,
106                   size_t size, const void *data)
107 {
108   struct GNUNET_GNS_PendingQuery *answer;
109   struct GNUNET_DNSPARSER_Record *record;
110   struct GNUNET_DNSPARSER_Packet *packet;
111   struct GNUNET_DNSPARSER_Flags dnsflags;
112   char *reply_buffer;
113   struct GNUNET_CONTAINER_SList_Iterator *i;
114   struct GNUNET_CONTAINER_SList_Iterator head;
115   int j;
116   size_t record_len;
117   
118   answer = (struct GNUNET_GNS_PendingQuery *) cls;
119
120
121   if (NULL == data)
122   {
123     /**
124      * Last result received (or none)
125      * FIXME extract to func
126      */
127     reply_buffer = GNUNET_malloc(6000); //FIXME magic number
128     packet = GNUNET_malloc(sizeof(struct GNUNET_DNSPARSER_Packet));
129     packet->answers =
130       GNUNET_malloc(sizeof(struct GNUNET_DNSPARSER_Record) * answer->num_records);
131
132     j = 0;
133     head = GNUNET_CONTAINER_slist_begin (answer->records);
134     i = &head;
135     record_len = sizeof(struct GNUNET_DNSPARSER_Record*);
136     for (;i != NULL; GNUNET_CONTAINER_slist_next (i))
137     {
138       memcpy(&packet->answers[j], 
139              GNUNET_CONTAINER_slist_get (i, &record_len),
140              sizeof (struct GNUNET_DNSPARSER_Record));
141       j++;
142     }
143
144
145     /* FIXME how to handle auth, additional etc */
146     packet->num_answers = answer->num_records;
147     packet->num_authority_records = answer->num_authority_records;
148     
149     dnsflags.authoritative_answer = 1;
150     dnsflags.return_code = GNUNET_DNSPARSER_RETURN_CODE_YXDOMAIN; //not sure
151     dnsflags.query_or_response = 1;
152     packet->flags = dnsflags;
153
154     packet->id = answer->id;
155     size_t bufsize = 6000;
156     int ret = GNUNET_DNSPARSER_pack (packet,
157                            600, /* FIXME max udp payload */
158                            &reply_buffer,
159                            &bufsize); // FIXME magic number bufsize
160     if (ret == GNUNET_OK)
161     {
162       GNUNET_DNS_request_answer(answer->request_handle,
163                                 6000, //FIXME what length here
164                                 reply_buffer);
165       //FIXME return code, free datastructures
166     }
167     else
168     {
169       //FIXME log
170     }
171   }
172   else
173   {
174     /**
175      * New result
176      */
177     record = GNUNET_malloc(sizeof(struct GNUNET_DNSPARSER_Record));
178     record->name = (char*)name;
179     /* FIXME for gns records this requires the dnsparser to be modified!*/
180     //record->data = data; FIXME!
181     record->expiration_time = expiration;
182     record->type = record_type;
183     record->class = GNUNET_DNSPARSER_CLASS_INTERNET; /* srsly? */
184     
185     if (flags == GNUNET_NAMESTORE_RF_AUTHORITY)
186     {
187       answer->num_authority_records++;
188     }
189     
190     answer->num_records++;
191     
192     //FIXME watch for leaks
193     GNUNET_CONTAINER_slist_add (answer->records,
194                                 GNUNET_CONTAINER_SLIST_DISPOSITION_DYNAMIC,
195                                 &record, sizeof(struct GNUNET_DNSPARSER_Record*));
196   }
197 }
198
199
200 /**
201  * The DNS request handler
202  * Is this supposed to happen here in the service (config option)
203  * or in an app that interfaces with the GNS API?
204  *
205  * @param cls closure
206  * @param rh request handle to user for reply
207  * @param request_length number of bytes in request
208  * @param request udp payload of the DNS request
209  */
210 void
211 handle_dns_request(void *cls,
212                    struct GNUNET_DNS_RequestHandle *rh,
213                    size_t request_length,
214                    const char *request)
215 {
216   /**
217    * parse request for tld
218    */
219   struct GNUNET_DNSPARSER_Packet *p;
220   struct GNUNET_GNS_PendingQuery *answer;
221   int namelen;
222   int i;
223   char *tail;
224   
225   p = GNUNET_DNSPARSER_parse (request, request_length);
226   if (NULL == p)
227   {
228     fprintf (stderr, "Received malformed DNS packet, leaving it untouched\n");
229     GNUNET_DNS_request_forward (rh);
230     return;
231   }
232   /**
233    * FIXME factor out
234    * Check tld and decide if we or
235    * legacy dns is responsible
236    */
237   for (i=0;i<p->num_queries;i++)
238   {
239     namelen = strlen(p->queries[i].name);
240     
241     if (namelen < 7) /* this can't be .gnunet */
242       continue;
243     /**
244      * FIXME off by 1?
245      * Move our tld/root to config file
246      * Generate fake DNS reply that replaces .gnunet with .org for testing?
247      */
248     tail = p->queries[i].name+(namelen-7);
249     if (0 == strcmp(tail, ".gnunet"))
250     {
251       /**
252        * Do db lookup here. Make dht lookup if necessary 
253        * FIXME for now only local lookups for our zone!
254        */
255       answer = GNUNET_malloc(sizeof (struct GNUNET_GNS_PendingQuery));
256       answer->records = GNUNET_CONTAINER_slist_create ();
257       answer->id = p->id;
258
259       GNUNET_NAMESTORE_lookup_name(namestore_handle,
260                                    my_zone,
261                                    p->queries[i].name,
262                                    p->queries[i].type,
263                                    &process_ns_result,
264                                    answer);
265       //GNUNET_DNS_request_answer(rh, 0 /*length*/, NULL/*reply*/);
266     }
267     else
268     {
269       GNUNET_DNS_request_forward (rh);
270     }
271   }
272 }
273
274 /*TODO*/
275 static void
276 handle_client_record_lookup(void *cls,
277                             struct GNUNET_SERVER_Client *client,
278                             const struct GNUNET_MessageHeader *message)
279 {
280 }
281
282 /**
283  * test function
284  */
285 void
286 put_some_records(void)
287 {
288   /* put a few records into namestore */
289   char* ipA = "1.2.3.4";
290   char* ipB = "5.6.7.8";
291   GNUNET_NAMESTORE_record_put (namestore_handle,
292                                my_zone,
293                                "alice.gnunet",
294                                GNUNET_GNS_RECORD_TYPE_A,
295                                GNUNET_TIME_absolute_get_forever(),
296                                GNUNET_NAMESTORE_RF_AUTHORITY,
297                                NULL, //sig loc
298                                strlen (ipA),
299                                ipA,
300                                NULL,
301                                NULL);
302   GNUNET_NAMESTORE_record_put (namestore_handle,
303                                my_zone,
304                                "bob.gnunet",
305                                GNUNET_GNS_RECORD_TYPE_A,
306                                GNUNET_TIME_absolute_get_forever(),
307                                GNUNET_NAMESTORE_RF_AUTHORITY,
308                                NULL, //sig loc
309                                strlen (ipB),
310                                ipB,
311                                NULL,
312                                NULL);
313 }
314
315 /**
316  * Process GNS requests.
317  *
318  * @param cls closure
319  * @param server the initialized server
320  * @param c configuration to use
321  */
322 static void
323 run (void *cls, struct GNUNET_SERVER_Handle *server,
324      const struct GNUNET_CONFIGURATION_Handle *c)
325 {
326   /* The IPC message types */
327   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
328     /* callback, cls, type, size */
329     {&handle_client_record_lookup, NULL, GNUNET_MESSAGE_TYPE_GNS_CLIENT_LOOKUP,
330       0},
331     {NULL, NULL, 0, 0}
332   };
333   
334   nc = GNUNET_SERVER_notification_context_create (server, 1);
335
336   /* FIXME - do some config parsing 
337    *       - Maybe only hijack dns if option is set (HIJACK_DNS=1)
338    */
339
340   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
341                                 NULL);
342   /**
343    * Do gnunet dns init here
344    */
345   dns_handle = GNUNET_DNS_connect(c,
346                                   GNUNET_DNS_FLAG_PRE_RESOLUTION,
347                                   &handle_dns_request, /* rh */
348                                   NULL); /* Closure */
349
350   /**
351    * handle to our local namestore
352    */
353   namestore_handle = GNUNET_NAMESTORE_connect(c);
354
355   if (NULL == namestore_handle)
356   {
357     //FIXME do error handling;
358   }
359
360   put_some_records();
361
362   GNUNET_SERVER_add_handlers (server, handlers);
363   /**
364    * Esp the lookup would require to keep track of the clients' context
365    * See dht.
366    * GNUNET_SERVER_disconnect_notify (server, &client_disconnect, NULL);
367    **/
368 }
369
370
371 /**
372  * The main function for the GNS service.
373  *
374  * @param argc number of arguments from the command line
375  * @param argv command line arguments
376  * @return 0 ok, 1 on error
377  */
378 int
379 main (int argc, char *const *argv)
380 {
381   int ret;
382
383   ret =
384       (GNUNET_OK ==
385        GNUNET_SERVICE_run (argc, argv, "gns", GNUNET_SERVICE_OPTION_NONE, &run,
386                            NULL)) ? 0 : 1;
387   return ret;
388 }
389
390 /* end of gnunet-service-gns.c */