From 89d58722a47b6568ec67e1930bae50efd1ae2115 Mon Sep 17 00:00:00 2001
From: pompolic <pompolic@special-circumstanc.es>
Date: Mon, 18 Apr 2022 18:59:06 +0200
Subject: [PATCH] Copy initialization code into separate file
---
gdb-port/initialize.py | 51 +++++++++++++++++++++
gdb-port/parser-name-instrumentation-gdb.py | 1 +
2 files changed, 52 insertions(+)
create mode 100644 gdb-port/initialize.py
diff --git a/gdb-port/initialize.py b/gdb-port/initialize.py
new file mode 100644
index 0000000..d0571a3
--- /dev/null
+++ b/gdb-port/initialize.py
@@ -0,0 +1,51 @@
+# Should be loaded last
+
+
+print(": Initializing BreakpointManager")
+breakpoint_manager = BreakpointManager(H_RULE_FUNCTIONS)
+
+
+print(": Registering exit handler")
+# Clean up by-address breakpoints in hammer when inferior exits.
+# Caveat: Assumes there's a single inferior, the debugged parser, so no checking is done
+# TODO: where to store breakpoints? TopLevelParse? A BreakpointManager class?i
+def exit_handler(event):
+ #breakpoints = [ perform_lowlevel_parse_ret, h_packrat_parse_ret ]
+ #del_hammer_retq_breakpoints(breakpoints)
+ breakpoint_manager.del_hammer_retq_breakpoints()
+
+gdb.events.exited.connect(exit_handler)
+
+
+print(": Setting PDFMain breakpoint")
+# Break on main so that libhammer.so gets to load
+main = PDFMainBreakpoint("main")
+
+
+print(": Setting Hammer library breakpoints")
+print(":: Normal breakpoints")
+breakpoint_manager.set_hammer_breakpoints()
+
+#TODO: there would be less complaining about pending breakpoints if vtable breakpoints were set after running to main
+print(":: Parser vtable breakpoints")
+breakpoint_manager.set_parser_virtual_breakpoints()
+
+# run until main
+print(": Running until main")
+gdb.execute("run")
+
+print(": Setting application breakpoints")
+print(":: init_parser breakpoint")
+breakpoint_manager.set_init_parser_breakpoint()
+
+print(":: RET breakpoints in functions with H_RULES")
+breakpoint_manager.set_h_rule_breakpoints()
+
+# TODO: the RET breakpoints in hammer break when "run" is executed again. figure out a way to automatically replace these
+
+# Run until stop position, if set. Finish parsing otherwise
+print(": Continuing execution")
+gdb.execute("continue")
+
+print(": Printing memory statistics")
+print([(p.name, hex(p.address), p.bytes_used) for p in top_level_parse.parser_objs.values()])
diff --git a/gdb-port/parser-name-instrumentation-gdb.py b/gdb-port/parser-name-instrumentation-gdb.py
index a4b5d3f..95bcbcb 100644
--- a/gdb-port/parser-name-instrumentation-gdb.py
+++ b/gdb-port/parser-name-instrumentation-gdb.py
@@ -621,6 +621,7 @@ breakpoint_manager.set_hammer_breakpoints()
#parse_get = ParserVirtualBreakpoint("parse_get")
#parse_whitespace = ParserVirtualBreakpoint("parse_whitespace")
#parse_xor = ParserVirtualBreakpoint("parse_xor")
+#TODO: there would be less complaining about pending breakpoints if vtable breakpoints were set after running to main
print(":: Parser vtable breakpoints")
breakpoint_manager.set_parser_virtual_breakpoints()
--
GitLab