From 70f413ec1bf13cfe36d08c1ba0a5d8066ea155b6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=B0=B8=E5=BF=A0?= <wangyongzhong2@huawei.com>
Date: Mon, 20 May 2024 04:03:15 +0000
Subject: [PATCH] update fuzz
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 王永忠 <wangyongzhong2@huawei.com>
---
 .../gethighestseclevelasync_fuzzer.cpp              | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/test/fuzztest/datatransmitmgr/gethighestseclevelasync_fuzzer/gethighestseclevelasync_fuzzer.cpp b/test/fuzztest/datatransmitmgr/gethighestseclevelasync_fuzzer/gethighestseclevelasync_fuzzer.cpp
index 6dff429..a39f3b5 100644
--- a/test/fuzztest/datatransmitmgr/gethighestseclevelasync_fuzzer/gethighestseclevelasync_fuzzer.cpp
+++ b/test/fuzztest/datatransmitmgr/gethighestseclevelasync_fuzzer/gethighestseclevelasync_fuzzer.cpp
@@ -17,15 +17,18 @@
 
 #include <cstddef>
 #include <cstdint>
+#include <mutex>
+#include <thread>
 
 #include "file_ex.h"
 #include "securec.h"
 #include "nativetoken_kit.h"
 #include "token_setproc.h"
 #include "accesstoken_kit.h"
-
 #include "dev_slinfo_mgr.h"
 
+#define DELAY_TIME 3000
+
 namespace OHOS {
 static bool g_isForcingFuzz1 = false;
 
@@ -69,8 +72,13 @@ static void EndFuzzCase1(void)
     }
 }
 
+static int32_t g_cnt = 0;
+static std::mutex g_mtx;
+static std::condition_variable g_cv;
+
 static void tmpCallbackFuzz1(DEVSLQueryParams *queryParams, int32_t result, uint32_t levelInfo)
 {
+    g_cnt++;
     (void)queryParams;
     (void)result;
     (void)levelInfo;
@@ -92,6 +100,9 @@ void FuzzDoGetHighestSecLevelAsync(const uint8_t *data, size_t size)
     (void)DATASL_GetHighestSecLevelAsync(&queryParams, tmpCallbackFuzz1);
     (void)DATASL_GetHighestSecLevelAsync(nullptr, tmpCallbackFuzz1);
     (void)DATASL_GetHighestSecLevelAsync(&queryParams, nullptr);
+
+    std::unique_lock<std::mutex> lck(g_mtx);
+    g_cv.wait_for(lck, std::chrono::milliseconds(DELAY_TIME), []() { return (g_cnt == 1); });
     DATASL_OnStop();
     EndFuzzCase1();
 }
-- 
Gitee