1 /* 2 * Copyright (c) 2023-2025 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_MEMORY_COLLECTOR_IMPL_H 17 #define HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_MEMORY_COLLECTOR_IMPL_H 18 19 #include "memory_collector.h" 20 21 namespace OHOS { 22 namespace HiviewDFX { 23 namespace UCollectUtil { 24 struct IHiaiInfra { 25 int32_t (*QueryAllUserAllocatedMemInfo)(struct IHiaiInfra* self, struct AIProcessMem* memInfos, 26 uint32_t* memInfosLen, int32_t* realSize); 27 28 int32_t (*GetVersion)(struct IHiaiInfra* self, uint32_t* majorVer, uint32_t* minorVer); 29 }; 30 31 class MemoryCollectorImpl : public MemoryCollector { 32 public: 33 MemoryCollectorImpl() = default; 34 virtual ~MemoryCollectorImpl() = default; 35 36 public: 37 virtual CollectResult<ProcessMemory> CollectProcessMemory(int32_t pid) override; 38 virtual CollectResult<SysMemory> CollectSysMemory() override; 39 virtual CollectResult<std::string> CollectRawMemInfo() override; 40 virtual CollectResult<std::string> ExportMemView() override; 41 virtual CollectResult<std::vector<ProcessMemory>> CollectAllProcessMemory() override; 42 virtual CollectResult<std::string> ExportAllProcessMemory() override; 43 virtual CollectResult<std::string> CollectRawSlabInfo() override; 44 virtual CollectResult<std::string> CollectRawPageTypeInfo() override; 45 virtual CollectResult<std::string> CollectRawDMA() override; 46 virtual CollectResult<std::vector<AIProcessMem>> CollectAllAIProcess() override; 47 virtual CollectResult<std::string> ExportAllAIProcess() override; 48 virtual CollectResult<std::string> CollectRawSmaps(int32_t pid) override; 49 virtual CollectResult<std::string> CollectHprof(int32_t pid) override; 50 virtual CollectResult<uint64_t> CollectProcessVss(int32_t pid) override; 51 virtual CollectResult<MemoryLimit> CollectMemoryLimit() override; 52 virtual CollectResult<uint32_t> CollectDdrFreq() override; 53 virtual CollectResult<ProcessMemoryDetail> CollectProcessMemoryDetail(int32_t pid, 54 GraphicMemOption option) override; 55 }; 56 } // namespace UCollectUtil 57 } // namespace HiviewDFX 58 } // namespace OHOS 59 #endif // HIVIEW_FRAMEWORK_NATIVE_UNIFIED_COLLECTION_MEMORY_COLLECTOR_IMPL_H