From f412b54de20d0fd1db89c7d404782307b683575d Mon Sep 17 00:00:00 2001
From: wu-liushuan <wuliushuan.30044128@h-partners.com>
Date: Sat, 28 Jun 2025 11:38:41 +0800
Subject: [PATCH] =?UTF-8?q?=E5=91=8A=E8=AD=A6=E6=B8=85=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: wu-liushuan <wuliushuan.30044128@h-partners.com>
Change-Id: I45dc320ee2ca04e0838fadada302cad3dd7ff872
---
 .../registertokensynccallback_fuzzer.cpp        | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/test/fuzztest/innerkits/accesstoken/registertokensynccallback_fuzzer/registertokensynccallback_fuzzer.cpp b/test/fuzztest/innerkits/accesstoken/registertokensynccallback_fuzzer/registertokensynccallback_fuzzer.cpp
index ef00997aa..c6c53cccc 100644
--- a/test/fuzztest/innerkits/accesstoken/registertokensynccallback_fuzzer/registertokensynccallback_fuzzer.cpp
+++ b/test/fuzztest/innerkits/accesstoken/registertokensynccallback_fuzzer/registertokensynccallback_fuzzer.cpp
@@ -16,11 +16,16 @@
 #include "registertokensynccallback_fuzzer.h"
 
 #include "accesstoken_kit.h"
+#include "fuzzer/FuzzedDataProvider.h"
 #include "token_setproc.h"
 #include "token_sync_kit_interface.h"
 
 using namespace std;
 using namespace OHOS::Security::AccessToken;
+#ifdef TOKEN_SYNC_ENABLE
+static constexpr int32_t NUMBER_TWO = 2;
+#endif // TOKEN_SYNC_ENABLE
+
 namespace {
 class TokenSyncCallback : public TokenSyncKitInterface {
 public:
@@ -62,10 +67,14 @@ bool RegisterTokenSyncCallbackFuzzTest(const uint8_t* data, size_t size)
     if (!NativeTokenGet()) {
         return false;
     }
-    std::shared_ptr<TokenSyncKitInterface> callback = std::make_shared<TokenSyncCallback>();
-    AccessTokenKit::RegisterTokenSyncCallback(callback);
-    AccessTokenKit::UnRegisterTokenSyncCallback();
-    #endif // TOKEN_SYNC_ENABLE
+    FuzzedDataProvider provider(data, size);
+    if ((provider.ConsumeIntegral<int32_t>() % NUMBER_TWO) == 0) {
+        std::shared_ptr<TokenSyncKitInterface> callback = std::make_shared<TokenSyncCallback>();
+        AccessTokenKit::RegisterTokenSyncCallback(callback);
+    } else {
+        AccessTokenKit::UnRegisterTokenSyncCallback();
+    }
+#endif // TOKEN_SYNC_ENABLE
     return true;
 }
 }
-- 
Gitee