MultiTool/IAT_VERSIONINFO.cpp at master · ShadowKnightMK4/MultiTool · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#include "IAT_DLLS.H"
#include "common.h"
#include "IAT_VERSIONINFO.H"

extern "C"
{
	GetVersionInfOExA_Ptr IAT_GetVersionInfoExA = 0;
	RtlGetVersion_Ptr IAT_RtlGetVersion = 0;

	DWORD IAT_DynamicLinkVersionInfo_Cleanup()
	{
		IAT_GetVersionInfoExA = 0;
		IAT_RtlGetVersion = 0;
		return 1;
	}
	 DWORD IAT_DynamicLinkVersionInfo(DWORD IAT_settings)
	{
		 DWORD ret = 0;
		 if (iatKernel32 == 0)
		 {
			 iatKernel32 = LoadLibraryA("kernel32.dll");
		 }
		 if ((IAT_settings & (IAT_VERSIONINFO_NTDLL_RTLVERSIONINFO)) == (IAT_VERSIONINFO_NTDLL_RTLVERSIONINFO))
		 {
			 if (iatNTDLL == 0)
			 {
				 iatNTDLL = LoadLibraryA("ntdll.dll");
			 }
		 }

		 if (iatKernel32 == 0)
		 {
			 return 0;
		 }
		 if ((IAT_settings & (IAT_VERSIONINFO_GETVERSIONEXA)) == (IAT_VERSIONINFO_GETVERSIONEXA))
		 {
			 IAT_GetVersionInfoExA = (GetVersionInfOExA_Ptr)GetProcAddress(iatKernel32, "GetVersionExA");
			 if (IAT_GetVersionInfoExA != 0)
			 {
				 ret |= IAT_VERSIONINFO_GETVERSIONEXA;
			 }
		 }
		 if ((IAT_settings & (IAT_VERSIONINFO_NTDLL_RTLVERSIONINFO)) == (IAT_VERSIONINFO_NTDLL_RTLVERSIONINFO))
		 {
			 IAT_RtlGetVersion = (RtlGetVersion_Ptr)GetProcAddress(iatNTDLL, "RtlGetVersion");
			 if (IAT_RtlGetVersion != 0)
			 {
				 ret |= IAT_VERSIONINFO_NTDLL_RTLVERSIONINFO;
			 }
		 }
		 else
		 {
			 IAT_RtlGetVersion = 0;
		 }

		 return ret;
	}
}