criu/include/protobuf.h

#ifndef PROTOBUF_H__
#define PROTOBUF_H__

#include "types.h"
#include "compiler.h"
#include "util.h"

/*
 * ATTENTION
 *
 * This typdefs represent "generic" prototypes for
 * pack/unpack/getsize functions generated by PB
 * engine, thus (!!!) if PB engine change arguments
 * order or their number we may meet serious problems.
 *
 * FIXME
 *
 * Find a way to verify PB generated functions statemens
 * to match this typedefs.
 */

typedef size_t (*pb_getpksize_t)(void *obj);
typedef size_t (*pb_pack_t)(void *obj, void *where);
typedef void  *(*pb_unpack_t)(void *allocator, size_t size, void *from);
typedef void   (*pb_free_t)(void *obj, void *allocator);

extern int pb_read_object_with_header(int fd, void **pobj,
				      pb_unpack_t unpack,
				      bool eof);

#define PB_UNPACK_TYPECHECK(__op, __fn)	({ if (0) *__op = __fn##__unpack(NULL, 0, NULL); (pb_unpack_t)&__fn##__unpack; })
#define PB_PACK_TYPECHECK(__o, __fn)	({ if (0) __fn##__pack(__o, NULL); (pb_pack_t)&__fn##__pack; })
#define PB_GPS_TYPECHECK(__o, __fn)	({ if (0) __fn##__get_packed_size(__o); (pb_getpksize_t)&__fn##__get_packed_size; })
#define PB_FREE_TYPECHECK(__o, __fn)	({ if (0) __fn##__free_unpacked(__o, NULL); (pb_free_t)&__fn##__free_unpacked; })

#define pb_read(__fd, __obj_pptr, __proto_message_name)					\
	pb_read_object_with_header(__fd, (void **)__obj_pptr,				\
		PB_UNPACK_TYPECHECK(__obj_pptr, __proto_message_name), false)

#define pb_read_eof(__fd, __obj_pptr, __proto_message_name)				\
	pb_read_object_with_header(__fd, (void **)__obj_pptr,				\
		PB_UNPACK_TYPECHECK(__obj_pptr, __proto_message_name), true)

extern int pb_write_object_with_header(int fd, void *obj,
				       pb_getpksize_t getpksize,
				       pb_pack_t pack);

#define pb_write(__fd, __obj, __proto_message_name)					\
	pb_write_object_with_header(__fd, __obj,					\
		PB_GPS_TYPECHECK(__obj, __proto_message_name),				\
		PB_PACK_TYPECHECK(__obj, __proto_message_name))

#define pb_pksize(__obj, __proto_message_name)						\
	(__proto_message_name ##__get_packed_size(__obj) + sizeof(u32))

#define pb_repeated_size(__obj, __member)						\
	(sizeof(*(__obj)->__member) * (__obj)->n_ ##__member)

#include <google/protobuf-c/protobuf-c.h>

extern void do_pb_show_plain(int fd, const ProtobufCMessageDescriptor *d,
		pb_unpack_t unpack, pb_free_t free, int single_entry,
		void (*payload_hadler)(int fd, void *obj, int flags),
		int flags);

/* Don't have objects at hands to also do typechecking here */
#define pb_show_plain_payload(__fd, __proto_message_name, payload_hadler, flags)	\
	do_pb_show_plain(__fd, &__proto_message_name##__descriptor,	\
			(pb_unpack_t)__proto_message_name##__unpack,			\
			(pb_free_t)__proto_message_name##__free_unpacked,		\
			0, payload_hadler, flags)

#define pb_show_plain(__fd, __proto_message_name)			\
	pb_show_plain_payload(__fd, __proto_message_name, NULL, 0)

#define pb_show_vertical(__fd, __proto_message_name)			\
	do_pb_show_plain(__fd, &__proto_message_name##__descriptor,	\
			(pb_unpack_t)__proto_message_name##__unpack,			\
			(pb_free_t)__proto_message_name##__free_unpacked,		\
			1, NULL, 0)

#endif /* PROTOBUF_H__ */