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