93 lines
2.8 KiB
C
93 lines
2.8 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stdint.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <pthread.h>
|
|
#include "libp2p/net/p2pnet.h"
|
|
#include "libp2p/record/message.h"
|
|
#include "libp2p/net/multistream.h"
|
|
#include "ipfs/core/daemon.h"
|
|
|
|
#define BUF_SIZE 4096
|
|
|
|
void *ipfs_null_connection (void *ptr)
|
|
{
|
|
struct null_connection_params *connection_param;
|
|
//char b[BUF_SIZE];
|
|
//int len;
|
|
|
|
connection_param = (struct null_connection_params*) ptr;
|
|
|
|
// TODO: multistream + secio + message.
|
|
fprintf(stderr, "Connection %d, count %d\n", connection_param->socket, *(connection_param->count));
|
|
|
|
for(;;) {
|
|
if (libp2p_net_multistream_negotiate(connection_param->socket)) {
|
|
// we negotiated, now find out what they want
|
|
libp2p_net_multistream_handle_message(connection_param->socket);
|
|
} else {
|
|
break;
|
|
}
|
|
/*
|
|
len = socket_read(connection_param->socket, b, sizeof(b)-1, 0);
|
|
if (len > 0) {
|
|
while (b[len-1] == '\r' || b[len-1] == '\n') len--;
|
|
b[len] = '\0';
|
|
fprintf(stderr, "Recv: '%s'\n", b);
|
|
if (strcmp (b, "ping") == 0) {
|
|
socket_write(connection_param->socket, "pong", 4, 0);
|
|
}
|
|
} else if(len < 0) {
|
|
break;
|
|
}
|
|
*/
|
|
}
|
|
|
|
close (connection_param->socket); // close socket.
|
|
(*(connection_param->count))--; // update counter.
|
|
free (connection_param);
|
|
return (void*) 1;
|
|
}
|
|
|
|
void *ipfs_null_listen (void *ptr)
|
|
{
|
|
int socketfd, s, count = 0;
|
|
pthread_t pth_connection;
|
|
struct IpfsNodeListenParams *listen_param;
|
|
struct null_connection_params *connection_param;
|
|
|
|
listen_param = (struct IpfsNodeListenParams*) ptr;
|
|
|
|
if ((socketfd = socket_listen(socket_tcp4(), &(listen_param->ipv4), &(listen_param->port))) <= 0) {
|
|
perror("fail to init null router.");
|
|
exit (1);
|
|
}
|
|
|
|
fprintf(stderr, "Null listening on %d\n", listen_param->port);
|
|
|
|
for (;;) {
|
|
s = socket_accept4(socketfd, &(listen_param->ipv4), &(listen_param->port));
|
|
if (count >= CONNECTIONS) { // limit reached.
|
|
close (s);
|
|
continue;
|
|
}
|
|
|
|
count++;
|
|
connection_param = malloc (sizeof (struct null_connection_params));
|
|
if (connection_param) {
|
|
connection_param->socket = s;
|
|
connection_param->count = &count;
|
|
connection_param->local_node = listen_param->local_node;
|
|
// Create pthread for ipfs_null_connection.
|
|
if (pthread_create(&pth_connection, NULL, ipfs_null_connection, connection_param)) {
|
|
fprintf(stderr, "Error creating thread for connection %d\n", count);
|
|
close (s);
|
|
} else {
|
|
pthread_detach (pth_connection);
|
|
}
|
|
}
|
|
}
|
|
|
|
return (void*) 2;
|
|
}
|