def invoke(self, arg, from_tty):
breakpoints = arg
current_pc_int = int(
SysUtils.extract_address(str(gdb.parse_and_eval("$pc"))), 16)
try:
disas_output = gdb.execute("disas $pc-30,$pc", to_string=True)
# Just before the line "End of assembler dump"
last_instruction = disas_output.splitlines()[-2]
previous_pc_address = SysUtils.extract_address(last_instruction)
except:
previous_pc_address = hex(current_pc_int)
global track_watchpoint_dict
try:
count = track_watchpoint_dict[breakpoints][current_pc_int][0] + 1
except KeyError:
if breakpoints not in track_watchpoint_dict:
track_watchpoint_dict[breakpoints] = OrderedDict()
count = 1
register_info = ScriptUtils.get_general_registers()
register_info.update(ScriptUtils.get_flag_registers())
register_info.update(ScriptUtils.get_segment_registers())
float_info = ScriptUtils.get_float_registers()
disas_info = gdb.execute("disas " + previous_pc_address + ",+40",
to_string=True).replace("=>", " ")
track_watchpoint_dict[breakpoints][current_pc_int] = [
count, previous_pc_address, register_info, float_info, disas_info
]
track_watchpoint_file = SysUtils.get_track_watchpoint_file(
pid, breakpoints)
pickle.dump(track_watchpoint_dict[breakpoints],
open(track_watchpoint_file, "wb"))
def invoke(self, arg, from_tty):
(breakpoint, max_trace_count, stop_condition, step_mode,
stop_after_trace, collect_general_registers, collect_flag_registers,
collect_segment_registers, collect_float_registers) = eval(arg)
gdb.execute("delete " + breakpoint)
trace_status_file = SysUtils.get_trace_instructions_status_file(
pid, breakpoint)
# The reason we don't use a tree class is to make the tree json-compatible
# tree format-->[node1, node2, node3, ...]
# node-->[(line_info, register_dict), parent_index, child_index_list]
tree = []
current_index = 0 # Avoid calling len()
current_root_index = 0
root_index = 0
# Root always be an empty node, it's up to you to use or delete it
tree.append([("", None), None, []])
for x in range(max_trace_count):
try:
output = pickle.load(open(trace_status_file, "rb"))
if output[0] == type_defs.TRACE_STATUS.STATUS_CANCELED:
break
except:
pass
line_info = gdb.execute("x/i $pc",
to_string=True).split(maxsplit=1)[1]
collect_dict = OrderedDict()
if collect_general_registers:
collect_dict.update(ScriptUtils.get_general_registers())
if collect_flag_registers:
collect_dict.update(ScriptUtils.get_flag_registers())
if collect_segment_registers:
collect_dict.update(ScriptUtils.get_segment_registers())
if collect_float_registers:
collect_dict.update(ScriptUtils.get_float_registers())
current_index += 1
tree.append([(line_info, collect_dict), current_root_index, []])
tree[current_root_index][2].append(current_index) # Add a child
status_info = (type_defs.TRACE_STATUS.STATUS_TRACING,
line_info + " (" + str(x + 1) + "/" +
str(max_trace_count) + ")")
pickle.dump(status_info, open(trace_status_file, "wb"))
if common_regexes.trace_instructions_ret.search(line_info):
if tree[current_root_index][1] is None: # If no parents exist
current_index += 1
tree.append([("", None), None, [current_root_index]])
tree[current_root_index][
1] = current_index # Set new parent
current_root_index = current_index # current_node=current_node.parent
root_index = current_root_index # set new root
else:
current_root_index = tree[current_root_index][
1] # current_node=current_node.parent
elif step_mode == type_defs.STEP_MODE.SINGLE_STEP:
if common_regexes.trace_instructions_call.search(line_info):
current_root_index = current_index
if stop_condition:
try:
if str(gdb.parse_and_eval(stop_condition)) == "1":
break
except:
pass
if step_mode == type_defs.STEP_MODE.SINGLE_STEP:
gdb.execute("stepi", to_string=True)
elif step_mode == type_defs.STEP_MODE.STEP_OVER:
gdb.execute("nexti", to_string=True)
status_info = (type_defs.TRACE_STATUS.STATUS_PROCESSING,
"Processing the collected data")
pickle.dump(status_info, open(trace_status_file, "wb"))
trace_instructions_file = SysUtils.get_trace_instructions_file(
pid, breakpoint)
json.dump((tree, root_index), open(trace_instructions_file, "w"))
status_info = (type_defs.TRACE_STATUS.STATUS_FINISHED,
"Tracing has been completed")
pickle.dump(status_info, open(trace_status_file, "wb"))
if not stop_after_trace:
gdb.execute("c")
def invoke(self, arg, from_tty):
registers = ScriptUtils.get_general_registers()
registers.update(ScriptUtils.get_flag_registers())
registers.update(ScriptUtils.get_segment_registers())
send_to_pince(registers)