1 #include	"mk.h"
2 
3 enum {
4 	ENVQUANTA=10
5 };
6 
7 Envy	*envy;
8 static int nextv;
9 
10 static char	*myenv[] =
11 {
12 	"target",
13 	"stem",
14 	"prereq",
15 	"pid",
16 	"nproc",
17 	"newprereq",
18 	"alltarget",
19 	"newmember",
20 	"stem0",		/* must be in order from here */
21 	"stem1",
22 	"stem2",
23 	"stem3",
24 	"stem4",
25 	"stem5",
26 	"stem6",
27 	"stem7",
28 	"stem8",
29 	"stem9",
30 	0
31 };
32 
33 void
34 initenv(void)
35 {
36 	char **p;
37 
38 	for(p = myenv; *p; p++)
39 		symlook(*p, S_INTERNAL, (void *)"");
40 	readenv();				/* o.s. dependent */
41 }
42 
43 static void
44 envinsert(char *name, Word *value)
45 {
46 	static int envsize;
47 
48 	if (nextv >= envsize) {
49 		envsize += ENVQUANTA;
50 		envy = (Envy *) Realloc((char *) envy, envsize*sizeof(Envy));
51 	}
52 	envy[nextv].name = name;
53 	envy[nextv++].values = value;
54 }
55 
56 static void
57 envupd(char *name, Word *value)
58 {
59 	Envy *e;
60 
61 	for(e = envy; e->name; e++)
62 		if(strcmp(name, e->name) == 0){
63 			delword(e->values);
64 			e->values = value;
65 			return;
66 		}
67 	e->name = name;
68 	e->values = value;
69 	envinsert(0,0);
70 }
71 
72 static void
73 ecopy(Symtab *s)
74 {
75 	char **p;
76 
77 	if(symlook(s->name, S_NOEXPORT, 0))
78 		return;
79 	for(p = myenv; *p; p++)
80 		if(strcmp(*p, s->name) == 0)
81 			return;
82 	envinsert(s->name, s->u.ptr);
83 }
84 
85 void
86 execinit(void)
87 {
88 	char **p;
89 
90 	nextv = 0;
91 	for(p = myenv; *p; p++)
92 		envinsert(*p, stow(""));
93 
94 	symtraverse(S_VAR, ecopy);
95 	envinsert(0, 0);
96 }
97 
98 Envy*
99 buildenv(Job *j, int slot)
100 {
101 	char **p, *cp, *qp;
102 	Word *w, *v, **l;
103 	int i;
104 	char buf[256];
105 
106 	envupd("target", wdup(j->t));
107 	if(j->r->attr&REGEXP)
108 		envupd("stem",newword(""));
109 	else
110 		envupd("stem", newword(j->stem));
111 	envupd("prereq", wdup(j->p));
112 	sprint(buf, "%d", getpid());
113 	envupd("pid", newword(buf));
114 	sprint(buf, "%d", slot);
115 	envupd("nproc", newword(buf));
116 	envupd("newprereq", wdup(j->np));
117 	envupd("alltarget", wdup(j->at));
118 	l = &v;
119 	v = w = wdup(j->np);
120 	while(w){
121 		cp = strchr(w->s, '(');
122 		if(cp){
123 			qp = strchr(cp+1, ')');
124 			if(qp){
125 				*qp = 0;
126 				/* strcpy, but might overlap */
127 				memmove(w->s, cp+1, strlen(cp+1)+1);
128 				l = &w->next;
129 				w = w->next;
130 				continue;
131 			}
132 		}
133 		*l = w->next;
134 		free(w->s);
135 		free(w);
136 		w = *l;
137 	}
138 	envupd("newmember", v);
139 		/* update stem0 -> stem9 */
140 	for(p = myenv; *p; p++)
141 		if(strcmp(*p, "stem0") == 0)
142 			break;
143 	for(i = 0; *p; i++, p++){
144 		if((j->r->attr&REGEXP) && j->match[i])
145 			envupd(*p, newword(j->match[i]));
146 		else
147 			envupd(*p, newword(""));
148 	}
149 	return envy;
150 }