2014-12-04 01:28:02 -08:00
|
|
|
package core
|
2014-08-11 07:23:17 -07:00
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
2014-09-26 04:32:54 -07:00
|
|
|
"math"
|
2014-08-11 07:23:17 -07:00
|
|
|
|
2014-12-04 01:28:02 -08:00
|
|
|
"github.com/ethereum/go-ethereum/core/types"
|
2014-10-31 06:43:14 -07:00
|
|
|
"github.com/ethereum/go-ethereum/state"
|
2014-08-11 07:23:17 -07:00
|
|
|
)
|
|
|
|
|
2014-10-18 04:20:06 -07:00
|
|
|
type AccountChange struct {
|
|
|
|
Address, StateAddress []byte
|
2014-08-15 07:19:10 -07:00
|
|
|
}
|
|
|
|
|
2015-01-29 07:52:00 -08:00
|
|
|
type FilterOptions struct {
|
|
|
|
Earliest int64
|
|
|
|
Latest int64
|
|
|
|
|
2015-02-17 07:12:55 -08:00
|
|
|
Address [][]byte
|
2015-01-29 07:52:00 -08:00
|
|
|
Topics [][]byte
|
|
|
|
|
|
|
|
Skip int
|
|
|
|
Max int
|
|
|
|
}
|
|
|
|
|
2014-08-11 07:23:17 -07:00
|
|
|
// Filtering interface
|
|
|
|
type Filter struct {
|
2015-02-17 03:24:51 -08:00
|
|
|
eth Backend
|
2014-09-26 04:32:54 -07:00
|
|
|
earliest int64
|
|
|
|
latest int64
|
2014-08-11 07:23:17 -07:00
|
|
|
skip int
|
2015-02-17 07:12:55 -08:00
|
|
|
address [][]byte
|
2014-08-11 07:23:17 -07:00
|
|
|
max int
|
2015-01-28 01:23:18 -08:00
|
|
|
topics [][]byte
|
2014-08-15 07:19:10 -07:00
|
|
|
|
2015-02-05 11:55:03 -08:00
|
|
|
BlockCallback func(*types.Block)
|
|
|
|
PendingCallback func(*types.Block)
|
|
|
|
LogsCallback func(state.Logs)
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// Create a new filter which uses a bloom filter on blocks to figure out whether a particular block
|
|
|
|
// is interesting or not.
|
2015-02-17 03:24:51 -08:00
|
|
|
func NewFilter(eth Backend) *Filter {
|
2014-08-11 07:23:17 -07:00
|
|
|
return &Filter{eth: eth}
|
|
|
|
}
|
|
|
|
|
2015-01-29 07:52:00 -08:00
|
|
|
func (self *Filter) SetOptions(options FilterOptions) {
|
|
|
|
self.earliest = options.Earliest
|
|
|
|
self.latest = options.Latest
|
|
|
|
self.skip = options.Skip
|
|
|
|
self.max = options.Max
|
|
|
|
self.address = options.Address
|
|
|
|
self.topics = options.Topics
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2014-08-11 07:23:17 -07:00
|
|
|
// Set the earliest and latest block for filtering.
|
|
|
|
// -1 = latest block (i.e., the current block)
|
|
|
|
// hash = particular hash from-to
|
2014-09-26 04:32:54 -07:00
|
|
|
func (self *Filter) SetEarliestBlock(earliest int64) {
|
|
|
|
self.earliest = earliest
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2014-09-26 04:32:54 -07:00
|
|
|
func (self *Filter) SetLatestBlock(latest int64) {
|
|
|
|
self.latest = latest
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2015-02-17 07:12:55 -08:00
|
|
|
func (self *Filter) SetAddress(addr [][]byte) {
|
2015-01-28 01:23:18 -08:00
|
|
|
self.address = addr
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
func (self *Filter) SetTopics(topics [][]byte) {
|
|
|
|
self.topics = topics
|
2014-08-14 08:02:39 -07:00
|
|
|
}
|
|
|
|
|
2014-08-11 07:23:17 -07:00
|
|
|
func (self *Filter) SetMax(max int) {
|
|
|
|
self.max = max
|
|
|
|
}
|
|
|
|
|
|
|
|
func (self *Filter) SetSkip(skip int) {
|
|
|
|
self.skip = skip
|
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
// Run filters logs with the current parameters set
|
|
|
|
func (self *Filter) Find() state.Logs {
|
2014-12-23 04:48:44 -08:00
|
|
|
earliestBlock := self.eth.ChainManager().CurrentBlock()
|
2014-09-26 04:32:54 -07:00
|
|
|
var earliestBlockNo uint64 = uint64(self.earliest)
|
|
|
|
if self.earliest == -1 {
|
2014-12-23 04:48:44 -08:00
|
|
|
earliestBlockNo = earliestBlock.NumberU64()
|
2014-09-26 04:32:54 -07:00
|
|
|
}
|
|
|
|
var latestBlockNo uint64 = uint64(self.latest)
|
|
|
|
if self.latest == -1 {
|
2014-12-23 04:48:44 -08:00
|
|
|
latestBlockNo = earliestBlock.NumberU64()
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2014-09-26 04:32:54 -07:00
|
|
|
var (
|
2015-01-28 01:23:18 -08:00
|
|
|
logs state.Logs
|
|
|
|
block = self.eth.ChainManager().GetBlockByNumber(latestBlockNo)
|
|
|
|
quit bool
|
2014-09-26 04:32:54 -07:00
|
|
|
)
|
|
|
|
for i := 0; !quit && block != nil; i++ {
|
|
|
|
// Quit on latest
|
|
|
|
switch {
|
2014-12-23 04:48:44 -08:00
|
|
|
case block.NumberU64() == earliestBlockNo, block.NumberU64() == 0:
|
2014-08-11 07:23:17 -07:00
|
|
|
quit = true
|
2015-01-28 01:23:18 -08:00
|
|
|
case self.max <= len(logs):
|
2014-09-26 04:32:54 -07:00
|
|
|
break
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// Use bloom filtering to see if this block is interesting given the
|
|
|
|
// current parameters
|
|
|
|
if self.bloomFilter(block) {
|
2015-01-28 01:23:18 -08:00
|
|
|
// Get the logs of the block
|
2015-02-22 04:12:01 -08:00
|
|
|
unfiltered, err := self.eth.BlockProcessor().GetLogs(block)
|
2014-08-11 07:23:17 -07:00
|
|
|
if err != nil {
|
2015-01-28 01:23:18 -08:00
|
|
|
chainlogger.Warnln("err: filter get logs ", err)
|
2014-08-11 07:23:17 -07:00
|
|
|
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
2015-02-22 04:12:01 -08:00
|
|
|
logs = append(logs, self.FilterLogs(unfiltered)...)
|
2014-08-14 15:24:37 -07:00
|
|
|
}
|
2014-08-14 08:02:39 -07:00
|
|
|
|
2014-12-23 04:48:44 -08:00
|
|
|
block = self.eth.ChainManager().GetBlock(block.ParentHash())
|
2014-08-14 15:24:37 -07:00
|
|
|
}
|
2014-08-11 07:23:17 -07:00
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
skip := int(math.Min(float64(len(logs)), float64(self.skip)))
|
2014-09-26 04:32:54 -07:00
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
return logs[skip:]
|
2014-08-14 15:24:37 -07:00
|
|
|
}
|
2014-08-11 07:23:17 -07:00
|
|
|
|
2015-02-04 17:28:54 -08:00
|
|
|
func includes(addresses [][]byte, a []byte) bool {
|
2014-08-15 07:19:10 -07:00
|
|
|
for _, addr := range addresses {
|
2015-02-04 17:28:54 -08:00
|
|
|
if !bytes.Equal(addr, a) {
|
|
|
|
return false
|
2014-08-15 07:19:10 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-02-04 17:28:54 -08:00
|
|
|
return true
|
2014-08-15 07:19:10 -07:00
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
func (self *Filter) FilterLogs(logs state.Logs) state.Logs {
|
|
|
|
var ret state.Logs
|
2014-08-11 07:23:17 -07:00
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
// Filter the logs for interesting stuff
|
2015-02-04 17:28:54 -08:00
|
|
|
Logs:
|
2015-01-28 01:23:18 -08:00
|
|
|
for _, log := range logs {
|
2015-02-17 07:12:55 -08:00
|
|
|
if !includes(self.address, log.Address()) {
|
2014-08-14 15:24:37 -07:00
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
2015-02-04 17:28:54 -08:00
|
|
|
max := int(math.Min(float64(len(self.topics)), float64(len(log.Topics()))))
|
|
|
|
for i := 0; i < max; i++ {
|
|
|
|
if !bytes.Equal(log.Topics()[i], self.topics[i]) {
|
|
|
|
continue Logs
|
2014-08-15 07:19:10 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
ret = append(ret, log)
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
return ret
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2014-11-18 07:58:22 -08:00
|
|
|
func (self *Filter) bloomFilter(block *types.Block) bool {
|
2015-02-17 07:12:55 -08:00
|
|
|
if len(self.address) > 0 {
|
|
|
|
var included bool
|
|
|
|
for _, addr := range self.address {
|
|
|
|
if types.BloomLookup(block.Bloom(), addr) {
|
|
|
|
included = true
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if !included {
|
|
|
|
return false
|
|
|
|
}
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
for _, topic := range self.topics {
|
|
|
|
if !types.BloomLookup(block.Bloom(), topic) {
|
|
|
|
return false
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-01-28 01:23:18 -08:00
|
|
|
return true
|
2014-08-11 07:23:17 -07:00
|
|
|
}
|