在c中的nanopb中为protobuf消息中的重复字段创建回调和结构

时间:2017-08-31 11:06:01

标签: c protocol-buffers nanopb

我的proto消息定义为:

message SimpleMessage {
repeated int32 number = 1;}

现在,在编译之后,该字段为pb_callback_t,我想要编写该函数。 (没有.options文件)

现在,该功能应包含在哪里以及应该包含哪些内容?数据本身存储在何处以及如何访问/为其分配新数据?

*编辑*

根据@Groo的回答,这是我试过的代码:

typedef struct {
    int numbers_decoded;
} DecodingState;

bool read_single_number(pb_istream_t *istream, const pb_field_t *field, void **arg)
{
    // get the pointer to the custom state
    DecodingState *state = (DecodingState*)(*arg);

    int32_t value;
    if (!pb_decode_varint32(istream, &value))
    {
        const char * error = PB_GET_ERROR(istream);
        printf("Protobuf error: %s", error);
        return false;
    }

    printf("Decoded successfully: %d", value);
    state->numbers_decoded++;

    return true;
}

int main(void) {
    int32_t arr[3] = {10, 22, 342};
    uint8_t buffer[128];
    size_t message_length;
    bool status;
    SimpleMessage simple = SimpleMessage_init_zero;

    printf("\nbefore : arr[0] = %d\n",arr[0]);

    // set the argument and the callback fn
    simple.number.arg = &arr;
    simple.number.funcs.decode = read_single_number;

    pb_ostream_t ostream = pb_ostream_from_buffer(buffer, sizeof(buffer));
    status = pb_encode(&ostream, SimpleMessage_fields, &simple);

    message_length = ostream.bytes_written;
    SimpleMessage simple1 = SimpleMessage_init_zero;
    simple = simple1;
    arr[0] = 0;
    pb_istream_t istream = pb_istream_from_buffer(buffer, message_length);
    // this function will call read_single_number several times
    status = pb_decode(&istream, SimpleMessage_fields, &simple);
    printf("\nafter : arr[0] = %d\n",arr[0]);

    return EXIT_SUCCESS;
}

,输出为:

  

之前:arr [0] = 10

     

成功解码:17

     

之后:arr [0] = 0

我做错了什么?

2 个答案:

答案 0 :(得分:3)

您可以使用一些nanopb-specific proto flags强制nanopb生成具有静态分配数组的结构。

然而,nanopb原型的默认行为是生成一个回调函数,在编码期间由nanopb调用(整个列表一次)和解码(一次为列表中的每个项目)。这在低内存嵌入式系统中有时是首选,因为您不需要一次分配多个项目。

因此,对于您的.proto文件:

message SimpleMessage {
    repeated int32 number = 1;
}

你可能会得到类似的东西:

typedef struct _SimpleMessage {
    pb_callback_t number;
} SimpleMessage;

这意味着您必须创建自己的回调函数,该函数将连续为每个项目调用。

为简单起见,我们假设你有一个简单的可变长度"像这样的列表:

#define MAX_NUMBERS 32

typedef struct
{
    int32_t numbers[MAX_NUMBERS];
    int32_t numbers_count;
}
IntList;

// add a number to the int list
void IntList_add_number(IntList * list, int32_t number)
{
    if (list->numbers_count < MAX_NUMBERS)
    {
        list->numbers[list->numbers_count] = number;
        list->numbers_count++;
    }
}

显然,对于这样一个例子,使用回调并没有任何意义,但它使示例变得简单。

编码回调必须遍历列表,并编写protobuf标记和列表中每个项目的值:

bool SimpleMessage_encode_numbers(pb_ostream_t *ostream, const pb_field_t *field, void * const *arg)
{
    IntList * source = (IntList*)(*arg);

    // encode all numbers
    for (int i = 0; i < source->numbers_count; i++)
    {
        if (!pb_encode_tag_for_field(ostream, field))
        {
            const char * error = PB_GET_ERROR(ostream);
            printf("SimpleMessage_encode_numbers error: %s", error);
            return false;
        }

        if (!pb_encode_svarint(ostream, source->numbers[i]))
        {
            const char * error = PB_GET_ERROR(ostream);
            printf("SimpleMessage_encode_numbers error: %s", error);
            return false;
        }
    }

    return true;
}

为每个项目调用一次解码回调,并且&#34;追加&#34;到列表中:

bool SimpleMessage_decode_single_number(pb_istream_t *istream, const pb_field_t *field, void **arg)
{
    IntList * dest = (IntList*)(*arg);

    // decode single number
    int64_t number;
    if (!pb_decode_svarint(istream, &number))
    {
        const char * error = PB_GET_ERROR(istream);
        printf("SimpleMessage_decode_single_number error: %s", error);
        return false;
    }

    // add to destination list
    IntList_add_number(dest, (int32_t)number);
    return true;
}

有了这两个,你必须小心地将正确的回调分配给正确的功能:

uint8_t buffer[128];
size_t total_bytes_encoded = 0;

// encoding
{
    // prepare the actual "variable" array
    IntList actualData = { 0 };
    IntList_add_number(&actualData, 123);
    IntList_add_number(&actualData, 456);
    IntList_add_number(&actualData, 789);

    // prepare the nanopb ENCODING callback
    SimpleMessage msg = SimpleMessage_init_zero;
    msg.number.arg = &actualData;
    msg.number.funcs.encode = SimpleMessage_encode_numbers;

    // call nanopb
    pb_ostream_t ostream = pb_ostream_from_buffer(buffer, sizeof(buffer));
    if (!pb_encode(&ostream, SimpleMessage_fields, &msg))
    {
        const char * error = PB_GET_ERROR(&ostream);
        printf("pb_encode error: %s", error);
        return;
    }

    total_bytes_encoded = ostream.bytes_written;
    printf("Encoded size: %d", total_bytes_encoded);
}

类似于解码:

// decoding
{
    // empty array for decoding
    IntList decodedData = { 0 };

    // prepare the nanopb DECODING callback
    SimpleMessage msg = SimpleMessage_init_zero;
    msg.number.arg = &decodedData;
    msg.number.funcs.decode = SimpleMessage_decode_single_number;

    // call nanopb
    pb_istream_t istream = pb_istream_from_buffer(buffer, total_bytes_encoded);
    if (!pb_decode(&istream, SimpleMessage_fields, &msg))
    {
        const char * error = PB_GET_ERROR(&istream);
        printf("pb_decode error: %s", error);
        return;
    }

    printf("Bytes decoded: %d", total_bytes_encoded - istream.bytes_left);
}

如果您的邮件中有重复的结构,则不会使用您的回调 nanopb原语函数(如上面的pb_decode_varint32),但每个具体的消息类型再次pb_decode。如果需要,您的回调还可以将新回调附加到那些嵌套结构。

答案 1 :(得分:3)

为了补充Groo的答案,以下是您具体问题的答案。

<强> 1。现在,该功能应包含在何处以及应包含哪些内容?

Groo为回调函数提供了很好的解释。 nanopb存储库中的network_server示例也使用回调,可以作为有用的参考:network_server/server.c network_server/client.c

<强> 2。数据本身存储在哪里?

无论你想要什么! nanopb回调的重点在于它为您提供了决定如何存储数据的完全灵活性。在某些情况下,您可能希望即时处理数据,而不是将其存储在任何地方。

例如,上面的network_server示例从文件系统获取文件名并直接将它们发送到网络 - 这样它就可以处理任意数量的文件,而不需要太多内存。

第3。如何访问/为其分配新数据?

现在这是回调的缺点 - 你必须为你使用的任何存储实现自己的访问和分配功能。这就是为什么对于最常见的情况,静态分配(具有固定的最大大小)或动态分配(malloc()所需的内存量)更方便。