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-06-26 21:56:15 +08:00
|
|
|
#include "network.h"
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-06-24 22:42:55 +08:00
|
|
|
//Current time, unix format
|
|
|
|
#define unix_time() ((uint32_t)time(NULL))
|
|
|
|
|
2013-06-25 04:50:43 +08:00
|
|
|
//size of the client_id in bytes
|
2013-06-24 22:42:55 +08:00
|
|
|
#define CLIENT_ID_SIZE 32
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
typedef struct
|
|
|
|
{
|
2013-07-06 05:00:39 +08:00
|
|
|
uint8_t client_id[CLIENT_ID_SIZE];
|
2013-06-24 10:27:29 +08:00
|
|
|
IP_Port ip_port;
|
|
|
|
uint32_t timestamp;
|
2013-06-27 19:37:06 +08:00
|
|
|
uint32_t last_pinged;
|
2013-06-24 10:27:29 +08:00
|
|
|
}Client_data;
|
2013-06-25 08:51:07 +08:00
|
|
|
//maximum number of clients stored per friend.
|
|
|
|
#define MAX_FRIEND_CLIENTS 8
|
2013-06-24 10:27:29 +08:00
|
|
|
typedef struct
|
|
|
|
{
|
2013-07-06 05:00:39 +08:00
|
|
|
uint8_t client_id[CLIENT_ID_SIZE];
|
2013-06-25 08:51:07 +08:00
|
|
|
Client_data client_list[MAX_FRIEND_CLIENTS];
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
}Friend;
|
|
|
|
|
2013-06-25 08:51:07 +08:00
|
|
|
|
|
|
|
typedef struct
|
|
|
|
{
|
2013-07-06 05:00:39 +08:00
|
|
|
uint8_t client_id[CLIENT_ID_SIZE];
|
2013-06-25 08:51:07 +08:00
|
|
|
IP_Port ip_port;
|
|
|
|
}Node_format;
|
|
|
|
|
2013-06-24 12:40:33 +08:00
|
|
|
typedef struct
|
|
|
|
{
|
|
|
|
IP_Port ip_port;
|
|
|
|
uint32_t ping_id;
|
|
|
|
uint32_t timestamp;
|
|
|
|
|
|
|
|
}Pinged;
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
|
|
|
|
//Add a new friend to the friends list
|
2013-06-24 22:42:55 +08:00
|
|
|
//client_id must be CLIENT_ID_SIZE bytes long.
|
2013-06-26 00:11:11 +08:00
|
|
|
//returns 0 if success
|
|
|
|
//returns 1 if failure (friends list is full)
|
2013-07-06 05:00:39 +08:00
|
|
|
int addfriend(uint8_t * client_id);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
//Delete a friend from the friends list
|
2013-06-24 22:42:55 +08:00
|
|
|
//client_id must be CLIENT_ID_SIZE bytes long.
|
2013-06-24 10:27:29 +08:00
|
|
|
//returns 0 if success
|
|
|
|
//returns 1 if failure (client_id not in friends list)
|
2013-07-06 05:00:39 +08:00
|
|
|
int delfriend(uint8_t * client_id);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-06-24 11:57:06 +08:00
|
|
|
|
2013-06-24 10:27:29 +08:00
|
|
|
//Get ip of friend
|
2013-06-24 22:42:55 +08:00
|
|
|
//client_id must be CLIENT_ID_SIZE bytes long.
|
2013-06-24 10:27:29 +08:00
|
|
|
//ip must be 4 bytes long.
|
|
|
|
//port must be 2 bytes long.
|
|
|
|
//returns ip if success
|
2013-06-25 00:23:46 +08:00
|
|
|
//returns ip of 0 if failure (This means the friend is either offline or we have not found him yet.)
|
2013-06-25 21:19:01 +08:00
|
|
|
//returns ip of 1 if friend is not in list.
|
2013-07-06 05:00:39 +08:00
|
|
|
IP_Port getfriendip(uint8_t * client_id);
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
|
|
|
|
//Run this function at least a couple times per second (It's the main loop)
|
|
|
|
void doDHT();
|
|
|
|
|
2013-06-24 11:23:28 +08:00
|
|
|
//if we recieve a DHT packet we call this function so it can be handled.
|
2013-06-26 19:40:07 +08:00
|
|
|
//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-06-24 11:57:06 +08:00
|
|
|
//Use this function to bootstrap the client
|
|
|
|
//Sends a get nodes request to the given ip port
|
|
|
|
void bootstrap(IP_Port ip_port);
|
|
|
|
|
|
|
|
|
2013-06-24 11:23:28 +08:00
|
|
|
//TODO:
|
|
|
|
//Add functions to save and load the state(client list, friends list)
|
2013-06-24 10:27:29 +08:00
|
|
|
|
|
|
|
|
|
|
|
//Global variables
|
|
|
|
|
2013-06-24 12:40:33 +08:00
|
|
|
//Our client id
|
2013-07-06 05:00:39 +08:00
|
|
|
extern uint8_t self_client_id[CLIENT_ID_SIZE];
|
2013-06-24 12:40:33 +08:00
|
|
|
|
2013-06-24 11:23:28 +08:00
|
|
|
|
2013-06-26 00:11:11 +08:00
|
|
|
//TODO: Move these out of here and put them into the .c file.
|
2013-06-24 20:59:42 +08:00
|
|
|
//A list of the clients mathematically closest to ours.
|
2013-06-24 20:28:19 +08:00
|
|
|
#define LCLIENT_LIST 32
|
2013-06-24 20:59:42 +08:00
|
|
|
Client_data close_clientlist[LCLIENT_LIST];
|
2013-06-24 20:28:19 +08:00
|
|
|
|
2013-06-24 10:27:29 +08:00
|
|
|
|
2013-06-26 00:11:11 +08:00
|
|
|
//Hard maximum number of friends
|
|
|
|
#define MAX_FRIENDS 256
|
|
|
|
|
2013-06-24 10:27:29 +08:00
|
|
|
//Let's start with a static array for testing.
|
2013-06-26 00:11:11 +08:00
|
|
|
Friend friends_list[MAX_FRIENDS];
|
2013-06-24 12:40:33 +08:00
|
|
|
uint16_t num_friends;
|
|
|
|
|
|
|
|
//The list of ip ports along with the ping_id of what we sent them and a timestamp
|
2013-06-24 20:28:19 +08:00
|
|
|
#define LPING_ARRAY 128
|
|
|
|
|
|
|
|
Pinged pings[LPING_ARRAY];
|
|
|
|
|
|
|
|
#define LSEND_NODES_ARRAY LPING_ARRAY/2
|
2013-06-24 12:40:33 +08:00
|
|
|
|
2013-06-24 20:28:19 +08:00
|
|
|
Pinged send_nodes[LSEND_NODES_ARRAY];
|
2013-06-24 19:26:30 +08:00
|
|
|
|
|
|
|
|
2013-06-25 08:51:07 +08:00
|
|
|
#endif
|