[CRIU] [PATCH 4/5] protobuf: added support for optional fields

[Kinsbursky Stanislav]

Signed-off-by: Stanislav Kinsbursky <skinsbursky at openvz.org>

---
 protobuf.c |   26 +++++++++++++++++++++++---
 1 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/protobuf.c b/protobuf.c
index a056440..fc1d6b4 100644
--- a/protobuf.c
+++ b/protobuf.c
@@ -163,6 +163,24 @@ static void pb_show_field(const ProtobufCFieldDescriptor *fd, void *where,
 		pr_msg(" ");
 }
 
+static int pb_optional_field_present(const ProtobufCFieldDescriptor *field,
+				    void *msg)
+{
+	if ((field->type == PROTOBUF_C_TYPE_MESSAGE) ||
+		(field->type == PROTOBUF_C_TYPE_STRING)) {
+		const void *opt_flag = * (const void * const *)(msg + field->offset);
+
+		if ((opt_flag == NULL) || (opt_flag == field->default_value))
+			return 0;
+	} else {
+		const protobuf_c_boolean *has = msg + field->quantifier_offset;
+
+		if (!*has)
+			return 0;
+	}
+	return 1;
+}
+
 static void pb_show_msg(const void *msg, pb_pr_ctl_t *ctl)
 {
 	int i;
@@ -175,12 +193,14 @@ static void pb_show_msg(const void *msg, pb_pr_ctl_t *ctl)
 		unsigned long *data;
 		size_t nr_fields;
 
-		if (fd.label == PROTOBUF_C_LABEL_OPTIONAL)
-			continue;
-
 		nr_fields = 1;
 		data = (unsigned long *)(msg + fd.offset);
 
+		if (fd.label == PROTOBUF_C_LABEL_OPTIONAL) {
+			if (!pb_optional_field_present(&fd, data))
+				continue;
+		}
+
 		if (fd.label == PROTOBUF_C_LABEL_REPEATED) {
 			nr_fields = *(size_t *)(msg + fd.quantifier_offset);
 			data = (unsigned long *)*data;