Logo AND Algorithmique Numérique Distribuée

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