106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/ipfs/go-blockservice"
|
|
"github.com/ipfs/go-cid"
|
|
offline "github.com/ipfs/go-ipfs-exchange-offline"
|
|
logging "github.com/ipfs/go-log/v2"
|
|
"github.com/ipfs/go-merkledag"
|
|
"github.com/ipld/go-car"
|
|
"github.com/mitchellh/go-homedir"
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/chain/gen"
|
|
genesis2 "github.com/filecoin-project/lotus/chain/gen/genesis"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/vm"
|
|
"github.com/filecoin-project/lotus/genesis"
|
|
"github.com/filecoin-project/lotus/node/modules"
|
|
"github.com/filecoin-project/lotus/node/modules/dtypes"
|
|
)
|
|
|
|
var glog = logging.Logger("genesis")
|
|
|
|
func MakeGenesisMem(out io.Writer, template genesis.Template) func(bs dtypes.ChainBlockstore, syscalls vm.SyscallBuilder) modules.Genesis {
|
|
return func(bs dtypes.ChainBlockstore, syscalls vm.SyscallBuilder) modules.Genesis {
|
|
return func() (*types.BlockHeader, error) {
|
|
glog.Warn("Generating new random genesis block, note that this SHOULD NOT happen unless you are setting up new network")
|
|
b, err := genesis2.MakeGenesisBlock(context.TODO(), bs, syscalls, template)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("make genesis block failed: %w", err)
|
|
}
|
|
offl := offline.Exchange(bs)
|
|
blkserv := blockservice.New(bs, offl)
|
|
dserv := merkledag.NewDAGService(blkserv)
|
|
|
|
if err := car.WriteCarWithWalker(context.TODO(), dserv, []cid.Cid{b.Genesis.Cid()}, out, gen.CarWalkFunc); err != nil {
|
|
return nil, xerrors.Errorf("failed to write car file: %w", err)
|
|
}
|
|
|
|
return b.Genesis, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func MakeGenesis(outFile, genesisTemplate string) func(bs dtypes.ChainBlockstore, syscalls vm.SyscallBuilder) modules.Genesis {
|
|
return func(bs dtypes.ChainBlockstore, syscalls vm.SyscallBuilder) modules.Genesis {
|
|
return func() (*types.BlockHeader, error) {
|
|
glog.Warn("Generating new random genesis block, note that this SHOULD NOT happen unless you are setting up new network")
|
|
genesisTemplate, err := homedir.Expand(genesisTemplate)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fdata, err := ioutil.ReadFile(genesisTemplate)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("reading preseals json: %w", err)
|
|
}
|
|
|
|
var template genesis.Template
|
|
if err := json.Unmarshal(fdata, &template); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if template.Timestamp == 0 {
|
|
template.Timestamp = uint64(build.Clock.Now().Unix())
|
|
}
|
|
|
|
b, err := genesis2.MakeGenesisBlock(context.TODO(), bs, syscalls, template)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("make genesis block: %w", err)
|
|
}
|
|
|
|
fmt.Printf("GENESIS MINER ADDRESS: t0%d\n", genesis2.MinerStart)
|
|
|
|
f, err := os.OpenFile(outFile, os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
offl := offline.Exchange(bs)
|
|
blkserv := blockservice.New(bs, offl)
|
|
dserv := merkledag.NewDAGService(blkserv)
|
|
|
|
if err := car.WriteCarWithWalker(context.TODO(), dserv, []cid.Cid{b.Genesis.Cid()}, f, gen.CarWalkFunc); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
glog.Warnf("WRITING GENESIS FILE AT %s", f.Name())
|
|
|
|
if err := f.Close(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return b.Genesis, nil
|
|
}
|
|
}
|
|
}
|