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