tinc/src/subnet.c

541 lines
11 KiB
C
Raw Normal View History

/*
subnet.c -- handle subnet lookups and lists
Copyright (C) 2000-2009 Guus Sliepen <guus@tinc-vpn.org>,
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
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
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$
*/
#include "system.h"
#include "avl_tree.h"
2004-12-01 20:06:05 +00:00
#include "device.h"
#include "logger.h"
#include "net.h"
#include "netutl.h"
#include "node.h"
2004-12-01 20:06:05 +00:00
#include "process.h"
#include "subnet.h"
#include "utils.h"
#include "xalloc.h"
/* lists type of subnet */
avl_tree_t *subnet_tree;
/* Subnet lookup cache */
static ipv4_t cache_ipv4_address[2];
static subnet_t *cache_ipv4_subnet[2];
static bool cache_ipv4_valid[2];
static int cache_ipv4_slot;
static ipv6_t cache_ipv6_address[2];
static subnet_t *cache_ipv6_subnet[2];
static bool cache_ipv6_valid[2];
static int cache_ipv6_slot;
void subnet_cache_flush() {
cache_ipv4_valid[0] = cache_ipv4_valid[1] = false;
cache_ipv6_valid[0] = cache_ipv6_valid[1] = false;
}
2000-11-20 19:12:17 +00:00
/* Subnet comparison */
static int subnet_compare_mac(const subnet_t *a, const subnet_t *b)
{
2002-09-09 21:25:28 +00:00
int result;
result = memcmp(&a->net.mac.address, &b->net.mac.address, sizeof(mac_t));
if(result)
return result;
result = a->weight - b->weight;
2002-09-09 21:25:28 +00:00
if(result || !a->owner || !b->owner)
return result;
return strcmp(a->owner->name, b->owner->name);
}
static int subnet_compare_ipv4(const subnet_t *a, const subnet_t *b)
{
2002-09-09 21:25:28 +00:00
int result;
result = b->net.ipv4.prefixlength - a->net.ipv4.prefixlength;
2002-09-09 21:25:28 +00:00
if(result)
return result;
result = memcmp(&a->net.ipv4.address, &b->net.ipv4.address, sizeof(ipv4_t));
if(result)
return result;
result = a->weight - b->weight;
2002-09-09 21:25:28 +00:00
if(result || !a->owner || !b->owner)
return result;
return strcmp(a->owner->name, b->owner->name);
2000-11-20 19:12:17 +00:00
}
static int subnet_compare_ipv6(const subnet_t *a, const subnet_t *b)
{
2002-09-09 21:25:28 +00:00
int result;
result = b->net.ipv6.prefixlength - a->net.ipv6.prefixlength;
2002-09-09 21:25:28 +00:00
if(result)
return result;
result = memcmp(&a->net.ipv6.address, &b->net.ipv6.address, sizeof(ipv6_t));
2002-09-09 21:25:28 +00:00
if(result)
return result;
result = a->weight - b->weight;
2002-09-09 21:25:28 +00:00
if(result || !a->owner || !b->owner)
return result;
return strcmp(a->owner->name, b->owner->name);
2000-11-20 19:12:17 +00:00
}
int subnet_compare(const subnet_t *a, const subnet_t *b)
2000-11-20 19:12:17 +00:00
{
2002-09-09 21:25:28 +00:00
int result;
result = a->type - b->type;
if(result)
return result;
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:
logger(LOG_ERR, _("subnet_compare() was called with unknown subnet type %d, exitting!"),
2002-09-09 21:25:28 +00:00
a->type);
cp_trace();
exit(0);
}
return 0;
}
/* Initialising trees */
void init_subnets(void)
{
2002-09-09 21:25:28 +00:00
cp();
subnet_tree = avl_alloc_tree((avl_compare_t) subnet_compare, (avl_action_t) free_subnet);
subnet_cache_flush();
}
void exit_subnets(void)
{
2002-09-09 21:25:28 +00:00
cp();
avl_delete_tree(subnet_tree);
}
avl_tree_t *new_subnet_tree(void)
{
2002-09-09 21:25:28 +00:00
cp();
return avl_alloc_tree((avl_compare_t) subnet_compare, NULL);
}
2002-09-09 22:33:31 +00:00
void free_subnet_tree(avl_tree_t *subnet_tree)
{
2002-09-09 21:25:28 +00:00
cp();
avl_delete_tree(subnet_tree);
}
2000-11-20 19:12:17 +00:00
/* Allocating and freeing space for subnets */
subnet_t *new_subnet(void)
{
2002-09-09 21:25:28 +00:00
cp();
2003-08-28 21:05:11 +00:00
return xmalloc_and_zero(sizeof(subnet_t));
2000-11-20 19:12:17 +00:00
}
2002-09-09 22:33:31 +00:00
void free_subnet(subnet_t *subnet)
2000-11-20 19:12:17 +00:00
{
2002-09-09 21:25:28 +00:00
cp();
free(subnet);
2000-11-20 19:12:17 +00:00
}
/* Adding and removing subnets */
2002-09-09 22:33:31 +00:00
void subnet_add(node_t *n, subnet_t *subnet)
2000-11-20 19:12:17 +00:00
{
2002-09-09 21:25:28 +00:00
cp();
subnet->owner = n;
2002-09-09 21:25:28 +00:00
avl_insert(subnet_tree, subnet);
avl_insert(n->subnet_tree, subnet);
subnet_cache_flush();
2000-11-20 19:12:17 +00:00
}
2002-09-09 22:33:31 +00:00
void subnet_del(node_t *n, subnet_t *subnet)
2000-11-20 19:12:17 +00:00
{
2002-09-09 21:25:28 +00:00
cp();
avl_delete(n->subnet_tree, subnet);
avl_delete(subnet_tree, subnet);
subnet_cache_flush();
}
/* Ascii representation of subnets */
bool str2net(subnet_t *subnet, const char *subnetstr)
{
2002-09-09 21:25:28 +00:00
int i, l;
uint16_t x[8];
int weight = 10;
2002-09-09 21:25:28 +00:00
cp();
if(sscanf(subnetstr, "%hu.%hu.%hu.%hu/%d#%d",
&x[0], &x[1], &x[2], &x[3], &l, &weight) >= 5) {
if(l < 0 || l > 32)
return false;
2002-09-09 21:25:28 +00:00
subnet->type = SUBNET_IPV4;
subnet->net.ipv4.prefixlength = l;
subnet->weight = weight;
2002-09-09 21:25:28 +00:00
for(i = 0; i < 4; i++) {
if(x[i] > 255)
return false;
2002-09-09 21:25:28 +00:00
subnet->net.ipv4.address.x[i] = x[i];
}
2002-09-09 21:25:28 +00:00
return true;
2002-09-09 21:25:28 +00:00
}
if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx/%d#%d",
2002-09-09 21:25:28 +00:00
&x[0], &x[1], &x[2], &x[3], &x[4], &x[5], &x[6], &x[7],
&l, &weight) >= 9) {
if(l < 0 || l > 128)
return false;
2002-09-09 21:25:28 +00:00
subnet->type = SUBNET_IPV6;
subnet->net.ipv6.prefixlength = l;
subnet->weight = weight;
2002-09-09 21:25:28 +00:00
for(i = 0; i < 8; i++)
subnet->net.ipv6.address.x[i] = htons(x[i]);
return true;
2002-09-09 21:25:28 +00:00
}
if(sscanf(subnetstr, "%hu.%hu.%hu.%hu#%d", &x[0], &x[1], &x[2], &x[3], &weight) >= 4) {
2002-09-09 21:25:28 +00:00
subnet->type = SUBNET_IPV4;
subnet->net.ipv4.prefixlength = 32;
subnet->weight = weight;
2002-09-09 21:25:28 +00:00
for(i = 0; i < 4; i++) {
if(x[i] > 255)
return false;
2002-09-09 21:25:28 +00:00
subnet->net.ipv4.address.x[i] = x[i];
}
2002-09-09 21:25:28 +00:00
return true;
2002-09-09 21:25:28 +00:00
}
if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx#%d",
&x[0], &x[1], &x[2], &x[3], &x[4], &x[5], &x[6], &x[7], &weight) >= 8) {
2002-09-09 21:25:28 +00:00
subnet->type = SUBNET_IPV6;
subnet->net.ipv6.prefixlength = 128;
subnet->weight = weight;
2002-09-09 21:25:28 +00:00
for(i = 0; i < 8; i++)
subnet->net.ipv6.address.x[i] = htons(x[i]);
return true;
2002-09-09 21:25:28 +00:00
}
if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx#%d",
&x[0], &x[1], &x[2], &x[3], &x[4], &x[5], &weight) >= 6) {
2002-09-09 21:25:28 +00:00
subnet->type = SUBNET_MAC;
subnet->weight = weight;
2002-09-09 21:25:28 +00:00
for(i = 0; i < 6; i++)
subnet->net.mac.address.x[i] = x[i];
return true;
2002-09-09 21:25:28 +00:00
}
return false;
}
bool net2str(char *netstr, int len, const subnet_t *subnet)
{
2002-09-09 21:25:28 +00:00
cp();
if(!netstr || !subnet) {
logger(LOG_ERR, _("net2str() was called with netstr=%p, subnet=%p!\n"), netstr, subnet);
return false;
}
2002-09-09 21:25:28 +00:00
switch (subnet->type) {
case SUBNET_MAC:
snprintf(netstr, len, "%hx:%hx:%hx:%hx:%hx:%hx#%d",
2002-09-09 21:25:28 +00:00
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],
subnet->weight);
2002-09-09 21:25:28 +00:00
break;
case SUBNET_IPV4:
snprintf(netstr, len, "%hu.%hu.%hu.%hu/%d#%d",
2002-09-09 21:25:28 +00:00
subnet->net.ipv4.address.x[0],
subnet->net.ipv4.address.x[1],
subnet->net.ipv4.address.x[2],
subnet->net.ipv4.address.x[3],
subnet->net.ipv4.prefixlength,
subnet->weight);
2002-09-09 21:25:28 +00:00
break;
case SUBNET_IPV6:
snprintf(netstr, len, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx/%d#%d",
2002-09-09 21:25:28 +00:00
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.prefixlength,
subnet->weight);
2002-09-09 21:25:28 +00:00
break;
default:
logger(LOG_ERR,
2002-09-09 21:25:28 +00:00
_("net2str() was called with unknown subnet type %d, exiting!"),
subnet->type);
cp_trace();
exit(0);
}
2004-11-01 15:16:12 +00:00
return true;
}
/* Subnet lookup routines */
subnet_t *lookup_subnet(const node_t *owner, const subnet_t *subnet)
{
2002-09-09 21:25:28 +00:00
cp();
return avl_search(owner->subnet_tree, subnet);
}
subnet_t *lookup_subnet_mac(const mac_t *address)
{
subnet_t *p, subnet = {0};
2002-09-09 21:25:28 +00:00
cp();
subnet.type = SUBNET_MAC;
subnet.net.mac.address = *address;
subnet.owner = NULL;
2003-08-28 21:05:11 +00:00
p = avl_search(subnet_tree, &subnet);
2002-09-09 21:25:28 +00:00
return p;
}
subnet_t *lookup_subnet_ipv4(const ipv4_t *address)
{
subnet_t *p, *r = NULL, subnet = {0};
avl_node_t *n;
int i;
2002-09-09 21:25:28 +00:00
cp();
// Check if this address is cached
for(i = 0; i < 2; i++) {
if(!cache_ipv4_valid[i])
continue;
if(!memcmp(address, &cache_ipv4_address[i], sizeof *address))
return cache_ipv4_subnet[i];
}
// Search all subnets for a matching one
subnet.type = SUBNET_IPV4;
subnet.net.ipv4.address = *address;
subnet.net.ipv4.prefixlength = 32;
subnet.owner = NULL;
for(n = subnet_tree->head; n; n = n->next) {
p = n->data;
if(!p || p->type != subnet.type)
continue;
2002-09-09 21:25:28 +00:00
if(!maskcmp(address, &p->net.ipv4.address, p->net.ipv4.prefixlength)) {
r = p;
if(p->owner->status.reachable)
2002-09-09 21:25:28 +00:00
break;
}
}
2002-09-09 21:25:28 +00:00
// Cache the result
cache_ipv4_slot = !cache_ipv4_slot;
memcpy(&cache_ipv4_address[cache_ipv4_slot], address, sizeof *address);
cache_ipv4_subnet[cache_ipv4_slot] = r;
cache_ipv4_valid[cache_ipv4_slot] = true;
return r;
}
subnet_t *lookup_subnet_ipv6(const ipv6_t *address)
{
subnet_t *p, *r = NULL, subnet = {0};
avl_node_t *n;
int i;
2002-09-09 21:25:28 +00:00
cp();
// Check if this address is cached
for(i = 0; i < 2; i++) {
if(!cache_ipv6_valid[i])
continue;
if(!memcmp(address, &cache_ipv6_address[i], sizeof *address))
return cache_ipv6_subnet[i];
}
// Search all subnets for a matching one
subnet.type = SUBNET_IPV6;
subnet.net.ipv6.address = *address;
subnet.net.ipv6.prefixlength = 128;
subnet.owner = NULL;
for(n = subnet_tree->head; n; n = n->next) {
p = n->data;
if(!p || p->type != subnet.type)
continue;
2002-09-09 21:25:28 +00:00
if(!maskcmp(address, &p->net.ipv6.address, p->net.ipv6.prefixlength)) {
r = p;
if(p->owner->status.reachable)
2002-09-09 21:25:28 +00:00
break;
}
}
2002-09-09 21:25:28 +00:00
// Cache the result
cache_ipv6_slot = !cache_ipv6_slot;
memcpy(&cache_ipv6_address[cache_ipv6_slot], address, sizeof *address);
cache_ipv6_subnet[cache_ipv6_slot] = r;
cache_ipv6_valid[cache_ipv6_slot] = true;
return r;
}
2004-12-01 20:06:05 +00:00
void subnet_update(node_t *owner, subnet_t *subnet, bool up) {
avl_node_t *node;
int i;
char *envp[8];
char netstr[MAXNETSTR + 7] = "SUBNET=";
char *name, *address, *port;
2009-09-08 16:18:36 +00:00
xasprintf(&envp[0], "NETNAME=%s", netname ? : "");
xasprintf(&envp[1], "DEVICE=%s", device ? : "");
xasprintf(&envp[2], "INTERFACE=%s", iface ? : "");
xasprintf(&envp[3], "NODE=%s", owner->name);
2004-12-01 21:26:51 +00:00
2004-12-01 20:06:05 +00:00
if(owner != myself) {
sockaddr2str(&owner->address, &address, &port);
2009-09-08 16:18:36 +00:00
xasprintf(&envp[4], "REMOTEADDRESS=%s", address);
xasprintf(&envp[5], "REMOTEPORT=%s", port);
2004-12-01 20:06:05 +00:00
envp[6] = netstr;
envp[7] = NULL;
} else {
envp[4] = netstr;
envp[5] = NULL;
}
name = up ? "subnet-up" : "subnet-down";
if(!subnet) {
for(node = owner->subnet_tree->head; node; node = node->next) {
subnet = node->data;
if(!net2str(netstr + 7, sizeof netstr - 7, subnet))
continue;
execute_script(name, envp);
}
} else {
if(net2str(netstr + 7, sizeof netstr - 7, subnet))
execute_script(name, envp);
}
for(i = 0; i < (owner != myself ? 6 : 4); i++)
free(envp[i]);
2004-12-01 21:26:51 +00:00
if(owner != myself) {
free(address);
free(port);
}
2004-12-01 20:06:05 +00:00
}
2001-10-27 13:13:35 +00:00
void dump_subnets(void)
{
char netstr[MAXNETSTR];
2002-09-09 21:25:28 +00:00
subnet_t *subnet;
avl_node_t *node;
cp();
logger(LOG_DEBUG, _("Subnet list:"));
2002-09-09 21:25:28 +00:00
for(node = subnet_tree->head; node; node = node->next) {
2003-08-28 21:05:11 +00:00
subnet = node->data;
if(!net2str(netstr, sizeof netstr, subnet))
continue;
logger(LOG_DEBUG, _(" %s owner %s"), netstr, subnet->owner->name);
2002-09-09 21:25:28 +00:00
}
logger(LOG_DEBUG, _("End of subnet list."));
}