From fbce16e02dbdecf765b971a0398f5feac3e35b85 Mon Sep 17 00:00:00 2001 From: Wagner Bruna Date: Sat, 17 Jan 2026 14:14:56 -0300 Subject: [PATCH] fix: avoid undefined behavior on image mask allocation failure (#1198) --- examples/cli/main.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp index 0a7da7a..ddc2828 100644 --- a/examples/cli/main.cpp +++ b/examples/cli/main.cpp @@ -594,12 +594,12 @@ int main(int argc, const char* argv[]) { } } else { mask_image.data = (uint8_t*)malloc(gen_params.width * gen_params.height); - memset(mask_image.data, 255, gen_params.width * gen_params.height); if (mask_image.data == nullptr) { LOG_ERROR("malloc mask image failed"); release_all_resources(); return 1; } + memset(mask_image.data, 255, gen_params.width * gen_params.height); } if (gen_params.control_image_path.size() > 0) {