obj.c 25 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465
  1. #define EXTERN
  2. #include "l.h"
  3. #include <ar.h>
  4. #ifndef DEFAULT
  5. #define DEFAULT '9'
  6. #endif
  7. char *noname = "<none>";
  8. char symname[] = SYMDEF;
  9. char thechar = 'q';
  10. char *thestring = "power";
  11. /*
  12. * -H0 -T0x200000 -R0 is boot
  13. * -H1 -T0x100000 -R4 is Be boot
  14. * -H2 -T4128 -R4096 is plan9 format
  15. * -H3 -T0x02010000 -D0x00001000 is raw
  16. * -H4 -T0x1000200 -D0x20000e00 -R4 is aix xcoff executable
  17. * -H5 -T0x80010000 -t0x10000 ELF, phys = 10000, vaddr = 0x8001...
  18. */
  19. static int
  20. isobjfile(char *f)
  21. {
  22. int n, v;
  23. Biobuf *b;
  24. char buf1[5], buf2[SARMAG];
  25. b = Bopen(f, OREAD);
  26. if(b == nil)
  27. return 0;
  28. n = Bread(b, buf1, 5);
  29. if(n == 5 && (buf1[2] == 1 && buf1[3] == '<' || buf1[3] == 1 && buf1[4] == '<'))
  30. v = 1; /* good enough for our purposes */
  31. else{
  32. Bseek(b, 0, 0);
  33. n = Bread(b, buf2, SARMAG);
  34. v = n == SARMAG && strncmp(buf2, ARMAG, SARMAG) == 0;
  35. }
  36. Bterm(b);
  37. return v;
  38. }
  39. void
  40. main(int argc, char *argv[])
  41. {
  42. int c;
  43. char *a;
  44. Binit(&bso, 1, OWRITE);
  45. cout = -1;
  46. listinit();
  47. outfile = 0;
  48. nerrors = 0;
  49. curtext = P;
  50. HEADTYPE = -1;
  51. INITTEXT = -1;
  52. INITDAT = -1;
  53. INITRND = -1;
  54. INITENTRY = 0;
  55. ARGBEGIN {
  56. default:
  57. c = ARGC();
  58. if(c >= 0 && c < sizeof(debug))
  59. debug[c]++;
  60. break;
  61. case 'o':
  62. outfile = ARGF();
  63. break;
  64. case 'E':
  65. a = ARGF();
  66. if(a)
  67. INITENTRY = a;
  68. break;
  69. case 'T':
  70. a = ARGF();
  71. if(a)
  72. INITTEXT = atolwhex(a);
  73. break;
  74. case 'D':
  75. a = ARGF();
  76. if(a)
  77. INITDAT = atolwhex(a);
  78. break;
  79. case 'R':
  80. a = ARGF();
  81. if(a)
  82. INITRND = atolwhex(a);
  83. break;
  84. case 'H':
  85. a = ARGF();
  86. if(a)
  87. HEADTYPE = atolwhex(a);
  88. break;
  89. case 'x': /* produce export table */
  90. doexp = 1;
  91. if(argv[1] != nil && argv[1][0] != '-' && !isobjfile(argv[1]))
  92. readundefs(ARGF(), SEXPORT);
  93. break;
  94. case 'u': /* produce dynamically loadable module */
  95. dlm = 1;
  96. if(argv[1] != nil && argv[1][0] != '-' && !isobjfile(argv[1]))
  97. readundefs(ARGF(), SIMPORT);
  98. break;
  99. } ARGEND
  100. USED(argc);
  101. if(*argv == 0) {
  102. diag("usage: ql [-options] objects");
  103. errorexit();
  104. }
  105. if(!debug['9'] && !debug['U'] && !debug['B'])
  106. debug[DEFAULT] = 1;
  107. r0iszero = debug['0'] == 0;
  108. if(HEADTYPE == -1) {
  109. if(debug['U'])
  110. HEADTYPE = 0;
  111. if(debug['B'])
  112. HEADTYPE = 1;
  113. if(debug['9'])
  114. HEADTYPE = 2;
  115. }
  116. switch(HEADTYPE) {
  117. default:
  118. diag("unknown -H option");
  119. errorexit();
  120. case 0: /* boot */
  121. HEADR = 32L;
  122. if(INITTEXT == -1)
  123. INITTEXT = 0x200000L;
  124. if(INITDAT == -1)
  125. INITDAT = 0;
  126. if(INITRND == -1)
  127. INITRND = 4096L;
  128. break;
  129. case 1: /* Be boot format (PEF) */
  130. HEADR = 208L;
  131. if(INITTEXT == -1)
  132. INITTEXT = 0x100000;
  133. if(INITDAT == -1)
  134. INITDAT = 0;
  135. if(INITRND == -1)
  136. INITRND = 4;
  137. break;
  138. case 2: /* plan 9 */
  139. HEADR = 32L;
  140. if(INITTEXT == -1)
  141. INITTEXT = 4128;
  142. if(INITDAT == -1)
  143. INITDAT = 0;
  144. if(INITRND == -1)
  145. INITRND = 4096;
  146. break;
  147. case 3: /* raw */
  148. HEADR = 0;
  149. if(INITTEXT == -1)
  150. INITTEXT = 4128;
  151. if(INITDAT == -1) {
  152. INITDAT = 0;
  153. INITRND = 4;
  154. }
  155. if(INITRND == -1)
  156. INITRND = 0;
  157. break;
  158. case 4: /* aix unix xcoff executable */
  159. HEADR = 20L+72L+3*40L;
  160. if(INITTEXT == -1)
  161. INITTEXT = 0x1000000L+HEADR;
  162. if(INITDAT == -1)
  163. INITDAT = 0x20000000;
  164. if(INITRND == -1)
  165. INITRND = 0;
  166. break;
  167. case 5: /* elf executable */
  168. HEADR = rnd(52L+3*32L, 16);
  169. if(INITTEXT == -1)
  170. INITTEXT = 0x00400000L+HEADR;
  171. if(INITDAT == -1)
  172. INITDAT = 0x10000000;
  173. if(INITRND == -1)
  174. INITRND = 0;
  175. break;
  176. }
  177. if(INITDAT != 0 && INITRND != 0)
  178. print("warning: -D0x%lux is ignored because of -R0x%lux\n",
  179. INITDAT, INITRND);
  180. if(debug['v'])
  181. Bprint(&bso, "HEADER = -H0x%x -T0x%lux -D0x%lux -R0x%lux\n",
  182. HEADTYPE, INITTEXT, INITDAT, INITRND);
  183. Bflush(&bso);
  184. zprg.as = AGOK;
  185. zprg.reg = NREG;
  186. zprg.from.name = D_NONE;
  187. zprg.from.type = D_NONE;
  188. zprg.from.reg = NREG;
  189. zprg.from3 = zprg.from;
  190. zprg.to = zprg.from;
  191. buildop();
  192. histgen = 0;
  193. textp = P;
  194. datap = P;
  195. pc = 0;
  196. dtype = 4;
  197. if(outfile == 0)
  198. outfile = "q.out";
  199. cout = create(outfile, 1, 0775);
  200. if(cout < 0) {
  201. diag("%s: cannot create", outfile);
  202. errorexit();
  203. }
  204. nuxiinit();
  205. version = 0;
  206. cbp = buf.cbuf;
  207. cbc = sizeof(buf.cbuf);
  208. firstp = prg();
  209. lastp = firstp;
  210. if(INITENTRY == 0) {
  211. INITENTRY = "_main";
  212. if(debug['p'])
  213. INITENTRY = "_mainp";
  214. if(!debug['l'])
  215. lookup(INITENTRY, 0)->type = SXREF;
  216. } else
  217. lookup(INITENTRY, 0)->type = SXREF;
  218. while(*argv)
  219. objfile(*argv++);
  220. if(!debug['l'])
  221. loadlib();
  222. firstp = firstp->link;
  223. if(firstp == P)
  224. goto out;
  225. if(doexp || dlm){
  226. EXPTAB = "_exporttab";
  227. zerosig(EXPTAB);
  228. zerosig("etext");
  229. zerosig("edata");
  230. zerosig("end");
  231. if(dlm){
  232. import();
  233. HEADTYPE = 2;
  234. INITTEXT = INITDAT = 0;
  235. INITRND = 8;
  236. INITENTRY = EXPTAB;
  237. }
  238. export();
  239. }
  240. patch();
  241. if(debug['p'])
  242. if(debug['1'])
  243. doprof1();
  244. else
  245. doprof2();
  246. dodata();
  247. follow();
  248. if(firstp == P)
  249. goto out;
  250. noops();
  251. span();
  252. asmb();
  253. undef();
  254. out:
  255. if(debug['v']) {
  256. Bprint(&bso, "%5.2f cpu time\n", cputime());
  257. Bprint(&bso, "%ld memory used\n", tothunk);
  258. Bprint(&bso, "%d sizeof adr\n", sizeof(Adr));
  259. Bprint(&bso, "%d sizeof prog\n", sizeof(Prog));
  260. }
  261. errorexit();
  262. }
  263. void
  264. loadlib(void)
  265. {
  266. int i;
  267. long h;
  268. Sym *s;
  269. loop:
  270. xrefresolv = 0;
  271. for(i=0; i<libraryp; i++) {
  272. if(debug['v'])
  273. Bprint(&bso, "%5.2f autolib: %s (from %s)\n", cputime(), library[i], libraryobj[i]);
  274. objfile(library[i]);
  275. }
  276. if(xrefresolv)
  277. for(h=0; h<nelem(hash); h++)
  278. for(s = hash[h]; s != S; s = s->link)
  279. if(s->type == SXREF)
  280. goto loop;
  281. }
  282. void
  283. errorexit(void)
  284. {
  285. Bflush(&bso);
  286. if(nerrors) {
  287. if(cout >= 0)
  288. remove(outfile);
  289. exits("error");
  290. }
  291. exits(0);
  292. }
  293. void
  294. objfile(char *file)
  295. {
  296. long off, esym, cnt, l;
  297. int f, work;
  298. Sym *s;
  299. char magbuf[SARMAG];
  300. char name[100], pname[150];
  301. struct ar_hdr arhdr;
  302. char *e, *start, *stop;
  303. if(file[0] == '-' && file[1] == 'l') {
  304. if(debug['9'])
  305. sprint(name, "/%s/lib/lib", thestring);
  306. else
  307. sprint(name, "/usr/%clib/lib", thechar);
  308. strcat(name, file+2);
  309. strcat(name, ".a");
  310. file = name;
  311. }
  312. if(debug['v'])
  313. Bprint(&bso, "%5.2f ldobj: %s\n", cputime(), file);
  314. Bflush(&bso);
  315. f = open(file, 0);
  316. if(f < 0) {
  317. diag("cannot open file: %s", file);
  318. errorexit();
  319. }
  320. l = read(f, magbuf, SARMAG);
  321. if(l != SARMAG || strncmp(magbuf, ARMAG, SARMAG)){
  322. /* load it as a regular file */
  323. l = seek(f, 0L, 2);
  324. seek(f, 0L, 0);
  325. ldobj(f, l, file);
  326. close(f);
  327. return;
  328. }
  329. l = read(f, &arhdr, SAR_HDR);
  330. if(l != SAR_HDR) {
  331. diag("%s: short read on archive file symbol header", file);
  332. goto out;
  333. }
  334. if(strncmp(arhdr.name, symname, strlen(symname))) {
  335. diag("%s: first entry not symbol header", file);
  336. goto out;
  337. }
  338. esym = SARMAG + SAR_HDR + atolwhex(arhdr.size);
  339. off = SARMAG + SAR_HDR;
  340. /*
  341. * just bang the whole symbol file into memory
  342. */
  343. seek(f, off, 0);
  344. cnt = esym - off;
  345. start = malloc(cnt + 10);
  346. cnt = read(f, start, cnt);
  347. if(cnt <= 0){
  348. close(f);
  349. return;
  350. }
  351. stop = &start[cnt];
  352. memset(stop, 0, 10);
  353. work = 1;
  354. while(work){
  355. if(debug['v'])
  356. Bprint(&bso, "%5.2f library pass: %s\n", cputime(), file);
  357. Bflush(&bso);
  358. work = 0;
  359. for(e = start; e < stop; e = strchr(e+5, 0) + 1) {
  360. s = lookup(e+5, 0);
  361. if(s->type != SXREF)
  362. continue;
  363. sprint(pname, "%s(%s)", file, s->name);
  364. if(debug['v'])
  365. Bprint(&bso, "%5.2f library: %s\n", cputime(), pname);
  366. Bflush(&bso);
  367. l = e[1] & 0xff;
  368. l |= (e[2] & 0xff) << 8;
  369. l |= (e[3] & 0xff) << 16;
  370. l |= (e[4] & 0xff) << 24;
  371. seek(f, l, 0);
  372. l = read(f, &arhdr, SAR_HDR);
  373. if(l != SAR_HDR)
  374. goto bad;
  375. if(strncmp(arhdr.fmag, ARFMAG, sizeof(arhdr.fmag)))
  376. goto bad;
  377. l = atolwhex(arhdr.size);
  378. ldobj(f, l, pname);
  379. if(s->type == SXREF) {
  380. diag("%s: failed to load: %s", file, s->name);
  381. errorexit();
  382. }
  383. work = 1;
  384. xrefresolv = 1;
  385. }
  386. }
  387. return;
  388. bad:
  389. diag("%s: bad or out of date archive", file);
  390. out:
  391. close(f);
  392. }
  393. int
  394. zaddr(uchar *p, Adr *a, Sym *h[])
  395. {
  396. int i, c;
  397. long l;
  398. Sym *s;
  399. Auto *u;
  400. c = p[2];
  401. if(c < 0 || c > NSYM){
  402. print("sym out of range: %d\n", c);
  403. p[0] = AEND+1;
  404. return 0;
  405. }
  406. a->type = p[0];
  407. a->reg = p[1];
  408. a->sym = h[c];
  409. a->name = p[3];
  410. c = 4;
  411. if(a->reg > NREG) {
  412. print("register out of range %d\n", a->reg);
  413. p[0] = AEND+1;
  414. return 0; /* force real diagnostic */
  415. }
  416. switch(a->type) {
  417. default:
  418. print("unknown type %d\n", a->type);
  419. p[0] = AEND+1;
  420. return 0; /* force real diagnostic */
  421. case D_NONE:
  422. case D_REG:
  423. case D_FREG:
  424. case D_CREG:
  425. case D_FPSCR:
  426. case D_MSR:
  427. case D_SREG:
  428. case D_OPT:
  429. break;
  430. case D_SPR:
  431. case D_DCR:
  432. case D_BRANCH:
  433. case D_OREG:
  434. case D_CONST:
  435. a->offset = p[4] | (p[5]<<8) |
  436. (p[6]<<16) | (p[7]<<24);
  437. c += 4;
  438. break;
  439. case D_SCONST:
  440. memmove(a->sval, p+4, NSNAME);
  441. c += NSNAME;
  442. break;
  443. case D_FCONST:
  444. a->ieee.l = p[4] | (p[5]<<8) |
  445. (p[6]<<16) | (p[7]<<24);
  446. a->ieee.h = p[8] | (p[9]<<8) |
  447. (p[10]<<16) | (p[11]<<24);
  448. c += 8;
  449. break;
  450. }
  451. s = a->sym;
  452. if(s == S)
  453. goto out;
  454. i = a->name;
  455. if(i != D_AUTO && i != D_PARAM)
  456. goto out;
  457. l = a->offset;
  458. for(u=curauto; u; u=u->link)
  459. if(u->sym == s)
  460. if(u->type == i) {
  461. if(u->aoffset > l)
  462. u->aoffset = l;
  463. goto out;
  464. }
  465. u = malloc(sizeof(Auto));
  466. u->link = curauto;
  467. curauto = u;
  468. u->sym = s;
  469. u->aoffset = l;
  470. u->type = i;
  471. out:
  472. return c;
  473. }
  474. void
  475. addlib(char *obj)
  476. {
  477. char name[1024], comp[256], *p;
  478. int i;
  479. if(histfrogp <= 0)
  480. return;
  481. if(histfrog[0]->name[1] == '/') {
  482. sprint(name, "");
  483. i = 1;
  484. } else
  485. if(histfrog[0]->name[1] == '.') {
  486. sprint(name, ".");
  487. i = 0;
  488. } else {
  489. if(debug['9'])
  490. sprint(name, "/%s/lib", thestring);
  491. else
  492. sprint(name, "/usr/%clib", thechar);
  493. i = 0;
  494. }
  495. for(; i<histfrogp; i++) {
  496. snprint(comp, sizeof comp, histfrog[i]->name+1);
  497. for(;;) {
  498. p = strstr(comp, "$O");
  499. if(p == 0)
  500. break;
  501. memmove(p+1, p+2, strlen(p+2)+1);
  502. p[0] = thechar;
  503. }
  504. for(;;) {
  505. p = strstr(comp, "$M");
  506. if(p == 0)
  507. break;
  508. if(strlen(comp)+strlen(thestring)-2+1 >= sizeof comp) {
  509. diag("library component too long");
  510. return;
  511. }
  512. memmove(p+strlen(thestring), p+2, strlen(p+2)+1);
  513. memmove(p, thestring, strlen(thestring));
  514. }
  515. if(strlen(name) + strlen(comp) + 3 >= sizeof(name)) {
  516. diag("library component too long");
  517. return;
  518. }
  519. strcat(name, "/");
  520. strcat(name, comp);
  521. }
  522. for(i=0; i<libraryp; i++)
  523. if(strcmp(name, library[i]) == 0)
  524. return;
  525. if(libraryp == nelem(library)){
  526. diag("too many autolibs; skipping %s", name);
  527. return;
  528. }
  529. p = malloc(strlen(name) + 1);
  530. strcpy(p, name);
  531. library[libraryp] = p;
  532. p = malloc(strlen(obj) + 1);
  533. strcpy(p, obj);
  534. libraryobj[libraryp] = p;
  535. libraryp++;
  536. }
  537. void
  538. addhist(long line, int type)
  539. {
  540. Auto *u;
  541. Sym *s;
  542. int i, j, k;
  543. u = malloc(sizeof(Auto));
  544. s = malloc(sizeof(Sym));
  545. s->name = malloc(2*(histfrogp+1) + 1);
  546. u->sym = s;
  547. u->type = type;
  548. u->aoffset = line;
  549. u->link = curhist;
  550. curhist = u;
  551. j = 1;
  552. for(i=0; i<histfrogp; i++) {
  553. k = histfrog[i]->value;
  554. s->name[j+0] = k>>8;
  555. s->name[j+1] = k;
  556. j += 2;
  557. }
  558. }
  559. void
  560. histtoauto(void)
  561. {
  562. Auto *l;
  563. while(l = curhist) {
  564. curhist = l->link;
  565. l->link = curauto;
  566. curauto = l;
  567. }
  568. }
  569. void
  570. collapsefrog(Sym *s)
  571. {
  572. int i;
  573. /*
  574. * bad encoding of path components only allows
  575. * MAXHIST components. if there is an overflow,
  576. * first try to collapse xxx/..
  577. */
  578. for(i=1; i<histfrogp; i++)
  579. if(strcmp(histfrog[i]->name+1, "..") == 0) {
  580. memmove(histfrog+i-1, histfrog+i+1,
  581. (histfrogp-i-1)*sizeof(histfrog[0]));
  582. histfrogp--;
  583. goto out;
  584. }
  585. /*
  586. * next try to collapse .
  587. */
  588. for(i=0; i<histfrogp; i++)
  589. if(strcmp(histfrog[i]->name+1, ".") == 0) {
  590. memmove(histfrog+i, histfrog+i+1,
  591. (histfrogp-i-1)*sizeof(histfrog[0]));
  592. goto out;
  593. }
  594. /*
  595. * last chance, just truncate from front
  596. */
  597. memmove(histfrog+0, histfrog+1,
  598. (histfrogp-1)*sizeof(histfrog[0]));
  599. out:
  600. histfrog[histfrogp-1] = s;
  601. }
  602. void
  603. nopout(Prog *p)
  604. {
  605. p->as = ANOP;
  606. p->from.type = D_NONE;
  607. p->to.type = D_NONE;
  608. }
  609. uchar*
  610. readsome(int f, uchar *buf, uchar *good, uchar *stop, int max)
  611. {
  612. int n;
  613. n = stop - good;
  614. memmove(buf, good, stop - good);
  615. stop = buf + n;
  616. n = MAXIO - n;
  617. if(n > max)
  618. n = max;
  619. n = read(f, stop, n);
  620. if(n <= 0)
  621. return 0;
  622. return stop + n;
  623. }
  624. void
  625. ldobj(int f, long c, char *pn)
  626. {
  627. Prog *p, *t;
  628. Sym *h[NSYM], *s, *di;
  629. int v, o, r, skip;
  630. long ipc;
  631. uchar *bloc, *bsize, *stop;
  632. ulong sig;
  633. static int files;
  634. static char **filen;
  635. char **nfilen;
  636. if((files&15) == 0){
  637. nfilen = malloc((files+16)*sizeof(char*));
  638. memmove(nfilen, filen, files*sizeof(char*));
  639. free(filen);
  640. filen = nfilen;
  641. }
  642. filen[files++] = strdup(pn);
  643. bsize = buf.xbuf;
  644. bloc = buf.xbuf;
  645. di = S;
  646. newloop:
  647. memset(h, 0, sizeof(h));
  648. histfrogp = 0;
  649. version++;
  650. ipc = pc;
  651. skip = 0;
  652. loop:
  653. if(c <= 0)
  654. goto eof;
  655. r = bsize - bloc;
  656. if(r < 100 && r < c) { /* enough for largest prog */
  657. bsize = readsome(f, buf.xbuf, bloc, bsize, c);
  658. if(bsize == 0)
  659. goto eof;
  660. bloc = buf.xbuf;
  661. goto loop;
  662. }
  663. o = bloc[0]; /* as */
  664. if(o <= 0 || o >= ALAST) {
  665. diag("%s: opcode out of range %d", pn, o);
  666. print(" probably not a .q file\n");
  667. errorexit();
  668. }
  669. if(o == ANAME || o == ASIGNAME) {
  670. sig = 0;
  671. if(o == ASIGNAME) {
  672. sig = bloc[1] | (bloc[2]<<8) | (bloc[3]<<16) | (bloc[4]<<24);
  673. bloc += 4;
  674. c -= 4;
  675. }
  676. stop = memchr(&bloc[3], 0, bsize-&bloc[3]);
  677. if(stop == 0){
  678. bsize = readsome(f, buf.xbuf, bloc, bsize, c);
  679. if(bsize == 0)
  680. goto eof;
  681. bloc = buf.xbuf;
  682. stop = memchr(&bloc[3], 0, bsize-&bloc[3]);
  683. if(stop == 0){
  684. fprint(2, "%s: name too long\n", pn);
  685. errorexit();
  686. }
  687. }
  688. v = bloc[1]; /* type */
  689. o = bloc[2]; /* sym */
  690. bloc += 3;
  691. c -= 3;
  692. r = 0;
  693. if(v == D_STATIC)
  694. r = version;
  695. s = lookup((char*)bloc, r);
  696. c -= &stop[1] - bloc;
  697. bloc = stop + 1;
  698. if(sig != 0){
  699. if(s->sig != 0 && s->sig != sig)
  700. diag("incompatible type signatures %lux(%s) and %lux(%s) for %s", s->sig, filen[s->file], sig, pn, s->name);
  701. s->sig = sig;
  702. s->file = files-1;
  703. }
  704. if(debug['W'])
  705. print(" ANAME %s\n", s->name);
  706. h[o] = s;
  707. if((v == D_EXTERN || v == D_STATIC) && s->type == 0)
  708. s->type = SXREF;
  709. if(v == D_FILE) {
  710. if(s->type != SFILE) {
  711. histgen++;
  712. s->type = SFILE;
  713. s->value = histgen;
  714. }
  715. if(histfrogp < MAXHIST) {
  716. histfrog[histfrogp] = s;
  717. histfrogp++;
  718. } else
  719. collapsefrog(s);
  720. }
  721. goto loop;
  722. }
  723. if(nhunk < sizeof(Prog))
  724. gethunk();
  725. p = (Prog*)hunk;
  726. nhunk -= sizeof(Prog);
  727. hunk += sizeof(Prog);
  728. p->as = o;
  729. p->reg = bloc[1] & 0x3f;
  730. if(bloc[1] & 0x80)
  731. p->mark = NOSCHED;
  732. p->line = bloc[2] | (bloc[3]<<8) | (bloc[4]<<16) | (bloc[5]<<24);
  733. r = zaddr(bloc+6, &p->from, h) + 6;
  734. if(bloc[1] & 0x40)
  735. r += zaddr(bloc+r, &p->from3, h);
  736. else
  737. p->from3 = zprg.from3;
  738. r += zaddr(bloc+r, &p->to, h);
  739. bloc += r;
  740. c -= r;
  741. if(p->reg < 0 || p->reg > NREG)
  742. diag("register out of range %d", p->reg);
  743. p->link = P;
  744. p->cond = P;
  745. if(debug['W'])
  746. print("%P\n", p);
  747. switch(o) {
  748. case AHISTORY:
  749. if(p->to.offset == -1) {
  750. addlib(pn);
  751. histfrogp = 0;
  752. goto loop;
  753. }
  754. addhist(p->line, D_FILE); /* 'z' */
  755. if(p->to.offset)
  756. addhist(p->to.offset, D_FILE1); /* 'Z' */
  757. histfrogp = 0;
  758. goto loop;
  759. case AEND:
  760. histtoauto();
  761. if(curtext != P)
  762. curtext->to.autom = curauto;
  763. curauto = 0;
  764. curtext = P;
  765. if(c)
  766. goto newloop;
  767. return;
  768. case AGLOBL:
  769. s = p->from.sym;
  770. if(s == S) {
  771. diag("GLOBL must have a name\n%P", p);
  772. errorexit();
  773. }
  774. if(s->type == 0 || s->type == SXREF) {
  775. s->type = SBSS;
  776. s->value = 0;
  777. }
  778. if(s->type != SBSS) {
  779. diag("redefinition: %s\n%P", s->name, p);
  780. s->type = SBSS;
  781. s->value = 0;
  782. }
  783. if(p->to.offset > s->value)
  784. s->value = p->to.offset;
  785. break;
  786. case ADYNT:
  787. if(p->to.sym == S) {
  788. diag("DYNT without a sym\n%P", p);
  789. break;
  790. }
  791. di = p->to.sym;
  792. p->reg = 4;
  793. if(di->type == SXREF) {
  794. if(debug['z'])
  795. Bprint(&bso, "%P set to %d\n", p, dtype);
  796. di->type = SCONST;
  797. di->value = dtype;
  798. dtype += 4;
  799. }
  800. if(p->from.sym == S)
  801. break;
  802. p->from.offset = di->value;
  803. p->from.sym->type = SDATA;
  804. if(curtext == P) {
  805. diag("DYNT not in text: %P", p);
  806. break;
  807. }
  808. p->to.sym = curtext->from.sym;
  809. p->to.type = D_CONST;
  810. p->link = datap;
  811. datap = p;
  812. break;
  813. case AINIT:
  814. if(p->from.sym == S) {
  815. diag("INIT without a sym\n%P", p);
  816. break;
  817. }
  818. if(di == S) {
  819. diag("INIT without previous DYNT\n%P", p);
  820. break;
  821. }
  822. p->from.offset = di->value;
  823. p->from.sym->type = SDATA;
  824. p->link = datap;
  825. datap = p;
  826. break;
  827. case ADATA:
  828. p->link = datap;
  829. datap = p;
  830. break;
  831. case AGOK:
  832. diag("unknown opcode\n%P", p);
  833. p->pc = pc;
  834. pc++;
  835. break;
  836. case ATEXT:
  837. if(curtext != P) {
  838. histtoauto();
  839. curtext->to.autom = curauto;
  840. curauto = 0;
  841. }
  842. curtext = p;
  843. autosize = (p->to.offset+3L) & ~3L;
  844. p->to.offset = autosize;
  845. autosize += 4;
  846. s = p->from.sym;
  847. if(s == S) {
  848. diag("TEXT must have a name\n%P", p);
  849. errorexit();
  850. }
  851. if(s->type != 0 && s->type != SXREF) {
  852. if(p->reg & DUPOK) {
  853. skip = 1;
  854. goto casedef;
  855. }
  856. diag("redefinition: %s\n%P", s->name, p);
  857. }
  858. s->type = STEXT;
  859. s->value = pc;
  860. if(textp != P) {
  861. for(t = textp; t->cond != P; t = t->cond)
  862. ;
  863. t->cond = p;
  864. } else
  865. textp = p;
  866. lastp->link = p;
  867. lastp = p;
  868. p->pc = pc;
  869. pc++;
  870. break;
  871. case AFMOVS:
  872. if(skip)
  873. goto casedef;
  874. if(p->from.type == D_FCONST) {
  875. /* size sb 9 max */
  876. sprint(literal, "$%lux", ieeedtof(&p->from.ieee));
  877. s = lookup(literal, 0);
  878. if(s->type == 0) {
  879. s->type = SBSS;
  880. s->value = 4;
  881. t = prg();
  882. t->as = ADATA;
  883. t->line = p->line;
  884. t->from.type = D_OREG;
  885. t->from.sym = s;
  886. t->from.name = D_EXTERN;
  887. t->reg = 4;
  888. t->to = p->from;
  889. t->link = datap;
  890. datap = t;
  891. }
  892. p->from.type = D_OREG;
  893. p->from.sym = s;
  894. p->from.name = D_EXTERN;
  895. p->from.offset = 0;
  896. }
  897. goto casedef;
  898. case AFMOVD:
  899. if(skip)
  900. goto casedef;
  901. if(p->from.type == D_FCONST) {
  902. /* size sb 18 max */
  903. sprint(literal, "$%lux.%lux",
  904. p->from.ieee.l, p->from.ieee.h);
  905. s = lookup(literal, 0);
  906. if(s->type == 0) {
  907. s->type = SBSS;
  908. s->value = 8;
  909. t = prg();
  910. t->as = ADATA;
  911. t->line = p->line;
  912. t->from.type = D_OREG;
  913. t->from.sym = s;
  914. t->from.name = D_EXTERN;
  915. t->reg = 8;
  916. t->to = p->from;
  917. t->link = datap;
  918. datap = t;
  919. }
  920. p->from.type = D_OREG;
  921. p->from.sym = s;
  922. p->from.name = D_EXTERN;
  923. p->from.offset = 0;
  924. }
  925. goto casedef;
  926. case ASUBC:
  927. if(p->from.type == D_CONST) {
  928. p->from.offset = -p->from.offset;
  929. p->as = AADDC;
  930. }
  931. goto casedef;
  932. case ASUBCCC:
  933. if(p->from.type == D_CONST) {
  934. p->from.offset = -p->from.offset;
  935. p->as = AADDCCC;
  936. }
  937. goto casedef;
  938. case ASUB:
  939. if(p->from.type == D_CONST) {
  940. p->from.offset = -p->from.offset;
  941. p->as = AADD;
  942. }
  943. goto casedef;
  944. default:
  945. casedef:
  946. if(skip)
  947. nopout(p);
  948. if(p->to.type == D_BRANCH)
  949. p->to.offset += ipc;
  950. lastp->link = p;
  951. lastp = p;
  952. p->pc = pc;
  953. pc++;
  954. break;
  955. }
  956. goto loop;
  957. eof:
  958. diag("truncated object file: %s", pn);
  959. }
  960. Sym*
  961. lookup(char *symb, int v)
  962. {
  963. Sym *s;
  964. char *p;
  965. long h;
  966. int c, l;
  967. h = v;
  968. for(p=symb; c = *p; p++)
  969. h = h+h+h + c;
  970. l = (p - symb) + 1;
  971. if(h < 0)
  972. h = ~h;
  973. h %= NHASH;
  974. for(s = hash[h]; s != S; s = s->link)
  975. if(s->version == v)
  976. if(memcmp(s->name, symb, l) == 0)
  977. return s;
  978. while(nhunk < sizeof(Sym))
  979. gethunk();
  980. s = (Sym*)hunk;
  981. nhunk -= sizeof(Sym);
  982. hunk += sizeof(Sym);
  983. s->name = malloc(l + 1);
  984. memmove(s->name, symb, l);
  985. s->link = hash[h];
  986. s->type = 0;
  987. s->version = v;
  988. s->value = 0;
  989. s->sig = 0;
  990. hash[h] = s;
  991. return s;
  992. }
  993. Prog*
  994. prg(void)
  995. {
  996. Prog *p;
  997. int n;
  998. n = (sizeof(Prog) + 3) & ~3;
  999. while(nhunk < n)
  1000. gethunk();
  1001. p = (Prog*)hunk;
  1002. nhunk -= n;
  1003. hunk += n;
  1004. *p = zprg;
  1005. return p;
  1006. }
  1007. void
  1008. gethunk(void)
  1009. {
  1010. char *h;
  1011. long nh;
  1012. nh = NHUNK;
  1013. if(tothunk >= 5L*NHUNK) {
  1014. nh = 5L*NHUNK;
  1015. if(tothunk >= 25L*NHUNK)
  1016. nh = 25L*NHUNK;
  1017. }
  1018. h = mysbrk(nh);
  1019. if(h == (char *)-1) {
  1020. diag("out of memory");
  1021. errorexit();
  1022. }
  1023. hunk = h;
  1024. nhunk = nh;
  1025. tothunk += nh;
  1026. }
  1027. void
  1028. doprof1(void)
  1029. {
  1030. Sym *s;
  1031. long n;
  1032. Prog *p, *q;
  1033. if(debug['v'])
  1034. Bprint(&bso, "%5.2f profile 1\n", cputime());
  1035. Bflush(&bso);
  1036. s = lookup("__mcount", 0);
  1037. n = 1;
  1038. for(p = firstp->link; p != P; p = p->link) {
  1039. if(p->as == ATEXT) {
  1040. q = prg();
  1041. q->line = p->line;
  1042. q->link = datap;
  1043. datap = q;
  1044. q->as = ADATA;
  1045. q->from.type = D_OREG;
  1046. q->from.name = D_EXTERN;
  1047. q->from.offset = n*4;
  1048. q->from.sym = s;
  1049. q->reg = 4;
  1050. q->to = p->from;
  1051. q->to.type = D_CONST;
  1052. q = prg();
  1053. q->line = p->line;
  1054. q->pc = p->pc;
  1055. q->link = p->link;
  1056. p->link = q;
  1057. p = q;
  1058. p->as = AMOVW;
  1059. p->from.type = D_OREG;
  1060. p->from.name = D_EXTERN;
  1061. p->from.sym = s;
  1062. p->from.offset = n*4 + 4;
  1063. p->to.type = D_REG;
  1064. p->to.reg = REGTMP;
  1065. q = prg();
  1066. q->line = p->line;
  1067. q->pc = p->pc;
  1068. q->link = p->link;
  1069. p->link = q;
  1070. p = q;
  1071. p->as = AADD;
  1072. p->from.type = D_CONST;
  1073. p->from.offset = 1;
  1074. p->to.type = D_REG;
  1075. p->to.reg = REGTMP;
  1076. q = prg();
  1077. q->line = p->line;
  1078. q->pc = p->pc;
  1079. q->link = p->link;
  1080. p->link = q;
  1081. p = q;
  1082. p->as = AMOVW;
  1083. p->from.type = D_REG;
  1084. p->from.reg = REGTMP;
  1085. p->to.type = D_OREG;
  1086. p->to.name = D_EXTERN;
  1087. p->to.sym = s;
  1088. p->to.offset = n*4 + 4;
  1089. n += 2;
  1090. continue;
  1091. }
  1092. }
  1093. q = prg();
  1094. q->line = 0;
  1095. q->link = datap;
  1096. datap = q;
  1097. q->as = ADATA;
  1098. q->from.type = D_OREG;
  1099. q->from.name = D_EXTERN;
  1100. q->from.sym = s;
  1101. q->reg = 4;
  1102. q->to.type = D_CONST;
  1103. q->to.offset = n;
  1104. s->type = SBSS;
  1105. s->value = n*4;
  1106. }
  1107. void
  1108. doprof2(void)
  1109. {
  1110. Sym *s2, *s4;
  1111. Prog *p, *q, *ps2, *ps4;
  1112. if(debug['v'])
  1113. Bprint(&bso, "%5.2f profile 2\n", cputime());
  1114. Bflush(&bso);
  1115. s2 = lookup("_profin", 0);
  1116. s4 = lookup("_profout", 0);
  1117. if(s2->type != STEXT || s4->type != STEXT) {
  1118. diag("_profin/_profout not defined");
  1119. return;
  1120. }
  1121. ps2 = P;
  1122. ps4 = P;
  1123. for(p = firstp; p != P; p = p->link) {
  1124. if(p->as == ATEXT) {
  1125. if(p->from.sym == s2) {
  1126. p->reg = 1;
  1127. ps2 = p;
  1128. }
  1129. if(p->from.sym == s4) {
  1130. p->reg = 1;
  1131. ps4 = p;
  1132. }
  1133. }
  1134. }
  1135. for(p = firstp; p != P; p = p->link) {
  1136. if(p->as == ATEXT) {
  1137. curtext = p;
  1138. if(p->reg & NOPROF) { /* dont profile */
  1139. for(;;) {
  1140. q = p->link;
  1141. if(q == P)
  1142. break;
  1143. if(q->as == ATEXT)
  1144. break;
  1145. p = q;
  1146. }
  1147. continue;
  1148. }
  1149. /*
  1150. * BL profin
  1151. */
  1152. q = prg();
  1153. q->line = p->line;
  1154. q->pc = p->pc;
  1155. q->link = p->link;
  1156. p->link = q;
  1157. p = q;
  1158. p->as = ABL;
  1159. p->to.type = D_BRANCH;
  1160. p->cond = ps2;
  1161. p->to.sym = s2;
  1162. continue;
  1163. }
  1164. if(p->as == ARETURN) {
  1165. /*
  1166. * RETURN
  1167. */
  1168. q = prg();
  1169. q->as = ARETURN;
  1170. q->from = p->from;
  1171. q->to = p->to;
  1172. q->link = p->link;
  1173. p->link = q;
  1174. /*
  1175. * BL profout
  1176. */
  1177. p->as = ABL;
  1178. p->from = zprg.from;
  1179. p->to = zprg.to;
  1180. p->to.type = D_BRANCH;
  1181. p->cond = ps4;
  1182. p->to.sym = s4;
  1183. p = q;
  1184. continue;
  1185. }
  1186. }
  1187. }
  1188. void
  1189. nuxiinit(void)
  1190. {
  1191. int i, c;
  1192. for(i=0; i<4; i++) {
  1193. c = find1(0x01020304L, i+1);
  1194. if(i >= 2)
  1195. inuxi2[i-2] = c;
  1196. if(i >= 3)
  1197. inuxi1[i-3] = c;
  1198. inuxi4[i] = c;
  1199. fnuxi8[i] = c+4;
  1200. fnuxi8[i+4] = c;
  1201. }
  1202. if(debug['v']) {
  1203. Bprint(&bso, "inuxi = ");
  1204. for(i=0; i<1; i++)
  1205. Bprint(&bso, "%d", inuxi1[i]);
  1206. Bprint(&bso, " ");
  1207. for(i=0; i<2; i++)
  1208. Bprint(&bso, "%d", inuxi2[i]);
  1209. Bprint(&bso, " ");
  1210. for(i=0; i<4; i++)
  1211. Bprint(&bso, "%d", inuxi4[i]);
  1212. Bprint(&bso, "\nfnuxi = ");
  1213. for(i=0; i<8; i++)
  1214. Bprint(&bso, "%d", fnuxi8[i]);
  1215. Bprint(&bso, "\n");
  1216. }
  1217. Bflush(&bso);
  1218. }
  1219. int
  1220. find1(long l, int c)
  1221. {
  1222. char *p;
  1223. int i;
  1224. p = (char*)&l;
  1225. for(i=0; i<4; i++)
  1226. if(*p++ == c)
  1227. return i;
  1228. return 0;
  1229. }
  1230. long
  1231. ieeedtof(Ieee *ieeep)
  1232. {
  1233. int exp;
  1234. long v;
  1235. if(ieeep->h == 0)
  1236. return 0;
  1237. exp = (ieeep->h>>20) & ((1L<<11)-1L);
  1238. exp -= (1L<<10) - 2L;
  1239. v = (ieeep->h & 0xfffffL) << 3;
  1240. v |= (ieeep->l >> 29) & 0x7L;
  1241. if((ieeep->l >> 28) & 1) {
  1242. v++;
  1243. if(v & 0x800000L) {
  1244. v = (v & 0x7fffffL) >> 1;
  1245. exp++;
  1246. }
  1247. }
  1248. if(exp <= -126 || exp >= 130)
  1249. diag("double fp to single fp overflow");
  1250. v |= ((exp + 126) & 0xffL) << 23;
  1251. v |= ieeep->h & 0x80000000L;
  1252. return v;
  1253. }
  1254. double
  1255. ieeedtod(Ieee *ieeep)
  1256. {
  1257. Ieee e;
  1258. double fr;
  1259. int exp;
  1260. if(ieeep->h & (1L<<31)) {
  1261. e.h = ieeep->h & ~(1L<<31);
  1262. e.l = ieeep->l;
  1263. return -ieeedtod(&e);
  1264. }
  1265. if(ieeep->l == 0 && ieeep->h == 0)
  1266. return 0;
  1267. fr = ieeep->l & ((1L<<16)-1L);
  1268. fr /= 1L<<16;
  1269. fr += (ieeep->l>>16) & ((1L<<16)-1L);
  1270. fr /= 1L<<16;
  1271. fr += (ieeep->h & (1L<<20)-1L) | (1L<<20);
  1272. fr /= 1L<<21;
  1273. exp = (ieeep->h>>20) & ((1L<<11)-1L);
  1274. exp -= (1L<<10) - 2L;
  1275. return ldexp(fr, exp);
  1276. }
  1277. void
  1278. undefsym(Sym *s)
  1279. {
  1280. int n;
  1281. n = imports;
  1282. if(s->value != 0)
  1283. diag("value != 0 on SXREF");
  1284. if(n >= 1<<Rindex)
  1285. diag("import index %d out of range", n);
  1286. s->value = n<<Roffset;
  1287. s->type = SUNDEF;
  1288. imports++;
  1289. }
  1290. void
  1291. zerosig(char *sp)
  1292. {
  1293. Sym *s;
  1294. s = lookup(sp, 0);
  1295. s->sig = 0;
  1296. }
  1297. void
  1298. readundefs(char *f, int t)
  1299. {
  1300. int i, n;
  1301. Sym *s;
  1302. Biobuf *b;
  1303. char *l, buf[256], *fields[64];
  1304. if(f == nil)
  1305. return;
  1306. b = Bopen(f, OREAD);
  1307. if(b == nil){
  1308. diag("could not open %s: %r", f);
  1309. errorexit();
  1310. }
  1311. while((l = Brdline(b, '\n')) != nil){
  1312. n = Blinelen(b);
  1313. if(n >= sizeof(buf)){
  1314. diag("%s: line too long", f);
  1315. errorexit();
  1316. }
  1317. memmove(buf, l, n);
  1318. buf[n-1] = '\0';
  1319. n = getfields(buf, fields, nelem(fields), 1, " \t\r\n");
  1320. if(n == nelem(fields)){
  1321. diag("%s: bad format", f);
  1322. errorexit();
  1323. }
  1324. for(i = 0; i < n; i++){
  1325. s = lookup(fields[i], 0);
  1326. s->type = SXREF;
  1327. s->subtype = t;
  1328. if(t == SIMPORT)
  1329. nimports++;
  1330. else
  1331. nexports++;
  1332. }
  1333. }
  1334. Bterm(b);
  1335. }