ce188d4d |
1 | /* DllSecur.c -- DLL loading security\r |
2 | 2016-10-04 : Igor Pavlov : Public domain */\r |
3 | \r |
4 | #include "Precomp.h"\r |
5 | \r |
6 | #ifdef _WIN32\r |
7 | \r |
8 | #include <windows.h>\r |
9 | \r |
10 | #include "DllSecur.h"\r |
11 | \r |
12 | #ifndef UNDER_CE\r |
13 | \r |
14 | typedef BOOL (WINAPI *Func_SetDefaultDllDirectories)(DWORD DirectoryFlags);\r |
15 | \r |
16 | #define MY_LOAD_LIBRARY_SEARCH_USER_DIRS 0x400\r |
17 | #define MY_LOAD_LIBRARY_SEARCH_SYSTEM32 0x800\r |
18 | \r |
19 | static const char * const g_Dlls =\r |
20 | #ifndef _CONSOLE\r |
21 | "UXTHEME\0"\r |
22 | #endif\r |
23 | "USERENV\0"\r |
24 | "SETUPAPI\0"\r |
25 | "APPHELP\0"\r |
26 | "PROPSYS\0"\r |
27 | "DWMAPI\0"\r |
28 | "CRYPTBASE\0"\r |
29 | "OLEACC\0"\r |
30 | "CLBCATQ\0"\r |
31 | ;\r |
32 | \r |
33 | #endif\r |
34 | \r |
35 | void LoadSecurityDlls()\r |
36 | {\r |
37 | #ifndef UNDER_CE\r |
38 | \r |
39 | wchar_t buf[MAX_PATH + 100];\r |
40 | \r |
41 | {\r |
42 | // at Vista (ver 6.0) : CoCreateInstance(CLSID_ShellLink, ...) doesn't work after SetDefaultDllDirectories() : Check it ???\r |
43 | OSVERSIONINFO vi;\r |
44 | vi.dwOSVersionInfoSize = sizeof(vi);\r |
45 | if (!GetVersionEx(&vi) || vi.dwMajorVersion != 6 || vi.dwMinorVersion != 0)\r |
46 | {\r |
47 | Func_SetDefaultDllDirectories setDllDirs = (Func_SetDefaultDllDirectories)\r |
48 | GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "SetDefaultDllDirectories");\r |
49 | if (setDllDirs)\r |
50 | if (setDllDirs(MY_LOAD_LIBRARY_SEARCH_SYSTEM32 | MY_LOAD_LIBRARY_SEARCH_USER_DIRS))\r |
51 | return;\r |
52 | }\r |
53 | }\r |
54 | \r |
55 | {\r |
56 | unsigned len = GetSystemDirectoryW(buf, MAX_PATH + 2);\r |
57 | if (len == 0 || len > MAX_PATH)\r |
58 | return;\r |
59 | }\r |
60 | {\r |
61 | const char *dll;\r |
62 | unsigned pos = (unsigned)lstrlenW(buf);\r |
63 | \r |
64 | if (buf[pos - 1] != '\\')\r |
65 | buf[pos++] = '\\';\r |
66 | \r |
67 | for (dll = g_Dlls; dll[0] != 0;)\r |
68 | {\r |
69 | unsigned k = 0;\r |
70 | for (;;)\r |
71 | {\r |
72 | char c = *dll++;\r |
73 | buf[pos + k] = c;\r |
74 | k++;\r |
75 | if (c == 0)\r |
76 | break;\r |
77 | }\r |
78 | \r |
79 | lstrcatW(buf, L".dll");\r |
80 | LoadLibraryExW(buf, NULL, LOAD_WITH_ALTERED_SEARCH_PATH);\r |
81 | }\r |
82 | }\r |
83 | \r |
84 | #endif\r |
85 | }\r |
86 | \r |
87 | #endif\r |