1 #!/usr/bin/env python 2 # Copyright 2013 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 procfs 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_file(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_file(cStringIO.StringIO(self._TEST_PROCMAPS)) 86 selected = [0, 2, 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_file(cStringIO.StringIO(self._TEST_PROCMAPS)) 93 selected = [1, 3, 6, 9] 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_file(cStringIO.StringIO(self._TEST_PROCMAPS)) 100 selected = [0, 1, 2, 3, 4, 6, 7, 9] 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() 111
