src/surf/workstation.c

   1 /*      $Id$     */
   2
   3 /* Copyright (c) 2004 Arnaud Legrand. All rights reserved.                  */
   4
   5 /* This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the license (GNU LGPL) which comes with this package. */
   7
   8 #include "xbt/ex.h"
   9 #include "xbt/dict.h"
  10 #include "workstation_private.h"
  11 #include "cpu_private.h"
  12 #include "network_private.h"
  13
  14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(workstation, surf,
  15                                 "Logging specific to the SURF workstation module");
  16
  17 surf_workstation_resource_t surf_workstation_resource = NULL;
  18
  19 xbt_dict_t workstation_set = NULL;
  20
  21 static workstation_CLM03_t workstation_new(const char *name,
  22                                      void *cpu, void *card)
  23 {
  24   workstation_CLM03_t workstation = xbt_new0(s_workstation_CLM03_t, 1);
  25
  26   workstation->resource = (surf_resource_t) surf_workstation_resource;
  27   workstation->name = xbt_strdup(name);
  28   workstation->cpu = cpu;
  29   workstation->network_card = card;
  30
  31   return workstation;
  32 }
  33
  34 static void workstation_free(void *workstation)
  35 {
  36   free(((workstation_CLM03_t)workstation)->name);
  37   free(workstation);
  38 }
  39
  40 static void create_workstations(void)
  41 {
  42   xbt_dict_cursor_t cursor = NULL;
  43   char *name = NULL;
  44   void *cpu = NULL;
  45   void *nw_card = NULL;
  46
  47   xbt_dict_foreach(cpu_set, cursor, name, cpu) {
  48     nw_card = xbt_dict_get_or_null(network_card_set, name);
  49     xbt_assert1(nw_card,
  50                 "No corresponding card found for %s",name);
  51
  52     xbt_dict_set(workstation_set, name,
  53                  workstation_new(name, cpu, nw_card), workstation_free);
  54   }
  55 }
  56
  57 static void *name_service(const char *name)
  58 {
  59   return xbt_dict_get_or_null(workstation_set, name);
  60 }
  61
  62 static const char *get_resource_name(void *resource_id)
  63 {
  64   return ((workstation_CLM03_t) resource_id)->name;
  65 }
  66
  67 static int resource_used(void *resource_id)
  68 {
  69   xbt_assert0(0,
  70               "Workstation is a virtual resource. I should not be there!");
  71   return 0;
  72 }
  73
  74 static int parallel_action_free(surf_action_t action)
  75 {
  76   action->using--;
  77   if(!action->using) {
  78     xbt_swag_remove(action, action->state_set);
  79     if(((surf_action_parallel_task_CSL05_t)action)->variable)
  80       lmm_variable_free(maxmin_system, ((surf_action_parallel_task_CSL05_t)action)->variable);
  81     free(action);
  82     return 1;
  83   }
  84   return 0;
  85 }
  86
  87 static void parallel_action_use(surf_action_t action)
  88 {
  89   action->using++;
  90 }
  91
  92 static int action_free(surf_action_t action)
  93 {
  94   if(action->resource_type==(surf_resource_t)surf_network_resource)
  95     return surf_network_resource->common_public->action_free(action);
  96   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
  97     return surf_cpu_resource->common_public->action_free(action);
  98   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
  99     return parallel_action_free(action);
 100   else DIE_IMPOSSIBLE;
 101   return 0;
 102 }
 103
 104 static void action_use(surf_action_t action)
 105 {
 106   if(action->resource_type==(surf_resource_t)surf_network_resource)
 107     surf_network_resource->common_public->action_use(action);
 108   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 109     surf_cpu_resource->common_public->action_use(action);
 110   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
 111     parallel_action_use(action);
 112   else DIE_IMPOSSIBLE;
 113   return;
 114 }
 115
 116 static void action_cancel(surf_action_t action)
 117 {
 118   if(action->resource_type==(surf_resource_t)surf_network_resource)
 119     surf_network_resource->common_public->action_cancel(action);
 120   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 121     surf_cpu_resource->common_public->action_cancel(action);
 122   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
 123     parallel_action_use(action);
 124   else DIE_IMPOSSIBLE;
 125   return;
 126 }
 127
 128 static void action_recycle(surf_action_t action)
 129 {
 130   DIE_IMPOSSIBLE;
 131   return;
 132 }
 133
 134 static void action_change_state(surf_action_t action,
 135                                 e_surf_action_state_t state)
 136 {
 137   if(action->resource_type==(surf_resource_t)surf_network_resource)
 138     surf_network_resource->common_public->action_change_state(action,state);
 139   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 140     surf_cpu_resource->common_public->action_change_state(action,state);
 141   else if(action->resource_type==(surf_resource_t)surf_workstation_resource)
 142     surf_action_change_state(action, state);
 143   else DIE_IMPOSSIBLE;
 144   return;
 145 }
 146
 147 static double share_resources(double now)
 148 {
 149   s_surf_action_parallel_task_CSL05_t action;
 150   return generic_maxmin_share_resources(surf_workstation_resource->common_public->
 151                                         states.running_action_set,
 152                                         xbt_swag_offset(action, variable));
 153 }
 154
 155 static void update_actions_state(double now, double delta)
 156 {
 157   surf_action_parallel_task_CSL05_t action = NULL;
 158   surf_action_parallel_task_CSL05_t next_action = NULL;
 159   xbt_swag_t running_actions =
 160       surf_workstation_resource->common_public->states.running_action_set;
 161   /* FIXME: unused
 162   xbt_swag_t failed_actions =
 163       surf_workstation_resource->common_public->states.failed_action_set;
 164   */
 165
 166   xbt_swag_foreach_safe(action, next_action, running_actions) {
 167     surf_double_update(&(action->generic_action.remains),
 168         lmm_variable_getvalue(action->variable) * delta);
 169     if (action->generic_action.max_duration != NO_MAX_DURATION)
 170       surf_double_update(&(action->generic_action.max_duration), delta);
 171     if ((action->generic_action.remains <= 0) &&
 172         (lmm_get_variable_weight(action->variable)>0)) {
 173       action->generic_action.finish = surf_get_clock();
 174       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
 175     } else if ((action->generic_action.max_duration != NO_MAX_DURATION) &&
 176                (action->generic_action.max_duration <= 0)) {
 177       action->generic_action.finish = surf_get_clock();
 178       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
 179     } else {                    /* Need to check that none of the resource has failed */
 180       lmm_constraint_t cnst = NULL;
 181       int i = 0;
 182       surf_resource_t resource = NULL;
 183
 184       while ((cnst =
 185               lmm_get_cnst_from_var(maxmin_system, action->variable,
 186                                     i++))) {
 187         resource = (surf_resource_t) lmm_constraint_id(cnst);
 188         if(resource== (surf_resource_t) surf_cpu_resource) {
 189           cpu_Cas01_t cpu = lmm_constraint_id(cnst);
 190           if (cpu->state_current == SURF_CPU_OFF) {
 191             action->generic_action.finish = surf_get_clock();
 192             action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
 193             break;
 194           }
 195         } else if (resource== (surf_resource_t) surf_network_resource) {
 196           network_link_CM02_t nw_link = lmm_constraint_id(cnst);
 197
 198           if (nw_link->state_current == SURF_NETWORK_LINK_OFF) {
 199             action->generic_action.finish = surf_get_clock();
 200             action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
 201             break;
 202           }
 203         }
 204       }
 205     }
 206   }
 207
 208   return;
 209 }
 210
 211 static void update_resource_state(void *id,
 212                                   tmgr_trace_event_t event_type,
 213                                   double value)
 214 {
 215   return;
 216 }
 217
 218 static surf_action_t execute(void *workstation, double size)
 219 {
 220   return surf_cpu_resource->extension_public->
 221       execute(((workstation_CLM03_t) workstation)->cpu, size);
 222 }
 223
 224 static surf_action_t action_sleep(void *workstation, double duration)
 225 {
 226   return surf_cpu_resource->extension_public->
 227       sleep(((workstation_CLM03_t) workstation)->cpu, duration);
 228 }
 229
 230 static void action_suspend(surf_action_t action)
 231 {
 232   if(action->resource_type==(surf_resource_t)surf_network_resource)
 233     surf_network_resource->common_public->suspend(action);
 234   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 235     surf_cpu_resource->common_public->suspend(action);
 236   else DIE_IMPOSSIBLE;
 237 }
 238
 239 static void action_resume(surf_action_t action)
 240 {
 241   if(action->resource_type==(surf_resource_t)surf_network_resource)
 242     surf_network_resource->common_public->resume(action);
 243   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 244     surf_cpu_resource->common_public->resume(action);
 245   else DIE_IMPOSSIBLE;
 246 }
 247
 248 static int action_is_suspended(surf_action_t action)
 249 {
 250   if(action->resource_type==(surf_resource_t)surf_network_resource)
 251     return surf_network_resource->common_public->is_suspended(action);
 252   if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 253     return surf_cpu_resource->common_public->is_suspended(action);
 254   DIE_IMPOSSIBLE;
 255 }
 256
 257 static void action_set_max_duration(surf_action_t action, double duration)
 258 {
 259   if(action->resource_type==(surf_resource_t)surf_network_resource)
 260     surf_network_resource->common_public->set_max_duration(action,duration);
 261   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 262     surf_cpu_resource->common_public->set_max_duration(action,duration);
 263   else  DIE_IMPOSSIBLE;
 264 }
 265
 266 static void action_set_priority(surf_action_t action, double priority)
 267 {
 268   if(action->resource_type==(surf_resource_t)surf_network_resource)
 269     surf_network_resource->common_public->set_priority(action,priority);
 270   else if(action->resource_type==(surf_resource_t)surf_cpu_resource)
 271     surf_cpu_resource->common_public->set_priority(action,priority);
 272   else  DIE_IMPOSSIBLE;
 273 }
 274
 275 static surf_action_t communicate(void *workstation_src,
 276                                  void *workstation_dst, double size,
 277                                  double rate)
 278 {
 279   return surf_network_resource->extension_public->
 280       communicate(((workstation_CLM03_t) workstation_src)->network_card,
 281                   ((workstation_CLM03_t) workstation_dst)->network_card, size, rate);
 282 }
 283
 284 static e_surf_cpu_state_t get_state(void *workstation)
 285 {
 286   return surf_cpu_resource->extension_public->
 287       get_state(((workstation_CLM03_t) workstation)->cpu);
 288 }
 289
 290 static double get_speed(void *workstation, double load)
 291 {
 292   return surf_cpu_resource->extension_public->
 293       get_speed(((workstation_CLM03_t) workstation)->cpu, load);
 294 }
 295
 296 static surf_action_t execute_parallel_task (int workstation_nb,
 297                                             void **workstation_list,
 298                                             double *computation_amount,
 299                                             double *communication_amount,
 300                                             double amount,
 301                                             double rate)
 302 {
 303   surf_action_parallel_task_CSL05_t action = NULL;
 304   int i, j, k;
 305   xbt_dict_t network_link_set = xbt_dict_new();
 306   xbt_dict_cursor_t cursor = NULL;
 307   char *name = NULL;
 308   int nb_link = 0;
 309   int nb_host = 0;
 310   network_link_CM02_t link;
 311
 312   /* Compute the number of affected resources... */
 313   for(i=0; i< workstation_nb; i++) {
 314     for(j=0; j< workstation_nb; j++) {
 315       network_card_CM02_t card_src = ((workstation_CLM03_t*)workstation_list)[i]->network_card;
 316       network_card_CM02_t card_dst = ((workstation_CLM03_t*)workstation_list)[j]->network_card;
 317       int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
 318       network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
 319
 320       if(communication_amount[i*workstation_nb+j]>0)
 321         for(k=0; k< route_size; k++) {
 322           xbt_dict_set(network_link_set, route[k]->name, route[k], NULL);
 323         }
 324     }
 325   }
 326
 327   xbt_dict_foreach(network_link_set, cursor, name, link) {
 328     nb_link++;
 329   }
 330
 331   xbt_dict_free(&network_link_set);
 332
 333   for (i = 0; i<workstation_nb; i++)
 334     if(computation_amount[i]>0) nb_host++;
 335
 336   if(nb_link + workstation_nb == 0)
 337     return NULL;
 338
 339   action = xbt_new0(s_surf_action_parallel_task_CSL05_t, 1);
 340   action->generic_action.using = 1;
 341   action->generic_action.cost = amount;
 342   action->generic_action.remains = amount;
 343   action->generic_action.max_duration = NO_MAX_DURATION;
 344   action->generic_action.start = -1.0;
 345   action->generic_action.finish = -1.0;
 346   action->generic_action.resource_type =
 347       (surf_resource_t) surf_workstation_resource;
 348   action->suspended = 0;  /* Should be useless because of the
 349                              calloc but it seems to help valgrind... */
 350   action->generic_action.state_set =
 351       surf_workstation_resource->common_public->states.running_action_set;
 352
 353   xbt_swag_insert(action, action->generic_action.state_set);
 354   action->rate = rate;
 355
 356   if(action->rate>0)
 357     action->variable = lmm_variable_new(maxmin_system, action, 1.0, -1.0,
 358                                         nb_host + nb_link);
 359   else
 360     action->variable = lmm_variable_new(maxmin_system, action, 1.0, action->rate,
 361                                         nb_host + nb_link);
 362
 363   for (i = 0; i<workstation_nb; i++)
 364     if(computation_amount[i]>0)
 365       lmm_expand(maxmin_system, ((cpu_Cas01_t) ((workstation_CLM03_t) workstation_list[i])->cpu)->constraint,
 366                  action->variable, computation_amount[i]);
 367
 368   for (i=0; i<workstation_nb; i++) {
 369     for(j=0; j< workstation_nb; j++) {
 370       network_card_CM02_t card_src = ((workstation_CLM03_t*)workstation_list)[i]->network_card;
 371       network_card_CM02_t card_dst = ((workstation_CLM03_t*)workstation_list)[j]->network_card;
 372       int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
 373       network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
 374
 375       for(k=0; k< route_size; k++) {
 376         if(communication_amount[i*workstation_nb+j]>0) {
 377           lmm_expand_add(maxmin_system, route[k]->constraint,
 378                        action->variable, communication_amount[i*workstation_nb+j]);
 379         }
 380       }
 381     }
 382   }
 383
 384   return (surf_action_t) action;
 385 }
 386
 387 static void finalize(void)
 388 {
 389   xbt_dict_free(&workstation_set);
 390   xbt_swag_free(surf_workstation_resource->common_public->states.ready_action_set);
 391   xbt_swag_free(surf_workstation_resource->common_public->states.
 392                 running_action_set);
 393   xbt_swag_free(surf_workstation_resource->common_public->states.
 394                 failed_action_set);
 395   xbt_swag_free(surf_workstation_resource->common_public->states.done_action_set);
 396
 397   free(surf_workstation_resource->common_public);
 398   free(surf_workstation_resource->common_private);
 399   free(surf_workstation_resource->extension_public);
 400
 401   free(surf_workstation_resource);
 402   surf_workstation_resource = NULL;
 403 }
 404
 405 static void surf_workstation_resource_init_internal(void)
 406 {
 407   s_surf_action_t action;
 408
 409   surf_workstation_resource = xbt_new0(s_surf_workstation_resource_t, 1);
 410
 411   surf_workstation_resource->common_private =
 412       xbt_new0(s_surf_resource_private_t, 1);
 413   surf_workstation_resource->common_public =
 414       xbt_new0(s_surf_resource_public_t, 1);
 415 /*   surf_workstation_resource->extension_private = xbt_new0(s_surf_workstation_resource_extension_private_t,1); */
 416   surf_workstation_resource->extension_public =
 417       xbt_new0(s_surf_workstation_resource_extension_public_t, 1);
 418
 419   surf_workstation_resource->common_public->states.ready_action_set =
 420       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 421   surf_workstation_resource->common_public->states.running_action_set =
 422       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 423   surf_workstation_resource->common_public->states.failed_action_set =
 424       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 425   surf_workstation_resource->common_public->states.done_action_set =
 426       xbt_swag_new(xbt_swag_offset(action, state_hookup));
 427
 428   surf_workstation_resource->common_public->name_service = name_service;
 429   surf_workstation_resource->common_public->get_resource_name =
 430       get_resource_name;
 431   surf_workstation_resource->common_public->action_get_state =
 432       surf_action_get_state;
 433   surf_workstation_resource->common_public->action_free = action_free;
 434   surf_workstation_resource->common_public->action_use = action_use;
 435   surf_workstation_resource->common_public->action_cancel = action_cancel;
 436   surf_workstation_resource->common_public->action_recycle =
 437       action_recycle;
 438   surf_workstation_resource->common_public->action_change_state =
 439       action_change_state;
 440   surf_workstation_resource->common_public->action_set_data = surf_action_set_data;
 441   surf_workstation_resource->common_public->name = "Workstation";
 442
 443   surf_workstation_resource->common_private->resource_used = resource_used;
 444   surf_workstation_resource->common_private->share_resources =
 445       share_resources;
 446   surf_workstation_resource->common_private->update_actions_state =
 447       update_actions_state;
 448   surf_workstation_resource->common_private->update_resource_state =
 449       update_resource_state;
 450   surf_workstation_resource->common_private->finalize = finalize;
 451
 452   surf_workstation_resource->common_public->suspend = action_suspend;
 453   surf_workstation_resource->common_public->resume = action_resume;
 454   surf_workstation_resource->common_public->is_suspended = action_is_suspended;
 455   surf_workstation_resource->common_public->set_max_duration = action_set_max_duration;
 456   surf_workstation_resource->common_public->set_priority = action_set_priority;
 457
 458   surf_workstation_resource->extension_public->execute = execute;
 459   surf_workstation_resource->extension_public->sleep = action_sleep;
 460   surf_workstation_resource->extension_public->get_state = get_state;
 461   surf_workstation_resource->extension_public->get_speed = get_speed;
 462   surf_workstation_resource->extension_public->communicate = communicate;
 463   surf_workstation_resource->extension_public->execute_parallel_task =
 464     execute_parallel_task;
 465
 466   workstation_set = xbt_dict_new();
 467
 468   xbt_assert0(maxmin_system, "surf_init has to be called first!");
 469 }
 470
 471 /********************************************************************/
 472 /* The model used in MSG and presented at CCGrid03                  */
 473 /********************************************************************/
 474 /* @InProceedings{Casanova.CLM_03, */
 475 /*   author = {Henri Casanova and Arnaud Legrand and Loris Marchal}, */
 476 /*   title = {Scheduling Distributed Applications: the SimGrid Simulation Framework}, */
 477 /*   booktitle = {Proceedings of the third IEEE International Symposium on Cluster Computing and the Grid (CCGrid'03)}, */
 478 /*   publisher = {"IEEE Computer Society Press"}, */
 479 /*   month = {may}, */
 480 /*   year = {2003} */
 481 /* } */
 482 void surf_workstation_resource_init_CLM03(const char *filename)
 483 {
 484 /*   int i ; */
 485 /*   surf_resource_t resource =  NULL; */
 486
 487   surf_workstation_resource_init_internal();
 488   surf_cpu_resource_init_Cas01(filename);
 489   surf_network_resource_init_CM02(filename);
 490   create_workstations();
 491   xbt_dynar_push(resource_list, &surf_workstation_resource);
 492 /*   xbt_dynar_foreach(resource_list, i, resource) { */
 493 /*     if(resource==surf_cpu_resource) { */
 494 /*       xbt_dynar_remove_at(resource_list, i, NULL); */
 495 /*       i--;  */
 496 /*       continue; */
 497 /*     } */
 498 /*     if(resource==surf_network_resource) { */
 499 /*       xbt_dynar_remove_at(resource_list, i, NULL); */
 500 /*       i--;  */
 501 /*       continue; */
 502 /*     } */
 503 /*   } */
 504 }