memory_grow_impl: use overflow checking macro

yamt · Jul 25, 2024 · 0175981 · 0175981
1 parent d9784ff
commit 0175981
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/exec_insn_subr.c b/lib/exec_insn_subr.c
@@ -433,11 +433,11 @@ memory_grow_impl(struct exec_context *ctx, struct meminst *mi, uint32_t sz)
 retry:
 #endif
         orig_size = mi->size_in_pages;
-        if (sz > UINT32_MAX - orig_size) {
+        uint32_t new_size;
+        if (ADD_U32_OVERFLOW(orig_size, sz, &new_size)) {
                 memory_unlock(mi);
                 return (uint32_t)-1; /* fail */
         }
-        uint32_t new_size = orig_size + sz;
         assert(lim->max <= WASM_MAX_MEMORY_SIZE >> page_shift);
         if (new_size > lim->max) {
                 memory_unlock(mi);