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
|
// 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"
HRESULT DetectNetCoreSdk(
__in NETFX_NET_CORE_PLATFORM platform,
__in LPCWSTR wzVersion,
__in LPCWSTR wzBaseDirectory,
__inout LPWSTR* psczLatestVersion
)
{
HRESULT hr = S_OK;
LPCWSTR wzPlatformName = NULL;
LPWSTR sczExePath = NULL;
LPWSTR sczCommandLine = NULL;
HANDLE hProcess = NULL;
HANDLE hStdOutErr = INVALID_HANDLE_VALUE;
BYTE* rgbOutput = NULL;
DWORD cbOutput = 0;
DWORD cbTotalRead = 0;
DWORD cbRead = 0;
DWORD dwExitCode = 0;
ReleaseNullStr(*psczLatestVersion);
switch (platform)
{
case NETFX_NET_CORE_PLATFORM_ARM64:
wzPlatformName = L"arm64";
break;
case NETFX_NET_CORE_PLATFORM_X64:
wzPlatformName = L"x64";
break;
case NETFX_NET_CORE_PLATFORM_X86:
wzPlatformName = L"x86";
break;
default:
BextExitWithRootFailure(hr, E_INVALIDARG, "Unknown platform: %u", platform);
break;
}
hr = StrAllocFormatted(&sczExePath, L"%ls%ls\\netcoresearch.exe", wzBaseDirectory, wzPlatformName);
BextExitOnFailure(hr, "Failed to build netcoresearch.exe path.");
hr = StrAllocFormatted(&sczCommandLine, L"\"%ls\" sdk %ls", sczExePath, wzVersion);
BextExitOnFailure(hr, "Failed to build netcoresearch.exe command line.");
hr = ProcExecute(sczExePath, sczCommandLine, &hProcess, NULL, &hStdOutErr);
if (HRESULT_FROM_WIN32(ERROR_EXE_MACHINE_TYPE_MISMATCH) == hr)
{
ExitFunction1(hr = S_FALSE);
}
BextExitOnFailure(hr, "Failed to run: %ls", sczCommandLine);
cbOutput = 64;
rgbOutput = static_cast<BYTE*>(MemAlloc(cbOutput, TRUE));
BextExitOnNull(rgbOutput, hr, E_OUTOFMEMORY, "Failed to alloc output string.");
while (::ReadFile(hStdOutErr, rgbOutput + cbTotalRead, cbOutput - cbTotalRead, &cbRead, NULL))
{
cbTotalRead += cbRead;
if (cbTotalRead == cbOutput)
{
cbOutput *= 2;
const LPVOID pvNew = MemReAlloc(rgbOutput, cbOutput, TRUE);
BextExitOnNull(pvNew, hr, E_OUTOFMEMORY, "Failed to realloc output string.");
rgbOutput = static_cast<BYTE*>(pvNew);
}
}
if (ERROR_BROKEN_PIPE != ::GetLastError())
{
BextExitWithLastError(hr, "Failed to read netcoresearch.exe output.");
}
hr = ProcWaitForCompletion(hProcess, INFINITE, &dwExitCode);
BextExitOnFailure(hr, "Failed to wait for netcoresearch.exe to exit.");
if (0 != dwExitCode)
{
BextExitWithRootFailure(hr, E_UNEXPECTED, "netcoresearch.exe failed with exit code: 0x%x\r\nOutput:\r\n%hs", dwExitCode, rgbOutput);
}
if (*rgbOutput)
{
hr = StrAllocStringAnsi(psczLatestVersion, reinterpret_cast<LPSTR>(rgbOutput), 0, CP_UTF8);
BextExitOnFailure(hr, "Failed to widen output string: %hs", rgbOutput);
}
LExit:
ReleaseFileHandle(hStdOutErr);
ReleaseHandle(hProcess);
ReleaseMem(rgbOutput);
ReleaseStr(sczCommandLine);
ReleaseStr(sczExePath);
return hr;
}
HRESULT NetfxPerformDetectNetCoreSdk(
__in LPCWSTR wzVariable,
__in NETFX_SEARCH* pSearch,
__in IBundleExtensionEngine* pEngine,
__in LPCWSTR wzBaseDirectory
)
{
HRESULT hr = S_OK;
LPWSTR sczLatestVersion = nullptr;
const auto& searchParams = pSearch->NetCoreSdkSearch;
hr = DetectNetCoreSdk(searchParams.platform, searchParams.sczVersion, wzBaseDirectory, &sczLatestVersion);
BextExitOnFailure(hr, "DetectNetCoreSdk failed.");
hr = pEngine->SetVariableVersion(wzVariable, sczLatestVersion);
BextExitOnFailure(hr, "Failed to set variable '%ls' to '%ls'", wzVariable, sczLatestVersion);
LExit:
return hr;
}
|
