obj.c 21 KB

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