package main import ( "encoding/json" "errors" "fmt" "math/big" "strconv" "time" "github.com/ethereum/go-ethereum/cmd/utils" "github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/common/compiler" "github.com/ethereum/go-ethereum/common/natspec" "github.com/ethereum/go-ethereum/common/resolver" "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/crypto" "github.com/ethereum/go-ethereum/logger/glog" "github.com/ethereum/go-ethereum/rlp" "github.com/ethereum/go-ethereum/rpc" "github.com/ethereum/go-ethereum/xeth" "github.com/robertkrimen/otto" "gopkg.in/fatih/set.v0" ) /* node admin bindings */ func (js *jsre) adminBindings() { ethO, _ := js.re.Get("eth") eth := ethO.Object() eth.Set("pendingTransactions", js.pendingTransactions) eth.Set("resend", js.resend) js.re.Set("admin", struct{}{}) t, _ := js.re.Get("admin") admin := t.Object() admin.Set("addPeer", js.addPeer) admin.Set("startRPC", js.startRPC) admin.Set("stopRPC", js.stopRPC) admin.Set("nodeInfo", js.nodeInfo) admin.Set("peers", js.peers) admin.Set("newAccount", js.newAccount) admin.Set("unlock", js.unlock) admin.Set("import", js.importChain) admin.Set("export", js.exportChain) admin.Set("verbosity", js.verbosity) admin.Set("progress", js.downloadProgress) admin.Set("setSolc", js.setSolc) admin.Set("contractInfo", struct{}{}) t, _ = admin.Get("contractInfo") cinfo := t.Object() // newRegistry officially not documented temporary option cinfo.Set("start", js.startNatSpec) cinfo.Set("stop", js.stopNatSpec) cinfo.Set("newRegistry", js.newRegistry) cinfo.Set("get", js.getContractInfo) cinfo.Set("register", js.register) cinfo.Set("registerUrl", js.registerUrl) // cinfo.Set("verify", js.verify) admin.Set("miner", struct{}{}) t, _ = admin.Get("miner") miner := t.Object() miner.Set("start", js.startMining) miner.Set("stop", js.stopMining) miner.Set("hashrate", js.hashrate) miner.Set("setExtra", js.setExtra) miner.Set("setGasPrice", js.setGasPrice) admin.Set("debug", struct{}{}) t, _ = admin.Get("debug") debug := t.Object() js.re.Set("sleep", js.sleep) debug.Set("backtrace", js.backtrace) debug.Set("printBlock", js.printBlock) debug.Set("dumpBlock", js.dumpBlock) debug.Set("getBlockRlp", js.getBlockRlp) debug.Set("setHead", js.setHead) debug.Set("processBlock", js.debugBlock) // undocumented temporary debug.Set("waitForBlocks", js.waitForBlocks) } // generic helper to getBlock by Number/Height or Hex depending on autodetected input // if argument is missing the current block is returned // if block is not found or there is problem with decoding // the appropriate value is returned and block is guaranteed to be nil func (js *jsre) getBlock(call otto.FunctionCall) (*types.Block, error) { var block *types.Block if len(call.ArgumentList) > 0 { if call.Argument(0).IsNumber() { num, _ := call.Argument(0).ToInteger() block = js.ethereum.ChainManager().GetBlockByNumber(uint64(num)) } else if call.Argument(0).IsString() { hash, _ := call.Argument(0).ToString() block = js.ethereum.ChainManager().GetBlock(common.HexToHash(hash)) } else { return nil, errors.New("invalid argument for dump. Either hex string or number") } } else { block = js.ethereum.ChainManager().CurrentBlock() } if block == nil { return nil, errors.New("block not found") } return block, nil } func (js *jsre) pendingTransactions(call otto.FunctionCall) otto.Value { txs := js.ethereum.TxPool().GetTransactions() // grab the accounts from the account manager. This will help with determening which // transactions should be returned. accounts, err := js.ethereum.AccountManager().Accounts() if err != nil { fmt.Println(err) return otto.UndefinedValue() } // Add the accouns to a new set accountSet := set.New() for _, account := range accounts { accountSet.Add(common.BytesToAddress(account.Address)) } //ltxs := make([]*tx, len(txs)) var ltxs []*tx for _, tx := range txs { // no need to check err if from, _ := tx.From(); accountSet.Has(from) { ltxs = append(ltxs, newTx(tx)) } } return js.re.ToVal(ltxs) } func (js *jsre) resend(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) == 0 { fmt.Println("first argument must be a transaction") return otto.FalseValue() } v, err := call.Argument(0).Export() if err != nil { fmt.Println(err) return otto.FalseValue() } if tx, ok := v.(*tx); ok { gl, gp := tx.GasLimit, tx.GasPrice if len(call.ArgumentList) > 1 { gp = call.Argument(1).String() } if len(call.ArgumentList) > 2 { gl = call.Argument(2).String() } ret, err := js.xeth.Transact(tx.From, tx.To, tx.Nonce, tx.Value, gl, gp, tx.Data) if err != nil { fmt.Println(err) return otto.FalseValue() } js.ethereum.TxPool().RemoveTransactions(types.Transactions{tx.tx}) return js.re.ToVal(ret) } fmt.Println("first argument must be a transaction") return otto.FalseValue() } func (js *jsre) debugBlock(call otto.FunctionCall) otto.Value { block, err := js.getBlock(call) if err != nil { fmt.Println(err) return otto.UndefinedValue() } old := vm.Debug vm.Debug = true _, err = js.ethereum.BlockProcessor().RetryProcess(block) if err != nil { glog.Infoln(err) } vm.Debug = old return otto.UndefinedValue() } func (js *jsre) setHead(call otto.FunctionCall) otto.Value { block, err := js.getBlock(call) if err != nil { fmt.Println(err) return otto.UndefinedValue() } js.ethereum.ChainManager().SetHead(block) return otto.UndefinedValue() } func (js *jsre) downloadProgress(call otto.FunctionCall) otto.Value { current, max := js.ethereum.Downloader().Stats() return js.re.ToVal(fmt.Sprintf("%d/%d", current, max)) } func (js *jsre) getBlockRlp(call otto.FunctionCall) otto.Value { block, err := js.getBlock(call) if err != nil { fmt.Println(err) return otto.UndefinedValue() } encoded, _ := rlp.EncodeToBytes(block) return js.re.ToVal(fmt.Sprintf("%x", encoded)) } func (js *jsre) setExtra(call otto.FunctionCall) otto.Value { extra, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.UndefinedValue() } if len(extra) > 1024 { fmt.Println("error: cannot exceed 1024 bytes") return otto.UndefinedValue() } js.ethereum.Miner().SetExtra([]byte(extra)) return otto.UndefinedValue() } func (js *jsre) setGasPrice(call otto.FunctionCall) otto.Value { gasPrice, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.UndefinedValue() } js.ethereum.Miner().SetGasPrice(common.String2Big(gasPrice)) return otto.UndefinedValue() } func (js *jsre) hashrate(otto.FunctionCall) otto.Value { return js.re.ToVal(js.ethereum.Miner().HashRate()) } func (js *jsre) backtrace(call otto.FunctionCall) otto.Value { tracestr, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.UndefinedValue() } glog.GetTraceLocation().Set(tracestr) return otto.UndefinedValue() } func (js *jsre) verbosity(call otto.FunctionCall) otto.Value { v, err := call.Argument(0).ToInteger() if err != nil { fmt.Println(err) return otto.UndefinedValue() } glog.SetV(int(v)) return otto.UndefinedValue() } func (js *jsre) startMining(call otto.FunctionCall) otto.Value { threads, err := call.Argument(0).ToInteger() if err != nil { fmt.Println(err) return otto.FalseValue() } err = js.ethereum.StartMining(int(threads)) if err != nil { fmt.Println(err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) stopMining(call otto.FunctionCall) otto.Value { js.ethereum.StopMining() return otto.TrueValue() } func (js *jsre) startRPC(call otto.FunctionCall) otto.Value { addr, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } port, err := call.Argument(1).ToInteger() if err != nil { fmt.Println(err) return otto.FalseValue() } corsDomain := js.corsDomain if len(call.ArgumentList) > 2 { corsDomain, err = call.Argument(2).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } } config := rpc.RpcConfig{ ListenAddress: addr, ListenPort: uint(port), CorsDomain: corsDomain, } xeth := xeth.New(js.ethereum, nil) err = rpc.Start(xeth, config) if err != nil { fmt.Println(err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) stopRPC(call otto.FunctionCall) otto.Value { if rpc.Stop() == nil { return otto.TrueValue() } return otto.FalseValue() } func (js *jsre) addPeer(call otto.FunctionCall) otto.Value { nodeURL, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } err = js.ethereum.AddPeer(nodeURL) if err != nil { fmt.Println(err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) unlock(call otto.FunctionCall) otto.Value { addr, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } seconds, err := call.Argument(2).ToInteger() if err != nil { fmt.Println(err) return otto.FalseValue() } arg := call.Argument(1) var passphrase string if arg.IsUndefined() { fmt.Println("Please enter a passphrase now.") passphrase, err = readPassword("Passphrase: ", true) if err != nil { fmt.Println(err) return otto.FalseValue() } } else { passphrase, err = arg.ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } } am := js.ethereum.AccountManager() err = am.TimedUnlock(common.FromHex(addr), passphrase, time.Duration(seconds)*time.Second) if err != nil { fmt.Printf("Unlock account failed '%v'\n", err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) newAccount(call otto.FunctionCall) otto.Value { arg := call.Argument(0) var passphrase string if arg.IsUndefined() { fmt.Println("The new account will be encrypted with a passphrase.") fmt.Println("Please enter a passphrase now.") auth, err := readPassword("Passphrase: ", true) if err != nil { fmt.Println(err) return otto.FalseValue() } confirm, err := readPassword("Repeat Passphrase: ", false) if err != nil { fmt.Println(err) return otto.FalseValue() } if auth != confirm { fmt.Println("Passphrases did not match.") return otto.FalseValue() } passphrase = auth } else { var err error passphrase, err = arg.ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } } acct, err := js.ethereum.AccountManager().NewAccount(passphrase) if err != nil { fmt.Printf("Could not create the account: %v", err) return otto.UndefinedValue() } return js.re.ToVal(common.ToHex(acct.Address)) } func (js *jsre) nodeInfo(call otto.FunctionCall) otto.Value { return js.re.ToVal(js.ethereum.NodeInfo()) } func (js *jsre) peers(call otto.FunctionCall) otto.Value { return js.re.ToVal(js.ethereum.PeersInfo()) } func (js *jsre) importChain(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) == 0 { fmt.Println("require file name. admin.importChain(filename)") return otto.FalseValue() } fn, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } if err := utils.ImportChain(js.ethereum.ChainManager(), fn); err != nil { fmt.Println("Import error: ", err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) exportChain(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) == 0 { fmt.Println("require file name: admin.exportChain(filename)") return otto.FalseValue() } fn, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } if err := utils.ExportChain(js.ethereum.ChainManager(), fn); err != nil { fmt.Println(err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) printBlock(call otto.FunctionCall) otto.Value { block, err := js.getBlock(call) if err != nil { fmt.Println(err) return otto.UndefinedValue() } fmt.Println(block) return otto.UndefinedValue() } func (js *jsre) dumpBlock(call otto.FunctionCall) otto.Value { block, err := js.getBlock(call) if err != nil { fmt.Println(err) return otto.UndefinedValue() } statedb := state.New(block.Root(), js.ethereum.StateDb()) dump := statedb.RawDump() return js.re.ToVal(dump) } func (js *jsre) waitForBlocks(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) > 2 { fmt.Println("requires 0, 1 or 2 arguments: admin.debug.waitForBlock(minHeight, timeout)") return otto.FalseValue() } var n, timeout int64 var timer <-chan time.Time var height *big.Int var err error args := len(call.ArgumentList) if args == 2 { timeout, err = call.Argument(1).ToInteger() if err != nil { fmt.Println(err) return otto.UndefinedValue() } timer = time.NewTimer(time.Duration(timeout) * time.Second).C } if args >= 1 { n, err = call.Argument(0).ToInteger() if err != nil { fmt.Println(err) return otto.UndefinedValue() } height = big.NewInt(n) } if args == 0 { height = js.xeth.CurrentBlock().Number() height.Add(height, common.Big1) } wait := js.wait js.wait <- height select { case <-timer: // if times out make sure the xeth loop does not block go func() { select { case wait <- nil: case <-wait: } }() return otto.UndefinedValue() case height = <-wait: } return js.re.ToVal(height.Uint64()) } func (js *jsre) sleep(call otto.FunctionCall) otto.Value { sec, err := call.Argument(0).ToInteger() if err != nil { fmt.Println(err) return otto.FalseValue() } time.Sleep(time.Duration(sec) * time.Second) return otto.UndefinedValue() } func (js *jsre) setSolc(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) != 1 { fmt.Println("needs 1 argument: admin.contractInfo.setSolc(solcPath)") return otto.FalseValue() } solcPath, err := call.Argument(0).ToString() if err != nil { return otto.FalseValue() } solc, err := js.xeth.SetSolc(solcPath) if err != nil { fmt.Println(err) return otto.FalseValue() } fmt.Println(solc.Info()) return otto.TrueValue() } func (js *jsre) register(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) != 4 { fmt.Println("requires 4 arguments: admin.contractInfo.register(fromaddress, contractaddress, contract, filename)") return otto.UndefinedValue() } sender, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.UndefinedValue() } address, err := call.Argument(1).ToString() if err != nil { fmt.Println(err) return otto.UndefinedValue() } raw, err := call.Argument(2).Export() if err != nil { fmt.Println(err) return otto.UndefinedValue() } jsonraw, err := json.Marshal(raw) if err != nil { fmt.Println(err) return otto.UndefinedValue() } var contract compiler.Contract err = json.Unmarshal(jsonraw, &contract) if err != nil { fmt.Println(err) return otto.UndefinedValue() } filename, err := call.Argument(3).ToString() if err != nil { fmt.Println(err) return otto.UndefinedValue() } contenthash, err := compiler.ExtractInfo(&contract, filename) if err != nil { fmt.Println(err) return otto.UndefinedValue() } // sender and contract address are passed as hex strings codeb := js.xeth.CodeAtBytes(address) codehash := common.BytesToHash(crypto.Sha3(codeb)) if err != nil { fmt.Println(err) return otto.UndefinedValue() } registry := resolver.New(js.xeth) _, err = registry.RegisterContentHash(common.HexToAddress(sender), codehash, contenthash) if err != nil { fmt.Println(err) return otto.UndefinedValue() } return js.re.ToVal(contenthash.Hex()) } func (js *jsre) registerUrl(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) != 3 { fmt.Println("requires 3 arguments: admin.contractInfo.register(fromaddress, contenthash, filename)") return otto.FalseValue() } sender, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } contenthash, err := call.Argument(1).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } url, err := call.Argument(2).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } registry := resolver.New(js.xeth) _, err = registry.RegisterUrl(common.HexToAddress(sender), common.HexToHash(contenthash), url) if err != nil { fmt.Println(err) return otto.FalseValue() } return otto.TrueValue() } func (js *jsre) getContractInfo(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) != 1 { fmt.Println("requires 1 argument: admin.contractInfo.register(contractaddress)") return otto.FalseValue() } addr, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } infoDoc, err := natspec.FetchDocsForContract(addr, js.xeth, ds) if err != nil { fmt.Println(err) return otto.UndefinedValue() } var info compiler.ContractInfo err = json.Unmarshal(infoDoc, &info) if err != nil { fmt.Println(err) return otto.UndefinedValue() } return js.re.ToVal(info) } func (js *jsre) startNatSpec(call otto.FunctionCall) otto.Value { js.ethereum.NatSpec = true return otto.TrueValue() } func (js *jsre) stopNatSpec(call otto.FunctionCall) otto.Value { js.ethereum.NatSpec = false return otto.TrueValue() } func (js *jsre) newRegistry(call otto.FunctionCall) otto.Value { if len(call.ArgumentList) != 1 { fmt.Println("requires 1 argument: admin.contractInfo.newRegistry(adminaddress)") return otto.FalseValue() } addr, err := call.Argument(0).ToString() if err != nil { fmt.Println(err) return otto.FalseValue() } registry := resolver.New(js.xeth) err = registry.CreateContracts(common.HexToAddress(addr)) if err != nil { fmt.Println(err) return otto.FalseValue() } return otto.TrueValue() } // internal transaction type which will allow us to resend transactions using `eth.resend` type tx struct { tx *types.Transaction To string From string Nonce string Value string Data string GasLimit string GasPrice string } func newTx(t *types.Transaction) *tx { from, _ := t.From() var to string if t := t.To(); t != nil { to = t.Hex() } return &tx{ tx: t, To: to, From: from.Hex(), Value: t.Amount.String(), Nonce: strconv.Itoa(int(t.Nonce())), Data: "0x" + common.Bytes2Hex(t.Data()), GasLimit: t.GasLimit.String(), GasPrice: t.GasPrice().String(), } }