lotus/chain/vm/mkactor.go

86 lines
2.2 KiB
Go
Raw Normal View History

2019-07-26 04:54:22 +00:00
package vm
import (
"context"
2019-10-15 04:33:29 +00:00
"github.com/ipfs/go-cid"
dstore "github.com/ipfs/go-datastore"
hamt "github.com/ipfs/go-hamt-ipld"
bstore "github.com/ipfs/go-ipfs-blockstore"
2019-07-26 04:54:22 +00:00
"github.com/filecoin-project/go-lotus/chain/actors"
"github.com/filecoin-project/go-lotus/chain/actors/aerrors"
2019-07-26 04:54:22 +00:00
"github.com/filecoin-project/go-lotus/chain/address"
"github.com/filecoin-project/go-lotus/chain/state"
"github.com/filecoin-project/go-lotus/chain/types"
)
2019-07-25 22:15:33 +00:00
func init() {
bs := bstore.NewBlockstore(dstore.NewMapDatastore())
cst := hamt.CSTFromBstore(bs)
emptyobject, err := cst.Put(context.TODO(), map[string]string{})
if err != nil {
panic(err)
}
EmptyObjectCid = emptyobject
}
var EmptyObjectCid cid.Cid
func TryCreateAccountActor(st *state.StateTree, addr address.Address) (*types.Actor, aerrors.ActorError) {
2019-07-26 04:54:22 +00:00
act, err := makeActor(st, addr)
if err != nil {
return nil, err
}
if _, err := st.RegisterNewAddress(addr, act); err != nil {
return nil, aerrors.Escalate(err, "registering actor address")
2019-07-26 04:54:22 +00:00
}
return act, nil
}
func makeActor(st *state.StateTree, addr address.Address) (*types.Actor, aerrors.ActorError) {
2019-07-26 04:54:22 +00:00
switch addr.Protocol() {
case address.BLS:
return NewBLSAccountActor(st, addr)
case address.SECP256K1:
return NewSecp256k1AccountActor(st, addr)
case address.ID:
return nil, aerrors.New(1, "no actor with given ID")
2019-07-26 04:54:22 +00:00
case address.Actor:
return nil, aerrors.Newf(1, "no such actor: %s", addr)
2019-07-26 04:54:22 +00:00
default:
return nil, aerrors.Newf(1, "address has unsupported protocol: %d", addr.Protocol())
2019-07-26 04:54:22 +00:00
}
}
func NewBLSAccountActor(st *state.StateTree, addr address.Address) (*types.Actor, aerrors.ActorError) {
2019-07-26 04:54:22 +00:00
var acstate actors.AccountActorState
acstate.Address = addr
c, err := st.Store.Put(context.TODO(), &acstate)
2019-07-26 04:54:22 +00:00
if err != nil {
return nil, aerrors.Escalate(err, "serializing account actor state")
2019-07-26 04:54:22 +00:00
}
nact := &types.Actor{
Code: actors.AccountActorCodeCid,
Balance: types.NewInt(0),
Head: c,
}
return nact, nil
}
func NewSecp256k1AccountActor(st *state.StateTree, addr address.Address) (*types.Actor, aerrors.ActorError) {
2019-07-26 04:54:22 +00:00
nact := &types.Actor{
Code: actors.AccountActorCodeCid,
Balance: types.NewInt(0),
2019-07-25 22:15:33 +00:00
Head: EmptyObjectCid,
2019-07-26 04:54:22 +00:00
}
return nact, nil
}