2 This file is part of GNUnet.
3 (C) 2009, 2010 Christian Grothoff (and other contributing authors)
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.
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.
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.
24 * @brief library to access the GNS service
25 * @author Martin Schanzenbach
29 #include "gnunet_util_lib.h"
30 #include "gnunet_constants.h"
31 #include "gnunet_arm_service.h"
32 #include "gnunet_hello_lib.h"
33 #include "gnunet_protocols.h"
34 #include "gnunet_dht_service.h"
36 #include "gnunet_gns_service.h"
38 /* TODO into gnunet_protocols */
39 #define GNUNET_MESSAGE_TYPE_GNS_LOOKUP 23
40 #define GNUNET_MESSAGE_TYPE_GNS_LOOKUP_RESULT 24
41 #define GNUNET_MESSAGE_TYPE_GNS_SHORTEN 25
42 #define GNUNET_MESSAGE_TYPE_GNS_SHORTEN_RESULT 26
43 #define GNUNET_MESSAGE_TYPE_GNS_GET_AUTH 27
44 #define GNUNET_MESSAGE_TYPE_GNS_GET_AUTH_RESULT 28
49 struct GNUNET_GNS_QueueEntry
54 struct GNUNET_GNS_QueueEntry *next;
59 struct GNUNET_GNS_QueueEntry *prev;
65 struct GNUNET_GNS_Handle *gns_handle;
67 /* processor to call on shorten result */
68 GNUNET_GNS_ShortenResultProcessor shorten_proc;
70 /* processor to call on lookup result */
71 GNUNET_GNS_LookupResultProcessor lookup_proc;
73 /* processor to call on authority lookup result */
74 GNUNET_GNS_GetAuthResultProcessor auth_proc;
76 /* processor closure */
83 * Entry in our list of messages to be (re-)transmitted.
88 * This is a doubly-linked list.
90 struct PendingMessage *prev;
93 * This is a doubly-linked list.
95 struct PendingMessage *next;
98 * Size of the message.
106 * Connection to the GNS service.
108 struct GNUNET_GNS_Handle
112 * Configuration to use.
114 const struct GNUNET_CONFIGURATION_Handle *cfg;
117 * Socket (if available).
119 struct GNUNET_CLIENT_Connection *client;
122 * Currently pending transmission request (or NULL).
124 struct GNUNET_CLIENT_TransmitHandle *th;
129 * Head of linked list of shorten messages we would like to transmit.
131 struct PendingMessage *pending_head;
134 * Tail of linked list of shorten messages we would like to transmit.
136 struct PendingMessage *pending_tail;
139 * Head of linked list of shorten messages we would like to transmit.
141 struct GNUNET_GNS_QueueEntry *shorten_head;
144 * Tail of linked list of shorten messages we would like to transmit.
146 struct GNUNET_GNS_QueueEntry *shorten_tail;
149 * Head of linked list of lookup messages we would like to transmit.
151 struct GNUNET_GNS_QueueEntry *lookup_head;
154 * Tail of linked list of lookup messages we would like to transmit.
156 struct GNUNET_GNS_QueueEntry *lookup_tail;
159 * Head of linked list of authority lookup messages we would like to transmit.
161 struct GNUNET_GNS_QueueEntry *get_auth_head;
164 * Tail of linked list of authority lookup messages we would like to transmit.
166 struct GNUNET_GNS_QueueEntry *get_auth_tail;
171 GNUNET_SCHEDULER_TaskIdentifier reconnect_task;
174 * Did we start our receive loop yet?
179 * Reconnect necessary
185 * Try to send messages from list of messages to send
186 * @param handle GNS_Handle
189 process_pending_messages (struct GNUNET_GNS_Handle *handle);
193 * Reconnect to GNS service.
195 * @param h the handle to the namestore service
198 reconnect (struct GNUNET_GNS_Handle *h)
200 GNUNET_assert (NULL == h->client);
201 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
202 "Trying to connect to GNS...\n");
203 h->client = GNUNET_CLIENT_connect ("gns", h->cfg);
204 GNUNET_assert (NULL != h->client);
210 * @param cls the handle
211 * @param tc task context
214 reconnect_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
216 struct GNUNET_GNS_Handle *h = cls;
218 h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
224 * Disconnect from service and then reconnect.
226 * @param h our handle
229 force_reconnect (struct GNUNET_GNS_Handle *h)
231 h->reconnect = GNUNET_NO;
232 GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
234 h->reconnect_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
240 * Transmit the next pending message, called by notify_transmit_ready
243 transmit_pending (void *cls, size_t size, void *buf);
246 * Handler for messages received from the GNS service
248 * @param cls the 'struct GNUNET_GNS_Handle'
249 * @param msg the incoming message
252 process_message (void *cls, const struct GNUNET_MessageHeader *msg);
255 * Try to send messages from list of messages to send
258 process_pending_messages (struct GNUNET_GNS_Handle *handle)
260 struct PendingMessage *p;
262 if (handle->client == NULL)
264 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
265 "process_pending_messages called, but client is null\n");
269 if (handle->th != NULL)
272 if (NULL == (p = handle->pending_head))
275 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
276 "Trying to transmit %d bytes...\n", p->size);
279 GNUNET_CLIENT_notify_transmit_ready (handle->client,
281 GNUNET_TIME_UNIT_FOREVER_REL,
282 GNUNET_NO, &transmit_pending,
284 if (NULL != handle->th)
287 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
288 "notify_transmit_ready returned NULL!\n");
293 * Transmit the next pending message, called by notify_transmit_ready
296 transmit_pending (void *cls, size_t size, void *buf)
298 struct GNUNET_GNS_Handle *handle = cls;
299 struct PendingMessage *p;
305 if ((size == 0) || (buf == NULL))
307 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
308 "Transmission to GNS service failed!\n");
309 force_reconnect(handle);
316 if (NULL == (p = handle->pending_head))
319 while ((NULL != (p = handle->pending_head)) && (p->size <= size))
321 memcpy (&cbuf[tsize], &p[1], p->size);
324 GNUNET_CONTAINER_DLL_remove (handle->pending_head, handle->pending_tail, p);
325 if (GNUNET_YES != handle->in_receive)
327 GNUNET_CLIENT_receive (handle->client, &process_message, handle,
328 GNUNET_TIME_UNIT_FOREVER_REL);
329 handle->in_receive = GNUNET_YES;
334 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
335 "Sending %d bytes\n", tsize);
337 process_pending_messages(handle);
342 * Process a given reply that might match the given
345 * @param cls the 'struct GNUNET_GNS_ClientResultMessage'
346 * @param key query of the request
347 * @param value the 'struct GNUNET_GNS_LookupHandle' of a request matching the same key
350 process_shorten_reply (struct GNUNET_GNS_QueueEntry *qe,
351 const struct GNUNET_GNS_ClientShortenResultMessage *msg)
353 struct GNUNET_GNS_Handle *h = qe->gns_handle;
354 const char *short_name;
356 GNUNET_CONTAINER_DLL_remove(h->shorten_head, h->shorten_tail, qe);
358 short_name = (char*)(&msg[1]);
360 if (ntohs (((struct GNUNET_MessageHeader*)msg)->size) <
361 sizeof (struct GNUNET_GNS_ClientShortenResultMessage))
369 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
370 "Received shortened reply `%s' from GNS service\n",
373 GNUNET_CLIENT_receive (h->client, &process_message, h,
374 GNUNET_TIME_UNIT_FOREVER_REL);
375 qe->shorten_proc(qe->proc_cls, short_name);
382 * Process a given reply that might match the given
385 * @param qe the handle to the request
386 * @param msg the message to process
389 process_get_auth_reply (struct GNUNET_GNS_QueueEntry *qe,
390 const struct GNUNET_GNS_ClientGetAuthResultMessage *msg)
392 struct GNUNET_GNS_Handle *h = qe->gns_handle;
393 const char *auth_name;
395 GNUNET_CONTAINER_DLL_remove(h->get_auth_head, h->get_auth_tail, qe);
397 auth_name = (char*)(&msg[1]);
399 if (ntohs (((struct GNUNET_MessageHeader*)msg)->size) <
400 sizeof (struct GNUNET_GNS_ClientGetAuthResultMessage))
408 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
409 "Received GET_AUTH reply `%s' from GNS service\n",
412 GNUNET_CLIENT_receive (h->client, &process_message, h,
413 GNUNET_TIME_UNIT_FOREVER_REL);
414 qe->auth_proc(qe->proc_cls, auth_name);
419 * Process a given reply to the lookup request
421 * @param cls the 'struct GNUNET_GNS_ClientResultMessage'
422 * @param key query of the request
423 * @param value the 'struct GNUNET_GNS_LookupHandle' of a request matching the same key
424 * @return GNUNET_YES to continue to iterate over all results,
425 * GNUNET_NO if the reply is malformed
428 process_lookup_reply (struct GNUNET_GNS_QueueEntry *qe,
429 const struct GNUNET_GNS_ClientLookupResultMessage *msg)
431 struct GNUNET_GNS_Handle *h = qe->gns_handle;
432 int rd_count = ntohl(msg->rd_count);
433 size_t len = ntohs (((struct GNUNET_MessageHeader*)msg)->size);
434 struct GNUNET_NAMESTORE_RecordData rd[rd_count];
436 GNUNET_CONTAINER_DLL_remove(h->lookup_head, h->lookup_tail, qe);
438 if (len < sizeof (struct GNUNET_GNS_ClientLookupResultMessage))
446 len -= sizeof(struct GNUNET_GNS_ClientLookupResultMessage);
448 GNUNET_CLIENT_receive (h->client, &process_message, h,
449 GNUNET_TIME_UNIT_FOREVER_REL);
450 if (GNUNET_SYSERR == GNUNET_NAMESTORE_records_deserialize (len,
455 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
456 "Failed to serialize lookup reply from GNS service!\n");
457 qe->lookup_proc(qe->proc_cls, 0, NULL);
462 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
463 "Received lookup reply from GNS service (count=%d)\n",
464 ntohl(msg->rd_count));
465 qe->lookup_proc(qe->proc_cls, rd_count, rd);
471 * Handler for messages received from the GNS service
473 * @param cls the 'struct GNUNET_GNS_Handle'
474 * @param msg the incoming message
477 process_message (void *cls, const struct GNUNET_MessageHeader *msg)
479 struct GNUNET_GNS_Handle *handle = cls;
480 struct GNUNET_GNS_QueueEntry *qe;
481 const struct GNUNET_GNS_ClientLookupResultMessage *lookup_msg;
482 const struct GNUNET_GNS_ClientShortenResultMessage *shorten_msg;
483 const struct GNUNET_GNS_ClientGetAuthResultMessage *get_auth_msg;
487 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
491 GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
492 "Error receiving data from GNS service, reconnecting\n");
493 force_reconnect (handle);
497 type = ntohs (msg->type);
499 if (type == GNUNET_MESSAGE_TYPE_GNS_LOOKUP_RESULT)
501 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
503 lookup_msg = (const struct GNUNET_GNS_ClientLookupResultMessage *) msg;
504 r_id = ntohl (lookup_msg->id);
506 if (r_id > handle->r_id)
508 /** no request found */
510 GNUNET_CLIENT_receive (handle->client, &process_message, handle,
511 GNUNET_TIME_UNIT_FOREVER_REL);
515 for (qe = handle->lookup_head; qe != NULL; qe = qe->next)
517 if (qe->r_id == r_id)
521 process_lookup_reply(qe, lookup_msg);
526 else if (type == GNUNET_MESSAGE_TYPE_GNS_SHORTEN_RESULT)
528 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
529 "Got SHORTEN_RESULT msg\n");
530 shorten_msg = (struct GNUNET_GNS_ClientShortenResultMessage *) msg;
532 r_id = ntohl (shorten_msg->id);
534 if (r_id > handle->r_id)
536 /** no request found */
538 GNUNET_CLIENT_receive (handle->client, &process_message, handle,
539 GNUNET_TIME_UNIT_FOREVER_REL);
543 for (qe = handle->shorten_head; qe != NULL; qe = qe->next)
545 if (qe->r_id == r_id)
549 process_shorten_reply(qe, shorten_msg);
552 else if (type == GNUNET_MESSAGE_TYPE_GNS_GET_AUTH_RESULT)
554 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
555 "Got GET_AUTH_RESULT msg\n");
556 get_auth_msg = (struct GNUNET_GNS_ClientGetAuthResultMessage *) msg;
558 r_id = ntohl (get_auth_msg->id);
560 if (r_id > handle->r_id)
562 /** no request found */
564 GNUNET_CLIENT_receive (handle->client, &process_message, handle,
565 GNUNET_TIME_UNIT_FOREVER_REL);
569 for (qe = handle->get_auth_head; qe != NULL; qe = qe->next)
571 if (qe->r_id == r_id)
575 process_get_auth_reply(qe, get_auth_msg);
580 if (GNUNET_YES == handle->reconnect)
581 force_reconnect (handle);
587 * Initialize the connection with the GNS service.
589 * @param cfg configuration to use
590 * @param ht_len size of the internal hash table to use for parallel requests
591 * @return handle to the GNS service, or NULL on error
593 struct GNUNET_GNS_Handle *
594 GNUNET_GNS_connect (const struct GNUNET_CONFIGURATION_Handle *cfg)
596 struct GNUNET_GNS_Handle *handle;
598 handle = GNUNET_malloc (sizeof (struct GNUNET_GNS_Handle));
599 handle->reconnect = GNUNET_NO;
602 //handle->reconnect_task = GNUNET_SCHEDULER_add_now (&reconnect_task, handle);
603 handle->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
605 handle->in_receive = GNUNET_NO;
611 * Shutdown connection with the GNS service.
613 * @param handle handle of the GNS connection to stop
616 GNUNET_GNS_disconnect (struct GNUNET_GNS_Handle *h)
618 GNUNET_CLIENT_disconnect (h->client, GNUNET_NO);
619 if (GNUNET_SCHEDULER_NO_TASK != h->reconnect_task)
621 GNUNET_SCHEDULER_cancel (h->reconnect_task);
622 h->reconnect_task = GNUNET_SCHEDULER_NO_TASK;
629 * Helper function to generate request ids
635 get_request_id (struct GNUNET_GNS_Handle *h)
637 uint32_t r_id = h->r_id;
643 * Perform an asynchronous Lookup operation on the GNS.
645 * @param handle handle to the GNS service
646 * @param name the name to look up
647 * @param iter function to call on each result
648 * @param iter_cls closure for iter
649 * @return handle to stop the async get
651 struct GNUNET_GNS_QueueEntry *
652 GNUNET_GNS_lookup (struct GNUNET_GNS_Handle *handle,
654 enum GNUNET_GNS_RecordType type,
655 GNUNET_GNS_LookupResultProcessor proc,
658 /* IPC to shorten gns names, return shorten_handle */
659 struct GNUNET_GNS_ClientLookupMessage *lookup_msg;
660 struct GNUNET_GNS_QueueEntry *qe;
662 struct PendingMessage *pending;
669 msize = sizeof (struct GNUNET_GNS_ClientLookupMessage) + strlen(name) + 1;
670 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Trying to lookup %s in GNS\n", name);
672 qe = GNUNET_malloc(sizeof (struct GNUNET_GNS_QueueEntry));
673 qe->gns_handle = handle;
674 qe->lookup_proc = proc;
675 qe->proc_cls = proc_cls;
676 qe->r_id = get_request_id(handle);
677 GNUNET_CONTAINER_DLL_insert_tail(handle->lookup_head,
678 handle->lookup_tail, qe);
680 pending = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
681 memset(pending, 0, (sizeof (struct PendingMessage) + msize));
683 pending->size = msize;
685 lookup_msg = (struct GNUNET_GNS_ClientLookupMessage *) &pending[1];
686 lookup_msg->header.type = htons (GNUNET_MESSAGE_TYPE_GNS_LOOKUP);
687 lookup_msg->header.size = htons (msize);
688 lookup_msg->id = htonl(qe->r_id);
689 lookup_msg->type = htonl(type);
691 memcpy(&lookup_msg[1], name, strlen(name));
693 GNUNET_CONTAINER_DLL_insert (handle->pending_head, handle->pending_tail,
696 process_pending_messages (handle);
702 * Perform a name shortening operation on the GNS.
704 * @param handle handle to the GNS service
705 * @param name the name to look up
706 * @param proc function to call on result
707 * @param proc_cls closure for processor
708 * @return handle to the operation
710 struct GNUNET_GNS_QueueEntry *
711 GNUNET_GNS_shorten (struct GNUNET_GNS_Handle *handle,
713 GNUNET_GNS_ShortenResultProcessor proc,
716 /* IPC to shorten gns names, return shorten_handle */
717 struct GNUNET_GNS_ClientShortenMessage *shorten_msg;
718 struct GNUNET_GNS_QueueEntry *qe;
720 struct PendingMessage *pending;
727 msize = sizeof (struct GNUNET_GNS_ClientShortenMessage) + strlen(name) + 1;
728 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Trying to shorten %s in GNS\n", name);
730 qe = GNUNET_malloc(sizeof (struct GNUNET_GNS_QueueEntry));
731 qe->gns_handle = handle;
732 qe->shorten_proc = proc;
733 qe->proc_cls = proc_cls;
734 qe->r_id = get_request_id(handle);
735 GNUNET_CONTAINER_DLL_insert_tail(handle->shorten_head,
736 handle->shorten_tail, qe);
738 pending = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
739 memset(pending, 0, (sizeof (struct PendingMessage) + msize));
741 pending->size = msize;
743 shorten_msg = (struct GNUNET_GNS_ClientShortenMessage *) &pending[1];
744 shorten_msg->header.type = htons (GNUNET_MESSAGE_TYPE_GNS_SHORTEN);
745 shorten_msg->header.size = htons (msize);
746 shorten_msg->id = htonl(qe->r_id);
748 memcpy(&shorten_msg[1], name, strlen(name));
750 GNUNET_CONTAINER_DLL_insert (handle->pending_head, handle->pending_tail,
753 process_pending_messages (handle);
759 * Perform an authority lookup for a given name.
761 * @param handle handle to the GNS service
762 * @param name the name to look up authority for
763 * @param proc function to call on result
764 * @param proc_cls closure for processor
765 * @return handle to the operation
767 struct GNUNET_GNS_QueueEntry *
768 GNUNET_GNS_get_authority (struct GNUNET_GNS_Handle *handle,
770 GNUNET_GNS_GetAuthResultProcessor proc,
773 struct GNUNET_GNS_ClientGetAuthMessage *get_auth_msg;
774 struct GNUNET_GNS_QueueEntry *qe;
776 struct PendingMessage *pending;
783 msize = sizeof (struct GNUNET_GNS_ClientGetAuthMessage) + strlen(name) + 1;
784 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
785 "Trying to look up authority for %s in GNS\n", name);
787 qe = GNUNET_malloc(sizeof (struct GNUNET_GNS_QueueEntry));
788 qe->gns_handle = handle;
789 qe->auth_proc = proc;
790 qe->proc_cls = proc_cls;
791 qe->r_id = get_request_id(handle);
792 GNUNET_CONTAINER_DLL_insert_tail(handle->get_auth_head,
793 handle->get_auth_tail, qe);
795 pending = GNUNET_malloc (sizeof (struct PendingMessage) + msize);
796 memset(pending, 0, (sizeof (struct PendingMessage) + msize));
798 pending->size = msize;
800 get_auth_msg = (struct GNUNET_GNS_ClientGetAuthMessage *) &pending[1];
801 get_auth_msg->header.type = htons (GNUNET_MESSAGE_TYPE_GNS_GET_AUTH);
802 get_auth_msg->header.size = htons (msize);
803 get_auth_msg->id = htonl(qe->r_id);
805 memcpy(&get_auth_msg[1], name, strlen(name));
807 GNUNET_CONTAINER_DLL_insert (handle->pending_head, handle->pending_tail,
810 process_pending_messages (handle);
815 /* end of gns_api.c */