155 lines
3.9 KiB
Go
155 lines
3.9 KiB
Go
package randomx
|
|
|
|
import (
|
|
"git.gammaspectra.live/P2Pool/go-randomx/v3/argon2"
|
|
"git.gammaspectra.live/P2Pool/go-randomx/v3/blake2"
|
|
"git.gammaspectra.live/P2Pool/go-randomx/v3/keys"
|
|
"runtime"
|
|
"slices"
|
|
"unsafe"
|
|
)
|
|
|
|
type MemoryBlock [128]uint64
|
|
|
|
func (m *MemoryBlock) GetLine(addr uint64) *RegisterLine {
|
|
addr >>= 3
|
|
return (*RegisterLine)(unsafe.Pointer(unsafe.SliceData(m[addr : addr+8 : addr+8])))
|
|
}
|
|
|
|
type Cache struct {
|
|
Blocks []MemoryBlock
|
|
|
|
Programs [RANDOMX_PROGRAM_COUNT]SuperScalarProgram
|
|
|
|
JitPrograms [RANDOMX_PROGRAM_COUNT]SuperScalarProgramFunc
|
|
|
|
Flags uint64
|
|
}
|
|
|
|
func NewCache(flags uint64) *Cache {
|
|
if flags == RANDOMX_FLAG_DEFAULT {
|
|
flags = RANDOMX_FLAG_JIT
|
|
}
|
|
return &Cache{
|
|
Flags: flags,
|
|
}
|
|
}
|
|
|
|
func (cache *Cache) HasJIT() bool {
|
|
return cache.Flags&RANDOMX_FLAG_JIT > 0 && cache.JitPrograms[0] != nil
|
|
}
|
|
|
|
func (cache *Cache) Close() error {
|
|
for _, p := range cache.JitPrograms {
|
|
if p != nil {
|
|
err := p.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cache *Cache) Init(key []byte) {
|
|
if cache.Flags&RANDOMX_FLAG_JIT > 0 {
|
|
// Lock due to external JIT madness
|
|
runtime.LockOSThread()
|
|
defer runtime.UnlockOSThread()
|
|
}
|
|
|
|
kkey := slices.Clone(key)
|
|
|
|
argonBlocks := argon2.BuildBlocks(kkey, []byte(RANDOMX_ARGON_SALT), []byte{}, []byte{}, RANDOMX_ARGON_ITERATIONS, RANDOMX_ARGON_MEMORY, RANDOMX_ARGON_LANES, 0)
|
|
|
|
memoryBlocks := unsafe.Slice((*MemoryBlock)(unsafe.Pointer(unsafe.SliceData(argonBlocks))), int(unsafe.Sizeof(argon2.Block{}))/int(unsafe.Sizeof(MemoryBlock{}))*len(argonBlocks))
|
|
|
|
cache.Blocks = memoryBlocks
|
|
|
|
const nonce uint32 = 0
|
|
|
|
gen := blake2.New(key, nonce)
|
|
for i := 0; i < 8; i++ {
|
|
cache.Programs[i] = BuildSuperScalarProgram(gen) // build a superscalar program
|
|
if cache.Flags&RANDOMX_FLAG_JIT > 0 {
|
|
cache.JitPrograms[i] = generateSuperscalarCode(cache.Programs[i])
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
const Mask = CacheSize/CacheLineSize - 1
|
|
|
|
// GetMixBlock fetch a 64 byte block in uint64 form
|
|
func (cache *Cache) GetMixBlock(addr uint64) *RegisterLine {
|
|
|
|
addr = (addr & Mask) * CacheLineSize
|
|
|
|
block := addr / 1024
|
|
return cache.Blocks[block].GetLine(addr % 1024)
|
|
}
|
|
|
|
func (cache *Cache) InitDatasetItem(rl *RegisterLine, itemNumber uint64) {
|
|
registerValue := itemNumber
|
|
|
|
rl[0] = (itemNumber + 1) * keys.SuperScalar_Constants[0]
|
|
rl[1] = rl[0] ^ keys.SuperScalar_Constants[1]
|
|
rl[2] = rl[0] ^ keys.SuperScalar_Constants[2]
|
|
rl[3] = rl[0] ^ keys.SuperScalar_Constants[3]
|
|
rl[4] = rl[0] ^ keys.SuperScalar_Constants[4]
|
|
rl[5] = rl[0] ^ keys.SuperScalar_Constants[5]
|
|
rl[6] = rl[0] ^ keys.SuperScalar_Constants[6]
|
|
rl[7] = rl[0] ^ keys.SuperScalar_Constants[7]
|
|
|
|
for i := 0; i < RANDOMX_CACHE_ACCESSES; i++ {
|
|
mix := cache.GetMixBlock(registerValue)
|
|
|
|
program := cache.Programs[i]
|
|
|
|
executeSuperscalar(program.Program(), rl)
|
|
|
|
for q := range rl {
|
|
rl[q] ^= mix[q]
|
|
}
|
|
|
|
registerValue = rl[program.AddressRegister()]
|
|
|
|
}
|
|
}
|
|
|
|
func (cache *Cache) InitDatasetItemJIT(rl *RegisterLine, itemNumber uint64) {
|
|
registerValue := itemNumber
|
|
|
|
rl[0] = (itemNumber + 1) * keys.SuperScalar_Constants[0]
|
|
rl[1] = rl[0] ^ keys.SuperScalar_Constants[1]
|
|
rl[2] = rl[0] ^ keys.SuperScalar_Constants[2]
|
|
rl[3] = rl[0] ^ keys.SuperScalar_Constants[3]
|
|
rl[4] = rl[0] ^ keys.SuperScalar_Constants[4]
|
|
rl[5] = rl[0] ^ keys.SuperScalar_Constants[5]
|
|
rl[6] = rl[0] ^ keys.SuperScalar_Constants[6]
|
|
rl[7] = rl[0] ^ keys.SuperScalar_Constants[7]
|
|
|
|
for i := 0; i < RANDOMX_CACHE_ACCESSES; i++ {
|
|
mix := cache.GetMixBlock(registerValue)
|
|
|
|
cache.JitPrograms[i].Execute(uintptr(unsafe.Pointer(rl)))
|
|
|
|
for q := range rl {
|
|
rl[q] ^= mix[q]
|
|
}
|
|
|
|
registerValue = rl[cache.Programs[i].AddressRegister()]
|
|
|
|
}
|
|
}
|
|
|
|
func (cache *Cache) InitDataset(dataset []RegisterLine, startItem, endItem uint64) {
|
|
for itemNumber := startItem; itemNumber < endItem; itemNumber, dataset = itemNumber+1, dataset[1:] {
|
|
if cache.HasJIT() {
|
|
cache.InitDatasetItemJIT(&dataset[0], itemNumber)
|
|
} else {
|
|
cache.InitDatasetItem(&dataset[0], itemNumber)
|
|
}
|
|
}
|
|
}
|