src/simix/smx_host.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 "smx_private.h"
   8 #include "xbt/sysdep.h"
   9 #include "xbt/log.h"
  10 #include "xbt/dict.h"
  11 #include "mc/mc.h"
  12
  13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(simix_host, simix,
  14                                 "Logging specific to SIMIX (hosts)");
  15
  16
  17 static void SIMIX_execution_finish(smx_action_t action);
  18
  19 /**
  20  * \brief Internal function to create a SIMIX host.
  21  * \param name name of the host to create
  22  * \param workstation the SURF workstation to encapsulate
  23  * \param data some user data (may be NULL)
  24  */
  25 smx_host_t SIMIX_host_create(const char *name,
  26                                void *workstation, void *data)
  27 {
  28   smx_host_t smx_host = xbt_new0(s_smx_host_t, 1);
  29   s_smx_process_t proc;
  30
  31   /* Host structure */
  32   smx_host->name = xbt_strdup(name);
  33   smx_host->data = data;
  34   smx_host->host = workstation;
  35   smx_host->process_list =
  36       xbt_swag_new(xbt_swag_offset(proc, host_proc_hookup));
  37
  38   /* Update global variables */
  39   xbt_lib_set(host_lib,smx_host->name,SIMIX_HOST_LEVEL,smx_host);
  40
  41   return smx_host;
  42 }
  43
  44 /**
  45  * \brief Internal function to destroy a SIMIX host.
  46  *
  47  * \param h the host to destroy (a smx_host_t)
  48  */
  49 void SIMIX_host_destroy(void *h)
  50 {
  51   smx_host_t host = (smx_host_t) h;
  52
  53   xbt_assert((host != NULL), "Invalid parameters");
  54
  55   /* Clean Simulator data */
  56   if (xbt_swag_size(host->process_list) != 0) {
  57     char *msg =
  58         bprintf("Shutting down host %s, but it's not empty:", host->name);
  59     char *tmp;
  60     smx_process_t process = NULL;
  61
  62     xbt_swag_foreach(process, host->process_list) {
  63       tmp = bprintf("%s\n\t%s", msg, process->name);
  64       free(msg);
  65       msg = tmp;
  66     }
  67     SIMIX_display_process_status();
  68     THROWF(arg_error, 0, "%s", msg);
  69   }
  70
  71   xbt_swag_free(host->process_list);
  72
  73   /* Clean host structure */
  74   free(host->name);
  75   free(host);
  76
  77   return;
  78 }
  79
  80 /**
  81  * \brief Returns a dict of all hosts.
  82  *
  83  * \return List of all hosts (as a #xbt_dict_t)
  84  */
  85 xbt_dict_t SIMIX_host_get_dict(void)
  86 {
  87   xbt_dict_t host_dict = xbt_dict_new_homogeneous(NULL);
  88   xbt_lib_cursor_t cursor = NULL;
  89   char *name = NULL;
  90   void **host = NULL;
  91
  92   xbt_lib_foreach(host_lib, cursor, name, host){
  93     if(host[SIMIX_HOST_LEVEL])
  94             xbt_dict_set(host_dict,name,host[SIMIX_HOST_LEVEL], NULL);
  95   }
  96   return host_dict;
  97 }
  98
  99 smx_host_t SIMIX_host_get_by_name(const char *name)
 100 {
 101   xbt_assert(((simix_global != NULL)
 102                && (host_lib != NULL)),
 103               "Environment not set yet");
 104
 105   return xbt_lib_get_or_null(host_lib, name, SIMIX_HOST_LEVEL);
 106 }
 107
 108 smx_host_t SIMIX_host_self(void)
 109 {
 110   smx_process_t process = SIMIX_process_self();
 111   return (process == NULL) ? NULL : SIMIX_process_get_host(process);
 112 }
 113
 114 /* needs to be public and without simcall because it is called
 115    by exceptions and logging events */
 116 const char* SIMIX_host_self_get_name(void)
 117 {
 118   smx_host_t host = SIMIX_host_self();
 119   if (host == NULL || SIMIX_process_self() == simix_global->maestro_process)
 120     return "";
 121
 122   return SIMIX_host_get_name(host);
 123 }
 124
 125 const char* SIMIX_host_get_name(smx_host_t host)
 126 {
 127   xbt_assert((host != NULL), "Invalid parameters");
 128
 129   return host->name;
 130 }
 131
 132 xbt_dict_t SIMIX_host_get_properties(smx_host_t host)
 133 {
 134   xbt_assert((host != NULL), "Invalid parameters (simix host is NULL)");
 135
 136   return surf_workstation_model->extension.workstation.get_properties(host->host);
 137 }
 138
 139 double SIMIX_host_get_speed(smx_host_t host)
 140 {
 141   xbt_assert((host != NULL), "Invalid parameters (simix host is NULL)");
 142
 143   return surf_workstation_model->extension.workstation.
 144       get_speed(host->host, 1.0);
 145 }
 146
 147 double SIMIX_host_get_available_speed(smx_host_t host)
 148 {
 149   xbt_assert((host != NULL), "Invalid parameters (simix host is NULL)");
 150
 151   return surf_workstation_model->extension.workstation.
 152       get_available_speed(host->host);
 153 }
 154
 155 int SIMIX_host_get_state(smx_host_t host)
 156 {
 157   xbt_assert((host != NULL), "Invalid parameters (simix host is NULL)");
 158
 159   return surf_workstation_model->extension.workstation.
 160       get_state(host->host);
 161 }
 162
 163 void* SIMIX_host_self_get_data(void)
 164 {
 165   return SIMIX_host_get_data(SIMIX_host_self());
 166 }
 167
 168 void SIMIX_host_self_set_data(void *data)
 169 {
 170   SIMIX_host_set_data(SIMIX_host_self(), data);
 171 }
 172
 173 void* SIMIX_host_get_data(smx_host_t host)
 174 {
 175   xbt_assert((host != NULL), "Invalid parameters (simix host is NULL)");
 176
 177   return host->data;
 178 }
 179
 180 void SIMIX_host_set_data(smx_host_t host, void *data)
 181 {
 182   xbt_assert((host != NULL), "Invalid parameters");
 183   xbt_assert((host->data == NULL), "Data already set");
 184
 185   host->data = data;
 186 }
 187
 188 smx_action_t SIMIX_host_execute(const char *name, smx_host_t host,
 189                                 double computation_amount,
 190                                 double priority)
 191 {
 192   /* alloc structures and initialize */
 193   smx_action_t action = xbt_mallocator_get(simix_global->action_mallocator);
 194   action->type = SIMIX_ACTION_EXECUTE;
 195   action->name = xbt_strdup(name);
 196   action->state = SIMIX_RUNNING;
 197   action->execution.host = host;
 198
 199 #ifdef HAVE_TRACING
 200   action->category = NULL;
 201 #endif
 202
 203   /* set surf's action */
 204   if (!MC_IS_ENABLED) {
 205     action->execution.surf_exec =
 206       surf_workstation_model->extension.workstation.execute(host->host,
 207     computation_amount);
 208     surf_workstation_model->action_data_set(action->execution.surf_exec, action);
 209     surf_workstation_model->set_priority(action->execution.surf_exec, priority);
 210   }
 211
 212   XBT_DEBUG("Create execute action %p", action);
 213
 214   return action;
 215 }
 216
 217 smx_action_t SIMIX_host_parallel_execute( const char *name,
 218     int host_nb, smx_host_t *host_list,
 219     double *computation_amount, double *communication_amount,
 220     double amount, double rate)
 221 {
 222   void **workstation_list = NULL;
 223   int i;
 224
 225   /* alloc structures and initialize */
 226   smx_action_t action = xbt_mallocator_get(simix_global->action_mallocator);
 227   action->type = SIMIX_ACTION_PARALLEL_EXECUTE;
 228   action->name = xbt_strdup(name);
 229   action->state = SIMIX_RUNNING;
 230   action->execution.host = NULL; /* FIXME: do we need the list of hosts? */
 231
 232 #ifdef HAVE_TRACING
 233   action->category = NULL;
 234 #endif
 235
 236   /* set surf's action */
 237   workstation_list = xbt_new0(void *, host_nb);
 238   for (i = 0; i < host_nb; i++)
 239     workstation_list[i] = host_list[i]->host;
 240
 241   /* set surf's action */
 242   if (!MC_IS_ENABLED) {
 243     action->execution.surf_exec =
 244       surf_workstation_model->extension.workstation.
 245       execute_parallel_task(host_nb, workstation_list, computation_amount,
 246                       communication_amount, rate);
 247
 248     surf_workstation_model->action_data_set(action->execution.surf_exec, action);
 249   }
 250   XBT_DEBUG("Create parallel execute action %p", action);
 251
 252   return action;
 253 }
 254
 255 void SIMIX_host_execution_destroy(smx_action_t action)
 256 {
 257   int destroyed=0;
 258   XBT_DEBUG("Destroy action %p", action);
 259
 260
 261   if (action->execution.surf_exec) {
 262     destroyed = surf_workstation_model->action_unref(action->execution.surf_exec);
 263     action->execution.surf_exec = NULL;
 264   }
 265
 266   if (destroyed) {
 267     xbt_free(action->name);
 268     xbt_mallocator_release(simix_global->action_mallocator, action);
 269   }
 270 }
 271
 272 void SIMIX_host_execution_cancel(smx_action_t action)
 273 {
 274   XBT_DEBUG("Cancel action %p", action);
 275
 276   if (action->execution.surf_exec)
 277     surf_workstation_model->action_cancel(action->execution.surf_exec);
 278 }
 279
 280 double SIMIX_host_execution_get_remains(smx_action_t action)
 281 {
 282   double result = 0.0;
 283
 284   if (action->state == SIMIX_RUNNING)
 285     result = surf_workstation_model->get_remains(action->execution.surf_exec);
 286
 287   return result;
 288 }
 289
 290 e_smx_state_t SIMIX_host_execution_get_state(smx_action_t action)
 291 {
 292   return action->state;
 293 }
 294
 295 void SIMIX_host_execution_set_priority(smx_action_t action, double priority)
 296 {
 297   if(action->execution.surf_exec)
 298     surf_workstation_model->set_priority(action->execution.surf_exec, priority);
 299 }
 300
 301 void SIMIX_pre_host_execution_wait(smx_simcall_t simcall)
 302 {
 303   smx_action_t action = simcall->host_execution_wait.execution;
 304
 305   XBT_DEBUG("Wait for execution of action %p, state %d", action, (int)action->state);
 306
 307   /* Associate this simcall to the action */
 308   xbt_fifo_push(action->simcalls, simcall);
 309   simcall->issuer->waiting_action = action;
 310
 311   /* set surf's action */
 312   if (MC_IS_ENABLED) {
 313     action->state = SIMIX_DONE;
 314     SIMIX_execution_finish(action);
 315     return;
 316   }
 317
 318   /* If the action is already finished then perform the error handling */
 319   if (action->state != SIMIX_RUNNING)
 320     SIMIX_execution_finish(action);
 321 }
 322
 323 void SIMIX_host_execution_suspend(smx_action_t action)
 324 {
 325   if(action->execution.surf_exec)
 326     surf_workstation_model->suspend(action->execution.surf_exec);
 327 }
 328
 329 void SIMIX_host_execution_resume(smx_action_t action)
 330 {
 331   if(action->execution.surf_exec)
 332     surf_workstation_model->resume(action->execution.surf_exec);
 333 }
 334
 335 void SIMIX_execution_finish(smx_action_t action)
 336 {
 337   xbt_fifo_item_t item;
 338   smx_simcall_t simcall;
 339
 340   xbt_fifo_foreach(action->simcalls, item, simcall, smx_simcall_t) {
 341
 342     switch (action->state) {
 343
 344       case SIMIX_DONE:
 345         /* do nothing, action done */
 346   XBT_DEBUG("SIMIX_execution_finished: execution successful");
 347         break;
 348
 349       case SIMIX_FAILED:
 350         XBT_DEBUG("SIMIX_execution_finished: host '%s' failed", simcall->issuer->smx_host->name);
 351         simcall->issuer->context->iwannadie = 1;
 352         //SMX_EXCEPTION(simcall->issuer, host_error, 0, "Host failed");
 353         break;
 354
 355       case SIMIX_CANCELED:
 356         XBT_DEBUG("SIMIX_execution_finished: execution canceled");
 357         SMX_EXCEPTION(simcall->issuer, cancel_error, 0, "Canceled");
 358         break;
 359
 360       default:
 361         xbt_die("Internal error in SIMIX_execution_finish: unexpected action state %d",
 362             (int)action->state);
 363     }
 364     /* check if the host is down */
 365     if (surf_workstation_model->extension.
 366         workstation.get_state(simcall->issuer->smx_host->host) != SURF_RESOURCE_ON) {
 367       simcall->issuer->context->iwannadie = 1;
 368     }
 369
 370     simcall->issuer->waiting_action =    NULL;
 371     simcall->host_execution_wait.result = action->state;
 372     SIMIX_simcall_answer(simcall);
 373   }
 374
 375   /* We no longer need it */
 376   SIMIX_host_execution_destroy(action);
 377 }
 378
 379 void SIMIX_post_host_execute(smx_action_t action)
 380 {
 381   if (surf_workstation_model->extension.workstation.get_state(action->execution.host->host)==SURF_RESOURCE_OFF) {
 382     /* if the host running the action failed, notice it so that the asking process can be killed if it runs on that host itself */
 383     action->state = SIMIX_FAILED;
 384   } else if (surf_workstation_model->action_state_get(action->execution.surf_exec) == SURF_ACTION_FAILED) {
 385     /* If the host running the action didn't fail, then the action was canceled */
 386      action->state = SIMIX_CANCELED;
 387   } else {
 388      action->state = SIMIX_DONE;
 389   }
 390
 391   if (action->execution.surf_exec) {
 392     surf_workstation_model->action_unref(action->execution.surf_exec);
 393     action->execution.surf_exec = NULL;
 394   }
 395
 396   /* If there are simcalls associated with the action, then answer them */
 397   if (xbt_fifo_size(action->simcalls)) {
 398     SIMIX_execution_finish(action);
 399   }
 400 }
 401
 402
 403 #ifdef HAVE_TRACING
 404 void SIMIX_set_category(smx_action_t action, const char *category)
 405 {
 406   if (action->state != SIMIX_RUNNING) return;
 407   if (action->type == SIMIX_ACTION_EXECUTE){
 408     surf_workstation_model->set_category(action->execution.surf_exec, category);
 409   }else if (action->type == SIMIX_ACTION_COMMUNICATE){
 410     surf_workstation_model->set_category(action->comm.surf_comm, category);
 411   }
 412 }
 413 #endif
 414