lotus/chain/gen/genesis/util.go

85 lines
2.1 KiB
Go

package genesis
import (
"context"
"github.com/filecoin-project/go-address"
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/go-state-types/network"
cbg "github.com/whyrusleeping/cbor-gen"
"golang.org/x/xerrors"
"github.com/filecoin-project/lotus/chain/actors"
"github.com/filecoin-project/lotus/chain/actors/builtin"
"github.com/filecoin-project/lotus/chain/state"
"github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/lotus/chain/vm"
)
func mustEnc(i cbg.CBORMarshaler) []byte {
enc, err := actors.SerializeParams(i)
if err != nil {
panic(err) // ok
}
return enc
}
func doExecValue(ctx context.Context, vm *vm.LegacyVM, to, from address.Address, value types.BigInt, method abi.MethodNum, params []byte) ([]byte, error) {
act, err := vm.StateTree().GetActor(from)
if err != nil {
return nil, xerrors.Errorf("doExec failed to get from actor (%s): %w", from, err)
}
ret, err := vm.ApplyImplicitMessage(ctx, &types.Message{
To: to,
From: from,
Method: method,
Params: params,
GasLimit: 1_000_000_000_000_000,
Value: value,
Nonce: act.Nonce,
})
if err != nil {
return nil, xerrors.Errorf("doExec apply message failed: %w", err)
}
if ret.ExitCode != 0 {
return nil, xerrors.Errorf("failed to call method: %w", ret.ActorErr)
}
return ret.Return, nil
}
func patchManifestCodeCids(st *state.StateTree, nv network.Version) error {
av, err := actors.VersionForNetwork(nv)
if err != nil {
return err
}
var acts []address.Address
err = st.ForEach(func(a address.Address, _ *types.Actor) error {
acts = append(acts, a)
return nil
})
if err != nil {
return xerrors.Errorf("error collecting actors: %w", err)
}
for _, a := range acts {
err = st.MutateActor(a, func(act *types.Actor) error {
name := actors.CanonicalName(builtin.ActorNameByCode(act.Code))
code, ok := actors.GetActorCodeID(av, name)
if ok {
act.Code = code
}
return nil
})
if err != nil {
return xerrors.Errorf("error mutating actor %s: %w", a, err)
}
}
return nil
}