language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-01 15:51:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

add refcount check

Browse source
This commit is contained in:
Folkert 2021-02-14 00:42:42 +01:00
parent 27c3a7c8f2
commit c9e2958947
1 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
compiler/builtins/bitcode/src/dict.zig
View file

@ -7,6 +7,8 @@ const Allocator = mem.Allocator;
const level_size = 32; const level_size = 32;
const INITIAL_SEED = 0xc70f6907; const INITIAL_SEED = 0xc70f6907;
const REFCOUNT_ONE_ISIZE: comptime isize = std.math.minInt(isize);
const REFCOUNT_ONE: usize = @bitCast(usize, REFCOUNT_ONE_ISIZE);
const InPlace = packed enum(u8) { const InPlace = packed enum(u8) {
InPlace, InPlace,
@ -136,8 +138,7 @@ pub const RocDict = extern struct {
if (result_in_place == InPlace.InPlace) { if (result_in_place == InPlace.InPlace) {
as_usize_array[0] = @intCast(usize, number_of_slots); as_usize_array[0] = @intCast(usize, number_of_slots);
} else { } else {
const v: isize = std.math.minInt(isize); as_usize_array[0] = REFCOUNT_ONE;
as_usize_array[0] = @bitCast(usize, v);
} }
var as_u8_array = @ptrCast([*]u8, new_bytes); var as_u8_array = @ptrCast([*]u8, new_bytes);
@ -157,9 +158,8 @@ pub const RocDict = extern struct {
as_usize_array[0] = 0; as_usize_array[0] = 0;
as_usize_array[1] = @intCast(usize, number_of_slots); as_usize_array[1] = @intCast(usize, number_of_slots);
} else { } else {
const v: isize = std.math.minInt(isize);
as_usize_array[0] = 0; as_usize_array[0] = 0;
as_usize_array[1] = @bitCast(usize, v); as_usize_array[1] = REFCOUNT_ONE;
} }
var as_u8_array = @ptrCast([*]u8, new_bytes); var as_u8_array = @ptrCast([*]u8, new_bytes);
@ -197,8 +197,7 @@ pub const RocDict = extern struct {
var new_bytes: []align(8) u8 = allocator.alignedAlloc(u8, 8, length) catch unreachable; var new_bytes: []align(8) u8 = allocator.alignedAlloc(u8, 8, length) catch unreachable;
var as_usize_array = @ptrCast([*]usize, new_bytes); var as_usize_array = @ptrCast([*]usize, new_bytes);
const v: isize = std.math.minInt(isize); as_usize_array[0] = REFCOUNT_ONE;
as_usize_array[0] = @bitCast(usize, v);
var as_u8_array = @ptrCast([*]u8, new_bytes); var as_u8_array = @ptrCast([*]u8, new_bytes);
const first_slot = as_u8_array + @sizeOf(usize); const first_slot = as_u8_array + @sizeOf(usize);
@ -257,9 +256,16 @@ pub const RocDict = extern struct {
} }
pub fn refcountIsOne(self: RocDict) bool { pub fn refcountIsOne(self: RocDict) bool {
if (self.isEmpty()) {
// empty dicts are not refcounted at all
return false; return false;
} }
const ptr: [*]usize = @ptrCast([*]usize, @alignCast(8, self.dict_bytes));
return (ptr - 1)[0] == REFCOUNT_ONE;
}
pub fn capacity(self: RocDict) usize { pub fn capacity(self: RocDict) usize {
return total_slots_at_level(self.number_of_levels); return total_slots_at_level(self.number_of_levels);
} }