95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"github.com/filecoin-project/lotus/build"
|
|
"github.com/filecoin-project/lotus/chain/actors"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/vm"
|
|
"golang.org/x/xerrors"
|
|
"math/big"
|
|
)
|
|
|
|
var zero = types.NewInt(0)
|
|
|
|
func (cs *ChainStore) Weight(ctx context.Context, ts *types.TipSet) (types.BigInt, error) {
|
|
if ts == nil {
|
|
return types.NewInt(0), nil
|
|
}
|
|
// >>> w[r] <<< + wFunction(totalPowerAtTipset(ts)) * 2^8 + (wFunction(totalPowerAtTipset(ts)) * len(ts.blocks) * wRatio_num * 2^8) / (e * wRatio_den)
|
|
|
|
var out = new(big.Int).Set(ts.Blocks()[0].ParentWeight.Int)
|
|
|
|
// >>> wFunction(totalPowerAtTipset(ts)) * 2^8 <<< + (wFunction(totalPowerAtTipset(ts)) * len(ts.blocks) * wRatio_num * 2^8) / (e * wRatio_den)
|
|
|
|
ret, err := cs.call(ctx, &types.Message{
|
|
From: actors.StoragePowerAddress,
|
|
To: actors.StoragePowerAddress,
|
|
Method: actors.SPAMethods.GetTotalStorage,
|
|
}, ts)
|
|
if err != nil {
|
|
return types.EmptyInt, xerrors.Errorf("failed to get total power from chain: %w", err)
|
|
}
|
|
if ret.ExitCode != 0 {
|
|
return types.EmptyInt, xerrors.Errorf("failed to get total power from chain (exit code %d)", ret.ExitCode)
|
|
}
|
|
log2P := int64(0)
|
|
tpow := types.BigFromBytes(ret.Return)
|
|
if tpow.GreaterThan(zero) {
|
|
log2P = int64(tpow.BitLen() - 1)
|
|
}
|
|
|
|
out.Add(out, big.NewInt(log2P*256))
|
|
|
|
// (wFunction(totalPowerAtTipset(ts)) * len(ts.blocks) * wRatio_num * 2^8) / (e * wRatio_den)
|
|
|
|
wRatioNum := int64(1)
|
|
wRatioDen := 2
|
|
|
|
eWeight := big.NewInt((log2P * int64(len(ts.Blocks())) * wRatioNum) << 8)
|
|
eWeight.Div(eWeight, big.NewInt(int64(build.BlocksPerEpoch*wRatioDen)))
|
|
out.Add(out, eWeight)
|
|
|
|
return types.BigInt{Int: out}, nil
|
|
}
|
|
|
|
// todo: dedupe with state manager
|
|
func (cs *ChainStore) call(ctx context.Context, msg *types.Message, ts *types.TipSet) (*types.MessageReceipt, error) {
|
|
bstate := ts.ParentState()
|
|
|
|
r := NewChainRand(cs, ts.Cids(), ts.Height(), nil)
|
|
|
|
vmi, err := vm.NewVM(bstate, ts.Height(), r, actors.NetworkAddress, cs.bs)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("failed to set up vm: %w", err)
|
|
}
|
|
|
|
if msg.GasLimit == types.EmptyInt {
|
|
msg.GasLimit = types.NewInt(10000000000)
|
|
}
|
|
if msg.GasPrice == types.EmptyInt {
|
|
msg.GasPrice = types.NewInt(0)
|
|
}
|
|
if msg.Value == types.EmptyInt {
|
|
msg.Value = types.NewInt(0)
|
|
}
|
|
|
|
fromActor, err := vmi.StateTree().GetActor(msg.From)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("call raw get actor: %s", err)
|
|
}
|
|
|
|
msg.Nonce = fromActor.Nonce
|
|
|
|
// TODO: maybe just use the invoker directly?
|
|
ret, err := vmi.ApplyMessage(ctx, msg)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("apply message failed: %w", err)
|
|
}
|
|
|
|
if ret.ActorErr != nil {
|
|
log.Warnf("chain call failed: %s", ret.ActorErr)
|
|
}
|
|
return &ret.MessageReceipt, nil
|
|
}
|