[kotlin] fix NPE from lockWithPCECheck when action lambda returns a nullable value

analysis/low-level-api-fir/src/org/jetbrains/kotlin/analysis/low/level/api/fir/util/utils.kt

@@ -25,21 +25,17 @@ internal inline fun <T> executeWithoutPCE(crossinline action: () -> T): T {
 }
 
 internal inline fun <T> Lock.lockWithPCECheck(lockingIntervalMs: Long, action: () -> T): T {
-    var needToRun = true
+    while (true) {
-    var result: T? = null
-    while (needToRun) {
         checkCanceled()
         if (tryLock(lockingIntervalMs, TimeUnit.MILLISECONDS)) {
             try {
                 checkCanceled()
-                needToRun = false
+                return action()
-                result = action()
             } finally {
                 unlock()
             }
         }
     }
-    return result!!
 }
 
 @Suppress("NOTHING_TO_INLINE")