diff options
| -rw-r--r-- | compiler/optimizing/stack_map_stream.cc | 58 | ||||
| -rw-r--r-- | compiler/optimizing/stack_map_stream.h | 3 | ||||
| -rw-r--r-- | compiler/optimizing/stack_map_test.cc | 74 | ||||
| -rw-r--r-- | runtime/entrypoints/entrypoint_utils-inl.h | 10 | ||||
| -rw-r--r-- | runtime/entrypoints/entrypoint_utils.cc | 5 | ||||
| -rw-r--r-- | runtime/entrypoints/quick/quick_trampoline_entrypoints.cc | 3 | ||||
| -rw-r--r-- | runtime/oat.h | 2 | ||||
| -rw-r--r-- | runtime/stack.cc | 19 | ||||
| -rw-r--r-- | runtime/stack_map.cc | 30 | ||||
| -rw-r--r-- | runtime/stack_map.h | 195 |
10 files changed, 252 insertions, 147 deletions
diff --git a/compiler/optimizing/stack_map_stream.cc b/compiler/optimizing/stack_map_stream.cc index c571312faa..11a254ef63 100644 --- a/compiler/optimizing/stack_map_stream.cc +++ b/compiler/optimizing/stack_map_stream.cc @@ -137,8 +137,9 @@ uint32_t StackMapStream::ComputeMaxNativePcOffset() const { size_t StackMapStream::PrepareForFillIn() { int stack_mask_number_of_bits = stack_mask_max_ + 1; // Need room for max element too. - inline_info_size_ = ComputeInlineInfoSize(); dex_register_maps_size_ = ComputeDexRegisterMapsSize(); + ComputeInlineInfoEncoding(); // needs dex_register_maps_size_. + inline_info_size_ = inline_infos_.size() * inline_info_encoding_.GetEntrySize(); uint32_t max_native_pc_offset = ComputeMaxNativePcOffset(); size_t stack_map_size = stack_map_encoding_.SetFromSizes(max_native_pc_offset, dex_pc_max_, @@ -158,9 +159,10 @@ size_t StackMapStream::PrepareForFillIn() { // Prepare the CodeInfo variable-sized encoding. CodeInfoEncoding code_info_encoding; code_info_encoding.non_header_size = non_header_size; - code_info_encoding.stack_map_encoding = stack_map_encoding_; code_info_encoding.number_of_stack_maps = stack_maps_.size(); code_info_encoding.stack_map_size_in_bytes = stack_map_size; + code_info_encoding.stack_map_encoding = stack_map_encoding_; + code_info_encoding.inline_info_encoding = inline_info_encoding_; code_info_encoding.number_of_location_catalog_entries = location_catalog_entries_.size(); code_info_encoding.Compress(&code_info_encoding_); @@ -224,10 +226,26 @@ size_t StackMapStream::ComputeDexRegisterMapsSize() const { return size; } -size_t StackMapStream::ComputeInlineInfoSize() const { - return inline_infos_.size() * InlineInfo::SingleEntrySize() - // For encoding the depth. - + (number_of_stack_maps_with_inline_info_ * InlineInfo::kFixedSize); +void StackMapStream::ComputeInlineInfoEncoding() { + uint32_t method_index_max = 0; + uint32_t dex_pc_max = 0; + uint32_t invoke_type_max = 0; + + uint32_t inline_info_index = 0; + for (const StackMapEntry& entry : stack_maps_) { + for (size_t j = 0; j < entry.inlining_depth; ++j) { + InlineInfoEntry inline_entry = inline_infos_[inline_info_index++]; + method_index_max = std::max(method_index_max, inline_entry.method_index); + dex_pc_max = std::max(dex_pc_max, inline_entry.dex_pc); + invoke_type_max = std::max(invoke_type_max, static_cast<uint32_t>(inline_entry.invoke_type)); + } + } + DCHECK_EQ(inline_info_index, inline_infos_.size()); + + inline_info_encoding_.SetFromSizes(method_index_max, + dex_pc_max, + invoke_type_max, + dex_register_maps_size_); } void StackMapStream::FillIn(MemoryRegion region) { @@ -321,7 +339,7 @@ void StackMapStream::FillIn(MemoryRegion region) { if (entry.inlining_depth != 0) { MemoryRegion inline_region = inline_infos_region.Subregion( next_inline_info_offset, - InlineInfo::kFixedSize + entry.inlining_depth * InlineInfo::SingleEntrySize()); + entry.inlining_depth * inline_info_encoding_.GetEntrySize()); next_inline_info_offset += inline_region.size(); InlineInfo inline_info(inline_region); @@ -329,16 +347,18 @@ void StackMapStream::FillIn(MemoryRegion region) { stack_map.SetInlineDescriptorOffset( stack_map_encoding_, inline_region.start() - dex_register_locations_region.start()); - inline_info.SetDepth(entry.inlining_depth); + inline_info.SetDepth(inline_info_encoding_, entry.inlining_depth); DCHECK_LE(entry.inline_infos_start_index + entry.inlining_depth, inline_infos_.size()); for (size_t depth = 0; depth < entry.inlining_depth; ++depth) { InlineInfoEntry inline_entry = inline_infos_[depth + entry.inline_infos_start_index]; - inline_info.SetMethodIndexAtDepth(depth, inline_entry.method_index); - inline_info.SetDexPcAtDepth(depth, inline_entry.dex_pc); - inline_info.SetInvokeTypeAtDepth(depth, inline_entry.invoke_type); + inline_info.SetMethodIndexAtDepth(inline_info_encoding_, depth, inline_entry.method_index); + inline_info.SetDexPcAtDepth(inline_info_encoding_, depth, inline_entry.dex_pc); + inline_info.SetInvokeTypeAtDepth(inline_info_encoding_, depth, inline_entry.invoke_type); if (inline_entry.num_dex_registers == 0) { // No dex map available. - inline_info.SetDexRegisterMapOffsetAtDepth(depth, StackMap::kNoDexRegisterMap); + inline_info.SetDexRegisterMapOffsetAtDepth(inline_info_encoding_, + depth, + StackMap::kNoDexRegisterMap); DCHECK(inline_entry.live_dex_registers_mask == nullptr); } else { MemoryRegion register_region = dex_register_locations_region.Subregion( @@ -348,7 +368,8 @@ void StackMapStream::FillIn(MemoryRegion region) { next_dex_register_map_offset += register_region.size(); DexRegisterMap dex_register_map(register_region); inline_info.SetDexRegisterMapOffsetAtDepth( - depth, register_region.start() - dex_register_locations_region.start()); + inline_info_encoding_, + depth, register_region.start() - dex_register_locations_region.start()); FillInDexRegisterMap(dex_register_map, inline_entry.num_dex_registers, @@ -513,14 +534,17 @@ void StackMapStream::CheckCodeInfo(MemoryRegion region) const { DCHECK_EQ(stack_map.HasInlineInfo(stack_map_encoding), (entry.inlining_depth != 0)); if (entry.inlining_depth != 0) { InlineInfo inline_info = code_info.GetInlineInfoOf(stack_map, encoding); - DCHECK_EQ(inline_info.GetDepth(), entry.inlining_depth); + DCHECK_EQ(inline_info.GetDepth(encoding.inline_info_encoding), entry.inlining_depth); for (size_t d = 0; d < entry.inlining_depth; ++d) { size_t inline_info_index = entry.inline_infos_start_index + d; DCHECK_LT(inline_info_index, inline_infos_.size()); InlineInfoEntry inline_entry = inline_infos_[inline_info_index]; - DCHECK_EQ(inline_info.GetDexPcAtDepth(d), inline_entry.dex_pc); - DCHECK_EQ(inline_info.GetMethodIndexAtDepth(d), inline_entry.method_index); - DCHECK_EQ(inline_info.GetInvokeTypeAtDepth(d), inline_entry.invoke_type); + DCHECK_EQ(inline_info.GetDexPcAtDepth(encoding.inline_info_encoding, d), + inline_entry.dex_pc); + DCHECK_EQ(inline_info.GetMethodIndexAtDepth(encoding.inline_info_encoding, d), + inline_entry.method_index); + DCHECK_EQ(inline_info.GetInvokeTypeAtDepth(encoding.inline_info_encoding, d), + inline_entry.invoke_type); CheckDexRegisterMap(code_info, code_info.GetDexRegisterMapAtDepth( diff --git a/compiler/optimizing/stack_map_stream.h b/compiler/optimizing/stack_map_stream.h index b686748802..41f72f508b 100644 --- a/compiler/optimizing/stack_map_stream.h +++ b/compiler/optimizing/stack_map_stream.h @@ -156,7 +156,7 @@ class StackMapStream : public ValueObject { size_t ComputeDexRegisterMapSize(uint32_t num_dex_registers, const BitVector* live_dex_registers_mask) const; size_t ComputeDexRegisterMapsSize() const; - size_t ComputeInlineInfoSize() const; + void ComputeInlineInfoEncoding(); // Returns the index of an entry with the same dex register map as the current_entry, // or kNoSameDexMapFound if no such entry exists. @@ -200,6 +200,7 @@ class StackMapStream : public ValueObject { StackMapEntry current_entry_; InlineInfoEntry current_inline_info_; StackMapEncoding stack_map_encoding_; + InlineInfoEncoding inline_info_encoding_; ArenaVector<uint8_t> code_info_encoding_; size_t inline_info_size_; size_t dex_register_maps_size_; diff --git a/compiler/optimizing/stack_map_test.cc b/compiler/optimizing/stack_map_test.cc index 35524877e3..967fd96561 100644 --- a/compiler/optimizing/stack_map_test.cc +++ b/compiler/optimizing/stack_map_test.cc @@ -237,13 +237,13 @@ TEST(StackMapTest, Test2) { ASSERT_TRUE(stack_map.HasInlineInfo(encoding.stack_map_encoding)); InlineInfo inline_info = code_info.GetInlineInfoOf(stack_map, encoding); - ASSERT_EQ(2u, inline_info.GetDepth()); - ASSERT_EQ(82u, inline_info.GetMethodIndexAtDepth(0)); - ASSERT_EQ(42u, inline_info.GetMethodIndexAtDepth(1)); - ASSERT_EQ(3u, inline_info.GetDexPcAtDepth(0)); - ASSERT_EQ(2u, inline_info.GetDexPcAtDepth(1)); - ASSERT_EQ(kDirect, inline_info.GetInvokeTypeAtDepth(0)); - ASSERT_EQ(kStatic, inline_info.GetInvokeTypeAtDepth(1)); + ASSERT_EQ(2u, inline_info.GetDepth(encoding.inline_info_encoding)); + ASSERT_EQ(82u, inline_info.GetMethodIndexAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(42u, inline_info.GetMethodIndexAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(3u, inline_info.GetDexPcAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(2u, inline_info.GetDexPcAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(kDirect, inline_info.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(kStatic, inline_info.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 1)); } // Second stack map. @@ -741,13 +741,13 @@ TEST(StackMapTest, InlineTest) { ASSERT_EQ(4, dex_registers0.GetConstant(1, 2, ci, encoding)); InlineInfo if0 = ci.GetInlineInfoOf(sm0, encoding); - ASSERT_EQ(2u, if0.GetDepth()); - ASSERT_EQ(2u, if0.GetDexPcAtDepth(0)); - ASSERT_EQ(42u, if0.GetMethodIndexAtDepth(0)); - ASSERT_EQ(kStatic, if0.GetInvokeTypeAtDepth(0)); - ASSERT_EQ(3u, if0.GetDexPcAtDepth(1)); - ASSERT_EQ(82u, if0.GetMethodIndexAtDepth(1)); - ASSERT_EQ(kStatic, if0.GetInvokeTypeAtDepth(1)); + ASSERT_EQ(2u, if0.GetDepth(encoding.inline_info_encoding)); + ASSERT_EQ(2u, if0.GetDexPcAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(42u, if0.GetMethodIndexAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(kStatic, if0.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(3u, if0.GetDexPcAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(82u, if0.GetMethodIndexAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(kStatic, if0.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 1)); DexRegisterMap dex_registers1 = ci.GetDexRegisterMapAtDepth(0, if0, encoding, 1); ASSERT_EQ(8, dex_registers1.GetStackOffsetInBytes(0, 1, ci, encoding)); @@ -767,16 +767,16 @@ TEST(StackMapTest, InlineTest) { ASSERT_EQ(0, dex_registers0.GetConstant(1, 2, ci, encoding)); InlineInfo if1 = ci.GetInlineInfoOf(sm1, encoding); - ASSERT_EQ(3u, if1.GetDepth()); - ASSERT_EQ(2u, if1.GetDexPcAtDepth(0)); - ASSERT_EQ(42u, if1.GetMethodIndexAtDepth(0)); - ASSERT_EQ(kDirect, if1.GetInvokeTypeAtDepth(0)); - ASSERT_EQ(3u, if1.GetDexPcAtDepth(1)); - ASSERT_EQ(82u, if1.GetMethodIndexAtDepth(1)); - ASSERT_EQ(kStatic, if1.GetInvokeTypeAtDepth(1)); - ASSERT_EQ(5u, if1.GetDexPcAtDepth(2)); - ASSERT_EQ(52u, if1.GetMethodIndexAtDepth(2)); - ASSERT_EQ(kVirtual, if1.GetInvokeTypeAtDepth(2)); + ASSERT_EQ(3u, if1.GetDepth(encoding.inline_info_encoding)); + ASSERT_EQ(2u, if1.GetDexPcAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(42u, if1.GetMethodIndexAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(kDirect, if1.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(3u, if1.GetDexPcAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(82u, if1.GetMethodIndexAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(kStatic, if1.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(5u, if1.GetDexPcAtDepth(encoding.inline_info_encoding, 2)); + ASSERT_EQ(52u, if1.GetMethodIndexAtDepth(encoding.inline_info_encoding, 2)); + ASSERT_EQ(kVirtual, if1.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 2)); DexRegisterMap dex_registers1 = ci.GetDexRegisterMapAtDepth(0, if1, encoding, 1); ASSERT_EQ(12, dex_registers1.GetStackOffsetInBytes(0, 1, ci, encoding)); @@ -786,7 +786,7 @@ TEST(StackMapTest, InlineTest) { ASSERT_EQ(10, dex_registers2.GetConstant(1, 3, ci, encoding)); ASSERT_EQ(5, dex_registers2.GetMachineRegister(2, 3, ci, encoding)); - ASSERT_FALSE(if1.HasDexRegisterMapAtDepth(2)); + ASSERT_FALSE(if1.HasDexRegisterMapAtDepth(encoding.inline_info_encoding, 2)); } { @@ -808,18 +808,18 @@ TEST(StackMapTest, InlineTest) { ASSERT_EQ(0, dex_registers0.GetConstant(1, 2, ci, encoding)); InlineInfo if2 = ci.GetInlineInfoOf(sm3, encoding); - ASSERT_EQ(3u, if2.GetDepth()); - ASSERT_EQ(2u, if2.GetDexPcAtDepth(0)); - ASSERT_EQ(42u, if2.GetMethodIndexAtDepth(0)); - ASSERT_EQ(kVirtual, if2.GetInvokeTypeAtDepth(0)); - ASSERT_EQ(5u, if2.GetDexPcAtDepth(1)); - ASSERT_EQ(52u, if2.GetMethodIndexAtDepth(1)); - ASSERT_EQ(kInterface, if2.GetInvokeTypeAtDepth(1)); - ASSERT_EQ(10u, if2.GetDexPcAtDepth(2)); - ASSERT_EQ(52u, if2.GetMethodIndexAtDepth(2)); - ASSERT_EQ(kStatic, if2.GetInvokeTypeAtDepth(2)); - - ASSERT_FALSE(if2.HasDexRegisterMapAtDepth(0)); + ASSERT_EQ(3u, if2.GetDepth(encoding.inline_info_encoding)); + ASSERT_EQ(2u, if2.GetDexPcAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(42u, if2.GetMethodIndexAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(kVirtual, if2.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 0)); + ASSERT_EQ(5u, if2.GetDexPcAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(52u, if2.GetMethodIndexAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(kInterface, if2.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 1)); + ASSERT_EQ(10u, if2.GetDexPcAtDepth(encoding.inline_info_encoding, 2)); + ASSERT_EQ(52u, if2.GetMethodIndexAtDepth(encoding.inline_info_encoding, 2)); + ASSERT_EQ(kStatic, if2.GetInvokeTypeAtDepth(encoding.inline_info_encoding, 2)); + + ASSERT_FALSE(if2.HasDexRegisterMapAtDepth(encoding.inline_info_encoding, 0)); DexRegisterMap dex_registers1 = ci.GetDexRegisterMapAtDepth(1, if2, encoding, 1); ASSERT_EQ(2, dex_registers1.GetMachineRegister(0, 1, ci, encoding)); diff --git a/runtime/entrypoints/entrypoint_utils-inl.h b/runtime/entrypoints/entrypoint_utils-inl.h index 116261b32b..16fbfaad32 100644 --- a/runtime/entrypoints/entrypoint_utils-inl.h +++ b/runtime/entrypoints/entrypoint_utils-inl.h @@ -42,11 +42,12 @@ namespace art { template <bool kResolve = true> inline ArtMethod* GetResolvedMethod(ArtMethod* outer_method, const InlineInfo& inline_info, + const InlineInfoEncoding& encoding, uint8_t inlining_depth) SHARED_REQUIRES(Locks::mutator_lock_) { - uint32_t method_index = inline_info.GetMethodIndexAtDepth(inlining_depth); + uint32_t method_index = inline_info.GetMethodIndexAtDepth(encoding, inlining_depth); InvokeType invoke_type = static_cast<InvokeType>( - inline_info.GetInvokeTypeAtDepth(inlining_depth)); + inline_info.GetInvokeTypeAtDepth(encoding, inlining_depth)); ArtMethod* caller = outer_method->GetDexCacheResolvedMethod(method_index, sizeof(void*)); if (!caller->IsRuntimeMethod()) { return caller; @@ -68,7 +69,10 @@ inline ArtMethod* GetResolvedMethod(ArtMethod* outer_method, if (inlining_depth == 0) { class_loader.Assign(outer_method->GetClassLoader()); } else { - caller = GetResolvedMethod<kResolve>(outer_method, inline_info, inlining_depth - 1); + caller = GetResolvedMethod<kResolve>(outer_method, + inline_info, + encoding, + inlining_depth - 1); class_loader.Assign(caller->GetClassLoader()); } diff --git a/runtime/entrypoints/entrypoint_utils.cc b/runtime/entrypoints/entrypoint_utils.cc index 336841145b..e46576e884 100644 --- a/runtime/entrypoints/entrypoint_utils.cc +++ b/runtime/entrypoints/entrypoint_utils.cc @@ -280,7 +280,10 @@ ArtMethod* GetCalleeSaveMethodCaller(ArtMethod** sp, DCHECK(stack_map.IsValid()); if (stack_map.HasInlineInfo(encoding.stack_map_encoding)) { InlineInfo inline_info = code_info.GetInlineInfoOf(stack_map, encoding); - caller = GetResolvedMethod(outer_method, inline_info, inline_info.GetDepth() - 1); + caller = GetResolvedMethod(outer_method, + inline_info, + encoding.inline_info_encoding, + inline_info.GetDepth(encoding.inline_info_encoding) - 1); } } } diff --git a/runtime/entrypoints/quick/quick_trampoline_entrypoints.cc b/runtime/entrypoints/quick/quick_trampoline_entrypoints.cc index 1a96d0fc61..f3e8dbadbe 100644 --- a/runtime/entrypoints/quick/quick_trampoline_entrypoints.cc +++ b/runtime/entrypoints/quick/quick_trampoline_entrypoints.cc @@ -332,7 +332,8 @@ class QuickArgumentVisitor { DCHECK(stack_map.IsValid()); if (stack_map.HasInlineInfo(encoding.stack_map_encoding)) { InlineInfo inline_info = code_info.GetInlineInfoOf(stack_map, encoding); - return inline_info.GetDexPcAtDepth(inline_info.GetDepth() - 1); + return inline_info.GetDexPcAtDepth(encoding.inline_info_encoding, + inline_info.GetDepth(encoding.inline_info_encoding)-1); } else { return stack_map.GetDexPc(encoding.stack_map_encoding); } diff --git a/runtime/oat.h b/runtime/oat.h index 68e71c450b..469a65f2de 100644 --- a/runtime/oat.h +++ b/runtime/oat.h @@ -32,7 +32,7 @@ class InstructionSetFeatures; class PACKED(4) OatHeader { public: static constexpr uint8_t kOatMagic[] = { 'o', 'a', 't', '\n' }; - static constexpr uint8_t kOatVersion[] = { '0', '7', '7', '\0' }; + static constexpr uint8_t kOatVersion[] = { '0', '7', '8', '\0' }; static constexpr const char* kImageLocationKey = "image-location"; static constexpr const char* kDex2OatCmdLineKey = "dex2oat-cmdline"; diff --git a/runtime/stack.cc b/runtime/stack.cc index 23363654d2..c22eb92f54 100644 --- a/runtime/stack.cc +++ b/runtime/stack.cc @@ -130,11 +130,19 @@ ArtMethod* StackVisitor::GetMethod() const { if (IsInInlinedFrame()) { size_t depth_in_stack_map = current_inlining_depth_ - 1; InlineInfo inline_info = GetCurrentInlineInfo(); + const OatQuickMethodHeader* method_header = GetCurrentOatQuickMethodHeader(); + CodeInfoEncoding encoding = method_header->GetOptimizedCodeInfo().ExtractEncoding(); DCHECK(walk_kind_ != StackWalkKind::kSkipInlinedFrames); bool allow_resolve = walk_kind_ != StackWalkKind::kIncludeInlinedFramesNoResolve; return allow_resolve - ? GetResolvedMethod<true>(*GetCurrentQuickFrame(), inline_info, depth_in_stack_map) - : GetResolvedMethod<false>(*GetCurrentQuickFrame(), inline_info, depth_in_stack_map); + ? GetResolvedMethod<true>(*GetCurrentQuickFrame(), + inline_info, + encoding.inline_info_encoding, + depth_in_stack_map) + : GetResolvedMethod<false>(*GetCurrentQuickFrame(), + inline_info, + encoding.inline_info_encoding, + depth_in_stack_map); } else { return *cur_quick_frame_; } @@ -148,7 +156,10 @@ uint32_t StackVisitor::GetDexPc(bool abort_on_failure) const { } else if (cur_quick_frame_ != nullptr) { if (IsInInlinedFrame()) { size_t depth_in_stack_map = current_inlining_depth_ - 1; - return GetCurrentInlineInfo().GetDexPcAtDepth(depth_in_stack_map); + const OatQuickMethodHeader* method_header = GetCurrentOatQuickMethodHeader(); + CodeInfoEncoding encoding = method_header->GetOptimizedCodeInfo().ExtractEncoding(); + return GetCurrentInlineInfo().GetDexPcAtDepth(encoding.inline_info_encoding, + depth_in_stack_map); } else if (cur_oat_quick_method_header_ == nullptr) { return DexFile::kDexNoIndex; } else { @@ -875,7 +886,7 @@ void StackVisitor::WalkStack(bool include_transitions) { if (stack_map.IsValid() && stack_map.HasInlineInfo(encoding.stack_map_encoding)) { InlineInfo inline_info = code_info.GetInlineInfoOf(stack_map, encoding); DCHECK_EQ(current_inlining_depth_, 0u); - for (current_inlining_depth_ = inline_info.GetDepth(); + for (current_inlining_depth_ = inline_info.GetDepth(encoding.inline_info_encoding); current_inlining_depth_ != 0; --current_inlining_depth_) { bool should_continue = VisitFrame(); diff --git a/runtime/stack_map.cc b/runtime/stack_map.cc index b51baf1b0c..a7e7c21a42 100644 --- a/runtime/stack_map.cc +++ b/runtime/stack_map.cc @@ -101,6 +101,17 @@ void StackMapEncoding::Dump(VariableIndentationOutputStream* vios) const { << ")\n"; } +void InlineInfoEncoding::Dump(VariableIndentationOutputStream* vios) const { + vios->Stream() + << "InlineInfoEncoding" + << " (method_index_bit_offset=" << static_cast<uint32_t>(kMethodIndexBitOffset) + << ", dex_pc_bit_offset=" << static_cast<uint32_t>(dex_pc_bit_offset_) + << ", invoke_type_bit_offset=" << static_cast<uint32_t>(invoke_type_bit_offset_) + << ", dex_register_map_bit_offset=" << static_cast<uint32_t>(dex_register_map_bit_offset_) + << ", total_bit_size=" << static_cast<uint32_t>(total_bit_size_) + << ")\n"; +} + void CodeInfo::Dump(VariableIndentationOutputStream* vios, uint32_t code_offset, uint16_t number_of_dex_registers, @@ -113,6 +124,9 @@ void CodeInfo::Dump(VariableIndentationOutputStream* vios, << ")\n"; ScopedIndentation indent1(vios); encoding.stack_map_encoding.Dump(vios); + if (HasInlineInfo(encoding)) { + encoding.inline_info_encoding.Dump(vios); + } // Display the Dex register location catalog. GetDexRegisterLocationCatalog(encoding).Dump(vios, *this); // Display stack maps along with (live) Dex register maps. @@ -207,18 +221,22 @@ void StackMap::Dump(VariableIndentationOutputStream* vios, void InlineInfo::Dump(VariableIndentationOutputStream* vios, const CodeInfo& code_info, uint16_t number_of_dex_registers[]) const { - vios->Stream() << "InlineInfo with depth " << static_cast<uint32_t>(GetDepth()) << "\n"; + InlineInfoEncoding inline_info_encoding = code_info.ExtractEncoding().inline_info_encoding; + vios->Stream() << "InlineInfo with depth " + << static_cast<uint32_t>(GetDepth(inline_info_encoding)) + << "\n"; - for (size_t i = 0; i < GetDepth(); ++i) { + for (size_t i = 0; i < GetDepth(inline_info_encoding); ++i) { vios->Stream() << " At depth " << i << std::hex - << " (dex_pc=0x" << GetDexPcAtDepth(i) + << " (dex_pc=0x" << GetDexPcAtDepth(inline_info_encoding, i) << std::dec - << ", method_index=" << GetMethodIndexAtDepth(i) - << ", invoke_type=" << static_cast<InvokeType>(GetInvokeTypeAtDepth(i)) + << ", method_index=" << GetMethodIndexAtDepth(inline_info_encoding, i) + << ", invoke_type=" << static_cast<InvokeType>(GetInvokeTypeAtDepth(inline_info_encoding, + i)) << ")\n"; - if (HasDexRegisterMapAtDepth(i) && (number_of_dex_registers != nullptr)) { + if (HasDexRegisterMapAtDepth(inline_info_encoding, i) && (number_of_dex_registers != nullptr)) { CodeInfoEncoding encoding = code_info.ExtractEncoding(); DexRegisterMap dex_register_map = code_info.GetDexRegisterMapAtDepth(i, *this, encoding, number_of_dex_registers[i]); diff --git a/runtime/stack_map.h b/runtime/stack_map.h index 9e8884efd4..7c50f97d39 100644 --- a/runtime/stack_map.h +++ b/runtime/stack_map.h @@ -24,12 +24,6 @@ namespace art { -#define ELEMENT_BYTE_OFFSET_AFTER(PreviousElement) \ - k ## PreviousElement ## Offset + sizeof(PreviousElement ## Type) - -#define ELEMENT_BIT_OFFSET_AFTER(PreviousElement) \ - k ## PreviousElement ## BitOffset + PreviousElement ## BitSize - class VariableIndentationOutputStream; // Size of a frame slot, in bytes. This constant is a signed value, @@ -888,102 +882,139 @@ class StackMap { friend class StackMapStream; }; +class InlineInfoEncoding { + public: + void SetFromSizes(size_t method_index_max, + size_t dex_pc_max, + size_t invoke_type_max, + size_t dex_register_map_size) { + total_bit_size_ = kMethodIndexBitOffset; + total_bit_size_ += MinimumBitsToStore(method_index_max); + + dex_pc_bit_offset_ = dchecked_integral_cast<uint8_t>(total_bit_size_); + total_bit_size_ += MinimumBitsToStore(1 /* kNoDexPc */ + dex_pc_max); + + invoke_type_bit_offset_ = dchecked_integral_cast<uint8_t>(total_bit_size_); + total_bit_size_ += MinimumBitsToStore(invoke_type_max); + + // We also need +1 for kNoDexRegisterMap, but since the size is strictly + // greater than any offset we might try to encode, we already implicitly have it. + dex_register_map_bit_offset_ = dchecked_integral_cast<uint8_t>(total_bit_size_); + total_bit_size_ += MinimumBitsToStore(dex_register_map_size); + } + + ALWAYS_INLINE FieldEncoding GetMethodIndexEncoding() const { + return FieldEncoding(kMethodIndexBitOffset, dex_pc_bit_offset_); + } + ALWAYS_INLINE FieldEncoding GetDexPcEncoding() const { + return FieldEncoding(dex_pc_bit_offset_, invoke_type_bit_offset_, -1 /* min_value */); + } + ALWAYS_INLINE FieldEncoding GetInvokeTypeEncoding() const { + return FieldEncoding(invoke_type_bit_offset_, dex_register_map_bit_offset_); + } + ALWAYS_INLINE FieldEncoding GetDexRegisterMapEncoding() const { + return FieldEncoding(dex_register_map_bit_offset_, total_bit_size_, -1 /* min_value */); + } + ALWAYS_INLINE size_t GetEntrySize() const { + return RoundUp(total_bit_size_, kBitsPerByte) / kBitsPerByte; + } + + void Dump(VariableIndentationOutputStream* vios) const; + + private: + static constexpr uint8_t kIsLastBitOffset = 0; + static constexpr uint8_t kMethodIndexBitOffset = 1; + uint8_t dex_pc_bit_offset_; + uint8_t invoke_type_bit_offset_; + uint8_t dex_register_map_bit_offset_; + uint8_t total_bit_size_; +}; + /** * Inline information for a specific PC. The information is of the form: * - * [inlining_depth, entry+] - * - * where `entry` is of the form: - * - * [dex_pc, method_index, dex_register_map_offset]. + * [is_last, method_index, dex_pc, invoke_type, dex_register_map_offset]+. */ class InlineInfo { public: - // Memory layout: fixed contents. - typedef uint8_t DepthType; - // Memory layout: single entry contents. - typedef uint32_t MethodIndexType; - typedef uint32_t DexPcType; - typedef uint8_t InvokeTypeType; - typedef uint32_t DexRegisterMapType; - - explicit InlineInfo(MemoryRegion region) : region_(region) {} - - DepthType GetDepth() const { - return region_.LoadUnaligned<DepthType>(kDepthOffset); + explicit InlineInfo(MemoryRegion region) : region_(region) { } - void SetDepth(DepthType depth) { - region_.StoreUnaligned<DepthType>(kDepthOffset, depth); + ALWAYS_INLINE uint32_t GetDepth(const InlineInfoEncoding& encoding) const { + size_t depth = 0; + while (!GetRegionAtDepth(encoding, depth++).LoadBit(0)) { } // Check is_last bit. + return depth; } - MethodIndexType GetMethodIndexAtDepth(DepthType depth) const { - return region_.LoadUnaligned<MethodIndexType>( - kFixedSize + depth * SingleEntrySize() + kMethodIndexOffset); + ALWAYS_INLINE void SetDepth(const InlineInfoEncoding& encoding, uint32_t depth) { + DCHECK_GT(depth, 0u); + for (size_t d = 0; d < depth; ++d) { + GetRegionAtDepth(encoding, d).StoreBit(0, d == depth - 1); // Set is_last bit. + } } - void SetMethodIndexAtDepth(DepthType depth, MethodIndexType index) { - region_.StoreUnaligned<MethodIndexType>( - kFixedSize + depth * SingleEntrySize() + kMethodIndexOffset, index); + ALWAYS_INLINE uint32_t GetMethodIndexAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth) const { + return encoding.GetMethodIndexEncoding().Load(GetRegionAtDepth(encoding, depth)); } - DexPcType GetDexPcAtDepth(DepthType depth) const { - return region_.LoadUnaligned<DexPcType>( - kFixedSize + depth * SingleEntrySize() + kDexPcOffset); + ALWAYS_INLINE void SetMethodIndexAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth, + uint32_t index) { + encoding.GetMethodIndexEncoding().Store(GetRegionAtDepth(encoding, depth), index); } - void SetDexPcAtDepth(DepthType depth, DexPcType dex_pc) { - region_.StoreUnaligned<DexPcType>( - kFixedSize + depth * SingleEntrySize() + kDexPcOffset, dex_pc); + ALWAYS_INLINE uint32_t GetDexPcAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth) const { + return encoding.GetDexPcEncoding().Load(GetRegionAtDepth(encoding, depth)); } - InvokeTypeType GetInvokeTypeAtDepth(DepthType depth) const { - return region_.LoadUnaligned<InvokeTypeType>( - kFixedSize + depth * SingleEntrySize() + kInvokeTypeOffset); + ALWAYS_INLINE void SetDexPcAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth, + uint32_t dex_pc) { + encoding.GetDexPcEncoding().Store(GetRegionAtDepth(encoding, depth), dex_pc); } - void SetInvokeTypeAtDepth(DepthType depth, InvokeTypeType invoke_type) { - region_.StoreUnaligned<InvokeTypeType>( - kFixedSize + depth * SingleEntrySize() + kInvokeTypeOffset, invoke_type); + ALWAYS_INLINE uint32_t GetInvokeTypeAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth) const { + return encoding.GetInvokeTypeEncoding().Load(GetRegionAtDepth(encoding, depth)); } - DexRegisterMapType GetDexRegisterMapOffsetAtDepth(DepthType depth) const { - return region_.LoadUnaligned<DexRegisterMapType>( - kFixedSize + depth * SingleEntrySize() + kDexRegisterMapOffset); + ALWAYS_INLINE void SetInvokeTypeAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth, + uint32_t invoke_type) { + encoding.GetInvokeTypeEncoding().Store(GetRegionAtDepth(encoding, depth), invoke_type); } - void SetDexRegisterMapOffsetAtDepth(DepthType depth, DexRegisterMapType offset) { - region_.StoreUnaligned<DexRegisterMapType>( - kFixedSize + depth * SingleEntrySize() + kDexRegisterMapOffset, offset); + ALWAYS_INLINE uint32_t GetDexRegisterMapOffsetAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth) const { + return encoding.GetDexRegisterMapEncoding().Load(GetRegionAtDepth(encoding, depth)); } - bool HasDexRegisterMapAtDepth(DepthType depth) const { - return GetDexRegisterMapOffsetAtDepth(depth) != StackMap::kNoDexRegisterMap; + ALWAYS_INLINE void SetDexRegisterMapOffsetAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth, + uint32_t offset) { + encoding.GetDexRegisterMapEncoding().Store(GetRegionAtDepth(encoding, depth), offset); } - static size_t SingleEntrySize() { - return kFixedEntrySize; + ALWAYS_INLINE bool HasDexRegisterMapAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth) const { + return GetDexRegisterMapOffsetAtDepth(encoding, depth) != StackMap::kNoDexRegisterMap; } void Dump(VariableIndentationOutputStream* vios, - const CodeInfo& info, uint16_t* number_of_dex_registers) const; - + const CodeInfo& info, + uint16_t* number_of_dex_registers) const; private: - static constexpr int kDepthOffset = 0; - static constexpr int kFixedSize = ELEMENT_BYTE_OFFSET_AFTER(Depth); - - static constexpr int kMethodIndexOffset = 0; - static constexpr int kDexPcOffset = ELEMENT_BYTE_OFFSET_AFTER(MethodIndex); - static constexpr int kInvokeTypeOffset = ELEMENT_BYTE_OFFSET_AFTER(DexPc); - static constexpr int kDexRegisterMapOffset = ELEMENT_BYTE_OFFSET_AFTER(InvokeType); - static constexpr int kFixedEntrySize = ELEMENT_BYTE_OFFSET_AFTER(DexRegisterMap); + ALWAYS_INLINE MemoryRegion GetRegionAtDepth(const InlineInfoEncoding& encoding, + uint32_t depth) const { + size_t entry_size = encoding.GetEntrySize(); + DCHECK_GT(entry_size, 0u); + return region_.Subregion(depth * entry_size, entry_size); + } MemoryRegion region_; - - friend class CodeInfo; - friend class StackMap; - friend class StackMapStream; }; // Most of the fields are encoded as ULEB128 to save space. @@ -993,6 +1024,7 @@ struct CodeInfoEncoding { uint32_t stack_map_size_in_bytes; uint32_t number_of_location_catalog_entries; StackMapEncoding stack_map_encoding; + InlineInfoEncoding inline_info_encoding; uint8_t header_size; CodeInfoEncoding() { } @@ -1003,9 +1035,18 @@ struct CodeInfoEncoding { number_of_stack_maps = DecodeUnsignedLeb128(&ptr); stack_map_size_in_bytes = DecodeUnsignedLeb128(&ptr); number_of_location_catalog_entries = DecodeUnsignedLeb128(&ptr); - static_assert(alignof(StackMapEncoding) == 1, "StackMapEncoding should not require alignment"); + static_assert(alignof(StackMapEncoding) == 1, + "StackMapEncoding should not require alignment"); stack_map_encoding = *reinterpret_cast<const StackMapEncoding*>(ptr); ptr += sizeof(StackMapEncoding); + if (stack_map_encoding.GetInlineInfoEncoding().BitSize() > 0) { + static_assert(alignof(InlineInfoEncoding) == 1, + "InlineInfoEncoding should not require alignment"); + inline_info_encoding = *reinterpret_cast<const InlineInfoEncoding*>(ptr); + ptr += sizeof(InlineInfoEncoding); + } else { + inline_info_encoding = InlineInfoEncoding{}; // NOLINT. + } header_size = dchecked_integral_cast<uint8_t>(ptr - reinterpret_cast<const uint8_t*>(data)); } @@ -1015,8 +1056,12 @@ struct CodeInfoEncoding { EncodeUnsignedLeb128(dest, number_of_stack_maps); EncodeUnsignedLeb128(dest, stack_map_size_in_bytes); EncodeUnsignedLeb128(dest, number_of_location_catalog_entries); - const uint8_t* ptr = reinterpret_cast<const uint8_t*>(&stack_map_encoding); - dest->insert(dest->end(), ptr, ptr + sizeof(stack_map_encoding)); + const uint8_t* stack_map_ptr = reinterpret_cast<const uint8_t*>(&stack_map_encoding); + dest->insert(dest->end(), stack_map_ptr, stack_map_ptr + sizeof(StackMapEncoding)); + if (stack_map_encoding.GetInlineInfoEncoding().BitSize() > 0) { + const uint8_t* inline_info_ptr = reinterpret_cast<const uint8_t*>(&inline_info_encoding); + dest->insert(dest->end(), inline_info_ptr, inline_info_ptr + sizeof(InlineInfoEncoding)); + } } }; @@ -1110,11 +1155,11 @@ class CodeInfo { InlineInfo inline_info, const CodeInfoEncoding& encoding, uint32_t number_of_dex_registers) const { - if (!inline_info.HasDexRegisterMapAtDepth(depth)) { + if (!inline_info.HasDexRegisterMapAtDepth(encoding.inline_info_encoding, depth)) { return DexRegisterMap(); } else { - uint32_t offset = GetDexRegisterMapsOffset(encoding) - + inline_info.GetDexRegisterMapOffsetAtDepth(depth); + uint32_t offset = GetDexRegisterMapsOffset(encoding) + + inline_info.GetDexRegisterMapOffsetAtDepth(encoding.inline_info_encoding, depth); size_t size = ComputeDexRegisterMapSizeOf(encoding, offset, number_of_dex_registers); return DexRegisterMap(region_.Subregion(offset, size)); } @@ -1124,9 +1169,7 @@ class CodeInfo { DCHECK(stack_map.HasInlineInfo(encoding.stack_map_encoding)); uint32_t offset = stack_map.GetInlineDescriptorOffset(encoding.stack_map_encoding) + GetDexRegisterMapsOffset(encoding); - uint8_t depth = region_.LoadUnaligned<uint8_t>(offset); - return InlineInfo(region_.Subregion(offset, - InlineInfo::kFixedSize + depth * InlineInfo::SingleEntrySize())); + return InlineInfo(region_.Subregion(offset, region_.size() - offset)); } StackMap GetStackMapForDexPc(uint32_t dex_pc, const CodeInfoEncoding& encoding) const { |