mirror of
https://github.com/roc-lang/roc.git
synced 2025-09-28 22:34:45 +00:00
Have examples with zig hosts use roc_alloc etc
This commit is contained in:
parent
2a68c0c2cd
commit
175115ee4e
5 changed files with 85 additions and 8 deletions
|
@ -29,6 +29,22 @@ extern fn roc__mainForHost_1_Fx_caller(*const u8, *const u8, [*]u8, [*]u8) void;
|
||||||
extern fn roc__mainForHost_1_Fx_size() i64;
|
extern fn roc__mainForHost_1_Fx_size() i64;
|
||||||
extern fn roc__mainForHost_1_Fx_result_size() i64;
|
extern fn roc__mainForHost_1_Fx_result_size() i64;
|
||||||
|
|
||||||
|
extern fn malloc(size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
|
||||||
|
|
||||||
|
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
|
||||||
|
return malloc(size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
|
||||||
|
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
|
||||||
|
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
|
||||||
|
}
|
||||||
|
|
||||||
const Unit = extern struct {};
|
const Unit = extern struct {};
|
||||||
|
|
||||||
pub export fn main() u8 {
|
pub export fn main() u8 {
|
||||||
|
|
|
@ -28,6 +28,10 @@ extern fn roc__mainForHost_1_Fx_caller(*const u8, *const u8, [*]u8, [*]u8) void;
|
||||||
extern fn roc__mainForHost_1_Fx_size() i64;
|
extern fn roc__mainForHost_1_Fx_size() i64;
|
||||||
extern fn roc__mainForHost_1_Fx_result_size() i64;
|
extern fn roc__mainForHost_1_Fx_result_size() i64;
|
||||||
|
|
||||||
|
extern fn malloc(size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
|
||||||
|
|
||||||
const Unit = extern struct {};
|
const Unit = extern struct {};
|
||||||
|
|
||||||
pub export fn main() u8 {
|
pub export fn main() u8 {
|
||||||
|
@ -60,28 +64,33 @@ pub export fn main() u8 {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
pub export fn malloc(size: usize) callconv(.C) ?*c_void {
|
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
|
||||||
const stdout = std.io.getStdOut().writer();
|
const stdout = std.io.getStdOut().writer();
|
||||||
const allocator = testing.allocator;
|
const allocator = testing.allocator;
|
||||||
|
|
||||||
// Perform the actual malloc
|
// Perform the actual malloc
|
||||||
const startNs = std.time.nanoTimestamp();
|
const startNs = std.time.nanoTimestamp();
|
||||||
const ptr = allocator.alignedAlloc(u8, 16, size) catch unreachable;
|
const ptr = malloc(size) orelse unreachable;
|
||||||
const endNs = std.time.nanoTimestamp();
|
const endNs = std.time.nanoTimestamp();
|
||||||
|
|
||||||
const totalMs = @divTrunc(endNs - startNs, 1000);
|
const totalMs = @divTrunc(endNs - startNs, 1000);
|
||||||
|
|
||||||
stdout.print("\x1B[36m{} | \x1B[39m Custom malloc allocated {} bytes in {} ms!\n", .{startNs, size, totalMs}) catch unreachable;
|
stdout.print("\x1B[36m{} | \x1B[39m Custom malloc allocated {} bytes in {} ms!\n", .{startNs, size, totalMs}) catch unreachable;
|
||||||
|
|
||||||
return @ptrCast(?*c_void, ptr);
|
return ptr;
|
||||||
}
|
}
|
||||||
pub export fn free(c_ptr: *u128) callconv(.C) void {
|
|
||||||
|
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
|
||||||
|
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
|
||||||
const stdout = std.io.getStdOut().writer();
|
const stdout = std.io.getStdOut().writer();
|
||||||
const allocator = testing.allocator;
|
const allocator = testing.allocator;
|
||||||
|
|
||||||
// Perform the actual free
|
// Perform the actual free
|
||||||
const startNs = std.time.nanoTimestamp();
|
const startNs = std.time.nanoTimestamp();
|
||||||
allocator.destroy(c_ptr);
|
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
|
||||||
const endNs = std.time.nanoTimestamp();
|
const endNs = std.time.nanoTimestamp();
|
||||||
|
|
||||||
const totalMs = @divTrunc(endNs - startNs, 1000);
|
const totalMs = @divTrunc(endNs - startNs, 1000);
|
||||||
|
@ -149,7 +158,7 @@ pub export fn roc_fx_readAllUtf8(rocPath: RocStr) callconv(.C) ReadResult {
|
||||||
};
|
};
|
||||||
|
|
||||||
var str_ptr = @ptrCast([*]u8, content);
|
var str_ptr = @ptrCast([*]u8, content);
|
||||||
var roc_str3 = RocStr.init(testing.allocator, str_ptr, content.len);
|
var roc_str3 = RocStr.init(str_ptr, content.len);
|
||||||
|
|
||||||
return .{ .bytes = roc_str3, .errno = 0 };
|
return .{ .bytes = roc_str3, .errno = 0 };
|
||||||
}
|
}
|
||||||
|
|
|
@ -19,6 +19,24 @@ comptime {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
extern fn malloc(size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
|
||||||
|
|
||||||
|
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
|
||||||
|
// TODO orelse cxa_throw instead of orelse unreachable
|
||||||
|
return malloc(size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
|
||||||
|
// TODO orelse cxa_throw instead of orelse unreachable
|
||||||
|
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
|
||||||
|
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
|
||||||
|
}
|
||||||
|
|
||||||
const mem = std.mem;
|
const mem = std.mem;
|
||||||
const Allocator = mem.Allocator;
|
const Allocator = mem.Allocator;
|
||||||
|
|
||||||
|
@ -45,7 +63,7 @@ pub export fn main() i32 {
|
||||||
// stdout the result
|
// stdout the result
|
||||||
stdout.print("{}\n", .{callresult.content.asSlice()}) catch unreachable;
|
stdout.print("{}\n", .{callresult.content.asSlice()}) catch unreachable;
|
||||||
|
|
||||||
callresult.content.deinit(std.heap.c_allocator);
|
callresult.content.deinit();
|
||||||
|
|
||||||
// end time
|
// end time
|
||||||
var ts2: std.os.timespec = undefined;
|
var ts2: std.os.timespec = undefined;
|
||||||
|
|
|
@ -22,6 +22,23 @@ const Allocator = mem.Allocator;
|
||||||
|
|
||||||
extern fn roc__mainForHost_1_exposed(RocList, *RocCallResult) void;
|
extern fn roc__mainForHost_1_exposed(RocList, *RocCallResult) void;
|
||||||
|
|
||||||
|
extern fn malloc(size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
|
||||||
|
|
||||||
|
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
|
||||||
|
return malloc(size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
|
||||||
|
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
|
||||||
|
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
// warning! the array is currently stack-allocated so don't make this too big
|
// warning! the array is currently stack-allocated so don't make this too big
|
||||||
const NUM_NUMS = 100;
|
const NUM_NUMS = 100;
|
||||||
|
|
||||||
|
|
|
@ -28,6 +28,23 @@ extern fn roc__mainForHost_1_Fx_caller(*const u8, *const u8, [*]u8, [*]u8) void;
|
||||||
extern fn roc__mainForHost_1_Fx_size() i64;
|
extern fn roc__mainForHost_1_Fx_size() i64;
|
||||||
extern fn roc__mainForHost_1_Fx_result_size() i64;
|
extern fn roc__mainForHost_1_Fx_result_size() i64;
|
||||||
|
|
||||||
|
extern fn malloc(size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn realloc(c_ptr: [*]align(@alignOf(u128)) u8, size: usize) callconv(.C) ?*c_void;
|
||||||
|
extern fn free(c_ptr: [*]align(@alignOf(u128)) u8) callconv(.C) void;
|
||||||
|
|
||||||
|
export fn roc_alloc(alignment: usize, size: usize) callconv(.C) *c_void {
|
||||||
|
return malloc(size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_realloc(alignment: usize, c_ptr: *c_void, old_size: usize, new_size: usize) callconv(.C) *c_void {
|
||||||
|
return realloc(@alignCast(16, @ptrCast([*]u8, c_ptr)), new_size) orelse unreachable;
|
||||||
|
}
|
||||||
|
|
||||||
|
export fn roc_dealloc(alignment: usize, c_ptr: *c_void) callconv(.C) void {
|
||||||
|
free(@alignCast(16, @ptrCast([*]u8, c_ptr)));
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
const Unit = extern struct {};
|
const Unit = extern struct {};
|
||||||
|
|
||||||
pub export fn main() u8 {
|
pub export fn main() u8 {
|
||||||
|
@ -120,7 +137,7 @@ pub export fn roc_fx_readAllUtf8(rocPath: RocStr) callconv(.C) ReadResult {
|
||||||
};
|
};
|
||||||
|
|
||||||
var str_ptr = @ptrCast([*]u8, content);
|
var str_ptr = @ptrCast([*]u8, content);
|
||||||
var roc_str3 = RocStr.init(testing.allocator, str_ptr, content.len);
|
var roc_str3 = RocStr.init(str_ptr, content.len);
|
||||||
|
|
||||||
return .{ .bytes = roc_str3, .errno = 0 };
|
return .{ .bytes = roc_str3, .errno = 0 };
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue