1 #include <u.h>
2 #include <libc.h>
3
4
5 typedef struct Objtype {
6 char *name;
7 char *cc;
8 char *ld;
9 char *o;
10 char *oname;
11 } Objtype;
12
13 /* sync with /sys/src/ape/cmd/cc.c */
14 Objtype objtype[] = {
15 {"spim", "0c", "0l", "0", "0.out"},
16 {"mips64", "4c", "4l", "4", "4.out"},
17 {"arm", "5c", "5l", "5", "5.out"},
18 {"amd64", "6c", "6l", "6", "6.out"},
19 {"386", "8c", "8l", "8", "8.out"},
20 {"power64", "9c", "9l", "9", "9.out"},
21 {"sparc", "kc", "kl", "k", "k.out"},
22 {"power", "qc", "ql", "q", "q.out"},
23 {"mips", "vc", "vl", "v", "v.out"},
24 {"riscv", "ic", "il", "i", "i.out"},
25 {"riscv64", "jc", "jl", "j", "j.out"},
26 {"spim64", "xc", "xl", "x", "x.out"},
27 };
28 char *allos = "05689kqvij";
29
30 enum {
31 Nobjs = (sizeof objtype)/(sizeof objtype[0]),
32 Maxlist = 2000,
33 };
34
35 typedef struct List {
36 char *strings[Maxlist];
37 int n;
38 } List;
39
40 List srcs, objs, cpp, cc, ld, ldargs;
41 int cflag, vflag, Eflag, Pflag;
42
43 void append(List *, char *);
44 char *changeext(char *, char *);
45 void doexec(char *, List *);
46 void dopipe(char *, List *, char *, List *);
47 void fatal(char *);
48 Objtype *findoty(void);
49 void printlist(List *);
50
51 void
main(int argc,char * argv[])52 main(int argc, char *argv[])
53 {
54 Objtype *ot;
55 char *s, *suf, *ccpath;
56 char *oname;
57 int i, cppn, ccn, oflag;
58
59 oflag = 0;
60 ot = findoty();
61 oname = ot->oname;
62 append(&cpp, "cpp");
63 append(&cpp, "-D__STDC__=1"); /* ANSI says so */
64 append(&cpp, "-N"); /* turn off standard includes */
65 append(&cc, ot->cc);
66 append(&ld, ot->ld);
67 while(argc > 0) {
68 ARGBEGIN {
69 case '+':
70 append(&cpp, smprint("-%c", ARGC()));
71 break;
72 case 'c':
73 cflag = 1;
74 break;
75 case 'l':
76 append(&objs, smprint("/%s/lib/ape/lib%s.a", ot->name, ARGF()));
77 break;
78 case 'o':
79 oflag = 1;
80 oname = ARGF();
81 if(!oname)
82 fatal("no -o argument");
83 break;
84 case 'w':
85 case 'B':
86 case 'F':
87 case 'N':
88 case 'S':
89 case 'T':
90 case 'V':
91 case 'W':
92 append(&cc, smprint("-%c", ARGC()));
93 break;
94 case 's':
95 append(&cc, smprint("-s%s", ARGF()));
96 break;
97 case 'D':
98 case 'I':
99 case 'U':
100 append(&cpp, smprint("-%c%s", ARGC(), ARGF()));
101 break;
102 case 'v':
103 vflag = 1;
104 append(&ldargs, "-v");
105 break;
106 case 'P':
107 Pflag = 1;
108 cflag = 1;
109 break;
110 case 'E':
111 Eflag = 1;
112 cflag = 1;
113 break;
114 case 'p':
115 append(&ldargs, "-p");
116 break;
117 case 'f':
118 if(strcmp(ot->name, "arm") == 0)
119 append(&ldargs, "-f");
120 break;
121 case 'x':
122 s = ARGF();
123 if(s == nil || *s == '-')
124 fatal("no -x argument");
125 append(&ldargs, smprint("-x %s", s));
126 break;
127 case 'a':
128 /* hacky look inside ARGBEGIN insides, to see if we have -aa */
129 if(*_args == 'a') {
130 append(&cc, "-aa");
131 _args++;
132 } else
133 append(&cc, "-a");
134 cflag = 1;
135 break;
136 default:
137 fprint(2, "pcc: flag -%c ignored\n", ARGC());
138 break;
139 } ARGEND
140 if(argc > 0) {
141 s = argv[0];
142 suf = utfrrune(s, '.');
143 if(suf) {
144 suf++;
145 if(strcmp(suf, "c") == 0) {
146 append(&srcs, s);
147 append(&objs, changeext(s, ot->o));
148 } else if(strcmp(suf, ot->o) == 0 ||
149 strcmp(suf, "a") == 0 ||
150 (suf[0] == 'a' && strcmp(suf+1, ot->o) == 0)) {
151 append(&objs, s);
152 } else if(utfrune(allos, suf[0]) != 0) {
153 fprint(2, "pcc: argument %s ignored: wrong architecture\n",
154 s);
155 }
156 }
157 }
158 }
159 if(objs.n == 0)
160 fatal("no files to compile or load");
161 ccpath = smprint("/bin/%s", ot->cc);
162 append(&cpp, smprint("-I/%s/include/ape", ot->name));
163 append(&cpp, "-I/sys/include/ape");
164 cppn = cpp.n;
165 ccn = cc.n;
166 for(i = 0; i < srcs.n; i++) {
167 append(&cpp, srcs.strings[i]);
168 if(Pflag)
169 append(&cpp, changeext(objs.strings[i], "i"));
170 if(Eflag || Pflag)
171 doexec("/bin/cpp", &cpp);
172 else {
173 append(&cc, "-o");
174 if(oflag && cflag)
175 append(&cc, oname);
176 else
177 append(&cc, changeext(srcs.strings[i], ot->o));
178 dopipe("/bin/cpp", &cpp, ccpath, &cc);
179 }
180 cpp.n = cppn;
181 cc.n = ccn;
182 }
183 if(!cflag) {
184 append(&ld, "-o");
185 append(&ld, oname);
186 for(i = 0; i < ldargs.n; i++)
187 append(&ld, ldargs.strings[i]);
188 for(i = 0; i < objs.n; i++)
189 append(&ld, objs.strings[i]);
190 append(&ld, smprint("/%s/lib/ape/libap.a", ot->name));
191 doexec(smprint("/bin/%s", ot->ld), &ld);
192 if(objs.n == 1){
193 /* prevent removal of a library */
194 if(strstr(objs.strings[0], ".a") == 0)
195 remove(objs.strings[0]);
196 }
197 }
198
199 exits(0);
200 }
201
202 void
append(List * l,char * s)203 append(List *l, char *s)
204 {
205 if(l->n >= Maxlist-1)
206 fatal("too many arguments");
207 l->strings[l->n++] = s;
208 l->strings[l->n] = 0;
209 }
210
211 void
doexec(char * c,List * a)212 doexec(char *c, List *a)
213 {
214 Waitmsg *w;
215
216 if(vflag) {
217 printlist(a);
218 fprint(2, "\n");
219 }
220 switch(fork()) {
221 case -1:
222 fatal("fork failed");
223 case 0:
224 exec(c, a->strings);
225 fatal("exec failed");
226 }
227 w = wait();
228 if(w == nil)
229 fatal("wait failed");
230 if(w->msg[0])
231 fatal(smprint("%s: %s", a->strings[0], w->msg));
232 free(w);
233 }
234
235 void
dopipe(char * c1,List * a1,char * c2,List * a2)236 dopipe(char *c1, List *a1, char *c2, List *a2)
237 {
238 Waitmsg *w;
239 int pid1, got;
240 int fd[2];
241
242 if(vflag) {
243 printlist(a1);
244 fprint(2, " | ");
245 printlist(a2);
246 fprint(2, "\n");
247 }
248 if(pipe(fd) < 0)
249 fatal("pipe failed");
250 switch((pid1 = fork())) {
251 case -1:
252 fatal("fork failed");
253 case 0:
254 dup(fd[0], 0);
255 close(fd[0]);
256 close(fd[1]);
257 exec(c2, a2->strings);
258 fatal("exec failed");
259 }
260 switch(fork()) {
261 case -1:
262 fatal("fork failed");
263 case 0:
264 close(0);
265 dup(fd[1], 1);
266 close(fd[0]);
267 close(fd[1]);
268 exec(c1, a1->strings);
269 fatal("exec failed");
270 }
271 close(fd[0]);
272 close(fd[1]);
273 for(got = 0; got < 2; got++) {
274 w = wait();
275 if(w == nil)
276 fatal("wait failed");
277 if(w->msg[0])
278 fatal(smprint("%s: %s",
279 (w->pid == pid1) ? a1->strings[0] : a2->strings[0], w->msg));
280 free(w);
281 }
282 }
283
284 Objtype *
findoty(void)285 findoty(void)
286 {
287 char *o;
288 Objtype *oty;
289
290 o = getenv("objtype");
291 if(!o)
292 fatal("no $objtype in environment");
293 for(oty = objtype; oty < &objtype[Nobjs]; oty++)
294 if(strcmp(o, oty->name) == 0)
295 return oty;
296 fatal("unknown $objtype");
297 return 0; /* shut compiler up */
298 }
299
300 void
fatal(char * msg)301 fatal(char *msg)
302 {
303 fprint(2, "pcc: %s\n", msg);
304 exits(msg);
305 }
306
307 /* src ends in .something; return copy of basename with .ext added */
308 char *
changeext(char * src,char * ext)309 changeext(char *src, char *ext)
310 {
311 char *b, *e, *ans;
312
313 b = utfrrune(src, '/');
314 if(b)
315 b++;
316 else
317 b = src;
318 e = utfrrune(src, '.');
319 if(!e)
320 return 0;
321 *e = 0;
322 ans = smprint("%s.%s", b, ext);
323 *e = '.';
324 return ans;
325 }
326
327 void
printlist(List * l)328 printlist(List *l)
329 {
330 int i;
331
332 for(i = 0; i < l->n; i++) {
333 fprint(2, "%s", l->strings[i]);
334 if(i < l->n - 1)
335 fprint(2, " ");
336 }
337 }
338