2 This file is part of GNUnet.
3 Copyright (C) 2012 GNUnet e.V.
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., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
21 * @file dht/gnunet-dht-monitor.c
22 * @brief search for data in DHT
23 * @author Christian Grothoff
24 * @author Bartlomiej Polot
27 #include "gnunet_dht_service.h"
30 * The type of the query
32 static unsigned int block_type;
35 * The key to be monitored
37 static char *query_key;
40 * User supplied timeout value (in seconds)
42 static struct GNUNET_TIME_Relative timeout_request = { 60000 };
52 static struct GNUNET_DHT_Handle *dht_handle;
55 * Global handle of the configuration
57 static const struct GNUNET_CONFIGURATION_Handle *cfg;
60 * Handle for the get request
62 static struct GNUNET_DHT_MonitorHandle *monitor_handle;
65 * Count of messages received
67 static unsigned int result_count;
75 * Task scheduled to handle timeout.
77 static struct GNUNET_SCHEDULER_Task *tt;
81 * Stop monitoring request and start shutdown
83 * @param cls closure (unused)
86 cleanup_task (void *cls)
89 FPRINTF (stderr, "%s", "Cleaning up!\n");
90 if (NULL != monitor_handle)
92 GNUNET_DHT_monitor_stop (monitor_handle);
93 monitor_handle = NULL;
95 if (NULL != dht_handle)
97 GNUNET_DHT_disconnect (dht_handle);
102 GNUNET_SCHEDULER_cancel (tt);
109 * We hit a timeout. Stop monitoring request and start shutdown
111 * @param cls closure (unused)
114 timeout_task (void *cls)
117 GNUNET_SCHEDULER_shutdown ();
123 * Callback called on each GET request going through the DHT.
125 * @param cls Closure.
126 * @param options Options, for instance RecordRoute, DemultiplexEverywhere.
127 * @param type The type of data in the request.
128 * @param hop_count Hop count so far.
129 * @param path_length number of entries in path (or 0 if not recorded).
130 * @param path peers on the GET path (or NULL if not recorded).
131 * @param desired_replication_level Desired replication level.
132 * @param key Key of the requested data.
135 get_callback (void *cls,
136 enum GNUNET_DHT_RouteOption options,
137 enum GNUNET_BLOCK_Type type,
139 uint32_t desired_replication_level,
140 unsigned int path_length,
141 const struct GNUNET_PeerIdentity *path,
142 const struct GNUNET_HashCode * key)
144 FPRINTF (stdout, "GET #%u: type %d, key `%s'\n",
147 GNUNET_h2s_full(key));
153 * Callback called on each GET reply going through the DHT.
155 * @param cls Closure.
156 * @param type The type of data in the result.
157 * @param get_path Peers on GET path (or NULL if not recorded).
158 * @param get_path_length number of entries in get_path.
159 * @param put_path peers on the PUT path (or NULL if not recorded).
160 * @param put_path_length number of entries in get_path.
161 * @param exp Expiration time of the data.
162 * @param key Key of the data.
163 * @param data Pointer to the result data.
164 * @param size Number of bytes in data.
167 get_resp_callback (void *cls,
168 enum GNUNET_BLOCK_Type type,
169 const struct GNUNET_PeerIdentity *get_path,
170 unsigned int get_path_length,
171 const struct GNUNET_PeerIdentity *put_path,
172 unsigned int put_path_length,
173 struct GNUNET_TIME_Absolute exp,
174 const struct GNUNET_HashCode * key,
179 "RESPONSE #%u: type %d, key `%s', data `%.*s'\n",
182 GNUNET_h2s_full (key),
190 * Callback called on each PUT request going through the DHT.
192 * @param cls Closure.
193 * @param options Options, for instance RecordRoute, DemultiplexEverywhere.
194 * @param type The type of data in the request.
195 * @param hop_count Hop count so far.
196 * @param path_length number of entries in path (or 0 if not recorded).
197 * @param path peers on the PUT path (or NULL if not recorded).
198 * @param desired_replication_level Desired replication level.
199 * @param exp Expiration time of the data.
200 * @param key Key under which data is to be stored.
201 * @param data Pointer to the data carried.
202 * @param size Number of bytes in data.
205 put_callback (void *cls,
206 enum GNUNET_DHT_RouteOption options,
207 enum GNUNET_BLOCK_Type type,
209 uint32_t desired_replication_level,
210 unsigned int path_length,
211 const struct GNUNET_PeerIdentity *path,
212 struct GNUNET_TIME_Absolute exp,
213 const struct GNUNET_HashCode * key,
218 "PUT %u: type %d, key `%s', data `%.*s'\n",
221 GNUNET_h2s_full(key),
229 * Main function that will be run by the scheduler.
232 * @param args remaining command-line arguments
233 * @param cfgfile name of the configuration file used (for saving, can be NULL!)
234 * @param c configuration
237 run (void *cls, char *const *args, const char *cfgfile,
238 const struct GNUNET_CONFIGURATION_Handle *c)
240 struct GNUNET_HashCode *key;
241 struct GNUNET_HashCode hc;
245 if (NULL == (dht_handle = GNUNET_DHT_connect (cfg, 1)))
247 FPRINTF (stderr, "%s",
248 _("Failed to connect to DHT service!\n"));
252 if (GNUNET_BLOCK_TYPE_ANY == block_type) /* Type of data not set */
253 block_type = GNUNET_BLOCK_TYPE_TEST;
254 if (NULL != query_key)
258 GNUNET_CRYPTO_hash_from_string (query_key, key))
259 GNUNET_CRYPTO_hash (query_key, strlen (query_key), key);
267 "Monitoring for %s\n",
268 GNUNET_STRINGS_relative_time_to_string (timeout_request, GNUNET_NO));
269 tt = GNUNET_SCHEDULER_add_delayed (timeout_request,
272 GNUNET_SCHEDULER_add_shutdown (&cleanup_task,
274 monitor_handle = GNUNET_DHT_monitor_start (dht_handle,
285 * gnunet-dht-monitor command line options
287 static struct GNUNET_GETOPT_CommandLineOption options[] = {
289 gettext_noop ("the query key"),
290 1, &GNUNET_GETOPT_set_string, &query_key},
291 {'t', "type", "TYPE",
292 gettext_noop ("the type of data to look for"),
293 1, &GNUNET_GETOPT_set_uint, &block_type},
294 {'T', "timeout", "TIMEOUT",
295 gettext_noop ("how long should the monitor command run"),
296 1, &GNUNET_GETOPT_set_relative_time, &timeout_request},
297 {'V', "verbose", NULL,
298 gettext_noop ("be verbose (print progress information)"),
299 0, &GNUNET_GETOPT_set_one, &verbose},
300 GNUNET_GETOPT_OPTION_END
305 * Entry point for gnunet-dht-monitor
307 * @param argc number of arguments from the command line
308 * @param argv command line arguments
309 * @return 0 ok, 1 on error
312 main (int argc, char *const *argv)
314 if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
318 GNUNET_PROGRAM_run (argc, argv, "gnunet-dht-monitor",
320 ("Prints all packets that go through the DHT."),
321 options, &run, NULL)) ? ret : 1;
324 /* end of gnunet-dht-monitor.c */