using System;
using System.Runtime.InteropServices;
using System.Text;

namespace ParsecVDisplay
{
    internal static unsafe class Device
    {
        public enum Status
        {
            OK,
            INACCESSIBLE,
            UNKNOW,
            UNKNOW_PROBLEM,
            DISABLED,
            DRIVER_ERROR,
            RESTART_REQUIRED,
            DISABLED_SERVICE,
            NOT_INSTALLED
        }

        public static Status QueryStatus(string guid, string devId)
        {
            var status = Status.INACCESSIBLE;

            var devInfoData = new Native.SP_DEVINFO_DATA();
            devInfoData.cbSize = sizeof(Native.SP_DEVINFO_DATA);

            var classGuid = Guid.Parse(guid);
            var devInfo = Native.SetupDiGetClassDevsA(ref classGuid, null, null, Native.DIGCF_PRESENT);

            if (devInfo != Native.INVALID_HANDLE_VALUE)
            {
                bool foundProp = false;
                uint deviceIndex = 0;

                do
                {
                    if (!Native.SetupDiEnumDeviceInfo(devInfo, deviceIndex, &devInfoData))
                        break;

                    int requiredSize = 0;
                    Native.SetupDiGetDeviceRegistryPropertyA(devInfo, &devInfoData,
                        Native.SPDRP_HARDWAREID, null, null, 0, &requiredSize);

                    if (requiredSize > 0)
                    {
                        uint regDataType = 0;
                        IntPtr propBuffer = Marshal.AllocHGlobal(requiredSize);

                        if (Native.SetupDiGetDeviceRegistryPropertyA(
                            devInfo,
                            &devInfoData,
                            Native.SPDRP_HARDWAREID,
                            &regDataType,
                            (void*)propBuffer,
                            requiredSize,
                            &requiredSize))
                        {
                            if (regDataType == Native.REG_SZ || regDataType == Native.REG_MULTI_SZ)
                            {
                                for (IntPtr cp = propBuffer; ; cp += Native.lstrlenA(cp) + 1)
                                {
                                    if (cp == (IntPtr)(0) || *(byte*)cp == 0 || (ulong)cp >= (ulong)(propBuffer + requiredSize))
                                    {
                                        status = Status.NOT_INSTALLED;
                                        goto except;
                                    }

                                    if (devId.Equals(Marshal.PtrToStringAnsi(cp)))
                                        break;
                                }

                                foundProp = true;
                                uint devStatus, devProblemNum;

                                if (Native.CM_Get_DevNode_Status(&devStatus, &devProblemNum, devInfoData.DevInst, 0) != Native.CR_SUCCESS)
                                {
                                    status = Status.NOT_INSTALLED;
                                    goto except;
                                }

                                if ((devStatus & (Native.DN_DRIVER_LOADED | Native.DN_STARTED)) != 0)
                                {
                                    status = Status.OK;
                                }
                                else if ((devStatus & Native.DN_HAS_PROBLEM) != 0)
                                {
                                    switch (devProblemNum)
                                    {
                                        case Native.CM_PROB_NEED_RESTART:
                                            status = Status.RESTART_REQUIRED;
                                            break;
                                        case Native.CM_PROB_DISABLED:
                                        case Native.CM_PROB_HARDWARE_DISABLED:
                                            status = Status.DISABLED;
                                            break;
                                        case Native.CM_PROB_DISABLED_SERVICE:
                                            status = Status.DISABLED_SERVICE;
                                            break;
                                        default:
                                            if (devProblemNum == Native.CM_PROB_FAILED_POST_START)
                                                status = Status.DRIVER_ERROR;
                                            else
                                                status = Status.UNKNOW_PROBLEM;
                                            break;
                                    }
                                }
                                else
                                {
                                    status = Status.UNKNOW;
                                }
                            }
                        }

                    except:
                        Marshal.FreeHGlobal(propBuffer);
                    }

                    ++deviceIndex;
                } while (!foundProp);

                if (!foundProp && Marshal.GetLastWin32Error() != 0)
                    status = Status.NOT_INSTALLED;

                Native.SetupDiDestroyDeviceInfoList(devInfo);
            }

            return status;
        }

        public static bool OpenHandle(string guid, out IntPtr handle)
        {
            handle = IntPtr.Zero;

            var interfaceGuid = Guid.Parse(guid);
            var devInfo = Native.SetupDiGetClassDevsA(ref interfaceGuid,
                null, null, Native.DIGCF_PRESENT | Native.DIGCF_DEVICEINTERFACE);

            if (devInfo != Native.INVALID_HANDLE_VALUE)
            {
                var devInterface = new Native.SP_DEVICE_INTERFACE_DATA();
                devInterface.cbSize = sizeof(Native.SP_DEVICE_INTERFACE_DATA);

                for (uint i = 0; Native.SetupDiEnumDeviceInterfaces(devInfo, null, ref interfaceGuid, i, &devInterface); ++i)
                {
                    int detailSize = 0;
                    Native.SetupDiGetDeviceInterfaceDetailA(devInfo, &devInterface, null, 0, &detailSize, null);

                    var detail = (Native.SP_DEVICE_INTERFACE_DETAIL_DATA_A*)Marshal.AllocHGlobal(detailSize);
                    detail->cbSize = sizeof(Native.SP_DEVICE_INTERFACE_DETAIL_DATA_A);

                    if (Native.SetupDiGetDeviceInterfaceDetailA(devInfo, &devInterface, detail, detailSize, &detailSize, null))
                    {
                        handle = Native.CreateFileA(&detail->DevicePath,
                            Native.GENERIC_READ | Native.GENERIC_WRITE,
                            Native.FILE_SHARE_READ | Native.FILE_SHARE_WRITE,
                            null,
                            Native.OPEN_EXISTING,
                            Native.FILE_ATTRIBUTE_NORMAL | Native.FILE_FLAG_NO_BUFFERING | Native.FILE_FLAG_OVERLAPPED | Native.FILE_FLAG_WRITE_THROUGH,
                            null);

                        if (handle != IntPtr.Zero && handle != Native.INVALID_HANDLE_VALUE)
                            break;
                    }

                    Marshal.FreeHGlobal((IntPtr)detail);
                }

                Native.SetupDiDestroyDeviceInfoList(devInfo);
            }

            return handle != IntPtr.Zero
                && handle != Native.INVALID_HANDLE_VALUE;
        }

        public static void CloseHandle(IntPtr handle)
        {
            if (handle != IntPtr.Zero && handle != Native.INVALID_HANDLE_VALUE)
            {
                Native.CloseHandle(handle);
            }
        }

        public static string GetDeviceDescription(uint devInst)
        {
            uint propType;
            int length = 128 * sizeof(ushort);
            var buffer = stackalloc byte[length];

            Native.CM_Get_DevNode_PropertyW(devInst,
                ref Native.DEVPROPKEY.Device_DeviceDesc, &propType, buffer, &length, 0);

            return Marshal.PtrToStringUni((IntPtr)buffer);
        }

        public static DateTime GetDeviceLastArrival(uint devInst)
        {
            uint propType;
            long lastArrival;
            int bufferLength = sizeof(long);

            Native.CM_Get_DevNode_PropertyW(devInst,
                ref Native.DEVPROPKEY.Device_LastArrivalDate, &propType, &lastArrival, &bufferLength, 0);

            return DateTime.FromFileTime(lastArrival);
        }

        public static bool GetParentDeviceInstance(uint devInst, out uint parentInst, out string parentId)
        {
            if (Native.CM_Get_Parent(out parentInst, devInst, 0) == 0)
            {
                byte[] idBuffer = new byte[Native.MAX_DEVICE_ID_LEN];
                Native.CM_Get_Device_IDA(parentInst, idBuffer, idBuffer.Length, 0);

                parentId = Encoding.ASCII.GetString(idBuffer).TrimEnd('\0');
                return true;
            }

            parentInst = 0;
            parentId = string.Empty;
            return false;
        }

        public static bool GetDeviceInstance(string deviceId, out uint devInst)
        {
            return Native.CM_Locate_DevNodeA(out devInst, deviceId, 0) == 0;
        }

        static class Native
        {
            public const int MAX_DEVICE_ID_LEN = 200;
            public static readonly IntPtr INVALID_HANDLE_VALUE = (IntPtr)(-1);

            public const uint GENERIC_READ = 0x80000000;
            public const uint GENERIC_WRITE = 0x40000000;

            public const uint FILE_SHARE_READ = 0x1;
            public const uint FILE_SHARE_WRITE = 0x2;

            public const uint OPEN_EXISTING = 3;

            public const uint FILE_ATTRIBUTE_NORMAL = 0x00000080;
            public const uint FILE_FLAG_NO_BUFFERING = 0x20000000;
            public const uint FILE_FLAG_OVERLAPPED = 0x40000000;
            public const uint FILE_FLAG_WRITE_THROUGH = 0x80000000;

            public const uint DIGCF_PRESENT = 0x2;
            public const uint DIGCF_DEVICEINTERFACE = 0x10;

            public const uint SPDRP_HARDWAREID = 0x1;

            public const uint REG_SZ = 1;
            public const uint REG_MULTI_SZ = 7;

            public const uint CR_SUCCESS = 0;

            public const uint CM_PROB_NEED_RESTART = 0x0000000E; // requires restart
            public const uint CM_PROB_DISABLED = 0x00000016; // devinst is disabled
            public const uint CM_PROB_HARDWARE_DISABLED = 0x0000001D; // device disabled
            public const uint CM_PROB_DISABLED_SERVICE = 0x00000020; // service's Start = 4
            public const uint CM_PROB_FAILED_POST_START = 0x0000002B; // The drivers set the device state to failed

            public const uint DN_DRIVER_LOADED = 0x00000002; // Has Register_Device_Driver
            public const uint DN_STARTED = 0x00000008; // Is currently configured
            public const uint DN_HAS_PROBLEM = 0x00000400; // Need device installer

            public static bool IsValidHandle(IntPtr handle)
            {
                return handle != IntPtr.Zero
                    && handle != (IntPtr)(-1);
            }

            [DllImport("kernel32.dll", CallingConvention = CallingConvention.Cdecl)]
            public static extern int lstrlenA(
                IntPtr lpString);

            [DllImport("kernel32.dll")]
            public static extern IntPtr CreateFileA(
                char* lpFileName,
                uint dwDesiredAccess,
                uint dwShareMode,
                void* lpSecurityAttributes,
                uint dwCreationDisposition,
                uint dwFlagsAndAttributes,
                void* hTemplateFile);

            [DllImport("kernel32.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool CloseHandle(IntPtr handle);

            [DllImport("kernel32.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool DeviceIoControl(
                IntPtr device, uint code,
                void* lpInBuffer, int nInBufferSize,
                void* lpOutBuffer, int nOutBufferSize,
                IntPtr lpBytesReturned,
                ref OVERLAPPED lpOverlapped
            );

            [DllImport("kernel32.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool GetOverlappedResult(
                IntPtr handle,
                ref OVERLAPPED lpOverlapped,
                out uint lpNumberOfBytesTransferred,
                [MarshalAs(UnmanagedType.Bool)] bool bWait
            );

            [StructLayout(LayoutKind.Sequential)]
            public struct OVERLAPPED
            {
                public IntPtr Internal;
                public IntPtr InternalHigh;
                public IntPtr Pointer;
                public IntPtr hEvent;
            }

            [StructLayout(LayoutKind.Sequential)]
            public struct RECT
            {
                public int left;
                public int top;
                public int right;
                public int bottom;
            }

            [StructLayout(LayoutKind.Sequential)]
            public struct SP_DEVICE_INTERFACE_DATA
            {
                public int cbSize;
                public Guid InterfaceClassGuid;
                public uint Flags;
                IntPtr _Reserved;
            }

            [StructLayout(LayoutKind.Sequential)]
            public struct SP_DEVINFO_DATA
            {
                public int cbSize;
                public Guid ClassGuid;
                public uint DevInst;
                IntPtr _Reserved;
            }

            [StructLayout(LayoutKind.Sequential)]
            public struct SP_DEVICE_INTERFACE_DETAIL_DATA_A
            {
                public int cbSize;
                public char DevicePath;
            }

            [DllImport("setupapi.dll")]
            public static extern IntPtr SetupDiGetClassDevsA(
                ref Guid ClassGuid,
                void* Enumerator,
                void* hwndParent,
                uint Flags);

            [DllImport("setupapi.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool SetupDiEnumDeviceInterfaces(
                IntPtr DeviceInfoSet,
                SP_DEVINFO_DATA* DeviceInfoData,
                ref Guid InterfaceClassGuid,
                uint MemberIndex,
                SP_DEVICE_INTERFACE_DATA* DeviceInterfaceData);

            [DllImport("setupapi.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool SetupDiGetDeviceInterfaceDetailA(
                IntPtr DeviceInfoSet,
                SP_DEVICE_INTERFACE_DATA* DeviceInterfaceData,
                void* DeviceInterfaceDetailData,
                int DeviceInterfaceDetailDataSize,
                int* RequiredSize,
                SP_DEVINFO_DATA* DeviceInfoData);

            [DllImport("setupapi.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool SetupDiDestroyDeviceInfoList(
                IntPtr DeviceInfoSet);

            [DllImport("setupapi.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool SetupDiEnumDeviceInfo(
                IntPtr DeviceInfoSet,
                uint MemberIndex,
                SP_DEVINFO_DATA* DeviceInfoData);

            [DllImport("setupapi.dll")]
            [return: MarshalAs(UnmanagedType.Bool)]
            public static extern bool SetupDiGetDeviceRegistryPropertyA(
                IntPtr DeviceInfoSet,
                SP_DEVINFO_DATA* DeviceInfoData,
                uint Property,
                uint* PropertyRegDataType,
                void* PropertyBuffer,
                int PropertyBufferSize,
                int* RequiredSize);

            [DllImport("setupapi.dll")]
            public static extern uint CM_Get_DevNode_Status(
                uint* pulStatus,
                uint* pulProblemNumber,
                uint dnDevInst,
                uint ulFlags);

            [DllImport("setupapi.dll")]
            public static extern uint CM_Get_Parent(out uint pdnDevInst, uint dnDevInst, uint ulFlags);

            [DllImport("setupapi.dll")]
            public static extern uint CM_Get_Device_IDA(uint dnDevInst, byte[] Buffer, int BufferLen, uint ulFlags);

            [StructLayout(LayoutKind.Sequential)]
            public struct DEVPROPKEY
            {
                public Guid fmtid;
                public uint pid;

                public static DEVPROPKEY Device_LastArrivalDate = new DEVPROPKEY
                {
                    fmtid = Guid.Parse("{83DA6326-97A6-4088-9453-A1923F573B29}"),
                    pid = 102,
                };

                public static DEVPROPKEY Device_DeviceDesc = new DEVPROPKEY
                {
                    fmtid = Guid.Parse("{A45C254E-DF1C-4EFD-8020-67D146A850E0}"),
                    pid = 2
                };
            }

            [DllImport("CfgMgr32.dll")]
            public static extern uint CM_Get_DevNode_PropertyW(
                uint dnDevInst,
                ref DEVPROPKEY PropertyKey,
                uint* PropertyType,
                void* PropertyBuffer,
                int* PropertyBufferSize,
                uint ulFlags);

            [DllImport("CfgMgr32.dll", CharSet = CharSet.Ansi)]
            public static extern uint CM_Locate_DevNodeA(out uint devInst, string deviceId, uint flags);
        }
    }
}