class based connect behaviour

[meshlink] / src / meshlink.c

diff --git a/src/meshlink.c b/src/meshlink.c
index 65a82c97f192f65b48fdd6a125c5d7d7c36c42b7..95ab41c7ab03f072fa1547b42be6eee92b20cadf 100644 (file)
--- a/src/meshlink.c
+++ b/src/meshlink.c
@@ -1679,39 +1679,21 @@ void meshlink_whitelist(meshlink_handle_t *mesh, meshlink_node_t *node) {
 /* Hint that a hostname may be found at an address
  * See header file for detailed comment.
  */
-extern void meshlink_hint_address(meshlink_handle_t *mesh, meshlink_node_t *node, struct sockaddr *addr) {
+extern void meshlink_hint_address(meshlink_handle_t *mesh, meshlink_node_t *node, const struct sockaddr *addr) {
        if(!mesh || !node || !addr)
                return;
        
-       char *addr_str = malloc(MAX_ADDRESS_LENGTH*sizeof(char));
-       memset(addr_str, 0, MAX_ADDRESS_LENGTH*sizeof(char));
+       char *host = NULL, *port = NULL, *str = NULL;
+       sockaddr2str((const sockaddr_t *)addr, &host, &port);
 
-       char *port_str = malloc(MAX_PORT_LENGTH*sizeof(char));
-       memset(port_str, 0, MAX_PORT_LENGTH*sizeof(char));
-       
-       // extra byte for a space, and one to make sure string is null-terminated
-       int full_addr_len = MAX_ADDRESS_LENGTH + MAX_PORT_LENGTH + 2;
-
-       char *full_addr_str = malloc(full_addr_len*sizeof(char));
-       memset(full_addr_str, 0, full_addr_len*sizeof(char));
-       
-       // get address and port number
-       if(!get_ip_str(addr, addr_str, MAX_ADDRESS_LENGTH))
-               goto fail;
-       if(!get_port_str(addr, port_str, MAX_ADDRESS_LENGTH))
-               goto fail;
-
-       // append_config_file expects an address, a space, and then a port number
-       strcat(full_addr_str, addr_str);
-       strcat(full_addr_str, " ");
-       strcat(full_addr_str, port_str);
-       
-       append_config_file(mesh, node->name, "Address", full_addr_str);
+       if(host && port) {
+               xasprintf(&str, "%s %s", host, port);
+               append_config_file(mesh, node->name, "Address", str);
+       }
 
-fail:
-       free(addr_str);
-       free(port_str);
-       free(full_addr_str);
+       free(str);
+       free(host);
+       free(port);
 
        // @TODO do we want to fire off a connection attempt right away?
 }
@@ -1724,7 +1706,7 @@ static void __attribute__((destructor)) meshlink_exit(void) {
        crypto_exit();
 }
 
-int weight_from_dclass(dclass_t dclass)
+int cweight_from_dclass(dclass_t dclass)
 {
        switch(dclass)
        {
@@ -1740,3 +1722,84 @@ int weight_from_dclass(dclass_t dclass)
 
        return 9;
 }
+
+int max_ccount_from_dclass(dclass_t dclass)
+{
+       switch(dclass)
+       {
+       case BACKBONE:
+               return 10000;
+
+       case STATIONARY:
+               return 100;
+
+       case PORTABLE:
+               return 3;
+       }
+
+       return 3;
+}
+
+bool dclass_ccounts_satisfied(dclass_t dclass, splay_tree_t* counts, int total_count)
+{
+       // lookup keys
+       dclass_ccount_t key_portable;
+       key_portable.dclass = PORTABLE;
+
+       dclass_ccount_t key_stationary;
+       key_stationary.dclass = STATIONARY;
+
+       dclass_ccount_t key_backbone;
+       key_backbone.dclass = BACKBONE;
+
+       // check num of portable devices
+       dclass_ccount_t* portable = splay_search(counts, &key_portable);
+
+       if(portable)
+       {
+               if(portable->ccount > 9)
+                       return true;
+       }
+
+       // check num of stationary devices
+       dclass_ccount_t* stationary = splay_search(counts, &key_stationary);
+
+       if(stationary)
+       {
+               if(stationary->ccount > 6)
+                       return true;
+       }
+
+       // check num of backbone devices
+       dclass_ccount_t* backbone = splay_search(counts, &key_backbone);
+
+       if(backbone)
+       {
+               if(backbone->ccount > 3)
+                       return true;
+       }
+
+       // fallback
+       if(total_count >= max_ccount_from_dclass(dclass))
+               return true;
+
+       return false;
+}
+
+int dclass_ccount_compare(const void *a, const void *b)
+{
+       const dclass_ccount_t* da = a;
+       const dclass_ccount_t* db = b;
+
+       return da->dclass - db->dclass;
+}
+
+dclass_ccount_t* dclass_ccount_alloc()
+{
+       return xzalloc(sizeof(dclass_ccount_t));
+}
+
+void dclass_ccount_delete(void *c)
+{
+       free(c);
+}