pykd

1
import unittest
2
import pykd
3
import sys
4
import os
5

6
class ARMDumpTest(unittest.TestCase):
7
    def setUp(self):
8
        dump_file = os.path.join( os.path.dirname(sys.argv[0]),
9
                                  r"..\..\kdlibcpp\kdlib\tests\dumps\win10_arm_rpi3_mem\win10_arm_rpi3_mem.cab" )
10
        self.dump_id = pykd.loadDump( dump_file )
11

12
    def tearDown(self):
13
        pykd.closeDump( self.dump_id )
14

15
    def testArm(self):
16
        # dbgmem
17
        self.assertEqual( 0, pykd.addr64(0) )
18
        self.assertEqual( 0xffffffff80000000, pykd.addr64(0x80000000) )
19

20
        # dbgeng
21
        self.assertEqual( pykd.CPUType.ARM, pykd.getCPUType() )
22
        self.assertEqual( pykd.CPUType.ARM, pykd.getCPUMode() )
23

24
        self.assertEqual( False, pykd.is64bitSystem() )
25
        self.assertEqual( 4, pykd.ptrSize() )
26

27
        self.assertEqual( pykd.addr64(pykd.expr("@$retreg")), pykd.addr64(pykd.reg("r0")) )
28

29
        # typeinfo
30
        self.assertEqual( 4, pykd.typeInfo("nt!_IRP*").size() )
31

32
        # cpu context
33
        self.assertEqual( pykd.expr("@$csp"), pykd.cpu().sp )
34

35
        # stack
36
        stack = pykd.getStack()
37

38
        self.assertTrue( len(stack) > 1 )
39

40
        self.assertEqual( "Wdf01000!FxRequest::CompleteInternal", pykd.findSymbol(stack[0].ip, False) )
41

42
        self.assertEqual( "Wdf01000!imp_WdfRequestCompleteWithInformation", pykd.findSymbol(stack[1].ip, False) )
43
        self.assertEqual( 0x102, stack[1].getParam("RequestStatus") )
44