| ... | ... | @@ -19,6 +19,7 @@ |
| 19 | 19 | #include "target.hpp" |
| 20 | 20 | #include "util.hpp" |
| 21 | 21 | #include "zig_llvm.h" |
| 22 | #include "blake2.h" |
| 22 | 23 | |
| 23 | 24 | #include <stdio.h> |
| 24 | 25 | #include <errno.h> |
| ... | ... | @@ -183,10 +184,6 @@ CodeGen *codegen_create(Buf *root_src_path, const ZigTarget *target, OutType out |
| 183 | 184 | return g; |
| 184 | 185 | } |
| 185 | 186 | |
| 186 | | void codegen_destroy(CodeGen *codegen) { |
| 187 | | LLVMDisposeTargetMachine(codegen->target_machine); |
| 188 | | } |
| 189 | | |
| 190 | 187 | void codegen_set_output_h_path(CodeGen *g, Buf *h_path) { |
| 191 | 188 | g->out_h_path = h_path; |
| 192 | 189 | } |
| ... | ... | @@ -7112,23 +7109,30 @@ static void create_test_compile_var_and_add_test_runner(CodeGen *g) { |
| 7112 | 7109 | g->test_runner_import = add_special_code(g, g->test_runner_package, "test_runner.zig"); |
| 7113 | 7110 | } |
| 7114 | 7111 | |
| 7115 | | static void gen_root_source(CodeGen *g) { |
| 7112 | static Buf *get_resolved_root_src_path(CodeGen *g) { |
| 7113 | // TODO memoize |
| 7116 | 7114 | if (buf_len(&g->root_package->root_src_path) == 0) |
| 7117 | | return; |
| 7118 | | |
| 7119 | | codegen_add_time_event(g, "Semantic Analysis"); |
| 7115 | return nullptr; |
| 7120 | 7116 | |
| 7121 | | Buf *rel_full_path = buf_alloc(); |
| 7122 | | os_path_join(&g->root_package->root_src_dir, &g->root_package->root_src_path, rel_full_path); |
| 7117 | Buf rel_full_path = BUF_INIT; |
| 7118 | os_path_join(&g->root_package->root_src_dir, &g->root_package->root_src_path, &rel_full_path); |
| 7123 | 7119 | |
| 7124 | 7120 | Buf *resolved_path = buf_alloc(); |
| 7125 | | Buf *resolve_paths[] = {rel_full_path}; |
| 7121 | Buf *resolve_paths[] = {&rel_full_path}; |
| 7126 | 7122 | *resolved_path = os_path_resolve(resolve_paths, 1); |
| 7127 | 7123 | |
| 7124 | return resolved_path; |
| 7125 | } |
| 7126 | |
| 7127 | static void gen_root_source(CodeGen *g) { |
| 7128 | Buf *resolved_path = get_resolved_root_src_path(g); |
| 7129 | if (resolved_path == nullptr) |
| 7130 | return; |
| 7131 | |
| 7128 | 7132 | Buf *source_code = buf_alloc(); |
| 7129 | 7133 | int err; |
| 7130 | | if ((err = os_fetch_file_path(rel_full_path, source_code, true))) { |
| 7131 | | fprintf(stderr, "unable to open '%s': %s\n", buf_ptr(rel_full_path), err_str(err)); |
| 7134 | if ((err = os_fetch_file_path(resolved_path, source_code, true))) { |
| 7135 | fprintf(stderr, "unable to open '%s': %s\n", buf_ptr(resolved_path), err_str(err)); |
| 7132 | 7136 | exit(1); |
| 7133 | 7137 | } |
| 7134 | 7138 | |
| ... | ... | @@ -7671,10 +7675,155 @@ void codegen_add_time_event(CodeGen *g, const char *name) { |
| 7671 | 7675 | g->timing_events.append({os_get_time(), name}); |
| 7672 | 7676 | } |
| 7673 | 7677 | |
| 7678 | static void add_cache_str(blake2b_state *blake, const char *ptr) { |
| 7679 | assert(ptr != nullptr); |
| 7680 | // + 1 to include the null byte |
| 7681 | blake2b_update(blake, ptr, strlen(ptr) + 1); |
| 7682 | } |
| 7683 | |
| 7684 | static void add_cache_int(blake2b_state *blake, int x) { |
| 7685 | // + 1 to include the null byte |
| 7686 | uint8_t buf[sizeof(int) + 1]; |
| 7687 | memcpy(buf, &x, sizeof(int)); |
| 7688 | buf[sizeof(int)] = 0; |
| 7689 | blake2b_update(blake, buf, sizeof(int) + 1); |
| 7690 | } |
| 7691 | |
| 7692 | static void add_cache_buf(blake2b_state *blake, Buf *buf) { |
| 7693 | assert(buf != nullptr); |
| 7694 | // + 1 to include the null byte |
| 7695 | blake2b_update(blake, buf_ptr(buf), buf_len(buf) + 1); |
| 7696 | } |
| 7697 | |
| 7698 | static void add_cache_buf_opt(blake2b_state *blake, Buf *buf) { |
| 7699 | if (buf == nullptr) { |
| 7700 | add_cache_str(blake, ""); |
| 7701 | add_cache_str(blake, ""); |
| 7702 | } else { |
| 7703 | add_cache_buf(blake, buf); |
| 7704 | } |
| 7705 | } |
| 7706 | |
| 7707 | static void add_cache_list_of_link_lib(blake2b_state *blake, LinkLib **ptr, size_t len) { |
| 7708 | for (size_t i = 0; i < len; i += 1) { |
| 7709 | LinkLib *lib = ptr[i]; |
| 7710 | if (lib->provided_explicitly) { |
| 7711 | add_cache_buf(blake, lib->name); |
| 7712 | } |
| 7713 | } |
| 7714 | add_cache_str(blake, ""); |
| 7715 | } |
| 7716 | |
| 7717 | static void add_cache_list_of_buf(blake2b_state *blake, Buf **ptr, size_t len) { |
| 7718 | for (size_t i = 0; i < len; i += 1) { |
| 7719 | Buf *buf = ptr[i]; |
| 7720 | add_cache_buf(blake, buf); |
| 7721 | } |
| 7722 | add_cache_str(blake, ""); |
| 7723 | } |
| 7724 | |
| 7725 | //static void add_cache_file(CodeGen *g, blake2b_state *blake, Buf *resolved_path) { |
| 7726 | // assert(file_name != nullptr); |
| 7727 | // g->cache_files.append(resolved_path); |
| 7728 | //} |
| 7729 | // |
| 7730 | //static void add_cache_file_opt(CodeGen *g, blake2b_state *blake, Buf *resolved_path) { |
| 7731 | // if (resolved_path == nullptr) { |
| 7732 | // add_cache_str(blake, ""); |
| 7733 | // add_cache_str(blake, ""); |
| 7734 | // } else { |
| 7735 | // add_cache_file(g, blake, resolved_path); |
| 7736 | // } |
| 7737 | //} |
| 7738 | |
| 7739 | // Ported from std/base64.zig |
| 7740 | static uint8_t base64_fs_alphabet[] = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_"; |
| 7741 | static void base64_encode(Slice<uint8_t> dest, Slice<uint8_t> source) { |
| 7742 | size_t dest_len = ((source.len + 2) / 3) * 4; |
| 7743 | assert(dest.len == dest_len); |
| 7744 | |
| 7745 | size_t i = 0; |
| 7746 | size_t out_index = 0; |
| 7747 | for (; i + 2 < source.len; i += 3) { |
| 7748 | dest.ptr[out_index] = base64_fs_alphabet[(source.ptr[i] >> 2) & 0x3f]; |
| 7749 | out_index += 1; |
| 7750 | |
| 7751 | dest.ptr[out_index] = base64_fs_alphabet[((source.ptr[i] & 0x3) << 4) | ((source.ptr[i + 1] & 0xf0) >> 4)]; |
| 7752 | out_index += 1; |
| 7753 | |
| 7754 | dest.ptr[out_index] = base64_fs_alphabet[((source.ptr[i + 1] & 0xf) << 2) | ((source.ptr[i + 2] & 0xc0) >> 6)]; |
| 7755 | out_index += 1; |
| 7756 | |
| 7757 | dest.ptr[out_index] = base64_fs_alphabet[source.ptr[i + 2] & 0x3f]; |
| 7758 | out_index += 1; |
| 7759 | } |
| 7760 | |
| 7761 | // Assert that we never need pad characters. |
| 7762 | assert(i == source.len); |
| 7763 | //if (i < source.len) { |
| 7764 | // dest.ptr[out_index] = base64_fs_alphabet[(source.ptr[i] >> 2) & 0x3f]; |
| 7765 | // out_index += 1; |
| 7766 | |
| 7767 | // if (i + 1 == source.len) { |
| 7768 | // dest.ptr[out_index] = base64_fs_alphabet[(source.ptr[i] & 0x3) << 4]; |
| 7769 | // out_index += 1; |
| 7770 | |
| 7771 | // dest.ptr[out_index] = encoder.pad_char; |
| 7772 | // out_index += 1; |
| 7773 | // } else { |
| 7774 | // dest.ptr[out_index] = base64_fs_alphabet[((source.ptr[i] & 0x3) << 4) | ((source.ptr[i + 1] & 0xf0) >> 4)]; |
| 7775 | // out_index += 1; |
| 7776 | |
| 7777 | // dest.ptr[out_index] = base64_fs_alphabet[(source.ptr[i + 1] & 0xf) << 2]; |
| 7778 | // out_index += 1; |
| 7779 | // } |
| 7780 | |
| 7781 | // dest.ptr[out_index] = encoder.pad_char; |
| 7782 | // out_index += 1; |
| 7783 | //} |
| 7784 | } |
| 7785 | |
| 7786 | // Called before init() |
| 7787 | static bool build_with_cache(CodeGen *g) { |
| 7788 | blake2b_state blake; |
| 7789 | int rc = blake2b_init(&blake, 48); |
| 7790 | assert(rc == 0); |
| 7791 | |
| 7792 | // TODO zig exe & dynamic libraries |
| 7793 | |
| 7794 | add_cache_buf(&blake, g->root_out_name); |
| 7795 | add_cache_buf_opt(&blake, get_resolved_root_src_path(g)); // Root source file |
| 7796 | add_cache_list_of_link_lib(&blake, g->link_libs_list.items, g->link_libs_list.length); |
| 7797 | add_cache_list_of_buf(&blake, g->darwin_frameworks.items, g->darwin_frameworks.length); |
| 7798 | add_cache_list_of_buf(&blake, g->rpath_list.items, g->rpath_list.length); |
| 7799 | add_cache_int(&blake, g->emit_file_type); |
| 7800 | add_cache_int(&blake, g->build_mode); |
| 7801 | add_cache_int(&blake, g->out_type); |
| 7802 | // TODO the rest of the struct CodeGen fields |
| 7803 | |
| 7804 | uint8_t bin_digest[48]; |
| 7805 | rc = blake2b_final(&blake, bin_digest, 48); |
| 7806 | assert(rc == 0); |
| 7807 | |
| 7808 | Buf b64_digest = BUF_INIT; |
| 7809 | buf_resize(&b64_digest, 64); |
| 7810 | base64_encode(buf_to_slice(&b64_digest), {bin_digest, 48}); |
| 7811 | |
| 7812 | fprintf(stderr, "input params hash: %s\n", buf_ptr(&b64_digest)); |
| 7813 | // TODO next look for a manifest file that has all the files from the input parameters |
| 7814 | // use that to construct the real hash, which looks up the output directory and another manifest file |
| 7815 | |
| 7816 | return false; |
| 7817 | } |
| 7818 | |
| 7674 | 7819 | void codegen_build(CodeGen *g) { |
| 7675 | 7820 | assert(g->out_type != OutTypeUnknown); |
| 7821 | if (build_with_cache(g)) |
| 7822 | return; |
| 7676 | 7823 | init(g); |
| 7677 | 7824 | |
| 7825 | codegen_add_time_event(g, "Semantic Analysis"); |
| 7826 | |
| 7678 | 7827 | gen_global_asm(g); |
| 7679 | 7828 | gen_root_source(g); |
| 7680 | 7829 | do_code_gen(g); |