Made elfheader run info files

joeleong · May 31, 2016 · 19b1bb2 · 19b1bb2
1 parent 0ab7574
commit 19b1bb2
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/pwndbg/__init__.py b/pwndbg/__init__.py
@@ -61,6 +61,7 @@
 import pwndbg.commands.peda
 import pwndbg.commands.gdbinit
 import pwndbg.commands.defcon
+import pwndbg.commands.elfheader
 
 
 

diff --git a/pwndbg/commands/elfheader.py b/pwndbg/commands/elfheader.py
@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+import gdb
+import pwndbg.commands
+
+@pwndbg.commands.Command
+def elfheader():
+    """
+    Prints the section mappings contained in the ELF header
+    """
+    gdb.execute('info files')
diff --git a/pwndbg/commands/vmmap.py b/pwndbg/commands/vmmap.py
@@ -36,11 +36,3 @@ def vmmap(map=None):
             continue
 
         print(pwndbg.color.get(page.vaddr, text=str(page)))
-
-@pwndbg.commands.OnlyWhenRunning
-@pwndbg.commands.QuietSloppyParsedCommand
-def elfheader(map=None):
-    """
-    Provides PEDA compatibility.  See vmmap command.
-    """
-    return vmmap(map)