X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fsubnet.c;h=f4b147411dd287deb32af5a35e5cb30ea23c5800;hb=3a149f7521dfff67e6a790c1a830afc649ae083e;hp=9c225f2129bfbd0df31d571d6712d48830ecb3f1;hpb=2228b16159a7aff64e6559ee1635716154e67fe6;p=tinc diff --git a/src/subnet.c b/src/subnet.c index 9c225f21..e0d1acbc 100644 --- a/src/subnet.c +++ b/src/subnet.c @@ -1,7 +1,7 @@ /* subnet.c -- handle subnet lookups and lists - Copyright (C) 2000 Guus Sliepen , - 2000 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,74 +13,288 @@ 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.1 2000/10/01 03:21:49 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 "subnet.h" +#include "system.h" + +#include "splay_tree.h" +#include "control_common.h" +#include "hash.h" +#include "logger.h" #include "net.h" +#include "netutl.h" +#include "node.h" +#include "script.h" +#include "subnet.h" +#include "xalloc.h" -/* Allocating and freeing space for subnets */ +/* lists type of subnet */ + +splay_tree_t subnet_tree = { + .compare = (splay_compare_t) subnet_compare, + .delete = (splay_action_t) free_subnet, +}; + +/* Subnet lookup cache */ + +static hash_t *ipv4_cache; +static hash_t *ipv6_cache; +static hash_t *mac_cache; + +void subnet_cache_flush(void) { + hash_clear(ipv4_cache); + hash_clear(ipv6_cache); + hash_clear(mac_cache); +} -subnet_t *new_subnet(void) -{ -cp -cp +/* Initialising trees */ + +void init_subnets(void) { + ipv4_cache = hash_alloc(0x100, sizeof(ipv4_t)); + ipv6_cache = hash_alloc(0x100, sizeof(ipv6_t)); + mac_cache = hash_alloc(0x100, sizeof(mac_t)); +} + +void exit_subnets(void) { + splay_empty_tree(&subnet_tree); + + hash_free(ipv4_cache); + hash_free(ipv6_cache); + hash_free(mac_cache); +} + +splay_tree_t *new_subnet_tree(void) { + return splay_alloc_tree((splay_compare_t) subnet_compare, NULL); } -void free_subnet(subnet_t *subnet) -{ -cp -cp +void free_subnet_tree(splay_tree_t *subnet_tree) { + splay_delete_tree(subnet_tree); } -/* Linked list management */ - -int subnet_add(conn_list_t *cl, subnet_t *subnet) -{ -cp - subnet->next = cl->subnets->next; - subnet->prev = NULL; - subnet->next->prev = subnet - cl->subnets = subnet; -cp - return 0; +/* Allocating and freeing space for subnets */ + +subnet_t *new_subnet(void) { + return xzalloc(sizeof(subnet_t)); } -int subnet_del(conn_list_t *cl, subnet_t *subnet) -{ -cp - if(subnet->prev) - { - subnet->prev->next = subnet->next; - } - else - { - subnet->owner->subnets = subnet->next; - } - - subnet->next->prev = subnet->prev; - free_subnet(subnet); -cp - return 0; +void free_subnet(subnet_t *subnet) { + free(subnet); } -/* Ascii representation of subnets */ +/* Adding and removing subnets */ + +void subnet_add(node_t *n, subnet_t *subnet) { + subnet->owner = n; + + splay_insert(&subnet_tree, subnet); -subnet_t *str2net(char *subnetstr) -{ -cp -cp + if(n) { + splay_insert(n->subnet_tree, subnet); + } + + subnet_cache_flush(); } -char *net2str(subnet_t *subnet) -{ -cp -cp +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 lookup routines */ + +subnet_t *lookup_subnet(const 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_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_cache, address, r); + } + + return r; +} + +subnet_t *lookup_subnet_ipv4(const ipv4_t *address) { + subnet_t *r = NULL; + + // Check if this address is cached + + if((r = hash_search(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; + } + + 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_cache, address, r); + } + + return r; +} + +subnet_t *lookup_subnet_ipv6(const ipv6_t *address) { + subnet_t *r = NULL; + + // Check if this address is cached + + if((r = hash_search(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_cache, address, r); + } + + return r; +} + +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); +} + +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); +}