wash/ivy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

asm: implement an asm parser and emitter

Browse Source
This commit is contained in:
max 🍓
2024-12-13 12:26:01 +00:00
parent 0a8d913fdd
commit 5fe1a21978
19 changed files with 1106 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
asm/assembler/assembler.c
View File

@@ -8,9 +8,11 @@
#include "assembler.h"
extern struct assembler_scope_type class_scope_type;
extern struct assembler_scope_type constpool_scope_type;
static const struct assembler_scope_type *scope_types[] = {
[IVY_ASM_SCOPE_CLASS] = &class_scope_type,
[IVY_ASM_SCOPE_CONSTPOOL] = &constpool_scope_type,
};
static const size_t nr_scope_types = sizeof scope_types / sizeof scope_types[0];
@@ -44,9 +46,14 @@ enum ivy_status ivy_assembler_create(FILE *fp, struct ivy_assembler **as)
out->as_xdat = tmpfile();
struct ivy_bin_header header = {0};
header.h_magic = b_i32_htob(0xAABBCCDD);
fseek(fp, 0, SEEK_SET);
fwrite(&header, sizeof header, 1, fp);
out->as_data_offset = sizeof header;
char blank[16] = {0};
assembler_write_extended_data(out, blank, sizeof blank);
*as = out;
return IVY_OK;
@@ -66,7 +73,7 @@ enum ivy_status ivy_assembler_finish(struct ivy_assembler *as)
size_t xdat_len = as->as_next_xdata_key;
fseek(as->as_xdat, 0, SEEK_SET);
char *buf = malloc(buf_len);
unsigned char *buf = malloc(buf_len);
if (!buf) {
return IVY_ERR_NO_MEMORY;
@@ -81,7 +88,7 @@ enum ivy_status ivy_assembler_finish(struct ivy_assembler *as)
while (1) {
size_t r = fread(buf, 1, buf_len, as->as_xdat);
fwrite(buf, 1, r, as->as_xdat);
fwrite(buf, 1, r, as->as_data);
as->as_data_offset += r;
if (r == buf_len) {
@@ -98,7 +105,7 @@ enum ivy_status ivy_assembler_finish(struct ivy_assembler *as)
free(buf);
unsigned int nr_table_entries = 0;
unsigned short nr_table_entries = 0;
struct ivy_bin_header header = {0};
header.h_table_offset = b_i64_htob(as->as_data_offset);
@@ -122,7 +129,7 @@ enum ivy_status ivy_assembler_finish(struct ivy_assembler *as)
fwrite(&xdat, 1, sizeof xdat, as->as_data);
nr_table_entries++;
header.h_table_len = b_i16_htob(nr_table_entries * sizeof(struct ivy_bin_table_entry));
header.h_table_len = b_i16_htob(nr_table_entries);
fseek(as->as_data, 0, SEEK_SET);
fwrite(&header, 1, sizeof header, as->as_data);
@@ -153,6 +160,11 @@ extended_data_key assembler_write_extended_data(struct ivy_assembler *as, const
return key;
}
struct assembler_scope *assembler_get_scope(struct ivy_assembler *as)
{
return as->as_scope;
}
enum ivy_status ivy_assembler_begin_scope(
struct ivy_assembler *as, enum ivy_assembler_scope_type type,
ivy_assembler_attrib_table attrib)
@@ -175,6 +187,8 @@ enum ivy_status ivy_assembler_begin_scope(
return IVY_ERR_NO_MEMORY;
}
memset(scope, 0x0, type_info->s_scope_size);
while ((as->as_data_offset % 16) != 0) {
fputc(0, as->as_data);
as->as_data_offset++;

2
asm/assembler/assembler.h
View File

@@ -48,4 +48,6 @@ struct assembler_scope_type {
extern size_t assembler_write_data(struct ivy_assembler *as, const void *p, size_t len);
extern extended_data_key assembler_write_extended_data(struct ivy_assembler *as, const void *p, size_t len);
extern struct assembler_scope *assembler_get_scope(struct ivy_assembler *as);
#endif

2
asm/assembler/class.c
View File

@@ -17,7 +17,7 @@ static enum ivy_status put_xval(
struct ivy_assembler *as,
struct assembler_scope *scope,
enum ivy_assembler_xval_type type,
ivy_assembler_attrib_table attrib)
const ivy_assembler_attrib_table attrib)
{
struct ivy_bin_class_table_entry entry = {0};
switch (type) {

184
asm/assembler/constpool.c
View File

@@ -0,0 +1,184 @@
#include "assembler.h"
#include <ivy/asm/bin.h>
#include <ivy/ident.h>
#include <ivy/selector.h>
#include <blue/object/dict.h>
#include <blue/object/number.h>
#include <blue/object/string.h>
#include <blue/core/hash.h>
#include <stdlib.h>
#include <string.h>
struct constpool_assembler_scope {
struct assembler_scope s_base;
b_dict *s_strings;
};
static enum ivy_status init_scope(
struct ivy_assembler *as, struct assembler_scope *scope,
const ivy_assembler_attrib_table attrib)
{
struct ivy_bin_constpool header = {0};
assembler_write_data(as, &header, sizeof header);
struct constpool_assembler_scope *c = (struct constpool_assembler_scope *)scope;
c->s_strings = b_dict_create();
return IVY_OK;
}
static ivy_bin_data_handle get_cached_string(struct constpool_assembler_scope *scope, const char *s)
{
b_number *key = b_dict_at(scope->s_strings, s);
if (!key) {
return IVY_BIN_NULL_HANDLE;
}
return (ivy_bin_data_handle)b_number_get_int32(key);
}
static void put_cached_string(struct constpool_assembler_scope *scope, const char *s, ivy_bin_data_handle key)
{
b_dict_put(scope->s_strings, s, B_RV_INT32(key));
}
static ivy_bin_data_handle write_string(struct ivy_assembler *as, const char *s)
{
struct constpool_assembler_scope *scope = (struct constpool_assembler_scope *)assembler_get_scope(as);
ivy_bin_data_handle key = get_cached_string(scope, s);
if (key != IVY_BIN_NULL_HANDLE) {
return key;
}
size_t len = strlen(s);
struct ivy_bin_string str = { 0 };
str.s_hash = b_i32_htob(b_hash_string(s));
str.s_len = b_i32_htob(len);
key = assembler_write_extended_data(as, &str, sizeof str);
/* include the trailing null terminator. */
assembler_write_extended_data(as, s, len + 1);
put_cached_string(scope, s, key);
return key;
}
static ivy_bin_data_handle write_selector(struct ivy_assembler *as, const struct ivy_selector *sel)
{
struct ivy_bin_selector dat = {0};
switch (sel->sel_recipient) {
case IVY_SEL_CLASS:
dat.sel_flags = IVY_BIN_SELECTOR_CLASS;
break;
case IVY_SEL_OBJECT:
dat.sel_flags = IVY_BIN_SELECTOR_OBJECT;
break;
default:
return IVY_BIN_NULL_HANDLE;
}
if (sel->sel_name) {
dat.sel_name = b_i32_htob(write_string(as, sel->sel_name));
}
unsigned int i = 0;
unsigned int nr_args = b_queue_length(&sel->sel_args);
/* TODO hash. */
ivy_bin_data_handle *arg_handles = calloc(nr_args, sizeof(ivy_bin_data_handle));
b_queue_iterator it = {0};
b_queue_foreach (&it, &sel->sel_args) {
struct ivy_selector_arg *arg = b_unbox(struct ivy_selector_arg, it.entry, arg_entry);
arg_handles[i++] = write_string(as, arg->arg_label);
}
ivy_bin_data_handle selector_handle = assembler_write_extended_data(as, &dat, sizeof dat);
for (i = 0; i < nr_args; i++) {
b_i32 arg_handle = b_i32_htob(arg_handles[i]);
assembler_write_extended_data(as, &arg_handle, sizeof arg_handle);
}
free(arg_handles);
return selector_handle;
}
static ivy_bin_data_handle write_ident(struct ivy_assembler *as, const struct ivy_ident *id)
{
struct ivy_bin_ident dat = {0};
/* TODO hash. */
unsigned int nr_parts = b_queue_length(&id->id_parts);
unsigned int i = 0;
ivy_bin_data_handle *part_handles = calloc(nr_parts, sizeof(ivy_bin_data_handle));
b_queue_iterator it = {0};
b_queue_foreach (&it, &id->id_parts) {
struct ivy_ident_part *arg = b_unbox(struct ivy_ident_part, it.entry, p_entry);
part_handles[i++] = write_string(as, arg->p_str);
}
dat.id_nr_parts = (uint8_t)nr_parts;
ivy_bin_data_handle ident_handle = assembler_write_extended_data(as, &dat, sizeof dat);
for (i = 0; i < nr_parts; i++) {
b_i32 part_handle = b_i32_htob(part_handles[i]);
assembler_write_extended_data(as, &part_handle, sizeof part_handle);
}
free(part_handles);
return ident_handle;
}
static enum ivy_status put_pval(
struct ivy_assembler *as, struct assembler_scope *scope,
enum ivy_assembler_pval_type type, unsigned long index, const void *val)
{
struct ivy_bin_constpool_table_entry entry = {0};
uintptr_t i = *(uintptr_t *)val;
switch (type) {
case IVY_ASM_PVAL_STRING:
entry.e_type = b_i32_htob(IVY_CONSTPOOL_TABLE_STRING);
entry.e_ex_handle = b_i32_htob(write_string(as, val));
break;
case IVY_ASM_PVAL_IDENT:
entry.e_type = b_i32_htob(IVY_CONSTPOOL_TABLE_IDENT);
entry.e_ex_handle = b_i32_htob(write_ident(as, val));
break;
case IVY_ASM_PVAL_ATOM:
entry.e_type = b_i32_htob(IVY_CONSTPOOL_TABLE_ATOM);
entry.e_ex_handle = b_i32_htob(write_string(as, val));
break;
case IVY_ASM_PVAL_SINT:
entry.e_type = b_i32_htob(IVY_CONSTPOOL_TABLE_INT);
entry.e_int = b_i32_htob((uint32_t)i);
break;
case IVY_ASM_PVAL_UINT:
entry.e_type = b_i32_htob(IVY_CONSTPOOL_TABLE_UINT);
entry.e_int = b_i32_htob((uint32_t)i);
break;
case IVY_ASM_PVAL_SELECTOR:
entry.e_type = b_i32_htob(IVY_CONSTPOOL_TABLE_SELECTOR);
entry.e_ex_handle = b_i32_htob(write_selector(as, val));
break;
default:
return IVY_ERR_NOT_SUPPORTED;
}
assembler_write_data(as, &entry, sizeof entry);
return IVY_OK;
}
const struct assembler_scope_type constpool_scope_type = {
.s_scope_size = sizeof(struct constpool_assembler_scope),
.s_init_scope = init_scope,
.s_put_pval = put_pval,
};