146 lines
3.6 KiB
Go
146 lines
3.6 KiB
Go
package randomx
|
|
|
|
import (
|
|
"unsafe"
|
|
)
|
|
|
|
type MemoryBlock [128]uint64
|
|
|
|
func (m *MemoryBlock) GetLine(addr uint64) *RegisterLine {
|
|
addr >>= 3
|
|
//[addr : addr+8 : addr+8]
|
|
return (*RegisterLine)(unsafe.Add(unsafe.Pointer(m), addr*8))
|
|
}
|
|
|
|
type Randomx_Cache struct {
|
|
Blocks []MemoryBlock
|
|
|
|
Programs [RANDOMX_PROGRAM_COUNT]SuperScalarProgram
|
|
|
|
JitPrograms [RANDOMX_PROGRAM_COUNT]ProgramFunc
|
|
|
|
Flags uint64
|
|
}
|
|
|
|
func Randomx_alloc_cache(flags uint64) *Randomx_Cache {
|
|
if flags == RANDOMX_FLAG_DEFAULT {
|
|
flags = RANDOMX_FLAG_JIT
|
|
}
|
|
return &Randomx_Cache{
|
|
Flags: flags,
|
|
}
|
|
}
|
|
|
|
func (cache *Randomx_Cache) VM_Initialize() *VM {
|
|
|
|
return &VM{
|
|
Dataset: &Randomx_DatasetLight{
|
|
Cache: cache,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (cache *Randomx_Cache) Close() error {
|
|
for _, p := range cache.JitPrograms {
|
|
if p != nil {
|
|
err := p.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (cache *Randomx_Cache) Init(key []byte) {
|
|
//fmt.Printf("appending null byte is not necessary but only done for testing")
|
|
kkey := append([]byte{}, key...)
|
|
//kkey = append(kkey,0)
|
|
//cache->initialize(cache, key, keySize);
|
|
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(argonBlock{}))/int(unsafe.Sizeof(MemoryBlock{}))*len(argonBlocks))
|
|
|
|
cache.Blocks = memoryBlocks
|
|
|
|
nonce := uint32(0) //uint32(len(key))
|
|
gen := Init_Blake2Generator(key, nonce)
|
|
for i := 0; i < 8; i++ {
|
|
cache.Programs[i] = Build_SuperScalar_Program(gen) // build a superscalar program
|
|
if cache.Flags&RANDOMX_FLAG_JIT > 0 {
|
|
cache.JitPrograms[i] = generateSuperscalarCode(cache.Programs[i])
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// GetMixBlock fetch a 64 byte block in uint64 form
|
|
func (cache *Randomx_Cache) GetMixBlock(addr uint64) *RegisterLine {
|
|
|
|
mask := CacheSize/CacheLineSize - 1
|
|
|
|
addr = (addr & mask) * CacheLineSize
|
|
|
|
block := addr / 1024
|
|
return cache.Blocks[block].GetLine(addr % 1024)
|
|
}
|
|
|
|
func (cache *Randomx_Cache) InitDatasetItem(rl *RegisterLine, itemNumber uint64) {
|
|
const superscalarMul0 uint64 = 6364136223846793005
|
|
const superscalarAdd1 uint64 = 9298411001130361340
|
|
const superscalarAdd2 uint64 = 12065312585734608966
|
|
const superscalarAdd3 uint64 = 9306329213124626780
|
|
const superscalarAdd4 uint64 = 5281919268842080866
|
|
const superscalarAdd5 uint64 = 10536153434571861004
|
|
const superscalarAdd6 uint64 = 3398623926847679864
|
|
const superscalarAdd7 uint64 = 9549104520008361294
|
|
|
|
registerValue := itemNumber
|
|
|
|
rl[0] = (itemNumber + 1) * superscalarMul0
|
|
rl[1] = rl[0] ^ superscalarAdd1
|
|
rl[2] = rl[0] ^ superscalarAdd2
|
|
rl[3] = rl[0] ^ superscalarAdd3
|
|
rl[4] = rl[0] ^ superscalarAdd4
|
|
rl[5] = rl[0] ^ superscalarAdd5
|
|
rl[6] = rl[0] ^ superscalarAdd6
|
|
rl[7] = rl[0] ^ superscalarAdd7
|
|
|
|
if cache.JitPrograms[0] != nil {
|
|
for i := 0; i < RANDOMX_CACHE_ACCESSES; i++ {
|
|
mix := cache.GetMixBlock(registerValue)
|
|
|
|
cache.JitPrograms[i].Execute(rl)
|
|
|
|
for q := range rl {
|
|
rl[q] ^= mix[q]
|
|
}
|
|
|
|
registerValue = rl[cache.Programs[i].AddressRegister()]
|
|
|
|
}
|
|
} else {
|
|
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 *Randomx_Cache) initDataset(dataset []RegisterLine, startItem, endItem uint64) {
|
|
panic("todo")
|
|
for itemNumber := startItem; itemNumber < endItem; itemNumber, dataset = itemNumber+1, dataset[1:] {
|
|
cache.InitDatasetItem(&dataset[0], itemNumber)
|
|
}
|
|
}
|