xref: /xnu-11215.41.3/tools/lldbmacros/core/operating_system.py (revision 33de042d024d46de5ff4e89f2471de6608e37fa4)
1*33de042dSApple OSS Distributions#!/usr/bin/python
2*33de042dSApple OSS Distributions#
3*33de042dSApple OSS Distributions
4*33de042dSApple OSS Distributions#source of register info is from http://opensource.apple.com/source/gdb/gdb-962/src/gdb/arm-tdep.c
5*33de042dSApple OSS Distributionsimport struct
6*33de042dSApple OSS Distributionsimport lldb
7*33de042dSApple OSS Distributions
8*33de042dSApple OSS Distributions
9*33de042dSApple OSS Distributionsosplugin_target_obj = None
10*33de042dSApple OSS Distributions
11*33de042dSApple OSS Distributionsclass PluginValue(lldb.SBValue):
12*33de042dSApple OSS Distributions    def GetChildMemberWithName(val, name):
13*33de042dSApple OSS Distributions        val_type = val.GetType()
14*33de042dSApple OSS Distributions        if val_type.IsPointerType():
15*33de042dSApple OSS Distributions            val_type = val_type.GetPointeeType()
16*33de042dSApple OSS Distributions        for i in range(val_type.GetNumberOfFields()):
17*33de042dSApple OSS Distributions            if name == val_type.GetFieldAtIndex(i).GetName():
18*33de042dSApple OSS Distributions                return PluginValue(val.GetChildAtIndex(i))
19*33de042dSApple OSS Distributions        return None
20*33de042dSApple OSS Distributions
21*33de042dSApple OSS Distributionsclass Armv8_RegisterSet(object):
22*33de042dSApple OSS Distributions    """ register info set for armv8 64 bit architecture"""
23*33de042dSApple OSS Distributions    register_info = { 'sets' : ['GPR'],
24*33de042dSApple OSS Distributions                  'registers': [
25*33de042dSApple OSS Distributions    {'name': 'x0'  , 'bitsize':64, 'offset':  0, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 0, 'dwarf': 0, 'alt-name':'arg1', 'generic':'arg1'},
26*33de042dSApple OSS Distributions    {'name': 'x1'  , 'bitsize':64, 'offset':  8, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 1, 'dwarf': 1, 'alt-name':'arg2', 'generic':'arg2'},
27*33de042dSApple OSS Distributions    {'name': 'x2'  , 'bitsize':64, 'offset': 16, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 2, 'dwarf': 2, 'alt-name':'arg3', 'generic':'arg3'},
28*33de042dSApple OSS Distributions    {'name': 'x3'  , 'bitsize':64, 'offset': 24, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 3, 'dwarf': 3, 'alt-name':'arg4', 'generic':'arg4'},
29*33de042dSApple OSS Distributions    {'name': 'x4'  , 'bitsize':64, 'offset': 32, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 4, 'dwarf': 4, 'alt-name':'arg5', 'generic':'arg5'},
30*33de042dSApple OSS Distributions    {'name': 'x5'  , 'bitsize':64, 'offset': 40, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 5, 'dwarf': 5, 'alt-name':'arg6', 'generic':'arg6'},
31*33de042dSApple OSS Distributions    {'name': 'x6'  , 'bitsize':64, 'offset': 48, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 6, 'dwarf': 6, 'alt-name':'arg7', 'generic':'arg7'},
32*33de042dSApple OSS Distributions    {'name': 'x7'  , 'bitsize':64, 'offset': 56, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 7, 'dwarf': 7, 'alt-name':'arg8', 'generic':'arg8'},
33*33de042dSApple OSS Distributions    {'name': 'x8'  , 'bitsize':64, 'offset': 64, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 8, 'dwarf': 8},
34*33de042dSApple OSS Distributions    {'name': 'x9'  , 'bitsize':64, 'offset': 72, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 9, 'dwarf': 9},
35*33de042dSApple OSS Distributions    {'name': 'x10' , 'bitsize':64, 'offset': 80, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':10, 'dwarf':10},
36*33de042dSApple OSS Distributions    {'name': 'x11' , 'bitsize':64, 'offset': 88, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':11, 'dwarf':11},
37*33de042dSApple OSS Distributions    {'name': 'x12' , 'bitsize':64, 'offset': 96, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':12, 'dwarf':12},
38*33de042dSApple OSS Distributions    {'name': 'x13' , 'bitsize':64, 'offset':104, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':13, 'dwarf':13},
39*33de042dSApple OSS Distributions    {'name': 'x14' , 'bitsize':64, 'offset':112, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':14, 'dwarf':14},
40*33de042dSApple OSS Distributions    {'name': 'x15' , 'bitsize':64, 'offset':120, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':15, 'dwarf':15},
41*33de042dSApple OSS Distributions    {'name': 'x16' , 'bitsize':64, 'offset':128, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':16, 'dwarf':16},
42*33de042dSApple OSS Distributions    {'name': 'x17' , 'bitsize':64, 'offset':136, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':17, 'dwarf':17},
43*33de042dSApple OSS Distributions    {'name': 'x18' , 'bitsize':64, 'offset':144, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':18, 'dwarf':18},
44*33de042dSApple OSS Distributions    {'name': 'x19' , 'bitsize':64, 'offset':152, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':19, 'dwarf':19},
45*33de042dSApple OSS Distributions    {'name': 'x20' , 'bitsize':64, 'offset':160, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':20, 'dwarf':20},
46*33de042dSApple OSS Distributions    {'name': 'x21' , 'bitsize':64, 'offset':168, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':21, 'dwarf':21},
47*33de042dSApple OSS Distributions    {'name': 'x22' , 'bitsize':64, 'offset':176, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':22, 'dwarf':22},
48*33de042dSApple OSS Distributions    {'name': 'x23' , 'bitsize':64, 'offset':184, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':23, 'dwarf':23},
49*33de042dSApple OSS Distributions    {'name': 'x24' , 'bitsize':64, 'offset':192, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':24, 'dwarf':24},
50*33de042dSApple OSS Distributions    {'name': 'x25' , 'bitsize':64, 'offset':200, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':25, 'dwarf':25},
51*33de042dSApple OSS Distributions    {'name': 'x26' , 'bitsize':64, 'offset':208, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':26, 'dwarf':26},
52*33de042dSApple OSS Distributions    {'name': 'x27' , 'bitsize':64, 'offset':216, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':27, 'dwarf':27},
53*33de042dSApple OSS Distributions    {'name': 'x28' , 'bitsize':64, 'offset':224, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':28, 'dwarf':28},
54*33de042dSApple OSS Distributions    {'name': 'fp'  , 'bitsize':64, 'offset':232, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':29, 'dwarf':29, 'alt-name': 'fp', 'generic':'fp'},
55*33de042dSApple OSS Distributions    {'name': 'lr'  , 'bitsize':64, 'offset':240, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':30, 'dwarf':30, 'alt-name': 'lr', 'generic':'lr'},
56*33de042dSApple OSS Distributions    {'name': 'sp'  , 'bitsize':64, 'offset':248, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':31, 'dwarf':31, 'alt-name': 'sp', 'generic':'sp'},
57*33de042dSApple OSS Distributions    {'name': 'pc'  , 'bitsize':64, 'offset':256, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':32, 'dwarf':32, 'alt-name': 'pc', 'generic':'pc'},
58*33de042dSApple OSS Distributions    {'name': 'far' , 'bitsize':64, 'offset':264, 'encoding':'uint', 'format':'hex', 'set':0},
59*33de042dSApple OSS Distributions    {'name': 'cpsr', 'bitsize':32, 'offset':272, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':33, 'dwarf':33, 'generic':'flags'},
60*33de042dSApple OSS Distributions    {'name': 'esr' , 'bitsize':32, 'offset':276, 'encoding':'uint', 'format':'hex', 'set':0},
61*33de042dSApple OSS Distributions    ]
62*33de042dSApple OSS Distributions    }
63*33de042dSApple OSS Distributions
64*33de042dSApple OSS Distributions    def __init__(self):
65*33de042dSApple OSS Distributions        self.switch_context_address = osplugin_target_obj.FindSymbols('Switch_context')[0].GetSymbol().GetStartAddress().GetLoadAddress(osplugin_target_obj)
66*33de042dSApple OSS Distributions        self.ResetRegisterValues()
67*33de042dSApple OSS Distributions    def ResetRegisterValues(self):
68*33de042dSApple OSS Distributions        self.x0 = 0
69*33de042dSApple OSS Distributions        self.x1 = 0
70*33de042dSApple OSS Distributions        self.x2 = 0
71*33de042dSApple OSS Distributions        self.x3 = 0
72*33de042dSApple OSS Distributions        self.x4 = 0
73*33de042dSApple OSS Distributions        self.x5 = 0
74*33de042dSApple OSS Distributions        self.x6 = 0
75*33de042dSApple OSS Distributions        self.x7 = 0
76*33de042dSApple OSS Distributions        self.x8 = 0
77*33de042dSApple OSS Distributions        self.x9 = 0
78*33de042dSApple OSS Distributions        self.x10 = 0
79*33de042dSApple OSS Distributions        self.x11 = 0
80*33de042dSApple OSS Distributions        self.x12 = 0
81*33de042dSApple OSS Distributions        self.x13 = 0
82*33de042dSApple OSS Distributions        self.x14 = 0
83*33de042dSApple OSS Distributions        self.x15 = 0
84*33de042dSApple OSS Distributions        self.x16 = 0
85*33de042dSApple OSS Distributions        self.x17 = 0
86*33de042dSApple OSS Distributions        self.x18 = 0
87*33de042dSApple OSS Distributions        self.x19 = 0
88*33de042dSApple OSS Distributions        self.x20 = 0
89*33de042dSApple OSS Distributions        self.x21 = 0
90*33de042dSApple OSS Distributions        self.x22 = 0
91*33de042dSApple OSS Distributions        self.x23 = 0
92*33de042dSApple OSS Distributions        self.x24 = 0
93*33de042dSApple OSS Distributions        self.x25 = 0
94*33de042dSApple OSS Distributions        self.x26 = 0
95*33de042dSApple OSS Distributions        self.x27 = 0
96*33de042dSApple OSS Distributions        self.x28 = 0
97*33de042dSApple OSS Distributions        self.fp = 0
98*33de042dSApple OSS Distributions        self.lr = 0
99*33de042dSApple OSS Distributions        self.sp = 0
100*33de042dSApple OSS Distributions        self.pc = 0
101*33de042dSApple OSS Distributions        self.far = 0
102*33de042dSApple OSS Distributions        self.cpsr = 0
103*33de042dSApple OSS Distributions        self.esr = 0
104*33de042dSApple OSS Distributions
105*33de042dSApple OSS Distributions    def __str__(self):
106*33de042dSApple OSS Distributions        return """ pc = """
107*33de042dSApple OSS Distributions
108*33de042dSApple OSS Distributions    def GetPackedRegisterState(self):
109*33de042dSApple OSS Distributions        return struct.pack('34QII', self.x0, self.x1, self.x2, self.x3, self.x4, self.x5,
110*33de042dSApple OSS Distributions            self.x6, self.x7, self.x8, self.x9, self.x10, self.x11, self.x12, self.x13,
111*33de042dSApple OSS Distributions            self.x14, self.x15, self.x16, self.x17, self.x18, self.x19, self.x20, self.x21,
112*33de042dSApple OSS Distributions            self.x22, self.x23, self.x24, self.x25, self.x26, self.x27, self.x28, self.fp,
113*33de042dSApple OSS Distributions            self.lr, self.sp, self.pc, self.far, self.cpsr, self.esr)
114*33de042dSApple OSS Distributions
115*33de042dSApple OSS Distributions    def ReadRegisterDataFromKDPSavedState(self, kdp_state, kernel_version):
116*33de042dSApple OSS Distributions        """ Setup register values from KDP saved information.
117*33de042dSApple OSS Distributions        """
118*33de042dSApple OSS Distributions        saved_state = kernel_version.CreateValueFromExpression(None, '(struct arm_saved_state64 *) ' + str(kdp_state.GetValueAsUnsigned()))
119*33de042dSApple OSS Distributions        saved_state = saved_state.Dereference()
120*33de042dSApple OSS Distributions        saved_state = PluginValue(saved_state)
121*33de042dSApple OSS Distributions        self.ResetRegisterValues()
122*33de042dSApple OSS Distributions        self.x0 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(0).GetValueAsUnsigned()
123*33de042dSApple OSS Distributions        self.x1 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(1).GetValueAsUnsigned()
124*33de042dSApple OSS Distributions        self.x2 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(2).GetValueAsUnsigned()
125*33de042dSApple OSS Distributions        self.x3 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(3).GetValueAsUnsigned()
126*33de042dSApple OSS Distributions        self.x4 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(4).GetValueAsUnsigned()
127*33de042dSApple OSS Distributions        self.x5 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(5).GetValueAsUnsigned()
128*33de042dSApple OSS Distributions        self.x6 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(6).GetValueAsUnsigned()
129*33de042dSApple OSS Distributions        self.x7 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(7).GetValueAsUnsigned()
130*33de042dSApple OSS Distributions        self.x8 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(8).GetValueAsUnsigned()
131*33de042dSApple OSS Distributions        self.x9 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(9).GetValueAsUnsigned()
132*33de042dSApple OSS Distributions        self.x10 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(10).GetValueAsUnsigned()
133*33de042dSApple OSS Distributions        self.x11 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(11).GetValueAsUnsigned()
134*33de042dSApple OSS Distributions        self.x12 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(12).GetValueAsUnsigned()
135*33de042dSApple OSS Distributions        self.x13 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(13).GetValueAsUnsigned()
136*33de042dSApple OSS Distributions        self.x14 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(14).GetValueAsUnsigned()
137*33de042dSApple OSS Distributions        self.x15 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(15).GetValueAsUnsigned()
138*33de042dSApple OSS Distributions        self.x16 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(16).GetValueAsUnsigned()
139*33de042dSApple OSS Distributions        self.x17 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(17).GetValueAsUnsigned()
140*33de042dSApple OSS Distributions        self.x18 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(18).GetValueAsUnsigned()
141*33de042dSApple OSS Distributions        self.x19 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(19).GetValueAsUnsigned()
142*33de042dSApple OSS Distributions        self.x20 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(20).GetValueAsUnsigned()
143*33de042dSApple OSS Distributions        self.x21 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(21).GetValueAsUnsigned()
144*33de042dSApple OSS Distributions        self.x22 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(22).GetValueAsUnsigned()
145*33de042dSApple OSS Distributions        self.x23 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(23).GetValueAsUnsigned()
146*33de042dSApple OSS Distributions        self.x24 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(24).GetValueAsUnsigned()
147*33de042dSApple OSS Distributions        self.x25 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(25).GetValueAsUnsigned()
148*33de042dSApple OSS Distributions        self.x26 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(26).GetValueAsUnsigned()
149*33de042dSApple OSS Distributions        self.x27 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(27).GetValueAsUnsigned()
150*33de042dSApple OSS Distributions        self.x28 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(28).GetValueAsUnsigned()
151*33de042dSApple OSS Distributions        self.fp = saved_state.GetChildMemberWithName('fp').GetValueAsUnsigned()
152*33de042dSApple OSS Distributions        self.lr = saved_state.GetChildMemberWithName('lr').GetValueAsUnsigned()
153*33de042dSApple OSS Distributions        self.sp = saved_state.GetChildMemberWithName('sp').GetValueAsUnsigned()
154*33de042dSApple OSS Distributions        self.pc = saved_state.GetChildMemberWithName('pc').GetValueAsUnsigned()
155*33de042dSApple OSS Distributions        self.far = saved_state.GetChildMemberWithName('far').GetValueAsUnsigned()
156*33de042dSApple OSS Distributions        self.cpsr = saved_state.GetChildMemberWithName('cpsr').GetValueAsUnsigned()
157*33de042dSApple OSS Distributions        self.esr = saved_state.GetChildMemberWithName('esr').GetValueAsUnsigned()
158*33de042dSApple OSS Distributions        return self
159*33de042dSApple OSS Distributions
160*33de042dSApple OSS Distributions    def ReadRegisterDataFromKernelStack(self, kstack_saved_state_addr, kernel_version):
161*33de042dSApple OSS Distributions        saved_state = kernel_version.CreateValueFromExpression(None, '(arm_kernel_saved_state_t *) '+ str(kstack_saved_state_addr))
162*33de042dSApple OSS Distributions        saved_state = saved_state.Dereference()
163*33de042dSApple OSS Distributions        saved_state = PluginValue(saved_state)
164*33de042dSApple OSS Distributions        self.ResetRegisterValues()
165*33de042dSApple OSS Distributions        self.x19 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(0).GetValueAsUnsigned()
166*33de042dSApple OSS Distributions        self.x20 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(1).GetValueAsUnsigned()
167*33de042dSApple OSS Distributions        self.x21 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(2).GetValueAsUnsigned()
168*33de042dSApple OSS Distributions        self.x22 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(3).GetValueAsUnsigned()
169*33de042dSApple OSS Distributions        self.x23 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(4).GetValueAsUnsigned()
170*33de042dSApple OSS Distributions        self.x24 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(5).GetValueAsUnsigned()
171*33de042dSApple OSS Distributions        self.x25 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(6).GetValueAsUnsigned()
172*33de042dSApple OSS Distributions        self.x26 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(7).GetValueAsUnsigned()
173*33de042dSApple OSS Distributions        self.x27 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(8).GetValueAsUnsigned()
174*33de042dSApple OSS Distributions        self.x28 = saved_state.GetChildMemberWithName('x').GetChildAtIndex(9).GetValueAsUnsigned()
175*33de042dSApple OSS Distributions        self.fp = saved_state.GetChildMemberWithName('fp').GetValueAsUnsigned()
176*33de042dSApple OSS Distributions        self.lr = saved_state.GetChildMemberWithName('lr').GetValueAsUnsigned()
177*33de042dSApple OSS Distributions        self.sp = saved_state.GetChildMemberWithName('sp').GetValueAsUnsigned()
178*33de042dSApple OSS Distributions        # pc for a blocked thread is treated to be the next instruction it would run after thread switch.
179*33de042dSApple OSS Distributions        self.pc = self.switch_context_address
180*33de042dSApple OSS Distributions        return self
181*33de042dSApple OSS Distributions
182*33de042dSApple OSS Distributions    def ReadRegisterDataFromContinuation(self, continuation_ptr):
183*33de042dSApple OSS Distributions        self.ResetRegisterValues()
184*33de042dSApple OSS Distributions        self.pc = continuation_ptr
185*33de042dSApple OSS Distributions        return self
186*33de042dSApple OSS Distributions
187*33de042dSApple OSS Distributions    @classmethod
188*33de042dSApple OSS Distributions    def GetRegisterInfo(cls, regnum):
189*33de042dSApple OSS Distributions        if regnum < 0 or regnum > len(cls.register_info['registers']):
190*33de042dSApple OSS Distributions            return ''
191*33de042dSApple OSS Distributions
192*33de042dSApple OSS Distributions        reginfo = cls.register_info['registers'][regnum]
193*33de042dSApple OSS Distributions        retval = ''
194*33de042dSApple OSS Distributions        for i in list(reginfo.keys()):
195*33de042dSApple OSS Distributions            v_str = str(reginfo[i])
196*33de042dSApple OSS Distributions            if i == 'set':
197*33de042dSApple OSS Distributions                v_str = 'General Purpose Registers'
198*33de042dSApple OSS Distributions            retval += "%s:%s;" % (str(i), v_str)
199*33de042dSApple OSS Distributions        return retval
200*33de042dSApple OSS Distributions
201*33de042dSApple OSS Distributions
202*33de042dSApple OSS Distributions
203*33de042dSApple OSS Distributionsclass Armv7_RegisterSet(object):
204*33de042dSApple OSS Distributions    """ register info set for armv7 32 bit architecture """
205*33de042dSApple OSS Distributions    register_info = { 'sets' : ['GPR'],
206*33de042dSApple OSS Distributions                  'registers': [
207*33de042dSApple OSS Distributions        { 'name':'r0'   , 'bitsize' : 32, 'offset' :  0, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 0, 'dwarf' : 0},
208*33de042dSApple OSS Distributions        { 'name':'r1'   , 'bitsize' : 32, 'offset' :  4, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 1, 'dwarf' : 1},
209*33de042dSApple OSS Distributions        { 'name':'r2'   , 'bitsize' : 32, 'offset' :  8, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 2, 'dwarf' : 2},
210*33de042dSApple OSS Distributions        { 'name':'r3'   , 'bitsize' : 32, 'offset' : 12, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 3, 'dwarf' : 3},
211*33de042dSApple OSS Distributions        { 'name':'r4'   , 'bitsize' : 32, 'offset' : 16, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 4, 'dwarf' : 4},
212*33de042dSApple OSS Distributions        { 'name':'r5'   , 'bitsize' : 32, 'offset' : 20, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 5, 'dwarf' : 5},
213*33de042dSApple OSS Distributions        { 'name':'r6'   , 'bitsize' : 32, 'offset' : 24, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 6, 'dwarf' : 6},
214*33de042dSApple OSS Distributions        { 'name':'r7'   , 'bitsize' : 32, 'offset' : 28, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 7, 'dwarf' : 7},
215*33de042dSApple OSS Distributions        { 'name':'r8'   , 'bitsize' : 32, 'offset' : 32, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 8, 'dwarf' : 8},
216*33de042dSApple OSS Distributions        { 'name':'r9'   , 'bitsize' : 32, 'offset' : 36, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc': 9, 'dwarf' : 9},
217*33de042dSApple OSS Distributions        { 'name':'r10'  , 'bitsize' : 32, 'offset' : 40, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':10, 'dwarf' :10},
218*33de042dSApple OSS Distributions        { 'name':'r11'  , 'bitsize' : 32, 'offset' : 44, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':11, 'dwarf' :11, 'alt-name': 'fp', 'generic': 'fp'},
219*33de042dSApple OSS Distributions        { 'name':'r12'  , 'bitsize' : 32, 'offset' : 48, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':12, 'dwarf' :12},
220*33de042dSApple OSS Distributions        { 'name':'sp'   , 'bitsize' : 32, 'offset' : 52, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':13, 'dwarf' :13, 'generic': 'sp'},
221*33de042dSApple OSS Distributions        { 'name':'lr'   , 'bitsize' : 32, 'offset' : 56, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':14, 'dwarf' :14, 'generic': 'lr'},
222*33de042dSApple OSS Distributions        { 'name':'pc'   , 'bitsize' : 32, 'offset' : 60, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':15, 'dwarf' :15, 'generic': 'pc'},
223*33de042dSApple OSS Distributions        { 'name':'cpsr' , 'bitsize' : 32, 'offset' : 64, 'encoding':'uint', 'format':'hex', 'set':0, 'gcc':16, 'dwarf' :16, 'generic':'flags'},
224*33de042dSApple OSS Distributions        { 'name':'fsr'  , 'bitsize' : 32, 'offset' : 68, 'encoding':'uint', 'format':'hex', 'set':0},
225*33de042dSApple OSS Distributions        { 'name':'far'  , 'bitsize' : 32, 'offset' : 72, 'encoding':'uint', 'format':'hex', 'set':0}
226*33de042dSApple OSS Distributions        ]
227*33de042dSApple OSS Distributions        }
228*33de042dSApple OSS Distributions
229*33de042dSApple OSS Distributions    def __init__(self):
230*33de042dSApple OSS Distributions        self.switch_context_address = osplugin_target_obj.FindSymbols('load_reg')[0].GetSymbol().GetStartAddress().GetLoadAddress(osplugin_target_obj) + 8
231*33de042dSApple OSS Distributions        self.ResetRegisterValues()
232*33de042dSApple OSS Distributions
233*33de042dSApple OSS Distributions    @classmethod
234*33de042dSApple OSS Distributions    def GetRegisterInfo(cls, regnum):
235*33de042dSApple OSS Distributions        if regnum < 0 or regnum > len(cls.register_info['registers']):
236*33de042dSApple OSS Distributions            return ''
237*33de042dSApple OSS Distributions
238*33de042dSApple OSS Distributions        reginfo = cls.register_info['registers'][regnum]
239*33de042dSApple OSS Distributions        retval = ''
240*33de042dSApple OSS Distributions        for i in list(reginfo.keys()):
241*33de042dSApple OSS Distributions            v_str = str(reginfo[i])
242*33de042dSApple OSS Distributions            if i == 'set':
243*33de042dSApple OSS Distributions                v_str = 'General Purpose Registers'
244*33de042dSApple OSS Distributions            retval += "%s:%s;" % (str(i), v_str)
245*33de042dSApple OSS Distributions        return retval
246*33de042dSApple OSS Distributions
247*33de042dSApple OSS Distributions    def ResetRegisterValues(self):
248*33de042dSApple OSS Distributions        self.r0 = 0
249*33de042dSApple OSS Distributions        self.r1 = 0
250*33de042dSApple OSS Distributions        self.r2 = 0
251*33de042dSApple OSS Distributions        self.r3 = 0
252*33de042dSApple OSS Distributions        self.r4 = 0
253*33de042dSApple OSS Distributions        self.r5 = 0
254*33de042dSApple OSS Distributions        self.r6 = 0
255*33de042dSApple OSS Distributions        self.r7 = 0
256*33de042dSApple OSS Distributions        self.r8 = 0
257*33de042dSApple OSS Distributions        self.r9 = 0
258*33de042dSApple OSS Distributions        self.r10 = 0
259*33de042dSApple OSS Distributions        self.r11 = 0
260*33de042dSApple OSS Distributions        self.r12 = 0
261*33de042dSApple OSS Distributions        self.sp = 0
262*33de042dSApple OSS Distributions        self.lr = 0
263*33de042dSApple OSS Distributions        self.pc = 0
264*33de042dSApple OSS Distributions        self.cpsr = 0
265*33de042dSApple OSS Distributions        self.fsr = 0
266*33de042dSApple OSS Distributions        self.far = 0
267*33de042dSApple OSS Distributions
268*33de042dSApple OSS Distributions    def __str__(self):
269*33de042dSApple OSS Distributions        return """
270*33de042dSApple OSS Distributions            r0 = {o.r0: <#010x}
271*33de042dSApple OSS Distributions            r1 = {o.r1: <#010x}
272*33de042dSApple OSS Distributions            r2 = {o.r2: <#010x}
273*33de042dSApple OSS Distributions            r3 = {o.r3: <#010x}
274*33de042dSApple OSS Distributions            r4 = {o.r4: <#010x}
275*33de042dSApple OSS Distributions            r5 = {o.r5: <#010x}
276*33de042dSApple OSS Distributions            r6 = {o.r6: <#010x}
277*33de042dSApple OSS Distributions            r7 = {o.r7: <#010x}
278*33de042dSApple OSS Distributions            r8 = {o.r8: <#010x}
279*33de042dSApple OSS Distributions            r9 = {o.r9: <#010x}
280*33de042dSApple OSS Distributions            r10 = {o.r10: <#010x}
281*33de042dSApple OSS Distributions            r11 = {o.r11: <#010x}
282*33de042dSApple OSS Distributions            r12 = {o.r12: <#010x}
283*33de042dSApple OSS Distributions            sp = {o.sp: <#010x}
284*33de042dSApple OSS Distributions            lr = {o.lr: <#010x}
285*33de042dSApple OSS Distributions            pc = {o.pc: <#010x}
286*33de042dSApple OSS Distributions            cpsr = {o.cpsr: <#010x}
287*33de042dSApple OSS Distributions            fsr = {o.fsr : <#010x}
288*33de042dSApple OSS Distributions            far = {o.far : <#010x}
289*33de042dSApple OSS Distributions            """.format(o=self)
290*33de042dSApple OSS Distributions
291*33de042dSApple OSS Distributions    def GetPackedRegisterState(self):
292*33de042dSApple OSS Distributions        return struct.pack('19I', self.r0, self.r1, self.r2, self.r3,
293*33de042dSApple OSS Distributions            self.r4, self.r5, self.r6, self.r7,
294*33de042dSApple OSS Distributions            self.r8, self.r9, self.r10, self.r11,
295*33de042dSApple OSS Distributions            self.r12, self.sp, self.lr, self.pc,
296*33de042dSApple OSS Distributions            self.cpsr, self.fsr, self.far)
297*33de042dSApple OSS Distributions
298*33de042dSApple OSS Distributions    def ReadRegisterDataFromKDPSavedState(self, kdp_state, kernel_version):
299*33de042dSApple OSS Distributions        saved_state = kernel_version.CreateValueFromExpression(None, '(struct arm_saved_state *) ' + str(kdp_state.GetValueAsUnsigned()))
300*33de042dSApple OSS Distributions        saved_state = saved_state.Dereference()
301*33de042dSApple OSS Distributions        saved_state = PluginValue(saved_state)
302*33de042dSApple OSS Distributions        self.ResetRegisterValues()
303*33de042dSApple OSS Distributions        self.r0 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(0).GetValueAsUnsigned()
304*33de042dSApple OSS Distributions        self.r1 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(1).GetValueAsUnsigned()
305*33de042dSApple OSS Distributions        self.r2 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(2).GetValueAsUnsigned()
306*33de042dSApple OSS Distributions        self.r3 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(3).GetValueAsUnsigned()
307*33de042dSApple OSS Distributions        self.r4 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(4).GetValueAsUnsigned()
308*33de042dSApple OSS Distributions        self.r5 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(5).GetValueAsUnsigned()
309*33de042dSApple OSS Distributions        self.r6 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(6).GetValueAsUnsigned()
310*33de042dSApple OSS Distributions        self.r7 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(7).GetValueAsUnsigned()
311*33de042dSApple OSS Distributions        self.r8 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(8).GetValueAsUnsigned()
312*33de042dSApple OSS Distributions        self.r9 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(9).GetValueAsUnsigned()
313*33de042dSApple OSS Distributions        self.r10 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(10).GetValueAsUnsigned()
314*33de042dSApple OSS Distributions        self.r11 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(11).GetValueAsUnsigned()
315*33de042dSApple OSS Distributions        self.r12 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(12).GetValueAsUnsigned()
316*33de042dSApple OSS Distributions        self.sp = saved_state.GetChildMemberWithName('sp').GetValueAsUnsigned()
317*33de042dSApple OSS Distributions        self.lr = saved_state.GetChildMemberWithName('lr').GetValueAsUnsigned()
318*33de042dSApple OSS Distributions        self.pc = saved_state.GetChildMemberWithName('pc').GetValueAsUnsigned()
319*33de042dSApple OSS Distributions        self.cpsr = saved_state.GetChildMemberWithName('cpsr').GetValueAsUnsigned()
320*33de042dSApple OSS Distributions        self.fsr = saved_state.GetChildMemberWithName('fsr').GetValueAsUnsigned()
321*33de042dSApple OSS Distributions        self.far = saved_state.GetChildMemberWithName('far').GetValueAsUnsigned()
322*33de042dSApple OSS Distributions        return self
323*33de042dSApple OSS Distributions
324*33de042dSApple OSS Distributions    def ReadRegisterDataFromKernelStack(self, kstack_saved_state_addr, kernel_version):
325*33de042dSApple OSS Distributions        saved_state = kernel_version.CreateValueFromExpression(None, '(struct arm_saved_state *) '+ str(kstack_saved_state_addr))
326*33de042dSApple OSS Distributions        saved_state = saved_state.Dereference()
327*33de042dSApple OSS Distributions        saved_state = PluginValue(saved_state)
328*33de042dSApple OSS Distributions        self.ResetRegisterValues()
329*33de042dSApple OSS Distributions        self.r0 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(0).GetValueAsUnsigned()
330*33de042dSApple OSS Distributions        self.r1 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(1).GetValueAsUnsigned()
331*33de042dSApple OSS Distributions        self.r2 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(2).GetValueAsUnsigned()
332*33de042dSApple OSS Distributions        self.r3 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(3).GetValueAsUnsigned()
333*33de042dSApple OSS Distributions        self.r4 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(4).GetValueAsUnsigned()
334*33de042dSApple OSS Distributions        self.r5 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(5).GetValueAsUnsigned()
335*33de042dSApple OSS Distributions        self.r6 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(6).GetValueAsUnsigned()
336*33de042dSApple OSS Distributions        self.r7 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(7).GetValueAsUnsigned()
337*33de042dSApple OSS Distributions        self.r8 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(8).GetValueAsUnsigned()
338*33de042dSApple OSS Distributions        self.r9 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(9).GetValueAsUnsigned()
339*33de042dSApple OSS Distributions        self.r10 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(10).GetValueAsUnsigned()
340*33de042dSApple OSS Distributions        self.r11 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(11).GetValueAsUnsigned()
341*33de042dSApple OSS Distributions        self.r12 = saved_state.GetChildMemberWithName('r').GetChildAtIndex(12).GetValueAsUnsigned()
342*33de042dSApple OSS Distributions        self.sp = saved_state.GetChildMemberWithName('sp').GetValueAsUnsigned()
343*33de042dSApple OSS Distributions        self.lr = saved_state.GetChildMemberWithName('lr').GetValueAsUnsigned()
344*33de042dSApple OSS Distributions        # pc for a blocked thread is treated to be the next instruction it would run after thread switch.
345*33de042dSApple OSS Distributions        self.pc = self.switch_context_address
346*33de042dSApple OSS Distributions        self.cpsr = saved_state.GetChildMemberWithName('cpsr').GetValueAsUnsigned()
347*33de042dSApple OSS Distributions        self.fsr = saved_state.GetChildMemberWithName('fsr').GetValueAsUnsigned()
348*33de042dSApple OSS Distributions        self.far = saved_state.GetChildMemberWithName('far').GetValueAsUnsigned()
349*33de042dSApple OSS Distributions        return self
350*33de042dSApple OSS Distributions
351*33de042dSApple OSS Distributions    def ReadRegisterDataFromContinuation(self, continuation_ptr):
352*33de042dSApple OSS Distributions        self.ResetRegisterValues()
353*33de042dSApple OSS Distributions        self.pc = continuation_ptr
354*33de042dSApple OSS Distributions        return self
355*33de042dSApple OSS Distributions
356*33de042dSApple OSS Distributions
357*33de042dSApple OSS Distributionsclass I386_RegisterSet(object):
358*33de042dSApple OSS Distributions    """ register info set for i386 architecture
359*33de042dSApple OSS Distributions    """
360*33de042dSApple OSS Distributions    register_info = { 'sets' : ['GPR'],
361*33de042dSApple OSS Distributions                  'registers': [
362*33de042dSApple OSS Distributions        { 'name': 'eax'   , 'bitsize': 32, 'offset' : 0, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 0, 'dwarf': 0},
363*33de042dSApple OSS Distributions        { 'name': 'ebx'   , 'bitsize': 32, 'offset' : 4, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 3, 'dwarf': 3},
364*33de042dSApple OSS Distributions        { 'name': 'ecx'   , 'bitsize': 32, 'offset' : 8, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 1, 'dwarf': 1},
365*33de042dSApple OSS Distributions        { 'name': 'edx'   , 'bitsize': 32, 'offset' :12, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 2, 'dwarf': 2},
366*33de042dSApple OSS Distributions        { 'name': 'edi'   , 'bitsize': 32, 'offset' :16, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 7, 'dwarf': 7},
367*33de042dSApple OSS Distributions        { 'name': 'esi'   , 'bitsize': 32, 'offset' :20, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 6, 'dwarf': 6},
368*33de042dSApple OSS Distributions        { 'name': 'ebp'   , 'bitsize': 32, 'offset' :24, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 4, 'dwarf': 5, 'generic': 'fp', 'alt-name': 'fp'},
369*33de042dSApple OSS Distributions        { 'name': 'esp'   , 'bitsize': 32, 'offset' :28, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 5, 'dwarf': 4, 'generic': 'sp', 'alt-name': 'sp'},
370*33de042dSApple OSS Distributions        { 'name': 'ss'    , 'bitsize': 32, 'offset' :32, 'encoding': 'uint' , 'format':'hex' , 'set': 0},
371*33de042dSApple OSS Distributions        { 'name': 'eflags', 'bitsize': 32, 'offset' :36, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' : 9, 'dwarf': 9, 'generic': 'flags'},
372*33de042dSApple OSS Distributions        { 'name': 'eip'   , 'bitsize': 32, 'offset' :40, 'encoding': 'uint' , 'format':'hex' , 'set': 0, 'gcc' :8, 'dwarf':8, 'generic': 'pc', 'alt-name': 'pc'},
373*33de042dSApple OSS Distributions        { 'name': 'cs'    , 'bitsize': 32, 'offset' :44, 'encoding': 'uint' , 'format':'hex' , 'set': 0},
374*33de042dSApple OSS Distributions        { 'name': 'ds'    , 'bitsize': 32, 'offset' :48, 'encoding': 'uint' , 'format':'hex' , 'set': 0},
375*33de042dSApple OSS Distributions        { 'name': 'es'    , 'bitsize': 32, 'offset' :52, 'encoding': 'uint' , 'format':'hex' , 'set': 0},
376*33de042dSApple OSS Distributions        { 'name': 'fs'    , 'bitsize': 32, 'offset' :56, 'encoding': 'uint' , 'format':'hex' , 'set': 0},
377*33de042dSApple OSS Distributions        { 'name': 'gs'    , 'bitsize': 32, 'offset' :60, 'encoding': 'uint' , 'format':'hex' , 'set': 0},
378*33de042dSApple OSS Distributions        ]
379*33de042dSApple OSS Distributions        }
380*33de042dSApple OSS Distributions
381*33de042dSApple OSS Distributions    def __init__(self):
382*33de042dSApple OSS Distributions        self.ResetRegisterValues()
383*33de042dSApple OSS Distributions
384*33de042dSApple OSS Distributions    @classmethod
385*33de042dSApple OSS Distributions    def GetRegisterInfo(cls, regnum):
386*33de042dSApple OSS Distributions        if regnum < 0 or regnum > len(cls.register_info['registers']):
387*33de042dSApple OSS Distributions            return ''
388*33de042dSApple OSS Distributions
389*33de042dSApple OSS Distributions        reginfo = cls.register_info['registers'][regnum]
390*33de042dSApple OSS Distributions        retval = ''
391*33de042dSApple OSS Distributions        for i in list(reginfo.keys()):
392*33de042dSApple OSS Distributions            v_str = str(reginfo[i])
393*33de042dSApple OSS Distributions            if i == 'set':
394*33de042dSApple OSS Distributions                v_str = 'General Purpose Registers'
395*33de042dSApple OSS Distributions            retval += "%s:%s;" % (str(i), v_str)
396*33de042dSApple OSS Distributions        return retval
397*33de042dSApple OSS Distributions
398*33de042dSApple OSS Distributions    def ResetRegisterValues(self):
399*33de042dSApple OSS Distributions        """ set all registers to zero """
400*33de042dSApple OSS Distributions        self.eax = 0
401*33de042dSApple OSS Distributions        self.ebx = 0
402*33de042dSApple OSS Distributions        self.ecx = 0
403*33de042dSApple OSS Distributions        self.edx = 0
404*33de042dSApple OSS Distributions        self.edi = 0
405*33de042dSApple OSS Distributions        self.esi = 0
406*33de042dSApple OSS Distributions        self.ebp = 0
407*33de042dSApple OSS Distributions        self.esp = 0
408*33de042dSApple OSS Distributions        self.ss  = 0
409*33de042dSApple OSS Distributions        self.eflags = 0
410*33de042dSApple OSS Distributions        self.eip = 0
411*33de042dSApple OSS Distributions        self.cs = 0
412*33de042dSApple OSS Distributions        self.ds = 0
413*33de042dSApple OSS Distributions        self.es = 0
414*33de042dSApple OSS Distributions        self.fs = 0
415*33de042dSApple OSS Distributions        self.gs = 0
416*33de042dSApple OSS Distributions
417*33de042dSApple OSS Distributions    def __str__(self):
418*33de042dSApple OSS Distributions        return """
419*33de042dSApple OSS Distributions            eax = {o.eax: #010x}
420*33de042dSApple OSS Distributions            ebx = {o.ebx: #010x}
421*33de042dSApple OSS Distributions            ecx = {o.ecx: #010x}
422*33de042dSApple OSS Distributions            edx = {o.edx: #010x}
423*33de042dSApple OSS Distributions            edi = {o.edi: #010x}
424*33de042dSApple OSS Distributions            esi = {o.esi: #010x}
425*33de042dSApple OSS Distributions            ebp = {o.ebp: #010x}
426*33de042dSApple OSS Distributions            esp = {o.esp: #010x}
427*33de042dSApple OSS Distributions            ss  = {o.ss: #010x}
428*33de042dSApple OSS Distributions         eflags = {o.eflags: #010x}
429*33de042dSApple OSS Distributions            eip = {o.eip: #010x}
430*33de042dSApple OSS Distributions            cs  = {o.cs: #010x}
431*33de042dSApple OSS Distributions            ds  = {o.ds: #010x}
432*33de042dSApple OSS Distributions            es  = {o.es: #010x}
433*33de042dSApple OSS Distributions            fs  = {o.fs: #010x}
434*33de042dSApple OSS Distributions            gs  = {o.gs: #010x}
435*33de042dSApple OSS Distributions            """.format(o=self)
436*33de042dSApple OSS Distributions
437*33de042dSApple OSS Distributions    def GetPackedRegisterState(self):
438*33de042dSApple OSS Distributions        """ get a struct.pack register data """
439*33de042dSApple OSS Distributions        return struct.pack('16I', self.eax, self.ebx, self.ecx,
440*33de042dSApple OSS Distributions            self.edx, self.edi, self.esi,
441*33de042dSApple OSS Distributions            self.ebp, self.esp, self.ss,
442*33de042dSApple OSS Distributions            self.eflags, self.eip, self.cs,
443*33de042dSApple OSS Distributions            self.ds, self.es, self.fs, self.gs
444*33de042dSApple OSS Distributions            )
445*33de042dSApple OSS Distributions
446*33de042dSApple OSS Distributions    def ReadRegisterDataFromKDPSavedState(self, kdp_state, kernel_version):
447*33de042dSApple OSS Distributions        """ to be implemented"""
448*33de042dSApple OSS Distributions        return None
449*33de042dSApple OSS Distributions
450*33de042dSApple OSS Distributions    def ReadRegisterDataFromKernelStack(self, kstack_saved_state_addr, kernel_version):
451*33de042dSApple OSS Distributions        """ to be implemented """
452*33de042dSApple OSS Distributions        return None
453*33de042dSApple OSS Distributions
454*33de042dSApple OSS Distributions    def ReadRegisterDataFromContinuation(self, continuation_ptr):
455*33de042dSApple OSS Distributions        self.ResetRegisterValues()
456*33de042dSApple OSS Distributions        self.eip = continuation_ptr
457*33de042dSApple OSS Distributions        return self
458*33de042dSApple OSS Distributions
459*33de042dSApple OSS Distributions
460*33de042dSApple OSS Distributionsclass X86_64RegisterSet(object):
461*33de042dSApple OSS Distributions    """ register info set for x86_64 architecture """
462*33de042dSApple OSS Distributions    register_info = { 'sets' : ['GPR'],
463*33de042dSApple OSS Distributions                  'registers': [
464*33de042dSApple OSS Distributions        { 'name':'rax'       , 'bitsize' :  64, 'offset' :   0, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 0, 'dwarf' : 0},
465*33de042dSApple OSS Distributions        { 'name':'rbx'       , 'bitsize' :  64, 'offset' :   8, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 3, 'dwarf' : 3},
466*33de042dSApple OSS Distributions        { 'name':'rcx'       , 'bitsize' :  64, 'offset' :  16, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 2, 'dwarf' : 2, 'generic':'arg4', 'alt-name':'arg4', },
467*33de042dSApple OSS Distributions        { 'name':'rdx'       , 'bitsize' :  64, 'offset' :  24, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 1, 'dwarf' : 1, 'generic':'arg3', 'alt-name':'arg3', },
468*33de042dSApple OSS Distributions        { 'name':'rdi'       , 'bitsize' :  64, 'offset' :  32, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 5, 'dwarf' : 5, 'generic':'arg1', 'alt-name':'arg1', },
469*33de042dSApple OSS Distributions        { 'name':'rsi'       , 'bitsize' :  64, 'offset' :  40, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 4, 'dwarf' : 4, 'generic':'arg2', 'alt-name':'arg2', },
470*33de042dSApple OSS Distributions        { 'name':'rbp'       , 'bitsize' :  64, 'offset' :  48, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 6, 'dwarf' : 6, 'generic':'fp'  , 'alt-name':'fp', },
471*33de042dSApple OSS Distributions        { 'name':'rsp'       , 'bitsize' :  64, 'offset' :  56, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 7, 'dwarf' : 7, 'generic':'sp'  , 'alt-name':'sp', },
472*33de042dSApple OSS Distributions        { 'name':'r8'        , 'bitsize' :  64, 'offset' :  64, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 8, 'dwarf' : 8, 'generic':'arg5', 'alt-name':'arg5', },
473*33de042dSApple OSS Distributions        { 'name':'r9'        , 'bitsize' :  64, 'offset' :  72, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 9, 'dwarf' : 9, 'generic':'arg6', 'alt-name':'arg6', },
474*33de042dSApple OSS Distributions        { 'name':'r10'       , 'bitsize' :  64, 'offset' :  80, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 10, 'dwarf' : 10},
475*33de042dSApple OSS Distributions        { 'name':'r11'       , 'bitsize' :  64, 'offset' :  88, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 11, 'dwarf' : 11},
476*33de042dSApple OSS Distributions        { 'name':'r12'       , 'bitsize' :  64, 'offset' :  96, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 12, 'dwarf' : 12},
477*33de042dSApple OSS Distributions        { 'name':'r13'       , 'bitsize' :  64, 'offset' : 104, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 13, 'dwarf' : 13},
478*33de042dSApple OSS Distributions        { 'name':'r14'       , 'bitsize' :  64, 'offset' : 112, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 14, 'dwarf' : 14},
479*33de042dSApple OSS Distributions        { 'name':'r15'       , 'bitsize' :  64, 'offset' : 120, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 15, 'dwarf' : 15},
480*33de042dSApple OSS Distributions        { 'name':'rip'       , 'bitsize' :  64, 'offset' : 128, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'gcc' : 16, 'dwarf' : 16, 'generic':'pc', 'alt-name':'pc' },
481*33de042dSApple OSS Distributions        { 'name':'rflags'    , 'bitsize' :  64, 'offset' : 136, 'encoding':'uint'  , 'format':'hex'         , 'set': 0, 'generic':'flags', 'alt-name':'flags' },
482*33de042dSApple OSS Distributions        { 'name':'cs'        , 'bitsize' :  64, 'offset' : 144, 'encoding':'uint'  , 'format':'hex'         , 'set': 0                          },
483*33de042dSApple OSS Distributions        { 'name':'fs'        , 'bitsize' :  64, 'offset' : 152, 'encoding':'uint'  , 'format':'hex'         , 'set': 0                          },
484*33de042dSApple OSS Distributions        { 'name':'gs'        , 'bitsize' :  64, 'offset' : 160, 'encoding':'uint'  , 'format':'hex'         , 'set': 0                          },
485*33de042dSApple OSS Distributions        ]
486*33de042dSApple OSS Distributions        }
487*33de042dSApple OSS Distributions    def __init__(self):
488*33de042dSApple OSS Distributions        self.ResetRegisterValues()
489*33de042dSApple OSS Distributions
490*33de042dSApple OSS Distributions    @classmethod
491*33de042dSApple OSS Distributions    def GetRegisterInfo(cls, regnum):
492*33de042dSApple OSS Distributions        if regnum < 0 or regnum > len(cls.register_info['registers']):
493*33de042dSApple OSS Distributions            return ''
494*33de042dSApple OSS Distributions
495*33de042dSApple OSS Distributions        reginfo = cls.register_info['registers'][regnum]
496*33de042dSApple OSS Distributions        retval = ''
497*33de042dSApple OSS Distributions        for i in list(reginfo.keys()):
498*33de042dSApple OSS Distributions            v_str = str(reginfo[i])
499*33de042dSApple OSS Distributions            if i == 'set':
500*33de042dSApple OSS Distributions                v_str = 'General Purpose Registers'
501*33de042dSApple OSS Distributions            retval += "%s:%s;" % (str(i), v_str)
502*33de042dSApple OSS Distributions        return retval
503*33de042dSApple OSS Distributions
504*33de042dSApple OSS Distributions
505*33de042dSApple OSS Distributions    def ResetRegisterValues(self):
506*33de042dSApple OSS Distributions        """ set all the registers to zero. """
507*33de042dSApple OSS Distributions        self.rax = 0
508*33de042dSApple OSS Distributions        self.rbx = 0
509*33de042dSApple OSS Distributions        self.rcx = 0
510*33de042dSApple OSS Distributions        self.rdx = 0
511*33de042dSApple OSS Distributions        self.rdi = 0
512*33de042dSApple OSS Distributions        self.rsi = 0
513*33de042dSApple OSS Distributions        self.rbp = 0
514*33de042dSApple OSS Distributions        self.rsp = 0
515*33de042dSApple OSS Distributions        self.r8  = 0
516*33de042dSApple OSS Distributions        self.r9  = 0
517*33de042dSApple OSS Distributions        self.r10 = 0
518*33de042dSApple OSS Distributions        self.r11 = 0
519*33de042dSApple OSS Distributions        self.r12 = 0
520*33de042dSApple OSS Distributions        self.r13 = 0
521*33de042dSApple OSS Distributions        self.r14 = 0
522*33de042dSApple OSS Distributions        self.r15 = 0
523*33de042dSApple OSS Distributions        self.rip = 0
524*33de042dSApple OSS Distributions        self.rflags = 0
525*33de042dSApple OSS Distributions        self.cs  = 0
526*33de042dSApple OSS Distributions        self.fs  = 0
527*33de042dSApple OSS Distributions        self.gs  = 0
528*33de042dSApple OSS Distributions
529*33de042dSApple OSS Distributions    def __str__(self):
530*33de042dSApple OSS Distributions        return """
531*33de042dSApple OSS Distributions            rax = {o.rax: <#018x}
532*33de042dSApple OSS Distributions            rbx = {o.rbx: <#018x}
533*33de042dSApple OSS Distributions            rcx = {o.rcx: <#018x}
534*33de042dSApple OSS Distributions            rdx = {o.rdx: <#018x}
535*33de042dSApple OSS Distributions            rdi = {o.rdi: <#018x}
536*33de042dSApple OSS Distributions            rsi = {o.rsi: <#018x}
537*33de042dSApple OSS Distributions            rbp = {o.rbp: <#018x}
538*33de042dSApple OSS Distributions            rsp = {o.rsp: <#018x}
539*33de042dSApple OSS Distributions            r8  = {o.r8: <#018x}
540*33de042dSApple OSS Distributions            r9  = {o.r9: <#018x}
541*33de042dSApple OSS Distributions            r10 = {o.r10: <#018x}
542*33de042dSApple OSS Distributions            r11 = {o.r11: <#018x}
543*33de042dSApple OSS Distributions            r12 = {o.r12: <#018x}
544*33de042dSApple OSS Distributions            r13 = {o.r13: <#018x}
545*33de042dSApple OSS Distributions            r14 = {o.r14: <#018x}
546*33de042dSApple OSS Distributions            r15 = {o.r15: <#018x}
547*33de042dSApple OSS Distributions            rip = {o.rip: <#018x}
548*33de042dSApple OSS Distributions            rflags =  {o.rflags: <#018x}
549*33de042dSApple OSS Distributions            cs = {o.cs: <#018x}
550*33de042dSApple OSS Distributions            fs = {o.fs: <#018x}
551*33de042dSApple OSS Distributions            gs = {o.gs: <#018x}
552*33de042dSApple OSS Distributions            """.format(o=self)
553*33de042dSApple OSS Distributions
554*33de042dSApple OSS Distributions    def GetPackedRegisterState(self):
555*33de042dSApple OSS Distributions        """ get a struct.pack register data for passing to C constructs """
556*33de042dSApple OSS Distributions        return struct.pack('21Q', self.rax, self.rbx, self.rcx, self.rdx, self.rdi,
557*33de042dSApple OSS Distributions            self.rsi, self.rbp, self.rsp, self.r8,  self.r9,
558*33de042dSApple OSS Distributions            self.r10, self.r11, self.r12, self.r13, self.r14,
559*33de042dSApple OSS Distributions            self.r15, self.rip, self.rflags, self.cs, self.fs, self.gs)
560*33de042dSApple OSS Distributions
561*33de042dSApple OSS Distributions    def ReadRegisterDataFromKDPSavedState(self, kdp_state, kernel_version):
562*33de042dSApple OSS Distributions        saved_state = kernel_version.CreateValueFromExpression(None, '(struct x86_saved_state64 *) '+ str(kdp_state.GetValueAsUnsigned()))
563*33de042dSApple OSS Distributions        saved_state = saved_state.Dereference()
564*33de042dSApple OSS Distributions        saved_state = PluginValue(saved_state)
565*33de042dSApple OSS Distributions        self.ResetRegisterValues()
566*33de042dSApple OSS Distributions        self.rdi = saved_state.GetChildMemberWithName('rdi').GetValueAsUnsigned()
567*33de042dSApple OSS Distributions        self.rsi = saved_state.GetChildMemberWithName('rsi').GetValueAsUnsigned()
568*33de042dSApple OSS Distributions        self.rdx = saved_state.GetChildMemberWithName('rdx').GetValueAsUnsigned()
569*33de042dSApple OSS Distributions        self.r10 = saved_state.GetChildMemberWithName('r10').GetValueAsUnsigned()
570*33de042dSApple OSS Distributions        self.r8 = saved_state.GetChildMemberWithName('r8').GetValueAsUnsigned()
571*33de042dSApple OSS Distributions        self.r9 = saved_state.GetChildMemberWithName('r9').GetValueAsUnsigned()
572*33de042dSApple OSS Distributions        self.r15 = saved_state.GetChildMemberWithName('r15').GetValueAsUnsigned()
573*33de042dSApple OSS Distributions        self.r14 = saved_state.GetChildMemberWithName('r14').GetValueAsUnsigned()
574*33de042dSApple OSS Distributions        self.r13 = saved_state.GetChildMemberWithName('r13').GetValueAsUnsigned()
575*33de042dSApple OSS Distributions        self.r12 = saved_state.GetChildMemberWithName('r12').GetValueAsUnsigned()
576*33de042dSApple OSS Distributions        self.r11 = saved_state.GetChildMemberWithName('r11').GetValueAsUnsigned()
577*33de042dSApple OSS Distributions        self.rbp = saved_state.GetChildMemberWithName('rbp').GetValueAsUnsigned()
578*33de042dSApple OSS Distributions        self.rbx = saved_state.GetChildMemberWithName('rbx').GetValueAsUnsigned()
579*33de042dSApple OSS Distributions        self.rcx = saved_state.GetChildMemberWithName('rcx').GetValueAsUnsigned()
580*33de042dSApple OSS Distributions        self.rax = saved_state.GetChildMemberWithName('rax').GetValueAsUnsigned()
581*33de042dSApple OSS Distributions        self.rip = saved_state.GetChildMemberWithName('isf').GetChildMemberWithName('rip').GetValueAsUnsigned()
582*33de042dSApple OSS Distributions        self.rflags = saved_state.GetChildMemberWithName('isf').GetChildMemberWithName('rflags').GetValueAsUnsigned()
583*33de042dSApple OSS Distributions        self.rsp = saved_state.GetChildMemberWithName('isf').GetChildMemberWithName('rsp').GetValueAsUnsigned()
584*33de042dSApple OSS Distributions        return self
585*33de042dSApple OSS Distributions
586*33de042dSApple OSS Distributions    def ReadRegisterDataFromKernelStack(self, kstack_saved_state_addr, kernel_version):
587*33de042dSApple OSS Distributions        saved_state = kernel_version.CreateValueFromExpression(None, '(struct x86_kernel_state *) '+ str(kstack_saved_state_addr))
588*33de042dSApple OSS Distributions        saved_state = saved_state.Dereference()
589*33de042dSApple OSS Distributions        saved_state = PluginValue(saved_state)
590*33de042dSApple OSS Distributions        self.ResetRegisterValues()
591*33de042dSApple OSS Distributions        self.rbx = saved_state.GetChildMemberWithName('k_rbx').GetValueAsUnsigned()
592*33de042dSApple OSS Distributions        self.rsp = saved_state.GetChildMemberWithName('k_rsp').GetValueAsUnsigned()
593*33de042dSApple OSS Distributions        self.rbp = saved_state.GetChildMemberWithName('k_rbp').GetValueAsUnsigned()
594*33de042dSApple OSS Distributions        self.r12 = saved_state.GetChildMemberWithName('k_r12').GetValueAsUnsigned()
595*33de042dSApple OSS Distributions        self.r13 = saved_state.GetChildMemberWithName('k_r13').GetValueAsUnsigned()
596*33de042dSApple OSS Distributions        self.r14 = saved_state.GetChildMemberWithName('k_r14').GetValueAsUnsigned()
597*33de042dSApple OSS Distributions        self.r15 = saved_state.GetChildMemberWithName('k_r15').GetValueAsUnsigned()
598*33de042dSApple OSS Distributions        self.rip = saved_state.GetChildMemberWithName('k_rip').GetValueAsUnsigned()
599*33de042dSApple OSS Distributions        return self
600*33de042dSApple OSS Distributions
601*33de042dSApple OSS Distributions    def ReadRegisterDataFromContinuation(self, continuation_ptr):
602*33de042dSApple OSS Distributions        self.ResetRegisterValues()
603*33de042dSApple OSS Distributions        self.rip = continuation_ptr
604*33de042dSApple OSS Distributions        return self
605*33de042dSApple OSS Distributions
606*33de042dSApple OSS Distributions
607*33de042dSApple OSS Distributions
608*33de042dSApple OSS Distributions
609*33de042dSApple OSS Distributionsdef IterateQueue(queue_head: lldb.SBValue, element_ptr_type: lldb.SBType, element_field_name: str):
610*33de042dSApple OSS Distributions    """ iterate over a queue in kernel of type queue_head_t. refer to osfmk/kern/queue.h
611*33de042dSApple OSS Distributions        params:
612*33de042dSApple OSS Distributions            queue_head         - lldb.SBValue : Value object for queue_head.
613*33de042dSApple OSS Distributions            element_type       - lldb.SBType : a pointer type of the element 'next' points to. Typically its structs like thread, task etc..
614*33de042dSApple OSS Distributions            element_field_name - str : name of the field in target struct.
615*33de042dSApple OSS Distributions        returns:
616*33de042dSApple OSS Distributions            A generator does not return. It is used for iterating.
617*33de042dSApple OSS Distributions            SBValue  : an object thats of type (element_type) queue_head->next. Always a pointer object
618*33de042dSApple OSS Distributions    """
619*33de042dSApple OSS Distributions    queue_head_addr = 0x0
620*33de042dSApple OSS Distributions    if queue_head.TypeIsPointerType():
621*33de042dSApple OSS Distributions        queue_head_addr = queue_head.GetValueAsUnsigned()
622*33de042dSApple OSS Distributions    else:
623*33de042dSApple OSS Distributions        queue_head_addr = queue_head.GetAddress().GetLoadAddress(osplugin_target_obj)
624*33de042dSApple OSS Distributions    cur_elt: lldb.SBValue = queue_head.GetChildMemberWithName('next')
625*33de042dSApple OSS Distributions    while True:
626*33de042dSApple OSS Distributions        if not (cur_elt.IsValid() and cur_elt.error.success) or cur_elt.GetValueAsUnsigned() == 0 or cur_elt.GetValueAsUnsigned() == queue_head_addr:
627*33de042dSApple OSS Distributions            break
628*33de042dSApple OSS Distributions        elt = cur_elt.Cast(element_ptr_type)
629*33de042dSApple OSS Distributions        yield elt
630*33de042dSApple OSS Distributions        cur_elt = elt.GetChildMemberWithName(element_field_name).GetChildMemberWithName('next')
631*33de042dSApple OSS Distributions
632*33de042dSApple OSS Distributionsdef GetUniqueSessionID(process_obj):
633*33de042dSApple OSS Distributions    """ Create a unique session identifier.
634*33de042dSApple OSS Distributions        params:
635*33de042dSApple OSS Distributions          process_obj: lldb.SBProcess object refering to connected process.
636*33de042dSApple OSS Distributions        returns:
637*33de042dSApple OSS Distributions          int - a unique number identified by processid and stopid.
638*33de042dSApple OSS Distributions    """
639*33de042dSApple OSS Distributions    session_key_str = ""
640*33de042dSApple OSS Distributions    if hasattr(process_obj, "GetUniqueID"):
641*33de042dSApple OSS Distributions        session_key_str += str(process_obj.GetUniqueID()) + ":"
642*33de042dSApple OSS Distributions    else:
643*33de042dSApple OSS Distributions        session_key_str += "0:"
644*33de042dSApple OSS Distributions
645*33de042dSApple OSS Distributions    if hasattr(process_obj, "GetStopID"):
646*33de042dSApple OSS Distributions        session_key_str += str(process_obj.GetStopID())
647*33de042dSApple OSS Distributions    else:
648*33de042dSApple OSS Distributions        session_key_str +="1"
649*33de042dSApple OSS Distributions
650*33de042dSApple OSS Distributions    return hash(session_key_str)
651*33de042dSApple OSS Distributions
652*33de042dSApple OSS Distributions
653*33de042dSApple OSS Distributions(archX86_64, archARMv7, archI386, archARMv8) = ("x86_64", "armv7", "i386", "arm64")
654*33de042dSApple OSS Distributions
655*33de042dSApple OSS Distributionsclass OperatingSystemPlugIn(object):
656*33de042dSApple OSS Distributions    """Class that provides data for an instance of a LLDB 'OperatingSystemPython' plug-in class"""
657*33de042dSApple OSS Distributions
658*33de042dSApple OSS Distributions    def __init__(self, process):
659*33de042dSApple OSS Distributions        '''Initialization needs a valid.SBProcess object'''
660*33de042dSApple OSS Distributions        self.process = None
661*33de042dSApple OSS Distributions        self.registers = None
662*33de042dSApple OSS Distributions        self.threads = None
663*33de042dSApple OSS Distributions        self.thread_cache = {}
664*33de042dSApple OSS Distributions        self.current_session_id = 0
665*33de042dSApple OSS Distributions        self.kdp_thread = None
666*33de042dSApple OSS Distributions        if type(process) is lldb.SBProcess and process.IsValid():
667*33de042dSApple OSS Distributions            global osplugin_target_obj
668*33de042dSApple OSS Distributions            self.process = process
669*33de042dSApple OSS Distributions            self._target = process.target
670*33de042dSApple OSS Distributions            osplugin_target_obj = self._target
671*33de042dSApple OSS Distributions            self.current_session_id = GetUniqueSessionID(self.process)
672*33de042dSApple OSS Distributions            self.version = self._target.FindGlobalVariables('version', 1).GetValueAtIndex(0)
673*33de042dSApple OSS Distributions
674*33de042dSApple OSS Distributions            # Configure explicit pointer stripping
675*33de042dSApple OSS Distributions            is_tagged = self._target.FindFirstGlobalVariable('kasan_tbi_enabled').GetValueAsUnsigned()
676*33de042dSApple OSS Distributions
677*33de042dSApple OSS Distributions            if is_tagged:
678*33de042dSApple OSS Distributions
679*33de042dSApple OSS Distributions                def strip_ptr(ptr):
680*33de042dSApple OSS Distributions                    if ptr != 0:
681*33de042dSApple OSS Distributions                        ptr |= (0xFF << 56)
682*33de042dSApple OSS Distributions                    return ptr
683*33de042dSApple OSS Distributions                self._strip_ptr = strip_ptr
684*33de042dSApple OSS Distributions
685*33de042dSApple OSS Distributions                def strip_thread_sbval(th):
686*33de042dSApple OSS Distributions                    addr = th.GetValueAsAddress()
687*33de042dSApple OSS Distributions                    return self.version.CreateValueFromExpression(str(addr), '(struct thread *)' + str(addr))
688*33de042dSApple OSS Distributions                self._strip_thread_sbval = strip_thread_sbval
689*33de042dSApple OSS Distributions
690*33de042dSApple OSS Distributions            else:
691*33de042dSApple OSS Distributions                self._strip_ptr = lambda ptr: ptr
692*33de042dSApple OSS Distributions                self._strip_thread_sbval = lambda val: val
693*33de042dSApple OSS Distributions
694*33de042dSApple OSS Distributions            self.kernel_stack_size = self._target.FindGlobalVariables('kernel_stack_size', 1).GetValueAtIndex(0).GetValueAsUnsigned()
695*33de042dSApple OSS Distributions            self.kernel_context_size = 0
696*33de042dSApple OSS Distributions            self.connected_over_kdp = False
697*33de042dSApple OSS Distributions            # connected_to_debugserver signifies if we are connected to astris or other gdbserver instance
698*33de042dSApple OSS Distributions            # that has the correct thread state for on core threads. For kdp and coredumps we rely on in memory
699*33de042dSApple OSS Distributions            # state of threads.
700*33de042dSApple OSS Distributions            self.connected_to_debugserver = True
701*33de042dSApple OSS Distributions            plugin_string = self.process.GetPluginName().lower()
702*33de042dSApple OSS Distributions            if plugin_string.find("kdp") >=0:
703*33de042dSApple OSS Distributions                self.connected_over_kdp = True
704*33de042dSApple OSS Distributions                self.connected_to_debugserver = False
705*33de042dSApple OSS Distributions            #print "version", self.version, "kernel_stack_size", self.kernel_stack_size, "context_size", self.kernel_context_size
706*33de042dSApple OSS Distributions            self.threads = None # Will be an dictionary containing info for each thread
707*33de042dSApple OSS Distributions            triple = self.process.target.triple
708*33de042dSApple OSS Distributions            arch = triple.split('-')[0].lower()
709*33de042dSApple OSS Distributions            self.target_arch = ""
710*33de042dSApple OSS Distributions            self.kernel_context_size = 0
711*33de042dSApple OSS Distributions            if arch == archX86_64 :
712*33de042dSApple OSS Distributions                self.target_arch = archX86_64
713*33de042dSApple OSS Distributions                print("Target arch: x86_64")
714*33de042dSApple OSS Distributions                self.register_set = X86_64RegisterSet()
715*33de042dSApple OSS Distributions                self.kernel_context_size = self._target.FindFirstType('x86_kernel_state').GetByteSize()
716*33de042dSApple OSS Distributions                self.kernel_thread_state_size = self._target.FindFirstType('struct thread_kernel_state').GetByteSize()
717*33de042dSApple OSS Distributions            elif arch.startswith(archARMv7) :
718*33de042dSApple OSS Distributions                self.target_arch = arch
719*33de042dSApple OSS Distributions                print("Target arch: " + self.target_arch)
720*33de042dSApple OSS Distributions                self.register_set = Armv7_RegisterSet()
721*33de042dSApple OSS Distributions            elif arch.startswith(archARMv8):
722*33de042dSApple OSS Distributions                self.target_arch = arch
723*33de042dSApple OSS Distributions                print("Target arch: " + self.target_arch)
724*33de042dSApple OSS Distributions                self.register_set = Armv8_RegisterSet()
725*33de042dSApple OSS Distributions            #  connection     intel         arm
726*33de042dSApple OSS Distributions            #  kdp            Memory        Memory
727*33de042dSApple OSS Distributions            #  gdb            Server        Server
728*33de042dSApple OSS Distributions            #  coredump       Memory        Server
729*33de042dSApple OSS Distributions            if not self.connected_over_kdp :
730*33de042dSApple OSS Distributions                if plugin_string.find('core') >= 0 and self.target_arch == archX86_64:
731*33de042dSApple OSS Distributions                    self.connected_to_debugserver = False
732*33de042dSApple OSS Distributions            self.registers = self.register_set.register_info
733*33de042dSApple OSS Distributions            if self.connected_to_debugserver:
734*33de042dSApple OSS Distributions                print("Connected to live debugserver or arm core. Will associate on-core threads to registers reported by server.")
735*33de042dSApple OSS Distributions            else:
736*33de042dSApple OSS Distributions                print("Instantiating threads completely from saved state in memory.")
737*33de042dSApple OSS Distributions
738*33de042dSApple OSS Distributions    def create_thread(self, tid, context):
739*33de042dSApple OSS Distributions
740*33de042dSApple OSS Distributions        # Strip TBI explicitly in case create_thread() is called externally.
741*33de042dSApple OSS Distributions        context = self._strip_ptr(context)
742*33de042dSApple OSS Distributions
743*33de042dSApple OSS Distributions        # tid == deadbeef means its a custom thread which kernel does not know of.
744*33de042dSApple OSS Distributions        if tid == 0xdeadbeef :
745*33de042dSApple OSS Distributions            # tid manipulation should be the same as in "switchtoregs" code in lldbmacros/process.py .
746*33de042dSApple OSS Distributions            tid = 0xdead0000 | (context & ~0xffff0000)
747*33de042dSApple OSS Distributions            tid = tid & 0xdeadffff
748*33de042dSApple OSS Distributions            thread_obj = { 'tid'   : tid,
749*33de042dSApple OSS Distributions                           'ptr'   : context,
750*33de042dSApple OSS Distributions                           'name'  : 'switchtoregs' + hex(context),
751*33de042dSApple OSS Distributions                           'queue' : 'None',
752*33de042dSApple OSS Distributions                           'state' : 'stopped',
753*33de042dSApple OSS Distributions                           'stop_reason' : 'none'
754*33de042dSApple OSS Distributions                         }
755*33de042dSApple OSS Distributions            self.thread_cache[tid] = thread_obj
756*33de042dSApple OSS Distributions            return thread_obj
757*33de042dSApple OSS Distributions
758*33de042dSApple OSS Distributions        th_ptr = context
759*33de042dSApple OSS Distributions        th = self.version.CreateValueFromExpression(str(th_ptr), '(struct thread *)' + str(th_ptr))
760*33de042dSApple OSS Distributions        thread_id = th.GetChildMemberWithName('thread_id').GetValueAsUnsigned()
761*33de042dSApple OSS Distributions        if tid != thread_id:
762*33de042dSApple OSS Distributions            print("FATAL ERROR: Creating thread from memory 0x%x with tid in mem=%d when requested tid = %d " % (context, thread_id, tid))
763*33de042dSApple OSS Distributions            return None
764*33de042dSApple OSS Distributions
765*33de042dSApple OSS Distributions        wait_queue = self._strip_ptr(th.GetChildMemberWithName('wait_queue').GetValueAsUnsigned())
766*33de042dSApple OSS Distributions        thread_obj = { 'tid'   : thread_id,
767*33de042dSApple OSS Distributions                       'ptr'   : th.GetValueAsUnsigned(),
768*33de042dSApple OSS Distributions                       'name'  : hex(th.GetValueAsUnsigned()).rstrip('L'),
769*33de042dSApple OSS Distributions                       'queue' : hex(wait_queue).rstrip('L'),
770*33de042dSApple OSS Distributions                       'state' : 'stopped',
771*33de042dSApple OSS Distributions                       'stop_reason' : 'none'
772*33de042dSApple OSS Distributions                     }
773*33de042dSApple OSS Distributions
774*33de042dSApple OSS Distributions        if self.current_session_id != GetUniqueSessionID(self.process):
775*33de042dSApple OSS Distributions            self.thread_cache = {}
776*33de042dSApple OSS Distributions            self.current_session_id = GetUniqueSessionID(self.process)
777*33de042dSApple OSS Distributions
778*33de042dSApple OSS Distributions        self.thread_cache[tid] = thread_obj
779*33de042dSApple OSS Distributions        return thread_obj
780*33de042dSApple OSS Distributions
781*33de042dSApple OSS Distributions    def get_thread_info(self):
782*33de042dSApple OSS Distributions        self.kdp_thread = None
783*33de042dSApple OSS Distributions        self.kdp_state = None
784*33de042dSApple OSS Distributions        if self.connected_over_kdp :
785*33de042dSApple OSS Distributions            kdp = self._target.FindGlobalVariables('kdp',1).GetValueAtIndex(0)
786*33de042dSApple OSS Distributions            kdp_state = kdp.GetChildMemberWithName('saved_state')
787*33de042dSApple OSS Distributions            kdp_thread = self._strip_thread_sbval(kdp.GetChildMemberWithName('kdp_thread'))
788*33de042dSApple OSS Distributions            if kdp_thread and kdp_thread.GetValueAsUnsigned() != 0:
789*33de042dSApple OSS Distributions                self.kdp_thread = kdp_thread
790*33de042dSApple OSS Distributions                self.kdp_state = kdp_state
791*33de042dSApple OSS Distributions                kdp_thid = kdp_thread.GetChildMemberWithName('thread_id').GetValueAsUnsigned()
792*33de042dSApple OSS Distributions                self.create_thread(kdp_thid, kdp_thread.GetValueAsUnsigned())
793*33de042dSApple OSS Distributions                self.thread_cache[kdp_thid]['core']=0
794*33de042dSApple OSS Distributions                retval = [self.thread_cache[kdp_thid]]
795*33de042dSApple OSS Distributions                return retval
796*33de042dSApple OSS Distributions            else:
797*33de042dSApple OSS Distributions                print("FATAL FAILURE: Unable to find kdp_thread state for this connection.")
798*33de042dSApple OSS Distributions                return []
799*33de042dSApple OSS Distributions
800*33de042dSApple OSS Distributions        num_threads = self._target.FindGlobalVariables('threads_count',1).GetValueAtIndex(0).GetValueAsUnsigned()
801*33de042dSApple OSS Distributions        #In case we are caught before threads are initialized. Fallback to threads known by astris/gdb server.
802*33de042dSApple OSS Distributions        if num_threads <=0 :
803*33de042dSApple OSS Distributions            return []
804*33de042dSApple OSS Distributions
805*33de042dSApple OSS Distributions        self.current_session_id = GetUniqueSessionID(self.process)
806*33de042dSApple OSS Distributions        self.threads = []
807*33de042dSApple OSS Distributions        self.thread_cache = {}
808*33de042dSApple OSS Distributions        self.processors = []
809*33de042dSApple OSS Distributions        try:
810*33de042dSApple OSS Distributions            processor_list_val = PluginValue(self._target.FindGlobalVariables('processor_list',1).GetValueAtIndex(0))
811*33de042dSApple OSS Distributions            while processor_list_val.IsValid() and processor_list_val.error.success and processor_list_val.GetValueAsUnsigned() !=0:
812*33de042dSApple OSS Distributions                th = self._strip_thread_sbval(processor_list_val.GetChildMemberWithName('active_thread'))
813*33de042dSApple OSS Distributions                th_id = th.GetChildMemberWithName('thread_id').GetValueAsUnsigned()
814*33de042dSApple OSS Distributions                cpu_id = processor_list_val.GetChildMemberWithName('cpu_id').GetValueAsUnsigned()
815*33de042dSApple OSS Distributions                self.processors.append({'active_thread': th.GetValueAsUnsigned(), 'cpu_id': cpu_id})
816*33de042dSApple OSS Distributions                self.create_thread(th_id, th.GetValueAsUnsigned())
817*33de042dSApple OSS Distributions                if self.connected_to_debugserver:
818*33de042dSApple OSS Distributions                    self.thread_cache[th_id]['core'] = cpu_id
819*33de042dSApple OSS Distributions                self.thread_cache[th_id]['queue'] = "cpu-%d" % int(cpu_id)
820*33de042dSApple OSS Distributions                nth = self.thread_cache[th_id]
821*33de042dSApple OSS Distributions                self.threads.append(nth)
822*33de042dSApple OSS Distributions                self.thread_cache[nth['tid']] = nth
823*33de042dSApple OSS Distributions                processor_list_val = processor_list_val.GetChildMemberWithName('processor_list')
824*33de042dSApple OSS Distributions        except KeyboardInterrupt as ke:
825*33de042dSApple OSS Distributions            print("OS Plugin Interrupted during thread loading process. \nWARNING:Thread registers and backtraces may not be accurate.")
826*33de042dSApple OSS Distributions            return self.threads
827*33de042dSApple OSS Distributions
828*33de042dSApple OSS Distributions        if hasattr(self.process, 'CreateOSPluginThread'):
829*33de042dSApple OSS Distributions            return self.threads
830*33de042dSApple OSS Distributions
831*33de042dSApple OSS Distributions        # FIXME remove legacy code
832*33de042dSApple OSS Distributions        try:
833*33de042dSApple OSS Distributions            thread_q_head = self._target.FindGlobalVariables('threads', 1).GetValueAtIndex(0)
834*33de042dSApple OSS Distributions            thread_type = self._target.FindFirstType('thread')
835*33de042dSApple OSS Distributions            thread_ptr_type = thread_type.GetPointerType()
836*33de042dSApple OSS Distributions            for th in IterateQueue(thread_q_head, thread_ptr_type, 'threads'):
837*33de042dSApple OSS Distributions                th = self._strip_thread_sbval(th)
838*33de042dSApple OSS Distributions                th_id = th.GetChildMemberWithName('thread_id').GetValueAsUnsigned()
839*33de042dSApple OSS Distributions                self.create_thread(th_id, th.GetValueAsUnsigned())
840*33de042dSApple OSS Distributions                nth = self.thread_cache[th_id]
841*33de042dSApple OSS Distributions                for cputhread in self.processors:
842*33de042dSApple OSS Distributions                    if cputhread['active_thread'] == nth['ptr']:
843*33de042dSApple OSS Distributions                        nth['core'] = cputhread['cpu_id']
844*33de042dSApple OSS Distributions                self.threads.append( nth )
845*33de042dSApple OSS Distributions        except KeyboardInterrupt as ke:
846*33de042dSApple OSS Distributions            print("OS Plugin Interrupted during thread loading process. \nWARNING:Thread registers and backtraces may not be accurate.")
847*33de042dSApple OSS Distributions            return self.threads
848*33de042dSApple OSS Distributions        # end legacy code
849*33de042dSApple OSS Distributions        return self.threads
850*33de042dSApple OSS Distributions
851*33de042dSApple OSS Distributions    def get_register_info(self):
852*33de042dSApple OSS Distributions        if self.registers == None:
853*33de042dSApple OSS Distributions            print("Register Information not found ")
854*33de042dSApple OSS Distributions        return self.register_set.register_info
855*33de042dSApple OSS Distributions
856*33de042dSApple OSS Distributions    def get_register_data(self, tid):
857*33de042dSApple OSS Distributions        thobj = None
858*33de042dSApple OSS Distributions        try:
859*33de042dSApple OSS Distributions            regs = self.register_set
860*33de042dSApple OSS Distributions            if self.current_session_id != GetUniqueSessionID(self.process):
861*33de042dSApple OSS Distributions                self.thread_cache = {}
862*33de042dSApple OSS Distributions                self.current_session_id = GetUniqueSessionID(self.process)
863*33de042dSApple OSS Distributions            if tid in self.thread_cache:
864*33de042dSApple OSS Distributions
865*33de042dSApple OSS Distributions                #Check if the thread is a fake one. Then create and return registers directly
866*33de042dSApple OSS Distributions                if self.thread_cache[tid]['name'].find('switchtoregs') == 0:
867*33de042dSApple OSS Distributions                    savedstateobj = self.version.CreateValueFromExpression(None, '(uintptr_t *) ' + str(self.thread_cache[tid]['ptr']))
868*33de042dSApple OSS Distributions                    regs.ReadRegisterDataFromKDPSavedState(savedstateobj, self.version)
869*33de042dSApple OSS Distributions                    return regs.GetPackedRegisterState()
870*33de042dSApple OSS Distributions
871*33de042dSApple OSS Distributions                thobj = self.version.CreateValueFromExpression(self.thread_cache[tid]['name'], '(struct thread *)' + str(self.thread_cache[tid]['ptr']))
872*33de042dSApple OSS Distributions
873*33de042dSApple OSS Distributions            if thobj == None :
874*33de042dSApple OSS Distributions                print("FATAL ERROR: Could not find thread with id %d" % tid)
875*33de042dSApple OSS Distributions                regs.ResetRegisterValues()
876*33de042dSApple OSS Distributions                return regs.GetPackedRegisterState()
877*33de042dSApple OSS Distributions
878*33de042dSApple OSS Distributions            if self.kdp_thread and self.kdp_thread.GetValueAsUnsigned() == thobj.GetValueAsUnsigned():
879*33de042dSApple OSS Distributions                regs.ReadRegisterDataFromKDPSavedState(self.kdp_state, self.version)
880*33de042dSApple OSS Distributions                return regs.GetPackedRegisterState()
881*33de042dSApple OSS Distributions            if int(PluginValue(thobj).GetChildMemberWithName('kernel_stack').GetValueAsUnsigned()) != 0 :
882*33de042dSApple OSS Distributions                if self.target_arch == archX86_64 :
883*33de042dSApple OSS Distributions                    # we do have a stack so lets get register information
884*33de042dSApple OSS Distributions                    saved_state_addr = PluginValue(thobj).GetChildMemberWithName('kernel_stack').GetValueAsUnsigned() + self.kernel_stack_size - self.kernel_thread_state_size
885*33de042dSApple OSS Distributions                    regs.ReadRegisterDataFromKernelStack(saved_state_addr, self.version)
886*33de042dSApple OSS Distributions                    return regs.GetPackedRegisterState()
887*33de042dSApple OSS Distributions                elif self.target_arch.startswith(archARMv7) and int(PluginValue(thobj).GetChildMemberWithName('machine').GetChildMemberWithName('kstackptr').GetValueAsUnsigned()) != 0:
888*33de042dSApple OSS Distributions                    #we have stack on the machine.kstackptr.
889*33de042dSApple OSS Distributions                    saved_state_addr = PluginValue(thobj).GetChildMemberWithName('machine').GetChildMemberWithName('kstackptr').GetValueAsUnsigned()
890*33de042dSApple OSS Distributions                    regs.ReadRegisterDataFromKernelStack(saved_state_addr, self.version)
891*33de042dSApple OSS Distributions                    return regs.GetPackedRegisterState()
892*33de042dSApple OSS Distributions                elif self.target_arch.startswith(archARMv8) and int(PluginValue(thobj).GetChildMemberWithName('machine').GetChildMemberWithName('kstackptr').GetValueAsUnsigned()) != 0:
893*33de042dSApple OSS Distributions                    saved_state_addr = PluginValue(thobj).GetChildMemberWithName('machine').GetChildMemberWithName('kstackptr').GetValueAsAddress()
894*33de042dSApple OSS Distributions                    arm_ctx = PluginValue(self.version.CreateValueFromExpression(None, '(struct arm_kernel_context *) ' + str(saved_state_addr)))
895*33de042dSApple OSS Distributions                    arm_ss_addr = arm_ctx.GetChildMemberWithName('ss').GetLoadAddress()
896*33de042dSApple OSS Distributions                    regs.ReadRegisterDataFromKernelStack(arm_ss_addr, self.version)
897*33de042dSApple OSS Distributions                    return regs.GetPackedRegisterState()
898*33de042dSApple OSS Distributions            elif self.target_arch == archX86_64 or self.target_arch.startswith(archARMv7) or self.target_arch.startswith(archARMv8):
899*33de042dSApple OSS Distributions                regs.ReadRegisterDataFromContinuation( PluginValue(thobj).GetChildMemberWithName('continuation').GetValueAsAddress())
900*33de042dSApple OSS Distributions                return regs.GetPackedRegisterState()
901*33de042dSApple OSS Distributions            #incase we failed very miserably
902*33de042dSApple OSS Distributions        except KeyboardInterrupt as ke:
903*33de042dSApple OSS Distributions            print("OS Plugin Interrupted during thread register load. \nWARNING:Thread registers and backtraces may not be accurate. for tid = %d" % tid)
904*33de042dSApple OSS Distributions        regs.ResetRegisterValues()
905*33de042dSApple OSS Distributions        print("FATAL ERROR: Failed to get register state for thread id 0x%x " % tid)
906*33de042dSApple OSS Distributions        print(thobj)
907*33de042dSApple OSS Distributions        return regs.GetPackedRegisterState()
908