/* * 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 #include #include using ::android::hardware::hidl_vec; using ::android::hardware::Return; namespace aidl { namespace android { namespace hardware { namespace memtrack { // Check Memtrack Flags static_assert(static_cast(V1_0_hidl::MemtrackFlag::SMAPS_ACCOUNTED) == static_cast(V1_aidl::MemtrackRecord::FLAG_SMAPS_ACCOUNTED)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::SMAPS_UNACCOUNTED) == static_cast(V1_aidl::MemtrackRecord::FLAG_SMAPS_UNACCOUNTED)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::SHARED) == static_cast(V1_aidl::MemtrackRecord::FLAG_SHARED)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::SHARED_PSS) == static_cast(V1_aidl::MemtrackRecord::FLAG_SHARED_PSS)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::PRIVATE) == static_cast(V1_aidl::MemtrackRecord::FLAG_PRIVATE)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::SYSTEM) == static_cast(V1_aidl::MemtrackRecord::FLAG_SYSTEM)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::DEDICATED) == static_cast(V1_aidl::MemtrackRecord::FLAG_DEDICATED)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::NONSECURE) == static_cast(V1_aidl::MemtrackRecord::FLAG_NONSECURE)); static_assert(static_cast(V1_0_hidl::MemtrackFlag::SECURE) == static_cast(V1_aidl::MemtrackRecord::FLAG_SECURE)); // Check Memtrack Types static_assert(static_cast(V1_0_hidl::MemtrackType::OTHER) == static_cast(V1_aidl::MemtrackType::OTHER)); static_assert(static_cast(V1_0_hidl::MemtrackType::GL) == static_cast(V1_aidl::MemtrackType::GL)); static_assert(static_cast(V1_0_hidl::MemtrackType::GRAPHICS) == static_cast(V1_aidl::MemtrackType::GRAPHICS)); static_assert(static_cast(V1_0_hidl::MemtrackType::MULTIMEDIA) == static_cast(V1_aidl::MemtrackType::MULTIMEDIA)); static_assert(static_cast(V1_0_hidl::MemtrackType::CAMERA) == static_cast(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::max() || in.sizeInBytes < 0) { return false; } out->sizeInBytes = static_cast(in.sizeInBytes); // It's ok to just assign directly, since this is a bitmap. out->flags = in.flags; return true; } sp MemtrackProxy::MemtrackHidlInstance() { return V1_0_hidl::IMemtrack::getService(); } std::shared_ptr 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* _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 ret = memtrack_hidl_instance_->getMemory( pid, static_cast(type), [&_aidl_return, &aidl_status](V1_0_hidl::MemtrackStatus status, hidl_vec 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* _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