266 lines · python
1""" This module implement Dwarf expression opcode parser. """2 3import lldb4 5# DWARF Expression operators.6DW_OP_addr = 0x037DW_OP_deref = 0x068DW_OP_const1u = 0x089DW_OP_const1s = 0x0910DW_OP_const2u = 0x0A11DW_OP_const2s = 0x0B12DW_OP_const4u = 0x0C13DW_OP_const4s = 0x0D14DW_OP_const8u = 0x0E15DW_OP_const8s = 0x0F16DW_OP_constu = 0x1017DW_OP_consts = 0x1118DW_OP_dup = 0x1219DW_OP_drop = 0x1320DW_OP_over = 0x1421DW_OP_pick = 0x1522DW_OP_swap = 0x1623DW_OP_rot = 0x1724DW_OP_xderef = 0x1825DW_OP_abs = 0x1926DW_OP_and = 0x1A27DW_OP_div = 0x1B28DW_OP_minus = 0x1C29DW_OP_mod = 0x1D30DW_OP_mul = 0x1E31DW_OP_neg = 0x1F32DW_OP_not = 0x2033DW_OP_or = 0x2134DW_OP_plus = 0x2235DW_OP_plus_uconst = 0x2336DW_OP_shl = 0x2437DW_OP_shr = 0x2538DW_OP_shra = 0x2639DW_OP_xor = 0x2740DW_OP_skip = 0x2F41DW_OP_bra = 0x2842DW_OP_eq = 0x2943DW_OP_ge = 0x2A44DW_OP_gt = 0x2B45DW_OP_le = 0x2C46DW_OP_lt = 0x2D47DW_OP_ne = 0x2E48DW_OP_lit0 = 0x3049DW_OP_lit1 = 0x3150DW_OP_lit2 = 0x3251DW_OP_lit3 = 0x3352DW_OP_lit4 = 0x3453DW_OP_lit5 = 0x3554DW_OP_lit6 = 0x3655DW_OP_lit7 = 0x3756DW_OP_lit8 = 0x3857DW_OP_lit9 = 0x3958DW_OP_lit10 = 0x3A59DW_OP_lit11 = 0x3B60DW_OP_lit12 = 0x3C61DW_OP_lit13 = 0x3D62DW_OP_lit14 = 0x3E63DW_OP_lit15 = 0x3F64DW_OP_lit16 = 0x4065DW_OP_lit17 = 0x4166DW_OP_lit18 = 0x4267DW_OP_lit19 = 0x4368DW_OP_lit20 = 0x4469DW_OP_lit21 = 0x4570DW_OP_lit22 = 0x4671DW_OP_lit23 = 0x4772DW_OP_lit24 = 0x4873DW_OP_lit25 = 0x4974DW_OP_lit26 = 0x4A75DW_OP_lit27 = 0x4B76DW_OP_lit28 = 0x4C77DW_OP_lit29 = 0x4D78DW_OP_lit30 = 0x4E79DW_OP_lit31 = 0x4F80DW_OP_reg0 = 0x5081DW_OP_reg1 = 0x5182DW_OP_reg2 = 0x5283DW_OP_reg3 = 0x5384DW_OP_reg4 = 0x5485DW_OP_reg5 = 0x5586DW_OP_reg6 = 0x5687DW_OP_reg7 = 0x5788DW_OP_reg8 = 0x5889DW_OP_reg9 = 0x5990DW_OP_reg10 = 0x5A91DW_OP_reg11 = 0x5B92DW_OP_reg12 = 0x5C93DW_OP_reg13 = 0x5D94DW_OP_reg14 = 0x5E95DW_OP_reg15 = 0x5F96DW_OP_reg16 = 0x6097DW_OP_reg17 = 0x6198DW_OP_reg18 = 0x6299DW_OP_reg19 = 0x63100DW_OP_reg20 = 0x64101DW_OP_reg21 = 0x65102DW_OP_reg22 = 0x66103DW_OP_reg23 = 0x67104DW_OP_reg24 = 0x68105DW_OP_reg25 = 0x69106DW_OP_reg26 = 0x6A107DW_OP_reg27 = 0x6B108DW_OP_reg28 = 0x6C109DW_OP_reg29 = 0x6D110DW_OP_reg30 = 0x6E111DW_OP_reg31 = 0x6F112DW_OP_breg0 = 0x70113DW_OP_breg1 = 0x71114DW_OP_breg2 = 0x72115DW_OP_breg3 = 0x73116DW_OP_breg4 = 0x74117DW_OP_breg5 = 0x75118DW_OP_breg6 = 0x76119DW_OP_breg7 = 0x77120DW_OP_breg8 = 0x78121DW_OP_breg9 = 0x79122DW_OP_breg10 = 0x7A123DW_OP_breg11 = 0x7B124DW_OP_breg12 = 0x7C125DW_OP_breg13 = 0x7D126DW_OP_breg14 = 0x7E127DW_OP_breg15 = 0x7F128DW_OP_breg16 = 0x80129DW_OP_breg17 = 0x81130DW_OP_breg18 = 0x82131DW_OP_breg19 = 0x83132DW_OP_breg20 = 0x84133DW_OP_breg21 = 0x85134DW_OP_breg22 = 0x86135DW_OP_breg23 = 0x87136DW_OP_breg24 = 0x88137DW_OP_breg25 = 0x89138DW_OP_breg26 = 0x8A139DW_OP_breg27 = 0x8B140DW_OP_breg28 = 0x8C141DW_OP_breg29 = 0x8D142DW_OP_breg30 = 0x8E143DW_OP_breg31 = 0x8F144DW_OP_regx = 0x90145DW_OP_fbreg = 0x91146DW_OP_bregx = 0x92147DW_OP_piece = 0x93148DW_OP_deref_size = 0x94149DW_OP_xderef_size = 0x95150DW_OP_nop = 0x96151DW_OP_push_object_address = 0x97152DW_OP_call2 = 0x98153DW_OP_call4 = 0x99154DW_OP_call_ref = 0x9A155DW_OP_form_tls_address = 0x9B156DW_OP_call_frame_cfa = 0x9C157DW_OP_bit_piece = 0x9D158DW_OP_implicit_value = 0x9E159DW_OP_stack_value = 0x9F160DW_OP_lo_user = 0xE0161DW_OP_GNU_push_tls_address = 0xE0162DW_OP_hi_user = 0xFF163 164 165class DwarfOpcodeParser(object):166 def updateRegInfoBitsize(self, reg_info, byte_order):167 """Update the regInfo bit size."""168 169 # Evaluate Dwarf Expression170 expr_result = self.evaluateDwarfExpression(171 reg_info["dynamic_size_dwarf_expr_bytes"], byte_order172 )173 174 if expr_result == 0:175 reg_info["bitsize"] = 32176 elif expr_result == 1:177 reg_info["bitsize"] = 64178 179 def evaluateDwarfExpression(self, dwarf_opcode, byte_order):180 """Evaluate Dwarf Expression."""181 182 dwarf_opcode = [dwarf_opcode[i : i + 2] for i in range(0, len(dwarf_opcode), 2)]183 dwarf_data = []184 for index in range(len(dwarf_opcode)):185 if index < len(dwarf_opcode):186 val = int(dwarf_opcode[index], 16)187 else:188 break189 190 if val == DW_OP_regx:191 # Read register number192 self.assertTrue(len(dwarf_opcode) > (index + 1))193 reg_no = int(dwarf_opcode.pop(index + 1), 16)194 195 self.reset_test_sequence()196 # Read register value197 self.test_sequence.add_log_lines(198 [199 "read packet: $p{0:x}#00".format(reg_no),200 {201 "direction": "send",202 "regex": r"^\$([0-9a-fA-F]+)#",203 "capture": {1: "p_response"},204 },205 ],206 True,207 )208 209 Context = self.expect_gdbremote_sequence()210 self.assertIsNotNone(Context)211 p_response = Context.get("p_response")212 self.assertIsNotNone(p_response)213 214 if byte_order == lldb.eByteOrderLittle:215 # In case of little endian216 # first decode the HEX ASCII bytes and then reverse it217 # to get actual value of SR register218 p_response = "".join(219 reversed(220 [221 p_response[i : i + 2]222 for i in range(0, len(p_response), 2)223 ]224 )225 )226 # Push register value227 dwarf_data.append(int(p_response, 16))228 229 elif val == DW_OP_lit1:230 # Push literal 1231 dwarf_data.append(1)232 233 elif val == DW_OP_lit26:234 # Push literal 26235 dwarf_data.append(26)236 237 elif val == DW_OP_shl:238 # left shift and push the result back239 self.assertTrue(len(dwarf_data) > 1)240 shift_amount = dwarf_data.pop()241 val_to_shift = dwarf_data.pop()242 result = val_to_shift << shift_amount243 dwarf_data.append(result)244 245 elif val == DW_OP_shr:246 # Right shift and push the result back247 self.assertTrue(len(dwarf_data) > 1)248 shift_amount = dwarf_data.pop()249 val_to_shift = dwarf_data.pop()250 result = val_to_shift >> shift_amount251 dwarf_data.append(result)252 253 elif val == DW_OP_and:254 # And of topmost 2 elements and push the result back255 first_ele = dwarf_data.pop()256 second_ele = dwarf_data.pop()257 result = first_ele & second_ele258 dwarf_data.append(result)259 260 else:261 self.assertTrue(False and "Unprocess Dwarf Opcode")262 263 self.assertTrue(len(dwarf_data) == 1)264 expr_result = dwarf_data.pop()265 return expr_result266