Search in sources :

Example 1 with IBlockEdge

use of com.google.security.zynamics.binnavi.disassembly.IBlockEdge in project binnavi by google.

the class BlockEdgeTest method testConstructor.

@Test
public void testConstructor() {
    final MockSqlProvider provider = new MockSqlProvider();
    final CModule internalModule = new CModule(1, "", "", new Date(), new Date(), "00000000000000000000000000000000", "0000000000000000000000000000000000000000", 0, 0, new CAddress(0), new CAddress(0), null, null, Integer.MAX_VALUE, false, provider);
    final CFunction parentFunction = new CFunction(internalModule, new MockView(), new CAddress(0x123), "Mock Function", "Mock Function", "Mock Description", 0, 0, 0, 0, FunctionType.NORMAL, "", 0, null, null, null, provider);
    final List<INaviInstruction> instructions = new ArrayList<INaviInstruction>();
    instructions.add(new MockInstruction(0x123));
    final INaviBasicBlock bblock = new CBasicBlock(1, "Hannes", instructions);
    final IBlockNode node = new CBlockNode(bblock);
    final Function pFunction = new Function(ModuleFactory.get(), parentFunction);
    final BasicBlock block1 = new BasicBlock(node, pFunction);
    final BasicBlock block2 = new BasicBlock(node, pFunction);
    final IBlockEdge mockEdge = new CFunctionEdge(node, node, com.google.security.zynamics.zylib.gui.zygraph.edges.EdgeType.JUMP_UNCONDITIONAL);
    final BlockEdge edge = new BlockEdge(mockEdge, block1, block2);
    assertEquals(block1, edge.getSource());
    assertEquals(block2, edge.getTarget());
    assertEquals(block1, block2.getParents().get(0));
    assertEquals(block2, block1.getChildren().get(0));
    assertEquals(EdgeType.JumpUnconditional, edge.getType());
    assertEquals("Block Edge [123 -> 123]", edge.toString());
}
Also used : CFunctionEdge(com.google.security.zynamics.binnavi.disassembly.CFunctionEdge) IBlockEdge(com.google.security.zynamics.binnavi.disassembly.IBlockEdge) CBlockNode(com.google.security.zynamics.binnavi.disassembly.CBlockNode) MockView(com.google.security.zynamics.binnavi.disassembly.MockView) IBlockNode(com.google.security.zynamics.binnavi.disassembly.IBlockNode) ArrayList(java.util.ArrayList) INaviBasicBlock(com.google.security.zynamics.binnavi.disassembly.INaviBasicBlock) CBasicBlock(com.google.security.zynamics.binnavi.disassembly.CBasicBlock) CFunction(com.google.security.zynamics.binnavi.disassembly.CFunction) Date(java.util.Date) CAddress(com.google.security.zynamics.zylib.disassembly.CAddress) CBasicBlock(com.google.security.zynamics.binnavi.disassembly.CBasicBlock) CFunction(com.google.security.zynamics.binnavi.disassembly.CFunction) MockInstruction(com.google.security.zynamics.binnavi.disassembly.MockInstruction) MockSqlProvider(com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider) CModule(com.google.security.zynamics.binnavi.disassembly.Modules.CModule) IBlockEdge(com.google.security.zynamics.binnavi.disassembly.IBlockEdge) INaviInstruction(com.google.security.zynamics.binnavi.disassembly.INaviInstruction) INaviBasicBlock(com.google.security.zynamics.binnavi.disassembly.INaviBasicBlock) Test(org.junit.Test)

Example 2 with IBlockEdge

use of com.google.security.zynamics.binnavi.disassembly.IBlockEdge in project binnavi by google.

the class Function method convertData.

/**
   * Converts internal function data to API function data.
   */
private void convertData() {
    final DirectedGraph<IBlockNode, IBlockEdge> graph = m_function.getGraph();
    final List<BasicBlock> blocks = new ArrayList<BasicBlock>();
    final List<BlockEdge> edges = new ArrayList<BlockEdge>();
    final HashMap<IBlockNode, BasicBlock> blockMap = new HashMap<IBlockNode, BasicBlock>();
    for (final IBlockNode block : graph.getNodes()) {
        final BasicBlock newBlock = new BasicBlock(block, this);
        blockMap.put(block, newBlock);
        blocks.add(newBlock);
    }
    for (final IBlockEdge edge : graph.getEdges()) {
        final BasicBlock source = blockMap.get(edge.getSource());
        final BasicBlock target = blockMap.get(edge.getTarget());
        edges.add(new BlockEdge(edge, source, target));
    }
    m_graph = new FlowGraph(blocks, edges);
}
Also used : IBlockEdge(com.google.security.zynamics.binnavi.disassembly.IBlockEdge) HashMap(java.util.HashMap) IBlockNode(com.google.security.zynamics.binnavi.disassembly.IBlockNode) ArrayList(java.util.ArrayList) IBlockEdge(com.google.security.zynamics.binnavi.disassembly.IBlockEdge)

Aggregations

IBlockEdge (com.google.security.zynamics.binnavi.disassembly.IBlockEdge)2 IBlockNode (com.google.security.zynamics.binnavi.disassembly.IBlockNode)2 ArrayList (java.util.ArrayList)2 MockSqlProvider (com.google.security.zynamics.binnavi.Database.MockClasses.MockSqlProvider)1 CBasicBlock (com.google.security.zynamics.binnavi.disassembly.CBasicBlock)1 CBlockNode (com.google.security.zynamics.binnavi.disassembly.CBlockNode)1 CFunction (com.google.security.zynamics.binnavi.disassembly.CFunction)1 CFunctionEdge (com.google.security.zynamics.binnavi.disassembly.CFunctionEdge)1 INaviBasicBlock (com.google.security.zynamics.binnavi.disassembly.INaviBasicBlock)1 INaviInstruction (com.google.security.zynamics.binnavi.disassembly.INaviInstruction)1 MockInstruction (com.google.security.zynamics.binnavi.disassembly.MockInstruction)1 MockView (com.google.security.zynamics.binnavi.disassembly.MockView)1 CModule (com.google.security.zynamics.binnavi.disassembly.Modules.CModule)1 CAddress (com.google.security.zynamics.zylib.disassembly.CAddress)1 Date (java.util.Date)1 HashMap (java.util.HashMap)1 Test (org.junit.Test)1