From 9f722db568512ffe3effda16e58d38a7f3ea0d3a Mon Sep 17 00:00:00 2001
From: Charles Lombardo <clombardo169@gmail.com>
Date: Wed, 30 Oct 2024 20:49:03 -0400
Subject: [PATCH] Request battery optimization exemption

Requests an exemption to battery optimization when enabling remote verification
---
 app/src/main/AndroidManifest.xml              |  1 +
 .../auditor/AttestationActivity.java          | 45 ++++++++++++++-----
 2 files changed, 36 insertions(+), 10 deletions(-)

diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index 64955b26..202f16fa 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -11,6 +11,7 @@
     <uses-permission android:name="android.permission.USE_FINGERPRINT" android:maxSdkVersion="28" />
     <uses-permission android:name="android.permission.USE_BIOMETRIC" />
     <uses-permission android:name="android.permission.QUERY_ALL_PACKAGES" />
+    <uses-permission android:name="android.permission.REQUEST_IGNORE_BATTERY_OPTIMIZATIONS" />
 
     <application android:label="@string/app_name"
             android:useEmbeddedDex="true"
diff --git a/app/src/main/java/app/attestation/auditor/AttestationActivity.java b/app/src/main/java/app/attestation/auditor/AttestationActivity.java
index 900f2fe4..397cb481 100644
--- a/app/src/main/java/app/attestation/auditor/AttestationActivity.java
+++ b/app/src/main/java/app/attestation/auditor/AttestationActivity.java
@@ -6,6 +6,7 @@
 import android.Manifest;
 import android.annotation.SuppressLint;
 import android.app.Activity;
+import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
 import android.content.pm.PackageManager;
@@ -13,6 +14,8 @@
 import android.net.Uri;
 import android.os.Build;
 import android.os.Bundle;
+import android.os.PowerManager;
+import android.provider.Settings;
 import android.text.Html;
 import android.text.Spanned;
 import android.util.Log;
@@ -294,6 +297,10 @@ protected void onResume() {
             stage = Stage.EnableRemoteVerify;
             startQrScanner();
         });
+
+        if (isIgnoringBatteryOptimizations() && stage == Stage.EnableRemoteVerify) {
+            startQrScanner();
+        }
     }
 
     @Override
@@ -434,17 +441,17 @@ private Bitmap createQrCode(final byte[] contents) {
 
     @SuppressLint("InlinedApi")
     private void startQrScanner() {
-        if (checkSelfPermission(Manifest.permission.CAMERA) != PackageManager.PERMISSION_GRANTED) {
+        if (!isIgnoringBatteryOptimizations()) {
+            requestIgnoreBatteryOptimizations();
+        } else if (checkSelfPermission(Manifest.permission.CAMERA) != PackageManager.PERMISSION_GRANTED) {
             requestPermissions(new String[]{Manifest.permission.CAMERA},
                     PERMISSIONS_REQUEST_CAMERA);
+        } else if (checkSelfPermission(Manifest.permission.POST_NOTIFICATIONS) != PackageManager.PERMISSION_GRANTED) {
+            requestPermissions(new String[]{Manifest.permission.POST_NOTIFICATIONS},
+                    PERMISSIONS_REQUEST_POST_NOTIFICATIONS_REMOTE_VERIFY);
         } else {
-            if (stage == Stage.EnableRemoteVerify &&
-                    checkSelfPermission(Manifest.permission.POST_NOTIFICATIONS) != PackageManager.PERMISSION_GRANTED) {
-                requestPermissions(new String[]{Manifest.permission.POST_NOTIFICATIONS},
-                        PERMISSIONS_REQUEST_POST_NOTIFICATIONS_REMOTE_VERIFY);
-            } else {
-                QRScannerActivityLauncher.launch(new Intent(this, QRScannerActivity.class));
-            }
+            QRScannerActivityLauncher.launch(new Intent(this, QRScannerActivity.class));
+            stage = Stage.None;
         }
     }
 
@@ -459,8 +466,6 @@ public void onRequestPermissionsResult(int requestCode,
             } else {
                 snackbar.setText(R.string.camera_permission_denied).show();
             }
-        } else if (requestCode == PERMISSIONS_REQUEST_POST_NOTIFICATIONS_REMOTE_VERIFY) {
-            QRScannerActivityLauncher.launch(new Intent(this, QRScannerActivity.class));
         } else if (requestCode == PERMISSIONS_REQUEST_POST_NOTIFICATIONS_SUBMIT_SAMPLE) {
             SubmitSampleJob.schedule(this);
             snackbar.setText(R.string.schedule_submit_sample_success).show();
@@ -577,4 +582,24 @@ public boolean onOptionsItemSelected(final MenuItem item) {
         }
         return super.onOptionsItemSelected(item);
     }
+
+    private boolean isIgnoringBatteryOptimizations() {
+        PowerManager powerManager = (PowerManager) getSystemService(Context.POWER_SERVICE);
+        if (powerManager != null) {
+            return powerManager.isIgnoringBatteryOptimizations(getPackageName());
+        }
+        return true;
+    }
+
+    @SuppressLint("BatteryLife")
+    private void requestIgnoreBatteryOptimizations() {
+        try {
+            Log.d(TAG, "Requesting battery optimization exemption");
+            Intent intent = new Intent(Settings.ACTION_REQUEST_IGNORE_BATTERY_OPTIMIZATIONS);
+            intent.setData(Uri.parse("package:" + getPackageName()));
+            startActivity(intent);
+        } catch (Exception e) {
+            Log.e(TAG, "Failed to request to ignore battery optimizations: " + e.getMessage());
+        }
+    }
 }