[lttng-dev] [PATCH babeltrace] Fix ctf-writer: Reject enumerations containing no mappings

[Jérémie Galarneau]

Signed-off-by: Jérémie Galarneau <jeremie.galarn...@efficios.com>
---
 formats/ctf/writer/event-fields.c                  |  4 +--
 formats/ctf/writer/event-types.c                   | 40 +++++++++++++++++++---
 .../babeltrace/ctf-writer/event-types-internal.h   |  3 ++
 3 files changed, 40 insertions(+), 7 deletions(-)

diff --git a/formats/ctf/writer/event-fields.c 
b/formats/ctf/writer/event-fields.c
index ff970c0..ad4fcb5 100644
--- a/formats/ctf/writer/event-fields.c
+++ b/formats/ctf/writer/event-fields.c
@@ -179,8 +179,8 @@ struct bt_ctf_field *bt_ctf_field_create(struct 
bt_ctf_field_type *type)
        }
 
        type_id = bt_ctf_field_type_get_type_id(type);
-       if (type_id <= CTF_TYPE_UNKNOWN ||
-               type_id >= NR_CTF_TYPES) {
+       if (type_id <= CTF_TYPE_UNKNOWN || type_id >= NR_CTF_TYPES ||
+               bt_ctf_field_type_validate(type)) {
                goto error;
        }
 
diff --git a/formats/ctf/writer/event-types.c b/formats/ctf/writer/event-types.c
index fa4e713..4ed9fed 100644
--- a/formats/ctf/writer/event-types.c
+++ b/formats/ctf/writer/event-types.c
@@ -238,6 +238,27 @@ end:
        return ret;
 }
 
+BT_HIDDEN
+int bt_ctf_field_type_validate(struct bt_ctf_field_type *type)
+{
+       int ret = 0;
+
+       if (!type) {
+               ret = -1;
+               goto end;
+       }
+
+       if (type->declaration->id == CTF_TYPE_ENUM) {
+               struct bt_ctf_field_type_enumeration *enumeration =
+                       container_of(type, struct bt_ctf_field_type_enumeration,
+                       parent);
+
+               ret = enumeration->entries->len ? 0 : -1;
+       }
+end:
+       return ret;
+}
+
 struct bt_ctf_field_type *bt_ctf_field_type_integer_create(unsigned int size)
 {
        struct bt_ctf_field_type_integer *integer =
@@ -517,7 +538,8 @@ int bt_ctf_field_type_structure_add_field(struct 
bt_ctf_field_type *type,
 
        if (!type || !field_type || type->frozen ||
                validate_identifier(field_name) ||
-               (type->declaration->id != CTF_TYPE_STRUCT)) {
+               (type->declaration->id != CTF_TYPE_STRUCT) ||
+               bt_ctf_field_type_validate(field_type)) {
                goto end;
        }
 
@@ -579,7 +601,8 @@ int bt_ctf_field_type_variant_add_field(struct 
bt_ctf_field_type *type,
 
        if (!type || !field_type || type->frozen ||
                validate_identifier(field_name) ||
-               (type->declaration->id != CTF_TYPE_VARIANT)) {
+               (type->declaration->id != CTF_TYPE_VARIANT) ||
+               bt_ctf_field_type_validate(field_type)) {
                ret = -1;
                goto end;
        }
@@ -611,7 +634,8 @@ struct bt_ctf_field_type *bt_ctf_field_type_array_create(
 {
        struct bt_ctf_field_type_array *array = NULL;
 
-       if (!element_type || length == 0) {
+       if (!element_type || length == 0 ||
+               bt_ctf_field_type_validate(element_type)) {
                goto error;
        }
 
@@ -639,7 +663,8 @@ struct bt_ctf_field_type *bt_ctf_field_type_sequence_create(
 {
        struct bt_ctf_field_type_sequence *sequence = NULL;
 
-       if (!element_type || validate_identifier(length_field_name)) {
+       if (!element_type || validate_identifier(length_field_name) ||
+               bt_ctf_field_type_validate(element_type)) {
                goto error;
        }
 
@@ -1154,10 +1179,15 @@ int bt_ctf_field_type_enumeration_serialize(struct 
bt_ctf_field_type *type,
                struct metadata_context *context)
 {
        size_t entry;
-       int ret = 0;
+       int ret;
        struct bt_ctf_field_type_enumeration *enumeration = container_of(type,
                struct bt_ctf_field_type_enumeration, parent);
 
+       ret = bt_ctf_field_type_validate(type);
+       if (ret) {
+               goto end;
+       }
+
        g_string_append(context->string, "enum : ");
        ret = bt_ctf_field_type_serialize(enumeration->container, context);
        if (ret) {
diff --git a/include/babeltrace/ctf-writer/event-types-internal.h 
b/include/babeltrace/ctf-writer/event-types-internal.h
index d9acdd3..a937c78 100644
--- a/include/babeltrace/ctf-writer/event-types-internal.h
+++ b/include/babeltrace/ctf-writer/event-types-internal.h
@@ -147,4 +147,7 @@ BT_HIDDEN
 int bt_ctf_field_type_serialize(struct bt_ctf_field_type *type,
                struct metadata_context *context);
 
+BT_HIDDEN
+int bt_ctf_field_type_validate(struct bt_ctf_field_type *type);
+
 #endif /* BABELTRACE_CTF_WRITER_EVENT_TYPES_INTERNAL_H */
-- 
1.8.4.2


_______________________________________________
lttng-dev mailing list
lttng-dev@lists.lttng.org
http://lists.lttng.org/cgi-bin/mailman/listinfo/lttng-dev