50 lines
1.5 KiB
C
50 lines
1.5 KiB
C
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "libp2p/net/stream.h"
|
|
#include "libp2p/conn/session.h"
|
|
|
|
int libp2p_nodeio_upgrade_stream(struct SessionContext* context) {
|
|
int retVal = 0;
|
|
char* protocol = "/nodeio/1.0.0\n";
|
|
unsigned char* results = NULL;
|
|
size_t results_size = 0;
|
|
if (!context->default_stream->write(context, (unsigned char*)protocol, strlen(protocol)))
|
|
goto exit;
|
|
if (!context->default_stream->read(context, &results, &results_size, 5))
|
|
goto exit;
|
|
if (results_size != strlen(protocol))
|
|
goto exit;
|
|
if (strncmp((char*)results, protocol, results_size) != 0)
|
|
goto exit;
|
|
retVal = 1;
|
|
exit:
|
|
if (results != NULL) {
|
|
free(results);
|
|
results = NULL;
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
/**
|
|
* Called by requestor to get a protobuf'd node from a hash
|
|
* @param context the session context
|
|
* @param hash the hash
|
|
* @param hash_size the length of the hash
|
|
* @param results where to put the buffer
|
|
* @param results_size the size of the results
|
|
* @returns true(1) on success, otherwise false(0)
|
|
*/
|
|
int libp2p_nodeio_get(struct SessionContext* context, unsigned char* hash, int hash_length, unsigned char** results, size_t* results_size) {
|
|
if (!context->default_stream->write(context, hash, hash_length))
|
|
return 0;
|
|
if (!context->default_stream->read(context, results, results_size, 5))
|
|
return 0;
|
|
return 1;
|
|
}
|
|
|
|
int libp2p_nodeio_handshake(struct SessionContext* context) {
|
|
char* protocol = "/nodeio/1.0.0\n";
|
|
return context->default_stream->write(context, (unsigned char*)protocol, strlen(protocol));
|
|
}
|