Added the MultiTree Control code.

BeneficialCode · Oct 12, 2024 · 085b7b4 · 085b7b4
1 parent 8f9043e
commit 085b7b4
Show file tree

Hide file tree

Showing 3 changed files with 545 additions and 7 deletions.
diff --git a/WinArk/ImportsHandling.cpp b/WinArk/ImportsHandling.cpp
@@ -1,4 +1,5 @@
 #include "stdafx.h"
+#include "resource.h"
 #include "ImportsHandling.h"
 #include "Architecture.h"
 
@@ -75,5 +76,20 @@ ImportsHandling::Icon ImportsHandling::GetAppropiateIcon(bool valid) {
 }
 
 ImportsHandling::ImportsHandling(CMultiSelectTreeViewCtrl& treeImports):TreeImports(treeImports) {
-
+	CDCHandle dc = CWindow(::GetDesktopWindow()).GetDC();
+	int bits = dc.GetDeviceCaps(BITSPIXEL);
+	const UINT flags = bits > 16 ? ILC_COLOR32 : (ILC_COLOR24 | ILC_MASK);
+	TreeIcons.Create(16, 16, flags, 3, 1);
+
+	UINT icons[] = {
+		IDI_CHECK,IDI_SCYLLA_WARNING,IDI_SCYLLA_ERROR
+	};
+
+	for (UINT icon : icons) {
+		TreeIcons.AddIcon(AtlLoadIconImage(icon, 0, 16, 16));
+	}
+}
+
+ImportsHandling::~ImportsHandling() {
+	TreeIcons.Destroy();
 }
diff --git a/WinArk/ImportsHandling.h b/WinArk/ImportsHandling.h
@@ -10,13 +10,13 @@ class ImportsHandling
 	std::map<DWORD_PTR, ImportModuleThunk> NewModuleMap;
 
 	ImportsHandling(CMultiSelectTreeViewCtrl& treeImports);
-
+	~ImportsHandling();
 private:
-	DWORD NumberOfFunctions;
+	DWORD NumberOfFunctions = 0;
 
-	unsigned int m_ThunkCount;
-	unsigned int m_InvalidThunkCount;
-	unsigned int m_SuspectThunkCount;
+	unsigned int m_ThunkCount = 0;
+	unsigned int m_InvalidThunkCount = 0;
+	unsigned int m_SuspectThunkCount = 0;
 
 	struct TreeItemData
 	{
@@ -34,7 +34,7 @@ class ImportsHandling
 
 	TreeItemData* GetItemData(CTreeItem item);
 
-	WCHAR Text[512];
+	WCHAR Text[512] = { 0 };
 
 	CMultiSelectTreeViewCtrl& TreeImports;
 	CImageList TreeIcons;