def test_delegate_add_node_to_the_underlying_graph_object(
digraph_add_node_mock):
definition = _a_mock_definition()
dep_graph = DepGraph()
dep_graph.add_node(definition)
digraph_add_node_mock.assert_called_once_with(definition)
def test_add_dependencies_for_concrete_pointers_of_adds_a_definition_with_codelocation_in_binary_if_data_in_readonly_memory(
self):
arch = self.ArchMock()
writable = False
loader = self.LoaderMock(
self.MainObjectMock(self.SectionMock(writable)))
memory_datum = self.MemoryDataMock(self.memory_address,
str.encode(self.string_in_memory),
len(self.string_in_memory),
'string')
cfg = self.CFGMock({self.memory_address: memory_datum})
register_definition = Definition(
Register(0, 4),
CodeLocation(0x42, 0),
)
dependency_graph = DepGraph()
dependency_graph.add_node(register_definition)
dependency_graph.add_dependencies_for_concrete_pointers_of(
[claripy.BVV(self.memory_address, arch.bits)],
register_definition,
cfg,
loader,
)
origin_codelocation = CodeLocation(0, 0, info={'readonly': True})
predecessor = list(
dependency_graph.graph.predecessors(register_definition))[0]
self.assertEqual(predecessor.codeloc, origin_codelocation)
def test_add_dependencies_for_concrete_pointers_of_adds_a_definition_for_data_pointed_to_by_given_definition(
self):
arch = self.ArchMock()
loader = self.LoaderMock(self.MainObjectMock(self.SectionMock(True)))
memory_datum = self.MemoryDataMock(self.memory_address,
str.encode(self.string_in_memory),
len(self.string_in_memory),
'string')
cfg = self.CFGMock({self.memory_address: memory_datum})
register_definition = Definition(
Register(0, 4),
None,
)
dependency_graph = DepGraph()
dependency_graph.add_node(register_definition)
dependency_graph.add_dependencies_for_concrete_pointers_of(
[claripy.BVV(self.memory_address, arch.bits)], register_definition,
cfg, loader)
memory_definition = Definition(
MemoryLocation(self.memory_address, self.string_in_memory_length),
ExternalCodeLocation(),
)
nodes = list(dependency_graph.nodes())
predecessors = list(
dependency_graph.graph.predecessors(register_definition))
self.assertEqual(nodes, [register_definition, memory_definition])
self.assertListEqual(predecessors, [memory_definition])
def test_add_dependencies_for_concrete_pointers_of_create_memory_location_with_undefined_data_if_data_pointed_to_by_definition_is_not_known(
self):
arch = self.ArchMock()
cfg = self.CFGMock({})
loader = self.LoaderMock(self.MainObjectMock(self.SectionMock(True)))
datum_content = None
datum_size = 0x4242
memory_datum = self.MemoryDataMock(self.memory_address, datum_content,
datum_size, 'unknown')
cfg = self.CFGMock({self.memory_address: memory_datum})
memory_definition = Definition(
MemoryLocation(self.memory_address, datum_size),
ExternalCodeLocation(), DataSet(UNDEFINED, datum_size * 8))
register_definition = Definition(
Register(0, 4), CodeLocation(0x42, 0),
DataSet(self.memory_address, arch.bits))
dependency_graph = DepGraph()
dependency_graph.add_node(register_definition)
dependency_graph.add_dependencies_for_concrete_pointers_of(
register_definition, cfg, loader)
nodes = list(dependency_graph.nodes())
predecessors = list(
dependency_graph.graph.predecessors(register_definition))
self.assertEqual(nodes, [register_definition, memory_definition])
self.assertListEqual(predecessors, [memory_definition])
def test_delegate_add_node_to_the_underlying_graph_object(self):
with mock.patch.object(networkx.DiGraph,
'add_node') as digraph_add_node_mock:
definition = _a_mock_definition()
dep_graph = DepGraph()
dep_graph.add_node(definition)
digraph_add_node_mock.assert_called_once_with(definition)
def test_add_dependencies_for_concrete_pointers_of_does_nothing_if_pointer_is_not_concrete(
self):
arch = self.ArchMock()
cfg = self.CFGMock({})
loader = self.LoaderMock(self.MainObjectMock(self.SectionMock(True)))
register_definition = Definition(Register(0, 4), CodeLocation(0x42, 0),
DataSet(UNDEFINED, arch.bits))
dependency_graph = DepGraph()
dependency_graph.add_node(register_definition)
nodes_before_call = dependency_graph.nodes()
dependency_graph.add_dependencies_for_concrete_pointers_of(
register_definition, cfg, loader)
self.assertEqual(nodes_before_call, dependency_graph.nodes())
