Auto-consent before phone verification

Author: opstic

play-services-constellation/core/src/main/kotlin/org/microg/gms/constellation/core/VerifyPhoneNumber.kt

@@ -23,15 +23,20 @@ import kotlinx.coroutines.withContext
 import org.microg.gms.common.Constants
 import org.microg.gms.constellation.core.proto.AsterismClient
 import org.microg.gms.constellation.core.proto.Consent
+import org.microg.gms.constellation.core.proto.ConsentVersion
 import org.microg.gms.constellation.core.proto.DeviceID
 import org.microg.gms.constellation.core.proto.GetConsentRequest
 import org.microg.gms.constellation.core.proto.GetConsentResponse
+import org.microg.gms.constellation.core.proto.Param
+import org.microg.gms.constellation.core.proto.RcsConsent
 import org.microg.gms.constellation.core.proto.RequestHeader
+import org.microg.gms.constellation.core.proto.SetConsentRequest
 import org.microg.gms.constellation.core.proto.SyncRequest
 import org.microg.gms.constellation.core.proto.Verification
 import org.microg.gms.constellation.core.proto.builder.RequestBuildContext
 import org.microg.gms.constellation.core.proto.builder.buildImsiToSubscriptionInfoMap
 import org.microg.gms.constellation.core.proto.builder.buildRequestContext
+import org.microg.gms.constellation.core.proto.builder.getList
 import org.microg.gms.constellation.core.proto.builder.invoke
 import org.microg.gms.constellation.core.verification.ChallengeProcessor
 import org.microg.gms.constellation.core.verification.MtSmsInboxRegistry
@@ -212,7 +217,6 @@ private suspend fun handleVerifyPhoneNumberRequest(
             e is PhoneNumberVerificationDisabledException -> Status(5000)
             readCallbackMode != ReadCallbackMode.NONE -> Status.INTERNAL_ERROR
             e is GrpcException -> handleRpcError(e)
-            e is NoConsentException -> Status(5001)
             else -> Status.INTERNAL_ERROR
         }
     }
@@ -254,8 +258,6 @@ private fun handleRpcError(error: GrpcException): Status {
     return Status(statusCode, error.message)
 }
 
-private class NoConsentException : Exception("No consent")
-
 private suspend fun runVerificationFlow(
     context: Context,
     request: VerifyPhoneNumberRequest,
@@ -287,8 +289,24 @@ private suspend fun runVerificationFlow(
                 }
 
         if (!consented) {
-            Log.e(TAG, "Consent has not been set. Not running verification.")
-            throw NoConsentException()
+            Log.e(TAG, "Consent has not been set. Auto-setting consent.")
+            val consentType = parseConsentVersion(request.extras)
+            val setRequest = SetConsentRequest(
+                header_ = RequestHeader(context, sessionId, buildContext, "setConsent"),
+                asterism_client = asterismClient,
+                rcs_consent = RcsConsent(
+                    consent = Consent.CONSENTED,
+                    consent_version = consentType
+                ),
+                consent_version = consentType,
+                api_params = Param.getList(request.extras)
+            )
+            try {
+                RpcClient.phoneDeviceVerificationClient.SetConsent().execute(setRequest)
+                Log.i(TAG, "Auto-consented for $asterismClient")
+            } catch (e: Exception) {
+                Log.w(TAG, "Auto-consent failed", e)
+            }
         }
     }
 
@@ -322,6 +340,14 @@ private suspend fun runVerificationFlow(
     return verifications
 }
 
+private fun parseConsentVersion(extras: Bundle): ConsentVersion {
+    val value = extras.getString("consent_type")
+    val parsed = value?.toIntOrNull()?.let(ConsentVersion::fromValue)
+        ?: value?.let { runCatching { ConsentVersion.valueOf(it) }.getOrNull() }
+    return parsed?.takeUnless { it == ConsentVersion.CONSENT_VERSION_UNSPECIFIED }
+        ?: ConsentVersion.RCS_DEFAULT_ON_LEGAL_FYI
+}
+
 private fun PhoneNumberVerification.toLegacyPhoneNumberInfoOrNull(): PhoneNumberInfo? {
     if (
         verificationStatus != Verification.Status.STATUS_VERIFIED.toClientStatus() ||