pcc.c 5.9 KB

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