Ryujinx.Horizon/Sdk/Sf/Cmif/DomainServiceObjectDispatchTable.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

using Ryujinx.Horizon.Common;
using System;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;

namespace Ryujinx.Horizon.Sdk.Sf.Cmif
{
    class DomainServiceObjectDispatchTable : ServiceDispatchTableBase
    {
        public override Result ProcessMessage(ref ServiceDispatchContext context, ReadOnlySpan<byte> inRawData)
        {
            return ProcessMessageImpl(ref context, ((DomainServiceObject)context.ServiceObject).GetServerDomain(), inRawData);
        }

        private Result ProcessMessageImpl(ref ServiceDispatchContext context, ServerDomainBase domain, ReadOnlySpan<byte> inRawData)
        {
            if (inRawData.Length < Unsafe.SizeOf<CmifDomainInHeader>())
            {
                return SfResult.InvalidHeaderSize;
            }

            var inHeader = MemoryMarshal.Cast<byte, CmifDomainInHeader>(inRawData)[0];

            ReadOnlySpan<byte> inDomainRawData = inRawData.Slice(Unsafe.SizeOf<CmifDomainInHeader>());

            int targetObjectId = inHeader.ObjectId;

            switch (inHeader.Type)
            {
                case CmifDomainRequestType.SendMessage:
                    var targetObject = domain.GetObject(targetObjectId);
                    if (targetObject == null)
                    {
                        return SfResult.TargetNotFound;
                    }

                    if (inHeader.DataSize + inHeader.ObjectsCount * sizeof(int) > inDomainRawData.Length)
                    {
                        return SfResult.InvalidHeaderSize;
                    }

                    ReadOnlySpan<byte> inMessageRawData = inDomainRawData.Slice(0, inHeader.DataSize);

                    if (inHeader.ObjectsCount > DomainServiceObjectProcessor.MaximumObjects)
                    {
                        return SfResult.InvalidInObjectsCount;
                    }

                    int[] inObjectIds = new int[inHeader.ObjectsCount];

                    var domainProcessor = new DomainServiceObjectProcessor(domain, inObjectIds);

                    if (context.Processor == null)
                    {
                        context.Processor = domainProcessor;
                    }
                    else
                    {
                        context.Processor.SetImplementationProcessor(domainProcessor);
                    }

                    context.ServiceObject = targetObject.ServiceObject;

                    return targetObject.ProcessMessage(ref context, inMessageRawData);

                case CmifDomainRequestType.Close:
                    domain.UnregisterObject(targetObjectId);
                    return Result.Success;

                default:
                    return SfResult.InvalidInHeader;
            }
        }
    }
}