FIR: convert Java type parameter bounds before reading annotations

This avoids a crash due to circular class references through annotation
arguments.
This commit is contained in:
pyos
2022-09-15 11:48:33 +02:00
committed by teamcity
parent 04dae17333
commit 5ba76ee757
11 changed files with 108 additions and 12 deletions
@@ -164,9 +164,27 @@ abstract class FirJavaFacade(
parentClassTypeParameterStackCache.remove(classSymbol)
parentClassEffectiveVisibilityCache.remove(classSymbol)
// There's a bit of an ordering restriction here:
// 1. annotations should be added after the symbol is bound, as annotations can refer to the class itself;
// 2. type enhancement requires annotations to be already present (and supertypes can refer to type parameters).
// This is where the problems begin. We need to enhance nullability of super types and type parameter bounds,
// for which we need the annotations of this class as they may specify default nullability.
// However, all three - annotations, type parameter bounds, and supertypes - can refer to other classes,
// which will cause the type parameter bounds and supertypes of *those* classes to get enhanced first,
// but they may refer back to this class again - which, thanks to the magic of symbol resolver caches,
// will be observed in a state where we've not done the enhancement yet. For those cases, we must publish
// at least unenhanced resolved types, or else FIR may crash upon encountering a FirJavaTypeRef where FirResolvedTypeRef
// is expected.
// TODO: some (all?) of those loops can be avoided, e.g. we don't actually need to resolve class arguments of annotations
// to determine whether they set default nullability - but without laziness, breaking those loops is somewhat hard,
// as we have a nested ordering here.
for (typeParameter in firJavaClass.typeParameters) {
if (typeParameter is FirTypeParameter) {
typeParameter.replaceBounds(typeParameter.bounds.map {
it.resolveIfJavaType(session, javaTypeParameterStack, FirJavaTypeConversionMode.TYPE_PARAMETER_BOUND)
})
}
}
// 1. Resolve annotations
// 2. Enhance type parameter bounds - may refer to each other, take default nullability from annotations
// 3. Enhance super types - may refer to type parameter bounds, take default nullability from annotations
firJavaClass.annotations.addFromJava(session, javaClass, javaTypeParameterStack)
val enhancement = FirSignatureEnhancement(firJavaClass, session) { emptyList() }
enhancement.enhanceTypeParameterBounds(firJavaClass.typeParameters)
@@ -333,16 +333,11 @@ class FirSignatureEnhancement(
// (`A : B, B : A` - invalid), the cycles can still appear when type parameters use each other in argument
// position (`A : C<B>, B : D<A>` - valid). In this case the precise enhancement of each bound depends on
// the others' nullability, for which we need to enhance at least its head type constructor.
//
// While this is straightforward to do within a single class/method (enhance all bounds' head type
// constructors, then enhance fully), it's not so simple when two classes depend on each other (we need
// to enhance *both* classes' type parameters' bounds' heads first). This is why we replace each bound
// with an unenhanced version first: this ensures that the frontend at least doesn't fail.
//
// TODO: find a way to partially enhance type parameters of all classes before fully enhancing anything.
// TODO: should this be done in topological order on head type constructors?
// I.e. for `A : B, B : C<A>` should we process `B` first?
typeParameters.replaceBounds { _, bound ->
// Resolve without enhancement so we don't crash the frontend if there is a restricted cycle (`A : B, B : A`)
// or if we visit type parameters in the wrong order (`A : B, B : C<A>` with `A` enhanced before `B`).
// TODO: the second case technically produces incorrect results - the loop below should visit type parameters
// in topological order, then a resolved-but-not-enhanced type will never be observable with valid code.
bound.resolveIfJavaType(session, javaTypeParameterStack, FirJavaTypeConversionMode.TYPE_PARAMETER_BOUND)
}
typeParameters.replaceBounds { typeParameter, bound ->