fix bound check in code chunking

gballet · Nov 26, 2021 · 6d40e11 · 6d40e11
1 parent 5ca9901
commit 6d40e11
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/core/vm/instructions.go b/core/vm/instructions.go
@@ -389,12 +389,13 @@ func touchEachChunks(start, end uint64, code []byte, contract *Contract, evm *EV
 	for chunk := start / 31; chunk <= end/31 && chunk <= uint64(len(code))/31; chunk++ {
 		index := trieUtils.GetTreeKeyCodeChunk(contract.Address().Bytes(), uint256.NewInt(chunk))
 		count := uint64(0)
+		end := (chunk + 1) * 31
+
 		// Look for the first code byte (i.e. no pushdata)
-		for ; count < 31 && !contract.IsCode(chunk*31+count); count++ {
+		for ; count < 31 && end+count < uint64(len(contract.Code)) && !contract.IsCode(chunk*31+count); count++ {
 		}
 		var value [32]byte
 		value[0] = byte(count)
-		end := (chunk + 1) * 31
 		if end > uint64(len(code)) {
 			end = uint64(len(code))
 		}