--- /dev/null
+/* Copyright (c) 2007-2017. The SimGrid Team. All rights reserved. */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
+#include "simgrid/msg.h"
+#include "simgrid/plugins/live_migration.h"
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test, "Messages specific for this msg example");
+
+/** @addtogroup MSG_examples
+ *
+ * - <b>cloud/bound.c</b>: Demonstrates the use of @ref MSG_task_set_bound
+ */
+
+static int worker_main(int argc, char* argv[])
+{
+ double computation_amount = xbt_str_parse_double(argv[1], "Invalid computation amount: %s");
+ int use_bound = xbt_str_parse_int(argv[2], "Second parameter (use_bound) should be 0 or 1 but is: %s");
+ double bound = xbt_str_parse_double(argv[3], "Invalid bound: %s");
+
+ double clock_sta = MSG_get_clock();
+
+ msg_task_t task = MSG_task_create("Task", computation_amount, 0, NULL);
+ if (use_bound)
+ MSG_task_set_bound(task, bound);
+ MSG_task_execute(task);
+ MSG_task_destroy(task);
+
+ double clock_end = MSG_get_clock();
+ double duration = clock_end - clock_sta;
+ double flops_per_sec = computation_amount / duration;
+
+ if (use_bound)
+ XBT_INFO("bound to %f => duration %f (%f flops/s)", bound, duration, flops_per_sec);
+ else
+ XBT_INFO("not bound => duration %f (%f flops/s)", duration, flops_per_sec);
+
+ return 0;
+}
+
+static void launch_worker(msg_host_t host, const char* pr_name, double computation_amount, int use_bound, double bound)
+{
+ char** argv = xbt_new(char*, 5);
+ argv[0] = xbt_strdup(pr_name);
+ argv[1] = bprintf("%f", computation_amount);
+ argv[2] = bprintf("%d", use_bound);
+ argv[3] = bprintf("%f", bound);
+ argv[4] = NULL;
+
+ MSG_process_create_with_arguments(pr_name, worker_main, NULL, host, 4, argv);
+}
+
+static int worker_busy_loop_main(int argc, char* argv[])
+{
+ msg_task_t* task = MSG_process_get_data(MSG_process_self());
+ MSG_task_execute(*task);
+ MSG_task_destroy(*task);
+
+ return 0;
+}
+
+#define DOUBLE_MAX 1e11
+
+static void test_dynamic_change(void)
+{
+ msg_host_t pm0 = MSG_host_by_name("Fafard");
+
+ msg_vm_t vm0 = MSG_vm_create_core(pm0, "VM0");
+ msg_vm_t vm1 = MSG_vm_create_core(pm0, "VM1");
+ MSG_vm_start(vm0);
+ MSG_vm_start(vm1);
+
+ msg_task_t task0 = MSG_task_create("Task0", DOUBLE_MAX, 0, NULL);
+ msg_task_t task1 = MSG_task_create("Task1", DOUBLE_MAX, 0, NULL);
+ MSG_process_create("worker0", worker_busy_loop_main, &task0, (msg_host_t)vm0);
+ MSG_process_create("worker1", worker_busy_loop_main, &task1, (msg_host_t)vm1);
+
+ double task0_remain_prev = MSG_task_get_flops_amount(task0);
+ double task1_remain_prev = MSG_task_get_flops_amount(task1);
+
+ const double cpu_speed = MSG_host_get_speed(pm0);
+ for (int i = 0; i < 10; i++) {
+ double new_bound = (cpu_speed / 10) * i;
+ XBT_INFO("set bound of VM1 to %f", new_bound);
+ MSG_vm_set_bound(vm1, new_bound);
+ MSG_process_sleep(100);
+
+ double task0_remain_now = MSG_task_get_flops_amount(task0);
+ double task1_remain_now = MSG_task_get_flops_amount(task1);
+
+ double task0_flops_per_sec = task0_remain_prev - task0_remain_now;
+ double task1_flops_per_sec = task1_remain_prev - task1_remain_now;
+
+ XBT_INFO("Task0@VM0: %f flops/s", task0_flops_per_sec / 100);
+ XBT_INFO("Task1@VM1: %f flops/s", task1_flops_per_sec / 100);
+
+ task0_remain_prev = task0_remain_now;
+ task1_remain_prev = task1_remain_now;
+ }
+
+ MSG_process_sleep(2000); // let the tasks end
+
+ MSG_vm_destroy(vm0);
+ MSG_vm_destroy(vm1);
+}
+
+static void test_one_task(msg_host_t hostA)
+{
+ const double cpu_speed = MSG_host_get_speed(hostA);
+ const double computation_amount = cpu_speed * 10;
+ const char* hostA_name = MSG_host_get_name(hostA);
+
+ XBT_INFO("### Test: with/without MSG_task_set_bound");
+
+ XBT_INFO("### Test: no bound for Task1@%s", hostA_name);
+ launch_worker(hostA, "worker0", computation_amount, 0, 0);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 50%% for Task1@%s", hostA_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed / 2);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 33%% for Task1@%s", hostA_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed / 3);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: zero for Task1@%s (i.e., unlimited)", hostA_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, 0);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 200%% for Task1@%s (i.e., meaningless)", hostA_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed * 2);
+
+ MSG_process_sleep(1000);
+}
+
+static void test_two_tasks(msg_host_t hostA, msg_host_t hostB)
+{
+ const double cpu_speed = MSG_host_get_speed(hostA);
+ xbt_assert(cpu_speed == MSG_host_get_speed(hostB));
+ const double computation_amount = cpu_speed * 10;
+ const char* hostA_name = MSG_host_get_name(hostA);
+ const char* hostB_name = MSG_host_get_name(hostB);
+
+ XBT_INFO("### Test: no bound for Task1@%s, no bound for Task2@%s", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 0, 0);
+ launch_worker(hostB, "worker1", computation_amount, 0, 0);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 0 for Task1@%s, 0 for Task2@%s (i.e., unlimited)", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, 0);
+ launch_worker(hostB, "worker1", computation_amount, 1, 0);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 50%% for Task1@%s, 50%% for Task2@%s", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed / 2);
+ launch_worker(hostB, "worker1", computation_amount, 1, cpu_speed / 2);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 25%% for Task1@%s, 25%% for Task2@%s", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed / 4);
+ launch_worker(hostB, "worker1", computation_amount, 1, cpu_speed / 4);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 75%% for Task1@%s, 100%% for Task2@%s", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed * 0.75);
+ launch_worker(hostB, "worker1", computation_amount, 1, cpu_speed);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: no bound for Task1@%s, 25%% for Task2@%s", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 0, 0);
+ launch_worker(hostB, "worker1", computation_amount, 1, cpu_speed / 4);
+
+ MSG_process_sleep(1000);
+
+ XBT_INFO("### Test: 75%% for Task1@%s, 25%% for Task2@%s", hostA_name, hostB_name);
+ launch_worker(hostA, "worker0", computation_amount, 1, cpu_speed * 0.75);
+ launch_worker(hostB, "worker1", computation_amount, 1, cpu_speed / 4);
+
+ MSG_process_sleep(1000);
+}
+
+static int master_main(int argc, char* argv[])
+{
+ msg_host_t pm0 = MSG_host_by_name("Fafard");
+ msg_host_t pm1 = MSG_host_by_name("Fafard");
+
+ XBT_INFO("# 1. Put a single task on a PM. ");
+ test_one_task(pm0);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 2. Put two tasks on a PM.");
+ test_two_tasks(pm0, pm0);
+ XBT_INFO(" ");
+
+ msg_vm_t vm0 = MSG_vm_create_core(pm0, "VM0");
+ MSG_vm_start(vm0);
+
+ XBT_INFO("# 3. Put a single task on a VM. ");
+ test_one_task((msg_host_t)vm0);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 4. Put two tasks on a VM.");
+ test_two_tasks((msg_host_t)vm0, (msg_host_t)vm0);
+ XBT_INFO(" ");
+
+ MSG_vm_destroy(vm0);
+
+ vm0 = MSG_vm_create_core(pm0, "VM0");
+ MSG_vm_start(vm0);
+
+ XBT_INFO("# 6. Put a task on a PM and a task on a VM.");
+ test_two_tasks(pm0, (msg_host_t)vm0);
+ XBT_INFO(" ");
+
+ MSG_vm_destroy(vm0);
+
+ vm0 = MSG_vm_create_core(pm0, "VM0");
+ double cpu_speed = MSG_host_get_speed(pm0);
+ MSG_vm_set_bound(vm0, cpu_speed / 10);
+ MSG_vm_start(vm0);
+
+ XBT_INFO("# 7. Put a single task on the VM capped by 10%%.");
+ test_one_task((msg_host_t)vm0);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 8. Put two tasks on the VM capped by 10%%.");
+ test_two_tasks((msg_host_t)vm0, (msg_host_t)vm0);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 9. Put a task on a PM and a task on the VM capped by 10%%.");
+ test_two_tasks(pm0, (msg_host_t)vm0);
+ XBT_INFO(" ");
+
+ MSG_vm_destroy(vm0);
+
+ vm0 = MSG_vm_create_core(pm0, "VM0");
+
+ MSG_vm_set_ramsize(vm0, 1e9); // 1GB
+ MSG_vm_start(vm0);
+
+ cpu_speed = MSG_host_get_speed(pm0);
+ MSG_vm_start(vm0);
+
+ XBT_INFO("# 10. Test migration");
+ const double computation_amount = cpu_speed * 10;
+
+ XBT_INFO("# 10. (a) Put a task on a VM without any bound.");
+ launch_worker((msg_host_t)vm0, "worker0", computation_amount, 0, 0);
+ MSG_process_sleep(1000);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 10. (b) set 10%% bound to the VM, and then put a task on the VM.");
+ MSG_vm_set_bound(vm0, cpu_speed / 10);
+ launch_worker((msg_host_t)vm0, "worker0", computation_amount, 0, 0);
+ MSG_process_sleep(1000);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 10. (c) migrate");
+ MSG_vm_migrate(vm0, pm1);
+ XBT_INFO(" ");
+
+ XBT_INFO("# 10. (d) Put a task again on the VM.");
+ launch_worker((msg_host_t)vm0, "worker0", computation_amount, 0, 0);
+ MSG_process_sleep(1000);
+ XBT_INFO(" ");
+
+ MSG_vm_destroy(vm0);
+
+ XBT_INFO("# 11. Change a bound dynamically.");
+ test_dynamic_change();
+
+ return 0;
+}
+
+static void launch_master(msg_host_t host)
+{
+ const char* pr_name = "master_";
+ char** argv = xbt_new(char*, 2);
+ argv[0] = xbt_strdup(pr_name);
+ argv[1] = NULL;
+
+ MSG_process_create_with_arguments(pr_name, master_main, NULL, host, 1, argv);
+}
+
+int main(int argc, char* argv[])
+{
+ /* Get the arguments */
+ MSG_init(&argc, argv);
+ MSG_vm_live_migration_plugin_init();
+
+ /* load the platform file */
+ xbt_assert(argc == 2, "Usage: %s platform_file\n\tExample: %s ../platforms/small_platform.xml\n", argv[0], argv[0]);
+
+ MSG_create_environment(argv[1]);
+
+ msg_host_t pm0 = MSG_host_by_name("Fafard");
+ launch_master(pm0);
+
+ int res = MSG_main();
+ XBT_INFO("Bye (simulation time %g)", MSG_get_clock());
+
+ return !(res == MSG_OK);
+}
