Added new columns: DPI Awareness, company, and description.

Author: trypsynth

CMakeLists.txt

@@ -8,4 +8,4 @@ if(MSVC)
 	target_compile_options(taskmon PRIVATE /GS- /W4 /Os /Oy)
 	target_link_options(taskmon PRIVATE /NODEFAULTLIB /ENTRY:WinMainCRTStartup /SUBSYSTEM:WINDOWS /OPT:REF /OPT:ICF /MERGE:.rdata=.text /MERGE:.pdata=.text)
 endif()
-target_link_libraries(taskmon PRIVATE kernel32 user32 gdi32 shell32 comctl32 ole32 shlwapi ntdll advapi32 dwmapi uxtheme comdlg32)
+target_link_libraries(taskmon PRIVATE kernel32 user32 gdi32 shell32 comctl32 ole32 shlwapi ntdll advapi32 dwmapi uxtheme comdlg32 version)

src/listview.c

@@ -153,6 +153,23 @@ static void format_column(const process_entry* e, column_id cid, wchar_t* buf, i
 			lstrcat(buf, L"/s");
 		} else buf[0] = L'\0';
 		break;
+	case COL_DESCRIPTION:
+		lstrcpyn(buf, e->description, len);
+		break;
+	case COL_COMPANY:
+		lstrcpyn(buf, e->company, len);
+		break;
+	case COL_DPI: {
+		const wchar_t* label;
+		switch (e->dpi_awareness) {
+		case TM_DPI_UNAWARE:            label = L"Unaware"; break;
+		case TM_DPI_SYSTEM_AWARE:       label = L"System"; break;
+		case TM_DPI_PER_MONITOR_AWARE:  label = L"Per-Monitor"; break;
+		default:                        label = L"Unknown"; break;
+		}
+		lstrcpyn(buf, label, len);
+		break;
+	}
 	default:
 		buf[0] = L'\0';
 		break;

src/process.c

@@ -200,6 +200,15 @@ static int compare_entries(const process_entry* a, const process_entry* b, sort_
 	case SORT_FIELD_IO_OTHER:
 		res = (a->io_other_rate < b->io_other_rate) ? -1 : (a->io_other_rate > b->io_other_rate);
 		break;
+	case SORT_FIELD_DESCRIPTION:
+		res = StrCmpI(a->description, b->description);
+		break;
+	case SORT_FIELD_COMPANY:
+		res = StrCmpI(a->company, b->company);
+		break;
+	case SORT_FIELD_DPI:
+		res = (int)a->dpi_awareness - (int)b->dpi_awareness;
+		break;
 	default:
 		break;
 	}
@@ -361,6 +370,53 @@ static void get_process_cmdline(DWORD pid, wchar_t* buf, int len) {
 	CloseHandle(h);
 }
 
+static void get_process_version_info(DWORD pid, wchar_t* desc, int desc_len, wchar_t* company, int comp_len) {
+	desc[0] = L'\0';
+	company[0] = L'\0';
+	wchar_t path[MAX_PATH];
+	get_process_path(pid, path, MAX_PATH);
+	if (!path[0]) return;
+	DWORD dummy;
+	DWORD size = GetFileVersionInfoSizeW(path, &dummy);
+	if (size == 0) return;
+	void* data = heap_alloc(size);
+	if (data) {
+		if (GetFileVersionInfoW(path, 0, size, data)) {
+			struct { USHORT lang; USHORT codepage; } *translate;
+			UINT tlen;
+			if (VerQueryValueW(data, L"\\VarFileInfo\\Translation", (LPVOID*)&translate, &tlen) && tlen >= sizeof(*translate)) {
+				wchar_t subblock[64];
+				wchar_t* value;
+				UINT vlen;
+				wnsprintf(subblock, 64, L"\\StringFileInfo\\%04x%04x\\FileDescription", translate[0].lang, translate[0].codepage);
+				if (VerQueryValueW(data, subblock, (LPVOID*)&value, &vlen)) lstrcpyn(desc, value, desc_len);
+				wnsprintf(subblock, 64, L"\\StringFileInfo\\%04x%04x\\CompanyName", translate[0].lang, translate[0].codepage);
+				if (VerQueryValueW(data, subblock, (LPVOID*)&value, &vlen)) lstrcpyn(company, value, comp_len);
+			}
+		}
+		heap_free(data);
+	}
+}
+
+static tm_dpi_awareness get_process_dpi_awareness(DWORD pid) {
+	typedef HRESULT (WINAPI *PFN_GPDA)(HANDLE, int*);
+	static PFN_GPDA fn = NULL;
+	static BOOL checked = FALSE;
+	if (!checked) {
+		HMODULE h = GetModuleHandle(L"shcore.dll");
+		if (!h) h = LoadLibrary(L"shcore.dll");
+		if (h) fn = (PFN_GPDA)GetProcAddress(h, "GetProcessDpiAwareness");
+		checked = TRUE;
+	}
+	if (!fn) return TM_DPI_UNAWARE;
+	HANDLE h = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pid);
+	if (!h) return TM_DPI_UNAWARE;
+	int awareness = 0;
+	fn(h, &awareness);
+	CloseHandle(h);
+	return (tm_dpi_awareness)awareness;
+}
+
 process_entry* snapshot_processes(snapshot_entry* snapshots, int* out_count, sort_field field, BOOL descending) {
 	FILETIME sys_idle_ft, sys_kernel_ft, sys_user_ft;
 	GetSystemTimes(&sys_idle_ft, &sys_kernel_ft, &sys_user_ft);
@@ -413,6 +469,8 @@ process_entry* snapshot_processes(snapshot_entry* snapshots, int* out_count, sor
 		e->integrity_level = get_process_integrity(pid);
 		get_process_user(pid, e->user, 64);
 		get_process_cmdline(pid, e->cmdline, 256);
+		get_process_version_info(pid, e->description, 128, e->company, 128);
+		e->dpi_awareness = get_process_dpi_awareness(pid);
 		e->arch_machine = get_process_arch(pid);
 		if (pid == 0) {
 			lstrcpy(e->name, L"System Idle Process");

src/process.h

@@ -29,13 +29,25 @@ typedef enum {
 	SORT_FIELD_IO_READ,
 	SORT_FIELD_IO_WRITE,
 	SORT_FIELD_IO_OTHER,
+	SORT_FIELD_DESCRIPTION,
+	SORT_FIELD_COMPANY,
+	SORT_FIELD_DPI,
 	SORT_FIELD_COUNT,
 } sort_field;
 
+typedef enum {
+	TM_DPI_UNAWARE = 0,
+	TM_DPI_SYSTEM_AWARE = 1,
+	TM_DPI_PER_MONITOR_AWARE = 2
+} tm_dpi_awareness;
+
 typedef struct {
 	DWORD pid;
 	DWORD parent_pid;
 	wchar_t name[64];
+	wchar_t description[128];
+	wchar_t company[128];
+	tm_dpi_awareness dpi_awareness;
 	double cpu_percent;
 	SIZE_T working_set;
 	SIZE_T private_working_set;

src/settings.c

@@ -18,21 +18,24 @@ const column_def COLUMNS[COL_COUNT] = {
 	{ L"Private Bytes", L"Private Bytes", 120, SORT_FIELD_PRIVATE_BYTES, FALSE, FALSE },
 	{ L"Page Faults",  L"Page Faults",  100, SORT_FIELD_PAGE_FAULTS,   FALSE, FALSE },
 	{ L"User",         L"User",         120, SORT_FIELD_USER,          FALSE, FALSE },
-	{ L"Command Line", L"Command Line", 300, SORT_FIELD_CMDLINE,       FALSE, FALSE },
+	{ L"Command Line", L"Command Line", 500, SORT_FIELD_CMDLINE,       FALSE, FALSE },
 	{ L"Architecture",    L"Architecture",  70,  SORT_FIELD_ARCH,           FALSE, FALSE },
 	{ L"Session",        L"Session",      60,  SORT_FIELD_SESSION,          FALSE, FALSE },
 	{ L"Peak Memory",    L"Peak Memory",  120, SORT_FIELD_PEAK_WORKING_SET,  FALSE, FALSE },
-	{ L"Virtual Memory", L"Virtual Mem",  120, SORT_FIELD_VIRTUAL_MEM,       FALSE, FALSE },
-	{ L"GDI Objects",    L"GDI Obj",      70,  SORT_FIELD_GDI_OBJECTS,       FALSE, FALSE },
-	{ L"USER Objects",   L"USER Obj",     70,  SORT_FIELD_USER_OBJECTS,      FALSE, FALSE },
+	{ L"Virtual Memory", L"Virtual Memory",  120, SORT_FIELD_VIRTUAL_MEM,       FALSE, FALSE },
+	{ L"GDI Objects",    L"GDI Objects",      70,  SORT_FIELD_GDI_OBJECTS,       FALSE, FALSE },
+	{ L"USER Objects",   L"USER Objects",     70,  SORT_FIELD_USER_OBJECTS,      FALSE, FALSE },
 	{ L"Integrity",      L"Integrity",    80,  SORT_FIELD_INTEGRITY,         FALSE, FALSE },
-	{ L"Parent PID",     L"PPID",         80,  SORT_FIELD_PPID,              FALSE, FALSE },
-	{ L"Private WS",     L"Private WS",   100, SORT_FIELD_PRIVATE_WS,        FALSE, FALSE },
+	{ L"Parent PID",     L"Parent PID",         80,  SORT_FIELD_PPID,              FALSE, FALSE },
+	{ L"Private Working Set",     L"Private Working Set",   100, SORT_FIELD_PRIVATE_WS,        FALSE, FALSE },
 	{ L"Paged Pool",     L"Paged Pool",   100, SORT_FIELD_PAGED_POOL,        FALSE, FALSE },
-	{ L"Non-paged Pool", L"NP Pool",      100, SORT_FIELD_NONPAGED_POOL,     FALSE, FALSE },
+	{ L"Non-paged Pool", L"Non-paged Pool",      100, SORT_FIELD_NONPAGED_POOL,     FALSE, FALSE },
 	{ L"I/O Read",       L"I/O Read",     100, SORT_FIELD_IO_READ,           FALSE, FALSE },
 	{ L"I/O Write",      L"I/O Write",    100, SORT_FIELD_IO_WRITE,          FALSE, FALSE },
 	{ L"I/O Other",      L"I/O Other",    100, SORT_FIELD_IO_OTHER,          FALSE, FALSE },
+	{ L"Description",    L"Description",  200, SORT_FIELD_DESCRIPTION,       FALSE, FALSE },
+	{ L"Company",        L"Company",      150, SORT_FIELD_COMPANY,           FALSE, FALSE },
+	{ L"DPI Awareness",  L"DPI Awareness",           90, SORT_FIELD_DPI,               FALSE, FALSE },
 };
 
 const UINT REFRESH_MS[REFRESH_OPTION_COUNT] = { 0, 5000, 10000, 30000, 60000 };

src/settings.h

@@ -29,6 +29,9 @@ typedef enum {
 	COL_IO_READ,
 	COL_IO_WRITE,
 	COL_IO_OTHER,
+	COL_DESCRIPTION,
+	COL_COMPANY,
+	COL_DPI,
 	COL_COUNT,
 } column_id;