Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
44ca9d5bf267c637fb9516ba0fe6775a2c4a7092
[simgrid.git] / src / gras / Virtu / sg_process.c
1 /* process_sg - GRAS process handling on simulator                          */
2
3 /* Copyright (c) 2004, 2005, 2006, 2007, 2008, 2009, 2010. The SimGrid Team.
4  * All rights reserved.                                                     */
5
6 /* This program is free software; you can redistribute it and/or modify it
7  * under the terms of the license (GNU LGPL) which comes with this package. */
8
9 #include "xbt/ex.h"
10 #include "xbt/dict.h"
11 #include "gras_modinter.h"      /* module initialization interface */
12 #include "gras/Virtu/virtu_sg.h"
13 #include "gras/Msg/msg_interface.h"     /* For some checks at simulation end */
14 #include "gras/Transport/transport_interface.h" /* For some checks at simulation end */
15 #if HAVE_LUA
16 #include <lua.h>
17 #include <lauxlib.h>
18 #include <lualib.h>
19 #endif
20 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_virtu_process);
21
22 static long int PID = 1;
23
24
25 void gras_agent_spawn(const char *name,
26                       xbt_main_func_t code, int argc, char *argv[],
27                       xbt_dict_t properties)
28 {
29
30   smx_process_t process;
31   simcall_process_create(&process, name, code, NULL,
32                            gras_os_myname(), argc, argv, properties);
33 }
34
35 /* **************************************************************************
36  * Process constructor/destructor (semi-public interface)
37  * **************************************************************************/
38
39 void gras_process_init()
40 {
41   smx_process_t self = SIMIX_process_self();
42   gras_hostdata_t *hd =
43       (gras_hostdata_t *) SIMIX_host_self_get_data();
44   gras_procdata_t *pd = xbt_new0(gras_procdata_t, 1);
45   gras_trp_procdata_t trp_pd;
46   long int pid = PID++; /* make sure the first process gets the first id */
47
48   if (!hd) {
49     /* First process on this host (FIXME: does not work if the SIMIX user contexts are truly parallel) */
50     hd = xbt_new(gras_hostdata_t, 1);
51     hd->refcount = 1;
52     hd->ports = xbt_dynar_new(sizeof(gras_sg_portrec_t), NULL);
53     SIMIX_host_self_set_data((void *) hd);
54   } else {
55     hd->refcount++;
56   }
57
58   SIMIX_process_self_set_data(self, (void *) pd);
59   gras_procdata_init();
60
61   trp_pd = (gras_trp_procdata_t) gras_libdata_by_name("gras_trp");
62   pd->pid = pid;
63
64   if (self != NULL) {
65     pd->ppid = gras_os_getpid();
66   } else
67     pd->ppid = -1;
68
69   trp_pd->msg_selectable_sockets = xbt_queue_new(0, sizeof(xbt_socket_t));
70
71   trp_pd->meas_selectable_sockets =
72       xbt_queue_new(0, sizeof(xbt_socket_t));
73
74   XBT_VERB("Creating process '%s' (%d)", SIMIX_process_self_get_name(),
75       gras_os_getpid());
76 }
77
78 void gras_process_exit()
79 {
80   xbt_dynar_t sockets =
81       ((gras_trp_procdata_t) gras_libdata_by_name("gras_trp"))->sockets;
82   xbt_socket_t sock_iter;
83   unsigned int cursor;
84   gras_hostdata_t *hd =
85       (gras_hostdata_t *) SIMIX_host_self_get_data();
86   gras_procdata_t *pd =
87       (gras_procdata_t *) simcall_process_get_data(SIMIX_process_self());
88
89   gras_msg_procdata_t msg_pd =
90       (gras_msg_procdata_t) gras_libdata_by_name("gras_msg");
91   gras_trp_procdata_t trp_pd =
92       (gras_trp_procdata_t) gras_libdata_by_name("gras_trp");
93
94   xbt_queue_free(&trp_pd->msg_selectable_sockets);
95
96   xbt_queue_free(&trp_pd->meas_selectable_sockets);
97
98
99   xbt_assert(hd, "Run gras_process_init (ie, gras_init)!!");
100
101   XBT_VERB("GRAS: Finalizing process '%s' (%d)",
102         simcall_process_get_name(SIMIX_process_self()), gras_os_getpid());
103
104   if (!xbt_dynar_is_empty(msg_pd->msg_queue)) {
105     unsigned int cpt;
106     s_gras_msg_t msg;
107     XBT_WARN
108         ("process %d terminated, but %lu messages are still queued. Message list:",
109          gras_os_getpid(), xbt_dynar_length(msg_pd->msg_queue));
110     xbt_dynar_foreach(msg_pd->msg_queue, cpt, msg) {
111       XBT_WARN("   Message %s (%s) from %s@%s:%d", msg.type->name,
112             e_gras_msg_kind_names[msg.kind],
113             xbt_socket_peer_proc(msg.expe),
114             xbt_socket_peer_name(msg.expe),
115             xbt_socket_peer_port(msg.expe));
116     }
117   }
118
119   /* if each process has its sockets list, we need to close them when the
120      process finish */
121   xbt_dynar_foreach(sockets, cursor, sock_iter) {
122     XBT_VERB("Closing the socket %p left open on exit. Maybe a socket leak?",
123           sock_iter);
124     gras_socket_close(sock_iter);
125   }
126   if (!--(hd->refcount)) {
127     xbt_dynar_free(&hd->ports);
128     free(hd);
129   }
130   gras_procdata_exit();
131   free(pd);
132 }
133
134 /* **************************************************************************
135  * Process data (public interface)
136  * **************************************************************************/
137
138 gras_procdata_t *gras_procdata_get(void)
139 {
140   gras_procdata_t *pd =
141       (gras_procdata_t *) simcall_process_get_data(SIMIX_process_self());
142
143   xbt_assert(pd, "Run gras_process_init! (ie, gras_init)");
144
145   return pd;
146 }
147
148 void *gras_libdata_by_name_from_remote(const char *name, smx_process_t p)
149 {
150   gras_procdata_t *pd = (gras_procdata_t *) simcall_process_get_data(p);
151
152   xbt_assert(pd,
153               "process '%s' on '%s' didn't run gras_process_init! (ie, gras_init)",
154               simcall_process_get_name(p),
155               simcall_host_get_name(simcall_process_get_host(p)));
156
157   return gras_libdata_by_name_from_procdata(name, pd);
158 }
159
160 /** @brief retrieve the value of a given process property (or NULL if not defined) */
161 const char *gras_process_property_value(const char *name)
162 {
163   return xbt_dict_get_or_null(simcall_process_get_properties
164                              (SIMIX_process_self()), name);
165 }
166
167 /** @brief retrieve the process properties dictionnary
168  *  @warning it's the original one, not a copy. Don't mess with it
169  */
170 xbt_dict_t gras_process_properties(void)
171 {
172   return simcall_process_get_properties(SIMIX_process_self());
173 }
174
175 /* **************************************************************************
176  * OS virtualization function
177  * **************************************************************************/
178
179
180 int gras_os_getpid(void)
181 {
182   gras_procdata_t *data;
183   data = (gras_procdata_t *) SIMIX_process_self_get_data(SIMIX_process_self());
184   if (data != NULL)
185     return data->pid;
186
187   return 0;
188 }
189
190 /** @brief retrieve the value of a given host property (or NULL if not defined) */
191 const char *gras_os_host_property_value(const char *name)
192 {
193   return
194       xbt_dict_get_or_null(simcall_host_get_properties
195                            (simcall_process_get_host(SIMIX_process_self())),
196                            name);
197 }
198
199 /** @brief retrieve the host properties dictionary
200  *  @warning it's the original one, not a copy. Don't mess with it
201  */
202 xbt_dict_t gras_os_host_properties(void)
203 {
204   return
205       simcall_host_get_properties(simcall_process_get_host
206                                 (SIMIX_process_self()));
207 }
208
209 /* **************************************************************************
210  * Interface with SIMIX
211  * (these functions are called by the stuff generated by gras_stub_generator)
212  * **************************************************************************/
213
214 void gras_global_init(int *argc, char **argv)
215 {
216   SIMIX_global_init(argc, argv);
217 }
218
219 void gras_create_environment(const char *file)
220 {
221   SIMIX_create_environment(file);
222 }
223
224 void gras_function_register(const char *name, xbt_main_func_t code)
225 {
226   SIMIX_function_register(name, code);
227 }
228
229 void gras_function_register_default(xbt_main_func_t code)
230 {
231   SIMIX_function_register_default(code);
232 }
233
234 void gras_main()
235 {
236   /* Clean IO before the run */
237   fflush(stdout);
238   fflush(stderr);
239   SIMIX_run();
240
241   return;
242 }
243
244 void gras_launch_application(const char *file)
245 {
246   SIMIX_launch_application(file);
247 }
248
249 void gras_load_environment_script(const char *script_file)
250 {
251 #ifdef HAVE_LUA
252   lua_State *L = lua_open();
253   luaL_openlibs(L);
254
255   if (luaL_loadfile(L, script_file) || lua_pcall(L, 0, 0, 0)) {
256     printf("error: %s\n", lua_tostring(L, -1));
257     return;
258   }
259 #else
260   xbt_die
261       ("Lua is not available!! to call gras_load_environment_script, lua should be available...");
262 #endif
263   return;
264 }
265
266 void gras_clean()
267 {
268   SIMIX_clean();
269 }