2013-06-26 21:56:15 +08:00
|
|
|
/* DHT.h
|
|
|
|
*
|
|
|
|
* An implementation of the DHT as seen in docs/DHT.txt
|
|
|
|
*
|
2013-06-27 03:02:42 +08:00
|
|
|
|
|
|
|
Copyright (C) 2013 Tox project All Rights Reserved.
|
|
|
|
|
|
|
|
This file is part of Tox.
|
|
|
|
|
|
|
|
Tox is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
Tox is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with Tox. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
2013-06-26 21:56:15 +08:00
|
|
|
*/
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
|
2013-06-26 21:56:15 +08:00
|
|
|
#ifndef DHT_H
|
|
|
|
#define DHT_H
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-07-13 04:27:19 +08:00
|
|
|
#include "net_crypto.h"
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* Current time, unix format */
|
2013-06-24 22:42:55 +08:00
|
|
|
#define unix_time() ((uint32_t)time(NULL))
|
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* size of the client_id in bytes */
|
2013-07-13 04:27:19 +08:00
|
|
|
#define CLIENT_ID_SIZE crypto_box_PUBLICKEYBYTES
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* Add a new friend to the friends list
|
|
|
|
client_id must be CLIENT_ID_SIZE bytes long.
|
|
|
|
returns 0 if success
|
|
|
|
returns 1 if failure (friends list is full) */
|
2013-07-07 23:54:25 +08:00
|
|
|
int DHT_addfriend(uint8_t * client_id);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* Delete a friend from the friends list
|
|
|
|
client_id must be CLIENT_ID_SIZE bytes long.
|
|
|
|
returns 0 if success
|
|
|
|
returns 1 if failure (client_id not in friends list) */
|
2013-07-07 23:54:25 +08:00
|
|
|
int DHT_delfriend(uint8_t * client_id);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-06-24 11:57:06 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* Get ip of friend
|
|
|
|
client_id must be CLIENT_ID_SIZE bytes long.
|
|
|
|
ip must be 4 bytes long.
|
|
|
|
port must be 2 bytes long.
|
|
|
|
returns ip if success
|
|
|
|
returns ip of 0 if failure (This means the friend is either offline or we have not found him yet.)
|
|
|
|
returns ip of 1 if friend is not in list. */
|
2013-07-07 23:54:25 +08:00
|
|
|
IP_Port DHT_getfriendip(uint8_t * client_id);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* Run this function at least a couple times per second (It's the main loop) */
|
2013-06-24 10:27:29 +08:00
|
|
|
void doDHT();
|
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* if we receive a DHT packet we call this function so it can be handled.
|
|
|
|
return 0 if packet is handled correctly.
|
|
|
|
return 1 if it didn't handle the packet or if the packet was shit. */
|
2013-07-06 05:00:39 +08:00
|
|
|
int DHT_handlepacket(uint8_t * packet, uint32_t length, IP_Port source);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* Use this function to bootstrap the client
|
|
|
|
Sends a get nodes request to the given node with ip port and public_key */
|
2013-07-13 04:27:19 +08:00
|
|
|
void DHT_bootstrap(IP_Port ip_port, uint8_t * public_key);
|
2013-06-24 11:57:06 +08:00
|
|
|
|
|
|
|
|
2013-07-22 01:14:36 +08:00
|
|
|
|
|
|
|
/* ROUTING FUNCTIONS */
|
|
|
|
|
|
|
|
/* send the given packet to node with client_id
|
|
|
|
returns -1 if failure */
|
|
|
|
int route_packet(uint8_t * client_id, uint8_t * packet, uint32_t length);
|
|
|
|
|
|
|
|
/* Send the following packet to everyone who tells us they are connected to friend_id
|
|
|
|
returns the number of nodes it sent the packet to */
|
|
|
|
int route_tofriend(uint8_t * friend_id, uint8_t * packet, uint32_t length);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* NAT PUNCHING FUNCTIONS */
|
|
|
|
|
|
|
|
/* Puts all the different ips returned by the nodes for a friend_id into array ip_portlist
|
|
|
|
ip_portlist must be at least MAX_FRIEND_CLIENTS big
|
|
|
|
returns the number of ips returned
|
|
|
|
returns -1 if no such friend*/
|
|
|
|
int friend_ips(IP_Port * ip_portlist, uint8_t * friend_id);
|
|
|
|
|
|
|
|
|
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* SAVE/LOAD functions */
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* get the size of the DHT (for saving) */
|
2013-07-17 01:20:45 +08:00
|
|
|
uint32_t DHT_size();
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* save the DHT in data where data is an array of size DHT_size() */
|
2013-07-17 01:20:45 +08:00
|
|
|
void DHT_save(uint8_t * data);
|
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* load the DHT from data of size size;
|
|
|
|
return -1 if failure
|
|
|
|
return 0 if success */
|
2013-07-17 01:20:45 +08:00
|
|
|
int DHT_load(uint8_t * data, uint32_t size);
|
2013-06-24 11:23:28 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
/* returns 0 if we are not connected to the DHT
|
|
|
|
returns 1 if we are */
|
2013-07-17 07:45:20 +08:00
|
|
|
int DHT_isconnected();
|
2013-06-24 19:26:30 +08:00
|
|
|
|
2013-07-20 11:00:10 +08:00
|
|
|
#endif
|