#include <simgrid/s4u/VirtualMachine.hpp>
#include <src/instr/instr_private.hpp>
#include <src/plugins/vm/VirtualMachineImpl.hpp>
+#include <src/plugins/vm/VmHostExt.hpp>
#include <src/plugins/vm/VmLiveMigration.hpp>
#include <xbt/ex.hpp>
namespace simgrid {
namespace vm {
+simgrid::xbt::Extension<s4u::Host, VmMigrationExt> VmMigrationExt::EXTENSION_ID;
+
+void VmMigrationExt::ensureVmMigrationExtInstalled()
+{
+ if (not EXTENSION_ID.valid())
+ EXTENSION_ID = simgrid::s4u::Host::extension_create<VmMigrationExt>();
+}
void MigrationRx::operator()()
{
vm_->resume();
// Now the VM is running on the new host (the migration is completed) (even if the SRC crash)
- vm_->pimpl_vm_->isMigrating = false;
+ vm_->getImpl()->isMigrating = false;
XBT_DEBUG("VM(%s) moved from PM(%s) to PM(%s)", vm_->getCname(), src_pm_->getCname(), dst_pm_->getCname());
if (TRACE_msg_vm_is_enabled()) {
XBT_VERB("timeout (%lf s) in sending_migration_data, remaining %llu bytes of %llu", timeout, remaining, size);
sent -= remaining;
}
+ delete msg;
}
double clock_end = s4u::Engine::getClock();
}
SG_BEGIN_DECL()
+
+static void onVirtualMachineShutdown(simgrid::s4u::VirtualMachine* vm)
+{
+ if (vm->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::onVmShutdown.connect(&onVirtualMachineShutdown);
+}
+
+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 */
+
+ msg_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->isMigrating();
+}
+
void sg_vm_migrate(simgrid::s4u::VirtualMachine* vm, simgrid::s4u::Host* dst_pm)
{
simgrid::s4u::Host* src_pm = vm->getPm();
if (vm->isMigrating())
THROWF(vm_error, 0, "Cannot migrate VM '%s' that is already migrating.", vm->getCname());
- vm->pimpl_vm_->isMigrating = true;
+ vm->getImpl()->isMigrating = true;
std::string rx_name =
std::string("__pr_mig_rx:") + vm->getCname() + "(" + src_pm->getCname() + "-" + dst_pm->getCname() + ")";
simgrid::s4u::ActorPtr tx =
simgrid::s4u::Actor::createActor(tx_name.c_str(), src_pm, simgrid::vm::MigrationTx(vm, dst_pm));
+ vm->extension_set<simgrid::vm::VmMigrationExt>(new simgrid::vm::VmMigrationExt(simgrid::s4u::Actor::self(), rx, tx));
+
/* wait until the migration have finished or on error has occurred */
XBT_DEBUG("wait for reception of the final ACK (i.e. migration has been correctly performed");
simgrid::s4u::MailboxPtr mbox_ctl = simgrid::s4u::Mailbox::byName(
std::string("__mbox_mig_ctl:") + vm->getCname() + "(" + src_pm->getCname() + "-" + dst_pm->getCname() + ")");
delete static_cast<std::string*>(mbox_ctl->get());
-
tx->join();
rx->join();
- vm->pimpl_vm_->isMigrating = false;
+ vm->getImpl()->isMigrating = false;
}
}