nasm.c: getkw -- use string helpers
[platform/upstream/nasm.git] / nasm.c
1 /* ----------------------------------------------------------------------- *
2  *
3  *   Copyright 1996-2009 The NASM Authors - All Rights Reserved
4  *   See the file AUTHORS included with the NASM distribution for
5  *   the specific copyright holders.
6  *
7  *   Redistribution and use in source and binary forms, with or without
8  *   modification, are permitted provided that the following
9  *   conditions are met:
10  *
11  *   * Redistributions of source code must retain the above copyright
12  *     notice, this list of conditions and the following disclaimer.
13  *   * Redistributions in binary form must reproduce the above
14  *     copyright notice, this list of conditions and the following
15  *     disclaimer in the documentation and/or other materials provided
16  *     with the distribution.
17  *
18  *     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
19  *     CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
20  *     INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
21  *     MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
22  *     DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
23  *     CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24  *     SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
25  *     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26  *     LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  *     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28  *     CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
29  *     OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
30  *     EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  *
32  * ----------------------------------------------------------------------- */
33
34 /*
35  * The Netwide Assembler main program module
36  */
37
38 #include "compiler.h"
39
40 #include <stdio.h>
41 #include <stdarg.h>
42 #include <stdlib.h>
43 #include <string.h>
44 #include <ctype.h>
45 #include <inttypes.h>
46 #include <limits.h>
47 #include <time.h>
48
49 #include "nasm.h"
50 #include "nasmlib.h"
51 #include "saa.h"
52 #include "raa.h"
53 #include "float.h"
54 #include "stdscan.h"
55 #include "insns.h"
56 #include "preproc.h"
57 #include "parser.h"
58 #include "eval.h"
59 #include "assemble.h"
60 #include "labels.h"
61 #include "output/outform.h"
62 #include "listing.h"
63
64 struct forwrefinfo {            /* info held on forward refs. */
65     int lineno;
66     int operand;
67 };
68
69 static int get_bits(char *value);
70 static uint32_t get_cpu(char *cpu_str);
71 static void parse_cmdline(int, char **);
72 static void assemble_file(char *, StrList **);
73 static void nasm_verror_gnu(int severity, const char *fmt, va_list args);
74 static void nasm_verror_vc(int severity, const char *fmt, va_list args);
75 static void nasm_verror_common(int severity, const char *fmt, va_list args);
76 static bool is_suppressed_warning(int severity);
77 static void usage(void);
78
79 static int using_debug_info, opt_verbose_info;
80 bool tasm_compatible_mode = false;
81 int pass0, passn;
82 int maxbits = 0;
83 int globalrel = 0;
84
85 static time_t official_compile_time;
86
87 static char inname[FILENAME_MAX];
88 static char outname[FILENAME_MAX];
89 static char listname[FILENAME_MAX];
90 static char errname[FILENAME_MAX];
91 static int globallineno;        /* for forward-reference tracking */
92 /* static int pass = 0; */
93 struct ofmt *ofmt = &OF_DEFAULT;
94 const struct dfmt *dfmt;
95
96 static FILE *error_file;        /* Where to write error messages */
97
98 FILE *ofile = NULL;
99 int optimizing = -1;            /* number of optimization passes to take */
100 static int sb, cmd_sb = 16;     /* by default */
101 static uint32_t cmd_cpu = IF_PLEVEL;       /* highest level by default */
102 static uint32_t cpu = IF_PLEVEL;   /* passed to insn_size & assemble.c */
103 int64_t global_offset_changed;      /* referenced in labels.c */
104 int64_t prev_offset_changed;
105 int32_t stall_count;
106
107 static struct location location;
108 int in_abs_seg;                 /* Flag we are in ABSOLUTE seg */
109 int32_t abs_seg;                   /* ABSOLUTE segment basis */
110 int32_t abs_offset;                /* ABSOLUTE offset */
111
112 static struct RAA *offsets;
113
114 static struct SAA *forwrefs;    /* keep track of forward references */
115 static const struct forwrefinfo *forwref;
116
117 static Preproc *preproc;
118 enum op_type {
119     op_normal,                  /* Preprocess and assemble */
120     op_preprocess,              /* Preprocess only */
121     op_depend,                  /* Generate dependencies */
122 };
123 static enum op_type operating_mode;
124 /* Dependency flags */
125 static bool depend_emit_phony = false;
126 static bool depend_missing_ok = false;
127 static const char *depend_target = NULL;
128 static const char *depend_file = NULL;
129
130 /*
131  * Which of the suppressible warnings are suppressed. Entry zero
132  * isn't an actual warning, but it used for -w+error/-Werror.
133  */
134
135 static bool warning_on[ERR_WARN_MAX+1]; /* Current state */
136 static bool warning_on_global[ERR_WARN_MAX+1]; /* Command-line state */
137
138 static const struct warning {
139     const char *name;
140     const char *help;
141     bool enabled;
142 } warnings[ERR_WARN_MAX+1] = {
143     {"error", "treat warnings as errors", false},
144     {"macro-params", "macro calls with wrong parameter count", true},
145     {"macro-selfref", "cyclic macro references", false},
146     {"macro-defaults", "macros with more default than optional parameters", true},
147     {"orphan-labels", "labels alone on lines without trailing `:'", true},
148     {"number-overflow", "numeric constant does not fit", true},
149     {"gnu-elf-extensions", "using 8- or 16-bit relocation in ELF32, a GNU extension", false},
150     {"float-overflow", "floating point overflow", true},
151     {"float-denorm", "floating point denormal", false},
152     {"float-underflow", "floating point underflow", false},
153     {"float-toolong", "too many digits in floating-point number", true},
154     {"user", "%warning directives", true},
155 };
156
157 /*
158  * This is a null preprocessor which just copies lines from input
159  * to output. It's used when someone explicitly requests that NASM
160  * not preprocess their source file.
161  */
162
163 static void no_pp_reset(char *, int, ListGen *, StrList **);
164 static char *no_pp_getline(void);
165 static void no_pp_cleanup(int);
166 static Preproc no_pp = {
167     no_pp_reset,
168     no_pp_getline,
169     no_pp_cleanup
170 };
171
172 /*
173  * get/set current offset...
174  */
175 #define GET_CURR_OFFS (in_abs_seg?abs_offset:\
176                       raa_read(offsets,location.segment))
177 #define SET_CURR_OFFS(x) (in_abs_seg?(void)(abs_offset=(x)):\
178                          (void)(offsets=raa_write(offsets,location.segment,(x))))
179
180 static bool want_usage;
181 static bool terminate_after_phase;
182 int user_nolist = 0;            /* fbk 9/2/00 */
183
184 static void nasm_fputs(const char *line, FILE * outfile)
185 {
186     if (outfile) {
187         fputs(line, outfile);
188         putc('\n', outfile);
189     } else
190         puts(line);
191 }
192
193 /* Convert a struct tm to a POSIX-style time constant */
194 static int64_t posix_mktime(struct tm *tm)
195 {
196     int64_t t;
197     int64_t y = tm->tm_year;
198
199     /* See IEEE 1003.1:2004, section 4.14 */
200
201     t = (y-70)*365 + (y-69)/4 - (y-1)/100 + (y+299)/400;
202     t += tm->tm_yday;
203     t *= 24;
204     t += tm->tm_hour;
205     t *= 60;
206     t += tm->tm_min;
207     t *= 60;
208     t += tm->tm_sec;
209
210     return t;
211 }
212
213 static void define_macros_early(void)
214 {
215     char temp[128];
216     struct tm lt, *lt_p, gm, *gm_p;
217     int64_t posix_time;
218
219     lt_p = localtime(&official_compile_time);
220     if (lt_p) {
221         lt = *lt_p;
222
223         strftime(temp, sizeof temp, "__DATE__=\"%Y-%m-%d\"", &lt);
224         pp_pre_define(temp);
225         strftime(temp, sizeof temp, "__DATE_NUM__=%Y%m%d", &lt);
226         pp_pre_define(temp);
227         strftime(temp, sizeof temp, "__TIME__=\"%H:%M:%S\"", &lt);
228         pp_pre_define(temp);
229         strftime(temp, sizeof temp, "__TIME_NUM__=%H%M%S", &lt);
230         pp_pre_define(temp);
231     }
232
233     gm_p = gmtime(&official_compile_time);
234     if (gm_p) {
235         gm = *gm_p;
236
237         strftime(temp, sizeof temp, "__UTC_DATE__=\"%Y-%m-%d\"", &gm);
238         pp_pre_define(temp);
239         strftime(temp, sizeof temp, "__UTC_DATE_NUM__=%Y%m%d", &gm);
240         pp_pre_define(temp);
241         strftime(temp, sizeof temp, "__UTC_TIME__=\"%H:%M:%S\"", &gm);
242         pp_pre_define(temp);
243         strftime(temp, sizeof temp, "__UTC_TIME_NUM__=%H%M%S", &gm);
244         pp_pre_define(temp);
245     }
246
247     if (gm_p)
248         posix_time = posix_mktime(&gm);
249     else if (lt_p)
250         posix_time = posix_mktime(&lt);
251     else
252         posix_time = 0;
253
254     if (posix_time) {
255         snprintf(temp, sizeof temp, "__POSIX_TIME__=%"PRId64, posix_time);
256         pp_pre_define(temp);
257     }
258 }
259
260 static void define_macros_late(void)
261 {
262     char temp[128];
263
264     snprintf(temp, sizeof(temp), "__OUTPUT_FORMAT__=%s\n",
265              ofmt->shortname);
266     pp_pre_define(temp);
267 }
268
269 static void emit_dependencies(StrList *list)
270 {
271     FILE *deps;
272     int linepos, len;
273     StrList *l, *nl;
274
275     if (depend_file && strcmp(depend_file, "-")) {
276         deps = fopen(depend_file, "w");
277         if (!deps) {
278             nasm_error(ERR_NONFATAL|ERR_NOFILE|ERR_USAGE,
279                          "unable to write dependency file `%s'", depend_file);
280             return;
281         }
282     } else {
283         deps = stdout;
284     }
285
286     linepos = fprintf(deps, "%s:", depend_target);
287     list_for_each(l, list) {
288         len = strlen(l->str);
289         if (linepos + len > 62) {
290             fprintf(deps, " \\\n ");
291             linepos = 1;
292         }
293         fprintf(deps, " %s", l->str);
294         linepos += len+1;
295     }
296     fprintf(deps, "\n\n");
297
298     list_for_each_safe(l, nl, list) {
299         if (depend_emit_phony)
300             fprintf(deps, "%s:\n\n", l->str);
301         nasm_free(l);
302     }
303
304     if (deps != stdout)
305         fclose(deps);
306 }
307
308 int main(int argc, char **argv)
309 {
310     StrList *depend_list = NULL, **depend_ptr;
311
312     time(&official_compile_time);
313
314     pass0 = 0;
315     want_usage = terminate_after_phase = false;
316     nasm_set_verror(nasm_verror_gnu);
317
318     error_file = stderr;
319
320     tolower_init();
321
322     nasm_init_malloc_error();
323     offsets = raa_init();
324     forwrefs = saa_init((int32_t)sizeof(struct forwrefinfo));
325
326     preproc = &nasmpp;
327     operating_mode = op_normal;
328
329     seg_init();
330
331     /* Define some macros dependent on the runtime, but not
332        on the command line. */
333     define_macros_early();
334
335     parse_cmdline(argc, argv);
336
337     if (terminate_after_phase) {
338         if (want_usage)
339             usage();
340         return 1;
341     }
342
343     /* If debugging info is disabled, suppress any debug calls */
344     if (!using_debug_info)
345         ofmt->current_dfmt = &null_debug_form;
346
347     if (ofmt->stdmac)
348         pp_extra_stdmac(ofmt->stdmac);
349     parser_global_info(&location);
350     eval_global_info(ofmt, lookup_label, &location);
351
352     /* define some macros dependent of command-line */
353     define_macros_late();
354
355     depend_ptr = (depend_file || (operating_mode == op_depend))
356         ? &depend_list : NULL;
357     if (!depend_target)
358         depend_target = outname;
359
360     switch (operating_mode) {
361     case op_depend:
362         {
363             char *line;
364
365             if (depend_missing_ok)
366                 pp_include_path(NULL);  /* "assume generated" */
367
368             preproc->reset(inname, 0, &nasmlist, depend_ptr);
369             if (outname[0] == '\0')
370                 ofmt->filename(inname, outname);
371             ofile = NULL;
372             while ((line = preproc->getline()))
373                 nasm_free(line);
374             preproc->cleanup(0);
375         }
376         break;
377
378     case op_preprocess:
379         {
380             char *line;
381             char *file_name = NULL;
382             int32_t prior_linnum = 0;
383             int lineinc = 0;
384
385             if (*outname) {
386                 ofile = fopen(outname, "w");
387                 if (!ofile)
388                     nasm_error(ERR_FATAL | ERR_NOFILE,
389                                  "unable to open output file `%s'",
390                                  outname);
391             } else
392                 ofile = NULL;
393
394             location.known = false;
395
396             /* pass = 1; */
397             preproc->reset(inname, 3, &nasmlist, depend_ptr);
398
399             while ((line = preproc->getline())) {
400                 /*
401                  * We generate %line directives if needed for later programs
402                  */
403                 int32_t linnum = prior_linnum += lineinc;
404                 int altline = src_get(&linnum, &file_name);
405                 if (altline) {
406                     if (altline == 1 && lineinc == 1)
407                         nasm_fputs("", ofile);
408                     else {
409                         lineinc = (altline != -1 || lineinc != 1);
410                         fprintf(ofile ? ofile : stdout,
411                                 "%%line %"PRId32"+%d %s\n", linnum, lineinc,
412                                 file_name);
413                     }
414                     prior_linnum = linnum;
415                 }
416                 nasm_fputs(line, ofile);
417                 nasm_free(line);
418             }
419             nasm_free(file_name);
420             preproc->cleanup(0);
421             if (ofile)
422                 fclose(ofile);
423             if (ofile && terminate_after_phase)
424                 remove(outname);
425             ofile = NULL;
426         }
427         break;
428
429     case op_normal:
430         {
431             /*
432              * We must call ofmt->filename _anyway_, even if the user
433              * has specified their own output file, because some
434              * formats (eg OBJ and COFF) use ofmt->filename to find out
435              * the name of the input file and then put that inside the
436              * file.
437              */
438             ofmt->filename(inname, outname);
439
440             ofile = fopen(outname, (ofmt->flags & OFMT_TEXT) ? "w" : "wb");
441             if (!ofile) {
442                 nasm_error(ERR_FATAL | ERR_NOFILE,
443                              "unable to open output file `%s'", outname);
444             }
445
446             /*
447              * We must call init_labels() before ofmt->init() since
448              * some object formats will want to define labels in their
449              * init routines. (eg OS/2 defines the FLAT group)
450              */
451             init_labels();
452
453             ofmt->init();
454             dfmt = ofmt->current_dfmt;
455             dfmt->init();
456
457             assemble_file(inname, depend_ptr);
458
459             if (!terminate_after_phase) {
460                 ofmt->cleanup(using_debug_info);
461                 cleanup_labels();
462                 fflush(ofile);
463                 if (ferror(ofile)) {
464                     nasm_error(ERR_NONFATAL|ERR_NOFILE,
465                                  "write error on output file `%s'", outname);
466                 }
467             }
468
469             if (ofile) {
470                 fclose(ofile);
471                 if (terminate_after_phase)
472                     remove(outname);
473                 ofile = NULL;
474             }
475         }
476         break;
477     }
478
479     if (depend_list && !terminate_after_phase)
480         emit_dependencies(depend_list);
481
482     if (want_usage)
483         usage();
484
485     raa_free(offsets);
486     saa_free(forwrefs);
487     eval_cleanup();
488     stdscan_cleanup();
489
490     return terminate_after_phase;
491 }
492
493 /*
494  * Get a parameter for a command line option.
495  * First arg must be in the form of e.g. -f...
496  */
497 static char *get_param(char *p, char *q, bool *advance)
498 {
499     *advance = false;
500     if (p[2]) {                 /* the parameter's in the option */
501         p += 2;
502         while (nasm_isspace(*p))
503             p++;
504         return p;
505     }
506     if (q && q[0]) {
507         *advance = true;
508         return q;
509     }
510     nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
511                  "option `-%c' requires an argument", p[1]);
512     return NULL;
513 }
514
515 /*
516  * Copy a filename
517  */
518 static void copy_filename(char *dst, const char *src)
519 {
520     size_t len = strlen(src);
521
522     if (len >= (size_t)FILENAME_MAX) {
523         nasm_error(ERR_FATAL | ERR_NOFILE, "file name too long");
524         return;
525     }
526     strncpy(dst, src, FILENAME_MAX);
527 }
528
529 /*
530  * Convert a string to Make-safe form
531  */
532 static char *quote_for_make(const char *str)
533 {
534     const char *p;
535     char *os, *q;
536
537     size_t n = 1;               /* Terminating zero */
538     size_t nbs = 0;
539
540     if (!str)
541         return NULL;
542
543     for (p = str; *p; p++) {
544         switch (*p) {
545         case ' ':
546         case '\t':
547             /* Convert N backslashes + ws -> 2N+1 backslashes + ws */
548             n += nbs + 2;
549             nbs = 0;
550             break;
551         case '$':
552         case '#':
553             nbs = 0;
554             n += 2;
555             break;
556         case '\\':
557             nbs++;
558             n++;
559             break;
560         default:
561             nbs = 0;
562             n++;
563         break;
564         }
565     }
566
567     /* Convert N backslashes at the end of filename to 2N backslashes */
568     if (nbs)
569         n += nbs;
570
571     os = q = nasm_malloc(n);
572
573     nbs = 0;
574     for (p = str; *p; p++) {
575         switch (*p) {
576         case ' ':
577         case '\t':
578             while (nbs--)
579                 *q++ = '\\';
580             *q++ = '\\';
581             *q++ = *p;
582             break;
583         case '$':
584             *q++ = *p;
585             *q++ = *p;
586             nbs = 0;
587             break;
588         case '#':
589             *q++ = '\\';
590             *q++ = *p;
591             nbs = 0;
592             break;
593         case '\\':
594             *q++ = *p;
595             nbs++;
596             break;
597         default:
598             *q++ = *p;
599             nbs = 0;
600         break;
601         }
602     }
603     while (nbs--)
604         *q++ = '\\';
605
606     *q = '\0';
607
608     return os;
609 }
610
611 struct textargs {
612     const char *label;
613     int value;
614 };
615
616 #define OPT_PREFIX 0
617 #define OPT_POSTFIX 1
618 struct textargs textopts[] = {
619     {"prefix", OPT_PREFIX},
620     {"postfix", OPT_POSTFIX},
621     {NULL, 0}
622 };
623
624 static bool stopoptions = false;
625 static bool process_arg(char *p, char *q)
626 {
627     char *param;
628     int i;
629     bool advance = false;
630     bool do_warn;
631
632     if (!p || !p[0])
633         return false;
634
635     if (p[0] == '-' && !stopoptions) {
636         if (strchr("oOfpPdDiIlFXuUZwW", p[1])) {
637             /* These parameters take values */
638             if (!(param = get_param(p, q, &advance)))
639                 return advance;
640         }
641
642         switch (p[1]) {
643         case 's':
644             error_file = stdout;
645             break;
646
647         case 'o':               /* output file */
648             copy_filename(outname, param);
649             break;
650
651         case 'f':               /* output format */
652             ofmt = ofmt_find(param);
653             if (!ofmt) {
654                 nasm_error(ERR_FATAL | ERR_NOFILE | ERR_USAGE,
655                              "unrecognised output format `%s' - "
656                              "use -hf for a list", param);
657             }
658             break;
659
660         case 'O':               /* Optimization level */
661         {
662             int opt;
663
664             if (!*param) {
665                 /* Naked -O == -Ox */
666                 optimizing = INT_MAX >> 1; /* Almost unlimited */
667             } else {
668                 while (*param) {
669                     switch (*param) {
670                     case '0': case '1': case '2': case '3': case '4':
671                     case '5': case '6': case '7': case '8': case '9':
672                         opt = strtoul(param, &param, 10);
673
674                         /* -O0 -> optimizing == -1, 0.98 behaviour */
675                         /* -O1 -> optimizing == 0, 0.98.09 behaviour */
676                         if (opt < 2)
677                             optimizing = opt - 1;
678                         else
679                             optimizing = opt;
680                         break;
681
682                     case 'v':
683                     case '+':
684                         param++;
685                         opt_verbose_info = true;
686                         break;
687
688                     case 'x':
689                         param++;
690                         optimizing = INT_MAX >> 1; /* Almost unlimited */
691                         break;
692
693                     default:
694                         nasm_error(ERR_FATAL,
695                                      "unknown optimization option -O%c\n",
696                                      *param);
697                         break;
698                     }
699                 }
700             }
701             break;
702         }
703
704         case 'p':                       /* pre-include */
705         case 'P':
706             pp_pre_include(param);
707             break;
708
709         case 'd':                       /* pre-define */
710         case 'D':
711             pp_pre_define(param);
712             break;
713
714         case 'u':                       /* un-define */
715         case 'U':
716             pp_pre_undefine(param);
717             break;
718
719         case 'i':                       /* include search path */
720         case 'I':
721             pp_include_path(param);
722             break;
723
724         case 'l':                       /* listing file */
725             copy_filename(listname, param);
726             break;
727
728         case 'Z':                       /* error messages file */
729             strcpy(errname, param);
730             break;
731
732         case 'F':                       /* specify debug format */
733             ofmt->current_dfmt = dfmt_find(ofmt, param);
734             if (!ofmt->current_dfmt) {
735                 nasm_error(ERR_FATAL | ERR_NOFILE | ERR_USAGE,
736                              "unrecognized debug format `%s' for"
737                              " output format `%s'",
738                              param, ofmt->shortname);
739             }
740             using_debug_info = true;
741             break;
742
743         case 'X':               /* specify error reporting format */
744             if (nasm_stricmp("vc", param) == 0)
745                 nasm_set_verror(nasm_verror_vc);
746             else if (nasm_stricmp("gnu", param) == 0)
747                 nasm_set_verror(nasm_verror_gnu);
748             else
749                 nasm_error(ERR_FATAL | ERR_NOFILE | ERR_USAGE,
750                              "unrecognized error reporting format `%s'",
751                              param);
752             break;
753
754         case 'g':
755             using_debug_info = true;
756             break;
757
758         case 'h':
759             printf
760                 ("usage: nasm [-@ response file] [-o outfile] [-f format] "
761                  "[-l listfile]\n"
762                  "            [options...] [--] filename\n"
763                  "    or nasm -v   for version info\n\n"
764                  "    -t          assemble in SciTech TASM compatible mode\n"
765                  "    -g          generate debug information in selected format\n");
766             printf
767                 ("    -E (or -e)  preprocess only (writes output to stdout by default)\n"
768                  "    -a          don't preprocess (assemble only)\n"
769                  "    -M          generate Makefile dependencies on stdout\n"
770                  "    -MG         d:o, missing files assumed generated\n"
771                  "    -MF <file>  set Makefile dependency file\n"
772                  "    -MD <file>  assemble and generate dependencies\n"
773                  "    -MT <file>  dependency target name\n"
774                  "    -MQ <file>  dependency target name (quoted)\n"
775                  "    -MP         emit phony target\n\n"
776                  "    -Z<file>    redirect error messages to file\n"
777                  "    -s          redirect error messages to stdout\n\n"
778                  "    -F format   select a debugging format\n\n"
779                  "    -I<path>    adds a pathname to the include file path\n");
780             printf
781                 ("    -O<digit>   optimize branch offsets\n"
782                  "                -O0: No optimization (default)\n"
783                  "                -O1: Minimal optimization\n"
784                  "                -Ox: Multipass optimization (recommended)\n\n"
785                  "    -P<file>    pre-includes a file\n"
786                  "    -D<macro>[=<value>] pre-defines a macro\n"
787                  "    -U<macro>   undefines a macro\n"
788                  "    -X<format>  specifies error reporting format (gnu or vc)\n"
789                  "    -w+foo      enables warning foo (equiv. -Wfoo)\n"
790                  "    -w-foo      disable warning foo (equiv. -Wno-foo)\n\n"
791                  "--prefix,--postfix\n"
792                  "  this options prepend or append the given argument to all\n"
793                  "  extern and global variables\n\n"
794                  "Warnings:\n");
795             for (i = 0; i <= ERR_WARN_MAX; i++)
796                 printf("    %-23s %s (default %s)\n",
797                        warnings[i].name, warnings[i].help,
798                        warnings[i].enabled ? "on" : "off");
799             printf
800                 ("\nresponse files should contain command line parameters"
801                  ", one per line.\n");
802             if (p[2] == 'f') {
803                 printf("\nvalid output formats for -f are"
804                        " (`*' denotes default):\n");
805                 ofmt_list(ofmt, stdout);
806             } else {
807                 printf("\nFor a list of valid output formats, use -hf.\n");
808                 printf("For a list of debug formats, use -f <form> -y.\n");
809             }
810             exit(0);            /* never need usage message here */
811             break;
812
813         case 'y':
814             printf("\nvalid debug formats for '%s' output format are"
815                    " ('*' denotes default):\n", ofmt->shortname);
816             dfmt_list(ofmt, stdout);
817             exit(0);
818             break;
819
820         case 't':
821             tasm_compatible_mode = true;
822             break;
823
824         case 'v':
825             printf("NASM version %s compiled on %s%s\n",
826                    nasm_version, nasm_date, nasm_compile_options);
827             exit(0);        /* never need usage message here */
828             break;
829
830         case 'e':              /* preprocess only */
831         case 'E':
832             operating_mode = op_preprocess;
833             break;
834
835         case 'a':              /* assemble only - don't preprocess */
836             preproc = &no_pp;
837             break;
838
839         case 'W':
840             if (param[0] == 'n' && param[1] == 'o' && param[2] == '-') {
841                 do_warn = false;
842                 param += 3;
843             } else {
844                 do_warn = true;
845             }
846             goto set_warning;
847
848         case 'w':
849             if (param[0] != '+' && param[0] != '-') {
850                 nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
851                              "invalid option to `-w'");
852                 break;
853             }
854             do_warn = (param[0] == '+');
855             param++;
856             goto set_warning;
857         set_warning:
858             for (i = 0; i <= ERR_WARN_MAX; i++)
859                 if (!nasm_stricmp(param, warnings[i].name))
860                     break;
861             if (i <= ERR_WARN_MAX)
862                 warning_on_global[i] = do_warn;
863             else if (!nasm_stricmp(param, "all"))
864                 for (i = 1; i <= ERR_WARN_MAX; i++)
865                     warning_on_global[i] = do_warn;
866             else if (!nasm_stricmp(param, "none"))
867                 for (i = 1; i <= ERR_WARN_MAX; i++)
868                     warning_on_global[i] = !do_warn;
869             else
870                 nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
871                              "invalid warning `%s'", param);
872             break;
873
874         case 'M':
875             switch (p[2]) {
876             case 0:
877                 operating_mode = op_depend;
878                 break;
879             case 'G':
880                 operating_mode = op_depend;
881                 depend_missing_ok = true;
882                 break;
883             case 'P':
884                 depend_emit_phony = true;
885                 break;
886             case 'D':
887                 depend_file = q;
888                 advance = true;
889                 break;
890             case 'T':
891                 depend_target = q;
892                 advance = true;
893                 break;
894             case 'Q':
895                 depend_target = quote_for_make(q);
896                 advance = true;
897                 break;
898             default:
899                 nasm_error(ERR_NONFATAL|ERR_NOFILE|ERR_USAGE,
900                              "unknown dependency option `-M%c'", p[2]);
901                 break;
902             }
903             if (advance && (!q || !q[0])) {
904                 nasm_error(ERR_NONFATAL|ERR_NOFILE|ERR_USAGE,
905                              "option `-M%c' requires a parameter", p[2]);
906                 break;
907             }
908             break;
909
910         case '-':
911             {
912                 int s;
913
914                 if (p[2] == 0) {        /* -- => stop processing options */
915                     stopoptions = 1;
916                     break;
917                 }
918                 for (s = 0; textopts[s].label; s++) {
919                     if (!nasm_stricmp(p + 2, textopts[s].label)) {
920                         break;
921                     }
922                 }
923
924                 switch (s) {
925
926                 case OPT_PREFIX:
927                 case OPT_POSTFIX:
928                     {
929                         if (!q) {
930                             nasm_error(ERR_NONFATAL | ERR_NOFILE |
931                                          ERR_USAGE,
932                                          "option `--%s' requires an argument",
933                                          p + 2);
934                             break;
935                         } else {
936                             advance = 1, param = q;
937                         }
938
939                         if (s == OPT_PREFIX) {
940                             strncpy(lprefix, param, PREFIX_MAX - 1);
941                             lprefix[PREFIX_MAX - 1] = 0;
942                             break;
943                         }
944                         if (s == OPT_POSTFIX) {
945                             strncpy(lpostfix, param, POSTFIX_MAX - 1);
946                             lpostfix[POSTFIX_MAX - 1] = 0;
947                             break;
948                         }
949                         break;
950                     }
951                 default:
952                     {
953                         nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
954                                      "unrecognised option `--%s'", p + 2);
955                         break;
956                     }
957                 }
958                 break;
959             }
960
961         default:
962             if (!ofmt->setinfo(GI_SWITCH, &p))
963                 nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
964                              "unrecognised option `-%c'", p[1]);
965             break;
966         }
967     } else {
968         if (*inname) {
969             nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
970                          "more than one input file specified");
971         } else {
972             copy_filename(inname, p);
973         }
974     }
975
976     return advance;
977 }
978
979 #define ARG_BUF_DELTA 128
980
981 static void process_respfile(FILE * rfile)
982 {
983     char *buffer, *p, *q, *prevarg;
984     int bufsize, prevargsize;
985
986     bufsize = prevargsize = ARG_BUF_DELTA;
987     buffer = nasm_malloc(ARG_BUF_DELTA);
988     prevarg = nasm_malloc(ARG_BUF_DELTA);
989     prevarg[0] = '\0';
990
991     while (1) {                 /* Loop to handle all lines in file */
992         p = buffer;
993         while (1) {             /* Loop to handle long lines */
994             q = fgets(p, bufsize - (p - buffer), rfile);
995             if (!q)
996                 break;
997             p += strlen(p);
998             if (p > buffer && p[-1] == '\n')
999                 break;
1000             if (p - buffer > bufsize - 10) {
1001                 int offset;
1002                 offset = p - buffer;
1003                 bufsize += ARG_BUF_DELTA;
1004                 buffer = nasm_realloc(buffer, bufsize);
1005                 p = buffer + offset;
1006             }
1007         }
1008
1009         if (!q && p == buffer) {
1010             if (prevarg[0])
1011                 process_arg(prevarg, NULL);
1012             nasm_free(buffer);
1013             nasm_free(prevarg);
1014             return;
1015         }
1016
1017         /*
1018          * Play safe: remove CRs, LFs and any spurious ^Zs, if any of
1019          * them are present at the end of the line.
1020          */
1021         *(p = &buffer[strcspn(buffer, "\r\n\032")]) = '\0';
1022
1023         while (p > buffer && nasm_isspace(p[-1]))
1024             *--p = '\0';
1025
1026         p = buffer;
1027         while (nasm_isspace(*p))
1028             p++;
1029
1030         if (process_arg(prevarg, p))
1031             *p = '\0';
1032
1033         if ((int) strlen(p) > prevargsize - 10) {
1034             prevargsize += ARG_BUF_DELTA;
1035             prevarg = nasm_realloc(prevarg, prevargsize);
1036         }
1037         strncpy(prevarg, p, prevargsize);
1038     }
1039 }
1040
1041 /* Function to process args from a string of args, rather than the
1042  * argv array. Used by the environment variable and response file
1043  * processing.
1044  */
1045 static void process_args(char *args)
1046 {
1047     char *p, *q, *arg, *prevarg;
1048     char separator = ' ';
1049
1050     p = args;
1051     if (*p && *p != '-')
1052         separator = *p++;
1053     arg = NULL;
1054     while (*p) {
1055         q = p;
1056         while (*p && *p != separator)
1057             p++;
1058         while (*p == separator)
1059             *p++ = '\0';
1060         prevarg = arg;
1061         arg = q;
1062         if (process_arg(prevarg, arg))
1063             arg = NULL;
1064     }
1065     if (arg)
1066         process_arg(arg, NULL);
1067 }
1068
1069 static void process_response_file(const char *file)
1070 {
1071     char str[2048];
1072     FILE *f = fopen(file, "r");
1073     if (!f) {
1074         perror(file);
1075         exit(-1);
1076     }
1077     while (fgets(str, sizeof str, f)) {
1078         process_args(str);
1079     }
1080     fclose(f);
1081 }
1082
1083 static void parse_cmdline(int argc, char **argv)
1084 {
1085     FILE *rfile;
1086     char *envreal, *envcopy = NULL, *p, *arg;
1087     int i;
1088
1089     *inname = *outname = *listname = *errname = '\0';
1090     for (i = 0; i <= ERR_WARN_MAX; i++)
1091         warning_on_global[i] = warnings[i].enabled;
1092
1093     /*
1094      * First, process the NASMENV environment variable.
1095      */
1096     envreal = getenv("NASMENV");
1097     arg = NULL;
1098     if (envreal) {
1099         envcopy = nasm_strdup(envreal);
1100         process_args(envcopy);
1101         nasm_free(envcopy);
1102     }
1103
1104     /*
1105      * Now process the actual command line.
1106      */
1107     while (--argc) {
1108         bool advance;
1109         argv++;
1110         if (argv[0][0] == '@') {
1111             /* We have a response file, so process this as a set of
1112              * arguments like the environment variable. This allows us
1113              * to have multiple arguments on a single line, which is
1114              * different to the -@resp file processing below for regular
1115              * NASM.
1116              */
1117             process_response_file(argv[0]+1);
1118             argc--;
1119             argv++;
1120         }
1121         if (!stopoptions && argv[0][0] == '-' && argv[0][1] == '@') {
1122             p = get_param(argv[0], argc > 1 ? argv[1] : NULL, &advance);
1123             if (p) {
1124                 rfile = fopen(p, "r");
1125                 if (rfile) {
1126                     process_respfile(rfile);
1127                     fclose(rfile);
1128                 } else
1129                     nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
1130                                  "unable to open response file `%s'", p);
1131             }
1132         } else
1133             advance = process_arg(argv[0], argc > 1 ? argv[1] : NULL);
1134         argv += advance, argc -= advance;
1135     }
1136
1137     /* Look for basic command line typos.  This definitely doesn't
1138        catch all errors, but it might help cases of fumbled fingers. */
1139     if (!*inname)
1140         nasm_error(ERR_NONFATAL | ERR_NOFILE | ERR_USAGE,
1141                      "no input file specified");
1142     else if (!strcmp(inname, errname) ||
1143              !strcmp(inname, outname) ||
1144              !strcmp(inname, listname) ||
1145              (depend_file && !strcmp(inname, depend_file)))
1146         nasm_error(ERR_FATAL | ERR_NOFILE | ERR_USAGE,
1147                      "file `%s' is both input and output file",
1148                      inname);
1149
1150     if (*errname) {
1151         error_file = fopen(errname, "w");
1152         if (!error_file) {
1153             error_file = stderr;        /* Revert to default! */
1154             nasm_error(ERR_FATAL | ERR_NOFILE | ERR_USAGE,
1155                          "cannot open file `%s' for error messages",
1156                          errname);
1157         }
1158     }
1159 }
1160
1161 static enum directives getkw(char **directive, char **value);
1162
1163 static void assemble_file(char *fname, StrList **depend_ptr)
1164 {
1165     char *directive, *value, *p, *q, *special, *line;
1166     insn output_ins;
1167     int i, validid;
1168     bool rn_error;
1169     int32_t seg;
1170     int64_t offs;
1171     struct tokenval tokval;
1172     expr *e;
1173     int pass_max;
1174
1175     if (cmd_sb == 32 && cmd_cpu < IF_386)
1176         nasm_error(ERR_FATAL, "command line: "
1177                      "32-bit segment size requires a higher cpu");
1178
1179     pass_max = prev_offset_changed = (INT_MAX >> 1) + 2; /* Almost unlimited */
1180     for (passn = 1; pass0 <= 2; passn++) {
1181         int pass1, pass2;
1182         ldfunc def_label;
1183
1184         pass1 = pass0 == 2 ? 2 : 1;     /* 1, 1, 1, ..., 1, 2 */
1185         pass2 = passn > 1  ? 2 : 1;     /* 1, 2, 2, ..., 2, 2 */
1186         /* pass0                           0, 0, 0, ..., 1, 2 */
1187
1188         def_label = passn > 1 ? redefine_label : define_label;
1189
1190         globalbits = sb = cmd_sb;   /* set 'bits' to command line default */
1191         cpu = cmd_cpu;
1192         if (pass0 == 2) {
1193             if (*listname)
1194                 nasmlist.init(listname, nasm_error);
1195         }
1196         in_abs_seg = false;
1197         global_offset_changed = 0;  /* set by redefine_label */
1198         location.segment = ofmt->section(NULL, pass2, &sb);
1199         globalbits = sb;
1200         if (passn > 1) {
1201             saa_rewind(forwrefs);
1202             forwref = saa_rstruct(forwrefs);
1203             raa_free(offsets);
1204             offsets = raa_init();
1205         }
1206         preproc->reset(fname, pass1, &nasmlist,
1207                        pass1 == 2 ? depend_ptr : NULL);
1208         memcpy(warning_on, warning_on_global, (ERR_WARN_MAX+1) * sizeof(bool));
1209
1210         globallineno = 0;
1211         if (passn == 1)
1212             location.known = true;
1213         location.offset = offs = GET_CURR_OFFS;
1214
1215         while ((line = preproc->getline())) {
1216             enum directives d;
1217             globallineno++;
1218
1219             /*
1220              * Here we parse our directives; this is not handled by the
1221              * 'real' parser.  This really should be a separate function.
1222              */
1223             directive = line;
1224             d = getkw(&directive, &value);
1225             if (d) {
1226                 int err = 0;
1227
1228                 switch (d) {
1229                 case D_SEGMENT:         /* [SEGMENT n] */
1230                 case D_SECTION:
1231                     seg = ofmt->section(value, pass2, &sb);
1232                     if (seg == NO_SEG) {
1233                         nasm_error(pass1 == 1 ? ERR_NONFATAL : ERR_PANIC,
1234                                      "segment name `%s' not recognized",
1235                                      value);
1236                     } else {
1237                         in_abs_seg = false;
1238                         location.segment = seg;
1239                     }
1240                     break;
1241                 case D_EXTERN:          /* [EXTERN label:special] */
1242                     if (*value == '$')
1243                         value++;        /* skip initial $ if present */
1244                     if (pass0 == 2) {
1245                         q = value;
1246                         while (*q && *q != ':')
1247                             q++;
1248                         if (*q == ':') {
1249                             *q++ = '\0';
1250                             ofmt->symdef(value, 0L, 0L, 3, q);
1251                         }
1252                     } else if (passn == 1) {
1253                         q = value;
1254                         validid = true;
1255                         if (!isidstart(*q))
1256                             validid = false;
1257                         while (*q && *q != ':') {
1258                             if (!isidchar(*q))
1259                                 validid = false;
1260                             q++;
1261                         }
1262                         if (!validid) {
1263                             nasm_error(ERR_NONFATAL,
1264                                          "identifier expected after EXTERN");
1265                             break;
1266                         }
1267                         if (*q == ':') {
1268                             *q++ = '\0';
1269                             special = q;
1270                         } else
1271                             special = NULL;
1272                         if (!is_extern(value)) {        /* allow re-EXTERN to be ignored */
1273                             int temp = pass0;
1274                             pass0 = 1;  /* fake pass 1 in labels.c */
1275                             declare_as_global(value, special);
1276                             define_label(value, seg_alloc(), 0L, NULL,
1277                                          false, true);
1278                             pass0 = temp;
1279                         }
1280                     }           /* else  pass0 == 1 */
1281                     break;
1282                 case D_BITS:            /* [BITS bits] */
1283                     globalbits = sb = get_bits(value);
1284                     break;
1285                 case D_GLOBAL:          /* [GLOBAL symbol:special] */
1286                     if (*value == '$')
1287                         value++;        /* skip initial $ if present */
1288                     if (pass0 == 2) {   /* pass 2 */
1289                         q = value;
1290                         while (*q && *q != ':')
1291                             q++;
1292                         if (*q == ':') {
1293                             *q++ = '\0';
1294                             ofmt->symdef(value, 0L, 0L, 3, q);
1295                         }
1296                     } else if (pass2 == 1) {    /* pass == 1 */
1297                         q = value;
1298                         validid = true;
1299                         if (!isidstart(*q))
1300                             validid = false;
1301                         while (*q && *q != ':') {
1302                             if (!isidchar(*q))
1303                                 validid = false;
1304                             q++;
1305                         }
1306                         if (!validid) {
1307                             nasm_error(ERR_NONFATAL,
1308                                          "identifier expected after GLOBAL");
1309                             break;
1310                         }
1311                         if (*q == ':') {
1312                             *q++ = '\0';
1313                             special = q;
1314                         } else
1315                             special = NULL;
1316                         declare_as_global(value, special);
1317                     }           /* pass == 1 */
1318                     break;
1319                 case D_COMMON:          /* [COMMON symbol size:special] */
1320                 {
1321                     int64_t size;
1322
1323                     if (*value == '$')
1324                         value++;        /* skip initial $ if present */
1325                     p = value;
1326                     validid = true;
1327                     if (!isidstart(*p))
1328                         validid = false;
1329                     while (*p && !nasm_isspace(*p)) {
1330                         if (!isidchar(*p))
1331                             validid = false;
1332                         p++;
1333                     }
1334                     if (!validid) {
1335                         nasm_error(ERR_NONFATAL,
1336                                      "identifier expected after COMMON");
1337                         break;
1338                     }
1339                     if (*p) {
1340                         while (*p && nasm_isspace(*p))
1341                             *p++ = '\0';
1342                         q = p;
1343                         while (*q && *q != ':')
1344                             q++;
1345                         if (*q == ':') {
1346                             *q++ = '\0';
1347                             special = q;
1348                         } else {
1349                             special = NULL;
1350                         }
1351                         size = readnum(p, &rn_error);
1352                         if (rn_error) {
1353                             nasm_error(ERR_NONFATAL,
1354                                          "invalid size specified"
1355                                          " in COMMON declaration");
1356                             break;
1357                         }
1358                     } else {
1359                         nasm_error(ERR_NONFATAL,
1360                                      "no size specified in"
1361                                      " COMMON declaration");
1362                         break;
1363                     }
1364
1365                     if (pass0 < 2) {
1366                         define_common(value, seg_alloc(), size, special);
1367                     } else if (pass0 == 2) {
1368                         if (special)
1369                             ofmt->symdef(value, 0L, 0L, 3, special);
1370                     }
1371                     break;
1372                 }
1373                 case D_ABSOLUTE:                /* [ABSOLUTE address] */
1374                     stdscan_reset();
1375                     stdscan_bufptr = value;
1376                     tokval.t_type = TOKEN_INVALID;
1377                     e = evaluate(stdscan, NULL, &tokval, NULL, pass2,
1378                                  nasm_error, NULL);
1379                     if (e) {
1380                         if (!is_reloc(e))
1381                             nasm_error(pass0 ==
1382                                          1 ? ERR_NONFATAL : ERR_PANIC,
1383                                          "cannot use non-relocatable expression as "
1384                                          "ABSOLUTE address");
1385                         else {
1386                             abs_seg = reloc_seg(e);
1387                             abs_offset = reloc_value(e);
1388                         }
1389                     } else if (passn == 1)
1390                         abs_offset = 0x100;     /* don't go near zero in case of / */
1391                     else
1392                         nasm_error(ERR_PANIC, "invalid ABSOLUTE address "
1393                                      "in pass two");
1394                     in_abs_seg = true;
1395                     location.segment = NO_SEG;
1396                     break;
1397                 case D_DEBUG:           /* [DEBUG] */
1398                 {
1399                     char debugid[128];
1400                     bool badid, overlong;
1401
1402                     p = value;
1403                     q = debugid;
1404                     badid = overlong = false;
1405                     if (!isidstart(*p)) {
1406                         badid = true;
1407                     } else {
1408                         while (*p && !nasm_isspace(*p)) {
1409                             if (q >= debugid + sizeof debugid - 1) {
1410                                 overlong = true;
1411                                 break;
1412                             }
1413                             if (!isidchar(*p))
1414                                 badid = true;
1415                             *q++ = *p++;
1416                         }
1417                         *q = 0;
1418                     }
1419                     if (badid) {
1420                         nasm_error(passn == 1 ? ERR_NONFATAL : ERR_PANIC,
1421                                    "identifier expected after DEBUG");
1422                         break;
1423                     }
1424                     if (overlong) {
1425                         nasm_error(passn == 1 ? ERR_NONFATAL : ERR_PANIC,
1426                                    "DEBUG identifier too long");
1427                         break;
1428                     }
1429                     while (*p && nasm_isspace(*p))
1430                         p++;
1431                     if (pass0 == 2)
1432                         dfmt->debug_directive(debugid, p);
1433                     break;
1434                 }
1435                 case D_WARNING:         /* [WARNING {+|-|*}warn-name] */
1436                     while (*value && nasm_isspace(*value))
1437                         value++;
1438
1439                     switch(*value) {
1440                     case '-': validid = 0; value++; break;
1441                     case '+': validid = 1; value++; break;
1442                     case '*': validid = 2; value++; break;
1443                     default:  validid = 1; break;
1444                     }
1445
1446                     for (i = 1; i <= ERR_WARN_MAX; i++)
1447                         if (!nasm_stricmp(value, warnings[i].name))
1448                             break;
1449                     if (i <= ERR_WARN_MAX) {
1450                         switch(validid) {
1451                         case 0:
1452                             warning_on[i] = false;
1453                             break;
1454                         case 1:
1455                             warning_on[i] = true;
1456                             break;
1457                         case 2:
1458                             warning_on[i] = warning_on_global[i];
1459                             break;
1460                         }
1461                     }
1462                     else
1463                         nasm_error(ERR_NONFATAL,
1464                                      "invalid warning id in WARNING directive");
1465                     break;
1466                 case D_CPU:             /* [CPU] */
1467                     cpu = get_cpu(value);
1468                     break;
1469                 case D_LIST:            /* [LIST {+|-}] */
1470                     while (*value && nasm_isspace(*value))
1471                         value++;
1472
1473                     if (*value == '+') {
1474                         user_nolist = 0;
1475                     } else {
1476                         if (*value == '-') {
1477                             user_nolist = 1;
1478                         } else {
1479                             err = 1;
1480                         }
1481                     }
1482                     break;
1483                 case D_DEFAULT:         /* [DEFAULT] */
1484                     stdscan_reset();
1485                     stdscan_bufptr = value;
1486                     tokval.t_type = TOKEN_INVALID;
1487                     if (stdscan(NULL, &tokval) == TOKEN_SPECIAL) {
1488                         switch ((int)tokval.t_integer) {
1489                         case S_REL:
1490                             globalrel = 1;
1491                             break;
1492                         case S_ABS:
1493                             globalrel = 0;
1494                             break;
1495                         default:
1496                             err = 1;
1497                             break;
1498                         }
1499                     } else {
1500                         err = 1;
1501                     }
1502                     break;
1503                 case D_FLOAT:
1504                     if (float_option(value)) {
1505                         nasm_error(pass1 == 1 ? ERR_NONFATAL : ERR_PANIC,
1506                                      "unknown 'float' directive: %s",
1507                                      value);
1508                     }
1509                     break;
1510                 default:
1511                     if (!d || !ofmt->directive(d, value, pass2))
1512                         nasm_error(pass1 == 1 ? ERR_NONFATAL : ERR_PANIC,
1513                                      "unrecognised directive [%s]",
1514                                      directive);
1515                     break;
1516                 }
1517                 if (err) {
1518                     nasm_error(ERR_NONFATAL,
1519                                  "invalid parameter to [%s] directive",
1520                                  directive);
1521                 }
1522             } else {            /* it isn't a directive */
1523                 parse_line(pass1, line, &output_ins, def_label);
1524
1525                 if (optimizing > 0) {
1526                     if (forwref != NULL && globallineno == forwref->lineno) {
1527                         output_ins.forw_ref = true;
1528                         do {
1529                             output_ins.oprs[forwref->operand].opflags |=
1530                                 OPFLAG_FORWARD;
1531                             forwref = saa_rstruct(forwrefs);
1532                         } while (forwref != NULL
1533                                  && forwref->lineno == globallineno);
1534                     } else
1535                         output_ins.forw_ref = false;
1536
1537                     if (output_ins.forw_ref) {
1538                         if (passn == 1) {
1539                             for (i = 0; i < output_ins.operands; i++) {
1540                                 if (output_ins.oprs[i].
1541                                     opflags & OPFLAG_FORWARD) {
1542                                     struct forwrefinfo *fwinf =
1543                                         (struct forwrefinfo *)
1544                                         saa_wstruct(forwrefs);
1545                                     fwinf->lineno = globallineno;
1546                                 fwinf->operand = i;
1547                                 }
1548                             }
1549                         }
1550                     }
1551                 }
1552
1553                 /*  forw_ref */
1554                 if (output_ins.opcode == I_EQU) {
1555                     if (pass1 == 1) {
1556                         /*
1557                          * Special `..' EQUs get processed in pass two,
1558                          * except `..@' macro-processor EQUs which are done
1559                          * in the normal place.
1560                          */
1561                         if (!output_ins.label)
1562                             nasm_error(ERR_NONFATAL,
1563                                          "EQU not preceded by label");
1564
1565                         else if (output_ins.label[0] != '.' ||
1566                                  output_ins.label[1] != '.' ||
1567                                  output_ins.label[2] == '@') {
1568                             if (output_ins.operands == 1 &&
1569                                 (output_ins.oprs[0].type & IMMEDIATE) &&
1570                                 output_ins.oprs[0].wrt == NO_SEG) {
1571                                 bool isext = !!(output_ins.oprs[0].opflags
1572                                                 & OPFLAG_EXTERN);
1573                                 def_label(output_ins.label,
1574                                           output_ins.oprs[0].segment,
1575                                           output_ins.oprs[0].offset, NULL,
1576                                           false, isext);
1577                             } else if (output_ins.operands == 2
1578                                        && (output_ins.oprs[0].type & IMMEDIATE)
1579                                        && (output_ins.oprs[0].type & COLON)
1580                                        && output_ins.oprs[0].segment == NO_SEG
1581                                        && output_ins.oprs[0].wrt == NO_SEG
1582                                        && (output_ins.oprs[1].type & IMMEDIATE)
1583                                        && output_ins.oprs[1].segment == NO_SEG
1584                                        && output_ins.oprs[1].wrt == NO_SEG) {
1585                                 def_label(output_ins.label,
1586                                           output_ins.oprs[0].offset | SEG_ABS,
1587                                           output_ins.oprs[1].offset,
1588                                           NULL, false, false);
1589                             } else
1590                                 nasm_error(ERR_NONFATAL,
1591                                              "bad syntax for EQU");
1592                         }
1593                     } else {
1594                         /*
1595                          * Special `..' EQUs get processed here, except
1596                          * `..@' macro processor EQUs which are done above.
1597                          */
1598                         if (output_ins.label[0] == '.' &&
1599                             output_ins.label[1] == '.' &&
1600                             output_ins.label[2] != '@') {
1601                             if (output_ins.operands == 1 &&
1602                                 (output_ins.oprs[0].type & IMMEDIATE)) {
1603                                 define_label(output_ins.label,
1604                                              output_ins.oprs[0].segment,
1605                                              output_ins.oprs[0].offset,
1606                                              NULL, false, false);
1607                             } else if (output_ins.operands == 2
1608                                        && (output_ins.oprs[0].
1609                                            type & IMMEDIATE)
1610                                        && (output_ins.oprs[0].type & COLON)
1611                                        && output_ins.oprs[0].segment ==
1612                                        NO_SEG
1613                                        && (output_ins.oprs[1].
1614                                            type & IMMEDIATE)
1615                                        && output_ins.oprs[1].segment ==
1616                                        NO_SEG) {
1617                                 define_label(output_ins.label,
1618                                              output_ins.oprs[0].
1619                                              offset | SEG_ABS,
1620                                              output_ins.oprs[1].offset,
1621                                              NULL, false, false);
1622                             } else
1623                                 nasm_error(ERR_NONFATAL,
1624                                              "bad syntax for EQU");
1625                         }
1626                     }
1627                 } else {        /* instruction isn't an EQU */
1628
1629                     if (pass1 == 1) {
1630
1631                         int64_t l = insn_size(location.segment, offs, sb, cpu,
1632                                            &output_ins, nasm_error);
1633
1634                         /* if (using_debug_info)  && output_ins.opcode != -1) */
1635                         if (using_debug_info)
1636                         {       /* fbk 03/25/01 */
1637                             /* this is done here so we can do debug type info */
1638                             int32_t typeinfo =
1639                                 TYS_ELEMENTS(output_ins.operands);
1640                             switch (output_ins.opcode) {
1641                             case I_RESB:
1642                                 typeinfo =
1643                                     TYS_ELEMENTS(output_ins.oprs[0].
1644                                                  offset) | TY_BYTE;
1645                                 break;
1646                             case I_RESW:
1647                                 typeinfo =
1648                                     TYS_ELEMENTS(output_ins.oprs[0].
1649                                                  offset) | TY_WORD;
1650                                 break;
1651                             case I_RESD:
1652                                 typeinfo =
1653                                     TYS_ELEMENTS(output_ins.oprs[0].
1654                                                  offset) | TY_DWORD;
1655                                 break;
1656                             case I_RESQ:
1657                                 typeinfo =
1658                                     TYS_ELEMENTS(output_ins.oprs[0].
1659                                                  offset) | TY_QWORD;
1660                                 break;
1661                             case I_REST:
1662                                 typeinfo =
1663                                     TYS_ELEMENTS(output_ins.oprs[0].
1664                                                  offset) | TY_TBYTE;
1665                                 break;
1666                             case I_RESO:
1667                                 typeinfo =
1668                                     TYS_ELEMENTS(output_ins.oprs[0].
1669                                                  offset) | TY_OWORD;
1670                                 break;
1671                             case I_RESY:
1672                                 typeinfo =
1673                                     TYS_ELEMENTS(output_ins.oprs[0].
1674                                                  offset) | TY_YWORD;
1675                                 break;
1676                             case I_DB:
1677                                 typeinfo |= TY_BYTE;
1678                                 break;
1679                             case I_DW:
1680                                 typeinfo |= TY_WORD;
1681                                 break;
1682                             case I_DD:
1683                                 if (output_ins.eops_float)
1684                                     typeinfo |= TY_FLOAT;
1685                                 else
1686                                     typeinfo |= TY_DWORD;
1687                                 break;
1688                             case I_DQ:
1689                                 typeinfo |= TY_QWORD;
1690                                 break;
1691                             case I_DT:
1692                                 typeinfo |= TY_TBYTE;
1693                                 break;
1694                             case I_DO:
1695                                 typeinfo |= TY_OWORD;
1696                                 break;
1697                             case I_DY:
1698                                 typeinfo |= TY_YWORD;
1699                                 break;
1700                             default:
1701                                 typeinfo = TY_LABEL;
1702
1703                             }
1704
1705                             dfmt->debug_typevalue(typeinfo);
1706                         }
1707                         if (l != -1) {
1708                             offs += l;
1709                             SET_CURR_OFFS(offs);
1710                         }
1711                         /*
1712                          * else l == -1 => invalid instruction, which will be
1713                          * flagged as an error on pass 2
1714                          */
1715
1716                     } else {
1717                         offs += assemble(location.segment, offs, sb, cpu,
1718                                          &output_ins, ofmt, nasm_error,
1719                                          &nasmlist);
1720                         SET_CURR_OFFS(offs);
1721
1722                     }
1723                 }               /* not an EQU */
1724                 cleanup_insn(&output_ins);
1725             }
1726             nasm_free(line);
1727             location.offset = offs = GET_CURR_OFFS;
1728         }                       /* end while (line = preproc->getline... */
1729
1730         if (pass0 == 2 && global_offset_changed && !terminate_after_phase)
1731             nasm_error(ERR_NONFATAL,
1732                          "phase error detected at end of assembly.");
1733
1734         if (pass1 == 1)
1735             preproc->cleanup(1);
1736
1737         if ((passn > 1 && !global_offset_changed) || pass0 == 2) {
1738             pass0++;
1739         } else if (global_offset_changed &&
1740                  global_offset_changed < prev_offset_changed) {
1741             prev_offset_changed = global_offset_changed;
1742             stall_count = 0;
1743         } else {
1744             stall_count++;
1745         }
1746
1747         if (terminate_after_phase)
1748             break;
1749
1750         if ((stall_count > 997) || (passn >= pass_max)) {
1751             /* We get here if the labels don't converge
1752              * Example: FOO equ FOO + 1
1753              */
1754              nasm_error(ERR_NONFATAL,
1755                           "Can't find valid values for all labels "
1756                           "after %d passes, giving up.", passn);
1757              nasm_error(ERR_NONFATAL,
1758                           "Possible causes: recursive EQUs, macro abuse.");
1759              break;
1760         }
1761     }
1762
1763     preproc->cleanup(0);
1764     nasmlist.cleanup();
1765     if (!terminate_after_phase && opt_verbose_info) {
1766         /*  -On and -Ov switches */
1767         fprintf(stdout, "info: assembly required 1+%d+1 passes\n", passn-3);
1768     }
1769 }
1770
1771 static enum directives getkw(char **directive, char **value)
1772 {
1773     char *p, *q, *buf;
1774
1775     buf = nasm_skip_spaces(*directive);
1776
1777     /* it should be enclosed in [ ] */
1778     if (*buf != '[')
1779         return D_NONE;
1780     q = strchr(buf, ']');
1781     if (!q)
1782         return D_NONE;
1783
1784     /* stip off the comments */
1785     p = strchr(buf, ';');
1786     if (p) {
1787         if (p < q) /* ouch! somwhere inside */
1788             return D_NONE;
1789         *p = '\0';
1790     }
1791
1792     /* no brace, no trailing spaces */
1793     *q = '\0';
1794     nasm_zap_spaces_rev(--q);
1795
1796     /* directive */
1797     p = nasm_skip_spaces(++buf);
1798     q = nasm_skip_word(p);
1799     if (!q)
1800         return D_NONE; /* sigh... no value there */
1801     *q = '\0';
1802     *directive = p;
1803
1804     /* and value finally */
1805     p = nasm_skip_spaces(++q);
1806     *value = p;
1807
1808     return find_directive(*directive);
1809 }
1810
1811 /**
1812  * gnu style error reporting
1813  * This function prints an error message to error_file in the
1814  * style used by GNU. An example would be:
1815  * file.asm:50: error: blah blah blah
1816  * where file.asm is the name of the file, 50 is the line number on
1817  * which the error occurs (or is detected) and "error:" is one of
1818  * the possible optional diagnostics -- it can be "error" or "warning"
1819  * or something else.  Finally the line terminates with the actual
1820  * error message.
1821  *
1822  * @param severity the severity of the warning or error
1823  * @param fmt the printf style format string
1824  */
1825 static void nasm_verror_gnu(int severity, const char *fmt, va_list ap)
1826 {
1827     char *currentfile = NULL;
1828     int32_t lineno = 0;
1829
1830     if (is_suppressed_warning(severity))
1831         return;
1832
1833     if (!(severity & ERR_NOFILE))
1834         src_get(&lineno, &currentfile);
1835
1836     if (currentfile) {
1837         fprintf(error_file, "%s:%"PRId32": ", currentfile, lineno);
1838         nasm_free(currentfile);
1839     } else {
1840         fputs("nasm: ", error_file);
1841     }
1842
1843     nasm_verror_common(severity, fmt, ap);
1844 }
1845
1846 /**
1847  * MS style error reporting
1848  * This function prints an error message to error_file in the
1849  * style used by Visual C and some other Microsoft tools. An example
1850  * would be:
1851  * file.asm(50) : error: blah blah blah
1852  * where file.asm is the name of the file, 50 is the line number on
1853  * which the error occurs (or is detected) and "error:" is one of
1854  * the possible optional diagnostics -- it can be "error" or "warning"
1855  * or something else.  Finally the line terminates with the actual
1856  * error message.
1857  *
1858  * @param severity the severity of the warning or error
1859  * @param fmt the printf style format string
1860  */
1861 static void nasm_verror_vc(int severity, const char *fmt, va_list ap)
1862 {
1863     char *currentfile = NULL;
1864     int32_t lineno = 0;
1865
1866     if (is_suppressed_warning(severity))
1867         return;
1868
1869     if (!(severity & ERR_NOFILE))
1870         src_get(&lineno, &currentfile);
1871
1872     if (currentfile) {
1873         fprintf(error_file, "%s(%"PRId32") : ", currentfile, lineno);
1874         nasm_free(currentfile);
1875     } else {
1876         fputs("nasm: ", error_file);
1877     }
1878
1879     nasm_verror_common(severity, fmt, ap);
1880 }
1881
1882 /**
1883  * check for supressed warning
1884  * checks for suppressed warning or pass one only warning and we're
1885  * not in pass 1
1886  *
1887  * @param severity the severity of the warning or error
1888  * @return true if we should abort error/warning printing
1889  */
1890 static bool is_suppressed_warning(int severity)
1891 {
1892     /*
1893      * See if it's a suppressed warning.
1894      */
1895     return (severity & ERR_MASK) == ERR_WARNING &&
1896         (((severity & ERR_WARN_MASK) != 0 &&
1897           !warning_on[(severity & ERR_WARN_MASK) >> ERR_WARN_SHR]) ||
1898          /* See if it's a pass-one only warning and we're not in pass one. */
1899          ((severity & ERR_PASS1) && pass0 != 1) ||
1900          ((severity & ERR_PASS2) && pass0 != 2));
1901 }
1902
1903 /**
1904  * common error reporting
1905  * This is the common back end of the error reporting schemes currently
1906  * implemented.  It prints the nature of the warning and then the
1907  * specific error message to error_file and may or may not return.  It
1908  * doesn't return if the error severity is a "panic" or "debug" type.
1909  *
1910  * @param severity the severity of the warning or error
1911  * @param fmt the printf style format string
1912  */
1913 static void nasm_verror_common(int severity, const char *fmt, va_list args)
1914 {
1915     char msg[1024];
1916     const char *pfx;
1917
1918     switch (severity & (ERR_MASK|ERR_NO_SEVERITY)) {
1919     case ERR_WARNING:
1920         pfx = "warning: ";
1921         break;
1922     case ERR_NONFATAL:
1923         pfx = "error: ";
1924         break;
1925     case ERR_FATAL:
1926         pfx = "fatal: ";
1927         break;
1928     case ERR_PANIC:
1929         pfx = "panic: ";
1930         break;
1931     case ERR_DEBUG:
1932         pfx = "debug: ";
1933         break;
1934     default:
1935         pfx = "";
1936         break;
1937     }
1938
1939     vsnprintf(msg, sizeof msg, fmt, args);
1940
1941     fprintf(error_file, "%s%s\n", pfx, msg);
1942
1943     if (*listname)
1944         nasmlist.error(severity, pfx, msg);
1945
1946     if (severity & ERR_USAGE)
1947         want_usage = true;
1948
1949     switch (severity & ERR_MASK) {
1950     case ERR_DEBUG:
1951         /* no further action, by definition */
1952         break;
1953     case ERR_WARNING:
1954         if (warning_on[0])      /* Treat warnings as errors */
1955             terminate_after_phase = true;
1956         break;
1957     case ERR_NONFATAL:
1958         terminate_after_phase = true;
1959         break;
1960     case ERR_FATAL:
1961         if (ofile) {
1962             fclose(ofile);
1963             remove(outname);
1964             ofile = NULL;
1965         }
1966         if (want_usage)
1967             usage();
1968         exit(1);                /* instantly die */
1969         break;                  /* placate silly compilers */
1970     case ERR_PANIC:
1971         fflush(NULL);
1972         /*      abort();        *//* halt, catch fire, and dump core */
1973         exit(3);
1974         break;
1975     }
1976 }
1977
1978 static void usage(void)
1979 {
1980     fputs("type `nasm -h' for help\n", error_file);
1981 }
1982
1983 #define BUF_DELTA 512
1984
1985 static FILE *no_pp_fp;
1986 static ListGen *no_pp_list;
1987 static int32_t no_pp_lineinc;
1988
1989 static void no_pp_reset(char *file, int pass, ListGen * listgen,
1990                         StrList **deplist)
1991 {
1992     src_set_fname(nasm_strdup(file));
1993     src_set_linnum(0);
1994     no_pp_lineinc = 1;
1995     no_pp_fp = fopen(file, "r");
1996     if (!no_pp_fp)
1997         nasm_error(ERR_FATAL | ERR_NOFILE,
1998                    "unable to open input file `%s'", file);
1999     no_pp_list = listgen;
2000     (void)pass;                 /* placate compilers */
2001
2002     if (deplist) {
2003         StrList *sl = nasm_malloc(strlen(file)+1+sizeof sl->next);
2004         sl->next = NULL;
2005         strcpy(sl->str, file);
2006         *deplist = sl;
2007     }
2008 }
2009
2010 static char *no_pp_getline(void)
2011 {
2012     char *buffer, *p, *q;
2013     int bufsize;
2014
2015     bufsize = BUF_DELTA;
2016     buffer = nasm_malloc(BUF_DELTA);
2017     src_set_linnum(src_get_linnum() + no_pp_lineinc);
2018
2019     while (1) {                 /* Loop to handle %line */
2020
2021         p = buffer;
2022         while (1) {             /* Loop to handle long lines */
2023             q = fgets(p, bufsize - (p - buffer), no_pp_fp);
2024             if (!q)
2025                 break;
2026             p += strlen(p);
2027             if (p > buffer && p[-1] == '\n')
2028                 break;
2029             if (p - buffer > bufsize - 10) {
2030                 int offset;
2031                 offset = p - buffer;
2032                 bufsize += BUF_DELTA;
2033                 buffer = nasm_realloc(buffer, bufsize);
2034                 p = buffer + offset;
2035             }
2036         }
2037
2038         if (!q && p == buffer) {
2039             nasm_free(buffer);
2040             return NULL;
2041         }
2042
2043         /*
2044          * Play safe: remove CRs, LFs and any spurious ^Zs, if any of
2045          * them are present at the end of the line.
2046          */
2047         buffer[strcspn(buffer, "\r\n\032")] = '\0';
2048
2049         if (!nasm_strnicmp(buffer, "%line", 5)) {
2050             int32_t ln;
2051             int li;
2052             char *nm = nasm_malloc(strlen(buffer));
2053             if (sscanf(buffer + 5, "%"PRId32"+%d %s", &ln, &li, nm) == 3) {
2054                 nasm_free(src_set_fname(nm));
2055                 src_set_linnum(ln);
2056                 no_pp_lineinc = li;
2057                 continue;
2058             }
2059             nasm_free(nm);
2060         }
2061         break;
2062     }
2063
2064     no_pp_list->line(LIST_READ, buffer);
2065
2066     return buffer;
2067 }
2068
2069 static void no_pp_cleanup(int pass)
2070 {
2071     (void)pass;                     /* placate GCC */
2072     fclose(no_pp_fp);
2073 }
2074
2075 static uint32_t get_cpu(char *value)
2076 {
2077     if (!strcmp(value, "8086"))
2078         return IF_8086;
2079     if (!strcmp(value, "186"))
2080         return IF_186;
2081     if (!strcmp(value, "286"))
2082         return IF_286;
2083     if (!strcmp(value, "386"))
2084         return IF_386;
2085     if (!strcmp(value, "486"))
2086         return IF_486;
2087     if (!strcmp(value, "586") || !nasm_stricmp(value, "pentium"))
2088         return IF_PENT;
2089     if (!strcmp(value, "686") ||
2090         !nasm_stricmp(value, "ppro") ||
2091         !nasm_stricmp(value, "pentiumpro") || !nasm_stricmp(value, "p2"))
2092         return IF_P6;
2093     if (!nasm_stricmp(value, "p3") || !nasm_stricmp(value, "katmai"))
2094         return IF_KATMAI;
2095     if (!nasm_stricmp(value, "p4") ||   /* is this right? -- jrc */
2096         !nasm_stricmp(value, "willamette"))
2097         return IF_WILLAMETTE;
2098     if (!nasm_stricmp(value, "prescott"))
2099         return IF_PRESCOTT;
2100     if (!nasm_stricmp(value, "x64") ||
2101         !nasm_stricmp(value, "x86-64"))
2102         return IF_X86_64;
2103     if (!nasm_stricmp(value, "ia64") ||
2104         !nasm_stricmp(value, "ia-64") ||
2105         !nasm_stricmp(value, "itanium") ||
2106         !nasm_stricmp(value, "itanic") || !nasm_stricmp(value, "merced"))
2107         return IF_IA64;
2108
2109     nasm_error(pass0 < 2 ? ERR_NONFATAL : ERR_FATAL,
2110                  "unknown 'cpu' type");
2111
2112     return IF_PLEVEL;           /* the maximum level */
2113 }
2114
2115 static int get_bits(char *value)
2116 {
2117     int i;
2118
2119     if ((i = atoi(value)) == 16)
2120         return i;               /* set for a 16-bit segment */
2121     else if (i == 32) {
2122         if (cpu < IF_386) {
2123             nasm_error(ERR_NONFATAL,
2124                          "cannot specify 32-bit segment on processor below a 386");
2125             i = 16;
2126         }
2127     } else if (i == 64) {
2128         if (cpu < IF_X86_64) {
2129             nasm_error(ERR_NONFATAL,
2130                          "cannot specify 64-bit segment on processor below an x86-64");
2131             i = 16;
2132         }
2133         if (i != maxbits) {
2134             nasm_error(ERR_NONFATAL,
2135                          "%s output format does not support 64-bit code",
2136                          ofmt->shortname);
2137             i = 16;
2138         }
2139     } else {
2140         nasm_error(pass0 < 2 ? ERR_NONFATAL : ERR_FATAL,
2141                      "`%s' is not a valid segment size; must be 16, 32 or 64",
2142                      value);
2143         i = 16;
2144     }
2145     return i;
2146 }
2147
2148 /* end of nasm.c */