sideros/src/mods/vm.zig
2025-08-04 20:48:35 +02:00

530 lines
23 KiB
Zig

const std = @import("std");
const wasm = @import("wasm.zig");
const Parser = @import("Parser.zig");
const IR = @import("ir.zig");
const Allocator = std.mem.Allocator;
const AllocationError = error{OutOfMemory};
pub const Memory = struct {
min: u32,
max: ?u32,
};
pub const Valtype = union(enum) {
val: std.wasm.Valtype,
ref: std.wasm.RefType,
};
pub const Functype = struct {
parameters: []Valtype,
returns: []Valtype,
pub fn deinit(self: Functype, allocator: Allocator) void {
allocator.free(self.parameters);
allocator.free(self.returns);
}
};
pub const Function = struct { func_type: Functype, typ: union(enum) {
internal: struct {
locals: []Valtype,
ir: IR,
},
external: void,
} };
pub const ExportFunction = enum {
preinit,
logErr,
logWarn,
logInfo,
logDebug,
};
pub const Exports = struct {
preinit: ?u32 = null,
logErr: ?u32 = null,
logWarn: ?u32 = null,
logInfo: ?u32 = null,
logDebug: ?u32 = null,
};
comptime {
std.debug.assert(@typeInfo(ExportFunction).@"enum".fields.len == @typeInfo(Exports).@"struct".fields.len);
}
pub const Module = struct {
memory: Memory,
functions: []Function,
exports: Exports,
exported_memory: u32,
imported_funcs: u32,
pub fn deinit(self: Module, allocator: Allocator) void {
// self.exports.deinit(allocator);
for (self.functions) |f| {
std.debug.print("Freeing function parameters at {*}\n", .{f.func_type.parameters.ptr});
allocator.free(f.func_type.parameters);
allocator.free(f.func_type.returns);
switch (f.typ) {
.internal => {
allocator.free(f.typ.internal.ir.opcodes);
allocator.free(f.typ.internal.ir.indices);
allocator.free(f.typ.internal.ir.select_valtypes);
allocator.free(f.typ.internal.locals);
},
.external => @panic("UNIMPLEMENTED"),
}
}
allocator.free(self.functions);
}
};
pub const CallFrame = struct {
program_counter: usize,
code: IR,
locals: []Value,
};
pub const Value = union(enum) {
i32: i32,
i64: i64,
f32: f32,
f64: f64,
};
pub const Runtime = struct {
module: Module,
stack: std.ArrayList(Value),
memory: []u8,
global_runtime: *wasm.GlobalRuntime,
pub fn init(allocator: Allocator, module: Module, global_runtime: *wasm.GlobalRuntime) !Runtime {
// if memory max is not set the memory is allowed to grow but it is not supported at the moment
const max = module.memory.max orelse module.memory.min;
if (module.memory.max == null) {
std.log.warn("Growing memory is not yet supported, usign the minimum memory\n", .{});
}
const memory = try allocator.alloc(u8, max);
return Runtime{
.module = module,
.stack = try std.ArrayList(Value).initCapacity(allocator, 10),
.memory = memory,
.global_runtime = global_runtime,
};
}
pub fn deinit(self: *Runtime, allocator: Allocator) void {
self.module.deinit(allocator);
self.stack.deinit();
allocator.free(self.memory);
}
pub fn executeFrame(self: *Runtime, allocator: Allocator, frame: *CallFrame) !void {
loop: while (frame.program_counter < frame.code.opcodes.len) {
const opcode: IR.Opcode = frame.code.opcodes[frame.program_counter];
const index = frame.code.indices[frame.program_counter];
switch (opcode) {
.@"unreachable" => {
std.debug.panic("Reached unreachable statement at IR counter {any}\n", .{frame.program_counter});
},
.nop => {},
.br => {
frame.program_counter = index.u32;
continue;
},
.br_if => {
if (self.stack.pop().?.i32 != 0) {
frame.program_counter = index.u32;
continue;
}
},
.br_table => @panic("UNIMPLEMENTED"),
.@"return" => break :loop,
.call => {
if (index.u32 == self.module.exports.logDebug) {
std.debug.print("TODO: logDebug\n", .{});
} else if (index.u32 == self.module.exports.logInfo) {
std.debug.print("TODO: logInfo\n", .{});
} else if (index.u32 == self.module.exports.logWarn) {
std.debug.print("TODO: logWarn\n", .{});
} else if (index.u32 == self.module.exports.logErr) {
std.debug.print("TODO: logErr\n", .{});
} else {
var parameters = std.ArrayList(Value).init(allocator);
defer parameters.deinit();
for (self.module.functions[index.u32 - self.module.imported_funcs].func_type.parameters) |_| {
try parameters.append(self.stack.pop().?);
}
try self.call(allocator, index.u32 - self.module.imported_funcs, parameters.items);
}
},
.call_indirect => @panic("UNIMPLEMENTED"),
.refnull => @panic("UNIMPLEMENTED"),
.refisnull => @panic("UNIMPLEMENTED"),
.reffunc => @panic("UNIMPLEMENTED"),
.drop => {
_ = self.stack.pop();
},
.select => {
const c = self.stack.pop().?.i32;
const val2 = self.stack.pop().?;
const val1 = self.stack.pop().?;
if (c != 0) {
try self.stack.append(val1);
} else {
try self.stack.append(val2);
}
},
.select_with_values => @panic("UNIMPLEMENTED"),
.localget => try self.stack.append(frame.locals[index.u32]),
.localset => frame.locals[index.u32] = self.stack.pop().?,
.localtee => frame.locals[index.u32] = self.stack.items[self.stack.items.len - 1],
.globalget => try self.stack.append(self.global_runtime.getGlobal(index.u32)),
.globalset => try self.global_runtime.updateGlobal(index.u32, self.stack.pop().?),
.tableget => @panic("UNIMPLEMENTED"),
.tableset => @panic("UNIMPLEMENTED"),
.tableinit => @panic("UNIMPLEMENTED"),
.elemdrop => @panic("UNIMPLEMENTED"),
.tablecopy => @panic("UNIMPLEMENTED"),
.tablegrow => @panic("UNIMPLEMENTED"),
.tablesize => @panic("UNIMPLEMENTED"),
.tablefill => @panic("UNIMPLEMENTED"),
// TODO(ernesto): This code is repeated...
.i32_load => {
const start = index.memarg.alignment + index.memarg.offset;
const end = start + @sizeOf(i32);
try self.stack.append(.{ .i32 = std.mem.littleToNative(i32, std.mem.bytesAsValue(i32, self.memory[start..end]).*) });
},
.i64_load => {
const start = index.memarg.alignment + index.memarg.offset;
const end = start + @sizeOf(i64);
try self.stack.append(.{ .i64 = std.mem.littleToNative(i64, std.mem.bytesAsValue(i64, self.memory[start..end]).*) });
},
.f32_load => {
const start = index.memarg.alignment + index.memarg.offset;
const end = start + @sizeOf(f32);
try self.stack.append(.{ .f32 = std.mem.littleToNative(f32, std.mem.bytesAsValue(f32, self.memory[start..end]).*) });
},
.f64_load => {
const start = index.memarg.alignment + index.memarg.offset;
const end = start + @sizeOf(f64);
try self.stack.append(.{ .f64 = std.mem.littleToNative(f64, std.mem.bytesAsValue(f64, self.memory[start..end]).*) });
},
.i32_load8_s => @panic("UNIMPLEMENTED"),
.i32_load8_u => @panic("UNIMPLEMENTED"),
.i32_load16_s => @panic("UNIMPLEMENTED"),
.i32_load16_u => @panic("UNIMPLEMENTED"),
.i64_load8_s => @panic("UNIMPLEMENTED"),
.i64_load8_u => @panic("UNIMPLEMENTED"),
.i64_load16_s => @panic("UNIMPLEMENTED"),
.i64_load16_u => @panic("UNIMPLEMENTED"),
.i64_load32_s => @panic("UNIMPLEMENTED"),
.i64_load32_u => @panic("UNIMPLEMENTED"),
.i32_store => @panic("UNIMPLEMENTED"),
.i64_store => {
// TODO(ernesto): I'm pretty sure this is wrong
const val = std.mem.nativeToLittle(i64, self.stack.pop().?.i64);
const offsetVal = self.stack.pop().?.i32;
if (offsetVal < 0) {
std.debug.panic("offsetVal is negative (val: {any})\n", .{offsetVal});
}
const offset: u64 = @intCast(offsetVal);
const start: usize = @intCast(@as(u64, index.memarg.offset) + offset);
const end = start + @sizeOf(u64);
@memcpy(self.memory[start..end], std.mem.asBytes(&val));
},
.f32_store => @panic("UNIMPLEMENTED"),
.f64_store => @panic("UNIMPLEMENTED"),
.i32_store8 => @panic("UNIMPLEMENTED"),
.i32_store16 => @panic("UNIMPLEMENTED"),
.i64_store8 => @panic("UNIMPLEMENTED"),
.i64_store16 => @panic("UNIMPLEMENTED"),
.i64_store32 => @panic("UNIMPLEMENTED"),
.memorysize => @panic("UNIMPLEMENTED"),
.memorygrow => @panic("UNIMPLEMENTED"),
.memoryinit => @panic("UNIMPLEMENTED"),
.datadrop => @panic("UNIMPLEMENTED"),
.memorycopy => @panic("UNIMPLEMENTED"),
.memoryfill => @panic("UNIMPLEMENTED"),
.i32_const => {
try self.stack.append(Value{ .i32 = frame.code.indices[frame.program_counter].i32 });
},
.i64_const => {
try self.stack.append(Value{ .i64 = frame.code.indices[frame.program_counter].i64 });
},
.f32_const => @panic("UNIMPLEMENTED"),
.f64_const => @panic("UNIMPLEMENTED"),
.i32_eqz => {
try self.stack.append(Value{ .i32 = @intCast(@as(u1, @bitCast(self.stack.pop().?.i32 == 0))) });
},
.i32_eq => @panic("UNIMPLEMENTED"),
.i32_ne => @panic("UNIMPLEMENTED"),
.i32_lt_s => @panic("UNIMPLEMENTED"),
.i32_lt_u => {
const b = self.stack.pop().?.i32;
const a = self.stack.pop().?.i32;
try self.stack.append(Value{ .i32 = @intCast(@as(u1, @bitCast(a < b))) });
},
.i32_gt_s => @panic("UNIMPLEMENTED"),
.i32_gt_u => @panic("UNIMPLEMENTED"),
.i32_le_s => @panic("UNIMPLEMENTED"),
.i32_le_u => @panic("UNIMPLEMENTED"),
.i32_ge_s => @panic("UNIMPLEMENTED"),
.i32_ge_u => {
const b = self.stack.pop().?.i32;
const a = self.stack.pop().?.i32;
try self.stack.append(Value{ .i32 = @intCast(@as(u1, @bitCast(a >= b))) });
},
.i64_eqz => {
try self.stack.append(Value{ .i32 = @intCast(@as(u1, @bitCast(self.stack.pop().?.i32 == 0))) });
},
.i64_eq => @panic("UNIMPLEMENTED"),
.i64_ne => @panic("UNIMPLEMENTED"),
.i64_lt_s => @panic("UNIMPLEMENTED"),
.i64_lt_u => {
const b = self.stack.pop().?.i32;
const a = self.stack.pop().?.i32;
try self.stack.append(Value{ .i32 = @intCast(@as(u1, @bitCast(a < b))) });
},
.i64_gt_s => @panic("UNIMPLEMENTED"),
.i64_gt_u => @panic("UNIMPLEMENTED"),
.i64_le_s => @panic("UNIMPLEMENTED"),
.i64_le_u => @panic("UNIMPLEMENTED"),
.i64_ge_s => @panic("UNIMPLEMENTED"),
.i64_ge_u => @panic("UNIMPLEMENTED"),
.f32_eq => @panic("UNIMPLEMENTED"),
.f32_ne => @panic("UNIMPLEMENTED"),
.f32_lt => @panic("UNIMPLEMENTED"),
.f32_gt => @panic("UNIMPLEMENTED"),
.f32_le => @panic("UNIMPLEMENTED"),
.f32_ge => @panic("UNIMPLEMENTED"),
.f64_eq => @panic("UNIMPLEMENTED"),
.f64_ne => @panic("UNIMPLEMENTED"),
.f64_lt => @panic("UNIMPLEMENTED"),
.f64_gt => @panic("UNIMPLEMENTED"),
.f64_le => @panic("UNIMPLEMENTED"),
.f64_ge => @panic("UNIMPLEMENTED"),
.i32_clz => @panic("UNIMPLEMENTED"),
.i32_ctz => @panic("UNIMPLEMENTED"),
.i32_popcnt => @panic("UNIMPLEMENTED"),
.i32_add => {
const a = self.stack.pop().?.i32;
const b = self.stack.pop().?.i32;
try self.stack.append(Value{ .i32 = a + b });
},
.i32_sub => {
const b = self.stack.pop().?.i32;
const a = self.stack.pop().?.i32;
try self.stack.append(Value{ .i32 = a - b });
},
.i32_and => {
const a = self.stack.pop().?.i32;
const b = self.stack.pop().?.i32;
try self.stack.append(Value{ .i32 = a & b });
},
.i32_mul => @panic("UNIMPLEMENTED"),
.i32_div_s => @panic("UNIMPLEMENTED"),
.i32_div_u => @panic("UNIMPLEMENTED"),
.i32_rem_s => @panic("UNIMPLEMENTED"),
.i32_rem_u => @panic("UNIMPLEMENTED"),
.i32_or => @panic("UNIMPLEMENTED"),
.i32_xor => @panic("UNIMPLEMENTED"),
.i32_shl => @panic("UNIMPLEMENTED"),
.i32_shr_s => @panic("UNIMPLEMENTED"),
.i32_shr_u => @panic("UNIMPLEMENTED"),
.i32_rotl => @panic("UNIMPLEMENTED"),
.i32_rotr => @panic("UNIMPLEMENTED"),
.i64_clz => @panic("UNIMPLEMENTED"),
.i64_ctz => @panic("UNIMPLEMENTED"),
.i64_popcnt => @panic("UNIMPLEMENTED"),
.i64_add => {
const a = self.stack.pop().?.i64;
const b = self.stack.pop().?.i64;
try self.stack.append(Value{ .i64 = a + b });
},
.i64_sub => @panic("UNIMPLEMENTED"),
.i64_mul => @panic("UNIMPLEMENTED"),
.i64_div_s => @panic("UNIMPLEMENTED"),
.i64_div_u => @panic("UNIMPLEMENTED"),
.i64_rem_s => @panic("UNIMPLEMENTED"),
.i64_rem_u => @panic("UNIMPLEMENTED"),
.i64_and => @panic("UNIMPLEMENTED"),
.i64_or => @panic("UNIMPLEMENTED"),
.i64_xor => @panic("UNIMPLEMENTED"),
.i64_shl => @panic("UNIMPLEMENTED"),
.i64_shr_s => @panic("UNIMPLEMENTED"),
.i64_shr_u => @panic("UNIMPLEMENTED"),
.i64_rotl => @panic("UNIMPLEMENTED"),
.i64_rotr => @panic("UNIMPLEMENTED"),
.f32_abs => @panic("UNIMPLEMENTED"),
.f32_neg => @panic("UNIMPLEMENTED"),
.f32_ceil => @panic("UNIMPLEMENTED"),
.f32_floor => @panic("UNIMPLEMENTED"),
.f32_trunc => @panic("UNIMPLEMENTED"),
.f32_nearest => @panic("UNIMPLEMENTED"),
.f32_sqrt => @panic("UNIMPLEMENTED"),
.f32_add => @panic("UNIMPLEMENTED"),
.f32_sub => @panic("UNIMPLEMENTED"),
.f32_mul => @panic("UNIMPLEMENTED"),
.f32_div => @panic("UNIMPLEMENTED"),
.f32_min => @panic("UNIMPLEMENTED"),
.f32_max => @panic("UNIMPLEMENTED"),
.f32_copysign => @panic("UNIMPLEMENTED"),
.f64_abs => @panic("UNIMPLEMENTED"),
.f64_neg => @panic("UNIMPLEMENTED"),
.f64_ceil => @panic("UNIMPLEMENTED"),
.f64_floor => @panic("UNIMPLEMENTED"),
.f64_trunc => @panic("UNIMPLEMENTED"),
.f64_nearest => @panic("UNIMPLEMENTED"),
.f64_sqrt => @panic("UNIMPLEMENTED"),
.f64_add => @panic("UNIMPLEMENTED"),
.f64_sub => @panic("UNIMPLEMENTED"),
.f64_mul => @panic("UNIMPLEMENTED"),
.f64_div => @panic("UNIMPLEMENTED"),
.f64_min => @panic("UNIMPLEMENTED"),
.f64_max => @panic("UNIMPLEMENTED"),
.f64_copysign => @panic("UNIMPLEMENTED"),
.i32_wrap_i64 => @panic("UNIMPLEMENTED"),
.i32_trunc_f32_s => @panic("UNIMPLEMENTED"),
.i32_trunc_f32_u => @panic("UNIMPLEMENTED"),
.i32_trunc_f64_s => @panic("UNIMPLEMENTED"),
.i32_trunc_f64_u => @panic("UNIMPLEMENTED"),
.i64_extend_i32_s => @panic("UNIMPLEMENTED"),
.i64_extend_i32_u => {
try self.stack.append(.{ .i64 = self.stack.pop().?.i32 });
},
.i64_trunc_f32_s => @panic("UNIMPLEMENTED"),
.i64_trunc_f32_u => @panic("UNIMPLEMENTED"),
.i64_trunc_f64_s => @panic("UNIMPLEMENTED"),
.i64_trunc_f64_u => @panic("UNIMPLEMENTED"),
.f32_convert_i32_s => @panic("UNIMPLEMENTED"),
.f32_convert_i32_u => @panic("UNIMPLEMENTED"),
.f32_convert_i64_s => @panic("UNIMPLEMENTED"),
.f32_convert_i64_u => @panic("UNIMPLEMENTED"),
.f32_demote_f64 => @panic("UNIMPLEMENTED"),
.f64_convert_i32_s => @panic("UNIMPLEMENTED"),
.f64_convert_i32_u => @panic("UNIMPLEMENTED"),
.f64_convert_i64_s => @panic("UNIMPLEMENTED"),
.f64_convert_i64_u => @panic("UNIMPLEMENTED"),
.f64_promote_f32 => @panic("UNIMPLEMENTED"),
.i32_reinterpret_f32 => @panic("UNIMPLEMENTED"),
.i64_reinterpret_f64 => @panic("UNIMPLEMENTED"),
.f32_reinterpret_i32 => @panic("UNIMPLEMENTED"),
.f64_reinterpret_i64 => @panic("UNIMPLEMENTED"),
.i32_extend8_s => @panic("UNIMPLEMENTED"),
.i32_extend16_s => @panic("UNIMPLEMENTED"),
.i64_extend8_s => @panic("UNIMPLEMENTED"),
.i64_extend16_s => @panic("UNIMPLEMENTED"),
.i64_extend32_s => @panic("UNIMPLEMENTED"),
.i32_trunc_sat_f32_s => @panic("UNIMPLEMENTED"),
.i32_trunc_sat_f32_u => @panic("UNIMPLEMENTED"),
.i32_trunc_sat_f64_s => @panic("UNIMPLEMENTED"),
.i32_trunc_sat_f64_u => @panic("UNIMPLEMENTED"),
.i64_trunc_sat_f32_s => @panic("UNIMPLEMENTED"),
.i64_trunc_sat_f32_u => @panic("UNIMPLEMENTED"),
.i64_trunc_sat_f64_s => @panic("UNIMPLEMENTED"),
.i64_trunc_sat_f64_u => @panic("UNIMPLEMENTED"),
.vecinst => @panic("UNIMPLEMENTED"),
}
frame.program_counter += 1;
}
}
// TODO: Do name resolution at parseTime
pub fn callExternal(self: *Runtime, allocator: Allocator, name: ExportFunction, parameters: []Value) !void {
switch (name) {
.preinit => {
if (self.module.exports.preinit) |func| {
try self.call(allocator, func, parameters);
} else {
std.debug.panic("Function preinit unavailable\n", .{});
}
},
else => {
std.debug.panic("Function {any} not handled\n", .{name});
},
}
}
pub fn call(self: *Runtime, allocator: Allocator, function: usize, parameters: []Value) AllocationError!void {
const f = self.module.functions[function];
switch (f.typ) {
.internal => {
const ir: IR = f.typ.internal.ir;
const function_type = f.func_type;
var frame = CallFrame{
.code = ir,
.program_counter = 0x0,
.locals = try allocator.alloc(Value, f.typ.internal.locals.len + function_type.parameters.len),
};
@memcpy(frame.locals[0..parameters.len], parameters);
for (f.typ.internal.locals, function_type.parameters.len..) |local, i| {
switch (local) {
.val => |v| switch (v) {
.i32 => {
frame.locals[i] = .{ .i32 = 0 };
},
.i64 => {
frame.locals[i] = .{ .i64 = 0 };
},
else => unreachable,
},
.ref => unreachable,
}
}
try self.executeFrame(allocator, &frame);
allocator.free(frame.locals);
},
.external => {
std.debug.panic("TODO: Handle external function {any}\n", .{function});
// TODO(ernesto): handle external functions
// const name = self.module.imports[f.external].name;
// if (self.global_runtime.functions.get(name)) |external| {
// external(&self.stack);
// }
},
}
}
};
pub fn handleGlobalInit(allocator: Allocator, ir: IR) !Value {
var instruction_pointer: usize = 0;
var stack = try std.ArrayList(Value).initCapacity(allocator, 10);
defer stack.deinit();
while (instruction_pointer < ir.opcodes.len) {
const opcode: IR.Opcode = ir.opcodes[instruction_pointer];
const index = ir.indices[instruction_pointer];
switch (opcode) {
.i32_const => try stack.append(Value{ .i32 = index.i32 }),
else => {
std.debug.panic("TODO: Handle opcode {any}\n", .{opcode});
},
}
instruction_pointer += 1;
}
if (stack.items.len != 1) {
std.debug.panic("Improper amount of variables at end\n", .{});
}
return stack.pop().?;
}