aboutsummaryrefslogtreecommitdiffstats
path: root/vm/analysis.go
blob: 501fbfc4a325df8475ab41f881b95faf5019fc4f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
package vm

import "gopkg.in/fatih/set.v0"

func analyseJumpDests(code []byte) (dests *set.Set) {
    dests = set.New()

    for pc := uint64(0); pc < uint64(len(code)); pc++ {
        var op OpCode = OpCode(code[pc])
        switch op {
        case PUSH1, PUSH2, PUSH3, PUSH4, PUSH5, PUSH6, PUSH7, PUSH8, PUSH9, PUSH10, PUSH11, PUSH12, PUSH13, PUSH14, PUSH15, PUSH16, PUSH17, PUSH18, PUSH19, PUSH20, PUSH21, PUSH22, PUSH23, PUSH24, PUSH25, PUSH26, PUSH27, PUSH28, PUSH29, PUSH30, PUSH31, PUSH32:
            a := uint64(op) - uint64(PUSH1) + 1

            pc += a
            //lp = true
        case JUMPDEST:
            dests.Add(pc)
        }
    }
    return
}