2 This file is part of GNUnet.
3 (C) 2010 Christian Grothoff
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.
22 * @file core/test_core_api_send_to_self.c
24 * @author Philipp Toelke
27 #include <gnunet_common.h>
28 #include <gnunet_program_lib.h>
29 #include <gnunet_protocols.h>
30 #include <gnunet_core_service.h>
31 #include <gnunet_constants.h>
39 * Handle to the cleanup task.
41 GNUNET_SCHEDULER_TaskIdentifier die_task;
43 static struct GNUNET_PeerIdentity myself;
48 struct GNUNET_CORE_Handle *core;
51 * Handle to gnunet-service-arm.
53 struct GNUNET_OS_Process *arm_proc;
56 * Function scheduled as very last function, cleans up after us
59 cleanup (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tskctx)
61 die_task = GNUNET_SCHEDULER_NO_TASK;
65 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Disconnecting core.\n");
66 GNUNET_CORE_disconnect (core);
70 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
72 if (0 != GNUNET_OS_process_kill (arm_proc, SIGTERM))
73 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
75 if (GNUNET_OS_process_wait(arm_proc) != GNUNET_OK)
76 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "waitpid");
78 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
79 "ARM process %u stopped\n", GNUNET_OS_process_get_pid (arm_proc));
80 GNUNET_OS_process_close (arm_proc);
83 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Ending test.\n");
87 receive(void* cls, const struct GNUNET_PeerIdentity* other, const struct GNUNET_MessageHeader* message, const struct GNUNET_TRANSPORT_ATS_Information* atsi)
89 if (die_task != GNUNET_SCHEDULER_NO_TASK)
90 GNUNET_SCHEDULER_cancel(die_task);
91 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Received message from peer %s\n", GNUNET_i2s(other));
92 GNUNET_SCHEDULER_add_now(&cleanup, NULL);
98 send_message (void* cls, size_t size, void* buf)
100 if (size == 0 || buf == NULL)
102 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Could not send; got 0 buffer\n");
105 GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Sending!\n");
106 struct GNUNET_MessageHeader *hdr = buf;
107 hdr->size = htons(sizeof(struct GNUNET_MessageHeader));
108 hdr->type = htons(GNUNET_MESSAGE_TYPE_SERVICE_UDP);
109 return ntohs(hdr->size);
113 init (void *cls, struct GNUNET_CORE_Handle *core,
114 const struct GNUNET_PeerIdentity *my_identity,
115 const struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *pk)
119 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Could NOT connect to CORE;\n");
122 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
123 "Correctly connected to CORE; we are the peer %s.\n",
124 GNUNET_i2s (my_identity));
125 memcpy (&myself, my_identity, sizeof (struct GNUNET_PeerIdentity));
129 connect_cb (void *cls, const struct GNUNET_PeerIdentity *peer,
130 const struct GNUNET_TRANSPORT_ATS_Information *atsi)
132 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connected to peer %s.\n",
134 if (0 == memcmp (peer, &myself, sizeof (struct GNUNET_PeerIdentity)))
136 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
137 "Connected to myself; sending message!\n");
138 GNUNET_CORE_notify_transmit_ready (core,
139 0, GNUNET_TIME_UNIT_FOREVER_REL,
141 sizeof (struct GNUNET_MessageHeader),
148 * Main function that will be run by the scheduler.
151 * @param args remaining command-line arguments
152 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
153 * @param cfg configuration
159 const struct GNUNET_CONFIGURATION_Handle *cfg)
161 const static struct GNUNET_CORE_MessageHandler handlers[] = {
162 {&receive, GNUNET_MESSAGE_TYPE_SERVICE_UDP, 0},
166 arm_proc = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-arm",
167 "gnunet-service-arm",
171 "-c", "test_core_api_data.conf", NULL);
173 core = GNUNET_CORE_connect (cfg,
178 NULL, NULL, NULL, 0, NULL, 0, handlers);
180 die_task = GNUNET_SCHEDULER_add_delayed(GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 60), &cleanup, cls);
184 * The main function to obtain template from gnunetd.
186 * @param argc number of arguments from the command line
187 * @param argv command line arguments
188 * @return 0 ok, 1 on error
191 main (int argc, char *const *argv)
193 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
194 GNUNET_GETOPT_OPTION_END
198 GNUNET_PROGRAM_run (argc,
200 "test_core_api_send_to_self",
201 gettext_noop ("help text"),
202 options, &run, NULL)) ? ret : 1;
205 /* end of test_core_api_send_to_self.c */