79 lines
2.3 KiB
C
79 lines
2.3 KiB
C
#include <stdlib.h>
|
|
#include <netdb.h>
|
|
#include <arpa/inet.h>
|
|
|
|
#include "multiaddr/multiaddr.h"
|
|
#include "libp2p/net/p2pnet.h"
|
|
#include "libp2p/conn/connection.h"
|
|
#include "libp2p/conn/transport_dialer.h"
|
|
|
|
/**
|
|
* An implementation of a tcp transport dialer
|
|
*/
|
|
|
|
|
|
struct TcpIp {
|
|
char* ip;
|
|
int port;
|
|
};
|
|
|
|
struct TcpIp* libp2p_conn_parse_ip_multiaddress(struct MultiAddress* addr) {
|
|
struct TcpIp* out = (struct TcpIp*)malloc(sizeof(struct TcpIp));
|
|
char* address = malloc(strlen(addr->string) + 1);
|
|
strcpy(address, addr->string);
|
|
char* tok = strtok(address, "/");
|
|
int pos = 0;
|
|
while (tok != NULL) {
|
|
switch (pos) {
|
|
case 1: {
|
|
out->ip = malloc(strlen(tok) + 1);
|
|
strcpy(out->ip, tok);
|
|
break;
|
|
}
|
|
case 3: {
|
|
out->port = strtol(tok, NULL, 10);
|
|
break;
|
|
}
|
|
}
|
|
tok = strtok(NULL, "/");
|
|
pos++;
|
|
}
|
|
//TODO: do a better job of parsing the results
|
|
return out;
|
|
}
|
|
|
|
int libp2p_conn_tcp_can_handle(struct MultiAddress* addr) {
|
|
return 1;
|
|
}
|
|
|
|
int libp2p_conn_tcp_read(const struct Connection* connection, char** out, size_t* num_bytes) {
|
|
int buffer_size = 65535;
|
|
*out = (char*)malloc(buffer_size);
|
|
ssize_t bytes = socket_read(connection->socket_handle, *out, buffer_size, 0);
|
|
*num_bytes = bytes;
|
|
return bytes > 0;
|
|
}
|
|
|
|
int libp2p_conn_tcp_write(const struct Connection* connection, const char* in, size_t num_bytes) {
|
|
ssize_t bytes = socket_write(connection->socket_handle, in, num_bytes, 0);
|
|
return bytes == num_bytes;
|
|
}
|
|
|
|
struct Connection* libp2p_conn_tcp_dial(struct TransportDialer* transport_dialer, struct MultiAddress* addr) {
|
|
struct Connection* conn = (struct Connection*) malloc(sizeof(struct Connection*));
|
|
conn->socket_handle = socket_open4();
|
|
struct TcpIp* results = libp2p_conn_parse_ip_multiaddress(addr);
|
|
struct hostent* host = gethostbyname(results->ip);
|
|
struct in_addr** addr_list = (struct in_addr**)host->h_addr_list;
|
|
socket_connect4(conn->socket_handle, (*addr_list[0]).s_addr, results->port);
|
|
conn->read = libp2p_conn_tcp_read;
|
|
conn->write = libp2p_conn_tcp_write;
|
|
return conn;
|
|
}
|
|
|
|
struct TransportDialer* libp2p_conn_tcp_transport_dialer_new(char* peer_id, struct PrivateKey* private_key) {
|
|
struct TransportDialer* out = libp2p_conn_transport_dialer_new(peer_id, private_key);
|
|
out->can_handle = libp2p_conn_tcp_can_handle;
|
|
out->dial = libp2p_conn_tcp_dial;
|
|
return out;
|
|
}
|