2014-12-04 09:28:02 +00:00
|
|
|
package core
|
2014-02-14 22:56:09 +00:00
|
|
|
|
|
|
|
import (
|
2015-01-31 16:22:17 +00:00
|
|
|
"errors"
|
2014-02-14 22:56:09 +00:00
|
|
|
"fmt"
|
2015-04-08 18:47:32 +00:00
|
|
|
"math/big"
|
2015-04-21 20:01:04 +00:00
|
|
|
"sort"
|
2015-02-19 21:33:22 +00:00
|
|
|
"sync"
|
2015-04-21 20:01:04 +00:00
|
|
|
"time"
|
2014-07-29 22:31:15 +00:00
|
|
|
|
2015-03-16 10:27:38 +00:00
|
|
|
"github.com/ethereum/go-ethereum/common"
|
2015-04-08 18:47:32 +00:00
|
|
|
"github.com/ethereum/go-ethereum/core/state"
|
2015-03-17 10:59:26 +00:00
|
|
|
"github.com/ethereum/go-ethereum/core/types"
|
2014-12-18 12:12:54 +00:00
|
|
|
"github.com/ethereum/go-ethereum/event"
|
2014-10-31 11:56:05 +00:00
|
|
|
"github.com/ethereum/go-ethereum/logger"
|
2015-04-08 18:47:32 +00:00
|
|
|
"github.com/ethereum/go-ethereum/logger/glog"
|
2015-04-07 22:31:23 +00:00
|
|
|
"gopkg.in/fatih/set.v0"
|
2014-02-14 22:56:09 +00:00
|
|
|
)
|
|
|
|
|
2015-01-31 16:22:17 +00:00
|
|
|
var (
|
2015-04-08 18:47:32 +00:00
|
|
|
ErrInvalidSender = errors.New("Invalid sender")
|
2015-04-21 20:01:04 +00:00
|
|
|
ErrNonce = errors.New("Nonce too low")
|
2015-04-26 09:19:40 +00:00
|
|
|
ErrBalance = errors.New("Insufficient balance")
|
2015-05-10 23:28:15 +00:00
|
|
|
ErrNonExistentAccount = errors.New("Account does not exist or account balance too low")
|
2015-04-26 09:19:40 +00:00
|
|
|
ErrInsufficientFunds = errors.New("Insufficient funds for gas * price + value")
|
2015-04-08 18:47:32 +00:00
|
|
|
ErrIntrinsicGas = errors.New("Intrinsic gas too low")
|
2015-04-24 15:45:51 +00:00
|
|
|
ErrGasLimit = errors.New("Exceeds block gas limit")
|
2015-05-26 17:50:42 +00:00
|
|
|
ErrNegativeValue = errors.New("Negative value")
|
2015-01-31 16:22:17 +00:00
|
|
|
)
|
2014-06-23 11:54:10 +00:00
|
|
|
|
2014-10-29 02:50:20 +00:00
|
|
|
const txPoolQueueSize = 50
|
2014-02-14 22:56:09 +00:00
|
|
|
|
2014-11-18 15:58:22 +00:00
|
|
|
type TxPoolHook chan *types.Transaction
|
2015-03-17 10:59:26 +00:00
|
|
|
type TxMsg struct{ Tx *types.Transaction }
|
2014-02-25 10:22:27 +00:00
|
|
|
|
2015-04-21 09:27:12 +00:00
|
|
|
type stateFn func() *state.StateDB
|
|
|
|
|
2014-02-25 10:22:27 +00:00
|
|
|
const (
|
2014-06-10 13:02:41 +00:00
|
|
|
minGasPrice = 1000000
|
2014-02-25 10:22:27 +00:00
|
|
|
)
|
|
|
|
|
2014-02-23 00:57:04 +00:00
|
|
|
type TxProcessor interface {
|
2014-11-18 15:58:22 +00:00
|
|
|
ProcessTransaction(tx *types.Transaction)
|
2014-02-23 00:57:04 +00:00
|
|
|
}
|
|
|
|
|
2014-02-14 22:56:09 +00:00
|
|
|
// The tx pool a thread safe transaction pool handler. In order to
|
|
|
|
// guarantee a non blocking pool we use a queue channel which can be
|
2015-01-02 11:09:38 +00:00
|
|
|
// independently read without needing access to the actual pool.
|
2014-02-14 22:56:09 +00:00
|
|
|
type TxPool struct {
|
2015-02-19 21:33:22 +00:00
|
|
|
mu sync.RWMutex
|
2014-02-14 22:56:09 +00:00
|
|
|
// Queueing channel for reading and writing incoming
|
|
|
|
// transactions to
|
2014-11-18 15:58:22 +00:00
|
|
|
queueChan chan *types.Transaction
|
2014-02-14 22:56:09 +00:00
|
|
|
// Quiting channel
|
|
|
|
quit chan bool
|
2015-04-08 18:47:32 +00:00
|
|
|
// The state function which will allow us to do some pre checkes
|
2015-04-21 09:27:12 +00:00
|
|
|
currentState stateFn
|
2015-04-24 15:45:51 +00:00
|
|
|
// The current gas limit function callback
|
|
|
|
gasLimit func() *big.Int
|
2014-02-14 22:56:09 +00:00
|
|
|
// The actual pool
|
2015-04-07 22:31:23 +00:00
|
|
|
txs map[common.Hash]*types.Transaction
|
|
|
|
invalidHashes *set.Set
|
2014-02-14 22:56:09 +00:00
|
|
|
|
2015-06-03 12:06:20 +00:00
|
|
|
queue map[common.Address]map[common.Hash]*types.Transaction
|
2015-04-21 20:01:04 +00:00
|
|
|
|
2014-02-25 10:22:27 +00:00
|
|
|
subscribers []chan TxMsg
|
2014-12-18 12:12:54 +00:00
|
|
|
|
2015-01-02 11:26:55 +00:00
|
|
|
eventMux *event.TypeMux
|
2014-02-14 22:56:09 +00:00
|
|
|
}
|
|
|
|
|
2015-04-24 15:45:51 +00:00
|
|
|
func NewTxPool(eventMux *event.TypeMux, currentStateFn stateFn, gasLimitFn func() *big.Int) *TxPool {
|
2015-04-21 20:01:04 +00:00
|
|
|
txPool := &TxPool{
|
2015-04-07 22:31:23 +00:00
|
|
|
txs: make(map[common.Hash]*types.Transaction),
|
2015-06-03 12:06:20 +00:00
|
|
|
queue: make(map[common.Address]map[common.Hash]*types.Transaction),
|
2015-04-07 22:31:23 +00:00
|
|
|
queueChan: make(chan *types.Transaction, txPoolQueueSize),
|
|
|
|
quit: make(chan bool),
|
|
|
|
eventMux: eventMux,
|
|
|
|
invalidHashes: set.New(),
|
2015-04-08 18:47:32 +00:00
|
|
|
currentState: currentStateFn,
|
2015-04-24 15:45:51 +00:00
|
|
|
gasLimit: gasLimitFn,
|
2014-02-14 22:56:09 +00:00
|
|
|
}
|
2015-04-21 20:01:04 +00:00
|
|
|
return txPool
|
|
|
|
}
|
|
|
|
|
|
|
|
func (pool *TxPool) Start() {
|
2015-04-23 09:09:58 +00:00
|
|
|
// Queue timer will tick so we can attempt to move items from the queue to the
|
|
|
|
// main transaction pool.
|
|
|
|
queueTimer := time.NewTicker(300 * time.Millisecond)
|
|
|
|
// Removal timer will tick and attempt to remove bad transactions (account.nonce>tx.nonce)
|
|
|
|
removalTimer := time.NewTicker(1 * time.Second)
|
2015-04-21 20:01:04 +00:00
|
|
|
done:
|
|
|
|
for {
|
|
|
|
select {
|
2015-04-23 09:09:58 +00:00
|
|
|
case <-queueTimer.C:
|
2015-04-21 20:01:04 +00:00
|
|
|
pool.checkQueue()
|
2015-04-23 09:09:58 +00:00
|
|
|
case <-removalTimer.C:
|
|
|
|
pool.validatePool()
|
2015-04-21 20:01:04 +00:00
|
|
|
case <-pool.quit:
|
|
|
|
break done
|
|
|
|
}
|
|
|
|
}
|
2014-02-14 22:56:09 +00:00
|
|
|
}
|
|
|
|
|
2014-11-18 15:58:22 +00:00
|
|
|
func (pool *TxPool) ValidateTransaction(tx *types.Transaction) error {
|
2015-03-17 10:59:26 +00:00
|
|
|
// Validate sender
|
2015-04-08 18:47:32 +00:00
|
|
|
var (
|
|
|
|
from common.Address
|
|
|
|
err error
|
|
|
|
)
|
|
|
|
|
|
|
|
if from, err = tx.From(); err != nil {
|
2015-03-18 12:38:47 +00:00
|
|
|
return ErrInvalidSender
|
2014-12-01 23:14:34 +00:00
|
|
|
}
|
2015-04-08 18:47:32 +00:00
|
|
|
|
|
|
|
if !pool.currentState().HasAccount(from) {
|
|
|
|
return ErrNonExistentAccount
|
2014-06-10 13:02:41 +00:00
|
|
|
}
|
2015-04-08 18:47:32 +00:00
|
|
|
|
2015-04-24 15:45:51 +00:00
|
|
|
if pool.gasLimit().Cmp(tx.GasLimit) < 0 {
|
|
|
|
return ErrGasLimit
|
|
|
|
}
|
|
|
|
|
2015-05-26 17:50:42 +00:00
|
|
|
if tx.Amount.Cmp(common.Big0) < 0 {
|
|
|
|
return ErrNegativeValue
|
|
|
|
}
|
|
|
|
|
2015-04-26 09:19:40 +00:00
|
|
|
total := new(big.Int).Mul(tx.Price, tx.GasLimit)
|
|
|
|
total.Add(total, tx.Value())
|
|
|
|
if pool.currentState().GetBalance(from).Cmp(total) < 0 {
|
2015-04-08 18:47:32 +00:00
|
|
|
return ErrInsufficientFunds
|
|
|
|
}
|
|
|
|
|
|
|
|
if tx.GasLimit.Cmp(IntrinsicGas(tx)) < 0 {
|
|
|
|
return ErrIntrinsicGas
|
|
|
|
}
|
|
|
|
|
|
|
|
if pool.currentState().GetNonce(from) > tx.Nonce() {
|
2015-04-21 20:01:04 +00:00
|
|
|
return ErrNonce
|
2015-04-08 18:47:32 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2014-02-14 22:56:09 +00:00
|
|
|
}
|
|
|
|
|
2015-02-19 21:33:22 +00:00
|
|
|
func (self *TxPool) add(tx *types.Transaction) error {
|
2015-03-17 11:16:21 +00:00
|
|
|
hash := tx.Hash()
|
2015-04-07 22:31:23 +00:00
|
|
|
|
2015-04-08 18:47:32 +00:00
|
|
|
/* XXX I'm unsure about this. This is extremely dangerous and may result
|
|
|
|
in total black listing of certain transactions
|
2015-04-07 22:31:23 +00:00
|
|
|
if self.invalidHashes.Has(hash) {
|
|
|
|
return fmt.Errorf("Invalid transaction (%x)", hash[:4])
|
|
|
|
}
|
2015-04-08 18:47:32 +00:00
|
|
|
*/
|
2015-03-17 11:16:21 +00:00
|
|
|
if self.txs[hash] != nil {
|
2015-04-07 22:31:23 +00:00
|
|
|
return fmt.Errorf("Known transaction (%x)", hash[:4])
|
2015-01-07 00:21:45 +00:00
|
|
|
}
|
2014-12-01 19:18:09 +00:00
|
|
|
err := self.ValidateTransaction(tx)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
self.queueTx(hash, tx)
|
2015-04-08 18:47:32 +00:00
|
|
|
|
|
|
|
if glog.V(logger.Debug) {
|
2015-06-03 12:06:20 +00:00
|
|
|
var toname string
|
|
|
|
if to := tx.To(); to != nil {
|
|
|
|
toname = common.Bytes2Hex(to[:4])
|
|
|
|
} else {
|
|
|
|
toname = "[NEW_CONTRACT]"
|
|
|
|
}
|
|
|
|
// we can ignore the error here because From is
|
|
|
|
// verified in ValidateTransaction.
|
|
|
|
f, _ := tx.From()
|
|
|
|
from := common.Bytes2Hex(f[:4])
|
|
|
|
glog.Infof("(t) %x => %s (%v) %x\n", from, toname, tx.Value, hash)
|
2015-04-08 18:47:32 +00:00
|
|
|
}
|
2014-12-01 19:18:09 +00:00
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2014-12-04 22:54:07 +00:00
|
|
|
func (self *TxPool) Size() int {
|
2015-01-05 16:10:42 +00:00
|
|
|
return len(self.txs)
|
2014-12-04 22:54:07 +00:00
|
|
|
}
|
|
|
|
|
2015-02-19 21:33:22 +00:00
|
|
|
func (self *TxPool) Add(tx *types.Transaction) error {
|
|
|
|
self.mu.Lock()
|
|
|
|
defer self.mu.Unlock()
|
2015-04-07 22:31:23 +00:00
|
|
|
|
2015-02-19 21:33:22 +00:00
|
|
|
return self.add(tx)
|
|
|
|
}
|
2015-03-17 10:59:26 +00:00
|
|
|
|
2014-12-14 18:15:48 +00:00
|
|
|
func (self *TxPool) AddTransactions(txs []*types.Transaction) {
|
2015-02-19 21:33:22 +00:00
|
|
|
self.mu.Lock()
|
|
|
|
defer self.mu.Unlock()
|
|
|
|
|
2014-12-14 18:15:48 +00:00
|
|
|
for _, tx := range txs {
|
2015-02-19 21:33:22 +00:00
|
|
|
if err := self.add(tx); err != nil {
|
2015-04-26 09:19:40 +00:00
|
|
|
glog.V(logger.Debug).Infoln("tx error:", err)
|
2014-12-14 18:15:48 +00:00
|
|
|
} else {
|
2015-03-17 11:16:21 +00:00
|
|
|
h := tx.Hash()
|
2015-04-08 18:47:32 +00:00
|
|
|
glog.V(logger.Debug).Infof("tx %x\n", h[:4])
|
2014-12-14 18:15:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-05-07 15:27:17 +00:00
|
|
|
// GetTransaction allows you to check the pending and queued transaction in the
|
|
|
|
// transaction pool.
|
|
|
|
// It has two stategies, first check the pool (map) then check the queue
|
|
|
|
func (tp *TxPool) GetTransaction(hash common.Hash) *types.Transaction {
|
|
|
|
// check the txs first
|
|
|
|
if tx, ok := tp.txs[hash]; ok {
|
|
|
|
return tx
|
|
|
|
}
|
|
|
|
// check queue
|
|
|
|
for _, txs := range tp.queue {
|
2015-06-03 12:06:20 +00:00
|
|
|
if tx, ok := txs[hash]; ok {
|
|
|
|
return tx
|
2015-05-07 15:27:17 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-01-05 16:10:42 +00:00
|
|
|
func (self *TxPool) GetTransactions() (txs types.Transactions) {
|
2015-02-19 21:33:22 +00:00
|
|
|
self.mu.RLock()
|
|
|
|
defer self.mu.RUnlock()
|
|
|
|
|
2015-01-05 16:10:42 +00:00
|
|
|
txs = make(types.Transactions, self.Size())
|
2014-02-14 22:56:09 +00:00
|
|
|
i := 0
|
2015-01-05 16:10:42 +00:00
|
|
|
for _, tx := range self.txs {
|
|
|
|
txs[i] = tx
|
2014-02-14 22:56:09 +00:00
|
|
|
i++
|
2015-01-05 16:10:42 +00:00
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
return txs
|
2014-03-28 10:20:07 +00:00
|
|
|
}
|
|
|
|
|
2015-04-23 08:51:13 +00:00
|
|
|
func (self *TxPool) GetQueuedTransactions() types.Transactions {
|
|
|
|
self.mu.RLock()
|
|
|
|
defer self.mu.RUnlock()
|
|
|
|
|
2015-06-03 12:06:20 +00:00
|
|
|
var ret types.Transactions
|
|
|
|
for _, txs := range self.queue {
|
|
|
|
for _, tx := range txs {
|
|
|
|
ret = append(ret, tx)
|
|
|
|
}
|
2015-04-23 08:51:13 +00:00
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
sort.Sort(types.TxByNonce{ret})
|
|
|
|
return ret
|
2015-04-23 08:51:13 +00:00
|
|
|
}
|
|
|
|
|
2015-04-21 20:01:04 +00:00
|
|
|
func (self *TxPool) RemoveTransactions(txs types.Transactions) {
|
2015-02-19 21:33:22 +00:00
|
|
|
self.mu.Lock()
|
|
|
|
defer self.mu.Unlock()
|
2014-10-27 15:52:58 +00:00
|
|
|
for _, tx := range txs {
|
2015-05-05 21:09:18 +00:00
|
|
|
self.removeTx(tx.Hash())
|
2014-10-27 15:52:58 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-02-19 21:33:22 +00:00
|
|
|
func (pool *TxPool) Flush() {
|
2015-03-17 10:59:26 +00:00
|
|
|
pool.txs = make(map[common.Hash]*types.Transaction)
|
2014-02-14 22:56:09 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (pool *TxPool) Stop() {
|
|
|
|
pool.Flush()
|
2015-04-21 20:01:04 +00:00
|
|
|
close(pool.quit)
|
2014-03-17 09:33:03 +00:00
|
|
|
|
2015-04-08 18:47:32 +00:00
|
|
|
glog.V(logger.Info).Infoln("TX Pool stopped")
|
2014-02-14 22:56:09 +00:00
|
|
|
}
|
2015-04-21 20:01:04 +00:00
|
|
|
|
2015-06-03 12:06:20 +00:00
|
|
|
func (self *TxPool) queueTx(hash common.Hash, tx *types.Transaction) {
|
2015-06-01 20:00:48 +00:00
|
|
|
from, _ := tx.From() // already validated
|
2015-06-03 12:06:20 +00:00
|
|
|
if self.queue[from] == nil {
|
|
|
|
self.queue[from] = make(map[common.Hash]*types.Transaction)
|
|
|
|
}
|
|
|
|
self.queue[from][hash] = tx
|
2015-04-21 21:20:27 +00:00
|
|
|
}
|
|
|
|
|
2015-06-03 12:06:20 +00:00
|
|
|
func (pool *TxPool) addTx(hash common.Hash, tx *types.Transaction) {
|
|
|
|
if _, ok := pool.txs[hash]; !ok {
|
|
|
|
pool.txs[hash] = tx
|
2015-04-22 15:56:06 +00:00
|
|
|
// Notify the subscribers. This event is posted in a goroutine
|
|
|
|
// because it's possible that somewhere during the post "Remove transaction"
|
|
|
|
// gets called which will then wait for the global tx pool lock and deadlock.
|
|
|
|
go pool.eventMux.Post(TxPreEvent{tx})
|
2015-04-21 21:20:27 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-21 20:01:04 +00:00
|
|
|
// check queue will attempt to insert
|
|
|
|
func (pool *TxPool) checkQueue() {
|
|
|
|
pool.mu.Lock()
|
|
|
|
defer pool.mu.Unlock()
|
|
|
|
|
2015-04-23 09:09:58 +00:00
|
|
|
statedb := pool.currentState()
|
2015-06-03 12:06:20 +00:00
|
|
|
var addq txQueue
|
2015-04-21 20:01:04 +00:00
|
|
|
for address, txs := range pool.queue {
|
2015-06-03 12:06:20 +00:00
|
|
|
curnonce := statedb.GetNonce(address)
|
|
|
|
addq := addq[:0]
|
|
|
|
for hash, tx := range txs {
|
|
|
|
if tx.AccountNonce < curnonce {
|
|
|
|
// Drop queued transactions whose nonce is lower than
|
|
|
|
// the account nonce because they have been processed.
|
|
|
|
delete(txs, hash)
|
|
|
|
} else {
|
|
|
|
// Collect the remaining transactions for the next pass.
|
|
|
|
addq = append(addq, txQueueEntry{hash, tx})
|
2015-04-21 20:01:04 +00:00
|
|
|
}
|
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
// Find the next consecutive nonce range starting at the
|
|
|
|
// current account nonce.
|
|
|
|
sort.Sort(addq)
|
|
|
|
for _, e := range addq {
|
|
|
|
if e.AccountNonce != curnonce {
|
2015-04-21 20:01:04 +00:00
|
|
|
break
|
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
curnonce++
|
|
|
|
delete(txs, e.hash)
|
|
|
|
pool.addTx(e.hash, e.Transaction)
|
2015-04-21 20:01:04 +00:00
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
// Delete the entire queue entry if it became empty.
|
|
|
|
if len(txs) == 0 {
|
2015-04-21 20:01:04 +00:00
|
|
|
delete(pool.queue, address)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-04-23 09:09:58 +00:00
|
|
|
|
2015-04-29 22:20:59 +00:00
|
|
|
func (pool *TxPool) removeTx(hash common.Hash) {
|
|
|
|
// delete from pending pool
|
|
|
|
delete(pool.txs, hash)
|
|
|
|
// delete from queue
|
|
|
|
for address, txs := range pool.queue {
|
2015-06-03 12:06:20 +00:00
|
|
|
if _, ok := txs[hash]; ok {
|
|
|
|
if len(txs) == 1 {
|
|
|
|
// if only one tx, remove entire address entry.
|
|
|
|
delete(pool.queue, address)
|
|
|
|
} else {
|
|
|
|
delete(txs, hash)
|
2015-04-29 22:20:59 +00:00
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
break
|
2015-04-29 22:20:59 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-23 09:09:58 +00:00
|
|
|
func (pool *TxPool) validatePool() {
|
|
|
|
pool.mu.Lock()
|
|
|
|
defer pool.mu.Unlock()
|
|
|
|
|
|
|
|
for hash, tx := range pool.txs {
|
2015-04-26 09:19:40 +00:00
|
|
|
if err := pool.ValidateTransaction(tx); err != nil {
|
|
|
|
if glog.V(logger.Info) {
|
|
|
|
glog.Infof("removed tx (%x) from pool: %v\n", hash[:4], err)
|
2015-04-23 09:09:58 +00:00
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
delete(pool.txs, hash)
|
2015-04-23 09:09:58 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-06-03 12:06:20 +00:00
|
|
|
|
|
|
|
type txQueue []txQueueEntry
|
|
|
|
|
|
|
|
type txQueueEntry struct {
|
|
|
|
hash common.Hash
|
|
|
|
*types.Transaction
|
|
|
|
}
|
|
|
|
|
|
|
|
func (q txQueue) Len() int { return len(q) }
|
|
|
|
func (q txQueue) Swap(i, j int) { q[i], q[j] = q[j], q[i] }
|
|
|
|
func (q txQueue) Less(i, j int) bool { return q[i].AccountNonce < q[j].AccountNonce }
|