test.c

Go to the documentation of this file.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pb_encode.h>
#include "test.pb.h"
#include "unittests.h"
 
const char STR[] = "test str";
#define ALIGN 0x100
 
int main(int argc, char **argv)
{
    int status = 0;
    uint8_t buffer[512] = {0};
    int i;
 
    {
        pb_ostream_t ostream;
        MyMessage msg = MyMessage_init_zero;
        char *pStr, *pStrAligned;
 
        COMMENT("Test for false negatives with pointer value low byte 0x00")
        ostream = pb_ostream_from_buffer(buffer, sizeof(buffer));
 
        /* copy STR to a malloced 0x100 aligned address */
        pStr = malloc(sizeof(STR) + ALIGN);
        pStrAligned = (char*)((uintptr_t)(pStr + ALIGN) & ~(ALIGN - 1));
        memcpy(pStrAligned, STR, sizeof(STR));
 
        msg.submessage.somestring = pStrAligned;
        printf("%p: '%s'\n", msg.submessage.somestring, msg.submessage.somestring);
 
        if (!pb_encode(&ostream, MyMessage_fields, &msg)) {
            fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&ostream));
            return 1;
        }
 
        free(pStr);
        msg.submessage.somestring = NULL;
 
        printf("response payload (%d):", (int)ostream.bytes_written);
        for (i = 0; i < ostream.bytes_written; i++) {
            printf("%02X", buffer[i]);
        }
        printf("\n");
 
        TEST(ostream.bytes_written != 0);
    }
 
    {
        pb_ostream_t ostream;
        struct {
            MyMessage msg;
            uint32_t bar;
         } msg = {MyMessage_init_zero, 0};
 
        COMMENT("Test for false positives with data after end of struct")
        ostream = pb_ostream_from_buffer(buffer, sizeof(buffer));
 
        msg.bar = 0xFFFFFFFF;
 
        if (!pb_encode(&ostream, MyMessage_fields, &msg)) {
            fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&ostream));
            return 1;
        }
 
        printf("response payload (%d):", (int)ostream.bytes_written);
        for (i = 0; i < ostream.bytes_written; i++) {
            printf("%02X", buffer[i]);
        }
        printf("\n");
 
        TEST(ostream.bytes_written == 0);
    }
 
    return status;
}