265 lines
7.1 KiB
Go
265 lines
7.1 KiB
Go
|
package xxh3
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/bits"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Hash128Seed returns the 128-bit hash of the byte slice.
|
||
|
|
func Hash128Seed(b []byte, seed uint64) Uint128 {
|
||
|
|
return hashAny128Seed(*(*str)(ptr(&b)), seed)
|
||
|
|
}
|
||
|
|
|
||
|
|
// HashString128Seed returns the 128-bit hash of the string slice.
|
||
|
|
func HashString128Seed(s string, seed uint64) Uint128 {
|
||
|
|
return hashAny128Seed(*(*str)(ptr(&s)), seed)
|
||
|
|
}
|
||
|
|
|
||
|
|
func hashAny128Seed(s str, seed uint64) (acc u128) {
|
||
|
|
p, l := s.p, s.l
|
||
|
|
|
||
|
|
switch {
|
||
|
|
case l <= 16:
|
||
|
|
switch {
|
||
|
|
case l > 8: // 9-16
|
||
|
|
bitflipl := (key64_032 ^ key64_040) - seed
|
||
|
|
bitfliph := (key64_048 ^ key64_056) + seed
|
||
|
|
|
||
|
|
input_lo := readU64(p, 0)
|
||
|
|
input_hi := readU64(p, ui(l)-8)
|
||
|
|
|
||
|
|
m128_h, m128_l := bits.Mul64(input_lo^input_hi^bitflipl, prime64_1)
|
||
|
|
|
||
|
|
m128_l += uint64(l-1) << 54
|
||
|
|
input_hi ^= bitfliph
|
||
|
|
|
||
|
|
m128_h += input_hi + uint64(uint32(input_hi))*(prime32_2-1)
|
||
|
|
|
||
|
|
m128_l ^= bits.ReverseBytes64(m128_h)
|
||
|
|
|
||
|
|
acc.Hi, acc.Lo = bits.Mul64(m128_l, prime64_2)
|
||
|
|
acc.Hi += m128_h * prime64_2
|
||
|
|
|
||
|
|
acc.Lo = xxh3Avalanche(acc.Lo)
|
||
|
|
acc.Hi = xxh3Avalanche(acc.Hi)
|
||
|
|
|
||
|
|
return acc
|
||
|
|
|
||
|
|
case l > 3: // 4-8
|
||
|
|
seed ^= u64(bits.ReverseBytes32(u32(seed))) << 32
|
||
|
|
bitflip := (key64_016 ^ key64_024) + seed
|
||
|
|
input_lo := readU32(p, 0)
|
||
|
|
input_hi := readU32(p, ui(l)-4)
|
||
|
|
input_64 := u64(input_lo) + u64(input_hi)<<32
|
||
|
|
keyed := input_64 ^ bitflip
|
||
|
|
|
||
|
|
acc.Hi, acc.Lo = bits.Mul64(keyed, prime64_1+(uint64(l)<<2))
|
||
|
|
|
||
|
|
acc.Hi += acc.Lo << 1
|
||
|
|
acc.Lo ^= acc.Hi >> 3
|
||
|
|
|
||
|
|
acc.Lo ^= acc.Lo >> 35
|
||
|
|
acc.Lo *= 0x9fb21c651e98df25
|
||
|
|
acc.Lo ^= acc.Lo >> 28
|
||
|
|
acc.Hi = xxh3Avalanche(acc.Hi)
|
||
|
|
|
||
|
|
return acc
|
||
|
|
|
||
|
|
case l == 3: // 3
|
||
|
|
c12 := u64(readU16(p, 0))
|
||
|
|
c3 := u64(readU8(p, 2))
|
||
|
|
acc.Lo = c12<<16 + c3 + 3<<8
|
||
|
|
|
||
|
|
case l > 1: // 2
|
||
|
|
c12 := u64(readU16(p, 0))
|
||
|
|
acc.Lo = c12*(1<<24+1)>>8 + 2<<8
|
||
|
|
|
||
|
|
case l == 1: // 1
|
||
|
|
c1 := u64(readU8(p, 0))
|
||
|
|
acc.Lo = c1*(1<<24+1<<16+1) + 1<<8
|
||
|
|
|
||
|
|
default: // 0
|
||
|
|
bitflipl := key64_064 ^ key64_072 ^ seed
|
||
|
|
bitfliph := key64_080 ^ key64_088 ^ seed
|
||
|
|
return u128{Lo: xxh64AvalancheFull(bitflipl), Hi: xxh64AvalancheFull(bitfliph)}
|
||
|
|
}
|
||
|
|
|
||
|
|
acc.Hi = uint64(bits.RotateLeft32(bits.ReverseBytes32(uint32(acc.Lo)), 13))
|
||
|
|
acc.Lo ^= uint64(key32_000^key32_004) + seed
|
||
|
|
acc.Hi ^= uint64(key32_008^key32_012) - seed
|
||
|
|
|
||
|
|
acc.Lo = xxh64AvalancheFull(acc.Lo)
|
||
|
|
acc.Hi = xxh64AvalancheFull(acc.Hi)
|
||
|
|
|
||
|
|
return acc
|
||
|
|
|
||
|
|
case l <= 128:
|
||
|
|
acc.Lo = u64(l) * prime64_1
|
||
|
|
|
||
|
|
if l > 32 {
|
||
|
|
if l > 64 {
|
||
|
|
if l > 96 {
|
||
|
|
in8, in7 := readU64(p, ui(l)-8*8), readU64(p, ui(l)-7*8)
|
||
|
|
i6, i7 := readU64(p, 6*8), readU64(p, 7*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(in8^(key64_112+seed), in7^(key64_120-seed))
|
||
|
|
acc.Hi ^= i6 + i7
|
||
|
|
acc.Lo += mulFold64(i6^(key64_096+seed), i7^(key64_104-seed))
|
||
|
|
acc.Lo ^= in8 + in7
|
||
|
|
|
||
|
|
} // 96
|
||
|
|
|
||
|
|
in6, in5 := readU64(p, ui(l)-6*8), readU64(p, ui(l)-5*8)
|
||
|
|
i4, i5 := readU64(p, 4*8), readU64(p, 5*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(in6^(key64_080+seed), in5^(key64_088-seed))
|
||
|
|
acc.Hi ^= i4 + i5
|
||
|
|
acc.Lo += mulFold64(i4^(key64_064+seed), i5^(key64_072-seed))
|
||
|
|
acc.Lo ^= in6 + in5
|
||
|
|
|
||
|
|
} // 64
|
||
|
|
|
||
|
|
in4, in3 := readU64(p, ui(l)-4*8), readU64(p, ui(l)-3*8)
|
||
|
|
i2, i3 := readU64(p, 2*8), readU64(p, 3*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(in4^(key64_048+seed), in3^(key64_056-seed))
|
||
|
|
acc.Hi ^= i2 + i3
|
||
|
|
acc.Lo += mulFold64(i2^(key64_032+seed), i3^(key64_040-seed))
|
||
|
|
acc.Lo ^= in4 + in3
|
||
|
|
|
||
|
|
} // 32
|
||
|
|
|
||
|
|
in2, in1 := readU64(p, ui(l)-2*8), readU64(p, ui(l)-1*8)
|
||
|
|
i0, i1 := readU64(p, 0*8), readU64(p, 1*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(in2^(key64_016+seed), in1^(key64_024-seed))
|
||
|
|
acc.Hi ^= i0 + i1
|
||
|
|
acc.Lo += mulFold64(i0^(key64_000+seed), i1^(key64_008-seed))
|
||
|
|
acc.Lo ^= in2 + in1
|
||
|
|
|
||
|
|
acc.Hi, acc.Lo = (acc.Lo*prime64_1)+(acc.Hi*prime64_4)+((u64(l)-seed)*prime64_2), acc.Hi+acc.Lo
|
||
|
|
|
||
|
|
acc.Hi = -xxh3Avalanche(acc.Hi)
|
||
|
|
acc.Lo = xxh3Avalanche(acc.Lo)
|
||
|
|
|
||
|
|
return acc
|
||
|
|
|
||
|
|
case l <= 240:
|
||
|
|
acc.Lo = u64(l) * prime64_1
|
||
|
|
|
||
|
|
{
|
||
|
|
i0, i1, i2, i3 := readU64(p, 0*8), readU64(p, 1*8), readU64(p, 2*8), readU64(p, 3*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(i2^(key64_016+seed), i3^(key64_024-seed))
|
||
|
|
acc.Hi ^= i0 + i1
|
||
|
|
acc.Lo += mulFold64(i0^(key64_000+seed), i1^(key64_008-seed))
|
||
|
|
acc.Lo ^= i2 + i3
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
i0, i1, i2, i3 := readU64(p, 4*8), readU64(p, 5*8), readU64(p, 6*8), readU64(p, 7*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(i2^(key64_048+seed), i3^(key64_056-seed))
|
||
|
|
acc.Hi ^= i0 + i1
|
||
|
|
acc.Lo += mulFold64(i0^(key64_032+seed), i1^(key64_040-seed))
|
||
|
|
acc.Lo ^= i2 + i3
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
i0, i1, i2, i3 := readU64(p, 8*8), readU64(p, 9*8), readU64(p, 10*8), readU64(p, 11*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(i2^(key64_080+seed), i3^(key64_088-seed))
|
||
|
|
acc.Hi ^= i0 + i1
|
||
|
|
acc.Lo += mulFold64(i0^(key64_064+seed), i1^(key64_072-seed))
|
||
|
|
acc.Lo ^= i2 + i3
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
i0, i1, i2, i3 := readU64(p, 12*8), readU64(p, 13*8), readU64(p, 14*8), readU64(p, 15*8)
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(i2^(key64_112+seed), i3^(key64_120-seed))
|
||
|
|
acc.Hi ^= i0 + i1
|
||
|
|
acc.Lo += mulFold64(i0^(key64_096+seed), i1^(key64_104-seed))
|
||
|
|
acc.Lo ^= i2 + i3
|
||
|
|
}
|
||
|
|
|
||
|
|
// avalanche
|
||
|
|
acc.Hi = xxh3Avalanche(acc.Hi)
|
||
|
|
acc.Lo = xxh3Avalanche(acc.Lo)
|
||
|
|
|
||
|
|
// trailing groups after 128
|
||
|
|
top := ui(l) &^ 31
|
||
|
|
for i := ui(4 * 32); i < top; i += 32 {
|
||
|
|
i0, i1, i2, i3 := readU64(p, i+0), readU64(p, i+8), readU64(p, i+16), readU64(p, i+24)
|
||
|
|
k0, k1, k2, k3 := readU64(key, i-125)+seed, readU64(key, i-117)-seed, readU64(key, i-109)+seed, readU64(key, i-101)-seed
|
||
|
|
|
||
|
|
acc.Hi += mulFold64(i2^k2, i3^k3)
|
||
|
|
acc.Hi ^= i0 + i1
|
||
|
|
acc.Lo += mulFold64(i0^k0, i1^k1)
|
||
|
|
acc.Lo ^= i2 + i3
|
||
|
|
}
|
||
|
|
|
||
|
|
// last 32 bytes
|
||
|
|
{
|
||
|
|
i0, i1, i2, i3 := readU64(p, ui(l)-32), readU64(p, ui(l)-24), readU64(p, ui(l)-16), readU64(p, ui(l)-8)
|
||
|
|
|
||
|
|
seed := 0 - seed
|
||
|
|
acc.Hi += mulFold64(i0^(key64_119+seed), i1^(key64_127-seed))
|
||
|
|
acc.Hi ^= i2 + i3
|
||
|
|
acc.Lo += mulFold64(i2^(key64_103+seed), i3^(key64_111-seed))
|
||
|
|
acc.Lo ^= i0 + i1
|
||
|
|
}
|
||
|
|
|
||
|
|
acc.Hi, acc.Lo = (acc.Lo*prime64_1)+(acc.Hi*prime64_4)+((u64(l)-seed)*prime64_2), acc.Hi+acc.Lo
|
||
|
|
|
||
|
|
acc.Hi = -xxh3Avalanche(acc.Hi)
|
||
|
|
acc.Lo = xxh3Avalanche(acc.Lo)
|
||
|
|
|
||
|
|
return acc
|
||
|
|
|
||
|
|
default:
|
||
|
|
acc.Lo = u64(l) * prime64_1
|
||
|
|
acc.Hi = ^(u64(l) * prime64_2)
|
||
|
|
|
||
|
|
secret := key
|
||
|
|
if seed != 0 {
|
||
|
|
secret = ptr(&[secretSize]byte{})
|
||
|
|
initSecret(secret, seed)
|
||
|
|
}
|
||
|
|
|
||
|
|
accs := [8]u64{
|
||
|
|
prime32_3, prime64_1, prime64_2, prime64_3,
|
||
|
|
prime64_4, prime32_2, prime64_5, prime32_1,
|
||
|
|
}
|
||
|
|
|
||
|
|
if hasAVX512 && l >= avx512Switch {
|
||
|
|
accumAVX512(&accs, p, secret, u64(l))
|
||
|
|
} else if hasAVX2 {
|
||
|
|
accumAVX2(&accs, p, secret, u64(l))
|
||
|
|
} else if hasSSE2 {
|
||
|
|
accumSSE(&accs, p, secret, u64(l))
|
||
|
|
} else {
|
||
|
|
accumScalar(&accs, p, secret, u64(l))
|
||
|
|
}
|
||
|
|
|
||
|
|
// merge accs
|
||
|
|
const hi_off = 117 - 11
|
||
|
|
|
||
|
|
acc.Lo += mulFold64(accs[0]^readU64(secret, 11), accs[1]^readU64(secret, 19))
|
||
|
|
acc.Hi += mulFold64(accs[0]^readU64(secret, 11+hi_off), accs[1]^readU64(secret, 19+hi_off))
|
||
|
|
|
||
|
|
acc.Lo += mulFold64(accs[2]^readU64(secret, 27), accs[3]^readU64(secret, 35))
|
||
|
|
acc.Hi += mulFold64(accs[2]^readU64(secret, 27+hi_off), accs[3]^readU64(secret, 35+hi_off))
|
||
|
|
|
||
|
|
acc.Lo += mulFold64(accs[4]^readU64(secret, 43), accs[5]^readU64(secret, 51))
|
||
|
|
acc.Hi += mulFold64(accs[4]^readU64(secret, 43+hi_off), accs[5]^readU64(secret, 51+hi_off))
|
||
|
|
|
||
|
|
acc.Lo += mulFold64(accs[6]^readU64(secret, 59), accs[7]^readU64(secret, 67))
|
||
|
|
acc.Hi += mulFold64(accs[6]^readU64(secret, 59+hi_off), accs[7]^readU64(secret, 67+hi_off))
|
||
|
|
|
||
|
|
acc.Lo = xxh3Avalanche(acc.Lo)
|
||
|
|
acc.Hi = xxh3Avalanche(acc.Hi)
|
||
|
|
|
||
|
|
return acc
|
||
|
|
}
|
||
|
|
}
|