Implement enum Register and three high level wrappers

Author: marmistrz

src/sys/ptrace.rs

@@ -3,6 +3,13 @@ use {Errno, Error, Result};
 use libc::{c_void, c_long, siginfo_t};
 use ::unistd::Pid;
 
+//------------------ First part: a low-level wrapper for ptrace -----------------//
+
+#[cfg(all(target_os = "linux",
+          any(target_arch = "x86",
+              target_arch = "x86_64",
+              target_arch = "arm")),
+          )]
 pub mod ptrace {
     use libc::c_int;
 
@@ -140,3 +147,60 @@ pub fn ptrace_setsiginfo(pid: Pid, sig: &siginfo_t) -> Result<()> {
         Err(e) => Err(e),
     }
 }
+
+//-------------------------- Second part: a low-level wrapper for ptrace ----------------------//
+
+#[cfg(target_arch = "x86_64")]
+// We're going to export it anyway
+#[allow(dead_code)]
+#[allow(non_camel_case_types)]
+pub enum Register {
+    R15 = 0 * 8,
+    R14 = 1 * 8,
+    R13 = 2 * 8,
+    R12 = 3 * 8,
+    RBP = 4 * 8,
+    RBX = 5 * 8,
+    R11 = 6 * 8,
+    R10 = 7 * 8,
+    R9 = 8 * 8,
+    R8 = 9 * 8,
+    RAX = 10 * 8,
+    RCX = 11 * 8,
+    RDX = 12 * 8,
+    RSI = 13 * 8,
+    RDI = 14 * 8,
+    ORIG_RAX = 15 * 8,
+    RIP = 16 * 8,
+    CS = 17 * 8,
+    EFLAGS = 18 * 8,
+    RSP = 19 * 8,
+    SS = 20 * 8,
+    FS_BASE = 21 * 8,
+    GS_BASE = 22 * 8,
+    DS = 23 * 8,
+    ES = 24 * 8,
+    FS = 25 * 8,
+    GS = 26 * 8,
+}
+
+/// Makes the `PTRACE_SYSCALL` request to ptrace
+pub fn syscall(pid: Pid) -> Result<()> {
+    ptrace(ptrace::PTRACE_SYSCALL, pid, ptr::null_mut(), ptr::null_mut()).map(|_| ()) // ignore the useless return value
+}
+
+/// Makes the `PTRACE_PEEKUSER` request to ptrace
+pub fn peekuser(pid: Pid, reg: Register) -> Result<c_long> {
+    let reg_arg = (reg as i32) as *mut c_void;
+    ptrace(ptrace::PTRACE_PEEKUSER, pid, reg_arg, ptr::null_mut())
+}
+
+/// Sets the process as traceable with `PTRACE_TRACEME`
+pub fn traceme() -> Result<()> {
+    ptrace(
+        ptrace::PTRACE_TRACEME,
+        Pid::from_raw(0),
+        ptr::null_mut(),
+        ptr::null_mut(),
+    ).map(|_| ()) // ignore the useless return value
+}