Add out-of-bound checks

[babeltrace.git] / types / struct.c

diff --git a/types/struct.c b/types/struct.c
index 211d1b02dd16bbc199695f29a976c6eb3c4c8759..6e82b57fb845e13d1c69d837efca679928dedd12 100644 (file)
--- a/types/struct.c
+++ b/types/struct.c
@@ -30,17 +30,21 @@ struct definition *_struct_definition_new(struct declaration *declaration,
 static
 void _struct_definition_free(struct definition *definition);
 
-void struct_rw(struct stream_pos *ppos, struct definition *definition)
+int struct_rw(struct stream_pos *ppos, struct definition *definition)
 {
        struct definition_struct *struct_definition =
                container_of(definition, struct definition_struct, p);
        unsigned long i;
+       int ret;
 
        for (i = 0; i < struct_definition->fields->len; i++) {
                struct field *field = &g_array_index(struct_definition->fields,
                                                     struct field, i);
-               generic_rw(ppos, field->definition);
+               ret = generic_rw(ppos, field->definition);
+               if (ret)
+                       return ret;
        }
+       return 0;
 }
 
 static