53e520e92f8d4980ac5d073f9601e409738fb2fc
[tinc] / src / node.c
1 /*
2     node.c -- node tree management
3     Copyright (C) 2001 Guus Sliepen <guus@sliepen.warande.net>,
4                   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: node.c,v 1.1.2.2 2001/10/27 12:13:17 guus Exp $
21 */
22
23 #include "config.h"
24
25 #include <string.h>
26 #include <syslog.h>
27
28 #include <avl_tree.h>
29 #include "node.h"
30 #include "net.h"
31 #include <utils.h>
32 #include <xalloc.h>
33
34 #include "system.h"
35
36 avl_tree_t *node_tree;          /* Known nodes, sorted by name */
37 avl_tree_t *node_udp_tree;      /* Known nodes, sorted by address and port */
38
39 int node_compare(node_t *a, node_t *b)
40 {
41   return strcmp(a->name, b->name);
42 }
43
44 int node_udp_compare(connection_t *a, connection_t *b)
45 {
46   if(a->address < b->address)
47     return -1;
48   else if (a->address > b->address)
49     return 1;
50   else
51     return a->port - b->port;
52 }
53
54 void init_nodes(void)
55 {
56 cp
57   node_tree = avl_alloc_tree((avl_compare_t)node_compare, NULL);
58   node_udp_tree = avl_alloc_tree((avl_compare_t)node_udp_compare, NULL);
59 cp
60 }
61
62 void exit_nodes(void)
63 {
64 cp
65   avl_delete_tree(node_tree);
66   avl_delete_tree(node_udp_tree);
67 cp
68 }
69
70 node_t *new_node(void)
71 {
72   node_t *n = (node_t *)xmalloc_and_zero(sizeof(*n));
73 cp
74   n->subnet_tree = avl_alloc_tree((avl_compare_t)subnet_compare, NULL);
75   n->queue = list_alloc((list_action_t)free);
76 cp
77   return n;
78 }
79
80 void free_node(node_t *n)
81 {
82 cp
83   if(n->queue)
84     list_delete_list(n->queue);
85   if(n->name)
86     free(n->name);
87   if(n->hostname)
88     free(n->hostname);
89   if(n->key)
90     free(n->key);
91   free(n);
92 cp
93 }
94
95 void node_add(node_t *n)
96 {
97 cp
98   avl_insert(node_tree, n);
99   avl_insert(node_udp_tree, n);
100 cp
101 }
102
103 void node_del(node_t *n)
104 {
105 cp
106   avl_delete(node_tree, n);
107   avl_delete(node_udp_tree, n);
108 cp
109 }
110
111 node_t *lookup_node(char *name)
112 {
113   node_t n;
114 cp
115   n.name = name;
116   return avl_search(node_tree, &n);
117 }
118
119 node_t *lookup_node_udp(ipv4_t address, port_t port)
120 {
121   node_t n;
122 cp
123   n.address = address;
124   n.port = port;
125   return avl_search(node_udp_tree, &n);
126 }
127
128 void dump_nodes(void)
129 {
130   avl_node_t *node;
131   node_t *n;
132 cp
133   syslog(LOG_DEBUG, _("Nodes:"));
134
135   for(node = node_tree->head; node; node = node->next)
136     {
137       n = (node_t *)node->data;
138       syslog(LOG_DEBUG, _(" %s at %s port %hd options %ld status %04x"),
139              n->name, n->hostname, n->port, n->options,
140              n->status);
141     }
142     
143   syslog(LOG_DEBUG, _("End of nodes."));
144 cp
145 }