Merge 746c5d0 into 3548754

Author: markushi

sentry-android-core/src/main/java/io/sentry/android/core/internal/util/ScreenshotUtils.java

@@ -5,7 +5,12 @@
 import android.graphics.Bitmap;
 import android.graphics.Canvas;
 import android.os.Build;
+import android.os.Handler;
+import android.os.HandlerThread;
+import android.view.PixelCopy;
 import android.view.View;
+import android.view.Window;
+
 import androidx.annotation.Nullable;
 import io.sentry.ILogger;
 import io.sentry.SentryLevel;
@@ -14,6 +19,8 @@
 import java.io.ByteArrayOutputStream;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicBoolean;
+
 import org.jetbrains.annotations.ApiStatus;
 import org.jetbrains.annotations.NotNull;
 
@@ -38,13 +45,14 @@ public class ScreenshotUtils {
 
     if (!isActivityValid(activity, buildInfoProvider)
         || activity.getWindow() == null
-        || activity.getWindow().getDecorView() == null
-        || activity.getWindow().getDecorView().getRootView() == null) {
+        || activity.getWindow().peekDecorView() == null
+        || activity.getWindow().peekDecorView().getRootView() == null) {
       logger.log(SentryLevel.DEBUG, "Activity isn't valid, not taking screenshot.");
       return null;
     }
 
-    final View view = activity.getWindow().getDecorView().getRootView();
+    final Window window = activity.getWindow();
+    final View view = window.peekDecorView().getRootView();
     if (view.getWidth() <= 0 || view.getHeight() <= 0) {
       logger.log(SentryLevel.DEBUG, "View's width and height is zeroed, not taking screenshot.");
       return null;
@@ -55,20 +63,55 @@ public class ScreenshotUtils {
       final Bitmap bitmap =
           Bitmap.createBitmap(view.getWidth(), view.getHeight(), Bitmap.Config.ARGB_8888);
 
-      final Canvas canvas = new Canvas(bitmap);
-      if (mainThreadChecker.isMainThread()) {
-        view.draw(canvas);
+      final @NotNull CountDownLatch latch = new CountDownLatch(1);
+
+      // Use Pixel Copy API on new devices, fallback to canvas rendering on older ones
+      if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.O) {
+
+        final HandlerThread thread = new HandlerThread("screenshot");
+        thread.start();
+
+        boolean success = false;
+        try {
+          final Handler handler = new Handler(thread.getLooper());
+          final AtomicBoolean copyResultSuccess = new AtomicBoolean(false);
+
+          PixelCopy.request(
+            window,
+            bitmap, copyResult -> {
+              copyResultSuccess.set(copyResult == PixelCopy.SUCCESS);
+              latch.countDown();
+            },
+            handler);
+
+          success = latch.await(CAPTURE_TIMEOUT_MS, TimeUnit.MILLISECONDS) && copyResultSuccess.get();
+        } catch (InterruptedException e) {
+          // ignored
+        } finally {
+          thread.quit();
+        }
+
+        if (!success) {
+          return null;
+        }
       } else {
-        final @NotNull CountDownLatch latch = new CountDownLatch(1);
-        activity.runOnUiThread(
-            () -> {
-              try {
-                view.draw(canvas);
-                latch.countDown();
-              } catch (Throwable e) {
-                logger.log(SentryLevel.ERROR, "Taking screenshot failed (view.draw).", e);
-              }
-            });
+        final Canvas canvas = new Canvas(bitmap);
+        if (mainThreadChecker.isMainThread()) {
+          view.draw(canvas);
+          latch.countDown();
+        } else {
+          activity.runOnUiThread(
+              () -> {
+                try {
+                  view.draw(canvas);
+                } catch (Throwable e) {
+                  logger.log(SentryLevel.ERROR, "Taking screenshot failed (view.draw).", e);
+                } finally {
+                  latch.countDown();
+                }
+              });
+        }
+
         if (!latch.await(CAPTURE_TIMEOUT_MS, TimeUnit.MILLISECONDS)) {
           return null;
         }

sentry-android-core/src/test/java/io/sentry/android/core/ScreenshotEventProcessorTest.kt

@@ -46,6 +46,7 @@ class ScreenshotEventProcessorTest {
             whenever(rootView.height).thenReturn(1)
             whenever(view.rootView).thenReturn(rootView)
             whenever(window.decorView).thenReturn(view)
+            whenever(window.peekDecorView()).thenReturn(view)
             whenever(activity.window).thenReturn(window)
             whenever(activity.runOnUiThread(any())).then {
                 it.getArgument<Runnable>(0).run()