mirror of
https://github.com/EmberEmu/Ember
synced 2026-08-14 22:32:05 -04:00
109 lines
3 KiB
C++
109 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2016 - 2024 Ember
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#include <shared/IPBanCache.h>
|
|
#include <gtest/gtest.h>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
using namespace ember;
|
|
|
|
class IPBanTest : public ::testing::Test {
|
|
public:
|
|
virtual void SetUp() {
|
|
std::vector<IPEntry> entries {
|
|
{ "2001:db8::", 64 },
|
|
{ "198.51.106.51", 8 },
|
|
{ "172.16.125.134", 16 },
|
|
{ "169.254.26.21", 24 },
|
|
{ "203.62.113.82", 31 },
|
|
{ "192.88.99.62", 32 }
|
|
};
|
|
|
|
bans = std::make_unique<IPBanCache>(entries);
|
|
}
|
|
|
|
virtual void TearDown() {}
|
|
|
|
std::unique_ptr<IPBanCache> bans;
|
|
};
|
|
|
|
|
|
TEST_F(IPBanTest, Mask32) {
|
|
EXPECT_FALSE(bans->is_banned("192.88.99.61"))
|
|
<< "Banned below range";
|
|
EXPECT_TRUE(bans->is_banned("192.88.99.62"));
|
|
EXPECT_FALSE(bans->is_banned("192.88.99.63"))
|
|
<< "Banned above range";
|
|
}
|
|
|
|
TEST_F(IPBanTest, Mask31) {
|
|
EXPECT_FALSE(bans->is_banned("203.62.113.81"))
|
|
<< "Banned below range";
|
|
EXPECT_TRUE(bans->is_banned("203.62.113.82"));
|
|
EXPECT_TRUE(bans->is_banned("203.62.113.83"));
|
|
EXPECT_FALSE(bans->is_banned("203.62.113.84"))
|
|
<< "Banned above range";
|
|
}
|
|
|
|
TEST_F(IPBanTest, Mask24) {
|
|
EXPECT_FALSE(bans->is_banned("169.254.25.255"))
|
|
<< "Banned below range";
|
|
EXPECT_TRUE(bans->is_banned("169.254.26.0"));
|
|
EXPECT_TRUE(bans->is_banned("169.254.26.21"));
|
|
EXPECT_TRUE(bans->is_banned("169.254.26.3"));
|
|
EXPECT_TRUE(bans->is_banned("169.254.26.128"));
|
|
EXPECT_TRUE(bans->is_banned("169.254.26.255"));
|
|
EXPECT_FALSE(bans->is_banned("169.254.27.0"))
|
|
<< "Banned above range";
|
|
}
|
|
|
|
TEST_F(IPBanTest, Mask16) {
|
|
EXPECT_FALSE(bans->is_banned("172.15.255.255"))
|
|
<< "Banned below range";
|
|
EXPECT_TRUE(bans->is_banned("172.16.0.0"));
|
|
EXPECT_TRUE(bans->is_banned("172.16.125.134"));
|
|
EXPECT_TRUE(bans->is_banned("172.16.117.92"));
|
|
EXPECT_TRUE(bans->is_banned("172.16.4.92"));
|
|
EXPECT_TRUE(bans->is_banned("172.16.255.255"));
|
|
EXPECT_FALSE(bans->is_banned("172.17.0.0"))
|
|
<< "Banned above range";
|
|
}
|
|
|
|
TEST_F(IPBanTest, Mask8) {
|
|
EXPECT_FALSE(bans->is_banned("197.255.255.255"))
|
|
<< "Banned below range";
|
|
EXPECT_TRUE(bans->is_banned("198.0.0.0"));
|
|
EXPECT_TRUE(bans->is_banned("198.51.106.51"));
|
|
EXPECT_TRUE(bans->is_banned("198.51.106.162"));
|
|
EXPECT_TRUE(bans->is_banned("198.51.42.162"));
|
|
EXPECT_TRUE(bans->is_banned("198.43.42.162"));
|
|
EXPECT_TRUE(bans->is_banned("198.255.255.255"));
|
|
EXPECT_FALSE(bans->is_banned("199.0.0.0"))
|
|
<< "Banned above range";
|
|
}
|
|
|
|
TEST_F(IPBanTest, IPv6NotBannedLocal) {
|
|
EXPECT_FALSE(bans->is_banned("::1"));
|
|
}
|
|
|
|
TEST_F(IPBanTest, IPv6BannedBegin) {
|
|
EXPECT_TRUE(bans->is_banned("2001:db8::"));
|
|
}
|
|
|
|
TEST_F(IPBanTest, IPv6BannedEnd) {
|
|
EXPECT_TRUE(bans->is_banned("2001:0db8:0000:0000:ffff:ffff:ffff:ffff"));
|
|
}
|
|
|
|
TEST_F(IPBanTest, IPv6BannedInRange) {
|
|
EXPECT_TRUE(bans->is_banned("2001:0db8:0000:0000:ffff:ffff:ffff:fffe"));
|
|
}
|
|
|
|
TEST_F(IPBanTest, IPv6NotBanned) {
|
|
EXPECT_FALSE(bans->is_banned("2001:0db9::"));
|
|
}
|