7a5b571c67
This implements a generic approach to enabling soft forks by allowing anyone to put in hashes of contracts that should not be interacted from. This will help "The DAO" in their endevour to stop any whithdrawals from any DAO contract by convincing the mining community to accept their code hash.
278 lines
7.8 KiB
Go
278 lines
7.8 KiB
Go
// Copyright 2014 The go-ethereum Authors
|
|
// This file is part of go-ethereum.
|
|
//
|
|
// go-ethereum is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// go-ethereum is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with go-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
// evm executes EVM code snippets.
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math/big"
|
|
"os"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/cmd/utils"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core"
|
|
"github.com/ethereum/go-ethereum/core/state"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/core/vm"
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
"github.com/ethereum/go-ethereum/logger/glog"
|
|
"gopkg.in/urfave/cli.v1"
|
|
)
|
|
|
|
var (
|
|
app *cli.App
|
|
DebugFlag = cli.BoolFlag{
|
|
Name: "debug",
|
|
Usage: "output full trace logs",
|
|
}
|
|
ForceJitFlag = cli.BoolFlag{
|
|
Name: "forcejit",
|
|
Usage: "forces jit compilation",
|
|
}
|
|
DisableJitFlag = cli.BoolFlag{
|
|
Name: "nojit",
|
|
Usage: "disabled jit compilation",
|
|
}
|
|
CodeFlag = cli.StringFlag{
|
|
Name: "code",
|
|
Usage: "EVM code",
|
|
}
|
|
GasFlag = cli.StringFlag{
|
|
Name: "gas",
|
|
Usage: "gas limit for the evm",
|
|
Value: "10000000000",
|
|
}
|
|
PriceFlag = cli.StringFlag{
|
|
Name: "price",
|
|
Usage: "price set for the evm",
|
|
Value: "0",
|
|
}
|
|
ValueFlag = cli.StringFlag{
|
|
Name: "value",
|
|
Usage: "value set for the evm",
|
|
Value: "0",
|
|
}
|
|
DumpFlag = cli.BoolFlag{
|
|
Name: "dump",
|
|
Usage: "dumps the state after the run",
|
|
}
|
|
InputFlag = cli.StringFlag{
|
|
Name: "input",
|
|
Usage: "input for the EVM",
|
|
}
|
|
SysStatFlag = cli.BoolFlag{
|
|
Name: "sysstat",
|
|
Usage: "display system stats",
|
|
}
|
|
VerbosityFlag = cli.IntFlag{
|
|
Name: "verbosity",
|
|
Usage: "sets the verbosity level",
|
|
}
|
|
CreateFlag = cli.BoolFlag{
|
|
Name: "create",
|
|
Usage: "indicates the action should be create rather than call",
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
app = utils.NewApp("0.2", "the evm command line interface")
|
|
app.Flags = []cli.Flag{
|
|
CreateFlag,
|
|
DebugFlag,
|
|
VerbosityFlag,
|
|
ForceJitFlag,
|
|
DisableJitFlag,
|
|
SysStatFlag,
|
|
CodeFlag,
|
|
GasFlag,
|
|
PriceFlag,
|
|
ValueFlag,
|
|
DumpFlag,
|
|
InputFlag,
|
|
}
|
|
app.Action = run
|
|
}
|
|
|
|
func run(ctx *cli.Context) error {
|
|
glog.SetToStderr(true)
|
|
glog.SetV(ctx.GlobalInt(VerbosityFlag.Name))
|
|
|
|
db, _ := ethdb.NewMemDatabase()
|
|
statedb, _ := state.New(common.Hash{}, db)
|
|
sender := statedb.CreateAccount(common.StringToAddress("sender"))
|
|
|
|
vmenv := NewEnv(statedb, common.StringToAddress("evmuser"), common.Big(ctx.GlobalString(ValueFlag.Name)), vm.Config{
|
|
Debug: ctx.GlobalBool(DebugFlag.Name),
|
|
ForceJit: ctx.GlobalBool(ForceJitFlag.Name),
|
|
EnableJit: !ctx.GlobalBool(DisableJitFlag.Name),
|
|
})
|
|
|
|
tstart := time.Now()
|
|
|
|
var (
|
|
ret []byte
|
|
err error
|
|
)
|
|
|
|
if ctx.GlobalBool(CreateFlag.Name) {
|
|
input := append(common.Hex2Bytes(ctx.GlobalString(CodeFlag.Name)), common.Hex2Bytes(ctx.GlobalString(InputFlag.Name))...)
|
|
ret, _, err = vmenv.Create(
|
|
sender,
|
|
input,
|
|
common.Big(ctx.GlobalString(GasFlag.Name)),
|
|
common.Big(ctx.GlobalString(PriceFlag.Name)),
|
|
common.Big(ctx.GlobalString(ValueFlag.Name)),
|
|
)
|
|
} else {
|
|
receiver := statedb.CreateAccount(common.StringToAddress("receiver"))
|
|
receiver.SetCode(common.Hex2Bytes(ctx.GlobalString(CodeFlag.Name)))
|
|
ret, err = vmenv.Call(
|
|
sender,
|
|
receiver.Address(),
|
|
common.Hex2Bytes(ctx.GlobalString(InputFlag.Name)),
|
|
common.Big(ctx.GlobalString(GasFlag.Name)),
|
|
common.Big(ctx.GlobalString(PriceFlag.Name)),
|
|
common.Big(ctx.GlobalString(ValueFlag.Name)),
|
|
)
|
|
}
|
|
vmdone := time.Since(tstart)
|
|
|
|
if ctx.GlobalBool(DumpFlag.Name) {
|
|
statedb.Commit()
|
|
fmt.Println(string(statedb.Dump()))
|
|
}
|
|
vm.StdErrFormat(vmenv.StructLogs())
|
|
|
|
if ctx.GlobalBool(SysStatFlag.Name) {
|
|
var mem runtime.MemStats
|
|
runtime.ReadMemStats(&mem)
|
|
fmt.Printf("vm took %v\n", vmdone)
|
|
fmt.Printf(`alloc: %d
|
|
tot alloc: %d
|
|
no. malloc: %d
|
|
heap alloc: %d
|
|
heap objs: %d
|
|
num gc: %d
|
|
`, mem.Alloc, mem.TotalAlloc, mem.Mallocs, mem.HeapAlloc, mem.HeapObjects, mem.NumGC)
|
|
}
|
|
|
|
fmt.Printf("OUT: 0x%x", ret)
|
|
if err != nil {
|
|
fmt.Printf(" error: %v", err)
|
|
}
|
|
fmt.Println()
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
if err := app.Run(os.Args); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
type VMEnv struct {
|
|
state *state.StateDB
|
|
block *types.Block
|
|
|
|
transactor *common.Address
|
|
value *big.Int
|
|
|
|
depth int
|
|
Gas *big.Int
|
|
time *big.Int
|
|
logs []vm.StructLog
|
|
|
|
evm *vm.EVM
|
|
}
|
|
|
|
func NewEnv(state *state.StateDB, transactor common.Address, value *big.Int, cfg vm.Config) *VMEnv {
|
|
env := &VMEnv{
|
|
state: state,
|
|
transactor: &transactor,
|
|
value: value,
|
|
time: big.NewInt(time.Now().Unix()),
|
|
}
|
|
cfg.Logger.Collector = env
|
|
|
|
env.evm = vm.New(env, cfg)
|
|
return env
|
|
}
|
|
|
|
// ruleSet implements vm.RuleSet and will always default to the homestead rule set.
|
|
type ruleSet struct{}
|
|
|
|
func (ruleSet) IsHomestead(*big.Int) bool { return true }
|
|
|
|
func (self *VMEnv) MarkCodeHash(common.Hash) {}
|
|
func (self *VMEnv) RuleSet() vm.RuleSet { return ruleSet{} }
|
|
func (self *VMEnv) Vm() vm.Vm { return self.evm }
|
|
func (self *VMEnv) Db() vm.Database { return self.state }
|
|
func (self *VMEnv) MakeSnapshot() vm.Database { return self.state.Copy() }
|
|
func (self *VMEnv) SetSnapshot(db vm.Database) { self.state.Set(db.(*state.StateDB)) }
|
|
func (self *VMEnv) Origin() common.Address { return *self.transactor }
|
|
func (self *VMEnv) BlockNumber() *big.Int { return common.Big0 }
|
|
func (self *VMEnv) Coinbase() common.Address { return *self.transactor }
|
|
func (self *VMEnv) Time() *big.Int { return self.time }
|
|
func (self *VMEnv) Difficulty() *big.Int { return common.Big1 }
|
|
func (self *VMEnv) BlockHash() []byte { return make([]byte, 32) }
|
|
func (self *VMEnv) Value() *big.Int { return self.value }
|
|
func (self *VMEnv) GasLimit() *big.Int { return big.NewInt(1000000000) }
|
|
func (self *VMEnv) VmType() vm.Type { return vm.StdVmTy }
|
|
func (self *VMEnv) Depth() int { return 0 }
|
|
func (self *VMEnv) SetDepth(i int) { self.depth = i }
|
|
func (self *VMEnv) GetHash(n uint64) common.Hash {
|
|
if self.block.Number().Cmp(big.NewInt(int64(n))) == 0 {
|
|
return self.block.Hash()
|
|
}
|
|
return common.Hash{}
|
|
}
|
|
func (self *VMEnv) AddStructLog(log vm.StructLog) {
|
|
self.logs = append(self.logs, log)
|
|
}
|
|
func (self *VMEnv) StructLogs() []vm.StructLog {
|
|
return self.logs
|
|
}
|
|
func (self *VMEnv) AddLog(log *vm.Log) {
|
|
self.state.AddLog(log)
|
|
}
|
|
func (self *VMEnv) CanTransfer(from common.Address, balance *big.Int) bool {
|
|
return self.state.GetBalance(from).Cmp(balance) >= 0
|
|
}
|
|
func (self *VMEnv) Transfer(from, to vm.Account, amount *big.Int) {
|
|
core.Transfer(from, to, amount)
|
|
}
|
|
|
|
func (self *VMEnv) Call(caller vm.ContractRef, addr common.Address, data []byte, gas, price, value *big.Int) ([]byte, error) {
|
|
self.Gas = gas
|
|
return core.Call(self, caller, addr, data, gas, price, value)
|
|
}
|
|
|
|
func (self *VMEnv) CallCode(caller vm.ContractRef, addr common.Address, data []byte, gas, price, value *big.Int) ([]byte, error) {
|
|
return core.CallCode(self, caller, addr, data, gas, price, value)
|
|
}
|
|
|
|
func (self *VMEnv) DelegateCall(caller vm.ContractRef, addr common.Address, data []byte, gas, price *big.Int) ([]byte, error) {
|
|
return core.DelegateCall(self, caller, addr, data, gas, price)
|
|
}
|
|
|
|
func (self *VMEnv) Create(caller vm.ContractRef, data []byte, gas, price, value *big.Int) ([]byte, common.Address, error) {
|
|
return core.Create(self, caller, data, gas, price, value)
|
|
}
|