Add out-of-bound checks

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

diff --git a/formats/ctf-text/types/struct.c b/formats/ctf-text/types/struct.c
index 4270a21afa65bb99e87b6e5d8955c026e6606a56..760f1727d4f37d4968b8f2456d2e9dacec2cc168 100644 (file)
--- a/formats/ctf-text/types/struct.c
+++ b/formats/ctf-text/types/struct.c
@@ -19,9 +19,10 @@
 #include <babeltrace/ctf-text/types.h>
 #include <stdio.h>
 
 #include <babeltrace/ctf-text/types.h>
 #include <stdio.h>
 

-void ctf_text_struct_write(struct stream_pos *ppos, struct definition *definition)
+int ctf_text_struct_write(struct stream_pos *ppos, struct definition *definition)

 {
        struct ctf_text_stream_pos *pos = ctf_text_pos(ppos);
 {
        struct ctf_text_stream_pos *pos = ctf_text_pos(ppos);

+       int ret;

 
        if (!pos->dummy) {
                if (pos->depth >= 0) {
 
        if (!pos->dummy) {
                if (pos->depth >= 0) {


@@ -30,7 +31,7 @@ void ctf_text_struct_write(struct stream_pos *ppos, struct definition *definitio
                }
                pos->depth++;
        }
                }
                pos->depth++;
        }

-       struct_rw(ppos, definition);
+       ret = struct_rw(ppos, definition);

        if (!pos->dummy) {
                pos->depth--;
                if (pos->depth >= 0) {
        if (!pos->dummy) {
                pos->depth--;
                if (pos->depth >= 0) {


@@ -38,4 +39,5 @@ void ctf_text_struct_write(struct stream_pos *ppos, struct definition *definitio
                        fprintf(pos->fp, "}\n");
                }
        }
                        fprintf(pos->fp, "}\n");
                }
        }

+       return ret;

 }
 }