#include "net.h"
#include "netutl.h"
#include "node.h"
-#include "splay_tree.h"
#include "utils.h"
#include "xalloc.h"
-static int node_compare(const node_t *a, const node_t *b) {
- return strcmp(a->name, b->name);
-}
-
void init_nodes(meshlink_handle_t *mesh) {
- mesh->nodes = splay_alloc_tree((splay_compare_t) node_compare, (splay_action_t) free_node);
- mesh->node_udp_cache = hash_alloc(0x100, sizeof(sockaddr_t));
+ mesh->peer = NULL;
}
void exit_nodes(meshlink_handle_t *mesh) {
- if(mesh->node_udp_cache) {
- hash_free(mesh->node_udp_cache);
+ if(mesh->peer) {
+ free_node(mesh->peer);
}
- if(mesh->nodes) {
- splay_delete_tree(mesh->nodes);
- }
-
- mesh->node_udp_cache = NULL;
- mesh->nodes = NULL;
+ mesh->peer = NULL;
}
node_t *new_node(void) {
node_t *n = xzalloc(sizeof(*n));
- n->edge_tree = new_edge_tree();
- n->mtu = MTU;
- n->maxmtu = MTU;
n->devclass = DEV_CLASS_UNKNOWN;
return n;
void free_node(node_t *n) {
n->status.destroyed = true;
- utcp_exit(n->utcp);
-
- if(n->edge_tree) {
- free_edge_tree(n->edge_tree);
- }
-
- sockaddrfree(&n->address);
-
ecdsa_free(n->ecdsa);
- sptps_stop(&n->sptps);
-
- if(n->mtutimeout.cb) {
- abort();
- }
free(n->name);
free(n->canonical_address);
}
void node_add(meshlink_handle_t *mesh, node_t *n) {
- n->mesh = mesh;
- splay_insert(mesh->nodes, n);
-}
-
-void node_del(meshlink_handle_t *mesh, node_t *n) {
- timeout_del(&mesh->loop, &n->mtutimeout);
-
- for splay_each(edge_t, e, n->edge_tree) {
- edge_del(mesh, e);
+ if(n == mesh->self) {
+ return;
}
- splay_delete(mesh->nodes, n);
-}
-
-node_t *lookup_node(meshlink_handle_t *mesh, const char *name) {
- const node_t n = {.name = (char *)name};
- node_t *result;
-
- result = splay_search(mesh->nodes, &n);
-
- return result;
-}
-
-node_t *lookup_node_udp(meshlink_handle_t *mesh, const sockaddr_t *sa) {
- return hash_search(mesh->node_udp_cache, sa);
+ assert(!mesh->peer);
+ n->mesh = mesh;
+ mesh->peer = n;
}
-void update_node_udp(meshlink_handle_t *mesh, node_t *n, const sockaddr_t *sa) {
+void node_del(meshlink_handle_t *mesh, node_t *n) {
if(n == mesh->self) {
- logger(mesh, MESHLINK_WARNING, "Trying to update UDP address of mesh->self!");
return;
}
- hash_insert(mesh->node_udp_cache, &n->address, NULL);
-
- if(sa) {
- n->address = *sa;
- n->sock = 0;
-
- for(int i = 0; i < mesh->listen_sockets; i++) {
- if(mesh->listen_socket[i].sa.sa.sa_family == sa->sa.sa_family) {
- n->sock = i;
- break;
- }
- }
-
- hash_insert(mesh->node_udp_cache, sa, n);
-
- node_add_recent_address(mesh, n, sa);
+ assert(mesh->peer && mesh->peer == n);
+ free_node(n);
+ mesh->peer = NULL;
+}
- if(mesh->log_level <= MESHLINK_DEBUG) {
- char *hostname = sockaddr2hostname(&n->address);
- logger(mesh, MESHLINK_DEBUG, "UDP address of %s set to %s", n->name, hostname);
- free(hostname);
- }
+node_t *lookup_node(meshlink_handle_t *mesh, const char *name) {
+ if(mesh->peer && !strcmp(name, mesh->peer->name)) {
+ return mesh->peer;
+ } else if(!strcmp(name, mesh->self->name)) {
+ return mesh->self;
+ } else {
+ return NULL;
}
}