142 lines
4.6 KiB
C++
142 lines
4.6 KiB
C++
#include "handler.hpp"
|
|
#include <gtest/gtest.h>
|
|
|
|
// Test fixture for process function tests
|
|
class ProcessTest : public ::testing::Test {
|
|
protected:
|
|
void SetUp() override {}
|
|
void TearDown() override {}
|
|
};
|
|
|
|
// Valid username tests
|
|
TEST_F(ProcessTest, ValidUsernameSimple) {
|
|
std::string result = process("john");
|
|
EXPECT_EQ(result, "john");
|
|
}
|
|
|
|
TEST_F(ProcessTest, ValidUsernameWithNumbers) {
|
|
std::string result = process("user123");
|
|
EXPECT_EQ(result, "user123");
|
|
}
|
|
|
|
TEST_F(ProcessTest, ValidUsernameWithUnderscore) {
|
|
std::string result = process("user_name");
|
|
EXPECT_EQ(result, "user_name");
|
|
}
|
|
|
|
TEST_F(ProcessTest, ValidUsernameWithHyphen) {
|
|
std::string result = process("user-name");
|
|
EXPECT_EQ(result, "user-name");
|
|
}
|
|
|
|
TEST_F(ProcessTest, ValidUsernameEmptyString) {
|
|
std::string result = process("");
|
|
EXPECT_EQ(result, "");
|
|
}
|
|
|
|
// Directory traversal tests
|
|
TEST_F(ProcessTest, DirectoryTraversalBasicDotDotSlash) {
|
|
std::string result = process("user../file");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalBasicDotDotBackslash) {
|
|
std::string result = process("user..\\file");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalURLEncodedLowercase2e2e2f) {
|
|
std::string result = process("user%2e%2e%2ffile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalURLEncodedLowercase2e2e5c) {
|
|
std::string result = process("user%2e%2e%5cfile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalURLEncodedUppercase2E2E2F) {
|
|
std::string result = process("user%2E%2E%2Ffile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalURLEncodedUppercase2E2E5C) {
|
|
std::string result = process("user%2E%2E%5Cfile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalMixedDotDot2f) {
|
|
std::string result = process("user..%2ffile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalMixedDotDot5c) {
|
|
std::string result = process("user..%5cfile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalMixedDotDot2F) {
|
|
std::string result = process("user..%2Ffile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, DirectoryTraversalMixedDotDot5C) {
|
|
std::string result = process("user..%5Cfile");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Directory traversal detected") != std::string::npos);
|
|
}
|
|
|
|
// Path detection tests
|
|
TEST_F(ProcessTest, PathDetectionForwardSlash) {
|
|
std::string result = process("user/name");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Path detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, PathDetectionForwardSlashAtStart) {
|
|
std::string result = process("/username");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Path detected") != std::string::npos);
|
|
}
|
|
|
|
TEST_F(ProcessTest, PathDetectionForwardSlashAtEnd) {
|
|
std::string result = process("username/");
|
|
EXPECT_TRUE(result.find("InvalidInput:") == 0);
|
|
EXPECT_TRUE(result.find("Path detected") != std::string::npos);
|
|
}
|
|
|
|
// Edge cases
|
|
TEST_F(ProcessTest, SingleDot) {
|
|
std::string result = process(".");
|
|
EXPECT_EQ(result, ".");
|
|
}
|
|
|
|
TEST_F(ProcessTest, DoubleDotWithoutSlash) {
|
|
std::string result = process("..");
|
|
EXPECT_EQ(result, "..");
|
|
}
|
|
|
|
TEST_F(ProcessTest, ContainsDotButNotTraversal) {
|
|
std::string result = process("user.name");
|
|
EXPECT_EQ(result, "user.name");
|
|
}
|
|
|
|
TEST_F(ProcessTest, BackslashWithoutDots) {
|
|
std::string result = process("user\\name");
|
|
EXPECT_EQ(result, "user\\name");
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|