+static void onVirtualMachineShutdown(simgrid::s4u::VirtualMachine& vm)
+{
+ if (vm.getImpl()->isMigrating) {
+ vm.extension<simgrid::vm::VmMigrationExt>()->rx_->kill();
+ vm.extension<simgrid::vm::VmMigrationExt>()->tx_->kill();
+ vm.extension<simgrid::vm::VmMigrationExt>()->issuer_->kill();
+ vm.getImpl()->isMigrating = false;
+ }
+}
+
+void sg_vm_live_migration_plugin_init()
+{
+ sg_vm_dirty_page_tracking_init();
+ simgrid::vm::VmMigrationExt::ensureVmMigrationExtInstalled();
+ simgrid::s4u::VirtualMachine::on_shutdown.connect(&onVirtualMachineShutdown);
+}
+
+/* Deprecated. Please use MSG_vm_create_migratable() instead */
+msg_vm_t MSG_vm_create(msg_host_t ind_pm, const char* name, int coreAmount, int ramsize, int mig_netspeed,
+ int dp_intensity)
+{
+ return sg_vm_create_migratable(ind_pm, name, coreAmount, ramsize, mig_netspeed, dp_intensity);
+}
+
+simgrid::s4u::VirtualMachine* sg_vm_create_migratable(simgrid::s4u::Host* pm, const char* name, int coreAmount,
+ int ramsize, int mig_netspeed, int dp_intensity)
+{
+ simgrid::vm::VmHostExt::ensureVmExtInstalled();
+
+ /* For the moment, intensity_rate is the percentage against the migration bandwidth */
+
+ sg_vm_t vm = new simgrid::s4u::VirtualMachine(name, pm, coreAmount, static_cast<sg_size_t>(ramsize) * 1024 * 1024);
+ sg_vm_set_dirty_page_intensity(vm, dp_intensity / 100.0);
+ sg_vm_set_working_set_memory(vm, vm->getRamsize() * 0.9); // assume working set memory is 90% of ramsize
+ sg_vm_set_migration_speed(vm, mig_netspeed * 1024 * 1024.0);
+
+ XBT_DEBUG("migspeed : %f intensity mem : %d", mig_netspeed * 1024 * 1024.0, dp_intensity);
+
+ return vm;
+}
+
+int sg_vm_is_migrating(simgrid::s4u::VirtualMachine* vm)
+{
+ return vm->getImpl()->isMigrating;
+}
+