From 4e790c4a6e23c951f2cc14694c19a0c3a838fd25 Mon Sep 17 00:00:00 2001
From: Isaac Marovitz <isaacryu@icloud.com>
Date: Sat, 21 Oct 2023 10:46:45 -0400
Subject: [PATCH] Correct signatures

---
 .../Am/Ipc/Controllers/DisplayController.cs               | 8 +++++---
 .../Sdk/Am/Controllers/IDisplayController.cs              | 7 ++++---
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/src/Ryujinx.Horizon/Am/Ipc/Controllers/DisplayController.cs b/src/Ryujinx.Horizon/Am/Ipc/Controllers/DisplayController.cs
index 1bad24931..f7ba2c9e5 100644
--- a/src/Ryujinx.Horizon/Am/Ipc/Controllers/DisplayController.cs
+++ b/src/Ryujinx.Horizon/Am/Ipc/Controllers/DisplayController.cs
@@ -2,13 +2,15 @@ using Ryujinx.Common.Logging;
 using Ryujinx.Horizon.Common;
 using Ryujinx.Horizon.Sdk.Am.Controllers;
 using Ryujinx.Horizon.Sdk.Sf;
+using Ryujinx.Horizon.Sdk.Sf.Hipc;
+using System;
 
 namespace Ryujinx.Horizon.Am.Ipc.Controllers
 {
     partial class DisplayController : IDisplayController
     {
         [CmifCommand(0)]
-        public Result GetLastForegroundCaptureImage()
+        public Result GetLastForegroundCaptureImage([Buffer(HipcBufferFlags.Out | HipcBufferFlags.MapAlias)] Span<byte> capture)
         {
             Logger.Stub?.PrintStub(LogClass.ServiceAm);
 
@@ -16,7 +18,7 @@ namespace Ryujinx.Horizon.Am.Ipc.Controllers
         }
 
         [CmifCommand(1)]
-        public Result UpdateLastForegroundCaptureImage()
+        public Result UpdateLastForegroundCaptureImage([Buffer(HipcBufferFlags.Out | HipcBufferFlags.MapAlias)] Span<byte> capture)
         {
             Logger.Stub?.PrintStub(LogClass.ServiceAm);
 
@@ -24,7 +26,7 @@ namespace Ryujinx.Horizon.Am.Ipc.Controllers
         }
 
         [CmifCommand(2)]
-        public Result GetLastApplicationCaptureImage()
+        public Result GetLastApplicationCaptureImage([Buffer(HipcBufferFlags.Out | HipcBufferFlags.MapAlias)] Span<byte> capture)
         {
             Logger.Stub?.PrintStub(LogClass.ServiceAm);
 
diff --git a/src/Ryujinx.Horizon/Sdk/Am/Controllers/IDisplayController.cs b/src/Ryujinx.Horizon/Sdk/Am/Controllers/IDisplayController.cs
index 8c7d94a20..263093003 100644
--- a/src/Ryujinx.Horizon/Sdk/Am/Controllers/IDisplayController.cs
+++ b/src/Ryujinx.Horizon/Sdk/Am/Controllers/IDisplayController.cs
@@ -1,12 +1,13 @@
 using Ryujinx.Horizon.Common;
+using System;
 
 namespace Ryujinx.Horizon.Sdk.Am.Controllers
 {
     public interface IDisplayController
     {
-        Result GetLastForegroundCaptureImage();
-        Result UpdateLastForegroundCaptureImage();
-        Result GetLastApplicationCaptureImage();
+        Result GetLastForegroundCaptureImage(Span<byte> capture);
+        Result UpdateLastForegroundCaptureImage(Span<byte> capture);
+        Result GetLastApplicationCaptureImage(Span<byte> capture);
         Result GetCallerAppletCaptureImage();
         Result UpdateCallerAppletCaptureImage();
         Result GetLastForegroundCaptureImageEx();