1 files changed, 164 insertions, 10 deletions

diff --git a/Ryujinx.HLE/HOS/Services/Ns/IAddOnContentManager.cs b/Ryujinx.HLE/HOS/Services/Ns/IAddOnContentManager.cs
index 13d56934..20d95cbb 100644
--- a/Ryujinx.HLE/HOS/Services/Ns/IAddOnContentManager.cs
+++ b/Ryujinx.HLE/HOS/Services/Ns/IAddOnContentManager.cs
@@ -1,32 +1,186 @@
 using Ryujinx.Common.Logging;
+using Ryujinx.HLE.HOS.Ipc;
+using Ryujinx.HLE.HOS.Kernel.Common;
+using Ryujinx.HLE.HOS.Kernel.Threading;
+using System;
 
 namespace Ryujinx.HLE.HOS.Services.Ns
 {
     [Service("aoc:u")]
     class IAddOnContentManager : IpcService
     {
-        public IAddOnContentManager(ServiceCtx context) { }
+        KEvent _addOnContentListChangedEvent;
+
+        public IAddOnContentManager(ServiceCtx context)
+        {
+            _addOnContentListChangedEvent = new KEvent(context.Device.System.KernelContext);
+        }
 
         [Command(2)]
-        // CountAddOnContent(u64, pid) -> u32
-        public static ResultCode CountAddOnContent(ServiceCtx context)
+        // CountAddOnContent(pid) -> u32
+        public ResultCode CountAddOnContent(ServiceCtx context)
         {
-            context.ResponseData.Write(0);
+            long pid = context.Process.Pid;
 
-            Logger.PrintStub(LogClass.ServiceNs);
+            // Official code checks ApplicationControlProperty.RuntimeAddOnContentInstall 
+            // if true calls ns:am ListAvailableAddOnContent again to get updated count
+
+            byte runtimeAddOnContentInstall = context.Device.Application.ControlData.Value.RuntimeAddOnContentInstall;
+            if (runtimeAddOnContentInstall != 0)
+            {
+                Logger.PrintWarning(LogClass.ServiceNs, $"RuntimeAddOnContentInstall is true. Some DLC may be missing");
+            }
+
+            uint aocCount = CountAddOnContentImpl(context);
+
+            context.ResponseData.Write(aocCount);
+
+            Logger.PrintStub(LogClass.ServiceNs, new { aocCount, runtimeAddOnContentInstall });
 
             return ResultCode.Success;
         }
 
+        private static uint CountAddOnContentImpl(ServiceCtx context)
+        {
+            return (uint)context.Device.System.ContentManager.GetAocCount();
+        }
+
         [Command(3)]
-        // ListAddOnContent(u32, u32, u64, pid) -> (u32, buffer<u32, 6>)
-        public static ResultCode ListAddOnContent(ServiceCtx context)
+        // ListAddOnContent(u32, u32, pid) -> (u32, buffer<u32>)
+        public ResultCode ListAddOnContent(ServiceCtx context)
+        {
+            uint startIndex = context.RequestData.ReadUInt32();
+            uint bufferSize = context.RequestData.ReadUInt32();
+            long pid = context.Process.Pid;
+
+            var aocTitleIds = context.Device.System.ContentManager.GetAocTitleIds();
+
+            uint aocCount = CountAddOnContentImpl(context);
+
+            if (aocCount <= startIndex)
+            {
+                context.ResponseData.Write((uint)0);
+
+                return ResultCode.Success;
+            }
+
+            aocCount = Math.Min(aocCount - startIndex, bufferSize);
+
+            context.ResponseData.Write(aocCount);
+
+            ulong bufAddr = (ulong)context.Request.ReceiveBuff[0].Position;
+
+            ulong aocBaseId = GetAddOnContentBaseIdImpl(context);
+
+            for (int i = 0; i < aocCount; ++i)
+            {
+                context.Memory.Write(bufAddr + (ulong)i * 4, (int)(aocTitleIds[i + (int)startIndex] - aocBaseId));
+            }
+
+            Logger.PrintStub(LogClass.ServiceNs, new { bufferSize, startIndex, aocCount });
+
+            return ResultCode.Success;
+        }
+
+        [Command(5)]
+        // GetAddOnContentBaseId(pid) -> u64
+        public ResultCode GetAddonContentBaseId(ServiceCtx context)
+        {
+            long pid = context.Process.Pid;
+
+            // Official code calls arp:r GetApplicationControlProperty to get AddOnContentBaseId
+            // If the call fails, calls arp:r GetApplicationLaunchProperty to get App TitleId
+            ulong aocBaseId = GetAddOnContentBaseIdImpl(context);
+
+            context.ResponseData.Write(aocBaseId);
+
+            Logger.PrintStub(LogClass.ServiceNs, $"aocBaseId={aocBaseId:X16}");
+
+            // ResultCode will be error code of GetApplicationLaunchProperty if it fails
+            return ResultCode.Success;
+        }
+
+        private static ulong GetAddOnContentBaseIdImpl(ServiceCtx context)
+        {
+            ulong aocBaseId = context.Device.Application.ControlData.Value.AddOnContentBaseId;
+
+            if (aocBaseId == 0)
+            {
+                aocBaseId = context.Device.Application.TitleId + 0x1000;
+            }
+
+            return aocBaseId;
+        }
+
+        [Command(7)]
+        // PrepareAddOnContent(u32, pid)
+        public ResultCode PrepareAddOnContent(ServiceCtx context)
+        {
+            uint aocIndex = context.RequestData.ReadUInt32();
+            long pid = context.Process.Pid;
+
+            // Official Code calls a bunch of functions from arp:r for aocBaseId
+            // and ns:am RegisterContentsExternalKey?, GetOwnedApplicationContentMetaStatus? etc...
+
+            // Ideally, this should probably initialize the AocData values for the specified index
+
+            Logger.PrintStub(LogClass.ServiceNs, new { aocIndex });
+
+            return ResultCode.Success;
+        }
+
+        [Command(8)]
+        // GetAddOnContentListChangedEvent() -> handle<copy>
+        public ResultCode GetAddOnContentListChangedEvent(ServiceCtx context)
+        {
+            // Official code seems to make an internal call to ns:am Cmd 84 GetDynamicCommitEvent()
+
+            if (context.Process.HandleTable.GenerateHandle(_addOnContentListChangedEvent.ReadableEvent, out int handle) != KernelResult.Success)
+            {
+                throw new InvalidOperationException("Out of handles!");
+            }
+
+            context.Response.HandleDesc = IpcHandleDesc.MakeCopy(handle);
+
+            Logger.PrintStub(LogClass.ServiceNs);
+
+            return ResultCode.Success;
+        }
+
+
+        [Command(9)] // [10.0.0+] 
+        // GetAddOnContentLostErrorCode() -> u64
+        public ResultCode GetAddOnContentLostErrorCode(ServiceCtx context)
+        {
+            // Seems to calculate ((value & 0x1ff)) + 2000 on 0x7d0a4
+            // which gives 0x874 (2000+164). 164 being Module ID of `EC (Shop)`
+            context.ResponseData.Write(2164L);
+
+            Logger.PrintStub(LogClass.ServiceNs);
+
+            return ResultCode.Success;
+        }
+
+        [Command(100)]
+        // CreateEcPurchasedEventManager() -> object<nn::ec::IPurchaseEventManager>
+        public ResultCode CreateEcPurchasedEventManager(ServiceCtx context)
         {
+            MakeObject(context, new IPurchaseEventManager());
+
             Logger.PrintStub(LogClass.ServiceNs);
 
-            // TODO: This is supposed to write a u32 array aswell.
-            // It's unknown what it contains.
-            context.ResponseData.Write(0);
+            return ResultCode.Success;
+        }
+
+        [Command(101)]
+        // CreatePermanentEcPurchasedEventManager() -> object<nn::ec::IPurchaseEventManager>
+        public ResultCode CreatePermanentEcPurchasedEventManager(ServiceCtx context)
+        {
+            // Very similar to CreateEcPurchasedEventManager but with some extra code
+
+            MakeObject(context, new IPurchaseEventManager());
+
+            Logger.PrintStub(LogClass.ServiceNs);
 
             return ResultCode.Success;
         }