#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;
}