mirror of
https://github.com/roc-lang/roc.git
synced 2025-07-24 15:03:46 +00:00
106 lines
No EOL
3.5 KiB
Zig
106 lines
No EOL
3.5 KiB
Zig
const std = @import("std");
|
|
const builtin = @import("builtin");
|
|
const str = @import("str");
|
|
const RocStr = str.RocStr;
|
|
const testing = std.testing;
|
|
const expectEqual = testing.expectEqual;
|
|
const expect = testing.expect;
|
|
|
|
comptime {
|
|
// This is a workaround for https://github.com/ziglang/zig/issues/8218
|
|
// which is only necessary on macOS.
|
|
//
|
|
// Once that issue is fixed, we can undo the changes in
|
|
// 177cf12e0555147faa4d436e52fc15175c2c4ff0 and go back to passing
|
|
// -fcompiler-rt in link.rs instead of doing this. Note that this
|
|
// workaround is present in many host.zig files, so make sure to undo
|
|
// it everywhere!
|
|
if (builtin.os.tag == .macos) {
|
|
_ = @import("compiler_rt");
|
|
}
|
|
}
|
|
|
|
const Align = 2 * @alignOf(usize);
|
|
extern fn malloc(size: usize) callconv(.C) ?*align(Align) anyopaque;
|
|
extern fn realloc(c_ptr: [*]align(Align) u8, size: usize) callconv(.C) ?*anyopaque;
|
|
extern fn free(c_ptr: [*]align(Align) u8) callconv(.C) void;
|
|
extern fn memcpy(dst: [*]u8, src: [*]u8, size: usize) callconv(.C) void;
|
|
extern fn memset(dst: [*]u8, value: i32, size: usize) callconv(.C) void;
|
|
|
|
const DEBUG: bool = false;
|
|
|
|
export fn roc_alloc(size: usize, alignment: u32) callconv(.C) ?*anyopaque {
|
|
if (DEBUG) {
|
|
var ptr = malloc(size);
|
|
const stdout = std.io.getStdOut().writer();
|
|
stdout.print("alloc: {d} (alignment {d}, size {d})\n", .{ ptr, alignment, size }) catch unreachable;
|
|
return ptr;
|
|
} else {
|
|
return malloc(size);
|
|
}
|
|
}
|
|
|
|
export fn roc_realloc(c_ptr: *anyopaque, new_size: usize, old_size: usize, alignment: u32) callconv(.C) ?*anyopaque {
|
|
if (DEBUG) {
|
|
const stdout = std.io.getStdOut().writer();
|
|
stdout.print("realloc: {d} (alignment {d}, old_size {d})\n", .{ c_ptr, alignment, old_size }) catch unreachable;
|
|
}
|
|
|
|
return realloc(@alignCast(Align, @ptrCast([*]u8, c_ptr)), new_size);
|
|
}
|
|
|
|
export fn roc_dealloc(c_ptr: *anyopaque, alignment: u32) callconv(.C) void {
|
|
if (DEBUG) {
|
|
const stdout = std.io.getStdOut().writer();
|
|
stdout.print("dealloc: {d} (alignment {d})\n", .{ c_ptr, alignment }) catch unreachable;
|
|
}
|
|
|
|
free(@alignCast(Align, @ptrCast([*]u8, c_ptr)));
|
|
}
|
|
|
|
export fn roc_panic(c_ptr: *anyopaque, tag_id: u32) callconv(.C) void {
|
|
_ = tag_id;
|
|
|
|
const stderr = std.io.getStdErr().writer();
|
|
const msg = @ptrCast([*:0]const u8, c_ptr);
|
|
stderr.print("Application crashed with message\n\n {s}\n\nShutting down\n", .{msg}) catch unreachable;
|
|
std.process.exit(0);
|
|
}
|
|
|
|
export fn roc_memcpy(dst: [*]u8, src: [*]u8, size: usize) callconv(.C) void {
|
|
return memcpy(dst, src, size);
|
|
}
|
|
|
|
export fn roc_memset(dst: [*]u8, value: i32, size: usize) callconv(.C) void {
|
|
return memset(dst, value, size);
|
|
}
|
|
|
|
const mem = std.mem;
|
|
const Allocator = mem.Allocator;
|
|
|
|
extern fn roc__mainForHost_1_exposed_generic(*RocStr) void;
|
|
|
|
const Unit = extern struct {};
|
|
|
|
pub fn main() u8 {
|
|
const stdout = std.io.getStdOut().writer();
|
|
const stderr = std.io.getStdErr().writer();
|
|
|
|
var timer = std.time.Timer.start() catch unreachable;
|
|
|
|
// actually call roc to populate the callresult
|
|
var callresult = RocStr.empty();
|
|
roc__mainForHost_1_exposed_generic(&callresult);
|
|
|
|
const nanos = timer.read();
|
|
const seconds = (@intToFloat(f64, nanos) / 1_000_000_000.0);
|
|
|
|
// stdout the result
|
|
stdout.print("{s}", .{callresult.asSlice()}) catch unreachable;
|
|
|
|
callresult.deinit();
|
|
|
|
stderr.print("runtime: {d:.3}ms\n", .{seconds * 1000}) catch unreachable;
|
|
|
|
return 0;
|
|
} |