kicad/thirdparty/sentry-native/external/libunwindstack-ndk/Regs.cpp

251 lines
6.7 KiB
C++

/*
* Copyright (C) 2016 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 <errno.h>
#include <stdint.h>
#include <string.h>
#include <sys/ptrace.h>
#include <sys/uio.h>
#include <algorithm>
#include <vector>
#include <unwindstack/Elf.h>
#include <unwindstack/Log.h>
#include <unwindstack/MapInfo.h>
#include <unwindstack/Regs.h>
#include <unwindstack/RegsArm.h>
#include <unwindstack/RegsArm64.h>
#ifdef SENTRY_REMOVED
#include <unwindstack/RegsRiscv64.h>
#endif // SENTRY_REMOVED
#include <unwindstack/RegsX86.h>
#include <unwindstack/RegsX86_64.h>
#include <unwindstack/UserArm.h>
#include <unwindstack/UserArm64.h>
#ifdef SENTRY_REMOVED
#include <unwindstack/UserRiscv64.h>
#endif // SENTRY_REMOVED
#include <unwindstack/UserX86.h>
#include <unwindstack/UserX86_64.h>
namespace unwindstack {
// The largest user structure.
#ifdef SENTRY_REMOVED
// constexpr size_t MAX_USER_REGS_SIZE = sizeof(mips64_user_regs) + 10;
#endif // SENTRY_REMOVED
static constexpr size_t kMaxUserRegsSize = std::max(
sizeof(arm_user_regs),
std::max(sizeof(arm64_user_regs), std::max(sizeof(x86_user_regs), sizeof(x86_64_user_regs))));
// This function assumes that reg_data is already aligned to a 64 bit value.
// If not this could crash with an unaligned access.
Regs* Regs::RemoteGet(pid_t pid, ErrorCode* error_code) {
// Make the buffer large enough to contain the largest registers type.
std::vector<uint64_t> buffer(kMaxUserRegsSize / sizeof(uint64_t));
struct iovec io;
io.iov_base = buffer.data();
io.iov_len = buffer.size() * sizeof(uint64_t);
if (ptrace(PTRACE_GETREGSET, pid, NT_PRSTATUS, reinterpret_cast<void*>(&io)) == -1) {
Log::Error("PTRACE_GETREGSET failed for pid %d: %s", pid, strerror(errno));
if (error_code != nullptr) {
*error_code = ERROR_PTRACE_CALL;
}
return nullptr;
}
// Infer the process architecture from the size of its register structure.
switch (io.iov_len) {
case sizeof(x86_user_regs):
return RegsX86::Read(buffer.data());
case sizeof(x86_64_user_regs):
return RegsX86_64::Read(buffer.data());
case sizeof(arm_user_regs):
return RegsArm::Read(buffer.data());
case sizeof(arm64_user_regs):
return RegsArm64::Read(buffer.data());
#ifdef SENTRY_REMOVED
case sizeof(riscv64_user_regs):
return RegsRiscv64::Read(buffer.data());
#endif // SENTRY_REMOVED
}
Log::Error("No matching size of user regs structure for pid %d: size %zu", pid, io.iov_len);
if (error_code != nullptr) {
*error_code = ERROR_UNSUPPORTED;
}
return nullptr;
}
ArchEnum Regs::RemoteGetArch(pid_t pid, ErrorCode* error_code) {
// Make the buffer large enough to contain the largest registers type.
std::vector<uint64_t> buffer(kMaxUserRegsSize / sizeof(uint64_t));
struct iovec io;
io.iov_base = buffer.data();
io.iov_len = buffer.size() * sizeof(uint64_t);
if (ptrace(PTRACE_GETREGSET, pid, NT_PRSTATUS, reinterpret_cast<void*>(&io)) == -1) {
Log::Error("PTRACE_GETREGSET failed for pid %d: %s", pid, strerror(errno));
if (error_code != nullptr) {
*error_code = ERROR_PTRACE_CALL;
}
return ARCH_UNKNOWN;
}
// Infer the process architecture from the size of its register structure.
switch (io.iov_len) {
case sizeof(x86_user_regs):
return ARCH_X86;
case sizeof(x86_64_user_regs):
return ARCH_X86_64;
case sizeof(arm_user_regs):
return ARCH_ARM;
case sizeof(arm64_user_regs):
return ARCH_ARM64;
}
Log::Error("No matching size of user regs structure for pid %d: size %zu", pid, io.iov_len);
if (error_code != nullptr) {
*error_code = ERROR_UNSUPPORTED;
}
return ARCH_UNKNOWN;
}
Regs* Regs::CreateFromUcontext(ArchEnum arch, void* ucontext) {
switch (arch) {
case ARCH_X86:
return RegsX86::CreateFromUcontext(ucontext);
case ARCH_X86_64:
return RegsX86_64::CreateFromUcontext(ucontext);
case ARCH_ARM:
return RegsArm::CreateFromUcontext(ucontext);
case ARCH_ARM64:
return RegsArm64::CreateFromUcontext(ucontext);
#ifdef SENTRY_REMOVED
case ARCH_RISCV64:
return RegsRiscv64::CreateFromUcontext(ucontext);
#endif // SENTRY_REMOVED
case ARCH_UNKNOWN:
default:
return nullptr;
}
}
ArchEnum Regs::CurrentArch() {
#if defined(__arm__)
return ARCH_ARM;
#elif defined(__aarch64__)
return ARCH_ARM64;
#elif defined(__i386__)
return ARCH_X86;
#elif defined(__x86_64__)
return ARCH_X86_64;
#ifdef SENTRY_REMOVED
//#elif defined(__riscv)
return ARCH_RISCV64;
#endif // SENTRY_REMOVED
#else
abort();
#endif
}
Regs* Regs::CreateFromLocal() {
Regs* regs;
#if defined(__arm__)
regs = new RegsArm();
#elif defined(__aarch64__)
regs = new RegsArm64();
#elif defined(__i386__)
regs = new RegsX86();
#elif defined(__x86_64__)
regs = new RegsX86_64();
#ifdef SENTRY_REMOVED
//#elif defined(__riscv)
regs = new RegsRiscv64();
#endif // SENTRY_REMOVED
#else
abort();
#endif
return regs;
}
uint64_t GetPcAdjustment(uint64_t rel_pc, Elf* elf, ArchEnum arch) {
switch (arch) {
case ARCH_ARM: {
if (!elf->valid()) {
return 2;
}
uint64_t load_bias = elf->GetLoadBias();
if (rel_pc < load_bias) {
if (rel_pc < 2) {
return 0;
}
return 2;
}
uint64_t adjusted_rel_pc = rel_pc - load_bias;
if (adjusted_rel_pc < 5) {
if (adjusted_rel_pc < 2) {
return 0;
}
return 2;
}
if (adjusted_rel_pc & 1) {
// This is a thumb instruction, it could be 2 or 4 bytes.
uint32_t value;
if (!elf->memory()->ReadFully(adjusted_rel_pc - 5, &value, sizeof(value)) ||
(value & 0xe000f000) != 0xe000f000) {
return 2;
}
}
return 4;
}
case ARCH_ARM64: {
#ifdef SENTRY_REMOVED
case ARCH_RISCV64: {
#endif // SENTRY_REMOVED
if (rel_pc < 4) {
return 0;
}
return 4;
#ifdef SENTRY_REMOVED
}
case ARCH_MIPS:
case ARCH_MIPS64: {
if (rel_pc < 8) {
return 0;
}
// For now, just assume no compact branches
return 8;
#endif // SENTRY_REMOVED
}
case ARCH_X86:
case ARCH_X86_64: {
if (rel_pc == 0) {
return 0;
}
return 1;
}
case ARCH_UNKNOWN:
return 0;
}
}
} // namespace unwindstack