放弃 cgo

2026-06-24 05:18:32 +08:00 · 2021-10-28 14:02:36 +08:00
parent e989e44588
commit b764251f95
9 changed files with 257 additions and 69 deletions
--- a/base14.go
+++ b/base14.go
@@ -0,0 +1,153 @@
+// +build amd64 arm64 ppc64le mips64le 386 arm mipsle
+
+// Package base14 base16384 的 go 接口
+package base14
+
+import (
+	"unsafe"
+
+	"github.com/wdvxdr1123/ZeroBot/utils/helper"
+)
+
+func EncodeString(s string) []byte {
+	return Encode(helper.StringToBytes(s))
+}
+
+func DecodeString(s string) []byte {
+	return Decode(helper.StringToBytes(s))
+}
+
+func Encode(b []byte) (encd []byte) {
+	outlen := len(b) / 7 * 8
+	offset := len(b) % 7
+	switch offset { //算上偏移标志字符占用的2字节
+	case 0:
+		break
+	case 1:
+		outlen += 4
+	case 2, 3:
+		outlen += 6
+	case 4, 5:
+		outlen += 8
+	case 6:
+		outlen += 10
+	}
+	encd = make([]byte, outlen, outlen+8) //冗余的8B用于可能的结尾的覆盖
+	vals := (*slice)(unsafe.Pointer(&encd)).Data
+	var n uintptr
+	i := 0
+	for ; i <= len(b)-7; i += 7 {
+		sum := 0x000000000000003f & ((uint64)(b[i]) >> 2)
+		sum |= (((uint64)(b[i+1]) << 6) | ((uint64)(b[i]) << 14)) & 0x000000000000ff00
+		sum |= (((uint64)(b[i+1]) << 20) | ((uint64)(b[i+2]) << 12)) & 0x00000000003f0000
+		sum |= (((uint64)(b[i+2]) << 28) | ((uint64)(b[i+3]) << 20)) & 0x00000000ff000000
+		sum |= (((uint64)(b[i+3]) << 34) | ((uint64)(b[i+4]) << 26)) & 0x0000003f00000000
+		sum |= (((uint64)(b[i+4]) << 42) | ((uint64)(b[i+5]) << 34)) & 0x0000ff0000000000
+		sum |= ((uint64)(b[i+5]) << 48) & 0x003f000000000000
+		sum |= ((uint64)(b[i+6]) << 56) & 0xff00000000000000
+		sum += 0x004e004e004e004e
+		*(*uint64)(unsafe.Pointer(uintptr(vals) + n)) = sum
+		n += 8
+	}
+	if offset > 0 {
+		sum := 0x000000000000003f & ((uint64)(b[i]) >> 2)
+		sum |= ((uint64)(b[i]) << 14) & 0x000000000000c000
+		if offset > 1 {
+			sum |= ((uint64)(b[i+1]) << 6) & 0x0000000000003f00
+			sum |= ((uint64)(b[i+1]) << 20) & 0x0000000000300000
+			if offset > 2 {
+				sum |= ((uint64)(b[i+2]) << 12) & 0x00000000000f0000
+				sum |= ((uint64)(b[i+2]) << 28) & 0x00000000f0000000
+				if offset > 3 {
+					sum |= ((uint64)(b[i+3]) << 20) & 0x000000000f000000
+					sum |= ((uint64)(b[i+3]) << 34) & 0x0000003c00000000
+					if offset > 4 {
+						sum |= ((uint64)(b[i+4]) << 26) & 0x0000000300000000
+						sum |= ((uint64)(b[i+4]) << 42) & 0x0000fc0000000000
+						if offset > 5 {
+							sum |= ((uint64)(b[i+5]) << 34) & 0x0000030000000000
+							sum |= ((uint64)(b[i+5]) << 48) & 0x003f000000000000
+						}
+					}
+				}
+			}
+		}
+		sum += 0x004e004e004e004e
+		*(*uint64)(unsafe.Pointer(uintptr(vals) + n)) = sum
+		encd[outlen-2] = '='
+		encd[outlen-1] = byte(offset)
+	}
+	return
+}
+
+func Decode(b []byte) (decd []byte) {
+	outlen := len(b)
+	offset := 0
+	if b[len(b)-2] == '=' {
+		offset = int(b[len(b)-1])
+		switch offset { //算上偏移标志字符占用的2字节
+		case 0:
+			break
+		case 1:
+			outlen -= 4
+		case 2, 3:
+			outlen -= 6
+		case 4, 5:
+			outlen -= 8
+		case 6:
+			outlen -= 10
+		}
+	}
+	outlen = outlen/8*7 + offset
+	decd = make([]byte, outlen)
+	vals := (*slice)(unsafe.Pointer(&b)).Data
+	var n uintptr
+	i := 0
+	for ; i <= len(decd)-7; n += 8 {
+		sum := *(*uint64)(unsafe.Pointer(uintptr(vals) + n)) - 0x004e004e004e004e
+		decd[i] = byte(((sum & 0x000000000000003f) << 2) | ((sum & 0x000000000000c000) >> 14))
+		i++
+		decd[i] = byte(((sum & 0x0000000000003f00) >> 6) | ((sum & 0x0000000000300000) >> 20))
+		i++
+		decd[i] = byte(((sum & 0x00000000000f0000) >> 12) | ((sum & 0x00000000f0000000) >> 28))
+		i++
+		decd[i] = byte(((sum & 0x000000000f000000) >> 20) | ((sum & 0x0000003c00000000) >> 34))
+		i++
+		decd[i] = byte(((sum & 0x0000000300000000) >> 26) | ((sum & 0x0000fc0000000000) >> 42))
+		i++
+		decd[i] = byte(((sum & 0x0000030000000000) >> 34) | ((sum & 0x003f000000000000) >> 48))
+		i++
+		decd[i] = byte(((sum & 0xff00000000000000) >> 56))
+		i++
+	}
+	if offset > 0 {
+		sum := *(*uint64)(unsafe.Pointer(uintptr(vals) + n)) - 0x000000000000004e
+		decd[i] = byte(((sum & 0x000000000000003f) << 2) | ((sum & 0x000000000000c000) >> 14))
+		i++
+		if offset > 1 {
+			sum -= 0x00000000004e0000
+			decd[i] = byte(((sum & 0x0000000000003f00) >> 6) | ((sum & 0x0000000000300000) >> 20))
+			i++
+			if offset > 2 {
+				decd[i] = byte(((sum & 0x00000000000f0000) >> 12) | ((sum & 0x00000000f0000000) >> 28))
+				i++
+				if offset > 3 {
+					sum -= 0x0000004e00000000
+					decd[i] = byte(((sum & 0x000000000f000000) >> 20) | ((sum & 0x0000003c00000000) >> 34))
+					i++
+					if offset > 4 {
+						decd[i] = byte(((sum & 0x0000000300000000) >> 26) | ((sum & 0x0000fc0000000000) >> 42))
+						i++
+						if offset > 5 {
+							sum -= 0x004e000000000000
+							decd[i] = byte(((sum & 0x0000030000000000) >> 34) | ((sum & 0x003f000000000000) >> 48))
+							i++
+						}
+
+					}
+				}
+			}
+		}
+	}
+	return
+}