Merge branch 'yamux' of https://github.com/Agorise/c-libp2p into yamux

This commit is contained in:
John Jones 2017-11-19 13:37:21 -05:00
commit db67210257
2 changed files with 234 additions and 0 deletions

View file

@ -1,6 +1,7 @@
#include <string.h>
#include "varint.h"
#include "protobuf.h"
#include "libp2p/net/protocol.h"
#include "libp2p/net/protocol.h"
#include "libp2p/net/multistream.h"
@ -74,6 +75,229 @@ int libp2p_identify_receive_protocol(struct Stream* stream) {
return 1;
}
/**
* Create a Identify struct
* @returns the newly allocated record struct
*/
Identify* libp2p_identify_new() {
Identify* out = (Identify*)malloc(sizeof(Identify));
if (out != NULL) {
out->PublicKey = NULL;
out->ListenAddrs = NULL;
out->Protocols = NULL;
out->ObservedAddr = NULL;
out->ProtocolVersion = IDENTIFY_PROTOCOL_VERSION;
out->AgentVersion = IDENTIFY_AGENT_VERSION;
out->XXX_unrecognized = NULL;
}
return out;
}
/**
* Free the resources from a identify struct
* @param in the struct to free
*/
void libp2p_identify_free(Identify* in) {
int i;
if (in != NULL) {
if (in->PublicKey != NULL)
free(in->PublicKey);
if (in->ListenAddrs != NULL) {
// free every item
for (i = 0 ; in->ListenAddrs[i] ; i++)
free(in->ListenAddrs[i]);
// free array
free(in->ListenAddrs);
}
if (in->Protocols != NULL) {
for (i = 0 ; in->Protocols[i] ; i++)
free(in->Protocols[i]);
free(in->Protocols);
}
if (in->ObservedAddr != NULL)
free(in->ObservedAddr);
if (in->XXX_unrecognized != NULL)
free(in->XXX_unrecognized);
free(in);
}
}
/* helper to alloc and copy an item.
* @param item the item.
* @returns a pointer on success, otherwise NULL.
*/
char *libp2p_identify_new_item(char *item, size_t size) {
char *dst = NULL;
if (item) {
dst = malloc(size);
if (dst) {
memcpy(dst, item, size);
}
}
return dst;
}
/* helper to add an item to an array.
* @param array the array.
* @param item the item.
* @returns count itens on success, otherwise zero.
*/
int libp2p_identify_array_add_item(char ***array, char *item) {
int count = 0;
if (array && item) {
if (*array) {
// count itens already in.
while (*array[count++]);
}
// alloc the necessary array count or realloc more if alread allocated.
*array = realloc(*array, sizeof(char*) * (count + 2)); // 2, the new and the NULL
*array[count++] = item;
*array[count] = NULL;
}
return count;
}
/**
* Convert a identify struct into protobuf format
* @param in the Identify struct to convert
* @param buffer where to store the protobuf
* @param max_buffer_size the size of the allocated buffer
* @param bytes_written the size written into buffer
* @returns true(1) on success, otherwise false(0)
*/
int libp2p_identify_protobuf_encode(const Identify* in, unsigned char* buffer, size_t max_buffer_size, size_t* bytes_written) {
// data & data_size
size_t bytes_used = 0;
*bytes_written = 0;
int i, retVal = 0;
// field 1
retVal = protobuf_encode_string(1, WIRETYPE_LENGTH_DELIMITED, in->PublicKey, &buffer[*bytes_written], max_buffer_size - *bytes_written, &bytes_used);
if (retVal == 0)
return 0;
*bytes_written += bytes_used;
// field 2
for (i = 0 ; in->ListenAddrs[i] ; i++) {
retVal = protobuf_encode_string(2, WIRETYPE_LENGTH_DELIMITED, in->ListenAddrs[i], &buffer[*bytes_written], max_buffer_size - *bytes_written, &bytes_used);
if (retVal == 0)
return 0;
*bytes_written += bytes_used;
}
// field 3
for (i = 0 ; in->Protocols[i] ; i++) {
retVal = protobuf_encode_string(3, WIRETYPE_LENGTH_DELIMITED, in->Protocols[i], &buffer[*bytes_written], max_buffer_size - *bytes_written, &bytes_used);
if (retVal == 0)
return 0;
*bytes_written += bytes_used;
}
// field 4
retVal = protobuf_encode_string(4, WIRETYPE_LENGTH_DELIMITED, in->ObservedAddr, &buffer[*bytes_written], max_buffer_size - *bytes_written, &bytes_used);
if (retVal == 0)
return 0;
*bytes_written += bytes_used;
// field 5
retVal = protobuf_encode_string(5, WIRETYPE_LENGTH_DELIMITED, in->ProtocolVersion, &buffer[*bytes_written], max_buffer_size - *bytes_written, &bytes_used);
if (retVal == 0)
return 0;
*bytes_written += bytes_used;
// field 6
retVal = protobuf_encode_string(6, WIRETYPE_LENGTH_DELIMITED, in->AgentVersion, &buffer[*bytes_written], max_buffer_size - *bytes_written, &bytes_used);
if (retVal == 0)
return 0;
*bytes_written += bytes_used;
return 1;
}
/**
* Convert a protobuf byte array into a Identify struct
* @param in the byte array
* @param in_size the size of the byte array
* @param out a pointer to the new Identify struct
* @returns true(1) on success, otherwise false(0)
*/
int libp2p_identify_protobuf_decode(const unsigned char* in, size_t in_size, Identify** out) {
size_t pos = 0;
int count, retVal = 0;
char *item;
if ( (*out = libp2p_identify_new()) == NULL)
goto exit;
while(pos < in_size) {
size_t bytes_read = 0;
int field_no;
enum WireType field_type;
if (!protobuf_decode_field_and_type(&in[pos], in_size, &field_no, &field_type, &bytes_read)) {
goto exit;
}
pos += bytes_read;
switch(field_no) {
case (1): // PublicKey
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&((*out)->PublicKey), &bytes_read))
goto exit;
pos += bytes_read;
break;
case (2): // ListenAddrs
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&(item), &bytes_read))
goto exit;
if (!libp2p_identify_array_add_item(&((*out)->ListenAddrs), item))
goto exit;
pos += bytes_read;
break;
case (3): // Protocols
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&(item), &bytes_read))
goto exit;
if (!libp2p_identify_array_add_item(&((*out)->Protocols), item))
goto exit;
pos += bytes_read;
break;
case (4): // ObservedAddr
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&((*out)->ObservedAddr), &bytes_read))
goto exit;
pos += bytes_read;
break;
case (5): // ProtocolVersion
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&((*out)->ProtocolVersion), &bytes_read))
goto exit;
pos += bytes_read;
break;
case (6): // AgentVersion
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&((*out)->AgentVersion), &bytes_read))
goto exit;
pos += bytes_read;
break;
default: // XXX_unrecognized
// create a comma-separated string of each unrecognized item.
if (!protobuf_decode_string(&in[pos], in_size - pos, (char**)&(item), &bytes_read))
goto exit;
count = 0;
if ((*out)->XXX_unrecognized) {
count = strlen((*out)->XXX_unrecognized);
(*out)->XXX_unrecognized[count++] = ','; // null terminator is now a comma.
}
(*out)->XXX_unrecognized = realloc((*out)->XXX_unrecognized, count + bytes_read);
if (!(*out)->XXX_unrecognized)
goto exit;
// append after the comma.
memcpy((*out)->XXX_unrecognized+count, item, bytes_read);
free(item);
pos += bytes_read;
break;
}
}
retVal = 1;
exit:
if (retVal == 0) {
libp2p_identify_free(*out);
*out = NULL;
}
return retVal;
}
/**
* A remote node is attempting to send us an Identify message
* @param msg the message sent

View file

@ -2,6 +2,9 @@
#include "libp2p/utils/vector.h"
#define IDENTIFY_PROTOCOL_VERSION "ipfs/0.1.0"
#define IDENTIFY_AGENT_VERSION "c-ipfs/1.0"
typedef struct {
// publicKey is this node's public key (which also gives its node.ID)
// - may not need to be sent, as secure channel implies it has been sent.
@ -31,6 +34,13 @@ struct IdentifyContext {
int libp2p_identify_can_handle(const struct StreamMessage* msg);
int libp2p_identify_send_protocol(struct Stream* stream);
int libp2p_identify_receive_protocol(struct Stream* stream);
Identify* libp2p_identify_new();
void libp2p_identify_free(Identify* in);
char *libp2p_identify_new_item(char *item, size_t size);
int libp2p_identify_array_add_item(char ***array, char *item);
int libp2p_identify_protobuf_encode(const Identify* in, unsigned char* buffer, size_t max_buffer_size, size_t* bytes_written);
int libp2p_identify_protobuf_decode(const unsigned char* in, size_t in_size, Identify** out);
int libp2p_identify_handle_message(const struct StreamMessage* msg, struct Stream* stream, void* protocol_context);
int libp2p_identify_shutdown(void* protocol_context);
struct Libp2pProtocolHandler* libp2p_identify_build_protocol_handler(struct Libp2pVector* handlers);