Provide equals/hashCode based on original PSI to avoid potential PCEs

#KT-59445

analysis/analysis-api-impl-base/src/org/jetbrains/kotlin/analysis/api/impl/base/java/source/JavaElementPsiSourceWithSmartPointer.kt

@@ -15,10 +15,19 @@ internal class JavaElementPsiSourceWithSmartPointer<PSI : PsiElement>(
     override val factory: JavaElementSourceFactory,
 ) : JavaElementPsiSource<PSI>() {
 
+    // is used only for the purposes of equals/hashCode to avoid underlying PCE
+    private val originalPsi: PSI = psi
+
     override val psi: PSI
         get() {
             return pointer.element
                 ?: error("Cannot restore a PsiElement from $pointer")
         }
+
+    override fun equals(other: Any?): Boolean {
+        return if (other === this) true else other is JavaElementPsiSourceWithSmartPointer<*> && originalPsi == other.originalPsi
+    }
+
+    override fun hashCode(): Int = originalPsi.hashCode()
 }
 

compiler/resolution.common.jvm/src/org/jetbrains/kotlin/load/java/structure/impl/source/JavaElementPsiSource.kt

@@ -18,6 +18,12 @@ class JavaElementPsiSourceWithFixedPsi<PSI : PsiElement>(
     override val factory: JavaElementSourceFactory
         get() = JavaElementSourceFactory.getInstance(psi.project)
 
+    override fun equals(other: Any?): Boolean {
+        return if (other === this) true else other is JavaElementPsiSourceWithFixedPsi<*> && psi == other.psi
+    }
+
+    override fun hashCode(): Int = psi.hashCode()
+
     override fun toString(): String {
         return psi.toString()
     }