When purging nodes, only delete them if nobody references them anymore.
[tinc] / src / node.c
1 /*
2     node.c -- node tree management
3     Copyright (C) 2001-2003 Guus Sliepen <guus@sliepen.eu.org>,
4                   2001-2003 Ivo Timmermans <ivo@o2w.nl>
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.27 2003/08/22 11:18:42 guus Exp $
21 */
22
23 #include "system.h"
24
25 #include "avl_tree.h"
26 #include "logger.h"
27 #include "net.h"
28 #include "netutl.h"
29 #include "node.h"
30 #include "utils.h"
31 #include "xalloc.h"
32
33 avl_tree_t *node_tree;                  /* Known nodes, sorted by name */
34 avl_tree_t *node_udp_tree;              /* Known nodes, sorted by address and port */
35
36 node_t *myself;
37
38 static int node_compare(const node_t *a, const node_t *b)
39 {
40         return strcmp(a->name, b->name);
41 }
42
43 static int node_udp_compare(const node_t *a, const node_t *b)
44 {
45         int result;
46
47         cp();
48
49         result = sockaddrcmp(&a->address, &b->address);
50
51         if(result)
52                 return result;
53
54         return (a->name && b->name) ? strcmp(a->name, b->name) : 0;
55 }
56
57 void init_nodes(void)
58 {
59         cp();
60
61         node_tree = avl_alloc_tree((avl_compare_t) node_compare, (avl_action_t) free_node);
62         node_udp_tree = avl_alloc_tree((avl_compare_t) node_udp_compare, NULL);
63 }
64
65 void exit_nodes(void)
66 {
67         cp();
68
69         avl_delete_tree(node_udp_tree);
70         avl_delete_tree(node_tree);
71 }
72
73 node_t *new_node(void)
74 {
75         node_t *n = (node_t *) xmalloc_and_zero(sizeof(*n));
76
77         cp();
78
79         n->subnet_tree = new_subnet_tree();
80         n->edge_tree = new_edge_tree();
81         n->queue = list_alloc((list_action_t) free);
82         EVP_CIPHER_CTX_init(&n->packet_ctx);
83
84         return n;
85 }
86
87 void free_node(node_t *n)
88 {
89         cp();
90
91         if(n->queue)
92                 list_delete_list(n->queue);
93
94         if(n->name)
95                 free(n->name);
96
97         if(n->hostname)
98                 free(n->hostname);
99
100         if(n->key)
101                 free(n->key);
102
103         if(n->subnet_tree)
104                 free_subnet_tree(n->subnet_tree);
105
106         if(n->edge_tree)
107                 free_edge_tree(n->edge_tree);
108
109         sockaddrfree(&n->address);
110
111         EVP_CIPHER_CTX_cleanup(&n->packet_ctx);
112         
113         free(n);
114 }
115
116 void node_add(node_t *n)
117 {
118         cp();
119
120         avl_insert(node_tree, n);
121         avl_insert(node_udp_tree, n);
122 }
123
124 void node_del(node_t *n)
125 {
126         avl_node_t *node, *next;
127         edge_t *e;
128         subnet_t *s;
129
130         cp();
131
132         for(node = n->subnet_tree->head; node; node = next) {
133                 next = node->next;
134                 s = (subnet_t *) node->data;
135                 subnet_del(n, s);
136         }
137
138         for(node = n->edge_tree->head; node; node = next) {
139                 next = node->next;
140                 e = (edge_t *) node->data;
141                 edge_del(e);
142         }
143
144         avl_delete(node_tree, n);
145         avl_delete(node_udp_tree, n);
146 }
147
148 node_t *lookup_node(char *name)
149 {
150         node_t n = {0};
151
152         cp();
153         
154         n.name = name;
155
156         return avl_search(node_tree, &n);
157 }
158
159 node_t *lookup_node_udp(const sockaddr_t *sa)
160 {
161         node_t n = {0};
162
163         cp();
164
165         n.address = *sa;
166         n.name = NULL;
167
168         return avl_search(node_udp_tree, &n);
169 }
170
171 void dump_nodes(void)
172 {
173         avl_node_t *node;
174         node_t *n;
175
176         cp();
177
178         logger(LOG_DEBUG, _("Nodes:"));
179
180         for(node = node_tree->head; node; node = node->next) {
181                 n = (node_t *) node->data;
182                 logger(LOG_DEBUG, _(" %s at %s cipher %d digest %d maclength %d compression %d options %lx status %04x nexthop %s via %s"),
183                            n->name, n->hostname, n->cipher ? n->cipher->nid : 0,
184                            n->digest ? n->digest->type : 0, n->maclength, n->compression,
185                            n->options, *(uint32_t *)&n->status, n->nexthop ? n->nexthop->name : "-",
186                            n->via ? n->via->name : "-");
187         }
188
189         logger(LOG_DEBUG, _("End of nodes."));
190 }