libiptc.c revision 073df8feb0a8c4023ce40138e519ac9b341b1ca2
1073df8feb0a8c4023ce40138e519ac9b341b1ca2Karsten Desler/* Library which manipulates firewall rules. Version $Revision: 1.43 $ */ 2e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 3e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Architecture of firewall rules is as follows: 4e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * 5e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Chains go INPUT, FORWARD, OUTPUT then user chains. 6e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Each user chain starts with an ERROR node. 7e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Every chain ends with an unconditional jump: a RETURN for user chains, 8e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * and a POLICY for built-ins. 9e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher */ 10e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 113ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte/* (C) 1999 Paul ``Rusty'' Russell - Placed under the GNU GPL (See 123ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte * COPYING for details). 133ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte * (C) 2000-2003 by the Netfilter Core Team <coreteam@netfilter.org> 143ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte * 15fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte * 2003-Jun-20: Harald Welte <laforge@netfilter.org>: 163ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte * - Reimplementation of chain cache to use offsets instead of entries 17fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte * 2003-Jun-23: Harald Welte <laforge@netfilter.org>: 180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * - performance optimization, sponsored by Astaro AG (http://www.astaro.com/) 19fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte * don't rebuild the chain cache after every operation, instead fix it 20fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte * up after a ruleset change. 213ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte */ 22e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 23e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#ifndef IPT_LIB_DIR 24e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#define IPT_LIB_DIR "/usr/local/lib/iptables" 25e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#endif 26e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte#ifndef __OPTIMIZE__ 280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald WelteSTRUCT_ENTRY_TARGET * 290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald WelteGET_TARGET(STRUCT_ENTRY *e) 300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (void *)e + e->target_offset; 320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte#endif 340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 35e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic int sockfd = -1; 36e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic void *iptc_fn = NULL; 37e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 38e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic const char *hooknames[] 3979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell= { [HOOK_PRE_ROUTING] "PREROUTING", 4079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell [HOOK_LOCAL_IN] "INPUT", 4179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell [HOOK_FORWARD] "FORWARD", 4279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell [HOOK_LOCAL_OUT] "OUTPUT", 4310758b743d6aa076ebe2c3e8f855e73826841e71Rusty Russell [HOOK_POST_ROUTING] "POSTROUTING", 4410758b743d6aa076ebe2c3e8f855e73826841e71Rusty Russell#ifdef HOOK_DROPPING 4510758b743d6aa076ebe2c3e8f855e73826841e71Rusty Russell [HOOK_DROPPING] "DROPPING" 4610758b743d6aa076ebe2c3e8f855e73826841e71Rusty Russell#endif 47e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher}; 48e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 49e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstruct counter_map 50e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 51e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher enum { 52e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher COUNTER_MAP_NOMAP, 53e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher COUNTER_MAP_NORMAL_MAP, 541cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte COUNTER_MAP_ZEROED, 551cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte COUNTER_MAP_SET 56e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } maptype; 57e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int mappos; 58e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher}; 59e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 60e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Convenience structures */ 61e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstruct ipt_error_target 62e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 6379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY_TARGET t; 6479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell char error[TABLE_MAXNAMELEN]; 65e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher}; 66e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 670113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestruct chain_cache 6830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell{ 6979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell char name[TABLE_MAXNAMELEN]; 700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* This is the first rule in chain. */ 710113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int start_off; 720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Last rule in chain */ 730113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int end_off; 7430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell}; 7530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 7679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellSTRUCT_TC_HANDLE 77e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 78e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Have changes been made? */ 79e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher int changed; 800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Size in here reflects original state. */ 8179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_GETINFO info; 82e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct counter_map *counter_map; 84e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Array of hook names */ 85e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher const char **hooknames; 86e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 8730fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell /* Cached position of chain heads (NULL = no cache). */ 8830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell unsigned int cache_num_chains; 8930fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell unsigned int cache_num_builtins; 9030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell struct chain_cache *cache_chain_heads; 9130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 9230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell /* Chain iterator: current chain cache entry. */ 9330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell struct chain_cache *cache_chain_iteration; 9430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 9530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell /* Rule iterator: terminal rule */ 9679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *cache_rule_end; 97175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell 98e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Number in here reflects current state. */ 99e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int new_number; 10079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_GET_ENTRIES entries; 101e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher}; 102e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 103175f64177743e5a417e98d483ef995bf7151f3bcRusty Russellstatic void 10479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellset_changed(TC_HANDLE_T h) 105175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell{ 106175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell h->changed = 1; 107175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell} 108175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell 109380ba5f3074a16fbaa8869d9594962d58b5f8608Harald Welte#ifdef IPTC_DEBUG 11079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellstatic void do_check(TC_HANDLE_T h, unsigned int line); 111849779c4adf8dd65c83fffb65e6b7898df2a55c6Rusty Russell#define CHECK(h) do { if (!getenv("IPTC_NO_CHECK")) do_check((h), __LINE__); } while(0) 11230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell#else 11330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell#define CHECK(h) 11430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell#endif 115e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 116e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic inline int 11779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellget_number(const STRUCT_ENTRY *i, 11879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY *seek, 119e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int *pos) 120e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 121e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (i == seek) 122e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 123e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher (*pos)++; 124e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 125e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 126e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 127e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic unsigned int 12879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellentry2index(const TC_HANDLE_T h, const STRUCT_ENTRY *seek) 129e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 130e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int pos = 0; 131e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 132725d97a79cf0b332ed45cb7d254915178328427dRusty Russell if (ENTRY_ITERATE(h->entries.entrytable, h->entries.size, 13379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell get_number, seek, &pos) == 0) { 134e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher fprintf(stderr, "ERROR: offset %i not an entry!\n", 135725d97a79cf0b332ed45cb7d254915178328427dRusty Russell (char *)seek - (char *)h->entries.entrytable); 136e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher abort(); 137e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 138e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return pos; 139e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 140e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 141e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic inline int 14279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellget_entry_n(STRUCT_ENTRY *i, 143e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int number, 144e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int *pos, 14579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY **pe) 146e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 147e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (*pos == number) { 148e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher *pe = i; 149e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 150e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 151e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher (*pos)++; 152e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 153e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 154e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 15579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellstatic STRUCT_ENTRY * 15679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellindex2entry(TC_HANDLE_T h, unsigned int index) 157e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 158e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int pos = 0; 15979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *ret = NULL; 160e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 161725d97a79cf0b332ed45cb7d254915178328427dRusty Russell ENTRY_ITERATE(h->entries.entrytable, h->entries.size, 16279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell get_entry_n, index, &pos, &ret); 163e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 164e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return ret; 165e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 166e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1670113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic inline STRUCT_ENTRY * 1680113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteget_entry(TC_HANDLE_T h, unsigned int offset) 169e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (STRUCT_ENTRY *)((char *)h->entries.entrytable + offset); 171e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 172e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1730113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic inline unsigned long 1740113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteentry2offset(const TC_HANDLE_T h, const STRUCT_ENTRY *e) 175e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1760113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (void *)e - (void *)h->entries.entrytable; 1770113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 1789e03380e9f78ae347ae4f3f041c4eca50348f2e8Harald Welte 1790113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic inline unsigned long 1800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteindex2offset(TC_HANDLE_T h, unsigned int index) 1810113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 1820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return entry2offset(h, index2entry(h, index)); 183e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 184e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1853ea8f40262386e6b1445a617841f28702fe74d9dHarald Weltestatic inline STRUCT_ENTRY * 1863ea8f40262386e6b1445a617841f28702fe74d9dHarald Welteoffset2entry(TC_HANDLE_T h, unsigned int offset) 1873ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte{ 1883ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte return (STRUCT_ENTRY *) ((void *)h->entries.entrytable+offset); 1893ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte} 1903ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte 1913ea8f40262386e6b1445a617841f28702fe74d9dHarald Weltestatic inline unsigned int 1923ea8f40262386e6b1445a617841f28702fe74d9dHarald Welteoffset2index(const TC_HANDLE_T h, unsigned int offset) 1933ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte{ 1943ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte return entry2index(h, offset2entry(h, offset)); 1953ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte} 1963ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte 1973ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte 1980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic const char * 1990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteget_errorlabel(TC_HANDLE_T h, unsigned int offset) 2000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 2010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *e; 2020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 2030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = get_entry(h, offset); 2040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (strcmp(GET_TARGET(e)->u.user.name, ERROR_TARGET) != 0) { 2050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte fprintf(stderr, "ERROR: offset %u not an error node!\n", 2060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset); 2070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte abort(); 2080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 2090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 2100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (const char *)GET_TARGET(e)->data; 2110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 212e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 213e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Allocate handle of given size */ 21479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellstatic TC_HANDLE_T 2150113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltealloc_handle(const char *tablename, unsigned int size, unsigned int num_rules) 216e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 217e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher size_t len; 21879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T h; 219e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 22079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell len = sizeof(STRUCT_TC_HANDLE) 221e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher + size 222e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher + num_rules * sizeof(struct counter_map); 223e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 224e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if ((h = malloc(len)) == NULL) { 225e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOMEM; 226e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 227e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 228e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 229e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher h->changed = 0; 2300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains = 0; 2310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads = NULL; 2320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->counter_map = (void *)h 2330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte + sizeof(STRUCT_TC_HANDLE) 2340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte + size; 235e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher strcpy(h->info.name, tablename); 236e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher strcpy(h->entries.name, tablename); 237e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 238e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return h; 239e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 240e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 24179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_HANDLE_T 24279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_INIT(const char *tablename) 243e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 24479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T h; 24579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_GETINFO info; 2460113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int i; 247e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher int tmp; 248e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher socklen_t s; 249e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 25079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_INIT; 251e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 252e560fd604284180f3ab522993c5b8e6f424ef1d9Martin Josefsson if (sockfd != -1) { 253366454bc69f781fdafc3a30eb6dd77155ee4efb6Harald Welte close(sockfd); 254e560fd604284180f3ab522993c5b8e6f424ef1d9Martin Josefsson sockfd = -1; 255e560fd604284180f3ab522993c5b8e6f424ef1d9Martin Josefsson } 256366454bc69f781fdafc3a30eb6dd77155ee4efb6Harald Welte 257841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson if (strlen(tablename) >= TABLE_MAXNAMELEN) { 258841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson errno = EINVAL; 259841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson return NULL; 260841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson } 261841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson 26279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell sockfd = socket(TC_AF, SOCK_RAW, IPPROTO_RAW); 263e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (sockfd < 0) 264e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 265e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 266e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher s = sizeof(info); 267841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson 268e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher strcpy(info.name, tablename); 26979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (getsockopt(sockfd, TC_IPPROTO, SO_GET_INFO, &info, &s) < 0) 270e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 271e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 2720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((h = alloc_handle(info.name, info.size, info.num_entries)) 273841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson == NULL) { 274841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson close(sockfd); 275e560fd604284180f3ab522993c5b8e6f424ef1d9Martin Josefsson sockfd = -1; 276e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 277841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson } 278e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 279e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Too hard --RR */ 280e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#if 0 281e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher sprintf(pathname, "%s/%s", IPT_LIB_DIR, info.name); 282e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher dynlib = dlopen(pathname, RTLD_NOW); 283e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!dynlib) { 284e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 285e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 286e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 287e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher h->hooknames = dlsym(dynlib, "hooknames"); 288e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!h->hooknames) { 289e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 290e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 291e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 292e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#else 293e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher h->hooknames = hooknames; 294e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#endif 295e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 296e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Initialize current state */ 297e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher h->info = info; 2980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->new_number = h->info.num_entries; 2990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < h->info.num_entries; i++) 3000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->counter_map[i] 3010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = ((struct counter_map){COUNTER_MAP_NORMAL_MAP, i}); 3020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 303e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher h->entries.size = h->info.size; 304e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 30579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell tmp = sizeof(STRUCT_GET_ENTRIES) + h->info.size; 306e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 30779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (getsockopt(sockfd, TC_IPPROTO, SO_GET_ENTRIES, &h->entries, 308e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher &tmp) < 0) { 309841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson close(sockfd); 310e560fd604284180f3ab522993c5b8e6f424ef1d9Martin Josefsson sockfd = -1; 311e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(h); 312e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 313e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 3147e53bf9c2a697abdb6f1385557338423a86612a3Rusty Russell 315e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher CHECK(h); 316e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return h; 317e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 318e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 319841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefssonvoid 320841e4aed2349046eb2c0b1375139c06569a93bd0Martin JosefssonTC_FREE(TC_HANDLE_T *h) 321841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson{ 322841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson close(sockfd); 323e560fd604284180f3ab522993c5b8e6f424ef1d9Martin Josefsson sockfd = -1; 3240113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*h)->cache_chain_heads) 3250113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte free((*h)->cache_chain_heads); 326841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson free(*h); 327841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson *h = NULL; 328841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson} 329841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson 330e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic inline int 33179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellprint_match(const STRUCT_ENTRY_MATCH *m) 332e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 333228e98dd6303af11925235af4cf3c3ec450f3f41Rusty Russell printf("Match name: `%s'\n", m->u.user.name); 334e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 335e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 336e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 33779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellstatic int dump_entry(STRUCT_ENTRY *e, const TC_HANDLE_T handle); 33879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell 339e6869a8f59d779ff4d5a0984c86d80db7078496Marc Bouchervoid 34079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_DUMP_ENTRIES(const TC_HANDLE_T handle) 341e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 342e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher CHECK(handle); 343e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 344e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher printf("libiptc v%s. %u entries, %u bytes.\n", 34580fe35d6339b53a12ddaec41885613e4e37ed031Harald Welte IPTABLES_VERSION, 346e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher handle->new_number, handle->entries.size); 347e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher printf("Table `%s'\n", handle->info.name); 348e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher printf("Hooks: pre/in/fwd/out/post = %u/%u/%u/%u/%u\n", 34967088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.hook_entry[HOOK_PRE_ROUTING], 35067088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.hook_entry[HOOK_LOCAL_IN], 35167088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.hook_entry[HOOK_FORWARD], 35267088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.hook_entry[HOOK_LOCAL_OUT], 35367088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.hook_entry[HOOK_POST_ROUTING]); 354e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher printf("Underflows: pre/in/fwd/out/post = %u/%u/%u/%u/%u\n", 35567088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.underflow[HOOK_PRE_ROUTING], 35667088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.underflow[HOOK_LOCAL_IN], 35767088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.underflow[HOOK_FORWARD], 35867088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.underflow[HOOK_LOCAL_OUT], 35967088e73ce7707229c56987868f112051defca5aRusty Russell handle->info.underflow[HOOK_POST_ROUTING]); 360e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 361725d97a79cf0b332ed45cb7d254915178328427dRusty Russell ENTRY_ITERATE(handle->entries.entrytable, handle->entries.size, 36279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell dump_entry, handle); 363e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 364e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 36530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell/* Returns 0 if not hook entry, else hooknumber + 1 */ 36630fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russellstatic inline unsigned int 36779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellis_hook_entry(STRUCT_ENTRY *e, TC_HANDLE_T h) 36830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell{ 36930fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell unsigned int i; 37030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 37179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell for (i = 0; i < NUMHOOKS; i++) { 37230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if ((h->info.valid_hooks & (1 << i)) 37330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell && get_entry(h, h->info.hook_entry[i]) == e) 37430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return i+1; 37530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell } 37630fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return 0; 37730fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell} 37830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 3790113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic inline int 3800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteadd_chain(STRUCT_ENTRY *e, TC_HANDLE_T h, STRUCT_ENTRY **prev) 3810113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 3820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int builtin; 3830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 3840113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Last entry. End it. */ 3850113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (entry2offset(h, e) + e->next_offset == h->entries.size) { 3860113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* This is the ERROR node at end of the table */ 3870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads[h->cache_num_chains-1].end_off = 3880113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte entry2offset(h, *prev); 3890113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 3900113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 3910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 3920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* We know this is the start of a new chain if it's an ERROR 3930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte target, or a hook entry point */ 3940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (strcmp(GET_TARGET(e)->u.user.name, ERROR_TARGET) == 0) { 3950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* prev was last entry in previous chain */ 3960113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads[h->cache_num_chains-1].end_off 3970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = entry2offset(h, *prev); 3980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 3990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strcpy(h->cache_chain_heads[h->cache_num_chains].name, 4000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (const char *)GET_TARGET(e)->data); 4010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads[h->cache_num_chains].start_off 4020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = entry2offset(h, (void *)e + e->next_offset); 4030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains++; 4040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } else if ((builtin = is_hook_entry(e, h)) != 0) { 4050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (h->cache_num_chains > 0) 4060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* prev was last entry in previous chain */ 4070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads[h->cache_num_chains-1].end_off 4080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = entry2offset(h, *prev); 4090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strcpy(h->cache_chain_heads[h->cache_num_chains].name, 4110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->hooknames[builtin-1]); 4120113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads[h->cache_num_chains].start_off 4130113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = entry2offset(h, (void *)e); 4140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains++; 4150113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 4160113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4170113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte *prev = e; 4180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 4190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 420e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 42130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russellstatic int alphasort(const void *a, const void *b) 42230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell{ 42330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return strcmp(((struct chain_cache *)a)->name, 42430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell ((struct chain_cache *)b)->name); 42530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell} 4260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic int populate_cache(TC_HANDLE_T h) 4280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 4290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int i; 4300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *prev; 4310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* # chains < # rules / 2 + num builtins - 1 */ 4330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads = malloc((h->new_number / 2 + 4) 4340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * sizeof(struct chain_cache)); 4350113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!h->cache_chain_heads) { 4360113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = ENOMEM; 4370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 4380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 4390113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4400113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains = 0; 4410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_builtins = 0; 4420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4430113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Count builtins */ 4440113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < NUMHOOKS; i++) { 4450113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (h->info.valid_hooks & (1 << i)) 4460113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_builtins++; 4470113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 4480113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4490113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte prev = NULL; 4500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ENTRY_ITERATE(h->entries.entrytable, h->entries.size, 4510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte add_chain, h, &prev); 4520113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte qsort(h->cache_chain_heads + h->cache_num_builtins, 4540113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains - h->cache_num_builtins, 4550113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte sizeof(struct chain_cache), alphasort); 4560113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4570113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 1; 4580113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 4590113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4600113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic int 4610113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltecorrect_cache(TC_HANDLE_T h, unsigned int offset, int delta) 4620113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 4630113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int i; /* needs to be signed because deleting first 4640113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chain can make it drop to -1 */ 4650113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4660113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!delta) 4670113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 1; 4680113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4690113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < h->cache_num_chains; i++) { 4700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *cc = &h->cache_chain_heads[i]; 4710113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (delta < 0) { 4730113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* take care about deleted chains */ 4740113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (cc->start_off > offset+delta 4750113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte && cc->end_off < offset) { 4760113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* this chain is within the deleted range, 4770113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * let's remove it from the cache */ 4780113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte void *start; 4790113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int size; 4800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4810113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains--; 4820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* no need for memmove since we are 4840113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * removing the last entry */ 4850113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (i >= h->cache_num_chains) 4860113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte continue; 4870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4880113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte start = &h->cache_chain_heads[i+1]; 4890113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte size = (h->cache_num_chains-i) 4900113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * sizeof(struct chain_cache); 4910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memmove(cc, start, size); 4920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 4930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* iterate over same index again, since 4940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * it is now a different chain */ 4950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte i--; 4960113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte continue; 4970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 4980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 4990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (cc->start_off > offset) 5010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte cc->start_off += delta; 5020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (cc->end_off >= offset) 5040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte cc->end_off += delta; 5050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 5060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* HW_FIXME: sorting might be needed, but just in case a new chain was 5070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * added */ 5080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 1; 5100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 5110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5120113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic int 5130113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteadd_chain_cache(TC_HANDLE_T h, const char *name, unsigned int start_off, 5140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int end_off) 5150113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 5160113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *ccs = realloc(h->cache_chain_heads, 5170113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (h->new_number / 2 + 4 + 1) 5180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * sizeof(struct chain_cache)); 5190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *newcc; 5200113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5210113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!ccs) 5220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 5230113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5240113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_chain_heads = ccs; 5250113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newcc = &h->cache_chain_heads[h->cache_num_chains]; 5260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte h->cache_num_chains++; 5270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strncpy(newcc->name, name, TABLE_MAXNAMELEN-1); 529073df8feb0a8c4023ce40138e519ac9b341b1ca2Karsten Desler newcc->name[TABLE_MAXNAMELEN-1] = '\0'; 5300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newcc->start_off = start_off; 5310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newcc->end_off = end_off; 5320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 1; 5340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 5350113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5360113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte/* Returns cache ptr if found, otherwise NULL. */ 5370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic struct chain_cache * 5380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltefind_label(const char *name, TC_HANDLE_T handle) 5390113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 5400113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int i; 5410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (handle->cache_chain_heads == NULL 5430113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte && !populate_cache(handle)) 5440113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return NULL; 5450113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5460113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* FIXME: Linear search through builtins, then binary --RR */ 5470113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < handle->cache_num_chains; i++) { 5480113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (strcmp(handle->cache_chain_heads[i].name, name) == 0) 5490113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return &handle->cache_chain_heads[i]; 5500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 5510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 5520113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return NULL; 5530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 554e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 555e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Does this chain exist? */ 55679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellint TC_IS_CHAIN(const char *chain, const TC_HANDLE_T handle) 557e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 55830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return find_label(chain, handle) != NULL; 559e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 560e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 561e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Returns the position of the final (ie. unconditional) element. */ 562e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic unsigned int 56379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellget_chain_end(const TC_HANDLE_T handle, unsigned int start) 564e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 565e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int last_off, off; 56679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *e; 567e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 568e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher last_off = start; 569e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher e = get_entry(handle, start); 570e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 571e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Terminate when we meet a error label or a hook entry. */ 572e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher for (off = start + e->next_offset; 573e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher off < handle->entries.size; 574e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher last_off = off, off += e->next_offset) { 57579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY_TARGET *t; 576e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int i; 577e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 578e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher e = get_entry(handle, off); 579e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 580e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* We hit an entry point. */ 58179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell for (i = 0; i < NUMHOOKS; i++) { 582e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if ((handle->info.valid_hooks & (1 << i)) 583e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher && off == handle->info.hook_entry[i]) 584e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return last_off; 585e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 586e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 587e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* We hit a user chain label */ 58879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell t = GET_TARGET(e); 58967088e73ce7707229c56987868f112051defca5aRusty Russell if (strcmp(t->u.user.name, ERROR_TARGET) == 0) 590e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return last_off; 591e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 592e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* SHOULD NEVER HAPPEN */ 593e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher fprintf(stderr, "ERROR: Off end (%u) of chain from %u!\n", 594e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher handle->entries.size, off); 595e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher abort(); 596e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 597e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 59830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell/* Iterator functions to run through the chains. */ 599e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherconst char * 6008c700900e2a0cf87d7917cb62578583a60ad1210Philip BlundellTC_FIRST_CHAIN(TC_HANDLE_T *handle) 601e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 6020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->cache_chain_heads == NULL 6030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte && !populate_cache(*handle)) 6040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return NULL; 6050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 6060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->cache_chain_iteration 6070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = &(*handle)->cache_chain_heads[0]; 60830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 6090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (*handle)->cache_chain_iteration->name; 61030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell} 61130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 61230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell/* Iterator functions to run through the chains. Returns NULL at end. */ 61330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russellconst char * 61479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_NEXT_CHAIN(TC_HANDLE_T *handle) 61530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell{ 6160113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->cache_chain_iteration++; 61730fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 6180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->cache_chain_iteration - (*handle)->cache_chain_heads 6190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte == (*handle)->cache_num_chains) 62030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return NULL; 62130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 6220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (*handle)->cache_chain_iteration->name; 62330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell} 62430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 62530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell/* Get first rule in the given chain: NULL for empty chain. */ 62679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellconst STRUCT_ENTRY * 6278c700900e2a0cf87d7917cb62578583a60ad1210Philip BlundellTC_FIRST_RULE(const char *chain, TC_HANDLE_T *handle) 62830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell{ 6290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 63030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 63130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell c = find_label(chain, *handle); 63230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!c) { 63330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell errno = ENOENT; 63430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return NULL; 635e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 636e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 63730fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell /* Empty chain: single return/policy rule */ 6380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (c->start_off == c->end_off) 63930fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return NULL; 64030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 6410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->cache_rule_end = offset2entry(*handle, c->end_off); 6420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return offset2entry(*handle, c->start_off); 643e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 644e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 64530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell/* Returns NULL when rules run out. */ 64679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellconst STRUCT_ENTRY * 6478c700900e2a0cf87d7917cb62578583a60ad1210Philip BlundellTC_NEXT_RULE(const STRUCT_ENTRY *prev, TC_HANDLE_T *handle) 64830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell{ 6490113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((void *)prev + prev->next_offset 6500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte == (void *)(*handle)->cache_rule_end) 65130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell return NULL; 65230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 6530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return (void *)prev + prev->next_offset; 65430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell} 65530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 65630fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell#if 0 657e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* How many rules in this chain? */ 658e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherunsigned int 65979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_NUM_RULES(const char *chain, TC_HANDLE_T *handle) 660e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 661e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int off = 0; 66279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *start, *end; 663e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 664e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher CHECK(*handle); 665e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!find_label(&off, chain, *handle)) { 666e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 667e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return (unsigned int)-1; 668e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 669e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 670e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher start = get_entry(*handle, off); 671e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher end = get_entry(*handle, get_chain_end(*handle, off)); 672e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 673e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return entry2index(*handle, end) - entry2index(*handle, start); 674e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 675e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 676e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Get n'th rule in this chain. */ 67779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellconst STRUCT_ENTRY *TC_GET_RULE(const char *chain, 67879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell unsigned int n, 67979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 680e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 681e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int pos = 0, chainindex; 682e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 683e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher CHECK(*handle); 684e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!find_label(&pos, chain, *handle)) { 685e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 686e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 687e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 688e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 689e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher chainindex = entry2index(*handle, get_entry(*handle, pos)); 690e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 691e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return index2entry(*handle, chainindex + n); 692e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 69330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell#endif 694e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 69530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russellstatic const char * 69679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russelltarget_name(TC_HANDLE_T handle, const STRUCT_ENTRY *ce) 697e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 698e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher int spos; 6990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int labelidx; 7000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *jumpto; 701e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 70230fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell /* To avoid const warnings */ 70379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *e = (STRUCT_ENTRY *)ce; 70430fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell 70579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (strcmp(GET_TARGET(e)->u.user.name, STANDARD_TARGET) != 0) 70679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell return GET_TARGET(e)->u.user.name; 707e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 708e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Standard target: evaluate */ 70979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell spos = *(int *)GET_TARGET(e)->data; 710e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (spos < 0) { 71179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (spos == RETURN) 71279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell return LABEL_RETURN; 713e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher else if (spos == -NF_ACCEPT-1) 71479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell return LABEL_ACCEPT; 715e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher else if (spos == -NF_DROP-1) 71679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell return LABEL_DROP; 7172f4e5d92c73906e0dc2ae42fee5c05740528e92bJames Morris else if (spos == -NF_QUEUE-1) 71867088e73ce7707229c56987868f112051defca5aRusty Russell return LABEL_QUEUE; 719e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 7200113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte fprintf(stderr, "ERROR: off %lu/%u not a valid target (%i)\n", 7210113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte entry2offset(handle, e), handle->entries.size, 7220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte spos); 723e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher abort(); 724e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 725e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 7260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte jumpto = get_entry(handle, spos); 727e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 728e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Fall through rule */ 729e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (jumpto == (void *)e + e->next_offset) 730e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return ""; 731e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 732e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Must point to head of a chain: ie. after error rule */ 733e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher labelidx = entry2index(handle, jumpto) - 1; 734e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return get_errorlabel(handle, index2offset(handle, labelidx)); 735e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 736e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 737e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Returns a pointer to the target name of this position. */ 73879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellconst char *TC_GET_TARGET(const STRUCT_ENTRY *e, 73979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 740e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 741e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return target_name(*handle, e); 742e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 743e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 744e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Is this a built-in chain? Actually returns hook + 1. */ 745e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 74679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_BUILTIN(const char *chain, const TC_HANDLE_T handle) 747e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 748e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int i; 749e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 75079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell for (i = 0; i < NUMHOOKS; i++) { 751e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if ((handle->info.valid_hooks & (1 << i)) 752e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher && handle->hooknames[i] 753e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher && strcmp(handle->hooknames[i], chain) == 0) 754e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return i+1; 755e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 756e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 757e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 758e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 759e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Get the policy of a given built-in chain */ 760e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherconst char * 76179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_GET_POLICY(const char *chain, 76279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_COUNTERS *counters, 76379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 764e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 7650113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int start; 76679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *e; 767e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher int hook; 768e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 76979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell hook = TC_BUILTIN(chain, *handle); 7700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (hook != 0) 7710113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte start = (*handle)->info.hook_entry[hook-1]; 7720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte else 7739e03380e9f78ae347ae4f3f041c4eca50348f2e8Harald Welte return NULL; 7749e03380e9f78ae347ae4f3f041c4eca50348f2e8Harald Welte 7750113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = get_entry(*handle, get_chain_end(*handle, start)); 776e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher *counters = e->counters; 777e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 778e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return target_name(*handle, e); 779e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 780e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 7810113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic inline int 78279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellcorrect_verdict(STRUCT_ENTRY *e, 783725d97a79cf0b332ed45cb7d254915178328427dRusty Russell char *base, 784e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int offset, int delta_offset) 785e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 78679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_STANDARD_TARGET *t = (void *)GET_TARGET(e); 787725d97a79cf0b332ed45cb7d254915178328427dRusty Russell unsigned int curr = (char *)e - base; 788e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 789e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Trap: insert of fall-through rule. Don't change fall-through 790e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher verdict to jump-over-next-rule. */ 79179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (strcmp(t->target.u.user.name, STANDARD_TARGET) == 0 792e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher && t->verdict > (int)offset 793e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher && !(curr == offset && 794e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher t->verdict == curr + e->next_offset)) { 795e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher t->verdict += delta_offset; 796e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 797e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 798e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 799e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 800e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 801e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Adjusts standard verdict jump positions after an insertion/deletion. */ 802e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic int 80379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellset_verdict(unsigned int offset, int delta_offset, TC_HANDLE_T *handle) 804e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 805725d97a79cf0b332ed45cb7d254915178328427dRusty Russell ENTRY_ITERATE((*handle)->entries.entrytable, 80679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell (*handle)->entries.size, 807725d97a79cf0b332ed45cb7d254915178328427dRusty Russell correct_verdict, (char *)(*handle)->entries.entrytable, 80879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell offset, delta_offset); 809e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 810175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell set_changed(*handle); 811e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 812e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 813e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 8140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte/* If prepend is set, then we are prepending to a chain: if the 8150113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * insertion position is an entry point, keep the entry point. */ 8160113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic int 8170113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welteinsert_rules(unsigned int num_rules, unsigned int rules_size, 8180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte const STRUCT_ENTRY *insert, 8190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int offset, unsigned int num_rules_offset, 8200113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int prepend, 8210113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte TC_HANDLE_T *handle) 8220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 8230113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte TC_HANDLE_T newh; 8240113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_GETINFO newinfo; 8250113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int i; 8260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (offset >= (*handle)->entries.size) { 8280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = EINVAL; 8290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 8300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 8310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newinfo = (*handle)->info; 8330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Fix up entry points. */ 8350113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < NUMHOOKS; i++) { 8360113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Entry points to START of chain, so keep same if 8370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte inserting on at that point. */ 8380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->info.hook_entry[i] > offset) 8390113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newinfo.hook_entry[i] += rules_size; 8400113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Underflow always points to END of chain (policy), 8420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte so if something is inserted at same point, it 8430113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte should be advanced. */ 8440113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->info.underflow[i] >= offset) 8450113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newinfo.underflow[i] += rules_size; 8460113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 8470113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8480113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh = alloc_handle((*handle)->info.name, 8490113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->entries.size + rules_size, 8500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->new_number + num_rules); 8510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!newh) 8520113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 8530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->info = newinfo; 8540113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8550113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Copy pre... */ 8560113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy(newh->entries.entrytable, (*handle)->entries.entrytable,offset); 8570113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* ... Insert new ... */ 8580113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy((char *)newh->entries.entrytable + offset, insert, rules_size); 8590113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* ... copy post */ 8600113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy((char *)newh->entries.entrytable + offset + rules_size, 8610113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (char *)(*handle)->entries.entrytable + offset, 8620113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->entries.size - offset); 8630113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8640113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Move counter map. */ 8650113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Copy pre... */ 8660113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy(newh->counter_map, (*handle)->counter_map, 8670113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte sizeof(struct counter_map) * num_rules_offset); 8680113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* ... copy post */ 8690113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy(newh->counter_map + num_rules_offset + num_rules, 8700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->counter_map + num_rules_offset, 8710113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte sizeof(struct counter_map) * ((*handle)->new_number 8720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte - num_rules_offset)); 8730113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Set intermediates to no counter copy */ 8740113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < num_rules; i++) 8750113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->counter_map[num_rules_offset+i] 8760113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = ((struct counter_map){ COUNTER_MAP_SET, 0 }); 8770113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8780113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->new_number = (*handle)->new_number + num_rules; 8790113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->entries.size = (*handle)->entries.size + rules_size; 8800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->hooknames = (*handle)->hooknames; 8810113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->cache_chain_heads = (*handle)->cache_chain_heads; 8830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->cache_num_builtins = (*handle)->cache_num_builtins; 8840113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->cache_num_chains = (*handle)->cache_num_chains; 8850113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->cache_rule_end = (*handle)->cache_rule_end; 8860113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newh->cache_chain_iteration = (*handle)->cache_chain_iteration; 8870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!correct_cache(newh, offset, rules_size)) { 8880113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte free(newh); 8890113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 8900113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 8910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte free(*handle); 8930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte *handle = newh; 8940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return set_verdict(offset, rules_size, handle); 8960113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 8970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 8980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltestatic int 8990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Weltedelete_rules(unsigned int num_rules, unsigned int rules_size, 9000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int offset, unsigned int num_rules_offset, 9010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte TC_HANDLE_T *handle) 9020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte{ 9030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int i; 9040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (offset + rules_size > (*handle)->entries.size) { 9060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = EINVAL; 9070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 9080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 909fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte 9100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Fix up entry points. */ 9110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (i = 0; i < NUMHOOKS; i++) { 9120113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* In practice, we never delete up to a hook entry, 9130113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte since the built-in chains are always first, 9140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte so these two are never equal */ 9150113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->info.hook_entry[i] >= offset + rules_size) 9160113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->info.hook_entry[i] -= rules_size; 9170113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte else if ((*handle)->info.hook_entry[i] > offset) { 9180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte fprintf(stderr, "ERROR: Deleting entry %u %u %u\n", 9190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte i, (*handle)->info.hook_entry[i], offset); 9200113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte abort(); 9210113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 9220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9230113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Underflow points to policy (terminal) rule in 9240113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte built-in, so sequality is valid here (when deleting 9250113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte the last rule). */ 9260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->info.underflow[i] >= offset + rules_size) 9270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->info.underflow[i] -= rules_size; 9280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte else if ((*handle)->info.underflow[i] > offset) { 9290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte fprintf(stderr, "ERROR: Deleting uflow %u %u %u\n", 9300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte i, (*handle)->info.underflow[i], offset); 9310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte abort(); 9320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 9330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 9340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9350113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Move the rules down. */ 9360113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memmove((char *)(*handle)->entries.entrytable + offset, 9370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (char *)(*handle)->entries.entrytable + offset + rules_size, 9380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->entries.size - (offset + rules_size)); 9390113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9400113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Move the counter map down. */ 9410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memmove(&(*handle)->counter_map[num_rules_offset], 9420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte &(*handle)->counter_map[num_rules_offset + num_rules], 9430113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte sizeof(struct counter_map) 9440113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte * ((*handle)->new_number - (num_rules + num_rules_offset))); 9450113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9460113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Fix numbers */ 9470113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->new_number -= num_rules; 9480113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->entries.size -= rules_size; 9490113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Fix the chain cache */ 9510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!correct_cache(*handle, offset+rules_size, -(int)rules_size)) 9520113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 9530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 9540113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return set_verdict(offset, -(int)rules_size, handle); 9550113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte} 956e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 957e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic int 95879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellstandard_map(STRUCT_ENTRY *e, int verdict) 959e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 96079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_STANDARD_TARGET *t; 961e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 96279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell t = (STRUCT_STANDARD_TARGET *)GET_TARGET(e); 963e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 96467088e73ce7707229c56987868f112051defca5aRusty Russell if (t->target.u.target_size 9658c700900e2a0cf87d7917cb62578583a60ad1210Philip Blundell != ALIGN(sizeof(STRUCT_STANDARD_TARGET))) { 966e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EINVAL; 967e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 968e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 969e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* memset for memcmp convenience on delete/replace */ 97079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell memset(t->target.u.user.name, 0, FUNCTION_MAXNAMELEN); 97179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell strcpy(t->target.u.user.name, STANDARD_TARGET); 972e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher t->verdict = verdict; 973e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 974e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 975e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 9767e53bf9c2a697abdb6f1385557338423a86612a3Rusty Russell 977e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic int 97879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellmap_target(const TC_HANDLE_T handle, 97979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *e, 980e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int offset, 98179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY_TARGET *old) 982e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 9830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY_TARGET *t = GET_TARGET(e); 984e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 985e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Save old target (except data, which we don't change, except for 986e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher standard case, where we don't care). */ 987e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher *old = *t; 988e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 989e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Maybe it's empty (=> fall through) */ 990228e98dd6303af11925235af4cf3c3ec450f3f41Rusty Russell if (strcmp(t->u.user.name, "") == 0) 991e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return standard_map(e, offset + e->next_offset); 992e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Maybe it's a standard target name... */ 99379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell else if (strcmp(t->u.user.name, LABEL_ACCEPT) == 0) 994e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return standard_map(e, -NF_ACCEPT - 1); 99579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell else if (strcmp(t->u.user.name, LABEL_DROP) == 0) 996e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return standard_map(e, -NF_DROP - 1); 99767088e73ce7707229c56987868f112051defca5aRusty Russell else if (strcmp(t->u.user.name, LABEL_QUEUE) == 0) 998e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return standard_map(e, -NF_QUEUE - 1); 99979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell else if (strcmp(t->u.user.name, LABEL_RETURN) == 0) 100079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell return standard_map(e, RETURN); 100179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell else if (TC_BUILTIN(t->u.user.name, handle)) { 1002e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Can't jump to builtins. */ 1003e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EINVAL; 1004e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1005e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } else { 1006e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Maybe it's an existing chain name. */ 10070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 1008e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1009228e98dd6303af11925235af4cf3c3ec450f3f41Rusty Russell c = find_label(t->u.user.name, handle); 101030fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (c) 10113ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte return standard_map(e, c->start_off); 1012e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1013e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1014e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Must be a module? If not, kernel will reject... */ 1015e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* memset to all 0 for your memcmp convenience. */ 1016228e98dd6303af11925235af4cf3c3ec450f3f41Rusty Russell memset(t->u.user.name + strlen(t->u.user.name), 1017e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 0, 101879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell FUNCTION_MAXNAMELEN - strlen(t->u.user.name)); 1019e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1020e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1021e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1022e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic void 102379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellunmap_target(STRUCT_ENTRY *e, STRUCT_ENTRY_TARGET *old) 1024e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 102579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY_TARGET *t = GET_TARGET(e); 1026e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1027e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Save old target (except data, which we don't change, except for 1028e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher standard case, where we don't care). */ 1029e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher *t = *old; 1030e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1031e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 10320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte/* Insert the entry `fw' in chain `chain' into position `rulenum'. */ 1033e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 103479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_INSERT_ENTRY(const IPT_CHAINLABEL chain, 103579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY *e, 103679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell unsigned int rulenum, 103779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1038e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 10390113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int chainindex, offset; 10400113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY_TARGET old; 10410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 10420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *tmp; 10430113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 1044e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 104579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_INSERT_ENTRY; 104630fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(chain, *handle))) { 1047e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1048e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1049e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1050e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 10510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex = offset2index(*handle, c->start_off); 10520113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 10530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte tmp = index2entry(*handle, chainindex + rulenum); 10540113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!tmp || tmp > offset2entry(*handle, c->end_off)) { 1055e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = E2BIG; 1056e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1057e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 10580113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset = index2offset(*handle, chainindex + rulenum); 1059e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 10600113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Mapping target actually alters entry, but that's 10610113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte transparent to the caller. */ 10620113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!map_target(*handle, (STRUCT_ENTRY *)e, offset, &old)) 10630113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 1064e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 10650113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = insert_rules(1, e->next_offset, e, offset, 10660113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex + rulenum, rulenum == 0, handle); 10670113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unmap_target((STRUCT_ENTRY *)e, &old); 10680113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1069e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1070e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1071e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Atomically replace rule `rulenum' in `chain' with `fw'. */ 1072e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 107379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_REPLACE_ENTRY(const IPT_CHAINLABEL chain, 107479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY *e, 107579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell unsigned int rulenum, 107679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1077e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 10780113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int chainindex, offset; 10790113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY_TARGET old; 10800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 10810113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *tmp; 10820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 1083e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 108479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_REPLACE_ENTRY; 1085e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 108630fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(chain, *handle))) { 1087e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1088e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1089e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1090e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 10910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex = offset2index(*handle, c->start_off); 10920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 10930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte tmp = index2entry(*handle, chainindex + rulenum); 10940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!tmp || tmp >= offset2entry(*handle, c->end_off)) { 1095e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = E2BIG; 1096e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1097e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1098e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 10990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset = index2offset(*handle, chainindex + rulenum); 11000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Replace = delete and insert. */ 11010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!delete_rules(1, get_entry(*handle, offset)->next_offset, 11020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset, chainindex + rulenum, handle)) 1103e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1104e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 11050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!map_target(*handle, (STRUCT_ENTRY *)e, offset, &old)) 11060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 11070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 11080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = insert_rules(1, e->next_offset, e, offset, 11090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex + rulenum, 1, handle); 11100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unmap_target((STRUCT_ENTRY *)e, &old); 11110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1112e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1113e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 11140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte/* Append entry `fw' to chain `chain'. Equivalent to insert with 1115e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher rulenum = length of chain. */ 1116e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 111779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_APPEND_ENTRY(const IPT_CHAINLABEL chain, 111879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY *e, 111979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1120e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 11210113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 11220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY_TARGET old; 11230113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 1124e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 112579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_APPEND_ENTRY; 112630fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(chain, *handle))) { 1127e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1128e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1129e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1130e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 11310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!map_target(*handle, (STRUCT_ENTRY *)e, 11320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte c->end_off, &old)) 11330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 11340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 11350113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = insert_rules(1, e->next_offset, e, c->end_off, 11360113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset2index(*handle, c->end_off), 0, handle); 11370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unmap_target((STRUCT_ENTRY *)e, &old); 11380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1139e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1140e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1141e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic inline int 114279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellmatch_different(const STRUCT_ENTRY_MATCH *a, 1143edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell const unsigned char *a_elems, 1144edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell const unsigned char *b_elems, 1145edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell unsigned char **maskptr) 1146e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 114779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY_MATCH *b; 1148edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell unsigned int i; 1149e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1150e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Offset of b is the same as a. */ 115130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell b = (void *)b_elems + ((unsigned char *)a - a_elems); 1152e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1153228e98dd6303af11925235af4cf3c3ec450f3f41Rusty Russell if (a->u.match_size != b->u.match_size) 1154e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1155e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1156228e98dd6303af11925235af4cf3c3ec450f3f41Rusty Russell if (strcmp(a->u.user.name, b->u.user.name) != 0) 1157e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1158e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 115973ef09b21573ddb17ff75e5fd06fd4b52ec8ea40Rusty Russell *maskptr += ALIGN(sizeof(*a)); 1160edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell 116173ef09b21573ddb17ff75e5fd06fd4b52ec8ea40Rusty Russell for (i = 0; i < a->u.match_size - ALIGN(sizeof(*a)); i++) 1162edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell if (((a->data[i] ^ b->data[i]) & (*maskptr)[i]) != 0) 1163edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell return 1; 1164edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell *maskptr += i; 1165edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell return 0; 1166edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell} 1167edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell 1168edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russellstatic inline int 1169edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russelltarget_different(const unsigned char *a_targdata, 1170edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell const unsigned char *b_targdata, 1171edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell unsigned int tdatasize, 1172edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell const unsigned char *mask) 1173edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell{ 1174edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell unsigned int i; 1175edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell for (i = 0; i < tdatasize; i++) 1176edf14cf4b5edb148d7473f067d95e7bd1316900bRusty Russell if (((a_targdata[i] ^ b_targdata[i]) & mask[i]) != 0) 117790e712a00913fe2a2f885142439c392392dc08a8Rusty Russell return 1; 1178e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1179e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1180e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1181e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 118279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellstatic int 118379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellis_same(const STRUCT_ENTRY *a, 118479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY *b, 118579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell unsigned char *matchmask); 1186e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 11870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte/* Delete the first rule in `chain' which matches `fw'. */ 1188e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 118979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_DELETE_ENTRY(const IPT_CHAINLABEL chain, 119079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_ENTRY *origfw, 119179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell unsigned char *matchmask, 119279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1193e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 11940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int offset; 11950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 11960113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *e, *fw; 1197e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 119879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_DELETE_ENTRY; 119930fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(chain, *handle))) { 1200e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1201e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1202e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1203e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 12040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte fw = malloc(origfw->next_offset); 12050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (fw == NULL) { 12060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = ENOMEM; 12070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 12080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 12090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 12100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (offset = c->start_off; offset < c->end_off; 12110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset += e->next_offset) { 12120113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY_TARGET discard; 12130113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 12140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy(fw, origfw, origfw->next_offset); 12150113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 12160113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* FIXME: handle this in is_same --RR */ 12170113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!map_target(*handle, fw, offset, &discard)) { 12180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte free(fw); 12190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 12200113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 12210113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = get_entry(*handle, offset); 12220113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 12230113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte#if 0 12240113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte printf("Deleting:\n"); 12250113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte dump_entry(newe); 12260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte#endif 12270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (is_same(e, fw, matchmask)) { 12280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 12290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = delete_rules(1, e->next_offset, 12300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte offset, entry2index(*handle, e), 12310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte handle); 12320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte free(fw); 12330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1234e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1235e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1236e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 12370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte free(fw); 1238e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1239e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 12407e53bf9c2a697abdb6f1385557338423a86612a3Rusty Russell} 1241e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1242e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Delete the rule in position `rulenum' in `chain'. */ 1243e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 124479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_DELETE_NUM_ENTRY(const IPT_CHAINLABEL chain, 124579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell unsigned int rulenum, 124679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1247e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 12480113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int index; 12490113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 12500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *e; 12510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 1252e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 125379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_DELETE_NUM_ENTRY; 12540113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!(c = find_label(chain, *handle))) { 1255e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1256e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1257e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1258e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 12590113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte index = offset2index(*handle, c->start_off) + rulenum; 12600113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 12610113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (index >= offset2index(*handle, c->end_off)) { 1262e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = E2BIG; 1263e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1264e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1265e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 12660113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = index2entry(*handle, index); 12670113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (e == NULL) { 12680113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = EINVAL; 12690113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 12700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 1271e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 12720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = delete_rules(1, e->next_offset, entry2offset(*handle, e), 12730113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte index, handle); 12740113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1275e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1276e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1277e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Check the packet `fw' on chain `chain'. Returns the verdict, or 1278e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher NULL and sets errno. */ 1279e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherconst char * 128079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_CHECK_PACKET(const IPT_CHAINLABEL chain, 128179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *entry, 128279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1283e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1284e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOSYS; 1285e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return NULL; 1286e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1287e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1288e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Flushes the entries in the given chain (ie. empties chain). */ 1289e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 129079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_FLUSH_ENTRIES(const IPT_CHAINLABEL chain, TC_HANDLE_T *handle) 1291e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 12920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int startindex, endindex; 12930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *startentry, *endentry; 12940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 12950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 1296e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 12970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte iptc_fn = TC_FLUSH_ENTRIES; 12980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!(c = find_label(chain, *handle))) { 1299e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1300e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1301e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 13020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte startindex = offset2index(*handle, c->start_off); 13030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte endindex = offset2index(*handle, c->end_off); 13040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte startentry = offset2entry(*handle, c->start_off); 13050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte endentry = offset2entry(*handle, c->end_off); 1306e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 13070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = delete_rules(endindex - startindex, 13080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (char *)endentry - (char *)startentry, 13090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte c->start_off, startindex, 13100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte handle); 13110113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1312e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1313e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1314e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Zeroes the counters in a chain. */ 1315e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 131679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_ZERO_ENTRIES(const IPT_CHAINLABEL chain, TC_HANDLE_T *handle) 1317e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 13180113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int i, end; 13190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 13207e53bf9c2a697abdb6f1385557338423a86612a3Rusty Russell 132130fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(chain, *handle))) { 1322e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1323e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1324e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1325e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 13260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte i = offset2index(*handle, c->start_off); 13270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte end = offset2index(*handle, c->end_off); 13280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte for (; i <= end; i++) { 13300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->counter_map[i].maptype ==COUNTER_MAP_NORMAL_MAP) 13310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->counter_map[i].maptype = COUNTER_MAP_ZEROED; 1332e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1333175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell set_changed(*handle); 1334e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1335e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1336e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1337e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 13381cef74d943055668b5e356eebea877fdaa1ce3e0Harald WelteSTRUCT_COUNTERS * 13391cef74d943055668b5e356eebea877fdaa1ce3e0Harald WelteTC_READ_COUNTER(const IPT_CHAINLABEL chain, 13401cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte unsigned int rulenum, 13411cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte TC_HANDLE_T *handle) 13421cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte{ 13431cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte STRUCT_ENTRY *e; 13440113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 13450113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int chainindex, end; 13461cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13471cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte iptc_fn = TC_READ_COUNTER; 13481cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte CHECK(*handle); 13491cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13500113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!(c = find_label(chain, *handle))) { 13511cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte errno = ENOENT; 13521cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte return NULL; 13531cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte } 13541cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13550113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex = offset2index(*handle, c->start_off); 13560113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte end = offset2index(*handle, c->end_off); 13570113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13580113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (chainindex + rulenum > end) { 13590113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = E2BIG; 13600113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return NULL; 13610113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 13620113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13630113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = index2entry(*handle, chainindex + rulenum); 13640113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13650113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return &e->counters; 13661cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte} 13671cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13681cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welteint 13691cef74d943055668b5e356eebea877fdaa1ce3e0Harald WelteTC_ZERO_COUNTER(const IPT_CHAINLABEL chain, 13701cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte unsigned int rulenum, 13711cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte TC_HANDLE_T *handle) 13721cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte{ 13731cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte STRUCT_ENTRY *e; 13740113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 13750113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int chainindex, end; 13761cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13771cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte iptc_fn = TC_ZERO_COUNTER; 13781cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte CHECK(*handle); 13791cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13800113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!(c = find_label(chain, *handle))) { 13811cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte errno = ENOENT; 13821cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte return 0; 13831cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte } 13841cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 13850113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex = offset2index(*handle, c->start_off); 13860113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte end = offset2index(*handle, c->end_off); 13870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13880113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (chainindex + rulenum > end) { 13890113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = E2BIG; 13900113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 13910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 13920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = index2entry(*handle, chainindex + rulenum); 13940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 13950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if ((*handle)->counter_map[chainindex + rulenum].maptype 13960113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte == COUNTER_MAP_NORMAL_MAP) { 13970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->counter_map[chainindex + rulenum].maptype 13980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = COUNTER_MAP_ZEROED; 13990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 14001cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14011cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte set_changed(*handle); 14021cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14031cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte return 1; 14041cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte} 14051cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14061cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welteint 14071cef74d943055668b5e356eebea877fdaa1ce3e0Harald WelteTC_SET_COUNTER(const IPT_CHAINLABEL chain, 14081cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte unsigned int rulenum, 14091cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte STRUCT_COUNTERS *counters, 14101cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte TC_HANDLE_T *handle) 14111cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte{ 14121cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte STRUCT_ENTRY *e; 14130113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 14140113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int chainindex, end; 14151cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14161cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte iptc_fn = TC_SET_COUNTER; 14171cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte CHECK(*handle); 14181cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14190113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!(c = find_label(chain, *handle))) { 14201cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte errno = ENOENT; 14211cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte return 0; 14221cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte } 14230113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 14240113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chainindex = offset2index(*handle, c->start_off); 14250113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte end = offset2index(*handle, c->end_off); 14260113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 14270113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (chainindex + rulenum > end) { 14280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte errno = E2BIG; 14290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return 0; 14300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } 14310113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 14320113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = index2entry(*handle, chainindex + rulenum); 14330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 14340113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->counter_map[chainindex + rulenum].maptype 14350113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = COUNTER_MAP_SET; 14360113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 14370113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memcpy(&e->counters, counters, sizeof(STRUCT_COUNTERS)); 14381cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14391cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte set_changed(*handle); 14401cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 14411cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte return 1; 14421cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte} 14431cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 1444e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Creates a new chain. */ 1445e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* To create a chain, create two rules: error node and unconditional 1446e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * return. */ 1447e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 144879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_CREATE_CHAIN(const IPT_CHAINLABEL chain, TC_HANDLE_T *handle) 1449e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1450e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher int ret; 14510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct { 145279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY head; 1453e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher struct ipt_error_target name; 145479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY ret; 145579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_STANDARD_TARGET target; 14560113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte } newc; 14570113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int destination; 1458e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 145979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_CREATE_CHAIN; 1460e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1461e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* find_label doesn't cover built-in targets: DROP, ACCEPT, 1462e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher QUEUE, RETURN. */ 146330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (find_label(chain, *handle) 146479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || strcmp(chain, LABEL_DROP) == 0 146579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || strcmp(chain, LABEL_ACCEPT) == 0 146667088e73ce7707229c56987868f112051defca5aRusty Russell || strcmp(chain, LABEL_QUEUE) == 0 146779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || strcmp(chain, LABEL_RETURN) == 0) { 1468e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EEXIST; 1469e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1470e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1471e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 147279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (strlen(chain)+1 > sizeof(IPT_CHAINLABEL)) { 1473e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EINVAL; 1474e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1475e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1476e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 14770113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memset(&newc, 0, sizeof(newc)); 14780113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.head.target_offset = sizeof(STRUCT_ENTRY); 14790113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.head.next_offset 148067088e73ce7707229c56987868f112051defca5aRusty Russell = sizeof(STRUCT_ENTRY) 14818c700900e2a0cf87d7917cb62578583a60ad1210Philip Blundell + ALIGN(sizeof(struct ipt_error_target)); 14820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strcpy(newc.name.t.u.user.name, ERROR_TARGET); 14830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.name.t.u.target_size = ALIGN(sizeof(struct ipt_error_target)); 14840113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strcpy(newc.name.error, chain); 1485e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 14860113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.ret.target_offset = sizeof(STRUCT_ENTRY); 14870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.ret.next_offset 148867088e73ce7707229c56987868f112051defca5aRusty Russell = sizeof(STRUCT_ENTRY) 14898c700900e2a0cf87d7917cb62578583a60ad1210Philip Blundell + ALIGN(sizeof(STRUCT_STANDARD_TARGET)); 14900113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strcpy(newc.target.target.u.user.name, STANDARD_TARGET); 14910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.target.target.u.target_size 14928c700900e2a0cf87d7917cb62578583a60ad1210Philip Blundell = ALIGN(sizeof(STRUCT_STANDARD_TARGET)); 14930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte newc.target.verdict = RETURN; 1494fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte 14950113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte destination = index2offset(*handle, (*handle)->new_number -1); 1496fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte 14970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Add just before terminal entry */ 14980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = insert_rules(2, sizeof(newc), &newc.head, 14990113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte destination, 15000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->new_number - 1, 15010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 0, handle); 15020113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 15030113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte set_changed(*handle); 15040113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 15050113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* add chain cache info for this chain */ 15060113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte add_chain_cache(*handle, chain, 15070113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte destination+newc.head.next_offset, 15080113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte destination+newc.head.next_offset); 15090113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 15100113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1511e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1512e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1513e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic int 151479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellcount_ref(STRUCT_ENTRY *e, unsigned int offset, unsigned int *ref) 1515e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 151679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_STANDARD_TARGET *t; 1517e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 151879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (strcmp(GET_TARGET(e)->u.user.name, STANDARD_TARGET) == 0) { 151979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell t = (STRUCT_STANDARD_TARGET *)GET_TARGET(e); 1520e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1521e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (t->verdict == offset) 1522e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher (*ref)++; 1523e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1524e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1525e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1526e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1527e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1528e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Get the number of references to this chain. */ 1529e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 153079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_GET_REFERENCES(unsigned int *ref, const IPT_CHAINLABEL chain, 153179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1532e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 153330fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell struct chain_cache *c; 1534e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 153530fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(chain, *handle))) { 1536e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1537e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1538e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1539e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1540e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher *ref = 0; 1541725d97a79cf0b332ed45cb7d254915178328427dRusty Russell ENTRY_ITERATE((*handle)->entries.entrytable, 154279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell (*handle)->entries.size, 15433ea8f40262386e6b1445a617841f28702fe74d9dHarald Welte count_ref, c->start_off, ref); 1544e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1545e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1546e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1547e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Deletes a chain. */ 1548e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 154979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_DELETE_CHAIN(const IPT_CHAINLABEL chain, TC_HANDLE_T *handle) 1550e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 15510113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int labelidx, labeloff; 1552e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int references; 15530113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 15540113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte int ret; 15550113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte STRUCT_ENTRY *start; 1556e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 155779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (!TC_GET_REFERENCES(&references, chain, handle)) 1558e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 15597e53bf9c2a697abdb6f1385557338423a86612a3Rusty Russell 156079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_DELETE_CHAIN; 1561e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 156279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (TC_BUILTIN(chain, *handle)) { 1563e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EINVAL; 1564e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1565e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1566e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1567e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (references > 0) { 1568e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EMLINK; 1569e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1570e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1571e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 15720113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (!(c = find_label(chain, *handle))) { 1573e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1574e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1575e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1576e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 15770113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (c->start_off != c->end_off) { 1578e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOTEMPTY; 1579e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1580e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1581e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 15820113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Need label index: preceeds chain start */ 15830113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte labelidx = offset2index(*handle, c->start_off) - 1; 15840113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte labeloff = index2offset(*handle, labelidx); 15850113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 15860113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte start = offset2entry(*handle, c->start_off); 15870113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 15880113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte ret = delete_rules(2, 15890113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte get_entry(*handle, labeloff)->next_offset 15900113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte + start->next_offset, 15910113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte labeloff, labelidx, handle); 15920113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte return ret; 1593e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1594e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1595e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Renames a chain. */ 159679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellint TC_RENAME_CHAIN(const IPT_CHAINLABEL oldname, 159779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const IPT_CHAINLABEL newname, 159879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1599e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 16000113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int labeloff, labelidx; 16010113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte struct chain_cache *c; 1602e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher struct ipt_error_target *t; 1603e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 160479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_RENAME_CHAIN; 1605e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 16061de804642d4c8e9c71b7e225a1528fff15fa7faaHarald Welte /* find_label doesn't cover built-in targets: DROP, ACCEPT, 16071de804642d4c8e9c71b7e225a1528fff15fa7faaHarald Welte QUEUE, RETURN. */ 160830fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (find_label(newname, *handle) 160979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || strcmp(newname, LABEL_DROP) == 0 161079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || strcmp(newname, LABEL_ACCEPT) == 0 16111de804642d4c8e9c71b7e225a1528fff15fa7faaHarald Welte || strcmp(newname, LABEL_QUEUE) == 0 161279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || strcmp(newname, LABEL_RETURN) == 0) { 1613e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EEXIST; 1614e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1615e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1616e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 161730fd6e5d45e6013f4df10a226787c7a9f49369c1Rusty Russell if (!(c = find_label(oldname, *handle)) 161879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell || TC_BUILTIN(oldname, *handle)) { 1619e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOENT; 1620e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1621e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1622e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 162379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (strlen(newname)+1 > sizeof(IPT_CHAINLABEL)) { 1624e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EINVAL; 1625e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1626e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1627e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 16280113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* Need label index: preceeds chain start */ 16290113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte labelidx = offset2index(*handle, c->start_off) - 1; 16300113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte labeloff = index2offset(*handle, labelidx); 1631e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1632e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher t = (struct ipt_error_target *) 16330113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte GET_TARGET(get_entry(*handle, labeloff)); 1634e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1635e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher memset(t->error, 0, sizeof(t->error)); 1636e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher strcpy(t->error, newname); 1637fbc85236a6140918ab1d0fb0e07e2d72da46ce45Harald Welte 16380113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte /* update chain cache */ 16390113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte memset(c->name, 0, sizeof(c->name)); 16400113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte strcpy(c->name, newname); 16410113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 16420113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte set_changed(*handle); 16430113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte 1644e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1645e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1646e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1647e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Sets the policy on a built-in chain. */ 1648e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 164979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_SET_POLICY(const IPT_CHAINLABEL chain, 165079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const IPT_CHAINLABEL policy, 16511cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte STRUCT_COUNTERS *counters, 165279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell TC_HANDLE_T *handle) 1653e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1654e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int hook; 16550113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte unsigned int policyoff, ctrindex; 165679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_ENTRY *e; 165779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_STANDARD_TARGET *t; 1658e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 165979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell iptc_fn = TC_SET_POLICY; 1660e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Figure out which chain. */ 166179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell hook = TC_BUILTIN(chain, *handle); 1662e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (hook == 0) { 1663c8264991454b5e77279830736f80ea3153b6f814Marc Boucher errno = ENOENT; 1664e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1665e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } else 1666e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher hook--; 1667e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 16680113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte policyoff = get_chain_end(*handle, (*handle)->info.hook_entry[hook]); 16690113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte if (policyoff != (*handle)->info.underflow[hook]) { 16700113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte printf("ERROR: Policy for `%s' offset %u != underflow %u\n", 16710113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte chain, policyoff, (*handle)->info.underflow[hook]); 16729e03380e9f78ae347ae4f3f041c4eca50348f2e8Harald Welte return 0; 16739e03380e9f78ae347ae4f3f041c4eca50348f2e8Harald Welte } 16749e03380e9f78ae347ae4f3f041c4eca50348f2e8Harald Welte 16750113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte e = get_entry(*handle, policyoff); 16760113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte t = (STRUCT_STANDARD_TARGET *)GET_TARGET(e); 1677e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 167879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (strcmp(policy, LABEL_ACCEPT) == 0) 1679e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher t->verdict = -NF_ACCEPT - 1; 168079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell else if (strcmp(policy, LABEL_DROP) == 0) 1681e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher t->verdict = -NF_DROP - 1; 1682e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher else { 1683e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = EINVAL; 1684e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1685e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 16861cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 16871cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte ctrindex = entry2index(*handle, e); 16881cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 16891cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte if (counters) { 16901cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte /* set byte and packet counters */ 16911cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte memcpy(&e->counters, counters, sizeof(STRUCT_COUNTERS)); 16921cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 16930113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->counter_map[ctrindex].maptype 16940113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = COUNTER_MAP_SET; 16951cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 16961cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte } else { 16970113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte (*handle)->counter_map[ctrindex] 16980113fe75ff05e09e6f3d251534d9ae32e9aa717cHarald Welte = ((struct counter_map){ COUNTER_MAP_NOMAP, 0 }); 16991cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte } 17001cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 1701175f64177743e5a417e98d483ef995bf7151f3bcRusty Russell set_changed(*handle); 1702e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1703e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1704e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1705e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1706e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Without this, on gcc 2.7.2.3, we get: 170779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell libiptc.c: In function `TC_COMMIT': 1708e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher libiptc.c:833: fixed or forbidden register was spilled. 1709e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher This may be due to a compiler bug or to impossible asm 1710e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher statements or clauses. 1711e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher*/ 1712e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherstatic void 171379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russellsubtract_counters(STRUCT_COUNTERS *answer, 171479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_COUNTERS *a, 171579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell const STRUCT_COUNTERS *b) 1716e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1717e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher answer->pcnt = a->pcnt - b->pcnt; 1718e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher answer->bcnt = a->bcnt - b->bcnt; 1719e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1720e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1721e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 172279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_COMMIT(TC_HANDLE_T *handle) 1723e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1724e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Replace, then map back the counters. */ 172579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_REPLACE *repl; 172679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell STRUCT_COUNTERS_INFO *newcounters; 1727e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int i; 1728841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson size_t counterlen; 1729e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1730e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher CHECK(*handle); 1731841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson 1732841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson counterlen = sizeof(STRUCT_COUNTERS_INFO) 1733841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson + sizeof(STRUCT_COUNTERS) * (*handle)->new_number; 1734841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson 1735e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#if 0 173654c307e0ff401f40a6fe382af4ae5bff0f5b40baRusty Russell TC_DUMP_ENTRIES(*handle); 1737e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher#endif 1738e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1739e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Don't commit if nothing changed. */ 1740e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!(*handle)->changed) 1741e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher goto finished; 1742e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1743e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher repl = malloc(sizeof(*repl) + (*handle)->entries.size); 1744e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!repl) { 1745e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOMEM; 1746e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1747e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1748e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1749e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* These are the old counters we will get from kernel */ 175079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell repl->counters = malloc(sizeof(STRUCT_COUNTERS) 1751e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * (*handle)->info.num_entries); 1752e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!repl->counters) { 1753e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl); 1754e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOMEM; 1755e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1756e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 17577e53bf9c2a697abdb6f1385557338423a86612a3Rusty Russell 1758e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* These are the counters we're going to put back, later. */ 1759e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher newcounters = malloc(counterlen); 1760e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if (!newcounters) { 1761e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl->counters); 1762e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl); 1763e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher errno = ENOMEM; 1764e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1765e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1766e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1767e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher strcpy(repl->name, (*handle)->info.name); 1768e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher repl->num_entries = (*handle)->new_number; 1769e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher repl->size = (*handle)->entries.size; 1770e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher memcpy(repl->hook_entry, (*handle)->info.hook_entry, 1771e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher sizeof(repl->hook_entry)); 1772e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher memcpy(repl->underflow, (*handle)->info.underflow, 1773e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher sizeof(repl->underflow)); 1774e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher repl->num_counters = (*handle)->info.num_entries; 1775e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher repl->valid_hooks = (*handle)->info.valid_hooks; 1776725d97a79cf0b332ed45cb7d254915178328427dRusty Russell memcpy(repl->entries, (*handle)->entries.entrytable, 1777e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher (*handle)->entries.size); 1778e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 177979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (setsockopt(sockfd, TC_IPPROTO, SO_SET_REPLACE, repl, 1780e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher sizeof(*repl) + (*handle)->entries.size) < 0) { 1781e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl->counters); 1782e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl); 1783e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(newcounters); 1784e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1785e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1786e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1787e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Put counters back. */ 1788e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher strcpy(newcounters->name, (*handle)->info.name); 1789e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher newcounters->num_counters = (*handle)->new_number; 1790e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher for (i = 0; i < (*handle)->new_number; i++) { 1791e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int mappos = (*handle)->counter_map[i].mappos; 1792e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher switch ((*handle)->counter_map[i].maptype) { 1793e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher case COUNTER_MAP_NOMAP: 1794e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher newcounters->counters[i] 179579dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell = ((STRUCT_COUNTERS){ 0, 0 }); 1796e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher break; 1797e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1798e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher case COUNTER_MAP_NORMAL_MAP: 1799e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Original read: X. 1800e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Atomic read on replacement: X + Y. 1801e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Currently in kernel: Z. 1802e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Want in kernel: X + Y + Z. 1803e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * => Add in X + Y 1804e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * => Add in replacement read. 1805e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher */ 1806e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher newcounters->counters[i] = repl->counters[mappos]; 1807e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher break; 1808e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1809e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher case COUNTER_MAP_ZEROED: 1810e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* Original read: X. 1811e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Atomic read on replacement: X + Y. 1812e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Currently in kernel: Z. 1813e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * Want in kernel: Y + Z. 1814e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * => Add in Y. 1815e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher * => Add in (replacement read - original read). 1816e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher */ 1817e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher subtract_counters(&newcounters->counters[i], 1818e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher &repl->counters[mappos], 1819e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher &index2entry(*handle, i)->counters); 1820e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher break; 18211cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 18221cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte case COUNTER_MAP_SET: 18231cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte /* Want to set counter (iptables-restore) */ 18241cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 18251cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte memcpy(&newcounters->counters[i], 18261cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte &index2entry(*handle, i)->counters, 18271cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte sizeof(STRUCT_COUNTERS)); 18281cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte 18291cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte break; 1830e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1831e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 183262527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell 183362527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell#ifdef KERNEL_64_USERSPACE_32 183462527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell { 183562527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell /* Kernel will think that pointer should be 64-bits, and get 183662527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell padding. So we accomodate here (assumption: alignment of 183762527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell `counters' is on 64-bit boundary). */ 183862527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell u_int64_t *kernptr = (u_int64_t *)&newcounters->counters; 183962527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell if ((unsigned long)&newcounters->counters % 8 != 0) { 184062527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell fprintf(stderr, 184162527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell "counters alignment incorrect! Mail rusty!\n"); 184262527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell abort(); 184362527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell } 184462527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell *kernptr = newcounters->counters; 184554c307e0ff401f40a6fe382af4ae5bff0f5b40baRusty Russell } 184662527ce5f0ffaa5b18aa118f64c21af238ddc156Rusty Russell#endif /* KERNEL_64_USERSPACE_32 */ 1847e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 184879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell if (setsockopt(sockfd, TC_IPPROTO, SO_SET_ADD_COUNTERS, 184979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell newcounters, counterlen) < 0) { 1850e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl->counters); 1851e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl); 1852e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(newcounters); 1853e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 0; 1854e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1855e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1856e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl->counters); 1857e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(repl); 1858e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher free(newcounters); 1859e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1860e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher finished: 1861841e4aed2349046eb2c0b1375139c06569a93bd0Martin Josefsson TC_FREE(handle); 1862e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return 1; 1863e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1864e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1865e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Get raw socket. */ 1866e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherint 186779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_GET_RAW_SOCKET() 1868e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1869e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return sockfd; 1870e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1871e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1872e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher/* Translates errno numbers into more human-readable form than strerror. */ 1873e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucherconst char * 187479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty RussellTC_STRERROR(int err) 1875e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher{ 1876e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher unsigned int i; 1877e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher struct table_struct { 1878e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher void *fn; 1879e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher int err; 1880e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher const char *message; 1881e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } table [] = 18824ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { { TC_INIT, EPERM, "Permission denied (you must be root)" }, 188379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_INIT, EINVAL, "Module is wrong version" }, 18844ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { TC_INIT, ENOENT, 18854ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte "Table does not exist (do you need to insmod?)" }, 188679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_DELETE_CHAIN, ENOTEMPTY, "Chain is not empty" }, 188779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_DELETE_CHAIN, EINVAL, "Can't delete built-in chain" }, 188879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_DELETE_CHAIN, EMLINK, 1889e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher "Can't delete chain with references left" }, 189079dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_CREATE_CHAIN, EEXIST, "Chain already exists" }, 189179dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_INSERT_ENTRY, E2BIG, "Index of insertion too big" }, 189279dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_REPLACE_ENTRY, E2BIG, "Index of replacement too big" }, 189379dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_DELETE_NUM_ENTRY, E2BIG, "Index of deletion too big" }, 18941cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte { TC_READ_COUNTER, E2BIG, "Index of counter too big" }, 18951cef74d943055668b5e356eebea877fdaa1ce3e0Harald Welte { TC_ZERO_COUNTER, E2BIG, "Index of counter too big" }, 189679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_INSERT_ENTRY, ELOOP, "Loop found in table" }, 189779dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_INSERT_ENTRY, EINVAL, "Target problem" }, 1898e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* EINVAL for CHECK probably means bad interface. */ 189979dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_CHECK_PACKET, EINVAL, 1900c8264991454b5e77279830736f80ea3153b6f814Marc Boucher "Bad arguments (does that interface exist?)" }, 19014ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { TC_CHECK_PACKET, ENOSYS, 19024ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte "Checking will most likely never get implemented" }, 1903e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher /* ENOENT for DELETE probably means no matching rule */ 190479dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_DELETE_ENTRY, ENOENT, 1905c8264991454b5e77279830736f80ea3153b6f814Marc Boucher "Bad rule (does a matching rule exist in that chain?)" }, 190679dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_SET_POLICY, ENOENT, 1907c8264991454b5e77279830736f80ea3153b6f814Marc Boucher "Bad built-in chain name" }, 190879dee0702b18c8ea1d1f7a2b1f6b29349466986bRusty Russell { TC_SET_POLICY, EINVAL, 1909c8264991454b5e77279830736f80ea3153b6f814Marc Boucher "Bad policy name" }, 19104ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte 19114ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { NULL, 0, "Incompatible with this kernel" }, 19124ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { NULL, ENOPROTOOPT, "iptables who? (do you need to insmod?)" }, 19134ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { NULL, ENOSYS, "Will be implemented real soon. I promise ;)" }, 19144ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { NULL, ENOMEM, "Memory allocation problem" }, 19154ccfa630d9a588d4b852abef8bc467642427c8cfHarald Welte { NULL, ENOENT, "No chain/target/match by that name" }, 1916e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher }; 1917e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1918e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher for (i = 0; i < sizeof(table)/sizeof(struct table_struct); i++) { 1919e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher if ((!table[i].fn || table[i].fn == iptc_fn) 1920e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher && table[i].err == err) 1921e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return table[i].message; 1922e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher } 1923e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher 1924e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher return strerror(err); 1925e6869a8f59d779ff4d5a0984c86d80db7078496Marc Boucher} 1926