YANNIS
78 lines
2.2 KiB
C++
78 lines
2.2 KiB
C++
// Breakpoint.cpp
|
|
#include "Breakpoint.hpp"
|
|
#include <sys/ptrace.h>
|
|
#include <unistd.h>
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <stdexcept>
|
|
#include <cerrno>
|
|
#include <cstdio>
|
|
|
|
Breakpoint::Breakpoint(pid_t pid, std::intptr_t addr)
|
|
: m_pid(pid), m_addr(addr) {
|
|
std::cout << "Creating breakpoint at address: " << std::hex << addr << std::dec << " for process: " << pid << std::endl;
|
|
if (addr % 4 != 0) {
|
|
throw std::runtime_error("[!] Breakpoint address is not 4-byte aligned (AArch64 requires alignment).");
|
|
}
|
|
m_enabled = false;
|
|
m_saved_data = 0;
|
|
}
|
|
|
|
void Breakpoint::enable() {
|
|
if (m_addr % 4 != 0) {
|
|
throw std::runtime_error("[!] Breakpoint address is not 4-byte aligned (AArch64 requires alignment).");
|
|
}
|
|
|
|
errno = 0;
|
|
long data = ptrace(PTRACE_PEEKDATA, m_pid, m_addr, nullptr);
|
|
if (data == -1 && errno != 0) {
|
|
perror("ptrace PEEKDATA failed");
|
|
throw std::runtime_error("ptrace PEEKDATA failed");
|
|
}
|
|
|
|
std::memcpy(&m_saved_data, &data, sizeof(uint32_t));
|
|
|
|
uint32_t brk_instr = 0xD4200000;
|
|
long data_with_brk;
|
|
std::memcpy(&data_with_brk, &brk_instr, sizeof(uint32_t));
|
|
data_with_brk |= (data & ~0xFFFFFFFF);
|
|
|
|
if (ptrace(PTRACE_POKEDATA, m_pid, m_addr, data_with_brk) == -1) {
|
|
perror("ptrace POKEDATA failed");
|
|
throw std::runtime_error("ptrace POKEDATA failed");
|
|
}
|
|
|
|
m_enabled = true;
|
|
}
|
|
|
|
void Breakpoint::disable() {
|
|
errno = 0;
|
|
long data = ptrace(PTRACE_PEEKDATA, m_pid, m_addr, nullptr);
|
|
if (data == -1 && errno != 0) {
|
|
perror("ptrace PEEKDATA failed");
|
|
throw std::runtime_error("ptrace PEEKDATA failed");
|
|
}
|
|
|
|
long restored_data = (data & ~0xFFFFFFFF) | m_saved_data;
|
|
if (ptrace(PTRACE_POKEDATA, m_pid, m_addr, restored_data) == -1) {
|
|
perror("ptrace POKEDATA failed");
|
|
throw std::runtime_error("ptrace POKEDATA failed");
|
|
}
|
|
|
|
m_enabled = false;
|
|
}
|
|
|
|
std::intptr_t Breakpoint::get_address() const {
|
|
return m_addr;
|
|
}
|
|
|
|
bool Breakpoint::is_enabled() const {
|
|
return m_enabled;
|
|
}
|
|
|
|
std::ostream& operator<<(std::ostream& os, const Breakpoint& bp) {
|
|
os << "pid: " << bp.m_pid << "Breakpoint at " << std::hex << bp.m_addr << (bp.m_enabled ? " (enabled)" : " (disabled)");
|
|
return os;
|
|
}
|
|
|