Sprinkling the source with static and attributes.
[tinc] / src / subnet.c
index fbcfba7..6a7cf78 100644 (file)
@@ -1,7 +1,7 @@
 /*
     subnet.c -- handle subnet lookups and lists
-    Copyright (C) 2000 Guus Sliepen <guus@sliepen.warande.net>,
-                  2000 Ivo Timmermans <itimmermans@bigfoot.com>
+    Copyright (C) 2000-2002 Guus Sliepen <guus@sliepen.eu.org>,
+                  2000-2002 Ivo Timmermans <ivo@o2w.nl>
 
     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
     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.5 2000/10/24 15:46:18 guus Exp $
+    $Id: subnet.c,v 1.1.2.45 2003/07/06 23:16:29 guus Exp $
 */
 
 #include "config.h"
-#include <utils.h>
 
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <netdb.h>
+#include <netinet/in.h>
+
+#include <utils.h>
 #include <xalloc.h>
-#include "subnet.h"
+#include <avl_tree.h>
+
+#include "conf.h"
 #include "net.h"
+#include "node.h"
+#include "subnet.h"
+#include "netutl.h"
+#include "logger.h"
+
+#include "system.h"
+
+/* lists type of subnet */
+
+avl_tree_t *subnet_tree;
+
+/* Subnet comparison */
+
+static int subnet_compare_mac(subnet_t *a, subnet_t *b)
+{
+       int result;
+
+       result = memcmp(&a->net.mac.address, &b->net.mac.address, sizeof(mac_t));
+
+       if(result || !a->owner || !b->owner)
+               return result;
+
+       return strcmp(a->owner->name, b->owner->name);
+}
+
+static int subnet_compare_ipv4(subnet_t *a, subnet_t *b)
+{
+       int result;
+
+       result = memcmp(&a->net.ipv4.address, &b->net.ipv4.address, sizeof(ipv4_t));
+
+       if(result)
+               return result;
+
+       result = a->net.ipv4.prefixlength - b->net.ipv4.prefixlength;
+
+       if(result || !a->owner || !b->owner)
+               return result;
+
+       return strcmp(a->owner->name, b->owner->name);
+}
+
+static int subnet_compare_ipv6(subnet_t *a, subnet_t *b)
+{
+       int result;
+
+       result = memcmp(&a->net.ipv6.address, &b->net.ipv6.address, sizeof(ipv6_t));
+
+       if(result)
+               return result;
+
+       result = a->net.ipv6.prefixlength - b->net.ipv6.prefixlength;
+
+       if(result || !a->owner || !b->owner)
+               return result;
+
+       return strcmp(a->owner->name, b->owner->name);
+}
+
+static int subnet_compare(subnet_t *a, subnet_t *b)
+{
+       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(DEBUG_ALWAYS, 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);
+}
+
+void exit_subnets(void)
+{
+       cp();
+
+       avl_delete_tree(subnet_tree);
+}
+
+avl_tree_t *new_subnet_tree(void)
+{
+       cp();
+
+       return avl_alloc_tree((avl_compare_t) subnet_compare, NULL);
+}
+
+void free_subnet_tree(avl_tree_t *subnet_tree)
+{
+       cp();
+
+       avl_delete_tree(subnet_tree);
+}
 
 /* Allocating and freeing space for subnets */
 
 subnet_t *new_subnet(void)
 {
-cp
-  return (subnet_t *)xmalloc(sizeof(subnet_t));
+       cp();
+
+       return (subnet_t *) xmalloc_and_zero(sizeof(subnet_t));
 }
 
 void free_subnet(subnet_t *subnet)
 {
-cp
-  free(subnet);
+       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)
 {
-  subnet_t *p = NULL;
-  subnet_t *q = NULL;
-cp
-  subnet->owner = cl;
-
-  /* Sort on size of subnet mask (IPv4 only at the moment!)
-  
-     Three cases: cl->subnets = NULL -> just add this subnet
-                  insert before first -> add it in front of list
-                  rest: insert after another subnet
-   */
-
-  if(cl->subnets)
-    {
-      p = q = cl->subnets;
-
-      for(; p; p = p->next)
-        {
-          if(subnet->net.ipv4.mask >= p->net.ipv4.mask)
-            break;
-
-          q = p;
-        }
-     }
-     
-  if(!cl->subnets || p == cl->subnets) /* First two cases */
-    {
-      /* Insert in front */
-      subnet->next = cl->subnets;
-      subnet->prev = NULL;
-      cl->subnets = subnet;
-    }
-  else                                 /* Third case */
-    {
-      /* Insert after q */
-      subnet->next = q->next;
-      subnet->prev = q;
-      q->next = subnet;
-    }
-
-  if(subnet->next)
-    subnet->next->prev = subnet;
-cp
+       cp();
+
+       subnet->owner = n;
+
+       avl_insert(subnet_tree, subnet);
+       avl_insert(n->subnet_tree, subnet);
 }
 
-void subnet_del(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;
+       cp();
 
-  if(subnet->next)
-    subnet->next->prev = subnet->prev;
-
-  free_subnet(subnet);
-cp
+       avl_delete(n->subnet_tree, subnet);
+       avl_delete(subnet_tree, subnet);
 }
 
 /* Ascii representation of subnets */
 
 subnet_t *str2net(char *subnetstr)
 {
-  int type;
-  subnet_t *subnet;
-cp
-  if(sscanf(subnetstr, "%d,", &type) != 1)
-    return NULL;
-
-  subnet = new_subnet();
-
-  switch(type)
-    {
-      case SUBNET_MAC:
-        if(sscanf(subnetstr, "%d,%hhx:%hhx:%hhx:%hhx:%hhx:%hhx", &subnet->type,
-                   &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]) != 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(subnetstr, "%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;
+       int i, l;
+       subnet_t *subnet;
+       uint16_t x[8];
+
+       cp();
+
+       subnet = new_subnet();
+
+       if(sscanf(subnetstr, "%hu.%hu.%hu.%hu/%d",
+                         &x[0], &x[1], &x[2], &x[3], &l) == 5) {
+               subnet->type = SUBNET_IPV4;
+               subnet->net.ipv4.prefixlength = l;
+
+               for(i = 0; i < 4; i++)
+                       subnet->net.ipv4.address.x[i] = x[i];
+
+               return subnet;
+       }
+
+       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],
+                         &l) == 9) {
+               subnet->type = SUBNET_IPV6;
+               subnet->net.ipv6.prefixlength = l;
+
+               for(i = 0; i < 8; i++)
+                       subnet->net.ipv6.address.x[i] = htons(x[i]);
+
+               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.prefixlength = 32;
+
+               for(i = 0; i < 4; i++)
+                       subnet->net.ipv4.address.x[i] = x[i];
+
+               return subnet;
+       }
+
+       if(sscanf(subnetstr, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx",
+                         &x[0], &x[1], &x[2], &x[3], &x[4], &x[5], &x[6], &x[7]) == 8) {
+               subnet->type = SUBNET_IPV6;
+               subnet->net.ipv6.prefixlength = 128;
+
+               for(i = 0; i < 8; i++)
+                       subnet->net.ipv6.address.x[i] = htons(x[i]);
+
+               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;
+
+               for(i = 0; i < 6; i++)
+                       subnet->net.mac.address.x[i] = x[i];
+
+               return subnet;
+       }
+
+       free(subnet);
+
+       return NULL;
 }
 
 char *net2str(subnet_t *subnet)
 {
-  char *netstr;
-cp
-  switch(subnet->type)
-    {
-      case SUBNET_MAC:
-        asprintf(&netstr, "%d,%hhx:%hhx:%hhx:%hhx:%hhx:%hhx", subnet->type,
-                   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, "%d,%lx/%lx", subnet->type, subnet->net.ipv4.address, subnet->net.ipv4.mask);
-        break;
-      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]);
-        break;
-      default:
-        netstr = NULL;
-    }
-cp
-  return netstr;
+       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",
+                                        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);
+                       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.prefixlength);
+                       break;
+
+               default:
+                       logger(DEBUG_ALWAYS, LOG_ERR,
+                                  _("net2str() was called with unknown subnet type %d, exiting!"),
+                                  subnet->type);
+                       cp_trace();
+                       exit(0);
+       }
+
+       return netstr;
 }
 
 /* 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
-  for(subnet = subnets; subnet != NULL; subnet = subnet->next)
-    {
-      if(subnet->type == SUBNET_MAC)
-        if(memcmp(&address, &subnet->net.mac.address, sizeof(address)) == 0)
-          break;
-    }
-cp
-  return subnet;
+       cp();
+
+       return avl_search(owner->subnet_tree, subnet);
 }
 
-subnet_t *lookup_subnet_ipv4(subnet_t *subnets, ipv4_t address)
+subnet_t *lookup_subnet_mac(mac_t *address)
 {
-  subnet_t *subnet;
-cp
-  for(subnet = subnets; subnet != NULL; subnet = subnet->next)
-    {
-      if(subnet->type == SUBNET_IPV4)
-        if((address & subnet->net.ipv4.mask) == subnet->net.ipv4.address)
-          break;
-    }
-cp
-  return subnet;
+       subnet_t subnet, *p;
+
+       cp();
+
+       subnet.type = SUBNET_MAC;
+       memcpy(&subnet.net.mac.address, address, sizeof(mac_t));
+       subnet.owner = NULL;
+
+       p = (subnet_t *) avl_search(subnet_tree, &subnet);
+
+       return p;
 }
 
-subnet_t *lookup_subnet_ipv6(subnet_t *subnets, ipv6_t address)
+subnet_t *lookup_subnet_ipv4(ipv4_t *address)
 {
-  subnet_t *subnet;
-  int i;
-cp
-  for(subnet = subnets; subnet != NULL; subnet = subnet->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;
-        }
-    }
-cp
-  return subnet;
+       subnet_t subnet, *p;
+
+       cp();
+
+       subnet.type = SUBNET_IPV4;
+       memcpy(&subnet.net.ipv4.address, address, sizeof(ipv4_t));
+       subnet.net.ipv4.prefixlength = 32;
+       subnet.owner = NULL;
+
+       do {
+               /* Go find subnet */
+
+               p = (subnet_t *) avl_search_closest_smaller(subnet_tree, &subnet);
+
+               /* Check if the found subnet REALLY matches */
+
+               if(p) {
+                       if(p->type != SUBNET_IPV4) {
+                               p = NULL;
+                               break;
+                       }
+
+                       if(!maskcmp(address, &p->net.ipv4.address, p->net.ipv4.prefixlength, sizeof(ipv4_t)))
+                               break;
+                       else {
+                               /* Otherwise, see if there is a bigger enclosing subnet */
+
+                               subnet.net.ipv4.prefixlength = p->net.ipv4.prefixlength - 1;
+                               maskcpy(&subnet.net.ipv4.address, &p->net.ipv4.address, subnet.net.ipv4.prefixlength, sizeof(ipv4_t));
+                       }
+               }
+       } while(p);
+
+       return p;
+}
+
+subnet_t *lookup_subnet_ipv6(ipv6_t *address)
+{
+       subnet_t subnet, *p;
+
+       cp();
+
+       subnet.type = SUBNET_IPV6;
+       memcpy(&subnet.net.ipv6.address, address, sizeof(ipv6_t));
+       subnet.net.ipv6.prefixlength = 128;
+       subnet.owner = NULL;
+
+       do {
+               /* Go find subnet */
+
+               p = (subnet_t *) avl_search_closest_smaller(subnet_tree, &subnet);
+
+               /* Check if the found subnet REALLY matches */
+
+               if(p) {
+                       if(p->type != SUBNET_IPV6)
+                               return NULL;
+
+                       if(!maskcmp(address, &p->net.ipv6.address, p->net.ipv6.prefixlength, sizeof(ipv6_t)))
+                               break;
+                       else {
+                               /* Otherwise, see if there is a bigger enclosing subnet */
+
+                               subnet.net.ipv6.prefixlength = p->net.ipv6.prefixlength - 1;
+                               maskcpy(&subnet.net.ipv6.address, &p->net.ipv6.address, subnet.net.ipv6.prefixlength, sizeof(ipv6_t));
+                       }
+               }
+       } while(p);
+
+       return p;
+}
+
+void dump_subnets(void)
+{
+       char *netstr;
+       subnet_t *subnet;
+       avl_node_t *node;
+
+       cp();
+
+       logger(DEBUG_ALWAYS, LOG_DEBUG, _("Subnet list:"));
+
+       for(node = subnet_tree->head; node; node = node->next) {
+               subnet = (subnet_t *) node->data;
+               netstr = net2str(subnet);
+               logger(DEBUG_ALWAYS, LOG_DEBUG, _(" %s owner %s"), netstr, subnet->owner->name);
+               free(netstr);
+       }
+
+       logger(DEBUG_ALWAYS, LOG_DEBUG, _("End of subnet list."));
 }