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))
