1 files changed, 177 insertions, 0 deletions

diff --git a/src/ext/Iis/ca/scaweblog.cpp b/src/ext/Iis/ca/scaweblog.cpp
new file mode 100644
index 00000000..01147848
--- /dev/null
+++ b/src/ext/Iis/ca/scaweblog.cpp
@@ -0,0 +1,177 @@
+// 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"
+
+enum eWebLogQuery { wlqLog = 1, wlqFormat };
+
+/* ****************************************************************
+ * LookupLogFormatGUID -Looks up a given IIS Log format type in 
+ * the metabase and returns the GUID for it.
+ * ****************************************************************/
+static HRESULT LookupLogFormatGUID(
+    __in IMSAdminBase* piMetabase,
+    __in LPCWSTR wzLogFormat,
+    __out_ecount(cchGUID) LPWSTR wzGUID,
+    __in int cchGUID
+    )
+{
+    WCHAR wzKey[METADATA_MAX_NAME_LEN];
+    HRESULT hr = S_OK;
+
+    METADATA_RECORD mrKeyType;
+    ::ZeroMemory(&mrKeyType, sizeof(mrKeyType));
+
+    mrKeyType.dwMDIdentifier = MD_KEY_TYPE;
+    mrKeyType.dwMDAttributes = METADATA_NO_ATTRIBUTES;
+    mrKeyType.dwMDUserType = IIS_MD_UT_SERVER;
+    mrKeyType.dwMDDataType = ALL_METADATA;
+    mrKeyType.dwMDDataLen = 0;
+    mrKeyType.pbMDData = NULL;
+
+    METADATA_RECORD mrPluginId;
+    ::ZeroMemory(&mrPluginId, sizeof(mrPluginId));
+
+    mrPluginId.dwMDIdentifier = MD_LOG_PLUGIN_MOD_ID;
+    mrPluginId.dwMDAttributes = METADATA_INHERIT;
+    mrPluginId.dwMDUserType = IIS_MD_UT_SERVER;
+    mrPluginId.dwMDDataType = ALL_METADATA;
+    mrPluginId.dwMDDataLen = 0;
+    mrPluginId.pbMDData = NULL;
+    
+    hr = ::StringCchPrintfW(wzKey, countof(wzKey), L"/LM/Logging/%s", wzLogFormat);
+    ExitOnFailure(hr, "failed to format logging metabase key name");
+
+    // verify that we have this log format available in IIS
+    hr = MetaGetValue(piMetabase, METADATA_MASTER_ROOT_HANDLE, wzKey, &mrKeyType);
+    ExitOnFailure(hr, "Failed to find specified Log format key in IIS for log format: %ls", wzLogFormat);
+    ExitOnNull(mrKeyType.pbMDData, hr, E_POINTER, "Request for Log format key in IIS for log format returned success, but value was NULL");
+
+    if (0 != lstrcmpW(L"IIsLogModule", (LPCWSTR)mrKeyType.pbMDData))
+    {
+        ExitOnFailure(hr = E_UNEXPECTED, "Found invalid log format in IIS: %ls", (LPCWSTR)mrKeyType.pbMDData);
+    }
+
+    // find the GUID for that log format
+    hr = MetaGetValue(piMetabase, METADATA_MASTER_ROOT_HANDLE, wzKey, &mrPluginId);
+    ExitOnFailure(hr, "Failed to retrieve IISLog format GUID. Key: %ls", wzKey);
+    ExitOnNull(mrPluginId.pbMDData, hr, E_POINTER, "Retrieval of IISLog format GUID returned success, but value was NULL");
+
+    hr = ::StringCchCopyW(wzGUID, cchGUID, (LPCWSTR)mrPluginId.pbMDData);
+    ExitOnFailure(hr, "failed to copy metabase value: %ls", (LPCWSTR)mrPluginId.pbMDData);
+
+LExit:
+
+    if (mrKeyType.pbMDData)
+    {
+        MetaFreeValue(&mrKeyType);
+    }
+
+    if (mrPluginId.pbMDData)
+    {
+        MetaFreeValue(&mrPluginId);
+    }
+
+    return hr;
+}
+
+
+/* ****************************************************************
+ * ScaGetWebLog -Retrieves Log table data for the specified Log key
+ *
+ * ****************************************************************/
+HRESULT ScaGetWebLog(
+    IMSAdminBase* piMetabase,
+    LPCWSTR wzLog,
+    __in WCA_WRAPQUERY_HANDLE hWebLogQuery,
+    SCA_WEB_LOG* pswl
+    )
+{
+    HRESULT hr = S_OK;
+    LPWSTR pwzData = NULL;
+    MSIHANDLE hRec;
+
+    if (0 == WcaGetQueryRecords(hWebLogQuery))
+    {
+        WcaLog(LOGMSG_VERBOSE, "Skipping ScaGetWebLog() - no records to process");
+        ExitFunction1(hr = S_FALSE);
+    }
+
+    WcaFetchWrappedReset(hWebLogQuery);
+
+    hr = WcaFetchWrappedRecordWhereString(hWebLogQuery, wlqLog, wzLog, &hRec);
+    if (E_NOMOREITEMS == hr)
+    {
+        ExitOnFailure(hr, "cannot locate IIsWebLog.Log='%ls'", wzLog);
+    }
+    HRESULT hrTemp = WcaFetchWrappedRecordWhereString(hWebLogQuery, wlqLog, wzLog, &hRec);
+
+    if (SUCCEEDED(hrTemp))
+    {
+        ExitOnFailure(hr, "error - found multiple matching IIsWebLog rows");
+    }
+
+    ::ZeroMemory(pswl, sizeof(SCA_WEB_LOG));
+
+    // check that log key matches
+    hr = WcaGetRecordString(hRec, wlqLog, &pwzData);
+    ExitOnFailure(hr, "failed to get IIsWebLog.Log for Log: %ls", wzLog);
+    hr = ::StringCchCopyW(pswl->wzLog, countof(pswl->wzLog), pwzData);
+    ExitOnFailure(hr, "failed to copy log name: %ls", pwzData);
+
+    hr = WcaGetRecordString(hRec, wlqFormat, &pwzData);
+    ExitOnFailure(hr, "failed to get IIsWebLog.Format for Log:", wzLog);
+    hr = ::StringCchCopyW(pswl->wzFormat, countof(pswl->wzFormat), pwzData);
+    ExitOnFailure(hr, "failed to copy log format: %ls", pwzData);
+
+    // if they specified a log format, look up its GUID in the metabase
+    if (*pswl->wzFormat && 0 != lstrcmpW(pswl->wzFormat, L"none"))
+    {
+        hr = LookupLogFormatGUID(piMetabase, pswl->wzFormat, pswl->wzFormatGUID, countof(pswl->wzFormatGUID));
+        ExitOnFailure(hr, "Failed to get Log Format GUID for Log: %ls", wzLog);
+    }
+
+LExit:
+    ReleaseStr(pwzData);
+
+    return hr;
+}
+
+
+/* ****************************************************************
+ * ScaWriteWebLog -Writes the IIS log values to the metabase.
+ *
+ * ****************************************************************/
+HRESULT ScaWriteWebLog(
+    IMSAdminBase* piMetabase,
+    LPCWSTR wzWebBase,
+    SCA_WEB_LOG *pswl
+    )
+{
+    HRESULT hr = S_OK;
+
+    if (*pswl->wzFormat)
+    {
+        if (0 == lstrcmpW(pswl->wzFormat, L"none"))
+        {
+            // user wishes for Logging to be turned 'off'
+            hr = ScaWriteMetabaseValue(piMetabase, wzWebBase, L"", MD_LOG_TYPE, METADATA_INHERIT, IIS_MD_UT_SERVER, DWORD_METADATA, (LPVOID)((DWORD_PTR)0));
+            ExitOnFailure(hr, "Failed to write Log Type for Web: %ls", wzWebBase);
+        }
+        else
+        {
+            Assert(*pswl->wzFormatGUID);
+            
+            // write the GUID for the log format for the web to the metabase
+            hr = ScaWriteMetabaseValue(piMetabase, wzWebBase, L"", MD_LOG_PLUGIN_ORDER, METADATA_INHERIT, IIS_MD_UT_SERVER, STRING_METADATA, pswl->wzFormatGUID);
+            ExitOnFailure(hr, "Failed to write Log GUID for Web: %ls", wzWebBase);
+
+            hr = ScaWriteMetabaseValue(piMetabase, wzWebBase, L"", MD_LOG_TYPE, METADATA_INHERIT, IIS_MD_UT_SERVER, DWORD_METADATA, (LPVOID)((DWORD_PTR)1));
+            ExitOnFailure(hr, "Failed to write Log Type for Web: %ls", wzWebBase);
+        }
+    }
+
+LExit:
+    return hr;
+}
+
+