diff options
Diffstat (limited to 'dev/LibDebugger')
| -rw-r--r-- | dev/LibDebugger/Debugger.h | 102 |
1 files changed, 102 insertions, 0 deletions
diff --git a/dev/LibDebugger/Debugger.h b/dev/LibDebugger/Debugger.h new file mode 100644 index 0000000..af3217c --- /dev/null +++ b/dev/LibDebugger/Debugger.h @@ -0,0 +1,102 @@ +/*** + (C) 2025 Amlal El Mahrouss + */ + +#include <iostream> +#include <unordered_map> + +#include <sys/ptrace.h> +#include <sys/types.h> +#include <sys/wait.h> +#include <sys/user.h> +#include <unistd.h> +#include <stdint.h> + +namespace LibDebugger +{ + /// \brief Debug IDebugger 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, nullptr) == -1) + { + perror("dbg: Attach"); + exit(1); + } + + waitpid(m_pid, nullptr, 0); + + std::cout << "[+] Attached to process: " << m_pid << std::endl; + } + + void SetBreakpoint(void* addr) + { + long original_data = ptrace(PTRACE_PEEKTEXT, m_pid, addr, nullptr); + if (original_data == -1) + { + perror("dbg: Peek"); + exit(1); + } + + 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"); + exit(1); + } + + std::cout << "[+] Breakpoint set at: " << addr << std::endl; + + m_breakpoints[reinterpret_cast<uintptr_t>(addr)] = original_data; // Store original data + } + + void ContinueExecution() + { + if (ptrace(PTRACE_CONT, m_pid, nullptr, nullptr) == -1) + { + perror("dbg: Cont"); + exit(1); + } + + 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, nullptr) == -1) + { + perror("dbg: Detach"); + exit(1); + } + + std::cout << "[-] Detached from process: " << m_pid << std::endl; + } + + std::unordered_map<uintptr_t, long>& Breakpoints() + { + return m_breakpoints; + } + + private: + pid_t m_pid; + std::unordered_map<uintptr_t, long> m_breakpoints; + }; +} // namespace LibDebugger |
