tinc-vpn.org Git - tinc/blob - src/edge.c

   1 /*
   2     edge.c -- edge tree management
   3     Copyright (C) 2000,2001 Guus Sliepen <guus@sliepen.warande.net>,
   4                   2000,2001 Ivo Timmermans <itimmermans@bigfoot.com>
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program; if not, write to the Free Software
  18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19
  20     $Id: edge.c,v 1.1.2.5 2001/11/16 12:21:49 zarq Exp $
  21 */
  22
  23 #include "config.h"
  24
  25 #include <stdio.h>
  26 #include <syslog.h>
  27 #include <string.h>
  28
  29 #include <avl_tree.h>
  30 #include <list.h>
  31
  32 #include "net.h"        /* Don't ask. */
  33 #include "config.h"
  34 #include "conf.h"
  35 #include <utils.h>
  36 #include "subnet.h"
  37
  38 #include "xalloc.h"
  39 #include "system.h"
  40
  41 avl_tree_t *edge_tree;        /* Tree with all known edges (replaces active_tree) */
  42 avl_tree_t *edge_weight_tree; /* Tree with all edges, sorted on weight */
  43
  44 int edge_compare(edge_t *a, edge_t *b)
  45 {
  46   int result;
  47
  48   result = strcmp(a->from->name, b->from->name);
  49
  50   if(result)
  51     return result;
  52   else
  53     return strcmp(a->to->name, b->to->name);
  54 }
  55
  56 /* Evil edge_compare() from a parallel universe ;)
  57
  58 int edge_compare(edge_t *a, edge_t *b)
  59 {
  60   int result;
  61
  62   return (result = strcmp(a->from->name, b->from->name)) || (result = strcmp(a->to->name, b->to->name)), result;
  63 }
  64
  65 */
  66
  67 int edge_name_compare(edge_t *a, edge_t *b)
  68 {
  69   int result;
  70   char *name_a1, *name_a2, *name_b1, *name_b2;
  71
  72   if(strcmp(a->from->name, a->to->name) < 0)
  73     name_a1 = a->from->name, name_a2 = a->to->name;
  74   else
  75     name_a1 = a->to->name, name_a2 = a->from->name;
  76
  77   if(strcmp(b->from->name, b->to->name) < 0)
  78     name_b1 = b->from->name, name_b2 = b->to->name;
  79   else
  80     name_b1 = b->to->name, name_b2 = b->from->name;
  81
  82   result = strcmp(name_a1, name_b1);
  83
  84   if(result)
  85     return result;
  86   else
  87     return strcmp(name_a2, name_b2);
  88 }
  89
  90 int edge_weight_compare(edge_t *a, edge_t *b)
  91 {
  92   int result;
  93
  94   result = a->weight - b->weight;
  95
  96   if(result)
  97     return result;
  98   else
  99     return edge_name_compare(a, b);
 100 }
 101
 102 void init_edges(void)
 103 {
 104 cp
 105   edge_tree = avl_alloc_tree((avl_compare_t)edge_compare, NULL);
 106   edge_weight_tree = avl_alloc_tree((avl_compare_t)edge_weight_compare, NULL);
 107 cp
 108 }
 109
 110 avl_tree_t *new_edge_tree(void)
 111 {
 112 cp
 113   return avl_alloc_tree((avl_compare_t)edge_name_compare, NULL);
 114 cp
 115 }
 116
 117 void free_edge_tree(avl_tree_t *edge_tree)
 118 {
 119 cp
 120   avl_delete_tree(edge_tree);
 121 cp
 122 }
 123
 124 void exit_edges(void)
 125 {
 126 cp
 127   avl_delete_tree(edge_tree);
 128 cp
 129 }
 130
 131 /* Creation and deletion of connection elements */
 132
 133 edge_t *new_edge(void)
 134 {
 135   edge_t *e;
 136 cp
 137   e = (edge_t *)xmalloc_and_zero(sizeof(*e));
 138 cp
 139   return e;
 140 }
 141
 142 void free_edge(edge_t *e)
 143 {
 144 cp
 145   free(e);
 146 cp
 147 }
 148
 149 void edge_add(edge_t *e)
 150 {
 151 cp
 152   avl_insert(edge_tree, e);
 153   avl_insert(edge_weight_tree, e);
 154   avl_insert(e->from->edge_tree, e);
 155   avl_insert(e->to->edge_tree, e);
 156 cp
 157 }
 158
 159 void edge_del(edge_t *e)
 160 {
 161 cp
 162   avl_delete(edge_tree, e);
 163   avl_delete(edge_weight_tree, e);
 164   avl_delete(e->from->edge_tree, e);
 165   avl_delete(e->to->edge_tree, e);
 166 cp
 167 }
 168
 169 edge_t *lookup_edge(node_t *from, node_t *to)
 170 {
 171   edge_t v, *result;
 172 cp
 173   v.from = from;
 174   v.to = to;
 175
 176   result = avl_search(edge_tree, &v);
 177
 178   if(result)
 179     return result;
 180 cp
 181   v.from = to;
 182   v.to = from;
 183
 184   return avl_search(edge_tree, &v);
 185 }
 186
 187 void dump_edges(void)
 188 {
 189   avl_node_t *node;
 190   edge_t *e;
 191 cp
 192   syslog(LOG_DEBUG, _("Edges:"));
 193
 194   for(node = edge_tree->head; node; node = node->next)
 195     {
 196       e = (edge_t *)node->data;
 197       syslog(LOG_DEBUG, _(" %s - %s options %ld weight %d"),
 198              e->from->name, e->to->name, e->options, e->weight);
 199     }
 200
 201   syslog(LOG_DEBUG, _("End of edges."));
 202 cp
 203 }