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
125
126
127
128
129
130
131
132
133
|
// 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"
#include "BalBaseBAFunctions.h"
#include "BalBaseBAFunctionsProc.h"
const DWORD VARIABLE_GROW_FACTOR = 80;
const LPCWSTR STRING_VARIABLE = L"AString";
const LPCWSTR NUMBER_VARIABLE = L"ANumber";
static void CALLBACK BafRelatedBundleVariableTestingTraceError(
__in_z LPCSTR szFile,
__in int iLine,
__in REPORT_LEVEL rl,
__in UINT source,
__in HRESULT hrError,
__in_z __format_string LPCSTR szFormat,
__in va_list args
);
class CBafRelatedBundleVariableTesting : public CBalBaseBAFunctions
{
public: // IBAFunctions
public: //IBootstrapperApplication
virtual STDMETHODIMP OnDetectRelatedBundle(
__in_z LPCWSTR wzBundleId,
__in BOOTSTRAPPER_RELATION_TYPE relationType,
__in_z LPCWSTR wzBundleTag,
__in BOOL fPerMachine,
__in LPCWSTR wzVersion,
__in BOOL fMissingFromCache,
__inout BOOL* pfCancel
)
{
HRESULT hr = S_OK;
LPWSTR wzValue = NULL;
hr = BalGetRelatedBundleVariable(wzBundleId, STRING_VARIABLE, &wzValue);
ExitOnFailure(hr, "Failed to get related bundle string variable.");
if (wzValue)
{
BalLog(BOOTSTRAPPER_LOG_LEVEL_STANDARD, "AString = %ws", wzValue);
}
hr = BalGetRelatedBundleVariable(wzBundleId, NUMBER_VARIABLE, &wzValue);
if (wzValue)
{
BalLog(BOOTSTRAPPER_LOG_LEVEL_STANDARD, "ANumber = %ws", wzValue);
}
hr = __super::OnDetectRelatedBundle(wzBundleId, relationType, wzBundleTag, fPerMachine, wzVersion, fMissingFromCache, pfCancel);
LExit:
ReleaseStr(wzValue);
return hr;
}
private:
public:
//
// Constructor - initialize member variables.
//
CBafRelatedBundleVariableTesting(
__in HMODULE hModule,
__in IBootstrapperEngine* pEngine,
__in const BA_FUNCTIONS_CREATE_ARGS* pArgs
) : CBalBaseBAFunctions(hModule, pEngine, pArgs)
{
}
//
// Destructor - release member variables.
//
~CBafRelatedBundleVariableTesting()
{
}
private:
};
HRESULT WINAPI CreateBAFunctions(
__in HMODULE hModule,
__in const BA_FUNCTIONS_CREATE_ARGS* pArgs,
__inout BA_FUNCTIONS_CREATE_RESULTS* pResults
)
{
HRESULT hr = S_OK;
CBafRelatedBundleVariableTesting* pBAFunctions = NULL;
IBootstrapperEngine* pEngine = NULL;
DutilInitialize(&BafRelatedBundleVariableTestingTraceError);
hr = BalInitializeFromCreateArgs(pArgs->pBootstrapperCreateArgs, &pEngine);
ExitOnFailure(hr, "Failed to initialize Bal.");
pBAFunctions = new CBafRelatedBundleVariableTesting(hModule, pEngine, pArgs);
ExitOnNull(pBAFunctions, hr, E_OUTOFMEMORY, "Failed to create new CBafRelatedBundleVariableTesting object.");
pResults->pfnBAFunctionsProc = BalBaseBAFunctionsProc;
pResults->pvBAFunctionsProcContext = pBAFunctions;
pBAFunctions = NULL;
LExit:
ReleaseObject(pBAFunctions);
ReleaseObject(pEngine);
return hr;
}
static void CALLBACK BafRelatedBundleVariableTestingTraceError(
__in_z LPCSTR /*szFile*/,
__in int /*iLine*/,
__in REPORT_LEVEL /*rl*/,
__in UINT source,
__in HRESULT hrError,
__in_z __format_string LPCSTR szFormat,
__in va_list args
)
{
// BalLogError currently uses the Exit... macros,
// so if expanding the scope need to ensure this doesn't get called recursively.
if (DUTIL_SOURCE_THMUTIL == source)
{
BalLogErrorArgs(hrError, szFormat, args);
}
}
|
