src/core/test_core_api_mq.c

   1 /*
   2   This file is part of GNUnet.
   3   Copyright (C) 2013 GNUnet e.V.
   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., 51 Franklin Street, Fifth Floor,
  18   Boston, MA 02110-1301, USA.
  19 */
  20
  21 #include "platform.h"
  22 #include "gnunet_util_lib.h"
  23 #include "gnunet_testing_lib.h"
  24 #include "gnunet_core_service.h"
  25
  26
  27 #define NUM_MSG 5
  28
  29 /**
  30  * Has the test been successful?
  31  */
  32 int result;
  33
  34 unsigned int num_received;
  35
  36 struct GNUNET_CORE_Handle *core;
  37
  38 struct GNUNET_MQ_Handle *mq;
  39
  40 struct GNUNET_PeerIdentity myself;
  41
  42
  43 static void
  44 init_cb (void *cls,
  45          const struct GNUNET_PeerIdentity *my_identity)
  46 {
  47   if (NULL == my_identity)
  48   {
  49     GNUNET_break (0);
  50     return;
  51   }
  52   myself = *my_identity;
  53   mq = GNUNET_CORE_mq_create (core, my_identity);
  54 }
  55
  56
  57 static void
  58 connect_cb (void *cls, const struct GNUNET_PeerIdentity *peer)
  59 {
  60   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connected to peer %s.\n",
  61               GNUNET_i2s (peer));
  62   if (0 == memcmp (peer, &myself, sizeof (struct GNUNET_PeerIdentity)))
  63   {
  64     unsigned int i;
  65     struct GNUNET_MQ_Envelope *ev;
  66     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Queueing messages.\n");
  67     for (i = 0; i < NUM_MSG; i++)
  68     {
  69       ev = GNUNET_MQ_msg_header (GNUNET_MESSAGE_TYPE_TEST);
  70       GNUNET_MQ_send (mq, ev);
  71     }
  72   }
  73 }
  74
  75
  76 static int
  77 handle_test (void *cls,
  78              const struct GNUNET_PeerIdentity *other,
  79              const struct GNUNET_MessageHeader *message)
  80 {
  81   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got test message %d\n", num_received);
  82   num_received++;
  83   if (NUM_MSG == num_received)
  84   {
  85     result = GNUNET_OK;
  86     GNUNET_SCHEDULER_shutdown ();
  87     return GNUNET_SYSERR;
  88   }
  89   if (num_received > NUM_MSG)
  90   {
  91     GNUNET_assert (0);
  92     return GNUNET_SYSERR;
  93   }
  94   return GNUNET_OK;
  95 }
  96
  97
  98 static void
  99 shutdown_task (void *cls)
 100 {
 101   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutting down\n");
 102   GNUNET_MQ_destroy (mq);
 103   GNUNET_CORE_disconnect (core);
 104 }
 105
 106
 107 /**
 108  * Initialize framework and start test
 109  *
 110  * @param cls Closure (unused).
 111  * @param cfg Configuration handle.
 112  * @param peer Testing peer handle.
 113  */
 114 static void
 115 run (void *cls,
 116      const struct GNUNET_CONFIGURATION_Handle *cfg,
 117      struct GNUNET_TESTING_Peer *peer)
 118 {
 119   static const struct GNUNET_CORE_MessageHandler handlers[] = {
 120     {&handle_test, GNUNET_MESSAGE_TYPE_TEST, 0},
 121     {NULL, 0, 0}
 122   };
 123   core = GNUNET_CORE_connect (cfg,
 124                               NULL, &init_cb, &connect_cb, NULL,
 125                               NULL, GNUNET_NO, NULL,
 126                               GNUNET_NO, handlers);
 127   if (NULL == core)
 128   {
 129     GNUNET_assert (0);
 130     return;
 131   }
 132   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task, NULL);
 133 }
 134
 135 int
 136 main (int argc, char *argv1[])
 137 {
 138   if (0 != GNUNET_TESTING_peer_run ("test-core-api-mq",
 139                                     "test_core_api_peer1.conf",
 140                                     &run, NULL))
 141     return 2;
 142   return (result == GNUNET_OK) ? 0 : 1;
 143 }