X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fsubnet.c;h=4e528e1af387f9e83cd6944a2ed00d5bb127c0a7;hb=6debc6c79ba385d35f646e0958f84ace5b8f4b4d;hp=6745e2f4766790f0be304aa3a017715952bab7ac;hpb=f0aa9641e82fb6e09c1e485366d14dddaa7f7c36;p=tinc diff --git a/src/subnet.c b/src/subnet.c index 6745e2f4..fe7d23ef 100644 --- a/src/subnet.c +++ b/src/subnet.c @@ -1,7 +1,7 @@ /* subnet.c -- handle subnet lookups and lists - Copyright (C) 2000-2002 Guus Sliepen , - 2000-2002 Ivo Timmermans + Copyright (C) 2000-2017 Guus Sliepen , + 2000-2005 Ivo Timmermans This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -13,409 +13,392 @@ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. - You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. - - $Id: subnet.c,v 1.1.2.29 2002/02/10 21:57:54 guus Exp $ + You should have received a copy of the GNU General Public License along + with this program; if not, write to the Free Software Foundation, Inc., + 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */ -#include "config.h" - -#include -#include -#include -#include -#include -#include -#include +#include "system.h" -#include "conf.h" +#include "splay_tree.h" +#include "control_common.h" +#include "crypto.h" +#include "hash.h" +#include "logger.h" #include "net.h" +#include "netutl.h" #include "node.h" +#include "script.h" #include "subnet.h" -#include "system.h" - -#include -#include -#include +#include "xalloc.h" /* lists type of subnet */ +uint32_t hash_seed; +splay_tree_t subnet_tree = { + .compare = (splay_compare_t) subnet_compare, + .delete = (splay_action_t) free_subnet, +}; -avl_tree_t *subnet_tree; +/* Subnet lookup cache */ -/* Subnet comparison */ +static uint32_t wrapping_add32(uint32_t a, uint32_t b) { + return (uint32_t)((uint64_t)a + b); +} -int subnet_compare_mac(subnet_t *a, subnet_t *b) -{ -cp - return memcmp(&a->net.mac.address, &b->net.mac.address, sizeof(mac_t)); +static uint32_t wrapping_mul32(uint32_t a, uint32_t b) { + return (uint32_t)((uint64_t)a * b); } -int subnet_compare_ipv4(subnet_t *a, subnet_t *b) -{ -cp - /* We compare as if a subnet is a number that equals (address << 32 + netmask). */ - - if(a->net.ipv4.address < b->net.ipv4.address) - return -1; - else if(a->net.ipv4.address > b->net.ipv4.address) - return 1; - - if(a->net.ipv4.mask < b->net.ipv4.mask) - return -1; - else if(a->net.ipv4.mask > b->net.ipv4.mask) - return 1; - - return 0; +static uint32_t hash_function_ipv4_t(const ipv4_t *p) { + /* + This basic hash works because + a) Most IPv4 networks routed via tinc are not /0 + b) Most IPv4 networks have more unique low order bits + */ + uint16_t *halfwidth = (uint16_t *)p; + uint32_t hash = hash_seed; + +#if __BYTE_ORDER == __LITTLE_ENDIAN + // 10.0.x.x/16 part + hash = wrapping_add32(hash, wrapping_mul32(halfwidth[1], 0x9e370001U)); + + // x.x.0.[0-255] part +#if SUBNET_HASH_SIZE >= 0x10000 + return hash ^ halfwidth[0]; +#else + // ensure that we have a /24 with no collisions on 32bit + return hash ^ ntohs(halfwidth[0]); +#endif // _____LP64_____ +#else + // 10.0.x.x/16 part + hash = wrapping_add32(hash, wrapping_mul32(halfwidth[0], 0x9e370001U)); + + // x.x.0.[0-255] part (ntohs is nop on big endian) + return hash ^ halfwidth[1]; +#endif // __BYTE_ORDER == __LITTLE_ENDIAN } -int subnet_compare_ipv6(subnet_t *a, subnet_t *b) -{ - int result; -cp - /* Same as ipv4 case, but with nasty 128 bit addresses */ - - result = memcmp(a->net.ipv6.address.x, b->net.ipv6.address.x, sizeof(ipv6_t)); - - if(result) - return result; - - result = memcmp(a->net.ipv6.mask.x, b->net.ipv6.mask.x, sizeof(ipv6_t)); - - if(result) - return result; - - return 0; + +static uint32_t hash_function_ipv6_t(const ipv6_t *p) { + uint32_t *fullwidth = (uint32_t *)p; + uint32_t hash = hash_seed; + + for(int i = 0; i < 4; i++) { + hash = wrapping_add32(hash, fullwidth[i]); + hash = wrapping_mul32(hash, 0x9e370001U); + } + + return hash; } -int subnet_compare(subnet_t *a, subnet_t *b) -{ - int x; -cp - x = a->type - b->type; - if(x) - return x; - - switch(a->type) - { - case SUBNET_MAC: - return subnet_compare_mac(a, b); - case SUBNET_IPV4: - return subnet_compare_ipv4(a, b); - case SUBNET_IPV6: - return subnet_compare_ipv6(a, b); - default: - syslog(LOG_ERR, _("subnet_compare() was called with unknown subnet type %d, exitting!"), a->type); - cp_trace(); - exit(0); - } - - return 0; +static uint32_t hash_function_mac_t(const mac_t *p) { + uint16_t *halfwidth = (uint16_t *)p; + uint32_t hash = hash_seed; + + for(int i = 0; i < 3; i++) { + hash = wrapping_add32(hash, halfwidth[i]); + hash = wrapping_mul32(hash, 0x9e370001U); + } + + return hash; } -/* Initialising trees */ +hash_define(ipv4_t, SUBNET_HASH_SIZE) +hash_define(ipv6_t, SUBNET_HASH_SIZE) +hash_define(mac_t, SUBNET_HASH_SIZE) + +hash_new(ipv4_t, ipv4_cache); +hash_new(ipv6_t, ipv6_cache); +hash_new(mac_t, mac_cache); + + +void subnet_cache_flush_table(subnet_type_t stype) { + // NOTE: a subnet type of SUBNET_TYPES can be used to clear all hash tables + + if(stype != SUBNET_IPV6) { // ipv4 + hash_clear(ipv4_t, &ipv4_cache); + } -void init_subnets(void) -{ -cp - subnet_tree = avl_alloc_tree((avl_compare_t)subnet_compare, (avl_action_t)free_subnet); -cp + if(stype != SUBNET_IPV4) { // ipv6 + hash_clear(ipv6_t, &ipv6_cache); + } + + hash_clear(mac_t, &mac_cache); } -void exit_subnets(void) -{ -cp - avl_delete_tree(subnet_tree); -cp +/* Initialising trees */ + +void init_subnets(void) { + hash_seed = prng(UINT32_MAX); + + // tables need to be cleared on startup + subnet_cache_flush_tables(); } -avl_tree_t *new_subnet_tree(void) -{ -cp - return avl_alloc_tree((avl_compare_t)subnet_compare, NULL); -cp +void exit_subnets(void) { + splay_empty_tree(&subnet_tree); + subnet_cache_flush_tables(); } -void free_subnet_tree(avl_tree_t *subnet_tree) -{ -cp - avl_delete_tree(subnet_tree); -cp +void init_subnet_tree(splay_tree_t *tree) { + memset(tree, 0, sizeof(*tree)); + tree->compare = (splay_compare_t) subnet_compare; } /* Allocating and freeing space for subnets */ -subnet_t *new_subnet(void) -{ -cp - return (subnet_t *)xmalloc(sizeof(subnet_t)); +subnet_t *new_subnet(void) { + return xzalloc(sizeof(subnet_t)); } -void free_subnet(subnet_t *subnet) -{ -cp - free(subnet); +void free_subnet(subnet_t *subnet) { + free(subnet); } -/* Adding and removing subnets */ +void subnet_cache_flush_tables(void) { + // flushes all the tables + hash_clear(ipv4_t, &ipv4_cache); + hash_clear(ipv6_t, &ipv6_cache); + hash_clear(mac_t, &mac_cache); +} -void subnet_add(node_t *n, subnet_t *subnet) -{ -cp - subnet->owner = n; +void subnet_cache_flush(subnet_t *subnet) { + switch(subnet->type) { + case SUBNET_IPV4: + if(subnet->net.ipv4.prefixlength == 32) { + hash_delete(ipv4_t, &ipv4_cache, &subnet->net.ipv4.address); + return; + } - avl_insert(subnet_tree, subnet); -cp - avl_insert(n->subnet_tree, subnet); -cp -} + break; + + case SUBNET_IPV6: + if(subnet->net.ipv4.prefixlength == 128) { + hash_delete(ipv6_t, &ipv6_cache, &subnet->net.ipv6.address); + return; + } -void subnet_del(node_t *n, subnet_t *subnet) -{ -cp - avl_delete(n->subnet_tree, subnet); -cp - avl_delete(subnet_tree, subnet); -cp + break; + + case SUBNET_MAC: + hash_delete(mac_t, &mac_cache, &subnet->net.mac.address); + return; + } + + subnet_cache_flush_table(subnet->type); } -/* Ascii representation of subnets */ - -subnet_t *str2net(char *subnetstr) -{ - int i, l; - subnet_t *subnet; - unsigned short int x[6]; -cp - subnet = new_subnet(); -cp - if(sscanf(subnetstr, "%hu.%hu.%hu.%hu/%d", - &x[0], - &x[1], - &x[2], - &x[3], - &subnet->net.ipv4.masklength) == 5) - { - subnet->type = SUBNET_IPV4; - subnet->net.ipv4.address = (((((x[0] << 8) + x[1]) << 8) + x[2]) << 8) + x[3]; - subnet->net.ipv4.mask = ~((1 << (32 - subnet->net.ipv4.masklength)) - 1); - return subnet; - } - - if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx/%d", - &subnet->net.ipv6.address.x[0], - &subnet->net.ipv6.address.x[1], - &subnet->net.ipv6.address.x[2], - &subnet->net.ipv6.address.x[3], - &subnet->net.ipv6.address.x[4], - &subnet->net.ipv6.address.x[5], - &subnet->net.ipv6.address.x[6], - &subnet->net.ipv6.address.x[7], - &subnet->net.ipv6.masklength) == 9) - { - subnet->type = SUBNET_IPV6; - for(l = subnet->net.ipv6.masklength, i = 0; i < 8; l -= 16, i++) - { - subnet->net.ipv6.address.x[i] = htons(subnet->net.ipv6.address.x[i]); - if(l >= 16) - subnet->net.ipv6.mask.x[i] = 65535; - else if (l > 0) - subnet->net.ipv6.mask.x[i] = htons(65536 - (1 << l)); - else - subnet->net.ipv6.mask.x[i] = 0; - } - return subnet; - } - - if(sscanf(subnetstr, "%hu.%hu.%hu.%hu", - &x[0], - &x[1], - &x[2], - &x[3]) == 4) - { - subnet->type = SUBNET_IPV4; - subnet->net.ipv4.address = (((((x[0] << 8) + x[1]) << 8) + x[2]) << 8) + x[3]; - subnet->net.ipv4.mask = ~0; - subnet->net.ipv4.masklength = 32; - return subnet; - } - - if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx", - &subnet->net.ipv6.address.x[0], - &subnet->net.ipv6.address.x[1], - &subnet->net.ipv6.address.x[2], - &subnet->net.ipv6.address.x[3], - &subnet->net.ipv6.address.x[4], - &subnet->net.ipv6.address.x[5], - &subnet->net.ipv6.address.x[6], - &subnet->net.ipv6.address.x[7]) == 8) - { - subnet->type = SUBNET_IPV6; - subnet->net.ipv6.masklength = 128; - for(l = subnet->net.ipv6.masklength, i = 0; i < 8; l -= 16, i++) - { - subnet->net.ipv6.address.x[i] = htons(subnet->net.ipv6.address.x[i]); - if(l >= 16) - subnet->net.ipv6.mask.x[i] = 65535; - else if (l > 0) - subnet->net.ipv6.mask.x[i] = htons(65536 - (1 << l)); - else - subnet->net.ipv6.mask.x[i] = 0; - } - return subnet; - } - - if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx", - &x[0], - &x[1], - &x[2], - &x[3], - &x[4], - &x[5]) == 6) - { - subnet->type = SUBNET_MAC; - subnet->net.mac.address.x[0] = x[0]; - subnet->net.mac.address.x[1] = x[1]; - subnet->net.mac.address.x[2] = x[2]; - subnet->net.mac.address.x[3] = x[3]; - subnet->net.mac.address.x[4] = x[4]; - subnet->net.mac.address.x[5] = x[5]; - return subnet; - } - - free(subnet); - return NULL; +/* Adding and removing subnets */ + +void subnet_add(node_t *n, subnet_t *subnet) { + subnet->owner = n; + + splay_insert(&subnet_tree, subnet); + + if(n) { + splay_insert(&n->subnet_tree, subnet); + } + + subnet_cache_flush(subnet); } -char *net2str(subnet_t *subnet) -{ - char *netstr; -cp - switch(subnet->type) - { - case SUBNET_MAC: - asprintf(&netstr, "%hx:%hx:%hx:%hx:%hx:%hx", - subnet->net.mac.address.x[0], - subnet->net.mac.address.x[1], - subnet->net.mac.address.x[2], - subnet->net.mac.address.x[3], - subnet->net.mac.address.x[4], - subnet->net.mac.address.x[5]); - break; - case SUBNET_IPV4: - asprintf(&netstr, "%hu.%hu.%hu.%hu/%d", - (unsigned short int)((subnet->net.ipv4.address >> 24) & 255), - (unsigned short int)((subnet->net.ipv4.address >> 16) & 255), - (unsigned short int)((subnet->net.ipv4.address >> 8) & 255), - (unsigned short int)(subnet->net.ipv4.address & 255), - subnet->net.ipv4.masklength); - break; - case SUBNET_IPV6: - asprintf(&netstr, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx/%d", - ntohs(subnet->net.ipv6.address.x[0]), - ntohs(subnet->net.ipv6.address.x[1]), - ntohs(subnet->net.ipv6.address.x[2]), - ntohs(subnet->net.ipv6.address.x[3]), - ntohs(subnet->net.ipv6.address.x[4]), - ntohs(subnet->net.ipv6.address.x[5]), - ntohs(subnet->net.ipv6.address.x[6]), - ntohs(subnet->net.ipv6.address.x[7]), - subnet->net.ipv6.masklength); - break; - default: - asprintf(&netstr, _("unknown subnet type")); - } -cp - return netstr; +void subnet_del(node_t *n, subnet_t *subnet) { + if(n) { + splay_delete(&n->subnet_tree, subnet); + } + + splay_delete(&subnet_tree, subnet); + + subnet_cache_flush(subnet); } /* Subnet lookup routines */ -subnet_t *lookup_subnet(node_t *owner, subnet_t *subnet) -{ -cp - return avl_search(owner->subnet_tree, subnet); +subnet_t *lookup_subnet(node_t *owner, const subnet_t *subnet) { + return splay_search(&owner->subnet_tree, subnet); +} + +subnet_t *lookup_subnet_mac(const node_t *owner, const mac_t *address) { + subnet_t *r = NULL; + + // Check if this address is cached + + if((r = hash_search(mac_t, &mac_cache, address))) { + return r; + } + + // Search all subnets for a matching one + + for splay_each(subnet_t, p, owner ? &owner->subnet_tree : &subnet_tree) { + if(!p || p->type != SUBNET_MAC) { + continue; + } + + if(!memcmp(address, &p->net.mac.address, sizeof(*address))) { + r = p; + + if(!p->owner || p->owner->status.reachable) { + break; + } + } + } + + // Cache the result + + if(r) { + hash_insert(mac_t, &mac_cache, address, r); + } + + return r; } -subnet_t *lookup_subnet_mac(mac_t *address) -{ - subnet_t subnet, *p; -cp - subnet.type = SUBNET_MAC; - memcpy(&subnet.net.mac.address, address, sizeof(mac_t)); +subnet_t *lookup_subnet_ipv4(const ipv4_t *address) { + subnet_t *r = NULL; + + // Check if this address is cached + + if((r = hash_search(ipv4_t, &ipv4_cache, address))) { + return r; + } + + // Search all subnets for a matching one + + for splay_each(subnet_t, p, &subnet_tree) { + if(!p || p->type != SUBNET_IPV4) { + continue; + } - p = (subnet_t *)avl_search(subnet_tree, &subnet); -cp - return p; + if(!maskcmp(address, &p->net.ipv4.address, p->net.ipv4.prefixlength)) { + r = p; + + if(!p->owner || p->owner->status.reachable) { + break; + } + } + } + + // Cache the result + + if(r) { + hash_insert(ipv4_t, &ipv4_cache, address, r); + } + + return r; } -subnet_t *lookup_subnet_ipv4(ipv4_t *address) -{ - subnet_t subnet, *p; -cp - subnet.type = SUBNET_IPV4; - subnet.net.ipv4.address = *address; - subnet.net.ipv4.mask = 0xFFFFFFFF; - - do - { - /* Go find subnet */ - - p = (subnet_t *)avl_search_closest_smaller(subnet_tree, &subnet); - - /* Check if the found subnet REALLY matches */ -cp - if(p) - { - if ((*address & p->net.ipv4.mask) == p->net.ipv4.address) - break; - else - { - /* Otherwise, see if there is a bigger enclosing subnet */ - - subnet.net.ipv4.mask = p->net.ipv4.mask << 1; - subnet.net.ipv4.address = p->net.ipv4.address & subnet.net.ipv4.mask; - } - } - } while (p); - - return p; +subnet_t *lookup_subnet_ipv6(const ipv6_t *address) { + subnet_t *r = NULL; + + // Check if this address is cached + + if((r = hash_search(ipv6_t, &ipv6_cache, address))) { + return r; + } + + // Search all subnets for a matching one + + for splay_each(subnet_t, p, &subnet_tree) { + if(!p || p->type != SUBNET_IPV6) { + continue; + } + + if(!maskcmp(address, &p->net.ipv6.address, p->net.ipv6.prefixlength)) { + r = p; + + if(!p->owner || p->owner->status.reachable) { + break; + } + } + } + + // Cache the result + + if(r) { + hash_insert(ipv6_t, &ipv6_cache, address, r); + } + + return r; } -subnet_t *lookup_subnet_ipv6(ipv6_t *address) -{ - subnet_t subnet, *p; - int i; -cp - subnet.type = SUBNET_IPV6; - memcpy(subnet.net.ipv6.address.x, address, sizeof(ipv6_t)); - memset(subnet.net.ipv6.mask.x, 0xFF, 16); - - p = (subnet_t *)avl_search_closest_greater(subnet_tree, &subnet); - - if(p) - for(i=0; i<8; i++) - if((address->x[i] & p->net.ipv6.address.x[i]) != p->net.ipv6.address.x[i]) - return NULL; - - return p; +void subnet_update(node_t *owner, subnet_t *subnet, bool up) { + char netstr[MAXNETSTR]; + char *name, *address, *port; + char empty[] = ""; + + // Prepare environment variables to be passed to the script + + environment_t env; + environment_init(&env); + environment_add(&env, "NODE=%s", owner->name); + + if(owner != myself) { + sockaddr2str(&owner->address, &address, &port); + environment_add(&env, "REMOTEADDRESS=%s", address); + environment_add(&env, "REMOTEPORT=%s", port); + free(port); + free(address); + } + + int env_subnet = environment_add(&env, NULL); + int env_weight = environment_add(&env, NULL); + + name = up ? "subnet-up" : "subnet-down"; + + if(!subnet) { + for splay_each(subnet_t, subnet, &owner->subnet_tree) { + if(!net2str(netstr, sizeof(netstr), subnet)) { + continue; + } + + // Strip the weight from the subnet, and put it in its own environment variable + char *weight = strchr(netstr, '#'); + + if(weight) { + *weight++ = 0; + } else { + weight = empty; + } + + // Prepare the SUBNET and WEIGHT variables + environment_update(&env, env_subnet, "SUBNET=%s", netstr); + environment_update(&env, env_weight, "WEIGHT=%s", weight); + + execute_script(name, &env); + } + } else { + if(net2str(netstr, sizeof(netstr), subnet)) { + // Strip the weight from the subnet, and put it in its own environment variable + char *weight = strchr(netstr, '#'); + + if(weight) { + *weight++ = 0; + } else { + weight = empty; + } + + // Prepare the SUBNET and WEIGHT variables + environment_update(&env, env_subnet, "SUBNET=%s", netstr); + environment_update(&env, env_weight, "WEIGHT=%s", weight); + + execute_script(name, &env); + } + } + + environment_exit(&env); } -void dump_subnets(void) -{ - char *netstr; - subnet_t *subnet; - avl_node_t *node; -cp - syslog(LOG_DEBUG, _("Subnet list:")); - for(node = subnet_tree->head; node; node = node->next) - { - subnet = (subnet_t *)node->data; - netstr = net2str(subnet); - syslog(LOG_DEBUG, _(" %s owner %s"), netstr, subnet->owner->name); - free(netstr); - } - syslog(LOG_DEBUG, _("End of subnet list.")); -cp +bool dump_subnets(connection_t *c) { + for splay_each(subnet_t, subnet, &subnet_tree) { + char netstr[MAXNETSTR]; + + if(!net2str(netstr, sizeof(netstr), subnet)) { + continue; + } + + send_request(c, "%d %d %s %s", + CONTROL, REQ_DUMP_SUBNETS, + netstr, subnet->owner ? subnet->owner->name : "(broadcast)"); + } + + return send_request(c, "%d %d", CONTROL, REQ_DUMP_SUBNETS); }