OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. |
| 5 |
| 6 import cStringIO |
| 7 import logging |
| 8 import os |
| 9 import sys |
| 10 import unittest |
| 11 |
| 12 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
| 13 sys.path.insert(0, ROOT_DIR) |
| 14 |
| 15 from proc_maps import ProcMaps |
| 16 |
| 17 |
| 18 class ProcMapsTest(unittest.TestCase): |
| 19 _TEST_PROCMAPS = '\n'.join([ |
| 20 '00000000-00001000 r--p 00000000 fc:00 0', |
| 21 '0080b000-0080c000 r-xp 0020b000 fc:00 2231329' |
| 22 ' /usr/bin/some', |
| 23 '0080c000-0080f000 ---p 0020c000 fc:00 2231329' |
| 24 ' /usr/bin/some', |
| 25 '0100a000-0100c000 r-xp 0120a000 fc:00 22381' |
| 26 ' /usr/bin/chrome', |
| 27 '0100c000-0100f000 ---p 0120c000 fc:00 22381' |
| 28 ' /usr/bin/chrome', |
| 29 '0237d000-02a9b000 rw-p 00000000 00:00 0' |
| 30 ' [heap]', |
| 31 '7fb920e6d000-7fb920e85000 r-xp 00000000 fc:00 263482' |
| 32 ' /lib/x86_64-linux-gnu/libpthread-2.15.so', |
| 33 '7fb920e85000-7fb921084000 ---p 00018000 fc:00 263482' |
| 34 ' /lib/x86_64-linux-gnu/libpthread-2.15.so', |
| 35 '7fb9225f4000-7fb922654000 rw-s 00000000 00:04 19660808' |
| 36 ' /SYSV00000000 (deleted)', |
| 37 'ffffffffff600000-ffffffffff601000 r-xp 00000000 00:00 0' |
| 38 ' [vsyscall]', |
| 39 ]) |
| 40 |
| 41 _EXPECTED = [ |
| 42 (0x0, 0x1000, 'r', '-', '-', 'p', 0x0, 'fc', '00', 0, ''), |
| 43 (0x80b000, 0x80c000, 'r', '-', 'x', 'p', 0x20b000, |
| 44 'fc', '00', 2231329, '/usr/bin/some'), |
| 45 (0x80c000, 0x80f000, '-', '-', '-', 'p', 0x20c000, |
| 46 'fc', '00', 2231329, '/usr/bin/some'), |
| 47 (0x100a000, 0x100c000, 'r', '-', 'x', 'p', 0x120a000, |
| 48 'fc', '00', 22381, '/usr/bin/chrome'), |
| 49 (0x100c000, 0x100f000, '-', '-', '-', 'p', 0x120c000, |
| 50 'fc', '00', 22381, '/usr/bin/chrome'), |
| 51 (0x237d000, 0x2a9b000, 'r', 'w', '-', 'p', 0x0, |
| 52 '00', '00', 0, '[heap]'), |
| 53 (0x7fb920e6d000, 0x7fb920e85000, 'r', '-', 'x', 'p', 0x0, |
| 54 'fc', '00', 263482, '/lib/x86_64-linux-gnu/libpthread-2.15.so'), |
| 55 (0x7fb920e85000, 0x7fb921084000, '-', '-', '-', 'p', 0x18000, |
| 56 'fc', '00', 263482, '/lib/x86_64-linux-gnu/libpthread-2.15.so'), |
| 57 (0x7fb9225f4000, 0x7fb922654000, 'r', 'w', '-', 's', 0x0, |
| 58 '00', '04', 19660808, '/SYSV00000000 (deleted)'), |
| 59 (0xffffffffff600000, 0xffffffffff601000, 'r', '-', 'x', 'p', 0x0, |
| 60 '00', '00', 0, '[vsyscall]'), |
| 61 ] |
| 62 |
| 63 @staticmethod |
| 64 def _expected_as_dict(index): |
| 65 return { |
| 66 'begin': ProcMapsTest._EXPECTED[index][0], |
| 67 'end': ProcMapsTest._EXPECTED[index][1], |
| 68 'readable': ProcMapsTest._EXPECTED[index][2], |
| 69 'writable': ProcMapsTest._EXPECTED[index][3], |
| 70 'executable': ProcMapsTest._EXPECTED[index][4], |
| 71 'private': ProcMapsTest._EXPECTED[index][5], |
| 72 'offset': ProcMapsTest._EXPECTED[index][6], |
| 73 'major': ProcMapsTest._EXPECTED[index][7], |
| 74 'minor': ProcMapsTest._EXPECTED[index][8], |
| 75 'inode': ProcMapsTest._EXPECTED[index][9], |
| 76 'name': ProcMapsTest._EXPECTED[index][10], |
| 77 } |
| 78 |
| 79 def test_load(self): |
| 80 maps = ProcMaps.load(cStringIO.StringIO(self._TEST_PROCMAPS)) |
| 81 for index, entry in enumerate(maps): |
| 82 self.assertEqual(entry.as_dict(), self._expected_as_dict(index)) |
| 83 |
| 84 def test_constants(self): |
| 85 maps = ProcMaps.load(cStringIO.StringIO(self._TEST_PROCMAPS)) |
| 86 selected = [4, 7] |
| 87 for index, entry in enumerate(maps.iter(ProcMaps.constants)): |
| 88 self.assertEqual(entry.as_dict(), |
| 89 self._expected_as_dict(selected[index])) |
| 90 |
| 91 def test_executable(self): |
| 92 maps = ProcMaps.load(cStringIO.StringIO(self._TEST_PROCMAPS)) |
| 93 selected = [3, 6] |
| 94 for index, entry in enumerate(maps.iter(ProcMaps.executable)): |
| 95 self.assertEqual(entry.as_dict(), |
| 96 self._expected_as_dict(selected[index])) |
| 97 |
| 98 def test_executable_and_constants(self): |
| 99 maps = ProcMaps.load(cStringIO.StringIO(self._TEST_PROCMAPS)) |
| 100 selected = [3, 4, 6, 7] |
| 101 for index, entry in enumerate(maps.iter(ProcMaps.executable_and_constants)): |
| 102 self.assertEqual(entry.as_dict(), |
| 103 self._expected_as_dict(selected[index])) |
| 104 |
| 105 |
| 106 if __name__ == '__main__': |
| 107 logging.basicConfig( |
| 108 level=logging.DEBUG if '-v' in sys.argv else logging.ERROR, |
| 109 format='%(levelname)5s %(filename)15s(%(lineno)3d): %(message)s') |
| 110 unittest.main() |
OLD | NEW |