|
@@ -383,6 +383,17 @@ test_geoip_with_pt(void *arg)
|
|
|
#undef SET_TEST_IPV6
|
|
|
#undef CHECK_COUNTRY
|
|
|
|
|
|
+static const char GEOIP_CONTENT[] =
|
|
|
+ "134445936,134445939,MP\n"
|
|
|
+ "134445940,134447103,GU\n"
|
|
|
+ "134447104,134738943,US\n"
|
|
|
+ "134738944,134739199,CA\n"
|
|
|
+ "134739200,135192575,US\n"
|
|
|
+ "135192576,135200767,MX\n"
|
|
|
+ "135200768,135430143,US\n"
|
|
|
+ "135430144,135430399,CA\n"
|
|
|
+ "135430400,135432191,US\n";
|
|
|
+
|
|
|
static void
|
|
|
test_geoip_load_file(void *arg)
|
|
|
{
|
|
@@ -403,16 +414,18 @@ test_geoip_load_file(void *arg)
|
|
|
tt_str_op("0000000000000000000000000000000000000000", OP_EQ,
|
|
|
geoip_db_digest(AF_INET));
|
|
|
|
|
|
- const char FNAME[] = SRCDIR "/src/config/geoip";
|
|
|
- int rv = geoip_load_file(AF_INET, FNAME);
|
|
|
+ const char *fname = get_fname("geoip");
|
|
|
+ tt_int_op(0, OP_EQ, write_str_to_file(fname, GEOIP_CONTENT, 0));
|
|
|
+
|
|
|
+ int rv = geoip_load_file(AF_INET, fname);
|
|
|
if (rv != 0) {
|
|
|
- TT_GRIPE(("Unable to load geoip from %s", escaped(FNAME)));
|
|
|
+ TT_GRIPE(("Unable to load geoip from %s", escaped(fname)));
|
|
|
}
|
|
|
tt_int_op(0, OP_EQ, rv);
|
|
|
|
|
|
|
|
|
- * fewer than 50 countries in the world. */
|
|
|
- tt_int_op(geoip_get_n_countries(), OP_GE, 50);
|
|
|
+ * fewer than 5 countries in our test above. */
|
|
|
+ tt_int_op(geoip_get_n_countries(), OP_GE, 5);
|
|
|
|
|
|
|
|
|
int country = geoip_get_country_by_ipv4(0x08080808);
|
|
@@ -425,7 +438,7 @@ test_geoip_load_file(void *arg)
|
|
|
geoip_db_digest(AF_INET));
|
|
|
|
|
|
|
|
|
- contents = read_file_to_str(FNAME, RFTS_BIN, NULL);
|
|
|
+ contents = read_file_to_str(fname, RFTS_BIN, NULL);
|
|
|
uint8_t d[DIGEST_LEN];
|
|
|
crypto_digest((char*)d, contents, strlen(contents));
|
|
|
dhex = tor_strdup(hex_str((char*)d, DIGEST_LEN));
|
|
@@ -461,12 +474,30 @@ test_geoip6_load_file(void *arg)
|
|
|
tt_int_op(-1, OP_EQ, geoip_get_country_by_ipv6(&iaddr6));
|
|
|
|
|
|
|
|
|
- const char FNAME6[] = SRCDIR "/src/config/geoip6";
|
|
|
- tt_int_op(0, OP_EQ, geoip_load_file(AF_INET6, FNAME6));
|
|
|
+ const char *fname6 = get_fname("geoip6");
|
|
|
+ const char CONTENT[] =
|
|
|
+ "2001:4830:6010::,2001:4830:601f:ffff:ffff:ffff:ffff:ffff,GB\n"
|
|
|
+ "2001:4830:6020::,2001:4830:ffff:ffff:ffff:ffff:ffff:ffff,US\n"
|
|
|
+ "2001:4838::,2001:4838:ffff:ffff:ffff:ffff:ffff:ffff,US\n"
|
|
|
+ "2001:4840::,2001:4840:ffff:ffff:ffff:ffff:ffff:ffff,XY\n"
|
|
|
+ "2001:4848::,2001:4848:ffff:ffff:ffff:ffff:ffff:ffff,ZD\n"
|
|
|
+ "2001:4850::,2001:4850:ffff:ffff:ffff:ffff:ffff:ffff,RO\n"
|
|
|
+ "2001:4858::,2001:4858:ffff:ffff:ffff:ffff:ffff:ffff,TC\n"
|
|
|
+ "2001:4860::,2001:4860:ffff:ffff:ffff:ffff:ffff:ffff,US\n"
|
|
|
+ "2001:4868::,2001:4868:ffff:ffff:ffff:ffff:ffff:ffff,US\n"
|
|
|
+ "2001:4870::,2001:4871:ffff:ffff:ffff:ffff:ffff:ffff,NB\n"
|
|
|
+ "2001:4878::,2001:4878:128:ffff:ffff:ffff:ffff:ffff,US\n"
|
|
|
+ "2001:4878:129::,2001:4878:129:ffff:ffff:ffff:ffff:ffff,CR\n"
|
|
|
+ "2001:4878:12a::,2001:4878:203:ffff:ffff:ffff:ffff:ffff,US\n"
|
|
|
+ "2001:4878:204::,2001:4878:204:ffff:ffff:ffff:ffff:ffff,DE\n"
|
|
|
+ "2001:4878:205::,2001:4878:214:ffff:ffff:ffff:ffff:ffff,US\n";
|
|
|
+ tt_int_op(0, OP_EQ, write_str_to_file(fname6, CONTENT, 0));
|
|
|
+
|
|
|
+ tt_int_op(0, OP_EQ, geoip_load_file(AF_INET6, fname6));
|
|
|
|
|
|
|
|
|
- * fewer than 50 countries in the world. */
|
|
|
- tt_int_op(geoip_get_n_countries(), OP_GE, 50);
|
|
|
+ * fewer than 5 countries in our test data above. */
|
|
|
+ tt_int_op(geoip_get_n_countries(), OP_GE, 5);
|
|
|
|
|
|
|
|
|
const char *caddr6 = "2001:4860:4860::8888";
|
|
@@ -482,7 +513,7 @@ test_geoip6_load_file(void *arg)
|
|
|
geoip_db_digest(AF_INET6));
|
|
|
|
|
|
|
|
|
- contents = read_file_to_str(FNAME6, RFTS_BIN, NULL);
|
|
|
+ contents = read_file_to_str(fname6, RFTS_BIN, NULL);
|
|
|
uint8_t d[DIGEST_LEN];
|
|
|
crypto_digest((char*)d, contents, strlen(contents));
|
|
|
dhex = tor_strdup(hex_str((char*)d, DIGEST_LEN));
|
|
@@ -507,21 +538,27 @@ test_geoip_load_2nd_file(void *arg)
|
|
|
{
|
|
|
(void)arg;
|
|
|
|
|
|
+ char *fname_geoip = tor_strdup(get_fname("geoip_data"));
|
|
|
+ char *fname_empty = tor_strdup(get_fname("geoip_empty"));
|
|
|
+
|
|
|
+ tt_int_op(0, OP_EQ, write_str_to_file(fname_geoip, GEOIP_CONTENT, 0));
|
|
|
+ tt_int_op(0, OP_EQ, write_str_to_file(fname_empty, "\n", 0));
|
|
|
+
|
|
|
|
|
|
- const char FNAME[] = SRCDIR "/src/config/geoip";
|
|
|
- tt_int_op(0, OP_EQ, geoip_load_file(AF_INET, FNAME));
|
|
|
+ tt_int_op(0, OP_EQ, geoip_load_file(AF_INET, fname_geoip));
|
|
|
|
|
|
|
|
|
|
|
|
- const char FNAME2[] = SRCDIR "/src/test/geoip_dummy";
|
|
|
- tt_int_op(0, OP_EQ, geoip_load_file(AF_INET, FNAME2));
|
|
|
+ tt_int_op(0, OP_EQ, geoip_load_file(AF_INET, fname_empty));
|
|
|
|
|
|
|
|
|
|
|
|
int country = geoip_get_country_by_ipv4(0x08080808);
|
|
|
tt_int_op(country, OP_EQ, 0);
|
|
|
|
|
|
- done: ;
|
|
|
+ done:
|
|
|
+ tor_free(fname_geoip);
|
|
|
+ tor_free(fname_empty);
|
|
|
}
|
|
|
|
|
|
#define ENT(name) \
|
|
@@ -529,20 +566,12 @@ test_geoip_load_2nd_file(void *arg)
|
|
|
#define FORK(name) \
|
|
|
{ #name, test_ ## name , TT_FORK, NULL, NULL }
|
|
|
|
|
|
-#ifdef _WIN32
|
|
|
-#define SKIP_ON_WINDOWS TT_SKIP
|
|
|
-#else
|
|
|
-#define SKIP_ON_WINDOWS 0
|
|
|
-#endif
|
|
|
-
|
|
|
struct testcase_t geoip_tests[] = {
|
|
|
{ "geoip", test_geoip, TT_FORK, NULL, NULL },
|
|
|
{ "geoip_with_pt", test_geoip_with_pt, TT_FORK, NULL, NULL },
|
|
|
- { "load_file", test_geoip_load_file, TT_FORK|SKIP_ON_WINDOWS, NULL, NULL },
|
|
|
- { "load_file6", test_geoip6_load_file, TT_FORK | SKIP_ON_WINDOWS,
|
|
|
- NULL, NULL },
|
|
|
- { "load_2nd_file", test_geoip_load_2nd_file, TT_FORK | SKIP_ON_WINDOWS,
|
|
|
- NULL, NULL },
|
|
|
+ { "load_file", test_geoip_load_file, TT_FORK, NULL, NULL },
|
|
|
+ { "load_file6", test_geoip6_load_file, TT_FORK, NULL, NULL },
|
|
|
+ { "load_2nd_file", test_geoip_load_2nd_file, TT_FORK, NULL, NULL },
|
|
|
|
|
|
END_OF_TESTCASES
|
|
|
};
|