path: root/xeth/xeth.go

package xeth

/*
 * eXtended ETHereum
 */

import (
    "bytes"
    "encoding/json"
    "fmt"

    "github.com/ethereum/go-ethereum/core"
    "github.com/ethereum/go-ethereum/core/types"
    "github.com/ethereum/go-ethereum/crypto"
    "github.com/ethereum/go-ethereum/ethutil"
    "github.com/ethereum/go-ethereum/event"
    "github.com/ethereum/go-ethereum/logger"
    "github.com/ethereum/go-ethereum/miner"
    "github.com/ethereum/go-ethereum/p2p"
    "github.com/ethereum/go-ethereum/whisper"
)

var pipelogger = logger.NewLogger("XETH")

// to resolve the import cycle
type Backend interface {
    BlockProcessor() *core.BlockProcessor
    ChainManager() *core.ChainManager
    TxPool() *core.TxPool
    PeerCount() int
    IsListening() bool
    Peers() []*p2p.Peer
    KeyManager() *crypto.KeyManager
    Db() ethutil.Database
    EventMux() *event.TypeMux
    Whisper() *whisper.Whisper
    Miner() *miner.Miner
}

type XEth struct {
    eth            Backend
    blockProcessor *core.BlockProcessor
    chainManager   *core.ChainManager
    state          *State
    whisper        *Whisper
    miner          *miner.Miner
}

func New(eth Backend) *XEth {
    xeth := &XEth{
        eth:            eth,
        blockProcessor: eth.BlockProcessor(),
        chainManager:   eth.ChainManager(),
        whisper:        NewWhisper(eth.Whisper()),
        miner:          eth.Miner(),
    }
    xeth.state = NewState(xeth)

    return xeth
}

func (self *XEth) Backend() Backend    { return self.eth }
func (self *XEth) State() *State       { return self.state }
func (self *XEth) Whisper() *Whisper   { return self.whisper }
func (self *XEth) Miner() *miner.Miner { return self.miner }

func (self *XEth) BlockByHash(strHash string) *Block {
    hash := fromHex(strHash)
    block := self.chainManager.GetBlock(hash)

    return NewBlock(block)
}

func (self *XEth) BlockByNumber(num int32) *Block {
    if num == -1 {
        return NewBlock(self.chainManager.CurrentBlock())
    }

    return NewBlock(self.chainManager.GetBlockByNumber(uint64(num)))
}

func (self *XEth) Block(v interface{}) *Block {
    if n, ok := v.(int32); ok {
        return self.BlockByNumber(n)
    } else if str, ok := v.(string); ok {
        return self.BlockByHash(str)
    } else if f, ok := v.(float64); ok { // Don't ask ...
        return self.BlockByNumber(int32(f))
    }

    return nil
}

func (self *XEth) Accounts() []string {
    return []string{toHex(self.eth.KeyManager().Address())}
}

func (self *XEth) PeerCount() int {
    return self.eth.PeerCount()
}

func (self *XEth) IsMining() bool {
    return self.miner.Mining()
}

func (self *XEth) SetMining(shouldmine bool) bool {
    ismining := self.miner.Mining()
    if shouldmine && !ismining {
        self.miner.Start()
    }
    if ismining && !shouldmine {
        self.miner.Stop()
    }
    return self.miner.Mining()
}

func (self *XEth) IsListening() bool {
    return self.eth.IsListening()
}

func (self *XEth) Coinbase() string {
    return toHex(self.eth.KeyManager().Address())
}

func (self *XEth) NumberToHuman(balance string) string {
    b := ethutil.Big(balance)

    return ethutil.CurrencyToString(b)
}

func (self *XEth) StorageAt(addr, storageAddr string) string {
    storage := self.State().SafeGet(addr).StorageString(storageAddr)

    return toHex(storage.Bytes())
}

func (self *XEth) BalanceAt(addr string) string {
    return self.State().SafeGet(addr).Balance().String()
}

func (self *XEth) TxCountAt(address string) int {
    return int(self.State().SafeGet(address).Nonce())
}

func (self *XEth) CodeAt(address string) string {
    return toHex(self.State().SafeGet(address).Code())
}

func (self *XEth) IsContract(address string) bool {
    return len(self.State().SafeGet(address).Code()) > 0
}

func (self *XEth) SecretToAddress(key string) string {
    pair, err := crypto.NewKeyPairFromSec(fromHex(key))
    if err != nil {
        return ""
    }

    return toHex(pair.Address())
}

func (self *XEth) Execute(addr, value, gas, price, data string) (string, error) {
    return "", nil
}

type KeyVal struct {
    Key   string `json:"key"`
    Value string `json:"value"`
}

func (self *XEth) EachStorage(addr string) string {
    var values []KeyVal
    object := self.State().SafeGet(addr)
    it := object.Trie().Iterator()
    for it.Next() {
        values = append(values, KeyVal{toHex(it.Key), toHex(it.Value)})
    }

    valuesJson, err := json.Marshal(values)
    if err != nil {
        return ""
    }

    return string(valuesJson)
}

func (self *XEth) ToAscii(str string) string {
    padded := ethutil.RightPadBytes([]byte(str), 32)

    return "0x" + toHex(padded)
}

func (self *XEth) FromAscii(str string) string {
    if ethutil.IsHex(str) {
        str = str[2:]
    }

    return string(bytes.Trim(fromHex(str), "\x00"))
}

func (self *XEth) FromNumber(str string) string {
    if ethutil.IsHex(str) {
        str = str[2:]
    }

    return ethutil.BigD(fromHex(str)).String()
}

func (self *XEth) PushTx(encodedTx string) (string, error) {
    tx := types.NewTransactionFromBytes(fromHex(encodedTx))
    err := self.eth.TxPool().Add(tx)
    if err != nil {
        return "", err
    }

    if tx.To() == nil {
        addr := core.AddressFromMessage(tx)
        return toHex(addr), nil
    }
    return toHex(tx.Hash()), nil
}

func (self *XEth) Call(toStr, valueStr, gasStr, gasPriceStr, dataStr string) (string, error) {
    if len(gasStr) == 0 {
        gasStr = "100000"
    }
    if len(gasPriceStr) == 0 {
        gasPriceStr = "1"
    }

    var (
        statedb = self.chainManager.TransState()
        key     = self.eth.KeyManager().KeyPair()
        from    = statedb.GetOrNewStateObject(key.Address())
        block   = self.chainManager.CurrentBlock()
        to      = statedb.GetOrNewStateObject(fromHex(toStr))
        data    = fromHex(dataStr)
        gas     = ethutil.Big(gasStr)
        price   = ethutil.Big(gasPriceStr)
        value   = ethutil.Big(valueStr)
    )

    msg := types.NewTransactionMessage(fromHex(toStr), value, gas, price, data)
    msg.Sign(key.PrivateKey)
    vmenv := core.NewEnv(statedb, self.chainManager, msg, block)

    res, err := vmenv.Call(from, to.Address(), data, gas, price, value)
    if err != nil {
        return "", err
    }

    return toHex(res), nil
}

func (self *XEth) Transact(toStr, valueStr, gasStr, gasPriceStr, codeStr string) (string, error) {
    var (
        to               []byte
        value            = ethutil.NewValue(valueStr)
        gas              = ethutil.NewValue(gasStr)
        price            = ethutil.NewValue(gasPriceStr)
        data             []byte
        key              = self.eth.KeyManager().KeyPair()
        contractCreation bool
    )

    data = fromHex(codeStr)
    to = fromHex(toStr)
    if len(to) == 0 {
        contractCreation = true
    }

    var tx *types.Transaction
    if contractCreation {
        tx = types.NewContractCreationTx(value.BigInt(), gas.BigInt(), price.BigInt(), data)
    } else {
        tx = types.NewTransactionMessage(to, value.BigInt(), gas.BigInt(), price.BigInt(), data)
    }

    var err error
    state := self.eth.ChainManager().TransState()
    if balance := state.GetBalance(key.Address()); balance.Cmp(tx.Value()) < 0 {
        return "", fmt.Errorf("insufficient balance. balance=%v tx=%v", balance, tx.Value())
    }
    nonce := state.GetNonce(key.Address())

    tx.SetNonce(nonce)
    tx.Sign(key.PrivateKey)

    //fmt.Printf("create tx: %x %v\n", tx.Hash()[:4], tx.Nonce())

    // Do some pre processing for our "pre" events  and hooks
    block := self.chainManager.NewBlock(key.Address())
    coinbase := state.GetOrNewStateObject(key.Address())
    coinbase.SetGasPool(block.GasLimit())
    self.blockProcessor.ApplyTransactions(coinbase, state, block, types.Transactions{tx}, true)

    err = self.eth.TxPool().Add(tx)
    if err != nil {
        return "", err
    }
    state.SetNonce(key.Address(), nonce+1)

    if types.IsContractAddr(to) {
        return toHex(core.AddressFromMessage(tx)), nil
    }

    return toHex(tx.Hash()), nil
}