Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AbstractRewardMine._handleStakePadding calls totalDeclaredReward and this way balanceOf function twice #128

Open
code423n4 opened this issue Nov 29, 2021 · 1 comment
Labels
bug Something isn't working G (Gas Optimization) sponsor confirmed Sponsor agrees this is a problem and intends to fix it (OK to use w/ "disagree with severity")

Comments

@code423n4
Copy link
Contributor

Handle

hyh

Vulnerability details

Impact

Gas is overspent on access and function calls.

Proof of Concept

totalDeclaredReward is called by _handleStakePadding twice:
https://github.com/code-423n4/2021-11-malt/blob/main/src/contracts/AbstractRewardMine.sol#L180

While totalDeclaredReward does expensive balanceOf call:
https://github.com/code-423n4/2021-11-malt/blob/main/src/contracts/AbstractRewardMine.sol#L97

Recommended Mitigation Steps

It is viable to at least remove its double usage:

Now:

uint256 totalRewardedWithStakePadding = totalDeclaredReward().add(totalStakePadding());
...
uint256 newStakePadding = bondedTotal == 0 ?
	totalDeclaredReward() == 0 ? amount.mul(INITIAL_STAKE_SHARE_MULTIPLE) : 0 :
	totalRewardedWithStakePadding.mul(amount).div(bondedTotal);

To be:

uint256 declaredRewardTotal =  rewardToken.balanceOf(address(this));
uint256 totalRewardedWithStakePadding = declaredRewardTotal.add(totalStakePadding());
...
uint256 newStakePadding = bondedTotal == 0 ?
	declaredRewardTotal == 0 ? amount.mul(INITIAL_STAKE_SHARE_MULTIPLE) : 0 : totalRewardedWithStakePadding.mul(amount).div(bondedTotal);
@code423n4 code423n4 added bug Something isn't working G (Gas Optimization) labels Nov 29, 2021
code423n4 added a commit that referenced this issue Nov 29, 2021
@0xScotch 0xScotch added the sponsor confirmed Sponsor agrees this is a problem and intends to fix it (OK to use w/ "disagree with severity") label Dec 10, 2021
@GalloDaSballo
Copy link
Collaborator

Agree, whenever you are reading from storage more than once, you want to use a supporting memory variable to save gas

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working G (Gas Optimization) sponsor confirmed Sponsor agrees this is a problem and intends to fix it (OK to use w/ "disagree with severity")
Projects
None yet
Development

No branches or pull requests

3 participants