fumiama/go-base16384
1
0
Fork 0
mirror of https://github.com/fumiama/go-base16384.git synced 2026-06-25 06:09:55 +08:00
Code Activity

fix: amd64 asm read out of range

Browse Source
This commit is contained in:
fumiama
2022-08-22 13:11:31 +08:00
parent ebe73ccc6c
commit ceb3054caa
3 changed files with 40 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
base14.go
View File

@@ -40,7 +40,7 @@ func Encode(b []byte) (encd []byte) {
outlen += 10 outlen += 10
} }
encd = make([]byte, outlen) encd = make([]byte, outlen)
encode(offset, outlen, append(b, 0), encd) encode(offset, outlen, b, encd)
return return
} }
@@ -62,7 +62,7 @@ func EncodeTo(b, encd []byte) error {
if len(encd) < outlen { if len(encd) < outlen {
return errors.New("encd too small") return errors.New("encd too small")
} }
encode(offset, outlen, append(b, 0), encd) encode(offset, outlen, b, encd)
return nil return nil
} }
@@ -105,7 +105,7 @@ func Decode(b []byte) (decd []byte) {
} }
} }
outlen = outlen/8*7 + offset outlen = outlen/8*7 + offset
decd = make([]byte, outlen+1) decd = make([]byte, outlen+8)
decode(offset, outlen, b, decd) decode(offset, outlen, b, decd)
return decd[:outlen] return decd[:outlen]
} }

14
base14_amd64.s
View File

@@ -3,8 +3,8 @@
#include "textflag.h" #include "textflag.h"
// func encode(offset, outlen int, b, encd []byte) // func _encode(offset, outlen int, b, encd []byte) (sum uint64, n uint64)
TEXT ·encode(SB), NOSPLIT, $0-64 TEXT ·_encode(SB), NOSPLIT, $0-81
MOVQ ·offset+0(FP), R10 MOVQ ·offset+0(FP), R10
MOVQ ·outlen+8(FP), AX MOVQ ·outlen+8(FP), AX
MOVQ ·data+16(FP), DI MOVQ ·data+16(FP), DI
@@ -116,16 +116,16 @@ encrem:
encsav: encsav:
MOVQ $21955383195992142, CX MOVQ $21955383195992142, CX
ADDQ CX, DX ADDQ CX, DX
MOVQ DX, 0(R9)(SI*8) SHLQ $3, SI
MOVB $61, -2(R9)(AX*1) MOVQ DX, ·sum+64(FP)
MOVB R10, -1(R9)(AX*1) MOVQ SI, ·sum+72(FP)
encend: encend:
RET RET
// func decode(offset, outlen int, b, decd []byte) // func _decode(offset, outlen int, b, decd []byte)
TEXT ·decode(SB), NOSPLIT, $0-64 TEXT ·_decode(SB), NOSPLIT, $0-64
MOVQ ·offset+0(FP), BX MOVQ ·offset+0(FP), BX
MOVQ ·outlen+8(FP), R8 MOVQ ·outlen+8(FP), R8
MOVQ ·data+16(FP), DI MOVQ ·data+16(FP), DI

34
base14_asm.go
View File

@@ -3,10 +3,36 @@
package base14 package base14
//go:noescape import (
//go:nosplit "encoding/binary"
func encode(offset, outlen int, b, encd []byte) )
//go:noescape //go:noescape
//go:nosplit //go:nosplit
func decode(offset, outlen int, b, decd []byte) func _encode(offset, outlen int, b, encd []byte) (sum uint64, n uint64)
//go:noescape
//go:nosplit
func _decode(offset, outlen int, b, decd []byte)
func encode(offset, outlen int, b, encd []byte) {
if len(b) == 7 {
b = append(b, 0)
}
sum, n := _encode(offset, outlen, b, encd)
if offset == 0 {
return
}
var tmp [8]byte
binary.LittleEndian.PutUint64(tmp[:], sum)
copy(encd[n:], tmp[:])
encd[outlen-2] = '='
encd[outlen-1] = byte(offset)
}
func decode(offset, outlen int, b, decd []byte) {
if offset != 0 && cap(b) == len(b) {
b = append(b, make([]byte, 8)...)
}
_decode(offset, outlen, b, decd)
}