Add out-of-bound checks

[babeltrace.git] / formats / ctf / types / struct.c

diff --git a/formats/ctf/types/struct.c b/formats/ctf/types/struct.c
index 70c08219953200be6baa3931659ca3f505fbb51b..2913f4d0e2d41d6099e6cfda5ce18339826b224c 100644 (file)
--- a/formats/ctf/types/struct.c
+++ b/formats/ctf/types/struct.c
@@ -18,13 +18,11 @@
 
 #include <babeltrace/ctf/types.h>
 
-void ctf_struct_begin(struct stream_pos *pos,
-                     const struct type_class_struct *struct_class)
+int ctf_struct_rw(struct stream_pos *ppos, struct definition *definition)
 {
-       align_pos(pos, struct_class->p.alignment);
-}
+       struct declaration *declaration = definition->declaration;
+       struct ctf_stream_pos *pos = ctf_pos(ppos);
 
-void ctf_struct_end(struct stream_pos *pos,
-                   const struct type_class_struct *struct_class)
-{
+       ctf_align_pos(pos, declaration->alignment);
+       return struct_rw(ppos, definition);
 }