195 lines
8.5 KiB
C++
195 lines
8.5 KiB
C++
/*
|
|
* 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
|