Elliott Hughes | 2faa5f1 | 2012-01-30 14:42:07 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2011 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 | */ |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 16 | |
| 17 | #include "intern_table.h" |
| 18 | |
| 19 | #include "common_test.h" |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 20 | #include "mirror/object.h" |
Ian Rogers | 1f53934 | 2012-10-03 21:09:42 -0700 | [diff] [blame] | 21 | #include "sirt_ref.h" |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 22 | |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 23 | namespace art { |
| 24 | |
Brian Carlstrom | f734cf5 | 2011-08-17 16:28:14 -0700 | [diff] [blame] | 25 | class InternTableTest : public CommonTest {}; |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 26 | |
| 27 | TEST_F(InternTableTest, Intern) { |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 28 | ScopedObjectAccess soa(Thread::Current()); |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 29 | InternTable intern_table; |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 30 | SirtRef<mirror::String> foo_1(soa.Self(), intern_table.InternStrong(3, "foo")); |
| 31 | SirtRef<mirror::String> foo_2(soa.Self(), intern_table.InternStrong(3, "foo")); |
| 32 | SirtRef<mirror::String> foo_3(soa.Self(), mirror::String::AllocFromModifiedUtf8(soa.Self(), "foo")); |
| 33 | SirtRef<mirror::String> bar(soa.Self(), intern_table.InternStrong(3, "bar")); |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 34 | EXPECT_TRUE(foo_1->Equals("foo")); |
| 35 | EXPECT_TRUE(foo_2->Equals("foo")); |
| 36 | EXPECT_TRUE(foo_3->Equals("foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 37 | EXPECT_TRUE(foo_1.get() != NULL); |
| 38 | EXPECT_TRUE(foo_2.get() != NULL); |
| 39 | EXPECT_EQ(foo_1.get(), foo_2.get()); |
| 40 | EXPECT_NE(foo_1.get(), bar.get()); |
| 41 | EXPECT_NE(foo_2.get(), bar.get()); |
| 42 | EXPECT_NE(foo_3.get(), bar.get()); |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 43 | } |
| 44 | |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 45 | TEST_F(InternTableTest, Size) { |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 46 | ScopedObjectAccess soa(Thread::Current()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 47 | InternTable t; |
| 48 | EXPECT_EQ(0U, t.Size()); |
| 49 | t.InternStrong(3, "foo"); |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 50 | SirtRef<mirror::String> foo(soa.Self(), mirror::String::AllocFromModifiedUtf8(soa.Self(), "foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 51 | t.InternWeak(foo.get()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 52 | EXPECT_EQ(1U, t.Size()); |
| 53 | t.InternStrong(3, "bar"); |
| 54 | EXPECT_EQ(2U, t.Size()); |
| 55 | } |
| 56 | |
Elliott Hughes | c33a32b | 2011-10-11 18:18:07 -0700 | [diff] [blame] | 57 | class TestPredicate { |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 58 | public: |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 59 | bool IsMarked(const mirror::Object* s) const { |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 60 | bool erased = false; |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 61 | typedef std::vector<const mirror::String*>::iterator It; // TODO: C++0x auto |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 62 | for (It it = expected_.begin(), end = expected_.end(); it != end; ++it) { |
| 63 | if (*it == s) { |
| 64 | expected_.erase(it); |
| 65 | erased = true; |
| 66 | break; |
| 67 | } |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 68 | } |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 69 | EXPECT_TRUE(erased); |
Elliott Hughes | c33a32b | 2011-10-11 18:18:07 -0700 | [diff] [blame] | 70 | return false; |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 71 | } |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 72 | |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 73 | void Expect(const mirror::String* s) { |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 74 | expected_.push_back(s); |
| 75 | } |
| 76 | |
| 77 | ~TestPredicate() { |
| 78 | EXPECT_EQ(0U, expected_.size()); |
| 79 | } |
| 80 | |
| 81 | private: |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 82 | mutable std::vector<const mirror::String*> expected_; |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 83 | }; |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 84 | |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 85 | bool IsMarked(const mirror::Object* object, void* arg) { |
Elliott Hughes | c33a32b | 2011-10-11 18:18:07 -0700 | [diff] [blame] | 86 | return reinterpret_cast<TestPredicate*>(arg)->IsMarked(object); |
| 87 | } |
| 88 | |
| 89 | TEST_F(InternTableTest, SweepInternTableWeaks) { |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 90 | ScopedObjectAccess soa(Thread::Current()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 91 | InternTable t; |
| 92 | t.InternStrong(3, "foo"); |
| 93 | t.InternStrong(3, "bar"); |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 94 | SirtRef<mirror::String> hello(soa.Self(), |
| 95 | mirror::String::AllocFromModifiedUtf8(soa.Self(), "hello")); |
| 96 | SirtRef<mirror::String> world(soa.Self(), |
| 97 | mirror::String::AllocFromModifiedUtf8(soa.Self(), "world")); |
| 98 | SirtRef<mirror::String> s0(soa.Self(), t.InternWeak(hello.get())); |
| 99 | SirtRef<mirror::String> s1(soa.Self(), t.InternWeak(world.get())); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 100 | |
| 101 | EXPECT_EQ(4U, t.Size()); |
| 102 | |
| 103 | // We should traverse only the weaks... |
Elliott Hughes | 410c0c8 | 2011-09-01 17:58:25 -0700 | [diff] [blame] | 104 | TestPredicate p; |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 105 | p.Expect(s0.get()); |
| 106 | p.Expect(s1.get()); |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 107 | { |
Ian Rogers | 1f53934 | 2012-10-03 21:09:42 -0700 | [diff] [blame] | 108 | ReaderMutexLock mu(soa.Self(), *Locks::heap_bitmap_lock_); |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 109 | t.SweepInternTableWeaks(IsMarked, &p); |
| 110 | } |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 111 | |
| 112 | EXPECT_EQ(2U, t.Size()); |
| 113 | |
Elliott Hughes | e5448b5 | 2012-01-18 16:44:06 -0800 | [diff] [blame] | 114 | // Just check that we didn't corrupt the map. |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 115 | SirtRef<mirror::String> still_here(soa.Self(), |
| 116 | mirror::String::AllocFromModifiedUtf8(soa.Self(), "still here")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 117 | t.InternWeak(still_here.get()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 118 | EXPECT_EQ(3U, t.Size()); |
| 119 | } |
| 120 | |
| 121 | TEST_F(InternTableTest, ContainsWeak) { |
Ian Rogers | 00f7d0e | 2012-07-19 15:28:27 -0700 | [diff] [blame] | 122 | ScopedObjectAccess soa(Thread::Current()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 123 | { |
| 124 | // Strongs are never weak. |
| 125 | InternTable t; |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 126 | SirtRef<mirror::String> interned_foo_1(soa.Self(), t.InternStrong(3, "foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 127 | EXPECT_FALSE(t.ContainsWeak(interned_foo_1.get())); |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 128 | SirtRef<mirror::String> interned_foo_2(soa.Self(), t.InternStrong(3, "foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 129 | EXPECT_FALSE(t.ContainsWeak(interned_foo_2.get())); |
| 130 | EXPECT_EQ(interned_foo_1.get(), interned_foo_2.get()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 131 | } |
| 132 | |
| 133 | { |
| 134 | // Weaks are always weak. |
| 135 | InternTable t; |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 136 | SirtRef<mirror::String> foo_1(soa.Self(), |
| 137 | mirror::String::AllocFromModifiedUtf8(soa.Self(), "foo")); |
| 138 | SirtRef<mirror::String> foo_2(soa.Self(), |
| 139 | mirror::String::AllocFromModifiedUtf8(soa.Self(), "foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 140 | EXPECT_NE(foo_1.get(), foo_2.get()); |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 141 | SirtRef<mirror::String> interned_foo_1(soa.Self(), t.InternWeak(foo_1.get())); |
| 142 | SirtRef<mirror::String> interned_foo_2(soa.Self(), t.InternWeak(foo_2.get())); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 143 | EXPECT_TRUE(t.ContainsWeak(interned_foo_2.get())); |
| 144 | EXPECT_EQ(interned_foo_1.get(), interned_foo_2.get()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 145 | } |
| 146 | |
| 147 | { |
| 148 | // A weak can be promoted to a strong. |
| 149 | InternTable t; |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 150 | SirtRef<mirror::String> foo(soa.Self(), mirror::String::AllocFromModifiedUtf8(soa.Self(), "foo")); |
| 151 | SirtRef<mirror::String> interned_foo_1(soa.Self(), t.InternWeak(foo.get())); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 152 | EXPECT_TRUE(t.ContainsWeak(interned_foo_1.get())); |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 153 | SirtRef<mirror::String> interned_foo_2(soa.Self(), t.InternStrong(3, "foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 154 | EXPECT_FALSE(t.ContainsWeak(interned_foo_2.get())); |
| 155 | EXPECT_EQ(interned_foo_1.get(), interned_foo_2.get()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 156 | } |
| 157 | |
| 158 | { |
| 159 | // Interning a weak after a strong gets you the strong. |
| 160 | InternTable t; |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 161 | SirtRef<mirror::String> interned_foo_1(soa.Self(), t.InternStrong(3, "foo")); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 162 | EXPECT_FALSE(t.ContainsWeak(interned_foo_1.get())); |
Ian Rogers | 2dd0e2c | 2013-01-24 12:42:14 -0800 | [diff] [blame] | 163 | SirtRef<mirror::String> foo(soa.Self(), |
| 164 | mirror::String::AllocFromModifiedUtf8(soa.Self(), "foo")); |
| 165 | SirtRef<mirror::String> interned_foo_2(soa.Self(), t.InternWeak(foo.get())); |
Brian Carlstrom | 40381fb | 2011-10-19 14:13:40 -0700 | [diff] [blame] | 166 | EXPECT_FALSE(t.ContainsWeak(interned_foo_2.get())); |
| 167 | EXPECT_EQ(interned_foo_1.get(), interned_foo_2.get()); |
Elliott Hughes | cf4c6c4 | 2011-09-01 15:16:42 -0700 | [diff] [blame] | 168 | } |
| 169 | } |
| 170 | |
Brian Carlstrom | 7e93b50 | 2011-08-04 14:16:22 -0700 | [diff] [blame] | 171 | } // namespace art |