consensus/ethash: fix usage of *reflect.SliceHeader (#21372)

* consensus/ethash: only use *reflect.SliceHeader, not reflect.SliceHeader. See comment here: https://github.com/golang/go/issues/40397\#issuecomment-663748689

* consensus/ethash: pr feedback from @mdempsky, makes a copy of dest such that is not mutated

* consensus/ethash: remove noop assign

* consensus/ethash: apply same fix to another location

Co-authored-by: Péter Szilágyi <peterke@gmail.com>
Co-authored-by: Martin Holst Swende <martin@swende.se>
This commit is contained in:
Preston Van Loon 2020-11-17 01:35:58 -08:00 committed by GitHub
parent 1ea7537997
commit 844485ec6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -151,10 +151,12 @@ func generateCache(dest []uint32, epoch uint64, seed []byte) {
logFn("Generated ethash verification cache", "elapsed", common.PrettyDuration(elapsed)) logFn("Generated ethash verification cache", "elapsed", common.PrettyDuration(elapsed))
}() }()
// Convert our destination slice to a byte buffer // Convert our destination slice to a byte buffer
header := *(*reflect.SliceHeader)(unsafe.Pointer(&dest)) var cache []byte
header.Len *= 4 cacheHdr := (*reflect.SliceHeader)(unsafe.Pointer(&cache))
header.Cap *= 4 dstHdr := (*reflect.SliceHeader)(unsafe.Pointer(&dest))
cache := *(*[]byte)(unsafe.Pointer(&header)) cacheHdr.Data = dstHdr.Data
cacheHdr.Len = dstHdr.Len * 4
cacheHdr.Cap = dstHdr.Cap * 4
// Calculate the number of theoretical rows (we'll store in one buffer nonetheless) // Calculate the number of theoretical rows (we'll store in one buffer nonetheless)
size := uint64(len(cache)) size := uint64(len(cache))
@ -283,10 +285,12 @@ func generateDataset(dest []uint32, epoch uint64, cache []uint32) {
swapped := !isLittleEndian() swapped := !isLittleEndian()
// Convert our destination slice to a byte buffer // Convert our destination slice to a byte buffer
header := *(*reflect.SliceHeader)(unsafe.Pointer(&dest)) var dataset []byte
header.Len *= 4 datasetHdr := (*reflect.SliceHeader)(unsafe.Pointer(&dataset))
header.Cap *= 4 destHdr := (*reflect.SliceHeader)(unsafe.Pointer(&dest))
dataset := *(*[]byte)(unsafe.Pointer(&header)) datasetHdr.Data = destHdr.Data
datasetHdr.Len = destHdr.Len * 4
datasetHdr.Cap = destHdr.Cap * 4
// Generate the dataset on many goroutines since it takes a while // Generate the dataset on many goroutines since it takes a while
threads := runtime.NumCPU() threads := runtime.NumCPU()