- Fixed all (except 2) compiler warnings gcc -Wall gave.

[tinc] / src / connection.c

diff --git a/src/connection.c b/src/connection.c
index dee0472..4d0b3a8 100644 (file)
--- a/src/connection.c
+++ b/src/connection.c
@@ -17,7 +17,7 @@
     along with this program; if not, write to the Free Software
     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 
-    $Id: connection.c,v 1.1.2.1 2000/11/20 19:12:11 guus Exp $
+    $Id: connection.c,v 1.1.2.5 2000/11/22 22:18:03 guus Exp $
 */
 
 #include "config.h"
@@ -32,6 +32,7 @@
 #include "config.h"
 #include "conf.h"
 #include <utils.h>
+#include "subnet.h"
 
 #include "xalloc.h"
 #include "system.h"
@@ -39,11 +40,23 @@
 /* Root of the connection list */
 
 rbltree_t *connection_tree;
+rbltree_t *id_tree;
+
 connection_t *myself = NULL;
 
 /* Initialization and callbacks */
 
 int connection_compare(connection_t *a, connection_t *b)
+{
+  ipv4_t result;
+  result = a->address - b->address;
+  if(result)
+    return result;
+  else
+    return a->port - b->port;
+}
+
+int id_compare(connection_t *a, connection_t *b)
 {
   return strcmp(a->name, b->name);
 }
@@ -51,6 +64,7 @@ int connection_compare(connection_t *a, connection_t *b)
 void init_connections(void)
 {
   connection_tree = new_rbltree((rbl_compare_t)connection_compare, (rbl_action_t)free_connection);
+  id_tree = new_rbltree((rbl_compare_t)id_compare, NULL);
 }
 
 /* Creation and deletion of connection elements */
@@ -61,6 +75,8 @@ connection_t *new_connection(void)
 cp
   /* initialise all those stupid pointers at once */
   memset(p, '\0', sizeof(*p));
+
+  p->subnet_tree = new_rbltree((rbl_compare_t)subnet_compare, NULL);
 cp
   return p;
 }
@@ -111,6 +127,7 @@ cp
 void destroy_connection_tree(void)
 {
 cp
+  rbl_delete_rbltree(id_tree);
   rbl_delete_rbltree(connection_tree);
 cp
 }
@@ -124,22 +141,43 @@ cp
 cp
 }
 
+void id_add(connection_t *cl)
+{
+cp
+  rbl_insert(id_tree, cl);
+cp
+}
+
 void connection_del(connection_t *cl)
 {
 cp
+  rbl_delete(id_tree, cl);
   rbl_delete(connection_tree, cl);
 cp
 }
 
 /* Lookup functions */
 
-connection_t *lookup_id(char *name)
+connection_t *lookup_connection(ipv4_t address, short unsigned int port)
 {
   connection_t cl;
 cp
-  cl.name = name;
+  cl.address = address;
+  cl.port = port;
+
   return rbl_search(connection_tree, &cl);
+}
+
+connection_t *lookup_id(char *name)
+{
+  connection_t cl, *p;
 cp
+  cl.name = name;
+  p = rbl_search(id_tree, &cl);
+  if(p && p->status.active)
+    return p;
+  else
+    return NULL;
 }
 
 /* Debugging */