Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
afdf097733681787f421db9ae4974a0a854e6db7
[simgrid.git] / src / simix / smx_global.c
1 /* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "private.h"
8 #include "xbt/heap.h"
9 #include "xbt/sysdep.h"
10 #include "xbt/log.h"
11 #include "xbt/str.h"
12 #include "xbt/ex.h"             /* ex_backtrace_display */
13
14 XBT_LOG_EXTERNAL_CATEGORY(simix);
15 XBT_LOG_EXTERNAL_CATEGORY(simix_action);
16 XBT_LOG_EXTERNAL_CATEGORY(simix_deployment);
17 XBT_LOG_EXTERNAL_CATEGORY(simix_environment);
18 XBT_LOG_EXTERNAL_CATEGORY(simix_host);
19 XBT_LOG_EXTERNAL_CATEGORY(simix_process);
20 XBT_LOG_EXTERNAL_CATEGORY(simix_synchro);
21 XBT_LOG_EXTERNAL_CATEGORY(simix_context);
22 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_kernel, simix,
23                                 "Logging specific to SIMIX (kernel)");
24
25 SIMIX_Global_t simix_global = NULL;
26 static xbt_heap_t simix_timers = NULL;
27
28 /* FIXME: Yeah, I'll do it in a portable maner one day [Mt] */
29 #include <signal.h>
30
31 static void _XBT_CALL inthandler(int ignored)
32 {
33   INFO0("CTRL-C pressed. Displaying status and bailing out");
34   SIMIX_display_process_status();
35   exit(1);
36 }
37
38 /********************************* SIMIX **************************************/
39
40 /**
41  * \brief Initialize SIMIX internal data.
42  *
43  * \param argc Argc
44  * \param argv Argv
45  */
46 void SIMIX_global_init(int *argc, char **argv)
47 {
48   s_smx_process_t proc;
49
50   if (!simix_global) {
51     /* Connect our log channels: that must be done manually under windows */
52     XBT_LOG_CONNECT(simix_action, simix);
53     XBT_LOG_CONNECT(simix_deployment, simix);
54     XBT_LOG_CONNECT(simix_environment, simix);
55     XBT_LOG_CONNECT(simix_host, simix);
56     XBT_LOG_CONNECT(simix_kernel, simix);
57     XBT_LOG_CONNECT(simix_process, simix);
58     XBT_LOG_CONNECT(simix_synchro, simix);
59     XBT_LOG_CONNECT(simix_context, simix);
60
61     simix_global = xbt_new0(s_SIMIX_Global_t, 1);
62
63     simix_global->host = xbt_dict_new();
64     simix_global->process_to_run =
65         xbt_swag_new(xbt_swag_offset(proc, synchro_hookup));
66     simix_global->process_list =
67         xbt_swag_new(xbt_swag_offset(proc, process_hookup));
68     simix_global->process_to_destroy =
69         xbt_swag_new(xbt_swag_offset(proc, destroy_hookup));
70
71     simix_global->current_process = NULL;
72     simix_global->maestro_process = NULL;
73     simix_global->registered_functions = xbt_dict_new();
74
75     simix_global->create_process_function = NULL;
76     simix_global->kill_process_function = NULL;
77     simix_global->cleanup_process_function = SIMIX_process_cleanup;
78
79 #ifdef HAVE_LATENCY_BOUND_TRACKING
80     simix_global->latency_limited_dict = xbt_dict_new();
81 #endif
82
83     SIMIX_context_mod_init();
84     SIMIX_create_maestro_process();
85
86     /* context exception handlers */
87     __xbt_running_ctx_fetch = SIMIX_process_get_running_context;
88     __xbt_ex_terminate = SIMIX_process_exception_terminate;
89
90     /* Initialize request mechanism */
91     SIMIX_request_init();
92
93     /* Initialize the SIMIX network module */
94     SIMIX_network_init();
95     
96     /* Prepare to display some more info when dying on Ctrl-C pressing */
97     signal(SIGINT, inthandler);
98     surf_init(argc, argv);      /* Initialize SURF structures */
99   }
100   if (!simix_timers) {
101     simix_timers = xbt_heap_new(8, &free);
102   }
103 }
104
105 /**
106  * \brief Clean the SIMIX simulation
107  *
108  * This functions remove the memory used by SIMIX
109  */
110 void SIMIX_clean(void)
111 {
112   /* Kill everyone (except maestro) */
113   SIMIX_process_killall();
114
115   /* Exit the SIMIX network module */
116   SIMIX_network_exit();
117   
118   /* Exit request mechanism */
119   SIMIX_request_destroy();
120   
121   xbt_heap_free(simix_timers);
122   /* Free the remaining data structures */
123   xbt_swag_free(simix_global->process_to_run);
124   xbt_swag_free(simix_global->process_to_destroy);
125   xbt_swag_free(simix_global->process_list);
126   simix_global->process_list = NULL;
127   simix_global->process_to_destroy = NULL;
128   xbt_dict_free(&(simix_global->registered_functions));
129   xbt_dict_free(&(simix_global->host));
130
131 #ifdef HAVE_LATENCY_BOUND_TRACKING
132   xbt_dict_free(&(simix_global->latency_limited_dict));
133 #endif
134
135   /* Let's free maestro now */
136   SIMIX_context_free(simix_global->maestro_process->context);
137   xbt_free(simix_global->maestro_process->running_ctx);
138   xbt_free(simix_global->maestro_process);
139   simix_global->maestro_process = NULL;
140
141   /* Restore the default exception setup */
142   __xbt_running_ctx_fetch = &__xbt_ex_ctx_default;
143   __xbt_ex_terminate = &__xbt_ex_terminate_default;
144
145   /* Finish context module and SURF */
146   SIMIX_context_mod_exit();
147
148   surf_exit();
149
150   xbt_free(simix_global);
151   simix_global = NULL;
152
153   return;
154 }
155
156
157 /**
158  * \brief A clock (in second).
159  *
160  * \return Return the clock.
161  */
162 XBT_INLINE double SIMIX_get_clock(void)
163 {
164   return surf_get_clock();
165 }
166
167 void SIMIX_run(void)
168 {
169   double time = 0;
170   smx_req_t req;
171   xbt_swag_t set;
172   surf_action_t action;
173   smx_timer_t timer;
174   surf_model_t model;
175   unsigned int iter;
176  
177   do {
178     do {
179       DEBUG0("New Schedule Round");
180       SIMIX_context_runall(simix_global->process_to_run);
181       while((req = SIMIX_request_pop())){
182         DEBUG1("Handling request %p", req);
183         SIMIX_request_pre(req);
184       }
185     } while (xbt_swag_size(simix_global->process_to_run));
186
187     time = surf_solve(SIMIX_timer_next());
188
189     /* Notify all the hosts that have failed */
190     /* FIXME: iterate through the list of failed host and mark each of them */
191     /* as failed. On each host, signal all the running processes with host_fail */
192     
193     /* Handle any pending timer */
194     while (xbt_heap_size(simix_timers) > 0 && SIMIX_get_clock() >= SIMIX_timer_next()) {
195        //FIXME: make the timers being real callbacks
196        // (i.e. provide dispatchers that read and expand the args) 
197        timer = xbt_heap_pop(simix_timers);
198        if (timer->func)
199          ((void (*)(void*))timer->func)(timer->args);
200     }
201     /* Wake up all process waiting for the action finish */
202     xbt_dynar_foreach(model_list, iter, model) {
203       for(set = model->states.failed_action_set;
204           set;
205           set = (set == model->states.failed_action_set)
206                 ? model->states.done_action_set
207                 : NULL) {
208         while ((action = xbt_swag_extract(set)))
209           SIMIX_request_post((smx_action_t)action->data);
210       }
211     }
212   } while(time != -1.0);
213
214 }
215
216 /**
217  *      \brief Set the date to execute a function
218  *
219  * Set the date to execute the function on the surf.
220  *      \param date Date to execute function
221  *      \param function Function to be executed
222  *      \param arg Parameters of the function
223  *
224  */
225 XBT_INLINE void SIMIX_timer_set(double date, void *function, void *arg)
226 {
227   smx_timer_t timer = xbt_new0(s_smx_timer_t, 1);
228
229   timer->date = date;
230   timer->func = function;
231   timer->args = arg;
232   xbt_heap_push(simix_timers, timer, date);
233 }
234
235 XBT_INLINE double SIMIX_timer_next(void)
236 {
237   return xbt_heap_size(simix_timers) > 0 ? xbt_heap_maxkey(simix_timers) : -1.0;
238 }
239
240 /**
241  *      \brief Registers a function to create a process.
242  *
243  *      This function registers an user function to be called when a new process is created. The user function have to call the SIMIX_create_process function.
244  *      \param function Create process function
245  *
246  */
247 XBT_INLINE void SIMIX_function_register_process_create(smx_creation_func_t
248                                                        function)
249 {
250   xbt_assert0((simix_global->create_process_function == NULL),
251               "Data already set");
252
253   simix_global->create_process_function = function;
254 }
255
256 /**
257  *      \brief Registers a function to kill a process.
258  *
259  *      This function registers an user function to be called when a new process is killed. The user function have to call the SIMIX_kill_process function.
260  *      \param function Kill process function
261  *
262  */
263 XBT_INLINE void SIMIX_function_register_process_kill(void_f_pvoid_t
264                                                      function)
265 {
266   xbt_assert0((simix_global->kill_process_function == NULL),
267               "Data already set");
268
269   simix_global->kill_process_function = function;
270 }
271
272 /**
273  *      \brief Registers a function to cleanup a process.
274  *
275  *      This function registers an user function to be called when a new process ends properly.
276  *      \param function cleanup process function
277  *
278  */
279 XBT_INLINE void SIMIX_function_register_process_cleanup(void_pfn_smxprocess_t
280                                                         function)
281 {
282   simix_global->cleanup_process_function = function;
283 }
284
285
286 void SIMIX_display_process_status(void)
287 {
288   if (simix_global->process_list == NULL) {
289     return;
290   }
291
292   smx_process_t process = NULL;
293   /*xbt_fifo_item_t item = NULL;
294   smx_action_t act;*/
295   int nbprocess = xbt_swag_size(simix_global->process_list);
296
297   INFO1("%d processes are still running, waiting for something.", nbprocess);
298   /*  List the process and their state */
299   INFO0
300     ("Legend of the following listing: \"<process> on <host>: <status>.\"");
301   xbt_swag_foreach(process, simix_global->process_list) {
302     char *who, *who2;
303
304     asprintf(&who, "%s on %s: %s",
305              process->name,
306              process->smx_host->name,
307              (process->blocked) ? "[BLOCKED] "
308              : ((process->suspended) ? "[SUSPENDED] " : ""));
309
310     if (process->waiting_action) {
311       who2 = bprintf("Waiting for action %p to finish", process->waiting_action);
312     }
313
314       /*
315     if (process->mutex) {
316       who2 =
317         bprintf("%s Blocked on mutex %p", who,
318                 (XBT_LOG_ISENABLED(simix_kernel, xbt_log_priority_verbose)) ?
319                 process->mutex : (void *) 0xdead);
320       free(who);
321       who = who2;
322     } else if (process->cond) {
323       who2 =
324         bprintf
325         ("%s Blocked on condition %p; Waiting for the following actions:",
326          who,
327          (XBT_LOG_ISENABLED(simix_kernel, xbt_log_priority_verbose)) ?
328          process->cond : (void *) 0xdead);
329       free(who);
330       who = who2;
331       xbt_fifo_foreach(process->cond->actions, item, act, smx_action_t) {
332         who2 =
333           bprintf("%s '%s'(%p)", who, act->name,
334                   (XBT_LOG_ISENABLED(simix_kernel, xbt_log_priority_verbose))
335                   ? act : (void *) 0xdead);
336         free(who);
337         who = who2;
338       }
339     } else if (process->sem) {
340       who2 =
341         bprintf
342         ("%s Blocked on semaphore %p; Waiting for the following actions:",
343          who,
344          (XBT_LOG_ISENABLED(simix_kernel, xbt_log_priority_verbose)) ?
345          process->sem : (void *) 0xdead);
346       free(who);
347       who = who2;
348       xbt_fifo_foreach(process->sem->actions, item, act, smx_action_t) {
349         who2 =
350           bprintf("%s '%s'(%p)", who, act->name,
351                   (XBT_LOG_ISENABLED(simix_kernel, xbt_log_priority_verbose))
352                   ? act : (void *) 0xdead);
353         free(who);
354         who = who2;
355       }
356
357     } else {
358       who2 =
359         bprintf
360         ("%s Blocked in an unknown status (please report this bug)", who);
361       free(who);
362       who = who2;
363     }
364     */
365     INFO1("%s.", who);
366     free(who);
367   }
368 }