The-Homebrew-Cloud/cemu-idapython
2
0
Fork 0
mirror of https://github.com/cemu-project/idapython.git synced 2024-12-01 05:24:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

idc.py: Implemented segment register support for {Get|Set}SegmentAttr()

Browse Source
This commit is contained in:
gergely.erdelyi 2008-06-15 15:44:36 +00:00
parent 0065fdce94
commit c6444d4ff5
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
python/idc.py
View File

@ -3060,12 +3060,12 @@ def GetSegmentAttr(segea, attr):
@param segea: any address within segment @param segea: any address within segment
@param attr: one of SEGATTR_... constants @param attr: one of SEGATTR_... constants
FIXME: add support for segment registers
""" """
seg = idaapi.getseg(segea) seg = idaapi.getseg(segea)
assert seg, "could not find segment at 0x%x" % segea
if seg: if attr in [ SEGATTR_ES, SEGATTR_CS, SEGATTR_SS, SEGATTR_DS, SEGATTR_FS, SEGATTR_GS ]:
return idaapi.get_defsr(seg, _SEGATTRMAP[attr])
else:
return _IDC_GetAttr(seg, _SEGATTRMAP, attr) return _IDC_GetAttr(seg, _SEGATTRMAP, attr)
@ -3079,14 +3079,14 @@ def SetSegmentAttr(segea, attr, value):
@note: Please note that not all segment attributes are modifiable. @note: Please note that not all segment attributes are modifiable.
Also some of them should be modified using special functions Also some of them should be modified using special functions
like SegAddrng, etc. like SegAddrng, etc.
FIXME: add support for segment registers
""" """
seg = idaapi.getseg(segea) seg = idaapi.getseg(segea)
assert seg, "could not find segment at 0x%x" % segea
if seg: if attr in [ SEGATTR_ES, SEGATTR_CS, SEGATTR_SS, SEGATTR_DS, SEGATTR_FS, SEGATTR_GS ]:
idaapi.set_defsr(seg, _SEGATTRMAP[attr], value)
else:
_IDC_SetAttr(seg, _SEGATTRMAP, attr, value) _IDC_SetAttr(seg, _SEGATTRMAP, attr, value)
return seg.update() return seg.update()
SEGATTR_START = 0 # starting address SEGATTR_START = 0 # starting address
@ -3120,12 +3120,12 @@ _SEGATTRMAP = {
SEGATTR_BITNESS : 'bitness', SEGATTR_BITNESS : 'bitness',
SEGATTR_FLAGS : 'flags', SEGATTR_FLAGS : 'flags',
SEGATTR_SEL : 'sel', SEGATTR_SEL : 'sel',
SEGATTR_ES : '', SEGATTR_ES : 0,
SEGATTR_CS : '', SEGATTR_CS : 1,
SEGATTR_SS : '', SEGATTR_SS : 2,
SEGATTR_DS : '', SEGATTR_DS : 3,
SEGATTR_FS : '', SEGATTR_FS : 4,
SEGATTR_GS : '', SEGATTR_GS : 5,
SEGATTR_TYPE : 'type', SEGATTR_TYPE : 'type',
SEGATTR_COLOR : 'color', SEGATTR_COLOR : 'color',
} }