1 /* Copyright (c) 2007, 2008, 2009, 2010. 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 "surf/surfxml_parse.h" /* to override surf_parse and bypass the parser */
11 /* Create a log channel to have nice outputs. */
13 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
14 "Messages specific for this msg example");
15 #define FINALIZE ((void*)221297) /* a magic number to tell people to stop working */
17 static int surf_parse_bypass_platform(void)
19 static int AX_ptr = 0;
20 static int surfxml_bufferstack_size = 2048;
22 /* allocating memory for the buffer, I think 2kB should be enough */
23 surfxml_bufferstack = xbt_new0(char, surfxml_bufferstack_size);
26 SURFXML_BUFFER_SET(platform_version, "2");
27 SURFXML_START_TAG(platform);
30 SURFXML_BUFFER_SET(AS_id, "AS0");
31 SURFXML_BUFFER_SET(AS_routing, "Full");
32 SURFXML_START_TAG(AS);
34 DEBUG0("<host id=\"host A\" power=\"100000000.00\"/>");
35 SURFXML_BUFFER_SET(host_id, "host A");
36 SURFXML_BUFFER_SET(host_power, "100000000.00");
37 SURFXML_BUFFER_SET(host_availability, "1.0");
38 SURFXML_BUFFER_SET(host_availability_file, "");
39 A_surfxml_host_state = A_surfxml_host_state_ON;
40 SURFXML_BUFFER_SET(host_state_file, "");
41 SURFXML_START_TAG(host);
42 SURFXML_END_TAG(host);
44 DEBUG0("<host id=\"host B\" power=\"100000000.00\"/>");
45 SURFXML_BUFFER_SET(host_id, "host B");
46 SURFXML_BUFFER_SET(host_power, "100000000.00");
47 SURFXML_BUFFER_SET(host_availability, "1.0");
48 SURFXML_BUFFER_SET(host_availability_file, "");
49 A_surfxml_host_state = A_surfxml_host_state_ON;
50 SURFXML_BUFFER_SET(host_state_file, "");
51 SURFXML_START_TAG(host);
52 SURFXML_END_TAG(host);
54 DEBUG0("<link id=\"LinkA\" bandwidth=\"10000000.0\" latency=\"0.2\"/>");
55 SURFXML_BUFFER_SET(link_id, "LinkA");
56 SURFXML_BUFFER_SET(link_bandwidth, "10000000.0");
57 SURFXML_BUFFER_SET(link_bandwidth_file, "");
58 SURFXML_BUFFER_SET(link_latency, "0.2");
59 SURFXML_BUFFER_SET(link_latency_file, "");
60 A_surfxml_link_state = A_surfxml_link_state_ON;
61 SURFXML_BUFFER_SET(link_state_file, "");
62 A_surfxml_link_sharing_policy = A_surfxml_link_sharing_policy_SHARED;
63 SURFXML_START_TAG(link);
64 SURFXML_END_TAG(link);
66 DEBUG0("<route src=\"host A\" dst=\"host B\">");
67 SURFXML_BUFFER_SET(route_src, "host A");
68 SURFXML_BUFFER_SET(route_dst, "host B");
69 SURFXML_START_TAG(route);
70 DEBUG0(" <link:ctn id=\"LinkA\"/>");
71 SURFXML_BUFFER_SET(link_ctn_id, "LinkA");
72 SURFXML_START_TAG(link_ctn);
73 SURFXML_END_TAG(link_ctn);
75 SURFXML_END_TAG(route);
79 DEBUG0("</platfrom>");
80 SURFXML_END_TAG(platform);
82 free(surfxml_bufferstack);
86 static int surf_parse_bypass_application(void)
89 static int surfxml_bufferstack_size = 2048;
90 static int surfxml_buffer_stack_stack_ptr = 0;
91 static int surfxml_buffer_stack_stack[1024];
92 /* allocating memory to the buffer, I think 2MB should be enough */
93 surfxml_bufferstack = xbt_new0(char, surfxml_bufferstack_size);
96 SURFXML_BUFFER_SET(platform_version, "2");
98 SURFXML_START_TAG(platform);
100 /* <process host="host A" function="master"> */
101 SURFXML_BUFFER_SET(process_host, "host A");
102 SURFXML_BUFFER_SET(process_function, "master");
103 SURFXML_BUFFER_SET(process_start_time, "-1.0");
104 SURFXML_BUFFER_SET(process_kill_time, "-1.0");
105 SURFXML_START_TAG(process);
107 /* <argument value="20"/> */
108 SURFXML_BUFFER_SET(argument_value, "20");
109 SURFXML_START_TAG(argument);
110 SURFXML_END_TAG(argument);
112 /* <argument value="5000000"/> */
113 SURFXML_BUFFER_SET(argument_value, "5000000");
114 SURFXML_START_TAG(argument);
115 SURFXML_END_TAG(argument);
117 /* <argument value="100000"/> */
118 SURFXML_BUFFER_SET(argument_value, "100000");
119 SURFXML_START_TAG(argument);
120 SURFXML_END_TAG(argument);
122 /* <argument value="host B"/> */
123 SURFXML_BUFFER_SET(argument_value, "host B");
124 SURFXML_START_TAG(argument);
125 SURFXML_END_TAG(argument);
128 SURFXML_END_TAG(process);
130 /* <process host="host B" function="slave"/> */
131 SURFXML_BUFFER_SET(process_host, "host B");
132 SURFXML_BUFFER_SET(process_function, "slave");
133 SURFXML_BUFFER_SET(process_start_time, "-1.0");
134 SURFXML_BUFFER_SET(process_kill_time, "-1.0");
135 SURFXML_START_TAG(process);
136 SURFXML_END_TAG(process);
139 SURFXML_END_TAG(platform);
141 free(surfxml_bufferstack);
145 int master(int argc, char *argv[]);
146 int slave(int argc, char *argv[]);
147 MSG_error_t test_all(void);
154 /** Emitter function */
155 int master(int argc, char *argv[])
157 int slaves_count = 0;
158 m_host_t *slaves = NULL;
159 m_task_t *todo = NULL;
160 int number_of_tasks = 0;
161 double task_comp_size = 0;
162 double task_comm_size = 0;
167 xbt_assert1(sscanf(argv[1], "%d", &number_of_tasks),
168 "Invalid argument %s\n", argv[1]);
169 xbt_assert1(sscanf(argv[2], "%lg", &task_comp_size),
170 "Invalid argument %s\n", argv[2]);
171 xbt_assert1(sscanf(argv[3], "%lg", &task_comm_size),
172 "Invalid argument %s\n", argv[3]);
174 { /* Task creation */
175 char sprintf_buffer[64];
177 todo = xbt_new0(m_task_t, number_of_tasks);
179 for (i = 0; i < number_of_tasks; i++) {
180 sprintf(sprintf_buffer, "Task_%d", i);
182 MSG_task_create(sprintf_buffer, task_comp_size, task_comm_size, NULL);
186 { /* Process organisation */
187 slaves_count = argc - 4;
188 slaves = xbt_new0(m_host_t, slaves_count);
190 for (i = 4; i < argc; i++) {
191 slaves[i - 4] = MSG_get_host_by_name(argv[i]);
192 if (slaves[i - 4] == NULL) {
193 INFO1("Unknown host %s. Stopping Now! ", argv[i]);
199 INFO1("Got %d slave(s) :", slaves_count);
200 for (i = 0; i < slaves_count; i++)
201 INFO1("\t %s", slaves[i]->name);
203 INFO1("Got %d task to process :", number_of_tasks);
205 for (i = 0; i < number_of_tasks; i++)
206 INFO1("\t\"%s\"", todo[i]->name);
208 for (i = 0; i < number_of_tasks; i++) {
209 INFO2("Sending \"%s\" to \"%s\"",
210 todo[i]->name, slaves[i % slaves_count]->name);
211 if (MSG_host_self() == slaves[i % slaves_count]) {
212 INFO0("Hey ! It's me ! :)");
214 MSG_task_put(todo[i], slaves[i % slaves_count], PORT_22);
215 INFO0("Send completed");
219 ("All tasks have been dispatched. Let's tell everybody the computation is over.");
220 for (i = 0; i < slaves_count; i++)
221 MSG_task_put(MSG_task_create("finalize", 0, 0, FINALIZE),
224 INFO0("Goodbye now!");
228 } /* end_of_master */
230 /** Receiver function */
231 int slave(int argc, char *argv[])
233 INFO0("I'm a slave");
235 m_task_t task = NULL;
237 a = MSG_task_get(&(task), PORT_22);
239 INFO1("Received \"%s\" ", MSG_task_get_name(task));
240 if (MSG_task_get_data(task) == FINALIZE) {
241 MSG_task_destroy(task);
244 INFO1("Processing \"%s\" ", MSG_task_get_name(task));
245 MSG_task_execute(task);
246 INFO1("\"%s\" done ", MSG_task_get_name(task));
247 MSG_task_destroy(task);
249 INFO0("Hey ?! What's up ? ");
250 xbt_assert0(0, "Unexpected behavior");
253 INFO0("I'm done. See you!");
258 MSG_error_t test_all(void)
260 MSG_error_t res = MSG_OK;
262 /* Simulation setting */
263 MSG_set_channel_number(MAX_CHANNEL);
264 surf_parse = surf_parse_bypass_platform;
265 MSG_create_environment(NULL);
267 /* Application deployment */
268 MSG_function_register("master", master);
269 MSG_function_register("slave", slave);
270 surf_parse = surf_parse_bypass_application;
271 MSG_launch_application(NULL);
275 INFO1("Simulation time %g", MSG_get_clock());
277 } /* end_of_test_all */
280 int main(int argc, char *argv[])
282 MSG_error_t res = MSG_OK;
284 MSG_global_init(&argc, argv);