1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
#-------------------------------------------------------------------------------
# elftools tests
#
# Eli Bendersky (eliben@gmail.com)
# This code is in the public domain
#-------------------------------------------------------------------------------
import unittest
from io import BytesIO
from elftools.dwarf.lineprogram import LineProgram, LineState, LineProgramEntry
from elftools.dwarf.structs import DWARFStructs
from elftools.dwarf.constants import *
class TestLineProgram(unittest.TestCase):
def _make_program_in_stream(self, stream):
""" Create a LineProgram from the given program encoded in a stream
"""
ds = DWARFStructs(little_endian=True, dwarf_format=32, address_size=4)
header = ds.Dwarf_lineprog_header.parse(
b'\x04\x10\x00\x00' + # initial length
b'\x03\x00' + # version
b'\x20\x00\x00\x00' + # header length
b'\x01\x01\x01\x0F' + # flags
b'\x0A' + # opcode_base
b'\x00\x01\x04\x08\x0C\x01\x01\x01\x00' + # standard_opcode_lengths
# 2 dir names followed by a NULL
b'\x61\x62\x00\x70\x00\x00' +
# a file entry
b'\x61\x72\x00\x0C\x0D\x0F' +
# and another entry
b'\x45\x50\x51\x00\x86\x12\x07\x08' +
# followed by NULL
b'\x00')
lp = LineProgram(header, stream, ds, 0, len(stream.getvalue()))
return lp
def assertLineState(self, state, **kwargs):
""" Assert that the state attributes specified in kwargs have the given
values (the rest are default).
"""
for k, v in kwargs.items():
self.assertEqual(getattr(state, k), v)
def test_spec_sample_59(self):
# Sample in figure 59 of DWARFv3
s = BytesIO()
s.write(
b'\x02\xb9\x04' +
b'\x0b' +
b'\x38' +
b'\x82' +
b'\x73' +
b'\x02\x02' +
b'\x00\x01\x01')
lp = self._make_program_in_stream(s)
linetable = lp.get_entries()
self.assertEqual(len(linetable), 7)
self.assertIs(linetable[0].state, None) # doesn't modify state
self.assertEqual(linetable[0].command, DW_LNS_advance_pc)
self.assertEqual(linetable[0].args, [0x239])
self.assertLineState(linetable[1].state, address=0x239, line=3)
self.assertEqual(linetable[1].command, 0xb)
self.assertEqual(linetable[1].args, [2, 0, 0])
self.assertLineState(linetable[2].state, address=0x23c, line=5)
self.assertLineState(linetable[3].state, address=0x244, line=6)
self.assertLineState(linetable[4].state, address=0x24b, line=7, end_sequence=False)
self.assertEqual(linetable[5].command, DW_LNS_advance_pc)
self.assertEqual(linetable[5].args, [2])
self.assertLineState(linetable[6].state, address=0x24d, line=7, end_sequence=True)
def test_spec_sample_60(self):
# Sample in figure 60 of DWARFv3
s = BytesIO()
s.write(
b'\x09\x39\x02' +
b'\x0b' +
b'\x09\x03\x00' +
b'\x0b' +
b'\x09\x08\x00' +
b'\x0a' +
b'\x09\x07\x00' +
b'\x0a' +
b'\x09\x02\x00' +
b'\x00\x01\x01')
lp = self._make_program_in_stream(s)
linetable = lp.get_entries()
self.assertEqual(len(linetable), 10)
self.assertIs(linetable[0].state, None) # doesn't modify state
self.assertEqual(linetable[0].command, DW_LNS_fixed_advance_pc)
self.assertEqual(linetable[0].args, [0x239])
self.assertLineState(linetable[1].state, address=0x239, line=3)
self.assertLineState(linetable[3].state, address=0x23c, line=5)
self.assertLineState(linetable[5].state, address=0x244, line=6)
self.assertLineState(linetable[7].state, address=0x24b, line=7, end_sequence=False)
self.assertLineState(linetable[9].state, address=0x24d, line=7, end_sequence=True)
def test_lne_set_discriminator(self):
"""
Tests the handling of DWARFv4's new DW_LNE_set_discriminator opcode.
"""
s = BytesIO()
s.write(
b'\x00\x02\x04\x05' + # DW_LNE_set_discriminator (discriminator=0x05)
b'\x01' + # DW_LNS_copy
b'\x00\x01\x01' # DW_LNE_end_sequence
)
lp = self._make_program_in_stream(s)
linetable = lp.get_entries()
# We expect two entries, since DW_LNE_set_discriminator does not add
# an entry of its own.
self.assertEqual(len(linetable), 2)
self.assertEqual(linetable[0].command, DW_LNS_copy)
self.assertLineState(linetable[0].state, discriminator=0x05)
self.assertLineState(linetable[1].state, discriminator=0x00, end_sequence=True)
if __name__ == '__main__':
unittest.main()
|
