Home | History | Annotate | Download | only in runtime
      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  */
     16 
     17 #ifndef ART_RUNTIME_INTERN_TABLE_H_
     18 #define ART_RUNTIME_INTERN_TABLE_H_
     19 
     20 #include <unordered_set>
     21 
     22 #include "atomic.h"
     23 #include "base/allocator.h"
     24 #include "base/hash_set.h"
     25 #include "base/mutex.h"
     26 #include "gc_root.h"
     27 #include "gc/weak_root_state.h"
     28 #include "object_callbacks.h"
     29 
     30 namespace art {
     31 
     32 namespace gc {
     33 namespace space {
     34 class ImageSpace;
     35 }  // namespace space
     36 }  // namespace gc
     37 
     38 enum VisitRootFlags : uint8_t;
     39 
     40 namespace mirror {
     41 class String;
     42 }  // namespace mirror
     43 class Transaction;
     44 
     45 /**
     46  * Used to intern strings.
     47  *
     48  * There are actually two tables: one that holds strong references to its strings, and one that
     49  * holds weak references. The former is used for string literals, for which there is an effective
     50  * reference from the constant pool. The latter is used for strings interned at runtime via
     51  * String.intern. Some code (XML parsers being a prime example) relies on being able to intern
     52  * arbitrarily many strings for the duration of a parse without permanently increasing the memory
     53  * footprint.
     54  */
     55 class InternTable {
     56  public:
     57   InternTable();
     58 
     59   // Interns a potentially new string in the 'strong' table. May cause thread suspension.
     60   mirror::String* InternStrong(int32_t utf16_length, const char* utf8_data)
     61       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(!Roles::uninterruptible_);
     62 
     63   // Only used by image writer. Special version that may not cause thread suspension since the GC
     64   // cannot be running while we are doing image writing. Maybe be called while while holding a
     65   // lock since there will not be thread suspension.
     66   mirror::String* InternStrongImageString(mirror::String* s)
     67       SHARED_REQUIRES(Locks::mutator_lock_);
     68 
     69   // Interns a potentially new string in the 'strong' table. May cause thread suspension.
     70   mirror::String* InternStrong(const char* utf8_data) SHARED_REQUIRES(Locks::mutator_lock_)
     71       REQUIRES(!Roles::uninterruptible_);
     72 
     73   // Interns a potentially new string in the 'strong' table. May cause thread suspension.
     74   mirror::String* InternStrong(mirror::String* s) SHARED_REQUIRES(Locks::mutator_lock_)
     75       REQUIRES(!Roles::uninterruptible_);
     76 
     77   // Interns a potentially new string in the 'weak' table. May cause thread suspension.
     78   mirror::String* InternWeak(mirror::String* s) SHARED_REQUIRES(Locks::mutator_lock_)
     79       REQUIRES(!Roles::uninterruptible_);
     80 
     81   void SweepInternTableWeaks(IsMarkedVisitor* visitor) SHARED_REQUIRES(Locks::mutator_lock_)
     82       REQUIRES(!Locks::intern_table_lock_);
     83 
     84   bool ContainsWeak(mirror::String* s) SHARED_REQUIRES(Locks::mutator_lock_)
     85       REQUIRES(!Locks::intern_table_lock_);
     86 
     87   // Lookup a strong intern, returns null if not found.
     88   mirror::String* LookupStrong(Thread* self, mirror::String* s)
     89       REQUIRES(!Locks::intern_table_lock_)
     90       SHARED_REQUIRES(Locks::mutator_lock_);
     91   mirror::String* LookupStrong(Thread* self, uint32_t utf16_length, const char* utf8_data)
     92       REQUIRES(!Locks::intern_table_lock_)
     93       SHARED_REQUIRES(Locks::mutator_lock_);
     94 
     95   // Lookup a weak intern, returns null if not found.
     96   mirror::String* LookupWeak(Thread* self, mirror::String* s)
     97       REQUIRES(!Locks::intern_table_lock_)
     98       SHARED_REQUIRES(Locks::mutator_lock_);
     99 
    100   // Total number of interned strings.
    101   size_t Size() const REQUIRES(!Locks::intern_table_lock_);
    102 
    103   // Total number of weakly live interned strings.
    104   size_t StrongSize() const REQUIRES(!Locks::intern_table_lock_);
    105 
    106   // Total number of strongly live interned strings.
    107   size_t WeakSize() const REQUIRES(!Locks::intern_table_lock_);
    108 
    109   void VisitRoots(RootVisitor* visitor, VisitRootFlags flags)
    110       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(!Locks::intern_table_lock_);
    111 
    112   void DumpForSigQuit(std::ostream& os) const REQUIRES(!Locks::intern_table_lock_);
    113 
    114   void BroadcastForNewInterns() SHARED_REQUIRES(Locks::mutator_lock_);
    115 
    116   // Adds all of the resolved image strings from the image spaces into the intern table. The
    117   // advantage of doing this is preventing expensive DexFile::FindStringId calls. Sets
    118   // images_added_to_intern_table_ to true.
    119   void AddImagesStringsToTable(const std::vector<gc::space::ImageSpace*>& image_spaces)
    120       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(!Locks::intern_table_lock_);
    121 
    122   // Add a new intern table for inserting to, previous intern tables are still there but no
    123   // longer inserted into and ideally unmodified. This is done to prevent dirty pages.
    124   void AddNewTable()
    125       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(!Locks::intern_table_lock_);
    126 
    127   // Read the intern table from memory. The elements aren't copied, the intern hash set data will
    128   // point to somewhere within ptr. Only reads the strong interns.
    129   size_t AddTableFromMemory(const uint8_t* ptr) REQUIRES(!Locks::intern_table_lock_)
    130       SHARED_REQUIRES(Locks::mutator_lock_);
    131 
    132   // Write the post zygote intern table to a pointer. Only writes the strong interns since it is
    133   // expected that there is no weak interns since this is called from the image writer.
    134   size_t WriteToMemory(uint8_t* ptr) SHARED_REQUIRES(Locks::mutator_lock_)
    135       REQUIRES(!Locks::intern_table_lock_);
    136 
    137   // Change the weak root state. May broadcast to waiters.
    138   void ChangeWeakRootState(gc::WeakRootState new_state)
    139       REQUIRES(!Locks::intern_table_lock_);
    140 
    141  private:
    142   // Modified UTF-8-encoded string treated as UTF16.
    143   class Utf8String {
    144    public:
    145     Utf8String(uint32_t utf16_length, const char* utf8_data, int32_t hash)
    146         : hash_(hash), utf16_length_(utf16_length), utf8_data_(utf8_data) { }
    147 
    148     int32_t GetHash() const { return hash_; }
    149     uint32_t GetUtf16Length() const { return utf16_length_; }
    150     const char* GetUtf8Data() const { return utf8_data_; }
    151 
    152    private:
    153     int32_t hash_;
    154     uint32_t utf16_length_;
    155     const char* utf8_data_;
    156   };
    157 
    158   class StringHashEquals {
    159    public:
    160     std::size_t operator()(const GcRoot<mirror::String>& root) const NO_THREAD_SAFETY_ANALYSIS;
    161     bool operator()(const GcRoot<mirror::String>& a, const GcRoot<mirror::String>& b) const
    162         NO_THREAD_SAFETY_ANALYSIS;
    163 
    164     // Utf8String can be used for lookup.
    165     std::size_t operator()(const Utf8String& key) const { return key.GetHash(); }
    166     bool operator()(const GcRoot<mirror::String>& a, const Utf8String& b) const
    167         NO_THREAD_SAFETY_ANALYSIS;
    168   };
    169   class GcRootEmptyFn {
    170    public:
    171     void MakeEmpty(GcRoot<mirror::String>& item) const {
    172       item = GcRoot<mirror::String>();
    173     }
    174     bool IsEmpty(const GcRoot<mirror::String>& item) const {
    175       return item.IsNull();
    176     }
    177   };
    178 
    179   // Table which holds pre zygote and post zygote interned strings. There is one instance for
    180   // weak interns and strong interns.
    181   class Table {
    182    public:
    183     Table();
    184     mirror::String* Find(mirror::String* s) SHARED_REQUIRES(Locks::mutator_lock_)
    185         REQUIRES(Locks::intern_table_lock_);
    186     mirror::String* Find(const Utf8String& string) SHARED_REQUIRES(Locks::mutator_lock_)
    187         REQUIRES(Locks::intern_table_lock_);
    188     void Insert(mirror::String* s) SHARED_REQUIRES(Locks::mutator_lock_)
    189         REQUIRES(Locks::intern_table_lock_);
    190     void Remove(mirror::String* s)
    191         SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    192     void VisitRoots(RootVisitor* visitor)
    193         SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    194     void SweepWeaks(IsMarkedVisitor* visitor)
    195         SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    196     // Add a new intern table that will only be inserted into from now on.
    197     void AddNewTable() REQUIRES(Locks::intern_table_lock_);
    198     size_t Size() const REQUIRES(Locks::intern_table_lock_);
    199     // Read and add an intern table from ptr.
    200     // Tables read are inserted at the front of the table array. Only checks for conflicts in
    201     // debug builds. Returns how many bytes were read.
    202     size_t AddTableFromMemory(const uint8_t* ptr)
    203         REQUIRES(Locks::intern_table_lock_) SHARED_REQUIRES(Locks::mutator_lock_);
    204     // Write the intern tables to ptr, if there are multiple tables they are combined into a single
    205     // one. Returns how many bytes were written.
    206     size_t WriteToMemory(uint8_t* ptr)
    207         REQUIRES(Locks::intern_table_lock_) SHARED_REQUIRES(Locks::mutator_lock_);
    208 
    209    private:
    210     typedef HashSet<GcRoot<mirror::String>, GcRootEmptyFn, StringHashEquals, StringHashEquals,
    211         TrackingAllocator<GcRoot<mirror::String>, kAllocatorTagInternTable>> UnorderedSet;
    212 
    213     void SweepWeaks(UnorderedSet* set, IsMarkedVisitor* visitor)
    214         SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    215 
    216     // We call AddNewTable when we create the zygote to reduce private dirty pages caused by
    217     // modifying the zygote intern table. The back of table is modified when strings are interned.
    218     std::vector<UnorderedSet> tables_;
    219   };
    220 
    221   // Insert if non null, otherwise return null. Must be called holding the mutator lock.
    222   // If holding_locks is true, then we may also hold other locks. If holding_locks is true, then we
    223   // require GC is not running since it is not safe to wait while holding locks.
    224   mirror::String* Insert(mirror::String* s, bool is_strong, bool holding_locks)
    225       REQUIRES(!Locks::intern_table_lock_) SHARED_REQUIRES(Locks::mutator_lock_);
    226 
    227   mirror::String* LookupStrongLocked(mirror::String* s)
    228       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    229   mirror::String* LookupWeakLocked(mirror::String* s)
    230       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    231   mirror::String* InsertStrong(mirror::String* s)
    232       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    233   mirror::String* InsertWeak(mirror::String* s)
    234       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    235   void RemoveStrong(mirror::String* s)
    236       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    237   void RemoveWeak(mirror::String* s)
    238       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    239 
    240   // Transaction rollback access.
    241   mirror::String* LookupStringFromImage(mirror::String* s)
    242       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    243   mirror::String* InsertStrongFromTransaction(mirror::String* s)
    244       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    245   mirror::String* InsertWeakFromTransaction(mirror::String* s)
    246       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    247   void RemoveStrongFromTransaction(mirror::String* s)
    248       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    249   void RemoveWeakFromTransaction(mirror::String* s)
    250       SHARED_REQUIRES(Locks::mutator_lock_) REQUIRES(Locks::intern_table_lock_);
    251 
    252   size_t AddTableFromMemoryLocked(const uint8_t* ptr)
    253       REQUIRES(Locks::intern_table_lock_) SHARED_REQUIRES(Locks::mutator_lock_);
    254 
    255   // Change the weak root state. May broadcast to waiters.
    256   void ChangeWeakRootStateLocked(gc::WeakRootState new_state)
    257       REQUIRES(Locks::intern_table_lock_);
    258 
    259   // Wait until we can read weak roots.
    260   void WaitUntilAccessible(Thread* self)
    261       REQUIRES(Locks::intern_table_lock_) SHARED_REQUIRES(Locks::mutator_lock_);
    262 
    263   bool images_added_to_intern_table_ GUARDED_BY(Locks::intern_table_lock_);
    264   bool log_new_roots_ GUARDED_BY(Locks::intern_table_lock_);
    265   ConditionVariable weak_intern_condition_ GUARDED_BY(Locks::intern_table_lock_);
    266   // Since this contains (strong) roots, they need a read barrier to
    267   // enable concurrent intern table (strong) root scan. Do not
    268   // directly access the strings in it. Use functions that contain
    269   // read barriers.
    270   Table strong_interns_ GUARDED_BY(Locks::intern_table_lock_);
    271   std::vector<GcRoot<mirror::String>> new_strong_intern_roots_
    272       GUARDED_BY(Locks::intern_table_lock_);
    273   // Since this contains (weak) roots, they need a read barrier. Do
    274   // not directly access the strings in it. Use functions that contain
    275   // read barriers.
    276   Table weak_interns_ GUARDED_BY(Locks::intern_table_lock_);
    277   // Weak root state, used for concurrent system weak processing and more.
    278   gc::WeakRootState weak_root_state_ GUARDED_BY(Locks::intern_table_lock_);
    279 
    280   friend class Transaction;
    281   DISALLOW_COPY_AND_ASSIGN(InternTable);
    282 };
    283 
    284 }  // namespace art
    285 
    286 #endif  // ART_RUNTIME_INTERN_TABLE_H_
    287