Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
More work on SG, almost done
[simgrid.git] / src / gras / Virtu / sg_process.c
1 /* $Id$ */
2
3 /* process_sg - GRAS process handling on simulator                          */
4
5 /* Authors: Martin Quinson                                                  */
6 /* Copyright (C) 2003,2004 da GRAS posse.                                   */
7
8 /* This program is free software; you can redistribute it and/or modify it
9    under the terms of the license (GNU LGPL) which comes with this package. */
10
11 #include "Virtu/virtu_sg.h"
12
13 GRAS_LOG_NEW_DEFAULT_SUBCATEGORY(process,GRAS);
14
15 gras_error_t
16 gras_process_init() {
17   gras_error_t errcode;
18   gras_hostdata_t *hd=(gras_hostdata_t *)MSG_host_get_data(MSG_host_self());
19   gras_procdata_t *pd;
20   int i;
21   
22   if (!(pd=(gras_procdata_t *)malloc(sizeof(gras_procdata_t)))) 
23     RAISE_MALLOC;
24
25   if (MSG_process_set_data(MSG_process_self(),(void*)pd) != MSG_OK) {
26     return unknown_error;
27   }
28   TRY(gras_procdata_init());
29
30   if (!hd) {
31     if (!(hd=(gras_hostdata_t *)malloc(sizeof(gras_hostdata_t)))) 
32       RAISE_MALLOC;
33
34     TRY(gras_dynar_new(&(hd->ports),sizeof(gras_sg_portrec_t),NULL));
35
36     memset(hd->proc, 0, sizeof(hd->proc[0]) * GRAS_MAX_CHANNEL); 
37
38     if (MSG_host_set_data(MSG_host_self(),(void*)hd) != MSG_OK) {
39       return unknown_error;
40     }
41   }
42   
43   /* take a free channel for this process */
44   for (i=0; i<GRAS_MAX_CHANNEL && hd->proc[i]; i++);
45   if (i == GRAS_MAX_CHANNEL) 
46     RAISE2(system_error,
47            "GRAS: Can't add a new process on %s, because all channel are already in use. Please increase MAX CHANNEL (which is %d for now) and recompile GRAS\n.",
48             MSG_host_get_name(MSG_host_self()),GRAS_MAX_CHANNEL);
49
50   pd->chan = i;
51   hd->proc[ i ] = MSG_process_self_PID();
52
53   /* take a free RAW channel for this process */
54   for (i=0; i<GRAS_MAX_CHANNEL && hd->proc[i]; i++);
55   if (i == GRAS_MAX_CHANNEL) {
56     RAISE2(system_error,
57            "GRAS: Can't add a new process on %s, because all channel are already in use. Please increase MAX CHANNEL (which is %d for now) and recompile GRAS\n.",
58             MSG_host_get_name(MSG_host_self()),GRAS_MAX_CHANNEL);
59   }
60   pd->rawChan = i;
61
62   hd->proc[ i ] = MSG_process_self_PID();
63
64   /* Connect a dummy socket to ourselves. It's returned by select() */
65   TRY(gras_socket_client(MSG_host_get_name(MSG_host_self()),
66                          pd->chan, &(pd->sock)));
67
68   VERB2("Creating process '%s' (%d)",
69            MSG_process_get_name(MSG_process_self()),
70            MSG_process_self_PID());
71   return no_error;
72 }
73
74 gras_error_t
75 gras_process_exit() {
76   gras_hostdata_t *hd=(gras_hostdata_t *)MSG_host_get_data(MSG_host_self());
77   gras_procdata_t *pd=gras_procdata_get();
78   int myPID=MSG_process_self_PID();
79   int cpt;
80   gras_sg_portrec_t pr;
81
82   gras_assert0(hd && pd,"Run gras_process_init!!");
83
84   INFO2("GRAS: Finalizing process '%s' (%d)",
85         MSG_process_get_name(MSG_process_self()),MSG_process_self_PID());
86
87   if (gras_dynar_length(pd->msg_queue))
88     WARN1("process %d terminated, but some messages are still queued",
89           MSG_process_self_PID());
90
91   for (cpt=0; cpt< GRAS_MAX_CHANNEL; cpt++)
92     if (myPID == hd->proc[cpt])
93       hd->proc[cpt] = 0;
94
95   gras_dynar_foreach(hd->ports, cpt, pr) {
96     if (pr.port == pd->chan || pr.port == pd->rawChan) {
97       gras_dynar_cursor_rm(hd->ports, &cpt);
98     }
99   }
100
101   return no_error;
102 }
103
104 /* **************************************************************************
105  * Process data
106  * **************************************************************************/
107
108 gras_procdata_t *gras_procdata_get(void) {
109   gras_procdata_t *pd=
110     (gras_procdata_t *)MSG_process_get_data(MSG_process_self());
111
112   gras_assert0(pd,"Run gras_process_init!");
113
114   return pd;
115 }
116