Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Catch up with lastest API breakage (xbt_ex_free)
[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 "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
16 XBT_LOG_EXTERNAL_CATEGORY(process);
17 XBT_LOG_DEFAULT_CATEGORY(process);
18
19 void
20 gras_process_init() {
21   gras_hostdata_t *hd=(gras_hostdata_t *)MSG_host_get_data(MSG_host_self());
22   gras_procdata_t *pd=xbt_new(gras_procdata_t,1);
23   gras_trp_procdata_t trp_pd;
24   gras_sg_portrec_t prmeas,pr;
25   int i;
26   
27   if (MSG_process_set_data(MSG_process_self(),(void*)pd) != MSG_OK)
28     THROW0(system_error,0,"Error in MSG_process_set_data()");
29    
30   gras_procdata_init();
31
32   if (!hd) {
33     /* First process on this host */
34     hd=xbt_new(gras_hostdata_t,1);
35     hd->refcount = 1;
36     hd->ports = xbt_dynar_new(sizeof(gras_sg_portrec_t),NULL);
37
38     memset(hd->proc, 0, sizeof(hd->proc[0]) * XBT_MAX_CHANNEL); 
39
40     if (MSG_host_set_data(MSG_host_self(),(void*)hd) != MSG_OK)
41       THROW0(system_error,0,"Error in MSG_host_set_data()");
42   } else {
43     hd->refcount++;
44   }
45   
46   /* take a free channel for this process */
47   trp_pd = (gras_trp_procdata_t)gras_libdata_by_name("gras_trp");
48   for (i=0; i<XBT_MAX_CHANNEL && hd->proc[i]; i++);
49   if (i == XBT_MAX_CHANNEL) 
50     THROW2(system_error,0,
51            "Can't add a new process on %s, because all channels are already in use. Please increase MAX CHANNEL (which is %d for now) and recompile GRAS.",
52             MSG_host_get_name(MSG_host_self()),XBT_MAX_CHANNEL);
53
54   trp_pd->chan = i;
55   hd->proc[ i ] = MSG_process_self_PID();
56
57   /* regiter it to the ports structure */
58   pr.port = -1;
59   pr.tochan = i;
60   pr.meas = 0;
61   xbt_dynar_push(hd->ports,&pr);
62
63   /* take a free meas channel for this process */
64   for (i=0; i<XBT_MAX_CHANNEL && hd->proc[i]; i++);
65   if (i == XBT_MAX_CHANNEL) {
66     THROW2(system_error,0,
67            "Can't add a new process on %s, because all channels are already in use. Please increase MAX CHANNEL (which is %d for now) and recompile GRAS.",
68             MSG_host_get_name(MSG_host_self()),XBT_MAX_CHANNEL);
69   }
70   trp_pd->measChan = i;
71
72   hd->proc[ i ] = MSG_process_self_PID();
73
74   /* register it to the ports structure */
75   prmeas.port = -1;
76   prmeas.tochan = i;
77   prmeas.meas = 1;
78   xbt_dynar_push(hd->ports,&prmeas);
79
80   VERB2("Creating process '%s' (%d)",
81            MSG_process_get_name(MSG_process_self()),
82            MSG_process_self_PID());
83 }
84
85 void
86 gras_process_exit() {
87   gras_hostdata_t *hd=(gras_hostdata_t *)MSG_host_get_data(MSG_host_self());
88   gras_procdata_t *pd=(gras_procdata_t*)MSG_process_get_data(MSG_process_self());
89
90   gras_msg_procdata_t msg_pd=(gras_msg_procdata_t)gras_libdata_by_name("gras_msg");
91   gras_trp_procdata_t trp_pd=(gras_trp_procdata_t)gras_libdata_by_name("gras_trp");
92   int myPID=MSG_process_self_PID();
93   int cpt;
94   gras_sg_portrec_t pr;
95
96   xbt_assert0(hd,"Run gras_process_init (ie, gras_init)!!");
97
98   VERB2("GRAS: Finalizing process '%s' (%d)",
99         MSG_process_get_name(MSG_process_self()),MSG_process_self_PID());
100
101   if (xbt_dynar_length(msg_pd->msg_queue))
102     WARN1("process %d terminated, but some messages are still queued",
103           MSG_process_self_PID());
104
105   for (cpt=0; cpt< XBT_MAX_CHANNEL; cpt++)
106     if (myPID == hd->proc[cpt])
107       hd->proc[cpt] = 0;
108
109   xbt_dynar_foreach(hd->ports, cpt, pr) {
110     if (pr.port == trp_pd->chan || pr.port == trp_pd->measChan) {
111       xbt_dynar_cursor_rm(hd->ports, &cpt);
112     }
113   }
114
115   if ( ! --(hd->refcount)) {
116     xbt_dynar_free(&hd->ports);
117     free(hd);
118   }
119   gras_procdata_exit();
120   free(pd);
121 }
122
123 /* **************************************************************************
124  * Process data
125  * **************************************************************************/
126
127 gras_procdata_t *gras_procdata_get(void) {
128   gras_procdata_t *pd=
129     (gras_procdata_t *)MSG_process_get_data(MSG_process_self());
130
131   xbt_assert0(pd,"Run gras_process_init! (ie, gras_init)");
132
133   return pd;
134 }
135
136 const char* xbt_procname(void) {
137   const char *res = NULL;
138   m_process_t process = MSG_process_self();
139   if ((process != NULL) && (process->simdata))
140     res = MSG_process_get_name(process);
141   if (res) 
142     return res;
143   else
144     return "";
145 }
146
147 int gras_os_getpid(void) {
148   m_process_t process = MSG_process_self();
149   if ((process != NULL) && (process->simdata))
150     return MSG_process_get_PID(MSG_process_self());
151   else
152     return 0;
153 }