Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
90cfaccf27798a23aeb6bbad687078bc38257f69
[simgrid.git] / src / msg / msg_vm.c
1 /* Copyright (c) 2012. The SimGrid Team. All rights reserved.               */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 // QUESTIONS:
7 // 1./ check how and where a new VM is added to the list of the hosts
8 // 2./ Diff between SIMIX_Actions and SURF_Actions
9 // => SIMIX_actions : point synchro entre processus de niveau (theoretically speaking I do not have to create such SIMIX_ACTION
10 // =>  Surf_Actions
11
12 // TODO
13 //      MSG_TRACE can be revisited in order to use  the host
14 //      To implement a mixed model between workstation and vm_workstation,
15 //     please give a look at surf_model_private_t model_private at SURF Level and to the share resource functions
16 //     double (*share_resources) (double now);
17 //      For the action into the vm workstation model, we should be able to leverage the usual one (and if needed, look at
18 //              the workstation model.
19
20 #include "msg_private.h"
21 #include "xbt/sysdep.h"
22 #include "xbt/log.h"
23
24 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(msg_vm, msg,
25                                 "Cloud-oriented parts of the MSG API");
26
27
28 /* **** ******** GENERAL ********* **** */
29
30 /** \ingroup m_vm_management
31  * \brief Returns the value of a given vm property
32  *
33  * \param vm a vm
34  * \param name a property name
35  * \return value of a property (or NULL if property not set)
36  */
37
38 const char *MSG_vm_get_property_value(msg_vm_t vm, const char *name)
39 {
40   return MSG_host_get_property_value(vm, name);
41 }
42
43 /** \ingroup m_vm_management
44  * \brief Returns a xbt_dict_t consisting of the list of properties assigned to this host
45  *
46  * \param vm a vm
47  * \return a dict containing the properties
48  */
49 xbt_dict_t MSG_vm_get_properties(msg_vm_t vm)
50 {
51   xbt_assert((vm != NULL), "Invalid parameters (vm is NULL)");
52
53   return (simcall_host_get_properties(vm));
54 }
55
56 /** \ingroup m_host_management
57  * \brief Change the value of a given host property
58  *
59  * \param host a host
60  * \param name a property name
61  * \param value what to change the property to
62  * \param free_ctn the freeing function to use to kill the value on need
63  */
64 void MSG_vm_set_property_value(msg_vm_t vm, const char *name, void *value, void_f_pvoid_t free_ctn)
65 {
66   xbt_dict_set(MSG_host_get_properties(vm), name, value, free_ctn);
67 }
68
69 /** \ingroup msg_vm_management
70  * \brief Finds a msg_vm_t using its name.
71  *
72  * This is a name directory service
73  * \param name the name of a vm.
74  * \return the corresponding vm
75  *
76  * Please note that a VM is a specific host. Hence, you should give a different name
77  * for each VM/PM.
78  */
79
80 msg_vm_t MSG_vm_get_by_name(const char *name)
81 {
82         return MSG_get_host_by_name(name);
83 }
84
85 /** \ingroup m_vm_management
86  *
87  * \brief Return the name of the #msg_host_t.
88  *
89  * This functions checks whether \a host is a valid pointer or not and return
90    its name.
91  */
92 const char *MSG_vm_get_name(msg_vm_t vm)
93 {
94   return MSG_host_get_name(vm);
95 }
96
97
98 /* **** Check state of a VM **** */
99 static inline int __MSG_vm_is_state(msg_vm_t vm, e_surf_vm_state_t state)
100 {
101   return simcall_vm_get_state(vm) == state;
102 }
103
104 /** @brief Returns whether the given VM has just reated, not running.
105  *  @ingroup msg_VMs
106  */
107 int MSG_vm_is_created(msg_vm_t vm)
108 {
109   return __MSG_vm_is_state(vm, SURF_VM_STATE_CREATED);
110 }
111
112 /** @brief Returns whether the given VM is currently running
113  *  @ingroup msg_VMs
114  */
115 int MSG_vm_is_running(msg_vm_t vm)
116 {
117   return __MSG_vm_is_state(vm, SURF_VM_STATE_RUNNING);
118 }
119
120 /** @brief Returns whether the given VM is currently migrating
121  *  @ingroup msg_VMs
122  */
123 int MSG_vm_is_migrating(msg_vm_t vm)
124 {
125   return __MSG_vm_is_state(vm, SURF_VM_STATE_MIGRATING);
126 }
127
128 /** @brief Returns whether the given VM is currently suspended, not running.
129  *  @ingroup msg_VMs
130  */
131 int MSG_vm_is_suspended(msg_vm_t vm)
132 {
133   return __MSG_vm_is_state(vm, SURF_VM_STATE_SUSPENDED);
134 }
135
136 /** @brief Returns whether the given VM is being saved (FIXME: live saving or not?).
137  *  @ingroup msg_VMs
138  */
139 int MSG_vm_is_saving(msg_vm_t vm)
140 {
141   return __MSG_vm_is_state(vm, SURF_VM_STATE_SAVING);
142 }
143
144 /** @brief Returns whether the given VM has been saved, not running.
145  *  @ingroup msg_VMs
146  */
147 int MSG_vm_is_saved(msg_vm_t vm)
148 {
149   return __MSG_vm_is_state(vm, SURF_VM_STATE_SAVED);
150 }
151
152 /** @brief Returns whether the given VM is being restored, not running.
153  *  @ingroup msg_VMs
154  */
155 int MSG_vm_is_restoring(msg_vm_t vm)
156 {
157   return __MSG_vm_is_state(vm, SURF_VM_STATE_RESTORING);
158 }
159
160
161
162 /* ------------------------------------------------------------------------- */
163 /* ------------------------------------------------------------------------- */
164
165 /* **** ******** MSG vm actions ********* **** */
166
167 /** @brief Create a new VM with specified parameters.
168  *  @ingroup msg_VMs*
169  *
170  */
171 msg_vm_t MSG_vm_create(msg_host_t ind_pm, const char *name,
172                                              int ncpus, int ramsize, int net_cap, char *disk_path, int disksize)
173 {
174   msg_vm_t vm = MSG_vm_create_core(ind_pm, name);
175
176   {
177     s_ws_params_t params;
178     memset(&params, 0, sizeof(params));
179     params.ramsize = ramsize;
180     params.overcommit = 0;
181     simcall_host_set_params(vm, &params);
182   }
183
184   /* TODO: Limit net capability, take into account disk considerations. */
185
186   return vm;
187 }
188
189
190 /** @brief Create a new VM object. The VM is not yet started. The resource of the VM is allocated upon MSG_vm_start().
191  *  @ingroup msg_VMs*
192  *
193  * A VM is treated as a host. The name of the VM must be unique among all hosts.
194  */
195 msg_vm_t MSG_vm_create_core(msg_host_t ind_pm, const char *name)
196 {
197   /* make sure the VM of the same name does not exit */
198   {
199     void *ind_host_tmp = xbt_lib_get_elm_or_null(host_lib, name);
200     if (ind_host_tmp) {
201       XBT_ERROR("host %s already exits", name);
202       return NULL;
203     }
204   }
205
206   /* Note: ind_vm and vm_workstation point to the same elm object. */
207   msg_vm_t ind_vm = NULL;
208   void *ind_vm_workstation =  NULL;
209
210   /* Ask the SIMIX layer to create the surf vm resource */
211   ind_vm_workstation = simcall_vm_create(name, ind_pm);
212   ind_vm = (msg_vm_t) __MSG_host_create(ind_vm_workstation);
213
214   XBT_DEBUG("A new VM (%s) has been created", name);
215
216   #ifdef HAVE_TRACING
217   TRACE_msg_vm_create(name, ind_pm);
218   #endif
219
220   return ind_vm;
221 }
222
223 /** @brief Destroy a VM. Destroy the VM object from the simulation.
224  *  @ingroup msg_VMs
225  */
226 void MSG_vm_destroy(msg_vm_t vm)
227 {
228   /* First, terminate all processes on the VM if necessary */
229   if (MSG_vm_is_running(vm))
230       simcall_vm_shutdown(vm);
231
232   if (!MSG_vm_is_created(vm)) {
233     XBT_CRITICAL("shutdown the given VM before destroying it");
234     DIE_IMPOSSIBLE;
235   }
236
237   /* Then, destroy the VM object */
238   simcall_vm_destroy(vm);
239
240   __MSG_host_destroy(vm);
241
242   #ifdef HAVE_TRACING
243   TRACE_msg_vm_end(vm);
244   #endif
245 }
246
247
248 /** @brief Start a vm (i.e., boot the guest operating system)
249  *  @ingroup msg_VMs
250  *
251  *  If the VM cannot be started, an exception is generated.
252  *
253  */
254 void MSG_vm_start(msg_vm_t vm)
255 {
256   simcall_vm_start(vm);
257
258   #ifdef HAVE_TRACING
259   TRACE_msg_vm_start(vm);
260   #endif
261 }
262
263
264
265 /** @brief Immediately kills all processes within the given VM. Any memory that they allocated will be leaked.
266  *  @ingroup msg_VMs
267  *
268  * FIXME: No extra delay occurs. If you want to simulate this too, you want to
269  * use a #MSG_process_sleep() or something. I'm not quite sure.
270  */
271 void MSG_vm_shutdown(msg_vm_t vm)
272 {
273   /* msg_vm_t equals to msg_host_t */
274   simcall_vm_shutdown(vm);
275
276   // #ifdef HAVE_TRACING
277   // TRACE_msg_vm_(vm);
278   // #endif
279 }
280
281
282
283 /* We have two mailboxes. mbox is used to transfer migration data between
284  * source and destiantion PMs. mbox_ctl is used to detect the completion of a
285  * migration. The names of these mailboxes must not conflict with others. */
286 static inline char *get_mig_mbox_src_dst(const char *vm_name, const char *src_pm_name, const char *dst_pm_name)
287 {
288   return bprintf("__mbox_mig_src_dst:%s(%s-%s)", vm_name, src_pm_name, dst_pm_name);
289 }
290
291 static inline char *get_mig_mbox_ctl(const char *vm_name, const char *src_pm_name, const char *dst_pm_name)
292 {
293   return bprintf("__mbox_mig_ctl:%s(%s-%s)", vm_name, src_pm_name, dst_pm_name);
294 }
295
296 static inline char *get_mig_process_tx_name(const char *vm_name, const char *src_pm_name, const char *dst_pm_name)
297 {
298   return bprintf("__pr_mig_tx:%s(%s-%s)", vm_name, src_pm_name, dst_pm_name);
299 }
300
301 static inline char *get_mig_process_rx_name(const char *vm_name, const char *src_pm_name, const char *dst_pm_name)
302 {
303   return bprintf("__pr_mig_rx:%s(%s-%s)", vm_name, src_pm_name, dst_pm_name);
304 }
305
306 static inline char *get_mig_task_name(const char *vm_name, const char *src_pm_name, const char *dst_pm_name, int stage)
307 {
308   return bprintf("__task_mig_stage%d:%s(%s-%s)", stage, vm_name, src_pm_name, dst_pm_name);
309 }
310
311 static int migration_rx_fun(int argc, char *argv[])
312 {
313   const char *pr_name = MSG_process_get_name(MSG_process_self());
314   const char *host_name = MSG_host_get_name(MSG_host_self());
315
316   XBT_DEBUG("mig: rx_start");
317
318   xbt_assert(argc == 4);
319   const char *vm_name = argv[1];
320   const char *src_pm_name  = argv[2];
321   const char *dst_pm_name  = argv[3];
322   msg_vm_t vm = MSG_get_host_by_name(vm_name);
323   msg_vm_t dst_pm = MSG_get_host_by_name(dst_pm_name);
324
325   int need_exit = 0;
326
327   char *mbox = get_mig_mbox_src_dst(vm_name, src_pm_name, dst_pm_name);
328   char *mbox_ctl = get_mig_mbox_ctl(vm_name, src_pm_name, dst_pm_name);
329   char *finalize_task_name = get_mig_task_name(vm_name, src_pm_name, dst_pm_name, 3);
330
331   for (;;) {
332     msg_task_t task = NULL;
333     MSG_task_recv(&task, mbox);
334
335     if (strcmp(task->name, finalize_task_name) == 0)
336       need_exit = 1;
337
338     MSG_task_destroy(task);
339
340     if (need_exit)
341       break;
342   }
343
344
345   simcall_vm_migrate(vm, dst_pm);
346   simcall_vm_resume(vm);
347
348   {
349     char *task_name = get_mig_task_name(vm_name, src_pm_name, dst_pm_name, 4);
350
351     msg_task_t task = MSG_task_create(task_name, 0, 0, NULL);
352     msg_error_t ret = MSG_task_send(task, mbox_ctl);
353     xbt_assert(ret == MSG_OK);
354
355     xbt_free(task_name);
356   }
357
358
359   xbt_free(mbox);
360   xbt_free(mbox_ctl);
361   xbt_free(finalize_task_name);
362
363   XBT_DEBUG("mig: rx_done");
364
365   return 0;
366 }
367
368
369 typedef struct dirty_page {
370   double prev_clock;
371   double prev_remaining;
372   msg_task_t task;
373 } s_dirty_page, *dirty_page_t;
374
375
376 static void reset_dirty_pages(msg_vm_t vm)
377 {
378   msg_host_priv_t priv = msg_host_resource_priv(vm);
379
380   char *key = NULL;
381   xbt_dict_cursor_t cursor = NULL;
382   dirty_page_t dp = NULL;
383   xbt_dict_foreach(priv->dp_objs, cursor, key, dp) {
384     double remaining = MSG_task_get_remaining_computation(dp->task);
385     dp->prev_clock = MSG_get_clock();
386     dp->prev_remaining = remaining;
387
388     // XBT_INFO("%s@%s remaining %f", key, sg_host_name(vm), remaining);
389   }
390 }
391
392 static void start_dirty_page_tracking(msg_vm_t vm)
393 {
394   msg_host_priv_t priv = msg_host_resource_priv(vm);
395   priv->dp_enabled = 1;
396
397   reset_dirty_pages(vm);
398 }
399
400 static void stop_dirty_page_tracking(msg_vm_t vm)
401 {
402   msg_host_priv_t priv = msg_host_resource_priv(vm);
403   priv->dp_enabled = 0;
404 }
405
406 #if 0
407 /* It might be natural that we define dp_rate for each task. But, we will also
408  * have to care about how each task behavior affects the memory update behavior
409  * at the operating system level. It may not be easy to model it with a simple algorithm. */
410 double calc_updated_pages(char *key, msg_vm_t vm, dirty_page_t dp, double remaining, double clock)
411 {
412     double computed = dp->prev_remaining - remaining;
413     double duration = clock - dp->prev_clock;
414     double updated = dp->task->dp_rate * computed;
415
416     XBT_INFO("%s@%s: computated %f ops (remaining %f -> %f) in %f secs (%f -> %f)",
417         key, sg_host_name(vm), computed, dp->prev_remaining, remaining, duration, dp->prev_clock, clock);
418     XBT_INFO("%s@%s: updated %f bytes, %f Mbytes/s",
419         key, sg_host_name(vm), updated, updated / duration / 1000 / 1000);
420
421     return updated;
422 }
423 #endif
424
425 double get_computed(char *key, msg_vm_t vm, dirty_page_t dp, double remaining, double clock)
426 {
427   double computed = dp->prev_remaining - remaining;
428   double duration = clock - dp->prev_clock;
429
430   XBT_DEBUG("%s@%s: computated %f ops (remaining %f -> %f) in %f secs (%f -> %f)",
431       key, sg_host_name(vm), computed, dp->prev_remaining, remaining, duration, dp->prev_clock, clock);
432
433   return computed;
434 }
435
436 static double lookup_computed_flop_counts(msg_vm_t vm, int stage2_round_for_fancy_debug)
437 {
438   msg_host_priv_t priv = msg_host_resource_priv(vm);
439   double total = 0;
440
441   char *key = NULL;
442   xbt_dict_cursor_t cursor = NULL;
443   dirty_page_t dp = NULL;
444   xbt_dict_foreach(priv->dp_objs, cursor, key, dp) {
445     double remaining = MSG_task_get_remaining_computation(dp->task);
446     double clock = MSG_get_clock();
447
448     // total += calc_updated_pages(key, vm, dp, remaining, clock);
449     total += get_computed(key, vm, dp, remaining, clock);
450
451     dp->prev_remaining = remaining;
452     dp->prev_clock = clock;
453   }
454
455   total += priv->dp_updated_by_deleted_tasks;
456
457   XBT_INFO("mig-stage2.%d: computed %f flop_counts (including %f by deleted tasks)",
458       stage2_round_for_fancy_debug,
459       total, priv->dp_updated_by_deleted_tasks);
460
461
462
463   priv->dp_updated_by_deleted_tasks = 0;
464
465
466   return total;
467 }
468
469 // TODO Is this code redundant with the information provided by
470 // msg_process_t MSG_process_create(const char *name, xbt_main_func_t code, void *data, msg_host_t host)
471 void MSG_host_add_task(msg_host_t host, msg_task_t task)
472 {
473   msg_host_priv_t priv = msg_host_resource_priv(host);
474   double remaining = MSG_task_get_remaining_computation(task);
475   char *key = bprintf("%s-%lld", task->name, task->counter);
476
477   dirty_page_t dp = xbt_new0(s_dirty_page, 1);
478   dp->task = task;
479
480   /* It should be okay that we add a task onto a migrating VM. */
481   if (priv->dp_enabled) {
482     dp->prev_clock = MSG_get_clock();
483     dp->prev_remaining = remaining;
484   }
485
486   xbt_assert(xbt_dict_get_or_null(priv->dp_objs, key) == NULL);
487   xbt_dict_set(priv->dp_objs, key, dp, NULL);
488   XBT_DEBUG("add %s on %s (remaining %f, dp_enabled %d)", key, sg_host_name(host), remaining, priv->dp_enabled);
489
490   xbt_free(key);
491 }
492
493 void MSG_host_del_task(msg_host_t host, msg_task_t task)
494 {
495   msg_host_priv_t priv = msg_host_resource_priv(host);
496
497   char *key = bprintf("%s-%lld", task->name, task->counter);
498
499   dirty_page_t dp = xbt_dict_get_or_null(priv->dp_objs, key);
500   xbt_assert(dp->task == task);
501
502   /* If we are in the middle of dirty page tracking, we record how much
503    * computaion has been done until now, and keep the information for the
504    * lookup_() function that will called soon. */
505   if (priv->dp_enabled) {
506     double remaining = MSG_task_get_remaining_computation(task);
507     double clock = MSG_get_clock();
508     // double updated = calc_updated_pages(key, host, dp, remaining, clock);
509     double updated = get_computed(key, host, dp, remaining, clock);
510
511     priv->dp_updated_by_deleted_tasks += updated;
512   }
513
514   xbt_dict_remove(priv->dp_objs, key);
515   xbt_free(dp);
516
517   XBT_DEBUG("del %s on %s", key, sg_host_name(host));
518
519   xbt_free(key);
520 }
521
522
523 static void send_migration_data(const char *vm_name, const char *src_pm_name, const char *dst_pm_name,
524     double size, char *mbox, int stage, int stage2_round)
525 {
526   char *task_name = get_mig_task_name(vm_name, src_pm_name, dst_pm_name, stage);
527   msg_task_t task = MSG_task_create(task_name, 0, size, NULL);
528   msg_error_t ret = MSG_task_send(task, mbox);
529   xbt_assert(ret == MSG_OK);
530
531   if (stage == 2)
532     XBT_INFO("mig-stage%d.%d: sent %f", stage, stage2_round, size);
533   else
534     XBT_INFO("mig-stage%d: sent %f", stage, size);
535
536   xbt_free(task_name);
537 }
538
539
540 static int migration_tx_fun(int argc, char *argv[])
541 {
542   const char *pr_name = MSG_process_get_name(MSG_process_self());
543   const char *host_name = MSG_host_get_name(MSG_host_self());
544
545   XBT_DEBUG("mig: tx_start");
546
547   xbt_assert(argc == 4);
548   const char *vm_name = argv[1];
549   const char *src_pm_name  = argv[2];
550   const char *dst_pm_name  = argv[3];
551   msg_vm_t vm = MSG_get_host_by_name(vm_name);
552
553
554   s_ws_params_t params;
555   simcall_host_get_params(vm, &params);
556   const long ramsize        = params.ramsize;
557   const long devsize        = params.devsize;
558   const int skip_stage2     = params.skip_stage2;
559   const double max_downtime = params.max_downtime;
560   const double dp_rate      = params.dp_rate;
561   const double dp_cap       = params.dp_cap;
562   double remaining_size = ramsize + devsize;
563   double threshold = max_downtime * 125 * 1000 * 1000;
564
565
566   if (ramsize == 0)
567     XBT_WARN("migrate a VM, but ramsize is zero");
568
569   char *mbox = get_mig_mbox_src_dst(vm_name, src_pm_name, dst_pm_name);
570
571   XBT_INFO("mig-stage1: remaining_size %f", remaining_size);
572
573   /* Stage1: send all memory pages to the destination. */
574   start_dirty_page_tracking(vm);
575
576   send_migration_data(vm_name, src_pm_name, dst_pm_name, ramsize, mbox, 1, 0);
577
578   remaining_size -= ramsize;
579
580
581
582   /* Stage2: send update pages iteratively until the size of remaining states
583    * becomes smaller than the threshold value. */
584   if (skip_stage2)
585     goto stage3;
586   if (max_downtime == 0) {
587     XBT_WARN("no max_downtime parameter, skip stage2");
588     goto stage3;
589   }
590
591
592   int stage2_round = 0;
593   for (;;) {
594     // long updated_size = lookup_dirty_pages(vm);
595     double updated_size = lookup_computed_flop_counts(vm, stage2_round) * dp_rate;
596     if (updated_size > dp_cap) {
597       XBT_INFO("mig-stage2.%d: %f bytes updated, but cap it with the working set size %f",
598           stage2_round, updated_size, dp_cap);
599       updated_size = dp_cap;
600     }
601
602     remaining_size += updated_size;
603
604     XBT_INFO("mig-stage2.%d: remaining_size %f (%s threshold %f)", stage2_round,
605         remaining_size, (remaining_size < threshold) ? "<" : ">", threshold);
606
607     if (remaining_size < threshold)
608       break;
609
610     send_migration_data(vm_name, src_pm_name, dst_pm_name, updated_size, mbox, 2, stage2_round);
611
612     remaining_size -= updated_size;
613     stage2_round += 1;
614   }
615
616
617 stage3:
618   /* Stage3: stop the VM and copy the rest of states. */
619   XBT_INFO("mig-stage3: remaining_size %f", remaining_size);
620   simcall_vm_suspend(vm);
621   stop_dirty_page_tracking(vm);
622
623   send_migration_data(vm_name, src_pm_name, dst_pm_name, remaining_size, mbox, 3, 0);
624
625   xbt_free(mbox);
626
627   XBT_DEBUG("mig: tx_done");
628
629   return 0;
630 }
631
632
633
634 static void do_migration(msg_vm_t vm, msg_host_t src_pm, msg_host_t dst_pm)
635 {
636   char *mbox_ctl = get_mig_mbox_ctl(sg_host_name(vm), sg_host_name(src_pm), sg_host_name(dst_pm));
637
638   {
639     char *pr_name = get_mig_process_rx_name(sg_host_name(vm), sg_host_name(src_pm), sg_host_name(dst_pm));
640     int nargvs = 5;
641     char **argv = xbt_new(char *, nargvs);
642     argv[0] = xbt_strdup(pr_name);
643     argv[1] = xbt_strdup(sg_host_name(vm));
644     argv[2] = xbt_strdup(sg_host_name(src_pm));
645     argv[3] = xbt_strdup(sg_host_name(dst_pm));
646     argv[4] = NULL;
647
648     msg_process_t pr = MSG_process_create_with_arguments(pr_name, migration_rx_fun, NULL, dst_pm, nargvs - 1, argv);
649
650     xbt_free(pr_name);
651   }
652
653   {
654     char *pr_name = get_mig_process_tx_name(sg_host_name(vm), sg_host_name(src_pm), sg_host_name(dst_pm));
655     int nargvs = 5;
656     char **argv = xbt_new(char *, nargvs);
657     argv[0] = xbt_strdup(pr_name);
658     argv[1] = xbt_strdup(sg_host_name(vm));
659     argv[2] = xbt_strdup(sg_host_name(src_pm));
660     argv[3] = xbt_strdup(sg_host_name(dst_pm));
661     argv[4] = NULL;
662
663     msg_process_t pr = MSG_process_create_with_arguments(pr_name, migration_tx_fun, NULL, src_pm, nargvs - 1, argv);
664
665     xbt_free(pr_name);
666   }
667
668   /* wait until the migration have finished */
669   {
670     msg_task_t task = NULL;
671     msg_error_t ret = MSG_task_recv(&task, mbox_ctl);
672     xbt_assert(ret == MSG_OK);
673
674     char *expected_task_name = get_mig_task_name(sg_host_name(vm), sg_host_name(src_pm), sg_host_name(dst_pm), 4);
675     xbt_assert(strcmp(task->name, expected_task_name) == 0);
676     xbt_free(expected_task_name);
677   }
678
679   xbt_free(mbox_ctl);
680 }
681
682
683 /** @brief Migrate the VM to the given host.
684  *  @ingroup msg_VMs
685  *
686  * FIXME: No migration cost occurs. If you want to simulate this too, you want to use a
687  * MSG_task_send() before or after, depending on whether you want to do cold or hot
688  * migration.
689  */
690 void MSG_vm_migrate(msg_vm_t vm, msg_host_t new_pm)
691 {
692   /* some thoughts:
693    * - One approach is ...
694    *   We first create a new VM (i.e., destination VM) on the destination
695    *   physical host. The destination VM will receive the state of the source
696    *   VM over network. We will finally destroy the source VM.
697    *   - This behavior is similar to the way of migration in the real world.
698    *     Even before a migration is completed, we will see a destination VM,
699    *     consuming resources.
700    *   - We have to relocate all processes. The existing process migraion code
701    *     will work for this?
702    *   - The name of the VM is a somewhat unique ID in the code. It is tricky
703    *     for the destination VM?
704    *
705    * - Another one is ...
706    *   We update the information of the given VM to place it to the destination
707    *   physical host.
708    *
709    * The second one would be easier.
710    *   
711    */
712
713   msg_host_t old_pm = simcall_vm_get_pm(vm);
714
715   if (simcall_vm_get_state(vm) != SURF_VM_STATE_RUNNING)
716     THROWF(vm_error, 0, "VM(%s) is not running", sg_host_name(vm));
717
718   do_migration(vm, old_pm, new_pm);
719
720
721
722   XBT_DEBUG("VM(%s) moved from PM(%s) to PM(%s)", vm->key, old_pm->key, new_pm->key);
723
724   #ifdef HAVE_TRACING
725   TRACE_msg_vm_change_host(vm, old_pm, new_pm);
726   #endif
727 }
728
729
730 /** @brief Immediately suspend the execution of all processes within the given VM.
731  *  @ingroup msg_VMs
732  *
733  * This function stops the exection of the VM. All the processes on this VM
734  * will pause. The state of the VM is perserved. We can later resume it again.
735  *
736  * No suspension cost occurs.
737  */
738 void MSG_vm_suspend(msg_vm_t vm)
739 {
740   simcall_vm_suspend(vm);
741
742   XBT_DEBUG("vm_suspend done");
743
744   #ifdef HAVE_TRACING
745   TRACE_msg_vm_suspend(vm);
746   #endif
747 }
748
749
750 /** @brief Resume the execution of the VM. All processes on the VM run again.
751  *  @ingroup msg_VMs
752  *
753  * No resume cost occurs.
754  */
755 void MSG_vm_resume(msg_vm_t vm)
756 {
757   simcall_vm_resume(vm);
758
759   #ifdef HAVE_TRACING
760   TRACE_msg_vm_resume(vm);
761   #endif
762 }
763
764
765 /** @brief Immediately save the execution of all processes within the given VM.
766  *  @ingroup msg_VMs
767  *
768  * This function stops the exection of the VM. All the processes on this VM
769  * will pause. The state of the VM is perserved. We can later resume it again.
770  *
771  * FIXME: No suspension cost occurs. If you want to simulate this too, you want to
772  * use a \ref MSG_file_write() before or after, depending on the exact semantic
773  * of VM save to you.
774  */
775 void MSG_vm_save(msg_vm_t vm)
776 {
777   simcall_vm_save(vm);
778   #ifdef HAVE_TRACING
779   TRACE_msg_vm_save(vm);
780   #endif
781 }
782
783 /** @brief Restore the execution of the VM. All processes on the VM run again.
784  *  @ingroup msg_VMs
785  *
786  * FIXME: No restore cost occurs. If you want to simulate this too, you want to
787  * use a \ref MSG_file_read() before or after, depending on the exact semantic
788  * of VM restore to you.
789  */
790 void MSG_vm_restore(msg_vm_t vm)
791 {
792   simcall_vm_restore(vm);
793
794   #ifdef HAVE_TRACING
795   TRACE_msg_vm_restore(vm);
796   #endif
797 }
798
799
800
801
802 /** @brief Get the physical host of a given VM.
803  *  @ingroup msg_VMs
804  */
805 msg_host_t MSG_vm_get_pm(msg_vm_t vm)
806 {
807   return simcall_vm_get_pm(vm);
808 }