| Eli Bendersky | 62ae328 | 2011-12-24 06:55:59 +0200 | [diff] [blame] | 1 | try: |
| 2 | import unittest2 as unittest |
| 3 | except ImportError: |
| 4 | import unittest |
| 5 | import sys |
| Eli Bendersky | ebe5116 | 2011-10-27 17:34:02 +0200 | [diff] [blame] | 6 | |
| 7 | sys.path.extend(('..', '.')) |
| Eli Bendersky | 3501f9f | 2011-12-09 12:29:06 +0200 | [diff] [blame] | 8 | from elftools.dwarf.descriptions import ExprDumper, set_global_machine_arch |
| Eli Bendersky | ebe5116 | 2011-10-27 17:34:02 +0200 | [diff] [blame] | 9 | from elftools.dwarf.structs import DWARFStructs |
| 10 | |
| 11 | |
| Eli Bendersky | 6c17c05 | 2011-12-07 14:43:47 +0200 | [diff] [blame] | 12 | class TestExprDumper(unittest.TestCase): |
| Eli Bendersky | ebe5116 | 2011-10-27 17:34:02 +0200 | [diff] [blame] | 13 | structs32 = DWARFStructs( |
| 14 | little_endian=True, |
| 15 | dwarf_format=32, |
| 16 | address_size=4) |
| 17 | |
| eliben | 84e27b4 | 2011-11-06 15:27:57 +0200 | [diff] [blame] | 18 | def setUp(self): |
| Eli Bendersky | 6c17c05 | 2011-12-07 14:43:47 +0200 | [diff] [blame] | 19 | self.visitor = ExprDumper(self.structs32) |
| Eli Bendersky | 3501f9f | 2011-12-09 12:29:06 +0200 | [diff] [blame] | 20 | set_global_machine_arch('x64') |
| eliben | 84e27b4 | 2011-11-06 15:27:57 +0200 | [diff] [blame] | 21 | |
| 22 | def test_basic_single(self): |
| 23 | self.visitor.process_expr([0x1b]) |
| 24 | self.assertEqual(self.visitor.get_str(), |
| 25 | 'DW_OP_div') |
| 26 | |
| 27 | self.setUp() |
| 28 | self.visitor.process_expr([0x74, 0x82, 0x01]) |
| 29 | self.assertEqual(self.visitor.get_str(), |
| Eli Bendersky | 3501f9f | 2011-12-09 12:29:06 +0200 | [diff] [blame] | 30 | 'DW_OP_breg4 (rsi): 130') |
| eliben | 4a4586a | 2011-11-06 15:49:37 +0200 | [diff] [blame] | 31 | |
| 32 | self.setUp() |
| 33 | self.visitor.process_expr([0x91, 0x82, 0x01]) |
| 34 | self.assertEqual(self.visitor.get_str(), |
| 35 | 'DW_OP_fbreg: 130') |
| eliben | 84e27b4 | 2011-11-06 15:27:57 +0200 | [diff] [blame] | 36 | |
| 37 | self.setUp() |
| Eli Bendersky | 3501f9f | 2011-12-09 12:29:06 +0200 | [diff] [blame] | 38 | self.visitor.process_expr([0x51]) |
| 39 | self.assertEqual(self.visitor.get_str(), |
| 40 | 'DW_OP_reg1 (rdx)') |
| 41 | |
| 42 | self.setUp() |
| 43 | self.visitor.process_expr([0x90, 16]) |
| 44 | self.assertEqual(self.visitor.get_str(), |
| 45 | 'DW_OP_regx: 16 (rip)') |
| 46 | |
| 47 | self.setUp() |
| eliben | 84e27b4 | 2011-11-06 15:27:57 +0200 | [diff] [blame] | 48 | self.visitor.process_expr([0x9d, 0x8f, 0x0A, 0x90, 0x01]) |
| 49 | self.assertEqual(self.visitor.get_str(), |
| 50 | 'DW_OP_bit_piece: 1295 144') |
| 51 | |
| 52 | def test_basic_sequence(self): |
| 53 | self.visitor.process_expr([0x03, 0x01, 0x02, 0, 0, 0x06, 0x06]) |
| 54 | self.assertEqual(self.visitor.get_str(), |
| 55 | 'DW_OP_addr: 201; DW_OP_deref; DW_OP_deref') |
| 56 | |
| 57 | self.setUp() |
| 58 | self.visitor.process_expr([0x15, 0xFF, 0x0b, 0xf1, 0xff]) |
| 59 | self.assertEqual(self.visitor.get_str(), |
| 60 | 'DW_OP_pick: 255; DW_OP_const2s: -15') |
| 61 | |
| 62 | self.setUp() |
| 63 | self.visitor.process_expr([0x1d, 0x1e, 0x1d, 0x1e, 0x1d, 0x1e]) |
| 64 | self.assertEqual(self.visitor.get_str(), |
| 65 | 'DW_OP_mod; DW_OP_mul; DW_OP_mod; DW_OP_mul; DW_OP_mod; DW_OP_mul') |
| 66 | |
| Eli Bendersky | ebe5116 | 2011-10-27 17:34:02 +0200 | [diff] [blame] | 67 | |
| 68 | if __name__ == '__main__': |
| 69 | unittest.main() |
| 70 | |
| eliben | ba447bc | 2011-11-06 11:29:49 +0200 | [diff] [blame] | 71 | |