Fix: add stricter checks on packet boundaries

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

diff --git a/formats/ctf/types/variant.c b/formats/ctf/types/variant.c
index 3c0c371f4d5922a708e300009fa910be3ab0d96d..fc7b7ad449d935ea4dcf9e32afda839463250141 100644 (file)
--- a/formats/ctf/types/variant.c
+++ b/formats/ctf/types/variant.c
@@ -28,11 +28,12 @@
 
 #include <babeltrace/ctf/types.h>
 
-int ctf_variant_rw(struct bt_stream_pos *ppos, struct definition *definition)
+int ctf_variant_rw(struct bt_stream_pos *ppos, struct bt_definition *definition)
 {
-       struct declaration *declaration = definition->declaration;
+       struct bt_declaration *declaration = definition->declaration;
        struct ctf_stream_pos *pos = ctf_pos(ppos);
 
-       ctf_align_pos(pos, declaration->alignment);
+       if (!ctf_align_pos(pos, declaration->alignment))
+               return -EFAULT;
        return bt_variant_rw(ppos, definition);
 }