00cd89750d
Updates dependencies for graphsync, fil-markets, data-transfer. Moves to new graphsync blockstore swapping capabilities, and also locks down graphsync impl so it does not accept arbitrary requests
44 lines
1.8 KiB
Go
44 lines
1.8 KiB
Go
package modules
|
|
|
|
import (
|
|
"github.com/filecoin-project/lotus/node/modules/dtypes"
|
|
"github.com/filecoin-project/lotus/node/modules/helpers"
|
|
"github.com/ipfs/go-graphsync"
|
|
graphsyncimpl "github.com/ipfs/go-graphsync/impl"
|
|
gsnet "github.com/ipfs/go-graphsync/network"
|
|
"github.com/ipfs/go-graphsync/storeutil"
|
|
"github.com/libp2p/go-libp2p-core/host"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
"go.uber.org/fx"
|
|
)
|
|
|
|
// Graphsync creates a graphsync instance from the given loader and storer
|
|
func Graphsync(mctx helpers.MetricsCtx, lc fx.Lifecycle, clientBs dtypes.ClientBlockstore, chainBs dtypes.ChainBlockstore, h host.Host) (dtypes.Graphsync, error) {
|
|
graphsyncNetwork := gsnet.NewFromLibp2pHost(h)
|
|
loader := storeutil.LoaderForBlockstore(clientBs)
|
|
storer := storeutil.StorerForBlockstore(clientBs)
|
|
gs := graphsyncimpl.New(helpers.LifecycleCtx(mctx, lc), graphsyncNetwork, loader, storer, graphsyncimpl.RejectAllRequestsByDefault())
|
|
chainLoader := storeutil.LoaderForBlockstore(chainBs)
|
|
chainStorer := storeutil.StorerForBlockstore(chainBs)
|
|
err := gs.RegisterPersistenceOption("chainstore", chainLoader, chainStorer)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
gs.RegisterIncomingRequestHook(func(p peer.ID, requestData graphsync.RequestData, hookActions graphsync.IncomingRequestHookActions) {
|
|
_, has := requestData.Extension("chainsync")
|
|
if has {
|
|
// TODO: we should confirm the selector is a reasonable one before we validate
|
|
// TODO: this code will get more complicated and should probably not live here eventually
|
|
hookActions.ValidateRequest()
|
|
hookActions.UsePersistenceOption("chainstore")
|
|
}
|
|
})
|
|
gs.RegisterOutgoingRequestHook(func(p peer.ID, requestData graphsync.RequestData, hookActions graphsync.OutgoingRequestHookActions) {
|
|
_, has := requestData.Extension("chainsync")
|
|
if has {
|
|
hookActions.UsePersistenceOption("chainstore")
|
|
}
|
|
})
|
|
return gs, nil
|
|
}
|