tx_pool.go 29.6 KB
Newer Older
F
Felix Lange 已提交
1
// Copyright 2014 The go-ethereum Authors
2
// This file is part of the go-ethereum library.
F
Felix Lange 已提交
3
//
4
// The go-ethereum library is free software: you can redistribute it and/or modify
F
Felix Lange 已提交
5 6 7 8
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
9
// The go-ethereum library is distributed in the hope that it will be useful,
F
Felix Lange 已提交
10
// but WITHOUT ANY WARRANTY; without even the implied warranty of
11
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
F
Felix Lange 已提交
12 13 14
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
15
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
F
Felix Lange 已提交
16

O
obscuren 已提交
17
package core
O
obscuren 已提交
18 19

import (
20
	"errors"
O
obscuren 已提交
21
	"fmt"
O
obscuren 已提交
22
	"math/big"
23
	"sort"
24
	"sync"
25
	"time"
26

O
obscuren 已提交
27
	"github.com/ethereum/go-ethereum/common"
O
obscuren 已提交
28
	"github.com/ethereum/go-ethereum/core/state"
29
	"github.com/ethereum/go-ethereum/core/types"
O
obscuren 已提交
30
	"github.com/ethereum/go-ethereum/event"
31
	"github.com/ethereum/go-ethereum/log"
32
	"github.com/ethereum/go-ethereum/metrics"
33
	"github.com/ethereum/go-ethereum/params"
34
	"gopkg.in/karalabe/cookiejar.v2/collections/prque"
O
obscuren 已提交
35 36
)

37
var (
38
	// Transaction Pool Errors
39 40 41 42 43 44 45 46 47
	ErrInvalidSender      = errors.New("invalid sender")
	ErrNonce              = errors.New("nonce too low")
	ErrUnderpriced        = errors.New("transaction underpriced")
	ErrReplaceUnderpriced = errors.New("replacement transaction underpriced")
	ErrBalance            = errors.New("insufficient balance")
	ErrInsufficientFunds  = errors.New("insufficient funds for gas * price + value")
	ErrIntrinsicGas       = errors.New("intrinsic gas too low")
	ErrGasLimit           = errors.New("exceeds block gas limit")
	ErrNegativeValue      = errors.New("negative value")
48
)
Z
zelig 已提交
49

50
var (
51 52
	evictionInterval    = time.Minute     // Time interval to check for evictable transactions
	statsReportInterval = 8 * time.Second // Time interval to report transaction pool stats
53 54
)

55 56 57 58 59 60 61 62 63 64 65 66 67 68
var (
	// Metrics for the pending pool
	pendingDiscardCounter = metrics.NewCounter("txpool/pending/discard")
	pendingReplaceCounter = metrics.NewCounter("txpool/pending/replace")
	pendingRLCounter      = metrics.NewCounter("txpool/pending/ratelimit") // Dropped due to rate limiting
	pendingNofundsCounter = metrics.NewCounter("txpool/pending/nofunds")   // Dropped due to out-of-funds

	// Metrics for the queued pool
	queuedDiscardCounter = metrics.NewCounter("txpool/queued/discard")
	queuedReplaceCounter = metrics.NewCounter("txpool/queued/replace")
	queuedRLCounter      = metrics.NewCounter("txpool/queued/ratelimit") // Dropped due to rate limiting
	queuedNofundsCounter = metrics.NewCounter("txpool/queued/nofunds")   // Dropped due to out-of-funds

	// General tx metrics
69 70
	invalidTxCounter     = metrics.NewCounter("txpool/invalid")
	underpricedTxCounter = metrics.NewCounter("txpool/underpriced")
71 72
)

73
type stateFn func() (*state.StateDB, error)
74

75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
// TxPoolConfig are the configuration parameters of the transaction pool.
type TxPoolConfig struct {
	PriceLimit uint64 // Minimum gas price to enforce for acceptance into the pool
	PriceBump  uint64 // Minimum price bump percentage to replace an already existing transaction (nonce)

	AccountSlots uint64 // Minimum number of executable transaction slots guaranteed per account
	GlobalSlots  uint64 // Maximum number of executable transaction slots for all accounts
	AccountQueue uint64 // Maximum number of non-executable transaction slots permitted per account
	GlobalQueue  uint64 // Maximum number of non-executable transaction slots for all accounts

	Lifetime time.Duration // Maximum amount of time non-executable transaction are queued
}

// DefaultTxPoolConfig contains the default configurations for the transaction
// pool.
var DefaultTxPoolConfig = TxPoolConfig{
	PriceLimit: 1,
	PriceBump:  10,

	AccountSlots: 16,
	GlobalSlots:  4096,
	AccountQueue: 64,
	GlobalQueue:  1024,

	Lifetime: 3 * time.Hour,
}

// sanitize checks the provided user configurations and changes anything that's
// unreasonable or unworkable.
func (config *TxPoolConfig) sanitize() TxPoolConfig {
	conf := *config
	if conf.PriceLimit < 1 {
		log.Warn("Sanitizing invalid txpool price limit", "provided", conf.PriceLimit, "updated", DefaultTxPoolConfig.PriceLimit)
		conf.PriceLimit = DefaultTxPoolConfig.PriceLimit
	}
	if conf.PriceBump < 1 {
		log.Warn("Sanitizing invalid txpool price bump", "provided", conf.PriceBump, "updated", DefaultTxPoolConfig.PriceBump)
		conf.PriceBump = DefaultTxPoolConfig.PriceBump
	}
	return conf
}

117 118 119 120 121 122 123
// TxPool contains all currently known transactions. Transactions
// enter the pool when they are received from the network or submitted
// locally. They exit the pool when they are included in the blockchain.
//
// The pool separates processable transactions (which can be applied to the
// current state) and future transactions. Transactions move between those
// two states over time as they are received and processed.
O
obscuren 已提交
124
type TxPool struct {
125 126
	config       TxPoolConfig
	chainconfig  *params.ChainConfig
127
	currentState stateFn // The state function which will allow us to do some pre checks
128
	pendingState *state.ManagedState
129
	gasLimit     func() *big.Int // The current gas limit function callback
130
	gasPrice     *big.Int
131
	eventMux     *event.TypeMux
132
	events       *event.TypeMuxSubscription
133
	locals       *txSet
J
Jeffrey Wilcke 已提交
134
	signer       types.Signer
135
	mu           sync.RWMutex
136

137 138
	pending map[common.Address]*txList         // All currently processable transactions
	queue   map[common.Address]*txList         // Queued but non-processable transactions
139
	beats   map[common.Address]time.Time       // Last heartbeat from each known account
140 141
	all     map[common.Hash]*types.Transaction // All transactions to allow lookups
	priced  *txPricedList                      // All transactions sorted by price
142

143 144
	wg   sync.WaitGroup // for shutdown sync
	quit chan struct{}
145

146
	homestead bool
O
obscuren 已提交
147 148
}

149 150 151 152 153 154 155
// NewTxPool creates a new transaction pool to gather, sort and filter inbound
// trnsactions from the network.
func NewTxPool(config TxPoolConfig, chainconfig *params.ChainConfig, eventMux *event.TypeMux, currentStateFn stateFn, gasLimitFn func() *big.Int) *TxPool {
	// Sanitize the input to ensure no vulnerable gas prices are set
	config = (&config).sanitize()

	// Create the transaction pool with its initial settings
156
	pool := &TxPool{
157
		config:       config,
158 159
		chainconfig:  chainconfig,
		signer:       types.NewEIP155Signer(chainconfig.ChainId),
160 161
		pending:      make(map[common.Address]*txList),
		queue:        make(map[common.Address]*txList),
162
		beats:        make(map[common.Address]time.Time),
163
		all:          make(map[common.Hash]*types.Transaction),
164 165 166
		eventMux:     eventMux,
		currentState: currentStateFn,
		gasLimit:     gasLimitFn,
167
		gasPrice:     new(big.Int).SetUint64(config.PriceLimit),
168
		pendingState: nil,
169 170
		locals:       newTxSet(),
		events:       eventMux.Subscribe(ChainHeadEvent{}, RemovedTransactionEvent{}),
171
		quit:         make(chan struct{}),
O
obscuren 已提交
172
	}
173
	pool.priced = newTxPricedList(&pool.all)
174 175
	pool.resetState()

176
	// Start the various events loops and return
177
	pool.wg.Add(2)
178
	go pool.eventLoop()
179
	go pool.expirationLoop()
180 181

	return pool
182 183
}

184
func (pool *TxPool) eventLoop() {
185 186
	defer pool.wg.Done()

187 188 189 190 191 192
	// Start a ticker and keep track of interesting pool stats to report
	var prevPending, prevQueued, prevStales int

	report := time.NewTicker(statsReportInterval)
	defer report.Stop()

O
obscuren 已提交
193 194 195
	// Track chain events. When a chain events occurs (new chain canon block)
	// we need to know the new state. The new state will help us determine
	// the nonces in the managed state
196 197 198 199 200 201 202 203 204 205 206
	for {
		select {
		// Handle any events fired by the system
		case ev, ok := <-pool.events.Chan():
			if !ok {
				return
			}
			switch ev := ev.Data.(type) {
			case ChainHeadEvent:
				pool.mu.Lock()
				if ev.Block != nil {
207
					if pool.chainconfig.IsHomestead(ev.Block.Number()) {
208 209
						pool.homestead = true
					}
J
Jeffrey Wilcke 已提交
210
				}
211 212 213 214 215
				pool.resetState()
				pool.mu.Unlock()

			case RemovedTransactionEvent:
				pool.AddBatch(ev.Txs)
216 217
			}

218 219 220 221 222 223 224 225 226 227 228
		// Handle stats reporting ticks
		case <-report.C:
			pool.mu.RLock()
			pending, queued := pool.stats()
			stales := pool.priced.stales
			pool.mu.RUnlock()

			if pending != prevPending || queued != prevQueued || stales != prevStales {
				log.Debug("Transaction pool status report", "executable", pending, "queued", queued, "stales", stales)
				prevPending, prevQueued, prevStales = pending, queued, stales
			}
229
		}
230
	}
O
obscuren 已提交
231 232
}

233
func (pool *TxPool) resetState() {
234 235
	currentState, err := pool.currentState()
	if err != nil {
236
		log.Error("Failed reset txpool state", "err", err)
237 238
		return
	}
239
	pool.pendingState = state.ManageState(currentState)
240 241 242 243 244

	// validate the pool of pending transactions, this will remove
	// any transactions that have been included in the block or
	// have been invalidated because of another transaction (e.g.
	// higher gas price)
245
	pool.demoteUnexecutables(currentState)
246 247 248

	// Update all accounts to the latest known pending nonce
	for addr, list := range pool.pending {
249 250
		txs := list.Flatten() // Heavy but will be cached and is needed by the miner anyway
		pool.pendingState.SetNonce(addr, txs[len(txs)-1].Nonce()+1)
251 252 253
	}
	// Check the queue and move transactions over to the pending if possible
	// or remove those that have become invalid
254
	pool.promoteExecutables(currentState, nil)
255 256
}

257
// Stop terminates the transaction pool.
258 259
func (pool *TxPool) Stop() {
	pool.events.Unsubscribe()
260
	close(pool.quit)
261
	pool.wg.Wait()
262 263

	log.Info("Transaction pool stopped")
264 265
}

266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286
// GasPrice returns the current gas price enforced by the transaction pool.
func (pool *TxPool) GasPrice() *big.Int {
	pool.mu.RLock()
	defer pool.mu.RUnlock()

	return new(big.Int).Set(pool.gasPrice)
}

// SetGasPrice updates the minimum price required by the transaction pool for a
// new transaction, and drops all transactions below this threshold.
func (pool *TxPool) SetGasPrice(price *big.Int) {
	pool.mu.Lock()
	defer pool.mu.Unlock()

	pool.gasPrice = price
	for _, tx := range pool.priced.Cap(price, pool.locals) {
		pool.removeTx(tx.Hash())
	}
	log.Info("Transaction pool price threshold updated", "price", price)
}

287
// State returns the virtual managed state of the transaction pool.
288
func (pool *TxPool) State() *state.ManagedState {
289 290
	pool.mu.RLock()
	defer pool.mu.RUnlock()
291

292
	return pool.pendingState
293 294
}

295 296
// Stats retrieves the current pool stats, namely the number of pending and the
// number of queued (non-executable) transactions.
297
func (pool *TxPool) Stats() (int, int) {
298 299 300
	pool.mu.RLock()
	defer pool.mu.RUnlock()

301 302 303 304 305 306 307
	return pool.stats()
}

// stats retrieves the current pool stats, namely the number of pending and the
// number of queued (non-executable) transactions.
func (pool *TxPool) stats() (int, int) {
	pending := 0
308 309
	for _, list := range pool.pending {
		pending += list.Len()
310
	}
311
	queued := 0
312 313
	for _, list := range pool.queue {
		queued += list.Len()
314
	}
315
	return pending, queued
316 317
}

318
// Content retrieves the data content of the transaction pool, returning all the
319 320
// pending as well as queued transactions, grouped by account and sorted by nonce.
func (pool *TxPool) Content() (map[common.Address]types.Transactions, map[common.Address]types.Transactions) {
321 322 323
	pool.mu.RLock()
	defer pool.mu.RUnlock()

324 325 326 327 328 329 330
	pending := make(map[common.Address]types.Transactions)
	for addr, list := range pool.pending {
		pending[addr] = list.Flatten()
	}
	queued := make(map[common.Address]types.Transactions)
	for addr, list := range pool.queue {
		queued[addr] = list.Flatten()
331 332 333 334
	}
	return pending, queued
}

335 336 337
// Pending retrieves all currently processable transactions, groupped by origin
// account and sorted by nonce. The returned transaction set is a copy and can be
// freely modified by calling code.
338
func (pool *TxPool) Pending() (map[common.Address]types.Transactions, error) {
339 340 341 342 343 344 345
	pool.mu.Lock()
	defer pool.mu.Unlock()

	pending := make(map[common.Address]types.Transactions)
	for addr, list := range pool.pending {
		pending[addr] = list.Flatten()
	}
346
	return pending, nil
347 348
}

349 350 351 352 353
// SetLocal marks a transaction as local, skipping gas price
//  check against local miner minimum in the future
func (pool *TxPool) SetLocal(tx *types.Transaction) {
	pool.mu.Lock()
	defer pool.mu.Unlock()
354
	pool.locals.add(tx.Hash())
355 356
}

357 358 359
// validateTx checks whether a transaction is valid according
// to the consensus rules.
func (pool *TxPool) validateTx(tx *types.Transaction) error {
360
	local := pool.locals.contains(tx.Hash())
361
	// Drop transactions under our own minimal accepted gas price
362 363
	if !local && pool.gasPrice.Cmp(tx.GasPrice()) > 0 {
		return ErrUnderpriced
364 365
	}

366 367 368 369 370
	currentState, err := pool.currentState()
	if err != nil {
		return err
	}

J
Jeffrey Wilcke 已提交
371
	from, err := types.Sender(pool.signer, tx)
372
	if err != nil {
F
Felix Lange 已提交
373
		return ErrInvalidSender
O
obscuren 已提交
374
	}
375
	// Last but not least check for nonce errors
376
	if currentState.GetNonce(from) > tx.Nonce() {
377 378 379
		return ErrNonce
	}

O
obscuren 已提交
380 381
	// Check the transaction doesn't exceed the current
	// block limit gas.
382
	if pool.gasLimit().Cmp(tx.Gas()) < 0 {
383 384 385
		return ErrGasLimit
	}

O
obscuren 已提交
386 387 388
	// Transactions can't be negative. This may never happen
	// using RLP decoded transactions but may occur if you create
	// a transaction using the RPC for example.
389
	if tx.Value().Sign() < 0 {
390 391 392
		return ErrNegativeValue
	}

O
obscuren 已提交
393 394
	// Transactor should have enough funds to cover the costs
	// cost == V + GP * GL
395
	if currentState.GetBalance(from).Cmp(tx.Cost()) < 0 {
O
obscuren 已提交
396 397 398
		return ErrInsufficientFunds
	}

J
Jeffrey Wilcke 已提交
399
	intrGas := IntrinsicGas(tx.Data(), tx.To() == nil, pool.homestead)
400
	if tx.Gas().Cmp(intrGas) < 0 {
O
obscuren 已提交
401 402 403 404
		return ErrIntrinsicGas
	}

	return nil
O
obscuren 已提交
405 406
}

407
// add validates a transaction and inserts it into the non-executable queue for
408 409 410 411
// later pending promotion and execution. If the transaction is a replacement for
// an already pending or queued one, it overwrites the previous and returns this
// so outer code doesn't uselessly call promote.
func (pool *TxPool) add(tx *types.Transaction) (bool, error) {
P
Péter Szilágyi 已提交
412
	// If the transaction is already known, discard it
413
	hash := tx.Hash()
414
	if pool.all[hash] != nil {
415
		log.Trace("Discarding already known transaction", "hash", hash)
416
		return false, fmt.Errorf("known transaction: %x", hash)
417
	}
418
	// If the transaction fails basic validation, discard it
419
	if err := pool.validateTx(tx); err != nil {
420
		log.Trace("Discarding invalid transaction", "hash", hash, "err", err)
421
		invalidTxCounter.Inc(1)
422 423 424
		return false, err
	}
	// If the transaction pool is full, discard underpriced transactions
425
	if uint64(len(pool.all)) >= pool.config.GlobalSlots+pool.config.GlobalQueue {
426 427 428 429 430 431 432
		// If the new transaction is underpriced, don't accept it
		if pool.priced.Underpriced(tx, pool.locals) {
			log.Trace("Discarding underpriced transaction", "hash", hash, "price", tx.GasPrice())
			underpricedTxCounter.Inc(1)
			return false, ErrUnderpriced
		}
		// New transaction is better than our worse ones, make room for it
433
		drop := pool.priced.Discard(len(pool.all)-int(pool.config.GlobalSlots+pool.config.GlobalQueue-1), pool.locals)
434 435 436 437 438
		for _, tx := range drop {
			log.Trace("Discarding freshly underpriced transaction", "hash", tx.Hash(), "price", tx.GasPrice())
			underpricedTxCounter.Inc(1)
			pool.removeTx(tx.Hash())
		}
439
	}
440 441 442 443
	// If the transaction is replacing an already pending one, do directly
	from, _ := types.Sender(pool.signer, tx) // already validated
	if list := pool.pending[from]; list != nil && list.Overlaps(tx) {
		// Nonce already pending, check if required price bump is met
444
		inserted, old := list.Add(tx, pool.config.PriceBump)
445 446 447 448 449 450 451 452 453 454 455 456
		if !inserted {
			pendingDiscardCounter.Inc(1)
			return false, ErrReplaceUnderpriced
		}
		// New transaction is better, replace old one
		if old != nil {
			delete(pool.all, old.Hash())
			pool.priced.Removed()
			pendingReplaceCounter.Inc(1)
		}
		pool.all[tx.Hash()] = tx
		pool.priced.Put(tx)
O
obscuren 已提交
457

458 459 460 461 462 463 464 465 466 467
		log.Trace("Pooled new executable transaction", "hash", hash, "from", from, "to", tx.To())
		return old != nil, nil
	}
	// New transaction isn't replacing a pending one, push into queue
	replace, err := pool.enqueueTx(hash, tx)
	if err != nil {
		return false, err
	}
	log.Trace("Pooled new future transaction", "hash", hash, "from", from, "to", tx.To())
	return replace, nil
468 469
}

470
// enqueueTx inserts a new transaction into the non-executable transaction queue.
471 472
//
// Note, this method assumes the pool lock is held!
473
func (pool *TxPool) enqueueTx(hash common.Hash, tx *types.Transaction) (bool, error) {
474
	// Try to insert the transaction into the future queue
J
Jeffrey Wilcke 已提交
475
	from, _ := types.Sender(pool.signer, tx) // already validated
476 477
	if pool.queue[from] == nil {
		pool.queue[from] = newTxList(false)
478
	}
479
	inserted, old := pool.queue[from].Add(tx, pool.config.PriceBump)
480
	if !inserted {
481
		// An older transaction was better, discard this
482
		queuedDiscardCounter.Inc(1)
483
		return false, ErrReplaceUnderpriced
484 485 486 487
	}
	// Discard any previous transaction and mark this
	if old != nil {
		delete(pool.all, old.Hash())
488
		pool.priced.Removed()
489
		queuedReplaceCounter.Inc(1)
490
	}
491
	pool.all[hash] = tx
492 493
	pool.priced.Put(tx)
	return old != nil, nil
494 495
}

496 497 498 499 500 501 502 503 504
// promoteTx adds a transaction to the pending (processable) list of transactions.
//
// Note, this method assumes the pool lock is held!
func (pool *TxPool) promoteTx(addr common.Address, hash common.Hash, tx *types.Transaction) {
	// Try to insert the transaction into the pending queue
	if pool.pending[addr] == nil {
		pool.pending[addr] = newTxList(true)
	}
	list := pool.pending[addr]
505

506
	inserted, old := list.Add(tx, pool.config.PriceBump)
507 508 509
	if !inserted {
		// An older transaction was better, discard this
		delete(pool.all, hash)
510 511
		pool.priced.Removed()

512
		pendingDiscardCounter.Inc(1)
513
		return
514
	}
515 516 517
	// Otherwise discard any previous transaction and mark this
	if old != nil {
		delete(pool.all, old.Hash())
518 519
		pool.priced.Removed()

520
		pendingReplaceCounter.Inc(1)
521
	}
522 523 524 525 526
	// Failsafe to work around direct pending inserts (tests)
	if pool.all[hash] == nil {
		pool.all[hash] = tx
		pool.priced.Put(tx)
	}
527
	// Set the potentially new pending nonce and notify any subsystems of the new tx
528
	pool.beats[addr] = time.Now()
529
	pool.pendingState.SetNonce(addr, tx.Nonce()+1)
530
	go pool.eventMux.Post(TxPreEvent{tx})
531 532
}

533
// Add queues a single transaction in the pool if it is valid.
534 535 536
func (pool *TxPool) Add(tx *types.Transaction) error {
	pool.mu.Lock()
	defer pool.mu.Unlock()
537

538 539 540
	// Try to inject the transaction and update any state
	replace, err := pool.add(tx)
	if err != nil {
541
		return err
542
	}
543 544
	// If we added a new transaction, run promotion checks and return
	if !replace {
545 546 547 548
		state, err := pool.currentState()
		if err != nil {
			return err
		}
549 550
		from, _ := types.Sender(pool.signer, tx) // already validated
		pool.promoteExecutables(state, []common.Address{from})
551
	}
552
	return nil
553
}
554

555
// AddBatch attempts to queue a batch of transactions.
556
func (pool *TxPool) AddBatch(txs []*types.Transaction) error {
557 558
	pool.mu.Lock()
	defer pool.mu.Unlock()
559

560
	// Add the batch of transaction, tracking the accepted ones
561
	dirty := make(map[common.Address]struct{})
Z
zelig 已提交
562
	for _, tx := range txs {
563 564
		if replace, err := pool.add(tx); err == nil {
			if !replace {
565 566
				from, _ := types.Sender(pool.signer, tx) // already validated
				dirty[from] = struct{}{}
567
			}
Z
zelig 已提交
568 569
		}
	}
570
	// Only reprocess the internal state if something was actually added
571 572 573 574 575 576 577 578
	if len(dirty) > 0 {
		state, err := pool.currentState()
		if err != nil {
			return err
		}
		addrs := make([]common.Address, 0, len(dirty))
		for addr, _ := range dirty {
			addrs = append(addrs, addr)
579
		}
580
		pool.promoteExecutables(state, addrs)
581 582
	}
	return nil
Z
zelig 已提交
583 584
}

585
// Get returns a transaction if it is contained in the pool
586
// and nil otherwise.
587 588 589
func (pool *TxPool) Get(hash common.Hash) *types.Transaction {
	pool.mu.RLock()
	defer pool.mu.RUnlock()
590

591
	return pool.all[hash]
592 593
}

594 595 596 597
// Remove removes the transaction with the given hash from the pool.
func (pool *TxPool) Remove(hash common.Hash) {
	pool.mu.Lock()
	defer pool.mu.Unlock()
598

599
	pool.removeTx(hash)
600 601
}

602 603
// RemoveBatch removes all given transactions from the pool.
func (pool *TxPool) RemoveBatch(txs types.Transactions) {
604 605
	pool.mu.Lock()
	defer pool.mu.Unlock()
606

607 608 609
	for _, tx := range txs {
		pool.removeTx(tx.Hash())
	}
610 611
}

612 613
// removeTx removes a single transaction from the queue, moving all subsequent
// transactions back to the future queue.
614
func (pool *TxPool) removeTx(hash common.Hash) {
615 616 617 618 619
	// Fetch the transaction we wish to delete
	tx, ok := pool.all[hash]
	if !ok {
		return
	}
J
Jeffrey Wilcke 已提交
620
	addr, _ := types.Sender(pool.signer, tx) // already validated during insertion
621

622
	// Remove it from the list of known transactions
623
	delete(pool.all, hash)
624
	pool.priced.Removed()
625

626 627 628
	// Remove the transaction from the pending lists and reset the account nonce
	if pending := pool.pending[addr]; pending != nil {
		if removed, invalids := pending.Remove(tx); removed {
629
			// If no more transactions are left, remove the list
630 631
			if pending.Empty() {
				delete(pool.pending, addr)
632
				delete(pool.beats, addr)
633
			} else {
634
				// Otherwise postpone any invalidated transactions
635 636 637 638
				for _, tx := range invalids {
					pool.enqueueTx(tx.Hash(), tx)
				}
			}
639 640 641 642
			// Update the account nonce if needed
			if nonce := tx.Nonce(); pool.pendingState.GetNonce(addr) > nonce {
				pool.pendingState.SetNonce(addr, tx.Nonce())
			}
643
		}
644
	}
645 646 647 648 649 650
	// Transaction is in the future queue
	if future := pool.queue[addr]; future != nil {
		future.Remove(tx)
		if future.Empty() {
			delete(pool.queue, addr)
		}
651 652 653
	}
}

654 655 656
// promoteExecutables moves transactions that have become processable from the
// future queue to the set of pending transactions. During this process, all
// invalidated transactions (low nonce, low balance) are deleted.
657
func (pool *TxPool) promoteExecutables(state *state.StateDB, accounts []common.Address) {
658 659
	gaslimit := pool.gasLimit()

660 661 662 663 664 665 666
	// Gather all the accounts potentially needing updates
	if accounts == nil {
		accounts = make([]common.Address, 0, len(pool.queue))
		for addr, _ := range pool.queue {
			accounts = append(accounts, addr)
		}
	}
667
	// Iterate over all accounts and promote any executable transactions
668
	queued := uint64(0)
669 670 671 672 673
	for _, addr := range accounts {
		list := pool.queue[addr]
		if list == nil {
			continue // Just in case someone calls with a non existing account
		}
674 675
		// Drop all transactions that are deemed too old (low nonce)
		for _, tx := range list.Forward(state.GetNonce(addr)) {
676
			hash := tx.Hash()
677
			log.Trace("Removed old queued transaction", "hash", hash)
678
			delete(pool.all, hash)
679
			pool.priced.Removed()
680
		}
681 682
		// Drop all transactions that are too costly (low balance or out of gas)
		drops, _ := list.Filter(state.GetBalance(addr), gaslimit)
683
		for _, tx := range drops {
684
			hash := tx.Hash()
685
			log.Trace("Removed unpayable queued transaction", "hash", hash)
686
			delete(pool.all, hash)
687
			pool.priced.Removed()
688
			queuedNofundsCounter.Inc(1)
689
		}
690 691
		// Gather all executable transactions and promote them
		for _, tx := range list.Ready(pool.pendingState.GetNonce(addr)) {
692
			hash := tx.Hash()
693
			log.Trace("Promoting queued transaction", "hash", hash)
694
			pool.promoteTx(addr, hash, tx)
695 696
		}
		// Drop all transactions over the allowed limit
697
		for _, tx := range list.Cap(int(pool.config.AccountQueue)) {
698
			hash := tx.Hash()
699
			log.Trace("Removed cap-exceeding queued transaction", "hash", hash)
700
			delete(pool.all, hash)
701
			pool.priced.Removed()
702
			queuedRLCounter.Inc(1)
703
		}
704 705
		queued += uint64(list.Len())

706
		// Delete the entire queue entry if it became empty.
707 708
		if list.Empty() {
			delete(pool.queue, addr)
709 710
		}
	}
711 712 713 714 715
	// If the pending limit is overflown, start equalizing allowances
	pending := uint64(0)
	for _, list := range pool.pending {
		pending += uint64(list.Len())
	}
716
	if pending > pool.config.GlobalSlots {
717
		pendingBeforeCap := pending
718 719 720 721
		// Assemble a spam order to penalize large transactors first
		spammers := prque.New()
		for addr, list := range pool.pending {
			// Only evict transactions from high rollers
722
			if uint64(list.Len()) > pool.config.AccountSlots {
723 724
				// Skip local accounts as pools should maintain backlogs for themselves
				for _, tx := range list.txs.items {
725
					if !pool.locals.contains(tx.Hash()) {
726 727 728 729 730 731 732 733
						spammers.Push(addr, float32(list.Len()))
					}
					break // Checking on transaction for locality is enough
				}
			}
		}
		// Gradually drop transactions from offenders
		offenders := []common.Address{}
734
		for pending > pool.config.GlobalSlots && !spammers.Empty() {
735 736 737 738 739 740 741 742 743 744
			// Retrieve the next offender if not local address
			offender, _ := spammers.Pop()
			offenders = append(offenders, offender.(common.Address))

			// Equalize balances until all the same or below threshold
			if len(offenders) > 1 {
				// Calculate the equalization threshold for all current offenders
				threshold := pool.pending[offender.(common.Address)].Len()

				// Iteratively reduce all offenders until below limit or threshold reached
745
				for pending > pool.config.GlobalSlots && pool.pending[offenders[len(offenders)-2]].Len() > threshold {
746 747 748 749 750 751 752 753 754
					for i := 0; i < len(offenders)-1; i++ {
						list := pool.pending[offenders[i]]
						list.Cap(list.Len() - 1)
						pending--
					}
				}
			}
		}
		// If still above threshold, reduce to limit or min allowance
755 756
		if pending > pool.config.GlobalSlots && len(offenders) > 0 {
			for pending > pool.config.GlobalSlots && uint64(pool.pending[offenders[len(offenders)-1]].Len()) > pool.config.AccountSlots {
757 758 759 760 761 762 763
				for _, addr := range offenders {
					list := pool.pending[addr]
					list.Cap(list.Len() - 1)
					pending--
				}
			}
		}
764
		pendingRLCounter.Inc(int64(pendingBeforeCap - pending))
765
	}
766
	// If we've queued more transactions than the hard limit, drop oldest ones
767
	if queued > pool.config.GlobalQueue {
768 769
		// Sort all accounts with queued transactions by heartbeat
		addresses := make(addresssByHeartbeat, 0, len(pool.queue))
F
Felix Lange 已提交
770
		for addr := range pool.queue {
771 772 773 774 775
			addresses = append(addresses, addressByHeartbeat{addr, pool.beats[addr]})
		}
		sort.Sort(addresses)

		// Drop transactions until the total is below the limit
776
		for drop := queued - pool.config.GlobalQueue; drop > 0; {
777 778 779 780 781 782 783 784 785 786 787
			addr := addresses[len(addresses)-1]
			list := pool.queue[addr.address]

			addresses = addresses[:len(addresses)-1]

			// Drop all transactions if they are less than the overflow
			if size := uint64(list.Len()); size <= drop {
				for _, tx := range list.Flatten() {
					pool.removeTx(tx.Hash())
				}
				drop -= size
788
				queuedRLCounter.Inc(int64(size))
789 790 791 792 793 794 795
				continue
			}
			// Otherwise drop only last few transactions
			txs := list.Flatten()
			for i := len(txs) - 1; i >= 0 && drop > 0; i-- {
				pool.removeTx(txs[i].Hash())
				drop--
796
				queuedRLCounter.Inc(1)
797 798 799
			}
		}
	}
800
}
801

802 803 804
// demoteUnexecutables removes invalid and processed transactions from the pools
// executable/pending queue and any subsequent transactions that become unexecutable
// are moved back into the future queue.
805
func (pool *TxPool) demoteUnexecutables(state *state.StateDB) {
806 807
	gaslimit := pool.gasLimit()

808 809 810
	// Iterate over all accounts and demote any non-executable transactions
	for addr, list := range pool.pending {
		nonce := state.GetNonce(addr)
811

812 813
		// Drop all transactions that are deemed too old (low nonce)
		for _, tx := range list.Forward(nonce) {
814
			hash := tx.Hash()
815
			log.Trace("Removed old pending transaction", "hash", hash)
816
			delete(pool.all, hash)
817
			pool.priced.Removed()
818
		}
819 820
		// Drop all transactions that are too costly (low balance or out of gas), and queue any invalids back for later
		drops, invalids := list.Filter(state.GetBalance(addr), gaslimit)
821
		for _, tx := range drops {
822
			hash := tx.Hash()
823
			log.Trace("Removed unpayable pending transaction", "hash", hash)
824
			delete(pool.all, hash)
825
			pool.priced.Removed()
826
			pendingNofundsCounter.Inc(1)
827
		}
828
		for _, tx := range invalids {
829
			hash := tx.Hash()
830
			log.Trace("Demoting pending transaction", "hash", hash)
831
			pool.enqueueTx(hash, tx)
832 833 834 835
		}
		// Delete the entire queue entry if it became empty.
		if list.Empty() {
			delete(pool.pending, addr)
836
			delete(pool.beats, addr)
837 838 839
		}
	}
}
840

841 842 843 844 845 846 847 848 849 850 851 852 853 854
// expirationLoop is a loop that periodically iterates over all accounts with
// queued transactions and drop all that have been inactive for a prolonged amount
// of time.
func (pool *TxPool) expirationLoop() {
	defer pool.wg.Done()

	evict := time.NewTicker(evictionInterval)
	defer evict.Stop()

	for {
		select {
		case <-evict.C:
			pool.mu.Lock()
			for addr := range pool.queue {
855
				if time.Since(pool.beats[addr]) > pool.config.Lifetime {
856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880
					for _, tx := range pool.queue[addr].Flatten() {
						pool.removeTx(tx.Hash())
					}
				}
			}
			pool.mu.Unlock()

		case <-pool.quit:
			return
		}
	}
}

// addressByHeartbeat is an account address tagged with its last activity timestamp.
type addressByHeartbeat struct {
	address   common.Address
	heartbeat time.Time
}

type addresssByHeartbeat []addressByHeartbeat

func (a addresssByHeartbeat) Len() int           { return len(a) }
func (a addresssByHeartbeat) Less(i, j int) bool { return a[i].heartbeat.Before(a[j].heartbeat) }
func (a addresssByHeartbeat) Swap(i, j int)      { a[i], a[j] = a[j], a[i] }

881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906
// txSet represents a set of transaction hashes in which entries
//  are automatically dropped after txSetDuration time
type txSet struct {
	txMap          map[common.Hash]struct{}
	txOrd          map[uint64]txOrdType
	addPtr, delPtr uint64
}

const txSetDuration = time.Hour * 2

// txOrdType represents an entry in the time-ordered list of transaction hashes
type txOrdType struct {
	hash common.Hash
	time time.Time
}

// newTxSet creates a new transaction set
func newTxSet() *txSet {
	return &txSet{
		txMap: make(map[common.Hash]struct{}),
		txOrd: make(map[uint64]txOrdType),
	}
}

// contains returns true if the set contains the given transaction hash
// (not thread safe, should be called from a locked environment)
C
changhong 已提交
907 908
func (ts *txSet) contains(hash common.Hash) bool {
	_, ok := ts.txMap[hash]
909 910 911 912 913
	return ok
}

// add adds a transaction hash to the set, then removes entries older than txSetDuration
// (not thread safe, should be called from a locked environment)
C
changhong 已提交
914 915
func (ts *txSet) add(hash common.Hash) {
	ts.txMap[hash] = struct{}{}
916
	now := time.Now()
C
changhong 已提交
917 918
	ts.txOrd[ts.addPtr] = txOrdType{hash: hash, time: now}
	ts.addPtr++
919
	delBefore := now.Add(-txSetDuration)
C
changhong 已提交
920 921 922 923
	for ts.delPtr < ts.addPtr && ts.txOrd[ts.delPtr].time.Before(delBefore) {
		delete(ts.txMap, ts.txOrd[ts.delPtr].hash)
		delete(ts.txOrd, ts.delPtr)
		ts.delPtr++
924 925
	}
}