lotus/chain/gen/genesis/f01_init.go

172 lines
4.5 KiB
Go
Raw Normal View History

2020-02-11 20:48:03 +00:00
package genesis
import (
"context"
2020-02-12 00:58:55 +00:00
"encoding/json"
2020-02-21 17:13:50 +00:00
"fmt"
2020-07-28 17:51:47 +00:00
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/go-state-types/abi"
2020-05-14 02:32:04 +00:00
2020-02-25 20:54:58 +00:00
"github.com/filecoin-project/specs-actors/actors/builtin"
"github.com/filecoin-project/specs-actors/actors/util/adt"
2020-02-11 20:48:03 +00:00
2020-02-25 20:35:15 +00:00
init_ "github.com/filecoin-project/specs-actors/actors/builtin/init"
2020-02-11 20:48:03 +00:00
cbor "github.com/ipfs/go-ipld-cbor"
cbg "github.com/whyrusleeping/cbor-gen"
2020-02-11 20:48:03 +00:00
"golang.org/x/xerrors"
bstore "github.com/filecoin-project/lotus/blockstore"
2020-02-11 20:48:03 +00:00
"github.com/filecoin-project/lotus/chain/types"
2020-02-12 00:58:55 +00:00
"github.com/filecoin-project/lotus/genesis"
2020-02-11 20:48:03 +00:00
)
func SetupInitActor(bs bstore.Blockstore, netname string, initialActors []genesis.Actor, rootVerifier genesis.Actor, remainder genesis.Actor) (int64, *types.Actor, map[address.Address]address.Address, error) {
2020-02-12 00:58:55 +00:00
if len(initialActors) > MaxAccounts {
return 0, nil, nil, xerrors.New("too many initial actors")
2020-02-12 00:58:55 +00:00
}
2020-02-25 20:35:15 +00:00
var ias init_.State
2020-02-12 00:58:55 +00:00
ias.NextID = MinerStart
ias.NetworkName = netname
2020-02-11 20:48:03 +00:00
store := adt.WrapStore(context.TODO(), cbor.NewCborStore(bs))
amap := adt.MakeEmptyMap(store)
2020-02-11 20:48:03 +00:00
2020-07-28 17:51:47 +00:00
keyToId := map[address.Address]address.Address{}
counter := int64(AccountStart)
for _, a := range initialActors {
2020-07-15 09:26:38 +00:00
if a.Type == genesis.TMultisig {
var ainfo genesis.MultisigMeta
if err := json.Unmarshal(a.Meta, &ainfo); err != nil {
return 0, nil, nil, xerrors.Errorf("unmarshaling account meta: %w", err)
}
for _, e := range ainfo.Signers {
2020-08-19 09:25:38 +00:00
if _, ok := keyToId[e]; ok {
continue
}
fmt.Printf("init set %s t0%d\n", e, counter)
value := cbg.CborInt(counter)
if err := amap.Put(abi.AddrKey(e), &value); err != nil {
return 0, nil, nil, err
}
counter = counter + 1
var err error
keyToId[e], err = address.NewIDAddress(uint64(value))
if err != nil {
return 0, nil, nil, err
}
}
// Need to add actors for all multisigs too
2020-07-15 09:26:38 +00:00
continue
}
2020-02-12 00:58:55 +00:00
if a.Type != genesis.TAccount {
2020-08-19 20:32:53 +00:00
return 0, nil, nil, xerrors.Errorf("unsupported account type: %s", a.Type)
2020-02-12 00:58:55 +00:00
}
var ainfo genesis.AccountMeta
if err := json.Unmarshal(a.Meta, &ainfo); err != nil {
return 0, nil, nil, xerrors.Errorf("unmarshaling account meta: %w", err)
2020-02-12 00:58:55 +00:00
}
fmt.Printf("init set %s t0%d\n", ainfo.Owner, counter)
2020-02-21 17:13:50 +00:00
value := cbg.CborInt(counter)
if err := amap.Put(abi.AddrKey(ainfo.Owner), &value); err != nil {
return 0, nil, nil, err
2020-07-28 17:51:47 +00:00
}
counter = counter + 1
2020-07-28 17:51:47 +00:00
var err error
keyToId[ainfo.Owner], err = address.NewIDAddress(uint64(value))
if err != nil {
return 0, nil, nil, err
2020-02-11 20:48:03 +00:00
}
}
setupMsig := func(meta json.RawMessage) error {
var ainfo genesis.MultisigMeta
if err := json.Unmarshal(meta, &ainfo); err != nil {
return xerrors.Errorf("unmarshaling account meta: %w", err)
}
for _, e := range ainfo.Signers {
2020-08-19 10:03:59 +00:00
if _, ok := keyToId[e]; ok {
continue
}
fmt.Printf("init set %s t0%d\n", e, counter)
value := cbg.CborInt(counter)
if err := amap.Put(abi.AddrKey(e), &value); err != nil {
return err
}
counter = counter + 1
var err error
keyToId[e], err = address.NewIDAddress(uint64(value))
if err != nil {
return err
}
}
return nil
}
if rootVerifier.Type == genesis.TAccount {
var ainfo genesis.AccountMeta
if err := json.Unmarshal(rootVerifier.Meta, &ainfo); err != nil {
return 0, nil, nil, xerrors.Errorf("unmarshaling account meta: %w", err)
}
value := cbg.CborInt(80)
if err := amap.Put(abi.AddrKey(ainfo.Owner), &value); err != nil {
return 0, nil, nil, err
}
} else if rootVerifier.Type == genesis.TMultisig {
err := setupMsig(rootVerifier.Meta)
if err != nil {
return 0, nil, nil, xerrors.Errorf("setting up root verifier msig: %w", err)
}
}
if remainder.Type == genesis.TAccount {
var ainfo genesis.AccountMeta
if err := json.Unmarshal(remainder.Meta, &ainfo); err != nil {
return 0, nil, nil, xerrors.Errorf("unmarshaling account meta: %w", err)
}
// TODO: Use builtin.ReserveAddress...
value := cbg.CborInt(90)
if err := amap.Put(abi.AddrKey(ainfo.Owner), &value); err != nil {
return 0, nil, nil, err
}
} else if remainder.Type == genesis.TMultisig {
err := setupMsig(remainder.Meta)
if err != nil {
return 0, nil, nil, xerrors.Errorf("setting up remainder msig: %w", err)
}
2020-05-14 02:32:04 +00:00
}
amapaddr, err := amap.Root()
2020-02-11 20:48:03 +00:00
if err != nil {
return 0, nil, nil, err
2020-02-11 20:48:03 +00:00
}
ias.AddressMap = amapaddr
2020-02-11 20:48:03 +00:00
statecid, err := store.Put(store.Context(), &ias)
2020-02-11 20:48:03 +00:00
if err != nil {
return 0, nil, nil, err
2020-02-11 20:48:03 +00:00
}
act := &types.Actor{
2020-02-25 20:54:58 +00:00
Code: builtin.InitActorCodeID,
2020-02-11 20:48:03 +00:00
Head: statecid,
}
return counter, act, keyToId, nil
2020-02-11 20:48:03 +00:00
}