Evaluate arguments passed to vmmap

hugsy · Apr 18, 2024 · bb47c27 · bb47c27
1 parent d211658
commit bb47c27
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/gef.py b/gef.py
@@ -8711,6 +8711,10 @@ def do_invoke(self, argv: List[str]) -> None:
                 addr = int(argv[0], 0)
                 if addr >= entry.page_start and addr < entry.page_end:
                     self.print_entry(entry)
+            else:
+                addr = safe_parse_and_eval(argv[0])
+                if addr is not None and addr >= entry.page_start and addr < entry.page_end:
+                    self.print_entry(entry)
         return
 
     def print_entry(self, entry: Section) -> None:

diff --git a/tests/commands/vmmap.py b/tests/commands/vmmap.py
@@ -21,3 +21,6 @@ def test_cmd_vmmap(self):
 
         res = gdb.execute("vmmap stack", to_string=True)
         self.assertGreater(len(res.splitlines()), 1)
+
+        res = gdb.execute("vmmap $rip", to_string=True)
+        self.assertEqual(len(res.splitlines()), 2)