JinxRyu/Ryujinx.Tests.Unicorn/Native/Interface.cs

using Ryujinx.Tests.Unicorn.Native.Const;
using System;
using System.Diagnostics.CodeAnalysis;
using System.IO;
using System.Reflection;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;

namespace Ryujinx.Tests.Unicorn.Native
{
    public static partial class Interface
    {
        public static bool IsUnicornAvailable { get; private set; } = true;

        private static IntPtr ImportResolver(string libraryName, Assembly assembly, DllImportSearchPath? searchPath)
        {
            if (libraryName == "unicorn")
            {
                string loadPath = $"{Path.GetDirectoryName(assembly.Location)}/";
                loadPath += OperatingSystem.IsWindows() ? $"{libraryName}.dll" : $"lib{libraryName}.so";

                if (!NativeLibrary.TryLoad(loadPath, out IntPtr libraryPtr))
                {
                    IsUnicornAvailable = false;
                    Console.Error.WriteLine($"ERROR: Could not find unicorn at: {loadPath}");
                }

                return libraryPtr;
            }

            // Otherwise, fallback to default import resolver.
            return IntPtr.Zero;
        }

        static Interface()
        {
            NativeLibrary.SetDllImportResolver(Assembly.GetExecutingAssembly(), ImportResolver);
        }

        public static void Checked(Error error)
        {
            if (error != Error.OK)
            {
                throw new UnicornException(error);
            }
        }

        public static void MarshalArrayOf<[DynamicallyAccessedMembers(DynamicallyAccessedMemberTypes.PublicConstructors | DynamicallyAccessedMemberTypes.NonPublicConstructors)] T>(IntPtr input, int length, out T[] output)
        {
            int size = Marshal.SizeOf<T>();

            output = new T[length];

            for (int i = 0; i < length; i++)
            {
                IntPtr item = new IntPtr(input.ToInt64() + i * size);

                output[i] = Marshal.PtrToStructure<T>(item);
            }
        }

        [LibraryImport("unicorn")]
        public static partial uint uc_version(out uint major, out uint minor);

        [LibraryImport("unicorn")]
        public static partial Error uc_open(Arch arch, Mode mode, out IntPtr uc);

        [LibraryImport("unicorn")]
        public static partial Error uc_close(IntPtr uc);

        [LibraryImport("unicorn")]
        public static partial IntPtr uc_strerror(Error err);

        [LibraryImport("unicorn")]
        public static partial Error uc_reg_write(IntPtr uc, int regid, byte[] value);

        [LibraryImport("unicorn")]
        public static partial Error uc_reg_read(IntPtr uc, int regid, byte[] value);

        [LibraryImport("unicorn")]
        public static partial Error uc_mem_write(IntPtr uc, ulong address, byte[] bytes, ulong size);

        [LibraryImport("unicorn")]
        public static partial Error uc_mem_read(IntPtr uc, ulong address, byte[] bytes, ulong size);

        [LibraryImport("unicorn")]
        public static partial Error uc_emu_start(IntPtr uc, ulong begin, ulong until, ulong timeout, ulong count);

        [LibraryImport("unicorn")]
        public static partial Error uc_mem_map(IntPtr uc, ulong address, ulong size, uint perms);

        [LibraryImport("unicorn")]
        public static partial Error uc_mem_unmap(IntPtr uc, ulong address, ulong size);

        [LibraryImport("unicorn")]
        public static partial Error uc_mem_protect(IntPtr uc, ulong address, ulong size, uint perms);

        [LibraryImport("unicorn")]
        public static partial Error uc_mem_regions(IntPtr uc, out IntPtr regions, out uint count);
    }
}