From 4af5ec2a405bc83fa2f96d583ad26c18c680e73b Mon Sep 17 00:00:00 2001
From: ganchuantao <ganchuantao1@huawei.com>
Date: Tue, 29 Jul 2025 21:04:26 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8C=91=E5=8D=95=E4=B8=BB=E5=B9=B2=EF=BC=9A?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E6=99=BA=E8=83=BD=E8=A1=A8ccm=E5=AE=9A?=
 =?UTF-8?q?=E5=88=B6=E5=8C=96=E9=85=8D=E7=BD=AE=20Signed-off-by:ganchuanta?=
 =?UTF-8?q?o=20<ganchuantao1@huawei.com>?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: ganchuantao <ganchuantao1@huawei.com>
---
 device/ohos_test.xml                          |  2 +
 device/plugins/native_daemon/BUILD.gn         |  1 +
 .../native_daemon/include/hook_manager.h      |  1 +
 .../plugins/native_daemon/include/utilities.h |  2 +
 .../native_memory_profiler_sa_service.h       |  2 +
 .../src/native_memory_profiler_sa_service.cpp |  6 ++-
 .../native_memory_profiler_sa/test/BUILD.gn   |  1 +
 .../native_daemon/src/hook_manager.cpp        |  8 ++--
 .../plugins/native_daemon/src/utilities.cpp   | 28 +++++++++++
 device/plugins/native_daemon/test/BUILD.gn    |  1 +
 .../unittest/common/native/utilities_test.cpp | 48 +++++++++++++++++++
 .../resource/testdata/hiprofiler_cfg.json     |  3 ++
 .../resource/testdata/hiprofiler_cfg2.json    |  4 ++
 device/plugins/native_hook/BUILD.gn           |  1 +
 device/plugins/native_hook/test/BUILD.gn      |  1 +
 15 files changed, 104 insertions(+), 5 deletions(-)
 create mode 100644 device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg.json
 create mode 100644 device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg2.json

diff --git a/device/ohos_test.xml b/device/ohos_test.xml
index 254d8b20e..c8cca99d3 100644
--- a/device/ohos_test.xml
+++ b/device/ohos_test.xml
@@ -117,6 +117,8 @@
             <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/syms_from_readelf_64 -> /data/test/resource/testdata/" src="res"/>
             <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/symbols_file_test_elf32 -> /data/test/resource/testdata/" src="res"/>
             <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/symbols_file_test_elf64 -> /data/test/resource/testdata/" src="res"/>
+            <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg.json -> /data/test/resource/testdata/" src="res"/>
+            <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg_2.json -> /data/test/resource/testdata/" src="res"/>
             <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/dwarf/user_regs.dump -> /data/test/resource/testdata/dwarf/" src="res"/>
             <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/dwarf/user_data.dump -> /data/test/resource/testdata/dwarf/" src="res"/>
             <option name="push" value="plugins/native_daemon/test/unittest/resource/testdata/dwarf/hiperf_example_cmd -> /data/test/resource/testdata/dwarf/" src="res"/>
diff --git a/device/plugins/native_daemon/BUILD.gn b/device/plugins/native_daemon/BUILD.gn
index 85fd18368..772938c62 100644
--- a/device/plugins/native_daemon/BUILD.gn
+++ b/device/plugins/native_daemon/BUILD.gn
@@ -130,6 +130,7 @@ ohos_executable("native_daemon") {
     "access_token:libaccesstoken_sdk",
     "access_token:libtokensetproc_shared",
     "bounds_checking_function:libsec_shared",
+    "cJSON:cjson",
     "c_utils:utils",
     "hicollie:libhicollie",
     "hisysevent:libhisysevent",
diff --git a/device/plugins/native_daemon/include/hook_manager.h b/device/plugins/native_daemon/include/hook_manager.h
index 9f744c1de..a565caf31 100644
--- a/device/plugins/native_daemon/include/hook_manager.h
+++ b/device/plugins/native_daemon/include/hook_manager.h
@@ -171,6 +171,7 @@ private:
     uint32_t largestSize_ = 0;
     uint32_t secondLargestSize_ = 0;
     uint32_t maxGrowthSize_ = 0;
+    int32_t maxProcessCount_ = 4;
     std::shared_ptr<HookRecordFactory> factory_{nullptr};
 };
 }
diff --git a/device/plugins/native_daemon/include/utilities.h b/device/plugins/native_daemon/include/utilities.h
index 25d39c3f6..00cbd3083 100644
--- a/device/plugins/native_daemon/include/utilities.h
+++ b/device/plugins/native_daemon/include/utilities.h
@@ -21,6 +21,7 @@
 #include <algorithm>
 #include <cctype>
 #include <cinttypes>
+#include "cJSON.h"
 #include <cstdio>
 #include <fstream>
 #include <iomanip>
@@ -299,6 +300,7 @@ float Percentage(const T &a, const T &b)
 }
 
 bool PowerOfTwo(int n);
+int GetValueFromJsonFile(const std::string& filePath, const std::string key);
 
 #define INDENT_ONE_LEVEL (indent + 1)
 #define INDENT_TWO_LEVEL (indent + 2)
diff --git a/device/plugins/native_daemon/native_memory_profiler_sa/include/native_memory_profiler_sa_service.h b/device/plugins/native_daemon/native_memory_profiler_sa/include/native_memory_profiler_sa_service.h
index 351e60d88..0d9bb99e8 100644
--- a/device/plugins/native_daemon/native_memory_profiler_sa/include/native_memory_profiler_sa_service.h
+++ b/device/plugins/native_daemon/native_memory_profiler_sa/include/native_memory_profiler_sa_service.h
@@ -74,6 +74,8 @@ private:
 
 private:
     std::mutex nmdMtx_;
+    int32_t taskNum_{0};
+    int32_t taskMaxNum_ = 4;
     std::string startupModeProcessName_;
 };
 } // namespace OHOS::Developtools::NativeDaemon
diff --git a/device/plugins/native_daemon/native_memory_profiler_sa/src/native_memory_profiler_sa_service.cpp b/device/plugins/native_daemon/native_memory_profiler_sa/src/native_memory_profiler_sa_service.cpp
index d79b29334..84ed1e0d7 100644
--- a/device/plugins/native_daemon/native_memory_profiler_sa/src/native_memory_profiler_sa_service.cpp
+++ b/device/plugins/native_daemon/native_memory_profiler_sa/src/native_memory_profiler_sa_service.cpp
@@ -30,9 +30,9 @@
 namespace OHOS::Developtools::NativeDaemon {
 namespace {
 constexpr int32_t TIME_BASE = 1000;
-constexpr int32_t MAX_TASK_NUM = 4;
 const std::string FILE_PATH_HEAD = "/data/local/tmp/native_memory_";
 const std::string FILE_PATH_TAIL = ".htrace";
+const std::string PRODUCT_CONFIG_PATH = "/sys_prod/etc/hiview/hiprofiler/hiprofiler_cfg.json";
 constexpr int32_t DELAYED_SHUTDOWN_TIME = 20;
 constexpr int FILE_MODE = 0644;
 constexpr int32_t SIMP_NMD = 3;
@@ -431,7 +431,9 @@ int32_t NativeMemoryProfilerSaService::StartHook(std::shared_ptr<NativeMemoryPro
 
 bool NativeMemoryProfilerSaService::CheckConfig(std::shared_ptr<NativeMemoryProfilerSaConfig>& config, uint32_t fd)
 {
-    if (g_taskNum + 1 > MAX_TASK_NUM) {
+    auto maxTaskCount = GetValueFromJsonFile(PRODUCT_CONFIG_PATH, "hiprofiler_hook_process_count");
+    taskMaxNum_ = maxTaskCount > 0 ? maxTaskCount : taskMaxNum_;
+    if (taskNum_ + 1 > taskMaxNum_) {
         PROFILER_LOG_INFO(LOG_CORE, "NativeMemoryProfilerSaService: Support up to 4 tasks at the same time");
         return false;
     }
diff --git a/device/plugins/native_daemon/native_memory_profiler_sa/test/BUILD.gn b/device/plugins/native_daemon/native_memory_profiler_sa/test/BUILD.gn
index 508dcb537..761438734 100644
--- a/device/plugins/native_daemon/native_memory_profiler_sa/test/BUILD.gn
+++ b/device/plugins/native_daemon/native_memory_profiler_sa/test/BUILD.gn
@@ -129,6 +129,7 @@ ohos_unittest("native_memory_profiler_sa_ut") {
     "access_token:libaccesstoken_sdk",
     "access_token:libtokensetproc_shared",
     "bounds_checking_function:libsec_shared",
+    "cJSON:cjson",
     "c_utils:utils",
     "faultloggerd:libdfx_dumpcatcher",
     "googletest:gtest",
diff --git a/device/plugins/native_daemon/src/hook_manager.cpp b/device/plugins/native_daemon/src/hook_manager.cpp
index 24f4419da..fc7705176 100644
--- a/device/plugins/native_daemon/src/hook_manager.cpp
+++ b/device/plugins/native_daemon/src/hook_manager.cpp
@@ -42,10 +42,10 @@ constexpr uint32_t PAGE_BYTES = 4096;
 std::shared_ptr<Writer> g_buffWriter;
 const std::string STARTUP = "startup:";
 const std::string PARAM_NAME = "libc.hook_mode";
+const std::string PRODUCT_CONFIG_PATH = "/sys_prod/etc/hiview/hiprofiler/hiprofiler_cfg.json";
 constexpr int SIGNAL_START_HOOK = 36;
 const std::string VERSION = "1.02";
 constexpr int32_t RESPONSE_MAX_PID_COUNT = 8;
-constexpr int32_t MAX_PID_COUNT = 4;
 constexpr int32_t SIMP_NMD = 3;
 constexpr int DOUBLE = 2;
 static thread_local std::array<std::shared_ptr<HookRecord>, CACHE_ARRAY_SIZE> g_rawDataArray = {};
@@ -106,8 +106,10 @@ bool HookManager::CheckProcess()
             return false;
         }
     } else {
-        if (hookCtx_.size() > static_cast<size_t>(MAX_PID_COUNT)) {
-            PROFILER_LOG_ERROR(LOG_CORE, "%s: The maximum allowed is to set %d PIDs.", __func__, MAX_PID_COUNT);
+        auto maxPidCount = GetValueFromJsonFile(PRODUCT_CONFIG_PATH, "hiprofiler_hook_process_count");
+        maxProcessCount_ = maxPidCount > 0 ? maxPidCount : maxProcessCount_;
+        if (hookCtx_.size() > maxProcessCount_) {
+            PROFILER_LOG_ERROR(LOG_CORE, "%s: The maximum allowed is to set %d PIDs.", __func__, maxProcessCount_);
             return false;
         }
     }
diff --git a/device/plugins/native_daemon/src/utilities.cpp b/device/plugins/native_daemon/src/utilities.cpp
index 8bdff1fc2..d45126781 100644
--- a/device/plugins/native_daemon/src/utilities.cpp
+++ b/device/plugins/native_daemon/src/utilities.cpp
@@ -54,6 +54,34 @@ std::string StringReplace(std::string source, const std::string &from, const std
     return source;
 }
 
+int GetValueFromJsonFile(const std::string& filePath, const std::string key)
+{
+    std::ifstream inFile(filePath, std::ios::in);
+    if (!inFile.is_open()) {
+        HLOGE("parse json file: %s is not existed.", filePath.c_str());
+        return -1;
+    }
+    std::string fileContent((std::istreambuf_iterator<char>(inFile)), std::istreambuf_iterator<char>());
+    cJSON* jsonNode = cJSON_Parse(fileContent.c_str());
+    inFile.close();
+
+    if (jsonNode == nullptr) {
+        HLOGE("parse json file: %s failed.", filePath.c_str());
+        return -1;
+    }
+
+    cJSON* vaulezNode = cJSON_GetObjectItem(jsonNode, key.c_str());
+    if (vaulezNode == nullptr) {
+        HLOGE("ParseJson hiprofiler_hook_process_count json node not found.");
+        return -1;
+    }
+    if (!cJSON_IsNumber(vaulezNode)) {
+        HLOGE("ParseJson: hiprofiler_hook_process_count item is illegal.");
+        return -1;
+    }
+    return vaulezNode->valueint;
+}
+
 size_t SubStringCount(const std::string &source, const std::string &sub)
 {
     size_t count(0);
diff --git a/device/plugins/native_daemon/test/BUILD.gn b/device/plugins/native_daemon/test/BUILD.gn
index 73e4b78dc..b2285b4eb 100644
--- a/device/plugins/native_daemon/test/BUILD.gn
+++ b/device/plugins/native_daemon/test/BUILD.gn
@@ -131,6 +131,7 @@ ohos_unittest("native_daemon_ut") {
     "access_token:libaccesstoken_sdk",
     "access_token:libtokensetproc_shared",
     "bounds_checking_function:libsec_shared",
+    "cJSON:cjson",
     "c_utils:utils",
     "faultloggerd:libunwinder",
     "googletest:gtest",
diff --git a/device/plugins/native_daemon/test/unittest/common/native/utilities_test.cpp b/device/plugins/native_daemon/test/unittest/common/native/utilities_test.cpp
index f48ef32e3..251f1ef00 100644
--- a/device/plugins/native_daemon/test/unittest/common/native/utilities_test.cpp
+++ b/device/plugins/native_daemon/test/unittest/common/native/utilities_test.cpp
@@ -628,6 +628,54 @@ HWTEST_F(UtilitiesTest, GetSubthreadIDs, TestSize.Level1)
     }
     ExitThreads();
 }
+
+/**
+ * @tc.name: test funtion GetValueFromJsonFile return normal result
+ * @tc.desc:
+ * @tc.type: FUNC
+ */
+HWTEST_F(UtilitiesTest, GetValueFromJsonFile, TestSize.Level1)
+{
+    std::string filePath = "/data/test/resource/testdata/hiprofiler_cfg.json";
+    auto res = GetValueFromJsonFile(filePath, "hiprofiler_hook_process_count");
+    EXPECT_EQ(res, 2);
+}
+
+/**
+ * @tc.name: test funtion GetValueFromJsonFile when path is not exit
+ * @tc.desc:
+ * @tc.type: FUNC
+ */
+HWTEST_F(UtilitiesTest, GetValueFromJsonFile002, TestSize.Level1)
+{
+    std::string filePath = "/data/test/resource/testdata/hiprofiler_cfg22.json";
+    auto res = GetValueFromJsonFile(filePath, "hiprofiler_hook_process_count");
+    EXPECT_EQ(res, -1);
+}
+
+/**
+ * @tc.name: test funtion GetValueFromJsonFile when key is not exit
+ * @tc.desc:
+ * @tc.type: FUNC
+ */
+HWTEST_F(UtilitiesTest, GetValueFromJsonFile003, TestSize.Level1)
+{
+    std::string filePath = "/data/test/resource/testdata/hiprofiler_cfg.json";
+    auto res = GetValueFromJsonFile(filePath, "test");
+    EXPECT_EQ(res, -1);
+}
+
+/**
+ * @tc.name: test funtion GetValueFromJsonFile when value is not int
+ * @tc.desc:
+ * @tc.type: FUNC
+ */
+HWTEST_F(UtilitiesTest, GetValueFromJsonFile004, TestSize.Level1)
+{
+    std::string filePath = "/data/test/resource/testdata/hiprofiler_cfg_2.json";
+    auto res = GetValueFromJsonFile(filePath, "hiprofiler_hook_process_count");
+    EXPECT_EQ(res, -1);
+}
 } // namespace NativeDaemon
 } // namespace Developtools
 } // namespace OHOS
diff --git a/device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg.json b/device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg.json
new file mode 100644
index 000000000..e700aa16d
--- /dev/null
+++ b/device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg.json
@@ -0,0 +1,3 @@
+{
+    "hiprofiler_hook_process_count": 2
+}
\ No newline at end of file
diff --git a/device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg2.json b/device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg2.json
new file mode 100644
index 000000000..6365e9e30
--- /dev/null
+++ b/device/plugins/native_daemon/test/unittest/resource/testdata/hiprofiler_cfg2.json
@@ -0,0 +1,4 @@
+{
+    "hiprofiler_hook_process_count": "4",
+    "test": "native_daemon"
+}
\ No newline at end of file
diff --git a/device/plugins/native_hook/BUILD.gn b/device/plugins/native_hook/BUILD.gn
index b9ce8ff29..0d64a4376 100644
--- a/device/plugins/native_hook/BUILD.gn
+++ b/device/plugins/native_hook/BUILD.gn
@@ -66,6 +66,7 @@ ohos_source_set("native_hook_source") {
   }
   external_deps = [
     "bounds_checking_function:libsec_shared",
+    "cJSON:cjson",
     "c_utils:utils",
     "ffrt:libffrt",
     "hitrace:libhitracechain",
diff --git a/device/plugins/native_hook/test/BUILD.gn b/device/plugins/native_hook/test/BUILD.gn
index 13d5cfb80..06e1da659 100644
--- a/device/plugins/native_hook/test/BUILD.gn
+++ b/device/plugins/native_hook/test/BUILD.gn
@@ -76,6 +76,7 @@ ohos_unittest("nativehook_ut") {
     "access_token:libaccesstoken_sdk",
     "access_token:libtokensetproc_shared",
     "bounds_checking_function:libsec_shared",
+    "cJSON:cjson",
     "c_utils:utils",
     "faultloggerd:libunwinder",
     "googletest:gtest",
-- 
Gitee