Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
85148457fd55e7cff6a8c9af2c7eca2909ed9a3f
[simgrid.git] / src / gras / Virtu / sg_process.c
1 /* $Id$ */
2
3 /* process_sg - GRAS process handling on simulator                          */
4
5 /* Copyright (c) 2003, 2004 Martin Quinson. All rights reserved.            */
6
7 /* This program is free software; you can redistribute it and/or modify it
8  * under the terms of the license (GNU LGPL) which comes with this package. */
9
10 #include "xbt/ex.h"
11 #include "xbt/dict.h"
12 #include "gras_modinter.h"      /* module initialization interface */
13 #include "gras/Virtu/virtu_sg.h"
14 #include "gras/Msg/msg_interface.h"     /* For some checks at simulation end */
15 #include "gras/Transport/transport_interface.h" /* For some checks at simulation end */
16
17 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_virtu_process);
18
19 static long int PID = 1;
20
21
22 void gras_agent_spawn(const char *name, void *data,
23                       xbt_main_func_t code, int argc, char *argv[],
24                       xbt_dict_t properties)
25 {
26
27   SIMIX_process_create(name, code,
28                        data, gras_os_myname(), argc, argv, properties);
29 }
30
31 /* **************************************************************************
32  * Process constructor/destructor (semi-public interface)
33  * **************************************************************************/
34
35 void gras_process_init()
36 {
37   gras_hostdata_t *hd =
38     (gras_hostdata_t *) SIMIX_host_get_data(SIMIX_host_self());
39   gras_procdata_t *pd = xbt_new0(gras_procdata_t, 1);
40   gras_trp_procdata_t trp_pd;
41
42   SIMIX_process_set_data(SIMIX_process_self(), (void *) pd);
43
44
45   gras_procdata_init();
46
47   if (!hd) {
48     /* First process on this host */
49     hd = xbt_new(gras_hostdata_t, 1);
50     hd->refcount = 1;
51     hd->ports = xbt_dynar_new(sizeof(gras_sg_portrec_t), NULL);
52     SIMIX_host_set_data(SIMIX_host_self(), (void *) hd);
53   } else {
54     hd->refcount++;
55   }
56
57   trp_pd = (gras_trp_procdata_t) gras_libdata_by_name("gras_trp");
58   pd->pid = PID++;
59
60   if (SIMIX_process_self() != NULL) {
61     pd->ppid = gras_os_getpid();
62   } else
63     pd->ppid = -1;
64
65   trp_pd->msg_selectable_sockets = xbt_queue_new(0, sizeof(gras_socket_t));
66
67   trp_pd->meas_selectable_sockets = xbt_queue_new(0, sizeof(gras_socket_t));
68
69   VERB2("Creating process '%s' (%d)",
70         SIMIX_process_get_name(SIMIX_process_self()), gras_os_getpid());
71 }
72
73 void gras_process_exit()
74 {
75   xbt_dynar_t sockets =
76     ((gras_trp_procdata_t) gras_libdata_by_name("gras_trp"))->sockets;
77   gras_socket_t sock_iter;
78   unsigned int cursor;
79   gras_hostdata_t *hd =
80     (gras_hostdata_t *) SIMIX_host_get_data(SIMIX_host_self());
81   gras_procdata_t *pd =
82     (gras_procdata_t *) SIMIX_process_get_data(SIMIX_process_self());
83
84   gras_msg_procdata_t msg_pd =
85     (gras_msg_procdata_t) gras_libdata_by_name("gras_msg");
86   gras_trp_procdata_t trp_pd =
87     (gras_trp_procdata_t) gras_libdata_by_name("gras_trp");
88
89   xbt_queue_free(&trp_pd->msg_selectable_sockets);
90
91   xbt_queue_free(&trp_pd->meas_selectable_sockets);
92
93
94   xbt_assert0(hd, "Run gras_process_init (ie, gras_init)!!");
95
96   VERB2("GRAS: Finalizing process '%s' (%d)",
97         SIMIX_process_get_name(SIMIX_process_self()), gras_os_getpid());
98
99   if (xbt_dynar_length(msg_pd->msg_queue))
100     WARN1("process %d terminated, but some messages are still queued",
101           gras_os_getpid());
102
103   /* if each process has its sockets list, we need to close them when the
104      process finish */
105   xbt_dynar_foreach(sockets, cursor, sock_iter) {
106     VERB1("Closing the socket %p left open on exit. Maybe a socket leak?",
107           sock_iter);
108     gras_socket_close(sock_iter);
109   }
110   if (!--(hd->refcount)) {
111     xbt_dynar_free(&hd->ports);
112     free(hd);
113   }
114   gras_procdata_exit();
115   free(pd);
116 }
117
118 /* **************************************************************************
119  * Process data (public interface)
120  * **************************************************************************/
121
122 gras_procdata_t *gras_procdata_get(void)
123 {
124   gras_procdata_t *pd =
125     (gras_procdata_t *) SIMIX_process_get_data(SIMIX_process_self());
126
127   xbt_assert0(pd, "Run gras_process_init! (ie, gras_init)");
128
129   return pd;
130 }
131
132 void *gras_libdata_by_name_from_remote(const char *name, smx_process_t p)
133 {
134   gras_procdata_t *pd = (gras_procdata_t *) SIMIX_process_get_data(p);
135
136   xbt_assert2(pd,
137               "process '%s' on '%s' didn't run gras_process_init! (ie, gras_init)",
138               SIMIX_process_get_name(p),
139               SIMIX_host_get_name(SIMIX_process_get_host(p)));
140
141   return gras_libdata_by_name_from_procdata(name, pd);
142 }
143
144 /** @brief retrieve the value of a given process property (or NULL if not defined) */
145 const char *gras_process_property_value(const char *name)
146 {
147   return
148     xbt_dict_get_or_null(SIMIX_process_get_properties(SIMIX_process_self()),
149                          name);
150 }
151
152 /** @brief retrieve the process properties dictionnary
153  *  @warning it's the original one, not a copy. Don't mess with it
154  */
155 xbt_dict_t gras_process_properties(void)
156 {
157   return SIMIX_process_get_properties(SIMIX_process_self());
158 }
159
160 /* **************************************************************************
161  * OS virtualization function
162  * **************************************************************************/
163
164 const char *xbt_procname(void)
165 {
166   const char *res = NULL;
167   smx_process_t process = SIMIX_process_self();
168   if ((process != NULL) && (process->simdata))
169     res = SIMIX_process_get_name(process);
170   if (res)
171     return res;
172   else
173     return "";
174 }
175
176 int gras_os_getpid(void)
177 {
178
179   smx_process_t process = SIMIX_process_self();
180
181   if ((process != NULL) && (process->data))
182     return ((gras_procdata_t *) process->data)->pid;
183   else
184     return 0;
185 }
186
187
188 /** @brief retrieve the value of a given host property (or NULL if not defined) */
189 const char *gras_os_host_property_value(const char *name)
190 {
191   return
192     xbt_dict_get_or_null(SIMIX_host_get_properties
193                          (SIMIX_process_get_host(SIMIX_process_self())),
194                          name);
195 }
196
197 /** @brief retrieve the host properties dictionnary
198  *  @warning it's the original one, not a copy. Don't mess with it
199  */
200 xbt_dict_t gras_os_host_properties(void)
201 {
202   return
203     SIMIX_host_get_properties(SIMIX_process_get_host(SIMIX_process_self()));
204 }
205
206 /* **************************************************************************
207  * Interface with SIMIX
208  * (these functions are called by the stuff generated by gras_stub_generator)
209  * **************************************************************************/
210
211 XBT_LOG_EXTERNAL_CATEGORY(gras_trp);
212 XBT_LOG_EXTERNAL_CATEGORY(gras_trp_sg);
213
214 void gras_global_init(int *argc, char **argv)
215 {
216   XBT_LOG_CONNECT(gras_trp_sg, gras_trp);
217   SIMIX_global_init(argc, argv);
218 }
219
220 void gras_create_environment(const char *file)
221 {
222   SIMIX_create_environment(file);
223 }
224
225 void gras_function_register(const char *name, xbt_main_func_t code)
226 {
227   SIMIX_function_register(name, code);
228 }
229
230 void gras_main()
231 {
232   smx_cond_t cond = NULL;
233   smx_action_t action;
234   xbt_fifo_t actions_done = xbt_fifo_new();
235   xbt_fifo_t actions_failed = xbt_fifo_new();
236
237   /* Clean IO before the run */
238   fflush(stdout);
239   fflush(stderr);
240   SIMIX_init();
241
242   while (SIMIX_solve(actions_done, actions_failed) != -1.0) {
243     while ((action = xbt_fifo_pop(actions_failed))) {
244       DEBUG1("** %s failed **", action->name);
245       while ((cond = xbt_fifo_pop(action->cond_list))) {
246         SIMIX_cond_broadcast(cond);
247       }
248       /* action finished, destroy it */
249       //        SIMIX_action_destroy(action);
250     }
251
252     while ((action = xbt_fifo_pop(actions_done))) {
253       DEBUG1("** %s done **", action->name);
254       while ((cond = xbt_fifo_pop(action->cond_list))) {
255         SIMIX_cond_broadcast(cond);
256       }
257       /* action finished, destroy it */
258       //SIMIX_action_destroy(action);
259     }
260   }
261   xbt_fifo_free(actions_failed);
262   xbt_fifo_free(actions_done);
263   return;
264 }
265
266 void gras_launch_application(const char *file)
267 {
268   SIMIX_launch_application(file);
269 }
270
271 void gras_clean()
272 {
273   SIMIX_clean();
274 }