pcc.c 5.9 KB

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