Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Add levels to lib host_lib, link_lib and as_router_lib
[simgrid.git] / src / simix / smx_global.c
index 2b72da8..a486e41 100644 (file)
-/*     $Id$     */
-
-/* Copyright (c) 2007 Arnaud Legrand, Bruno Donassolo.
-   All rights reserved.                                          */
+/* Copyright (c) 2007, 2008, 2009, 2010. The SimGrid Team.
+ * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
 #include "private.h"
+#include "xbt/heap.h"
 #include "xbt/sysdep.h"
 #include "xbt/log.h"
-#include "xbt/ex.h" /* ex_backtrace_display */
+#include "xbt/str.h"
+#include "xbt/ex.h"             /* ex_backtrace_display */
+#include "mc/mc.h"
+
+XBT_LOG_EXTERNAL_CATEGORY(simix);
+XBT_LOG_EXTERNAL_CATEGORY(simix_action);
+XBT_LOG_EXTERNAL_CATEGORY(simix_deployment);
+XBT_LOG_EXTERNAL_CATEGORY(simix_environment);
+XBT_LOG_EXTERNAL_CATEGORY(simix_host);
+XBT_LOG_EXTERNAL_CATEGORY(simix_process);
+XBT_LOG_EXTERNAL_CATEGORY(simix_synchro);
+XBT_LOG_EXTERNAL_CATEGORY(simix_context);
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_kernel, simix,
-                               "Logging specific to SIMIX (kernel)");
+                                "Logging specific to SIMIX (kernel)");
 
+smx_global_t simix_global = NULL;
+static xbt_heap_t simix_timers = NULL;
 
-SIMIX_Global_t simix_global = NULL;
+static void* SIMIX_action_mallocator_new_f(void);
+static void SIMIX_action_mallocator_free_f(void* action);
+static void SIMIX_action_mallocator_reset_f(void* action);
 
+extern void smx_ctx_raw_new_sr(void);
 
-/** \defgroup simix_simulation   SIMIX simulation Functions
- *  \brief This section describes the functions you need to know to
- *  set up a simulation. You should have a look at \ref SIMIX_examples 
- *  to have an overview of their usage.
- *    \htmlonly <!-- DOXYGEN_NAVBAR_LABEL="Simulation functions" --> \endhtmlonly
- */
+/* FIXME: Yeah, I'll do it in a portable maner one day [Mt] */
+#include <signal.h>
+
+static void _XBT_CALL inthandler(int ignored)
+{
+  XBT_INFO("CTRL-C pressed. Displaying status and bailing out");
+  SIMIX_display_process_status();
+  exit(1);
+}
 
 /********************************* SIMIX **************************************/
 
-/** \ingroup simix_simulation
- * \brief Initialize some SIMIX internal data.
- */
-void SIMIX_global_init_args(int *argc, char **argv)
+XBT_INLINE double SIMIX_timer_next(void)
 {
-  SIMIX_global_init(argc,argv);
+  return xbt_heap_size(simix_timers) > 0 ? xbt_heap_maxkey(simix_timers) : -1.0;
 }
 
-/** \ingroup simix_simulation
- * \brief Initialize some SIMIX internal data.
+/**
+ * \brief Initialize SIMIX internal data.
+ *
+ * \param argc Argc
+ * \param argv Argv
  */
 void SIMIX_global_init(int *argc, char **argv)
 {
-       s_smx_process_t proc;
+  s_smx_process_t proc;
+
+  if (!simix_global) {
+    /* Connect our log channels: that must be done manually under windows */
+    XBT_LOG_CONNECT(simix_action, simix);
+    XBT_LOG_CONNECT(simix_deployment, simix);
+    XBT_LOG_CONNECT(simix_environment, simix);
+    XBT_LOG_CONNECT(simix_host, simix);
+    XBT_LOG_CONNECT(simix_kernel, simix);
+    XBT_LOG_CONNECT(simix_process, simix);
+    XBT_LOG_CONNECT(simix_synchro, simix);
+    XBT_LOG_CONNECT(simix_context, simix);
+
+    simix_global = xbt_new0(s_smx_global_t, 1);
+
+    simix_global->host = xbt_dict_new();
+    simix_global->process_to_run = xbt_dynar_new(sizeof(void *), NULL);
+    simix_global->process_list =
+        xbt_swag_new(xbt_swag_offset(proc, process_hookup));
+    simix_global->process_to_destroy =
+        xbt_swag_new(xbt_swag_offset(proc, destroy_hookup));
+
+    simix_global->maestro_process = NULL;
+    simix_global->registered_functions = xbt_dict_new();
+
+    simix_global->create_process_function = SIMIX_process_create;
+    simix_global->kill_process_function = SIMIX_process_kill;
+    simix_global->cleanup_process_function = SIMIX_process_cleanup;
+    simix_global->action_mallocator = xbt_mallocator_new(65536,
+        SIMIX_action_mallocator_new_f, SIMIX_action_mallocator_free_f,
+        SIMIX_action_mallocator_reset_f);
+
+    surf_init(argc, argv);      /* Initialize SURF structures */
+    SIMIX_context_mod_init();
+    SIMIX_create_maestro_process();
+
+    /* context exception handlers */
+    __xbt_running_ctx_fetch = SIMIX_process_get_running_context;
+    __xbt_ex_terminate = SIMIX_process_exception_terminate;
+
+    /* Initialize request mechanism */
+    SIMIX_request_init();
+
+    /* Initialize the SIMIX network module */
+    SIMIX_network_init();
+
+    /* Prepare to display some more info when dying on Ctrl-C pressing */
+    signal(SIGINT, inthandler);
+  }
+  if (!simix_timers) {
+    simix_timers = xbt_heap_new(8, &free);
+  }
+
+  XBT_DEBUG("ADD SIMIX LEVELS");
+  SIMIX_HOST_LEVEL = xbt_lib_add_level(host_lib,free);
+}
+
+/**
+ * \brief Clean the SIMIX simulation
+ *
+ * This functions remove the memory used by SIMIX
+ */
+void SIMIX_clean(void)
+{
+#ifdef TIME_BENCH
+  smx_ctx_raw_new_sr();
+#endif
 
-       if (!simix_global) {
-               surf_init(argc, argv);  /* Initialize some common structures. Warning, it sets simix_global=NULL */
+  /* Kill everyone (except maestro) */
+  SIMIX_process_killall(simix_global->maestro_process);
 
-               simix_global = xbt_new0(s_SIMIX_Global_t,1);
+  /* Exit the SIMIX network module */
+  SIMIX_network_exit();
 
-               simix_global->host = xbt_fifo_new();
-               simix_global->process_to_run = xbt_swag_new(xbt_swag_offset(proc,synchro_hookup));
-               simix_global->process_list = xbt_swag_new(xbt_swag_offset(proc,process_hookup));
-               simix_global->current_process = NULL;
-               simix_global->registered_functions = xbt_dict_new();
-       }
+  /* Exit request mechanism */
+  SIMIX_request_destroy();
+
+  xbt_heap_free(simix_timers);
+  /* Free the remaining data structures */
+  xbt_dynar_free(&simix_global->process_to_run);
+  xbt_swag_free(simix_global->process_to_destroy);
+  xbt_swag_free(simix_global->process_list);
+  simix_global->process_list = NULL;
+  simix_global->process_to_destroy = NULL;
+  xbt_dict_free(&(simix_global->registered_functions));
+  xbt_dict_free(&(simix_global->host));
+
+  /* Let's free maestro now */
+  SIMIX_context_free(simix_global->maestro_process->context);
+  xbt_free(simix_global->maestro_process->running_ctx);
+  xbt_free(simix_global->maestro_process);
+  simix_global->maestro_process = NULL;
+
+  /* Restore the default exception setup */
+  __xbt_running_ctx_fetch = &__xbt_ex_ctx_default;
+  __xbt_ex_terminate = &__xbt_ex_terminate_default;
+
+  /* Finish context module and SURF */
+  SIMIX_context_mod_exit();
+
+  surf_exit();
+
+  xbt_mallocator_free(simix_global->action_mallocator);
+  xbt_free(simix_global);
+  simix_global = NULL;
+
+  return;
 }
 
-/* Debug purpose, incomplete */
-void __SIMIX_display_process_status(void)
+
+/**
+ * \brief A clock (in second).
+ *
+ * \return Return the clock.
+ */
+XBT_INLINE double SIMIX_get_clock(void)
 {
-   smx_process_t process = NULL;
-   xbt_fifo_item_t item = NULL;
-        smx_action_t act;
-   int nbprocess=xbt_swag_size(simix_global->process_list);
-   
-   INFO1("SIMIX: %d processes are still running, waiting for something.",
-        nbprocess);
-   /*  List the process and their state */
-   INFO0("SIMIX: <process> on <host>: <status>.");
-   xbt_swag_foreach(process, simix_global->process_list) {
-      simdata_process_t p_simdata = (simdata_process_t) process->simdata;
-     // simdata_host_t h_simdata=(simdata_host_t)p_simdata->host->simdata;
-      char *who;
-       
-      asprintf(&who,"SIMIX:  %s on %s: %s",
-              process->name,
-                               p_simdata->host->name,
-              (process->simdata->blocked)?"[BLOCKED] "
-              :((process->simdata->suspended)?"[SUSPENDED] ":""));
-                       if (p_simdata->mutex) {
-                               DEBUG1("Block on a mutex: %s", who);                    
-                       }
-                       else if (p_simdata->cond) {
-                               DEBUG1("Block on a condition: %s", who);
-                               DEBUG0("Waiting actions:");
-                               xbt_fifo_foreach(p_simdata->cond->actions,item, act, smx_action_t) {
-                                       DEBUG1("\t %s", act->name);
-                               }
-                       }
-                       else DEBUG1("Unknown block status: %s", who);
-      free(who);
-   }
+  if(MC_IS_ENABLED){
+    return MC_process_clock_get(SIMIX_process_self());
+  }else{
+    return surf_get_clock();
+  }
 }
 
-/* FIXME: Yeah, I'll do it in a portable maner one day [Mt] */
-#include <signal.h>
-
-static void _XBT_CALL inthandler(int ignored)
+void SIMIX_run(void)
 {
-   INFO0("CTRL-C pressed. Displaying status and bailing out");
-   __SIMIX_display_process_status();
-   exit(1);
+  double time = 0;
+  smx_req_t req;
+  xbt_swag_t set;
+  surf_action_t action;
+  smx_timer_t timer;
+  surf_model_t model;
+  unsigned int iter;
+
+  do {
+    XBT_DEBUG("New Schedule Round; size(queue)=%lu",
+        xbt_dynar_length(simix_global->process_to_run));
+#ifdef TIME_BENCH
+    smx_ctx_raw_new_sr();
+#endif
+    do {
+      XBT_DEBUG("New Sub-Schedule Round; size(queue)=%lu",
+              xbt_dynar_length(simix_global->process_to_run));
+      SIMIX_context_runall(simix_global->process_to_run);
+      while ((req = SIMIX_request_pop())) {
+        XBT_DEBUG("Handling request %p", req);
+        SIMIX_request_pre(req, 0);
+      }
+    } while (xbt_dynar_length(simix_global->process_to_run));
+
+    time = surf_solve(SIMIX_timer_next());
+
+    /* Notify all the hosts that have failed */
+    /* FIXME: iterate through the list of failed host and mark each of them */
+    /* as failed. On each host, signal all the running processes with host_fail */
+
+    /* Handle any pending timer */
+    while (xbt_heap_size(simix_timers) > 0 && SIMIX_get_clock() >= SIMIX_timer_next()) {
+       //FIXME: make the timers being real callbacks
+       // (i.e. provide dispatchers that read and expand the args) 
+       timer = xbt_heap_pop(simix_timers);
+       if (timer->func)
+         ((void (*)(void*))timer->func)(timer->args);
+    }
+    /* Wake up all process waiting for the action finish */
+    xbt_dynar_foreach(model_list, iter, model) {
+      for (set = model->states.failed_action_set;
+           set;
+           set = (set == model->states.failed_action_set)
+                 ? model->states.done_action_set
+                 : NULL) {
+        while ((action = xbt_swag_extract(set)))
+          SIMIX_request_post((smx_action_t) action->data);
+      }
+    }
+  } while (time != -1.0);
+
+  if (xbt_swag_size(simix_global->process_list) != 0) {
+
+    XBT_WARN("Oops ! Deadlock or code not perfectly clean.");
+    SIMIX_display_process_status();
+    xbt_abort();
+  }
 }
 
-/** \ingroup msg_simulation
- * \brief Launch the SIMIX simulation
+/**
+ *     \brief Set the date to execute a function
+ *
+ * Set the date to execute the function on the surf.
+ *     \param date Date to execute function
+ *     \param function Function to be executed
+ *     \param arg Parameters of the function
+ *
  */
-void __SIMIX_main(void)
+XBT_INLINE void SIMIX_timer_set(double date, void *function, void *arg)
 {
-       smx_process_t process = NULL;
-       smx_cond_t cond = NULL;
-       smx_action_t smx_action;
-       xbt_fifo_t actions_done = xbt_fifo_new();
-       xbt_fifo_t actions_failed = xbt_fifo_new();
-
-       /* Prepare to display some more info when dying on Ctrl-C pressing */
-       signal(SIGINT,inthandler);
-
-       /* Clean IO before the run */
-       fflush(stdout);
-       fflush(stderr);
-
-       //surf_solve(); /* Takes traces into account. Returns 0.0 */
-       /* xbt_fifo_size(msg_global->process_to_run) */
-
-       while (SIMIX_solve(actions_done, actions_failed) != -1.0) {
-
-               while ( (smx_action = xbt_fifo_pop(actions_failed)) ) {
-
-                       xbt_fifo_item_t _cursor;
-
-                       DEBUG1("** %s failed **",smx_action->name);
-                       xbt_fifo_foreach(smx_action->simdata->cond_list,_cursor,cond,smx_cond_t) {
-                               xbt_swag_foreach(process,cond->sleeping) {
-                                       DEBUG2("\t preparing to wake up %s on %s",           
-                                                       process->name,  process->simdata->host->name);
-                               }
-                               SIMIX_cond_broadcast(cond);
-                               /* remove conditional from action */
-                               xbt_fifo_remove(smx_action->simdata->cond_list,cond);
-                       }
-               }
-
-               while ( (smx_action = xbt_fifo_pop(actions_done)) ) {
-                       xbt_fifo_item_t _cursor;
-
-                       DEBUG1("** %s done **",smx_action->name);
-                       xbt_fifo_foreach(smx_action->simdata->cond_list,_cursor,cond,smx_cond_t) {
-                               xbt_swag_foreach(process,cond->sleeping) {
-                                       DEBUG2("\t preparing to wake up %s on %s",           
-                                                       process->name,  process->simdata->host->name);
-                               }
-                               SIMIX_cond_broadcast(cond);
-                               /* remove conditional from action */
-                               xbt_fifo_remove(smx_action->simdata->cond_list,cond);
-                       }
-               }
-       }
-       return;
+  smx_timer_t timer = xbt_new0(s_smx_timer_t, 1);
+
+  timer->date = date;
+  timer->func = function;
+  timer->args = arg;
+  xbt_heap_push(simix_timers, timer, date);
 }
 
-/** \ingroup msg_simulation
- * \brief Kill all running process
+/**
+ * \brief Registers a function to create a process.
+ *
+ * This function registers a function to be called
+ * when a new process is created. The function has
+ * to call SIMIX_process_create().
+ * \param function create process function
+ */
+XBT_INLINE void SIMIX_function_register_process_create(smx_creation_func_t
+                                                       function)
+{
+  simix_global->create_process_function = function;
+}
 
- * \param reset_PIDs should we reset the PID numbers. A negative
- *   number means no reset and a positive number will be used to set the PID
- *   of the next newly created process.
+/**
+ * \brief Registers a function to kill a process.
+ *
+ * This function registers a function to be called when a
+ * process is killed. The function has to call the SIMIX_process_kill().
+ *
+ * \param function Kill process function
  */
-void SIMIX_process_killall()
+XBT_INLINE void SIMIX_function_register_process_kill(void_pfn_smxprocess_t
+                                                     function)
 {
-  smx_process_t p = NULL;
-  smx_process_t self = SIMIX_process_self();
+  simix_global->kill_process_function = function;
+}
 
-  while((p=xbt_swag_extract(simix_global->process_list))) {
-    if(p!=self) SIMIX_process_kill(p);
-  }
+/**
+ * \brief Registers a function to cleanup a process.
+ *
+ * This function registers a user function to be called when
+ * a process ends properly.
+ *
+ * \param function cleanup process function
+ */
+XBT_INLINE void SIMIX_function_register_process_cleanup(void_pfn_smxprocess_t
+                                                        function)
+{
+  simix_global->cleanup_process_function = function;
+}
 
-  xbt_context_empty_trash();
 
-  if(self) {
-    xbt_context_yield();
+void SIMIX_display_process_status(void)
+{
+  if (simix_global->process_list == NULL) {
+    return;
   }
 
-  return;
-}
+  smx_process_t process = NULL;
+  int nbprocess = xbt_swag_size(simix_global->process_list);
 
-/** \ingroup msg_simulation
- * \brief Clean the SIMIX simulation
- */
-void SIMIX_clean(void)
-{
-  xbt_fifo_item_t i = NULL;
-  smx_host_t h = NULL;
-  smx_process_t p = NULL;
+  XBT_INFO("%d processes are still running, waiting for something.", nbprocess);
+  /*  List the process and their state */
+  XBT_INFO
+    ("Legend of the following listing: \"<process> on <host>: <status>.\"");
+  xbt_swag_foreach(process, simix_global->process_list) {
 
+    if (process->waiting_action) {
 
-  while((p=xbt_swag_extract(simix_global->process_list))) {
-    SIMIX_process_kill(p);
-  }
+      const char* action_description = "unknown";
+      switch (process->waiting_action->type) {
 
-  xbt_fifo_foreach(simix_global->host,i,h,smx_host_t) {
-    __SIMIX_host_destroy(h);
-  }
-  xbt_fifo_free(simix_global->host);
-  xbt_swag_free(simix_global->process_to_run);
-  xbt_swag_free(simix_global->process_list);
-  xbt_dict_free(&(simix_global->registered_functions));
-  simix_config_finalize();
-  free(simix_global);
-  surf_exit();
+       case SIMIX_ACTION_EXECUTE:
+         action_description = "execution";
+         break;
 
-  return ;
-}
+       case SIMIX_ACTION_PARALLEL_EXECUTE:
+         action_description = "parallel execution";
+         break;
 
+       case SIMIX_ACTION_COMMUNICATE:
+         action_description = "communication";
+         break;
 
-/** \ingroup msg_easier_life
- * \brief A clock (in second).
- */
-double SIMIX_get_clock(void)
-{
-  return surf_get_clock();
-}
+       case SIMIX_ACTION_SLEEP:
+         action_description = "sleeping";
+         break;
 
-double SIMIX_solve(xbt_fifo_t actions_done, xbt_fifo_t actions_failed) 
-{
+       case SIMIX_ACTION_SYNCHRO:
+         action_description = "synchronization";
+         break;
 
-       smx_process_t process = NULL;
-       int i;
-       double elapsed_time = 0.0;
-       int state_modifications = 0;
-
-
-       //surf_solve(); /* Takes traces into account. Returns 0.0  I don't know what to do with this*/
-
-       xbt_context_empty_trash();
-       if(xbt_swag_size(simix_global->process_to_run) && (elapsed_time>0)) {
-               DEBUG0("**************************************************");
-       }
-
-       while ((process = xbt_swag_extract(simix_global->process_to_run))) {
-               DEBUG2("Scheduling %s on %s",        
-                               process->name,
-                               process->simdata->host->name);
-               simix_global->current_process = process;
-               xbt_context_schedule(process->simdata->context);
-               /*       fflush(NULL); */
-               simix_global->current_process = NULL;
-       }
-
-       {
-               surf_action_t action = NULL;
-               surf_resource_t resource = NULL;
-               smx_action_t smx_action = NULL;
-
-               void *fun = NULL;
-               void *arg = NULL;
-
-               xbt_dynar_foreach(resource_list, i, resource) {
-                       if(xbt_swag_size(resource->common_public->states.failed_action_set) ||
-                                       xbt_swag_size(resource->common_public->states.done_action_set))
-                               state_modifications = 1;
-               }
-
-               if(!state_modifications) {
-                       DEBUG1("%f : Calling surf_solve",SIMIX_get_clock());
-                       elapsed_time = surf_solve();
-                       DEBUG1("Elapsed_time %f",elapsed_time);
-               }
-
-               while (surf_timer_resource->extension_public->get(&fun,(void*)&arg)) {
-                       DEBUG2("got %p %p", fun, arg);
-                       if(fun==SIMIX_process_create_with_arguments) {
-                               process_arg_t args = arg;
-                               DEBUG2("Launching %s on %s", args->name, args->host->name);
-                               process = SIMIX_process_create_with_arguments(args->name, args->code, 
-                                               args->data, args->host,
-                                               args->argc,args->argv);
-                               if(args->kill_time > SIMIX_get_clock()) {
-                                       surf_timer_resource->extension_public->set(args->kill_time, 
-                                                       (void*) &SIMIX_process_kill,
-                                                       (void*) process);
-                               }
-                               xbt_free(args);
-                       }
-                       if(fun==SIMIX_process_kill) {
-                               process = arg;
-                               DEBUG2("Killing %s on %s", process->name, 
-                                               process->simdata->host->name);
-                               SIMIX_process_kill(process);
-                       }
-               }
-
-               /* Wake up all process waiting for the action finish */
-               xbt_dynar_foreach(resource_list, i, resource) {
-                       while ((action = xbt_swag_extract(resource->common_public->states.failed_action_set))) {
-                               smx_action = action->data;
-                               if (smx_action) {
-                                       xbt_fifo_unshift(actions_failed,smx_action);
-                               }
-                       }
-                       while ((action =xbt_swag_extract(resource->common_public->states.done_action_set))) {
-                               smx_action = action->data;
-                               if (smx_action) {
-                                       xbt_fifo_unshift(actions_done,smx_action);
-                               }
-                       }
-               }
-       }
-
-       if (elapsed_time == -1) {
-               if (xbt_swag_size(simix_global->process_list) == 0) {
-                       INFO0("Congratulations ! Simulation terminated : all processes are over");
-               } else {
-                       INFO0("Oops ! Deadlock or code not perfectly clean.");
-                       __SIMIX_display_process_status();
-                       if(XBT_LOG_ISENABLED(simix, xbt_log_priority_debug) ||
-                                       XBT_LOG_ISENABLED(simix_kernel, xbt_log_priority_debug)) {
-                               DEBUG0("Aborting!");
-                               xbt_abort();
-                       }
-                       INFO0("Return a Warning.");
-               }
-       }
-       return elapsed_time;
+       case SIMIX_ACTION_IO:
+         action_description = "I/O";
+         break;
+      }
+      XBT_INFO("Waiting for %s action %p to finish", action_description, process->waiting_action);
+    }
+  }
 }
 
+static void* SIMIX_action_mallocator_new_f(void) {
+  smx_action_t action = xbt_new(s_smx_action_t, 1);
+  action->request_list = xbt_fifo_new();
+  return action;
+}
 
-void SIMIX_timer_set (double date, void *function, void *arg)
-{
-       surf_timer_resource->extension_public->set(date, function, arg);
+static void SIMIX_action_mallocator_free_f(void* action) {
+  xbt_fifo_free(((smx_action_t) action)->request_list);
+  xbt_free(action);
 }
 
-int SIMIX_timer_get(void **function, void **arg)
-{
-       return surf_timer_resource->extension_public->get(function, arg);
+static void SIMIX_action_mallocator_reset_f(void* action) {
+
+  // we also recycle the request list
+  xbt_fifo_t fifo = ((smx_action_t) action)->request_list;
+  xbt_fifo_reset(fifo);
+  memset(action, 0, sizeof(s_smx_action_t));
+  ((smx_action_t) action)->request_list = fifo;
 }