mirror of
https://github.com/shadowsocks/shadowsocks-libev.git
synced 2026-10-21 06:23:52 +00:00
Max Lv
678a891ef9
Convert acl, acl_literal, aead2022, base64, buffer, cache (legacy reference), crypto, ipset, ipset_reference, json, netutils, ppbloom, process, ref, ssurl and utils tests to test_helpers.hpp. ctest names are unchanged; every assertion is kept. Build the ported sources with the strict C++ warning set.
78 lines
2.1 KiB
C++
78 lines
2.1 KiB
C++
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
// IP prefix set (src/ipset.cpp) checked against an exhaustive bitmap oracle.
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
|
|
#include "ipset.h"
|
|
#include "test_helpers.hpp"
|
|
|
|
extern "C" {
|
|
int verbose = 0;
|
|
}
|
|
|
|
namespace {
|
|
|
|
uint32_t g_state = 42;
|
|
|
|
uint32_t next_random()
|
|
{
|
|
g_state = g_state * 1664525u + 1013904223u;
|
|
return g_state;
|
|
}
|
|
|
|
void check_family(unsigned width)
|
|
{
|
|
ss_ipset set{};
|
|
uint8_t address[16] = {};
|
|
bool oracle[256] = {};
|
|
const unsigned last = width / 8 - 1;
|
|
// Independent exhaustive bitmap oracle over a /24 or /120. Prefix writes
|
|
// can cover the whole space; host deletion must punch holes in them.
|
|
for (unsigned round = 0; round < 2000; round++) {
|
|
const unsigned host = next_random() >> 24;
|
|
const unsigned bits = next_random() % 9;
|
|
const bool present = (next_random() >> 31) != 0;
|
|
const unsigned mask = bits == 0 ? 0 : (255u << (8 - bits)) & 255u;
|
|
address[last] = static_cast<uint8_t>(host);
|
|
ss_ipset_assign(&set, address, width, width - 8 + bits, present);
|
|
for (unsigned i = 0; i < 256; i++) {
|
|
if ((i & mask) == (host & mask)) {
|
|
oracle[i] = present;
|
|
}
|
|
address[last] = static_cast<uint8_t>(i);
|
|
SS_CHECK(ss_ipset_contains(&set, address, width) == oracle[i]);
|
|
}
|
|
}
|
|
ss_ipset_assign(&set, address, width, 0, true);
|
|
std::memset(address, 255, sizeof(address));
|
|
SS_CHECK(ss_ipset_contains(&set, address, width));
|
|
ss_ipset_assign(&set, address, width, width, false);
|
|
SS_CHECK(!ss_ipset_contains(&set, address, width));
|
|
address[last]--;
|
|
SS_CHECK(ss_ipset_contains(&set, address, width));
|
|
ss_ipset_assign(&set, address, width, width + 1, false);
|
|
SS_CHECK(ss_ipset_contains(&set, address, width));
|
|
ss_ipset_clear(&set);
|
|
ss_ipset_clear(&set);
|
|
SS_CHECK(!ss_ipset_contains(&set, address, width));
|
|
}
|
|
|
|
void test_ipv4()
|
|
{
|
|
check_family(32);
|
|
}
|
|
|
|
void test_ipv6()
|
|
{
|
|
check_family(128);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
int main()
|
|
{
|
|
SS_RUN(test_ipv4);
|
|
SS_RUN(test_ipv6);
|
|
return 0;
|
|
}
|