From 6530a2f0094b4773e5dc3be1152d5c5a0d6eda84 Mon Sep 17 00:00:00 2001
From: gongguilin <gongguilin@h-partners.com>
Date: Sat, 29 Mar 2025 19:42:50 +0800
Subject: [PATCH] =?UTF-8?q?fuzz=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: gongguilin <gongguilin@h-partners.com>
---
 .../samgrdumper_fuzzer/samgrdumper_fuzzer.cpp | 22 ++++++++++---------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/test/fuzztest/samgrdumper_fuzzer/samgrdumper_fuzzer.cpp b/test/fuzztest/samgrdumper_fuzzer/samgrdumper_fuzzer.cpp
index c81daef5..aa01728b 100644
--- a/test/fuzztest/samgrdumper_fuzzer/samgrdumper_fuzzer.cpp
+++ b/test/fuzztest/samgrdumper_fuzzer/samgrdumper_fuzzer.cpp
@@ -67,6 +67,13 @@ std::string BuildStringFromData(const uint8_t* data, size_t size)
     return strVal;
 }
 
+void InitData(const uint8_t* data, size_t size)
+{
+    g_baseFuzzData = data;
+    g_baseFuzzSize = size;
+    g_baseFuzzPos = 0;
+}
+
 void SamgrDumperFuzzTest(const uint8_t* data, size_t size)
 {
     SamMockPermission::MockProcess(HIDUMPER_PROCESS_NAME);
@@ -103,12 +110,9 @@ void SamgrDumperFuzzTest(const uint8_t* data, size_t size)
     manager->IpcDumpSingleProcess(fd, cmd, processName);
 }
 
-void FuzzListenerDumpProc(const uint8_t* data, size_t size)
+void FuzzListenerDumpProc()
 {
     SamMockPermission::MockProcess(HIDUMPER_PROCESS_NAME);
-    g_baseFuzzData = data;
-    g_baseFuzzSize = size;
-    g_baseFuzzPos = 0;
     int32_t pid1 = GetData<int32_t>();
     int32_t pid2 = GetData<int32_t>();
     std::map<int32_t, std::list<SAListener>> dumpListeners;
@@ -149,11 +153,8 @@ void FuzzListenerDumpProc(const uint8_t* data, size_t size)
     args.clear();
 }
 
-void FuzzFfrtLoadMetrics(const uint8_t* data, size_t size)
+void FuzzFfrtLoadMetrics()
 {
-    g_baseFuzzData = data;
-    g_baseFuzzSize = size;
-    g_baseFuzzPos = 0;
     int32_t pid = GetData<int32_t>();
     std::shared_ptr<SystemAbilityStateScheduler> scheduler = std::make_shared<SystemAbilityStateScheduler>();
     int32_t fd = -1;
@@ -182,9 +183,10 @@ extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
         return 0;
     }
 
+    OHOS::Samgr::InitData(data, size);
     OHOS::Samgr::SamgrDumperFuzzTest(data, size);
-    OHOS::Samgr::FuzzListenerDumpProc(data, size);
-    OHOS::Samgr::FuzzFfrtLoadMetrics(data, size);
+    OHOS::Samgr::FuzzListenerDumpProc();
+    OHOS::Samgr::FuzzFfrtLoadMetrics();
 
     return 0;
 }
-- 
Gitee