pcc.c 5.8 KB

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