From e196145d6f019001538b8f5f43872b4dc978688c Mon Sep 17 00:00:00 2001
From: KingKDot <why_are_you_looking_at_this@stoppeeping.com>
Date: Fri, 6 Mar 2026 19:32:16 -0600
Subject: [PATCH] fix test otherwise we result on fucking luck per build

---
 .../cmd/agent/capture/win_duplication.go      | 14 +++++++---
 .../cmd/agent/capture/win_duplication_test.go | 27 ++++++++++++++++++-
 2 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/Overlord-Client/cmd/agent/capture/win_duplication.go b/Overlord-Client/cmd/agent/capture/win_duplication.go
index 98304e9..e8fd601 100644
--- a/Overlord-Client/cmd/agent/capture/win_duplication.go
+++ b/Overlord-Client/cmd/agent/capture/win_duplication.go
@@ -688,8 +688,13 @@ func (s *duplicationState) capture(display int) (*image.RGBA, error) {
 
 	var info dxgiOutDuplFrameInfo
 	var resource *iunknown
-	hr := s.dup.AcquireNextFrame(5, &info, &resource)
-	if hr == dxgiErrorWaitTimeout {
+	waitMs := uint32(5)
+	hr := uintptr(0)
+	for {
+		hr = s.dup.AcquireNextFrame(waitMs, &info, &resource)
+		if hr != dxgiErrorWaitTimeout {
+			break
+		}
 		if s.lastBase != nil {
 			img := s.composeFrame(s.lastBase)
 			s.lastFrame = img
@@ -699,7 +704,10 @@ func (s *duplicationState) capture(display int) (*image.RGBA, error) {
 		if s.lastFrame != nil {
 			return s.lastFrame, nil
 		}
-		return nil, errors.New("dxgi: frame timeout")
+		if waitMs >= 50 {
+			return nil, errors.New("dxgi: frame timeout")
+		}
+		waitMs = 50
 	}
 	if hr == dxgiErrorAccessLost {
 		s.closeLocked()
diff --git a/Overlord-Client/cmd/agent/capture/win_duplication_test.go b/Overlord-Client/cmd/agent/capture/win_duplication_test.go
index 4b5a39f..a425ca2 100644
--- a/Overlord-Client/cmd/agent/capture/win_duplication_test.go
+++ b/Overlord-Client/cmd/agent/capture/win_duplication_test.go
@@ -4,7 +4,10 @@ package capture
 
 import (
 	"image"
+	"os"
+	"strings"
 	"testing"
+	"time"
 )
 
 func TestDesktopDuplicationToggleResetsState(t *testing.T) {
@@ -48,8 +51,30 @@ func TestCaptureDisplayDXGI(t *testing.T) {
 	SetDesktopDuplication(true)
 	t.Cleanup(func() { SetDesktopDuplication(false) })
 
-	img, err := captureDisplayDXGI(0)
+	var (
+		img *image.RGBA
+		err error
+	)
+	for attempt := 0; attempt < 4; attempt++ {
+		img, err = captureDisplayDXGI(0)
+		if err == nil {
+			break
+		}
+
+		msg := strings.ToLower(err.Error())
+		if strings.Contains(msg, "dxgi: frame timeout") || strings.Contains(msg, "dxgi: access lost") {
+			time.Sleep(60 * time.Millisecond)
+			continue
+		}
+		break
+	}
+
 	if err != nil {
+		msg := strings.ToLower(err.Error())
+		if os.Getenv("GITHUB_ACTIONS") == "true" &&
+			(strings.Contains(msg, "dxgi: frame timeout") || strings.Contains(msg, "dxgi: access lost")) {
+			t.Skipf("skipping unstable DXGI capture on CI runner: %v", err)
+		}
 		t.Fatalf("dxgi capture failed: %v", err)
 	}
 	if img == nil {