fix: detection when app launched during active recording

Author: yardexx

android/src/main/kotlin/com/aheaditec/freerasp/FreeraspPlugin.kt

@@ -16,17 +16,16 @@ import io.flutter.embedding.engine.plugins.lifecycle.FlutterLifecycleAdapter
 /** FreeraspPlugin */
 class FreeraspPlugin : FlutterPlugin, ActivityAware, LifecycleEventObserver {
     private var streamHandler: StreamHandler = StreamHandler()
-    private var methodCallHandler: MethodCallHandler = MethodCallHandler()
     private var screenProtector: ScreenProtector? =
         if (Build.VERSION.SDK_INT >= 34) ScreenProtector() else null
+    private var methodCallHandler: MethodCallHandler = MethodCallHandler(screenProtector)
 
     private var context: Context? = null
     private var lifecycle: Lifecycle? = null
 
     override fun onAttachedToEngine(flutterPluginBinding: FlutterPlugin.FlutterPluginBinding) {
         val messenger = flutterPluginBinding.binaryMessenger
         context = flutterPluginBinding.applicationContext
-        screenProtector?.enable()
         methodCallHandler.createMethodChannel(messenger, flutterPluginBinding.applicationContext)
         streamHandler.createEventChannel(messenger)
     }

android/src/main/kotlin/com/aheaditec/freerasp/ScreenProtector.kt

@@ -36,6 +36,7 @@ internal class ScreenProtector : DefaultLifecycleObserver {
     internal fun enable() {
         if (isEnabled) return
         isEnabled = true
+        activity?.let { register(it) }
     }
 
     override fun onStart(owner: LifecycleOwner) {

android/src/main/kotlin/com/aheaditec/freerasp/handlers/MethodCallHandler.kt

@@ -8,9 +8,10 @@ import android.os.Looper
 import androidx.lifecycle.Lifecycle
 import androidx.lifecycle.LifecycleEventObserver
 import androidx.lifecycle.LifecycleOwner
-import com.aheaditec.freerasp.runResultCatching
+import com.aheaditec.freerasp.ScreenProtector
 import com.aheaditec.freerasp.Utils
 import com.aheaditec.freerasp.generated.TalsecPigeonApi
+import com.aheaditec.freerasp.runResultCatching
 import com.aheaditec.freerasp.toPigeon
 import com.aheaditec.talsec_security.security.api.SuspiciousAppInfo
 import com.aheaditec.talsec_security.security.api.Talsec
@@ -23,7 +24,8 @@ import io.flutter.plugin.common.MethodChannel.MethodCallHandler
 /**
  * A method handler that creates and manages an [MethodChannel] for freeRASP methods.
  */
-internal class MethodCallHandler : MethodCallHandler, LifecycleEventObserver {
+internal class MethodCallHandler(private val screenProtector: ScreenProtector?) : MethodCallHandler,
+    LifecycleEventObserver {
     private var context: Context? = null
     private var methodChannel: MethodChannel? = null
     private var pigeonApi: TalsecPigeonApi? = null
@@ -137,6 +139,7 @@ internal class MethodCallHandler : MethodCallHandler, LifecycleEventObserver {
             val talsecConfig = Utils.toTalsecConfigThrowing(config)
             context?.let {
                 TalsecThreatHandler.start(it, talsecConfig)
+                screenProtector?.enable()
             } ?: throw IllegalStateException("Unable to run Talsec - context is null")
             result.success(null)
         }