Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2014 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "stack_map.h" |
Mathieu Chartier | e5d80f8 | 2015-10-15 17:47:48 -0700 | [diff] [blame] | 18 | |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 19 | #include "art_method.h" |
Mathieu Chartier | e5d80f8 | 2015-10-15 17:47:48 -0700 | [diff] [blame] | 20 | #include "base/arena_bit_vector.h" |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 21 | #include "base/malloc_arena_pool.h" |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 22 | #include "stack_map_stream.h" |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 23 | |
| 24 | #include "gtest/gtest.h" |
| 25 | |
| 26 | namespace art { |
| 27 | |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 28 | // Check that the stack mask of given stack map is identical |
| 29 | // to the given bit vector. Returns true if they are same. |
| 30 | static bool CheckStackMask( |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 31 | const CodeInfo& code_info, |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 32 | const StackMap& stack_map, |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 33 | const BitVector& bit_vector) { |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 34 | BitMemoryRegion stack_mask = code_info.GetStackMaskOf(stack_map); |
David Srbecky | 4b59d10 | 2018-05-29 21:46:10 +0000 | [diff] [blame] | 35 | if (bit_vector.GetNumberOfBits() > stack_mask.size_in_bits()) { |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 36 | return false; |
| 37 | } |
David Srbecky | 4b59d10 | 2018-05-29 21:46:10 +0000 | [diff] [blame] | 38 | for (size_t i = 0; i < stack_mask.size_in_bits(); ++i) { |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 39 | if (stack_mask.LoadBit(i) != bit_vector.IsBitSet(i)) { |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 40 | return false; |
| 41 | } |
| 42 | } |
| 43 | return true; |
| 44 | } |
| 45 | |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 46 | using Kind = DexRegisterLocation::Kind; |
| 47 | |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 48 | constexpr static uint32_t kPcAlign = GetInstructionSetInstructionAlignment(kRuntimeISA); |
| 49 | |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 50 | TEST(StackMapTest, Test1) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 51 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 52 | ArenaStack arena_stack(&pool); |
| 53 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 54 | StackMapStream stream(&allocator, kRuntimeISA); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 55 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 56 | ArenaBitVector sp_mask(&allocator, 0, false); |
Roland Levillain | 12baf47 | 2015-03-05 12:41:42 +0000 | [diff] [blame] | 57 | size_t number_of_dex_registers = 2; |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 58 | stream.BeginStackMapEntry(0, 64 * kPcAlign, 0x3, &sp_mask, number_of_dex_registers, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 59 | stream.AddDexRegisterEntry(Kind::kInStack, 0); // Short location. |
| 60 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Short location. |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 61 | stream.EndStackMapEntry(); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 62 | |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 63 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 64 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 65 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 66 | stream.FillInCodeInfo(region); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 67 | |
Nicolas Geoffray | 3946844 | 2014-09-02 15:17:15 +0100 | [diff] [blame] | 68 | CodeInfo code_info(region); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 69 | ASSERT_EQ(1u, code_info.GetNumberOfStackMaps()); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 70 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 71 | uint32_t number_of_catalog_entries = code_info.GetNumberOfLocationCatalogEntries(); |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 72 | ASSERT_EQ(2u, number_of_catalog_entries); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 73 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 74 | StackMap stack_map = code_info.GetStackMapAt(0); |
| 75 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(0))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 76 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(64 * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 77 | ASSERT_EQ(0u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 78 | ASSERT_EQ(64u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 79 | ASSERT_EQ(0x3u, code_info.GetRegisterMaskOf(stack_map)); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 80 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 81 | ASSERT_TRUE(CheckStackMask(code_info, stack_map, sp_mask)); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 82 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 83 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 84 | DexRegisterMap dex_register_map = code_info.GetDexRegisterMapOf(stack_map); |
| 85 | ASSERT_EQ(number_of_dex_registers, dex_register_map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 86 | ASSERT_TRUE(dex_register_map[0].IsLive()); |
| 87 | ASSERT_TRUE(dex_register_map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 88 | ASSERT_EQ(2u, dex_register_map.GetNumberOfLiveDexRegisters()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 89 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 90 | ASSERT_EQ(Kind::kInStack, dex_register_map[0].GetKind()); |
| 91 | ASSERT_EQ(Kind::kConstant, dex_register_map[1].GetKind()); |
| 92 | ASSERT_EQ(0, dex_register_map[0].GetStackOffsetInBytes()); |
| 93 | ASSERT_EQ(-2, dex_register_map[1].GetConstant()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 94 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 95 | DexRegisterLocation location0 = code_info.GetDexRegisterCatalogEntry(0); |
| 96 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(1); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 97 | ASSERT_EQ(Kind::kInStack, location0.GetKind()); |
| 98 | ASSERT_EQ(Kind::kConstant, location1.GetKind()); |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 99 | ASSERT_EQ(0, location0.GetValue()); |
| 100 | ASSERT_EQ(-2, location1.GetValue()); |
Roland Levillain | 12baf47 | 2015-03-05 12:41:42 +0000 | [diff] [blame] | 101 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 102 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 103 | } |
| 104 | |
| 105 | TEST(StackMapTest, Test2) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 106 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 107 | ArenaStack arena_stack(&pool); |
| 108 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 109 | StackMapStream stream(&allocator, kRuntimeISA); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 110 | ArtMethod art_method; |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 111 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 112 | ArenaBitVector sp_mask1(&allocator, 0, true); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 113 | sp_mask1.SetBit(2); |
| 114 | sp_mask1.SetBit(4); |
Roland Levillain | 12baf47 | 2015-03-05 12:41:42 +0000 | [diff] [blame] | 115 | size_t number_of_dex_registers = 2; |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 116 | size_t number_of_dex_registers_in_inline_info = 0; |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 117 | stream.BeginStackMapEntry(0, 64 * kPcAlign, 0x3, &sp_mask1, number_of_dex_registers, 2); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 118 | stream.AddDexRegisterEntry(Kind::kInStack, 0); // Short location. |
| 119 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 120 | stream.BeginInlineInfoEntry(&art_method, 3, number_of_dex_registers_in_inline_info); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 121 | stream.EndInlineInfoEntry(); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 122 | stream.BeginInlineInfoEntry(&art_method, 2, number_of_dex_registers_in_inline_info); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 123 | stream.EndInlineInfoEntry(); |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 124 | stream.EndStackMapEntry(); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 125 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 126 | ArenaBitVector sp_mask2(&allocator, 0, true); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 127 | sp_mask2.SetBit(3); |
David Brazdil | f10a25f | 2015-06-02 14:29:52 +0100 | [diff] [blame] | 128 | sp_mask2.SetBit(8); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 129 | stream.BeginStackMapEntry(1, 128 * kPcAlign, 0xFF, &sp_mask2, number_of_dex_registers, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 130 | stream.AddDexRegisterEntry(Kind::kInRegister, 18); // Short location. |
| 131 | stream.AddDexRegisterEntry(Kind::kInFpuRegister, 3); // Short location. |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 132 | stream.EndStackMapEntry(); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 133 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 134 | ArenaBitVector sp_mask3(&allocator, 0, true); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 135 | sp_mask3.SetBit(1); |
| 136 | sp_mask3.SetBit(5); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 137 | stream.BeginStackMapEntry(2, 192 * kPcAlign, 0xAB, &sp_mask3, number_of_dex_registers, 0); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 138 | stream.AddDexRegisterEntry(Kind::kInRegister, 6); // Short location. |
| 139 | stream.AddDexRegisterEntry(Kind::kInRegisterHigh, 8); // Short location. |
| 140 | stream.EndStackMapEntry(); |
| 141 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 142 | ArenaBitVector sp_mask4(&allocator, 0, true); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 143 | sp_mask4.SetBit(6); |
| 144 | sp_mask4.SetBit(7); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 145 | stream.BeginStackMapEntry(3, 256 * kPcAlign, 0xCD, &sp_mask4, number_of_dex_registers, 0); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 146 | stream.AddDexRegisterEntry(Kind::kInFpuRegister, 3); // Short location, same in stack map 2. |
| 147 | stream.AddDexRegisterEntry(Kind::kInFpuRegisterHigh, 1); // Short location. |
| 148 | stream.EndStackMapEntry(); |
| 149 | |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 150 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 151 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 152 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 153 | stream.FillInCodeInfo(region); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 154 | |
Nicolas Geoffray | 3946844 | 2014-09-02 15:17:15 +0100 | [diff] [blame] | 155 | CodeInfo code_info(region); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 156 | ASSERT_EQ(4u, code_info.GetNumberOfStackMaps()); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 157 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 158 | uint32_t number_of_catalog_entries = code_info.GetNumberOfLocationCatalogEntries(); |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 159 | ASSERT_EQ(7u, number_of_catalog_entries); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 160 | |
Roland Levillain | 12baf47 | 2015-03-05 12:41:42 +0000 | [diff] [blame] | 161 | // First stack map. |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 162 | { |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 163 | StackMap stack_map = code_info.GetStackMapAt(0); |
| 164 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(0))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 165 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(64 * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 166 | ASSERT_EQ(0u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 167 | ASSERT_EQ(64u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 168 | ASSERT_EQ(0x3u, code_info.GetRegisterMaskOf(stack_map)); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 169 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 170 | ASSERT_TRUE(CheckStackMask(code_info, stack_map, sp_mask1)); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 171 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 172 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 173 | DexRegisterMap dex_register_map = code_info.GetDexRegisterMapOf(stack_map); |
| 174 | ASSERT_EQ(number_of_dex_registers, dex_register_map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 175 | ASSERT_TRUE(dex_register_map[0].IsLive()); |
| 176 | ASSERT_TRUE(dex_register_map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 177 | ASSERT_EQ(2u, dex_register_map.GetNumberOfLiveDexRegisters()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 178 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 179 | ASSERT_EQ(Kind::kInStack, dex_register_map[0].GetKind()); |
| 180 | ASSERT_EQ(Kind::kConstant, dex_register_map[1].GetKind()); |
| 181 | ASSERT_EQ(0, dex_register_map[0].GetStackOffsetInBytes()); |
| 182 | ASSERT_EQ(-2, dex_register_map[1].GetConstant()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 183 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 184 | DexRegisterLocation location0 = code_info.GetDexRegisterCatalogEntry(0); |
| 185 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(1); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 186 | ASSERT_EQ(Kind::kInStack, location0.GetKind()); |
| 187 | ASSERT_EQ(Kind::kConstant, location1.GetKind()); |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 188 | ASSERT_EQ(0, location0.GetValue()); |
| 189 | ASSERT_EQ(-2, location1.GetValue()); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 190 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 191 | ASSERT_TRUE(stack_map.HasInlineInfo()); |
David Srbecky | 6e69e52 | 2018-06-03 12:00:14 +0100 | [diff] [blame] | 192 | InlineInfo inline_info0 = code_info.GetInlineInfoAtDepth(stack_map, 0); |
| 193 | InlineInfo inline_info1 = code_info.GetInlineInfoAtDepth(stack_map, 1); |
| 194 | ASSERT_EQ(2u, code_info.GetInlineDepthOf(stack_map)); |
| 195 | ASSERT_EQ(3u, inline_info0.GetDexPc()); |
| 196 | ASSERT_EQ(2u, inline_info1.GetDexPc()); |
| 197 | ASSERT_TRUE(inline_info0.EncodesArtMethod()); |
| 198 | ASSERT_TRUE(inline_info1.EncodesArtMethod()); |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 199 | } |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 200 | |
Roland Levillain | 12baf47 | 2015-03-05 12:41:42 +0000 | [diff] [blame] | 201 | // Second stack map. |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 202 | { |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 203 | StackMap stack_map = code_info.GetStackMapAt(1); |
| 204 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(1u))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 205 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(128u * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 206 | ASSERT_EQ(1u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 207 | ASSERT_EQ(128u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 208 | ASSERT_EQ(0xFFu, code_info.GetRegisterMaskOf(stack_map)); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 209 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 210 | ASSERT_TRUE(CheckStackMask(code_info, stack_map, sp_mask2)); |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 211 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 212 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 213 | DexRegisterMap dex_register_map = code_info.GetDexRegisterMapOf(stack_map); |
| 214 | ASSERT_EQ(number_of_dex_registers, dex_register_map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 215 | ASSERT_TRUE(dex_register_map[0].IsLive()); |
| 216 | ASSERT_TRUE(dex_register_map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 217 | ASSERT_EQ(2u, dex_register_map.GetNumberOfLiveDexRegisters()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 218 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 219 | ASSERT_EQ(Kind::kInRegister, dex_register_map[0].GetKind()); |
| 220 | ASSERT_EQ(Kind::kInFpuRegister, dex_register_map[1].GetKind()); |
| 221 | ASSERT_EQ(18, dex_register_map[0].GetMachineRegister()); |
| 222 | ASSERT_EQ(3, dex_register_map[1].GetMachineRegister()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 223 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 224 | DexRegisterLocation location0 = code_info.GetDexRegisterCatalogEntry(2); |
| 225 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(3); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 226 | ASSERT_EQ(Kind::kInRegister, location0.GetKind()); |
| 227 | ASSERT_EQ(Kind::kInFpuRegister, location1.GetKind()); |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 228 | ASSERT_EQ(18, location0.GetValue()); |
| 229 | ASSERT_EQ(3, location1.GetValue()); |
Roland Levillain | 12baf47 | 2015-03-05 12:41:42 +0000 | [diff] [blame] | 230 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 231 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
Roland Levillain | a2d8ec6 | 2015-03-12 15:25:29 +0000 | [diff] [blame] | 232 | } |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 233 | |
| 234 | // Third stack map. |
| 235 | { |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 236 | StackMap stack_map = code_info.GetStackMapAt(2); |
| 237 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(2u))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 238 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(192u * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 239 | ASSERT_EQ(2u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 240 | ASSERT_EQ(192u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 241 | ASSERT_EQ(0xABu, code_info.GetRegisterMaskOf(stack_map)); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 242 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 243 | ASSERT_TRUE(CheckStackMask(code_info, stack_map, sp_mask3)); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 244 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 245 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 246 | DexRegisterMap dex_register_map = code_info.GetDexRegisterMapOf(stack_map); |
| 247 | ASSERT_EQ(number_of_dex_registers, dex_register_map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 248 | ASSERT_TRUE(dex_register_map[0].IsLive()); |
| 249 | ASSERT_TRUE(dex_register_map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 250 | ASSERT_EQ(2u, dex_register_map.GetNumberOfLiveDexRegisters()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 251 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 252 | ASSERT_EQ(Kind::kInRegister, dex_register_map[0].GetKind()); |
| 253 | ASSERT_EQ(Kind::kInRegisterHigh, dex_register_map[1].GetKind()); |
| 254 | ASSERT_EQ(6, dex_register_map[0].GetMachineRegister()); |
| 255 | ASSERT_EQ(8, dex_register_map[1].GetMachineRegister()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 256 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 257 | DexRegisterLocation location0 = code_info.GetDexRegisterCatalogEntry(4); |
| 258 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(5); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 259 | ASSERT_EQ(Kind::kInRegister, location0.GetKind()); |
| 260 | ASSERT_EQ(Kind::kInRegisterHigh, location1.GetKind()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 261 | ASSERT_EQ(6, location0.GetValue()); |
| 262 | ASSERT_EQ(8, location1.GetValue()); |
| 263 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 264 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 265 | } |
| 266 | |
| 267 | // Fourth stack map. |
| 268 | { |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 269 | StackMap stack_map = code_info.GetStackMapAt(3); |
| 270 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(3u))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 271 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(256u * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 272 | ASSERT_EQ(3u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 273 | ASSERT_EQ(256u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 274 | ASSERT_EQ(0xCDu, code_info.GetRegisterMaskOf(stack_map)); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 275 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 276 | ASSERT_TRUE(CheckStackMask(code_info, stack_map, sp_mask4)); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 277 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 278 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 279 | DexRegisterMap dex_register_map = code_info.GetDexRegisterMapOf(stack_map); |
| 280 | ASSERT_EQ(number_of_dex_registers, dex_register_map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 281 | ASSERT_TRUE(dex_register_map[0].IsLive()); |
| 282 | ASSERT_TRUE(dex_register_map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 283 | ASSERT_EQ(2u, dex_register_map.GetNumberOfLiveDexRegisters()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 284 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 285 | ASSERT_EQ(Kind::kInFpuRegister, dex_register_map[0].GetKind()); |
| 286 | ASSERT_EQ(Kind::kInFpuRegisterHigh, dex_register_map[1].GetKind()); |
| 287 | ASSERT_EQ(3, dex_register_map[0].GetMachineRegister()); |
| 288 | ASSERT_EQ(1, dex_register_map[1].GetMachineRegister()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 289 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 290 | DexRegisterLocation location0 = code_info.GetDexRegisterCatalogEntry(3); |
| 291 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(6); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 292 | ASSERT_EQ(Kind::kInFpuRegister, location0.GetKind()); |
| 293 | ASSERT_EQ(Kind::kInFpuRegisterHigh, location1.GetKind()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 294 | ASSERT_EQ(3, location0.GetValue()); |
| 295 | ASSERT_EQ(1, location1.GetValue()); |
| 296 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 297 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
David Brazdil | d9cb68e | 2015-08-25 13:52:43 +0100 | [diff] [blame] | 298 | } |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 299 | } |
| 300 | |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 301 | TEST(StackMapTest, TestDeduplicateInlineInfoDexRegisterMap) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 302 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 303 | ArenaStack arena_stack(&pool); |
| 304 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 305 | StackMapStream stream(&allocator, kRuntimeISA); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 306 | ArtMethod art_method; |
| 307 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 308 | ArenaBitVector sp_mask1(&allocator, 0, true); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 309 | sp_mask1.SetBit(2); |
| 310 | sp_mask1.SetBit(4); |
| 311 | const size_t number_of_dex_registers = 2; |
| 312 | const size_t number_of_dex_registers_in_inline_info = 2; |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 313 | stream.BeginStackMapEntry(0, 64 * kPcAlign, 0x3, &sp_mask1, number_of_dex_registers, 1); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 314 | stream.AddDexRegisterEntry(Kind::kInStack, 0); // Short location. |
| 315 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
| 316 | stream.BeginInlineInfoEntry(&art_method, 3, number_of_dex_registers_in_inline_info); |
| 317 | stream.AddDexRegisterEntry(Kind::kInStack, 0); // Short location. |
| 318 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
| 319 | stream.EndInlineInfoEntry(); |
| 320 | stream.EndStackMapEntry(); |
| 321 | |
| 322 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 323 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 324 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 325 | stream.FillInCodeInfo(region); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 326 | |
| 327 | CodeInfo code_info(region); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 328 | ASSERT_EQ(1u, code_info.GetNumberOfStackMaps()); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 329 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 330 | uint32_t number_of_catalog_entries = code_info.GetNumberOfLocationCatalogEntries(); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 331 | ASSERT_EQ(2u, number_of_catalog_entries); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 332 | |
| 333 | // First stack map. |
| 334 | { |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 335 | StackMap stack_map = code_info.GetStackMapAt(0); |
| 336 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(0))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 337 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(64 * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 338 | ASSERT_EQ(0u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 339 | ASSERT_EQ(64u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 340 | ASSERT_EQ(0x3u, code_info.GetRegisterMaskOf(stack_map)); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 341 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 342 | ASSERT_TRUE(CheckStackMask(code_info, stack_map, sp_mask1)); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 343 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 344 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 345 | DexRegisterMap map(code_info.GetDexRegisterMapOf(stack_map)); |
| 346 | ASSERT_EQ(number_of_dex_registers, map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 347 | ASSERT_TRUE(map[0].IsLive()); |
| 348 | ASSERT_TRUE(map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 349 | ASSERT_EQ(2u, map.GetNumberOfLiveDexRegisters()); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 350 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 351 | ASSERT_EQ(Kind::kInStack, map[0].GetKind()); |
| 352 | ASSERT_EQ(Kind::kConstant, map[1].GetKind()); |
| 353 | ASSERT_EQ(0, map[0].GetStackOffsetInBytes()); |
| 354 | ASSERT_EQ(-2, map[1].GetConstant()); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 355 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 356 | DexRegisterLocation location0 = code_info.GetDexRegisterCatalogEntry(0); |
| 357 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(1); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 358 | ASSERT_EQ(Kind::kInStack, location0.GetKind()); |
| 359 | ASSERT_EQ(Kind::kConstant, location1.GetKind()); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 360 | ASSERT_EQ(0, location0.GetValue()); |
| 361 | ASSERT_EQ(-2, location1.GetValue()); |
Mathieu Chartier | 3228908 | 2017-02-09 15:57:37 -0800 | [diff] [blame] | 362 | } |
| 363 | } |
| 364 | |
Nicolas Geoffray | fead4e4 | 2015-03-13 14:39:40 +0000 | [diff] [blame] | 365 | TEST(StackMapTest, TestNonLiveDexRegisters) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 366 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 367 | ArenaStack arena_stack(&pool); |
| 368 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 369 | StackMapStream stream(&allocator, kRuntimeISA); |
Nicolas Geoffray | fead4e4 | 2015-03-13 14:39:40 +0000 | [diff] [blame] | 370 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 371 | ArenaBitVector sp_mask(&allocator, 0, false); |
Nicolas Geoffray | fead4e4 | 2015-03-13 14:39:40 +0000 | [diff] [blame] | 372 | uint32_t number_of_dex_registers = 2; |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 373 | stream.BeginStackMapEntry(0, 64 * kPcAlign, 0x3, &sp_mask, number_of_dex_registers, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 374 | stream.AddDexRegisterEntry(Kind::kNone, 0); // No location. |
| 375 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 376 | stream.EndStackMapEntry(); |
Nicolas Geoffray | fead4e4 | 2015-03-13 14:39:40 +0000 | [diff] [blame] | 377 | |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 378 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 379 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Nicolas Geoffray | fead4e4 | 2015-03-13 14:39:40 +0000 | [diff] [blame] | 380 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 381 | stream.FillInCodeInfo(region); |
Nicolas Geoffray | fead4e4 | 2015-03-13 14:39:40 +0000 | [diff] [blame] | 382 | |
| 383 | CodeInfo code_info(region); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 384 | ASSERT_EQ(1u, code_info.GetNumberOfStackMaps()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 385 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 386 | uint32_t number_of_catalog_entries = code_info.GetNumberOfLocationCatalogEntries(); |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 387 | ASSERT_EQ(1u, number_of_catalog_entries); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 388 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 389 | StackMap stack_map = code_info.GetStackMapAt(0); |
| 390 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(0))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 391 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(64 * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 392 | ASSERT_EQ(0u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 393 | ASSERT_EQ(64u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 394 | ASSERT_EQ(0x3u, code_info.GetRegisterMaskOf(stack_map)); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 395 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 396 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 397 | DexRegisterMap dex_register_map = code_info.GetDexRegisterMapOf(stack_map); |
| 398 | ASSERT_EQ(number_of_dex_registers, dex_register_map.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 399 | ASSERT_FALSE(dex_register_map[0].IsLive()); |
| 400 | ASSERT_TRUE(dex_register_map[1].IsLive()); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 401 | ASSERT_EQ(1u, dex_register_map.GetNumberOfLiveDexRegisters()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 402 | |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 403 | ASSERT_EQ(Kind::kNone, dex_register_map[0].GetKind()); |
| 404 | ASSERT_EQ(Kind::kConstant, dex_register_map[1].GetKind()); |
| 405 | ASSERT_EQ(-2, dex_register_map[1].GetConstant()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 406 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 407 | DexRegisterLocation location1 = code_info.GetDexRegisterCatalogEntry(0); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 408 | ASSERT_EQ(Kind::kConstant, location1.GetKind()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 409 | ASSERT_EQ(-2, location1.GetValue()); |
| 410 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 411 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
Nicolas Geoffray | 004c230 | 2015-03-20 10:06:38 +0000 | [diff] [blame] | 412 | } |
| 413 | |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 414 | TEST(StackMapTest, TestShareDexRegisterMap) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 415 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 416 | ArenaStack arena_stack(&pool); |
| 417 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 418 | StackMapStream stream(&allocator, kRuntimeISA); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 419 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 420 | ArenaBitVector sp_mask(&allocator, 0, false); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 421 | uint32_t number_of_dex_registers = 2; |
| 422 | // First stack map. |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 423 | stream.BeginStackMapEntry(0, 64 * kPcAlign, 0x3, &sp_mask, number_of_dex_registers, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 424 | stream.AddDexRegisterEntry(Kind::kInRegister, 0); // Short location. |
| 425 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 426 | stream.EndStackMapEntry(); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 427 | // Second stack map, which should share the same dex register map. |
David Srbecky | 50fac06 | 2018-06-13 18:55:35 +0100 | [diff] [blame] | 428 | stream.BeginStackMapEntry(0, 65 * kPcAlign, 0x3, &sp_mask, number_of_dex_registers, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 429 | stream.AddDexRegisterEntry(Kind::kInRegister, 0); // Short location. |
| 430 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 431 | stream.EndStackMapEntry(); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 432 | // Third stack map (doesn't share the dex register map). |
David Srbecky | 50fac06 | 2018-06-13 18:55:35 +0100 | [diff] [blame] | 433 | stream.BeginStackMapEntry(0, 66 * kPcAlign, 0x3, &sp_mask, number_of_dex_registers, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 434 | stream.AddDexRegisterEntry(Kind::kInRegister, 2); // Short location. |
| 435 | stream.AddDexRegisterEntry(Kind::kConstant, -2); // Large location. |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 436 | stream.EndStackMapEntry(); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 437 | |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 438 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 439 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 440 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 441 | stream.FillInCodeInfo(region); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 442 | |
| 443 | CodeInfo ci(region); |
David Brazdil | f677ebf | 2015-05-29 16:29:43 +0100 | [diff] [blame] | 444 | |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 445 | // Verify first stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 446 | StackMap sm0 = ci.GetStackMapAt(0); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 447 | DexRegisterMap dex_registers0 = ci.GetDexRegisterMapOf(sm0); |
| 448 | ASSERT_EQ(number_of_dex_registers, dex_registers0.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 449 | ASSERT_EQ(0, dex_registers0[0].GetMachineRegister()); |
| 450 | ASSERT_EQ(-2, dex_registers0[1].GetConstant()); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 451 | |
| 452 | // Verify second stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 453 | StackMap sm1 = ci.GetStackMapAt(1); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 454 | DexRegisterMap dex_registers1 = ci.GetDexRegisterMapOf(sm1); |
| 455 | ASSERT_EQ(number_of_dex_registers, dex_registers1.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 456 | ASSERT_EQ(0, dex_registers1[0].GetMachineRegister()); |
| 457 | ASSERT_EQ(-2, dex_registers1[1].GetConstant()); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 458 | |
| 459 | // Verify third stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 460 | StackMap sm2 = ci.GetStackMapAt(2); |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 461 | DexRegisterMap dex_registers2 = ci.GetDexRegisterMapOf(sm2); |
| 462 | ASSERT_EQ(number_of_dex_registers, dex_registers2.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 463 | ASSERT_EQ(2, dex_registers2[0].GetMachineRegister()); |
| 464 | ASSERT_EQ(-2, dex_registers2[1].GetConstant()); |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 465 | |
David Srbecky | 6de8833 | 2018-06-03 12:00:11 +0100 | [diff] [blame] | 466 | // Verify dex register mask offsets. |
| 467 | ASSERT_FALSE(sm1.HasDexRegisterMaskIndex()); // No delta. |
| 468 | ASSERT_TRUE(sm2.HasDexRegisterMaskIndex()); // Has delta. |
Calin Juravle | 6ae7096 | 2015-03-18 16:31:28 +0000 | [diff] [blame] | 469 | } |
| 470 | |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 471 | TEST(StackMapTest, TestNoDexRegisterMap) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 472 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 473 | ArenaStack arena_stack(&pool); |
| 474 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 475 | StackMapStream stream(&allocator, kRuntimeISA); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 476 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 477 | ArenaBitVector sp_mask(&allocator, 0, false); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 478 | uint32_t number_of_dex_registers = 0; |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 479 | stream.BeginStackMapEntry(0, 64 * kPcAlign, 0x3, &sp_mask, number_of_dex_registers, 0); |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 480 | stream.EndStackMapEntry(); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 481 | |
Nicolas Geoffray | 012fc4e | 2016-01-08 15:58:19 +0000 | [diff] [blame] | 482 | number_of_dex_registers = 1; |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 483 | stream.BeginStackMapEntry(1, 68 * kPcAlign, 0x4, &sp_mask, number_of_dex_registers, 0); |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 484 | stream.AddDexRegisterEntry(Kind::kNone, 0); |
Nicolas Geoffray | 012fc4e | 2016-01-08 15:58:19 +0000 | [diff] [blame] | 485 | stream.EndStackMapEntry(); |
| 486 | |
Calin Juravle | 4f46ac5 | 2015-04-23 18:47:21 +0100 | [diff] [blame] | 487 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 488 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 489 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 490 | stream.FillInCodeInfo(region); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 491 | |
| 492 | CodeInfo code_info(region); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 493 | ASSERT_EQ(2u, code_info.GetNumberOfStackMaps()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 494 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 495 | uint32_t number_of_catalog_entries = code_info.GetNumberOfLocationCatalogEntries(); |
David Srbecky | 09ed098 | 2016-02-12 21:58:43 +0000 | [diff] [blame] | 496 | ASSERT_EQ(0u, number_of_catalog_entries); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 497 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 498 | StackMap stack_map = code_info.GetStackMapAt(0); |
| 499 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(0))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 500 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(64 * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 501 | ASSERT_EQ(0u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 502 | ASSERT_EQ(64u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 503 | ASSERT_EQ(0x3u, code_info.GetRegisterMaskOf(stack_map)); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 504 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 505 | ASSERT_FALSE(stack_map.HasDexRegisterMap()); |
| 506 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
Nicolas Geoffray | 012fc4e | 2016-01-08 15:58:19 +0000 | [diff] [blame] | 507 | |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 508 | stack_map = code_info.GetStackMapAt(1); |
| 509 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForDexPc(1))); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 510 | ASSERT_TRUE(stack_map.Equals(code_info.GetStackMapForNativePcOffset(68 * kPcAlign))); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 511 | ASSERT_EQ(1u, stack_map.GetDexPc()); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 512 | ASSERT_EQ(68u * kPcAlign, stack_map.GetNativePcOffset(kRuntimeISA)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 513 | ASSERT_EQ(0x4u, code_info.GetRegisterMaskOf(stack_map)); |
Nicolas Geoffray | 012fc4e | 2016-01-08 15:58:19 +0000 | [diff] [blame] | 514 | |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 515 | ASSERT_TRUE(stack_map.HasDexRegisterMap()); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 516 | ASSERT_FALSE(stack_map.HasInlineInfo()); |
Roland Levillain | a552e1c | 2015-03-26 15:01:03 +0000 | [diff] [blame] | 517 | } |
| 518 | |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 519 | TEST(StackMapTest, InlineTest) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 520 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 521 | ArenaStack arena_stack(&pool); |
| 522 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 523 | StackMapStream stream(&allocator, kRuntimeISA); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 524 | ArtMethod art_method; |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 525 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 526 | ArenaBitVector sp_mask1(&allocator, 0, true); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 527 | sp_mask1.SetBit(2); |
| 528 | sp_mask1.SetBit(4); |
| 529 | |
| 530 | // First stack map. |
David Srbecky | 50fac06 | 2018-06-13 18:55:35 +0100 | [diff] [blame] | 531 | stream.BeginStackMapEntry(0, 10 * kPcAlign, 0x3, &sp_mask1, 2, 2); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 532 | stream.AddDexRegisterEntry(Kind::kInStack, 0); |
| 533 | stream.AddDexRegisterEntry(Kind::kConstant, 4); |
| 534 | |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 535 | stream.BeginInlineInfoEntry(&art_method, 2, 1); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 536 | stream.AddDexRegisterEntry(Kind::kInStack, 8); |
| 537 | stream.EndInlineInfoEntry(); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 538 | stream.BeginInlineInfoEntry(&art_method, 3, 3); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 539 | stream.AddDexRegisterEntry(Kind::kInStack, 16); |
| 540 | stream.AddDexRegisterEntry(Kind::kConstant, 20); |
| 541 | stream.AddDexRegisterEntry(Kind::kInRegister, 15); |
| 542 | stream.EndInlineInfoEntry(); |
| 543 | |
| 544 | stream.EndStackMapEntry(); |
| 545 | |
| 546 | // Second stack map. |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 547 | stream.BeginStackMapEntry(2, 22 * kPcAlign, 0x3, &sp_mask1, 2, 3); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 548 | stream.AddDexRegisterEntry(Kind::kInStack, 56); |
| 549 | stream.AddDexRegisterEntry(Kind::kConstant, 0); |
| 550 | |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 551 | stream.BeginInlineInfoEntry(&art_method, 2, 1); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 552 | stream.AddDexRegisterEntry(Kind::kInStack, 12); |
| 553 | stream.EndInlineInfoEntry(); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 554 | stream.BeginInlineInfoEntry(&art_method, 3, 3); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 555 | stream.AddDexRegisterEntry(Kind::kInStack, 80); |
| 556 | stream.AddDexRegisterEntry(Kind::kConstant, 10); |
| 557 | stream.AddDexRegisterEntry(Kind::kInRegister, 5); |
| 558 | stream.EndInlineInfoEntry(); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 559 | stream.BeginInlineInfoEntry(&art_method, 5, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 560 | stream.EndInlineInfoEntry(); |
| 561 | |
| 562 | stream.EndStackMapEntry(); |
| 563 | |
| 564 | // Third stack map. |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 565 | stream.BeginStackMapEntry(4, 56 * kPcAlign, 0x3, &sp_mask1, 2, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 566 | stream.AddDexRegisterEntry(Kind::kNone, 0); |
| 567 | stream.AddDexRegisterEntry(Kind::kConstant, 4); |
| 568 | stream.EndStackMapEntry(); |
| 569 | |
| 570 | // Fourth stack map. |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 571 | stream.BeginStackMapEntry(6, 78 * kPcAlign, 0x3, &sp_mask1, 2, 3); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 572 | stream.AddDexRegisterEntry(Kind::kInStack, 56); |
| 573 | stream.AddDexRegisterEntry(Kind::kConstant, 0); |
| 574 | |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 575 | stream.BeginInlineInfoEntry(&art_method, 2, 0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 576 | stream.EndInlineInfoEntry(); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 577 | stream.BeginInlineInfoEntry(&art_method, 5, 1); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 578 | stream.AddDexRegisterEntry(Kind::kInRegister, 2); |
| 579 | stream.EndInlineInfoEntry(); |
Nicolas Geoffray | 5d37c15 | 2017-01-12 13:25:19 +0000 | [diff] [blame] | 580 | stream.BeginInlineInfoEntry(&art_method, 10, 2); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 581 | stream.AddDexRegisterEntry(Kind::kNone, 0); |
| 582 | stream.AddDexRegisterEntry(Kind::kInRegister, 3); |
| 583 | stream.EndInlineInfoEntry(); |
| 584 | |
| 585 | stream.EndStackMapEntry(); |
| 586 | |
| 587 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 588 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 589 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 590 | stream.FillInCodeInfo(region); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 591 | |
| 592 | CodeInfo ci(region); |
| 593 | |
| 594 | { |
| 595 | // Verify first stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 596 | StackMap sm0 = ci.GetStackMapAt(0); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 597 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 598 | DexRegisterMap dex_registers0 = ci.GetDexRegisterMapOf(sm0); |
| 599 | ASSERT_EQ(2u, dex_registers0.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 600 | ASSERT_EQ(0, dex_registers0[0].GetStackOffsetInBytes()); |
| 601 | ASSERT_EQ(4, dex_registers0[1].GetConstant()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 602 | |
David Srbecky | 6e69e52 | 2018-06-03 12:00:14 +0100 | [diff] [blame] | 603 | InlineInfo if0_0 = ci.GetInlineInfoAtDepth(sm0, 0); |
| 604 | InlineInfo if0_1 = ci.GetInlineInfoAtDepth(sm0, 1); |
| 605 | ASSERT_EQ(2u, ci.GetInlineDepthOf(sm0)); |
| 606 | ASSERT_EQ(2u, if0_0.GetDexPc()); |
| 607 | ASSERT_TRUE(if0_0.EncodesArtMethod()); |
| 608 | ASSERT_EQ(3u, if0_1.GetDexPc()); |
| 609 | ASSERT_TRUE(if0_1.EncodesArtMethod()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 610 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 611 | DexRegisterMap dex_registers1 = ci.GetDexRegisterMapAtDepth(0, sm0); |
| 612 | ASSERT_EQ(1u, dex_registers1.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 613 | ASSERT_EQ(8, dex_registers1[0].GetStackOffsetInBytes()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 614 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 615 | DexRegisterMap dex_registers2 = ci.GetDexRegisterMapAtDepth(1, sm0); |
| 616 | ASSERT_EQ(3u, dex_registers2.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 617 | ASSERT_EQ(16, dex_registers2[0].GetStackOffsetInBytes()); |
| 618 | ASSERT_EQ(20, dex_registers2[1].GetConstant()); |
| 619 | ASSERT_EQ(15, dex_registers2[2].GetMachineRegister()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 620 | } |
| 621 | |
| 622 | { |
| 623 | // Verify second stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 624 | StackMap sm1 = ci.GetStackMapAt(1); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 625 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 626 | DexRegisterMap dex_registers0 = ci.GetDexRegisterMapOf(sm1); |
| 627 | ASSERT_EQ(2u, dex_registers0.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 628 | ASSERT_EQ(56, dex_registers0[0].GetStackOffsetInBytes()); |
| 629 | ASSERT_EQ(0, dex_registers0[1].GetConstant()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 630 | |
David Srbecky | 6e69e52 | 2018-06-03 12:00:14 +0100 | [diff] [blame] | 631 | InlineInfo if1_0 = ci.GetInlineInfoAtDepth(sm1, 0); |
| 632 | InlineInfo if1_1 = ci.GetInlineInfoAtDepth(sm1, 1); |
| 633 | InlineInfo if1_2 = ci.GetInlineInfoAtDepth(sm1, 2); |
| 634 | ASSERT_EQ(3u, ci.GetInlineDepthOf(sm1)); |
| 635 | ASSERT_EQ(2u, if1_0.GetDexPc()); |
| 636 | ASSERT_TRUE(if1_0.EncodesArtMethod()); |
| 637 | ASSERT_EQ(3u, if1_1.GetDexPc()); |
| 638 | ASSERT_TRUE(if1_1.EncodesArtMethod()); |
| 639 | ASSERT_EQ(5u, if1_2.GetDexPc()); |
| 640 | ASSERT_TRUE(if1_2.EncodesArtMethod()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 641 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 642 | DexRegisterMap dex_registers1 = ci.GetDexRegisterMapAtDepth(0, sm1); |
| 643 | ASSERT_EQ(1u, dex_registers1.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 644 | ASSERT_EQ(12, dex_registers1[0].GetStackOffsetInBytes()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 645 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 646 | DexRegisterMap dex_registers2 = ci.GetDexRegisterMapAtDepth(1, sm1); |
| 647 | ASSERT_EQ(3u, dex_registers2.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 648 | ASSERT_EQ(80, dex_registers2[0].GetStackOffsetInBytes()); |
| 649 | ASSERT_EQ(10, dex_registers2[1].GetConstant()); |
| 650 | ASSERT_EQ(5, dex_registers2[2].GetMachineRegister()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 651 | } |
| 652 | |
| 653 | { |
| 654 | // Verify third stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 655 | StackMap sm2 = ci.GetStackMapAt(2); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 656 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 657 | DexRegisterMap dex_registers0 = ci.GetDexRegisterMapOf(sm2); |
| 658 | ASSERT_EQ(2u, dex_registers0.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 659 | ASSERT_FALSE(dex_registers0[0].IsLive()); |
| 660 | ASSERT_EQ(4, dex_registers0[1].GetConstant()); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 661 | ASSERT_FALSE(sm2.HasInlineInfo()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 662 | } |
| 663 | |
| 664 | { |
| 665 | // Verify fourth stack map. |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 666 | StackMap sm3 = ci.GetStackMapAt(3); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 667 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 668 | DexRegisterMap dex_registers0 = ci.GetDexRegisterMapOf(sm3); |
| 669 | ASSERT_EQ(2u, dex_registers0.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 670 | ASSERT_EQ(56, dex_registers0[0].GetStackOffsetInBytes()); |
| 671 | ASSERT_EQ(0, dex_registers0[1].GetConstant()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 672 | |
David Srbecky | 6e69e52 | 2018-06-03 12:00:14 +0100 | [diff] [blame] | 673 | InlineInfo if2_0 = ci.GetInlineInfoAtDepth(sm3, 0); |
| 674 | InlineInfo if2_1 = ci.GetInlineInfoAtDepth(sm3, 1); |
| 675 | InlineInfo if2_2 = ci.GetInlineInfoAtDepth(sm3, 2); |
| 676 | ASSERT_EQ(3u, ci.GetInlineDepthOf(sm3)); |
| 677 | ASSERT_EQ(2u, if2_0.GetDexPc()); |
| 678 | ASSERT_TRUE(if2_0.EncodesArtMethod()); |
| 679 | ASSERT_EQ(5u, if2_1.GetDexPc()); |
| 680 | ASSERT_TRUE(if2_1.EncodesArtMethod()); |
| 681 | ASSERT_EQ(10u, if2_2.GetDexPc()); |
| 682 | ASSERT_TRUE(if2_2.EncodesArtMethod()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 683 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 684 | DexRegisterMap dex_registers1 = ci.GetDexRegisterMapAtDepth(1, sm3); |
| 685 | ASSERT_EQ(1u, dex_registers1.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 686 | ASSERT_EQ(2, dex_registers1[0].GetMachineRegister()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 687 | |
David Srbecky | fd89b07 | 2018-06-03 12:00:22 +0100 | [diff] [blame] | 688 | DexRegisterMap dex_registers2 = ci.GetDexRegisterMapAtDepth(2, sm3); |
| 689 | ASSERT_EQ(2u, dex_registers2.size()); |
David Srbecky | e140212 | 2018-06-13 18:20:45 +0100 | [diff] [blame] | 690 | ASSERT_FALSE(dex_registers2[0].IsLive()); |
| 691 | ASSERT_EQ(3, dex_registers2[1].GetMachineRegister()); |
Nicolas Geoffray | b1d0f3f | 2015-05-14 12:41:51 +0100 | [diff] [blame] | 692 | } |
| 693 | } |
| 694 | |
David Srbecky | d02b23f | 2018-05-29 23:27:22 +0100 | [diff] [blame] | 695 | TEST(StackMapTest, PackedNativePcTest) { |
David Srbecky | 71ec1cc | 2018-05-18 15:57:25 +0100 | [diff] [blame] | 696 | // Test minimum alignments, and decoding. |
David Srbecky | d02b23f | 2018-05-29 23:27:22 +0100 | [diff] [blame] | 697 | uint32_t packed_thumb2 = |
| 698 | StackMap::PackNativePc(kThumb2InstructionAlignment, InstructionSet::kThumb2); |
| 699 | uint32_t packed_arm64 = |
| 700 | StackMap::PackNativePc(kArm64InstructionAlignment, InstructionSet::kArm64); |
| 701 | uint32_t packed_x86 = |
| 702 | StackMap::PackNativePc(kX86InstructionAlignment, InstructionSet::kX86); |
| 703 | uint32_t packed_x86_64 = |
| 704 | StackMap::PackNativePc(kX86_64InstructionAlignment, InstructionSet::kX86_64); |
| 705 | uint32_t packed_mips = |
| 706 | StackMap::PackNativePc(kMipsInstructionAlignment, InstructionSet::kMips); |
| 707 | uint32_t packed_mips64 = |
| 708 | StackMap::PackNativePc(kMips64InstructionAlignment, InstructionSet::kMips64); |
| 709 | EXPECT_EQ(StackMap::UnpackNativePc(packed_thumb2, InstructionSet::kThumb2), |
| 710 | kThumb2InstructionAlignment); |
| 711 | EXPECT_EQ(StackMap::UnpackNativePc(packed_arm64, InstructionSet::kArm64), |
| 712 | kArm64InstructionAlignment); |
| 713 | EXPECT_EQ(StackMap::UnpackNativePc(packed_x86, InstructionSet::kX86), |
| 714 | kX86InstructionAlignment); |
| 715 | EXPECT_EQ(StackMap::UnpackNativePc(packed_x86_64, InstructionSet::kX86_64), |
| 716 | kX86_64InstructionAlignment); |
| 717 | EXPECT_EQ(StackMap::UnpackNativePc(packed_mips, InstructionSet::kMips), |
| 718 | kMipsInstructionAlignment); |
| 719 | EXPECT_EQ(StackMap::UnpackNativePc(packed_mips64, InstructionSet::kMips64), |
| 720 | kMips64InstructionAlignment); |
Mathieu Chartier | a2f526f | 2017-01-19 14:48:48 -0800 | [diff] [blame] | 721 | } |
| 722 | |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 723 | TEST(StackMapTest, TestDeduplicateStackMask) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 724 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 725 | ArenaStack arena_stack(&pool); |
| 726 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 727 | StackMapStream stream(&allocator, kRuntimeISA); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 728 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 729 | ArenaBitVector sp_mask(&allocator, 0, true); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 730 | sp_mask.SetBit(1); |
| 731 | sp_mask.SetBit(4); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 732 | stream.BeginStackMapEntry(0, 4 * kPcAlign, 0x3, &sp_mask, 0, 0); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 733 | stream.EndStackMapEntry(); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 734 | stream.BeginStackMapEntry(0, 8 * kPcAlign, 0x3, &sp_mask, 0, 0); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 735 | stream.EndStackMapEntry(); |
| 736 | |
| 737 | size_t size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 738 | void* memory = allocator.Alloc(size, kArenaAllocMisc); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 739 | MemoryRegion region(memory, size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 740 | stream.FillInCodeInfo(region); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 741 | |
| 742 | CodeInfo code_info(region); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 743 | ASSERT_EQ(2u, code_info.GetNumberOfStackMaps()); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 744 | |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 745 | StackMap stack_map1 = code_info.GetStackMapForNativePcOffset(4 * kPcAlign); |
| 746 | StackMap stack_map2 = code_info.GetStackMapForNativePcOffset(8 * kPcAlign); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 747 | EXPECT_EQ(stack_map1.GetStackMaskIndex(), |
| 748 | stack_map2.GetStackMaskIndex()); |
David Srbecky | 45aa598 | 2016-03-18 02:15:09 +0000 | [diff] [blame] | 749 | } |
| 750 | |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 751 | TEST(StackMapTest, TestInvokeInfo) { |
David Sehr | 3215fff | 2018-04-03 17:10:12 -0700 | [diff] [blame] | 752 | MallocArenaPool pool; |
Vladimir Marko | 174b2e2 | 2017-10-12 13:34:49 +0100 | [diff] [blame] | 753 | ArenaStack arena_stack(&pool); |
| 754 | ScopedArenaAllocator allocator(&arena_stack); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 755 | StackMapStream stream(&allocator, kRuntimeISA); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 756 | |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 757 | ArenaBitVector sp_mask(&allocator, 0, true); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 758 | sp_mask.SetBit(1); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 759 | stream.BeginStackMapEntry(0, 4 * kPcAlign, 0x3, &sp_mask, 0, 0); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 760 | stream.AddInvoke(kSuper, 1); |
| 761 | stream.EndStackMapEntry(); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 762 | stream.BeginStackMapEntry(0, 8 * kPcAlign, 0x3, &sp_mask, 0, 0); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 763 | stream.AddInvoke(kStatic, 3); |
| 764 | stream.EndStackMapEntry(); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 765 | stream.BeginStackMapEntry(0, 16 * kPcAlign, 0x3, &sp_mask, 0, 0); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 766 | stream.AddInvoke(kDirect, 65535); |
| 767 | stream.EndStackMapEntry(); |
| 768 | |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 769 | const size_t code_info_size = stream.PrepareForFillIn(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 770 | MemoryRegion code_info_region(allocator.Alloc(code_info_size, kArenaAllocMisc), code_info_size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 771 | stream.FillInCodeInfo(code_info_region); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 772 | |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 773 | const size_t method_info_size = stream.ComputeMethodInfoSize(); |
Vladimir Marko | 69d310e | 2017-10-09 14:12:23 +0100 | [diff] [blame] | 774 | MemoryRegion method_info_region(allocator.Alloc(method_info_size, kArenaAllocMisc), |
| 775 | method_info_size); |
Mathieu Chartier | cbcedbf | 2017-03-12 22:24:50 -0700 | [diff] [blame] | 776 | stream.FillInMethodInfo(method_info_region); |
| 777 | |
| 778 | CodeInfo code_info(code_info_region); |
| 779 | MethodInfo method_info(method_info_region.begin()); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 780 | ASSERT_EQ(3u, code_info.GetNumberOfStackMaps()); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 781 | |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 782 | InvokeInfo invoke1(code_info.GetInvokeInfoForNativePcOffset(4 * kPcAlign)); |
| 783 | InvokeInfo invoke2(code_info.GetInvokeInfoForNativePcOffset(8 * kPcAlign)); |
| 784 | InvokeInfo invoke3(code_info.GetInvokeInfoForNativePcOffset(16 * kPcAlign)); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 785 | InvokeInfo invoke_invalid(code_info.GetInvokeInfoForNativePcOffset(12)); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 786 | EXPECT_FALSE(invoke_invalid.IsValid()); // No entry for that index. |
| 787 | EXPECT_TRUE(invoke1.IsValid()); |
| 788 | EXPECT_TRUE(invoke2.IsValid()); |
| 789 | EXPECT_TRUE(invoke3.IsValid()); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 790 | EXPECT_EQ(invoke1.GetInvokeType(), kSuper); |
| 791 | EXPECT_EQ(invoke1.GetMethodIndex(method_info), 1u); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 792 | EXPECT_EQ(invoke1.GetNativePcOffset(kRuntimeISA), 4u * kPcAlign); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 793 | EXPECT_EQ(invoke2.GetInvokeType(), kStatic); |
| 794 | EXPECT_EQ(invoke2.GetMethodIndex(method_info), 3u); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 795 | EXPECT_EQ(invoke2.GetNativePcOffset(kRuntimeISA), 8u * kPcAlign); |
David Srbecky | 052f8ca | 2018-04-26 15:42:54 +0100 | [diff] [blame] | 796 | EXPECT_EQ(invoke3.GetInvokeType(), kDirect); |
| 797 | EXPECT_EQ(invoke3.GetMethodIndex(method_info), 65535u); |
David Srbecky | d775f96 | 2018-05-30 18:12:52 +0100 | [diff] [blame] | 798 | EXPECT_EQ(invoke3.GetNativePcOffset(kRuntimeISA), 16u * kPcAlign); |
Mathieu Chartier | d776ff0 | 2017-01-17 09:32:18 -0800 | [diff] [blame] | 799 | } |
| 800 | |
Nicolas Geoffray | 99ea58c | 2014-07-02 15:08:17 +0100 | [diff] [blame] | 801 | } // namespace art |