90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package builders
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/lotus/chain/actors/aerrors"
|
|
"github.com/multiformats/go-varint"
|
|
)
|
|
|
|
// AddressHandle encapsulates both the ID and Robust addresses of an actor.
|
|
type AddressHandle struct {
|
|
ID, Robust address.Address
|
|
}
|
|
|
|
// NextActorAddress predicts the address of the next actor created by this address.
|
|
//
|
|
// Code is adapted from vm.Runtime#NewActorAddress()
|
|
func (ah *AddressHandle) NextActorAddress(nonce, numActorsCreated uint64) address.Address {
|
|
var b bytes.Buffer
|
|
if err := ah.Robust.MarshalCBOR(&b); err != nil {
|
|
panic(aerrors.Fatalf("writing caller address into assert buffer: %v", err))
|
|
}
|
|
|
|
if err := binary.Write(&b, binary.BigEndian, nonce); err != nil {
|
|
panic(aerrors.Fatalf("writing nonce address into assert buffer: %v", err))
|
|
}
|
|
if err := binary.Write(&b, binary.BigEndian, numActorsCreated); err != nil {
|
|
panic(aerrors.Fatalf("writing callSeqNum address into assert buffer: %v", err))
|
|
}
|
|
addr, err := address.NewActorAddress(b.Bytes())
|
|
if err != nil {
|
|
panic(aerrors.Fatalf("create actor address: %v", err))
|
|
}
|
|
return addr
|
|
}
|
|
|
|
// MustNewIDAddr returns an address.Address of kind ID.
|
|
func MustNewIDAddr(id uint64) address.Address {
|
|
addr, err := address.NewIDAddress(id)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return addr
|
|
}
|
|
|
|
// MustNewSECP256K1Addr returns an address.Address of kind secp256k1.
|
|
func MustNewSECP256K1Addr(pubkey string) address.Address {
|
|
// the pubkey of assert secp256k1 address is hashed for consistent length.
|
|
addr, err := address.NewSecp256k1Address([]byte(pubkey))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return addr
|
|
}
|
|
|
|
// MustNewBLSAddr returns an address.Address of kind bls.
|
|
func MustNewBLSAddr(seed int64) address.Address {
|
|
buf := make([]byte, address.BlsPublicKeyBytes)
|
|
binary.PutVarint(buf, seed)
|
|
|
|
addr, err := address.NewBLSAddress(buf)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return addr
|
|
}
|
|
|
|
// MustNewActorAddr returns an address.Address of kind actor.
|
|
func MustNewActorAddr(data string) address.Address {
|
|
addr, err := address.NewActorAddress([]byte(data))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return addr
|
|
}
|
|
|
|
// MustIDFromAddress returns the integer ID from an ID address.
|
|
func MustIDFromAddress(a address.Address) uint64 {
|
|
if a.Protocol() != address.ID {
|
|
panic("must be ID protocol address")
|
|
}
|
|
id, _, err := varint.FromUvarint(a.Payload())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return id
|
|
}
|