1  /*
  2  * Copyright (c) 2020, 2025, Oracle and/or its affiliates. All rights reserved.
  3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  4  *
  5  * This code is free software; you can redistribute it and/or modify it
  6  * under the terms of the GNU General Public License version 2 only, as
  7  * published by the Free Software Foundation.
  8  *
  9  * This code is distributed in the hope that it will be useful, but WITHOUT
 10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 12  * version 2 for more details (a copy is included in the LICENSE file that
 13  * accompanied this code).
 14  *
 15  * You should have received a copy of the GNU General Public License version
 16  * 2 along with this work; if not, write to the Free Software Foundation,
 17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 18  *
 19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 20  * or visit www.oracle.com if you need additional information or have any
 21  * questions.
 22  *
 23  */
 24 
 25 #include "cds/aotLogging.hpp"
 26 #include "cds/cdsConfig.hpp"
 27 #include "cds/heapShared.inline.hpp"
 28 #include "cds/serializeClosure.hpp"
 29 #include "classfile/classLoaderData.inline.hpp"
 30 #include "classfile/classLoaderDataShared.hpp"
 31 #include "classfile/moduleEntry.hpp"
 32 #include "classfile/modules.hpp"
 33 #include "classfile/packageEntry.hpp"
 34 #include "classfile/systemDictionary.hpp"
 35 #include "logging/log.hpp"
 36 #include "runtime/handles.inline.hpp"
 37 #include "runtime/safepoint.hpp"
 38 
 39 #if INCLUDE_CDS_JAVA_HEAP
 40 
 41 bool ClassLoaderDataShared::_full_module_graph_loaded = false;
 42 
 43 class ArchivedClassLoaderData {
 44   Array<PackageEntry*>* _packages;
 45   Array<ModuleEntry*>* _modules;
 46   ModuleEntry* _unnamed_module;
 47 
 48   void assert_valid(ClassLoaderData* loader_data) {
 49     // loader_data may be null if the boot layer has loaded no modules for the platform or
 50     // system loaders (e.g., if you create a custom JDK image with only java.base).
 51     if (loader_data != nullptr) {
 52       assert(!loader_data->has_class_mirror_holder(),
 53              "loaders for non-strong hidden classes not supported");
 54     }
 55   }
 56 public:
 57   ArchivedClassLoaderData() : _packages(nullptr), _modules(nullptr), _unnamed_module(nullptr) {}
 58 
 59   void iterate_symbols(ClassLoaderData* loader_data, MetaspaceClosure* closure);
 60   void allocate(ClassLoaderData* loader_data);
 61   void init_archived_entries(ClassLoaderData* loader_data);
 62   ModuleEntry* unnamed_module() {
 63     return _unnamed_module;
 64   }
 65 
 66   void serialize(SerializeClosure* f) {
 67     f->do_ptr(&_packages);
 68     f->do_ptr(&_modules);
 69     f->do_ptr(&_unnamed_module);
 70   }
 71 
 72   void restore(ClassLoaderData* loader_data, bool do_entries, bool do_oops);
 73   void clear_archived_oops();
 74 };
 75 
 76 static ArchivedClassLoaderData _archived_boot_loader_data;
 77 static ArchivedClassLoaderData _archived_platform_loader_data;
 78 static ArchivedClassLoaderData _archived_system_loader_data;
 79 static ModuleEntry* _archived_javabase_moduleEntry = nullptr;
 80 static int _platform_loader_root_index = -1;
 81 static int _system_loader_root_index = -1;
 82 
 83 void ArchivedClassLoaderData::iterate_symbols(ClassLoaderData* loader_data, MetaspaceClosure* closure) {
 84   assert(CDSConfig::is_dumping_full_module_graph(), "must be");
 85   assert_valid(loader_data);
 86   if (loader_data != nullptr) {
 87     loader_data->packages()->iterate_symbols(closure);
 88     loader_data->modules() ->iterate_symbols(closure);
 89     loader_data->unnamed_module()->iterate_symbols(closure);
 90   }
 91 }
 92 
 93 void ArchivedClassLoaderData::allocate(ClassLoaderData* loader_data) {
 94   assert(CDSConfig::is_dumping_full_module_graph(), "must be");
 95   assert_valid(loader_data);
 96   if (loader_data != nullptr) {
 97     // We can't create hashtables at dump time because the hashcode depends on the
 98     // address of the Symbols, which may be relocated at runtime due to ASLR.
 99     // So we store the packages/modules in Arrays. At runtime, we create
100     // the hashtables using these arrays.
101     _packages = loader_data->packages()->allocate_archived_entries();
102     _modules  = loader_data->modules() ->allocate_archived_entries();
103     _unnamed_module = loader_data->unnamed_module()->allocate_archived_entry();
104   }
105 }
106 
107 void ArchivedClassLoaderData::init_archived_entries(ClassLoaderData* loader_data) {
108   assert(CDSConfig::is_dumping_full_module_graph(), "must be");
109   assert_valid(loader_data);
110   if (loader_data != nullptr) {
111     loader_data->packages()->init_archived_entries(_packages);
112     loader_data->modules() ->init_archived_entries(_modules);
113     _unnamed_module->init_as_archived_entry();
114   }
115 }
116 
117 void ArchivedClassLoaderData::restore(ClassLoaderData* loader_data, bool do_entries, bool do_oops) {
118   assert(CDSConfig::is_using_full_module_graph(), "must be");
119   assert_valid(loader_data);
120   if (_modules != nullptr) { // Could be null if we have archived no modules for platform/system loaders
121     ModuleEntryTable* modules = loader_data->modules();
122     PackageEntryTable* packages = loader_data->packages();
123 
124     MutexLocker m1(Module_lock);
125     if (do_entries) {
126       modules->load_archived_entries(loader_data, _modules);
127       packages->load_archived_entries(_packages);
128     }
129     if (do_oops) {
130       modules->restore_archived_oops(loader_data, _modules);
131       if (_unnamed_module != nullptr) {
132         oop module_oop = _unnamed_module->module_oop();
133         assert(module_oop != nullptr, "must be already set");
134         assert(_unnamed_module == java_lang_Module::module_entry(module_oop), "must be already set");
135         assert(loader_data->class_loader() == java_lang_Module::loader(module_oop), "must be set in dump time");
136       }
137     }
138   }
139 }
140 
141 void ArchivedClassLoaderData::clear_archived_oops() {
142   assert(!CDSConfig::is_using_full_module_graph(), "must be");
143   if (_modules != nullptr) {
144     for (int i = 0; i < _modules->length(); i++) {
145       _modules->at(i)->clear_archived_oops();
146     }
147     if (_unnamed_module != nullptr) {
148       _unnamed_module->clear_archived_oops();
149     }
150   }
151 }
152 
153 // ------------------------------
154 
155 void ClassLoaderDataShared::load_archived_platform_and_system_class_loaders() {
156 #if INCLUDE_CDS_JAVA_HEAP
157   // The streaming object loader prefers loading the class loader related objects before
158   //  the CLD constructor which has a NoSafepointVerifier.
159   if (!HeapShared::is_loading_streaming_mode()) {
160     return;
161   }
162 
163   // Ensure these class loaders are eagerly materialized before their CLDs are created.
164   HeapShared::get_root(_platform_loader_root_index, false /* clear */);
165   HeapShared::get_root(_system_loader_root_index, false /* clear */);
166 
167   if (Universe::is_module_initialized() || !CDSConfig::is_using_full_module_graph()) {
168     return;
169   }
170 
171   // When using the full module graph, we need to load unnamed modules too.
172   ModuleEntry* platform_loader_module_entry = _archived_platform_loader_data.unnamed_module();
173   if (platform_loader_module_entry != nullptr) {
174     platform_loader_module_entry->preload_archived_oops();
175   }
176 
177   ModuleEntry* system_loader_module_entry = _archived_system_loader_data.unnamed_module();
178   if (system_loader_module_entry != nullptr) {
179     system_loader_module_entry->preload_archived_oops();
180   }
181 #endif
182 }
183 
184 static ClassLoaderData* null_class_loader_data() {
185   ClassLoaderData* loader_data = ClassLoaderData::the_null_class_loader_data();
186   assert(loader_data != nullptr, "must be");
187   return loader_data;
188 }
189 
190 static ClassLoaderData* java_platform_loader_data_or_null() {
191   return ClassLoaderData::class_loader_data_or_null(SystemDictionary::java_platform_loader());
192 }
193 
194 static ClassLoaderData* java_system_loader_data_or_null() {
195   return ClassLoaderData::class_loader_data_or_null(SystemDictionary::java_system_loader());
196 }
197 
198 // ModuleEntryTables (even if empty) are required for iterate_symbols() to scan the
199 // platform/system loaders inside the CDS safepoint, but the tables can be created only
200 // when outside of safepoints. Let's do that now.
201 void ClassLoaderDataShared::ensure_module_entry_tables_exist() {
202   assert(!SafepointSynchronize::is_at_safepoint(), "sanity");
203   ensure_module_entry_table_exists(SystemDictionary::java_platform_loader());
204   ensure_module_entry_table_exists(SystemDictionary::java_system_loader());
205 }
206 
207 void ClassLoaderDataShared::ensure_module_entry_table_exists(oop class_loader) {
208   Handle h_loader(JavaThread::current(), class_loader);
209   ModuleEntryTable* met = Modules::get_module_entry_table(h_loader);
210   assert(met != nullptr, "sanity");
211 }
212 
213 void ClassLoaderDataShared::iterate_symbols(MetaspaceClosure* closure) {
214   assert(CDSConfig::is_dumping_full_module_graph(), "must be");
215   _archived_boot_loader_data.iterate_symbols    (null_class_loader_data(), closure);
216   _archived_platform_loader_data.iterate_symbols(java_platform_loader_data_or_null(), closure);
217   _archived_system_loader_data.iterate_symbols  (java_system_loader_data_or_null(), closure);
218 }
219 
220 void ClassLoaderDataShared::allocate_archived_tables() {
221   assert(CDSConfig::is_dumping_full_module_graph(), "must be");
222   _archived_boot_loader_data.allocate    (null_class_loader_data());
223   _archived_platform_loader_data.allocate(java_platform_loader_data_or_null());
224   _archived_system_loader_data.allocate  (java_system_loader_data_or_null());
225 }
226 
227 void ClassLoaderDataShared::init_archived_tables() {
228   assert(CDSConfig::is_dumping_full_module_graph(), "must be");
229 
230   _archived_boot_loader_data.init_archived_entries    (null_class_loader_data());
231   _archived_platform_loader_data.init_archived_entries(java_platform_loader_data_or_null());
232   _archived_system_loader_data.init_archived_entries  (java_system_loader_data_or_null());
233 
234   _archived_javabase_moduleEntry = ModuleEntry::get_archived_entry(ModuleEntryTable::javabase_moduleEntry());
235 
236   _platform_loader_root_index = HeapShared::append_root(SystemDictionary::java_platform_loader());
237   _system_loader_root_index = HeapShared::append_root(SystemDictionary::java_system_loader());
238 }
239 
240 void ClassLoaderDataShared::serialize(SerializeClosure* f) {
241   _archived_boot_loader_data.serialize(f);
242   _archived_platform_loader_data.serialize(f);
243   _archived_system_loader_data.serialize(f);
244   f->do_ptr(&_archived_javabase_moduleEntry);
245   f->do_int(&_platform_loader_root_index);
246   f->do_int(&_system_loader_root_index);
247 }
248 
249 ModuleEntry* ClassLoaderDataShared::archived_boot_unnamed_module() {
250   if (CDSConfig::is_using_full_module_graph()) {
251     return _archived_boot_loader_data.unnamed_module();
252   } else {
253     return nullptr;
254   }
255 }
256 
257 ModuleEntry* ClassLoaderDataShared::archived_unnamed_module(ClassLoaderData* loader_data) {
258   ModuleEntry* archived_module = nullptr;
259 
260   if (!Universe::is_module_initialized() && CDSConfig::is_using_full_module_graph()) {
261     precond(_platform_loader_root_index >= 0);
262     precond(_system_loader_root_index >= 0);
263 
264     if (loader_data->class_loader() == HeapShared::get_root(_platform_loader_root_index)) {
265       archived_module = _archived_platform_loader_data.unnamed_module();
266     } else if (loader_data->class_loader() == HeapShared::get_root(_system_loader_root_index)) {
267       archived_module = _archived_system_loader_data.unnamed_module();
268     }
269   }
270 
271   return archived_module;
272 }
273 
274 
275 void ClassLoaderDataShared::clear_archived_oops() {
276   assert(!CDSConfig::is_using_full_module_graph(), "must be");
277   _archived_boot_loader_data.clear_archived_oops();
278   _archived_platform_loader_data.clear_archived_oops();
279   _archived_system_loader_data.clear_archived_oops();
280   if (_platform_loader_root_index >= 0) {
281     HeapShared::clear_root(_platform_loader_root_index);
282     HeapShared::clear_root(_system_loader_root_index);
283   }
284 }
285 
286 // Must be done before ClassLoader::create_javabase()
287 void ClassLoaderDataShared::restore_archived_entries_for_null_class_loader_data() {
288   precond(CDSConfig::is_using_full_module_graph());
289   _archived_boot_loader_data.restore(null_class_loader_data(), true, false);
290   ModuleEntryTable::set_javabase_moduleEntry(_archived_javabase_moduleEntry);
291   aot_log_info(aot)("use_full_module_graph = true; java.base = " INTPTR_FORMAT,
292                     p2i(_archived_javabase_moduleEntry));
293 }
294 
295 oop ClassLoaderDataShared::restore_archived_oops_for_null_class_loader_data() {
296   assert(CDSConfig::is_using_full_module_graph(), "must be");
297   _archived_boot_loader_data.restore(null_class_loader_data(), false, true);
298   return _archived_javabase_moduleEntry->module_oop();
299 }
300 
301 void ClassLoaderDataShared::restore_java_platform_loader_from_archive(ClassLoaderData* loader_data) {
302   assert(CDSConfig::is_using_full_module_graph(), "must be");
303   _archived_platform_loader_data.restore(loader_data, true, true);
304 }
305 
306 void ClassLoaderDataShared::restore_java_system_loader_from_archive(ClassLoaderData* loader_data) {
307   assert(CDSConfig::is_using_full_module_graph(), "must be");
308   _archived_system_loader_data.restore(loader_data, true, true);
309   _full_module_graph_loaded = true;
310 }
311 
312 // This is called before AOTLinkedClassBulkLoader starts preloading classes. It makes sure that
313 // when we preload any class, its module is already valid.
314 void ClassLoaderDataShared::restore_archived_modules_for_preloading_classes(JavaThread* current) {
315   precond(CDSConfig::is_using_aot_linked_classes());
316 
317   precond(_platform_loader_root_index >= 0);
318   precond(_system_loader_root_index >= 0);
319 
320   Handle h_platform_loader(current, HeapShared::get_root(_platform_loader_root_index));
321   Handle h_system_loader(current, HeapShared::get_root(_system_loader_root_index));
322   Modules::init_archived_modules(current, h_platform_loader, h_system_loader);
323 }
324 
325 #endif // INCLUDE_CDS_JAVA_HEAP
--- EOF ---