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