#define __REGISTERS_H
 
 #include "sysdep/ptrace.h"
+#include "sysdep/archsetjmp.h"
 
 extern void init_thread_registers(union uml_pt_regs *to);
 extern int save_fp_registers(int pid, unsigned long *fp_regs);
 extern void restore_registers(int pid, union uml_pt_regs *regs);
 extern void init_registers(int pid);
 extern void get_safe_registers(unsigned long * regs, unsigned long * fp_regs);
-extern void get_thread_regs(union uml_pt_regs *uml_regs, void *buffer);
+extern unsigned long get_thread_reg(int reg, jmp_buf *buf);
 
 #endif
 
 #include <sched.h>
 #include <fcntl.h>
 #include <errno.h>
-#include <setjmp.h>
 #include <sys/time.h>
 #include <sys/wait.h>
 #include <sys/mman.h>
 
                       HOST_FP_SIZE * sizeof(unsigned long));
 }
 
-void get_thread_regs(union uml_pt_regs *uml_regs, void *buffer)
+unsigned long get_thread_reg(int reg, jmp_buf *buf)
 {
-       struct __jmp_buf *jmpbuf = buffer;
-
-       UPT_SET(uml_regs, EIP, jmpbuf->__eip);
-       UPT_SET(uml_regs, UESP, jmpbuf->__esp);
-       UPT_SET(uml_regs, EBP, jmpbuf->__ebp);
+       switch(reg){
+       case EIP: return buf[0]->__eip;
+       case UESP: return buf[0]->__esp;
+       case EBP: return buf[0]->__ebp;
+       default:
+               printk("get_thread_regs - unknown register %d\n", reg);
+               return 0;
+       }
 }
 
                       HOST_FP_SIZE * sizeof(unsigned long));
 }
 
-void get_thread_regs(union uml_pt_regs *uml_regs, void *buffer)
+unsigned long get_thread_reg(int reg, jmp_buf *buf)
 {
-       struct __jmp_buf *jmpbuf = buffer;
-
-       UPT_SET(uml_regs, RIP, jmpbuf->__rip);
-       UPT_SET(uml_regs, RSP, jmpbuf->__rsp);
-       UPT_SET(uml_regs, RBP, jmpbuf->__rbp);
+       switch(reg){
+       case RIP: return buf[0]->__rip;
+       case RSP: return buf[0]->__rsp;
+       case RBP: return buf[0]->__rbp;
+       default:
+               printk("get_thread_regs - unknown register %d\n", reg);
+               return 0;
+       }
 }
 
 
 #ifdef CONFIG_MODE_SKAS
 #define KSTK_REG(tsk, reg) \
-       ({ union uml_pt_regs regs; \
-          get_thread_regs(®s, tsk->thread.mode.skas.switch_buf); \
-          UPT_REG(®s, reg); })
+       get_thread_reg(reg, tsk->thread.mode.skas.switch_buf)
 #else
 #define KSTK_REG(tsk, reg) (0xbadbabe)
 #endif