|
|
|
@ -10,6 +10,7 @@ import ( |
|
|
|
|
"github.com/ethereum/go-ethereum/params" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
// Vm implements VirtualMachine
|
|
|
|
|
type Vm struct { |
|
|
|
|
env Environment |
|
|
|
|
|
|
|
|
@ -27,11 +28,13 @@ type Vm struct { |
|
|
|
|
After func(*Context, error) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// New returns a new Virtual Machine
|
|
|
|
|
func New(env Environment) *Vm { |
|
|
|
|
return &Vm{env: env, debug: Debug, Recoverable: true} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
// Run loops and evaluates the contract's code with the given input data
|
|
|
|
|
func (self *Vm) Run(context *Context, input []byte) (ret []byte, err error) { |
|
|
|
|
self.env.SetDepth(self.env.Depth() + 1) |
|
|
|
|
defer self.env.SetDepth(self.env.Depth() - 1) |
|
|
|
|
|
|
|
|
@ -59,7 +62,7 @@ func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
|
|
|
|
|
if context.CodeAddr != nil { |
|
|
|
|
if p := Precompiled[context.CodeAddr.Str()]; p != nil { |
|
|
|
|
return self.RunPrecompiled(p, callData, context) |
|
|
|
|
return self.RunPrecompiled(p, input, context) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -69,13 +72,15 @@ func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
var ( |
|
|
|
|
op OpCode |
|
|
|
|
codehash = crypto.Sha3Hash(code) |
|
|
|
|
mem = NewMemory() |
|
|
|
|
stack = newStack() |
|
|
|
|
pc = uint64(0) |
|
|
|
|
statedb = self.env.State() |
|
|
|
|
|
|
|
|
|
op OpCode // current opcode
|
|
|
|
|
codehash = crypto.Sha3Hash(code) // codehash is used when doing jump dest caching
|
|
|
|
|
mem = NewMemory() // bound memory
|
|
|
|
|
stack = newStack() // local stack
|
|
|
|
|
pc = uint64(0) // program counter
|
|
|
|
|
statedb = self.env.State() // current state
|
|
|
|
|
|
|
|
|
|
// jump evaluates and checks whether the given jump destination is a valid one
|
|
|
|
|
// if valid move the `pc` otherwise return an error.
|
|
|
|
|
jump = func(from uint64, to *big.Int) error { |
|
|
|
|
if !context.jumpdests.has(codehash, code, to) { |
|
|
|
|
nop := context.GetOp(to.Uint64()) |
|
|
|
@ -97,20 +102,22 @@ func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
|
|
|
|
|
self.log(pc, op, context.Gas, mem, stack, context) |
|
|
|
|
|
|
|
|
|
// calculate the new memory size and gas price for the current executing opcode
|
|
|
|
|
newMemSize, gas, err := self.calculateGasAndSize(context, caller, op, statedb, mem, stack) |
|
|
|
|
if err != nil { |
|
|
|
|
return nil, err |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Use the calculated gas. When insufficient gas is present, use all gas and return an
|
|
|
|
|
// Out Of Gas error
|
|
|
|
|
if !context.UseGas(gas) { |
|
|
|
|
|
|
|
|
|
tmp := new(big.Int).Set(context.Gas) |
|
|
|
|
|
|
|
|
|
context.UseGas(context.Gas) |
|
|
|
|
|
|
|
|
|
return context.Return(nil), OOG(gas, tmp) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Resize the memory calculated previously
|
|
|
|
|
mem.Resize(newMemSize.Uint64()) |
|
|
|
|
|
|
|
|
|
switch op { |
|
|
|
@ -364,11 +371,11 @@ func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
stack.push(value) |
|
|
|
|
|
|
|
|
|
case CALLDATALOAD: |
|
|
|
|
data := getData(callData, stack.pop(), common.Big32) |
|
|
|
|
data := getData(input, stack.pop(), common.Big32) |
|
|
|
|
|
|
|
|
|
stack.push(common.Bytes2Big(data)) |
|
|
|
|
case CALLDATASIZE: |
|
|
|
|
l := int64(len(callData)) |
|
|
|
|
l := int64(len(input)) |
|
|
|
|
stack.push(big.NewInt(l)) |
|
|
|
|
|
|
|
|
|
case CALLDATACOPY: |
|
|
|
@ -377,7 +384,7 @@ func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
cOff = stack.pop() |
|
|
|
|
l = stack.pop() |
|
|
|
|
) |
|
|
|
|
data := getData(callData, cOff, l) |
|
|
|
|
data := getData(input, cOff, l) |
|
|
|
|
|
|
|
|
|
mem.Set(mOff.Uint64(), l.Uint64(), data) |
|
|
|
|
|
|
|
|
@ -623,6 +630,8 @@ func (self *Vm) Run(context *Context, callData []byte) (ret []byte, err error) { |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// calculateGasAndSize calculates the required given the opcode and stack items calculates the new memorysize for
|
|
|
|
|
// the operation. This does not reduce gas or resizes the memory.
|
|
|
|
|
func (self *Vm) calculateGasAndSize(context *Context, caller ContextRef, op OpCode, statedb *state.StateDB, mem *Memory, stack *Stack) (*big.Int, *big.Int, error) { |
|
|
|
|
var ( |
|
|
|
|
gas = new(big.Int) |
|
|
|
@ -764,20 +773,22 @@ func (self *Vm) calculateGasAndSize(context *Context, caller ContextRef, op OpCo |
|
|
|
|
return newMemSize, gas, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (self *Vm) RunPrecompiled(p *PrecompiledAccount, callData []byte, context *Context) (ret []byte, err error) { |
|
|
|
|
gas := p.Gas(len(callData)) |
|
|
|
|
// RunPrecompile runs and evaluate the output of a precompiled contract defined in contracts.go
|
|
|
|
|
func (self *Vm) RunPrecompiled(p *PrecompiledAccount, input []byte, context *Context) (ret []byte, err error) { |
|
|
|
|
gas := p.Gas(len(input)) |
|
|
|
|
if context.UseGas(gas) { |
|
|
|
|
ret = p.Call(callData) |
|
|
|
|
ret = p.Call(input) |
|
|
|
|
|
|
|
|
|
return context.Return(ret), nil |
|
|
|
|
} else { |
|
|
|
|
|
|
|
|
|
tmp := new(big.Int).Set(context.Gas) |
|
|
|
|
|
|
|
|
|
return nil, OOG(gas, tmp) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// log emits a log event to the environment for each opcode encountered. This is not to be confused with the
|
|
|
|
|
// LOG* opcode.
|
|
|
|
|
func (self *Vm) log(pc uint64, op OpCode, gas *big.Int, memory *Memory, stack *Stack, context *Context) { |
|
|
|
|
if Debug { |
|
|
|
|
mem := make([]byte, len(memory.Data())) |
|
|
|
@ -795,6 +806,7 @@ func (self *Vm) log(pc uint64, op OpCode, gas *big.Int, memory *Memory, stack *S |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Environment returns the current workable state of the VM
|
|
|
|
|
func (self *Vm) Env() Environment { |
|
|
|
|
return self.env |
|
|
|
|
} |
|
|
|
|