Adds pre- and post-conditions to s2n_stuffer_resize (aws#2191)

Signed-off-by: Felipe R. Monteiro <[email protected]>
colmmacc · Jul 30, 2020 · 3083cf5 · 3083cf5
1 parent 34ee087
commit 3083cf5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/stuffer/s2n_stuffer.c b/stuffer/s2n_stuffer.c
@@ -90,6 +90,7 @@ int s2n_stuffer_free(struct s2n_stuffer *stuffer)
 
 int s2n_stuffer_resize(struct s2n_stuffer *stuffer, const uint32_t size)
 {
+    PRECONDITION_POSIX(s2n_stuffer_is_valid(stuffer));
     ENSURE_POSIX(!stuffer->tainted, S2N_ERR_RESIZE_TAINTED_STUFFER);
     ENSURE_POSIX(stuffer->growable, S2N_ERR_RESIZE_STATIC_STUFFER);
 
@@ -108,11 +109,12 @@ int s2n_stuffer_resize(struct s2n_stuffer *stuffer, const uint32_t size)
         if (stuffer->write_cursor > size) stuffer->write_cursor = size;
         if (stuffer->high_water_mark > size) stuffer->high_water_mark = size;
         stuffer->blob.size = size;
+        POSTCONDITION_POSIX(s2n_stuffer_is_valid(stuffer));
         return S2N_SUCCESS;
     }
 
     GUARD(s2n_realloc(&stuffer->blob, size));
-
+    POSTCONDITION_POSIX(s2n_stuffer_is_valid(stuffer));
     return S2N_SUCCESS;
 }