2 This file is part of GNUnet.
3 (C) 2009 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.
22 * @file experimentation/gnunet-daemon-experimentation_scheduler.c
23 * @brief experimentation daemon: execute experiments
24 * @author Christian Grothoff
25 * @author Matthias Wachs
28 #include "gnunet_getopt_lib.h"
29 #include "gnunet_util_lib.h"
30 #include "gnunet_core_service.h"
31 #include "gnunet_statistics_service.h"
32 #include "gnunet-daemon-experimentation.h"
35 * An experiment is added during startup as not running NOT_RUNNING
37 * The scheduler then decides to schedule it and sends a request to the
38 * remote peer, if core cannot send since it is busy we wait for some time
39 * and change state to BUSY, if we can send we change to REQUESTED and wait
40 * for remote peers ACK.
42 * When we receive an ACK we change to STARTED and when scheduler decides that
43 * the experiment is finished we change to STOPPED.
48 /* Experiment is added and waiting to be executed */
50 /* Cannot send request to remote peer, core is busy*/
52 /* We requested experiment and wait for remote peer to ACK */
54 /* Experiment is running */
56 /* Experiment is done */
60 struct ScheduledExperiment {
61 struct ScheduledExperiment *next;
62 struct ScheduledExperiment *prev;
68 GNUNET_SCHEDULER_TaskIdentifier task;
71 struct ScheduledExperiment *waiting_in_head;
72 struct ScheduledExperiment *waiting_in_tail;
74 struct ScheduledExperiment *running_in_head;
75 struct ScheduledExperiment *running_in_tail;
77 struct ScheduledExperiment *waiting_out_head;
78 struct ScheduledExperiment *waiting_out_tail;
80 struct ScheduledExperiment *running_out_head;
81 struct ScheduledExperiment *running_out_tail;
84 static unsigned int experiments_scheduled;
85 static unsigned int experiments_running;
86 static unsigned int experiments_requested;
89 static struct ScheduledExperiment *
90 find_experiment (struct ScheduledExperiment *head, struct ScheduledExperiment *tail,
91 struct Node *n, struct Experiment *e, int outbound)
93 struct ScheduledExperiment *cur;
94 for (cur = head; NULL != cur; cur = cur->next)
96 if ((cur->n == n) && (cur->e == e) && (cur->outbound == outbound)) /* Node and experiment are equal */
103 request_timeout (void *cls,const struct GNUNET_SCHEDULER_TaskContext* tc)
105 struct ScheduledExperiment *se = cls;
106 se->task = GNUNET_SCHEDULER_NO_TASK;
108 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Peer `%s' did not respond to request for experiment `%s'\n",
109 GNUNET_i2s (&se->n->id), se->e->name);
111 GNUNET_CONTAINER_DLL_remove (waiting_out_head, waiting_out_tail, se);
114 /* Remove experiment */
115 GNUNET_assert (experiments_requested > 0);
116 experiments_requested --;
117 GNUNET_STATISTICS_set (GED_stats, "# experiments requested", experiments_requested, GNUNET_NO);
120 static void run_experiment_inbound (void *cls,const struct GNUNET_SCHEDULER_TaskContext* tc)
122 struct ScheduledExperiment *se = cls;
123 struct GNUNET_TIME_Relative start;
124 struct GNUNET_TIME_Relative end;
126 se->task = GNUNET_SCHEDULER_NO_TASK;
130 /* Send START_ACK message */
131 GED_nodes_send_start_ack (se->n, se->e);
132 se->state = REQUESTED;
133 /* Schedule to run */
134 start = GNUNET_TIME_absolute_get_remaining(se->e->start);
135 if (0 == start.rel_value)
136 se->task = GNUNET_SCHEDULER_add_now (&run_experiment_inbound, se);
138 se->task = GNUNET_SCHEDULER_add_delayed (start, &run_experiment_inbound, se);
143 /* Experiment is running */
144 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Running %s experiment `%s' peer for `%s'\n",
145 "inbound", GNUNET_i2s (&se->n->id), se->e->name);
150 end = GNUNET_TIME_absolute_get_remaining(GNUNET_TIME_absolute_add (se->e->stop, se->e->frequency));
151 if (0 == end.rel_value)
154 return; /* End of experiment is reached */
157 se->task = GNUNET_SCHEDULER_add_delayed (se->e->frequency, &run_experiment_inbound, se);
160 /* Experiment expired */
168 static void run_experiment_outbound (void *cls,const struct GNUNET_SCHEDULER_TaskContext* tc)
170 struct ScheduledExperiment *se = cls;
171 struct GNUNET_TIME_Relative end;
173 se->task = GNUNET_SCHEDULER_NO_TASK;
177 /* Send START message */
178 GED_nodes_request_start (se->n, se->e);
179 se->state = REQUESTED;
180 se->task = GNUNET_SCHEDULER_add_delayed (EXP_RESPONSE_TIMEOUT, &request_timeout, se);
181 experiments_requested ++;
182 GNUNET_STATISTICS_set (GED_stats, "# experiments requested", experiments_requested, GNUNET_NO);
185 /* Expecting START_ACK */
189 /* Experiment is running */
190 GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Running %s experiment `%s' peer for `%s'\n",
191 "outbound", GNUNET_i2s (&se->n->id), se->e->name);
196 end = GNUNET_TIME_absolute_get_remaining(GNUNET_TIME_absolute_add (se->e->stop, se->e->frequency));
197 if (0 == end.rel_value)
200 return; /* End of experiment is reached */
203 se->task = GNUNET_SCHEDULER_add_delayed (se->e->frequency, &run_experiment_outbound, se);
206 /* Experiment expired */
215 * Handle a START message from a remote node
218 * @param e the experiment
221 GED_scheduler_handle_start (struct Node *n, struct Experiment *e)
223 struct ScheduledExperiment *se;
225 if ((NULL != (se = find_experiment (waiting_in_head, waiting_in_tail, n, e, GNUNET_NO))) ||
226 (NULL != (se = find_experiment (running_in_head, running_in_tail, n, e, GNUNET_NO))))
228 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Received duplicate %s message from peer %s for experiment `%s'\n"),
229 "START", GNUNET_i2s (&n->id), e->name);
234 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Received %s message from peer %s for experiment `%s'\n"),
235 "START", GNUNET_i2s (&n->id), e->name);
237 GED_scheduler_add (n, e, GNUNET_NO);
241 * Handle a START_ACK message from a remote node
244 * @param e the experiment
247 GED_scheduler_handle_start_ack (struct Node *n, struct Experiment *e)
249 struct ScheduledExperiment *se;
251 if (NULL == (se = find_experiment (waiting_out_head, waiting_out_tail, n, e, GNUNET_YES)))
257 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Received %s message from peer %s for requested experiment `%s'\n"),
258 "START_ACK", GNUNET_i2s (&n->id), e->name);
260 if (GNUNET_SCHEDULER_NO_TASK != se->task)
261 GNUNET_SCHEDULER_cancel (se->task); /* *Canceling timeout task */
263 /* Remove from waiting list, add to running list */
264 GNUNET_CONTAINER_DLL_remove (waiting_out_head, waiting_out_tail, se);
265 GNUNET_CONTAINER_DLL_insert (running_out_head, waiting_out_tail, se);
267 /* Change state and schedule to run */
269 se->task = GNUNET_SCHEDULER_add_now (&run_experiment_outbound, se);
274 * Handle a STOP message from a remote node
277 * @param e the experiment
280 GED_scheduler_handle_stop (struct Node *n, struct Experiment *e)
282 struct ScheduledExperiment *se;
284 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Received %s message from peer %s for experiment `%s'\n"),
285 "STOP", GNUNET_i2s (&n->id), e->name);
287 if (NULL != (se = find_experiment (waiting_in_head, waiting_in_tail, n, e, GNUNET_NO)))
289 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Received %s message from peer %s for waiting experiment `%s'\n"),
290 "STOP", GNUNET_i2s (&n->id), e->name);
293 if (NULL != (se = find_experiment (running_in_head, running_in_tail, n, e, GNUNET_NO)))
295 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Received %s message from peer %s for running experiment `%s'\n"),
296 "STOP", GNUNET_i2s (&n->id), e->name);
302 * Add a new experiment for a node
305 * @param e the experiment
306 * @param outbound are we initiator (GNUNET_YES) or client (GNUNET_NO)?
309 GED_scheduler_add (struct Node *n, struct Experiment *e, int outbound)
311 struct ScheduledExperiment *se;
312 struct GNUNET_TIME_Relative start;
313 struct GNUNET_TIME_Relative end;
315 GNUNET_assert ((GNUNET_YES == outbound) || (GNUNET_NO == outbound));
317 start = GNUNET_TIME_absolute_get_remaining(e->start);
318 end = GNUNET_TIME_absolute_get_remaining(e->stop);
319 if (0 == end.rel_value)
320 return; /* End of experiment is reached */
322 /* Add additional checks here if required */
323 se = GNUNET_malloc (sizeof (struct ScheduledExperiment));
324 se->state = NOT_RUNNING;
325 se->outbound = outbound;
329 if (GNUNET_YES == outbound)
331 if (0 == start.rel_value)
332 se->task = GNUNET_SCHEDULER_add_now (&run_experiment_outbound, se);
334 se->task = GNUNET_SCHEDULER_add_delayed (start, &run_experiment_outbound, se);
335 GNUNET_CONTAINER_DLL_insert (waiting_out_head, waiting_out_tail, se);
339 if (0 == start.rel_value)
340 se->task = GNUNET_SCHEDULER_add_now (&run_experiment_inbound, se);
342 se->task = GNUNET_SCHEDULER_add_delayed (start, &run_experiment_inbound, se);
343 GNUNET_CONTAINER_DLL_insert (waiting_in_head, waiting_in_tail, se);
346 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Added %s experiment `%s' for node to be scheduled\n",
347 (GNUNET_YES == outbound) ? "outbound" : "inbound", e->name, GNUNET_i2s(&se->n->id));
348 experiments_scheduled ++;
349 GNUNET_STATISTICS_set (GED_stats, "# experiments scheduled", experiments_scheduled, GNUNET_NO);
354 * Start the scheduler component
357 GED_scheduler_start ()
359 experiments_requested = 0;
360 experiments_scheduled = 0;
365 * Stop the scheduler component
368 GED_scheduler_stop ()
370 struct ScheduledExperiment *cur;
371 struct ScheduledExperiment *next;
373 next = waiting_in_head;
374 while (NULL != (cur = next))
377 GNUNET_CONTAINER_DLL_remove (waiting_in_head, waiting_in_tail, cur);
378 if (GNUNET_SCHEDULER_NO_TASK != cur->task)
380 GNUNET_SCHEDULER_cancel (cur->task);
381 cur->task = GNUNET_SCHEDULER_NO_TASK;
384 GNUNET_assert (experiments_scheduled > 0);
385 experiments_scheduled --;
386 GNUNET_STATISTICS_set (GED_stats, "# experiments scheduled", experiments_scheduled, GNUNET_NO);
389 next = running_in_head;
390 while (NULL != (cur = next))
393 GNUNET_CONTAINER_DLL_remove (running_in_head, running_in_tail, cur);
394 if (GNUNET_SCHEDULER_NO_TASK != cur->task)
396 GNUNET_SCHEDULER_cancel (cur->task);
397 cur->task = GNUNET_SCHEDULER_NO_TASK;
400 GNUNET_assert (experiments_running > 0);
401 experiments_running --;
402 GNUNET_STATISTICS_set (GED_stats, "# experiments running", experiments_running, GNUNET_NO);
405 next = waiting_out_head;
406 while (NULL != (cur = next))
409 GNUNET_CONTAINER_DLL_remove (waiting_out_head, waiting_out_tail, cur);
410 if (GNUNET_SCHEDULER_NO_TASK != cur->task)
412 GNUNET_SCHEDULER_cancel (cur->task);
413 cur->task = GNUNET_SCHEDULER_NO_TASK;
416 GNUNET_assert (experiments_scheduled > 0);
417 experiments_scheduled --;
418 GNUNET_STATISTICS_set (GED_stats, "# experiments scheduled", experiments_scheduled, GNUNET_NO);
421 next = running_out_head;
422 while (NULL != (cur = next))
425 GNUNET_CONTAINER_DLL_remove (running_out_head, running_out_tail, cur);
426 if (GNUNET_SCHEDULER_NO_TASK != cur->task)
428 GNUNET_SCHEDULER_cancel (cur->task);
429 cur->task = GNUNET_SCHEDULER_NO_TASK;
432 GNUNET_assert (experiments_running > 0);
433 experiments_running --;
434 GNUNET_STATISTICS_set (GED_stats, "# experiments running", experiments_running, GNUNET_NO);
438 /* end of gnunet-daemon-experimentation_scheduler.c */