/*
edge.c -- edge tree management
- Copyright (C) 2000-2002 Guus Sliepen <guus@sliepen.eu.org>,
- 2000-2002 Ivo Timmermans <ivo@o2w.nl>
+ Copyright (C) 2000-2006 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
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
- $Id: edge.c,v 1.1.2.12 2002/09/04 13:48:51 guus Exp $
+ $Id$
*/
-#include "config.h"
-
-#include <stdio.h>
-#include <syslog.h>
-#include <string.h>
-
-#include <avl_tree.h>
-#include <list.h>
+#include "system.h"
-#include "net.h" /* Don't ask. */
-#include "netutl.h"
-#include "config.h"
-#include "conf.h"
-#include <utils.h>
-#include "subnet.h"
+#include "splay_tree.h"
#include "edge.h"
+#include "logger.h"
+#include "netutl.h"
#include "node.h"
-
+#include "utils.h"
#include "xalloc.h"
-#include "system.h"
-avl_tree_t *edge_tree; /* Tree with all known edges (replaces active_tree) */
-avl_tree_t *edge_weight_tree; /* Tree with all edges, sorted on weight */
+splay_tree_t *edge_weight_tree; /* Tree with all edges, sorted on weight */
-int edge_compare(edge_t *a, edge_t *b)
-{
- int result;
+static int edge_compare(const edge_t *a, const edge_t *b) {
+ return strcmp(a->to->name, b->to->name);
+}
- result = strcmp(a->from->name, b->from->name);
+static int edge_weight_compare(const edge_t *a, const edge_t *b) {
+ int result;
- if(result)
- return result;
- else
- return strcmp(a->to->name, b->to->name);
-}
+ result = a->weight - b->weight;
-/* Evil edge_compare() from a parallel universe ;)
+ if(result)
+ return result;
-int edge_compare(edge_t *a, edge_t *b)
-{
- int result;
+ result = strcmp(a->from->name, b->from->name);
- return (result = strcmp(a->from->name, b->from->name)) || (result = strcmp(a->to->name, b->to->name)), result;
+ if(result)
+ return result;
+
+ return strcmp(a->to->name, b->to->name);
}
-*/
+void init_edges(void) {
+ cp();
-int edge_weight_compare(edge_t *a, edge_t *b)
-{
- int result;
+ edge_weight_tree = splay_alloc_tree((splay_compare_t) edge_weight_compare, NULL);
+}
- result = a->weight - b->weight;
+splay_tree_t *new_edge_tree(void) {
+ cp();
- if(result)
- return result;
- else
- return edge_compare(a, b);
+ return splay_alloc_tree((splay_compare_t) edge_compare, (splay_action_t) free_edge);
}
-void init_edges(void)
-{
-cp
- edge_tree = avl_alloc_tree((avl_compare_t)edge_compare, NULL);
- edge_weight_tree = avl_alloc_tree((avl_compare_t)edge_weight_compare, NULL);
-cp
-}
+void free_edge_tree(splay_tree_t *edge_tree) {
+ cp();
-avl_tree_t *new_edge_tree(void)
-{
-cp
- return avl_alloc_tree((avl_compare_t)edge_compare, NULL);
-cp
+ splay_delete_tree(edge_tree);
}
-void free_edge_tree(avl_tree_t *edge_tree)
-{
-cp
- avl_delete_tree(edge_tree);
-cp
-}
+void exit_edges(void) {
+ cp();
-void exit_edges(void)
-{
-cp
- avl_delete_tree(edge_tree);
-cp
+ splay_delete_tree(edge_weight_tree);
}
/* Creation and deletion of connection elements */
-edge_t *new_edge(void)
-{
- edge_t *e;
-cp
- e = (edge_t *)xmalloc_and_zero(sizeof(*e));
-cp
- return e;
+edge_t *new_edge(void) {
+ cp();
+
+ return xmalloc_and_zero(sizeof(edge_t));
}
-void free_edge(edge_t *e)
-{
-cp
- free(e);
-cp
+void free_edge(edge_t *e) {
+ cp();
+
+ sockaddrfree(&e->address);
+
+ free(e);
}
-void edge_add(edge_t *e)
-{
-cp
- avl_insert(edge_tree, e);
- avl_insert(edge_weight_tree, e);
- avl_insert(e->from->edge_tree, e);
-cp
- e->reverse = lookup_edge(e->to, e->from);
- if(e->reverse)
- e->reverse->reverse = e;
-cp
+void edge_add(edge_t *e) {
+ cp();
+
+ splay_insert(edge_weight_tree, e);
+ splay_insert(e->from->edge_tree, e);
+
+ e->reverse = lookup_edge(e->to, e->from);
+
+ if(e->reverse)
+ e->reverse->reverse = e;
}
-void edge_del(edge_t *e)
-{
-cp
- if(e->reverse)
- e->reverse->reverse = NULL;
-cp
- avl_delete(edge_tree, e);
- avl_delete(edge_weight_tree, e);
- avl_delete(e->from->edge_tree, e);
-cp
+void edge_del(edge_t *e) {
+ cp();
+
+ if(e->reverse)
+ e->reverse->reverse = NULL;
+
+ splay_delete(edge_weight_tree, e);
+ splay_delete(e->from->edge_tree, e);
}
-edge_t *lookup_edge(node_t *from, node_t *to)
-{
- edge_t v;
-cp
- v.from = from;
- v.to = to;
+edge_t *lookup_edge(node_t *from, node_t *to) {
+ edge_t v;
+
+ cp();
+
+ v.from = from;
+ v.to = to;
- return avl_search(edge_tree, &v);
+ return splay_search(from->edge_tree, &v);
}
-void dump_edges(void)
-{
- avl_node_t *node;
- edge_t *e;
- char *address;
-cp
- syslog(LOG_DEBUG, _("Edges:"));
-
- for(node = edge_tree->head; node; node = node->next)
- {
- e = (edge_t *)node->data;
- address = sockaddr2hostname(&e->address);
- syslog(LOG_DEBUG, _(" %s to %s at %s options %lx weight %d"),
- e->from->name, e->to->name, address,
- e->options, e->weight);
- free(address);
- }
-
- syslog(LOG_DEBUG, _("End of edges."));
-cp
+int dump_edges(struct evbuffer *out) {
+ splay_node_t *node, *node2;
+ node_t *n;
+ edge_t *e;
+ char *address;
+
+ cp();
+
+ for(node = node_tree->head; node; node = node->next) {
+ n = node->data;
+ for(node2 = n->edge_tree->head; node2; node2 = node2->next) {
+ e = node2->data;
+ address = sockaddr2hostname(&e->address);
+ if(evbuffer_add_printf(out,
+ _(" %s to %s at %s options %lx weight %d\n"),
+ e->from->name, e->to->name, address,
+ e->options, e->weight) == -1) {
+ free(address);
+ return errno;
+ }
+ free(address);
+ }
+ }
+
+ return 0;
}