github/tai

公开

Tai · Windows 软件与网站使用时长统计。上游 Planshit/Tai 的源码镜像，非灵栈原创；MIT 许可。来源：https://github.com/Planshit/Tai 。本次为手动同步，暂不提供安装包。

优化uwp进程信息获取

作者: noberumotto
时间: 2021/12/29 07:03
提交: a3d2cf2
上一个: 6a2bcca

修改Core/Servicers/Instances/Observer.cs+88 −39看文件

@@ -8,6 +8,7 @@ using System.IO;
8	8	using System.Linq;
9	9	using System.Runtime.InteropServices;
10	10	using System.Text;
	11	新增：using System.Threading;
11	12	using System.Threading.Tasks;
12	13
13	14	namespace Core.Servicers.Instances
@@ -32,60 +33,108 @@ namespace Core.Servicers.Instances
32	33	winEventDelegate, 0, 0, 0);
33	34	}
34	35
35		删除： private void WinEventProc(IntPtr hWinEventHook, uint eventType, IntPtr hwnd, int idObject, int idChild, uint dwEventThread, uint dwmsEventTime)
	36	新增： private async void WinEventProc(IntPtr hWinEventHook, uint eventType, IntPtr hwnd, int idObject, int idChild, uint dwEventThread, uint dwmsEventTime)
36	37	{
37	38	string processName = String.Empty, processFileName = String.Empty, processDescription = String.Empty;
	39	新增： var processResult = await Task.Run(() =>
	40	新增： {
	41	新增： try
	42	新增： {
	43	新增： string[] processInfo = GetProcessInfoByWindowHandle(hwnd);
	44	新增： int processID = int.Parse(processInfo[2]);
	45	新增： processName = processInfo[0];
	46	新增： processFileName = processInfo[1];
	47	新增：
	48	新增：
	49	新增： // handle uwp app
	50	新增： if (processFileName.IndexOf("ApplicationFrameHost.exe") != -1)
	51	新增： {
	52	新增： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
	53	新增： if (processFileName == null)
	54	新增： {
	55	新增： Thread.Sleep(500);
	56	新增：
	57	新增： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
	58	新增： processName = GetUWPAPPName(processFileName);
	59	新增：
	60	新增： IntPtr foregeroundHwnd = Win32API.GetForegroundWindow();
	61	新增：
	62	新增： if (foregeroundHwnd != hwnd)
	63	新增： {
	64	新增： // 获取后焦点变了
	65	新增： processInfo = GetProcessInfoByWindowHandle(foregeroundHwnd);
	66	新增： processID = int.Parse(processInfo[2]);
	67	新增： processName = processInfo[0];
	68	新增： processFileName = processInfo[1];
	69	新增： }
	70	新增：
	71	新增： }
	72	新增： else
	73	新增： {
	74	新增： processName = GetUWPAPPName(processFileName);
	75	新增： }
	76	新增：
	77	新增： }
	78	新增： Debug.WriteLine("file name: ---------> " + " \| " + hwnd + " \| " + processID + processFileName);
	79	新增： Debug.WriteLine("app name: ---------> " + processName);
	80	新增： FileVersionInfo info = FileVersionInfo.GetVersionInfo(processFileName);
	81	新增： processDescription = info.FileDescription;
	82	新增：
	83	新增：
	84	新增： //Debug.WriteLine("file name: ---------> " + processFileName);
	85	新增： //Debug.WriteLine("app name: ---------> " + processName);
	86	新增： //Debug.WriteLine("processDescription: ---------> " + processDescription);
	87	新增： }
	88	新增： catch (Exception e)
	89	新增： {
	90	新增： Logger.Warn(e.Message + " \| Process Name:" + processName + " \| Process File:" + processFileName + " \| Process Description:" + processDescription);
	91	新增： }
	92	新增： return processName;
	93	新增： });
38	94
39		删除： try
40		删除： {
41		删除： int processID = 0;
42		删除： Win32API.GetWindowThreadProcessId(hwnd, out processID);
43		删除： Process process = Process.GetProcessById(processID);
44		删除： processName = process.ProcessName;
45		删除：
46		删除： IntPtr processHandle = IntPtr.Zero;
47		删除： processHandle = Win32API.OpenProcess(0x001F0FFF, false, processID);
	95	新增： EventInvoke(processName, processDescription, processFileName);
	96	新增： }
48	97
49		删除： if (processName == "Snipaste")
50		删除： {
	98	新增： private string GetUWPAPPName(string processFileName)
	99	新增： {
	100	新增： string processName = string.Empty;
	101	新增： if (!string.IsNullOrEmpty(processFileName) && processFileName.IndexOf("\\") != -1)
	102	新增： {
	103	新增： processName = processFileName.Split('\\').Last();
	104	新增： processName = processName.Replace(".exe", "");
	105	新增： }
	106	新增： return processName;
	107	新增： }
	108	新增： /// <summary>
	109	新增： /// 通过窗口句柄获取进程信息
	110	新增： /// </summary>
	111	新增： /// <param name="hwnd"></param>
	112	新增： /// <returns></returns>
	113	新增： private string[] GetProcessInfoByWindowHandle(IntPtr hwnd)
	114	新增： {
	115	新增： string processName, processFileName = string.Empty;
51	116
52		删除： int error = Marshal.GetLastWin32Error();
	117	新增： int processID = 0;
	118	新增： Win32API.GetWindowThreadProcessId(hwnd, out processID);
	119	新增： Process process = Process.GetProcessById(processID);
	120	新增： processName = process.ProcessName;
53	121
54		删除： Debug.WriteLine("The last Win32 Error was: " + error);
	122	新增： IntPtr processHandle = IntPtr.Zero;
	123	新增： processHandle = Win32API.OpenProcess(0x001F0FFF, false, processID);
55	124
56		删除： }
	125	新增： if (processHandle != IntPtr.Zero)
	126	新增： {
57	127	int MaxPathLength = 4096;
58	128	var buffer = new StringBuilder(MaxPathLength);
59	129	Win32API.QueryFullProcessImageName(processHandle, 0, buffer, ref MaxPathLength);
60	130	processFileName = buffer.ToString();
61		删除：
62		删除： // handle uwp app
63		删除： if (processFileName.IndexOf("ApplicationFrameHost.exe") != -1)
64		删除： {
65		删除： processFileName = Win32API.UWP_AppName(hwnd, (uint)processID);
66		删除： if (processFileName != string.Empty && processFileName.IndexOf("\\") != -1)
67		删除： {
68		删除： processName = processFileName.Split('\\').Last();
69		删除： processName = processName.Replace(".exe", "");
70		删除： }
71		删除： }
72		删除：
73		删除： FileVersionInfo info = FileVersionInfo.GetVersionInfo(processFileName);
74		删除： processDescription = info.FileDescription;
75		删除：
76		删除：
77		删除： //Debug.WriteLine("file name: ---------> " + processFileName);
78		删除： //Debug.WriteLine("app name: ---------> " + processName);
79		删除： //Debug.WriteLine("processDescription: ---------> " + processDescription);
80	131	}
81		删除： catch (Exception e)
	132	新增： else
82	133	{
83		删除： Logger.Error(e.Message + " \| Process Name:" + processName + " \| Process File:" + processFileName + " \| Process Description:" + processDescription);
	134	新增： Logger.Warn($"无法获取进程：{processName}的句柄，请尝试以管理员身份运行Tai。");
84	135	}
85		删除：
86		删除： EventInvoke(processName, processDescription, processFileName);
	136	新增： return new string[] { processName, processFileName, processID.ToString() };
87	137	}
88		删除：
89	138	private void EventInvoke(string processName, string description, string filename)
90	139	{
91	140	// 防止重复和错误响应