diff --git a/pwndbg/commands/segments.py b/pwndbg/commands/segments.py index 0076944e9..6587fb725 100644 --- a/pwndbg/commands/segments.py +++ b/pwndbg/commands/segments.py @@ -1,6 +1,7 @@ from __future__ import print_function import gdb import pwndbg.regs +import pwndbg.commands class segment(gdb.Function): """Get the flat address of memory based off of the named segment register. @@ -14,3 +15,20 @@ class segment(gdb.Function): segment('fsbase') segment('gsbase') + +@pwndbg.commands.OnlyWhenRunning +@pwndbg.commands.ParsedCommand +def fsbase(): + """ + Prints out the FS base address. See also $fsbase. + """ + print(hex(pwndbg.regs.fsbase)) + + +@pwndbg.commands.OnlyWhenRunning +@pwndbg.commands.ParsedCommand +def gsbase(): + """ + Prints out the GS base address. See also $gsbase. + """ + print(hex(pwndbg.regs.gsbase))