From 21ae6e46b7de7d1bfa6c9fee441ca95ed115111a Mon Sep 17 00:00:00 2001
From: Pavel Kunyavskiy <pavel.kunyavskiy@jetbrains.com>
Date: Tue, 20 Feb 2024 12:53:14 +0100
Subject: [PATCH] [IRFakeOverrides] Also replace special symbols in annotations

This fixes some crashes on red code in diagnostic tests.
Also can be useful in the future, when we would allow things like
enum.name in annotation arguments.

^KT-65971
---
 .../SpecialFakeOverrideSymbolsResolver.kt     | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/compiler/ir/ir.actualization/src/main/kotlin/org/jetbrains/kotlin/backend/common/actualizer/SpecialFakeOverrideSymbolsResolver.kt b/compiler/ir/ir.actualization/src/main/kotlin/org/jetbrains/kotlin/backend/common/actualizer/SpecialFakeOverrideSymbolsResolver.kt
index ef06b01de7b..49c97290d38 100644
--- a/compiler/ir/ir.actualization/src/main/kotlin/org/jetbrains/kotlin/backend/common/actualizer/SpecialFakeOverrideSymbolsResolver.kt
+++ b/compiler/ir/ir.actualization/src/main/kotlin/org/jetbrains/kotlin/backend/common/actualizer/SpecialFakeOverrideSymbolsResolver.kt
@@ -20,6 +20,7 @@ import org.jetbrains.kotlin.ir.symbols.impl.IrFakeOverrideSymbolBase
 import org.jetbrains.kotlin.ir.util.SymbolRemapper
 import org.jetbrains.kotlin.ir.visitors.IrElementVisitorVoid
 import org.jetbrains.kotlin.ir.visitors.acceptChildrenVoid
+import org.jetbrains.kotlin.ir.visitors.acceptVoid
 import org.jetbrains.kotlin.utils.addToStdlib.shouldNotBeCalled
 
 /**
@@ -116,40 +117,46 @@ class SpecialFakeOverrideSymbolsResolver(private val expectActualMap: Map<IrSymb
 
 class SpecialFakeOverrideSymbolsResolverVisitor(private val resolver: SpecialFakeOverrideSymbolsResolver) : IrElementVisitorVoid {
     override fun visitElement(element: IrElement) {
+        // E.g. annotation can contain fake override of constant property
+        if (element is IrAnnotationContainer) {
+            for (annotation in element.annotations) {
+                annotation.acceptVoid(this)
+            }
+        }
         element.acceptChildrenVoid(this)
     }
 
     override fun visitSimpleFunction(declaration: IrSimpleFunction) {
         declaration.overriddenSymbols = declaration.overriddenSymbols.map(resolver::getReferencedSimpleFunction)
-        declaration.acceptChildrenVoid(this)
+        visitElement(declaration)
     }
 
     override fun visitProperty(declaration: IrProperty) {
         declaration.overriddenSymbols = declaration.overriddenSymbols.map(resolver::getReferencedProperty)
-        declaration.acceptChildrenVoid(this)
+        visitElement(declaration)
     }
 
     override fun visitCall(expression: IrCall) {
         expression.symbol = resolver.getReferencedSimpleFunction(expression.symbol)
-        expression.acceptChildrenVoid(this)
+        visitElement(expression)
     }
 
     override fun visitFunctionReference(expression: IrFunctionReference) {
         expression.symbol = resolver.getReferencedFunction(expression.symbol)
         expression.reflectionTarget = expression.reflectionTarget?.let(resolver::getReferencedFunction)
-        expression.acceptChildrenVoid(this)
+        visitElement(expression)
     }
 
     override fun visitPropertyReference(expression: IrPropertyReference) {
         expression.symbol = expression.symbol.let(resolver::getReferencedProperty)
         expression.getter = expression.getter?.let(resolver::getReferencedSimpleFunction)
         expression.setter = expression.setter?.let(resolver::getReferencedSimpleFunction)
-        expression.acceptChildrenVoid(this)
+        visitElement(expression)
     }
 
     override fun visitLocalDelegatedPropertyReference(expression: IrLocalDelegatedPropertyReference) {
         expression.getter = expression.getter.let(resolver::getReferencedSimpleFunction)
         expression.setter = expression.setter?.let(resolver::getReferencedSimpleFunction)
-        expression.acceptChildrenVoid(this)
+        visitElement(expression)
     }
 }