1 files changed, 55 insertions, 0 deletions

diff --git a/src/ext/NetFx/be/detectnetcoresdkfeatureband.cpp b/src/ext/NetFx/be/detectnetcoresdkfeatureband.cpp
new file mode 100644
index 00000000..d48c7a85
--- /dev/null
+++ b/src/ext/NetFx/be/detectnetcoresdkfeatureband.cpp
@@ -0,0 +1,55 @@
+// 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 DetectNetCoreSdkFeatureBand(
+    __in NETFX_NET_CORE_PLATFORM platform,
+    __in LPCWSTR wzMajorVersion,
+    __in LPCWSTR wzMinorVersion,
+    __in LPCWSTR wzPatchVersion,
+    __in LPCWSTR wzBaseDirectory,
+    __inout LPWSTR* psczLatestVersion
+    )
+{
+    HRESULT hr = S_OK;
+    LPWSTR sczArguments = NULL;
+
+    hr = StrAllocFormatted(&sczArguments, L"sdkfeatureband %ls %ls %ls", wzMajorVersion, wzMinorVersion, wzPatchVersion);
+    BextExitOnFailure(hr, "Failed to build sdkfeatureband netcoresearch.exe arguments.");
+
+    hr = RunNetCoreSearch(platform, wzBaseDirectory, sczArguments, psczLatestVersion);
+    BextExitOnFailure(hr, "Failed to run netcoresearch.exe for sdkfeatureband.");
+
+LExit:
+    ReleaseStr(sczArguments);
+
+    return hr;
+}
+
+HRESULT NetfxPerformDetectNetCoreSdkFeatureBand(
+    __in LPCWSTR wzVariable,
+    __in NETFX_SEARCH* pSearch,
+    __in IBundleExtensionEngine* pEngine,
+    __in LPCWSTR wzBaseDirectory
+    )
+{
+    HRESULT hr = S_OK;
+    LPWSTR sczLatestVersion = NULL;
+
+    hr = DetectNetCoreSdkFeatureBand(
+        pSearch->NetCoreSdkFeatureBandSearch.platform,
+        pSearch->NetCoreSdkFeatureBandSearch.sczMajorVersion,
+        pSearch->NetCoreSdkFeatureBandSearch.sczMinorVersion,
+        pSearch->NetCoreSdkFeatureBandSearch.sczPatchVersion,
+        wzBaseDirectory,
+        &sczLatestVersion);
+    BextExitOnFailure(hr, "DetectNetCoreSdkFeatureBand failed.");
+
+    hr = pEngine->SetVariableVersion(wzVariable, sczLatestVersion);
+    BextExitOnFailure(hr, "Failed to set variable '%ls' to '%ls'", wzVariable, sczLatestVersion);
+
+LExit:
+    ReleaseStr(sczLatestVersion);
+
+    return hr;
+}