examples/c/cloud-migration/cloud-migration.c

   1 /* Copyright (c) 2007-2022. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "simgrid/actor.h"
   8 #include "simgrid/engine.h"
   9 #include "simgrid/host.h"
  10 #include "simgrid/mailbox.h"
  11 #include "simgrid/plugins/live_migration.h"
  12 #include "simgrid/vm.h"
  13 #include "xbt/log.h"
  14 #include "xbt/str.h"
  15 #include "xbt/sysdep.h"
  16
  17 XBT_LOG_NEW_DEFAULT_CATEGORY(cloud_migration, "Messages specific for this example");
  18
  19 static void vm_migrate(sg_vm_t vm, sg_host_t dst_pm)
  20 {
  21   const_sg_host_t src_pm = sg_vm_get_pm(vm);
  22   double mig_sta         = simgrid_get_clock();
  23   sg_vm_migrate(vm, dst_pm);
  24   double mig_end = simgrid_get_clock();
  25
  26   XBT_INFO("%s migrated: %s->%s in %g s", sg_vm_get_name(vm), sg_host_get_name(src_pm), sg_host_get_name(dst_pm),
  27            mig_end - mig_sta);
  28 }
  29
  30 static void migration_worker_main(int argc, char* argv[])
  31 {
  32   xbt_assert(argc == 3);
  33   const char* vm_name     = argv[1];
  34   const char* dst_pm_name = argv[2];
  35
  36   sg_host_t src_pm = sg_host_self();
  37   sg_vm_t vm       = (sg_vm_t)sg_vm_by_name(src_pm, vm_name);
  38   sg_host_t dst_pm = sg_host_by_name(dst_pm_name);
  39
  40   vm_migrate(vm, dst_pm);
  41 }
  42
  43 static void vm_migrate_async(const_sg_vm_t vm, const_sg_host_t dst_pm)
  44 {
  45   const char* vm_name     = sg_vm_get_name(vm);
  46   const char* dst_pm_name = sg_host_get_name(dst_pm);
  47
  48   const char* argv[] = {"mig_work", vm_name, dst_pm_name, NULL};
  49   sg_actor_create_("mig_wrk", sg_host_self(), migration_worker_main, 3, argv);
  50 }
  51
  52 static void master_main(int argc, char* argv[])
  53 {
  54   sg_host_t pm0       = sg_host_by_name("Fafard");
  55   sg_host_t pm1       = sg_host_by_name("Tremblay");
  56   const_sg_host_t pm2 = sg_host_by_name("Bourassa");
  57
  58   sg_vm_t vm0 = sg_vm_create_core(pm0, "VM0");
  59   sg_vm_set_ramsize(vm0, 1e9); // 1Gbytes
  60   sg_vm_start(vm0);
  61
  62   XBT_INFO("Test: Migrate a VM with %zu Mbytes RAM", sg_vm_get_ramsize(vm0) / 1000 / 1000);
  63   vm_migrate(vm0, pm1);
  64
  65   sg_vm_destroy(vm0);
  66
  67   vm0 = sg_vm_create_core(pm0, "VM0");
  68   sg_vm_set_ramsize(vm0, 1e8); // 100Mbytes
  69   sg_vm_start(vm0);
  70
  71   XBT_INFO("Test: Migrate a VM with %zu Mbytes RAM", sg_vm_get_ramsize(vm0) / 1000 / 1000);
  72   vm_migrate(vm0, pm1);
  73
  74   sg_vm_destroy(vm0);
  75
  76   vm0         = sg_vm_create_core(pm0, "VM0");
  77   sg_vm_t vm1 = sg_vm_create_core(pm0, "VM1");
  78
  79   sg_vm_set_ramsize(vm0, 1e9); // 1Gbytes
  80   sg_vm_set_ramsize(vm1, 1e9); // 1Gbytes
  81   sg_vm_start(vm0);
  82   sg_vm_start(vm1);
  83
  84   XBT_INFO("Test: Migrate two VMs at once from PM0 to PM1");
  85   vm_migrate_async(vm0, pm1);
  86   vm_migrate_async(vm1, pm1);
  87   sg_actor_sleep_for(10000);
  88
  89   sg_vm_destroy(vm0);
  90   sg_vm_destroy(vm1);
  91
  92   vm0 = sg_vm_create_core(pm0, "VM0");
  93   vm1 = sg_vm_create_core(pm0, "VM1");
  94
  95   sg_vm_set_ramsize(vm0, 1e9); // 1Gbytes
  96   sg_vm_set_ramsize(vm1, 1e9); // 1Gbytes
  97   sg_vm_start(vm0);
  98   sg_vm_start(vm1);
  99
 100   XBT_INFO("Test: Migrate two VMs at once to different PMs");
 101   vm_migrate_async(vm0, pm1);
 102   vm_migrate_async(vm1, pm2);
 103   sg_actor_sleep_for(10000);
 104
 105   sg_vm_destroy(vm0);
 106   sg_vm_destroy(vm1);
 107 }
 108
 109 int main(int argc, char* argv[])
 110 {
 111   /* Get the arguments */
 112   simgrid_init(&argc, argv);
 113   sg_vm_live_migration_plugin_init();
 114
 115   /* load the platform file */
 116   simgrid_load_platform(argv[1]);
 117
 118   sg_actor_create("master_", sg_host_by_name("Fafard"), master_main, 0, NULL);
 119
 120   simgrid_run();
 121   XBT_INFO("Bye (simulation time %g)", simgrid_get_clock());
 122
 123   return 0;
 124 }