82 lines
2.6 KiB
C
82 lines
2.6 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "protobuf.h"
|
|
#include "varint.h"
|
|
#include "Test3_protobuf.h"
|
|
|
|
enum WireType Test3_message_fields[] = { WIRETYPE_LENGTH_DELIMITED, WIRETYPE_VARINT, WIRETYPE_LENGTH_DELIMITED };
|
|
|
|
int Test3_new(struct Test3** test3) {
|
|
*test3 = (struct Test3*)malloc(sizeof(struct Test3));
|
|
if (*test3 == NULL)
|
|
return 0;
|
|
(*test3)->a_string = NULL;
|
|
(*test3)->an_int = 0;
|
|
(*test3)->test2 = NULL;
|
|
return 1;
|
|
}
|
|
|
|
int Test3_free(struct Test3* test3) {
|
|
if (test3->a_string != NULL)
|
|
free(test3->a_string);
|
|
if (test3->test2 != NULL)
|
|
Test2_free(test3->test2);
|
|
free(test3);
|
|
return 1;
|
|
}
|
|
|
|
int Test3_protobuf_encode(struct Test3* incoming, unsigned char* buffer, size_t max_buffer_length, size_t* bytes_written) {
|
|
size_t bytes_used;
|
|
*bytes_written = 0;
|
|
protobuf_encode_string(1, Test3_message_fields[0], incoming->a_string, buffer, max_buffer_length, &bytes_used);
|
|
*bytes_written += bytes_used;
|
|
protobuf_encode_varint(2, Test3_message_fields[1], incoming->an_int, &buffer[*bytes_written], max_buffer_length - *bytes_written, &bytes_used);
|
|
*bytes_written += bytes_used;
|
|
// write Test2 to safe area
|
|
int pos = *bytes_written + 11;
|
|
size_t test2_size = 0;
|
|
int retVal = Test2_protobuf_encode(incoming->test2, &buffer[pos], max_buffer_length - pos, &test2_size);
|
|
protobuf_encode_varint(3, Test3_message_fields[2], test2_size, &buffer[*bytes_written], max_buffer_length - *bytes_written, &bytes_used);
|
|
*bytes_written += bytes_used;
|
|
// now move the encoded Test2 to the correct area
|
|
memcpy(&buffer[*bytes_written], &buffer[pos], test2_size);
|
|
*bytes_written += test2_size;
|
|
return retVal;
|
|
}
|
|
|
|
int Test3_protobuf_decode(unsigned char* buffer, size_t buffer_length, struct Test3** output) {
|
|
size_t pos = 0;
|
|
if (Test3_new(output) == 0)
|
|
return 0;
|
|
|
|
while (pos < buffer_length) { // loop through buffer
|
|
size_t bytes_read = 0;
|
|
int field_no;
|
|
enum WireType field_type;
|
|
if (protobuf_decode_field_and_type(&buffer[pos], buffer_length, &field_no, &field_type, &bytes_read) == 0) {
|
|
Test3_free(*output);
|
|
return 0;
|
|
}
|
|
pos += bytes_read;
|
|
switch (field_no) {
|
|
case(1):
|
|
protobuf_decode_string(&buffer[pos], buffer_length - pos, &((*output)->a_string), &bytes_read);
|
|
pos += bytes_read;
|
|
break;
|
|
case (2):
|
|
(*output)->an_int = varint_decode(&buffer[pos], buffer_length - pos, &bytes_read);
|
|
pos += bytes_read;
|
|
break;
|
|
case(3): {
|
|
size_t test2_size = varint_decode(&buffer[pos], buffer_length - pos, &bytes_read);
|
|
pos += bytes_read;
|
|
Test2_protobuf_decode(&buffer[pos], buffer_length - pos, &((*output)->test2));
|
|
pos += test2_size;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return 1;
|
|
}
|