From 16064ce66e8d79493d0f4289408aeaab7b84d478 Mon Sep 17 00:00:00 2001
From: Derek Parker <parkerderek86@gmail.com>
Date: Thu, 23 Apr 2015 11:40:20 -0500
Subject: [PATCH] Refactor save/restore register funcs

---
 proctl/threads_linux.go | 22 ++++++++--------------
 1 file changed, 8 insertions(+), 14 deletions(-)

diff --git a/proctl/threads_linux.go b/proctl/threads_linux.go
index d629d101..603a6cd8 100644
--- a/proctl/threads_linux.go
+++ b/proctl/threads_linux.go
@@ -50,24 +50,18 @@ func (t *ThreadContext) blocked() bool {
 	return false
 }
 
-func writeMemory(thread *ThreadContext, addr uintptr, data []byte) (int, error) {
-	return sys.PtracePokeData(thread.Id, addr, data)
+func (thread *ThreadContext) saveRegisters() error {
+	return sys.PtraceGetRegs(thread.Id, &thread.os.registers)
 }
 
-func readMemory(thread *ThreadContext, addr uintptr, data []byte) (int, error) {
-	return sys.PtracePeekData(thread.Id, addr, data)
+func (thread *ThreadContext) restoreRegisters() error {
+	return sys.PtraceSetRegs(thread.Id, &thread.os.registers)
 }
 
-func (thread *ThreadContext) saveRegisters() error {
-	var regs sys.PtraceRegs
-	err := sys.PtraceGetRegs(thread.Id, &regs)
-	if err != nil {
-		return err
-	}
-	thread.os.registers = regs
-	return nil
+func writeMemory(thread *ThreadContext, addr uintptr, data []byte) (int, error) {
+	return sys.PtracePokeData(thread.Id, addr, data)
 }
 
-func (thread *ThreadContext) restoreRegisters() error {
-	return sys.PtraceSetRegs(thread.Id, &thread.os.registers)
+func readMemory(thread *ThreadContext, addr uintptr, data []byte) (int, error) {
+	return sys.PtracePeekData(thread.Id, addr, data)
 }
-- 
GitLab