1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009 |
- /*
- * Program entry point, command line parsing
- *
- * Copyright (C) 2001-2008 Peter Johnson
- * Copyright (C) 2007-2008 Samuel Thibault
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND OTHER CONTRIBUTORS ``AS IS''
- * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR OTHER CONTRIBUTORS BE
- * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
- * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
- * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
- * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
- * POSSIBILITY OF SUCH DAMAGE.
- */
- #include <util.h>
- #include <ctype.h>
- #include <libyasm/compat-queue.h>
- #include <libyasm/bitvect.h>
- #include <libyasm.h>
- #ifdef HAVE_LIBGEN_H
- #include <libgen.h>
- #endif
- #include "tasm-options.h"
- #if defined(CMAKE_BUILD) && defined(BUILD_SHARED_LIBS)
- #include "yasm-plugin.h"
- #endif
- #include "license.c"
- #define DEFAULT_OBJFMT_MODULE "bin"
- #if defined(CMAKE_BUILD) && !defined(BUILD_SHARED_LIBS)
- void yasm_init_plugin(void);
- #endif
- /*@null@*/ /*@only@*/ static char *obj_filename = NULL, *in_filename = NULL;
- /*@null@*/ /*@only@*/ static char *list_filename = NULL, *xref_filename = NULL;
- /*@null@*/ /*@only@*/ static char *machine_name = NULL;
- static int special_options = 0;
- static int segment_ordering = 0;
- static int cross_reference = 0;
- static int floating_point = 0;
- static int listing = 0;
- static int expanded_listing = 0;
- static int case_sensitivity = 0;
- static int valid_length = -1;
- /*@null@*/ /*@dependent@*/ static yasm_arch *cur_arch = NULL;
- /*@null@*/ /*@dependent@*/ static const yasm_arch_module *
- cur_arch_module = NULL;
- /*@null@*/ /*@dependent@*/ static const yasm_parser_module *
- cur_parser_module = NULL;
- /*@null@*/ /*@dependent@*/ static yasm_preproc *cur_preproc = NULL;
- /*@null@*/ /*@dependent@*/ static const yasm_preproc_module *
- cur_preproc_module = NULL;
- /*@null@*/ static char *objfmt_keyword = NULL;
- /*@null@*/ /*@dependent@*/ static const yasm_objfmt_module *
- cur_objfmt_module = NULL;
- /*@null@*/ /*@dependent@*/ static const yasm_dbgfmt_module *
- cur_dbgfmt_module = NULL;
- /*@null@*/ /*@dependent@*/ static yasm_listfmt *cur_listfmt = NULL;
- /*@null@*/ /*@dependent@*/ static const yasm_listfmt_module *
- cur_listfmt_module = NULL;
- static int warning_error = 0; /* warnings being treated as errors */
- static FILE *errfile;
- /*@null@*/ /*@only@*/ static char *error_filename = NULL;
- /*@null@*/ /*@dependent@*/ static FILE *open_file(const char *filename,
- const char *mode);
- static void check_errors(/*@only@*/ yasm_errwarns *errwarns,
- /*@only@*/ yasm_object *object,
- /*@only@*/ yasm_linemap *linemap);
- static void cleanup(/*@null@*/ /*@only@*/ yasm_object *object);
- /* Forward declarations: cmd line parser handlers */
- static int opt_special_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_segment_ordering_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_cross_reference_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_floating_point_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_ignore(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_listing_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_case_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_valid_length_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_warning_handler(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_preproc_option(char *cmd, /*@null@*/ char *param, int extra);
- static int opt_exe_handler(char *cmd, /*@null@*/ char *param, int extra);
- static /*@only@*/ char *replace_extension(const char *orig, /*@null@*/
- const char *ext, const char *def);
- static void print_error(const char *fmt, ...);
- static /*@exits@*/ void handle_yasm_int_error(const char *file,
- unsigned int line,
- const char *message);
- static /*@exits@*/ void handle_yasm_fatal(const char *message, va_list va);
- static const char *handle_yasm_gettext(const char *msgid);
- static void print_yasm_error(const char *filename, unsigned long line,
- const char *msg, /*@null@*/ const char *xref_fn,
- unsigned long xref_line,
- /*@null@*/ const char *xref_msg);
- static void print_yasm_warning(const char *filename, unsigned long line,
- const char *msg);
- static void apply_preproc_builtins(void);
- static void apply_preproc_standard_macros(const yasm_stdmac *stdmacs);
- static void apply_preproc_saved_options(void);
- static void print_list_keyword_desc(const char *name, const char *keyword);
- /* values for special_options */
- #define SPECIAL_SHOW_HELP 0x01
- #define SPECIAL_SHOW_VERSION 0x02
- #define SPECIAL_SHOW_LICENSE 0x04
- #define SEGMENT_ORDERING_ALPHABETIC 0x01
- #define SEGMENT_ORDERING_SOURCE 0x02
- #define FP_EMULATED 0x01
- #define FP_REAL 0x02
- #define CASE_ALL 0x01
- #define CASE_GLOBALS 0x02
- #define CASE_NONE 0x04
- #define DEBUG_FULL 0x01
- #define DEBUG_LINES 0x02
- #define DEBUG_NONE 0x04
- /* command line options */
- static opt_option options[] =
- {
- { "version", 0, opt_special_handler, SPECIAL_SHOW_VERSION,
- N_("show version text"), NULL },
- { "license", 0, opt_special_handler, SPECIAL_SHOW_LICENSE,
- N_("show license text"), NULL },
- { "a", 0, opt_segment_ordering_handler, SEGMENT_ORDERING_ALPHABETIC,
- N_("Alphabetic segment ordering"), NULL },
- { "s", 0, opt_segment_ordering_handler, SEGMENT_ORDERING_SOURCE,
- N_("Source segment ordering"), NULL },
- { "c", 0, opt_cross_reference_handler, 0,
- N_("Generate cross-reference in listing"), NULL },
- { "d", 1, opt_preproc_option, 2,
- N_("pre-define a macro, optionally to value"), N_("macro[=value]") },
- { "e", 0, opt_floating_point_handler, FP_EMULATED,
- N_("Emulated floating-point instructions (not supported)"), NULL },
- { "r", 0, opt_floating_point_handler, FP_REAL,
- N_("Real floating-point instructions"), NULL },
- { "h", 0, opt_special_handler, SPECIAL_SHOW_HELP,
- N_("show help text"), NULL },
- { "?", 0, opt_special_handler, SPECIAL_SHOW_HELP,
- N_("show help text"), NULL },
- { "i", 1, opt_preproc_option, 0,
- N_("add include path"), N_("path") },
- { "j", 1, opt_ignore, 0,
- N_("Jam in an assemble directive CMD (eg. /jIDEAL) (not supported)"), NULL },
- { "k", 1, opt_ignore, 0,
- N_("Hash table capacity (ignored)"), N_("# symbols") },
- { "l", 0, opt_listing_handler, 0,
- N_("Generate listing"), N_("l=normal listing, la=expanded listing") },
- { "ml", 0, opt_case_handler, CASE_ALL,
- N_("Case sensitivity on all symbols"), NULL },
- { "mx", 0, opt_case_handler, CASE_GLOBALS,
- N_("Case sensitivity on global symbols"), NULL },
- { "mu", 0, opt_case_handler, CASE_NONE,
- N_("No case sensitivity on symbols"), NULL },
- { "mv", 0, opt_valid_length_handler, 0,
- N_("Set maximum valid length for symbols"), N_("length") },
- { "m", 1, opt_ignore, 0,
- N_("Allow multiple passes to resolve forward reference (ignored)"), N_("number of passes") },
- { "n", 0, opt_ignore, 0,
- N_("Suppress symbol tables in listing"), NULL },
- { "o", 0, opt_ignore, 0,
- N_("Object code"), N_("os: standard, o: standard w/overlays, op: Phar Lap, oi: IBM") },
- { "p", 0, opt_ignore, 0,
- N_("Check for code segment overrides in protected mode"), NULL },
- { "q", 0, opt_ignore, 0,
- N_("Suppress OBJ records not needed for linking (ignored)"), NULL },
- { "t", 0, opt_ignore, 0,
- N_("Suppress messages if successful assembly"), NULL },
- { "u", 0, opt_ignore, 0,
- N_("Set version emulation"), N_("Version") },
- { "w", 1, opt_warning_handler, 0,
- N_("Set warning level"), N_("w0=none, w1=w2=warnings on, w-xxx/w+xxx=disable/enable warning xxx") },
- { "x", 0, opt_ignore, 0,
- N_("Include false conditionals in listing"), NULL },
- { "zi", 0, opt_ignore, DEBUG_FULL,
- N_("Full debug info"), NULL },
- { "zd", 0, opt_ignore, DEBUG_LINES,
- N_("Line numbers debug info"), NULL },
- { "zn", 0, opt_ignore, DEBUG_NONE,
- N_("No debug info"), NULL },
- { "z", 0, opt_ignore, 0,
- N_("Display source line with error message (ignored)"), NULL },
- { "b", 0, opt_exe_handler, 0,
- N_("Build a (very) basic .exe file"), NULL },
- };
- /* version message */
- /*@observer@*/ static const char *version_msg[] = {
- PACKAGE_STRING,
- "Compiled on " __DATE__ ".",
- "Copyright (c) 2001-2010 Peter Johnson and other Yasm developers.",
- "Run yasm --license for licensing overview and summary."
- };
- /* help messages */
- /*@observer@*/ static const char *help_head = N_(
- "usage: tasm [option]* source [,object] [,listing] [,xref] \n"
- "Options:\n");
- /*@observer@*/ static const char *help_tail = N_(
- "\n"
- "source is asm source to be assembled.\n"
- "\n"
- "Sample invocation:\n"
- " tasm /zi source.asm\n"
- "\n"
- "Report bugs to bug-yasm@tortall.net\n");
- /* parsed command line storage until appropriate modules have been loaded */
- typedef STAILQ_HEAD(constcharparam_head, constcharparam) constcharparam_head;
- typedef struct constcharparam {
- STAILQ_ENTRY(constcharparam) link;
- const char *param;
- int id;
- } constcharparam;
- static constcharparam_head preproc_options;
- static int
- do_assemble(void)
- {
- yasm_object *object;
- const char *base_filename;
- /*@null@*/ FILE *obj = NULL;
- yasm_arch_create_error arch_error;
- yasm_linemap *linemap;
- yasm_errwarns *errwarns = yasm_errwarns_create();
- int i, matched;
- /* Initialize line map */
- linemap = yasm_linemap_create();
- yasm_linemap_set(linemap, in_filename, 0, 1, 1);
- /* determine the object filename if not specified */
- if (!obj_filename) {
- if (in_filename == NULL)
- /* Default to yasm.out if no obj filename specified */
- obj_filename = yasm__xstrdup("yasm.out");
- else {
- /* replace (or add) extension to base filename */
- yasm__splitpath(in_filename, &base_filename);
- if (base_filename[0] == '\0')
- obj_filename = yasm__xstrdup("yasm.out");
- else
- obj_filename = replace_extension(base_filename,
- "obj",
- "yasm.out");
- }
- }
- cur_arch = yasm_arch_create(cur_arch_module, machine_name,
- cur_parser_module->keyword, &arch_error);
- if (!cur_arch) {
- switch (arch_error) {
- case YASM_ARCH_CREATE_BAD_MACHINE:
- print_error(_("%s: `%s' is not a valid %s for %s `%s'"),
- _("FATAL"), machine_name, _("machine"),
- _("architecture"), cur_arch_module->keyword);
- break;
- case YASM_ARCH_CREATE_BAD_PARSER:
- print_error(_("%s: `%s' is not a valid %s for %s `%s'"),
- _("FATAL"), cur_parser_module->keyword,
- _("parser"), _("architecture"),
- cur_arch_module->keyword);
- break;
- default:
- print_error(_("%s: unknown architecture error"), _("FATAL"));
- }
- return EXIT_FAILURE;
- }
- /* Create object */
- object = yasm_object_create(in_filename, obj_filename, cur_arch,
- cur_objfmt_module, cur_dbgfmt_module);
- if (!object) {
- yasm_error_class eclass;
- unsigned long xrefline;
- /*@only@*/ /*@null@*/ char *estr, *xrefstr;
- yasm_error_fetch(&eclass, &estr, &xrefline, &xrefstr);
- print_error("%s: %s", _("FATAL"), estr);
- yasm_xfree(estr);
- yasm_xfree(xrefstr);
- cleanup(object);
- return EXIT_FAILURE;
- }
- /* Get a fresh copy of objfmt_module as it may have changed. */
- cur_objfmt_module = ((yasm_objfmt_base *)object->objfmt)->module;
- /* Check to see if the requested preprocessor is in the allowed list
- * for the active parser.
- */
- matched = 0;
- for (i=0; cur_parser_module->preproc_keywords[i]; i++)
- if (yasm__strcasecmp(cur_parser_module->preproc_keywords[i],
- cur_preproc_module->keyword) == 0)
- matched = 1;
- if (!matched) {
- print_error(_("%s: `%s' is not a valid %s for %s `%s'"), _("FATAL"),
- cur_preproc_module->keyword, _("preprocessor"),
- _("parser"), cur_parser_module->keyword);
- cleanup(object);
- return EXIT_FAILURE;
- }
- cur_preproc = yasm_preproc_create(cur_preproc_module, in_filename,
- object->symtab, linemap, errwarns);
- apply_preproc_builtins();
- apply_preproc_standard_macros(cur_parser_module->stdmacs);
- apply_preproc_standard_macros(cur_objfmt_module->stdmacs);
- apply_preproc_saved_options();
- /* Get initial x86 BITS setting from object format */
- if (strcmp(cur_arch_module->keyword, "x86") == 0) {
- yasm_arch_set_var(cur_arch, "mode_bits",
- cur_objfmt_module->default_x86_mode_bits);
- }
- /* Parse! */
- cur_parser_module->do_parse(object, cur_preproc, list_filename != NULL,
- linemap, errwarns);
- check_errors(errwarns, object, linemap);
- /* Finalize parse */
- yasm_object_finalize(object, errwarns);
- check_errors(errwarns, object, linemap);
- /* Optimize */
- yasm_object_optimize(object, errwarns);
- check_errors(errwarns, object, linemap);
- /* generate any debugging information */
- yasm_dbgfmt_generate(object, linemap, errwarns);
- check_errors(errwarns, object, linemap);
- /* open the object file for output (if not already opened by dbg objfmt) */
- if (!obj && strcmp(cur_objfmt_module->keyword, "dbg") != 0) {
- obj = open_file(obj_filename, "wb");
- if (!obj) {
- cleanup(object);
- return EXIT_FAILURE;
- }
- }
- /* Write the object file */
- yasm_objfmt_output(object, obj?obj:stderr,
- strcmp(cur_dbgfmt_module->keyword, "null"), errwarns);
- /* Close object file */
- if (obj)
- fclose(obj);
- /* If we had an error at this point, we also need to delete the output
- * object file (to make sure it's not left newer than the source).
- */
- if (yasm_errwarns_num_errors(errwarns, warning_error) > 0)
- remove(obj_filename);
- check_errors(errwarns, object, linemap);
- /* Open and write the list file */
- if (list_filename) {
- FILE *list = open_file(list_filename, "wt");
- if (!list) {
- cleanup(object);
- return EXIT_FAILURE;
- }
- /* Initialize the list format */
- cur_listfmt = yasm_listfmt_create(cur_listfmt_module, in_filename,
- obj_filename);
- yasm_listfmt_output(cur_listfmt, list, linemap, cur_arch);
- fclose(list);
- }
- yasm_errwarns_output_all(errwarns, linemap, warning_error,
- print_yasm_error, print_yasm_warning);
- yasm_linemap_destroy(linemap);
- yasm_errwarns_destroy(errwarns);
- cleanup(object);
- return EXIT_SUCCESS;
- }
- /* main function */
- /*@-globstate -unrecog@*/
- int
- main(int argc, char *argv[])
- {
- size_t i;
- errfile = stderr;
- #if defined(HAVE_SETLOCALE) && defined(HAVE_LC_MESSAGES)
- setlocale(LC_MESSAGES, "");
- #endif
- #if defined(LOCALEDIR)
- bindtextdomain(PACKAGE, LOCALEDIR);
- #endif
- textdomain(PACKAGE);
- /* Initialize errwarn handling */
- yasm_internal_error_ = handle_yasm_int_error;
- yasm_fatal = handle_yasm_fatal;
- yasm_gettext_hook = handle_yasm_gettext;
- yasm_errwarn_initialize();
- /* Initialize BitVector (needed for intnum/floatnum). */
- if (BitVector_Boot() != ErrCode_Ok) {
- print_error(_("%s: could not initialize BitVector"), _("FATAL"));
- return EXIT_FAILURE;
- }
- /* Initialize intnum and floatnum */
- yasm_intnum_initialize();
- yasm_floatnum_initialize();
- #ifdef CMAKE_BUILD
- /* Load standard modules */
- #ifdef BUILD_SHARED_LIBS
- if (!load_plugin("yasmstd")) {
- print_error(_("%s: could not load standard modules"), _("FATAL"));
- return EXIT_FAILURE;
- }
- #else
- yasm_init_plugin();
- #endif
- #endif
- /* Initialize parameter storage */
- STAILQ_INIT(&preproc_options);
- if (parse_cmdline(argc, argv, options, NELEMS(options), print_error))
- return EXIT_FAILURE;
- switch (special_options) {
- case SPECIAL_SHOW_HELP:
- /* Does gettext calls internally */
- help_msg(help_head, help_tail, options, NELEMS(options));
- return EXIT_SUCCESS;
- case SPECIAL_SHOW_VERSION:
- for (i=0; i<NELEMS(version_msg); i++)
- printf("%s\n", version_msg[i]);
- return EXIT_SUCCESS;
- case SPECIAL_SHOW_LICENSE:
- for (i=0; i<NELEMS(license_msg); i++)
- printf("%s\n", license_msg[i]);
- return EXIT_SUCCESS;
- }
- /* Open error file if specified. */
- if (error_filename) {
- errfile = open_file(error_filename, "wt");
- if (!errfile)
- return EXIT_FAILURE;
- }
- /* If not already specified, default to bin as the object format. */
- if (!cur_objfmt_module) {
- if (!objfmt_keyword)
- objfmt_keyword = yasm__xstrdup(DEFAULT_OBJFMT_MODULE);
- cur_objfmt_module = yasm_load_objfmt(objfmt_keyword);
- if (!cur_objfmt_module) {
- print_error(_("%s: could not load default %s"), _("FATAL"),
- _("object format"));
- return EXIT_FAILURE;
- }
- }
- /* TASM's architecture is x86 */
- cur_arch_module = yasm_load_arch("x86");
- if (!cur_arch_module) {
- print_error(_("%s: could not load %s"), _("FATAL"),
- _("architecture"));
- return EXIT_FAILURE;
- }
- machine_name =
- yasm__xstrdup(cur_arch_module->default_machine_keyword);
- /* Check for arch help */
- if (machine_name && strcmp(machine_name, "help") == 0) {
- const yasm_arch_machine *m = cur_arch_module->machines;
- printf(_("Available %s for %s `%s':\n"), _("machines"),
- _("architecture"), cur_arch_module->keyword);
- while (m->keyword && m->name) {
- print_list_keyword_desc(m->name, m->keyword);
- m++;
- }
- return EXIT_SUCCESS;
- }
- cur_parser_module = yasm_load_parser("tasm");
- if (!cur_parser_module) {
- print_error(_("%s: could not load %s"), _("FATAL"),
- _("parser"));
- cleanup(NULL);
- return EXIT_FAILURE;
- }
- /* If not already specified, default to the parser's default preproc. */
- if (!cur_preproc_module) {
- cur_preproc_module =
- yasm_load_preproc(cur_parser_module->default_preproc_keyword);
- if (!cur_preproc_module) {
- print_error(_("%s: could not load default %s"), _("FATAL"),
- _("preprocessor"));
- cleanup(NULL);
- return EXIT_FAILURE;
- }
- }
- /* Determine input filename and open input file. */
- if (!in_filename) {
- print_error(_("No input files specified"));
- return EXIT_FAILURE;
- }
- /* If list file enabled, make sure we have a list format loaded. */
- if (list_filename) {
- /* use nasm as the list format. */
- cur_listfmt_module = yasm_load_listfmt("nasm");
- }
- /* If not already specified, default to null as the debug format. */
- if (!cur_dbgfmt_module) {
- cur_dbgfmt_module = yasm_load_dbgfmt("null");
- if (!cur_dbgfmt_module) {
- print_error(_("%s: could not load default %s"), _("FATAL"),
- _("debug format"));
- return EXIT_FAILURE;
- }
- }
- return do_assemble();
- }
- /*@=globstate =unrecog@*/
- /* Open the object file. Returns 0 on failure. */
- static FILE *
- open_file(const char *filename, const char *mode)
- {
- FILE *f;
- f = fopen(filename, mode);
- if (!f)
- print_error(_("could not open file `%s'"), filename);
- return f;
- }
- static void
- check_errors(yasm_errwarns *errwarns, yasm_object *object,
- yasm_linemap *linemap)
- {
- if (yasm_errwarns_num_errors(errwarns, warning_error) > 0) {
- yasm_errwarns_output_all(errwarns, linemap, warning_error,
- print_yasm_error, print_yasm_warning);
- yasm_linemap_destroy(linemap);
- yasm_errwarns_destroy(errwarns);
- cleanup(object);
- exit(EXIT_FAILURE);
- }
- }
- /* Define DO_FREE to 1 to enable deallocation of all data structures.
- * Useful for detecting memory leaks, but slows down execution unnecessarily
- * (as the OS will free everything we miss here).
- */
- #define DO_FREE 1
- /* Cleans up all allocated structures. */
- static void
- cleanup(yasm_object *object)
- {
- if (DO_FREE) {
- if (cur_listfmt)
- yasm_listfmt_destroy(cur_listfmt);
- if (cur_preproc)
- yasm_preproc_destroy(cur_preproc);
- if (object)
- yasm_object_destroy(object);
- yasm_floatnum_cleanup();
- yasm_intnum_cleanup();
- yasm_errwarn_cleanup();
- BitVector_Shutdown();
- }
- if (DO_FREE) {
- if (in_filename)
- yasm_xfree(in_filename);
- if (obj_filename)
- yasm_xfree(obj_filename);
- if (list_filename)
- yasm_xfree(list_filename);
- if (xref_filename)
- yasm_xfree(xref_filename);
- if (machine_name)
- yasm_xfree(machine_name);
- if (objfmt_keyword)
- yasm_xfree(objfmt_keyword);
- }
- if (errfile != stderr && errfile != stdout)
- fclose(errfile);
- #if defined(CMAKE_BUILD) && defined(BUILD_SHARED_LIBS)
- unload_plugins();
- #endif
- }
- /*
- * Command line options handlers
- */
- static char ** const filenames[] = {
- &in_filename, &obj_filename, &list_filename, &xref_filename, NULL
- }, ** const * cur_filename = &filenames[0];
- static int filename_handler(char *param) {
- if (!*cur_filename) {
- print_error(_("error: too many files on command line."));
- return 1;
- }
- if (*param)
- **cur_filename = yasm__xstrdup(param);
- return 0;
- }
- int
- not_an_option_handler(char *param) {
- char *c, *d = param;
- while ((c = strchr(d, ','))) {
- *c = '\0';
- if (filename_handler(d))
- return 1;
- d = c + 1;
- cur_filename++;
- }
- filename_handler(d);
- return 0;
- }
- static int
- opt_special_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- if (special_options == 0)
- special_options = extra;
- return 0;
- }
- static int
- opt_segment_ordering_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- segment_ordering = extra;
- return 0;
- }
- static int
- opt_cross_reference_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- cross_reference = 1;
- return 0;
- }
- static int
- opt_floating_point_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- floating_point = extra;
- return 0;
- }
- static int
- opt_ignore(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- return 0;
- }
- static int
- opt_listing_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- if (param && param[0]) {
- if (param[0] != 'a')
- return 1;
- expanded_listing = 1;
- }
- listing = 1;
- return 0;
- }
- static int
- opt_case_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- case_sensitivity = extra;
- return 0;
- }
- static int
- opt_valid_length_handler(/*@unused@*/ char *cmd, /*@unused@*/ char *param, int extra)
- {
- valid_length = atoi(param);
- return 0;
- }
- static int
- opt_warning_handler(char *cmd, /*@unused@*/ char *param, int extra)
- {
- /* is it disabling the warning instead of enabling? */
- void (*action)(yasm_warn_class wclass) = NULL;
- if (cmd[0] == '0') {
- /* /w0, disable warnings */
- yasm_warn_disable_all();
- return 0;
- }
- if (cmd[0] == '1' || cmd[0] == '2') {
- /* /w[12], enable warnings */
- yasm_warn_enable(YASM_WARN_UNREC_CHAR);
- yasm_warn_enable(YASM_WARN_ORPHAN_LABEL);
- yasm_warn_enable(YASM_WARN_UNINIT_CONTENTS);
- return 0;
- }
- /* detect no- prefix to disable the warning */
- if (cmd[0] == '-') {
- action = yasm_warn_disable;
- } else if (cmd[0] == '+') {
- action = yasm_warn_enable;
- } else return 1;
- /* skip past '+/-' */
- cmd++;
- if (cmd[0] == '\0')
- /* just /w- or /w+, so definitely not valid */
- return 1;
- else if (strcmp(cmd, "error") == 0)
- warning_error = (action == yasm_warn_enable);
- else if (strcmp(cmd, "unrecognized-char") == 0)
- action(YASM_WARN_UNREC_CHAR);
- else if (strcmp(cmd, "orphan-labels") == 0)
- action(YASM_WARN_ORPHAN_LABEL);
- else if (strcmp(cmd, "uninit-contents") == 0)
- action(YASM_WARN_UNINIT_CONTENTS);
- else if (strcmp(cmd, "size-override") == 0)
- action(YASM_WARN_SIZE_OVERRIDE);
- else
- return 1;
- return 0;
- }
- static int
- opt_preproc_option(/*@unused@*/ char *cmd, char *param, int extra)
- {
- constcharparam *cp;
- cp = yasm_xmalloc(sizeof(constcharparam));
- cp->param = param;
- cp->id = extra;
- STAILQ_INSERT_TAIL(&preproc_options, cp, link);
- return 0;
- }
- static int
- opt_exe_handler(char *cmd, /*@unused@*/ char *param, int extra)
- {
- objfmt_keyword = yasm__xstrdup("dosexe");
- return 0;
- }
- static void
- apply_preproc_builtins()
- {
- char *predef;
- if (!objfmt_keyword)
- objfmt_keyword = yasm__xstrdup(DEFAULT_OBJFMT_MODULE);
- /* Define standard YASM assembly-time macro constants */
- predef = yasm_xmalloc(strlen("__YASM_OBJFMT__=")
- + strlen(objfmt_keyword) + 1);
- strcpy(predef, "__YASM_OBJFMT__=");
- strcat(predef, objfmt_keyword);
- yasm_preproc_define_builtin(cur_preproc, predef);
- yasm_xfree(predef);
- }
- static void
- apply_preproc_standard_macros(const yasm_stdmac *stdmacs)
- {
- int i, matched;
- if (!stdmacs)
- return;
- matched = -1;
- for (i=0; stdmacs[i].parser; i++)
- if (yasm__strcasecmp(stdmacs[i].parser,
- cur_parser_module->keyword) == 0 &&
- yasm__strcasecmp(stdmacs[i].preproc,
- cur_preproc_module->keyword) == 0)
- matched = i;
- if (matched >= 0 && stdmacs[matched].macros)
- yasm_preproc_add_standard(cur_preproc, stdmacs[matched].macros);
- }
- static void
- apply_preproc_saved_options()
- {
- constcharparam *cp, *cpnext;
- void (*funcs[3])(yasm_preproc *, const char *);
- funcs[0] = cur_preproc_module->add_include_file;
- funcs[1] = cur_preproc_module->predefine_macro;
- funcs[2] = cur_preproc_module->undefine_macro;
- STAILQ_FOREACH(cp, &preproc_options, link) {
- if (0 <= cp->id && cp->id < 3 && funcs[cp->id])
- funcs[cp->id](cur_preproc, cp->param);
- }
- cp = STAILQ_FIRST(&preproc_options);
- while (cp != NULL) {
- cpnext = STAILQ_NEXT(cp, link);
- yasm_xfree(cp);
- cp = cpnext;
- }
- STAILQ_INIT(&preproc_options);
- }
- /* Replace extension on a filename (or append one if none is present).
- * If output filename would be identical to input (same extension out as in),
- * returns (copy of) def.
- * A NULL ext means the trailing '.' should NOT be included, whereas a "" ext
- * means the trailing '.' should be included.
- */
- static char *
- replace_extension(const char *orig, /*@null@*/ const char *ext,
- const char *def)
- {
- char *out, *outext;
- size_t deflen, outlen;
- /* allocate enough space for full existing name + extension */
- outlen = strlen(orig) + 2;
- if (ext)
- outlen += strlen(ext) + 1;
- deflen = strlen(def) + 1;
- if (outlen < deflen)
- outlen = deflen;
- out = yasm_xmalloc(outlen);
- strcpy(out, orig);
- outext = strrchr(out, '.');
- if (outext) {
- /* Existing extension: make sure it's not the same as the replacement
- * (as we don't want to overwrite the source file).
- */
- outext++; /* advance past '.' */
- if (ext && strcmp(outext, ext) == 0) {
- outext = NULL; /* indicate default should be used */
- print_error(
- _("file name already ends in `.%s': output will be in `%s'"),
- ext, def);
- }
- } else {
- /* No extension: make sure the output extension is not empty
- * (again, we don't want to overwrite the source file).
- */
- if (!ext)
- print_error(
- _("file name already has no extension: output will be in `%s'"),
- def);
- else {
- outext = strrchr(out, '\0'); /* point to end of the string */
- *outext++ = '.'; /* append '.' */
- }
- }
- /* replace extension or use default name */
- if (outext) {
- if (!ext) {
- /* Back up and replace '.' with string terminator */
- outext--;
- *outext = '\0';
- } else
- strcpy(outext, ext);
- } else
- strcpy(out, def);
- return out;
- }
- void
- print_list_keyword_desc(const char *name, const char *keyword)
- {
- printf("%4s%-12s%s\n", "", keyword, name);
- }
- static void
- print_error(const char *fmt, ...)
- {
- va_list va;
- fprintf(errfile, "tasm: ");
- va_start(va, fmt);
- vfprintf(errfile, fmt, va);
- va_end(va);
- fputc('\n', errfile);
- }
- static /*@exits@*/ void
- handle_yasm_int_error(const char *file, unsigned int line, const char *message)
- {
- fprintf(stderr, _("INTERNAL ERROR at %s, line %u: %s\n"), file, line,
- gettext(message));
- #ifdef HAVE_ABORT
- abort();
- #else
- exit(EXIT_FAILURE);
- #endif
- }
- static /*@exits@*/ void
- handle_yasm_fatal(const char *fmt, va_list va)
- {
- fprintf(errfile, "**%s**: ", _("Fatal"));
- vfprintf(errfile, gettext(fmt), va);
- fputc('\n', errfile);
- exit(EXIT_FAILURE);
- }
- static const char *
- handle_yasm_gettext(const char *msgid)
- {
- return gettext(msgid);
- }
- static void
- print_yasm_error(const char *filename, unsigned long line, const char *msg,
- const char *xref_fn, unsigned long xref_line,
- const char *xref_msg)
- {
- if (line)
- fprintf(errfile, "**%s** %s(%lu) %s\n", _("Error"), filename, line, msg);
- else
- fprintf(errfile, "**%s** %s %s\n", _("Error"), filename, msg);
- if (/* xref_fn && */ xref_msg) {
- if (xref_line)
- fprintf(errfile, "**%s** %s(%lu) %s\n", _("Error"), filename, xref_line, xref_msg);
- else
- fprintf(errfile, "**%s** %s %s\n", _("Error"), filename, xref_msg);
- }
- }
- static void
- print_yasm_warning(const char *filename, unsigned long line, const char *msg)
- {
- if (line)
- fprintf(errfile, "*%s* %s(%lu) %s\n", _("Warning"), filename, line, msg);
- else
- fprintf(errfile, "*%s* %s %s\n", _("Warning"), filename, msg);
- }
|