diff --git a/Tests/Kernel/TestKernelPledge.cpp b/Tests/Kernel/TestKernelPledge.cpp index 02f9ce6f9c..3a90450c2e 100644 --- a/Tests/Kernel/TestKernelPledge.cpp +++ b/Tests/Kernel/TestKernelPledge.cpp @@ -4,7 +4,9 @@ * SPDX-License-Identifier: BSD-2-Clause */ +#include #include +#include #include TEST_CASE(test_nonexistent_pledge) @@ -14,6 +16,31 @@ TEST_CASE(test_nonexistent_pledge) FAIL("Pledging on existent promises should fail."); } +TEST_CASE(test_pledge_argument_validation) +{ + const auto long_argument = String::repeated('a', 2048); + + auto res = pledge(long_argument.characters(), "stdio"); + EXPECT_EQ(res, -1); + EXPECT_EQ(errno, E2BIG); + + res = pledge("stdio", long_argument.characters()); + EXPECT_EQ(res, -1); + EXPECT_EQ(errno, E2BIG); + + res = pledge(long_argument.characters(), long_argument.characters()); + EXPECT_EQ(res, -1); + EXPECT_EQ(errno, E2BIG); + + res = pledge("fake", "stdio"); + EXPECT_EQ(res, -1); + EXPECT_EQ(errno, EINVAL); + + res = pledge("stdio", "fake"); + EXPECT_EQ(res, -1); + EXPECT_EQ(errno, EINVAL); +} + TEST_CASE(test_pledge_failures) { auto res = pledge("stdio unix rpath", "stdio");