nasmlib.c
上传用户:yuppie_zhu
上传日期:2007-01-08
资源大小:535k
文件大小:26k
源码类别:

编译器/解释器

开发平台:

C/C++

  1. /* nasmlib.c library routines for the Netwide Assembler
  2.  *
  3.  * The Netwide Assembler is copyright (C) 1996 Simon Tatham and
  4.  * Julian Hall. All rights reserved. The software is
  5.  * redistributable under the licence given in the file "Licence"
  6.  * distributed in the NASM archive.
  7.  */
  8. #include <stdio.h>
  9. #include <stdlib.h>
  10. #include <string.h>
  11. #include <ctype.h>
  12. #include "nasm.h"
  13. #include "nasmlib.h"
  14. static efunc nasm_malloc_error;
  15. #ifdef LOGALLOC
  16. static FILE *logfp;
  17. #endif
  18. void nasm_set_malloc_error (efunc error) 
  19. {
  20.     nasm_malloc_error = error;
  21. #ifdef LOGALLOC
  22.     logfp = fopen ("malloc.log", "w");
  23.     setvbuf (logfp, NULL, _IOLBF, BUFSIZ);
  24.     fprintf (logfp, "null pointer is %pn", NULL);
  25. #endif
  26. }
  27. #ifdef LOGALLOC
  28. void *nasm_malloc_log (char *file, int line, size_t size)
  29. #else
  30. void *nasm_malloc (size_t size)
  31. #endif
  32. {
  33.     void *p = malloc(size);
  34.     if (!p)
  35. nasm_malloc_error (ERR_FATAL | ERR_NOFILE, "out of memory");
  36. #ifdef LOGALLOC
  37.     else
  38. fprintf(logfp, "%s %d malloc(%ld) returns %pn",
  39. file, line, (long)size, p);
  40. #endif
  41.     return p;
  42. }
  43. #ifdef LOGALLOC
  44. void *nasm_realloc_log (char *file, int line, void *q, size_t size)
  45. #else
  46. void *nasm_realloc (void *q, size_t size)
  47. #endif
  48. {
  49.     void *p = q ? realloc(q, size) : malloc(size);
  50.     if (!p)
  51. nasm_malloc_error (ERR_FATAL | ERR_NOFILE, "out of memory");
  52. #ifdef LOGALLOC
  53.     else if (q)
  54. fprintf(logfp, "%s %d realloc(%p,%ld) returns %pn",
  55. file, line, q, (long)size, p);
  56.     else
  57. fprintf(logfp, "%s %d malloc(%ld) returns %pn",
  58. file, line, (long)size, p);
  59. #endif
  60.     return p;
  61. }
  62. #ifdef LOGALLOC
  63. void nasm_free_log (char *file, int line, void *q)
  64. #else
  65. void nasm_free (void *q)
  66. #endif
  67. {
  68.     if (q) {
  69. free (q);
  70. #ifdef LOGALLOC
  71. fprintf(logfp, "%s %d free(%p)n",
  72. file, line, q);
  73. #endif
  74.     }
  75. }
  76. #ifdef LOGALLOC
  77. char *nasm_strdup_log (char *file, int line, char *s)
  78. #else
  79. char *nasm_strdup (char *s)
  80. #endif
  81. {
  82.     char *p;
  83.     int size = strlen(s)+1;
  84.     p = malloc(size);
  85.     if (!p)
  86. nasm_malloc_error (ERR_FATAL | ERR_NOFILE, "out of memory");
  87. #ifdef LOGALLOC
  88.     else
  89. fprintf(logfp, "%s %d strdup(%ld) returns %pn",
  90. file, line, (long)size, p);
  91. #endif
  92.     strcpy (p, s);
  93.     return p;
  94. }
  95. #ifdef LOGALLOC
  96. char *nasm_strndup_log (char *file, int line, char *s, size_t len)
  97. #else
  98. char *nasm_strndup (char *s, size_t len)
  99. #endif
  100. {
  101.     char *p;
  102.     int size = len+1;
  103.     p = malloc(size);
  104.     if (!p)
  105. nasm_malloc_error (ERR_FATAL | ERR_NOFILE, "out of memory");
  106. #ifdef LOGALLOC
  107.     else
  108. fprintf(logfp, "%s %d strndup(%ld) returns %pn",
  109. file, line, (long)size, p);
  110. #endif
  111.     strncpy (p, s, len);
  112.     p[len] = '';
  113.     return p;
  114. }
  115. int nasm_stricmp (const char *s1, const char *s2) 
  116. {
  117.     while (*s1 && toupper(*s1) == toupper(*s2))
  118. s1++, s2++;
  119.     if (!*s1 && !*s2)
  120. return 0;
  121.     else if (toupper(*s1) < toupper(*s2))
  122. return -1;
  123.     else
  124. return 1;
  125. }
  126. int nasm_strnicmp (const char *s1, const char *s2, int n) 
  127. {
  128.     while (n > 0 && *s1 && toupper(*s1) == toupper(*s2))
  129. s1++, s2++, n--;
  130.     if ((!*s1 && !*s2) || n==0)
  131. return 0;
  132.     else if (toupper(*s1) < toupper(*s2))
  133. return -1;
  134.     else
  135. return 1;
  136. }
  137. #define lib_isnumchar(c)   ( isalnum(c) || (c) == '$')
  138. #define numvalue(c)  ((c)>='a' ? (c)-'a'+10 : (c)>='A' ? (c)-'A'+10 : (c)-'0')
  139. long readnum (char *str, int *error) 
  140. {
  141.     char *r = str, *q;
  142.     long radix;
  143.     unsigned long result, checklimit;
  144.     int digit, last;
  145.     int warn = FALSE;
  146.     int sign = 1;
  147.     *error = FALSE;
  148.     while (isspace(*r)) r++;        /* find start of number */
  149.     /*
  150.      * If the number came from make_tok_num (as a result of an %assign), it
  151.      * might have a '-' built into it (rather than in a preceeding token).
  152.      */
  153.     if (*r == '-')
  154.     {
  155. r++;
  156. sign = -1;
  157.     }
  158.     q = r;
  159.     while (lib_isnumchar(*q)) q++;     /* find end of number */
  160.     /*
  161.      * If it begins 0x, 0X or $, or ends in H, it's in hex. if it
  162.      * ends in Q, it's octal. if it ends in B, it's binary.
  163.      * Otherwise, it's ordinary decimal.
  164.      */
  165.     if (*r=='0' && (r[1]=='x' || r[1]=='X'))
  166. radix = 16, r += 2;
  167.     else if (*r=='$')
  168. radix = 16, r++;
  169.     else if (q[-1]=='H' || q[-1]=='h')
  170. radix = 16 , q--;
  171.     else if (q[-1]=='Q' || q[-1]=='q')
  172. radix = 8 , q--;
  173.     else if (q[-1]=='B' || q[-1]=='b')
  174. radix = 2 , q--;
  175.     else
  176. radix = 10;
  177.     /*
  178.      * If this number has been found for us by something other than
  179.      * the ordinary scanners, then it might be malformed by having
  180.      * nothing between the prefix and the suffix. Check this case
  181.      * now.
  182.      */
  183.     if (r >= q) {
  184. *error = TRUE;
  185. return 0;
  186.     }
  187.     /*
  188.      * `checklimit' must be 2**32 / radix. We can't do that in
  189.      * 32-bit arithmetic, which we're (probably) using, so we
  190.      * cheat: since we know that all radices we use are even, we
  191.      * can divide 2**31 by radix/2 instead.
  192.      */
  193.     checklimit = 0x80000000UL / (radix>>1);
  194.     /*
  195.      * Calculate the highest allowable value for the last digit
  196.      * of a 32 bit constant... in radix 10, it is 6, otherwise it is 0
  197.      */
  198.     last = (radix == 10 ? 6 : 0);
  199.     result = 0;
  200.     while (*r && r < q) {
  201. if (*r<'0' || (*r>'9' && *r<'A') || (digit = numvalue(*r)) >= radix) 
  202. {
  203.     *error = TRUE;
  204.     return 0;
  205. }
  206. if (result > checklimit ||
  207.     (result == checklimit && digit >= last))
  208. {
  209.     warn = TRUE;
  210. }
  211. result = radix * result + digit;
  212. r++;
  213.     }
  214.     if (warn)
  215. nasm_malloc_error (ERR_WARNING | ERR_PASS1 | ERR_WARN_NOV,
  216.    "numeric constant %s does not fit in 32 bits",
  217.    str);
  218.     return result*sign;
  219. }
  220. long readstrnum (char *str, int length, int *warn) 
  221. {
  222.     long charconst = 0;
  223.     int i;
  224.     *warn = FALSE;
  225.     str += length;
  226.     for (i=0; i<length; i++) {
  227. if (charconst & 0xff000000UL) {
  228.     *warn = TRUE;
  229. }
  230. charconst = (charconst<<8) + (unsigned char) *--str;
  231.     }
  232.     return charconst;
  233. }
  234. static long next_seg;
  235. void seg_init(void) 
  236. {
  237.     next_seg = 0;
  238. }
  239. long seg_alloc(void) 
  240. {
  241.     return (next_seg += 2) - 2;
  242. }
  243. void fwriteshort (int data, FILE *fp) 
  244. {
  245.     fputc ((int) (data & 255), fp);
  246.     fputc ((int) ((data >> 8) & 255), fp);
  247. }
  248. void fwritelong (long data, FILE *fp) 
  249. {
  250.     fputc ((int) (data & 255), fp);
  251.     fputc ((int) ((data >> 8) & 255), fp);
  252.     fputc ((int) ((data >> 16) & 255), fp);
  253.     fputc ((int) ((data >> 24) & 255), fp);
  254. }
  255. void standard_extension (char *inname, char *outname, char *extension,
  256.  efunc error) 
  257. {
  258.     char *p, *q;
  259.     if (*outname)        /* file name already exists, */
  260. return;        /* so do nothing */
  261.     q = inname;
  262.     p = outname;
  263.     while (*q) *p++ = *q++;        /* copy, and find end of string */
  264.     *p = '';        /* terminate it */
  265.     while (p > outname && *--p != '.');/* find final period (or whatever) */
  266.     if (*p != '.') while (*p) p++;     /* go back to end if none found */
  267.     if (!strcmp(p, extension)) {       /* is the extension already there? */
  268. if (*extension)
  269.     error(ERR_WARNING | ERR_NOFILE,
  270.   "file name already ends in `%s': "
  271.   "output will be in `nasm.out'",
  272.   extension);
  273. else
  274.     error(ERR_WARNING | ERR_NOFILE,
  275.   "file name already has no extension: "
  276.   "output will be in `nasm.out'");
  277. strcpy(outname, "nasm.out");
  278.     } else
  279. strcpy(p, extension);
  280. }
  281. #define LEAFSIZ (sizeof(RAA)-sizeof(RAA_UNION)+sizeof(RAA_LEAF))
  282. #define BRANCHSIZ (sizeof(RAA)-sizeof(RAA_UNION)+sizeof(RAA_BRANCH))
  283. #define LAYERSIZ(r) ( (r)->layers==0 ? RAA_BLKSIZE : RAA_LAYERSIZE )
  284. static struct RAA *real_raa_init (int layers) 
  285. {
  286.     struct RAA *r;
  287.     if (layers == 0) {
  288. r = nasm_malloc (LEAFSIZ);
  289. memset (r->u.l.data, 0, sizeof(r->u.l.data));
  290. r->layers = 0;
  291. r->stepsize = 1L;
  292.     } else {
  293. r = nasm_malloc (BRANCHSIZ);
  294. memset (r->u.b.data, 0, sizeof(r->u.b.data));
  295. r->layers = layers;
  296. r->stepsize = RAA_BLKSIZE;
  297. while (--layers)
  298.     r->stepsize *= RAA_LAYERSIZE;
  299.     }
  300.     return r;
  301. }
  302. struct RAA *raa_init (void) 
  303. {
  304.     return real_raa_init (0);
  305. }
  306. void raa_free (struct RAA *r) 
  307. {
  308.     if (r->layers == 0)
  309. nasm_free (r);
  310.     else {
  311. struct RAA **p;
  312. for (p = r->u.b.data; p - r->u.b.data < RAA_LAYERSIZE; p++)
  313.     if (*p)
  314. raa_free (*p);
  315.     }
  316. }
  317. long raa_read (struct RAA *r, long posn) 
  318. {
  319.     if (posn > r->stepsize * LAYERSIZ(r))
  320. return 0L;
  321.     while (r->layers > 0) {
  322. ldiv_t l;
  323. l = ldiv (posn, r->stepsize);
  324. r = r->u.b.data[l.quot];
  325. posn = l.rem;
  326. if (!r)        /* better check this */
  327.     return 0L;
  328.     }
  329.     return r->u.l.data[posn];
  330. }
  331. struct RAA *raa_write (struct RAA *r, long posn, long value) 
  332. {
  333.     struct RAA *result;
  334.     if (posn < 0)
  335. nasm_malloc_error (ERR_PANIC, "negative position in raa_write");
  336.     while (r->stepsize * LAYERSIZ(r) < posn) {
  337. /*
  338.  * Must go up a layer.
  339.  */
  340. struct RAA *s;
  341. s = nasm_malloc (BRANCHSIZ);
  342. memset (s->u.b.data, 0, sizeof(r->u.b.data));
  343. s->layers = r->layers + 1;
  344. s->stepsize = RAA_LAYERSIZE * r->stepsize;
  345. s->u.b.data[0] = r;
  346. r = s;
  347.     }
  348.     result = r;
  349.     while (r->layers > 0) {
  350. ldiv_t l;
  351. struct RAA **s;
  352. l = ldiv (posn, r->stepsize);
  353. s = &r->u.b.data[l.quot];
  354. if (!*s)
  355.     *s = real_raa_init (r->layers - 1);
  356. r = *s;
  357. posn = l.rem;
  358.     }
  359.     r->u.l.data[posn] = value;
  360.     return result;
  361. }
  362. #define SAA_MAXLEN 8192
  363. struct SAA *saa_init (long elem_len) 
  364. {
  365.     struct SAA *s;
  366.     if (elem_len > SAA_MAXLEN)
  367. nasm_malloc_error (ERR_PANIC | ERR_NOFILE, "SAA with huge elements");
  368.     s = nasm_malloc (sizeof(struct SAA));
  369.     s->posn = s->start = 0L;
  370.     s->elem_len = elem_len;
  371.     s->length = SAA_MAXLEN - (SAA_MAXLEN % elem_len);
  372.     s->data = nasm_malloc (s->length);
  373.     s->next = NULL;
  374.     s->end = s;
  375.     return s;
  376. }
  377. void saa_free (struct SAA *s) 
  378. {
  379.     struct SAA *t;
  380.     while (s) {
  381. t = s->next;
  382. nasm_free (s->data);
  383. nasm_free (s);
  384. s = t;
  385.     }
  386. }
  387. void *saa_wstruct (struct SAA *s) 
  388. {
  389.     void *p;
  390.     if (s->end->length - s->end->posn < s->elem_len) {
  391. s->end->next = nasm_malloc (sizeof(struct SAA));
  392. s->end->next->start = s->end->start + s->end->posn;
  393. s->end = s->end->next;
  394. s->end->length = s->length;
  395. s->end->next = NULL;
  396. s->end->posn = 0L;
  397. s->end->data = nasm_malloc (s->length);
  398.     }
  399.     p = s->end->data + s->end->posn;
  400.     s->end->posn += s->elem_len;
  401.     return p;
  402. }
  403. void saa_wbytes (struct SAA *s, void *data, long len) 
  404. {
  405.     char *d = data;
  406.     while (len > 0) {
  407. long l = s->end->length - s->end->posn;
  408. if (l > len)
  409.     l = len;
  410. if (l > 0) {
  411.     if (d) {
  412. memcpy (s->end->data + s->end->posn, d, l);
  413. d += l;
  414.     } else
  415. memset (s->end->data + s->end->posn, 0, l);
  416.     s->end->posn += l;
  417.     len -= l;
  418. }
  419. if (len > 0) {
  420.     s->end->next = nasm_malloc (sizeof(struct SAA));
  421.     s->end->next->start = s->end->start + s->end->posn;
  422.     s->end = s->end->next;
  423.     s->end->length = s->length;
  424.     s->end->next = NULL;
  425.     s->end->posn = 0L;
  426.     s->end->data = nasm_malloc (s->length);
  427. }
  428.     }
  429. }
  430. void saa_rewind (struct SAA *s) 
  431. {
  432.     s->rptr = s;
  433.     s->rpos = 0L;
  434. }
  435. void *saa_rstruct (struct SAA *s) 
  436. {
  437.     void *p;
  438.     if (!s->rptr)
  439. return NULL;
  440.     if (s->rptr->posn - s->rpos < s->elem_len) {
  441. s->rptr = s->rptr->next;
  442. if (!s->rptr)
  443.     return NULL;        /* end of array */
  444. s->rpos = 0L;
  445.     }
  446.     p = s->rptr->data + s->rpos;
  447.     s->rpos += s->elem_len;
  448.     return p;
  449. }
  450. void *saa_rbytes (struct SAA *s, long *len) 
  451. {
  452.     void *p;
  453.     if (!s->rptr)
  454. return NULL;
  455.     p = s->rptr->data + s->rpos;
  456.     *len = s->rptr->posn - s->rpos;
  457.     s->rptr = s->rptr->next;
  458.     s->rpos = 0L;
  459.     return p;
  460. }
  461. void saa_rnbytes (struct SAA *s, void *data, long len) 
  462. {
  463.     char *d = data;
  464.     while (len > 0) {
  465. long l;
  466. if (!s->rptr)
  467.     return;
  468. l = s->rptr->posn - s->rpos;
  469. if (l > len)
  470.     l = len;
  471. if (l > 0) {
  472.     memcpy (d, s->rptr->data + s->rpos, l);
  473.     d += l;
  474.     s->rpos += l;
  475.     len -= l;
  476. }
  477. if (len > 0) {
  478.     s->rptr = s->rptr->next;
  479.     s->rpos = 0L;
  480. }
  481.     }
  482. }
  483. void saa_fread (struct SAA *s, long posn, void *data, long len) 
  484. {
  485.     struct SAA *p;
  486.     long pos;
  487.     char *cdata = data;
  488.     if (!s->rptr || posn < s->rptr->start)
  489. saa_rewind (s);
  490.     p = s->rptr;
  491.     while (posn >= p->start + p->posn) {
  492. p = p->next;
  493. if (!p)
  494.     return;        /* what else can we do?! */
  495.     }
  496.     pos = posn - p->start;
  497.     while (len) {
  498. long l = p->posn - pos;
  499. if (l > len)
  500.     l = len;
  501. memcpy (cdata, p->data+pos, l);
  502. len -= l;
  503. cdata += l;
  504. p = p->next;
  505. if (!p)
  506.     return;
  507. pos = 0L;
  508.     }
  509.     s->rptr = p;
  510. }
  511. void saa_fwrite (struct SAA *s, long posn, void *data, long len) 
  512. {
  513.     struct SAA *p;
  514.     long pos;
  515.     char *cdata = data;
  516.     if (!s->rptr || posn < s->rptr->start)
  517. saa_rewind (s);
  518.     p = s->rptr;
  519.     while (posn >= p->start + p->posn) {
  520. p = p->next;
  521. if (!p)
  522.     return;        /* what else can we do?! */
  523.     }
  524.     pos = posn - p->start;
  525.     while (len) {
  526. long l = p->posn - pos;
  527. if (l > len)
  528.     l = len;
  529. memcpy (p->data+pos, cdata, l);
  530. len -= l;
  531. cdata += l;
  532. p = p->next;
  533. if (!p)
  534.     return;
  535. pos = 0L;
  536.     }
  537.     s->rptr = p;
  538. }
  539. void saa_fpwrite (struct SAA *s, FILE *fp) 
  540. {
  541.     char *data;
  542.     long len;
  543.     saa_rewind (s);
  544.     while ( (data = saa_rbytes (s, &len)) )
  545. fwrite (data, 1, len, fp);
  546. }
  547. /*
  548.  * Register, instruction, condition-code and prefix keywords used
  549.  * by the scanner.
  550.  */
  551. #include "names.c"
  552. static char *special_names[] = {
  553.     "byte", "dword", "far", "long", "near", "nosplit", "qword",
  554.     "short", "to", "tword", "word"
  555. };
  556. static char *prefix_names[] = {
  557.     "a16", "a32", "lock", "o16", "o32", "rep", "repe", "repne",
  558.     "repnz", "repz", "times"
  559. };
  560. /*
  561.  * Standard scanner routine used by parser.c and some output
  562.  * formats. It keeps a succession of temporary-storage strings in
  563.  * stdscan_tempstorage, which can be cleared using stdscan_reset.
  564.  */
  565. static char **stdscan_tempstorage = NULL;
  566. static int stdscan_tempsize = 0, stdscan_templen = 0;
  567. #define STDSCAN_TEMP_DELTA 256
  568. static void stdscan_pop(void) 
  569. {
  570.     nasm_free (stdscan_tempstorage[--stdscan_templen]);
  571. }
  572. void stdscan_reset(void) 
  573. {
  574.     while (stdscan_templen > 0)
  575. stdscan_pop();
  576. }
  577. /*
  578.  * Unimportant cleanup is done to avoid confusing people who are trying
  579.  * to debug real memory leaks
  580.  */
  581. void nasmlib_cleanup (void) 
  582. {
  583.     stdscan_reset();
  584.     nasm_free (stdscan_tempstorage);
  585. }
  586. static char *stdscan_copy(char *p, int len) 
  587. {
  588.     char *text;
  589.     text = nasm_malloc(len+1);
  590.     strncpy (text, p, len);
  591.     text[len] = '';
  592.     if (stdscan_templen >= stdscan_tempsize) {
  593. stdscan_tempsize += STDSCAN_TEMP_DELTA;
  594. stdscan_tempstorage = nasm_realloc(stdscan_tempstorage,
  595.    stdscan_tempsize*sizeof(char *));
  596.     }
  597.     stdscan_tempstorage[stdscan_templen++] = text;
  598.     return text;
  599. }
  600. char *stdscan_bufptr = NULL;
  601. int stdscan (void *private_data, struct tokenval *tv) 
  602. {
  603.     char ourcopy[MAX_KEYWORD+1], *r, *s;
  604.     (void) private_data;  /* Don't warn that this parameter is unused */
  605.     while (isspace(*stdscan_bufptr)) stdscan_bufptr++;
  606.     if (!*stdscan_bufptr)
  607. return tv->t_type = 0;
  608.     /* we have a token; either an id, a number or a char */
  609.     if (isidstart(*stdscan_bufptr) ||
  610. (*stdscan_bufptr == '$' && isidstart(stdscan_bufptr[1]))) {
  611. /* now we've got an identifier */
  612. int i;
  613. int is_sym = FALSE;
  614. if (*stdscan_bufptr == '$') {
  615.     is_sym = TRUE;
  616.     stdscan_bufptr++;
  617. }
  618.   r = stdscan_bufptr++;
  619. while (isidchar(*stdscan_bufptr)) stdscan_bufptr++;
  620. tv->t_charptr = stdscan_copy(r, stdscan_bufptr - r);
  621. if (is_sym || stdscan_bufptr-r > MAX_KEYWORD)
  622.     return tv->t_type = TOKEN_ID;/* bypass all other checks */
  623.     
  624. for (s=tv->t_charptr, r=ourcopy; *s; s++)
  625.     *r++ = tolower (*s);
  626. *r = '';
  627. /* right, so we have an identifier sitting in temp storage. now,
  628.  * is it actually a register or instruction name, or what? */
  629. if ((tv->t_integer=bsi(ourcopy, reg_names,
  630.        elements(reg_names)))>=0) {
  631.     tv->t_integer += EXPR_REG_START;
  632.     return tv->t_type = TOKEN_REG;
  633. } else if ((tv->t_integer=bsi(ourcopy, insn_names,
  634.       elements(insn_names)))>=0) {
  635.     return tv->t_type = TOKEN_INSN;
  636. }
  637. for (i=0; i<elements(icn); i++)
  638.     if (!strncmp(ourcopy, icn[i], strlen(icn[i]))) {
  639. char *p = ourcopy + strlen(icn[i]);
  640. tv->t_integer = ico[i];
  641. if ((tv->t_inttwo=bsi(p, conditions,
  642.  elements(conditions)))>=0)
  643.     return tv->t_type = TOKEN_INSN;
  644.     }
  645. if ((tv->t_integer=bsi(ourcopy, prefix_names,
  646.   elements(prefix_names)))>=0) {
  647.     tv->t_integer += PREFIX_ENUM_START;
  648.     return tv->t_type = TOKEN_PREFIX;
  649. }
  650. if ((tv->t_integer=bsi(ourcopy, special_names,
  651.   elements(special_names)))>=0)
  652.     return tv->t_type = TOKEN_SPECIAL;
  653. if (!strcmp(ourcopy, "seg"))
  654.     return tv->t_type = TOKEN_SEG;
  655. if (!strcmp(ourcopy, "wrt"))
  656.     return tv->t_type = TOKEN_WRT;
  657. return tv->t_type = TOKEN_ID;
  658.     } else if (*stdscan_bufptr == '$' && !isnumchar(stdscan_bufptr[1])) {
  659. /*
  660.  * It's a $ sign with no following hex number; this must
  661.  * mean it's a Here token ($), evaluating to the current
  662.  * assembly location, or a Base token ($$), evaluating to
  663.  * the base of the current segment.
  664.  */
  665. stdscan_bufptr++;
  666. if (*stdscan_bufptr == '$') {
  667.     stdscan_bufptr++;
  668.     return tv->t_type = TOKEN_BASE;
  669. }
  670. return tv->t_type = TOKEN_HERE;
  671.     } else if (isnumstart(*stdscan_bufptr)) {  /* now we've got a number */
  672. int rn_error;
  673. r = stdscan_bufptr++;
  674. while (isnumchar(*stdscan_bufptr))
  675.     stdscan_bufptr++;
  676. if (*stdscan_bufptr == '.') {
  677.     /*
  678.      * a floating point constant
  679.      */
  680.     stdscan_bufptr++;
  681.     while (isnumchar(*stdscan_bufptr) ||
  682.    ((stdscan_bufptr[-1] == 'e' || stdscan_bufptr[-1] == 'E')
  683.     && (*stdscan_bufptr == '-' || *stdscan_bufptr == '+')) ) 
  684.     {
  685. stdscan_bufptr++;
  686.     }
  687.     tv->t_charptr = stdscan_copy(r, stdscan_bufptr - r);
  688.     return tv->t_type = TOKEN_FLOAT;
  689. }
  690. r = stdscan_copy(r, stdscan_bufptr - r);
  691. tv->t_integer = readnum(r, &rn_error);
  692. stdscan_pop();
  693. if (rn_error)
  694.     return tv->t_type = TOKEN_ERRNUM;/* some malformation occurred */
  695. tv->t_charptr = NULL;
  696. return tv->t_type = TOKEN_NUM;
  697.     } else if (*stdscan_bufptr == ''' ||
  698.        *stdscan_bufptr == '"') {/* a char constant */
  699.      char quote = *stdscan_bufptr++, *r;
  700. int rn_warn;
  701. r = tv->t_charptr = stdscan_bufptr;
  702. while (*stdscan_bufptr && *stdscan_bufptr != quote) stdscan_bufptr++;
  703. tv->t_inttwo = stdscan_bufptr - r;      /* store full version */
  704. if (!*stdscan_bufptr)
  705.     return tv->t_type = TOKEN_ERRNUM;       /* unmatched quotes */
  706. stdscan_bufptr++; /* skip over final quote */
  707. tv->t_integer = readstrnum(r, tv->t_inttwo, &rn_warn);
  708. /* FIXME: rn_warn is not checked! */
  709. return tv->t_type = TOKEN_NUM;
  710.     } else if (*stdscan_bufptr == ';') {  /* a comment has happened - stay */
  711. return tv->t_type = 0;
  712.     } else if (stdscan_bufptr[0] == '>' && stdscan_bufptr[1] == '>') {
  713. stdscan_bufptr += 2;
  714. return tv->t_type = TOKEN_SHR;
  715.     } else if (stdscan_bufptr[0] == '<' && stdscan_bufptr[1] == '<') {
  716. stdscan_bufptr += 2;
  717. return tv->t_type = TOKEN_SHL;
  718.     } else if (stdscan_bufptr[0] == '/' && stdscan_bufptr[1] == '/') {
  719. stdscan_bufptr += 2;
  720. return tv->t_type = TOKEN_SDIV;
  721.     } else if (stdscan_bufptr[0] == '%' && stdscan_bufptr[1] == '%') {
  722. stdscan_bufptr += 2;
  723. return tv->t_type = TOKEN_SMOD;
  724.     } else if (stdscan_bufptr[0] == '=' && stdscan_bufptr[1] == '=') {
  725. stdscan_bufptr += 2;
  726. return tv->t_type = TOKEN_EQ;
  727.     } else if (stdscan_bufptr[0] == '<' && stdscan_bufptr[1] == '>') {
  728. stdscan_bufptr += 2;
  729. return tv->t_type = TOKEN_NE;
  730.     } else if (stdscan_bufptr[0] == '!' && stdscan_bufptr[1] == '=') {
  731. stdscan_bufptr += 2;
  732. return tv->t_type = TOKEN_NE;
  733.     } else if (stdscan_bufptr[0] == '<' && stdscan_bufptr[1] == '=') {
  734. stdscan_bufptr += 2;
  735. return tv->t_type = TOKEN_LE;
  736.     } else if (stdscan_bufptr[0] == '>' && stdscan_bufptr[1] == '=') {
  737. stdscan_bufptr += 2;
  738. return tv->t_type = TOKEN_GE;
  739.     } else if (stdscan_bufptr[0] == '&' && stdscan_bufptr[1] == '&') {
  740. stdscan_bufptr += 2;
  741. return tv->t_type = TOKEN_DBL_AND;
  742.     } else if (stdscan_bufptr[0] == '^' && stdscan_bufptr[1] == '^') {
  743. stdscan_bufptr += 2;
  744. return tv->t_type = TOKEN_DBL_XOR;
  745.     } else if (stdscan_bufptr[0] == '|' && stdscan_bufptr[1] == '|') {
  746. stdscan_bufptr += 2;
  747. return tv->t_type = TOKEN_DBL_OR;
  748.     } else        /* just an ordinary char */
  749.      return tv->t_type = (unsigned char) (*stdscan_bufptr++);
  750. }
  751. /*
  752.  * Return TRUE if the argument is a simple scalar. (Or a far-
  753.  * absolute, which counts.)
  754.  */
  755. int is_simple (expr *vect) 
  756. {
  757.     while (vect->type && !vect->value)
  758.      vect++;
  759.     if (!vect->type)
  760. return 1;
  761.     if (vect->type != EXPR_SIMPLE)
  762. return 0;
  763.     do {
  764. vect++;
  765.     } while (vect->type && !vect->value);
  766.     if (vect->type && vect->type < EXPR_SEGBASE+SEG_ABS) return 0;
  767.     return 1;
  768. }
  769. /*
  770.  * Return TRUE if the argument is a simple scalar, _NOT_ a far-
  771.  * absolute.
  772.  */
  773. int is_really_simple (expr *vect) 
  774. {
  775.     while (vect->type && !vect->value)
  776.      vect++;
  777.     if (!vect->type)
  778. return 1;
  779.     if (vect->type != EXPR_SIMPLE)
  780. return 0;
  781.     do {
  782. vect++;
  783.     } while (vect->type && !vect->value);
  784.     if (vect->type) return 0;
  785.     return 1;
  786. }
  787. /*
  788.  * Return TRUE if the argument is relocatable (i.e. a simple
  789.  * scalar, plus at most one segment-base, plus possibly a WRT).
  790.  */
  791. int is_reloc (expr *vect) 
  792. {
  793.     while (vect->type && !vect->value) /* skip initial value-0 terms */
  794.      vect++;
  795.     if (!vect->type)        /* trivially return TRUE if nothing */
  796. return 1;        /* is present apart from value-0s */
  797.     if (vect->type < EXPR_SIMPLE)      /* FALSE if a register is present */
  798. return 0;
  799.     if (vect->type == EXPR_SIMPLE) {   /* skip over a pure number term... */
  800. do {
  801.     vect++;
  802. } while (vect->type && !vect->value);
  803. if (!vect->type)        /* ...returning TRUE if that's all */
  804.     return 1;
  805.     }
  806.     if (vect->type == EXPR_WRT) {      /* skip over a WRT term... */
  807. do {
  808.     vect++;
  809. } while (vect->type && !vect->value);
  810. if (!vect->type)        /* ...returning TRUE if that's all */
  811.     return 1;
  812.     }
  813.     if (vect->value != 0 && vect->value != 1)
  814. return 0;        /* segment base multiplier non-unity */
  815.     do {        /* skip over _one_ seg-base term... */
  816. vect++;
  817.     } while (vect->type && !vect->value);
  818.     if (!vect->type)        /* ...returning TRUE if that's all */
  819. return 1;
  820.     return 0;        /* And return FALSE if there's more */
  821. }
  822. /*
  823.  * Return TRUE if the argument contains an `unknown' part.
  824.  */
  825. int is_unknown(expr *vect) 
  826. {
  827.     while (vect->type && vect->type < EXPR_UNKNOWN)
  828. vect++;
  829.     return (vect->type == EXPR_UNKNOWN);
  830. }
  831. /*
  832.  * Return TRUE if the argument contains nothing but an `unknown'
  833.  * part.
  834.  */
  835. int is_just_unknown(expr *vect) 
  836. {
  837.     while (vect->type && !vect->value)
  838. vect++;
  839.     return (vect->type == EXPR_UNKNOWN);
  840. }
  841. /*
  842.  * Return the scalar part of a relocatable vector. (Including
  843.  * simple scalar vectors - those qualify as relocatable.)
  844.  */
  845. long reloc_value (expr *vect) 
  846. {
  847.     while (vect->type && !vect->value)
  848.      vect++;
  849.     if (!vect->type) return 0;
  850.     if (vect->type == EXPR_SIMPLE)
  851. return vect->value;
  852.     else
  853. return 0;
  854. }
  855. /*
  856.  * Return the segment number of a relocatable vector, or NO_SEG for
  857.  * simple scalars.
  858.  */
  859. long reloc_seg (expr *vect) 
  860. {
  861.     while (vect->type && (vect->type == EXPR_WRT || !vect->value))
  862.      vect++;
  863.     if (vect->type == EXPR_SIMPLE) {
  864. do {
  865.     vect++;
  866. } while (vect->type && (vect->type == EXPR_WRT || !vect->value));
  867.     }
  868.     if (!vect->type)
  869. return NO_SEG;
  870.     else
  871. return vect->type - EXPR_SEGBASE;
  872. }
  873. /*
  874.  * Return the WRT segment number of a relocatable vector, or NO_SEG
  875.  * if no WRT part is present.
  876.  */
  877. long reloc_wrt (expr *vect) 
  878. {
  879.     while (vect->type && vect->type < EXPR_WRT)
  880.      vect++;
  881.     if (vect->type == EXPR_WRT) {
  882. return vect->value;
  883.     } else
  884. return NO_SEG;
  885. }
  886. /*
  887.  * Binary search.
  888.  */
  889. int bsi (char *string, char **array, int size) 
  890. {
  891.     int i = -1, j = size;        /* always, i < index < j */
  892.     while (j-i >= 2) {
  893. int k = (i+j)/2;
  894. int l = strcmp(string, array[k]);
  895. if (l<0)        /* it's in the first half */
  896.     j = k;
  897. else if (l>0)        /* it's in the second half */
  898.     i = k;
  899. else        /* we've got it :) */
  900.     return k;
  901.     }
  902.     return -1;        /* we haven't got it :( */
  903. }
  904. static char *file_name = NULL;
  905. static long line_number = 0;
  906. char *src_set_fname(char *newname) 
  907. {
  908.     char *oldname = file_name;
  909.     file_name = newname;
  910.     return oldname;
  911. }
  912. long src_set_linnum(long newline) 
  913. {
  914.     long oldline = line_number;
  915.     line_number = newline;
  916.     return oldline;
  917. }
  918. long src_get_linnum(void) 
  919. {
  920.     return line_number;
  921. }
  922. int src_get(long *xline, char **xname) 
  923. {
  924.     if (!file_name || !*xname || strcmp(*xname, file_name)) 
  925.     {
  926. nasm_free(*xname);
  927. *xname = file_name ? nasm_strdup(file_name) : NULL;
  928. *xline = line_number;
  929. return -2;
  930.     }
  931.     if (*xline != line_number) 
  932.     {
  933. long tmp = line_number - *xline;
  934. *xline = line_number;
  935. return tmp;
  936.     }
  937.     return 0;
  938. }
  939. void nasm_quote(char **str) 
  940. {
  941.     int ln=strlen(*str);
  942.     char q=(*str)[0];
  943.     char *p;
  944.     if (ln>1 && (*str)[ln-1]==q && (q=='"' || q=='''))
  945. return;
  946.     q = '"';
  947.     if (strchr(*str,q))
  948. q = ''';
  949.     p = nasm_malloc(ln+3);
  950.     strcpy(p+1, *str);
  951.     nasm_free(*str);
  952.     p[ln+1] = p[0] = q;
  953.     p[ln+2] = 0;
  954.     *str = p;
  955. }
  956.     
  957. char *nasm_strcat(char *one, char *two) 
  958. {
  959.     char *rslt;
  960.     int l1=strlen(one);
  961.     rslt = nasm_malloc(l1+strlen(two)+1);
  962.     strcpy(rslt, one);
  963.     strcpy(rslt+l1, two);
  964.     return rslt;
  965. }
  966. void null_debug_routine()
  967. {
  968. }
  969. struct dfmt null_debug_form = {
  970.     "Null debug format",
  971.     "null",
  972.     null_debug_routine,
  973.     null_debug_routine,
  974.     null_debug_routine,
  975.     null_debug_routine,
  976.     null_debug_routine,
  977.     null_debug_routine,
  978.     null_debug_routine,
  979. };
  980. struct dfmt *null_debug_arr[2] = { &null_debug_form, NULL };