142 lines
3.9 KiB
C
142 lines
3.9 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <sys/time.h>
|
|
#include <arpa/inet.h>
|
|
#include "libp2p/net/p2pnet.h"
|
|
#include "libp2p/net/multistream.h"
|
|
#include "libp2p/record/message.h"
|
|
#include "libp2p/secio/secio.h"
|
|
#include "ipfs/repo/fsrepo/fs_repo.h"
|
|
|
|
#define BUF_SIZE 4096
|
|
|
|
int ipfs_ping (int argc, char **argv)
|
|
{
|
|
unsigned char* results = NULL;
|
|
size_t results_size = 0;
|
|
int port = 0;
|
|
char* ip = NULL;
|
|
struct SessionContext session;
|
|
|
|
// read the configuration
|
|
struct FSRepo* fs_repo;
|
|
if (!ipfs_repo_fsrepo_new(NULL, NULL, &fs_repo))
|
|
return 0;
|
|
|
|
// open the repository and read the file
|
|
if (!ipfs_repo_fsrepo_open(fs_repo)) {
|
|
ipfs_repo_fsrepo_free(fs_repo);
|
|
return 0;
|
|
}
|
|
|
|
//TODO: handle multiaddress
|
|
if (strstr(argv[2], "/ipfs/") != NULL) {
|
|
// look in peerstore
|
|
} else {
|
|
// the way using multistream
|
|
//TODO: Error checking
|
|
ip = argv[2];
|
|
port = atoi(argv[3]);
|
|
}
|
|
|
|
session.insecure_stream = libp2p_net_multistream_connect(ip, port);
|
|
session.default_stream = session.insecure_stream;
|
|
if (session.insecure_stream == NULL) {
|
|
fprintf(stderr, "Unable to connect to %s on port %s", ip, argv[3]);
|
|
}
|
|
|
|
// try to switch to secio
|
|
if (!libp2p_secio_handshake(&session, &fs_repo->config->identity->private_key, 0)) {
|
|
fprintf(stderr, "Unable to switch to secure connection. Attempting insecure ping...\n");
|
|
}
|
|
|
|
// prepare the PING message
|
|
struct Libp2pMessage* msg = libp2p_message_new();
|
|
msg->message_type = MESSAGE_TYPE_PING;
|
|
|
|
size_t protobuf_size = libp2p_message_protobuf_encode_size(msg);
|
|
unsigned char protobuf[protobuf_size];
|
|
libp2p_message_protobuf_encode(msg, &protobuf[0], protobuf_size, &protobuf_size);
|
|
session.default_stream->write(&session, protobuf, protobuf_size);
|
|
session.default_stream->read(&session, &results, &results_size);
|
|
|
|
// see if we can unprotobuf
|
|
struct Libp2pMessage* msg_returned = NULL;
|
|
libp2p_message_protobuf_decode(results, results_size, &msg_returned);
|
|
if (msg_returned->message_type != MESSAGE_TYPE_PING) {
|
|
fprintf(stderr, "Ping unsuccessful. Returned message was not a PING");
|
|
return 0;
|
|
}
|
|
|
|
if (results_size != protobuf_size) {
|
|
fprintf(stderr, "PING unsuccessful. Original size: %lu, returned size: %lu\n", protobuf_size, results_size);
|
|
return 0;
|
|
}
|
|
if (memcmp(results, protobuf, protobuf_size) != 0) {
|
|
fprintf(stderr, "PING unsuccessful. Results do not match.\n");
|
|
return 0;
|
|
}
|
|
|
|
if (msg != NULL)
|
|
libp2p_message_free(msg);
|
|
|
|
fprintf(stdout, "Ping of %s:%s successful.\n", ip, argv[3]);
|
|
|
|
return 0;
|
|
|
|
// the old way
|
|
/*
|
|
int socketfd, i, count=10, tcount = 0;
|
|
uint32_t ipv4;
|
|
uint16_t port;
|
|
char b[BUF_SIZE];
|
|
size_t len;
|
|
struct timeval time;
|
|
long cur_time, old_time;
|
|
double ms, total = 0;
|
|
|
|
if (inet_pton (AF_INET, argv[2], &ipv4) == 0) {
|
|
fprintf(stderr, "Unable to use '%s' as an IP address.\n", argv[1]);
|
|
return 1;
|
|
}
|
|
|
|
if ((port = atoi(argv[3])) == 0) {
|
|
fprintf(stderr, "Unable to use '%s' port.\n", argv[2]);
|
|
return 1;
|
|
}
|
|
|
|
if ((socketfd = socket_tcp4()) <= 0) {
|
|
perror("can't create socket");
|
|
return 1;
|
|
}
|
|
|
|
if (socket_connect4(socketfd, ipv4, port) < 0) {
|
|
perror("fail to connect");
|
|
return 1;
|
|
}
|
|
|
|
fprintf(stderr, "PING %s.\n", argv[2]);
|
|
|
|
for (i=0 ; i < count ; i++) {
|
|
if (gettimeofday (&time, 0)) return -1;
|
|
old_time = 1000000 * time.tv_sec + time.tv_usec;
|
|
|
|
socket_write(socketfd, "ping", 4, 0);
|
|
len = socket_read(socketfd, b, sizeof(b), 0);
|
|
|
|
if (len == 4 && memcmp(b, "pong", 4) == 0) {
|
|
if (gettimeofday (&time, 0)) return -1;
|
|
cur_time = 1000000 * time.tv_sec + time.tv_usec;
|
|
ms = (cur_time - old_time) / 1000.0;
|
|
total += ms; tcount++;
|
|
fprintf(stderr, "Pong received: time=%.2f ms\n", ms);
|
|
}
|
|
sleep (1);
|
|
}
|
|
fprintf(stderr, "Average latency: %.2fms\n", total / tcount);
|
|
|
|
return 0;
|
|
*/
|
|
}
|