85 lines
2.0 KiB
Go
85 lines
2.0 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 patchStateTree(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
|
|
}
|