From 106adcb98390d41ced28ecbbd0b661d3056023d2 Mon Sep 17 00:00:00 2001 From: Amlal El Mahrouss Date: Sun, 9 Feb 2025 09:15:30 +0100 Subject: ADD: Add LibDebugger as a separate dylib, fix depreciation warning for ptrace. Signed-off-by: Amlal El Mahrouss --- dev/LibDebugger/IDebugger.h | 113 --------------------------- dev/LibDebugger/IDebuggerEngine.h | 119 +++++++++++++++++++++++++++++ dev/LibDebugger/src/DebuggerEnginePOSIX.cc | 70 +++++++++++++++++ 3 files changed, 189 insertions(+), 113 deletions(-) delete mode 100644 dev/LibDebugger/IDebugger.h create mode 100644 dev/LibDebugger/IDebuggerEngine.h create mode 100644 dev/LibDebugger/src/DebuggerEnginePOSIX.cc (limited to 'dev/LibDebugger') diff --git a/dev/LibDebugger/IDebugger.h b/dev/LibDebugger/IDebugger.h deleted file mode 100644 index 957cea9..0000000 --- a/dev/LibDebugger/IDebugger.h +++ /dev/null @@ -1,113 +0,0 @@ -/*** - (C) 2025 Amlal El Mahrouss - */ - -#include -#include - -#include -#include -#include -#include -#include -#include - -#ifdef __APPLE__ -#define PTRACE_ATTACH PT_ATTACH -#define PTRACE_DETACH PT_DETACH -#define PTRACE_POKETEXT PT_WRITE_I -#define PTRACE_CONT PT_CONTINUE -#define PTRACE_PEEKTEXT PT_READ_I -#endif - -namespace LibDebugger -{ - typedef char* VmAddress; - - /// \brief Debugger interface class in C++ - /// \author Amlal El Mahrouss - class IDebugger final - { - public: - IDebugger() = default; - ~IDebugger() = default; - - IDebugger& operator=(const IDebugger&) = default; - IDebugger(const IDebugger&) = default; - - public: - void Attach(pid_t pid) - { - this->m_pid = pid; - - if (ptrace(PTRACE_ATTACH, this->m_pid, nullptr, 0) == -1) - { - perror("dbg: Attach"); - return; - } - - waitpid(m_pid, nullptr, 0); - - std::cout << "[+] Attached to process: " << m_pid << std::endl; - } - - void SetBreakpoint(VmAddress addr) - { - long original_data = ptrace(PTRACE_PEEKTEXT, m_pid, addr, 0); - - if (original_data == -1) - { - perror("dbg: Peek"); - return; - } - - long data_with_int3 = (original_data & ~0xFF) | 0xCC; // Insert INT3 (0xCC) - if (ptrace(PTRACE_POKETEXT, m_pid, addr, data_with_int3) == -1) - { - perror("dbg: Poke"); - return; - } - - std::cout << "[+] Breakpoint set at: " << addr << std::endl; - - m_breakpoints[reinterpret_cast(addr)] = original_data; // Store original data - } - - void ContinueExecution() - { - if (ptrace(PTRACE_CONT, m_pid, nullptr, 0) == -1) - { - perror("dbg: Cont"); - return; - } - - int status; - waitpid(m_pid, &status, 0); - - if (WIFSTOPPED(status) && WSTOPSIG(status) == SIGTRAP) - { - std::cout << "[!] Breakpoint hit." << std::endl; - } - } - - void Detach() - { - if (ptrace(PTRACE_DETACH, m_pid, nullptr, 0) == -1) - { - perror("dbg: Detach"); - return; - } - - std::cout << "[-] Detached from process: " << m_pid << std::endl; - } - - std::unordered_map& Breakpoints() - { - return m_breakpoints; - } - - private: - pid_t m_pid; - std::unordered_map m_breakpoints; - }; -} // namespace LibDebugger diff --git a/dev/LibDebugger/IDebuggerEngine.h b/dev/LibDebugger/IDebuggerEngine.h new file mode 100644 index 0000000..aefe117 --- /dev/null +++ b/dev/LibDebugger/IDebuggerEngine.h @@ -0,0 +1,119 @@ +/*** + (C) 2025 Amlal El Mahrouss + */ + +#include +#include + +#include +#include +#include +#include +#include +#include + +#ifdef __APPLE__ +#define PTRACE_ATTACH PT_ATTACHEXC +#define PTRACE_DETACH PT_DETACH +#define PTRACE_POKETEXT PT_WRITE_I +#define PTRACE_CONT PT_CONTINUE +#define PTRACE_PEEKTEXT PT_READ_I +#endif + +namespace LibDebugger +{ +#ifdef __APPLE__ + typedef caddr_t CAddr; +#else + typedef char* CAddr; +#endif + + /// \brief Debugger engine interface class in C++ + /// \author Amlal El Mahrouss + class IDebuggerEngine final + { + public: + explicit IDebuggerEngine() = default; + ~IDebuggerEngine() = default; + + public: + IDebuggerEngine& operator=(const IDebuggerEngine&) = default; + IDebuggerEngine(const IDebuggerEngine&) = default; + + public: + void Attach(pid_t pid) + { + if (ptrace(PTRACE_ATTACH, pid, nullptr, 0) == -1) + { + perror("dbg: Attach"); + return; + } + + this->m_pid = pid; + + waitpid(m_pid, nullptr, 0); + + std::cout << "[+] Attached to process: " << m_pid << std::endl; + } + + void SetBreakpoint(CAddr addr) + { + auto original_data = ptrace(PTRACE_PEEKTEXT, m_pid, addr, 0); + + if (original_data == -1) + { + perror("dbg: Peek"); + return; + } + + auto data_with_int3 = (original_data & ~0xFF) | 0xCC; // Insert INT3 (0xCC) + + if (ptrace(PTRACE_POKETEXT, m_pid, addr, data_with_int3) == -1) + { + perror("dbg: Poke"); + return; + } + + std::cout << "[+] Breakpoint set at: " << addr << std::endl; + + m_breakpoints[reinterpret_cast(addr)] = original_data; // Store original data + } + + void ContinueExecution() + { + if (ptrace(PTRACE_CONT, m_pid, nullptr, 0) == -1) + { + perror("dbg: Cont"); + return; + } + + int status; + waitpid(m_pid, &status, 0); + + if (WIFSTOPPED(status) && WSTOPSIG(status) == SIGTRAP) + { + std::cout << "[!] Breakpoint hit." << std::endl; + } + } + + void Detach() + { + if (ptrace(PTRACE_DETACH, m_pid, nullptr, 0) == -1) + { + perror("dbg: Detach"); + return; + } + + std::cout << "[-] Detached from process: " << m_pid << std::endl; + } + + std::unordered_map& Breakpoints() + { + return m_breakpoints; + } + + private: + pid_t m_pid; + std::unordered_map m_breakpoints; + }; +} // namespace LibDebugger diff --git a/dev/LibDebugger/src/DebuggerEnginePOSIX.cc b/dev/LibDebugger/src/DebuggerEnginePOSIX.cc new file mode 100644 index 0000000..e8a394e --- /dev/null +++ b/dev/LibDebugger/src/DebuggerEnginePOSIX.cc @@ -0,0 +1,70 @@ +/*** + (C) 2025 Amlal El Mahrouss + */ + +#include +#include + +#ifndef _WIN32 + +LIBCOMPILER_MODULE(DebuggerPOSIX) +{ + LibDebugger::IDebuggerEngine debugger; + pid_t pid = 0L; + + if (argc >= 3 && std::string(argv[1]) == "-p" && + argv[2] != nullptr) + { + pid = std::stoi(argv[2]); + debugger.Attach(pid); + } + + while (YES) + { + std::string cmd; + std::getline(std::cin, cmd); + + if (cmd == "c" || + cmd == "cont") + debugger.ContinueExecution(); + + if (cmd == "d" || + cmd == "detach") + debugger.Detach(); + + if (cmd == "attach") + { + std::cout << "[?] Enter a PID to attach on: "; + + std::getline(std::cin, cmd); + pid = std::stoi(cmd.c_str()); + + debugger.Attach(pid); + } + + if (cmd == "exit") + { + if (pid > 0) + debugger.Detach(); + + break; + } + + if (cmd == "break" || + cmd == "b") + { + std::cout << "[?] Enter an address to add a breakpoint on: "; + + std::getline(std::cin, cmd); + + LibDebugger::CAddr breakpoint_addr = reinterpret_cast(std::stoul(cmd.c_str(), nullptr, 16)); + + if (breakpoint_addr) + debugger.SetBreakpoint(breakpoint_addr); + } + } + + return 0; +} + +#endif \ No newline at end of file -- cgit v1.2.3