Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Damn code dupplication
[simgrid.git] / testsuite / msg / msg_test.c
1 /*      $Id$     */
2
3 /* Copyright (c) 2002,2003,2004 Arnaud Legrand. All rights reserved.        */
4
5 /* This program is free software; you can redistribute it and/or modify it
6  * under the terms of the license (GNU LGPL) which comes with this package. */
7
8 #include "msg/msg.h"
9
10 #define VERBOSE
11 #include "messages.h"
12
13 int master(int argc, char *argv[]);
14 int slave(int argc, char *argv[]);
15 int forwarder(int argc, char *argv[]);
16 void test_all(const char *platform_file, const char *application_file);
17
18
19 typedef enum {
20   PORT_22 = 0,
21   MAX_CHANNEL
22 } channel_t;
23
24 void print_args(int argc, char** argv);
25 void print_args(int argc, char** argv)
26 {
27   int i ; 
28
29   fprintf(stderr,"<");
30   for(i=0; i<argc; i++) 
31     fprintf(stderr,"%s ",argv[i]);
32   fprintf(stderr,">\n");
33 }
34
35 int master(int argc, char *argv[])
36 {
37   int slaves_count = 0;
38   m_host_t *slaves = NULL;
39   m_task_t *todo = NULL;
40   int number_of_tasks = 0;
41   double task_comp_size = 0;
42   double task_comm_size = 0;
43
44
45   int i;
46
47   print_args(argc,argv);
48
49   ASSERT(sscanf(argv[1],"%d", &number_of_tasks),
50          "Invalid argument %s\n",argv[1]);
51   ASSERT(sscanf(argv[2],"%lg", &task_comp_size),
52          "Invalid argument %s\n",argv[2]);
53   ASSERT(sscanf(argv[3],"%lg", &task_comm_size),
54          "Invalid argument %s\n",argv[3]);
55
56   {                  /*  Task creation */
57     char sprintf_buffer[64];
58
59     todo = calloc(number_of_tasks, sizeof(m_task_t));
60
61     for (i = 0; i < number_of_tasks; i++) {
62       sprintf(sprintf_buffer, "Task_%d", i);
63       todo[i] = MSG_task_create(sprintf_buffer, task_comp_size, task_comm_size, NULL);
64     }
65   }
66
67   {                  /* Process organisation */
68     slaves_count = argc - 4;
69     slaves = calloc(slaves_count, sizeof(m_host_t));
70     
71     for (i = 4; i < argc; i++) {
72       slaves[i-4] = MSG_get_host_by_name(argv[i]);
73       if(slaves[i-4]==NULL) {
74         PRINT_MESSAGE("Unknown host %s. Stopping Now! \n", argv[i]);
75         abort();
76       }
77     }
78   }
79
80   PRINT_MESSAGE("Got %d slave(s) :\n", slaves_count);
81   for (i = 0; i < slaves_count; i++)
82     PRINT_MESSAGE("\t %s\n", slaves[i]->name);
83
84   PRINT_MESSAGE("Got %d task to process :\n", number_of_tasks);
85
86   for (i = 0; i < number_of_tasks; i++)
87     PRINT_MESSAGE("\t\"%s\"\n", todo[i]->name);
88
89   for (i = 0; i < number_of_tasks; i++) {
90     PRINT_MESSAGE("Sending \"%s\" to \"%s\"\n",
91                   todo[i]->name,
92                   slaves[i % slaves_count]->name);
93     MSG_task_put(todo[i], slaves[i % slaves_count],
94                  PORT_22);
95     PRINT_MESSAGE("Send completed\n");
96   }
97   
98   PRINT_MESSAGE("All tasks have been dispatched. Bye!\n");
99   free(slaves);
100   free(todo);
101   return 0;
102 }
103
104 int slave(int argc, char *argv[])
105 {
106   print_args(argc,argv);
107
108   while(1) {
109     m_task_t task = NULL;
110     int a;
111     a = MSG_task_get(&(task), PORT_22);
112     if (a == MSG_OK) {
113       PRINT_MESSAGE("Received \"%s\" \n", task->name);
114       PRINT_MESSAGE("Processing \"%s\" \n", task->name);
115       MSG_task_execute(task);
116       PRINT_MESSAGE("\"%s\" done \n", task->name);
117       MSG_task_destroy(task);
118     } else {
119       PRINT_MESSAGE("Hey ?! What's up ? \n");
120       DIE("Unexpected behaviour");
121     }
122   }
123   PRINT_MESSAGE("I'm done. See you!\n");
124   return 0;
125 }
126
127
128 int forwarder(int argc, char *argv[])
129 {
130   int i;
131   int slaves_count = argc - 1;
132   m_host_t *slaves = calloc(slaves_count, sizeof(m_host_t));
133
134   print_args(argc,argv);
135
136   {                  /* Process organisation */
137     slaves_count = argc - 1;
138     slaves = calloc(slaves_count, sizeof(m_host_t));
139     
140     for (i = 1; i < argc; i++) {
141       slaves[i-1] = MSG_get_host_by_name(argv[i]);
142       if(slaves[i-1]==NULL) {
143         PRINT_MESSAGE("Unknown host %s. Stopping Now! \n", argv[i]);
144         abort();
145       }
146     }
147   }
148
149
150   while(1) {
151     m_task_t task = NULL;
152     int a;
153     a = MSG_task_get(&(task), PORT_22);
154     if (a == MSG_OK) {
155       PRINT_MESSAGE("Received \"%s\" \n", task->name);
156       PRINT_MESSAGE("Sending \"%s\" to \"%s\"\n",
157                     task->name,
158                     slaves[i % slaves_count]->name);
159       MSG_task_put(task, slaves[i % slaves_count],
160                    PORT_22);
161     } else {
162       PRINT_MESSAGE("Hey ?! What's up ? \n");
163       DIE("Unexpected behaviour");
164     }
165   }
166
167   PRINT_MESSAGE("I'm done. See you!\n");
168   return 0;
169 }
170
171
172 void test_all(const char *platform_file,const char *application_file)
173 {
174   {                             /*  Simulation setting */
175     MSG_set_channel_number(MAX_CHANNEL);
176     MSG_create_environment(platform_file);
177   }
178   {                            /*   Application deployment */
179     MSG_function_register("master", master);
180     MSG_function_register("slave", slave);
181     MSG_function_register("forwarder", forwarder);
182     MSG_launch_application(application_file);
183   }
184   MSG_main();
185   printf("Simulation time %g\n",MSG_getClock());
186 }
187
188 int main(int argc, char *argv[])
189 {
190   MSG_global_init_args(&argc,argv);
191   test_all("msg_platform.xml","msg_deployment.xml");
192   MSG_clean();
193   return (0);
194 }