def test_proxy_blocks(self): b = gtirb.ProxyBlock() self.assertEquals(set(b.references), set()) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) m = gtirb.Module() sym1 = gtirb.Symbol("test", payload=b) sym2 = gtirb.Symbol("test", payload=123) sym3 = gtirb.Symbol("test", payload=b) m.symbols |= {sym1, sym2, sym3} b.module = m self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) i = gtirb.IR() m.ir = i e1 = gtirb.Edge(b, gtirb.ProxyBlock()) e2 = gtirb.Edge(gtirb.ProxyBlock(), b) e3 = gtirb.Edge(gtirb.ProxyBlock(), gtirb.ProxyBlock()) e4 = gtirb.Edge(b, b) i.cfg |= {e1, e2, e3, e4} self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set(b.incoming_edges), {e2, e4}) self.assertEquals(set(b.outgoing_edges), {e1, e4})
def test_cfg(self): node = gtirb.CFG() node.add( gtirb.Edge( gtirb.CodeBlock(offset=1, size=2), gtirb.CodeBlock(offset=3, size=4), gtirb.Edge.Label( type=gtirb.Edge.Type.Fallthrough, conditional=True, direct=False, ), ) ) node.add( gtirb.Edge( gtirb.CodeBlock(offset=5, size=6), gtirb.CodeBlock(offset=7, size=8), gtirb.Edge.Label( type=gtirb.Edge.Type.Branch, conditional=True, direct=False, ), ) ) string = repr(node) new_node = eval(string) self.assertTrue(node.deep_eq(new_node))
def test_proxy_blocks(self): b = gtirb.ProxyBlock() self.assertEquals(set(b.references), set()) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) m = gtirb.Module(name="M") sym1 = gtirb.Symbol("test", payload=b) sym2 = gtirb.Symbol("test", payload=123) sym3 = gtirb.Symbol("test", payload=b) m.symbols |= {sym1, sym2, sym3} b.module = m self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) i = gtirb.IR() m.ir = i p1 = gtirb.ProxyBlock() p2 = gtirb.ProxyBlock() p3 = gtirb.ProxyBlock() p4 = gtirb.ProxyBlock() i.cfg.add(gtirb.Edge(b, p1)) i.cfg.add(gtirb.Edge(p2, b)) i.cfg.add(gtirb.Edge(p3, p4)) i.cfg.add(gtirb.Edge(b, b)) self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set((s, t) for s, t, l in b.incoming_edges), {(p2, b), (b, b)}) self.assertEquals(set((s, t) for s, t, l in b.outgoing_edges), {(b, p1), (b, b)})
def test_code_blocks(self): b = gtirb.CodeBlock() self.assertEquals(b.address, None) self.assertEquals(b.contents, b"") self.assertEquals(set(b.references), set()) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) bi = gtirb.ByteInterval(address=1, contents=b"abcd1234") b.offset = 2 b.size = 3 b.byte_interval = bi self.assertEquals(b.address, 3) self.assertEquals(b.contents, b"cd1") self.assertEquals(set(b.references), set()) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) s = gtirb.Section() bi.section = s m = gtirb.Module(name="M") sym1 = gtirb.Symbol("test", payload=b) sym2 = gtirb.Symbol("test", payload=123) sym3 = gtirb.Symbol("test", payload=b) m.symbols |= {sym1, sym2, sym3} s.module = m self.assertEquals(b.address, 3) self.assertEquals(b.contents, b"cd1") self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) i = gtirb.IR() m.ir = i p1 = gtirb.ProxyBlock() p2 = gtirb.ProxyBlock() p3 = gtirb.ProxyBlock() p4 = gtirb.ProxyBlock() i.cfg.add(gtirb.Edge(b, p1)) i.cfg.add(gtirb.Edge(p2, b)) i.cfg.add(gtirb.Edge(p3, p4)) i.cfg.add(gtirb.Edge(b, b)) self.assertEquals(b.address, 3) self.assertEquals(b.contents, b"cd1") self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals( set((s, t) for s, t, l in b.incoming_edges), {(p2, b), (b, b)} ) self.assertEquals( set((s, t) for s, t, l in b.outgoing_edges), {(b, p1), (b, b)} )
def test_add(self): b1, b2 = gtirb.ProxyBlock(), gtirb.ProxyBlock() cfg = gtirb.CFG() cfg.add(gtirb.Edge(b1, b2)) cfg.add(gtirb.Edge(b1, b2)) cfg.add(gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Branch))) self.assertEqual(len(cfg), 2) self.assertTrue(gtirb.Edge(b1, b2) in cfg) self.assertTrue( gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Branch)) in cfg)
def test_in_edges(self): b1, b2, b3 = gtirb.ProxyBlock(), gtirb.ProxyBlock(), gtirb.ProxyBlock() b4 = gtirb.CodeBlock(offset=0, size=1) cfg = gtirb.CFG([ gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Fallthrough)), gtirb.Edge(b3, b2, gtirb.Edge.Label(gtirb.Edge.Type.Branch)), gtirb.Edge(b1, b3), ]) self.assertEqual(sum(1 for _ in cfg.in_edges(b1)), 0) self.assertEqual(sum(1 for _ in cfg.in_edges(b2)), 2) self.assertEqual(sum(1 for _ in cfg.in_edges(b3)), 1) self.assertEqual(sum(1 for _ in cfg.in_edges(b4)), 0)
def test_contains(self): b1, b2 = gtirb.ProxyBlock(), gtirb.ProxyBlock() cfg = gtirb.CFG( [gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Branch))]) self.assertFalse( gtirb.Edge(gtirb.ProxyBlock(), gtirb.ProxyBlock()) in cfg) self.assertFalse( gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Fallthrough)) in cfg) self.assertTrue( gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Branch)) in cfg)
def test_clear(self): cfg = gtirb.CFG([ gtirb.Edge(gtirb.ProxyBlock(), gtirb.ProxyBlock()), gtirb.Edge( gtirb.CodeBlock(offset=0, size=1), gtirb.CodeBlock(offset=1, size=2), ), ]) self.assertEqual(len(cfg), 2) cfg.clear() self.assertEqual(len(cfg), 0)
def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) ir = gtirb.IR() m = gtirb.Module( binary_path="binary_path", file_format=gtirb.Module.FileFormat.RAW, isa=gtirb.Module.ISA.ValidButUnsupported, name="name", preferred_addr=1, rebase_delta=2, ) m.ir = ir s = gtirb.Section( name="name", flags=( gtirb.Section.Flag.Executable, gtirb.Section.Flag.Readable, gtirb.Section.Flag.Loaded, gtirb.Section.Flag.Initialized, ), ) s.module = m bi = gtirb.ByteInterval(address=0, size=10, contents=b"abcd") bi.section = s cb = gtirb.CodeBlock(size=4, offset=0, decode_mode=1) cb.byte_interval = bi db = gtirb.DataBlock(size=6, offset=4) db.byte_interval = bi sym = gtirb.Symbol(name="name", payload=cb) sym.module = m sac = gtirb.SymAddrConst(0, sym, {gtirb.SymbolicExpression.Attribute.Part1}) bi.symbolic_expressions[2] = sac p = gtirb.ProxyBlock() p.module = m ir.cfg.add( gtirb.Edge( cb, p, gtirb.Edge.Label(type=gtirb.Edge.Type.Branch, conditional=False, direct=True), )) ir.cfg.add(gtirb.Edge(p, p)) m.aux_data["key"] = gtirb.AuxData(gtirb.Offset(s, 777), "Offset") ir.aux_data["key"] = gtirb.AuxData("value", "string") self.ir = ir
def test_code_blocks(self): b = gtirb.CodeBlock() self.assertEquals(b.address, None) self.assertEquals(b.contents, b"") self.assertEquals(set(b.references), set()) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) bi = gtirb.ByteInterval(address=1, contents=b"abcd1234") b.offset = 2 b.size = 3 b.byte_interval = bi self.assertEquals(b.address, 3) self.assertEquals(b.contents, b"cd1") self.assertEquals(set(b.references), set()) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) s = gtirb.Section() bi.section = s m = gtirb.Module() sym1 = gtirb.Symbol("test", payload=b) sym2 = gtirb.Symbol("test", payload=123) sym3 = gtirb.Symbol("test", payload=b) m.symbols |= {sym1, sym2, sym3} s.module = m self.assertEquals(b.address, 3) self.assertEquals(b.contents, b"cd1") self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set(b.incoming_edges), set()) self.assertEquals(set(b.outgoing_edges), set()) i = gtirb.IR() m.ir = i e1 = gtirb.Edge(b, gtirb.ProxyBlock()) e2 = gtirb.Edge(gtirb.ProxyBlock(), b) e3 = gtirb.Edge(gtirb.ProxyBlock(), gtirb.ProxyBlock()) e4 = gtirb.Edge(b, b) i.cfg |= {e1, e2, e3, e4} self.assertEquals(b.address, 3) self.assertEquals(b.contents, b"cd1") self.assertEquals(set(b.references), {sym1, sym3}) self.assertEquals(set(b.incoming_edges), {e2, e4}) self.assertEquals(set(b.outgoing_edges), {e1, e4})
def test_nx(self): b1, b2 = gtirb.ProxyBlock(), gtirb.ProxyBlock() cfg = gtirb.CFG( [gtirb.Edge(b1, b2, gtirb.Edge.Label(gtirb.Edge.Type.Call))]) for n1, n2, lab in cfg.nx().edges(data="label"): self.assertEqual(n1, b1) self.assertEqual(n2, b2) self.assertEqual(lab, gtirb.Edge.Label(gtirb.Edge.Type.Call))
def test_edge(self): node = gtirb.Edge( source=gtirb.CodeBlock(offset=1, size=2), target=gtirb.CodeBlock(offset=3, size=4), label=gtirb.Edge.Label( gtirb.Edge.Type.Fallthrough, conditional=True, direct=False ), ) string = repr(node) new_node = eval(string) self.assertEqual(node, new_node)
def test_discard(self): b1, b2 = gtirb.ProxyBlock(), gtirb.CodeBlock(offset=0, size=1) cfg = gtirb.CFG([ gtirb.Edge(b1, b2), gtirb.Edge(gtirb.ProxyBlock(), gtirb.CodeBlock(offset=1, size=2)), ]) self.assertEqual(len(cfg), 2) cfg.discard(gtirb.Edge(b1, b2)) self.assertEqual(len(cfg), 1) self.assertFalse(gtirb.Edge(b1, b2) in cfg) cfg.discard(gtirb.Edge(b1, b2)) self.assertEqual(len(cfg), 1) self.assertFalse(gtirb.Edge(b1, b2) in cfg)
def test_cfg(self): id1 = uuid.uuid4() id2 = uuid.uuid4() e1 = gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id1), gtirb.CodeBlock(size=2, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Branch, conditional=True, direct=False), ) e2 = gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id1), gtirb.CodeBlock(size=2, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Branch, conditional=True, direct=False), ) self.assertTrue(e1.deep_eq(e2)) e2 = gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id1), gtirb.CodeBlock(size=2, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Branch, conditional=True, direct=False), ) self.assertFalse(e1.deep_eq(e2)) e2 = gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id1), gtirb.CodeBlock(size=3, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Branch, conditional=True, direct=False), ) self.assertFalse(e1.deep_eq(e2)) e2 = gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id1), gtirb.CodeBlock(size=2, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Fallthrough, conditional=True, direct=False), ) self.assertFalse(e1.deep_eq(e2)) e2 = gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id1), gtirb.CodeBlock(size=2, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Branch, conditional=False, direct=False), ) self.assertFalse(e1.deep_eq(e2)) e2 = gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id1), gtirb.CodeBlock(size=2, uuid=id2), gtirb.Edge.Label(gtirb.Edge.Type.Branch, conditional=True, direct=True), ) self.assertFalse(e1.deep_eq(e2))
def test_ir(self): id1 = uuid.uuid4() id2 = uuid.uuid4() id3 = uuid.uuid4() id4 = uuid.uuid4() id5 = uuid.uuid4() ir1 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) ir2 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) self.assertTrue(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("other_value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) self.assertTrue(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=( gtirb.Module(name="m11", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) self.assertFalse(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=(gtirb.Module(name="m1", uuid=id2), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) self.assertFalse(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=55, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) self.assertFalse(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=(gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id1, ) self.assertFalse(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=5, uuid=id1, ) self.assertFalse(ir1.deep_eq(ir2)) ir2 = gtirb.IR( modules=( gtirb.Module(name="m1", uuid=id2), gtirb.Module(name="m2", uuid=id3), ), aux_data={"key": gtirb.AuxData("value", "string")}, cfg=( gtirb.Edge( gtirb.CodeBlock(size=1, uuid=id4), gtirb.CodeBlock(size=2, uuid=id4), None, ), gtirb.Edge( gtirb.CodeBlock(size=3, uuid=id4), gtirb.CodeBlock(size=4, uuid=id4), None, ), ), version=1, uuid=id5, ) self.assertFalse(ir1.deep_eq(ir2))