src/core/core_api_peer_request.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009, 2010 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 2, 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 core/core_api_peer_request.c
  23  * @brief implementation of the peer_request functions
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_core_service.h"
  28 #include "core.h"
  29
  30
  31 /**
  32  * Handle for a request to the core to connect to
  33  * a particular peer.  Can be used to cancel the request
  34  * (before the 'cont'inuation is called).
  35  */
  36 struct GNUNET_CORE_PeerRequestHandle
  37 {
  38
  39   /**
  40    * Our connection to the service.
  41    */
  42   struct GNUNET_CLIENT_Connection *client;
  43
  44   /**
  45    * Scheduler.
  46    */
  47   struct GNUNET_SCHEDULER_Handle *sched;
  48
  49   /**
  50    * Function to call once done.
  51    */
  52   GNUNET_SCHEDULER_Task cont;
  53
  54   /**
  55    * Closure for 'cont'.
  56    */
  57   void *cont_cls;
  58
  59   /**
  60    * When to time out.
  61    */
  62   struct GNUNET_TIME_Absolute timeout;
  63
  64   /**
  65    * Identity of the peer to connect to.
  66    */
  67   struct GNUNET_PeerIdentity peer;
  68
  69   /**
  70    * Message type to use.
  71    */
  72   uint16_t type;
  73 };
  74
  75
  76 /**
  77  * Transmit the request to the core service.
  78  *
  79  * @param cls our 'struct GNUNET_CORE_PeerRequestHandle'
  80  * @param size number of bytes available in buf
  81  * @param buf where the callee should write the message
  82  * @return number of bytes written to buf
  83  */
  84 static size_t
  85 send_request (void *cls,
  86               size_t size,
  87               void *buf)
  88 {
  89   struct GNUNET_CORE_PeerRequestHandle * prh = cls;
  90   struct ConnectMessage msg;
  91
  92   if (buf == NULL)
  93     {
  94       GNUNET_SCHEDULER_add_continuation (prh->sched,
  95                                          prh->cont,
  96                                          prh->cont_cls,
  97                                          GNUNET_SCHEDULER_REASON_TIMEOUT);
  98       GNUNET_CLIENT_disconnect (prh->client, GNUNET_NO);
  99       GNUNET_free (prh);
 100       return 0;
 101     }
 102   GNUNET_assert (size >= sizeof (struct ConnectMessage));
 103   msg.header.type = htons (prh->type);
 104   msg.header.size = htons (sizeof (struct ConnectMessage));
 105   msg.reserved = htonl (0);
 106   msg.timeout = GNUNET_TIME_relative_hton (GNUNET_TIME_absolute_get_remaining (prh->timeout));
 107   msg.peer = prh->peer;
 108   memcpy (buf, &msg, sizeof (msg));
 109   GNUNET_SCHEDULER_add_continuation (prh->sched,
 110                                      prh->cont,
 111                                      prh->cont_cls,
 112                                      GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 113   GNUNET_CLIENT_disconnect (prh->client, GNUNET_YES);
 114   GNUNET_free (prh);
 115   return sizeof (msg);
 116 }
 117
 118
 119 /**
 120  * Request that the core should try to connect to a particular peer.
 121  * Once the request has been transmitted to the core, the continuation
 122  * function will be called.  Note that this does NOT mean that a
 123  * connection was successfully established -- it only means that the
 124  * core will now try.  Successful establishment of the connection
 125  * will be signalled to the 'connects' callback argument of
 126  * 'GNUNET_CORE_connect' only.  If the core service does not respond
 127  * to our connection attempt within the given time frame, 'cont' will
 128  * be called with the TIMEOUT reason code.
 129  *
 130  * @param sched scheduler to use
 131  * @param cfg configuration to use
 132  * @param timeout how long to try to talk to core
 133  * @param peer who should we connect to
 134  * @param cont function to call once the request has been completed (or timed out)
 135  * @param cont_cls closure for cont
 136  * @return NULL on error (cont will not be called), otherwise handle for cancellation
 137  */
 138 struct GNUNET_CORE_PeerRequestHandle *
 139 GNUNET_CORE_peer_request_connect (struct GNUNET_SCHEDULER_Handle *sched,
 140                                   const struct GNUNET_CONFIGURATION_Handle *cfg,
 141                                   struct GNUNET_TIME_Relative timeout,
 142                                   const struct GNUNET_PeerIdentity * peer,
 143                                   GNUNET_SCHEDULER_Task cont,
 144                                   void *cont_cls)
 145 {
 146   struct GNUNET_CORE_PeerRequestHandle *ret;
 147   struct GNUNET_CLIENT_Connection *client;
 148
 149   client = GNUNET_CLIENT_connect (sched, "core", cfg);
 150   if (client == NULL)
 151     return NULL;
 152   ret = GNUNET_malloc (sizeof (struct GNUNET_CORE_PeerRequestHandle));
 153   ret->client = client;
 154   ret->sched = sched;
 155   ret->cont = cont;
 156   ret->cont_cls = cont_cls;
 157   ret->peer = *peer;
 158   ret->type = GNUNET_MESSAGE_TYPE_CORE_REQUEST_CONNECT;
 159   ret->timeout = GNUNET_TIME_relative_to_absolute (timeout);
 160   GNUNET_CLIENT_notify_transmit_ready (client,
 161                                        sizeof (struct ConnectMessage),
 162                                        timeout,
 163                                        GNUNET_YES,
 164                                        &send_request,
 165                                        ret);
 166   return ret;
 167 }
 168
 169
 170 /**
 171  * Cancel a pending request to connect to a particular peer.  Must not
 172  * be called after the 'cont' function was invoked.
 173  *
 174  * @param req request handle that was returned for the original request
 175  */
 176 void
 177 GNUNET_CORE_peer_request_connect_cancel (struct GNUNET_CORE_PeerRequestHandle *req)
 178 {
 179   GNUNET_CLIENT_disconnect (req->client, GNUNET_NO);
 180   GNUNET_free (req);
 181 }
 182
 183
 184 /* end of core_api_peer_request.c */