Collab/WAI
2
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

feat: use types of values in stack and instructions

Browse source
This commit is contained in:
Orangerot 2025-05-12 00:52:51 +02:00
parent 47406b5def
commit 3f13610b96
1 changed files with 121 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

161
wai.c
View file

@ -27,7 +27,10 @@
#include <stddef.h> #include <stddef.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <inttypes.h>
#include <sys/types.h>
enum section { enum section {
Section_Custom, Section_Custom,
@ -47,11 +50,10 @@ enum section {
#define STACK_CAPACITY 1024 #define STACK_CAPACITY 1024
struct stack { struct stack {
double items[STACK_CAPACITY]; u_char items[STACK_CAPACITY];
size_t count; size_t bytes;
}; };
struct module { struct module {
struct type_t *types; struct type_t *types;
u_char *funcs[128]; u_char *funcs[128];
@ -72,7 +74,6 @@ enum INSTRUCTION {
INSTR_CALL = 0x10, INSTR_CALL = 0x10,
INSTR_ELSE = 0x05, INSTR_ELSE = 0x05,
INSTR_END = 0x0b, INSTR_END = 0x0b,
INSTR_F64 = 0x7C,
INSTR_F64_CONST = 0x44, INSTR_F64_CONST = 0x44,
INSTR_F64_LT = 0x63, INSTR_F64_LT = 0x63,
INSTR_F64_MUL = 0xa2, INSTR_F64_MUL = 0xa2,
@ -91,26 +92,81 @@ enum TYPE {
TYPE_EXTERNREF = 0x6F TYPE_EXTERNREF = 0x6F
}; };
static const int TYPE_SIZE[] = {
[TYPE_I32] = 4,
[TYPE_I64] = 8,
[TYPE_F32] = 4,
[TYPE_F64] = 8,
[TYPE_V128] = 16,
[TYPE_FUNCREF] = 16,
[TYPE_EXTERNREF] = 16
};
struct value_t {
enum TYPE type;
union {
int32_t i32;
int64_t i64;
float f32;
double f64;
__int128 v128;
int64_t funcref;
int64_t extref;
} value;
};
#define incr(i, len) i++; if (i >= len) {return -1;} #define incr(i, len) i++; if (i >= len) {return -1;}
void stack_push(struct stack *s, double a) { void stack_push(struct stack *s, const struct value_t *value) {
s->items[s->count++] = a; size_t type_size = TYPE_SIZE[value->type];
memcpy(&(s->items[s->bytes]), &value->value, type_size);
s->items[s->bytes + type_size] = value->type;
s->bytes += type_size + 1;
printf("stack: "); printf("stack: ");
for (int i = 0; i < s->count; i++) { for (int i = s->bytes - 1; i > 0; i -= TYPE_SIZE[s->items[i]] + 1) {
printf("%f, ", s->items[i]); enum TYPE t = s->items[i];
size_t type_size = TYPE_SIZE[t];
void *value = &s->items[i - type_size];
switch (t) {
case TYPE_I32:
printf("%d (I32)", *(int32_t*)value);
break;
case TYPE_I64:
printf("%ld (I32)", *(int64_t*)value);
break;
case TYPE_F32:
printf("%f (F32)", *(float*)value);
break;
case TYPE_F64:
printf("%f (F64)", *(double*)value);
break;
case TYPE_V128:
printf("%ld (V128)", *(__int128*)value);
break;
case TYPE_FUNCREF:
printf("%ld (EREF)", *(int64_t*)value);
break;
case TYPE_EXTERNREF:
printf("%ld (EREF)", *(int64_t*)value);
break;
}
printf(", ");
} }
printf("\n"); printf("\n");
} }
double stack_pop(struct stack *s) { void stack_top(struct stack *s, struct value_t *value) {
s->count--; value->type = s->items[s->bytes-1];
return s->items[s->count]; memcpy(&value->value, &(s->items[s->bytes - 1 - TYPE_SIZE[value->type]]), TYPE_SIZE[value->type]);
}
double stack_top(struct stack *s) {
return s->items[s->count-1];
} }
void stack_pop(struct stack *s, struct value_t *value) {
stack_top(s, value);
s->bytes -= TYPE_SIZE[value->type] + 1;
}
int parse_type(u_char *binary, int len) { int parse_type(u_char *binary, int len) {
int i = 0; int i = 0;
enum TYPE param = binary[i]; enum TYPE param = binary[i];
@ -136,47 +192,71 @@ int parse_instruction(struct module *module, u_char *binary, double param, int l
int i = 0; int i = 0;
enum INSTRUCTION instr = (u_char) binary[i]; enum INSTRUCTION instr = (u_char) binary[i];
u_char *instr_addr = &binary[i]; u_char *instr_addr = &binary[i];
struct value_t a = {0};
struct value_t b = {0};
struct value_t result = {0};
incr(i, len); incr(i, len);
switch (instr) { switch (instr) {
case INSTR_CALL: case INSTR_CALL: {
int a = binary[i]; int func_index = binary[i];
incr(i, len); incr(i, len);
parse_function(module, module->funcs[a], stack_pop(&module->stack), len); stack_pop(&module->stack, &a);
parse_function(module, module->funcs[func_index], a.value.f64, len);
break; break;
}
case INSTR_ELSE: case INSTR_ELSE:
printf("reached else instruction: impossible!\n"); printf("reached else instruction: impossible!\n");
case INSTR_END: case INSTR_END:
break; break;
case INSTR_F64:
break;
case INSTR_F64_CONST: case INSTR_F64_CONST:
double k = *(double*)&binary[i]; result.type = TYPE_F64;
stack_push(&module->stack, k); result.value.f64 = *(double*)&binary[i];
i += 8; i += 8;
stack_push(&module->stack, &result);
break; break;
case INSTR_F64_LT: { case INSTR_F64_LT: {
double b = stack_pop(&module->stack); stack_pop(&module->stack, &a);
double a = stack_pop(&module->stack); stack_pop(&module->stack, &b);
stack_push(&module->stack, a < b); if (a.type != TYPE_F64 || b.type != TYPE_F64)
printf("Wrong types!\n");
result.type = TYPE_F64;
result.value.f64 = b.value.f64 < a.value.f64;
stack_push(&module->stack, &result);
break; break;
} }
case INSTR_F64_MUL: { case INSTR_F64_MUL: {
double b = stack_pop(&module->stack); stack_pop(&module->stack, &a);
double a = stack_pop(&module->stack); stack_pop(&module->stack, &b);
stack_push(&module->stack, a * b); if (a.type != TYPE_F64 || b.type != TYPE_F64)
printf("Wrong types!\n");
result.type = TYPE_F64;
result.value.f64 = a.value.f64 * b.value.f64;
stack_push(&module->stack, &result);
break; break;
} }
case INSTR_F64_SUB: { case INSTR_F64_SUB: {
double b = stack_pop(&module->stack); stack_pop(&module->stack, &a);
double a = stack_pop(&module->stack); stack_pop(&module->stack, &b);
stack_push(&module->stack, a - b); if (a.type != TYPE_F64 || b.type != TYPE_F64)
printf("Wrong types!\n");
result.type = TYPE_F64;
result.value.f64 = b.value.f64 - a.value.f64;
stack_push(&module->stack, &result);
break; break;
} }
case INSTR_IF: { case INSTR_IF: {
double a = stack_pop(&module->stack); stack_pop(&module->stack, &a);
enum TYPE condition_type = binary[i];
incr(i, len);
if (a.type != condition_type)
printf("Wrong types!\n");
while (binary[i] != INSTR_ELSE) { while (binary[i] != INSTR_ELSE) {
if (a) { // TODO test condition with correct type.
// This might not matter since all types are false with 0x0
if (a.value.i64) {
i += parse_instruction(module, &binary[i], param, len); i += parse_instruction(module, &binary[i], param, len);
} else { } else {
incr(i, len); incr(i, len);
@ -184,7 +264,7 @@ int parse_instruction(struct module *module, u_char *binary, double param, int l
} }
incr(i, len); incr(i, len);
while (binary[i] != INSTR_END) { while (binary[i] != INSTR_END) {
if (a) { if (a.value.i64) {
incr(i, len); incr(i, len);
} else { } else {
i += parse_instruction(module, &binary[i], param, len); i += parse_instruction(module, &binary[i], param, len);
@ -193,11 +273,12 @@ int parse_instruction(struct module *module, u_char *binary, double param, int l
incr(i, len); incr(i, len);
break; break;
} }
case INSTR_LOCAL_GET: case INSTR_LOCAL_GET: {
int local_index = binary[i]; int local_index = binary[i];
incr(i, len); incr(i, len);
stack_push(&module->stack, param); stack_push(&module->stack, &(struct value_t) {.value.f64 = param, .type = TYPE_F64});
break; break;
}
default: default:
printf("unknown instruction! %x at %lx\n", instr, instr_addr - module->binary); printf("unknown instruction! %x at %lx\n", instr, instr_addr - module->binary);
exit(1); exit(1);
@ -299,7 +380,7 @@ int parse_section(struct module *module, u_char *binary, int len) {
module->funcs[function_i] = &binary[i]; module->funcs[function_i] = &binary[i];
i += parse_function(module, &binary[i], 4, len); i += parse_function(module, &binary[i], 4, len);
} }
printf("result: %f\n", module->stack.items[0]); // printf("result: %f\n", module->stack.items[0]);
break; break;
case Section_Data: case Section_Data:
break; break;
@ -316,9 +397,9 @@ int parse_section(struct module *module, u_char *binary, int len) {
int parse_module(u_char *binary, size_t len) { int parse_module(u_char *binary, size_t len) {
int i = 0; int i = 0;
u_char *magic = "\0asm"; char *magic = "\0asm";
while (i < 4) { while (i < 4) {
if (binary[i] != magic[i]) { if ((char) binary[i] != magic[i]) {
fprintf(stderr, "no wasm magic\n"); fprintf(stderr, "no wasm magic\n");
return 0; return 0;
} }
@ -355,7 +436,7 @@ int main(int argc, char **argv) {
stat(argv[1], &st); stat(argv[1], &st);
printf("size: %ld\n", st.st_size); printf("size: %ld\n", st.st_size);
unsigned char *binary = malloc(st.st_size); u_char *binary = malloc(st.st_size);
fread(binary, st.st_size, st.st_size, file); fread(binary, st.st_size, st.st_size, file);
if (parse_module(binary, st.st_size) == -1) { if (parse_module(binary, st.st_size) == -1) {