1 #include <u.h>
2 #include <libc.h>
3
4 /*
5 POSIX standard c89
6
7 standard options: -c, -D name[=val], -E (preprocess to stdout),
8 -g, -L dir, -o outfile, -O, -s, -U name
9 (and operands can have -l lib interspersed)
10
11 nonstandard but specified options: -S (assembly language left in .s),
12 -Wx,arg1[,arg2...] (pass arg(s) to phase x, where x is p (cpp)
13 0 (compiler), or l (loader)
14 nonstandard options: -v (echo real commands to stdout as they execute)
15 -A: turn on ANSI prototype warnings
16 */
17
18 typedef struct Objtype {
19 char *name;
20 char *cc;
21 char *ld;
22 char *o;
23 char *oname; /* compatibility with pcc.c; unused */
24 } Objtype;
25
26 /* sync with /sys/src/cmd/pcc.c */
27 Objtype objtype[] = {
28 {"spim", "0c", "0l", "0", "0.out"},
29 {"mips64", "4c", "4l", "4", "4.out"},
30 {"arm", "5c", "5l", "5", "5.out"},
31 {"amd64", "6c", "6l", "6", "6.out"},
32 {"386", "8c", "8l", "8", "8.out"},
33 {"power64", "9c", "9l", "9", "9.out"},
34 {"sparc", "kc", "kl", "k", "k.out"},
35 {"power", "qc", "ql", "q", "q.out"},
36 {"mips", "vc", "vl", "v", "v.out"},
37 {"riscv", "ic", "il", "i", "i.out"},
38 {"riscv64", "jc", "jl", "j", "j.out"},
39 {"spim64", "xc", "xl", "x", "x.out"},
40 };
41 char *allos = "05689kqv";
42
43 enum {
44 Nobjs = (sizeof objtype)/(sizeof objtype[0]),
45 Maxlist = 2000,
46 };
47
48 typedef struct List {
49 char *strings[Maxlist];
50 int n;
51 } List;
52
53 List srcs, objs, cpp, cc, ld, ldargs, srchlibs;
54 int cflag, vflag, Eflag, Sflag, Aflag;
55
56 void append(List *, char *);
57 char *changeext(char *, char *);
58 void doexec(char *, List *);
59 void dopipe(char *, List *, char *, List *);
60 void fatal(char *);
61 Objtype *findoty(void);
62 void printlist(List *);
63 char *searchlib(char *, char*);
64
65 void
main(int argc,char * argv[])66 main(int argc, char *argv[])
67 {
68 char *s, *suf, *ccpath, *lib;
69 char *oname;
70 int haveoname = 0;
71 int i, cppn, ccn;
72 Objtype *ot;
73
74 ot = findoty();
75 oname = "a.out";
76 append(&cpp, "cpp");
77 append(&cpp, "-D__STDC__=1"); /* ANSI says so */
78 append(&cpp, "-D_POSIX_SOURCE=");
79 append(&cpp, "-N"); /* turn off standard includes */
80 append(&cc, ot->cc);
81 append(&ld, ot->ld);
82 append(&srchlibs, smprint("/%s/lib/ape", ot->name));
83 while(argc > 0) {
84 ARGBEGIN {
85 case 'c':
86 cflag = 1;
87 break;
88 case 'l':
89 lib = searchlib(ARGF(), ot->name);
90 if(!lib)
91 fprint(2, "cc: can't find library for -l\n");
92 else
93 append(&objs, lib);
94 break;
95 case 'o':
96 oname = ARGF();
97 haveoname = 1;
98 if(!oname)
99 fatal("cc: no -o argument");
100 break;
101 case 'D':
102 case 'I':
103 case 'U':
104 append(&cpp, smprint("-%c%s", ARGC(), ARGF()));
105 break;
106 case 'E':
107 Eflag = 1;
108 cflag = 1;
109 break;
110 case 's':
111 case 'g':
112 break;
113 case 'L':
114 lib = ARGF();
115 if(!lib)
116 fprint(2, "cc: no -L argument\n");
117 else
118 append(&srchlibs, lib);
119 break;
120 case 'N':
121 case 'T':
122 case 'w':
123 append(&cc, smprint("-%c", ARGC()));
124 break;
125 case 'O':
126 break;
127 case 'W':
128 s = ARGF();
129 if(s && s[1]==',') {
130 switch (s[0]) {
131 case 'p':
132 append(&cpp, s+2);
133 break;
134 case '0':
135 append(&cc, s+2);
136 break;
137 case 'l':
138 append(&ldargs, s+2);
139 break;
140 default:
141 fprint(2, "cc: pass letter after -W should be one of p0l; ignored\n");
142 }
143 } else
144 fprint(2, "cc: bad option after -W; ignored\n");
145 break;
146 case 'v':
147 vflag = 1;
148 append(&ldargs, "-v");
149 break;
150 case 'A':
151 Aflag = 1;
152 break;
153 case 'S':
154 Sflag = 1;
155 break;
156 default:
157 fprint(2, "cc: flag -%c ignored\n", ARGC());
158 break;
159 } ARGEND
160 if(!Aflag) {
161 append(&cc, "-J"); /* old/new decl mixture hack */
162 append(&cc, "-B"); /* turn off non-prototype warnings */
163 }
164 if(argc > 0) {
165 s = argv[0];
166 suf = utfrrune(s, '.');
167 if(suf) {
168 suf++;
169 if(strcmp(suf, "c") == 0) {
170 append(&srcs, s);
171 append(&objs, changeext(s, "o"));
172 } else if(strcmp(suf, "o") == 0 ||
173 strcmp(suf, ot->o) == 0 ||
174 strcmp(suf, "a") == 0 ||
175 (suf[0] == 'a' && strcmp(suf+1, ot->o) == 0)) {
176 append(&objs, s);
177 } else if(utfrune(allos, suf[0]) != 0) {
178 fprint(2, "cc: argument %s ignored: wrong architecture\n",
179 s);
180 }
181 }
182 }
183 }
184 if(objs.n == 0)
185 fatal("no files to compile or load");
186 ccpath = smprint("/bin/%s", ot->cc);
187 append(&cpp, smprint("-I/%s/include/ape", ot->name));
188 append(&cpp, "-I/sys/include/ape");
189 cppn = cpp.n;
190 ccn = cc.n;
191 for(i = 0; i < srcs.n; i++) {
192 append(&cpp, srcs.strings[i]);
193 if(Eflag)
194 doexec("/bin/cpp", &cpp);
195 else {
196 if(Sflag)
197 append(&cc, "-S");
198 else {
199 append(&cc, "-o");
200 if (haveoname && cflag)
201 append(&cc, oname);
202 else
203 append(&cc, changeext(srcs.strings[i], "o"));
204 }
205 dopipe("/bin/cpp", &cpp, ccpath, &cc);
206 }
207 cpp.n = cppn;
208 cc.n = ccn;
209 }
210 if(!cflag) {
211 append(&ld, "-o");
212 append(&ld, oname);
213 for(i = 0; i < ldargs.n; i++)
214 append(&ld, ldargs.strings[i]);
215 for(i = 0; i < objs.n; i++)
216 append(&ld, objs.strings[i]);
217 append(&ld, smprint("/%s/lib/ape/libap.a", ot->name));
218 doexec(smprint("/bin/%s", ot->ld), &ld);
219 if(objs.n == 1)
220 remove(objs.strings[0]);
221 }
222
223 exits(0);
224 }
225
226 char *
searchlib(char * s,char * objtype)227 searchlib(char *s, char *objtype)
228 {
229 char *l;
230 int i;
231
232 if(!s)
233 return 0;
234 for(i = srchlibs.n-1; i>=0; i--) {
235 l = smprint("%s/lib%s.a", srchlibs.strings[i], s);
236 if(access(l, 0) >= 0)
237 return l;
238 }
239 if(s[1] == 0)
240 switch(s[0]) {
241 case 'c':
242 l = smprint("/%s/lib/ape/libap.a", objtype);
243 break;
244 case 'm':
245 l = smprint("/%s/lib/ape/libap.a", objtype);
246 break;
247 case 'l':
248 l = smprint("/%s/lib/ape/libl.a", objtype);
249 break;
250 case 'y':
251 l = smprint("/%s/lib/ape/liby.a", objtype);
252 break;
253 default:
254 l = 0;
255 }
256 else
257 l = 0;
258 return l;
259 }
260
261 void
append(List * l,char * s)262 append(List *l, char *s)
263 {
264 if(l->n >= Maxlist-1)
265 fatal("too many arguments");
266 l->strings[l->n++] = s;
267 l->strings[l->n] = 0;
268 }
269
270 void
doexec(char * c,List * a)271 doexec(char *c, List *a)
272 {
273 Waitmsg *w;
274
275 if(vflag) {
276 printlist(a);
277 fprint(2, "\n");
278 }
279 switch(fork()) {
280 case -1:
281 fatal("fork failed");
282 case 0:
283 exec(c, a->strings);
284 fatal("exec failed");
285 }
286 if((w = wait()) == nil)
287 fatal("wait failed");
288 if(w->msg[0])
289 fatal(smprint("%s: %s", a->strings[0], w->msg));
290 free(w);
291 }
292
293 void
dopipe(char * c1,List * a1,char * c2,List * a2)294 dopipe(char *c1, List *a1, char *c2, List *a2)
295 {
296 Waitmsg *w;
297 int pid1, got;
298 int fd[2];
299
300 if(vflag) {
301 printlist(a1);
302 fprint(2, " | ");
303 printlist(a2);
304 fprint(2, "\n");
305 }
306 if(pipe(fd) < 0)
307 fatal("pipe failed");
308 switch((pid1 = fork())) {
309 case -1:
310 fatal("fork failed");
311 case 0:
312 dup(fd[0], 0);
313 close(fd[0]);
314 close(fd[1]);
315 exec(c2, a2->strings);
316 fatal("exec failed");
317 }
318 switch(fork()) {
319 case -1:
320 fatal("fork failed");
321 case 0:
322 close(0);
323 dup(fd[1], 1);
324 close(fd[0]);
325 close(fd[1]);
326 exec(c1, a1->strings);
327 fatal("exec failed");
328 }
329 close(fd[0]);
330 close(fd[1]);
331 for(got = 0; got < 2; got++) {
332 if((w = wait()) == nil)
333 fatal("wait failed");
334 if(w->msg[0])
335 fatal(smprint("%s: %s", (w->pid == pid1) ? a1->strings[0] : a2->strings[0], w->msg));
336 free(w);
337 }
338 }
339
340 Objtype *
findoty(void)341 findoty(void)
342 {
343 char *o;
344 Objtype *oty;
345
346 o = getenv("objtype");
347 if(!o)
348 fatal("no $objtype in environment");
349 for(oty = objtype; oty < &objtype[Nobjs]; oty++)
350 if(strcmp(o, oty->name) == 0)
351 return oty;
352 fatal("unknown $objtype");
353 return 0; /* shut compiler up */
354 }
355
356 void
fatal(char * msg)357 fatal(char *msg)
358 {
359 fprint(2, "cc: %s\n", msg);
360 exits(msg);
361 }
362
363 /* src ends in .something; return copy of basename with .ext added */
364 char *
changeext(char * src,char * ext)365 changeext(char *src, char *ext)
366 {
367 char *b, *e, *ans;
368
369 b = utfrrune(src, '/');
370 if(b)
371 b++;
372 else
373 b = src;
374 e = utfrrune(src, '.');
375 if(!e)
376 return 0;
377 *e = 0;
378 ans = smprint("%s.%s", b, ext);
379 *e = '.';
380 return ans;
381 }
382
383 void
printlist(List * l)384 printlist(List *l)
385 {
386 int i;
387
388 for(i = 0; i < l->n; i++) {
389 fprint(2, "%s", l->strings[i]);
390 if(i < l->n - 1)
391 fprint(2, " ");
392 }
393 }
394