Some small improvements added.

This commit is contained in:
irungentoo 2014-01-19 14:14:23 -05:00
parent e6e2112350
commit 3268f18a32
2 changed files with 18 additions and 1 deletions

View File

@ -360,10 +360,16 @@ static int get_somewhat_close_nodes(DHT *dht, uint8_t *client_id, Node_format *n
get_close_nodes_inner(dht, client_id, nodes_list, sa_family, get_close_nodes_inner(dht, client_id, nodes_list, sa_family,
dht->close_clientlist, LCLIENT_LIST, &num_nodes, is_LAN, want_good); dht->close_clientlist, LCLIENT_LIST, &num_nodes, is_LAN, want_good);
/*TODO uncomment this when hardening is added to close friend clients
for (i = 0; i < dht->num_friends; ++i) for (i = 0; i < dht->num_friends; ++i)
get_close_nodes_inner(dht, client_id, nodes_list, sa_family, get_close_nodes_inner(dht, client_id, nodes_list, sa_family,
dht->friends_list[i].client_list, MAX_FRIEND_CLIENTS, dht->friends_list[i].client_list, MAX_FRIEND_CLIENTS,
&num_nodes, is_LAN, want_good); &num_nodes, is_LAN, want_good);
*/
for (i = 0; i < dht->num_friends; ++i)
get_close_nodes_inner(dht, client_id, nodes_list, sa_family,
dht->friends_list[i].client_list, MAX_FRIEND_CLIENTS,
&num_nodes, is_LAN, 0);
return num_nodes; return num_nodes;
} }
@ -2243,6 +2249,14 @@ DHT *new_DHT(Net_Crypto *c)
#ifdef ENABLE_ASSOC_DHT #ifdef ENABLE_ASSOC_DHT
dht->assoc = new_Assoc_default(dht->self_public_key); dht->assoc = new_Assoc_default(dht->self_public_key);
#endif #endif
uint32_t i;
for (i = 0; i < DHT_FAKE_FRIEND_NUMBER; ++i) {
uint8_t random_key_bytes[CLIENT_ID_SIZE];
randombytes(random_key_bytes, sizeof(random_key_bytes));
DHT_addfriend(dht, random_key_bytes);
}
return dht; return dht;
} }

View File

@ -56,6 +56,9 @@
#define TOX_AF_INET 2 #define TOX_AF_INET 2
#define TOX_AF_INET6 10 #define TOX_AF_INET6 10
/* The number of "fake" friends to add (for optimization purposes and so our paths for the onion part are more random) */
#define DHT_FAKE_FRIEND_NUMBER 4
/* Functions to transfer ips safely across wire. */ /* Functions to transfer ips safely across wire. */
void to_net_family(IP *ip); void to_net_family(IP *ip);
void to_host_family(IP *ip); void to_host_family(IP *ip);