123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227 |
- 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) {
- 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);
- }
- }
- 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, ", ");
- }
- ssa_print_type(f, m, elem_type);
- 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_type(f, m, elem_type);
- 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, ", ");
- }
- ssa_print_type(f, m, elem_type);
- 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_type(f, m, elem_type);
- ssa_fprintf(f, " ");
- 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, "<INVALID-BLOCK>");
- }
- }
- 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->id);
- 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->id);
- 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->id);
- } break;
- case ssaInstr_Store: {
- Type *type = ssa_type(instr);
- 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->id);
- 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->id);
- 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->id);
- ssa_print_type(f, m, instr->Phi.type);
- ssa_fprintf(f, " ", value->id);
- for (isize i = 0; i < instr->Phi.edges.count; i++) {
- ssaValue *edge = instr->Phi.edges[i];
- if (i > 0) {
- ssa_fprintf(f, ", ");
- }
- 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->id);
- 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->id);
- } 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->id);
- }
- 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->id, 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->id);
- 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->id);
- }
- 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->id);
- 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->id);
- 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->id);
- 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->id);
- 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("<unknown instr> %d\n", instr->kind);
- ssa_fprintf(f, "; <unknown instr> %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;
- }
- }
- }
- }
|