From d7c6f94f08492db2c5746018864f511714e64efc Mon Sep 17 00:00:00 2001
From: "elias.bachaalany"
 <elias.bachaalany@fccdda4b-c33c-0410-84de-61e1e3e5f415>
Date: Mon, 18 Jan 2010 10:55:33 +0000
Subject: [PATCH] bugfix: SetRegEx and GetReg were only working with x86
 segment registers

---
 python/idc.py | 35 ++++++++++++++---------------------
 1 file changed, 14 insertions(+), 21 deletions(-)

diff --git a/python/idc.py b/python/idc.py
index 790e576..e52567a 100644
--- a/python/idc.py
+++ b/python/idc.py
@@ -1449,15 +1449,6 @@ def SetFlags(ea, flags):
     """
     return idaapi.setFlags(ea, flags)
 
-_REGMAP = {
-    'es' : idaapi.R_es,
-    'cs' : idaapi.R_cs,
-    'ss' : idaapi.R_ss,
-    'ds' : idaapi.R_ds,
-    'fs' : idaapi.R_fs,
-    'gs' : idaapi.R_gs
-}
-
 def SetRegEx(ea, reg, value, tag):
     """
     Set value of a segment register.
@@ -1473,8 +1464,9 @@ def SetRegEx(ea, reg, value, tag):
 
            See also SetReg() compatibility macro.
     """
-    if reg in _REGMAP:
-        return idaapi.splitSRarea1(ea, _REGMAP[reg], value, tag)
+    reg = idaapi.str2reg(reg);
+    if reg >= 0:
+        return idaapi.splitSRarea1(ea, reg, value, tag)
     else:
         return False
 
@@ -1885,17 +1877,17 @@ def GetReg(ea, reg):
     @param ea: linear address
     @param reg: name of segment register
 
-    @return: the value of the segment register or 0xFFFF on error
+    @return: the value of the segment register or -1 on error
 
     @note: The segment registers in 32bit program usually contain selectors,
            so to get paragraph pointed by the segment register you need to
            call AskSelector() function.
     """
-    if reg in _REGMAP:
-        return idaapi.getSR(ea, _REGMAP[reg]) & 0xFFFF
+    reg = idaapi.str2reg(reg);
+    if reg >= 0:
+        return idaapi.getSR(ea, reg)
     else:
-        return False
-
+        return -1
 
 def NextAddr(ea):
     """
@@ -3342,8 +3334,9 @@ def SetSegDefReg(ea, reg, value):
     """
     seg = idaapi.getseg(ea)
 
-    if seg and reg in _REGMAP:
-        return idaapi.SetDefaultRegisterValue(seg, _REGMAP[reg], value)
+    reg = idaapi.str2reg(reg);
+    if seg and reg >= 0:
+        return idaapi.SetDefaultRegisterValue(seg, reg, value)
     else:
         return False
 
@@ -5110,7 +5103,7 @@ def AddStrucMember(sid, name, offset, flag, typeid, nbytes, target=-1, tdelta=0,
 
     """
     if isOff0(flag):
-        return Eval('AddStrucMember(%d, "%s", %d, %d, %d, %d, %d, %d, %d);' % (sid, name, offset, flag, typeid, nbytes, 
+        return Eval('AddStrucMember(%d, "%s", %d, %d, %d, %d, %d, %d, %d);' % (sid, name, offset, flag, typeid, nbytes,
                                                                                target, tdelta, reftype))
     else:
         return Eval('AddStrucMember(%d, "%s", %d, %d, %d, %d);' % (sid, name, offset, flag, typeid, nbytes))
@@ -5189,7 +5182,7 @@ def SetMemberType(sid, member_offset, flag, typeid, nitems, target=-1, tdelta=0,
     @return: !=0 - ok.
     """
     if isOff0(flag):
-        return Eval('SetMemberType(%d, %d, %d, %d, %d, %d, %d, %d);' % (sid, member_offset, flag, typeid, nitems, 
+        return Eval('SetMemberType(%d, %d, %d, %d, %d, %d, %d, %d);' % (sid, member_offset, flag, typeid, nitems,
                                                                               target, tdelta, reftype))
     else:
         return Eval('SetMemberType(%d, %d, %d, %d, %d);' % (sid, member_offset, flag, typeid, nitems))
@@ -6441,7 +6434,7 @@ def AttachProcess(pid, event_id):
                 will interactively ask the user for the process to attach to.
     @param event_id: reserved, must be -1
 
-    @return: 
+    @return:
              - -2: impossible to find a compatible process
              - -1: impossible to attach to the given process (process died, privilege
                needed, not supported by the debugger plugin, ...)