def _instrument_code_recursive(
self,
code: CodeType,
parent_code_object_id: Optional[int] = None,
) -> CodeType:
"""Instrument the given Code Object recursively.
Args:
code: The code object that should be instrumented
parent_code_object_id: The ID of the optional parent code object
Returns:
The instrumented code object
"""
self._logger.debug("Instrumenting Code Object for %s", code.co_name)
cfg = CFG.from_bytecode(Bytecode.from_code(code))
cdg = ControlDependenceGraph.compute(cfg)
code_object_id = self._tracer.register_code_object(
CodeObjectMetaData(
code_object=code,
parent_code_object_id=parent_code_object_id,
cfg=cfg,
cdg=cdg,
))
assert cfg.entry_node is not None, "Entry node cannot be None."
real_entry_node = cfg.get_successors(
cfg.entry_node).pop() # Only one exists!
assert real_entry_node.basic_block is not None, "Basic block cannot be None."
self._add_code_object_executed(real_entry_node.basic_block,
code_object_id)
self._instrument_cfg(cfg, code_object_id)
return self._instrument_inner_code_objects(
cfg.bytecode_cfg().to_code(), code_object_id)
def _instrument_code_recursive(
self,
code: CodeType,
) -> CodeType:
"""Instrument the given Code Object recursively.
Args:
code: The code object that should be instrumented
Returns:
The instrumented code object
"""
self._logger.debug(
"Instrumenting Code Object for dynamic seeding for %s",
code.co_name)
cfg = CFG.from_bytecode(Bytecode.from_code(code))
assert cfg.entry_node is not None, "Entry node cannot be None."
real_entry_node = cfg.get_successors(
cfg.entry_node).pop() # Only one exists!
assert real_entry_node.basic_block is not None, "Basic block cannot be None."
self._instrument_cfg(cfg)
return self._instrument_inner_code_objects(
cfg.bytecode_cfg().to_code())
def test_integration_create_cfg(conditional_jump_example_bytecode):
cfg = CFG.from_bytecode(conditional_jump_example_bytecode)
dot_representation = cfg.to_dot()
graph = """strict digraph {
"ProgramGraphNode(0)";
"ProgramGraphNode(1)";
"ProgramGraphNode(2)";
"ProgramGraphNode(3)";
"ProgramGraphNode(9223372036854775807)";
"ProgramGraphNode(-1)";
"ProgramGraphNode(0)" -> "ProgramGraphNode(1)";
"ProgramGraphNode(0)" -> "ProgramGraphNode(2)";
"ProgramGraphNode(1)" -> "ProgramGraphNode(3)";
"ProgramGraphNode(2)" -> "ProgramGraphNode(3)";
"ProgramGraphNode(3)" -> "ProgramGraphNode(9223372036854775807)";
"ProgramGraphNode(-1)" -> "ProgramGraphNode(0)";
}
"""
assert cfg.cyclomatic_complexity == 2
assert cfg.entry_node.is_artificial
assert len(cfg.exit_nodes) == 1
assert dot_representation == graph
def test_integration_while_loop():
while_loop_cfg = CFG.from_bytecode(Bytecode.from_code(Foo.receive.__code__))
assert len(while_loop_cfg.nodes) == 3
assert while_loop_cfg.entry_node.index == -1
assert while_loop_cfg.exit_nodes.pop().index == sys.maxsize
def test_integration_copy_cfg(conditional_jump_example_bytecode):
cfg = CFG.from_bytecode(conditional_jump_example_bytecode)
copied_cfg = cfg.copy()
assert copied_cfg.to_dot() == cfg.to_dot()