Dalamud/Dalamud/Hooking/Internal/ReloadedHook.cs

using System.Reflection;

using Reloaded.Hooks;

namespace Dalamud.Hooking.Internal;

/// <summary>
/// Class facilitating hooks via reloaded.
/// </summary>
/// <typeparam name="T">Delegate of the hook.</typeparam>
internal class ReloadedHook<T> : Hook<T> where T : Delegate
{
    private readonly Reloaded.Hooks.Definitions.IHook<T> hookImpl;

    /// <summary>
    /// Initializes a new instance of the <see cref="ReloadedHook{T}"/> class.
    /// </summary>
    /// <param name="address">A memory address to install a hook.</param>
    /// <param name="detour">Callback function. Delegate must have a same original function prototype.</param>
    /// <param name="callingAssembly">Calling assembly.</param>
    internal ReloadedHook(IntPtr address, T detour, Assembly callingAssembly)
        : base(address)
    {
        lock (HookManager.HookEnableSyncRoot)
        {
            var unhooker = HookManager.RegisterUnhooker(address);
            this.hookImpl = ReloadedHooks.Instance.CreateHook<T>(detour, address.ToInt64());
            this.hookImpl.Activate();
            this.hookImpl.Disable();

            unhooker.TrimAfterHook();

            HookManager.TrackedHooks.TryAdd(this.HookId, new HookInfo(this, detour, callingAssembly));
        }
    }

    /// <inheritdoc/>
    public override T Original
    {
        get
        {
            this.CheckDisposed();
            return this.hookImpl.OriginalFunction;
        }
    }

    /// <inheritdoc/>
    public override bool IsEnabled => !this.IsDisposed && this.hookImpl.IsHookEnabled;

    /// <inheritdoc/>
    public override string BackendName => "Reloaded";

    /// <inheritdoc/>
    public override void Dispose()
    {
        if (this.IsDisposed)
            return;

        HookManager.TrackedHooks.TryRemove(this.HookId, out _);

        this.Disable();

        base.Dispose();
    }

    /// <inheritdoc/>
    public override void Enable()
    {
        lock (HookManager.HookEnableSyncRoot)
        {
            this.CheckDisposed();

            if (!this.hookImpl.IsHookEnabled)
                this.hookImpl.Enable();
        }
    }

    /// <inheritdoc/>
    public override void Disable()
    {
        lock (HookManager.HookEnableSyncRoot)
        {
            if (this.IsDisposed)
                return;

            if (!this.hookImpl.IsHookActivated)
                return;

            if (this.hookImpl.IsHookEnabled)
                this.hookImpl.Disable();
        }
    }
}