1 /* Copyright (c) 2007-2012. The SimGrid Team. All rights reserved. */
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. */
8 #include "xbt/sysdep.h" /* calloc, printf */
10 /* Create a log channel to have nice outputs. */
12 #include "xbt/asserts.h"
13 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
14 "Messages specific for this msg example");
17 int computation_fun(int argc, char *argv[])
19 const char *pr_name = MSG_process_get_name(MSG_process_self());
20 const char *host_name = MSG_host_get_name(MSG_host_self());
22 msg_task_t task = MSG_task_create("Task", 1000000, 1000000, NULL);
24 double clock_sta = MSG_get_clock();
25 MSG_task_execute(task);
26 double clock_end = MSG_get_clock();
28 XBT_INFO("%s:%s task executed %g", host_name, pr_name, clock_end - clock_sta);
30 MSG_task_destroy(task);
35 void launch_computation_worker(msg_host_t host)
37 const char *pr_name = "compute";
38 char **argv = xbt_new(char *, 2);
39 argv[0] = xbt_strdup(pr_name);
42 MSG_process_create_with_arguments(pr_name, computation_fun, NULL, host, 1, argv);
47 msg_process_t tx_proc;
51 int communication_tx_fun(int argc, char *argv[])
53 xbt_assert(argc == 2);
54 const char *mbox = argv[1];
56 msg_task_t task = MSG_task_create("Task", 1000000, 1000000, NULL);
58 struct task_priv *priv = xbt_new(struct task_priv, 1);
59 priv->tx_proc = MSG_process_self();
60 priv->tx_host = MSG_host_self();
61 priv->clock_sta = MSG_get_clock();
63 MSG_task_set_data(task, priv);
65 MSG_task_send(task, mbox);
70 int communication_rx_fun(int argc, char *argv[])
72 const char *pr_name = MSG_process_get_name(MSG_process_self());
73 const char *host_name = MSG_host_get_name(MSG_host_self());
74 xbt_assert(argc == 2);
75 const char *mbox = argv[1];
77 msg_task_t task = NULL;
78 MSG_task_recv(&task, mbox);
80 struct task_priv *priv = MSG_task_get_data(task);
81 double clock_end = MSG_get_clock();
83 XBT_INFO("%s:%s to %s:%s => %g sec",
84 MSG_host_get_name(priv->tx_host),
85 MSG_process_get_name(priv->tx_proc),
86 host_name, pr_name, clock_end - priv->clock_sta);
88 MSG_task_destroy(task);
93 void launch_communication_worker(msg_host_t tx_host, msg_host_t rx_host)
95 char *mbox = bprintf("MBOX:%s-%s",
96 MSG_host_get_name(tx_host),
97 MSG_host_get_name(rx_host));
102 argv = xbt_new(char *, 3);
103 argv[0] = xbt_strdup(pr_name);
104 argv[1] = xbt_strdup(mbox);
107 MSG_process_create_with_arguments(pr_name, communication_tx_fun, NULL, tx_host, 2, argv);
110 argv = xbt_new(char *, 3);
111 argv[0] = xbt_strdup(pr_name);
112 argv[1] = xbt_strdup(mbox);
115 MSG_process_create_with_arguments(pr_name, communication_rx_fun, NULL, rx_host, 2, argv);
121 int master_main(int argc, char *argv[])
123 xbt_dynar_t hosts_dynar = MSG_hosts_as_dynar();
124 msg_host_t pm0 = xbt_dynar_get_as(hosts_dynar, 0, msg_host_t);
125 msg_host_t pm1 = xbt_dynar_get_as(hosts_dynar, 1, msg_host_t);
126 msg_host_t pm2 = xbt_dynar_get_as(hosts_dynar, 2, msg_host_t);
129 XBT_INFO("### Put a task on a PM");
130 launch_computation_worker(pm0);
131 MSG_process_sleep(2);
133 XBT_INFO("### Put two tasks on a PM");
134 launch_computation_worker(pm0);
135 launch_computation_worker(pm0);
136 MSG_process_sleep(2);
138 XBT_INFO("### Put a task on each PM");
139 launch_computation_worker(pm0);
140 launch_computation_worker(pm1);
141 MSG_process_sleep(2);
143 XBT_INFO("### Put a VM on a PM, and put a task to the VM");
144 vm0 = MSG_vm_create_core(pm0, "VM0");
146 launch_computation_worker(vm0);
147 MSG_process_sleep(2);
150 XBT_INFO("### Put a VM on a PM, and put a task to the PM (FIXME: broken)");
151 vm0 = MSG_vm_create_core(pm0, "VM0");
153 launch_computation_worker(pm0);
154 MSG_process_sleep(2);
157 XBT_INFO("### Put two VMs on a PM, and put a task to each VM");
158 vm0 = MSG_vm_create_core(pm0, "VM0");
159 vm1 = MSG_vm_create_core(pm0, "VM1");
162 launch_computation_worker(vm0);
163 launch_computation_worker(vm1);
164 MSG_process_sleep(2);
168 XBT_INFO("### Put a VM on each PM, and put a task to each VM");
169 vm0 = MSG_vm_create_core(pm0, "VM0");
170 vm1 = MSG_vm_create_core(pm1, "VM1");
173 launch_computation_worker(vm0);
174 launch_computation_worker(vm1);
175 MSG_process_sleep(2);
180 XBT_INFO("### Make a connection between PM0 and PM1");
181 launch_communication_worker(pm0, pm1);
182 MSG_process_sleep(5);
184 XBT_INFO("### Make two connection between PM0 and PM1");
185 launch_communication_worker(pm0, pm1);
186 launch_communication_worker(pm0, pm1);
187 MSG_process_sleep(5);
189 XBT_INFO("### Make a connection between PM0 and VM0@PM0");
190 vm0 = MSG_vm_create_core(pm0, "VM0");
192 launch_communication_worker(pm0, vm0);
193 MSG_process_sleep(5);
196 XBT_INFO("### Make a connection between PM0 and VM0@PM1");
197 vm0 = MSG_vm_create_core(pm1, "VM0");
199 launch_communication_worker(pm0, vm0);
200 MSG_process_sleep(5);
203 XBT_INFO("### Make two connections between PM0 and VM0@PM1");
204 vm0 = MSG_vm_create_core(pm1, "VM0");
206 launch_communication_worker(pm0, vm0);
207 launch_communication_worker(pm0, vm0);
208 MSG_process_sleep(5);
211 XBT_INFO("### Make a connection between PM0 and VM0@PM1, and also make a connection between PM0 and PM1");
212 vm0 = MSG_vm_create_core(pm1, "VM0");
214 launch_communication_worker(pm0, vm0);
215 launch_communication_worker(pm0, pm1);
216 MSG_process_sleep(5);
219 XBT_INFO("### Make a connection between VM0@PM0 and PM1@PM1, and also make a connection between VM0@PM0 and VM1@PM1");
220 vm0 = MSG_vm_create_core(pm0, "VM0");
221 vm1 = MSG_vm_create_core(pm1, "VM1");
224 launch_communication_worker(vm0, vm1);
225 launch_communication_worker(vm0, vm1);
226 MSG_process_sleep(5);
231 XBT_INFO("### Relocate VM0 between PM0 and PM1");
232 vm0 = MSG_vm_create_core(pm0, "VM0");
234 launch_communication_worker(vm0, pm2);
235 MSG_process_sleep(0.01);
236 MSG_vm_migrate(vm0, pm1);
237 MSG_process_sleep(0.01);
238 MSG_vm_migrate(vm0, pm0);
239 MSG_process_sleep(5);
245 void launch_master(msg_host_t host)
247 const char *pr_name = "master_";
248 char **argv = xbt_new(char *, 2);
249 argv[0] = xbt_strdup(pr_name);
252 msg_process_t pr = MSG_process_create_with_arguments(pr_name, master_main, NULL, host, 1, argv);
256 int main(int argc, char *argv[])
258 /* Get the arguments */
259 MSG_init(&argc, argv);
261 /* load the platform file */
262 MSG_create_environment(argv[1]);
264 xbt_dynar_t hosts_dynar = MSG_hosts_as_dynar();
265 msg_host_t pm0 = xbt_dynar_get_as(hosts_dynar, 0, msg_host_t);
268 int res = MSG_main();
269 XBT_INFO("Simulation time %g", MSG_get_clock());