123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214 |
- #define ROUTERSET_PRIVATE
- #include "or.h"
- #include "geoip.h"
- #include "routerset.h"
- #include "routerparse.h"
- #include "policies.h"
- #include "nodelist.h"
- #include "test.h"
- #define NS_MODULE routerset
- #define NS_SUBMODULE routerset_new
- /*
- * Functional (blackbox) test to determine that each member of the routerset
- * is non-NULL
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *rs;
- (void)arg;
- rs = routerset_new();
- tt_ptr_op(rs, OP_NE, NULL);
- tt_ptr_op(rs->list, OP_NE, NULL);
- tt_ptr_op(rs->names, OP_NE, NULL);
- tt_ptr_op(rs->digests, OP_NE, NULL);
- tt_ptr_op(rs->policies, OP_NE, NULL);
- tt_ptr_op(rs->country_names, OP_NE, NULL);
- done:
- routerset_free(rs);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_get_countryname
- /*
- * Functional test to strip the braces from a "{xx}" country code string.
- */
- static void
- NS(test_main)(void *arg)
- {
- const char *input;
- char *name;
- (void)arg;
- /* strlen(c) < 4 */
- input = "xxx";
- name = routerset_get_countryname(input);
- tt_ptr_op(name, OP_EQ, NULL);
- tor_free(name);
- /* c[0] != '{' */
- input = "xxx}";
- name = routerset_get_countryname(input);
- tt_ptr_op(name, OP_EQ, NULL);
- tor_free(name);
- /* c[3] != '}' */
- input = "{xxx";
- name = routerset_get_countryname(input);
- tt_ptr_op(name, OP_EQ, NULL);
- tor_free(name);
- /* tor_strlower */
- input = "{XX}";
- name = routerset_get_countryname(input);
- tt_str_op(name, OP_EQ, "xx");
- tor_free(name);
- input = "{xx}";
- name = routerset_get_countryname(input);
- tt_str_op(name, OP_EQ, "xx");
- done:
- tor_free(name);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_refresh_counties, geoip_not_loaded)
- /*
- * Structural (whitebox) test for routerset_refresh_counties, when the GeoIP DB
- * is not loaded.
- */
- NS_DECL(int, geoip_is_loaded, (sa_family_t family));
- NS_DECL(int, geoip_get_n_countries, (void));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- (void)arg;
- NS_MOCK(geoip_is_loaded);
- NS_MOCK(geoip_get_n_countries);
- routerset_refresh_countries(set);
- tt_ptr_op(set->countries, OP_EQ, NULL);
- tt_int_op(set->n_countries, OP_EQ, 0);
- tt_int_op(CALLED(geoip_is_loaded), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_n_countries), OP_EQ, 0);
- done:
- NS_UNMOCK(geoip_is_loaded);
- NS_UNMOCK(geoip_get_n_countries);
- routerset_free(set);
- }
- static int
- NS(geoip_is_loaded)(sa_family_t family)
- {
- (void)family;
- CALLED(geoip_is_loaded)++;
- return 0;
- }
- static int
- NS(geoip_get_n_countries)(void)
- {
- CALLED(geoip_get_n_countries)++;
- return 0;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_refresh_counties, no_countries)
- /*
- * Structural test for routerset_refresh_counties, when there are no countries.
- */
- NS_DECL(int, geoip_is_loaded, (sa_family_t family));
- NS_DECL(int, geoip_get_n_countries, (void));
- NS_DECL(country_t, geoip_get_country, (const char *country));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- (void)arg;
- NS_MOCK(geoip_is_loaded);
- NS_MOCK(geoip_get_n_countries);
- NS_MOCK(geoip_get_country);
- routerset_refresh_countries(set);
- tt_ptr_op(set->countries, OP_NE, NULL);
- tt_int_op(set->n_countries, OP_EQ, 1);
- tt_int_op((unsigned int)(*set->countries), OP_EQ, 0);
- tt_int_op(CALLED(geoip_is_loaded), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_n_countries), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_country), OP_EQ, 0);
- done:
- NS_UNMOCK(geoip_is_loaded);
- NS_UNMOCK(geoip_get_n_countries);
- NS_UNMOCK(geoip_get_country);
- routerset_free(set);
- }
- static int
- NS(geoip_is_loaded)(sa_family_t family)
- {
- (void)family;
- CALLED(geoip_is_loaded)++;
- return 1;
- }
- static int
- NS(geoip_get_n_countries)(void)
- {
- CALLED(geoip_get_n_countries)++;
- return 1;
- }
- static country_t
- NS(geoip_get_country)(const char *countrycode)
- {
- (void)countrycode;
- CALLED(geoip_get_country)++;
- return 1;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_refresh_counties, one_valid_country)
- /*
- * Structural test for routerset_refresh_counties, with one valid country.
- */
- NS_DECL(int, geoip_is_loaded, (sa_family_t family));
- NS_DECL(int, geoip_get_n_countries, (void));
- NS_DECL(country_t, geoip_get_country, (const char *country));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- (void)arg;
- NS_MOCK(geoip_is_loaded);
- NS_MOCK(geoip_get_n_countries);
- NS_MOCK(geoip_get_country);
- smartlist_add(set->country_names, tor_strndup("foo", 3));
- routerset_refresh_countries(set);
- tt_ptr_op(set->countries, OP_NE, NULL);
- tt_int_op(set->n_countries, OP_EQ, 2);
- tt_int_op(CALLED(geoip_is_loaded), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_n_countries), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_country), OP_EQ, 1);
- tt_int_op((unsigned int)(*set->countries), OP_NE, 0);
- done:
- NS_UNMOCK(geoip_is_loaded);
- NS_UNMOCK(geoip_get_n_countries);
- NS_UNMOCK(geoip_get_country);
- routerset_free(set);
- }
- static int
- NS(geoip_is_loaded)(sa_family_t family)
- {
- (void)family;
- CALLED(geoip_is_loaded)++;
- return 1;
- }
- static int
- NS(geoip_get_n_countries)(void)
- {
- CALLED(geoip_get_n_countries)++;
- return 2;
- }
- static country_t
- NS(geoip_get_country)(const char *countrycode)
- {
- (void)countrycode;
- CALLED(geoip_get_country)++;
- return 1;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_refresh_counties, one_invalid_country)
- /*
- * Structural test for routerset_refresh_counties, with one invalid
- * country code..
- */
- NS_DECL(int, geoip_is_loaded, (sa_family_t family));
- NS_DECL(int, geoip_get_n_countries, (void));
- NS_DECL(country_t, geoip_get_country, (const char *country));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- (void)arg;
- NS_MOCK(geoip_is_loaded);
- NS_MOCK(geoip_get_n_countries);
- NS_MOCK(geoip_get_country);
- smartlist_add(set->country_names, tor_strndup("foo", 3));
- routerset_refresh_countries(set);
- tt_ptr_op(set->countries, OP_NE, NULL);
- tt_int_op(set->n_countries, OP_EQ, 2);
- tt_int_op(CALLED(geoip_is_loaded), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_n_countries), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_country), OP_EQ, 1);
- tt_int_op((unsigned int)(*set->countries), OP_EQ, 0);
- done:
- NS_UNMOCK(geoip_is_loaded);
- NS_UNMOCK(geoip_get_n_countries);
- NS_UNMOCK(geoip_get_country);
- routerset_free(set);
- }
- static int
- NS(geoip_is_loaded)(sa_family_t family)
- {
- (void)family;
- CALLED(geoip_is_loaded)++;
- return 1;
- }
- static int
- NS(geoip_get_n_countries)(void)
- {
- CALLED(geoip_get_n_countries)++;
- return 2;
- }
- static country_t
- NS(geoip_get_country)(const char *countrycode)
- {
- (void)countrycode;
- CALLED(geoip_get_country)++;
- return -1;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, malformed)
- /*
- * Functional test, with a malformed string to parse.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- const char *s = "_";
- int r;
- (void)arg;
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, -1);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, valid_hexdigest)
- /*
- * Functional test for routerset_parse, that routerset_parse returns 0
- * on a valid hexdigest entry.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- const char *s;
- int r;
- (void)arg;
- set = routerset_new();
- s = "$0000000000000000000000000000000000000000";
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(digestmap_isempty(set->digests), OP_NE, 1);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, valid_nickname)
- /*
- * Functional test for routerset_parse, when given a valid nickname as input.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- const char *s;
- int r;
- (void)arg;
- set = routerset_new();
- s = "fred";
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(strmap_isempty(set->names), OP_NE, 1);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, get_countryname)
- /*
- * Functional test for routerset_parse, when given a valid countryname.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- const char *s;
- int r;
- (void)arg;
- set = routerset_new();
- s = "{cc}";
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(smartlist_len(set->country_names), OP_NE, 0);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, policy_wildcard)
- /*
- * Structural test for routerset_parse, when given a valid wildcard policy.
- */
- NS_DECL(addr_policy_t *, router_parse_addr_policy_item_from_string,
- (const char *s, int assume_action));
- addr_policy_t *NS(mock_addr_policy);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- const char *s;
- int r;
- (void)arg;
- NS_MOCK(router_parse_addr_policy_item_from_string);
- NS(mock_addr_policy) = tor_malloc_zero(sizeof(addr_policy_t));
- set = routerset_new();
- s = "*";
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(smartlist_len(set->policies), OP_NE, 0);
- tt_int_op(CALLED(router_parse_addr_policy_item_from_string), OP_EQ, 1);
- done:
- routerset_free(set);
- }
- addr_policy_t *
- NS(router_parse_addr_policy_item_from_string)(const char *s, int assume_action)
- {
- (void)s;
- (void)assume_action;
- CALLED(router_parse_addr_policy_item_from_string)++;
- return NS(mock_addr_policy);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, policy_ipv4)
- /*
- * Structural test for routerset_parse, when given a valid IPv4 address
- * literal policy.
- */
- NS_DECL(addr_policy_t *, router_parse_addr_policy_item_from_string,
- (const char *s, int assume_action));
- addr_policy_t *NS(mock_addr_policy);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- const char *s;
- int r;
- (void)arg;
- NS_MOCK(router_parse_addr_policy_item_from_string);
- NS(mock_addr_policy) = tor_malloc_zero(sizeof(addr_policy_t));
- set = routerset_new();
- s = "127.0.0.1";
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(smartlist_len(set->policies), OP_NE, 0);
- tt_int_op(CALLED(router_parse_addr_policy_item_from_string), OP_EQ, 1);
- done:
- routerset_free(set);
- }
- addr_policy_t *
- NS(router_parse_addr_policy_item_from_string)(const char *s, int assume_action)
- {
- (void)s;
- (void)assume_action;
- CALLED(router_parse_addr_policy_item_from_string)++;
- return NS(mock_addr_policy);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_parse, policy_ipv6)
- /*
- * Structural test for routerset_parse, when given a valid IPv6 address
- * literal policy.
- */
- NS_DECL(addr_policy_t *, router_parse_addr_policy_item_from_string,
- (const char *s, int assume_action));
- addr_policy_t *NS(mock_addr_policy);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- const char *s;
- int r;
- (void)arg;
- NS_MOCK(router_parse_addr_policy_item_from_string);
- NS(mock_addr_policy) = tor_malloc_zero(sizeof(addr_policy_t));
- set = routerset_new();
- s = "::1";
- r = routerset_parse(set, s, "");
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(smartlist_len(set->policies), OP_NE, 0);
- tt_int_op(CALLED(router_parse_addr_policy_item_from_string), OP_EQ, 1);
- done:
- routerset_free(set);
- }
- addr_policy_t *
- NS(router_parse_addr_policy_item_from_string)(const char *s, int assume_action)
- {
- (void)s;
- (void)assume_action;
- CALLED(router_parse_addr_policy_item_from_string)++;
- return NS(mock_addr_policy);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_union, source_bad)
- /*
- * Structural test for routerset_union, when given a bad source argument.
- */
- NS_DECL(smartlist_t *, smartlist_new, (void));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set, *bad_set;
- (void)arg;
- set = routerset_new();
- bad_set = routerset_new();
- smartlist_free(bad_set->list);
- bad_set->list = NULL;
- NS_MOCK(smartlist_new);
- routerset_union(set, NULL);
- tt_int_op(CALLED(smartlist_new), OP_EQ, 0);
- routerset_union(set, bad_set);
- tt_int_op(CALLED(smartlist_new), OP_EQ, 0);
- done:
- NS_UNMOCK(smartlist_new);
- routerset_free(set);
- /* Just recreate list, so we can simply use routerset_free. */
- bad_set->list = smartlist_new();
- routerset_free(bad_set);
- }
- static smartlist_t *
- NS(smartlist_new)(void)
- {
- CALLED(smartlist_new)++;
- return NULL;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_union, one)
- /*
- * Functional test for routerset_union.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *src = routerset_new();
- routerset_t *tgt;
- (void)arg;
- tgt = routerset_new();
- smartlist_add(src->list, tor_strdup("{xx}"));
- routerset_union(tgt, src);
- tt_int_op(smartlist_len(tgt->list), OP_NE, 0);
- done:
- routerset_free(src);
- routerset_free(tgt);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_is_list
- /*
- * Functional tests for routerset_is_list.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set;
- addr_policy_t *policy;
- int is_list;
- (void)arg;
- /* len(set->country_names) == 0, len(set->policies) == 0 */
- set = routerset_new();
- is_list = routerset_is_list(set);
- routerset_free(set);
- set = NULL;
- tt_int_op(is_list, OP_NE, 0);
- /* len(set->country_names) != 0, len(set->policies) == 0 */
- set = routerset_new();
- smartlist_add(set->country_names, tor_strndup("foo", 3));
- is_list = routerset_is_list(set);
- routerset_free(set);
- set = NULL;
- tt_int_op(is_list, OP_EQ, 0);
- /* len(set->country_names) == 0, len(set->policies) != 0 */
- set = routerset_new();
- policy = tor_malloc_zero(sizeof(addr_policy_t));
- smartlist_add(set->policies, (void *)policy);
- is_list = routerset_is_list(set);
- routerset_free(set);
- set = NULL;
- tt_int_op(is_list, OP_EQ, 0);
- /* len(set->country_names) != 0, len(set->policies) != 0 */
- set = routerset_new();
- smartlist_add(set->country_names, tor_strndup("foo", 3));
- policy = tor_malloc_zero(sizeof(addr_policy_t));
- smartlist_add(set->policies, (void *)policy);
- is_list = routerset_is_list(set);
- routerset_free(set);
- set = NULL;
- tt_int_op(is_list, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_needs_geoip
- /*
- * Functional tests for routerset_needs_geoip.
- */
- static void
- NS(test_main)(void *arg)
- {
- const routerset_t *set;
- int needs_geoip;
- (void)arg;
- set = NULL;
- needs_geoip = routerset_needs_geoip(set);
- tt_int_op(needs_geoip, OP_EQ, 0);
- set = routerset_new();
- needs_geoip = routerset_needs_geoip(set);
- routerset_free((routerset_t *)set);
- tt_int_op(needs_geoip, OP_EQ, 0);
- set = NULL;
- set = routerset_new();
- smartlist_add(set->country_names, tor_strndup("xx", 2));
- needs_geoip = routerset_needs_geoip(set);
- routerset_free((routerset_t *)set);
- set = NULL;
- tt_int_op(needs_geoip, OP_NE, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_is_empty
- /*
- * Functional tests for routerset_is_empty.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = NULL;
- int is_empty;
- (void)arg;
- is_empty = routerset_is_empty(set);
- tt_int_op(is_empty, OP_NE, 0);
- set = routerset_new();
- is_empty = routerset_is_empty(set);
- routerset_free(set);
- set = NULL;
- tt_int_op(is_empty, OP_NE, 0);
- set = routerset_new();
- smartlist_add(set->list, tor_strdup("{xx}"));
- is_empty = routerset_is_empty(set);
- routerset_free(set);
- set = NULL;
- tt_int_op(is_empty, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, null_set_or_null_set_list)
- /*
- * Functional test for routerset_contains, when given a NULL set or the
- * set has a NULL list.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = NULL;
- int contains;
- (void)arg;
- contains = routerset_contains(set, NULL, 0, NULL, NULL, 0);
- tt_int_op(contains, OP_EQ, 0);
- set = tor_malloc_zero(sizeof(routerset_t));
- set->list = NULL;
- contains = routerset_contains(set, NULL, 0, NULL, NULL, 0);
- tor_free(set);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_null_nickname)
- /*
- * Functional test for routerset_contains, when given a valid routerset but a
- * NULL nickname.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- char *nickname = NULL;
- int contains;
- (void)arg;
- contains = routerset_contains(set, NULL, 0, nickname, NULL, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_nickname)
- /*
- * Functional test for routerset_contains, when given a valid routerset
- * and the nickname is in the routerset.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- const char *nickname;
- int contains;
- (void)arg;
- nickname = "Foo"; /* This tests the lowercase comparison as well. */
- strmap_set_lc(set->names, nickname, (void *)1);
- contains = routerset_contains(set, NULL, 0, nickname, NULL, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 4);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_no_nickname)
- /*
- * Functional test for routerset_contains, when given a valid routerset
- * and the nickname is not in the routerset.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains;
- (void)arg;
- strmap_set_lc(set->names, "bar", (void *)1);
- contains = routerset_contains(set, NULL, 0, "foo", NULL, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_digest)
- /*
- * Functional test for routerset_contains, when given a valid routerset
- * and the digest is contained in the routerset.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains;
- uint8_t foo[20] = { 2, 3, 4 };
- (void)arg;
- digestmap_set(set->digests, (const char*)foo, (void *)1);
- contains = routerset_contains(set, NULL, 0, NULL, (const char*)foo, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 4);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_no_digest)
- /*
- * Functional test for routerset_contains, when given a valid routerset
- * and the digest is not contained in the routerset.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains;
- uint8_t bar[20] = { 9, 10, 11, 55 };
- uint8_t foo[20] = { 1, 2, 3, 4};
- (void)arg;
- digestmap_set(set->digests, (const char*)bar, (void *)1);
- contains = routerset_contains(set, NULL, 0, NULL, (const char*)foo, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_null_digest)
- /*
- * Functional test for routerset_contains, when given a valid routerset
- * and the digest is NULL.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains;
- uint8_t bar[20] = { 9, 10, 11, 55 };
- (void)arg;
- digestmap_set(set->digests, (const char*)bar, (void *)1);
- contains = routerset_contains(set, NULL, 0, NULL, NULL, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_addr)
- /*
- * Structural test for routerset_contains, when given a valid routerset
- * and the address is rejected by policy.
- */
- NS_DECL(addr_policy_result_t, compare_tor_addr_to_addr_policy,
- (const tor_addr_t *addr, uint16_t port, const smartlist_t *policy));
- static tor_addr_t MOCK_TOR_ADDR;
- #define MOCK_TOR_ADDR_PTR (&MOCK_TOR_ADDR)
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- tor_addr_t *addr = MOCK_TOR_ADDR_PTR;
- int contains;
- (void)arg;
- NS_MOCK(compare_tor_addr_to_addr_policy);
- contains = routerset_contains(set, addr, 0, NULL, NULL, 0);
- routerset_free(set);
- tt_int_op(CALLED(compare_tor_addr_to_addr_policy), OP_EQ, 1);
- tt_int_op(contains, OP_EQ, 3);
- done:
- ;
- }
- addr_policy_result_t
- NS(compare_tor_addr_to_addr_policy)(const tor_addr_t *addr, uint16_t port,
- const smartlist_t *policy)
- {
- (void)port;
- (void)policy;
- CALLED(compare_tor_addr_to_addr_policy)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- return ADDR_POLICY_REJECTED;
- done:
- return 0;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_no_addr)
- /*
- * Structural test for routerset_contains, when given a valid routerset
- * and the address is not rejected by policy.
- */
- NS_DECL(addr_policy_result_t, compare_tor_addr_to_addr_policy,
- (const tor_addr_t *addr, uint16_t port, const smartlist_t *policy));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- tor_addr_t *addr = MOCK_TOR_ADDR_PTR;
- int contains;
- (void)arg;
- NS_MOCK(compare_tor_addr_to_addr_policy);
- contains = routerset_contains(set, addr, 0, NULL, NULL, 0);
- routerset_free(set);
- tt_int_op(CALLED(compare_tor_addr_to_addr_policy), OP_EQ, 1);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- addr_policy_result_t
- NS(compare_tor_addr_to_addr_policy)(const tor_addr_t *addr, uint16_t port,
- const smartlist_t *policy)
- {
- (void)port;
- (void)policy;
- CALLED(compare_tor_addr_to_addr_policy)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- return ADDR_POLICY_ACCEPTED;
- done:
- return 0;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, set_and_null_addr)
- /*
- * Structural test for routerset_contains, when given a valid routerset
- * and the address is NULL.
- */
- NS_DECL(addr_policy_result_t, compare_tor_addr_to_addr_policy,
- (const tor_addr_t *addr, uint16_t port, const smartlist_t *policy));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains;
- (void)arg;
- NS_MOCK(compare_tor_addr_to_addr_policy);
- contains = routerset_contains(set, NULL, 0, NULL, NULL, 0);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 0);
- done:
- ;
- }
- addr_policy_result_t
- NS(compare_tor_addr_to_addr_policy)(const tor_addr_t *addr, uint16_t port,
- const smartlist_t *policy)
- {
- (void)port;
- (void)policy;
- CALLED(compare_tor_addr_to_addr_policy)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- return ADDR_POLICY_ACCEPTED;
- done:
- return 0;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, countries_no_geoip)
- /*
- * Structural test for routerset_contains, when there is no matching country
- * for the address.
- */
- NS_DECL(addr_policy_result_t, compare_tor_addr_to_addr_policy,
- (const tor_addr_t *addr, uint16_t port, const smartlist_t *policy));
- NS_DECL(int, geoip_get_country_by_addr, (const tor_addr_t *addr));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains = 1;
- (void)arg;
- NS_MOCK(compare_tor_addr_to_addr_policy);
- NS_MOCK(geoip_get_country_by_addr);
- set->countries = bitarray_init_zero(1);
- bitarray_set(set->countries, 1);
- contains = routerset_contains(set, MOCK_TOR_ADDR_PTR, 0, NULL, NULL, -1);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 0);
- tt_int_op(CALLED(compare_tor_addr_to_addr_policy), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_country_by_addr), OP_EQ, 1);
- done:
- ;
- }
- addr_policy_result_t
- NS(compare_tor_addr_to_addr_policy)(const tor_addr_t *addr, uint16_t port,
- const smartlist_t *policy)
- {
- (void)port;
- (void)policy;
- CALLED(compare_tor_addr_to_addr_policy)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- done:
- return ADDR_POLICY_ACCEPTED;
- }
- int
- NS(geoip_get_country_by_addr)(const tor_addr_t *addr)
- {
- CALLED(geoip_get_country_by_addr)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- done:
- return -1;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains, countries_geoip)
- /*
- * Structural test for routerset_contains, when there a matching country
- * for the address.
- */
- NS_DECL(addr_policy_result_t, compare_tor_addr_to_addr_policy,
- (const tor_addr_t *addr, uint16_t port, const smartlist_t *policy));
- NS_DECL(int, geoip_get_country_by_addr, (const tor_addr_t *addr));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int contains = 1;
- (void)arg;
- NS_MOCK(compare_tor_addr_to_addr_policy);
- NS_MOCK(geoip_get_country_by_addr);
- set->n_countries = 2;
- set->countries = bitarray_init_zero(1);
- bitarray_set(set->countries, 1);
- contains = routerset_contains(set, MOCK_TOR_ADDR_PTR, 0, NULL, NULL, -1);
- routerset_free(set);
- tt_int_op(contains, OP_EQ, 2);
- tt_int_op(CALLED(compare_tor_addr_to_addr_policy), OP_EQ, 1);
- tt_int_op(CALLED(geoip_get_country_by_addr), OP_EQ, 1);
- done:
- ;
- }
- addr_policy_result_t
- NS(compare_tor_addr_to_addr_policy)(const tor_addr_t *addr, uint16_t port,
- const smartlist_t *policy)
- {
- (void)port;
- (void)policy;
- CALLED(compare_tor_addr_to_addr_policy)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- done:
- return ADDR_POLICY_ACCEPTED;
- }
- int
- NS(geoip_get_country_by_addr)(const tor_addr_t *addr)
- {
- CALLED(geoip_get_country_by_addr)++;
- tt_ptr_op(addr, OP_EQ, MOCK_TOR_ADDR_PTR);
- done:
- return 1;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_add_unknown_ccs, only_flag_and_no_ccs)
- /*
- * Functional test for routerset_add_unknown_ccs, where only_if_some_cc_set
- * is set and there are no country names.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- routerset_t **setp = &set;
- int r;
- (void)arg;
- r = routerset_add_unknown_ccs(setp, 1);
- tt_int_op(r, OP_EQ, 0);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_add_unknown_ccs, creates_set)
- /*
- * Functional test for routerset_add_unknown_ccs, where the set argument
- * is created if passed in as NULL.
- */
- /* The mock is only used to stop the test from asserting erroneously. */
- NS_DECL(country_t, geoip_get_country, (const char *country));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = NULL;
- routerset_t **setp = &set;
- int r;
- (void)arg;
- NS_MOCK(geoip_get_country);
- r = routerset_add_unknown_ccs(setp, 0);
- tt_ptr_op(*setp, OP_NE, NULL);
- tt_int_op(r, OP_EQ, 0);
- done:
- if (set != NULL)
- routerset_free(set);
- }
- country_t
- NS(geoip_get_country)(const char *country)
- {
- (void)country;
- CALLED(geoip_get_country)++;
- return -1;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_add_unknown_ccs, add_unknown)
- /*
- * Structural test for routerset_add_unknown_ccs, that the "{??}"
- * country code is added to the list.
- */
- NS_DECL(country_t, geoip_get_country, (const char *country));
- NS_DECL(int, geoip_is_loaded, (sa_family_t family));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- routerset_t **setp = &set;
- int r;
- (void)arg;
- NS_MOCK(geoip_get_country);
- NS_MOCK(geoip_is_loaded);
- r = routerset_add_unknown_ccs(setp, 0);
- tt_int_op(r, OP_EQ, 1);
- tt_int_op(smartlist_contains_string(set->country_names, "??"), OP_EQ, 1);
- tt_int_op(smartlist_contains_string(set->list, "{??}"), OP_EQ, 1);
- done:
- if (set != NULL)
- routerset_free(set);
- }
- country_t
- NS(geoip_get_country)(const char *country)
- {
- int arg_is_qq, arg_is_a1;
- CALLED(geoip_get_country)++;
- arg_is_qq = !strcmp(country, "??");
- arg_is_a1 = !strcmp(country, "A1");
- tt_int_op(arg_is_qq || arg_is_a1, OP_EQ, 1);
- if (arg_is_qq)
- return 1;
- done:
- return -1;
- }
- int
- NS(geoip_is_loaded)(sa_family_t family)
- {
- CALLED(geoip_is_loaded)++;
- tt_int_op(family, OP_EQ, AF_INET);
- done:
- return 0;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_add_unknown_ccs, add_a1)
- /*
- * Structural test for routerset_add_unknown_ccs, that the "{a1}"
- * country code is added to the list.
- */
- NS_DECL(country_t, geoip_get_country, (const char *country));
- NS_DECL(int, geoip_is_loaded, (sa_family_t family));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- routerset_t **setp = &set;
- int r;
- (void)arg;
- NS_MOCK(geoip_get_country);
- NS_MOCK(geoip_is_loaded);
- r = routerset_add_unknown_ccs(setp, 0);
- tt_int_op(r, OP_EQ, 1);
- tt_int_op(smartlist_contains_string(set->country_names, "a1"), OP_EQ, 1);
- tt_int_op(smartlist_contains_string(set->list, "{a1}"), OP_EQ, 1);
- done:
- if (set != NULL)
- routerset_free(set);
- }
- country_t
- NS(geoip_get_country)(const char *country)
- {
- int arg_is_qq, arg_is_a1;
- CALLED(geoip_get_country)++;
- arg_is_qq = !strcmp(country, "??");
- arg_is_a1 = !strcmp(country, "A1");
- tt_int_op(arg_is_qq || arg_is_a1, OP_EQ, 1);
- if (arg_is_a1)
- return 1;
- done:
- return -1;
- }
- int
- NS(geoip_is_loaded)(sa_family_t family)
- {
- CALLED(geoip_is_loaded)++;
- tt_int_op(family, OP_EQ, AF_INET);
- done:
- return 0;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_contains_extendinfo
- /*
- * Functional test for routerset_contains_extendinfo.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- extend_info_t ei;
- int r;
- const char *nickname = "foo";
- (void)arg;
- memset(&ei, 0, sizeof(ei));
- strmap_set_lc(set->names, nickname, (void *)1);
- strncpy(ei.nickname, nickname, sizeof(ei.nickname) - 1);
- ei.nickname[sizeof(ei.nickname) - 1] = '\0';
- r = routerset_contains_extendinfo(set, &ei);
- tt_int_op(r, OP_EQ, 4);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_contains_router
- /*
- * Functional test for routerset_contains_router.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- routerinfo_t ri;
- country_t country = 1;
- int r;
- const char *nickname = "foo";
- (void)arg;
- memset(&ri, 0, sizeof(ri));
- strmap_set_lc(set->names, nickname, (void *)1);
- ri.nickname = (char *)nickname;
- r = routerset_contains_router(set, &ri, country);
- tt_int_op(r, OP_EQ, 4);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_contains_routerstatus
- /*
- * Functional test for routerset_contains_routerstatus.
- */
- // XXX: This is a bit brief. It only populates and tests the nickname fields
- // ie., enough to make the containment check succeed. Perhaps it should do
- // a bit more or test a bit more.
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- routerstatus_t rs;
- country_t country = 1;
- int r;
- const char *nickname = "foo";
- (void)arg;
- memset(&rs, 0, sizeof(rs));
- strmap_set_lc(set->names, nickname, (void *)1);
- strncpy(rs.nickname, nickname, sizeof(rs.nickname) - 1);
- rs.nickname[sizeof(rs.nickname) - 1] = '\0';
- r = routerset_contains_routerstatus(set, &rs, country);
- tt_int_op(r, OP_EQ, 4);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains_node, none)
- /*
- * Functional test for routerset_contains_node, when the node has no
- * routerset or routerinfo.
- */
- node_t NS(mock_node);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int r;
- (void)arg;
- NS(mock_node).ri = NULL;
- NS(mock_node).rs = NULL;
- r = routerset_contains_node(set, &NS(mock_node));
- tt_int_op(r, OP_EQ, 0);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains_node, routerstatus)
- /*
- * Functional test for routerset_contains_node, when the node has a
- * routerset and no routerinfo.
- */
- node_t NS(mock_node);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int r;
- const char *nickname = "foo";
- routerstatus_t rs;
- (void)arg;
- strmap_set_lc(set->names, nickname, (void *)1);
- strncpy(rs.nickname, nickname, sizeof(rs.nickname) - 1);
- rs.nickname[sizeof(rs.nickname) - 1] = '\0';
- NS(mock_node).ri = NULL;
- NS(mock_node).rs = &rs;
- r = routerset_contains_node(set, &NS(mock_node));
- tt_int_op(r, OP_EQ, 4);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_contains_node, routerinfo)
- /*
- * Functional test for routerset_contains_node, when the node has no
- * routerset and a routerinfo.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- int r;
- const char *nickname = "foo";
- routerinfo_t ri;
- node_t mock_node;
- (void)arg;
- strmap_set_lc(set->names, nickname, (void *)1);
- ri.nickname = (char *)nickname;
- mock_node.ri = &ri;
- mock_node.rs = NULL;
- r = routerset_contains_node(set, &mock_node);
- tt_int_op(r, OP_EQ, 4);
- done:
- routerset_free(set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_get_all_nodes, no_routerset)
- /*
- * Functional test for routerset_get_all_nodes, when routerset is NULL or
- * the routerset list is NULL.
- */
- static void
- NS(test_main)(void *arg)
- {
- smartlist_t *out = smartlist_new();
- routerset_t *set = NULL;
- (void)arg;
- tt_int_op(smartlist_len(out), OP_EQ, 0);
- routerset_get_all_nodes(out, NULL, NULL, 0);
- tt_int_op(smartlist_len(out), OP_EQ, 0);
- set = routerset_new();
- smartlist_free(set->list);
- routerset_get_all_nodes(out, NULL, NULL, 0);
- tt_int_op(smartlist_len(out), OP_EQ, 0);
- /* Just recreate list, so we can simply use routerset_free. */
- set->list = smartlist_new();
- done:
- routerset_free(set);
- smartlist_free(out);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_get_all_nodes, list_with_no_nodes)
- /*
- * Structural test for routerset_get_all_nodes, when the routerset list
- * is empty.
- */
- NS_DECL(const node_t *, node_get_by_nickname,
- (const char *nickname, int warn_if_unused));
- const char *NS(mock_nickname);
- static void
- NS(test_main)(void *arg)
- {
- smartlist_t *out = smartlist_new();
- routerset_t *set = routerset_new();
- int out_len;
- (void)arg;
- NS_MOCK(node_get_by_nickname);
- NS(mock_nickname) = "foo";
- smartlist_add(set->list, tor_strdup(NS(mock_nickname)));
- routerset_get_all_nodes(out, set, NULL, 0);
- out_len = smartlist_len(out);
- smartlist_free(out);
- routerset_free(set);
- tt_int_op(out_len, OP_EQ, 0);
- tt_int_op(CALLED(node_get_by_nickname), OP_EQ, 1);
- done:
- ;
- }
- const node_t *
- NS(node_get_by_nickname)(const char *nickname, int warn_if_unused)
- {
- CALLED(node_get_by_nickname)++;
- tt_str_op(nickname, OP_EQ, NS(mock_nickname));
- tt_int_op(warn_if_unused, OP_EQ, 1);
- done:
- return NULL;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_get_all_nodes, list_flag_not_running)
- /*
- * Structural test for routerset_get_all_nodes, with the running_only flag
- * is set but the nodes are not running.
- */
- NS_DECL(const node_t *, node_get_by_nickname,
- (const char *nickname, int warn_if_unused));
- const char *NS(mock_nickname);
- node_t NS(mock_node);
- static void
- NS(test_main)(void *arg)
- {
- smartlist_t *out = smartlist_new();
- routerset_t *set = routerset_new();
- int out_len;
- (void)arg;
- NS_MOCK(node_get_by_nickname);
- NS(mock_node).is_running = 0;
- NS(mock_nickname) = "foo";
- smartlist_add(set->list, tor_strdup(NS(mock_nickname)));
- routerset_get_all_nodes(out, set, NULL, 1);
- out_len = smartlist_len(out);
- smartlist_free(out);
- routerset_free(set);
- tt_int_op(out_len, OP_EQ, 0);
- tt_int_op(CALLED(node_get_by_nickname), OP_EQ, 1);
- done:
- ;
- }
- const node_t *
- NS(node_get_by_nickname)(const char *nickname, int warn_if_unused)
- {
- CALLED(node_get_by_nickname)++;
- tt_str_op(nickname, OP_EQ, NS(mock_nickname));
- tt_int_op(warn_if_unused, OP_EQ, 1);
- done:
- return &NS(mock_node);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_get_all_nodes, list)
- /*
- * Structural test for routerset_get_all_nodes.
- */
- NS_DECL(const node_t *, node_get_by_nickname,
- (const char *nickname, int warn_if_unused));
- char *NS(mock_nickname);
- node_t NS(mock_node);
- static void
- NS(test_main)(void *arg)
- {
- smartlist_t *out = smartlist_new();
- routerset_t *set = routerset_new();
- int out_len;
- node_t *ent;
- (void)arg;
- NS_MOCK(node_get_by_nickname);
- NS(mock_nickname) = tor_strdup("foo");
- smartlist_add(set->list, NS(mock_nickname));
- routerset_get_all_nodes(out, set, NULL, 0);
- out_len = smartlist_len(out);
- ent = (node_t *)smartlist_get(out, 0);
- smartlist_free(out);
- routerset_free(set);
- tt_int_op(out_len, OP_EQ, 1);
- tt_ptr_op(ent, OP_EQ, &NS(mock_node));
- tt_int_op(CALLED(node_get_by_nickname), OP_EQ, 1);
- done:
- ;
- }
- const node_t *
- NS(node_get_by_nickname)(const char *nickname, int warn_if_unused)
- {
- CALLED(node_get_by_nickname)++;
- tt_str_op(nickname, OP_EQ, NS(mock_nickname));
- tt_int_op(warn_if_unused, OP_EQ, 1);
- done:
- return &NS(mock_node);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_get_all_nodes, nodelist_with_no_nodes)
- /*
- * Structural test for routerset_get_all_nodes, when the nodelist has no nodes.
- */
- NS_DECL(smartlist_t *, nodelist_get_list, (void));
- smartlist_t *NS(mock_smartlist);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- smartlist_t *out = smartlist_new();
- int r;
- (void)arg;
- NS_MOCK(nodelist_get_list);
- smartlist_add(set->country_names, tor_strdup("{xx}"));
- NS(mock_smartlist) = smartlist_new();
- routerset_get_all_nodes(out, set, NULL, 1);
- r = smartlist_len(out);
- routerset_free(set);
- smartlist_free(out);
- smartlist_free(NS(mock_smartlist));
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(CALLED(nodelist_get_list), OP_EQ, 1);
- done:
- ;
- }
- smartlist_t *
- NS(nodelist_get_list)(void)
- {
- CALLED(nodelist_get_list)++;
- return NS(mock_smartlist);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_get_all_nodes, nodelist_flag_not_running)
- /*
- * Structural test for routerset_get_all_nodes, with a non-list routerset
- * the running_only flag is set, but the nodes are not running.
- */
- NS_DECL(smartlist_t *, nodelist_get_list, (void));
- smartlist_t *NS(mock_smartlist);
- node_t NS(mock_node);
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- smartlist_t *out = smartlist_new();
- int r;
- (void)arg;
- NS_MOCK(nodelist_get_list);
- smartlist_add(set->country_names, tor_strdup("{xx}"));
- NS(mock_smartlist) = smartlist_new();
- NS(mock_node).is_running = 0;
- smartlist_add(NS(mock_smartlist), (void *)&NS(mock_node));
- routerset_get_all_nodes(out, set, NULL, 1);
- r = smartlist_len(out);
- routerset_free(set);
- smartlist_free(out);
- smartlist_free(NS(mock_smartlist));
- tt_int_op(r, OP_EQ, 0);
- tt_int_op(CALLED(nodelist_get_list), OP_EQ, 1);
- done:
- ;
- }
- smartlist_t *
- NS(nodelist_get_list)(void)
- {
- CALLED(nodelist_get_list)++;
- return NS(mock_smartlist);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_subtract_nodes
- /*
- * Functional test for routerset_subtract_nodes.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = routerset_new();
- smartlist_t *list = smartlist_new();
- const char *nickname = "foo";
- routerinfo_t ri;
- node_t mock_node;
- (void)arg;
- strmap_set_lc(set->names, nickname, (void *)1);
- ri.nickname = (char *)nickname;
- mock_node.rs = NULL;
- mock_node.ri = &ri;
- smartlist_add(list, (void *)&mock_node);
- tt_int_op(smartlist_len(list), OP_NE, 0);
- routerset_subtract_nodes(list, set);
- tt_int_op(smartlist_len(list), OP_EQ, 0);
- done:
- routerset_free(set);
- smartlist_free(list);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_subtract_nodes, null_routerset)
- /*
- * Functional test for routerset_subtract_nodes, with a NULL routerset.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = NULL;
- smartlist_t *list = smartlist_new();
- const char *nickname = "foo";
- routerinfo_t ri;
- node_t mock_node;
- (void)arg;
- ri.nickname = (char *)nickname;
- mock_node.ri = &ri;
- smartlist_add(list, (void *)&mock_node);
- tt_int_op(smartlist_len(list), OP_NE, 0);
- routerset_subtract_nodes(list, set);
- tt_int_op(smartlist_len(list), OP_NE, 0);
- done:
- routerset_free(set);
- smartlist_free(list);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_to_string
- /*
- * Functional test for routerset_to_string.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *set = NULL;
- char *s = NULL;
- (void)arg;
- set = NULL;
- s = routerset_to_string(set);
- tt_str_op(s, OP_EQ, "");
- tor_free(s);
- set = routerset_new();
- s = routerset_to_string(set);
- tt_str_op(s, OP_EQ, "");
- tor_free(s);
- routerset_free(set); set = NULL;
- set = routerset_new();
- smartlist_add(set->list, tor_strndup("a", 1));
- s = routerset_to_string(set);
- tt_str_op(s, OP_EQ, "a");
- tor_free(s);
- routerset_free(set); set = NULL;
- set = routerset_new();
- smartlist_add(set->list, tor_strndup("a", 1));
- smartlist_add(set->list, tor_strndup("b", 1));
- s = routerset_to_string(set);
- tt_str_op(s, OP_EQ, "a,b");
- tor_free(s);
- routerset_free(set); set = NULL;
- done:
- tor_free(s);
- routerset_free((routerset_t *)set);
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_equal, empty_empty)
- /*
- * Functional test for routerset_equal, with both routersets empty.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *a = routerset_new(), *b = routerset_new();
- int r;
- (void)arg;
- r = routerset_equal(a, b);
- routerset_free(a);
- routerset_free(b);
- tt_int_op(r, OP_EQ, 1);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_equal, empty_not_empty)
- /*
- * Functional test for routerset_equal, with one routersets empty.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *a = routerset_new(), *b = routerset_new();
- int r;
- (void)arg;
- smartlist_add(b->list, tor_strdup("{xx}"));
- r = routerset_equal(a, b);
- routerset_free(a);
- routerset_free(b);
- tt_int_op(r, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_equal, differing_lengths)
- /*
- * Functional test for routerset_equal, with the routersets having
- * differing lengths.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *a = routerset_new(), *b = routerset_new();
- int r;
- (void)arg;
- smartlist_add(a->list, tor_strdup("{aa}"));
- smartlist_add(b->list, tor_strdup("{b1}"));
- smartlist_add(b->list, tor_strdup("{b2}"));
- r = routerset_equal(a, b);
- routerset_free(a);
- routerset_free(b);
- tt_int_op(r, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_equal, unequal)
- /*
- * Functional test for routerset_equal, with the routersets being
- * different.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *a = routerset_new(), *b = routerset_new();
- int r;
- (void)arg;
- smartlist_add(a->list, tor_strdup("foo"));
- smartlist_add(b->list, tor_strdup("bar"));
- r = routerset_equal(a, b);
- routerset_free(a);
- routerset_free(b);
- tt_int_op(r, OP_EQ, 0);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_equal, equal)
- /*
- * Functional test for routerset_equal, with the routersets being
- * equal.
- */
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *a = routerset_new(), *b = routerset_new();
- int r;
- (void)arg;
- smartlist_add(a->list, tor_strdup("foo"));
- smartlist_add(b->list, tor_strdup("foo"));
- r = routerset_equal(a, b);
- routerset_free(a);
- routerset_free(b);
- tt_int_op(r, OP_EQ, 1);
- done:
- ;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE ASPECT(routerset_free, null_routerset)
- /*
- * Structural test for routerset_free, where the routerset is NULL.
- */
- NS_DECL(void, smartlist_free, (smartlist_t *sl));
- static void
- NS(test_main)(void *arg)
- {
- (void)arg;
- NS_MOCK(smartlist_free);
- routerset_free(NULL);
- tt_int_op(CALLED(smartlist_free), OP_EQ, 0);
- done:
- ;
- }
- void
- NS(smartlist_free)(smartlist_t *s)
- {
- (void)s;
- CALLED(smartlist_free)++;
- }
- #undef NS_SUBMODULE
- #define NS_SUBMODULE routerset_free
- /*
- * Structural test for routerset_free.
- */
- NS_DECL(void, smartlist_free, (smartlist_t *sl));
- NS_DECL(void, strmap_free,(strmap_t *map, void (*free_val)(void*)));
- NS_DECL(void, digestmap_free, (digestmap_t *map, void (*free_val)(void*)));
- static void
- NS(test_main)(void *arg)
- {
- routerset_t *routerset = routerset_new();
- (void)arg;
- NS_MOCK(smartlist_free);
- NS_MOCK(strmap_free);
- NS_MOCK(digestmap_free);
- routerset_free(routerset);
- tt_int_op(CALLED(smartlist_free), OP_NE, 0);
- tt_int_op(CALLED(strmap_free), OP_NE, 0);
- tt_int_op(CALLED(digestmap_free), OP_NE, 0);
- done:
- ;
- }
- void
- NS(smartlist_free)(smartlist_t *s)
- {
- CALLED(smartlist_free)++;
- smartlist_free__real(s);
- }
- void
- NS(strmap_free)(strmap_t *map, void (*free_val)(void*))
- {
- CALLED(strmap_free)++;
- strmap_free__real(map, free_val);
- }
- void
- NS(digestmap_free)(digestmap_t *map, void (*free_val)(void*))
- {
- CALLED(digestmap_free)++;
- digestmap_free__real(map, free_val);
- }
- #undef NS_SUBMODULE
- struct testcase_t routerset_tests[] = {
- TEST_CASE(routerset_new),
- TEST_CASE(routerset_get_countryname),
- TEST_CASE(routerset_is_list),
- TEST_CASE(routerset_needs_geoip),
- TEST_CASE(routerset_is_empty),
- TEST_CASE_ASPECT(routerset_contains, null_set_or_null_set_list),
- TEST_CASE_ASPECT(routerset_contains, set_and_nickname),
- TEST_CASE_ASPECT(routerset_contains, set_and_null_nickname),
- TEST_CASE_ASPECT(routerset_contains, set_and_no_nickname),
- TEST_CASE_ASPECT(routerset_contains, set_and_digest),
- TEST_CASE_ASPECT(routerset_contains, set_and_no_digest),
- TEST_CASE_ASPECT(routerset_contains, set_and_null_digest),
- TEST_CASE_ASPECT(routerset_contains, set_and_addr),
- TEST_CASE_ASPECT(routerset_contains, set_and_no_addr),
- TEST_CASE_ASPECT(routerset_contains, set_and_null_addr),
- TEST_CASE_ASPECT(routerset_contains, countries_no_geoip),
- TEST_CASE_ASPECT(routerset_contains, countries_geoip),
- TEST_CASE_ASPECT(routerset_add_unknown_ccs, only_flag_and_no_ccs),
- TEST_CASE_ASPECT(routerset_add_unknown_ccs, creates_set),
- TEST_CASE_ASPECT(routerset_add_unknown_ccs, add_unknown),
- TEST_CASE_ASPECT(routerset_add_unknown_ccs, add_a1),
- TEST_CASE(routerset_contains_extendinfo),
- TEST_CASE(routerset_contains_router),
- TEST_CASE(routerset_contains_routerstatus),
- TEST_CASE_ASPECT(routerset_contains_node, none),
- TEST_CASE_ASPECT(routerset_contains_node, routerinfo),
- TEST_CASE_ASPECT(routerset_contains_node, routerstatus),
- TEST_CASE_ASPECT(routerset_get_all_nodes, no_routerset),
- TEST_CASE_ASPECT(routerset_get_all_nodes, list_with_no_nodes),
- TEST_CASE_ASPECT(routerset_get_all_nodes, list_flag_not_running),
- TEST_CASE_ASPECT(routerset_get_all_nodes, list),
- TEST_CASE_ASPECT(routerset_get_all_nodes, nodelist_with_no_nodes),
- TEST_CASE_ASPECT(routerset_get_all_nodes, nodelist_flag_not_running),
- TEST_CASE_ASPECT(routerset_refresh_counties, geoip_not_loaded),
- TEST_CASE_ASPECT(routerset_refresh_counties, no_countries),
- TEST_CASE_ASPECT(routerset_refresh_counties, one_valid_country),
- TEST_CASE_ASPECT(routerset_refresh_counties, one_invalid_country),
- TEST_CASE_ASPECT(routerset_union, source_bad),
- TEST_CASE_ASPECT(routerset_union, one),
- TEST_CASE_ASPECT(routerset_parse, malformed),
- TEST_CASE_ASPECT(routerset_parse, valid_hexdigest),
- TEST_CASE_ASPECT(routerset_parse, valid_nickname),
- TEST_CASE_ASPECT(routerset_parse, get_countryname),
- TEST_CASE_ASPECT(routerset_parse, policy_wildcard),
- TEST_CASE_ASPECT(routerset_parse, policy_ipv4),
- TEST_CASE_ASPECT(routerset_parse, policy_ipv6),
- TEST_CASE(routerset_subtract_nodes),
- TEST_CASE_ASPECT(routerset_subtract_nodes, null_routerset),
- TEST_CASE(routerset_to_string),
- TEST_CASE_ASPECT(routerset_equal, empty_empty),
- TEST_CASE_ASPECT(routerset_equal, empty_not_empty),
- TEST_CASE_ASPECT(routerset_equal, differing_lengths),
- TEST_CASE_ASPECT(routerset_equal, unequal),
- TEST_CASE_ASPECT(routerset_equal, equal),
- TEST_CASE_ASPECT(routerset_free, null_routerset),
- TEST_CASE(routerset_free),
- END_OF_TESTCASES
- };
|