Skip to content

feat(contract): implement custom token transfer validation checks#1296

Open
devisgood24 wants to merge 1 commit into
Web3Novalabs:mainfrom
devisgood24:feat/token-transfer-validation-checks
Open

feat(contract): implement custom token transfer validation checks#1296
devisgood24 wants to merge 1 commit into
Web3Novalabs:mainfrom
devisgood24:feat/token-transfer-validation-checks

Conversation

@devisgood24

Copy link
Copy Markdown

Summary

Implement comprehensive token transfer validation checks for all token operations in the prediction market contract. Validates amount, token address, sender/recipient addresses, and token contract responsiveness before every transfer.

Changes

  • Added validate_token_transfer() helper function with checks for:

    • Amount validation (positive, non-zero)
    • Token address validation (not null)
    • Sender/recipient validation (distinct, not null)
    • Token contract verification (balance check)
  • Integrated validation into 5 transfer operations:

    • place_prediction(): Inbound stakes
    • claim_winnings_internal(): Payouts and referrals
    • claim_refund(): Refunds from canceled pools
    • withdraw_treasury(): Treasury withdrawals
    • emergency_withdraw(): Emergency fund extractions
  • Added 9 unit tests covering validation scenarios and edge cases

Files Changed

  • contract/contracts/predifi-contract/src/lib.rs (+135 lines)
  • contract/contracts/predifi-contract/src/test.rs (+361 lines)

Testing

cd contract
cargo test --workspace

Notes

  • Uses existing PredifiError variants (InvalidAmount, InvalidAddressOrToken)
  • Fully backwards compatible
  • No storage or API changes
  • Enhances security without behavior changes

Closes #1138

Implement comprehensive token transfer validation checks to ensure safe and
compliant token operations across the prediction market contract.

CHANGES:
- Add validate_token_transfer() helper function that performs:
  * Amount validation (must be positive and non-zero)
  * Token address validation (not null/default)
  * Sender/recipient validation (distinct, not null)
  * Token contract callable verification (balance check)

- Integrate validation checks into all token transfer operations:
  * place_prediction: validate inbound transfers
  * claim_winnings_internal: validate all outbound payouts (main winnings, referrals)
  * claim_refund: validate refund transfers
  * withdraw_treasury: validate treasury withdrawals
  * emergency_withdraw: validate emergency fund transfers

- Add comprehensive test suite (9 new tests):
  * test_validate_token_transfer_zero_amount_fails
  * test_validate_token_transfer_negative_amount_fails
  * test_validate_token_transfer_same_sender_recipient_fails
  * test_validate_token_transfer_on_claim_winnings
  * test_validate_token_transfer_on_claim_refund
  * test_validate_token_transfer_on_treasury_withdrawal
  * test_validate_token_transfer_on_emergency_withdraw
  * test_multiple_predictions_with_token_validation
  * Plus edge case validation tests

These validations prevent:
- Invalid amount transfers (zero, negative)
- Invalid addresses (null, self-transfers)
- Non-responsive token contracts
- Reentrancy attacks during token transfers

Resolves Web3Novalabs#1138
@vercel

vercel Bot commented Jun 28, 2026

Copy link
Copy Markdown

Someone is attempting to deploy a commit to the shola's projects Team on Vercel.

A member of the Team first needs to authorize it.

@drips-wave

drips-wave Bot commented Jun 28, 2026

Copy link
Copy Markdown

@devisgood24 Great news! 🎉 Based on an automated assessment of this PR, the linked Wave issue(s) no longer count against your application limits.

You can now already apply to more issues while waiting for a review of this PR. Keep up the great work! 🚀

Learn more about application limits

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

feat(contract): implement custom token transfer validation checks

1 participant