From 4344acc50ebb2685cecc58b8606d838aa982fffc Mon Sep 17 00:00:00 2001
From: l30005037 <luyangyang7@huawei.com>
Date: Thu, 22 May 2025 10:23:20 +0800
Subject: [PATCH] Fuzz fix Signed-off-by: l30005037 <luyangyang7@huawei.com>

---
 .../fuzztest/rdbservicestub_fuzzer/rdbservicestub_fuzzer.cpp  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/services/distributeddataservice/service/test/fuzztest/rdbservicestub_fuzzer/rdbservicestub_fuzzer.cpp b/services/distributeddataservice/service/test/fuzztest/rdbservicestub_fuzzer/rdbservicestub_fuzzer.cpp
index eacc5de6a..4f126d96e 100644
--- a/services/distributeddataservice/service/test/fuzztest/rdbservicestub_fuzzer/rdbservicestub_fuzzer.cpp
+++ b/services/distributeddataservice/service/test/fuzztest/rdbservicestub_fuzzer/rdbservicestub_fuzzer.cpp
@@ -29,6 +29,8 @@ using namespace OHOS::DistributedRdb;
 
 namespace OHOS {
 const std::u16string INTERFACE_TOKEN = u"OHOS.DistributedRdb.IRdbService";
+constexpr uint32_t CODE_MIN = 0;
+constexpr uint32_t CODE_MAX = static_cast<uint32_t>(RdbServiceCode::RDB_SERVICE_CMD_MAX) + 1;
 constexpr size_t NUM_MIN = 5;
 constexpr size_t NUM_MAX = 12;
 
@@ -39,7 +41,7 @@ bool OnRemoteRequestFuzz(FuzzedDataProvider &provider)
     rdbServiceImpl->OnBind(
         { "RdbServiceStubFuzz", static_cast<uint32_t>(IPCSkeleton::GetSelfTokenID()), std::move(executor) });
 
-    uint32_t code = provider.ConsumeIntegral<uint32_t>();
+    uint32_t code = provider.ConsumeIntegralInRange<uint32_t>(CODE_MIN, CODE_MAX);
     std::vector<uint8_t> remainingData = provider.ConsumeRemainingBytes<uint8_t>();
     MessageParcel request;
     request.WriteInterfaceToken(INTERFACE_TOKEN);
-- 
Gitee