Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Deprecate functions MSG_global_init() / MSG_global_init_args() in flavor of MSG_init()
[simgrid.git] / examples / msg / cloud / masterslave_virtual_machines.c
1 /* Copyright (c) 2007-2012. The SimGrid Team. All rights reserved.                             */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include <stdio.h>
7 #include "msg/msg.h"            /* Yeah! If you want to use msg, you need to include msg/msg.h */
8 #include "xbt/sysdep.h"         /* calloc, printf */
9
10 /* Create a log channel to have nice outputs. */
11 #include "xbt/log.h"
12 #include "xbt/asserts.h"
13 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
14                              "Messages specific for this msg example");
15
16 /** @addtogroup MSG_examples
17  * 
18  *  - <b>cloud/masterslave_virtual_machines.c: Master/slaves
19  *    example, à la cloud</b>. The classical example revisited to demonstrate the use of virtual machines.
20  */
21
22 double task_comp_size = 10000000;
23 double task_comm_size = 10000000;
24
25
26 int master(int argc, char *argv[]);
27 int slave_fun(int argc, char *argv[]);
28
29 static void work_batch(int slaves_count) {
30   int i;
31   for (i = 0; i < slaves_count; i++) {
32     char taskname_buffer[64];
33     char mailbox_buffer[64];
34
35     sprintf(taskname_buffer, "Task_%d", i);
36     sprintf(mailbox_buffer,"Slave_%d",i);
37
38     XBT_INFO("Sending \"%s\" to \"%s\"",taskname_buffer,mailbox_buffer);
39     MSG_task_send(MSG_task_create(taskname_buffer, task_comp_size, task_comm_size,NULL),
40         mailbox_buffer);
41   }
42 }
43
44 int master(int argc, char *argv[]) {
45   int slaves_count = 10;
46   m_host_t *slaves = xbt_new(m_host_t,10);
47
48   int i;
49
50   /* Retrive the hostnames constituting our playground today */
51   for (i = 1; i < argc; i++) {
52     slaves[i - 1] = MSG_get_host_by_name(argv[i]);
53     xbt_assert(slaves[i - 1] != NULL, "Cannot use inexistent host %s", argv[i]);
54   }
55
56   /* Launch the sub processes: one VM per host, with one process inside each */
57
58   for (i=0;i<slaves_count;i++) {
59     char slavename[64];
60     sprintf(slavename,"Slave %d",i);
61     char**argv=xbt_new(char*,3);
62     argv[0] = xbt_strdup(slavename);
63     argv[1] = bprintf("%d",i);
64     argv[2] = NULL;
65     msg_vm_t vm = MSG_vm_start(slaves[i],2);
66     MSG_vm_bind(vm, MSG_process_create_with_arguments(slavename,slave_fun,NULL,slaves[i],2,argv));
67   }
68
69   xbt_dynar_t vms = MSG_vms_as_dynar();
70   XBT_INFO("Launched %ld VMs", xbt_dynar_length(vms));
71
72   /* Send a bunch of work to every one */
73   XBT_INFO("Send a first batch of work to every one");
74   work_batch(slaves_count);
75
76   XBT_INFO("Now suspend all VMs, just for fun");
77   for (i=0;i<xbt_dynar_length(vms);i++)
78     MSG_vm_suspend(xbt_dynar_get_as(vms,i,msg_vm_t));
79
80   XBT_INFO("Wait a while");
81   MSG_process_sleep(2);
82
83   XBT_INFO("Enough. Let's resume everybody.");
84   for (i=0;i<xbt_dynar_length(vms);i++)
85     MSG_vm_resume(xbt_dynar_get_as(vms,i,msg_vm_t));
86
87   XBT_INFO("Sleep long enough for everyone to be done with previous batch of work");
88   MSG_process_sleep(1000-MSG_get_clock());
89
90   XBT_INFO("Add one more process per VM");
91   for (i=0;i<xbt_dynar_length(vms);i++) {
92     msg_vm_t vm = xbt_dynar_get_as(vms,i,msg_vm_t);
93     char slavename[64];
94     sprintf(slavename,"Slave %ld",i+xbt_dynar_length(vms));
95     char**argv=xbt_new(char*,3);
96     argv[0] = xbt_strdup(slavename);
97     argv[1] = bprintf("%ld",i+xbt_dynar_length(vms));
98     argv[2] = NULL;
99     MSG_vm_bind(vm, MSG_process_create_with_arguments(slavename,slave_fun,NULL,slaves[i],2,argv));
100   }
101
102   work_batch(slaves_count*2);
103
104   XBT_INFO("Migrate everyone to the second host.");
105   for (i=0;i<xbt_dynar_length(vms);i++)
106     MSG_vm_migrate(xbt_dynar_get_as(vms,i,msg_vm_t),slaves[1]);
107
108   XBT_INFO("Suspend everyone, move them to the third host, and resume them.");
109   for (i=0;i<xbt_dynar_length(vms);i++) {
110     msg_vm_t vm = xbt_dynar_get_as(vms,i,msg_vm_t);
111     MSG_vm_suspend(vm);
112     MSG_vm_migrate(vm,slaves[2]);
113     MSG_vm_resume(vm);
114   }
115
116
117   XBT_INFO("Let's shut down the simulation. 10 first processes will be shut down cleanly while the second half will forcefully get killed");
118   for (i = 0; i < slaves_count; i++) {
119     char mailbox_buffer[64];
120     sprintf(mailbox_buffer,"Slave_%d",i);
121     m_task_t finalize = MSG_task_create("finalize", 0, 0, 0);
122     MSG_task_send(finalize, mailbox_buffer);
123   }
124
125   for (i=0;i<xbt_dynar_length(vms);i++) {
126     msg_vm_t vm = xbt_dynar_get_as(vms,i,msg_vm_t);
127     MSG_vm_shutdown(vm);
128     MSG_vm_destroy(vm);
129   }
130
131   XBT_INFO("Goodbye now!");
132   free(slaves);
133   xbt_dynar_free(&vms);
134   return 0;
135 }                               /* end_of_master */
136
137 /** Receiver function  */
138 int slave_fun(int argc, char *argv[])
139 {
140   char *mailbox_name;
141   m_task_t task = NULL;
142   _XBT_GNUC_UNUSED int res;
143
144   /* since the slaves will move around, use slave_%d as mailbox names instead of hostnames */
145   xbt_assert(argc>=2, "slave processes need to be given their rank as parameter");
146   mailbox_name=bprintf("Slave_%s",argv[1]);
147
148   while (1) {
149     res = MSG_task_receive(&(task),mailbox_name);
150     xbt_assert(res == MSG_OK, "MSG_task_get failed");
151
152     XBT_INFO("Received \"%s\" from mailbox %s", MSG_task_get_name(task),mailbox_name);
153     if (!strcmp(MSG_task_get_name(task), "finalize")) {
154       MSG_task_destroy(task);
155       break;
156     }
157
158     MSG_task_execute(task);
159     XBT_INFO("\"%s\" done", MSG_task_get_name(task));
160     MSG_task_destroy(task);
161     task = NULL;
162   }
163
164   free(mailbox_name);
165   return 0;
166 }                               /* end_of_slave */
167
168 /** Main function */
169 int main(int argc, char *argv[])
170 {
171   MSG_error_t res = MSG_OK;
172   xbt_dynar_t hosts_dynar;
173   m_host_t*hosts= xbt_new(m_host_t,10);
174   char**hostnames= xbt_new(char*,10);
175   char**masterargv=xbt_new(char*,12);
176   int i;
177
178   /* Get the arguments */
179   MSG_init(&argc, argv);
180   if (argc < 2) {
181     printf("Usage: %s platform_file\n", argv[0]);
182     printf("example: %s msg_platform.xml\n", argv[0]);
183     exit(1);
184   } if (argc>2) {
185     printf("Usage: %s platform_file\n", argv[0]);
186     printf("Other parameters (such as the deployment file) are ignored.");
187   }
188
189   /* load the platform file */
190   MSG_create_environment(argv[1]);
191   /* Retrieve the 10 first hosts of the platform file */
192   hosts_dynar = MSG_hosts_as_dynar();
193   xbt_assert(xbt_dynar_length(hosts_dynar)>10,
194       "I need at least 10 hosts in the platform file, but %s contains only %ld hosts_dynar.",
195       argv[1],xbt_dynar_length(hosts_dynar));
196   for (i=0;i<10;i++) {
197     hosts[i] = xbt_dynar_get_as(hosts_dynar,i,m_host_t);
198     hostnames[i] = xbt_strdup(MSG_host_get_name(hosts[i]));
199   }
200   masterargv[0]=xbt_strdup("master");
201   for (i=1;i<11;i++) {
202     masterargv[i] = xbt_strdup(MSG_host_get_name(hosts[i-1]));
203   }
204   masterargv[11]=NULL;
205   MSG_process_create_with_arguments("master",master,NULL,hosts[0],11,masterargv);
206   res = MSG_main();
207   XBT_INFO("Simulation time %g", MSG_get_clock());
208
209   MSG_clean();
210   free(hosts);
211   free(hostnames);
212   xbt_dynar_free(&hosts_dynar);
213
214   if (res == MSG_OK)
215     return 0;
216   else
217     return 1;
218 }                               /* end_of_main */