fix: avoid potential dangling pointer problem

leejet · leejet · commit 30b3ac8e6279 · 2025-03-01T16:58:26.000+08:00
diff --git a/examples/cli/main.cpp b/examples/cli/main.cpp
@@ -931,12 +931,12 @@ int main(int argc, const char* argv[]) {
         }
     }
 
+    std::vector<uint8_t> default_mask_image_vec(params.width * params.height, 255);
     if (params.mask_path != "") {
         int c             = 0;
         mask_image_buffer = stbi_load(params.mask_path.c_str(), &params.width, &params.height, &c, 1);
     } else {
-        std::vector<uint8_t> arr(params.width * params.height, 255);
-        mask_image_buffer = arr.data();
+        mask_image_buffer = default_mask_image_vec.data();
     }
     sd_image_t mask_image = {(uint32_t)params.width,
                              (uint32_t)params.height,

Original file line number	Diff line number	Diff line change
`@@ -931,12 +931,12 @@ int main(int argc, const char* argv[]) {`
`931`	`931`	`}`
`932`	`932`	`}`
`933`	`933`
	`934`	`+ std::vector<uint8_t> default_mask_image_vec(params.width * params.height, 255);`
`934`	`935`	`if (params.mask_path != "") {`
`935`	`936`	`int c = 0;`
`936`	`937`	`mask_image_buffer = stbi_load(params.mask_path.c_str(), &params.width, &params.height, &c, 1);`
`937`	`938`	`} else {`
`938`		`- std::vector<uint8_t> arr(params.width * params.height, 255);`
`939`		`- mask_image_buffer = arr.data();`
	`939`	`+ mask_image_buffer = default_mask_image_vec.data();`
`940`	`940`	`}`
`941`	`941`	`sd_image_t mask_image = {(uint32_t)params.width,`
`942`	`942`	`(uint32_t)params.height,`