src/Ryujinx.HLE/HOS/Kernel/Common/KAutoObject.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

using Ryujinx.Horizon.Common;
using System.Diagnostics;
using System.Threading;

namespace Ryujinx.HLE.HOS.Kernel.Common
{
    class KAutoObject
    {
        protected KernelContext KernelContext;

        private int _referenceCount;

        public KAutoObject(KernelContext context)
        {
            KernelContext = context;

            _referenceCount = 1;
        }

        public virtual Result SetName(string name)
        {
            if (!KernelContext.AutoObjectNames.TryAdd(name, this))
            {
                return KernelResult.InvalidState;
            }

            return Result.Success;
        }

        public static Result RemoveName(KernelContext context, string name)
        {
            if (!context.AutoObjectNames.TryRemove(name, out _))
            {
                return KernelResult.NotFound;
            }

            return Result.Success;
        }

        public static KAutoObject FindNamedObject(KernelContext context, string name)
        {
            if (context.AutoObjectNames.TryGetValue(name, out KAutoObject obj))
            {
                return obj;
            }

            return null;
        }

        public void IncrementReferenceCount()
        {
            int newRefCount = Interlocked.Increment(ref _referenceCount);

            Debug.Assert(newRefCount >= 2);
        }

        public void DecrementReferenceCount()
        {
            int newRefCount = Interlocked.Decrement(ref _referenceCount);

            Debug.Assert(newRefCount >= 0);

            if (newRefCount == 0)
            {
                Destroy();
            }
        }

        protected virtual void Destroy()
        {
        }
    }
}