/* * Copyright (c) 2015 Andrew Kelley * * This file is part of zig, which is MIT licensed. * See http://opensource.org/licenses/MIT */ #include "ast_render.hpp" #include "buffer.hpp" #include "codegen.hpp" #include "compiler.hpp" #include "config.h" #include "error.hpp" #include "os.hpp" #include "target.hpp" #include "libc_installation.hpp" #include static int print_error_usage(const char *arg0) { fprintf(stderr, "See `%s --help` for detailed usage information\n", arg0); return EXIT_FAILURE; } static int print_full_usage(const char *arg0, FILE *file, int return_code) { fprintf(file, "Usage: %s [command] [options]\n" "\n" "Commands:\n" " build build project from build.zig\n" " build-exe [source] create executable from source or object files\n" " build-lib [source] create library from source or object files\n" " build-obj [source] create object from source or assembly\n" " builtin show the source code of that @import(\"builtin\")\n" " cc C compiler\n" " fmt parse files and render in canonical zig format\n" " id print the base64-encoded compiler id\n" " init-exe initialize a `zig build` application in the cwd\n" " init-lib initialize a `zig build` library in the cwd\n" " libc [paths_file] Display native libc paths file or validate one\n" " run [source] create executable and run immediately\n" " translate-c [source] convert c code to zig code\n" " targets list available compilation targets\n" " test [source] create and run a test build\n" " version print version number and exit\n" " zen print zen of zig and exit\n" "\n" "Compile Options:\n" " --assembly [source] add assembly file to build\n" " --c-source [options] [file] compile C source code\n" " --cache-dir [path] override the cache directory\n" " --cache [auto|off|on] build in global cache, print out paths to stdout\n" " --color [auto|off|on] enable or disable colored error messages\n" " --disable-pic disable Position Independent Code for libraries\n" " --disable-valgrind omit valgrind client requests in debug builds\n" " --enable-valgrind include valgrind client requests release builds\n" " --emit [asm|bin|llvm-ir] emit a specific file format as compilation output\n" " -ftime-report print timing diagnostics\n" " --libc [file] Provide a file which specifies libc paths\n" " --name [name] override output name\n" " --output [file] override destination path\n" " --output-h [file] generate header file\n" " --output-lib [file] override import library path\n" " --pkg-begin [name] [path] make pkg available to import and push current pkg\n" " --pkg-end pop current pkg\n" " --release-fast build with optimizations on and safety off\n" " --release-safe build with optimizations on and safety on\n" " --release-small build with size optimizations on and safety off\n" " --single-threaded source may assume it is only used single-threaded\n" " --static output will be statically linked\n" " --strip exclude debug symbols\n" " --target-arch [name] specify target architecture\n" " --target-environ [name] specify target environment\n" " --target-os [name] specify target operating system\n" " --verbose-tokenize enable compiler debug output for tokenization\n" " --verbose-ast enable compiler debug output for AST parsing\n" " --verbose-link enable compiler debug output for linking\n" " --verbose-ir enable compiler debug output for Zig IR\n" " --verbose-llvm-ir enable compiler debug output for LLVM IR\n" " --verbose-cimport enable compiler debug output for C imports\n" " --verbose-cc enable compiler debug output for C compilation\n" " -dirafter [dir] same as -isystem but do it last\n" " -isystem [dir] add additional search path for other .h files\n" " -mllvm [arg] forward an arg to LLVM's option processing\n" " --override-std-dir [arg] use an alternate Zig standard library\n" "\n" "Link Options:\n" " --dynamic-linker [path] set the path to ld.so\n" " --each-lib-rpath add rpath for each used dynamic library\n" " --library [lib] link against lib\n" " --forbid-library [lib] make it an error to link against lib\n" " --library-path [dir] add a directory to the library search path\n" " --linker-script [path] use a custom linker script\n" " --object [obj] add object file to build\n" " -L[dir] alias for --library-path\n" " -rdynamic add all symbols to the dynamic symbol table\n" " -rpath [path] add directory to the runtime library search path\n" " --subsystem [subsystem] (windows) /SUBSYSTEM: to the linker\n" " -framework [name] (darwin) link against framework\n" " -mios-version-min [ver] (darwin) set iOS deployment target\n" " -mmacosx-version-min [ver] (darwin) set Mac OS X deployment target\n" " --ver-major [ver] dynamic library semver major version\n" " --ver-minor [ver] dynamic library semver minor version\n" " --ver-patch [ver] dynamic library semver patch version\n" "\n" "Test Options:\n" " --test-filter [text] skip tests that do not match filter\n" " --test-name-prefix [text] add prefix to all tests\n" " --test-cmd [arg] specify test execution command one arg at a time\n" " --test-cmd-bin appends test binary path to test cmd args\n" , arg0); return return_code; } static int print_libc_usage(const char *arg0, FILE *file, int return_code) { fprintf(file, "Usage: %s libc\n" "\n" "Detect the native libc installation and print the resulting paths to stdout.\n" "You can save this into a file and then edit the paths to create a cross\n" "compilation libc kit. Then you can pass `--libc [file]` for Zig to use it.\n" "\n" "When compiling natively and no `--libc` argument provided, Zig automatically\n" "creates zig-cache/native_libc.txt so that it does not have to detect libc\n" "on every invocation. You can remove this file to have Zig re-detect the\n" "native libc.\n" "\n\n" "Usage: %s libc [file]\n" "\n" "Parse a libc installation text file and validate it.\n" , arg0, arg0); return return_code; } static const char *ZIG_ZEN = "\n" " * Communicate intent precisely.\n" " * Edge cases matter.\n" " * Favor reading code over writing code.\n" " * Only one obvious way to do things.\n" " * Runtime crashes are better than bugs.\n" " * Compile errors are better than runtime crashes.\n" " * Incremental improvements.\n" " * Avoid local maximums.\n" " * Reduce the amount one must remember.\n" " * Minimize energy spent on coding style.\n" " * Together we serve end users.\n"; static int print_target_list(FILE *f) { ZigTarget native; get_native_target(&native); fprintf(f, "Architectures:\n"); size_t arch_count = target_arch_count(); for (size_t arch_i = 0; arch_i < arch_count; arch_i += 1) { const ArchType *arch = get_target_arch(arch_i); char arch_name[50]; get_arch_name(arch_name, arch); const char *native_str = (native.arch.arch == arch->arch && native.arch.sub_arch == arch->sub_arch) ? " (native)" : ""; fprintf(f, " %s%s\n", arch_name, native_str); } fprintf(f, "\nOperating Systems:\n"); size_t os_count = target_os_count(); for (size_t i = 0; i < os_count; i += 1) { Os os_type = get_target_os(i); const char *native_str = (native.os == os_type) ? " (native)" : ""; fprintf(f, " %s%s\n", get_target_os_name(os_type), native_str); } fprintf(f, "\nEnvironments:\n"); size_t environ_count = target_environ_count(); for (size_t i = 0; i < environ_count; i += 1) { ZigLLVM_EnvironmentType environ_type = get_target_environ(i); const char *native_str = (native.env_type == environ_type) ? " (native)" : ""; fprintf(f, " %s%s\n", ZigLLVMGetEnvironmentTypeName(environ_type), native_str); } return EXIT_SUCCESS; } enum Cmd { CmdNone, CmdBuild, CmdBuiltin, CmdRun, CmdTargets, CmdTest, CmdTranslateC, CmdVersion, CmdZen, CmdLibC, }; static const char *default_zig_cache_name = "zig-cache"; struct CliPkg { const char *name; const char *path; ZigList children; CliPkg *parent; }; static void add_package(CodeGen *g, CliPkg *cli_pkg, PackageTableEntry *pkg) { for (size_t i = 0; i < cli_pkg->children.length; i += 1) { CliPkg *child_cli_pkg = cli_pkg->children.at(i); Buf *dirname = buf_alloc(); Buf *basename = buf_alloc(); os_path_split(buf_create_from_str(child_cli_pkg->path), dirname, basename); PackageTableEntry *child_pkg = codegen_create_package(g, buf_ptr(dirname), buf_ptr(basename)); auto entry = pkg->package_table.put_unique(buf_create_from_str(child_cli_pkg->name), child_pkg); if (entry) { PackageTableEntry *existing_pkg = entry->value; Buf *full_path = buf_alloc(); os_path_join(&existing_pkg->root_src_dir, &existing_pkg->root_src_path, full_path); fprintf(stderr, "Unable to add package '%s'->'%s': already exists as '%s'\n", child_cli_pkg->name, child_cli_pkg->path, buf_ptr(full_path)); exit(EXIT_FAILURE); } add_package(g, child_cli_pkg, child_pkg); } } enum CacheOpt { CacheOptAuto, CacheOptOn, CacheOptOff, }; static bool get_cache_opt(CacheOpt opt, bool default_value) { switch (opt) { case CacheOptAuto: return default_value; case CacheOptOn: return true; case CacheOptOff: return false; } zig_unreachable(); } extern "C" int ZigClang_main(int argc, char **argv); int main(int argc, char **argv) { char *arg0 = argv[0]; Error err; if (argc == 2 && strcmp(argv[1], "BUILD_INFO") == 0) { printf("%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n", ZIG_CMAKE_BINARY_DIR, ZIG_CXX_COMPILER, ZIG_LLVM_CONFIG_EXE, ZIG_LLD_INCLUDE_PATH, ZIG_LLD_LIBRARIES, ZIG_STD_FILES, ZIG_C_HEADER_FILES, ZIG_DIA_GUIDS_LIB); return 0; } if (argc >= 2 && (strcmp(argv[1], "cc") == 0 || strcmp(argv[1], "-cc1") == 0 || strcmp(argv[1], "-cc1as") == 0)) { return ZigClang_main(argc, argv); } // Must be before all os.hpp function calls. os_init(); if (argc == 2 && strcmp(argv[1], "id") == 0) { Buf *compiler_id; if ((err = get_compiler_id(&compiler_id))) { fprintf(stderr, "Unable to determine compiler id: %s\n", err_str(err)); return EXIT_FAILURE; } printf("%s\n", buf_ptr(compiler_id)); return EXIT_SUCCESS; } enum InitKind { InitKindNone, InitKindExe, InitKindLib, }; InitKind init_kind = InitKindNone; if (argc >= 2) { const char *init_cmd = argv[1]; if (strcmp(init_cmd, "init-exe") == 0) { init_kind = InitKindExe; } else if (strcmp(init_cmd, "init-lib") == 0) { init_kind = InitKindLib; } if (init_kind != InitKindNone) { if (argc >= 3) { fprintf(stderr, "Unexpected extra argument: %s\n", argv[2]); return print_error_usage(arg0); } Buf *cmd_template_path = buf_alloc(); os_path_join(get_zig_special_dir(), buf_create_from_str(init_cmd), cmd_template_path); Buf *build_zig_path = buf_alloc(); os_path_join(cmd_template_path, buf_create_from_str("build.zig"), build_zig_path); Buf *src_dir_path = buf_alloc(); os_path_join(cmd_template_path, buf_create_from_str("src"), src_dir_path); Buf *main_zig_path = buf_alloc(); os_path_join(src_dir_path, buf_create_from_str("main.zig"), main_zig_path); Buf *cwd = buf_alloc(); if ((err = os_get_cwd(cwd))) { fprintf(stderr, "Unable to get cwd: %s\n", err_str(err)); return EXIT_FAILURE; } Buf *cwd_basename = buf_alloc(); os_path_split(cwd, nullptr, cwd_basename); Buf *build_zig_contents = buf_alloc(); if ((err = os_fetch_file_path(build_zig_path, build_zig_contents, false))) { fprintf(stderr, "Unable to read %s: %s\n", buf_ptr(build_zig_path), err_str(err)); return EXIT_FAILURE; } Buf *modified_build_zig_contents = buf_alloc(); for (size_t i = 0; i < buf_len(build_zig_contents); i += 1) { char c = buf_ptr(build_zig_contents)[i]; if (c == '$') { buf_append_buf(modified_build_zig_contents, cwd_basename); } else { buf_append_char(modified_build_zig_contents, c); } } Buf *main_zig_contents = buf_alloc(); if ((err = os_fetch_file_path(main_zig_path, main_zig_contents, false))) { fprintf(stderr, "Unable to read %s: %s\n", buf_ptr(main_zig_path), err_str(err)); return EXIT_FAILURE; } Buf *out_build_zig_path = buf_create_from_str("build.zig"); Buf *out_src_dir_path = buf_create_from_str("src"); Buf *out_main_zig_path = buf_alloc(); os_path_join(out_src_dir_path, buf_create_from_str("main.zig"), out_main_zig_path); bool already_exists; if ((err = os_file_exists(out_build_zig_path, &already_exists))) { fprintf(stderr, "Unable test existence of %s: %s\n", buf_ptr(out_build_zig_path), err_str(err)); return EXIT_FAILURE; } if (already_exists) { fprintf(stderr, "This file would be overwritten: %s\n", buf_ptr(out_build_zig_path)); return EXIT_FAILURE; } if ((err = os_make_dir(out_src_dir_path))) { fprintf(stderr, "Unable to make directory: %s: %s\n", buf_ptr(out_src_dir_path), err_str(err)); return EXIT_FAILURE; } os_write_file(out_build_zig_path, modified_build_zig_contents); os_write_file(out_main_zig_path, main_zig_contents); fprintf(stderr, "Created %s\n", buf_ptr(out_build_zig_path)); fprintf(stderr, "Created %s\n", buf_ptr(out_main_zig_path)); if (init_kind == InitKindExe) { fprintf(stderr, "\nNext, try `zig build --help` or `zig build run`\n"); } else if (init_kind == InitKindLib) { fprintf(stderr, "\nNext, try `zig build --help` or `zig build test`\n"); } else { zig_unreachable(); } return EXIT_SUCCESS; } } Cmd cmd = CmdNone; EmitFileType emit_file_type = EmitFileTypeBinary; const char *in_file = nullptr; const char *out_file = nullptr; const char *out_file_h = nullptr; const char *out_file_lib = nullptr; bool strip = false; bool is_static = false; OutType out_type = OutTypeUnknown; const char *out_name = nullptr; bool verbose_tokenize = false; bool verbose_ast = false; bool verbose_link = false; bool verbose_ir = false; bool verbose_llvm_ir = false; bool verbose_cimport = false; bool verbose_cc = false; ErrColor color = ErrColorAuto; CacheOpt enable_cache = CacheOptAuto; const char *libc_txt = nullptr; ZigList clang_argv = {0}; ZigList llvm_argv = {0}; ZigList lib_dirs = {0}; ZigList link_libs = {0}; ZigList forbidden_link_libs = {0}; ZigList frameworks = {0}; const char *target_arch = nullptr; const char *target_os = nullptr; const char *target_environ = nullptr; bool rdynamic = false; const char *mmacosx_version_min = nullptr; const char *mios_version_min = nullptr; const char *linker_script = nullptr; ZigList rpath_list = {0}; bool each_lib_rpath = false; ZigList objects = {0}; ZigList c_source_files = {0}; ZigList asm_files = {0}; const char *test_filter = nullptr; const char *test_name_prefix = nullptr; size_t ver_major = 0; size_t ver_minor = 0; size_t ver_patch = 0; bool timing_info = false; bool disable_pic = false; const char *cache_dir = nullptr; CliPkg *cur_pkg = allocate(1); BuildMode build_mode = BuildModeDebug; ZigList test_exec_args = {0}; int runtime_args_start = -1; bool system_linker_hack = false; TargetSubsystem subsystem = TargetSubsystemAuto; bool is_single_threaded = false; Buf *override_std_dir = nullptr; ValgrindSupport valgrind_support = ValgrindSupportAuto; if (argc >= 2 && strcmp(argv[1], "build") == 0) { Buf zig_exe_path_buf = BUF_INIT; if ((err = os_self_exe_path(&zig_exe_path_buf))) { fprintf(stderr, "Unable to determine path to zig's own executable\n"); return EXIT_FAILURE; } const char *zig_exe_path = buf_ptr(&zig_exe_path_buf); const char *build_file = "build.zig"; bool asked_for_help = false; init_all_targets(); ZigList args = {0}; args.append(zig_exe_path); args.append(NULL); // placeholder args.append(NULL); // placeholder for (int i = 2; i < argc; i += 1) { if (strcmp(argv[i], "--help") == 0) { asked_for_help = true; args.append(argv[i]); } else if (i + 1 < argc && strcmp(argv[i], "--build-file") == 0) { build_file = argv[i + 1]; i += 1; } else if (i + 1 < argc && strcmp(argv[i], "--cache-dir") == 0) { cache_dir = argv[i + 1]; i += 1; } else { args.append(argv[i]); } } Buf *build_runner_path = buf_alloc(); os_path_join(get_zig_special_dir(), buf_create_from_str("build_runner.zig"), build_runner_path); ZigTarget target; get_native_target(&target); CodeGen *g = codegen_create(build_runner_path, &target, OutTypeExe, BuildModeDebug, get_zig_lib_dir(), override_std_dir, nullptr); g->valgrind_support = valgrind_support; g->enable_time_report = timing_info; buf_init_from_str(&g->cache_dir, cache_dir ? cache_dir : default_zig_cache_name); codegen_set_out_name(g, buf_create_from_str("build")); Buf *build_file_buf = buf_create_from_str(build_file); Buf build_file_abs = os_path_resolve(&build_file_buf, 1); Buf build_file_basename = BUF_INIT; Buf build_file_dirname = BUF_INIT; os_path_split(&build_file_abs, &build_file_dirname, &build_file_basename); Buf full_cache_dir = BUF_INIT; if (cache_dir == nullptr) { os_path_join(&build_file_dirname, buf_create_from_str(default_zig_cache_name), &full_cache_dir); } else { Buf *cache_dir_buf = buf_create_from_str(cache_dir); full_cache_dir = os_path_resolve(&cache_dir_buf, 1); } args.items[1] = buf_ptr(&build_file_dirname); args.items[2] = buf_ptr(&full_cache_dir); bool build_file_exists; if ((err = os_file_exists(&build_file_abs, &build_file_exists))) { fprintf(stderr, "unable to open '%s': %s\n", buf_ptr(&build_file_abs), err_str(err)); return 1; } if (!build_file_exists) { if (asked_for_help) { // This usage text has to be synchronized with std/special/build_runner.zig fprintf(stdout, "Usage: %s build [options]\n" "\n" "General Options:\n" " --help Print this help and exit\n" " --verbose Print commands before executing them\n" " --prefix [path] Override default install prefix\n" " --search-prefix [path] Add a path to look for binaries, libraries, headers\n" "\n" "Project-specific options become available when the build file is found.\n" "\n" "Advanced Options:\n" " --build-file [file] Override path to build.zig\n" " --cache-dir [path] Override path to cache directory\n" " --verbose-tokenize Enable compiler debug output for tokenization\n" " --verbose-ast Enable compiler debug output for parsing into an AST\n" " --verbose-link Enable compiler debug output for linking\n" " --verbose-ir Enable compiler debug output for Zig IR\n" " --verbose-llvm-ir Enable compiler debug output for LLVM IR\n" " --verbose-cimport Enable compiler debug output for C imports\n" " --verbose-cc Enable compiler debug output for C compilation\n" "\n" , zig_exe_path); return EXIT_SUCCESS; } fprintf(stderr, "No 'build.zig' file found.\n" "Initialize a 'build.zig' template file with `zig init-lib` or `zig init-exe`,\n" "or build an executable directly with `zig build-exe $FILENAME.zig`.\n" "See: `zig build --help` or `zig --help` for more options.\n" ); return EXIT_FAILURE; } PackageTableEntry *build_pkg = codegen_create_package(g, buf_ptr(&build_file_dirname), buf_ptr(&build_file_basename)); g->root_package->package_table.put(buf_create_from_str("@build"), build_pkg); g->enable_cache = get_cache_opt(enable_cache, true); codegen_build_and_link(g); Termination term; os_spawn_process(buf_ptr(&g->output_file_path), args, &term); if (term.how != TerminationIdClean || term.code != 0) { fprintf(stderr, "\nBuild failed. The following command failed:\n"); fprintf(stderr, "%s", buf_ptr(&g->output_file_path)); for (size_t i = 0; i < args.length; i += 1) { fprintf(stderr, " %s", args.at(i)); } fprintf(stderr, "\n"); } return (term.how == TerminationIdClean) ? term.code : -1; } else if (argc >= 2 && strcmp(argv[1], "fmt") == 0) { init_all_targets(); ZigTarget target; get_native_target(&target); Buf *fmt_runner_path = buf_alloc(); os_path_join(get_zig_special_dir(), buf_create_from_str("fmt_runner.zig"), fmt_runner_path); CodeGen *g = codegen_create(fmt_runner_path, &target, OutTypeExe, BuildModeDebug, get_zig_lib_dir(), nullptr, nullptr); g->valgrind_support = valgrind_support; g->is_single_threaded = true; codegen_set_out_name(g, buf_create_from_str("fmt")); g->enable_cache = true; codegen_build_and_link(g); // TODO standardize os.cpp so that the args are supposed to have the exe ZigList args_with_exe = {0}; ZigList args_without_exe = {0}; const char *exec_path = buf_ptr(&g->output_file_path); args_with_exe.append(exec_path); for (int i = 2; i < argc; i += 1) { args_with_exe.append(argv[i]); args_without_exe.append(argv[i]); } args_with_exe.append(nullptr); os_execv(exec_path, args_with_exe.items); Termination term; os_spawn_process(exec_path, args_without_exe, &term); return term.code; } for (int i = 1; i < argc; i += 1) { char *arg = argv[i]; if (arg[0] == '-') { if (strcmp(arg, "--release-fast") == 0) { build_mode = BuildModeFastRelease; } else if (strcmp(arg, "--release-safe") == 0) { build_mode = BuildModeSafeRelease; } else if (strcmp(arg, "--release-small") == 0) { build_mode = BuildModeSmallRelease; } else if (strcmp(arg, "--help") == 0) { if (cmd == CmdLibC) { return print_libc_usage(arg0, stdout, EXIT_SUCCESS); } else { return print_full_usage(arg0, stdout, EXIT_SUCCESS); } } else if (strcmp(arg, "--strip") == 0) { strip = true; } else if (strcmp(arg, "--static") == 0) { is_static = true; } else if (strcmp(arg, "--verbose-tokenize") == 0) { verbose_tokenize = true; } else if (strcmp(arg, "--verbose-ast") == 0) { verbose_ast = true; } else if (strcmp(arg, "--verbose-link") == 0) { verbose_link = true; } else if (strcmp(arg, "--verbose-ir") == 0) { verbose_ir = true; } else if (strcmp(arg, "--verbose-llvm-ir") == 0) { verbose_llvm_ir = true; } else if (strcmp(arg, "--verbose-cimport") == 0) { verbose_cimport = true; } else if (strcmp(arg, "--verbose-cc") == 0) { verbose_cc = true; } else if (strcmp(arg, "-rdynamic") == 0) { rdynamic = true; } else if (strcmp(arg, "--each-lib-rpath") == 0) { each_lib_rpath = true; } else if (strcmp(arg, "-ftime-report") == 0) { timing_info = true; } else if (strcmp(arg, "--disable-pic") == 0) { disable_pic = true; } else if (strcmp(arg, "--enable-valgrind") == 0) { valgrind_support = ValgrindSupportEnabled; } else if (strcmp(arg, "--disable-valgrind") == 0) { valgrind_support = ValgrindSupportDisabled; } else if (strcmp(arg, "--system-linker-hack") == 0) { system_linker_hack = true; } else if (strcmp(arg, "--single-threaded") == 0) { is_single_threaded = true; } else if (strcmp(arg, "--test-cmd-bin") == 0) { test_exec_args.append(nullptr); } else if (arg[1] == 'L' && arg[2] != 0) { // alias for --library-path lib_dirs.append(&arg[2]); } else if (strcmp(arg, "--pkg-begin") == 0) { if (i + 2 >= argc) { fprintf(stderr, "Expected 2 arguments after --pkg-begin\n"); return print_error_usage(arg0); } CliPkg *new_cur_pkg = allocate(1); i += 1; new_cur_pkg->name = argv[i]; i += 1; new_cur_pkg->path = argv[i]; new_cur_pkg->parent = cur_pkg; cur_pkg->children.append(new_cur_pkg); cur_pkg = new_cur_pkg; } else if (strcmp(arg, "--pkg-end") == 0) { if (cur_pkg->parent == nullptr) { fprintf(stderr, "Encountered --pkg-end with no matching --pkg-begin\n"); return EXIT_FAILURE; } cur_pkg = cur_pkg->parent; } else if (i + 1 >= argc) { fprintf(stderr, "Expected another argument after %s\n", arg); return print_error_usage(arg0); } else { i += 1; if (strcmp(arg, "--output") == 0) { out_file = argv[i]; } else if (strcmp(arg, "--output-h") == 0) { out_file_h = argv[i]; } else if (strcmp(arg, "--output-lib") == 0) { out_file_lib = argv[i]; } else if (strcmp(arg, "--color") == 0) { if (strcmp(argv[i], "auto") == 0) { color = ErrColorAuto; } else if (strcmp(argv[i], "on") == 0) { color = ErrColorOn; } else if (strcmp(argv[i], "off") == 0) { color = ErrColorOff; } else { fprintf(stderr, "--color options are 'auto', 'on', or 'off'\n"); return print_error_usage(arg0); } } else if (strcmp(arg, "--cache") == 0) { if (strcmp(argv[i], "auto") == 0) { enable_cache = CacheOptAuto; } else if (strcmp(argv[i], "on") == 0) { enable_cache = CacheOptOn; } else if (strcmp(argv[i], "off") == 0) { enable_cache = CacheOptOff; } else { fprintf(stderr, "--cache options are 'auto', 'on', or 'off'\n"); return print_error_usage(arg0); } } else if (strcmp(arg, "--emit") == 0) { if (strcmp(argv[i], "asm") == 0) { emit_file_type = EmitFileTypeAssembly; } else if (strcmp(argv[i], "bin") == 0) { emit_file_type = EmitFileTypeBinary; } else if (strcmp(argv[i], "llvm-ir") == 0) { emit_file_type = EmitFileTypeLLVMIr; } else { fprintf(stderr, "--emit options are 'asm', 'bin', or 'llvm-ir'\n"); return print_error_usage(arg0); } } else if (strcmp(arg, "--name") == 0) { out_name = argv[i]; } else if (strcmp(arg, "--libc") == 0) { libc_txt = argv[i]; } else if (strcmp(arg, "-isystem") == 0) { clang_argv.append("-isystem"); clang_argv.append(argv[i]); } else if (strcmp(arg, "-dirafter") == 0) { clang_argv.append("-dirafter"); clang_argv.append(argv[i]); } else if (strcmp(arg, "-mllvm") == 0) { clang_argv.append("-mllvm"); clang_argv.append(argv[i]); llvm_argv.append(argv[i]); } else if (strcmp(arg, "--override-std-dir") == 0) { override_std_dir = buf_create_from_str(argv[i]); } else if (strcmp(arg, "--library-path") == 0 || strcmp(arg, "-L") == 0) { lib_dirs.append(argv[i]); } else if (strcmp(arg, "--library") == 0) { link_libs.append(argv[i]); } else if (strcmp(arg, "--forbid-library") == 0) { forbidden_link_libs.append(argv[i]); } else if (strcmp(arg, "--object") == 0) { objects.append(argv[i]); } else if (strcmp(arg, "--c-source") == 0) { CFile *c_file = allocate(1); for (;;) { if (argv[i][0] == '-') { c_file->args.append(argv[i]); i += 1; continue; } else { c_file->source_path = argv[i]; c_source_files.append(c_file); break; } } } else if (strcmp(arg, "--assembly") == 0) { asm_files.append(argv[i]); } else if (strcmp(arg, "--cache-dir") == 0) { cache_dir = argv[i]; } else if (strcmp(arg, "--target-arch") == 0) { target_arch = argv[i]; } else if (strcmp(arg, "--target-os") == 0) { target_os = argv[i]; } else if (strcmp(arg, "--target-environ") == 0) { target_environ = argv[i]; } else if (strcmp(arg, "-mmacosx-version-min") == 0) { mmacosx_version_min = argv[i]; } else if (strcmp(arg, "-mios-version-min") == 0) { mios_version_min = argv[i]; } else if (strcmp(arg, "-framework") == 0) { frameworks.append(argv[i]); } else if (strcmp(arg, "--linker-script") == 0) { linker_script = argv[i]; } else if (strcmp(arg, "-rpath") == 0) { rpath_list.append(argv[i]); } else if (strcmp(arg, "--test-filter") == 0) { test_filter = argv[i]; } else if (strcmp(arg, "--test-name-prefix") == 0) { test_name_prefix = argv[i]; } else if (strcmp(arg, "--ver-major") == 0) { ver_major = atoi(argv[i]); } else if (strcmp(arg, "--ver-minor") == 0) { ver_minor = atoi(argv[i]); } else if (strcmp(arg, "--ver-patch") == 0) { ver_patch = atoi(argv[i]); } else if (strcmp(arg, "--test-cmd") == 0) { test_exec_args.append(argv[i]); } else if (strcmp(arg, "--subsystem") == 0) { if (strcmp(argv[i], "console") == 0) { subsystem = TargetSubsystemConsole; } else if (strcmp(argv[i], "windows") == 0) { subsystem = TargetSubsystemWindows; } else if (strcmp(argv[i], "posix") == 0) { subsystem = TargetSubsystemPosix; } else if (strcmp(argv[i], "native") == 0) { subsystem = TargetSubsystemNative; } else if (strcmp(argv[i], "efi_application") == 0) { subsystem = TargetSubsystemEfiApplication; } else if (strcmp(argv[i], "efi_boot_service_driver") == 0) { subsystem = TargetSubsystemEfiBootServiceDriver; } else if (strcmp(argv[i], "efi_rom") == 0) { subsystem = TargetSubsystemEfiRom; } else if (strcmp(argv[i], "efi_runtime_driver") == 0) { subsystem = TargetSubsystemEfiRuntimeDriver; } else { fprintf(stderr, "invalid: --subsystem %s\n" "Options are:\n" " console\n" " windows\n" " posix\n" " native\n" " efi_application\n" " efi_boot_service_driver\n" " efi_rom\n" " efi_runtime_driver\n" , argv[i]); return EXIT_FAILURE; } } else { fprintf(stderr, "Invalid argument: %s\n", arg); return print_error_usage(arg0); } } } else if (cmd == CmdNone) { if (strcmp(arg, "build-exe") == 0) { cmd = CmdBuild; out_type = OutTypeExe; } else if (strcmp(arg, "build-obj") == 0) { cmd = CmdBuild; out_type = OutTypeObj; } else if (strcmp(arg, "build-lib") == 0) { cmd = CmdBuild; out_type = OutTypeLib; } else if (strcmp(arg, "run") == 0) { cmd = CmdRun; out_type = OutTypeExe; } else if (strcmp(arg, "version") == 0) { cmd = CmdVersion; } else if (strcmp(arg, "zen") == 0) { cmd = CmdZen; } else if (strcmp(arg, "libc") == 0) { cmd = CmdLibC; } else if (strcmp(arg, "translate-c") == 0) { cmd = CmdTranslateC; } else if (strcmp(arg, "test") == 0) { cmd = CmdTest; out_type = OutTypeExe; } else if (strcmp(arg, "targets") == 0) { cmd = CmdTargets; } else if (strcmp(arg, "builtin") == 0) { cmd = CmdBuiltin; } else { fprintf(stderr, "Unrecognized command: %s\n", arg); return print_error_usage(arg0); } } else { switch (cmd) { case CmdBuild: case CmdRun: case CmdTranslateC: case CmdTest: case CmdLibC: if (!in_file) { in_file = arg; if (cmd == CmdRun) { runtime_args_start = i + 1; break; // rest of the args are for the program } } else { fprintf(stderr, "Unexpected extra parameter: %s\n", arg); return print_error_usage(arg0); } break; case CmdBuiltin: case CmdVersion: case CmdZen: case CmdTargets: fprintf(stderr, "Unexpected extra parameter: %s\n", arg); return print_error_usage(arg0); case CmdNone: zig_unreachable(); } } } if (cur_pkg->parent != nullptr) { fprintf(stderr, "Unmatched --pkg-begin\n"); return EXIT_FAILURE; } init_all_targets(); ZigTarget target; if (!target_arch && !target_os && !target_environ) { get_native_target(&target); } else { get_unknown_target(&target); if (target_arch) { if (parse_target_arch(target_arch, &target.arch)) { fprintf(stderr, "invalid --target-arch argument\n"); return print_error_usage(arg0); } } if (target_os) { if (parse_target_os(target_os, &target.os)) { fprintf(stderr, "invalid --target-os argument\n"); return print_error_usage(arg0); } } if (target_environ) { if (parse_target_environ(target_environ, &target.env_type)) { fprintf(stderr, "invalid --target-environ argument\n"); return print_error_usage(arg0); } } } switch (cmd) { case CmdLibC: { if (in_file) { ZigLibCInstallation libc; if ((err = zig_libc_parse(&libc, buf_create_from_str(in_file), &target, true))) return EXIT_FAILURE; return EXIT_SUCCESS; } ZigLibCInstallation libc; if ((err = zig_libc_find_native(&libc, true))) return EXIT_FAILURE; zig_libc_render(&libc, stdout); return EXIT_SUCCESS; } case CmdBuiltin: { CodeGen *g = codegen_create(nullptr, &target, out_type, build_mode, get_zig_lib_dir(), override_std_dir, nullptr); g->valgrind_support = valgrind_support; g->is_single_threaded = is_single_threaded; Buf *builtin_source = codegen_generate_builtin_source(g); if (fwrite(buf_ptr(builtin_source), 1, buf_len(builtin_source), stdout) != buf_len(builtin_source)) { fprintf(stderr, "unable to write to stdout: %s\n", strerror(ferror(stdout))); return EXIT_FAILURE; } return EXIT_SUCCESS; } case CmdRun: case CmdBuild: case CmdTranslateC: case CmdTest: { if (cmd == CmdBuild && !in_file && objects.length == 0 && asm_files.length == 0 && c_source_files.length == 0) { fprintf(stderr, "Expected at least one of these things:\n" " * Zig root source file argument\n" " * --object argument\n" " * --assembly argument\n" " * --c-source argument\n"); return print_error_usage(arg0); } else if ((cmd == CmdTranslateC || cmd == CmdTest || cmd == CmdRun) && !in_file) { fprintf(stderr, "Expected source file argument.\n"); return print_error_usage(arg0); } else if (cmd == CmdBuild && out_type == OutTypeObj) { if (objects.length != 0) { fprintf(stderr, "When building an object file, --object arguments are invalid.\n" "Consider building a static library instead.\n"); return print_error_usage(arg0); } size_t zig_root_src_count = in_file ? 1 : 0; if (zig_root_src_count + c_source_files.length > 1) { fprintf(stderr, "When building an object file, only one of these allowed:\n" " * Zig root source file argument\n" " * --c-source argument\n" "Consider building a static library instead.\n"); return print_error_usage(arg0); } if (c_source_files.length != 0 && asm_files.length != 0) { fprintf(stderr, "When building an object file, only one of these allowed:\n" " * --assembly argument\n" " * --c-source argument\n" "Consider building a static library instead.\n"); return print_error_usage(arg0); } } assert(cmd != CmdBuild || out_type != OutTypeUnknown); bool need_name = (cmd == CmdBuild || cmd == CmdTranslateC); if (cmd == CmdRun) { out_name = "run"; } Buf *in_file_buf = nullptr; Buf *buf_out_name = (cmd == CmdTest) ? buf_create_from_str("test") : (out_name == nullptr) ? nullptr : buf_create_from_str(out_name); if (in_file) { in_file_buf = buf_create_from_str(in_file); if (need_name && buf_out_name == nullptr) { Buf basename = BUF_INIT; os_path_split(in_file_buf, nullptr, &basename); buf_out_name = buf_alloc(); os_path_extname(&basename, buf_out_name, nullptr); } } if (need_name && buf_out_name == nullptr && c_source_files.length == 1) { Buf basename = BUF_INIT; os_path_split(buf_create_from_str(c_source_files.at(0)->source_path), nullptr, &basename); buf_out_name = buf_alloc(); os_path_extname(&basename, buf_out_name, nullptr); } if (need_name && buf_out_name == nullptr) { fprintf(stderr, "--name [name] not provided and unable to infer\n\n"); return print_error_usage(arg0); } Buf *zig_root_source_file = (cmd == CmdTranslateC) ? nullptr : in_file_buf; if (cmd == CmdRun && buf_out_name == nullptr) { buf_out_name = buf_create_from_str("run"); } ZigLibCInstallation *libc = nullptr; if (libc_txt != nullptr) { libc = allocate(1); if ((err = zig_libc_parse(libc, buf_create_from_str(libc_txt), &target, true))) { fprintf(stderr, "Unable to parse --libc text file: %s\n", err_str(err)); return EXIT_FAILURE; } } CodeGen *g = codegen_create(zig_root_source_file, &target, out_type, build_mode, get_zig_lib_dir(), override_std_dir, libc); g->valgrind_support = valgrind_support; g->subsystem = subsystem; if (disable_pic) { if (out_type != OutTypeLib || !is_static) { fprintf(stderr, "--disable-pic only applies to static libraries"); return EXIT_FAILURE; } g->disable_pic = true; } g->enable_time_report = timing_info; buf_init_from_str(&g->cache_dir, cache_dir ? cache_dir : default_zig_cache_name); codegen_set_out_name(g, buf_out_name); codegen_set_lib_version(g, ver_major, ver_minor, ver_patch); codegen_set_is_test(g, cmd == CmdTest); g->is_single_threaded = is_single_threaded; codegen_set_linker_script(g, linker_script); if (each_lib_rpath) codegen_set_each_lib_rpath(g, each_lib_rpath); codegen_set_clang_argv(g, clang_argv.items, clang_argv.length); codegen_set_llvm_argv(g, llvm_argv.items, llvm_argv.length); codegen_set_strip(g, strip); codegen_set_is_static(g, is_static); g->verbose_tokenize = verbose_tokenize; g->verbose_ast = verbose_ast; g->verbose_link = verbose_link; g->verbose_ir = verbose_ir; g->verbose_llvm_ir = verbose_llvm_ir; g->verbose_cimport = verbose_cimport; g->verbose_cc = verbose_cc; codegen_set_errmsg_color(g, color); g->system_linker_hack = system_linker_hack; for (size_t i = 0; i < lib_dirs.length; i += 1) { codegen_add_lib_dir(g, lib_dirs.at(i)); } for (size_t i = 0; i < link_libs.length; i += 1) { LinkLib *link_lib = codegen_add_link_lib(g, buf_create_from_str(link_libs.at(i))); link_lib->provided_explicitly = true; } for (size_t i = 0; i < forbidden_link_libs.length; i += 1) { Buf *forbidden_link_lib = buf_create_from_str(forbidden_link_libs.at(i)); codegen_add_forbidden_lib(g, forbidden_link_lib); } for (size_t i = 0; i < frameworks.length; i += 1) { codegen_add_framework(g, frameworks.at(i)); } for (size_t i = 0; i < rpath_list.length; i += 1) { codegen_add_rpath(g, rpath_list.at(i)); } codegen_set_rdynamic(g, rdynamic); if (mmacosx_version_min && mios_version_min) { fprintf(stderr, "-mmacosx-version-min and -mios-version-min options not allowed together\n"); return EXIT_FAILURE; } if (mmacosx_version_min) { codegen_set_mmacosx_version_min(g, buf_create_from_str(mmacosx_version_min)); } if (mios_version_min) { codegen_set_mios_version_min(g, buf_create_from_str(mios_version_min)); } if (test_filter) { codegen_set_test_filter(g, buf_create_from_str(test_filter)); } if (test_name_prefix) { codegen_set_test_name_prefix(g, buf_create_from_str(test_name_prefix)); } if (out_file) codegen_set_output_path(g, buf_create_from_str(out_file)); if (out_file_h != nullptr && (out_type == OutTypeObj || out_type == OutTypeLib)) codegen_set_output_h_path(g, buf_create_from_str(out_file_h)); if (out_file_lib != nullptr && out_type == OutTypeLib && !is_static) codegen_set_output_lib_path(g, buf_create_from_str(out_file_lib)); add_package(g, cur_pkg, g->root_package); if (cmd == CmdBuild || cmd == CmdRun || cmd == CmdTest) { g->c_source_files = c_source_files; for (size_t i = 0; i < objects.length; i += 1) { codegen_add_object(g, buf_create_from_str(objects.at(i))); } for (size_t i = 0; i < asm_files.length; i += 1) { codegen_add_assembly(g, buf_create_from_str(asm_files.at(i))); } } if (cmd == CmdBuild || cmd == CmdRun) { codegen_set_emit_file_type(g, emit_file_type); g->enable_cache = get_cache_opt(enable_cache, cmd == CmdRun); codegen_build_and_link(g); if (timing_info) codegen_print_timing_report(g, stdout); if (cmd == CmdRun) { ZigList args = {0}; for (int i = runtime_args_start; i < argc; ++i) { args.append(argv[i]); } const char *exec_path = buf_ptr(&g->output_file_path); args.append(nullptr); os_execv(exec_path, args.items); args.pop(); Termination term; os_spawn_process(exec_path, args, &term); return term.code; } else if (cmd == CmdBuild) { if (g->enable_cache) { printf("%s\n", buf_ptr(&g->output_file_path)); if (g->out_h_path != nullptr) { printf("%s\n", buf_ptr(g->out_h_path)); } } return EXIT_SUCCESS; } else { zig_unreachable(); } } else if (cmd == CmdTranslateC) { codegen_translate_c(g, in_file_buf); ast_render(g, stdout, g->root_import->root, 4); if (timing_info) codegen_print_timing_report(g, stdout); return EXIT_SUCCESS; } else if (cmd == CmdTest) { codegen_set_emit_file_type(g, emit_file_type); ZigTarget native; get_native_target(&native); g->enable_cache = get_cache_opt(enable_cache, false); codegen_build_and_link(g); if (timing_info) { codegen_print_timing_report(g, stdout); } Buf *test_exe_path_unresolved = &g->output_file_path; Buf *test_exe_path = buf_alloc(); *test_exe_path = os_path_resolve(&test_exe_path_unresolved, 1); for (size_t i = 0; i < test_exec_args.length; i += 1) { if (test_exec_args.items[i] == nullptr) { test_exec_args.items[i] = buf_ptr(test_exe_path); } } if (!target_can_exec(&native, &target)) { fprintf(stderr, "Created %s but skipping execution because it is non-native.\n", buf_ptr(test_exe_path)); return 0; } Termination term; if (test_exec_args.length > 0) { ZigList rest_args = {0}; for (size_t i = 1; i < test_exec_args.length; i += 1) { rest_args.append(test_exec_args.at(i)); } os_spawn_process(test_exec_args.items[0], rest_args, &term); } else { ZigList no_args = {0}; os_spawn_process(buf_ptr(test_exe_path), no_args, &term); } if (term.how != TerminationIdClean || term.code != 0) { fprintf(stderr, "\nTests failed. Use the following command to reproduce the failure:\n"); fprintf(stderr, "%s\n", buf_ptr(test_exe_path)); } return (term.how == TerminationIdClean) ? term.code : -1; } else { zig_unreachable(); } } case CmdVersion: printf("%s\n", ZIG_VERSION_STRING); return EXIT_SUCCESS; case CmdZen: printf("%s\n", ZIG_ZEN); return EXIT_SUCCESS; case CmdTargets: return print_target_list(stdout); case CmdNone: return print_full_usage(arg0, stderr, EXIT_FAILURE); } }