forked from toolshed/abra
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
//go:build !noasm && gc && arm64 && !amd64
|
|
// +build !noasm,gc,arm64,!amd64
|
|
|
|
package sha1cd
|
|
|
|
import (
|
|
"runtime"
|
|
|
|
"github.com/klauspost/cpuid/v2"
|
|
shared "github.com/pjbgf/sha1cd/internal"
|
|
)
|
|
|
|
var hasSHA1 = (runtime.GOARCH == "arm64" && cpuid.CPU.Supports(cpuid.SHA1))
|
|
|
|
// blockARM64 hashes the message p into the current state in h.
|
|
// Both m1 and cs are used to store intermediate results which are used by the collision detection logic.
|
|
//
|
|
//go:noescape
|
|
func blockARM64(h []uint32, p []byte, m1 []uint32, cs [][5]uint32)
|
|
|
|
func block(dig *digest, p []byte) {
|
|
if forceGeneric || !hasSHA1 {
|
|
blockGeneric(dig, p)
|
|
return
|
|
}
|
|
|
|
m1 := [shared.Rounds]uint32{}
|
|
cs := [shared.PreStepState][shared.WordBuffers]uint32{}
|
|
|
|
for len(p) >= shared.Chunk {
|
|
// The assembly code only supports processing a block at a time,
|
|
// so adjust the chunk accordingly.
|
|
chunk := p[:shared.Chunk]
|
|
|
|
blockARM64(dig.h[:], chunk, m1[:], cs[:])
|
|
|
|
rectifyCompressionState(m1, &cs)
|
|
col := checkCollision(m1, cs, dig.h)
|
|
if col {
|
|
dig.col = true
|
|
|
|
blockARM64(dig.h[:], chunk, m1[:], cs[:])
|
|
blockARM64(dig.h[:], chunk, m1[:], cs[:])
|
|
}
|
|
|
|
p = p[shared.Chunk:]
|
|
}
|
|
}
|