From 40897e8dd27dc58ce009ff04a5dd50ccd49c6c33 Mon Sep 17 00:00:00 2001
From: Jared Goodwin <translucency@outlook.com>
Date: Sun, 2 Feb 2020 09:05:21 -0800
Subject: [PATCH] Add DX screen count check in Windows ScreenCaster.

---
 ScreenCast.Win/Program.cs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/ScreenCast.Win/Program.cs b/ScreenCast.Win/Program.cs
index c3635d72..e52abbd9 100644
--- a/ScreenCast.Win/Program.cs
+++ b/ScreenCast.Win/Program.cs
@@ -13,6 +13,7 @@ using Remotely.ScreenCast.Win.Capture;
 using Remotely.ScreenCast.Core.Communication;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
+using System.Windows.Forms;
 
 namespace Remotely.ScreenCast.Win
 {
@@ -81,7 +82,17 @@ namespace Remotely.ScreenCast.Win
             {
                 try
                 {
-                    return new DXCapture();
+                    var dxCapture = new DXCapture();
+                    if (dxCapture.GetScreenCount() == Screen.AllScreens.Length)
+                    {
+                        return dxCapture;
+                    }
+                    else
+                    {
+                        Logger.Write("DX screen count doesn't match.  Using CPU capturer instead.");
+                        dxCapture.Dispose();
+                        return new BitBltCapture();
+                    }
                 }
                 catch
                 {