1 files changed, 401 insertions, 0 deletions

diff --git a/src/ext/Iis/ca/scaapppool7.cpp b/src/ext/Iis/ca/scaapppool7.cpp
new file mode 100644
index 00000000..0fac4346
--- /dev/null
+++ b/src/ext/Iis/ca/scaapppool7.cpp
@@ -0,0 +1,401 @@
+// 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"
+
+// prototypes
+static HRESULT AppPoolExists(
+    __in LPCWSTR wzAppPool
+    );
+
+// functions
+HRESULT ScaFindAppPool7(
+    __in LPCWSTR wzAppPool,
+    __out_ecount(cchName) LPWSTR wzName,
+    __in DWORD cchName,
+    __in SCA_APPPOOL *psapList
+    )
+{
+    Assert(wzAppPool && *wzAppPool && wzName && *wzName);
+
+    HRESULT hr = S_OK;
+
+    // check memory first
+    SCA_APPPOOL* psap = psapList;
+    for (; psap; psap = psap->psapNext)
+    {
+        if (0 == wcscmp(psap->wzAppPool, wzAppPool))
+        {
+            break;
+        }
+    }
+    ExitOnNull(psap, hr, HRESULT_FROM_WIN32(ERROR_NOT_FOUND), "Could not find the app pool: %ls", wzAppPool);
+
+    // copy the web app pool name
+#pragma prefast(suppress:26037, "Source string is null terminated - it is populated as target of ::StringCchCopyW")
+    hr = ::StringCchCopyW(wzName, cchName, psap->wzName);
+    ExitOnFailure(hr, "failed to copy app pool name while finding app pool: %ls", psap->wzName);
+
+    // if it's not being installed now, check if it exists already
+    if (!psap->fHasComponent)
+    {
+        hr = AppPoolExists(psap->wzName);
+        ExitOnFailure(hr, "failed to check for existence of app pool: %ls", psap->wzName);
+    }
+
+LExit:
+    return hr;
+}
+
+
+static HRESULT AppPoolExists(
+    __in LPCWSTR /*wzAppPool*/
+    )
+{
+    HRESULT hr = S_OK;
+
+    //this function checks for existance of app pool in IIS7 config
+    //at schedule time, we will defer this to execute time.
+
+    return hr;
+}
+
+
+HRESULT ScaAppPoolInstall7(
+    __in SCA_APPPOOL* psapList
+    )
+{
+    HRESULT hr = S_OK;
+
+    for (SCA_APPPOOL* psap = psapList; psap; psap = psap->psapNext)
+    {
+        // if we are installing the app pool
+        if (psap->fHasComponent && WcaIsInstalling(psap->isInstalled, psap->isAction))
+        {
+            hr = ScaWriteAppPool7(psap);
+            ExitOnFailure(hr, "failed to write AppPool '%ls' to metabase", psap->wzAppPool);
+        }
+    }
+
+LExit:
+    return hr;
+}
+
+
+HRESULT ScaAppPoolUninstall7(
+    __in SCA_APPPOOL* psapList
+    )
+{
+
+    HRESULT hr = S_OK;
+
+    for (SCA_APPPOOL* psap = psapList; psap; psap = psap->psapNext)
+    {
+        // if we are uninstalling the app pool
+        if (psap->fHasComponent && WcaIsUninstalling(psap->isInstalled, psap->isAction))
+        {
+            hr = ScaRemoveAppPool7(psap);
+            ExitOnFailure(hr, "Failed to remove AppPool '%ls' from metabase", psap->wzAppPool);
+        }
+    }
+
+LExit:
+    return hr;
+}
+
+
+HRESULT ScaWriteAppPool7(
+    __in const SCA_APPPOOL* psap
+    )
+{
+    Assert(psap);
+
+    HRESULT hr = S_OK;
+    DWORD dwIdentity = 0xFFFFFFFF;
+    LPWSTR pwzValue = NULL;
+    LPWSTR wz = NULL;
+
+    //create the app pool
+    hr = ScaWriteConfigID(IIS_APPPOOL);
+    ExitOnFailure(hr, "failed to write AppPool key.");
+
+    hr = ScaWriteConfigID(IIS_CREATE);
+    ExitOnFailure(hr, "failed to write AppPool create action.");
+
+    hr = ScaWriteConfigString(psap->wzName);
+    ExitOnFailure(hr, "failed to write AppPool name: %ls", psap->wzName);
+
+    // Now do all the optional stuff
+
+    // Set the AppPool Recycling Tab
+    if (MSI_NULL_INTEGER != psap->iRecycleMinutes)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_MIN);
+        ExitOnFailure(hr, "failed to set periodic restart time id");
+        hr = ScaWriteConfigInteger(psap->iRecycleMinutes);
+        ExitOnFailure(hr, "failed to set periodic restart time");
+    }
+
+    if (MSI_NULL_INTEGER != psap->iRecycleRequests)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_REQ);
+        ExitOnFailure(hr, "failed to set periodic restart request count id");
+        hr = ScaWriteConfigInteger(psap->iRecycleRequests);
+        ExitOnFailure(hr, "failed to set periodic restart request count");
+    }
+
+    if (*psap->wzRecycleTimes)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_TIMES);
+        ExitOnFailure(hr, "failed to set periodic restart schedule id");
+        hr = ScaWriteConfigString(psap->wzRecycleTimes);
+        ExitOnFailure(hr, "failed to set periodic restart schedule");
+    }
+
+    if (MSI_NULL_INTEGER != psap->iVirtualMemory)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_VIRMEM);
+        ExitOnFailure(hr, "failed to set periodic restart memory count id");
+        hr = ScaWriteConfigInteger(psap->iVirtualMemory);
+        ExitOnFailure(hr, "failed to set periodic restart memory count");
+    }
+
+    if (MSI_NULL_INTEGER != psap->iPrivateMemory)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_PRIVMEM);
+        ExitOnFailure(hr, "failed to set periodic restart private memory count id");
+        hr = ScaWriteConfigInteger(psap->iPrivateMemory);
+        ExitOnFailure(hr, "failed to set periodic restart private memory count");
+    }
+
+    // Set AppPool Performance Tab
+    if (MSI_NULL_INTEGER != psap->iIdleTimeout)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_IDLTIMEOUT);
+        ExitOnFailure(hr, "failed to set idle timeout value id");
+        hr = ScaWriteConfigInteger(psap->iIdleTimeout);
+        ExitOnFailure(hr, "failed to set idle timeout value");
+    }
+
+    if (MSI_NULL_INTEGER != psap->iQueueLimit)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_QUEUELIMIT);
+        ExitOnFailure(hr, "failed to set request queue limit value id");
+        hr = ScaWriteConfigInteger(psap->iQueueLimit);
+        ExitOnFailure(hr, "failed to set request queue limit value");
+    }
+    if (*psap->wzCpuMon)
+    {
+#pragma prefast(suppress:26037, "Source string is null terminated - it is populated as target of ::StringCchCopyW")
+        hr = ::StrAllocString(&pwzValue, psap->wzCpuMon, 0);
+        ExitOnFailure(hr, "failed to allocate CPUMonitor string");
+
+        DWORD dwPercent = 0;
+        DWORD dwRefreshMinutes = 0;
+        DWORD dwAction = 0;
+
+        dwPercent = wcstoul(pwzValue, &wz, 10);
+        if (100  < dwPercent)
+        {
+            ExitOnFailure(hr = E_INVALIDARG, "invalid maximum cpu percentage value: %d", dwPercent);
+        }
+        if (wz && L',' == *wz)
+        {
+            ++wz;
+            dwRefreshMinutes = wcstoul(wz, &wz, 10);
+            if (wz && L',' == *wz)
+            {
+                ++wz;
+                dwAction = wcstoul(wz, &wz, 10);
+            }
+        }
+        if (dwPercent)
+        {
+            hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_CPU_PCT);
+            ExitOnFailure(hr, "failed to set recycle pct id");
+            hr = ScaWriteConfigInteger(dwPercent);
+            ExitOnFailure(hr, "failed to set CPU percentage max");
+        }
+        if (dwRefreshMinutes)
+        {
+            hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_CPU_REFRESH);
+            ExitOnFailure(hr, "failed to set recycle refresh id");
+            hr = ScaWriteConfigInteger(dwRefreshMinutes);
+            ExitOnFailure(hr, "failed to set refresh CPU minutes");
+        }
+        if (dwAction)
+        {
+            // 0 = No Action
+            // 1 = Shutdown
+            hr = ScaWriteConfigID(IIS_APPPOOL_RECYCLE_CPU_ACTION);
+            ExitOnFailure(hr, "failed to set recycle refresh id");
+            hr = ScaWriteConfigInteger(dwAction);
+            ExitOnFailure(hr, "failed to set CPU action");
+        }
+    }
+
+    if (MSI_NULL_INTEGER != psap->iMaxProcesses)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_MAXPROCESS);
+        ExitOnFailure(hr, "Failed to write max processes config ID");
+
+        hr = ScaWriteConfigInteger(psap->iMaxProcesses);
+        ExitOnFailure(hr, "failed to set web garden maximum worker processes");
+    }
+
+    hr = ScaWriteConfigID(IIS_APPPOOL_32BIT);
+    ExitOnFailure(hr, "Failed to write 32 bit app pool config ID");
+    hr = ScaWriteConfigInteger(psap->iCompAttributes & msidbComponentAttributes64bit ? 0 : 1);
+    ExitOnFailure(hr, "Failed to write 32 bit app pool config value");
+
+    //
+    // Set the AppPool Identity tab
+    //
+    if (psap->iAttributes & APATTR_APPPOOLIDENTITY)
+    {
+        dwIdentity = 4;
+    }
+    else if (psap->iAttributes & APATTR_NETSERVICE)
+    {
+        dwIdentity = 2;
+    }
+    else if (psap->iAttributes & APATTR_LOCSERVICE)
+    {
+        dwIdentity = 1;
+    }
+    else if (psap->iAttributes & APATTR_LOCSYSTEM)
+    {
+        dwIdentity = 0;
+    }
+    else if (psap->iAttributes & APATTR_OTHERUSER)
+    {
+        if (!*psap->suUser.wzDomain || 0 == _wcsicmp(psap->suUser.wzDomain, L"."))
+        {
+            if (0 == _wcsicmp(psap->suUser.wzName, L"NetworkService"))
+            {
+                dwIdentity = 2;
+            }
+            else if (0 == _wcsicmp(psap->suUser.wzName, L"LocalService"))
+            {
+                dwIdentity = 1;
+            }
+            else if (0 == _wcsicmp(psap->suUser.wzName, L"LocalSystem"))
+            {
+                dwIdentity = 0;
+            }
+            else
+            {
+                dwIdentity = 3;
+            }
+        }
+        else if (0 == _wcsicmp(psap->suUser.wzDomain, L"NT AUTHORITY"))
+        {
+            if (0 == _wcsicmp(psap->suUser.wzName, L"NETWORK SERVICE"))
+            {
+                dwIdentity = 2;
+            }
+            else if (0 == _wcsicmp(psap->suUser.wzName, L"SERVICE"))
+            {
+                dwIdentity = 1;
+            }
+            else if (0 == _wcsicmp(psap->suUser.wzName, L"SYSTEM"))
+            {
+                dwIdentity = 0;
+            }
+            else
+            {
+                dwIdentity = 3;
+            }
+        }
+        else
+        {
+            dwIdentity = 3;
+        }
+    }
+
+    if (-1 != dwIdentity)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_IDENTITY);
+        ExitOnFailure(hr, "failed to set app pool identity id");
+        hr = ScaWriteConfigInteger(dwIdentity);
+        ExitOnFailure(hr, "failed to set app pool identity");
+
+        if (3 == dwIdentity)
+        {
+            if (*psap->suUser.wzDomain)
+            {
+                hr = StrAllocFormatted(&pwzValue, L"%s\\%s", psap->suUser.wzDomain, psap->suUser.wzName);
+                ExitOnFailure(hr, "failed to format user name: %ls domain: %ls", psap->suUser.wzName, psap->suUser.wzDomain);
+            }
+            else
+            {
+                hr = StrAllocFormatted(&pwzValue, L"%s", psap->suUser.wzName);
+                ExitOnFailure(hr, "failed to format user name: %ls", psap->suUser.wzName);
+            }
+
+            hr = ScaWriteConfigID(IIS_APPPOOL_USER);
+            ExitOnFailure(hr, "failed to set app pool identity name id");
+            hr = ScaWriteConfigString(pwzValue);
+            ExitOnFailure(hr, "failed to set app pool identity name");
+
+            hr = ScaWriteConfigID(IIS_APPPOOL_PWD);
+            ExitOnFailure(hr, "failed to set app pool identity password id");
+            hr = ScaWriteConfigString(psap->suUser.wzPassword);
+            ExitOnFailure(hr, "failed to set app pool identity password");
+        }
+    }
+    
+    if (*psap->wzManagedPipelineMode)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_MANAGED_PIPELINE_MODE);
+        ExitOnFailure(hr, "failed to set app pool integrated mode");
+        hr = ScaWriteConfigString(psap->wzManagedPipelineMode);
+        ExitOnFailure(hr, "failed to set app pool managed pipeline mode value");
+    }
+
+    if (*psap->wzManagedRuntimeVersion)
+    {
+        hr = ScaWriteConfigID(IIS_APPPOOL_MANAGED_RUNTIME_VERSION);
+        ExitOnFailure(hr, "failed to set app pool managed runtime version mode");
+        hr = ScaWriteConfigString(psap->wzManagedRuntimeVersion);
+        ExitOnFailure(hr, "failed to set app pool managed runtime version value");
+    }
+
+    //
+    //The number of properties above is variable so we put an end tag in so the
+    //execute CA will know when to stop looking for AppPool properties
+    //
+    hr = ScaWriteConfigID(IIS_APPPOOL_END);
+    ExitOnFailure(hr, "failed to set app pool end of properties id");
+
+LExit:
+    ReleaseStr(pwzValue);
+
+    return hr;
+}
+
+
+HRESULT ScaRemoveAppPool7(
+    __in const SCA_APPPOOL* psap
+    )
+{
+    Assert(psap);
+
+    HRESULT hr = S_OK;
+
+    //do not delete the default App Pool
+    if (0 != _wcsicmp(psap->wzAppPool, L"DefaultAppPool"))
+    {
+        //delete the app pool
+        hr = ScaWriteConfigID(IIS_APPPOOL);
+        ExitOnFailure(hr, "failed to write AppPool key.");
+
+        hr = ScaWriteConfigID(IIS_DELETE);
+        ExitOnFailure(hr, "failed to write AppPool delete action.");
+
+        hr = ScaWriteConfigString(psap->wzName);
+        ExitOnFailure(hr, "failed to delete AppPool: %ls", psap->wzName);
+    }
+
+LExit:
+    return hr;
+}