agorise
/
c-libp2p
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

debugging daemon

yamux
John Jones 2017-03-24 14:11:54 -05:00
parent 41ef0e5492
commit 158012858a
3 changed files with 77 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/libp2p/routing/kademlia.h
View File

@ -1,6 +1,9 @@
#pragma once
#include "multiaddr/multiaddr.h"
int start_kademlia(int sock, int family, char* peer_id, int timeout);
int start_kademlia_multiaddress(struct MultiAddress* multiaddress, char* peer_id, int timeout);
void stop_kademlia (void);
void *kademlia_thread (void *ptr);

45
routing/dht.c
View File

@ -1924,11 +1924,21 @@ bucket_maintenance(int af)
return 0;
}
int
dht_periodic(const void *buf, size_t buflen,
const struct sockaddr *from, int fromlen,
time_t *tosleep,
dht_callback *callback, void *closure)
/***
* Called when something is received from the network or
* the network times out (things that should be done
* periodically
* @param buf what came in from the network
* @param buflen the size of buf
* @param from where it came from
* @param fromlen
* @param tosleep
* @param callback
* @param closure
* @returns ??
*/
int dht_periodic(const void *buf, size_t buflen, const struct sockaddr *from, int fromlen,
time_t *tosleep, dht_callback *callback, void *closure)
{
dht_gettimeofday(&now, NULL);
@ -2412,7 +2422,7 @@ send_ping(const struct sockaddr *sa, int salen,
COPY(buf, i, tid, tid_len, 512);
ADD_V(buf, i, 512);
rc = snprintf(buf + i, 512 - i, "1:y1:qe"); INC(i, rc, 512);
return dht_send(buf, i, 0, sa, salen);
return dht_send(buf, i, MSG_NOSIGNAL, sa, salen);
fail:
errno = ENOSPC;
@ -2671,8 +2681,20 @@ send_get_peers(const struct sockaddr *sa, int salen,
return -1;
}
int
send_announce_peer(const struct sockaddr *sa, int salen,
/***
* Announce to a peer that you can provide a hash
* @param sa who to announce to
* @param salen theh length of the sockaddr
* @param tid transaction id
* @param tid_len length of transaction id
* @param infohash
* @param port
* @param token
* @param token_len
* @param confirm
* @returns ???
*/
int send_announce_peer(const struct sockaddr *sa, int salen,
unsigned char *tid, int tid_len,
unsigned char *infohash, unsigned short port,
unsigned char *token, int token_len, int confirm)
@ -2701,6 +2723,13 @@ send_announce_peer(const struct sockaddr *sa, int salen,
return -1;
}
/**
* This looks like a response to a send_announce_peer
* @param sa
* @param salen
* @param tid transaction id
* @param tid_len length of transaction id
*/
static int
send_peer_announced(const struct sockaddr *sa, int salen,
unsigned char *tid, int tid_len)

50
routing/kademlia.c
View File

@ -34,7 +34,7 @@ struct bs_struct {
pthread_t pth_kademlia, pth_announce;
time_t tosleep = 0;
int ksock = -1;
int kfd = -1;
int net_family = 0;
volatile int8_t searching = 0; // search lock, -1 to busy, 0 to free, 1 to running.
volatile char hash[20]; // hash to be search or announce.
@ -166,12 +166,26 @@ static void callback(void *closure, int event, const unsigned char *info_hash, c
}
}
int start_kademlia(int sock, int family, char* peer_id, int timeout)
int start_kademlia_multiaddress(struct MultiAddress* address, char* peer_id, int timeout) {
int port = multiaddress_get_ip_port(address);
int family = multiaddress_get_ip_family(address);
int fd = socket(family, SOCK_STREAM, 0);
struct sockaddr_in serv_addr;
serv_addr.sin_family = family;
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_addr.sin_port = port;
bind (fd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
return start_kademlia(fd, family, peer_id, timeout);
}
int start_kademlia(int net_fd, int family, char* peer_id, int timeout)
{
int rc, i, len;
unsigned char id[sizeof hash];
struct sockaddr_in sa;
dht_debug = stderr;
len = sizeof(bootstrap_list) / sizeof(bootstrap_list[0]); // array length
if (len > MAX_BOOTSTRAP_NODES) {
@ -196,9 +210,9 @@ int start_kademlia(int sock, int family, char* peer_id, int timeout)
dht_hash (id, sizeof(id), peer_id, strlen(peer_id), NULL, 0, NULL, 0);
if (family == AF_INET6) {
rc = dht_init(-1, sock, id, NULL);
rc = dht_init(-1, net_fd, id, NULL);
} else {
rc = dht_init(sock, -1, id, NULL);
rc = dht_init(net_fd, -1, id, NULL);
}
if (rc < 0) {
return rc;
@ -214,14 +228,17 @@ int start_kademlia(int sock, int family, char* peer_id, int timeout)
a dump) and you already know their ids, it's better to use
dht_insert_node. If the ids are incorrect, the DHT will recover. */
for(i = 0; i < num_bootstrap_nodes; i++) {
// for debugging
int retVal =
dht_ping_node((struct sockaddr*)&bootstrap_nodes[i],
sizeof (bootstrap_nodes[i]));
fprintf(stderr, "ping returned %d\n", retVal);
usleep(random() % 100000);
}
// TODO: Read cache nodes from file and load using dht_insert_node.
ksock = sock;
kfd = net_fd;
net_family = family;
tosleep = timeout;
@ -235,7 +252,7 @@ int start_kademlia(int sock, int family, char* peer_id, int timeout)
void stop_kademlia (void)
{
if (ksock != -1) {
if (kfd != -1) {
closing = 1;
pthread_cancel(pth_announce);
@ -245,8 +262,8 @@ void stop_kademlia (void)
dht_uninit();
close (ksock);
ksock = -1;
close (kfd);
kfd = -1;
}
}
@ -264,8 +281,8 @@ void *kademlia_thread (void *ptr)
tv.tv_usec = random() % 1000000;
FD_ZERO(&readfds);
FD_SET(ksock, &readfds);
rc = select(ksock + 1, &readfds, NULL, NULL, &tv);
FD_SET(kfd, &readfds);
rc = select(kfd + 1, &readfds, NULL, NULL, &tv);
if(rc < 0) {
if(errno != EINTR) {
perror("select");
@ -273,10 +290,17 @@ void *kademlia_thread (void *ptr)
}
}
if(rc > 0 && FD_ISSET(ksock, &readfds)) {
if(rc > 0 && FD_ISSET(kfd, &readfds)) {
fromlen = sizeof(from);
rc = recvfrom(ksock, buf, sizeof(buf) - 1, 0,
rc = recvfrom(kfd, buf, sizeof(buf) - 1, 0,
(struct sockaddr*)&from, &fromlen);
if (rc < 0) {
if (errno == EAGAIN || errno == EWOULDBLOCK) {
continue;
} else {
fprintf(stderr, "kademlia_thread:recvfrom failed with %d\n", errno);
}
}
buf[rc] = '\0';
rc = dht_periodic(buf, rc, (struct sockaddr*)&from, fromlen,
&tosleep, callback, NULL);
@ -423,7 +447,7 @@ struct MultiAddress** search_kademlia(char* peer_id, int timeout)
struct search_struct *rp; // result pointer
struct MultiAddress **ret;
if (ksock == -1) {
if (kfd == -1) {
return NULL; // start thread first.
}