47 lines
1.4 KiB
C++
47 lines
1.4 KiB
C++
|
|
#include "../src/search.h"
|
|
|
|
std::map<int, std::string> prompt;
|
|
|
|
void test(xdb::search_t& s, const std::string& ip, const std::string& region) {
|
|
if (s.search(ip) != region)
|
|
xdb::log_exit("测试失败, ip " + ip + ", region " + region);
|
|
}
|
|
|
|
void test_ipv4(int policy) {
|
|
std::cout << "测试 IPv4 " << prompt[policy];
|
|
|
|
xdb::search_t s("../../data/ip2region_v4.xdb", xdb::ipv4, policy);
|
|
test(s, "0.0.0.0", "Reserved|Reserved|Reserved|0|0");
|
|
test(s, "1.2.3.4", "Australia|Queensland|Brisbane|0|AU");
|
|
|
|
std::cout << " 成功" << std::endl;
|
|
}
|
|
|
|
void test_ipv6(int policy) {
|
|
std::cout << "测试 IPv6 " << prompt[policy];
|
|
|
|
xdb::search_t s("../../data/ip2region_v6.xdb", xdb::ipv6, policy);
|
|
test(s, "::1", "");
|
|
test(s, "2001:200:124::", "Japan|Tokyo|Asagaya-minami|WIDE Project|JP");
|
|
test(s, "2001:200:124::", "Japan|Tokyo|Asagaya-minami|WIDE Project|JP");
|
|
test(s, "240e:3b7:3273:51d0:cd38:8ae1:e3c0:b708", "中国|广东省|深圳市|电信|CN");
|
|
|
|
std::cout << " 成功" << std::endl;
|
|
}
|
|
|
|
int main() {
|
|
prompt[xdb::policy_file] = " 不缓存:";
|
|
prompt[xdb::policy_vector] = "部分缓存:";
|
|
prompt[xdb::policy_content] = "全部缓存:";
|
|
|
|
test_ipv4(xdb::policy_file);
|
|
test_ipv4(xdb::policy_vector);
|
|
test_ipv4(xdb::policy_content);
|
|
|
|
test_ipv6(xdb::policy_file);
|
|
test_ipv6(xdb::policy_vector);
|
|
test_ipv6(xdb::policy_content);
|
|
return 0;
|
|
}
|