mirror/v
1
0
Fork 0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
Code Issues Projects Releases Wiki Activity

rand: fix edge case, when bit length is 31 and 63, add tests for rand.intn(2147483647)! etc (#18714)

Browse Source
This commit is contained in:
Subhomoy Haldar
2023-06-29 20:11:27 +01:00
committed by GitHub
parent cf323cd0ef
commit 7ab11097be
2 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
vlib/rand/rand.v
View File

@ -55,7 +55,7 @@ pub fn (mut rng PRNG) u32n(max u32) !u32 {
// the closest power of two. Then we loop until we find
// an int in the required range
bit_len := bits.len_32(max)
if bit_len == 32 {
if _unlikely_(bit_len == 32) {
for {
value := rng.u32()
if value < max {
@ -63,7 +63,11 @@ pub fn (mut rng PRNG) u32n(max u32) !u32 {
}
}
} else {
mask := (u32(1) << (bit_len + 1)) - 1
mask := if _unlikely_(bit_len == 31) {
u32(0x7FFFFFFF)
} else {
(u32(1) << (bit_len + 1)) - 1
}
for {
value := rng.u32() & mask
if value < max {
@ -81,7 +85,7 @@ pub fn (mut rng PRNG) u64n(max u64) !u64 {
return error('max must be positive integer')
}
bit_len := bits.len_64(max)
if bit_len == 64 {
if _unlikely_(bit_len == 64) {
for {
value := rng.u64()
if value < max {
@ -89,7 +93,11 @@ pub fn (mut rng PRNG) u64n(max u64) !u64 {
}
}
} else {
mask := (u64(1) << (bit_len + 1)) - 1
mask := if _unlikely_(bit_len == 63) {
u64(0x7FFFFFFFFFFFFFFF)
} else {
(u64(1) << (bit_len + 1)) - 1
}
for {
value := rng.u64() & mask
if value < max {