5733c71c50
We were ignoring quite a few error cases, and had one case where we weren't actually updating state where we wanted to. Unfortunately, if the linter doesn't pass, nobody has any reason to actually check lint failures in CI. There are three remaining XXXs marked in the code for lint.
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
|
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
)
|
|
|
|
type AddrUse int
|
|
|
|
const (
|
|
PreCommitAddr AddrUse = iota
|
|
CommitAddr
|
|
PoStAddr
|
|
)
|
|
|
|
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)
|
|
}
|
|
|
|
func AddressFor(ctx context.Context, a addrSelectApi, mi api.MinerInfo, use AddrUse, minFunds abi.TokenAmount) (address.Address, error) {
|
|
switch use {
|
|
case PreCommitAddr, CommitAddr:
|
|
// always use worker, at least for now
|
|
return mi.Worker, nil
|
|
}
|
|
|
|
for _, addr := range mi.ControlAddresses {
|
|
b, err := a.WalletBalance(ctx, addr)
|
|
if err != nil {
|
|
return address.Undef, xerrors.Errorf("checking control address balance: %w", err)
|
|
}
|
|
|
|
if b.GreaterThanEqual(minFunds) {
|
|
k, err := a.StateAccountKey(ctx, addr, types.EmptyTSK)
|
|
if err != nil {
|
|
log.Errorw("getting account key", "error", err)
|
|
continue
|
|
}
|
|
|
|
have, err := a.WalletHas(ctx, k)
|
|
if err != nil {
|
|
return address.Undef, xerrors.Errorf("failed to check control address: %w", err)
|
|
}
|
|
|
|
if !have {
|
|
log.Errorw("don't have key", "key", k)
|
|
continue
|
|
}
|
|
|
|
return addr, nil
|
|
}
|
|
|
|
log.Warnw("control address didn't have enough funds for PoSt message", "address", addr, "required", types.FIL(minFunds), "balance", types.FIL(b))
|
|
}
|
|
|
|
// Try to use the owner account if we can, fallback to worker if we can't
|
|
|
|
b, err := a.WalletBalance(ctx, mi.Owner)
|
|
if err != nil {
|
|
return address.Undef, xerrors.Errorf("checking owner balance: %w", err)
|
|
}
|
|
|
|
if !b.GreaterThanEqual(minFunds) {
|
|
return mi.Worker, nil
|
|
}
|
|
|
|
k, err := a.StateAccountKey(ctx, mi.Owner, types.EmptyTSK)
|
|
if err != nil {
|
|
log.Errorw("getting owner account key", "error", err)
|
|
return mi.Worker, nil
|
|
}
|
|
|
|
have, err := a.WalletHas(ctx, k)
|
|
if err != nil {
|
|
return address.Undef, xerrors.Errorf("failed to check owner address: %w", err)
|
|
}
|
|
|
|
if !have {
|
|
return mi.Worker, nil
|
|
}
|
|
|
|
return mi.Owner, nil
|
|
}
|