github.com/ontio/ontology@v1.14.4/vm/evm/analysis_test.go (about)

     1  // Copyright (C) 2021 The Ontology Authors
     2  // Copyright 2017 The go-ethereum Authors
     3  // This file is part of the go-ethereum library.
     4  //
     5  // The go-ethereum library is free software: you can redistribute it and/or modify
     6  // it under the terms of the GNU Lesser General Public License as published by
     7  // the Free Software Foundation, either version 3 of the License, or
     8  // (at your option) any later version.
     9  //
    10  // The go-ethereum library is distributed in the hope that it will be useful,
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    13  // GNU Lesser General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU Lesser General Public License
    16  // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
    17  
    18  package evm
    19  
    20  import (
    21  	"testing"
    22  
    23  	"github.com/ethereum/go-ethereum/crypto"
    24  )
    25  
    26  func TestJumpDestAnalysis(t *testing.T) {
    27  	tests := []struct {
    28  		code  []byte
    29  		exp   byte
    30  		which int
    31  	}{
    32  		{[]byte{byte(PUSH1), 0x01, 0x01, 0x01}, 0x40, 0},
    33  		{[]byte{byte(PUSH1), byte(PUSH1), byte(PUSH1), byte(PUSH1)}, 0x50, 0},
    34  		{[]byte{byte(PUSH8), byte(PUSH8), byte(PUSH8), byte(PUSH8), byte(PUSH8), byte(PUSH8), byte(PUSH8), byte(PUSH8), 0x01, 0x01, 0x01}, 0x7F, 0},
    35  		{[]byte{byte(PUSH8), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0x80, 1},
    36  		{[]byte{0x01, 0x01, 0x01, 0x01, 0x01, byte(PUSH2), byte(PUSH2), byte(PUSH2), 0x01, 0x01, 0x01}, 0x03, 0},
    37  		{[]byte{0x01, 0x01, 0x01, 0x01, 0x01, byte(PUSH2), 0x01, 0x01, 0x01, 0x01, 0x01}, 0x00, 1},
    38  		{[]byte{byte(PUSH3), 0x01, 0x01, 0x01, byte(PUSH1), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0x74, 0},
    39  		{[]byte{byte(PUSH3), 0x01, 0x01, 0x01, byte(PUSH1), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0x00, 1},
    40  		{[]byte{0x01, byte(PUSH8), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0x3F, 0},
    41  		{[]byte{0x01, byte(PUSH8), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0xC0, 1},
    42  		{[]byte{byte(PUSH16), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0x7F, 0},
    43  		{[]byte{byte(PUSH16), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0xFF, 1},
    44  		{[]byte{byte(PUSH16), 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01}, 0x80, 2},
    45  		{[]byte{byte(PUSH8), 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, byte(PUSH1), 0x01}, 0x7f, 0},
    46  		{[]byte{byte(PUSH8), 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, byte(PUSH1), 0x01}, 0xA0, 1},
    47  		{[]byte{byte(PUSH32)}, 0x7F, 0},
    48  		{[]byte{byte(PUSH32)}, 0xFF, 1},
    49  		{[]byte{byte(PUSH32)}, 0xFF, 2},
    50  	}
    51  	for _, test := range tests {
    52  		ret := codeBitmap(test.code)
    53  		if ret[test.which] != test.exp {
    54  			t.Fatalf("expected %x, got %02x", test.exp, ret[test.which])
    55  		}
    56  	}
    57  }
    58  
    59  func BenchmarkJumpdestAnalysis_1200k(bench *testing.B) {
    60  	// 1.4 ms
    61  	code := make([]byte, 1200000)
    62  	bench.ResetTimer()
    63  	for i := 0; i < bench.N; i++ {
    64  		codeBitmap(code)
    65  	}
    66  	bench.StopTimer()
    67  }
    68  func BenchmarkJumpdestHashing_1200k(bench *testing.B) {
    69  	// 4 ms
    70  	code := make([]byte, 1200000)
    71  	bench.ResetTimer()
    72  	for i := 0; i < bench.N; i++ {
    73  		crypto.Keccak256Hash(code)
    74  	}
    75  	bench.StopTimer()
    76  }