c-libp2p/test/test_multistream.h

72 lines
1.6 KiB
C

#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netdb.h>
#include "libp2p/net/multistream.h"
int test_multistream_connect() {
int retVal = 0;
char* response;
size_t response_size;
struct Stream* stream = libp2p_net_multistream_connect("www.jmjatlanta.com", 4001);
if (stream == NULL)
goto exit;
retVal = 1;
exit:
if (stream != NULL) {
struct SessionContext ctx;
ctx.insecure_stream = stream;
stream->close(&ctx);
libp2p_net_multistream_stream_free(stream);
}
return retVal;
}
int test_multistream_get_list() {
int retVal = 0;
struct StreamMessage* response;
char* filtered = NULL;
struct SessionContext session;
session.insecure_stream = libp2p_net_multistream_connect("10.211.55.2", 4001);
if (*((int*)session.insecure_stream->socket_descriptor) < 0)
goto exit;
// try to respond something, ls command
const unsigned char* out = (unsigned char*)"ls\n";
if (libp2p_net_multistream_write(&session, out, strlen((char*)out)) <= 0)
goto exit;
// retrieve response
retVal = libp2p_net_multistream_read(&session, &response, 5);
if (retVal <= 0)
goto exit;
filtered = malloc(response->data_size + 1);
strncpy(filtered, (char*)response->data, response->data_size);
filtered[response->data_size] = 0;
fprintf(stdout, "Response from multistream ls: %s", (char*)filtered);
retVal = 1;
exit:
if (session.insecure_stream != NULL) {
session.insecure_stream->close(&session);
libp2p_net_multistream_stream_free(session.insecure_stream);
}
libp2p_stream_message_free(response);
if (filtered != NULL)
free(filtered);
return retVal > 0;
}