-
Notifications
You must be signed in to change notification settings - Fork 324
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
399 additions
and
25 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
File renamed without changes.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,181 @@ | ||
| package staking | ||
|
|
||
| import ( | ||
| "context" | ||
| "math/big" | ||
|
|
||
| "github.com/pkg/errors" | ||
| "go.uber.org/zap" | ||
|
|
||
| "github.com/iotexproject/iotex-address/address" | ||
| "github.com/iotexproject/iotex-proto/golang/iotextypes" | ||
|
|
||
| "github.com/iotexproject/iotex-core/action" | ||
| "github.com/iotexproject/iotex-core/action/protocol" | ||
| accountutil "github.com/iotexproject/iotex-core/action/protocol/account/util" | ||
| "github.com/iotexproject/iotex-core/pkg/log" | ||
| "github.com/iotexproject/iotex-core/pkg/util/byteutil" | ||
| "github.com/iotexproject/iotex-core/state" | ||
| ) | ||
|
|
||
| const ( | ||
| executionProtocolID = "smart_contract" | ||
| ) | ||
|
|
||
| type ( | ||
| executionProtocol interface { | ||
| Handle(ctx context.Context, act action.Action, sm protocol.StateManager) (*action.Receipt, error) | ||
| } | ||
| ) | ||
|
|
||
| func (p *Protocol) handleStakeMigrate(ctx context.Context, act *action.MigrateStake, csm CandidateStateManager) ([]*action.Log, []*action.TransactionLog, uint64, error) { | ||
| actLogs := make([]*action.Log, 0) | ||
| transferLogs := make([]*action.TransactionLog, 0) | ||
| si := csm.SM().Snapshot() | ||
| revertSM := func() { | ||
| if revertErr := csm.SM().Revert(si); revertErr != nil { | ||
| log.L().Panic("failed to revert state", zap.Error(revertErr)) | ||
| } | ||
| } | ||
|
|
||
| // validate bucket index | ||
| bucket, rErr := p.fetchBucket(csm, act.BucketIndex()) | ||
| if rErr != nil { | ||
| return nil, nil, 0, rErr | ||
| } | ||
| if err := p.validateStakeMigrate(ctx, bucket, csm); err != nil { | ||
| return nil, nil, 0, err | ||
| } | ||
|
|
||
| // force-withdraw native bucket | ||
| staker, rerr := fetchCaller(ctx, csm, big.NewInt(0)) | ||
| if rerr != nil { | ||
| return nil, nil, 0, errors.Wrap(rerr, "failed to fetch caller") | ||
| } | ||
| candidate := csm.GetByIdentifier(bucket.Candidate) | ||
| if candidate == nil { | ||
| return nil, nil, 0, errCandNotExist | ||
| } | ||
| actLog, tLog, err := p.withdrawBucket(ctx, staker, bucket, candidate, csm) | ||
| if err != nil { | ||
| return nil, nil, 0, err | ||
| } | ||
| actLogs = append(actLogs, actLog.Build(ctx, nil)) | ||
| transferLogs = append(transferLogs, tLog) | ||
|
|
||
| // call staking contract to stake | ||
| duration := int64(bucket.StakedDuration / p.getBlockInterval(protocol.MustGetBlockCtx(ctx).BlockHeight)) | ||
| excReceipt, err := p.createNFTBucket(ctx, act, bucket.StakedAmount, big.NewInt(duration), candidate, csm.SM()) | ||
| if err != nil { | ||
| revertSM() | ||
| return nil, nil, 0, errors.Wrap(err, "failed to handle execution action") | ||
| } | ||
| if excReceipt.Status != uint64(iotextypes.ReceiptStatus_Success) { | ||
| // TODO: return err or handle error? | ||
| revertSM() | ||
| return nil, nil, 0, errors.Errorf("execution failed with status %d", excReceipt.Status) | ||
| } | ||
| // add sub-receipts logs | ||
| actLogs = append(actLogs, excReceipt.Logs()...) | ||
| transferLogs = append(transferLogs, excReceipt.TransactionLogs()...) | ||
| return actLogs, transferLogs, excReceipt.GasConsumed, nil | ||
| } | ||
|
|
||
| func (p *Protocol) validateStakeMigrate(ctx context.Context, bucket *VoteBucket, csm CandidateStateManager) error { | ||
| if err := validateBucketOwner(bucket, protocol.MustGetActionCtx(ctx).Caller); err != nil { | ||
| return err | ||
| } | ||
| if !bucket.AutoStake { | ||
| return &handleError{ | ||
| err: errors.New("cannot migrate non-auto-staked bucket"), | ||
| failureStatus: iotextypes.ReceiptStatus_ErrInvalidBucketType, | ||
| } | ||
| } | ||
| if bucket.isUnstaked() { | ||
| return &handleError{ | ||
| err: errors.New("cannot migrate unstaked bucket"), | ||
| failureStatus: iotextypes.ReceiptStatus_ErrInvalidBucketType, | ||
| } | ||
| } | ||
| if err := validateBucketWithoutEndorsement(NewEndorsementStateManager(csm.SM()), bucket, protocol.MustGetBlockCtx(ctx).BlockHeight); err != nil { | ||
| return err | ||
| } | ||
| return validateBucketSelfStake(protocol.MustGetFeatureCtx(ctx), csm, bucket, false) | ||
| } | ||
|
|
||
| func (p *Protocol) withdrawBucket(ctx context.Context, withdrawer *state.Account, bucket *VoteBucket, cand *Candidate, csm CandidateStateManager) (*receiptLog, *action.TransactionLog, error) { | ||
| // delete bucket and bucket index | ||
| if err := csm.delBucketAndIndex(bucket.Owner, bucket.Candidate, bucket.Index); err != nil { | ||
| return nil, nil, errors.Wrapf(err, "failed to delete bucket for candidate %s", bucket.Candidate.String()) | ||
| } | ||
|
|
||
| // update bucket pool | ||
| if err := csm.CreditBucketPool(bucket.StakedAmount); err != nil { | ||
| return nil, nil, errors.Wrapf(err, "failed to update staking bucket pool %s", err.Error()) | ||
| } | ||
| // update candidate vote | ||
| weightedVote := p.calculateVoteWeight(bucket, false) | ||
| if err := cand.SubVote(weightedVote); err != nil { | ||
| return nil, nil, &handleError{ | ||
| err: errors.Wrapf(err, "failed to subtract vote for candidate %s", bucket.Candidate.String()), | ||
| failureStatus: iotextypes.ReceiptStatus_ErrNotEnoughBalance, | ||
| } | ||
| } | ||
| // clear candidate's self stake if the | ||
| if cand.SelfStakeBucketIdx == bucket.Index { | ||
| cand.SelfStake = big.NewInt(0) | ||
| cand.SelfStakeBucketIdx = candidateNoSelfStakeBucketIndex | ||
| } | ||
| if err := csm.Upsert(cand); err != nil { | ||
| return nil, nil, csmErrorToHandleError(cand.GetIdentifier().String(), err) | ||
| } | ||
| // update withdrawer balance | ||
| if err := withdrawer.AddBalance(bucket.StakedAmount); err != nil { | ||
| return nil, nil, errors.Wrapf(err, "failed to add balance %s", bucket.StakedAmount) | ||
| } | ||
| // put updated withdrawer's account state to trie | ||
| actionCtx := protocol.MustGetActionCtx(ctx) | ||
| if err := accountutil.StoreAccount(csm.SM(), actionCtx.Caller, withdrawer); err != nil { | ||
| return nil, nil, errors.Wrapf(err, "failed to store account %s", actionCtx.Caller.String()) | ||
| } | ||
| // create receipt log | ||
| actLog := newReceiptLog(p.addr.String(), handleCandidateActivate, protocol.MustGetFeatureCtx(ctx).NewStakingReceiptFormat) | ||
| actLog.AddTopics(byteutil.Uint64ToBytesBigEndian(bucket.Index), bucket.Candidate.Bytes()) | ||
| actLog.AddAddress(actionCtx.Caller) | ||
| actLog.SetData(bucket.StakedAmount.Bytes()) | ||
| return actLog, &action.TransactionLog{ | ||
| Type: iotextypes.TransactionLogType_WITHDRAW_BUCKET, | ||
| Amount: bucket.StakedAmount, | ||
| Sender: address.StakingBucketPoolAddr, | ||
| Recipient: actionCtx.Caller.String(), | ||
| }, nil | ||
| } | ||
|
|
||
| func (p *Protocol) createNFTBucket(ctx context.Context, act *action.MigrateStake, amount, duration *big.Int, cand *Candidate, sm protocol.StateManager) (*action.Receipt, error) { | ||
| ptl, ok := protocol.MustGetRegistry(ctx).Find(executionProtocolID) | ||
| if !ok { | ||
| return nil, errors.New("execution protocol is not registered") | ||
| } | ||
| exctPtl, ok := ptl.(executionProtocol) | ||
| contractAddress := p.config.MigrateContractAddress | ||
| data, err := StakingContractABI.Pack("stake0", duration, cand.GetIdentifier()) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to pack data for contract call") | ||
| } | ||
| exeAct, err := action.NewExecution( | ||
| contractAddress, | ||
| act.Nonce(), | ||
| amount, | ||
| act.GasLimit(), | ||
| act.GasPrice(), | ||
| data, | ||
| ) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to create execution action") | ||
| } | ||
| excReceipt, err := exctPtl.Handle(ctx, exeAct, sm) | ||
| if err != nil { | ||
| return nil, errors.Wrap(err, "failed to handle execution action") | ||
| } | ||
| return excReceipt, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.