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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
#include "precomp.h"
// Exit macros
#define AppExitOnLastError(x, s, ...) ExitOnLastErrorSource(DUTIL_SOURCE_APPUTIL, x, s, __VA_ARGS__)
#define AppExitOnLastErrorDebugTrace(x, s, ...) ExitOnLastErrorDebugTraceSource(DUTIL_SOURCE_APPUTIL, x, s, __VA_ARGS__)
#define AppExitWithLastError(x, s, ...) ExitWithLastErrorSource(DUTIL_SOURCE_APPUTIL, x, s, __VA_ARGS__)
#define AppExitOnFailure(x, s, ...) ExitOnFailureSource(DUTIL_SOURCE_APPUTIL, x, s, __VA_ARGS__)
#define AppExitOnRootFailure(x, s, ...) ExitOnRootFailureSource(DUTIL_SOURCE_APPUTIL, x, s, __VA_ARGS__)
#define AppExitOnFailureDebugTrace(x, s, ...) ExitOnFailureDebugTraceSource(DUTIL_SOURCE_APPUTIL, x, s, __VA_ARGS__)
#define AppExitOnNull(p, x, e, s, ...) ExitOnNullSource(DUTIL_SOURCE_APPUTIL, p, x, e, s, __VA_ARGS__)
#define AppExitOnNullWithLastError(p, x, s, ...) ExitOnNullWithLastErrorSource(DUTIL_SOURCE_APPUTIL, p, x, s, __VA_ARGS__)
#define AppExitOnNullDebugTrace(p, x, e, s, ...) ExitOnNullDebugTraceSource(DUTIL_SOURCE_APPUTIL, p, x, e, s, __VA_ARGS__)
#define AppExitOnInvalidHandleWithLastError(p, x, s, ...) ExitOnInvalidHandleWithLastErrorSource(DUTIL_SOURCE_APPUTIL, p, x, s, __VA_ARGS__)
#define AppExitOnWin32Error(e, x, s, ...) ExitOnWin32ErrorSource(DUTIL_SOURCE_APPUTIL, e, x, s, __VA_ARGS__)
#define AppExitOnGdipFailure(g, x, s, ...) ExitOnGdipFailureSource(DUTIL_SOURCE_APPUTIL, g, x, s, __VA_ARGS__)
const DWORD PRIVATE_LOAD_LIBRARY_SEARCH_SYSTEM32 = 0x00000800;
typedef BOOL(WINAPI *LPFN_SETDEFAULTDLLDIRECTORIES)(DWORD);
typedef BOOL(WINAPI *LPFN_SETDLLDIRECTORYW)(LPCWSTR);
extern "C" void DAPI AppFreeCommandLineArgs(
__in LPWSTR* argv
)
{
// The "ignored" hack in AppParseCommandLine requires an adjustment.
LPWSTR* argvOriginal = argv - 1;
::LocalFree(argvOriginal);
}
/********************************************************************
AppInitialize - initializes the standard safety precautions for an
installation application.
********************************************************************/
extern "C" void DAPI AppInitialize(
__in_ecount(cSafelyLoadSystemDlls) LPCWSTR rgsczSafelyLoadSystemDlls[],
__in DWORD cSafelyLoadSystemDlls
)
{
HRESULT hr = S_OK;
HMODULE hIgnored = NULL;
BOOL fSetDefaultDllDirectories = FALSE;
::HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
// Best effort call to initialize default DLL directories to system only.
HMODULE hKernel32 = ::GetModuleHandleW(L"kernel32");
Assert(hKernel32);
LPFN_SETDEFAULTDLLDIRECTORIES pfnSetDefaultDllDirectories = (LPFN_SETDEFAULTDLLDIRECTORIES)::GetProcAddress(hKernel32, "SetDefaultDllDirectories");
if (pfnSetDefaultDllDirectories)
{
if (pfnSetDefaultDllDirectories(PRIVATE_LOAD_LIBRARY_SEARCH_SYSTEM32))
{
fSetDefaultDllDirectories = TRUE;
}
else
{
hr = HRESULT_FROM_WIN32(::GetLastError());
TraceError(hr, "Failed to call SetDefaultDllDirectories.");
}
}
// Only need to safely load if the default DLL directories was not
// able to be set.
if (!fSetDefaultDllDirectories)
{
// Remove current working directory from search order.
LPFN_SETDLLDIRECTORYW pfnSetDllDirectory = (LPFN_SETDLLDIRECTORYW)::GetProcAddress(hKernel32, "SetDllDirectoryW");
if (!pfnSetDllDirectory || !pfnSetDllDirectory(L""))
{
hr = HRESULT_FROM_WIN32(::GetLastError());
TraceError(hr, "Failed to call SetDllDirectory.");
}
for (DWORD i = 0; i < cSafelyLoadSystemDlls; ++i)
{
hr = LoadSystemLibrary(rgsczSafelyLoadSystemDlls[i], &hIgnored);
if (FAILED(hr))
{
TraceError(hr, "Failed to safety load: %ls", rgsczSafelyLoadSystemDlls[i]);
}
}
}
}
extern "C" void DAPI AppInitializeUnsafe()
{
::HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
}
extern "C" DAPI_(HRESULT) AppParseCommandLine(
__in LPCWSTR wzCommandLine,
__in int* pArgc,
__in LPWSTR** pArgv
)
{
HRESULT hr = S_OK;
LPWSTR sczCommandLine = NULL;
LPWSTR* argv = NULL;
int argc = 0;
// CommandLineToArgvW tries to treat the first argument as the path to the process,
// which fails pretty miserably if your first argument is something like
// FOO="C:\Program Files\My Company". So give it something harmless to play with.
hr = StrAllocConcat(&sczCommandLine, L"ignored ", 0);
AppExitOnFailure(hr, "Failed to initialize command line.");
hr = StrAllocConcat(&sczCommandLine, wzCommandLine, 0);
AppExitOnFailure(hr, "Failed to copy command line.");
argv = ::CommandLineToArgvW(sczCommandLine, &argc);
AppExitOnNullWithLastError(argv, hr, "Failed to parse command line.");
// Skip "ignored" argument/hack.
*pArgv = argv + 1;
*pArgc = argc - 1;
LExit:
ReleaseStr(sczCommandLine);
return hr;
}
|
