Fixed tests and bloom

This commit is contained in:
obscuren 2015-03-17 18:00:03 +01:00
parent c21293cd91
commit 86661de077
3 changed files with 35 additions and 19 deletions

View File

@ -20,15 +20,15 @@ func CreateBloom(receipts Receipts) Bloom {
func LogsBloom(logs state.Logs) *big.Int { func LogsBloom(logs state.Logs) *big.Int {
bin := new(big.Int) bin := new(big.Int)
for _, log := range logs { for _, log := range logs {
data := make([]common.Hash, len(log.Topics())+1) data := make([]common.Hash, len(log.Topics()))
data[0] = log.Address().Hash() bin.Or(bin, bloom9(log.Address().Bytes()))
for i, topic := range log.Topics() { for i, topic := range log.Topics() {
data[i+1] = topic data[i] = topic
} }
for _, b := range data { for _, b := range data {
bin.Or(bin, bloom9(crypto.Sha3(b[:]))) bin.Or(bin, bloom9(b[:]))
} }
} }
@ -36,21 +36,24 @@ func LogsBloom(logs state.Logs) *big.Int {
} }
func bloom9(b []byte) *big.Int { func bloom9(b []byte) *big.Int {
b = crypto.Sha3(b[:])
r := new(big.Int) r := new(big.Int)
for i := 0; i < 6; i += 2 { for i := 0; i < 6; i += 2 {
t := big.NewInt(1) t := big.NewInt(1)
//b := uint(b[i+1]) + 512*(uint(b[i])&1) b := (uint(b[i+1]) + (uint(b[i]) << 8)) & 2047
b := (uint(b[i+1]) + (uint(b[i]) << 8)) & 511
r.Or(r, t.Lsh(t, b)) r.Or(r, t.Lsh(t, b))
} }
return r return r
} }
var Bloom9 = bloom9
func BloomLookup(bin Bloom, topic common.Hash) bool { func BloomLookup(bin Bloom, topic common.Hash) bool {
bloom := bin.Big() bloom := bin.Big()
cmp := bloom9(crypto.Sha3(topic[:])) cmp := bloom9(topic[:])
return bloom.And(bloom, cmp).Cmp(cmp) == 0 return bloom.And(bloom, cmp).Cmp(cmp) == 0
} }

View File

@ -1,6 +1,7 @@
package types package types
import ( import (
"fmt"
"math/big" "math/big"
"github.com/ethereum/go-ethereum/common" "github.com/ethereum/go-ethereum/common"
@ -10,7 +11,9 @@ type BlockProcessor interface {
Process(*Block) (*big.Int, error) Process(*Block) (*big.Int, error)
} }
type Bloom [256]byte const bloomLength = 256
type Bloom [bloomLength]byte
func BytesToBloom(b []byte) Bloom { func BytesToBloom(b []byte) Bloom {
var bloom Bloom var bloom Bloom
@ -19,13 +22,13 @@ func BytesToBloom(b []byte) Bloom {
} }
func (b *Bloom) SetBytes(d []byte) { func (b *Bloom) SetBytes(d []byte) {
if len(b) > len(d) { if len(b) < len(d) {
panic("bloom bytes too big") panic(fmt.Sprintf("bloom bytes too big %d %d", len(b), len(d)))
} }
// reverse loop // reverse loop
for i := len(d) - 1; i >= 0; i-- { for i := len(d) - 1; i >= 0; i-- {
b[i] = b[i] b[bloomLength-len(d)+i] = b[i]
} }
} }

View File

@ -2,7 +2,6 @@ package vm
import ( import (
"bytes" "bytes"
"fmt"
"math/big" "math/big"
"strconv" "strconv"
"testing" "testing"
@ -82,9 +81,6 @@ func RunVmTest(p string, t *testing.T) {
helper.CreateFileTests(t, p, &tests) helper.CreateFileTests(t, p, &tests)
for name, test := range tests { for name, test := range tests {
if name != "log2_nonEmptyMem" {
continue
}
db, _ := ethdb.NewMemDatabase() db, _ := ethdb.NewMemDatabase()
statedb := state.New(common.Hash{}, db) statedb := state.New(common.Hash{}, db)
for addr, account := range test.Pre { for addr, account := range test.Pre {
@ -172,12 +168,26 @@ func RunVmTest(p string, t *testing.T) {
if len(test.Logs) != len(logs) { if len(test.Logs) != len(logs) {
t.Errorf("log length mismatch. Expected %d, got %d", len(test.Logs), len(logs)) t.Errorf("log length mismatch. Expected %d, got %d", len(test.Logs), len(logs))
} else { } else {
fmt.Println("A", test.Logs)
fmt.Println("B", logs)
for i, log := range test.Logs { for i, log := range test.Logs {
if common.HexToAddress(log.AddressF) != logs[i].Address() {
t.Errorf("'%s' log address expected %v got %x", name, log.AddressF, logs[i].Address())
}
if !bytes.Equal(logs[i].Data(), helper.FromHex(log.DataF)) {
t.Errorf("'%s' log data expected %v got %x", name, log.DataF, logs[i].Data())
}
if len(log.TopicsF) != len(logs[i].Topics()) {
t.Errorf("'%s' log topics length expected %d got %d", name, len(log.TopicsF), logs[i].Topics())
} else {
for j, topic := range log.TopicsF {
if common.HexToHash(topic) != logs[i].Topics()[j] {
t.Errorf("'%s' log topic[%d] expected %v got %x", name, j, topic, logs[i].Topics()[j])
}
}
}
genBloom := common.LeftPadBytes(types.LogsBloom(state.Logs{logs[i]}).Bytes(), 256) genBloom := common.LeftPadBytes(types.LogsBloom(state.Logs{logs[i]}).Bytes(), 256)
fmt.Println("A BLOOM", log.BloomF)
fmt.Printf("B BLOOM %x\n", genBloom)
if !bytes.Equal(genBloom, common.Hex2Bytes(log.BloomF)) { if !bytes.Equal(genBloom, common.Hex2Bytes(log.BloomF)) {
t.Errorf("'%s' bloom mismatch", name) t.Errorf("'%s' bloom mismatch", name)
} }