Santa's List

AI First Flight #3
Beginner FriendlyFoundry
EXP
View results
Submission Details
Severity: high
Valid

Anyone is able to call `checkList` function in SantasList contract; preventing any address from becoming NICE OR EXTRA_NICE

Root + Impact

Description

  • Only Santa is allowed to call the `SantasList:checkList` function and mark an address as NICE, EXTRA_NICE OR NAUGHTY

  • However, as per current code anyone can call the `SantasList:checkList` function and prevent any address from becoming NICE OR EXTRA_NICE

// Root cause in the codebase with @> marks to highlight the relevant section
@> function checkList(address person, Status status) external {
s_theListCheckedOnce[person] = status;
emit CheckedOnce(person, status);
}

Risk

Likelihood:

Likelihood is HIGH

This can occur when `SantasList:checkList` function is called by an address other than Santa's address. This can be considered as an access control vulnerability as according to documentation, only Santa is allowed to call the function.

Impact:

Impact is HIGH as this results in broken mechanism of the checkList system. Any user could be declared NAUGHTY for the first time, preventing present collection by the address even if SANTA considered them NICE or EXTRA_NICE.

A malicious actor can mark any address as NAUGHTY preventing that address from being NICE OR EXTRA_NICE.
A malicious actor can mark an already marked NICE or EXTRA_NICE address again as NAUGHTY or UNKNOWN.

Proof of Concept

Place the following test to the `SantasListTest.t.sol` file.

function testDosAttack() external {
vm.startPrank(makeAddr("attacker"));
// any user can call checLkist on any address and mark the status to NAUGHTY
// an attacker could front run Santa before the second check
santasList.checkList(makeAddr("user"), SantasList.Status.NAUGHTY);
vm.stopPrank();
vm.startPrank(santa);
vm.expectRevert();
// Santa is unable to call checkTwice for the user
santasList.checkTwice(makeAddr("user"), SantasList.Status.NICE);
vm.stopPrank();
}

Recommended Mitigation

I suggest adding onlySanta modifier to the `SantasList::checkList` function will resolve the issue. This will ensure that only Santa is allowed to call the function and no malicious actor can call the function. Thus, only Santa will be able to mark an address as NICE, EXTRA_NICE, NAUGHTY or UNKNOWN.

- function checkList(address person, Status status) external{
+ function checkList(address person, Status status) external onlySanta{
s_theListCheckedOnce[person] = status;
emit CheckedOnce(person, status);
}
Updates

Lead Judging Commences

ai-first-flight-judge Lead Judge 1 day ago
Submission Judgement Published
Validated
Assigned finding tags:

[H-01] Anyone is able to call `checkList` function in SantasList contract and prevent any address from becoming `NICE` or `EXTRA_NICE` and collect present.

## Description With the current design of the protocol, anyone is able to call `checkList` function in SantasList contract, while documentation says only Santa should be able to call it. This can be considered as an access control vulnerability, because not only santa is allowed to make the first check. ## Vulnerability Details An attacker could simply call the external `checkList` function, passing as parameter the address of someone else and the enum Status `NAUGHTY`(or `NOT_CHECKED_TWICE`, which should actually be `UNKNOWN` given documentation). By doing that, Santa will not be able to execute `checkTwice` function correctly for `NICE` and `EXTRA_NICE` people. Indeed, if Santa first checked a user and assigned the status `NICE` or `EXTRA_NICE`, anyone is able to call `checkList` function again, and by doing so modify the status. This could result in Santa unable to execute the second check. Moreover, any malicious actor could check the mempool and front run Santa just before calling `checkTwice` function to check users. This would result in a major denial of service issue. ## Impact The impact of this vulnerability is HIGH as it results in a broken mechanism of the check list system. Any user could be declared `NAUGHTY` for the first check at any time, preventing present collecting by users although Santa considered the user as `NICE` or `EXTRA_NICE`. Santa could still call `checkList` function again to reassigned the status to `NICE` or `EXTRA_NICE` before calling `checkTwice` function, but any malicious actor could front run the call to `checkTwice` function. In this scenario, it would be impossible for Santa to actually double check a `NICE` or `EXTRA_NICE` user. ## Proof of Concept Just copy paste this test in SantasListTest contract : ``` function testDosAttack() external { vm.startPrank(makeAddr("attacker")); // any user can checList any address and assigned status to naughty // an attacker could front run Santa before the second check santasList.checkList(makeAddr("user"), SantasList.Status.NAUGHTY); vm.stopPrank(); vm.startPrank(santa); vm.expectRevert(); // Santa is unable to check twice the user santasList.checkTwice(makeAddr("user"), SantasList.Status.NICE); vm.stopPrank(); } ``` ## Recommendations I suggest to add the `onlySanta` modifier to `checkList` function. This will ensure the first check can only be done by Santa, and prevent DOS attack on the contract. With this modifier, specification will be respected : "In this contract Only Santa to take the following actions: - checkList: A function that changes an address to a new Status of NICE, EXTRA_NICE, NAUGHTY, or UNKNOWN on the original s_theListCheckedOnce list." The following code will resolve this access control issue, simply by adding `onlySanta` modifier: ``` function checkList(address person, Status status) external onlySanta { s_theListCheckedOnce[person] = status; emit CheckedOnce(person, status); } ``` No malicious actor is now able to front run Santa before `checkTwice` function call. The following tests shows that doing the first check for another user is impossible after adding `onlySanta` modifier: ``` function testDosResolved() external { vm.startPrank(makeAddr("attacker")); // checklist function call will revert if a user tries to execute the first check for another user vm.expectRevert(SantasList.SantasList__NotSanta.selector); santasList.checkList(makeAddr("user"), SantasList.Status.NAUGHTY); vm.stopPrank(); } ```

Support

FAQs

Can't find an answer? Chat with us on Discord, Twitter or Linkedin.

Give us feedback!