home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / nasm097s.zip / NASM.C < prev    next >
C/C++ Source or Header  |  1997-11-02  |  29KB  |  1,108 lines
  1. /* The Netwide Assembler main program module
  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.  
  9. #include <stdio.h>
  10. #include <stdarg.h>
  11. #include <stdlib.h>
  12. #include <string.h>
  13. #include <ctype.h>
  14.  
  15. #include "nasm.h"
  16. #include "nasmlib.h"
  17. #include "preproc.h"
  18. #include "parser.h"
  19. #include "eval.h"
  20. #include "assemble.h"
  21. #include "labels.h"
  22. #include "outform.h"
  23. #include "listing.h"
  24.  
  25. static void report_error (int, char *, ...);
  26. static void parse_cmdline (int, char **);
  27. static void assemble_file (char *);
  28. static int getkw (char *buf, char **value);
  29. static void register_output_formats(void);
  30. static void usage(void);
  31.  
  32. static char *obuf;
  33. static char inname[FILENAME_MAX];
  34. static char outname[FILENAME_MAX];
  35. static char listname[FILENAME_MAX];
  36. static int lineno;               /* for error reporting */
  37. static int lineinc;               /* set by [LINE] or [ONELINE] */
  38. static int globallineno;           /* for forward-reference tracking */
  39. static int pass;
  40. static struct ofmt *ofmt = NULL;
  41.  
  42. static FILE *ofile = NULL;
  43. static int sb = 16;               /* by default */
  44.  
  45. static int use_stdout = FALSE;           /* by default, errors to stderr */
  46.  
  47. static long current_seg, abs_seg;
  48. static struct RAA *offsets;
  49. static long abs_offset;
  50.  
  51. static struct SAA *forwrefs;           /* keep track of forward references */
  52. static int forwline;
  53.  
  54. static Preproc *preproc;
  55. static int preprocess_only;
  56.  
  57. /* used by error function to report location */
  58. static char currentfile[FILENAME_MAX];
  59.  
  60. /*
  61.  * Which of the suppressible warnings are suppressed. Entry zero
  62.  * doesn't do anything. Initial defaults are given here.
  63.  */
  64. static char suppressed[1+ERR_WARN_MAX] = {
  65.     0, FALSE, TRUE, FALSE
  66. };
  67.  
  68. /*
  69.  * The option names for the suppressible warnings. As before, entry
  70.  * zero does nothing.
  71.  */
  72. static char *suppressed_names[1+ERR_WARN_MAX] = {
  73.     NULL, "macro-params", "orphan-labels", "number-overflow"
  74. };
  75.  
  76. /*
  77.  * The explanations for the suppressible warnings. As before, entry
  78.  * zero does nothing.
  79.  */
  80. static char *suppressed_what[1+ERR_WARN_MAX] = {
  81.     NULL, "macro calls with wrong no. of params",
  82.     "labels alone on lines without trailing `:'",
  83.     "numeric constants greater than 0xFFFFFFFF"
  84. };
  85.  
  86. /*
  87.  * This is a null preprocessor which just copies lines from input
  88.  * to output. It's used when someone explicitly requests that NASM
  89.  * not preprocess their source file.
  90.  */
  91.  
  92. static void no_pp_reset (char *, int, efunc, evalfunc, ListGen *);
  93. static char *no_pp_getline (void);
  94. static void no_pp_cleanup (void);
  95. static Preproc no_pp = {
  96.     no_pp_reset,
  97.     no_pp_getline,
  98.     no_pp_cleanup
  99. };
  100.  
  101. /*
  102.  * get/set current offset...
  103.  */
  104. #define get_curr_ofs (current_seg==NO_SEG?abs_offset:\
  105.               raa_read(offsets,current_seg))
  106. #define set_curr_ofs(x) (current_seg==NO_SEG?(void)(abs_offset=(x)):\
  107.              (void)(offsets=raa_write(offsets,current_seg,(x))))
  108.  
  109. static int want_usage;
  110. static int terminate_after_phase;
  111.  
  112. int main(int argc, char **argv) {
  113.     want_usage = terminate_after_phase = FALSE;
  114.  
  115.     nasm_set_malloc_error (report_error);
  116.     offsets = raa_init();
  117.     forwrefs = saa_init ((long)sizeof(int));
  118.  
  119.     preproc = &nasmpp;
  120.     preprocess_only = FALSE;
  121.  
  122.     seg_init();
  123.  
  124.     register_output_formats();
  125.  
  126.     parse_cmdline(argc, argv);
  127.  
  128.     if (terminate_after_phase) {
  129.     if (want_usage)
  130.         usage();
  131.     return 1;
  132.     }
  133.  
  134.     if (ofmt->stdmac)
  135.     pp_extra_stdmac (ofmt->stdmac);
  136.     eval_global_info (ofmt, lookup_label);
  137.  
  138.     if (preprocess_only) {
  139.     char *line;
  140.  
  141.     if (*outname) {
  142.         ofile = fopen(outname, "w");
  143.         if (!ofile)
  144.         report_error (ERR_FATAL | ERR_NOFILE,
  145.                   "unable to open output file `%s'", outname);
  146.     } else
  147.         ofile = NULL;
  148.  
  149.     eval_info ("%", 0L, 0L);       /* disallow labels, $ or $$ in exprs */
  150.  
  151.     preproc->reset (inname, 2, report_error, evaluate, &nasmlist);
  152.     strcpy(currentfile,inname);
  153.     lineno = 0;
  154.     lineinc = 1;
  155.     while ( (line = preproc->getline()) ) {
  156.         int ln, li;
  157.         char buf[FILENAME_MAX];
  158.  
  159.         lineno += lineinc;
  160.         /*
  161.          * We must still check for %line directives, so that we
  162.          * can report errors accurately.
  163.          */
  164.         if (!strncmp(line, "%line", 5) &&
  165.         sscanf(line, "%%line %d+%d %s", &ln, &li, buf) == 3) {
  166.         lineno = ln - li;
  167.         lineinc = li;
  168.         strncpy (currentfile, buf, FILENAME_MAX-1);
  169.         currentfile[FILENAME_MAX-1] = '\0';
  170.         }
  171.         if (ofile) {
  172.         fputs(line, ofile);
  173.         fputc('\n', ofile);
  174.         } else
  175.         puts(line);
  176.         nasm_free (line);
  177.     }
  178.     preproc->cleanup();
  179.     if (ofile)
  180.         fclose(ofile);
  181.     if (ofile && terminate_after_phase)
  182.         remove(outname);
  183.     } else {
  184.     /*
  185.      * We must call ofmt->filename _anyway_, even if the user
  186.      * has specified their own output file, because some
  187.      * formats (eg OBJ and COFF) use ofmt->filename to find out
  188.      * the name of the input file and then put that inside the
  189.      * file.
  190.      */
  191.     ofmt->filename (inname, outname, report_error);
  192.  
  193.     ofile = fopen(outname, "wb");
  194.     if (!ofile) {
  195.         report_error (ERR_FATAL | ERR_NOFILE,
  196.               "unable to open output file `%s'", outname);
  197.     }
  198.     /*
  199.      * We must call init_labels() before ofmt->init() since
  200.      * some object formats will want to define labels in their
  201.      * init routines. (eg OS/2 defines the FLAT group)
  202.      */
  203.     init_labels ();
  204.     ofmt->init (ofile, report_error, define_label, evaluate);
  205.     assemble_file (inname);
  206.     if (!terminate_after_phase) {
  207.         ofmt->cleanup ();
  208.         cleanup_labels ();
  209.     }
  210.     /*
  211.      * We had an fclose on the output file here, but we
  212.      * actually do that in all the object file drivers as well,
  213.      * so we're leaving out the one here.
  214.      *     fclose (ofile);
  215.      */
  216.     if (terminate_after_phase) {
  217.         remove(outname);
  218.         if (listname[0])
  219.         remove(listname);
  220.     }
  221.     }
  222.  
  223.     if (want_usage)
  224.     usage();
  225.     raa_free (offsets);
  226.     saa_free (forwrefs);
  227.  
  228.     if (terminate_after_phase)
  229.     return 1;
  230.     else
  231.     return 0;
  232. }
  233.  
  234. static int process_arg (char *p, char *q) {
  235.     char *param;
  236.     int i;
  237.     int advance = 0;
  238.  
  239.     if (!p || !p[0])
  240.     return 0;
  241.  
  242.     if (p[0]=='-') {
  243.     switch (p[1]) {
  244.       case 's':
  245.         use_stdout = TRUE;
  246.         break;
  247.       case 'o':               /* these parameters take values */
  248.       case 'f':
  249.       case 'p':
  250.       case 'd':
  251.       case 'i':
  252.       case 'l':
  253.         if (p[2])               /* the parameter's in the option */
  254.         param = p+2;
  255.         else if (!q) {
  256.         report_error (ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
  257.                   "option `-%c' requires an argument",
  258.                   p[1]);
  259.         break;
  260.         } else
  261.         advance = 1, param = q;
  262.         if (p[1]=='o') {           /* output file */
  263.         strcpy (outname, param);
  264.         } else if (p[1]=='f') {    /* output format */
  265.         ofmt = ofmt_find(param);
  266.         if (!ofmt) {
  267.             report_error (ERR_FATAL | ERR_NOFILE | ERR_USAGE,
  268.                   "unrecognised output format `%s'",
  269.                   param);
  270.         }
  271.         } else if (p[1]=='p') {    /* pre-include */
  272.         pp_pre_include (param);
  273.         } else if (p[1]=='d') {    /* pre-define */
  274.         pp_pre_define (param);
  275.         } else if (p[1]=='i') {    /* include search path */
  276.         pp_include_path (param);
  277.         } else if (p[1]=='l') {    /* listing file */
  278.         strcpy (listname, param);
  279.         }
  280.         break;
  281.       case 'h':
  282.         fprintf(use_stdout ? stdout : stderr,
  283.             "usage: nasm [-o outfile] [-f format] [-l listfile]"
  284.             " [options...] filename\n");
  285.         fprintf(use_stdout ? stdout : stderr,
  286.             "    or nasm -r   for version info\n\n");
  287.         fprintf(use_stdout ? stdout : stderr,
  288.             "    -e means preprocess only; "
  289.             "-a means don't preprocess\n");
  290.         fprintf(use_stdout ? stdout : stderr,
  291.             "    -s means send errors to stdout not stderr\n");
  292.         fprintf(use_stdout ? stdout : stderr,
  293.             "    -i<path> adds a pathname to the include file"
  294.             " path\n    -p<file> pre-includes a file;"
  295.             " -d<macro>[=<value] pre-defines a macro\n");
  296.         fprintf(use_stdout ? stdout : stderr,
  297.             "    -w+foo enables warnings about foo; "
  298.             "-w-foo disables them\n  where foo can be:\n");
  299.         for (i=1; i<=ERR_WARN_MAX; i++)
  300.         fprintf(use_stdout ? stdout : stderr,
  301.             "    %-16s%s (default %s)\n",
  302.             suppressed_names[i], suppressed_what[i],
  303.             suppressed[i] ? "off" : "on");
  304.         fprintf(use_stdout ? stdout : stderr,
  305.             "\nvalid output formats for -f are"
  306.             " (`*' denotes default):\n");
  307.         ofmt_list(ofmt, use_stdout ? stdout : stderr);
  308.         exit (0);               /* never need usage message here */
  309.         break;
  310.       case 'r':
  311.         fprintf(use_stdout ? stdout : stderr,
  312.             "NASM version %s\n", NASM_VER);
  313.         exit (0);               /* never need usage message here */
  314.         break;
  315.       case 'e':               /* preprocess only */
  316.         preprocess_only = TRUE;
  317.         break;
  318.       case 'a':               /* assemble only - don't preprocess */
  319.         preproc = &no_pp;
  320.         break;
  321.       case 'w':
  322.         if (p[2] != '+' && p[2] != '-') {
  323.         report_error (ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
  324.                   "invalid option to `-w'");
  325.         } else {
  326.         for (i=1; i<=ERR_WARN_MAX; i++)
  327.             if (!nasm_stricmp(p+3, suppressed_names[i]))
  328.             break;
  329.         if (i <= ERR_WARN_MAX)
  330.             suppressed[i] = (p[2] == '-');
  331.         else
  332.             report_error (ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
  333.                   "invalid option to `-w'");
  334.         }
  335.         break;
  336.       default:
  337.         report_error (ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
  338.               "unrecognised option `-%c'",
  339.               p[1]);
  340.         break;
  341.     }
  342.     } else {
  343.     if (*inname) {
  344.         report_error (ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
  345.               "more than one input file specified");
  346.     } else
  347.         strcpy(inname, p);
  348.     }
  349.  
  350.     return advance;
  351. }
  352.  
  353. static void parse_cmdline(int argc, char **argv) {
  354.     char *envreal, *envcopy, *p, *q, *arg, *prevarg;
  355.     char separator = ' ';
  356.  
  357.     *inname = *outname = *listname = '\0';
  358.  
  359.     /*
  360.      * First, process the NASM environment variable.
  361.      */
  362.     envreal = getenv("NASM");
  363.     arg = NULL;
  364.     if (envreal) {
  365.     envcopy = nasm_strdup(envreal);
  366.     p = envcopy;
  367.     if (*p && *p != '-')
  368.         separator = *p++;
  369.     while (*p) {
  370.         q = p;
  371.         while (*p && *p != separator) p++;
  372.         while (*p == separator) *p++ = '\0';
  373.         prevarg = arg;
  374.         arg = q;
  375.         if (process_arg (prevarg, arg))
  376.         arg = NULL;
  377.     }
  378.     nasm_free (envcopy);
  379.     }
  380.     if (arg)
  381.     process_arg (arg, NULL);
  382.  
  383.     /*
  384.      * Now process the actual command line.
  385.      */
  386.     while (--argc) {
  387.     int i;
  388.     argv++;
  389.     i = process_arg (argv[0], argc > 1 ? argv[1] : NULL);
  390.     argv += i, argc -= i;
  391.     }
  392.  
  393.     if (!*inname)
  394.     report_error (ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
  395.               "no input file specified");
  396. }
  397.  
  398. static void assemble_file (char *fname) {
  399.     char *value, *p, *q, *special, *line;
  400.     insn output_ins;
  401.     int i, rn_error, validid;
  402.     long seg, offs;
  403.     struct tokenval tokval;
  404.     expr *e;
  405.  
  406.     /* pass one */
  407.     pass = 1;
  408.     current_seg = ofmt->section(NULL, pass, &sb);
  409.     preproc->reset(fname, 1, report_error, evaluate, &nasmlist);
  410.     strcpy(currentfile,fname);
  411.     lineno = 0;
  412.     lineinc = 1;
  413.     globallineno = 0;
  414.     offs = get_curr_ofs;
  415.     eval_info (NULL, current_seg, offs);   /* set $ */
  416.     while ( (line = preproc->getline()) ) {
  417.     lineno += lineinc;
  418.     globallineno++;
  419.  
  420.     if (line[0] == '%') {
  421.         int ln, li;
  422.         char buf[FILENAME_MAX];
  423.  
  424.         /*
  425.          * This will be a line number directive. They come
  426.          * straight from the preprocessor, so we'll subject
  427.          * them to only minimal error checking.
  428.          */
  429.         if (strncmp(line, "%line", 5)) {
  430.         if (preproc == &no_pp)
  431.             report_error (ERR_WARNING, "unknown `%%' directive in "
  432.                   " preprocessed source");
  433.         } else if (sscanf(line, "%%line %d+%d %s", &ln, &li, buf) != 3) {
  434.         report_error (ERR_WARNING, "bogus line number directive in"
  435.                   " preprocessed source");
  436.         } else {
  437.         lineno = ln - li;
  438.         lineinc = li;
  439.         strncpy (currentfile, buf, FILENAME_MAX-1);
  440.         currentfile[FILENAME_MAX-1] = '\0';
  441.         }
  442.         continue;
  443.     }
  444.  
  445.     /* here we parse our directives; this is not handled by the 'real'
  446.      * parser. */
  447.     if ( (i = getkw (line, &value)) ) {
  448.         switch (i) {
  449.           case 1:           /* [SEGMENT n] */
  450.         seg = ofmt->section (value, pass, &sb);
  451.         if (seg == NO_SEG) {
  452.             report_error (ERR_NONFATAL,
  453.                   "segment name `%s' not recognised",
  454.                   value);
  455.         } else {
  456.             current_seg = seg;
  457.         }
  458.         break;
  459.           case 2:           /* [EXTERN label:special] */
  460.         if (*value == '$')
  461.             value++;           /* skip initial $ if present */
  462.         q = value;
  463.         validid = TRUE;
  464.         if (!isidstart(*q))
  465.             validid = FALSE;
  466.         while (*q && *q != ':') {
  467.             if (!isidchar(*q))
  468.             validid = FALSE;
  469.             q++;
  470.         }
  471.         if (!validid) {
  472.             report_error (ERR_NONFATAL,
  473.                   "identifier expected after EXTERN");
  474.             break;
  475.         }
  476.         if (*q == ':') {
  477.             *q++ = '\0';
  478.             special = q;
  479.         } else
  480.             special = NULL;
  481.         if (!is_extern(value)) {   /* allow re-EXTERN to be ignored */
  482.             declare_as_global (value, special, report_error);
  483.             define_label (value, seg_alloc(), 0L, NULL, FALSE, TRUE,
  484.                   ofmt, report_error);
  485.         }
  486.         break;
  487.           case 3:           /* [BITS bits] */
  488.         switch (atoi(value)) {
  489.           case 16:
  490.           case 32:
  491.             sb = atoi(value);
  492.             break;
  493.           default:
  494.             report_error(ERR_NONFATAL,
  495.                  "`%s' is not a valid argument to [BITS]",
  496.                  value);
  497.             break;
  498.         }
  499.         break;
  500.           case 4:           /* [GLOBAL symbol:special] */
  501.         if (*value == '$')
  502.             value++;           /* skip initial $ if present */
  503.         q = value;
  504.         validid = TRUE;
  505.         if (!isidstart(*q))
  506.             validid = FALSE;
  507.         while (*q && *q != ':') {
  508.             if (!isidchar(*q))
  509.             validid = FALSE;
  510.             q++;
  511.         }
  512.         if (!validid) {
  513.             report_error (ERR_NONFATAL,
  514.                   "identifier expected after GLOBAL");
  515.             break;
  516.         }
  517.         if (*q == ':') {
  518.             *q++ = '\0';
  519.             special = q;
  520.         } else
  521.             special = NULL;
  522.         declare_as_global (value, special, report_error);
  523.         break;
  524.           case 5:           /* [COMMON symbol size:special] */
  525.         p = value;
  526.         validid = TRUE;
  527.         if (!isidstart(*p))
  528.             validid = FALSE;
  529.         while (*p && !isspace(*p)) {
  530.             if (!isidchar(*p))
  531.             validid = FALSE;
  532.             p++;
  533.         }
  534.         if (!validid) {
  535.             report_error (ERR_NONFATAL,
  536.                   "identifier expected after COMMON");
  537.             break;
  538.         }
  539.         if (*p) {
  540.             long size;
  541.  
  542.             while (*p && isspace(*p))
  543.             *p++ = '\0';
  544.             q = p;
  545.             while (*q && *q != ':')
  546.             q++;
  547.             if (*q == ':') {
  548.             *q++ = '\0';
  549.             special = q;
  550.             } else
  551.             special = NULL;
  552.             size = readnum (p, &rn_error);
  553.             if (rn_error)
  554.             report_error (ERR_NONFATAL, "invalid size specified"
  555.                       " in COMMON declaration");
  556.             else
  557.             define_common (value, seg_alloc(), size,
  558.                        special, ofmt, report_error);
  559.         } else
  560.             report_error (ERR_NONFATAL, "no size specified in"
  561.                   " COMMON declaration");
  562.         break;
  563.           case 6:               /* [ABSOLUTE address] */
  564.         current_seg = NO_SEG;
  565.         stdscan_reset();
  566.         stdscan_bufptr = value;
  567.         tokval.t_type = TOKEN_INVALID;
  568.         e = evaluate(stdscan, NULL, &tokval, NULL, 1, report_error,
  569.                  NULL);
  570.         if (e) {
  571.             if (!is_reloc(e))
  572.             report_error (ERR_NONFATAL, "cannot use non-"
  573.                       "relocatable expression as ABSOLUTE"
  574.                       " address");
  575.             else {
  576.             abs_seg = reloc_seg(e);
  577.             abs_offset = reloc_value(e);
  578.             }
  579.         } else
  580.             abs_offset = 0x100;/* don't go near zero in case of / */
  581.         break;
  582.           default:
  583.         if (!ofmt->directive (line+1, value, 1))
  584.             report_error (ERR_NONFATAL, "unrecognised directive [%s]",
  585.                   line+1);
  586.         break;
  587.         }
  588.     } else {
  589.         parse_line (1, line, &output_ins,
  590.             report_error, evaluate, eval_info);
  591.         if (output_ins.forw_ref)
  592.         *(int *)saa_wstruct(forwrefs) = globallineno;
  593.  
  594.         /*
  595.          * Hack to prevent phase error in the code
  596.          *   rol ax,x
  597.          *   x equ 1
  598.          *
  599.          * We rule that the presence of a forward reference
  600.          * cancels out the UNITY property of the number 1. This
  601.          * isn't _strictly_ necessary in pass one, since the
  602.          * problem occurs in pass two, but for the sake of
  603.          * having the passes as near to identical as we can
  604.          * manage, we do it like this.
  605.          */
  606.         if (output_ins.forw_ref) {
  607.         int i;
  608.         for (i=0; i<output_ins.operands; i++)
  609.             output_ins.oprs[i].type &= ~ONENESS;
  610.         }
  611.  
  612.         if (output_ins.opcode == I_EQU) {
  613.         /*
  614.          * Special `..' EQUs get processed in pass two,
  615.          * except `..@' macro-processor EQUs which are done
  616.          * in the normal place.
  617.          */
  618.         if (!output_ins.label)
  619.             report_error (ERR_NONFATAL,
  620.                   "EQU not preceded by label");
  621.         else if (output_ins.label[0] != '.' ||
  622.              output_ins.label[1] != '.' ||
  623.              output_ins.label[2] == '@') {
  624.             if (output_ins.operands == 1 &&
  625.             (output_ins.oprs[0].type & IMMEDIATE) &&
  626.             output_ins.oprs[0].wrt == NO_SEG) {
  627.             define_label (output_ins.label,
  628.                       output_ins.oprs[0].segment,
  629.                       output_ins.oprs[0].offset,
  630.                       NULL, FALSE, FALSE, ofmt, report_error);
  631.             } else if (output_ins.operands == 2 &&
  632.                    (output_ins.oprs[0].type & IMMEDIATE) &&
  633.                    (output_ins.oprs[0].type & COLON) &&
  634.                    output_ins.oprs[0].segment == NO_SEG &&
  635.                    output_ins.oprs[0].wrt == NO_SEG &&
  636.                    (output_ins.oprs[1].type & IMMEDIATE) &&
  637.                    output_ins.oprs[1].segment == NO_SEG &&
  638.                    output_ins.oprs[1].wrt == NO_SEG) {
  639.             define_label (output_ins.label,
  640.                       output_ins.oprs[0].offset | SEG_ABS,
  641.                       output_ins.oprs[1].offset,
  642.                       NULL, FALSE, FALSE, ofmt, report_error);
  643.             } else
  644.             report_error(ERR_NONFATAL, "bad syntax for EQU");
  645.         }
  646.         } else {
  647.         if (output_ins.label)
  648.             define_label (output_ins.label,
  649.                   current_seg==NO_SEG ? abs_seg : current_seg,
  650.                   offs, NULL, TRUE, FALSE, ofmt, report_error);
  651.         offs += insn_size (current_seg, offs, sb,
  652.                    &output_ins, report_error);
  653.         set_curr_ofs (offs);
  654.         }
  655.         cleanup_insn (&output_ins);
  656.     }
  657.     nasm_free (line);
  658.     offs = get_curr_ofs;
  659.     eval_info (NULL, current_seg, offs);   /* set $ */
  660.     }
  661.     preproc->cleanup();
  662.  
  663.     if (terminate_after_phase) {
  664.     fclose(ofile);
  665.     remove(outname);
  666.     if (want_usage)
  667.         usage();
  668.     exit (1);
  669.     }
  670.  
  671.     /* pass two */
  672.     pass = 2;
  673.     saa_rewind (forwrefs);
  674.     if (*listname)
  675.     nasmlist.init(listname, report_error);
  676.     {
  677.     int *p = saa_rstruct (forwrefs);
  678.     if (p)
  679.         forwline = *p;
  680.     else
  681.         forwline = -1;
  682.     }
  683.     current_seg = ofmt->section(NULL, pass, &sb);
  684.     raa_free (offsets);
  685.     offsets = raa_init();
  686.     preproc->reset(fname, 2, report_error, evaluate, &nasmlist);
  687.     strcpy(currentfile,fname);
  688.     lineno = 0;
  689.     lineinc = 1;
  690.     globallineno = 0;
  691.     offs = get_curr_ofs;
  692.     eval_info (NULL, current_seg, offs);   /* set $ */
  693.     while ( (line = preproc->getline()) ) {
  694.     lineno += lineinc;
  695.     globallineno++;
  696.  
  697.     if (line[0] == '%') {
  698.         int ln, li;
  699.         char buf[FILENAME_MAX];
  700.  
  701.         /*
  702.          * This will be a line number directive. They come
  703.          * straight from the preprocessor, so we'll subject
  704.          * them to only minimal error checking.
  705.          */
  706.         if (!strncmp(line, "%line", 5) &&
  707.         sscanf(line, "%%line %d+%d %s", &ln, &li, buf) == 3) {
  708.         lineno = ln - li;
  709.         lineinc = li;
  710.         strncpy (currentfile, buf, FILENAME_MAX-1);
  711.         currentfile[FILENAME_MAX-1] = '\0';
  712.         }
  713.         continue;
  714.     }
  715.  
  716.     /* here we parse our directives; this is not handled by
  717.      * the 'real' parser. */
  718.     if ( (i = getkw (line, &value)) ) {
  719.         switch (i) {
  720.           case 1:           /* [SEGMENT n] */
  721.         seg = ofmt->section (value, pass, &sb);
  722.         if (seg == NO_SEG) {
  723.             report_error (ERR_PANIC,
  724.                   "invalid segment name on pass two");
  725.         } else
  726.             current_seg = seg;
  727.         break;
  728.           case 2:           /* [EXTERN label] */
  729.         q = value;
  730.         while (*q && *q != ':')
  731.             q++;
  732.         if (*q == ':') {
  733.             *q++ = '\0';
  734.             ofmt->symdef(value, 0L, 0L, 3, q);
  735.         }
  736.         break;
  737.           case 3:           /* [BITS bits] */
  738.         switch (atoi(value)) {
  739.           case 16:
  740.           case 32:
  741.             sb = atoi(value);
  742.             break;
  743.           default:
  744.             report_error(ERR_PANIC,
  745.                  "invalid [BITS] value on pass two",
  746.                  value);
  747.             break;
  748.         }
  749.         break;
  750.           case 4:               /* [GLOBAL symbol] */
  751.         q = value;
  752.         while (*q && *q != ':')
  753.             q++;
  754.         if (*q == ':') {
  755.             *q++ = '\0';
  756.             ofmt->symdef(value, 0L, 0L, 3, q);
  757.         }
  758.         break;
  759.           case 5:               /* [COMMON symbol size] */
  760.         q = value;
  761.         while (*q && *q != ':') {
  762.             if (isspace(*q))
  763.             *q = '\0';
  764.             q++;
  765.         }
  766.         if (*q == ':') {
  767.             *q++ = '\0';
  768.             ofmt->symdef(value, 0L, 0L, 3, q);
  769.         }
  770.         break;
  771.           case 6:               /* [ABSOLUTE addr] */
  772.         current_seg = NO_SEG;
  773.         stdscan_reset();
  774.         stdscan_bufptr = value;
  775.         tokval.t_type = TOKEN_INVALID;
  776.         e = evaluate(stdscan, NULL, &tokval, NULL, 2, report_error,
  777.                  NULL);
  778.         if (e) {
  779.             if (!is_reloc(e))
  780.             report_error (ERR_PANIC, "non-reloc ABSOLUTE address"
  781.                       " in pass two");
  782.             else {
  783.             abs_seg = reloc_seg(e);
  784.             abs_offset = reloc_value(e);
  785.             }
  786.         } else
  787.             report_error (ERR_PANIC, "invalid ABSOLUTE address "
  788.                   "in pass two");
  789.         break;
  790.           default:
  791.         if (!ofmt->directive (line+1, value, 2))
  792.             report_error (ERR_PANIC, "invalid directive on pass two");
  793.         break;
  794.         }
  795.     } else {
  796.         parse_line (2, line, &output_ins,
  797.             report_error, evaluate, eval_info);
  798.         if (globallineno == forwline) {
  799.         int *p = saa_rstruct (forwrefs);
  800.         if (p)
  801.             forwline = *p;
  802.         else
  803.             forwline = -1;
  804.         output_ins.forw_ref = TRUE;
  805.         } else
  806.         output_ins.forw_ref = FALSE;
  807.  
  808.         /*
  809.          * Hack to prevent phase error in the code
  810.          *   rol ax,x
  811.          *   x equ 1
  812.          */
  813.         if (output_ins.forw_ref) {
  814.         int i;
  815.         for (i=0; i<output_ins.operands; i++)
  816.             output_ins.oprs[i].type &= ~ONENESS;
  817.         }
  818.  
  819.         obuf = line;
  820.         if (output_ins.label)
  821.         define_label_stub (output_ins.label, report_error);
  822.         if (output_ins.opcode == I_EQU) {
  823.         /*
  824.          * Special `..' EQUs get processed here, except
  825.          * `..@' macro processor EQUs which are done above.
  826.          */
  827.         if (output_ins.label[0] == '.' &&
  828.             output_ins.label[1] == '.' &&
  829.             output_ins.label[2] != '@') {
  830.             if (output_ins.operands == 1 &&
  831.             (output_ins.oprs[0].type & IMMEDIATE)) {
  832.             define_label (output_ins.label,
  833.                       output_ins.oprs[0].segment,
  834.                       output_ins.oprs[0].offset,
  835.                       NULL, FALSE, FALSE, ofmt, report_error);
  836.             } else if (output_ins.operands == 2 &&
  837.                    (output_ins.oprs[0].type & IMMEDIATE) &&
  838.                    (output_ins.oprs[0].type & COLON) &&
  839.                    output_ins.oprs[0].segment == NO_SEG &&
  840.                    (output_ins.oprs[1].type & IMMEDIATE) &&
  841.                    output_ins.oprs[1].segment == NO_SEG) {
  842.             define_label (output_ins.label,
  843.                       output_ins.oprs[0].offset | SEG_ABS,
  844.                       output_ins.oprs[1].offset,
  845.                       NULL, FALSE, FALSE, ofmt, report_error);
  846.             } else
  847.             report_error(ERR_NONFATAL, "bad syntax for EQU");
  848.         }
  849.         }
  850.         offs += assemble (current_seg, offs, sb,
  851.                   &output_ins, ofmt, report_error, &nasmlist);
  852.         cleanup_insn (&output_ins);
  853.         set_curr_ofs (offs);
  854.     }
  855.     nasm_free (line);
  856.  
  857.     offs = get_curr_ofs;
  858.     eval_info (NULL, current_seg, offs);   /* set $ */
  859.     }
  860.     preproc->cleanup();
  861.     nasmlist.cleanup();
  862. }
  863.  
  864. static int getkw (char *buf, char **value) {
  865.     char *p, *q;
  866.  
  867.     if (*buf!='[')
  868.         return 0;
  869.     p = buf;
  870.     while (*p && *p != ']') p++;
  871.     if (!*p)
  872.     return 0;
  873.     q = p++;
  874.     while (*p && *p != ';') {
  875.     if (!isspace(*p))
  876.         return 0;
  877.     p++;
  878.     }
  879.     q[1] = '\0';
  880.  
  881.     p = buf+1;
  882.     while (*buf && *buf!=' ' && *buf!=']' && *buf!='\t')
  883.         buf++;
  884.     if (*buf==']') {
  885.     *buf = '\0';
  886.     *value = buf;
  887.     } else {
  888.     *buf++ = '\0';
  889.         while (isspace(*buf)) buf++;   /* beppu - skip leading whitespace */
  890.     *value = buf;
  891.     while (*buf!=']') buf++;
  892.     *buf++ = '\0';
  893.     }
  894.     for (q=p; *q; q++)
  895.     *q = tolower(*q);
  896.     if (!strcmp(p, "segment") || !strcmp(p, "section"))
  897.         return 1;
  898.     if (!strcmp(p, "extern"))
  899.         return 2;
  900.     if (!strcmp(p, "bits"))
  901.         return 3;
  902.     if (!strcmp(p, "global"))
  903.         return 4;
  904.     if (!strcmp(p, "common"))
  905.         return 5;
  906.     if (!strcmp(p, "absolute"))
  907.         return 6;
  908.     return -1;
  909. }
  910.  
  911. static void report_error (int severity, char *fmt, ...) {
  912.     va_list ap;
  913.  
  914.     /*
  915.      * See if it's a suppressed warning.
  916.      */
  917.     if ((severity & ERR_MASK) == ERR_WARNING &&
  918.     (severity & ERR_WARN_MASK) != 0 &&
  919.     suppressed[ (severity & ERR_WARN_MASK) >> ERR_WARN_SHR ])
  920.     return;                   /* and bail out if so */
  921.  
  922.     /*
  923.      * See if it's a pass-one only warning and we're not in pass one.
  924.      */
  925.     if ((severity & ERR_PASS1) && pass != 1)
  926.     return;
  927.  
  928.     if (severity & ERR_NOFILE)
  929.     fputs ("nasm: ", use_stdout ? stdout : stderr);
  930.     else
  931.     fprintf (use_stdout ? stdout : stderr, "%s:%d: ", currentfile,
  932.          lineno + (severity & ERR_OFFBY1 ? lineinc : 0));
  933.  
  934.     if ( (severity & ERR_MASK) == ERR_WARNING)
  935.     fputs ("warning: ", use_stdout ? stdout : stderr);
  936.     else if ( (severity & ERR_MASK) == ERR_PANIC)
  937.     fputs ("panic: ", use_stdout ? stdout : stderr);
  938.  
  939.     va_start (ap, fmt);
  940.     vfprintf (use_stdout ? stdout : stderr, fmt, ap);
  941.     fputc ('\n', use_stdout ? stdout : stderr);
  942.  
  943.     if (severity & ERR_USAGE)
  944.     want_usage = TRUE;
  945.  
  946.     switch (severity & ERR_MASK) {
  947.       case ERR_WARNING:
  948.     /* no further action, by definition */
  949.     break;
  950.       case ERR_NONFATAL:
  951.     terminate_after_phase = TRUE;
  952.     break;
  953.       case ERR_FATAL:
  954.     if (ofile) {
  955.         fclose(ofile);
  956.         remove(outname);
  957.     }
  958.     if (want_usage)
  959.         usage();
  960.     exit(1);               /* instantly die */
  961.     break;                   /* placate silly compilers */
  962.       case ERR_PANIC:
  963.     abort();               /* halt, catch fire, and dump core */
  964.     break;
  965.     }
  966. }
  967.  
  968. static void usage(void) {
  969.     fputs("type `nasm -h' for help\n", use_stdout ? stdout : stderr);
  970. }
  971.  
  972. static void register_output_formats(void) {
  973.     /* Flat-form binary format */
  974. #ifdef OF_BIN
  975.     extern struct ofmt of_bin;
  976. #endif
  977.     /* Unix formats: a.out, COFF, ELF */
  978. #ifdef OF_AOUT
  979.     extern struct ofmt of_aout;
  980. #endif
  981. #ifdef OF_AOUTB
  982.     extern struct ofmt of_aoutb;
  983. #endif
  984. #ifdef OF_COFF
  985.     extern struct ofmt of_coff;
  986. #endif
  987. #ifdef OF_ELF
  988.     extern struct ofmt of_elf;
  989. #endif
  990.     /* Linux strange format: as86 */
  991. #ifdef OF_AS86
  992.     extern struct ofmt of_as86;
  993. #endif
  994.     /* DOS and DOS-ish formats: OBJ, OS/2, Win32 */
  995. #ifdef OF_OBJ
  996.     extern struct ofmt of_obj;
  997. #endif
  998. #ifdef OF_WIN32
  999.     extern struct ofmt of_win32;
  1000. #endif
  1001. #ifdef OF_RDF
  1002.     extern struct ofmt of_rdf;
  1003. #endif
  1004. #ifdef OF_DBG     /* debug format must be included specifically */
  1005.     extern struct ofmt of_dbg;
  1006. #endif
  1007.  
  1008. #ifdef OF_BIN
  1009.     ofmt_register (&of_bin);
  1010. #endif
  1011. #ifdef OF_AOUT
  1012.     ofmt_register (&of_aout);
  1013. #endif
  1014. #ifdef OF_AOUTB
  1015.     ofmt_register (&of_aoutb);
  1016. #endif
  1017. #ifdef OF_COFF
  1018.     ofmt_register (&of_coff);
  1019. #endif
  1020. #ifdef OF_ELF
  1021.     ofmt_register (&of_elf);
  1022. #endif
  1023. #ifdef OF_AS86
  1024.     ofmt_register (&of_as86);
  1025. #endif
  1026. #ifdef OF_OBJ
  1027.     ofmt_register (&of_obj);
  1028. #endif
  1029. #ifdef OF_WIN32
  1030.     ofmt_register (&of_win32);
  1031. #endif
  1032. #ifdef OF_RDF
  1033.     ofmt_register (&of_rdf);
  1034. #endif
  1035. #ifdef OF_DBG
  1036.     ofmt_register (&of_dbg);
  1037. #endif
  1038.     /*
  1039.      * set the default format
  1040.      */
  1041.     ofmt = &OF_DEFAULT;
  1042. }
  1043.  
  1044. #define BUF_DELTA 512
  1045.  
  1046. static FILE *no_pp_fp;
  1047. static efunc no_pp_err;
  1048. static ListGen *no_pp_list;
  1049.  
  1050. static void no_pp_reset (char *file, int pass, efunc error, evalfunc eval,
  1051.              ListGen *listgen) {
  1052.     no_pp_err = error;
  1053.     no_pp_fp = fopen(file, "r");
  1054.     if (!no_pp_fp)
  1055.     no_pp_err (ERR_FATAL | ERR_NOFILE,
  1056.            "unable to open input file `%s'", file);
  1057.     no_pp_list = listgen;
  1058.     (void) pass;               /* placate compilers */
  1059.     (void) eval;               /* placate compilers */
  1060. }
  1061.  
  1062. static char *no_pp_getline (void) {
  1063.     char *buffer, *p, *q;
  1064.     int bufsize;
  1065.  
  1066.     bufsize = BUF_DELTA;
  1067.     buffer = nasm_malloc(BUF_DELTA);
  1068.     p = buffer;
  1069.     while (1) {
  1070.     q = fgets(p, bufsize-(p-buffer), no_pp_fp);
  1071.     if (!q)
  1072.         break;
  1073.     p += strlen(p);
  1074.     if (p > buffer && p[-1] == '\n')
  1075.         break;
  1076.     if (p-buffer > bufsize-10) {
  1077.         bufsize += BUF_DELTA;
  1078.         buffer = nasm_realloc(buffer, bufsize);
  1079.     }
  1080.     }
  1081.  
  1082.     if (!q && p == buffer) {
  1083.     nasm_free (buffer);
  1084.     return NULL;
  1085.     }
  1086.  
  1087.     /*
  1088.      * Play safe: remove CRs as well as LFs, if any of either are
  1089.      * present at the end of the line.
  1090.      */
  1091.     while (p > buffer && (p[-1] == '\n' || p[-1] == '\r'))
  1092.     *--p = '\0';
  1093.  
  1094.     /*
  1095.      * Handle spurious ^Z, which may be inserted into source files
  1096.      * by some file transfer utilities.
  1097.      */
  1098.     buffer[strcspn(buffer, "\032")] = '\0';
  1099.  
  1100.     no_pp_list->line (LIST_READ, buffer);
  1101.  
  1102.     return buffer;
  1103. }
  1104.  
  1105. static void no_pp_cleanup (void) {
  1106.     fclose(no_pp_fp);
  1107. }
  1108.