Unity：通过当前进程名称查找句柄

快来打我* 2024-04-20 12:15 74阅读 0赞

using System;
    using System.Collections;
    using System.Collections.Generic;
    using System.Linq;
    using System.Runtime.InteropServices;
    using System.Text;
    using System.Threading.Tasks;
    
    namespace WindowsHandler
    {
        
        public class User32API
        {
        
    
            private static Hashtable processWnd = null;
    
            static User32API()
            {
        
                if (processWnd == null)
                {
        
                    processWnd = new Hashtable();
                }
            }
    
            [StructLayout(LayoutKind.Sequential)]
            public struct ProcessEntry32
            {
        
                public uint dwSize;
                public uint cntUsage;
                public uint th32ProcessID;
                public IntPtr th32DefaultHeapID;
                public uint th32ModuleID;
                public uint cntThreads;
                public uint th32ParentProcessID;
                public int pcPriClassBase;
                public uint dwFlags;
    
    
                [MarshalAs(UnmanagedType.ByValTStr, SizeConst = 260)]
                public string szExeFile;
            }
    
            //得到窗口客户区大小
            [DllImport("user32.dll")]
            public static extern int GetClientRect(IntPtr hwnd, out Rect lpRect);
    
    
            //得到窗口大小和坐标
            [DllImport("user32.dll")]
            public static extern int GetWindowRect(IntPtr hwnd, out Rect lpRect);
    
            /// <summary>
            /// 一个窗口区域的结构体
            /// </summary>
            public struct Rect
            {
        
                public int Left;
                public int Top;
                public int Right;
                public int Bottom;
            }
    
            public delegate bool WNDENUMPROC(IntPtr hwnd, uint lParam);
    
    
            [DllImport("user32.dll")]
            public static extern long GetWindowThreadProcessId(IntPtr hWnd, ref uint lpdwProcessId);
    
            [DllImport("KERNEL32.DLL ")]
            public static extern IntPtr CreateToolhelp32Snapshot(uint flags, uint processid);
            [DllImport("KERNEL32.DLL ")]
            public static extern int CloseHandle(IntPtr handle);
            [DllImport("KERNEL32.DLL ")]
            public static extern int Process32First(IntPtr handle, ref ProcessEntry32 pe);
            [DllImport("KERNEL32.DLL ")]
            public static extern int Process32Next(IntPtr handle, ref ProcessEntry32 pe);
    
            [DllImport("User32.dll", EntryPoint = "SendMessage")]
            private static extern int SendMessage(int hWnd, int Msg, int wParam, string lParam);
    
            [DllImport("user32", EntryPoint = "IsWindow")]
            private static extern bool IsWindow(IntPtr hWnd);
    
            [DllImport("user32.dll")]
            public static extern int SetParent(IntPtr hWndChild, IntPtr hWndParent);
    
            [DllImport("user32.dll", ExactSpelling = true, CharSet = CharSet.Auto)]
            public static extern IntPtr GetParent(IntPtr hWnd);
    
            [DllImport("user32.dll", EntryPoint = "SetWindowPos")]
            public static extern bool SetWindowPos(IntPtr hWnd, int hWndInsertAfter,
                                            int X, int Y, int cx, int cy, uint uFlags);
    
            [DllImport("kernel32.dll", EntryPoint = "SetLastError")]
            public static extern void SetLastError(uint dwErrCode);
    
            [DllImport("user32.dll", EntryPoint = "EnumWindows", SetLastError = true)]
            public static extern bool EnumWindows(WNDENUMPROC lpEnumFunc, uint lParam);
    
            [DllImport("user32.dll", EntryPoint = "FindWindow")]
            public static extern IntPtr FindWindow(string IpClassName, string IpWindowName);
    
            public static IntPtr GetHandleByProcessName(string ProcessName)
            {
        
                List<ProcessEntry32> list = new List<ProcessEntry32>();
                IntPtr handle = CreateToolhelp32Snapshot(0x2, 0);
                IntPtr hh = IntPtr.Zero;
                if ((int)handle > 0)
                {
        
                    ProcessEntry32 pe32 = new ProcessEntry32();
                    pe32.dwSize = (uint)Marshal.SizeOf(pe32);
                    int bMore = Process32First(handle, ref pe32);
                    while (bMore == 1)
                    {
        
                        IntPtr temp = Marshal.AllocHGlobal((int)pe32.dwSize);
                        Marshal.StructureToPtr(pe32, temp, true);
                        ProcessEntry32 pe = (ProcessEntry32)Marshal.PtrToStructure(temp, typeof(ProcessEntry32));
                        Marshal.FreeHGlobal(temp);
                        list.Add(pe);
                        if (pe.szExeFile == ProcessName)
                        {
        
                            bMore = 2;
                            hh = GetCurrentWindowHandle(pe.th32ProcessID);
                            break;
                        }
                        bMore = Process32Next(handle, ref pe32);
                    }
                }
                return hh;
            }
    
            public static IntPtr GetCurrentWindowHandle(uint proid)
            {
        
                IntPtr ptrWnd = IntPtr.Zero;
                uint uiPid = proid;
                object objWnd = processWnd[uiPid];
                if (objWnd != null)
                {
        
                    ptrWnd = (IntPtr)objWnd;
                    if (ptrWnd != IntPtr.Zero && IsWindow(ptrWnd))  // 从缓存中获取句柄
                    {
        
                        return ptrWnd;
                    }
                    else
                    {
        
                        ptrWnd = IntPtr.Zero;
                    }
                }
                bool bResult = EnumWindows(new WNDENUMPROC(EnumWindowsProc), uiPid);
                // 枚举窗口返回 false 并且没有错误号时表明获取成功
                if (!bResult && Marshal.GetLastWin32Error() == 0)
                {
        
                    objWnd = processWnd[uiPid];
                    if (objWnd != null)
                    {
        
                        ptrWnd = (IntPtr)objWnd;
                    }
                }
                return ptrWnd;
            }
    
    
            private static bool EnumWindowsProc(IntPtr hwnd, uint lParam)
            {
        
                uint uiPid = 0;
                if (GetParent(hwnd) == IntPtr.Zero)
                {
        
                    GetWindowThreadProcessId(hwnd, ref uiPid);
                    if (uiPid == lParam)    // 找到进程对应的主窗口句柄
                    {
        
                        processWnd.Add(uiPid, hwnd);   // 把句柄缓存起来
                        SetLastError(0);    // 设置无错误
                        return false;   // 返回 false 以终止枚举窗口
                    }
                }
                return true;
            }
        }
    }