1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
package native
import (
"unsafe"
`github.com/klauspost/cpuid/v2`
"github.com/cloudwego/base64x/internal/rt"
"github.com/cloudwego/base64x/internal/native/avx2"
"github.com/cloudwego/base64x/internal/native/sse"
)
var (
hasAVX2 = cpuid.CPU.Has(cpuid.AVX2)
hasSSE = cpuid.CPU.Has(cpuid.SSE)
)
var (
S_b64decode uintptr
S_b64encode uintptr
)
var (
F_b64decode func(out unsafe.Pointer, src unsafe.Pointer, len int, mod int) (ret int)
F_b64encode func(out unsafe.Pointer, src unsafe.Pointer, mod int)
)
func useAVX2() {
avx2.Use()
S_b64decode = avx2.S_b64decode
S_b64encode = avx2.S_b64encode
F_b64decode = avx2.F_b64decode
F_b64encode = avx2.F_b64encode
}
func useSSE() {
sse.Use()
S_b64decode = sse.S_b64decode
S_b64encode = sse.S_b64encode
F_b64decode = sse.F_b64decode
F_b64encode = sse.F_b64encode
}
//go:nosplit
func B64Decode(out *[]byte, src unsafe.Pointer, len int, mod int) (ret int) {
return F_b64decode(rt.NoEscape(unsafe.Pointer(out)), rt.NoEscape(unsafe.Pointer(src)), len, mod)
}
//go:nosplit
func B64Encode(out *[]byte, src *[]byte, mod int) {
F_b64encode(rt.NoEscape(unsafe.Pointer(out)), rt.NoEscape(unsafe.Pointer(src)), mod)
}
func init() {
if hasAVX2 {
useAVX2()
} else if hasSSE {
useSSE()
} else {
panic("Unsupported CPU, lacks of AVX2 or SSE CPUID Flag. maybe it's too old to run Sonic.")
}
}
|
