lotus/storage/addresses.go

137 lines
4.0 KiB
Go
Raw Normal View History

package storage
import (
"context"
"github.com/filecoin-project/go-address"
2020-09-07 03:49:10 +00:00
"github.com/filecoin-project/go-state-types/abi"
2020-12-02 19:46:07 +00:00
"github.com/filecoin-project/lotus/api"
"github.com/filecoin-project/lotus/chain/actors/builtin/miner"
"github.com/filecoin-project/lotus/chain/types"
)
type addrSelectApi interface {
WalletBalance(context.Context, address.Address) (types.BigInt, error)
WalletHas(context.Context, address.Address) (bool, error)
StateAccountKey(context.Context, address.Address, types.TipSetKey) (address.Address, error)
StateLookupID(context.Context, address.Address, types.TipSetKey) (address.Address, error)
}
type AddressSelector struct {
2020-12-02 19:46:07 +00:00
api.AddressConfig
}
2020-12-02 20:47:45 +00:00
func (as *AddressSelector) AddressFor(ctx context.Context, a addrSelectApi, mi miner.MinerInfo, use api.AddrUse, goodFunds, minFunds abi.TokenAmount) (address.Address, abi.TokenAmount, error) {
var addrs []address.Address
switch use {
2020-12-02 20:47:45 +00:00
case api.PreCommitAddr:
addrs = append(addrs, as.PreCommitControl...)
2020-12-02 20:47:45 +00:00
case api.CommitAddr:
addrs = append(addrs, as.CommitControl...)
2021-01-14 11:37:23 +00:00
case api.TerminateSectorsAddr:
addrs = append(addrs, as.TerminateControl...)
default:
defaultCtl := map[address.Address]struct{}{}
for _, a := range mi.ControlAddresses {
defaultCtl[a] = struct{}{}
}
delete(defaultCtl, mi.Owner)
delete(defaultCtl, mi.Worker)
for _, addr := range append(append([]address.Address{}, as.PreCommitControl...), as.CommitControl...) {
if addr.Protocol() != address.ID {
var err error
addr, err = a.StateLookupID(ctx, addr, types.EmptyTSK)
if err != nil {
log.Warnw("looking up control address", "address", addr, "error", err)
continue
}
}
delete(defaultCtl, addr)
}
for a := range defaultCtl {
addrs = append(addrs, a)
}
}
addrs = append(addrs, mi.Owner, mi.Worker)
return pickAddress(ctx, a, mi, goodFunds, minFunds, addrs)
}
func pickAddress(ctx context.Context, a addrSelectApi, mi miner.MinerInfo, goodFunds, minFunds abi.TokenAmount, addrs []address.Address) (address.Address, abi.TokenAmount, error) {
2020-10-26 15:28:39 +00:00
leastBad := mi.Worker
bestAvail := minFunds
ctl := map[address.Address]struct{}{}
for _, a := range append(mi.ControlAddresses, mi.Owner, mi.Worker) {
ctl[a] = struct{}{}
}
for _, addr := range addrs {
2020-12-02 19:46:07 +00:00
if addr.Protocol() != address.ID {
2020-12-04 16:02:23 +00:00
var err error
addr, err = a.StateLookupID(ctx, addr, types.EmptyTSK)
2020-12-02 19:46:07 +00:00
if err != nil {
log.Warnw("looking up control address", "address", addr, "error", err)
continue
}
}
if _, ok := ctl[addr]; !ok {
log.Warnw("non-control address configured for sending messages", "address", addr)
continue
}
2020-10-26 15:28:39 +00:00
if maybeUseAddress(ctx, a, addr, goodFunds, &leastBad, &bestAvail) {
return leastBad, bestAvail, nil
}
2020-10-26 15:28:39 +00:00
}
2020-11-19 19:37:00 +00:00
log.Warnw("No address had enough funds to for full PoSt message Fee, selecting least bad address", "address", leastBad, "balance", types.FIL(bestAvail), "optimalFunds", types.FIL(goodFunds), "minFunds", types.FIL(minFunds))
2020-11-19 19:37:00 +00:00
return leastBad, bestAvail, nil
2020-10-26 15:28:39 +00:00
}
func maybeUseAddress(ctx context.Context, a addrSelectApi, addr address.Address, goodFunds abi.TokenAmount, leastBad *address.Address, bestAvail *abi.TokenAmount) bool {
b, err := a.WalletBalance(ctx, addr)
if err != nil {
2020-10-26 15:28:39 +00:00
log.Errorw("checking control address balance", "addr", addr, "error", err)
return false
}
2020-10-26 15:28:39 +00:00
if b.GreaterThanEqual(goodFunds) {
k, err := a.StateAccountKey(ctx, addr, types.EmptyTSK)
if err != nil {
2020-10-26 15:28:39 +00:00
log.Errorw("getting account key", "error", err)
return false
}
2020-10-26 15:28:39 +00:00
have, err := a.WalletHas(ctx, k)
if err != nil {
log.Errorw("failed to check control address", "addr", addr, "error", err)
return false
}
2020-10-26 15:28:39 +00:00
if !have {
log.Errorw("don't have key", "key", k, "address", addr)
2020-10-26 15:28:39 +00:00
return false
}
2020-10-26 15:28:39 +00:00
*leastBad = addr
*bestAvail = b
return true
}
2020-10-26 15:28:39 +00:00
if b.GreaterThan(*bestAvail) {
*leastBad = addr
*bestAvail = b
}
log.Warnw("address didn't have enough funds to send message", "address", addr, "required", types.FIL(goodFunds), "balance", types.FIL(b))
2020-10-26 15:28:39 +00:00
return false
2020-08-19 23:26:13 +00:00
}