1 /*
  2  * Copyright (c) 2022, 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/aotClassLinker.hpp"
 26 #include "cds/aotClassLocation.hpp"
 27 #include "cds/aotConstantPoolResolver.hpp"
 28 #include "cds/aotLogging.hpp"
 29 #include "cds/archiveBuilder.hpp"
 30 #include "cds/archiveUtils.inline.hpp"
 31 #include "cds/cdsConfig.hpp"
 32 #include "cds/classListWriter.hpp"
 33 #include "cds/finalImageRecipes.hpp"
 34 #include "cds/heapShared.hpp"
 35 #include "cds/lambdaFormInvokers.inline.hpp"
 36 #include "classfile/classLoader.hpp"
 37 #include "classfile/dictionary.hpp"
 38 #include "classfile/symbolTable.hpp"
 39 #include "classfile/systemDictionary.hpp"
 40 #include "classfile/systemDictionaryShared.hpp"
 41 #include "classfile/vmClasses.hpp"
 42 #include "interpreter/bytecodeStream.hpp"
 43 #include "interpreter/interpreterRuntime.hpp"
 44 #include "memory/resourceArea.hpp"
 45 #include "oops/constantPool.inline.hpp"
 46 #include "oops/fieldStreams.inline.hpp"
 47 #include "oops/instanceKlass.hpp"
 48 #include "oops/klass.inline.hpp"
 49 #include "runtime/handles.inline.hpp"
 50 #include "runtime/javaCalls.hpp"
 51 #include "runtime/signature.hpp"
 52 
 53 // Returns true if we CAN PROVE that cp_index will always resolve to
 54 // the same information at both dump time and run time. This is a
 55 // necessary (but not sufficient) condition for pre-resolving cp_index
 56 // during CDS archive assembly.
 57 bool AOTConstantPoolResolver::is_resolution_deterministic(ConstantPool* cp, int cp_index) {
 58   assert(!is_in_archivebuilder_buffer(cp), "sanity");
 59 
 60   if (cp->tag_at(cp_index).is_klass()) {
 61     // We require cp_index to be already resolved. This is fine for now, are we
 62     // currently archive only CP entries that are already resolved.
 63     Klass* resolved_klass = cp->resolved_klass_at(cp_index);
 64     return resolved_klass != nullptr && is_class_resolution_deterministic(cp->pool_holder(), resolved_klass);
 65   } else if (cp->tag_at(cp_index).is_invoke_dynamic()) {
 66     return is_indy_resolution_deterministic(cp, cp_index);
 67   } else if (cp->tag_at(cp_index).is_field() ||
 68              cp->tag_at(cp_index).is_method() ||
 69              cp->tag_at(cp_index).is_interface_method()) {
 70     int klass_cp_index = cp->uncached_klass_ref_index_at(cp_index);
 71     if (!cp->tag_at(klass_cp_index).is_klass()) {
 72       // Not yet resolved
 73       return false;
 74     }
 75     Klass* k = cp->resolved_klass_at(klass_cp_index);
 76     if (!is_class_resolution_deterministic(cp->pool_holder(), k)) {
 77       return false;
 78     }
 79 
 80     if (!k->is_instance_klass()) {
 81       // TODO: support non instance klasses as well.
 82       return false;
 83     }
 84 
 85     // Here, We don't check if this entry can actually be resolved to a valid Field/Method.
 86     // This method should be called by the ConstantPool to check Fields/Methods that
 87     // have already been successfully resolved.
 88     return true;
 89   } else {
 90     return false;
 91   }
 92 }
 93 
 94 bool AOTConstantPoolResolver::is_class_resolution_deterministic(InstanceKlass* cp_holder, Klass* resolved_class) {
 95   assert(!is_in_archivebuilder_buffer(cp_holder), "sanity");
 96   assert(!is_in_archivebuilder_buffer(resolved_class), "sanity");
 97 
 98   if (resolved_class->is_instance_klass()) {
 99     InstanceKlass* ik = InstanceKlass::cast(resolved_class);
100 
101     if (!ik->in_aot_cache() && SystemDictionaryShared::is_excluded_class(ik)) {
102       return false;
103     }
104 
105     if (cp_holder->is_subtype_of(ik)) {
106       // All super types of ik will be resolved in ik->class_loader() before
107       // ik is defined in this loader, so it's safe to archive the resolved klass reference.
108       return true;
109     }
110 
111     if (CDSConfig::is_dumping_aot_linked_classes()) {
112       // Need to call try_add_candidate instead of is_candidate, as this may be called
113       // before AOTClassLinker::add_candidates().
114       if (AOTClassLinker::try_add_candidate(ik)) {
115         return true;
116       } else {
117         return false;
118       }
119     } else if (AOTClassLinker::is_vm_class(ik)) {
120       if (ik->class_loader() != cp_holder->class_loader()) {
121         // At runtime, cp_holder() may not be able to resolve to the same
122         // ik. For example, a different version of ik may be defined in
123         // cp->pool_holder()'s loader using MethodHandles.Lookup.defineClass().
124         return false;
125       } else {
126         return true;
127       }
128     } else {
129       return false;
130     }
131   } else if (resolved_class->is_objArray_klass()) {
132     Klass* elem = ObjArrayKlass::cast(resolved_class)->bottom_klass();
133     if (elem->is_instance_klass()) {
134       return is_class_resolution_deterministic(cp_holder, InstanceKlass::cast(elem));
135     } else if (elem->is_typeArray_klass()) {
136       return true;
137     } else {
138       return false;
139     }
140   } else if (resolved_class->is_typeArray_klass()) {
141     return true;
142   } else {
143     return false;
144   }
145 }
146 
147 void AOTConstantPoolResolver::preresolve_string_cp_entries(InstanceKlass* ik, TRAPS) {
148   if (!ik->is_linked()) {
149     // The cp->resolved_referenced() array is not ready yet, so we can't call resolve_string().
150     return;
151   }
152   constantPoolHandle cp(THREAD, ik->constants());
153   for (int cp_index = 1; cp_index < cp->length(); cp_index++) { // Index 0 is unused
154     switch (cp->tag_at(cp_index).value()) {
155     case JVM_CONSTANT_String:
156       resolve_string(cp, cp_index, CHECK); // may throw OOM when interning strings.
157       break;
158     }
159   }
160 
161   // Normally, we don't want to archive any CP entries that were not resolved
162   // in the training run. Otherwise the AOT/JIT may inline too much code that has not
163   // been executed.
164   //
165   // However, we want to aggressively resolve all klass/field/method constants for
166   // LambdaForm Invoker Holder classes, Lambda Proxy classes, and LambdaForm classes,
167   // so that the compiler can inline through them.
168   if (SystemDictionaryShared::is_builtin_loader(ik->class_loader_data())) {
169     bool eager_resolve = false;
170 
171     if (LambdaFormInvokers::may_be_regenerated_class(ik->name())) {
172       eager_resolve = true;
173     }
174     if (ik->is_hidden() && HeapShared::is_archivable_hidden_klass(ik)) {
175       eager_resolve = true;
176     }
177 
178     if (eager_resolve) {
179       preresolve_class_cp_entries(THREAD, ik, nullptr);
180       preresolve_field_and_method_cp_entries(THREAD, ik, nullptr);
181     }
182   }
183 }
184 
185 // This works only for the boot/platform/app loaders
186 Klass* AOTConstantPoolResolver::find_loaded_class(Thread* current, oop class_loader, Symbol* name) {
187   HandleMark hm(current);
188   Handle h_loader(current, class_loader);
189   Klass* k = SystemDictionary::find_instance_or_array_klass(current, name, h_loader);
190   if (k != nullptr) {
191     return k;
192   }
193   if (h_loader() == SystemDictionary::java_system_loader()) {
194     return find_loaded_class(current, SystemDictionary::java_platform_loader(), name);
195   } else if (h_loader() == SystemDictionary::java_platform_loader()) {
196     return find_loaded_class(current, nullptr, name);
197   } else {
198     assert(h_loader() == nullptr, "This function only works for boot/platform/app loaders %p %p %p",
199            cast_from_oop<address>(h_loader()),
200            cast_from_oop<address>(SystemDictionary::java_system_loader()),
201            cast_from_oop<address>(SystemDictionary::java_platform_loader()));
202   }
203 
204   return nullptr;
205 }
206 
207 Klass* AOTConstantPoolResolver::find_loaded_class(Thread* current, ConstantPool* cp, int class_cp_index) {
208   Symbol* name = cp->klass_name_at(class_cp_index);
209   return find_loaded_class(current, cp->pool_holder()->class_loader(), name);
210 }
211 
212 #if INCLUDE_CDS_JAVA_HEAP
213 void AOTConstantPoolResolver::resolve_string(constantPoolHandle cp, int cp_index, TRAPS) {
214   if (CDSConfig::is_dumping_heap()) {
215     int cache_index = cp->cp_to_object_index(cp_index);
216     ConstantPool::string_at_impl(cp, cp_index, cache_index, CHECK);
217   }
218 }
219 #endif
220 
221 void AOTConstantPoolResolver::preresolve_class_cp_entries(JavaThread* current, InstanceKlass* ik, GrowableArray<bool>* preresolve_list) {
222   if (!SystemDictionaryShared::is_builtin_loader(ik->class_loader_data())) {
223     return;
224   }
225 
226   JavaThread* THREAD = current;
227   constantPoolHandle cp(THREAD, ik->constants());
228   for (int cp_index = 1; cp_index < cp->length(); cp_index++) {
229     if (cp->tag_at(cp_index).value() == JVM_CONSTANT_UnresolvedClass) {
230       if (preresolve_list != nullptr && preresolve_list->at(cp_index) == false) {
231         // This class was not resolved during trial run. Don't attempt to resolve it. Otherwise
232         // the compiler may generate less efficient code.
233         continue;
234       }
235       if (find_loaded_class(current, cp(), cp_index) == nullptr) {
236         // Do not resolve any class that has not been loaded yet
237         continue;
238       }
239       Klass* resolved_klass = cp->klass_at(cp_index, THREAD);
240       if (HAS_PENDING_EXCEPTION) {
241         CLEAR_PENDING_EXCEPTION; // just ignore
242       } else {
243         log_trace(aot, resolve)("Resolved class  [%3d] %s -> %s", cp_index, ik->external_name(),
244                                 resolved_klass->external_name());
245       }
246     }
247   }
248 }
249 
250 void AOTConstantPoolResolver::preresolve_field_and_method_cp_entries(JavaThread* current, InstanceKlass* ik, GrowableArray<bool>* preresolve_list) {
251   JavaThread* THREAD = current;
252   constantPoolHandle cp(THREAD, ik->constants());
253   if (cp->cache() == nullptr) {
254     return;
255   }
256   for (int i = 0; i < ik->methods()->length(); i++) {
257     Method* m = ik->methods()->at(i);
258     BytecodeStream bcs(methodHandle(THREAD, m));
259     while (!bcs.is_last_bytecode()) {
260       bcs.next();
261       Bytecodes::Code raw_bc = bcs.raw_code();
262       switch (raw_bc) {
263       case Bytecodes::_getstatic:
264       case Bytecodes::_putstatic:
265       case Bytecodes::_getfield:
266       case Bytecodes::_putfield:
267         maybe_resolve_fmi_ref(ik, m, raw_bc, bcs.get_index_u2(), preresolve_list, THREAD);
268         if (HAS_PENDING_EXCEPTION) {
269           CLEAR_PENDING_EXCEPTION; // just ignore
270         }
271         break;
272       case Bytecodes::_invokehandle:
273       case Bytecodes::_invokespecial:
274       case Bytecodes::_invokevirtual:
275       case Bytecodes::_invokeinterface:
276       case Bytecodes::_invokestatic:
277         maybe_resolve_fmi_ref(ik, m, raw_bc, bcs.get_index_u2(), preresolve_list, THREAD);
278         if (HAS_PENDING_EXCEPTION) {
279           CLEAR_PENDING_EXCEPTION; // just ignore
280         }
281         break;
282       default:
283         break;
284       }
285     }
286   }
287 }
288 
289 void AOTConstantPoolResolver::maybe_resolve_fmi_ref(InstanceKlass* ik, Method* m, Bytecodes::Code bc, int raw_index,
290                                            GrowableArray<bool>* preresolve_list, TRAPS) {
291   methodHandle mh(THREAD, m);
292   constantPoolHandle cp(THREAD, ik->constants());
293   HandleMark hm(THREAD);
294   int cp_index = cp->to_cp_index(raw_index, bc);
295 
296   if (cp->is_resolved(raw_index, bc)) {
297     return;
298   }
299 
300   if (preresolve_list != nullptr && preresolve_list->at(cp_index) == false) {
301     // This field wasn't resolved during the trial run. Don't attempt to resolve it. Otherwise
302     // the compiler may generate less efficient code.
303     return;
304   }
305 
306   int klass_cp_index = cp->uncached_klass_ref_index_at(cp_index);
307   if (find_loaded_class(THREAD, cp(), klass_cp_index) == nullptr) {
308     // Do not resolve any field/methods from a class that has not been loaded yet.
309     return;
310   }
311 
312   Klass* resolved_klass = cp->klass_ref_at(raw_index, bc, CHECK);
313   const char* is_static = "";
314 
315   switch (bc) {
316   case Bytecodes::_getstatic:
317   case Bytecodes::_putstatic:
318     if (!VM_Version::supports_fast_class_init_checks()) {
319       return; // Do not resolve since interpreter lacks fast clinit barriers support
320     }
321     InterpreterRuntime::resolve_get_put(bc, raw_index, mh, cp, false /*initialize_holder*/, CHECK);
322     is_static = " *** static";
323     break;
324   case Bytecodes::_getfield:
325   case Bytecodes::_putfield:
326     InterpreterRuntime::resolve_get_put(bc, raw_index, mh, cp, false /*initialize_holder*/, CHECK);
327     break;
328 
329   case Bytecodes::_invokestatic:
330     if (!VM_Version::supports_fast_class_init_checks()) {
331       return; // Do not resolve since interpreter lacks fast clinit barriers support
332     }
333     InterpreterRuntime::cds_resolve_invoke(bc, raw_index, cp, CHECK);
334     is_static = " *** static";
335     break;
336 
337   case Bytecodes::_invokevirtual:
338   case Bytecodes::_invokespecial:
339   case Bytecodes::_invokeinterface:
340     InterpreterRuntime::cds_resolve_invoke(bc, raw_index, cp, CHECK);
341     break;
342 
343   case Bytecodes::_invokehandle:
344     InterpreterRuntime::cds_resolve_invokehandle(raw_index, cp, CHECK);
345     break;
346 
347   default:
348     ShouldNotReachHere();
349   }
350 
351   if (log_is_enabled(Trace, aot, resolve)) {
352     ResourceMark rm(THREAD);
353     bool resolved = cp->is_resolved(raw_index, bc);
354     Symbol* name = cp->name_ref_at(raw_index, bc);
355     Symbol* signature = cp->signature_ref_at(raw_index, bc);
356     log_trace(aot, resolve)("%s %s [%3d] %s -> %s.%s:%s%s",
357                             (resolved ? "Resolved" : "Failed to resolve"),
358                             Bytecodes::name(bc), cp_index, ik->external_name(),
359                             resolved_klass->external_name(),
360                             name->as_C_string(), signature->as_C_string(), is_static);
361   }
362 }
363 
364 void AOTConstantPoolResolver::preresolve_indy_cp_entries(JavaThread* current, InstanceKlass* ik, GrowableArray<bool>* preresolve_list) {
365   JavaThread* THREAD = current;
366   constantPoolHandle cp(THREAD, ik->constants());
367   if (!CDSConfig::is_dumping_invokedynamic() || cp->cache() == nullptr) {
368     return;
369   }
370 
371   assert(preresolve_list != nullptr, "preresolve_indy_cp_entries() should not be called for "
372          "regenerated LambdaForm Invoker classes, which should not have indys anyway.");
373 
374   Array<ResolvedIndyEntry>* indy_entries = cp->cache()->resolved_indy_entries();
375   for (int i = 0; i < indy_entries->length(); i++) {
376     ResolvedIndyEntry* rie = indy_entries->adr_at(i);
377     int cp_index = rie->constant_pool_index();
378     if (preresolve_list->at(cp_index) == true) {
379       if (!rie->is_resolved() && is_indy_resolution_deterministic(cp(), cp_index)) {
380         InterpreterRuntime::cds_resolve_invokedynamic(i, cp, THREAD);
381         if (HAS_PENDING_EXCEPTION) {
382           CLEAR_PENDING_EXCEPTION; // just ignore
383         }
384       }
385       if (log_is_enabled(Trace, aot, resolve)) {
386         ResourceMark rm(THREAD);
387         log_trace(aot, resolve)("%s indy   [%3d] %s",
388                                 rie->is_resolved() ? "Resolved" : "Failed to resolve",
389                                 cp_index, ik->external_name());
390       }
391     }
392   }
393 }
394 
395 // Check the MethodType signatures used by parameters to the indy BSMs. Make sure we don't
396 // use types that have been excluded, or else we might end up creating MethodTypes that cannot be stored
397 // in the AOT cache.
398 bool AOTConstantPoolResolver::check_methodtype_signature(ConstantPool* cp, Symbol* sig, Klass** return_type_ret) {
399   ResourceMark rm;
400   for (SignatureStream ss(sig); !ss.is_done(); ss.next()) {
401     if (ss.is_reference()) {
402       Symbol* type = ss.as_symbol();
403       Klass* k = find_loaded_class(Thread::current(), cp->pool_holder()->class_loader(), type);
404       if (k == nullptr) {
405         return false;
406       }
407 
408       if (SystemDictionaryShared::should_be_excluded(k)) {
409         if (log_is_enabled(Warning, aot, resolve)) {
410           ResourceMark rm;
411           log_warning(aot, resolve)("Cannot aot-resolve Lambda proxy because %s is excluded", k->external_name());
412         }
413         return false;
414       }
415 
416       if (ss.at_return_type() && return_type_ret != nullptr) {
417         *return_type_ret = k;
418       }
419     }
420   }
421   return true;
422 }
423 
424 bool AOTConstantPoolResolver::check_lambda_metafactory_signature(ConstantPool* cp, Symbol* sig) {
425   Klass* k;
426   if (!check_methodtype_signature(cp, sig, &k)) {
427     return false;
428   }
429 
430   // <k> is the interface type implemented by the lambda proxy
431   if (!k->is_interface()) {
432     // cp->pool_holder() doesn't look like a valid class generated by javac
433     return false;
434   }
435 
436 
437   // The linked lambda callsite has an instance of the interface implemented by this lambda. If this
438   // interface requires its <clinit> to be executed, then we must delay the execution to the production run
439   // as <clinit> can have side effects ==> exclude such cases.
440   InstanceKlass* intf = InstanceKlass::cast(k);
441   bool exclude = intf->interface_needs_clinit_execution_as_super();
442   if (log_is_enabled(Debug, aot, resolve)) {
443     ResourceMark rm;
444     log_debug(aot, resolve)("%s aot-resolve Lambda proxy of interface type %s",
445                             exclude ? "Cannot" : "Can", k->external_name());
446   }
447   return !exclude;
448 }
449 
450 bool AOTConstantPoolResolver::check_lambda_metafactory_methodtype_arg(ConstantPool* cp, int bsms_attribute_index, int arg_i) {
451   int mt_index = cp->bsm_attribute_entry(bsms_attribute_index)->argument_index(arg_i);
452   if (!cp->tag_at(mt_index).is_method_type()) {
453     // malformed class?
454     return false;
455   }
456 
457   Symbol* sig = cp->method_type_signature_at(mt_index);
458   if (log_is_enabled(Debug, aot, resolve)) {
459     ResourceMark rm;
460     log_debug(aot, resolve)("Checking MethodType for LambdaMetafactory BSM arg %d: %s", arg_i, sig->as_C_string());
461   }
462 
463   return check_methodtype_signature(cp, sig);
464 }
465 
466 bool AOTConstantPoolResolver::check_lambda_metafactory_methodhandle_arg(ConstantPool* cp, int bsms_attribute_index, int arg_i) {
467   int mh_index = cp->bsm_attribute_entry(bsms_attribute_index)->argument_index(arg_i);
468   if (!cp->tag_at(mh_index).is_method_handle()) {
469     // malformed class?
470     return false;
471   }
472 
473   Symbol* sig = cp->method_handle_signature_ref_at(mh_index);
474   if (log_is_enabled(Debug, aot, resolve)) {
475     ResourceMark rm;
476     log_debug(aot, resolve)("Checking MethodType of MethodHandle for LambdaMetafactory BSM arg %d: %s", arg_i, sig->as_C_string());
477   }
478   return check_methodtype_signature(cp, sig);
479 }
480 
481 bool AOTConstantPoolResolver::is_indy_resolution_deterministic(ConstantPool* cp, int cp_index) {
482   assert(cp->tag_at(cp_index).is_invoke_dynamic(), "sanity");
483   if (!CDSConfig::is_dumping_invokedynamic()) {
484     return false;
485   }
486 
487   InstanceKlass* pool_holder = cp->pool_holder();
488   if (!SystemDictionaryShared::is_builtin(pool_holder)) {
489     return false;
490   }
491 
492   int bsm = cp->bootstrap_method_ref_index_at(cp_index);
493   int bsm_ref = cp->method_handle_index_at(bsm);
494   Symbol* bsm_name = cp->uncached_name_ref_at(bsm_ref);
495   Symbol* bsm_signature = cp->uncached_signature_ref_at(bsm_ref);
496   Symbol* bsm_klass = cp->klass_name_at(cp->uncached_klass_ref_index_at(bsm_ref));
497 
498   // We currently support only StringConcatFactory::makeConcatWithConstants() and LambdaMetafactory::metafactory()
499   // We should mark the allowed BSMs in the JDK code using a private annotation.
500   // See notes on RFE JDK-8342481.
501 
502   if (bsm_klass->equals("java/lang/invoke/StringConcatFactory") &&
503       bsm_name->equals("makeConcatWithConstants") &&
504       bsm_signature->equals("(Ljava/lang/invoke/MethodHandles$Lookup;"
505                              "Ljava/lang/String;"
506                              "Ljava/lang/invoke/MethodType;"
507                              "Ljava/lang/String;"
508                              "[Ljava/lang/Object;"
509                             ")Ljava/lang/invoke/CallSite;")) {
510     Symbol* factory_type_sig = cp->uncached_signature_ref_at(cp_index);
511     if (log_is_enabled(Debug, aot, resolve)) {
512       ResourceMark rm;
513       log_debug(aot, resolve)("Checking StringConcatFactory callsite signature [%d]: %s", cp_index, factory_type_sig->as_C_string());
514     }
515 
516     Klass* k;
517     if (!check_methodtype_signature(cp, factory_type_sig, &k)) {
518       return false;
519     }
520     if (k != vmClasses::String_klass()) {
521       // bad class file?
522       return false;
523     }
524 
525     return true;
526   }
527 
528   if (bsm_klass->equals("java/lang/invoke/LambdaMetafactory") &&
529       bsm_name->equals("metafactory") &&
530       bsm_signature->equals("(Ljava/lang/invoke/MethodHandles$Lookup;"
531                              "Ljava/lang/String;"
532                              "Ljava/lang/invoke/MethodType;"
533                              "Ljava/lang/invoke/MethodType;"
534                              "Ljava/lang/invoke/MethodHandle;"
535                              "Ljava/lang/invoke/MethodType;"
536                             ")Ljava/lang/invoke/CallSite;")) {
537     /*
538      * An indy callsite is associated with the following MethodType and MethodHandles:
539      *
540      * https://github.com/openjdk/jdk/blob/580eb62dc097efeb51c76b095c1404106859b673/src/java.base/share/classes/java/lang/invoke/LambdaMetafactory.java#L293-L309
541      *
542      * MethodType factoryType         The expected signature of the {@code CallSite}.  The
543      *                                parameter types represent the types of capture variables;
544      *                                the return type is the interface to implement.   When
545      *                                used with {@code invokedynamic}, this is provided by
546      *                                the {@code NameAndType} of the {@code InvokeDynamic}
547      *
548      * MethodType interfaceMethodType Signature and return type of method to be
549      *                                implemented by the function object.
550      *
551      * MethodHandle implementation    A direct method handle describing the implementation
552      *                                method which should be called (with suitable adaptation
553      *                                of argument types and return types, and with captured
554      *                                arguments prepended to the invocation arguments) at
555      *                                invocation time.
556      *
557      * MethodType dynamicMethodType   The signature and return type that should
558      *                                be enforced dynamically at invocation time.
559      *                                In simple use cases this is the same as
560      *                                {@code interfaceMethodType}.
561      */
562     Symbol* factory_type_sig = cp->uncached_signature_ref_at(cp_index);
563     if (log_is_enabled(Debug, aot, resolve)) {
564       ResourceMark rm;
565       log_debug(aot, resolve)("Checking lambda callsite signature [%d]: %s", cp_index, factory_type_sig->as_C_string());
566     }
567 
568     if (!check_lambda_metafactory_signature(cp, factory_type_sig)) {
569       return false;
570     }
571 
572     int bsms_attribute_index = cp->bootstrap_methods_attribute_index(cp_index);
573     int arg_count = cp->bsm_attribute_entry(bsms_attribute_index)->argument_count();
574     if (arg_count != 3) {
575       // Malformed class?
576       return false;
577     }
578 
579     // interfaceMethodType
580     if (!check_lambda_metafactory_methodtype_arg(cp, bsms_attribute_index, 0)) {
581       return false;
582     }
583 
584     // implementation
585     if (!check_lambda_metafactory_methodhandle_arg(cp, bsms_attribute_index, 1)) {
586       return false;
587     }
588 
589     // dynamicMethodType
590     if (!check_lambda_metafactory_methodtype_arg(cp, bsms_attribute_index, 2)) {
591       return false;
592     }
593 
594     return true;
595   }
596 
597   return false;
598 }
599 #ifdef ASSERT
600 bool AOTConstantPoolResolver::is_in_archivebuilder_buffer(address p) {
601   if (!Thread::current()->is_VM_thread() || ArchiveBuilder::current() == nullptr) {
602     return false;
603   } else {
604     return ArchiveBuilder::current()->is_in_buffer_space(p);
605   }
606 }
607 #endif
608 
609 // Paranoid check -- if any field/method signature in <ik> is referencing a class that is known to be
610 // excluded, do not archive any reflection data in <ik>.
611 bool AOTConstantPoolResolver::check_signature_for_reflection_data(InstanceKlass* ik, Symbol* sig, bool is_method) {
612   precond(SafepointSynchronize::is_at_safepoint());
613   ResourceMark rm;
614   for (SignatureStream ss(sig, is_method); !ss.is_done(); ss.next()) {
615     if (ss.is_reference()) {
616       Symbol* klass_name = ss.as_symbol(/*probe_only=*/true);
617       if (klass_name == nullptr) {
618         // This symbol has never been created during the assembly phase, so it can't
619         // possibly be referencing an excluded class.
620         continue;
621       }
622       if (Signature::is_array(klass_name)) {
623         SignatureStream ss2(klass_name, false);
624         ss2.skip_array_prefix();
625         if (ss2.is_reference() && ss2.as_symbol(/*probe_only=*/true) == nullptr) {
626           // klass_name is an array klass that has not been loaded during the assembly phase, so it can't
627           // possibly be referencing an excluded class.
628           continue;
629         }
630       }
631       Klass* k = find_loaded_class(Thread::current(), ik->class_loader(), klass_name);
632       if (k == nullptr) {
633         // No class of this name has been loaded for this loader, so this name can't
634         // possibly be referencing an excluded class.
635         continue;
636       }
637       if (SystemDictionaryShared::should_be_excluded(k)) {
638         if (log_is_enabled(Warning, aot, resolve)) {
639           ResourceMark rm;
640           log_warning(aot, resolve)("Cannot archive reflection data in %s because it refers to excluded class %s",
641                                     ik->external_name(), k->external_name());
642         }
643         return false;
644       }
645     }
646   }
647   return true;
648 }
649 
650 bool AOTConstantPoolResolver::can_archive_reflection_data_declared_fields(InstanceKlass* ik) {
651   for (JavaFieldStream fs(ik); !fs.done(); fs.next()) {
652     if (!check_signature_for_reflection_data(ik, fs.signature(), /*is_method=*/false)) {
653       return false;
654     }
655   }
656 
657   return true;
658 }
659 
660 bool AOTConstantPoolResolver::can_archive_reflection_data_declared_methods(InstanceKlass* ik) {
661   Array<Method*>* methods = ik->methods();
662   for (int i = 0; i < methods->length(); i++) {
663     Method* m = methods->at(i);
664     if (!check_signature_for_reflection_data(ik, m->signature(), /*is_method=*/true)) {
665       return false;
666     }
667   }
668 
669   return true;
670 }
671 
672 // This is called by AOT assembly phase to see if java_lang_Class::reflection_data(k) can be
673 // archived.
674 bool AOTConstantPoolResolver::can_archive_reflection_data(InstanceKlass* ik) {
675   // At this point, we have resolved the reflection data of *all* classes
676   // recorded by FinalImageRecipes.
677   // When we are in the AOT safepoint, we will archive the reflection
678   // data of <ik> *only if* it doesn't reference any excluded classes.
679   precond(SafepointSynchronize::is_at_safepoint());
680 
681   if (!can_archive_reflection_data_declared_fields(ik)) {
682     return false;
683   }
684   if (!can_archive_reflection_data_declared_methods(ik)) {
685     return false;
686   }
687 
688   if (log_is_enabled(Info, aot, resolve)) {
689     ResourceMark rm;
690     log_info(aot, resolve)("Archived reflection data in %s", ik->external_name());
691   }
692 
693   return true;
694 }
695 
696 int AOTConstantPoolResolver::class_reflection_data_flags(InstanceKlass* ik, TRAPS) {
697   assert(java_lang_Class::has_reflection_data(ik->java_mirror()), "must be");
698 
699   HandleMark hm(THREAD);
700   JavaCallArguments args(Handle(THREAD, ik->java_mirror()));
701   JavaValue result(T_INT);
702   JavaCalls::call_special(&result,
703                           vmClasses::Class_klass(),
704                           vmSymbols::encodeReflectionData_name(),
705                           vmSymbols::void_int_signature(),
706                           &args, CHECK_0);
707   int flags = result.get_jint();
708   aot_log_info(aot)("Encode ReflectionData: %s (flags=0x%x)", ik->external_name(), flags);
709   return flags;
710 }
711 
712 void AOTConstantPoolResolver::generate_reflection_data(JavaThread* current, InstanceKlass* ik, int rd_flags) {
713   aot_log_info(aot)("Generate ReflectionData: %s (flags=" INT32_FORMAT_X ")", ik->external_name(), rd_flags);
714   JavaThread* THREAD = current; // for exception macros
715   JavaCallArguments args(Handle(THREAD, ik->java_mirror()));
716   args.push_int(rd_flags);
717   JavaValue result(T_OBJECT);
718   JavaCalls::call_special(&result,
719                           vmClasses::Class_klass(),
720                           vmSymbols::generateReflectionData_name(),
721                           vmSymbols::int_void_signature(),
722                           &args, THREAD);
723   if (HAS_PENDING_EXCEPTION) {
724     Handle exc_handle(THREAD, PENDING_EXCEPTION);
725     CLEAR_PENDING_EXCEPTION;
726 
727     log_warning(aot)("Exception during Class::generateReflectionData() call for %s", ik->external_name());
728     LogStreamHandle(Debug, aot) log;
729     if (log.is_enabled()) {
730       java_lang_Throwable::print_stack_trace(exc_handle, &log);
731     }
732   }
733 }
734 
735 Klass* AOTConstantPoolResolver::resolve_boot_class_or_fail(const char* class_name, TRAPS) {
736   Handle class_loader;
737   TempNewSymbol class_name_sym = SymbolTable::new_symbol(class_name);
738   return SystemDictionary::resolve_or_fail(class_name_sym, class_loader, true, THREAD);
739 }
740 
741 void AOTConstantPoolResolver::trace_dynamic_proxy_class(oop loader, const char* proxy_name, objArrayOop interfaces, int access_flags) {
742   if (interfaces->length() < 1) {
743     return;
744   }
745   if (ClassListWriter::is_enabled()) {
746     const char* loader_name = ArchiveUtils::builtin_loader_name_or_null(loader);
747     if (loader_name != nullptr) {
748       stringStream ss;
749       ss.print("%s %s %d %d", loader_name, proxy_name, access_flags, interfaces->length());
750       for (int i = 0; i < interfaces->length(); i++) {
751         oop mirror = interfaces->obj_at(i);
752         Klass* k = java_lang_Class::as_Klass(mirror);
753         ss.print(" %s", k->name()->as_C_string());
754       }
755       ClassListWriter w; // This locks ClassListFile_lock
756       w.stream()->print_cr("@dynamic-proxy %s", ss.freeze());
757     }
758   }
759   if (CDSConfig::is_dumping_preimage_static_archive()) {
760     FinalImageRecipes::add_dynamic_proxy_class(loader, proxy_name, interfaces, access_flags);
761   }
762 }
763 
764 void AOTConstantPoolResolver::init_dynamic_proxy_cache(TRAPS) {
765   static bool inited = false;
766   if (inited) {
767     return;
768   }
769   inited = true;
770 
771   Klass* klass = resolve_boot_class_or_fail("java/lang/reflect/Proxy", CHECK);
772   TempNewSymbol method = SymbolTable::new_symbol("initCacheForCDS");
773   TempNewSymbol signature = SymbolTable::new_symbol("(Ljava/lang/ClassLoader;Ljava/lang/ClassLoader;)V");
774 
775   JavaCallArguments args;
776   args.push_oop(Handle(THREAD, SystemDictionary::java_platform_loader()));
777   args.push_oop(Handle(THREAD, SystemDictionary::java_system_loader()));
778   JavaValue result(T_VOID);
779   JavaCalls::call_static(&result,
780                          klass,
781                          method,
782                          signature,
783                          &args, CHECK);
784 }
785 
786 
787 void AOTConstantPoolResolver::define_dynamic_proxy_class(Handle loader, Handle proxy_name, Handle interfaces, int access_flags, TRAPS) {
788   if (!CDSConfig::is_dumping_dynamic_proxies()) {
789     return;
790   }
791   init_dynamic_proxy_cache(CHECK);
792 
793   Klass* klass = resolve_boot_class_or_fail("java/lang/reflect/Proxy$ProxyBuilder", CHECK);
794   TempNewSymbol method = SymbolTable::new_symbol("defineProxyClassForCDS");
795   TempNewSymbol signature = SymbolTable::new_symbol("(Ljava/lang/ClassLoader;Ljava/lang/String;[Ljava/lang/Class;I)Ljava/lang/Class;");
796 
797   JavaCallArguments args;
798   args.push_oop(Handle(THREAD, loader()));
799   args.push_oop(Handle(THREAD, proxy_name()));
800   args.push_oop(Handle(THREAD, interfaces()));
801   args.push_int(access_flags);
802   JavaValue result(T_OBJECT);
803   JavaCalls::call_static(&result,
804                          klass,
805                          method,
806                          signature,
807                          &args, CHECK);
808 
809   // Assumptions:
810   // FMG is archived, which means -modulepath and -Xbootclasspath are both not specified.
811   // All named modules are loaded from the system modules files.
812   // TODO: test support for -Xbootclasspath after JDK-8322322. Some of the code below need to be changed.
813   // TODO: we just give dummy shared_classpath_index for the generated class so that it will be archived.
814   //       The index is not used at runtime (see SystemDictionaryShared::load_shared_class_for_builtin_loader, which
815   //       uses a null ProtectionDomain for this class)
816   oop mirror = result.get_oop();
817   assert(mirror != nullptr, "class must have been generated if not OOM");
818   InstanceKlass* ik = InstanceKlass::cast(java_lang_Class::as_Klass(mirror));
819   if (ik->defined_by_boot_loader() || ik->defined_by_platform_loader()) {
820     assert(ik->module()->is_named(), "dynamic proxies defined in unnamed modules for boot/platform loaders not supported");
821     ik->set_shared_classpath_index(0);
822   } else {
823     assert(ik->defined_by_app_loader(), "must be");
824     ik->set_shared_classpath_index(AOTClassLocationConfig::dumptime()->app_cp_start_index());
825   }
826 
827   ArchiveBuilder::alloc_stats()->record_dynamic_proxy_class();
828   if (log_is_enabled(Info, cds, dynamic, proxy)) {
829     ResourceMark rm(THREAD);
830     stringStream ss;
831     const char* prefix = "";
832     ss.print("%s (%-7s, cp index = %d) implements ", ik->external_name(),
833              ArchiveUtils::builtin_loader_name(loader()), ik->shared_classpath_index());
834     objArrayOop intfs = (objArrayOop)interfaces();
835     for (int i = 0; i < intfs->length(); i++) {
836       oop intf_mirror = intfs->obj_at(i);
837       ss.print("%s%s", prefix, java_lang_Class::as_Klass(intf_mirror)->external_name());
838       prefix = ", ";
839     }
840 
841     log_info(cds, dynamic, proxy)("%s", ss.freeze());
842   }
843 }