diff options
Diffstat (limited to 'services/memtrackproxy/MemtrackProxy.cpp')
-rw-r--r-- | services/memtrackproxy/MemtrackProxy.cpp | 194 |
1 files changed, 194 insertions, 0 deletions
diff --git a/services/memtrackproxy/MemtrackProxy.cpp b/services/memtrackproxy/MemtrackProxy.cpp new file mode 100644 index 0000000000..467616724a --- /dev/null +++ b/services/memtrackproxy/MemtrackProxy.cpp @@ -0,0 +1,194 @@ +/* + * Copyright (C) 2021 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "MemtrackProxy.h" + +#include <android-base/logging.h> +#include <android/binder_manager.h> +#include <private/android_filesystem_config.h> + +using ::android::hardware::hidl_vec; +using ::android::hardware::Return; + +namespace aidl { +namespace android { +namespace hardware { +namespace memtrack { + +// Check Memtrack Flags +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::SMAPS_ACCOUNTED) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_SMAPS_ACCOUNTED)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::SMAPS_UNACCOUNTED) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_SMAPS_UNACCOUNTED)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::SHARED) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_SHARED)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::SHARED_PSS) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_SHARED_PSS)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::PRIVATE) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_PRIVATE)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::SYSTEM) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_SYSTEM)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::DEDICATED) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_DEDICATED)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::NONSECURE) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_NONSECURE)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackFlag::SECURE) == + static_cast<uint32_t>(V1_aidl::MemtrackRecord::FLAG_SECURE)); + +// Check Memtrack Types +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackType::OTHER) == + static_cast<uint32_t>(V1_aidl::MemtrackType::OTHER)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackType::GL) == + static_cast<uint32_t>(V1_aidl::MemtrackType::GL)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackType::GRAPHICS) == + static_cast<uint32_t>(V1_aidl::MemtrackType::GRAPHICS)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackType::MULTIMEDIA) == + static_cast<uint32_t>(V1_aidl::MemtrackType::MULTIMEDIA)); +static_assert(static_cast<uint32_t>(V1_0_hidl::MemtrackType::CAMERA) == + static_cast<uint32_t>(V1_aidl::MemtrackType::CAMERA)); + +__attribute__((warn_unused_result)) bool translate(const V1_0_hidl::MemtrackRecord& in, + V1_aidl::MemtrackRecord* out) { + // Convert uint64_t to int64_t (long in AIDL). AIDL doesn't support unsigned types. + if (in.sizeInBytes > std::numeric_limits<int64_t>::max() || in.sizeInBytes < 0) { + return false; + } + out->sizeInBytes = static_cast<int64_t>(in.sizeInBytes); + + // It's ok to just assign directly, since this is a bitmap. + out->flags = in.flags; + return true; +} + +sp<V1_0_hidl::IMemtrack> MemtrackProxy::MemtrackHidlInstance() { + return V1_0_hidl::IMemtrack::getService(); +} + +std::shared_ptr<V1_aidl::IMemtrack> MemtrackProxy::MemtrackAidlInstance() { + const auto instance = std::string() + V1_aidl::IMemtrack::descriptor + "/default"; + bool declared = AServiceManager_isDeclared(instance.c_str()); + if (!declared) { + return nullptr; + } + ndk::SpAIBinder memtrack_binder = + ndk::SpAIBinder(AServiceManager_waitForService(instance.c_str())); + return V1_aidl::IMemtrack::fromBinder(memtrack_binder); +} + +bool MemtrackProxy::CheckUid(uid_t calling_uid) { + // Allow AID_SHELL for adb shell dumpsys meminfo + return calling_uid == AID_SYSTEM || calling_uid == AID_ROOT || calling_uid == AID_SHELL; +} + +bool MemtrackProxy::CheckPid(pid_t calling_pid, pid_t request_pid) { + return calling_pid == request_pid; +} + +MemtrackProxy::MemtrackProxy() + : memtrack_hidl_instance_(MemtrackProxy::MemtrackHidlInstance()), + memtrack_aidl_instance_(MemtrackProxy::MemtrackAidlInstance()) {} + +ndk::ScopedAStatus MemtrackProxy::getMemory(int pid, MemtrackType type, + std::vector<MemtrackRecord>* _aidl_return) { + if (pid < 0) { + return ndk::ScopedAStatus::fromExceptionCode(EX_ILLEGAL_ARGUMENT); + } + + if (!MemtrackProxy::CheckPid(AIBinder_getCallingPid(), pid) && + !MemtrackProxy::CheckUid(AIBinder_getCallingUid())) { + return ndk::ScopedAStatus::fromExceptionCodeWithMessage( + EX_SECURITY, + "Only AID_ROOT, AID_SYSTEM and AID_SHELL can request getMemory() for PIDs other " + "than the calling PID"); + } + + if (type != MemtrackType::OTHER && type != MemtrackType::GL && type != MemtrackType::GRAPHICS && + type != MemtrackType::MULTIMEDIA && type != MemtrackType::CAMERA) { + return ndk::ScopedAStatus::fromExceptionCode(EX_UNSUPPORTED_OPERATION); + } + + _aidl_return->clear(); + + if (memtrack_aidl_instance_) { + return memtrack_aidl_instance_->getMemory(pid, type, _aidl_return); + } else if (memtrack_hidl_instance_) { + ndk::ScopedAStatus aidl_status; + + Return<void> ret = memtrack_hidl_instance_->getMemory( + pid, static_cast<V1_0_hidl::MemtrackType>(type), + [&_aidl_return, &aidl_status](V1_0_hidl::MemtrackStatus status, + hidl_vec<V1_0_hidl::MemtrackRecord> records) { + switch (status) { + case V1_0_hidl::MemtrackStatus::SUCCESS: + aidl_status = ndk::ScopedAStatus::ok(); + break; + case V1_0_hidl::MemtrackStatus::MEMORY_TRACKING_NOT_SUPPORTED: + [[fallthrough]]; + case V1_0_hidl::MemtrackStatus::TYPE_NOT_SUPPORTED: + [[fallthrough]]; + default: + aidl_status = + ndk::ScopedAStatus::fromExceptionCode(EX_UNSUPPORTED_OPERATION); + return; + } + + _aidl_return->resize(records.size()); + for (size_t i = 0; i < records.size(); i++) { + if (!translate(records[i], &(*_aidl_return)[i])) { + aidl_status = ndk::ScopedAStatus::fromExceptionCodeWithMessage( + EX_SERVICE_SPECIFIC, + "Failed to convert HIDL MemtrackRecord to AIDL"); + return; + } + } + }); + + // Check HIDL return + if (!ret.isOk()) { + const char* err_msg = "HIDL Memtrack::getMemory() failed"; + aidl_status = + ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_SERVICE_SPECIFIC, err_msg); + LOG(ERROR) << err_msg << ": " << ret.description(); + } + + return aidl_status; + } + + return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_NULL_POINTER, + "Memtrack HAL service not available"); +} + +ndk::ScopedAStatus MemtrackProxy::getGpuDeviceInfo(std::vector<DeviceInfo>* _aidl_return) { + if (!MemtrackProxy::CheckUid(AIBinder_getCallingUid())) { + return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_SECURITY, + "Only AID_ROOT, AID_SYSTEM and AID_SHELL can request getGpuDeviceInfo()"); + } + + _aidl_return->clear(); + + if (memtrack_aidl_instance_ || + (memtrack_aidl_instance_ = MemtrackProxy::MemtrackAidlInstance())) { + return memtrack_aidl_instance_->getGpuDeviceInfo(_aidl_return); + } + + return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_NULL_POINTER, + "Memtrack HAL service not available"); +} + +} // namespace memtrack +} // namespace hardware +} // namespace android +} // namespace aidl |