X-Git-Url: https://tinc-vpn.org/git/browse?p=tinc;a=blobdiff_plain;f=src%2Fsubnet.c;h=6745e2f4766790f0be304aa3a017715952bab7ac;hp=9a882570b7e3b8c9b228ba173f12e0688cffac9c;hb=1c34ba7fb8580949f3bd3a0d804747bbaea28e36;hpb=950fb8e916b0e248dcaa72c96859acd6046683aa diff --git a/src/subnet.c b/src/subnet.c index 9a882570..6745e2f4 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-2002 Guus Sliepen , + 2000-2002 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 @@ -17,15 +17,132 @@ 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.2 2000/10/11 10:35:17 guus Exp $ + $Id: subnet.c,v 1.1.2.29 2002/02/10 21:57:54 guus Exp $ */ #include "config.h" -#include -#include -#include "subnet.h" +#include +#include +#include +#include +#include +#include +#include + +#include "conf.h" #include "net.h" +#include "node.h" +#include "subnet.h" +#include "system.h" + +#include +#include +#include + +/* lists type of subnet */ + +avl_tree_t *subnet_tree; + +/* Subnet comparison */ + +int subnet_compare_mac(subnet_t *a, subnet_t *b) +{ +cp + return memcmp(&a->net.mac.address, &b->net.mac.address, sizeof(mac_t)); +} + +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; +} + +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; +} + +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; +} + +/* Initialising trees */ + +void init_subnets(void) +{ +cp + subnet_tree = avl_alloc_tree((avl_compare_t)subnet_compare, (avl_action_t)free_subnet); +cp +} + +void exit_subnets(void) +{ +cp + avl_delete_tree(subnet_tree); +cp +} + +avl_tree_t *new_subnet_tree(void) +{ +cp + return avl_alloc_tree((avl_compare_t)subnet_compare, NULL); +cp +} + +void free_subnet_tree(avl_tree_t *subnet_tree) +{ +cp + avl_delete_tree(subnet_tree); +cp +} /* Allocating and freeing space for subnets */ @@ -41,34 +158,25 @@ cp free(subnet); } -/* Linked list management */ +/* Adding and removing subnets */ -void subnet_add(conn_list_t *cl, subnet_t *subnet) +void subnet_add(node_t *n, subnet_t *subnet) { cp - /* FIXME: do sorting on netmask size if necessary */ + subnet->owner = n; - subnet->next = cl->subnets->next; - subnet->prev = NULL; - subnet->next->prev = subnet; - cl->subnets = subnet; + avl_insert(subnet_tree, subnet); +cp + avl_insert(n->subnet_tree, subnet); cp } -void subnet_del(conn_list_t *cl, subnet_t *subnet) +void subnet_del(node_t *n, 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); + avl_delete(n->subnet_tree, subnet); +cp + avl_delete(subnet_tree, subnet); cp } @@ -76,65 +184,108 @@ cp subnet_t *str2net(char *subnetstr) { - int type; + int i, l; subnet_t *subnet; + unsigned short int x[6]; cp - if(sscanf(subnetstr, "%d,", &type) != 1) - return NULL; - 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; + } - switch(type) + if(sscanf(subnetstr, "%hu.%hu.%hu.%hu", + &x[0], + &x[1], + &x[2], + &x[3]) == 4) { - case SUBNET_MAC: - if(sscanf(netstr, "%d,%hhx:%hhx:%hhx:%hhx:%hhx:%hhx", &subnet->type, - &subnet->net.mac.x[0], - &subnet->net.mac.x[1], - &subnet->net.mac.x[2], - &subnet->net.mac.x[3], - &subnet->net.mac.x[4], - &subnet->net.mac.x[5]) != 7) - { - free_subnet(subnet); - return NULL; - } - break; - case SUBNET_IPv4: - if(sscanf(subnetstr, "%d,%lx:%lx", &subnet->type, &subnet->net.ipv4.address, &subnet->net.ipv4.mask) != 3) - { - free_subnet(subnet); - return NULL; - } - break; - case SUBNET_IPv6: - if(sscanf(netstr, "%d,%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx/%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx", &subnet->type, - &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.mask.x[0], - &subnet->net.ipv6.mask.x[1], - &subnet->net.ipv6.mask.x[2], - &subnet->net.ipv6.mask.x[3], - &subnet->net.ipv6.mask.x[4], - &subnet->net.ipv6.mask.x[5], - &subnet->net.ipv6.mask.x[6], - &subnet->net.ipv6.mask.x[7]) != 17) - { - free_subnet(subnet); - return NULL; - } - break; - break; - default: - free_subnet(subnet); - return NULL; -cp - return subnet; + 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; } char *net2str(subnet_t *subnet) @@ -144,35 +295,36 @@ cp switch(subnet->type) { case SUBNET_MAC: - asprintf(netstr, "%d,%hhx:%hhx:%hhx:%hhx:%hhx:%hhx", subnet->type, - subnet->net.mac.x[0], - subnet->net.mac.x[1], - subnet->net.mac.x[2], - subnet->net.mac.x[3], - subnet->net.mac.x[4], - subnet->net.mac.x[5]); - case SUBNET_IPv4: - asprintf(netstr, "%d,%lx:%lx", subnet->type, subnet->net.ipv4.address, subnet->net.ipv4.mask); - case SUBNET_IPv6: - asprintf(netstr, "%d,%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx/%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], - subnet->net.ipv6.mask.x[0], - subnet->net.ipv6.mask.x[1], - subnet->net.ipv6.mask.x[2], - subnet->net.ipv6.mask.x[3], - subnet->net.ipv6.mask.x[4], - subnet->net.ipv6.mask.x[5], - subnet->net.ipv6.mask.x[6], - subnet->net.ipv6.mask.x[7]); + 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: - netstr = NULL; + asprintf(&netstr, _("unknown subnet type")); } cp return netstr; @@ -180,49 +332,90 @@ cp /* Subnet lookup routines */ -subnet_t *lookup_subnet_mac(subnet_t *subnets, mac_t address) +subnet_t *lookup_subnet(node_t *owner, subnet_t *subnet) { - subnet_t *subnet; +cp + return avl_search(owner->subnet_tree, subnet); +} + +subnet_t *lookup_subnet_mac(mac_t *address) +{ + subnet_t subnet, *p; cp - for(subnet = subnets; subnet != NULL; subnet = subnet->next) - { - if(subnet->type == SUBNET_MAC) - if(memcmp(&address, &subnet->net.mac.address, sizeof(address)) == 0) - break; - } + subnet.type = SUBNET_MAC; + memcpy(&subnet.net.mac.address, address, sizeof(mac_t)); + + p = (subnet_t *)avl_search(subnet_tree, &subnet); cp - return subnet; + return p; } -subnet_t *lookup_subnet_ipv4(subnet_t *subnets, ipv4_t address) +subnet_t *lookup_subnet_ipv4(ipv4_t *address) { - subnet_t *subnet; + subnet_t subnet, *p; cp - for(subnet = subnets; subnet != NULL; subnet = subnet->next) - { - if(subnet->type == SUBNET_IPV4) - if((address & subnet->net.ipv4.mask) == subnet->net.ipv4.address) + 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(ipv6_t *address) +{ + subnet_t subnet, *p; + int i; cp - return subnet; + 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; } -subnet_t *lookup_subnet_ipv6(subnet_t *subnets, ipv6_t address) +void dump_subnets(void) { + char *netstr; subnet_t *subnet; + avl_node_t *node; cp - for(subnet = subnets; subnet != NULL; subnet = subnet->next) + syslog(LOG_DEBUG, _("Subnet list:")); + for(node = subnet_tree->head; node; node = node->next) { - if(subnet->type == SUBNET_IPV6) - { - for(i=0; i<8; i++) - if((address.x[i] & subnet->net.ipv6.mask.x[i]) != subnet->net.ipv6.address.x[i]) - break; - if(i=8) - break; - } + 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 - return subnet; }