aboutsummaryrefslogtreecommitdiffstats
path: root/ethchain/filter.go
blob: 4f3160b901726a6a138ff482f8cc592befd0f1ca (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
package ethchain

import (
    "bytes"
    "fmt"

    "github.com/ethereum/eth-go/ethstate"
    "github.com/ethereum/eth-go/ethutil"
)

// Filtering interface
type Filter struct {
    eth      EthManager
    earliest []byte
    latest   []byte
    skip     int
    from, to [][]byte
    max      int
}

// Create a new filter which uses a bloom filter on blocks to figure out whether a particular block
// is interesting or not.
func NewFilter(eth EthManager) *Filter {
    return &Filter{eth: eth}
}

func NewFilterFromMap(object map[string]interface{}, eth EthManager) *Filter {
    filter := NewFilter(eth)

    if object["earliest"] != nil {
        earliest := object["earliest"]
        if e, ok := earliest.(string); ok {
            filter.SetEarliestBlock(ethutil.Hex2Bytes(e))
        } else {
            filter.SetEarliestBlock(earliest)
        }
    }

    if object["latest"] != nil {
        latest := object["latest"]
        if l, ok := latest.(string); ok {
            filter.SetLatestBlock(ethutil.Hex2Bytes(l))
        } else {
            filter.SetLatestBlock(latest)
        }
    }

    if object["to"] != nil {
        filter.AddTo(ethutil.Hex2Bytes(object["to"].(string)))
    }

    if object["from"] != nil {
        filter.AddFrom(ethutil.Hex2Bytes(object["from"].(string)))
    }

    if object["max"] != nil {
        filter.SetMax(object["max"].(int))
    }

    if object["skip"] != nil {
        filter.SetSkip(object["skip"].(int))
    }

    return filter
}

// Set the earliest and latest block for filtering.
// -1 = latest block (i.e., the current block)
// hash = particular hash from-to
func (self *Filter) SetEarliestBlock(earliest interface{}) {
    e := ethutil.NewValue(earliest)

    // Check for -1 (latest) otherwise assume bytes
    if e.Int() == -1 {
        self.earliest = self.eth.BlockChain().CurrentBlock.Hash()
    } else if e.Len() > 0 {
        self.earliest = e.Bytes()
    } else {
        panic(fmt.Sprintf("earliest has to be either -1 or a valid hash: %v (%T)", e, e.Val))
    }
}

func (self *Filter) SetLatestBlock(latest interface{}) {
    l := ethutil.NewValue(latest)

    // Check for -1 (latest) otherwise assume bytes
    if l.Int() == -1 {
        self.latest = self.eth.BlockChain().CurrentBlock.Hash()
    } else if l.Len() > 0 {
        self.latest = l.Bytes()
    } else {
        panic(fmt.Sprintf("latest has to be either -1 or a valid hash: %v", l))
    }
}

func (self *Filter) SetFrom(addr [][]byte) {
    self.from = addr
}

func (self *Filter) AddFrom(addr []byte) {
    self.from = append(self.from, addr)
}

func (self *Filter) SetTo(addr [][]byte) {
    self.to = addr
}

func (self *Filter) AddTo(addr []byte) {
    self.from = append(self.to, addr)
}

func (self *Filter) SetMax(max int) {
    self.max = max
}

func (self *Filter) SetSkip(skip int) {
    self.skip = skip
}

// Run filters messages with the current parameters set
func (self *Filter) Find() []*ethstate.Message {
    var messages []*ethstate.Message

    block := self.eth.BlockChain().GetBlock(self.latest)

    // skip N blocks (useful for pagination)
    if self.skip > 0 {
        for i := 0; i < i; i++ {
            block = self.eth.BlockChain().GetBlock(block.PrevHash)
        }
    }

    // Start block filtering
    quit := false
    for i := 1; !quit && block != nil; i++ {
        // Mark last check
        if self.max == i || (len(self.earliest) > 0 && bytes.Compare(block.Hash(), self.earliest) == 0) {
            quit = true
        }

        // Use bloom filtering to see if this block is interesting given the
        // current parameters
        if self.bloomFilter(block) {
            // Get the messages of the block
            msgs, err := self.eth.StateManager().GetMessages(block)
            if err != nil {
                chainlogger.Warnln("err: filter get messages ", err)

                break
            }

            messages = append(messages, self.FilterMessages(msgs)...)
        }

        block = self.eth.BlockChain().GetBlock(block.PrevHash)
    }

    return messages
}

func (self *Filter) FilterMessages(msgs []*ethstate.Message) []*ethstate.Message {
    var messages []*ethstate.Message
    includes := func(addresses [][]byte, a []byte) (found bool) {
        for _, addr := range addresses {
            if bytes.Compare(addr, a) == 0 {
                return true
            }
        }

        return
    }
    // Filter the messages for interesting stuff
    for _, message := range msgs {
        if len(self.to) > 0 && !includes(self.to, message.To) {
            continue
        }

        if len(self.from) > 0 && !includes(self.from, message.From) {
            continue
        }

        messages = append(messages, message)
    }

    return messages
}

func (self *Filter) bloomFilter(block *Block) bool {
    fk := append([]byte("bloom"), block.Hash()...)
    bin, err := self.eth.Db().Get(fk)
    if err != nil {
        panic(err)
    }

    bloom := NewBloomFilter(bin)

    var fromIncluded, toIncluded bool
    if len(self.from) > 0 {
        for _, from := range self.from {
            if bloom.Search(from) {
                fromIncluded = true
                break
            }
        }
    } else {
        fromIncluded = true
    }

    if len(self.to) > 0 {
        for _, to := range self.to {
            if bloom.Search(to) {
                toIncluded = true
                break
            }
        }
    } else {
        toIncluded = true
    }

    return fromIncluded && toIncluded
}