56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package state
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"compress/gzip"
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/lotus/chain/state"
|
||
|
|
bs "github.com/filecoin-project/lotus/lib/blockstore"
|
||
|
|
"github.com/ipfs/go-hamt-ipld"
|
||
|
|
cbor "github.com/ipfs/go-ipld-cbor"
|
||
|
|
"github.com/ipfs/go-ipld-format"
|
||
|
|
"github.com/ipld/go-car"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RecoverStateTree parses a car encoding of a state tree back to a structured format
|
||
|
|
func RecoverStateTree(ctx context.Context, raw []byte) (*state.StateTree, error) {
|
||
|
|
buf := bytes.NewBuffer(raw)
|
||
|
|
store := bs.NewTemporary()
|
||
|
|
gr, err := gzip.NewReader(buf)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer gr.Close()
|
||
|
|
|
||
|
|
ch, err := car.LoadCar(store, gr)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if len(ch.Roots) != 1 {
|
||
|
|
return nil, fmt.Errorf("car should have 1 root, has %d", len(ch.Roots))
|
||
|
|
}
|
||
|
|
cborstore := cbor.NewCborStore(store)
|
||
|
|
|
||
|
|
fmt.Printf("root is %s\n", ch.Roots[0])
|
||
|
|
|
||
|
|
nd, err := hamt.LoadNode(ctx, cborstore, ch.Roots[0], hamt.UseTreeBitWidth(5))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if err := nd.ForEach(ctx, func(k string, val interface{}) error {
|
||
|
|
n, ok := val.(format.Node)
|
||
|
|
if !ok {
|
||
|
|
fmt.Printf("hampt %s (not node): %+v\n", k, val)
|
||
|
|
} else {
|
||
|
|
fmt.Printf("%s: %#v\n", k, n)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return state.LoadStateTree(cborstore, ch.Roots[0])
|
||
|
|
}
|