struct ssaFileBuffer { gbVirtualMemory vm; isize offset; gbFile *output; }; void ssa_file_buffer_init(ssaFileBuffer *f, gbFile *output) { isize size = 8*gb_virtual_memory_page_size(NULL); f->vm = gb_vm_alloc(NULL, size); f->offset = 0; f->output = output; } void ssa_file_buffer_destroy(ssaFileBuffer *f) { if (f->offset > 0) { // NOTE(bill): finish writing buffered data gb_file_write(f->output, f->vm.data, f->offset); } gb_vm_free(f->vm); } void ssa_file_buffer_write(ssaFileBuffer *f, void *data, isize len) { if (len > f->vm.size) { gb_file_write(f->output, data, len); return; } if ((f->vm.size - f->offset) < len) { gb_file_write(f->output, f->vm.data, f->offset); f->offset = 0; } u8 *cursor = cast(u8 *)f->vm.data + f->offset; gb_memcopy(cursor, data, len); f->offset += len; } void ssa_fprintf(ssaFileBuffer *f, char *fmt, ...) { va_list va; va_start(va, fmt); char buf[4096] = {}; isize len = gb_snprintf_va(buf, gb_size_of(buf), fmt, va); ssa_file_buffer_write(f, buf, len-1); va_end(va); } void ssa_file_write(ssaFileBuffer *f, void *data, isize len) { ssa_file_buffer_write(f, data, len); } b32 ssa_valid_char(u8 c) { if (c >= 0x80) return false; if (gb_char_is_alphanumeric(c)) return true; switch (c) { case '$': case '-': case '.': case '_': return true; } return false; } void ssa_print_escape_string(ssaFileBuffer *f, String name, b32 print_quotes) { isize extra = 0; for (isize i = 0; i < name.len; i++) { u8 c = name.text[i]; if (!ssa_valid_char(c)) extra += 2; } if (extra == 0) { ssa_fprintf(f, "%.*s", LIT(name)); return; } char hex_table[] = "0123456789ABCDEF"; isize buf_len = name.len + extra + 2; gbTempArenaMemory tmp = gb_temp_arena_memory_begin(&string_buffer_arena); defer (gb_temp_arena_memory_end(tmp)); u8 *buf = gb_alloc_array(string_buffer_allocator, u8, buf_len); isize j = 0; if (print_quotes) { buf[j++] = '"'; } for (isize i = 0; i < name.len; i++) { u8 c = name.text[i]; if (ssa_valid_char(c)) { buf[j++] = c; } else { buf[j] = '\\'; buf[j+1] = hex_table[c >> 4]; buf[j+2] = hex_table[c & 0x0f]; j += 3; } } if (print_quotes) { buf[j++] = '"'; } ssa_file_write(f, buf, j); } void ssa_print_encoded_local(ssaFileBuffer *f, String name) { ssa_fprintf(f, "%%"); ssa_print_escape_string(f, name, true); } void ssa_print_encoded_global(ssaFileBuffer *f, String name, b32 global_scope) { ssa_fprintf(f, "@"); if (!global_scope && name != make_string("main")) { ssa_fprintf(f, "."); } ssa_print_escape_string(f, name, true); } void ssa_print_type(ssaFileBuffer *f, ssaModule *m, Type *t) { BaseTypeSizes s = m->sizes; i64 word_bits = 8*s.word_size; GB_ASSERT_NOT_NULL(t); t = default_type(t); switch (t->kind) { case Type_Basic: switch (t->Basic.kind) { case Basic_bool: ssa_fprintf(f, "i1"); break; case Basic_i8: ssa_fprintf(f, "i8"); break; case Basic_i16: ssa_fprintf(f, "i16"); break; case Basic_i32: ssa_fprintf(f, "i32"); break; case Basic_i64: ssa_fprintf(f, "i64"); break; // case Basic_i128: ssa_fprintf(f, "i128"); break; case Basic_u8: ssa_fprintf(f, "i8"); break; case Basic_u16: ssa_fprintf(f, "i16"); break; case Basic_u32: ssa_fprintf(f, "i32"); break; case Basic_u64: ssa_fprintf(f, "i64"); break; // case Basic_u128: ssa_fprintf(f, "i128"); break; case Basic_f32: ssa_fprintf(f, "float"); break; case Basic_f64: ssa_fprintf(f, "double"); break; case Basic_rawptr: ssa_fprintf(f, "%%..rawptr"); break; case Basic_string: ssa_fprintf(f, "%%..string"); break; case Basic_uint: ssa_fprintf(f, "i%lld", word_bits); break; case Basic_int: ssa_fprintf(f, "i%lld", word_bits); break; case Basic_any: ssa_fprintf(f, "%%..any"); break; } break; case Type_Pointer: ssa_print_type(f, m, t->Pointer.elem); ssa_fprintf(f, "*"); break; case Type_Maybe: ssa_fprintf(f, "{"); ssa_print_type(f, m, t->Maybe.elem); ssa_fprintf(f, ", "); ssa_print_type(f, m, t_bool); ssa_fprintf(f, "}"); break; case Type_Array: ssa_fprintf(f, "[%lld x ", t->Array.count); ssa_print_type(f, m, t->Array.elem); ssa_fprintf(f, "]"); break; case Type_Vector: { // TODO(bill): actually do correctly ssa_fprintf(f, "<%lld x ", t->Vector.count); ssa_print_type(f, m, t->Vector.elem); ssa_fprintf(f, ">"); } break; case Type_Slice: ssa_fprintf(f, "{"); ssa_print_type(f, m, t->Slice.elem); ssa_fprintf(f, "*, i%lld, i%lld}", word_bits, word_bits); break; case Type_Record: { switch (t->Record.kind) { case TypeRecord_Struct: if (t->Record.struct_is_packed) { ssa_fprintf(f, "<"); } ssa_fprintf(f, "{"); for (isize i = 0; i < t->Record.field_count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } Type *ft = t->Record.fields[i]->type; Type *bft = base_type(ft); if (!is_type_struct(bft)) { ft = bft; } ssa_print_type(f, m, ft); } ssa_fprintf(f, "}"); if (t->Record.struct_is_packed) { ssa_fprintf(f, ">"); } break; case TypeRecord_Union: { i64 size_of_union = type_size_of(s, gb_heap_allocator(), t) - s.word_size; ssa_fprintf(f, "{[%lld x i8], i%lld}", size_of_union, word_bits); } break; case TypeRecord_RawUnion: ssa_fprintf(f, "[%lld x i8]", type_size_of(s, gb_heap_allocator(), t)); break; case TypeRecord_Enum: ssa_print_type(f, m, t->Record.enum_base); break; } } break; case Type_Named: if (is_type_struct(t) || is_type_union(t)) { String *name = map_get(&m->type_names, hash_pointer(t)); GB_ASSERT(name != NULL); ssa_print_encoded_local(f, *name); // ssa_print_encoded_local(f, t->Named.name); } else { ssa_print_type(f, m, base_type(t)); } break; case Type_Tuple: if (t->Tuple.variable_count == 1) { ssa_print_type(f, m, t->Tuple.variables[0]->type); } else { ssa_fprintf(f, "{"); for (isize i = 0; i < t->Tuple.variable_count; i++) { if (i > 0) ssa_fprintf(f, ", "); ssa_print_type(f, m, t->Tuple.variables[i]->type); } ssa_fprintf(f, "}"); } break; case Type_Proc: { if (t->Proc.result_count == 0) { ssa_fprintf(f, "void"); } else { ssa_print_type(f, m, t->Proc.results); } ssa_fprintf(f, " ("); auto *params = &t->Proc.params->Tuple; for (isize i = 0; i < t->Proc.param_count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } ssa_print_type(f, m, params->variables[i]->type); } ssa_fprintf(f, ")*"); } break; } } void ssa_print_exact_value(ssaFileBuffer *f, ssaModule *m, ExactValue value, Type *type); void ssa_print_compound_element(ssaFileBuffer *f, ssaModule *m, ExactValue v, Type *elem_type) { ssa_print_type(f, m, elem_type); ssa_fprintf(f, " "); if (v.kind != ExactValue_Invalid && is_type_maybe(elem_type)) { Type *t = base_type(elem_type)->Maybe.elem; ssa_fprintf(f, "{"); ssa_print_type(f, m, t); ssa_fprintf(f, " "); } if (v.kind == ExactValue_Invalid) { ssa_fprintf(f, "zeroinitializer"); } else if (v.kind == ExactValue_String) { // HACK NOTE(bill): This is a hack but it works because strings are created at the very end // of the .ll file ssaValue *str_array = ssa_add_global_string_array(m, v.value_string); ssa_fprintf(f, "{i8* getelementptr inbounds ("); ssa_print_type(f, m, str_array->Global.entity->type); ssa_fprintf(f, ", "); ssa_print_type(f, m, str_array->Global.entity->type); ssa_fprintf(f, "* "); ssa_print_encoded_global(f, str_array->Global.entity->token.string, false); ssa_fprintf(f, ", "); ssa_print_type(f, m, t_int); ssa_fprintf(f, " 0, i32 0), "); ssa_print_type(f, m, t_int); ssa_fprintf(f, " %lld}", cast(i64)v.value_string.len); } else { ssa_print_exact_value(f, m, v, elem_type); } if (v.kind != ExactValue_Invalid && is_type_maybe(elem_type)) { ssa_fprintf(f, ", "); ssa_print_type(f, m, t_bool); ssa_fprintf(f, " "); ssa_fprintf(f, "true}"); } } void ssa_print_exact_value(ssaFileBuffer *f, ssaModule *m, ExactValue value, Type *type) { type = base_type(type); if (is_type_float(type)) { value = exact_value_to_float(value); } else if (is_type_integer(type)) { value = exact_value_to_integer(value); } else if (is_type_pointer(type)) { value = exact_value_to_integer(value); } switch (value.kind) { case ExactValue_Bool: ssa_fprintf(f, "%s", (value.value_bool ? "true" : "false")); break; case ExactValue_String: { ssa_fprintf(f, "c\""); ssa_print_escape_string(f, value.value_string, false); ssa_fprintf(f, "\""); } break; case ExactValue_Integer: { if (is_type_pointer(type)) { if (value.value_integer == 0) { ssa_fprintf(f, "null"); } else { ssa_fprintf(f, "inttoptr ("); ssa_print_type(f, m, t_int); ssa_fprintf(f, " %llu to ", value.value_integer); ssa_print_type(f, m, t_rawptr); ssa_fprintf(f, ")"); } } else { ssa_fprintf(f, "%lld", value.value_integer); } } break; case ExactValue_Float: { u64 u = *cast(u64*)&value.value_float; if (is_type_float(type) && type->Basic.kind == Basic_f32) { // IMPORTANT NOTE(bill): LLVM requires all floating point constants to be // a 64 bit number if bits_of(float type) <= 64. // For some bizarre reason, you need to clear the bottom 29 bits // https://groups.google.com/forum/#!topic/llvm-dev/IlqV3TbSk6M u >>= 29; u <<= 29; } ssa_fprintf(f, "0x%016llx", u); } break; case ExactValue_Pointer: if (value.value_pointer == NULL) { ssa_fprintf(f, "null"); } else { ssa_fprintf(f, "inttoptr ("); ssa_print_type(f, m, t_int); ssa_fprintf(f, " %llu to ", cast(u64)cast(uintptr)value.value_pointer); ssa_print_type(f, m, t_rawptr); ssa_fprintf(f, ")"); } break; case ExactValue_Compound: { type = base_type(type); if (is_type_array(type)) { ast_node(cl, CompoundLit, value.value_compound); isize elem_count = cl->elems.count; if (elem_count == 0) { ssa_fprintf(f, "zeroinitializer"); break; } ssa_fprintf(f, "["); Type *elem_type = type->Array.elem; for (isize i = 0; i < elem_count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } TypeAndValue *tav = type_and_value_of_expression(m->info, cl->elems[i]); GB_ASSERT(tav != NULL); ssa_print_compound_element(f, m, tav->value, elem_type); } for (isize i = elem_count; i < type->Array.count; i++) { if (i >= elem_count) { ssa_fprintf(f, ", "); } ssa_print_type(f, m, elem_type); ssa_fprintf(f, " zeroinitializer"); } ssa_fprintf(f, "]"); } else if (is_type_vector(type)) { ast_node(cl, CompoundLit, value.value_compound); isize elem_count = cl->elems.count; if (elem_count == 0) { ssa_fprintf(f, "zeroinitializer"); break; } ssa_fprintf(f, "<"); Type *elem_type = type->Vector.elem; if (elem_count == 1 && type->Vector.count > 1) { TypeAndValue *tav = type_and_value_of_expression(m->info, cl->elems[0]); GB_ASSERT(tav != NULL); for (isize i = 0; i < type->Vector.count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } ssa_print_compound_element(f, m, tav->value, elem_type); } } else { for (isize i = 0; i < elem_count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } TypeAndValue *tav = type_and_value_of_expression(m->info, cl->elems[i]); GB_ASSERT(tav != NULL); ssa_print_compound_element(f, m, tav->value, elem_type); } } ssa_fprintf(f, ">"); } else if (is_type_struct(type)) { gbTempArenaMemory tmp = gb_temp_arena_memory_begin(&m->tmp_arena); defer (gb_temp_arena_memory_end(tmp)); ast_node(cl, CompoundLit, value.value_compound); if (cl->elems.count == 0) { ssa_fprintf(f, "zeroinitializer"); break; } isize value_count = type->Record.field_count; ExactValue *values = gb_alloc_array(m->tmp_allocator, ExactValue, value_count); if (cl->elems[0]->kind == AstNode_FieldValue) { isize elem_count = cl->elems.count; for (isize i = 0; i < elem_count; i++) { ast_node(fv, FieldValue, cl->elems[i]); String name = fv->field->Ident.string; TypeAndValue *tav = type_and_value_of_expression(m->info, fv->value); GB_ASSERT(tav != NULL); Selection sel = lookup_field(m->allocator, type, name, false); Entity *f = type->Record.fields[sel.index[0]]; values[f->Variable.field_index] = tav->value; } } else { for (isize i = 0; i < value_count; i++) { TypeAndValue *tav = type_and_value_of_expression(m->info, cl->elems[i]); GB_ASSERT(tav != NULL); Entity *f = type->Record.fields_in_src_order[i]; values[f->Variable.field_index] = tav->value; } } if (type->Record.struct_is_packed) { ssa_fprintf(f, "<"); } ssa_fprintf(f, "{"); for (isize i = 0; i < value_count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } Type *elem_type = type->Record.fields[i]->type; ssa_print_compound_element(f, m, values[i], elem_type); } ssa_fprintf(f, "}"); if (type->Record.struct_is_packed) { ssa_fprintf(f, ">"); } } else { ssa_fprintf(f, "zeroinitializer"); } } break; default: ssa_fprintf(f, "zeroinitializer"); // GB_PANIC("Invalid ExactValue: %d", value.kind); break; } } void ssa_print_block_name(ssaFileBuffer *f, ssaBlock *b) { if (b != NULL) { ssa_print_escape_string(f, b->label, false); ssa_fprintf(f, "-%td", b->index); } else { ssa_fprintf(f, ""); } } void ssa_print_value(ssaFileBuffer *f, ssaModule *m, ssaValue *value, Type *type_hint) { if (value == NULL) { ssa_fprintf(f, "!!!NULL_VALUE"); return; } switch (value->kind) { default: GB_PANIC("Unknown ssaValue kind"); break; case ssaValue_Constant: ssa_print_exact_value(f, m, value->Constant.value, type_hint); break; case ssaValue_ConstantSlice: { auto *cs = &value->ConstantSlice; if (cs->backing_array == NULL || cs->count == 0) { ssa_fprintf(f, "zeroinitializer"); } else { Type *at = base_type(type_deref(ssa_type(cs->backing_array))); Type *et = at->Array.elem; ssa_fprintf(f, "{"); ssa_print_type(f, m, et); ssa_fprintf(f, "* getelementptr inbounds ("); ssa_print_type(f, m, at); ssa_fprintf(f, ", "); ssa_print_type(f, m, at); ssa_fprintf(f, "* "); ssa_print_value(f, m, cs->backing_array, at); ssa_fprintf(f, ", "); ssa_print_type(f, m, t_int); ssa_fprintf(f, " 0, i32 0), "); ssa_print_type(f, m, t_int); ssa_fprintf(f, " %lld, ", cs->count); ssa_print_type(f, m, t_int); ssa_fprintf(f, " %lld}", cs->count); } } break; case ssaValue_Nil: ssa_fprintf(f, "zeroinitializer"); break; case ssaValue_TypeName: ssa_print_encoded_local(f, value->TypeName.name); break; case ssaValue_Global: { Scope *scope = value->Global.entity->scope; b32 in_global_scope = false; if (scope != NULL) { in_global_scope = scope->is_global || scope->is_init; } ssa_print_encoded_global(f, value->Global.entity->token.string, in_global_scope); } break; case ssaValue_Param: ssa_print_encoded_local(f, value->Param.entity->token.string); break; case ssaValue_Proc: ssa_print_encoded_global(f, value->Proc.name, (value->Proc.tags & (ProcTag_foreign|ProcTag_link_name)) != 0); break; case ssaValue_Instr: ssa_fprintf(f, "%%%d", value->index); break; } } void ssa_print_instr(ssaFileBuffer *f, ssaModule *m, ssaValue *value) { GB_ASSERT(value->kind == ssaValue_Instr); ssaInstr *instr = &value->Instr; ssa_fprintf(f, "\t"); switch (instr->kind) { case ssaInstr_StartupRuntime: { ssa_fprintf(f, "call void "); ssa_print_encoded_global(f, make_string(SSA_STARTUP_RUNTIME_PROC_NAME), false); ssa_fprintf(f, "()\n"); } break; case ssaInstr_Comment: ssa_fprintf(f, "; %.*s\n", LIT(instr->Comment.text)); break; case ssaInstr_Local: { Type *type = instr->Local.entity->type; ssa_fprintf(f, "%%%d = alloca ", value->index); ssa_print_type(f, m, type); ssa_fprintf(f, ", align %lld\n", type_align_of(m->sizes, m->allocator, type)); } break; case ssaInstr_ZeroInit: { Type *type = type_deref(ssa_type(instr->ZeroInit.address)); ssa_fprintf(f, "store "); ssa_print_type(f, m, type); ssa_fprintf(f, " zeroinitializer, "); ssa_print_type(f, m, type); ssa_fprintf(f, "* %%%d\n", instr->ZeroInit.address->index); } break; case ssaInstr_Store: { Type *type = ssa_type(instr->Store.value); ssa_fprintf(f, "store "); ssa_print_type(f, m, type); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->Store.value, type); ssa_fprintf(f, ", "); ssa_print_type(f, m, type); ssa_fprintf(f, "* "); ssa_print_value(f, m, instr->Store.address, type); ssa_fprintf(f, "\n"); } break; case ssaInstr_Load: { Type *type = instr->Load.type; ssa_fprintf(f, "%%%d = load ", value->index); ssa_print_type(f, m, type); ssa_fprintf(f, ", "); ssa_print_type(f, m, type); ssa_fprintf(f, "* "); ssa_print_value(f, m, instr->Load.address, type); ssa_fprintf(f, ", align %lld\n", type_align_of(m->sizes, m->allocator, type)); } break; case ssaInstr_GetElementPtr: { Type *et = instr->GetElementPtr.elem_type; ssa_fprintf(f, "%%%d = getelementptr ", value->index); if (instr->GetElementPtr.inbounds) { ssa_fprintf(f, "inbounds "); } ssa_print_type(f, m, type_deref(et)); ssa_fprintf(f, ", "); ssa_print_type(f, m, et); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->GetElementPtr.address, et); for (isize i = 0; i < instr->GetElementPtr.index_count; i++) { ssaValue *index = instr->GetElementPtr.indices[i]; Type *t = ssa_type(index); ssa_fprintf(f, ", "); ssa_print_type(f, m, t); ssa_fprintf(f, " "); ssa_print_value(f, m, index, t); } ssa_fprintf(f, "\n"); } break; case ssaInstr_Phi: { ssa_fprintf(f, "%%%d = phi ", value->index); ssa_print_type(f, m, instr->Phi.type); ssa_fprintf(f, " ", value->index); for (isize i = 0; i < instr->Phi.edges.count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } ssaValue *edge = instr->Phi.edges[i]; ssaBlock *block = NULL; if (instr->parent != NULL && i < instr->parent->preds.count) { block = instr->parent->preds[i]; } ssa_fprintf(f, "[ "); ssa_print_value(f, m, edge, instr->Phi.type); ssa_fprintf(f, ", %%"); ssa_print_block_name(f, block); ssa_fprintf(f, " ]"); } ssa_fprintf(f, "\n"); } break; case ssaInstr_ExtractValue: { Type *et = instr->ExtractValue.elem_type; ssa_fprintf(f, "%%%d = extractvalue ", value->index); ssa_print_type(f, m, et); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->ExtractValue.address, et); ssa_fprintf(f, ", %d\n", instr->ExtractValue.index); } break; case ssaInstr_NoOp: {; ssa_fprintf(f, "%%%d = add i32 0, 0\n", value->index); } break; case ssaInstr_Br: {; ssa_fprintf(f, "br "); if (instr->Br.cond != NULL) { ssa_print_type(f, m, t_bool); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->Br.cond, t_bool); ssa_fprintf(f, ", ", instr->Br.cond->index); } ssa_fprintf(f, "label "); ssa_fprintf(f, "%%"); ssa_print_block_name(f, instr->Br.true_block); if (instr->Br.false_block != NULL) { ssa_fprintf(f, ", label "); ssa_fprintf(f, "%%"); ssa_print_block_name(f, instr->Br.false_block); } ssa_fprintf(f, "\n"); } break; case ssaInstr_Ret: { auto *ret = &instr->Ret; ssa_fprintf(f, "ret "); if (ret->value == NULL) { ssa_fprintf(f, "void"); } else { Type *t = ssa_type(ret->value); ssa_print_type(f, m, t); ssa_fprintf(f, " "); ssa_print_value(f, m, ret->value, t); } ssa_fprintf(f, "\n"); } break; case ssaInstr_Conv: { auto *c = &instr->Conv; ssa_fprintf(f, "%%%d = %.*s ", value->index, LIT(ssa_conv_strings[c->kind])); ssa_print_type(f, m, c->from); ssa_fprintf(f, " "); ssa_print_value(f, m, c->value, c->from); ssa_fprintf(f, " to "); ssa_print_type(f, m, c->to); ssa_fprintf(f, "\n"); } break; case ssaInstr_Unreachable: { ssa_fprintf(f, "unreachable\n"); } break; case ssaInstr_BinaryOp: { auto *bo = &value->Instr.BinaryOp; Type *type = base_type(ssa_type(bo->left)); Type *elem_type = type; while (elem_type->kind == Type_Vector) { elem_type = base_type(elem_type->Vector.elem); } ssa_fprintf(f, "%%%d = ", value->index); if (gb_is_between(bo->op, Token__ComparisonBegin+1, Token__ComparisonEnd-1)) { if (is_type_string(elem_type)) { ssa_fprintf(f, "call "); ssa_print_type(f, m, t_bool); char *runtime_proc = ""; switch (bo->op) { case Token_CmpEq: runtime_proc = "__string_eq"; break; case Token_NotEq: runtime_proc = "__string_ne"; break; case Token_Lt: runtime_proc = "__string_lt"; break; case Token_Gt: runtime_proc = "__string_gt"; break; case Token_LtEq: runtime_proc = "__string_le"; break; case Token_GtEq: runtime_proc = "__string_gt"; break; } ssa_fprintf(f, " "); ssa_print_encoded_global(f, make_string(runtime_proc), false); ssa_fprintf(f, "("); ssa_print_type(f, m, type); ssa_fprintf(f, " "); ssa_print_value(f, m, bo->left, type); ssa_fprintf(f, ", "); ssa_print_type(f, m, type); ssa_fprintf(f, " "); ssa_print_value(f, m, bo->right, type); ssa_fprintf(f, ")\n"); return; } else if (is_type_float(elem_type)) { ssa_fprintf(f, "fcmp "); switch (bo->op) { case Token_CmpEq: ssa_fprintf(f, "oeq"); break; case Token_NotEq: ssa_fprintf(f, "one"); break; case Token_Lt: ssa_fprintf(f, "olt"); break; case Token_Gt: ssa_fprintf(f, "ogt"); break; case Token_LtEq: ssa_fprintf(f, "ole"); break; case Token_GtEq: ssa_fprintf(f, "oge"); break; } } else { ssa_fprintf(f, "icmp "); if (bo->op != Token_CmpEq && bo->op != Token_NotEq) { if (is_type_unsigned(elem_type)) { ssa_fprintf(f, "u"); } else { ssa_fprintf(f, "s"); } } switch (bo->op) { case Token_CmpEq: ssa_fprintf(f, "eq"); break; case Token_NotEq: ssa_fprintf(f, "ne"); break; case Token_Lt: ssa_fprintf(f, "lt"); break; case Token_Gt: ssa_fprintf(f, "gt"); break; case Token_LtEq: ssa_fprintf(f, "le"); break; case Token_GtEq: ssa_fprintf(f, "ge"); break; } } } else { if (is_type_float(elem_type)) ssa_fprintf(f, "f"); switch (bo->op) { case Token_Add: ssa_fprintf(f, "add"); break; case Token_Sub: ssa_fprintf(f, "sub"); break; case Token_And: ssa_fprintf(f, "and"); break; case Token_Or: ssa_fprintf(f, "or"); break; case Token_Xor: ssa_fprintf(f, "xor"); break; case Token_Shl: ssa_fprintf(f, "shl"); break; case Token_Shr: ssa_fprintf(f, "lshr"); break; case Token_Mul: ssa_fprintf(f, "mul"); break; case Token_Not: ssa_fprintf(f, "xor"); break; case Token_AndNot: GB_PANIC("Token_AndNot Should never be called"); default: { if (!is_type_float(elem_type)) { if (is_type_unsigned(elem_type)) ssa_fprintf(f, "u"); else ssa_fprintf(f, "s"); } switch (bo->op) { case Token_Quo: ssa_fprintf(f, "div"); break; case Token_Mod: ssa_fprintf(f, "rem"); break; } } break; } } ssa_fprintf(f, " "); ssa_print_type(f, m, type); ssa_fprintf(f, " "); ssa_print_value(f, m, bo->left, type); ssa_fprintf(f, ", "); ssa_print_value(f, m, bo->right, type); ssa_fprintf(f, "\n"); } break; case ssaInstr_Call: { auto *call = &instr->Call; Type *result_type = call->type; if (result_type) { ssa_fprintf(f, "%%%d = ", value->index); } ssa_fprintf(f, "call "); if (result_type) { ssa_print_type(f, m, result_type); } else { ssa_fprintf(f, "void"); } ssa_fprintf(f, " "); ssa_print_value(f, m, call->value, call->type); ssa_fprintf(f, "("); if (call->arg_count > 0) { Type *proc_type = base_type(ssa_type(call->value)); GB_ASSERT(proc_type->kind == Type_Proc); auto *params = &proc_type->Proc.params->Tuple; for (isize i = 0; i < call->arg_count; i++) { Entity *e = params->variables[i]; GB_ASSERT(e != NULL); Type *t = e->type; if (i > 0) { ssa_fprintf(f, ", "); } ssa_print_type(f, m, t); ssa_fprintf(f, " "); ssaValue *arg = call->args[i]; ssa_print_value(f, m, arg, t); } } ssa_fprintf(f, ")\n"); } break; case ssaInstr_Select: { ssa_fprintf(f, "%%%d = select i1 ", value->index); ssa_print_value(f, m, instr->Select.cond, t_bool); ssa_fprintf(f, ", "); ssa_print_type(f, m, ssa_type(instr->Select.true_value)); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->Select.true_value, ssa_type(instr->Select.true_value)); ssa_fprintf(f, ", "); ssa_print_type(f, m, ssa_type(instr->Select.false_value)); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->Select.false_value, ssa_type(instr->Select.false_value)); ssa_fprintf(f, "\n"); } break; case ssaInstr_ExtractElement: { Type *vt = ssa_type(instr->ExtractElement.vector); ssa_fprintf(f, "%%%d = extractelement ", value->index); ssa_print_type(f, m, vt); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->ExtractElement.vector, vt); ssa_fprintf(f, ", "); Type *it = ssa_type(instr->ExtractElement.index); ssa_print_type(f, m, it); ssa_fprintf(f, " "); ssa_print_value(f, m, instr->ExtractElement.index, it); ssa_fprintf(f, "\n"); } break; case ssaInstr_InsertElement: { auto *ie = &instr->InsertElement; Type *vt = ssa_type(ie->vector); ssa_fprintf(f, "%%%d = insertelement ", value->index); ssa_print_type(f, m, vt); ssa_fprintf(f, " "); ssa_print_value(f, m, ie->vector, vt); ssa_fprintf(f, ", "); ssa_print_type(f, m, ssa_type(ie->elem)); ssa_fprintf(f, " "); ssa_print_value(f, m, ie->elem, ssa_type(ie->elem)); ssa_fprintf(f, ", "); ssa_print_type(f, m, ssa_type(ie->index)); ssa_fprintf(f, " "); ssa_print_value(f, m, ie->index, ssa_type(ie->index)); ssa_fprintf(f, "\n"); } break; case ssaInstr_ShuffleVector: { auto *sv = &instr->ShuffleVector; Type *vt = ssa_type(sv->vector); ssa_fprintf(f, "%%%d = shufflevector ", value->index); ssa_print_type(f, m, vt); ssa_fprintf(f, " "); ssa_print_value(f, m, sv->vector, vt); ssa_fprintf(f, ", "); ssa_print_type(f, m, vt); ssa_fprintf(f, " "); ssa_print_value(f, m, sv->vector, vt); ssa_fprintf(f, ", "); ssa_fprintf(f, "<%td x i32> <", sv->index_count); for (isize i = 0; i < sv->index_count; i++) { if (i > 0) { ssa_fprintf(f, ", "); } ssa_fprintf(f, "i32 %d", sv->indices[i]); } ssa_fprintf(f, ">"); ssa_fprintf(f, "\n"); } break; default: { GB_PANIC(" %d\n", instr->kind); ssa_fprintf(f, "; %d\n", instr->kind); } break; } } void ssa_print_proc(ssaFileBuffer *f, ssaModule *m, ssaProcedure *proc) { if (proc->body == NULL) { ssa_fprintf(f, "declare "); if (proc->tags & ProcTag_dll_import) { ssa_fprintf(f, "dllimport "); } if (proc->tags & ProcTag_dll_export) { ssa_fprintf(f, "dllexport "); } } else { ssa_fprintf(f, "\ndefine "); } if (proc->tags & ProcTag_stdcall) { ssa_fprintf(f, "cc 64 "); } else if (proc->tags & ProcTag_fastcall) { ssa_fprintf(f, "cc 65 "); } auto *proc_type = &proc->type->Proc; if (proc_type->result_count == 0) { ssa_fprintf(f, "void"); } else { ssa_print_type(f, m, proc_type->results); } ssa_fprintf(f, " "); ssa_print_encoded_global(f, proc->name, (proc->tags & (ProcTag_foreign|ProcTag_link_name)) != 0); ssa_fprintf(f, "("); if (proc_type->param_count > 0) { auto *params = &proc_type->params->Tuple; for (isize i = 0; i < params->variable_count; i++) { Entity *e = params->variables[i]; if (i > 0) { ssa_fprintf(f, ", "); } ssa_print_type(f, m, e->type); if (proc->body != NULL) { ssa_fprintf(f, " %%%.*s", LIT(e->token.string)); } } } ssa_fprintf(f, ") "); if (proc->tags & ProcTag_inline) { ssa_fprintf(f, "alwaysinline "); } if (proc->tags & ProcTag_no_inline) { ssa_fprintf(f, "noinline "); } if (proc->module->generate_debug_info && proc->entity != NULL) { if (proc->body != NULL) { ssaDebugInfo *di = *map_get(&proc->module->debug_info, hash_pointer(proc->entity)); GB_ASSERT(di->kind == ssaDebugInfo_Proc); ssa_fprintf(f, "!dbg !%d ", di->id); } } if (proc->body != NULL) { // ssa_fprintf(f, "nounwind uwtable {\n"); ssa_fprintf(f, "{\n"); for_array(i, proc->blocks) { ssaBlock *block = proc->blocks[i]; if (i > 0) ssa_fprintf(f, "\n"); ssa_print_block_name(f, block); ssa_fprintf(f, ":\n"); for_array(j, block->instrs) { ssaValue *value = block->instrs[j]; ssa_print_instr(f, m, value); } } ssa_fprintf(f, "}\n"); } else { ssa_fprintf(f, "\n"); } for_array(i, proc->children) { ssa_print_proc(f, m, proc->children[i]); } } void ssa_print_type_name(ssaFileBuffer *f, ssaModule *m, ssaValue *v) { GB_ASSERT(v->kind == ssaValue_TypeName); Type *bt = base_type(ssa_type(v)); if (!is_type_struct(bt) && !is_type_union(bt)) { return; } ssa_print_encoded_local(f, v->TypeName.name); ssa_fprintf(f, " = type "); ssa_print_type(f, m, base_type(v->TypeName.type)); ssa_fprintf(f, "\n"); } void ssa_print_llvm_ir(ssaFileBuffer *f, ssaModule *m) { if (m->layout.len > 0) { ssa_fprintf(f, "target datalayout = \"%.*s\"\n", LIT(m->layout)); } ssa_print_encoded_local(f, make_string("..string")); ssa_fprintf(f, " = type {i8*, "); ssa_print_type(f, m, t_int); ssa_fprintf(f, "} ; Basic_string\n"); ssa_print_encoded_local(f, make_string("..rawptr")); ssa_fprintf(f, " = type i8* ; Basic_rawptr\n"); ssa_print_encoded_local(f, make_string("..any")); ssa_fprintf(f, " = type {"); ssa_print_type(f, m, t_type_info_ptr); ssa_fprintf(f, ", "); ssa_print_type(f, m, t_rawptr); ssa_fprintf(f, "} ; Basic_any\n"); for_array(member_index, m->members.entries) { auto *entry = &m->members.entries[member_index]; ssaValue *v = entry->value; if (v->kind != ssaValue_TypeName) { continue; } ssa_print_type_name(f, m, v); } for_array(member_index, m->members.entries) { auto *entry = &m->members.entries[member_index]; ssaValue *v = entry->value; if (v->kind != ssaValue_Proc) { continue; } if (v->Proc.body == NULL) { ssa_print_proc(f, m, &v->Proc); } } for_array(member_index, m->members.entries) { auto *entry = &m->members.entries[member_index]; ssaValue *v = entry->value; if (v->kind != ssaValue_Proc) { continue; } if (v->Proc.body != NULL) { ssa_print_proc(f, m, &v->Proc); } } for_array(member_index, m->members.entries) { auto *entry = &m->members.entries[member_index]; ssaValue *v = entry->value; if (v->kind != ssaValue_Global) { continue; } auto *g = &v->Global; Scope *scope = g->entity->scope; b32 in_global_scope = false; if (scope != NULL) { in_global_scope = scope->is_global || scope->is_init; } ssa_print_encoded_global(f, g->entity->token.string, in_global_scope); ssa_fprintf(f, " = "); if (g->is_thread_local) { ssa_fprintf(f, "thread_local "); } if (g->is_constant) { if (g->is_private) { ssa_fprintf(f, "private "); } ssa_fprintf(f, "constant "); } else { ssa_fprintf(f, "global "); } ssa_print_type(f, m, g->entity->type); ssa_fprintf(f, " "); if (g->value != NULL) { ssa_print_value(f, m, g->value, g->entity->type); } else { ssa_fprintf(f, "zeroinitializer"); } ssa_fprintf(f, "\n"); } if (m->generate_debug_info) { ssa_fprintf(f, "\n"); ssa_fprintf(f, "!llvm.dbg.cu = !{!0}\n"); for_array(di_index, m->debug_info.entries) { auto *entry = &m->debug_info.entries[di_index]; ssaDebugInfo *di = entry->value; ssa_fprintf(f, "!%d = ", di->id); defer (ssa_fprintf(f, "\n")); switch (di->kind) { case ssaDebugInfo_CompileUnit: { auto *cu = &di->CompileUnit; ssaDebugInfo *file = *map_get(&m->debug_info, hash_pointer(cu->file)); ssa_fprintf(f, "distinct !DICompileUnit(" "language: DW_LANG_Go, " // Is this good enough? "file: !%d, " "producer: \"%.*s\", " "flags: \"\", " "runtimeVersion: 0, " "isOptimized: false, " "emissionKind: FullDebug" ")", file->id, LIT(cu->producer)); } break; case ssaDebugInfo_File: ssa_fprintf(f, "!DIFile(filename: \""); ssa_print_escape_string(f, di->File.filename, false); ssa_fprintf(f, "\", directory: \""); ssa_print_escape_string(f, di->File.directory, false); ssa_fprintf(f, "\")"); break; case ssaDebugInfo_Proc: ssa_fprintf(f, "distinct !DISubprogram(" "name: \"%.*s\", " // "linkageName: \"\", " "file: !%d, " "line: %td, " "isDefinition: true, " "isLocal: false, " "unit: !0" ")", LIT(di->Proc.name), di->Proc.file->id, di->Proc.pos.line); break; case ssaDebugInfo_AllProcs: ssa_fprintf(f, "!{"); for_array(proc_index, di->AllProcs.procs) { ssaDebugInfo *p = di->AllProcs.procs[proc_index]; if (proc_index > 0) {ssa_fprintf(f, ",");} ssa_fprintf(f, "!%d", p->id); } ssa_fprintf(f, "}"); break; } } } }