mods: ucmd: dm: also handle cookie for remove event

sid-project · Aug 7, 2024 · 2ab077b · 2ab077b
1 parent 26b0c28
commit 2ab077b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/modules/ucmd/type/dm/dm.c b/src/modules/ucmd/type/dm/dm.c
@@ -387,7 +387,8 @@ static int _get_cookie_props(sid_res_t *mod_res, struct sid_ucmd_ctx *ucmd_ctx,
 		return -1;
 	}
 
-	*cookie_flags = flags;
+	if (cookie_flags)
+		*cookie_flags = flags;
 	return 1;
 }
 
@@ -1017,6 +1018,9 @@ static int _dm_scan_remove_init(sid_res_t *mod_res, struct sid_ucmd_ctx *ucmd_ct
 {
 	sid_res_log_debug(mod_res, "scan-remove-init");
 
+	if (_get_cookie_props(mod_res, ucmd_ctx, NULL) < 0)
+		return -1;
+
 	if (_dm_submod_common_scan_init(mod_res, ucmd_ctx) < 0)
 		return -1;