1 /* Copyright (c) 2007-2013. The SimGrid Team.
2 * All rights reserved. */
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. */
8 #include "msg/msg.h" /* Yeah! If you want to use msg, you need to include msg/msg.h */
9 #include "xbt/sysdep.h" /* calloc, printf */
11 /* Create a log channel to have nice outputs. */
13 #include "xbt/asserts.h"
14 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
15 "Messages specific for this msg example");
17 /** @addtogroup MSG_examples
19 * - <b>cloud/masterslave_virtual_machines.c: Master/slaves
20 * example, à la cloud</b>. The classical example revisited to demonstrate the use of virtual machines.
23 double task_comp_size = 10000000;
24 double task_comm_size = 10000000;
27 int master(int argc, char *argv[]);
28 int slave_fun(int argc, char *argv[]);
30 static void work_batch(int slaves_count) {
32 for (i = 0; i < slaves_count; i++) {
33 char taskname_buffer[64];
34 char mailbox_buffer[64];
36 sprintf(taskname_buffer, "Task_%d", i);
37 sprintf(mailbox_buffer,"Slave_%d",i);
39 XBT_INFO("Sending \"%s\" to \"%s\"",taskname_buffer,mailbox_buffer);
40 MSG_task_send(MSG_task_create(taskname_buffer, task_comp_size, task_comm_size,NULL),
45 int master(int argc, char *argv[]) {
46 int slaves_count = 10;
47 msg_host_t *slaves = xbt_new(msg_host_t,10);
52 /* Retrive the hostnames constituting our playground today */
53 for (i = 1; i < argc; i++) {
54 slaves[i - 1] = MSG_get_host_by_name(argv[i]);
55 xbt_assert(slaves[i - 1] != NULL, "Cannot use inexistent host %s", argv[i]);
58 /* Launch the sub processes: one VM per host, with one process inside each */
60 for (i=0;i<slaves_count;i++) {
62 sprintf(slavename,"Slave %d",i);
63 char**argv=xbt_new(char*,3);
64 argv[0] = xbt_strdup(slavename);
65 argv[1] = bprintf("%d",i);
69 snprintf(vmName, 64, "vm_%d", i);
71 msg_vm_t vm = MSG_vm_start(slaves[i],vmName,2);
72 MSG_vm_bind(vm, MSG_process_create_with_arguments(slavename,slave_fun,NULL,slaves[i],2,argv));
76 xbt_dynar_t vms = MSG_vms_as_dynar();
77 XBT_INFO("Launched %ld VMs", xbt_dynar_length(vms));
79 /* Send a bunch of work to every one */
80 XBT_INFO("Send a first batch of work to every one");
81 work_batch(slaves_count);
83 XBT_INFO("Now suspend all VMs, just for fun");
85 xbt_dynar_foreach(vms,i,vm) {
89 XBT_INFO("Wait a while");
92 XBT_INFO("Enough. Let's resume everybody.");
93 xbt_dynar_foreach(vms,i,vm) {
96 XBT_INFO("Sleep long enough for everyone to be done with previous batch of work");
97 MSG_process_sleep(1000-MSG_get_clock());
99 XBT_INFO("Add one more process per VM");
100 xbt_dynar_foreach(vms,i,vm) {
101 msg_vm_t vm = xbt_dynar_get_as(vms,i,msg_vm_t);
103 sprintf(slavename,"Slave %ld",i+xbt_dynar_length(vms));
104 char**argv=xbt_new(char*,3);
105 argv[0] = xbt_strdup(slavename);
106 argv[1] = bprintf("%ld",i+xbt_dynar_length(vms));
108 MSG_vm_bind(vm, MSG_process_create_with_arguments(slavename,slave_fun,NULL,slaves[i],2,argv));
111 XBT_INFO("Reboot all the VMs");
112 xbt_dynar_foreach(vms,i,vm) {
116 work_batch(slaves_count*2);
118 XBT_INFO("Migrate everyone to the second host.");
119 xbt_dynar_foreach(vms,i,vm) {
120 MSG_vm_migrate(vm,slaves[1]);
122 XBT_INFO("Suspend everyone, move them to the third host, and resume them.");
123 xbt_dynar_foreach(vms,i,vm) {
125 MSG_vm_migrate(vm,slaves[2]);
130 XBT_INFO("Let's shut down the simulation. 10 first processes will be shut down cleanly while the second half will forcefully get killed");
131 for (i = 0; i < slaves_count; i++) {
132 char mailbox_buffer[64];
133 sprintf(mailbox_buffer,"Slave_%d",i);
134 msg_task_t finalize = MSG_task_create("finalize", 0, 0, 0);
135 MSG_task_send(finalize, mailbox_buffer);
138 XBT_INFO("Wait a while before effective shutdown.");
139 MSG_process_sleep(2);
141 xbt_dynar_foreach(vms,i,vm) {
146 XBT_INFO("Goodbye now!");
148 xbt_dynar_free(&vms);
150 } /* end_of_master */
152 /** Receiver function */
153 int slave_fun(int argc, char *argv[])
155 char mailbox_name[128];
156 msg_task_t task = NULL;
157 _XBT_GNUC_UNUSED int res;
158 /* since the slaves will move around, use slave_%d as mailbox names instead of hostnames */
159 xbt_assert(argc>=2, "slave processes need to be given their rank as parameter");
160 sprintf(mailbox_name,"Slave_%s",argv[1]);
161 XBT_INFO("Slave listenning on %s",argv[1]);
163 res = MSG_task_receive(&(task),mailbox_name);
164 xbt_assert(res == MSG_OK, "MSG_task_get failed");
166 XBT_INFO("Received \"%s\" from mailbox %s", MSG_task_get_name(task),mailbox_name);
167 if (!strcmp(MSG_task_get_name(task), "finalize")) {
168 MSG_task_destroy(task);
172 MSG_task_execute(task);
173 XBT_INFO("\"%s\" done", MSG_task_get_name(task));
174 MSG_task_destroy(task);
182 int main(int argc, char *argv[])
184 msg_error_t res = MSG_OK;
185 xbt_dynar_t hosts_dynar;
186 msg_host_t*hosts= xbt_new(msg_host_t,10);
187 char**hostnames= xbt_new(char*,10);
188 char**masterargv=xbt_new(char*,12);
191 /* Get the arguments */
192 MSG_init(&argc, argv);
194 printf("Usage: %s platform_file\n", argv[0]);
195 printf("example: %s msg_platform.xml\n", argv[0]);
198 printf("Usage: %s platform_file\n", argv[0]);
199 printf("Other parameters (such as the deployment file) are ignored.");
202 /* load the platform file */
203 MSG_create_environment(argv[1]);
204 /* Retrieve the 10 first hosts of the platform file */
205 hosts_dynar = MSG_hosts_as_dynar();
206 xbt_assert(xbt_dynar_length(hosts_dynar)>10,
207 "I need at least 10 hosts in the platform file, but %s contains only %ld hosts_dynar.",
208 argv[1],xbt_dynar_length(hosts_dynar));
210 hosts[i] = xbt_dynar_get_as(hosts_dynar,i,msg_host_t);
211 hostnames[i] = xbt_strdup(MSG_host_get_name(hosts[i]));
213 masterargv[0]=xbt_strdup("master");
215 masterargv[i] = xbt_strdup(MSG_host_get_name(hosts[i-1]));
218 MSG_process_create_with_arguments("master",master,NULL,hosts[0],11,masterargv);
220 XBT_INFO("Simulation time %g", MSG_get_clock());
226 xbt_dynar_free(&hosts_dynar);