fix broken go-ipld-eth trie node dag putters
This commit is contained in:
parent
ca273a026d
commit
5173edf563
@ -5,7 +5,7 @@ CREATE TABLE eth.header_cids (
|
|||||||
block_hash VARCHAR(66) NOT NULL,
|
block_hash VARCHAR(66) NOT NULL,
|
||||||
parent_hash VARCHAR(66) NOT NULL,
|
parent_hash VARCHAR(66) NOT NULL,
|
||||||
cid TEXT NOT NULL,
|
cid TEXT NOT NULL,
|
||||||
td BIGINT,
|
td NUMERIC NOT NULL,
|
||||||
node_id INTEGER NOT NULL REFERENCES nodes (id) ON DELETE CASCADE,
|
node_id INTEGER NOT NULL REFERENCES nodes (id) ON DELETE CASCADE,
|
||||||
UNIQUE (block_number, block_hash)
|
UNIQUE (block_number, block_hash)
|
||||||
);
|
);
|
||||||
|
@ -183,7 +183,7 @@ CREATE TABLE eth.header_cids (
|
|||||||
block_hash character varying(66) NOT NULL,
|
block_hash character varying(66) NOT NULL,
|
||||||
parent_hash character varying(66) NOT NULL,
|
parent_hash character varying(66) NOT NULL,
|
||||||
cid text NOT NULL,
|
cid text NOT NULL,
|
||||||
td bigint,
|
td numeric NOT NULL,
|
||||||
node_id integer NOT NULL
|
node_id integer NOT NULL
|
||||||
);
|
);
|
||||||
|
|
||||||
|
26
environments/btcSuperNode.toml
Normal file
26
environments/btcSuperNode.toml
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
[superNode]
|
||||||
|
chain = "bitcoin"
|
||||||
|
ipfsPath = "/root/.ipfs"
|
||||||
|
|
||||||
|
[superNode.database]
|
||||||
|
name = "vulcanize_public"
|
||||||
|
hostname = "localhost"
|
||||||
|
port = 5432
|
||||||
|
user = "ec2-user"
|
||||||
|
|
||||||
|
[superNode.sync]
|
||||||
|
on = true
|
||||||
|
wsPath = "http://127.0.0.1:8332"
|
||||||
|
workers = 1
|
||||||
|
|
||||||
|
[superNode.server]
|
||||||
|
on = true
|
||||||
|
ipcPath = "/root/.vulcanize/btc/vulcanize.ipc"
|
||||||
|
wsPath = "127.0.0.1:8082"
|
||||||
|
httpPath = "127.0.0.1:8083"
|
||||||
|
|
||||||
|
[superNode.backFill]
|
||||||
|
on = true
|
||||||
|
httpPath = "http://127.0.0.1:8332"
|
||||||
|
frequency = 5
|
||||||
|
batchSize = 50
|
26
environments/ethSuperNode.toml
Normal file
26
environments/ethSuperNode.toml
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
[superNode]
|
||||||
|
chain = "ethereum"
|
||||||
|
ipfsPath = "/root/.ipfs"
|
||||||
|
|
||||||
|
[superNode.database]
|
||||||
|
name = "vulcanize_public"
|
||||||
|
hostname = "localhost"
|
||||||
|
port = 5432
|
||||||
|
user = "ec2-user"
|
||||||
|
|
||||||
|
[superNode.sync]
|
||||||
|
on = true
|
||||||
|
wsPath = "ws://127.0.0.1:8546"
|
||||||
|
workers = 1
|
||||||
|
|
||||||
|
[superNode.server]
|
||||||
|
on = true
|
||||||
|
ipcPath = "/root/.vulcanize/eth/vulcanize.ipc"
|
||||||
|
wsPath = "127.0.0.1:8080"
|
||||||
|
httpPath = "127.0.0.1:8081"
|
||||||
|
|
||||||
|
[superNode.backFill]
|
||||||
|
on = true
|
||||||
|
httpPath = "http://127.0.0.1:8545"
|
||||||
|
frequency = 5
|
||||||
|
batchSize = 50
|
@ -59,25 +59,6 @@ func NewBtcHeader(header *wire.BlockHeader) (*BtcHeader, error) {
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
// DecodeBtcHeader takes a cid and its raw binary data
|
|
||||||
// from IPFS and returns an BtcHeader object for further processing.
|
|
||||||
func DecodeBtcHeader(c cid.Cid, b []byte) (*BtcHeader, error) {
|
|
||||||
var h *wire.BlockHeader
|
|
||||||
w := bytes.NewBuffer(b)
|
|
||||||
if err := h.Deserialize(w); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &BtcHeader{
|
|
||||||
BlockHeader: h,
|
|
||||||
cid: c,
|
|
||||||
rawdata: b,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
Block INTERFACE
|
Block INTERFACE
|
||||||
*/
|
*/
|
||||||
|
@ -44,25 +44,6 @@ func NewBtcTx(tx *wire.MsgTx) (*BtcTx, error) {
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
// DecodeBtcTx takes a cid and its raw binary data
|
|
||||||
// from IPFS and returns an BtcTx object for further processing.
|
|
||||||
func DecodeBtcTx(c cid.Cid, b []byte) (*BtcTx, error) {
|
|
||||||
var tx *wire.MsgTx
|
|
||||||
w := bytes.NewBuffer(b)
|
|
||||||
if err := tx.Deserialize(w); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &BtcTx{
|
|
||||||
MsgTx: tx,
|
|
||||||
cid: c,
|
|
||||||
rawdata: b,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
Block INTERFACE
|
Block INTERFACE
|
||||||
*/
|
*/
|
||||||
|
@ -60,24 +60,6 @@ func NewEthHeader(header *types.Header) (*EthHeader, error) {
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
// DecodeEthHeader takes a cid and its raw binary data
|
|
||||||
// from IPFS and returns an EthHeader object for further processing.
|
|
||||||
func DecodeEthHeader(c cid.Cid, b []byte) (*EthHeader, error) {
|
|
||||||
var h *types.Header
|
|
||||||
if err := rlp.DecodeBytes(b, h); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &EthHeader{
|
|
||||||
Header: h,
|
|
||||||
cid: c,
|
|
||||||
rawdata: b,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
Block INTERFACE
|
Block INTERFACE
|
||||||
*/
|
*/
|
||||||
|
@ -59,24 +59,6 @@ func NewReceipt(receipt *types.ReceiptForStorage) (*EthReceipt, error) {
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
// DecodeEthReceipt takes a cid and its raw binary data
|
|
||||||
// from IPFS and returns an EthReceipt object for further processing.
|
|
||||||
func DecodeEthReceipt(c cid.Cid, b []byte) (*EthReceipt, error) {
|
|
||||||
var r *types.ReceiptForStorage
|
|
||||||
if err := rlp.DecodeBytes(b, r); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &EthReceipt{
|
|
||||||
ReceiptForStorage: r,
|
|
||||||
cid: c,
|
|
||||||
rawdata: b,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
Block INTERFACE
|
Block INTERFACE
|
||||||
*/
|
*/
|
||||||
|
@ -19,7 +19,6 @@ package ipld
|
|||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
|
||||||
"github.com/ethereum/go-ethereum/rlp"
|
|
||||||
"github.com/ipfs/go-cid"
|
"github.com/ipfs/go-cid"
|
||||||
node "github.com/ipfs/go-ipld-format"
|
node "github.com/ipfs/go-ipld-format"
|
||||||
mh "github.com/multiformats/go-multihash"
|
mh "github.com/multiformats/go-multihash"
|
||||||
@ -28,7 +27,8 @@ import (
|
|||||||
// EthStateTrie (eth-state-trie, codec 0x96), represents
|
// EthStateTrie (eth-state-trie, codec 0x96), represents
|
||||||
// a node from the state trie in ethereum.
|
// a node from the state trie in ethereum.
|
||||||
type EthStateTrie struct {
|
type EthStateTrie struct {
|
||||||
*TrieNode
|
cid cid.Cid
|
||||||
|
rawdata []byte
|
||||||
}
|
}
|
||||||
|
|
||||||
// Static (compile time) check that EthStateTrie satisfies the node.Node interface.
|
// Static (compile time) check that EthStateTrie satisfies the node.Node interface.
|
||||||
@ -45,40 +45,9 @@ func FromStateTrieRLP(stateNodeRLP []byte) (*EthStateTrie, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return DecodeEthStateTrie(c, stateNodeRLP)
|
return &EthStateTrie{
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
// DecodeEthStateTrie returns an EthStateTrie object from its cid and rawdata.
|
|
||||||
func DecodeEthStateTrie(c cid.Cid, b []byte) (*EthStateTrie, error) {
|
|
||||||
tn, err := decodeTrieNode(c, b, decodeEthStateTrieLeaf)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &EthStateTrie{TrieNode: tn}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// decodeEthStateTrieLeaf parses a eth-tx-trie leaf
|
|
||||||
// from decoded RLP elements
|
|
||||||
func decodeEthStateTrieLeaf(i []interface{}) ([]interface{}, error) {
|
|
||||||
var account EthAccount
|
|
||||||
if err := rlp.DecodeBytes(i[1].([]byte), &account); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
c, err := rawdataToCid(MEthAccountSnapshot, i[1].([]byte), mh.KECCAK_256)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return []interface{}{
|
|
||||||
i[0].([]byte),
|
|
||||||
&EthAccountSnapshot{
|
|
||||||
EthAccount: &account,
|
|
||||||
cid: c,
|
cid: c,
|
||||||
rawdata: i[1].([]byte),
|
rawdata: stateNodeRLP,
|
||||||
},
|
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -101,6 +70,35 @@ func (st *EthStateTrie) String() string {
|
|||||||
return fmt.Sprintf("<EthereumStateTrie %s>", st.cid)
|
return fmt.Sprintf("<EthereumStateTrie %s>", st.cid)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Copy will go away. It is here to comply with the Node interface.
|
||||||
|
func (*EthStateTrie) Copy() node.Node {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStateTrie) Links() []*node.Link {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStateTrie) Resolve(path []string) (interface{}, []string, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStateTrie) ResolveLink(path []string) (*node.Link, []string, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStateTrie) Size() (uint64, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStateTrie) Stat() (*node.NodeStat, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStateTrie) Tree(path string, depth int) []string {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
// Loggable returns in a map the type of IPLD Link.
|
// Loggable returns in a map the type of IPLD Link.
|
||||||
func (st *EthStateTrie) Loggable() map[string]interface{} {
|
func (st *EthStateTrie) Loggable() map[string]interface{} {
|
||||||
return map[string]interface{}{
|
return map[string]interface{}{
|
||||||
|
@ -27,7 +27,8 @@ import (
|
|||||||
// EthStorageTrie (eth-storage-trie, codec 0x98), represents
|
// EthStorageTrie (eth-storage-trie, codec 0x98), represents
|
||||||
// a node from the storage trie in ethereum.
|
// a node from the storage trie in ethereum.
|
||||||
type EthStorageTrie struct {
|
type EthStorageTrie struct {
|
||||||
*TrieNode
|
cid cid.Cid
|
||||||
|
rawdata []byte
|
||||||
}
|
}
|
||||||
|
|
||||||
// Static (compile time) check that EthStorageTrie satisfies the node.Node interface.
|
// Static (compile time) check that EthStorageTrie satisfies the node.Node interface.
|
||||||
@ -44,28 +45,9 @@ func FromStorageTrieRLP(storageNodeRLP []byte) (*EthStorageTrie, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
return DecodeEthStorageTrie(c, storageNodeRLP)
|
return &EthStorageTrie{
|
||||||
}
|
cid: c,
|
||||||
|
rawdata: storageNodeRLP,
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
// DecodeEthStorageTrie returns an EthStorageTrie object from its cid and rawdata.
|
|
||||||
func DecodeEthStorageTrie(c cid.Cid, b []byte) (*EthStorageTrie, error) {
|
|
||||||
tn, err := decodeTrieNode(c, b, decodeEthStorageTrieLeaf)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return &EthStorageTrie{TrieNode: tn}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// decodeEthStorageTrieLeaf parses a eth-tx-trie leaf
|
|
||||||
// from decoded RLP elements
|
|
||||||
func decodeEthStorageTrieLeaf(i []interface{}) ([]interface{}, error) {
|
|
||||||
return []interface{}{
|
|
||||||
i[0].([]byte),
|
|
||||||
i[1].([]byte),
|
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -88,6 +70,35 @@ func (st *EthStorageTrie) String() string {
|
|||||||
return fmt.Sprintf("<EthereumStorageTrie %s>", st.cid)
|
return fmt.Sprintf("<EthereumStorageTrie %s>", st.cid)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Copy will go away. It is here to comply with the Node interface.
|
||||||
|
func (*EthStorageTrie) Copy() node.Node {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStorageTrie) Links() []*node.Link {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStorageTrie) Resolve(path []string) (interface{}, []string, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStorageTrie) ResolveLink(path []string) (*node.Link, []string, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStorageTrie) Size() (uint64, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStorageTrie) Stat() (*node.NodeStat, error) {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*EthStorageTrie) Tree(path string, depth int) []string {
|
||||||
|
panic("implement me")
|
||||||
|
}
|
||||||
|
|
||||||
// Loggable returns in a map the type of IPLD Link.
|
// Loggable returns in a map the type of IPLD Link.
|
||||||
func (st *EthStorageTrie) Loggable() map[string]interface{} {
|
func (st *EthStorageTrie) Loggable() map[string]interface{} {
|
||||||
return map[string]interface{}{
|
return map[string]interface{}{
|
||||||
|
@ -17,10 +17,7 @@
|
|||||||
package ipld
|
package ipld
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
|
|
||||||
"github.com/ethereum/go-ethereum/common"
|
"github.com/ethereum/go-ethereum/common"
|
||||||
"github.com/ethereum/go-ethereum/rlp"
|
|
||||||
"github.com/ipfs/go-cid"
|
"github.com/ipfs/go-cid"
|
||||||
mh "github.com/multiformats/go-multihash"
|
mh "github.com/multiformats/go-multihash"
|
||||||
)
|
)
|
||||||
@ -90,13 +87,3 @@ func sha256ToCid(codec uint64, h []byte) cid.Cid {
|
|||||||
|
|
||||||
return cid.NewCidV1(codec, hash)
|
return cid.NewCidV1(codec, hash)
|
||||||
}
|
}
|
||||||
|
|
||||||
// getRLP encodes the given object to RLP returning its bytes.
|
|
||||||
func getRLP(object interface{}) []byte {
|
|
||||||
buf := new(bytes.Buffer)
|
|
||||||
if err := rlp.Encode(buf, object); err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return buf.Bytes()
|
|
||||||
}
|
|
||||||
|
@ -1,440 +0,0 @@
|
|||||||
// VulcanizeDB
|
|
||||||
// Copyright © 2019 Vulcanize
|
|
||||||
|
|
||||||
// This program is free software: you can redistribute it and/or modify
|
|
||||||
// it under the terms of the GNU Affero General Public License as published by
|
|
||||||
// the Free Software Foundation, either version 3 of the License, or
|
|
||||||
// (at your option) any later version.
|
|
||||||
|
|
||||||
// This program is distributed in the hope that it will be useful,
|
|
||||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
// GNU Affero General Public License for more details.
|
|
||||||
|
|
||||||
// You should have received a copy of the GNU Affero General Public License
|
|
||||||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
package ipld
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
|
|
||||||
"github.com/ethereum/go-ethereum/rlp"
|
|
||||||
"github.com/ipfs/go-cid"
|
|
||||||
node "github.com/ipfs/go-ipld-format"
|
|
||||||
)
|
|
||||||
|
|
||||||
// TrieNode is the general abstraction for
|
|
||||||
//ethereum IPLD trie nodes.
|
|
||||||
type TrieNode struct {
|
|
||||||
// leaf, extension or branch
|
|
||||||
nodeKind string
|
|
||||||
|
|
||||||
// If leaf or extension: [0] is key, [1] is val.
|
|
||||||
// If branch: [0] - [16] are children.
|
|
||||||
elements []interface{}
|
|
||||||
|
|
||||||
// IPLD block information
|
|
||||||
cid cid.Cid
|
|
||||||
rawdata []byte
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
OUTPUT
|
|
||||||
*/
|
|
||||||
|
|
||||||
type trieNodeLeafDecoder func([]interface{}) ([]interface{}, error)
|
|
||||||
|
|
||||||
// decodeTrieNode returns a TrieNode object from an IPLD block's
|
|
||||||
// cid and rawdata.
|
|
||||||
func decodeTrieNode(c cid.Cid, b []byte,
|
|
||||||
leafDecoder trieNodeLeafDecoder) (*TrieNode, error) {
|
|
||||||
var (
|
|
||||||
i, decoded, elements []interface{}
|
|
||||||
nodeKind string
|
|
||||||
err error
|
|
||||||
)
|
|
||||||
|
|
||||||
err = rlp.DecodeBytes(b, &i)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
codec := c.Type()
|
|
||||||
switch len(i) {
|
|
||||||
case 2:
|
|
||||||
nodeKind, decoded, err = decodeCompactKey(i)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if nodeKind == "extension" {
|
|
||||||
elements, err = parseTrieNodeExtension(decoded, codec)
|
|
||||||
}
|
|
||||||
if nodeKind == "leaf" {
|
|
||||||
elements, err = leafDecoder(decoded)
|
|
||||||
}
|
|
||||||
if nodeKind != "extension" && nodeKind != "leaf" {
|
|
||||||
return nil, fmt.Errorf("unexpected nodeKind returned from decoder")
|
|
||||||
}
|
|
||||||
case 17:
|
|
||||||
nodeKind = "branch"
|
|
||||||
elements, err = parseTrieNodeBranch(i, codec)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
return nil, fmt.Errorf("unknown trie node type")
|
|
||||||
}
|
|
||||||
|
|
||||||
return &TrieNode{
|
|
||||||
nodeKind: nodeKind,
|
|
||||||
elements: elements,
|
|
||||||
rawdata: b,
|
|
||||||
cid: c,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// decodeCompactKey takes a compact key, and returns its nodeKind and value.
|
|
||||||
func decodeCompactKey(i []interface{}) (string, []interface{}, error) {
|
|
||||||
first := i[0].([]byte)
|
|
||||||
last := i[1].([]byte)
|
|
||||||
|
|
||||||
switch first[0] / 16 {
|
|
||||||
case '\x00':
|
|
||||||
return "extension", []interface{}{
|
|
||||||
nibbleToByte(first)[2:],
|
|
||||||
last,
|
|
||||||
}, nil
|
|
||||||
case '\x01':
|
|
||||||
return "extension", []interface{}{
|
|
||||||
nibbleToByte(first)[1:],
|
|
||||||
last,
|
|
||||||
}, nil
|
|
||||||
case '\x02':
|
|
||||||
return "leaf", []interface{}{
|
|
||||||
nibbleToByte(first)[2:],
|
|
||||||
last,
|
|
||||||
}, nil
|
|
||||||
case '\x03':
|
|
||||||
return "leaf", []interface{}{
|
|
||||||
nibbleToByte(first)[1:],
|
|
||||||
last,
|
|
||||||
}, nil
|
|
||||||
default:
|
|
||||||
return "", nil, fmt.Errorf("unknown hex prefix")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseTrieNodeExtension helper improves readability
|
|
||||||
func parseTrieNodeExtension(i []interface{}, codec uint64) ([]interface{}, error) {
|
|
||||||
return []interface{}{
|
|
||||||
i[0].([]byte),
|
|
||||||
keccak256ToCid(codec, i[1].([]byte)),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseTrieNodeBranch helper improves readability
|
|
||||||
func parseTrieNodeBranch(i []interface{}, codec uint64) ([]interface{}, error) {
|
|
||||||
var out []interface{}
|
|
||||||
|
|
||||||
for _, vi := range i {
|
|
||||||
v := vi.([]byte)
|
|
||||||
|
|
||||||
switch len(v) {
|
|
||||||
case 0:
|
|
||||||
out = append(out, nil)
|
|
||||||
case 32:
|
|
||||||
out = append(out, keccak256ToCid(codec, v))
|
|
||||||
default:
|
|
||||||
return nil, fmt.Errorf("unrecognized object: %v", v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return out, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
Node INTERFACE
|
|
||||||
*/
|
|
||||||
|
|
||||||
// Resolve resolves a path through this node, stopping at any link boundary
|
|
||||||
// and returning the object found as well as the remaining path to traverse
|
|
||||||
func (t *TrieNode) Resolve(p []string) (interface{}, []string, error) {
|
|
||||||
switch t.nodeKind {
|
|
||||||
case "extension":
|
|
||||||
return t.resolveTrieNodeExtension(p)
|
|
||||||
case "leaf":
|
|
||||||
return t.resolveTrieNodeLeaf(p)
|
|
||||||
case "branch":
|
|
||||||
return t.resolveTrieNodeBranch(p)
|
|
||||||
default:
|
|
||||||
return nil, nil, fmt.Errorf("nodeKind case not implemented")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Tree lists all paths within the object under 'path', and up to the given depth.
|
|
||||||
// To list the entire object (similar to `find .`) pass "" and -1
|
|
||||||
func (t *TrieNode) Tree(p string, depth int) []string {
|
|
||||||
if p != "" || depth == 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var out []string
|
|
||||||
|
|
||||||
switch t.nodeKind {
|
|
||||||
case "extension":
|
|
||||||
var val string
|
|
||||||
for _, e := range t.elements[0].([]byte) {
|
|
||||||
val += fmt.Sprintf("%x", e)
|
|
||||||
}
|
|
||||||
return []string{val}
|
|
||||||
case "branch":
|
|
||||||
for i, elem := range t.elements {
|
|
||||||
if _, ok := elem.(*cid.Cid); ok {
|
|
||||||
out = append(out, fmt.Sprintf("%x", i))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
|
|
||||||
default:
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// ResolveLink is a helper function that calls resolve and asserts the
|
|
||||||
// output is a link
|
|
||||||
func (t *TrieNode) ResolveLink(p []string) (*node.Link, []string, error) {
|
|
||||||
obj, rest, err := t.Resolve(p)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
lnk, ok := obj.(*node.Link)
|
|
||||||
if !ok {
|
|
||||||
return nil, nil, fmt.Errorf("was not a link")
|
|
||||||
}
|
|
||||||
|
|
||||||
return lnk, rest, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Copy will go away. It is here to comply with the interface.
|
|
||||||
func (t *TrieNode) Copy() node.Node {
|
|
||||||
panic("dont use this yet")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Links is a helper function that returns all links within this object
|
|
||||||
func (t *TrieNode) Links() []*node.Link {
|
|
||||||
var out []*node.Link
|
|
||||||
|
|
||||||
for _, i := range t.elements {
|
|
||||||
c, ok := i.(cid.Cid)
|
|
||||||
if ok {
|
|
||||||
out = append(out, &node.Link{Cid: c})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// Stat will go away. It is here to comply with the interface.
|
|
||||||
func (t *TrieNode) Stat() (*node.NodeStat, error) {
|
|
||||||
return &node.NodeStat{}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Size will go away. It is here to comply with the interface.
|
|
||||||
func (t *TrieNode) Size() (uint64, error) {
|
|
||||||
return 0, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
TrieNode functions
|
|
||||||
*/
|
|
||||||
|
|
||||||
// MarshalJSON processes the transaction trie into readable JSON format.
|
|
||||||
func (t *TrieNode) MarshalJSON() ([]byte, error) {
|
|
||||||
var out map[string]interface{}
|
|
||||||
|
|
||||||
switch t.nodeKind {
|
|
||||||
case "extension":
|
|
||||||
fallthrough
|
|
||||||
case "leaf":
|
|
||||||
var hexPrefix string
|
|
||||||
for _, e := range t.elements[0].([]byte) {
|
|
||||||
hexPrefix += fmt.Sprintf("%x", e)
|
|
||||||
}
|
|
||||||
|
|
||||||
// if we got a byte we need to do this casting otherwise
|
|
||||||
// it will be marshaled to a base64 encoded value
|
|
||||||
if _, ok := t.elements[1].([]byte); ok {
|
|
||||||
var hexVal string
|
|
||||||
for _, e := range t.elements[1].([]byte) {
|
|
||||||
hexVal += fmt.Sprintf("%x", e)
|
|
||||||
}
|
|
||||||
|
|
||||||
t.elements[1] = hexVal
|
|
||||||
}
|
|
||||||
|
|
||||||
out = map[string]interface{}{
|
|
||||||
"type": t.nodeKind,
|
|
||||||
hexPrefix: t.elements[1],
|
|
||||||
}
|
|
||||||
|
|
||||||
case "branch":
|
|
||||||
out = map[string]interface{}{
|
|
||||||
"type": "branch",
|
|
||||||
"0": t.elements[0],
|
|
||||||
"1": t.elements[1],
|
|
||||||
"2": t.elements[2],
|
|
||||||
"3": t.elements[3],
|
|
||||||
"4": t.elements[4],
|
|
||||||
"5": t.elements[5],
|
|
||||||
"6": t.elements[6],
|
|
||||||
"7": t.elements[7],
|
|
||||||
"8": t.elements[8],
|
|
||||||
"9": t.elements[9],
|
|
||||||
"a": t.elements[10],
|
|
||||||
"b": t.elements[11],
|
|
||||||
"c": t.elements[12],
|
|
||||||
"d": t.elements[13],
|
|
||||||
"e": t.elements[14],
|
|
||||||
"f": t.elements[15],
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
return nil, fmt.Errorf("nodeKind %s not supported", t.nodeKind)
|
|
||||||
}
|
|
||||||
|
|
||||||
return json.Marshal(out)
|
|
||||||
}
|
|
||||||
|
|
||||||
// nibbleToByte expands the nibbles of a byte slice into their own bytes.
|
|
||||||
func nibbleToByte(k []byte) []byte {
|
|
||||||
var out []byte
|
|
||||||
|
|
||||||
for _, b := range k {
|
|
||||||
out = append(out, b/16)
|
|
||||||
out = append(out, b%16)
|
|
||||||
}
|
|
||||||
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
|
|
||||||
// Resolve reading conveniences
|
|
||||||
func (t *TrieNode) resolveTrieNodeExtension(p []string) (interface{}, []string, error) {
|
|
||||||
nibbles := t.elements[0].([]byte)
|
|
||||||
idx, rest := shiftFromPath(p, len(nibbles))
|
|
||||||
if len(idx) < len(nibbles) {
|
|
||||||
return nil, nil, fmt.Errorf("not enough nibbles to traverse this extension")
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, i := range idx {
|
|
||||||
if getHexIndex(string(i)) == -1 {
|
|
||||||
return nil, nil, fmt.Errorf("invalid path element")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, n := range nibbles {
|
|
||||||
if string(idx[i]) != fmt.Sprintf("%x", n) {
|
|
||||||
return nil, nil, fmt.Errorf("no such link in this extension")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return &node.Link{Cid: t.elements[1].(cid.Cid)}, rest, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TrieNode) resolveTrieNodeLeaf(p []string) (interface{}, []string, error) {
|
|
||||||
nibbles := t.elements[0].([]byte)
|
|
||||||
|
|
||||||
if len(nibbles) != 0 {
|
|
||||||
idx, rest := shiftFromPath(p, len(nibbles))
|
|
||||||
if len(idx) < len(nibbles) {
|
|
||||||
return nil, nil, fmt.Errorf("not enough nibbles to traverse this leaf")
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, i := range idx {
|
|
||||||
if getHexIndex(string(i)) == -1 {
|
|
||||||
return nil, nil, fmt.Errorf("invalid path element")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for i, n := range nibbles {
|
|
||||||
if string(idx[i]) != fmt.Sprintf("%x", n) {
|
|
||||||
return nil, nil, fmt.Errorf("no such link in this extension")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
p = rest
|
|
||||||
}
|
|
||||||
|
|
||||||
link, ok := t.elements[1].(node.Node)
|
|
||||||
if !ok {
|
|
||||||
return nil, nil, fmt.Errorf("leaf children is not an IPLD node")
|
|
||||||
}
|
|
||||||
|
|
||||||
return link.Resolve(p)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TrieNode) resolveTrieNodeBranch(p []string) (interface{}, []string, error) {
|
|
||||||
idx, rest := shiftFromPath(p, 1)
|
|
||||||
hidx := getHexIndex(idx)
|
|
||||||
if hidx == -1 {
|
|
||||||
return nil, nil, fmt.Errorf("incorrect path")
|
|
||||||
}
|
|
||||||
|
|
||||||
child := t.elements[hidx]
|
|
||||||
if child != nil {
|
|
||||||
return &node.Link{Cid: child.(cid.Cid)}, rest, nil
|
|
||||||
}
|
|
||||||
return nil, nil, fmt.Errorf("no such link in this branch")
|
|
||||||
}
|
|
||||||
|
|
||||||
// shiftFromPath extracts from a given path (as a slice of strings)
|
|
||||||
// the given number of elements as a single string, returning whatever
|
|
||||||
// it has not taken.
|
|
||||||
//
|
|
||||||
// Examples:
|
|
||||||
// ["0", "a", "something"] and 1 -> "0" and ["a", "something"]
|
|
||||||
// ["ab", "c", "d", "1"] and 2 -> "ab" and ["c", "d", "1"]
|
|
||||||
// ["abc", "d", "1"] and 2 -> "ab" and ["c", "d", "1"]
|
|
||||||
func shiftFromPath(p []string, i int) (string, []string) {
|
|
||||||
var (
|
|
||||||
out string
|
|
||||||
rest []string
|
|
||||||
)
|
|
||||||
|
|
||||||
for _, pe := range p {
|
|
||||||
re := ""
|
|
||||||
for _, c := range pe {
|
|
||||||
if len(out) < i {
|
|
||||||
out += string(c)
|
|
||||||
} else {
|
|
||||||
re += string(c)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(out) == i && re != "" {
|
|
||||||
rest = append(rest, re)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return out, rest
|
|
||||||
}
|
|
||||||
|
|
||||||
// getHexIndex returns to you the integer 0 - 15 equivalent to your
|
|
||||||
// string character if applicable, or -1 otherwise.
|
|
||||||
func getHexIndex(s string) int {
|
|
||||||
if len(s) != 1 {
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
|
|
||||||
c := byte(s[0])
|
|
||||||
switch {
|
|
||||||
case '0' <= c && c <= '9':
|
|
||||||
return int(c - '0')
|
|
||||||
case 'a' <= c && c <= 'f':
|
|
||||||
return int(c - 'a' + 10)
|
|
||||||
}
|
|
||||||
|
|
||||||
return -1
|
|
||||||
}
|
|
@ -17,10 +17,11 @@
|
|||||||
package eth
|
package eth
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"github.com/ethereum/go-ethereum/statediff"
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
"github.com/vulcanize/vulcanizedb/pkg/super_node/shared"
|
|
||||||
|
|
||||||
"github.com/vulcanize/vulcanizedb/pkg/eth/core"
|
"github.com/vulcanize/vulcanizedb/pkg/eth/core"
|
||||||
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/shared"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@ -42,6 +43,15 @@ func NewPayloadStreamer(client core.RPCClient) *PayloadStreamer {
|
|||||||
// Stream is the main loop for subscribing to data from the Geth state diff process
|
// Stream is the main loop for subscribing to data from the Geth state diff process
|
||||||
// Satisfies the shared.PayloadStreamer interface
|
// Satisfies the shared.PayloadStreamer interface
|
||||||
func (ps *PayloadStreamer) Stream(payloadChan chan shared.RawChainData) (shared.ClientSubscription, error) {
|
func (ps *PayloadStreamer) Stream(payloadChan chan shared.RawChainData) (shared.ClientSubscription, error) {
|
||||||
|
stateDiffChan := make(chan statediff.Payload, PayloadChanBufferSize)
|
||||||
logrus.Info("streaming diffs from geth")
|
logrus.Info("streaming diffs from geth")
|
||||||
return ps.Client.Subscribe("statediff", payloadChan, "stream")
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case payload := <-stateDiffChan:
|
||||||
|
payloadChan <- payload
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
return ps.Client.Subscribe("statediff", stateDiffChan, "stream")
|
||||||
}
|
}
|
||||||
|
@ -17,10 +17,10 @@ package eth_test
|
|||||||
import (
|
import (
|
||||||
. "github.com/onsi/ginkgo"
|
. "github.com/onsi/ginkgo"
|
||||||
. "github.com/onsi/gomega"
|
. "github.com/onsi/gomega"
|
||||||
"github.com/vulcanize/vulcanizedb/pkg/super_node/shared"
|
|
||||||
|
|
||||||
"github.com/vulcanize/vulcanizedb/pkg/eth/fakes"
|
"github.com/vulcanize/vulcanizedb/pkg/eth/fakes"
|
||||||
"github.com/vulcanize/vulcanizedb/pkg/super_node/eth"
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/eth"
|
||||||
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/shared"
|
||||||
)
|
)
|
||||||
|
|
||||||
var _ = Describe("StateDiff Streamer", func() {
|
var _ = Describe("StateDiff Streamer", func() {
|
||||||
@ -30,6 +30,5 @@ var _ = Describe("StateDiff Streamer", func() {
|
|||||||
payloadChan := make(chan shared.RawChainData)
|
payloadChan := make(chan shared.RawChainData)
|
||||||
_, err := streamer.Stream(payloadChan)
|
_, err := streamer.Stream(payloadChan)
|
||||||
Expect(err).NotTo(HaveOccurred())
|
Expect(err).NotTo(HaveOccurred())
|
||||||
client.AssertSubscribeCalledWith("statediff", payloadChan, []interface{}{"stream"})
|
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
Loading…
Reference in New Issue
Block a user