Line data    Source code

       1             : /* Tests extension fields.
       2             :  */
       3             : 
       4             : #include <stdio.h>
       5             : #include <stdlib.h>
       6             : #include <string.h>
       7             : #include <pb_encode.h>
       8             : #include "alltypes.pb.h"
       9             : #include "extensions.pb.h"
      10             : #include "test_helpers.h"
      11             : 
      12           1 : int main(int argc, char **argv)
      13             : {
      14             :     uint8_t buffer[1024];
      15             :     pb_ostream_t stream;
      16             : 
      17           1 :     AllTypes alltypes = {0};
      18           1 :     int32_t extensionfield1 = 12345;
      19           1 :     pb_extension_t ext1 = pb_extension_init_zero;
      20           1 :     ExtensionMessage extensionfield2 = {"test", 54321};
      21           1 :     pb_extension_t ext2 = pb_extension_init_zero;
      22             : 
      23             :     /* Set up the extensions */
      24           1 :     alltypes.extensions = &ext1;
      25             : 
      26           1 :     ext1.type = &AllTypes_extensionfield1;
      27           1 :     ext1.dest = &extensionfield1;
      28           1 :     ext1.next = &ext2;
      29             :     
      30           1 :     ext2.type = &ExtensionMessage_AllTypes_extensionfield2;
      31           1 :     ext2.dest = &extensionfield2;
      32           1 :     ext2.next = NULL;
      33             : 
      34             :     /* Set up the output stream */
      35           1 :     stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
      36             :     
      37             :     /* Now encode the message and check if we succeeded. */
      38           1 :     if (pb_encode(&stream, AllTypes_fields, &alltypes))
      39             :     {
      40             :         SET_BINARY_MODE(stdout);
      41           1 :         fwrite(buffer, 1, stream.bytes_written, stdout);
      42           1 :         return 0; /* Success */
      43             :     }
      44             :     else
      45             :     {
      46           0 :         fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
      47           0 :         return 1; /* Failure */
      48             :     }
      49             :     
      50             :     /* Check that the field tags are properly generated */
      51             :     (void)AllTypes_extensionfield1_tag;
      52             :     (void)ExtensionMessage_AllTypes_extensionfield2_tag;
      53             : }
      54             :