Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Adding a function to set the max_duration of an action.
[simgrid.git] / src / surf / network.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 "network_private.h"
9
10 #define SG_TCP_CTE_GAMMA 20000.0
11
12 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(network, surf,
13                                 "Logging specific to the SURF network module");
14
15 surf_network_resource_t surf_network_resource = NULL;
16
17 static xbt_dict_t network_link_set = NULL;
18 xbt_dict_t network_card_set = NULL;
19
20 static int card_number = 0;
21 static network_link_CM02_t **routing_table = NULL;
22 static int *routing_table_size = NULL;
23
24 #define ROUTE(i,j) routing_table[(i)+(j)*card_number]
25 #define ROUTE_SIZE(i,j) routing_table_size[(i)+(j)*card_number]
26
27 static void create_routing_table(void)
28 {
29   routing_table = xbt_new0(network_link_CM02_t *, card_number * card_number);
30   routing_table_size = xbt_new0(int, card_number * card_number);
31 }
32
33 static void network_link_free(void *nw_link)
34 {
35   free(((network_link_CM02_t)nw_link)->name);
36   free(nw_link);
37 }
38
39 static network_link_CM02_t network_link_new(char *name,
40                                        double bw_initial,
41                                        tmgr_trace_t bw_trace,
42                                        double lat_initial,
43                                        tmgr_trace_t lat_trace,
44                                        e_surf_network_link_state_t
45                                        state_initial,
46                                        tmgr_trace_t state_trace)
47 {
48   network_link_CM02_t nw_link = xbt_new0(s_network_link_CM02_t, 1);
49
50
51   nw_link->resource = (surf_resource_t) surf_network_resource;
52   nw_link->name = name;
53   nw_link->bw_current = bw_initial;
54   if (bw_trace)
55     nw_link->bw_event =
56         tmgr_history_add_trace(history, bw_trace, 0.0, 0, nw_link);
57   nw_link->lat_current = lat_initial;
58   if (lat_trace)
59     nw_link->lat_event =
60         tmgr_history_add_trace(history, lat_trace, 0.0, 0, nw_link);
61   nw_link->state_current = state_initial;
62   if (state_trace)
63     nw_link->state_event =
64         tmgr_history_add_trace(history, state_trace, 0.0, 0, nw_link);
65
66   nw_link->constraint =
67       lmm_constraint_new(maxmin_system, nw_link, nw_link->bw_current);
68
69   xbt_dict_set(network_link_set, name, nw_link, network_link_free);
70
71   return nw_link;
72 }
73
74 static void network_card_free(void *nw_card)
75 {
76   free(((network_card_CM02_t)nw_card)->name);
77   free(nw_card);
78 }
79
80 static int network_card_new(const char *card_name)
81 {
82   network_card_CM02_t card = NULL;
83
84   xbt_dict_get(network_card_set, card_name, (void *) &card);
85
86   if (!card) {
87     card = xbt_new0(s_network_card_CM02_t, 1);
88     card->name = xbt_strdup(card_name);
89     card->id = card_number++;
90     xbt_dict_set(network_card_set, card_name, card, network_card_free);
91   }
92   return card->id;
93 }
94
95 static void route_new(int src_id, int dst_id, char **links, int nb_link)
96 {
97   network_link_CM02_t *link_list = NULL;
98   int i;
99
100   ROUTE_SIZE(src_id, dst_id) = nb_link;
101   link_list = (ROUTE(src_id, dst_id) = xbt_new0(network_link_CM02_t, nb_link));
102   for (i = 0; i < nb_link; i++) {
103     xbt_dict_get(network_link_set, links[i], (void *) &(link_list[i]));
104     free(links[i]);
105   }
106   free(links);
107 }
108
109 static void parse_network_link(void)
110 {
111   char *name;
112   double bw_initial;
113   tmgr_trace_t bw_trace;
114   double lat_initial;
115   tmgr_trace_t lat_trace;
116   e_surf_network_link_state_t state_initial = SURF_NETWORK_LINK_ON;
117   tmgr_trace_t state_trace;
118
119   name = xbt_strdup(A_network_link_name);
120   surf_parse_get_double(&bw_initial,A_network_link_bandwidth);
121   surf_parse_get_trace(&bw_trace, A_network_link_bandwidth_file);
122   surf_parse_get_double(&lat_initial,A_network_link_latency);
123   surf_parse_get_trace(&lat_trace, A_network_link_latency_file);
124
125   xbt_assert0((A_network_link_state==A_network_link_state_ON)||
126               (A_network_link_state==A_network_link_state_OFF),
127               "Invalid state")
128   if (A_network_link_state==A_network_link_state_ON) 
129     state_initial = SURF_NETWORK_LINK_ON;
130   if (A_network_link_state==A_network_link_state_OFF) 
131     state_initial = SURF_NETWORK_LINK_OFF;
132   surf_parse_get_trace(&state_trace,A_network_link_state_file);
133
134   network_link_new(name, bw_initial, bw_trace,
135                    lat_initial, lat_trace, state_initial, state_trace);
136 }
137
138 static int nb_link = 0;
139 static char **link_name = NULL;
140 static int src_id = -1;
141 static int dst_id = -1;
142
143 static void parse_route_set_endpoints(void)
144 {
145   src_id = network_card_new(A_route_src);
146   dst_id = network_card_new(A_route_dst);
147   nb_link = 0;
148   link_name = NULL;
149 }
150
151 static void parse_route_elem(void)
152 {
153   nb_link++;
154   link_name = xbt_realloc(link_name, (nb_link) * sizeof(char *));
155   link_name[(nb_link) - 1] = xbt_strdup(A_route_element_name);
156 }
157
158 static void parse_route_set_route(void)
159 {
160   route_new(src_id, dst_id, link_name, nb_link);
161 }
162
163 static void parse_file(const char *file)
164 {
165   /* Figuring out the network links */
166   surf_parse_reset_parser();
167   ETag_network_link_fun=parse_network_link;
168   surf_parse_open(file);
169   xbt_assert1((!surf_parse()),"Parse error in %s",file);
170   surf_parse_close();
171
172   /* Figuring out the network cards used */
173   surf_parse_reset_parser();
174   STag_route_fun=parse_route_set_endpoints;
175   surf_parse_open(file);
176   xbt_assert1((!surf_parse()),"Parse error in %s",file);
177   surf_parse_close();
178
179   create_routing_table();
180
181   /* Building the routes */
182   surf_parse_reset_parser();
183   STag_route_fun=parse_route_set_endpoints;
184   ETag_route_element_fun=parse_route_elem;
185   ETag_route_fun=parse_route_set_route;
186   surf_parse_open(file);
187   xbt_assert1((!surf_parse()),"Parse error in %s",file);
188   surf_parse_close();
189 }
190
191 static void *name_service(const char *name)
192 {
193   network_card_CM02_t card = NULL;
194
195   xbt_dict_get(network_card_set, name, (void *) &card);
196
197   return card;
198 }
199
200 static const char *get_resource_name(void *resource_id)
201 {
202   return ((network_card_CM02_t) resource_id)->name;
203 }
204
205 static int resource_used(void *resource_id)
206 {
207   return lmm_constraint_used(maxmin_system,
208                              ((network_link_CM02_t) resource_id)->constraint);
209 }
210
211 static int action_free(surf_action_t action)
212 {
213   action->using--;
214   if(!action->using) {
215     xbt_swag_remove(action, action->state_set);
216     if(((surf_action_network_CM02_t)action)->variable)
217       lmm_variable_free(maxmin_system, ((surf_action_network_CM02_t)action)->variable);
218     free(action);
219     return 1;
220   }
221   return 0;
222 }
223
224 static void action_use(surf_action_t action)
225 {
226   action->using++;
227 }
228
229 static void action_cancel(surf_action_t action)
230 {
231   return;
232 }
233
234 static void action_recycle(surf_action_t action)
235 {
236   return;
237 }
238
239 static void action_change_state(surf_action_t action,
240                                 e_surf_action_state_t state)
241 {
242 /*   if((state==SURF_ACTION_DONE) || (state==SURF_ACTION_FAILED)) */
243 /*     if(((surf_action_network_CM02_t)action)->variable) { */
244 /*       lmm_variable_disable(maxmin_system, ((surf_action_network_CM02_t)action)->variable); */
245 /*       ((surf_action_network_CM02_t)action)->variable = NULL; */
246 /*     } */
247
248   surf_action_change_state(action, state);
249   return;
250 }
251
252 static double share_resources(double now)
253 {
254   s_surf_action_network_CM02_t s_action;
255   surf_action_network_CM02_t action = NULL;
256   xbt_swag_t running_actions = surf_network_resource->common_public->states.running_action_set;
257   double min = generic_maxmin_share_resources(running_actions,
258                                               xbt_swag_offset(s_action, variable));
259
260   xbt_swag_foreach(action, running_actions) {
261     if(action->latency>0) {
262       if(min<0) min = action->latency;
263       else if (action->latency<min) min = action->latency;
264     }
265   }
266
267   return min;
268 }
269
270 static void update_actions_state(double now, double delta)
271 {
272   double deltap = 0.0;
273   surf_action_network_CM02_t action = NULL;
274   surf_action_network_CM02_t next_action = NULL;
275   xbt_swag_t running_actions =
276       surf_network_resource->common_public->states.running_action_set;
277   xbt_swag_t failed_actions =
278       surf_network_resource->common_public->states.failed_action_set;
279
280   xbt_swag_foreach_safe(action, next_action, running_actions) {
281     deltap = delta;
282     if (action->latency > 0) {
283       if (action->latency > deltap) {
284         surf_double_update(&(action->latency),deltap);
285         deltap = 0.0;
286       } else {
287         surf_double_update(&(deltap), action->latency);
288         action->latency = 0.0;
289       }
290       if ((action->latency == 0.0) && !(action->suspended)) 
291         lmm_update_variable_weight(maxmin_system, action->variable, 1.0);
292     }
293     surf_double_update(&(action->generic_action.remains),
294         lmm_variable_getvalue(action->variable) * deltap);
295     if (action->generic_action.max_duration != NO_MAX_DURATION)
296       surf_double_update(&(action->generic_action.max_duration), delta);
297
298     /*   if(action->generic_action.remains<.00001) action->generic_action.remains=0; */
299
300     if (action->generic_action.remains <= 0) {
301       action->generic_action.finish = surf_get_clock();
302       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
303     } else if ((action->generic_action.max_duration != NO_MAX_DURATION) &&
304                (action->generic_action.max_duration <= 0)) {
305       action->generic_action.finish = surf_get_clock();
306       action_change_state((surf_action_t) action, SURF_ACTION_DONE);
307     } else {                    /* Need to check that none of the resource has failed */
308       lmm_constraint_t cnst = NULL;
309       int i = 0;
310       network_link_CM02_t nw_link = NULL;
311
312       while ((cnst =
313               lmm_get_cnst_from_var(maxmin_system, action->variable,
314                                     i++))) {
315         nw_link = lmm_constraint_id(cnst);
316         if (nw_link->state_current == SURF_NETWORK_LINK_OFF) {
317           action->generic_action.finish = surf_get_clock();
318           action_change_state((surf_action_t) action, SURF_ACTION_FAILED);
319           break;
320         }
321       }
322     }
323   }
324
325   return;
326 }
327
328 static void update_resource_state(void *id,
329                                   tmgr_trace_event_t event_type,
330                                   double value)
331 {
332   network_link_CM02_t nw_link = id;
333   /*   printf("[" "%lg" "] Asking to update network card \"%s\" with value " */
334   /*     "%lg" " for event %p\n", surf_get_clock(), nw_link->name, */
335   /*     value, event_type); */
336
337   if (event_type == nw_link->bw_event) {
338     nw_link->bw_current = value;
339     lmm_update_constraint_bound(maxmin_system, nw_link->constraint,
340                                 nw_link->bw_current);
341   } else if (event_type == nw_link->lat_event) {
342     double delta = value - nw_link->lat_current;
343     lmm_variable_t var = NULL;
344     surf_action_network_CM02_t action = NULL;
345
346     nw_link->lat_current = value;
347     while (lmm_get_var_from_cnst(maxmin_system, nw_link->constraint, &var)) {
348       action = lmm_variable_id(var);
349       action->lat_current += delta;
350       if(action->rate<0)
351         lmm_update_variable_bound(maxmin_system, action->variable,
352                                   SG_TCP_CTE_GAMMA / action->lat_current);
353       else 
354         lmm_update_variable_bound(maxmin_system, action->variable,
355                                   min(action->rate,SG_TCP_CTE_GAMMA / action->lat_current));
356     }
357   } else if (event_type == nw_link->state_event) {
358     if (value > 0)
359       nw_link->state_current = SURF_NETWORK_LINK_ON;
360     else
361       nw_link->state_current = SURF_NETWORK_LINK_OFF;
362   } else {
363     CRITICAL0("Unknown event ! \n");
364     xbt_abort();
365   }
366
367   return;
368 }
369
370 static surf_action_t communicate(void *src, void *dst, double size, double rate)
371 {
372   surf_action_network_CM02_t action = NULL;
373   network_card_CM02_t card_src = src;
374   network_card_CM02_t card_dst = dst;
375   int route_size = ROUTE_SIZE(card_src->id, card_dst->id);
376   network_link_CM02_t *route = ROUTE(card_src->id, card_dst->id);
377   int i;
378
379   action = xbt_new0(s_surf_action_network_CM02_t, 1);
380
381   action->generic_action.using = 1;
382   action->generic_action.cost = size;
383   action->generic_action.remains = size;
384   action->generic_action.max_duration = NO_MAX_DURATION;
385   action->generic_action.start = -1.0;
386   action->generic_action.finish = -1.0;
387   action->generic_action.resource_type =
388       (surf_resource_t) surf_network_resource;
389   action->suspended = 0;  /* Should be useless because of the 
390                              calloc but it seems to help valgrind... */
391   action->generic_action.state_set =
392       surf_network_resource->common_public->states.running_action_set;
393
394   xbt_swag_insert(action, action->generic_action.state_set);
395   action->rate = rate;
396
397   action->latency = 0.0;
398   for (i = 0; i < route_size; i++)
399     action->latency += route[i]->lat_current;
400   action->lat_current = action->latency;
401
402   if(action->latency>0)
403     action->variable = lmm_variable_new(maxmin_system, action, 0.0, -1.0,
404                                         route_size);
405   else 
406     action->variable = lmm_variable_new(maxmin_system, action, 1.0, -1.0,
407                                         route_size);
408
409   if(action->rate<0)
410     lmm_update_variable_bound(maxmin_system, action->variable,
411                               SG_TCP_CTE_GAMMA / action->lat_current);
412   else 
413     lmm_update_variable_bound(maxmin_system, action->variable,
414                               min(action->rate,SG_TCP_CTE_GAMMA / action->lat_current));
415
416   if(route_size == 0) {
417     action_change_state((surf_action_t) action, SURF_ACTION_DONE);
418   }
419
420   for (i = 0; i < route_size; i++)
421     lmm_expand(maxmin_system, route[i]->constraint, action->variable, 1.0);
422
423   return (surf_action_t) action;
424 }
425
426 static void action_suspend(surf_action_t action)
427 {
428   ((surf_action_network_CM02_t) action)->suspended = 1;
429   lmm_update_variable_weight(maxmin_system,
430                              ((surf_action_network_CM02_t) action)->variable, 0.0);
431 }
432
433 static void action_resume(surf_action_t action)
434 {
435   lmm_update_variable_weight(maxmin_system,
436                              ((surf_action_network_CM02_t) action)->variable, 1.0);
437   ((surf_action_network_CM02_t) action)->suspended = 0;
438 }
439
440 static int action_is_suspended(surf_action_t action)
441 {
442   return ((surf_action_network_CM02_t) action)->suspended;
443 }
444
445 static void action_set_max_duration(surf_action_t action, double duration)
446 {
447   action->max_duration = duration;
448 }
449
450 static void finalize(void)
451 {
452   int i,j;
453
454   xbt_dict_free(&network_card_set);
455   xbt_dict_free(&network_link_set);
456   xbt_swag_free(surf_network_resource->common_public->states.
457                 ready_action_set);
458   xbt_swag_free(surf_network_resource->common_public->states.
459                 running_action_set);
460   xbt_swag_free(surf_network_resource->common_public->states.
461                 failed_action_set);
462   xbt_swag_free(surf_network_resource->common_public->states.
463                 done_action_set);
464   free(surf_network_resource->common_public);
465   free(surf_network_resource->common_private);
466   free(surf_network_resource->extension_public);
467
468   free(surf_network_resource);
469   surf_network_resource = NULL;
470
471   for (i = 0; i < card_number; i++) 
472     for (j = 0; j < card_number; j++) 
473       free(ROUTE(i,j));
474   free(routing_table);
475   routing_table = NULL;
476   free(routing_table_size);
477   routing_table_size = NULL;
478   card_number = 0;
479 }
480
481 static void surf_network_resource_init_internal(void)
482 {
483   s_surf_action_t action;
484
485   surf_network_resource = xbt_new0(s_surf_network_resource_t, 1);
486
487   surf_network_resource->common_private =
488       xbt_new0(s_surf_resource_private_t, 1);
489   surf_network_resource->common_public =
490       xbt_new0(s_surf_resource_public_t, 1);
491   surf_network_resource->extension_public =
492       xbt_new0(s_surf_network_resource_extension_public_t, 1);
493
494   surf_network_resource->common_public->states.ready_action_set =
495       xbt_swag_new(xbt_swag_offset(action, state_hookup));
496   surf_network_resource->common_public->states.running_action_set =
497       xbt_swag_new(xbt_swag_offset(action, state_hookup));
498   surf_network_resource->common_public->states.failed_action_set =
499       xbt_swag_new(xbt_swag_offset(action, state_hookup));
500   surf_network_resource->common_public->states.done_action_set =
501       xbt_swag_new(xbt_swag_offset(action, state_hookup));
502
503   surf_network_resource->common_public->name_service = name_service;
504   surf_network_resource->common_public->get_resource_name =
505       get_resource_name;
506   surf_network_resource->common_public->action_get_state =
507       surf_action_get_state;
508   surf_network_resource->common_public->action_free = action_free;
509   surf_network_resource->common_public->action_use = action_use;
510   surf_network_resource->common_public->action_cancel = action_cancel;
511   surf_network_resource->common_public->action_recycle = action_recycle;
512   surf_network_resource->common_public->action_change_state =
513       action_change_state;
514   surf_network_resource->common_public->action_set_data = surf_action_set_data;
515   surf_network_resource->common_public->name = "network";
516
517   surf_network_resource->common_private->resource_used = resource_used;
518   surf_network_resource->common_private->share_resources = share_resources;
519   surf_network_resource->common_private->update_actions_state =
520       update_actions_state;
521   surf_network_resource->common_private->update_resource_state =
522       update_resource_state;
523   surf_network_resource->common_private->finalize = finalize;
524
525   surf_network_resource->common_public->suspend = action_suspend;
526   surf_network_resource->common_public->resume = action_resume;
527   surf_network_resource->common_public->is_suspended = action_is_suspended;
528   surf_cpu_resource->common_public->set_max_duration = action_set_max_duration;
529
530   surf_network_resource->extension_public->communicate = communicate;
531
532   network_link_set = xbt_dict_new();
533   network_card_set = xbt_dict_new();
534
535   xbt_assert0(maxmin_system, "surf_init has to be called first!");
536 }
537
538 /***************************************************************************/
539 /* The nice TCP sharing model designed by Loris Marchal and Henri Casanova */
540 /***************************************************************************/
541 /* @TechReport{      rr-lip2002-40, */
542 /*   author        = {Henri Casanova and Loris Marchal}, */
543 /*   institution   = {LIP}, */
544 /*   title         = {A Network Model for Simulation of Grid Application}, */
545 /*   number        = {2002-40}, */
546 /*   month         = {oct}, */
547 /*   year          = {2002} */
548 /* } */
549 void surf_network_resource_init_CM02(const char *filename)
550 {
551   if (surf_network_resource)
552     return;
553   surf_network_resource_init_internal();
554   parse_file(filename);
555   xbt_dynar_push(resource_list, &surf_network_resource);
556 }