diff options
| author | Colin Davidson <colrdavidson@gmail.com> | 2022-08-29 02:03:12 -0700 |
|---|---|---|
| committer | Colin Davidson <colrdavidson@gmail.com> | 2022-08-29 02:03:12 -0700 |
| commit | 7a6fc3a93bf1be42de21a1af7af6e9e435438832 (patch) | |
| tree | d5e75cedc3700d252fd68c5b8448888e460c1c1a /core/container/queue/queue.odin | |
| parent | 83c002c197301bffdd2140966e98c89c4529b1b6 (diff) | |
Add bounds check for peeks
Diffstat (limited to 'core/container/queue/queue.odin')
| -rw-r--r-- | core/container/queue/queue.odin | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/core/container/queue/queue.odin b/core/container/queue/queue.odin index 9674b40be..b3a8ad43f 100644 --- a/core/container/queue/queue.odin +++ b/core/container/queue/queue.odin @@ -99,12 +99,14 @@ get_ptr :: proc(q: ^$Q/Queue($T), #any_int i: int, loc := #caller_location) -> ^ return &q.data[idx] } -peek_front :: proc(q: ^$Q/Queue($T)) -> ^T { +peek_front :: proc(q: ^$Q/Queue($T), loc := #caller_location) -> ^T { + runtime.bounds_check_error_loc(loc, 0, builtin.len(q.data)) idx := q.offset%builtin.len(q.data) return &q.data[idx] } -peek_back :: proc(q: ^$Q/Queue($T)) -> ^T { +peek_back :: proc(q: ^$Q/Queue($T), loc := #caller_location) -> ^T { + runtime.bounds_check_error_loc(loc, int(q.len - 1), builtin.len(q.data)) idx := (uint(q.len - 1)+q.offset)%builtin.len(q.data) return &q.data[idx] } |