/*
edge.c -- edge tree management
- Copyright (C) 2000-2002 Guus Sliepen <guus@sliepen.warande.net>,
- 2000-2002 Ivo Timmermans <itimmermans@bigfoot.com>
+ 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.8 2002/03/22 11:43:46 guus Exp $
+ $Id$
*/
-#include "config.h"
+#include "system.h"
-#include <stdio.h>
-#include <syslog.h>
-#include <string.h>
+#include "splay_tree.h"
+#include "edge.h"
+#include "logger.h"
+#include "netutl.h"
+#include "node.h"
+#include "utils.h"
+#include "xalloc.h"
-#include <avl_tree.h>
-#include <list.h>
+splay_tree_t *edge_weight_tree; /* Tree with all edges, sorted on weight */
-#include "net.h" /* Don't ask. */
-#include "netutl.h"
-#include "config.h"
-#include "conf.h"
-#include <utils.h>
-#include "subnet.h"
+static int edge_compare(const edge_t *a, const edge_t *b) {
+ return strcmp(a->to->name, b->to->name);
+}
-#include "xalloc.h"
-#include "system.h"
+static int edge_weight_compare(const edge_t *a, const edge_t *b) {
+ int result;
-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 */
+ result = a->weight - b->weight;
-int edge_compare(edge_t *a, edge_t *b)
-{
- int result;
+ if(result)
+ return result;
- result = strcmp(a->from.node->name, b->from.node->name);
+ result = strcmp(a->from->name, b->from->name);
- if(result)
- return result;
- else
- return strcmp(a->to.node->name, b->to.node->name);
-}
+ if(result)
+ return result;
-/* Evil edge_compare() from a parallel universe ;)
+ return strcmp(a->to->name, b->to->name);
+}
-int edge_compare(edge_t *a, edge_t *b)
-{
- int result;
+void init_edges(void) {
+ cp();
- return (result = strcmp(a->from.node->name, b->from.node->name)) || (result = strcmp(a->to.node->name, b->to.node->name)), result;
+ edge_weight_tree = splay_alloc_tree((splay_compare_t) edge_weight_compare, NULL);
}
-*/
+splay_tree_t *new_edge_tree(void) {
+ cp();
-int edge_name_compare(edge_t *a, edge_t *b)
-{
- int result;
- char *name_a1, *name_a2, *name_b1, *name_b2;
+ return splay_alloc_tree((splay_compare_t) edge_compare, (splay_action_t) free_edge);
+}
- if(strcmp(a->from.node->name, a->to.node->name) < 0)
- name_a1 = a->from.node->name, name_a2 = a->to.node->name;
- else
- name_a1 = a->to.node->name, name_a2 = a->from.node->name;
+void free_edge_tree(splay_tree_t *edge_tree) {
+ cp();
- if(strcmp(b->from.node->name, b->to.node->name) < 0)
- name_b1 = b->from.node->name, name_b2 = b->to.node->name;
- else
- name_b1 = b->to.node->name, name_b2 = b->from.node->name;
+ splay_delete_tree(edge_tree);
+}
- result = strcmp(name_a1, name_b1);
+void exit_edges(void) {
+ cp();
- if(result)
- return result;
- else
- return strcmp(name_a2, name_b2);
+ splay_delete_tree(edge_weight_tree);
}
-int edge_weight_compare(edge_t *a, edge_t *b)
-{
- int result;
+/* Creation and deletion of connection elements */
- result = a->weight - b->weight;
+edge_t *new_edge(void) {
+ cp();
- if(result)
- return result;
- else
- return edge_name_compare(a, b);
+ return xmalloc_and_zero(sizeof(edge_t));
}
-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(edge_t *e) {
+ cp();
+
+ sockaddrfree(&e->address);
-avl_tree_t *new_edge_tree(void)
-{
-cp
- return avl_alloc_tree((avl_compare_t)edge_name_compare, NULL);
-cp
+ free(e);
}
-void free_edge_tree(avl_tree_t *edge_tree)
-{
-cp
- avl_delete_tree(edge_tree);
-cp
-}
+void edge_add(edge_t *e) {
+ cp();
-void exit_edges(void)
-{
-cp
- avl_delete_tree(edge_tree);
-cp
-}
+ splay_insert(edge_weight_tree, e);
+ splay_insert(e->from->edge_tree, e);
-/* Creation and deletion of connection elements */
+ e->reverse = lookup_edge(e->to, e->from);
-edge_t *new_edge(void)
-{
- edge_t *e;
-cp
- e = (edge_t *)xmalloc_and_zero(sizeof(*e));
-cp
- return e;
+ if(e->reverse)
+ e->reverse->reverse = e;
}
-void free_edge(edge_t *e)
-{
-cp
- free(e);
-cp
-}
+void edge_del(edge_t *e) {
+ cp();
-void edge_add(edge_t *e)
-{
-cp
- avl_insert(edge_tree, e);
- avl_insert(edge_weight_tree, e);
- avl_insert(e->from.node->edge_tree, e);
- avl_insert(e->to.node->edge_tree, e);
-cp
-}
+ if(e->reverse)
+ e->reverse->reverse = NULL;
-void edge_del(edge_t *e)
-{
-cp
- avl_delete(edge_tree, e);
- avl_delete(edge_weight_tree, e);
- avl_delete(e->from.node->edge_tree, e);
- avl_delete(e->to.node->edge_tree, e);
-cp
+ 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, *result;
-cp
- v.from.node = from;
- v.to.node = to;
-
- result = avl_search(edge_tree, &v);
+edge_t *lookup_edge(node_t *from, node_t *to) {
+ edge_t v;
+
+ cp();
- if(result)
- return result;
-cp
- v.from.node = to;
- v.to.node = from;
+ 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 *from_tcp, *from_udp;
- char *to_tcp, *to_udp;
-cp
- syslog(LOG_DEBUG, _("Edges:"));
-
- for(node = edge_tree->head; node; node = node->next)
- {
- e = (edge_t *)node->data;
-// from_tcp = sockaddr2hostname(&e->from.tcpaddress);
- from_udp = sockaddr2hostname(&e->from.udpaddress);
-// to_tcp = sockaddr2hostname(&e->to.tcpaddress);
- to_udp = sockaddr2hostname(&e->to.udpaddress);
- syslog(LOG_DEBUG, _(" %s at %s - %s at %s options %ld weight %d"),
- e->from.node->name, from_udp,
- e->to.node->name, to_udp,
- e->options, e->weight);
-// free(from_tcp);
- free(from_udp);
-// free(to_tcp);
- free(to_udp);
- }
-
- 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;
}