1855 lines
46 KiB
C++
1855 lines
46 KiB
C++
//=============================================================================
|
|
|
|
//
|
|
|
|
// File: WMIlFile.cpp
|
|
|
|
// Copyright (c) 1994-2001 Microsoft Corporation, All Rights Reserved
|
|
// All rights reserved.
|
|
//
|
|
//=============================================================================
|
|
|
|
#include "stdafx.h"
|
|
#include <buildnum.h>
|
|
#include <helpids.h>
|
|
|
|
#include "WMIparse.h"
|
|
#include "resource.h"
|
|
#include "WMIlprs.h"
|
|
|
|
|
|
#include "WMIlfile.h"
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::CWMILocFile
|
|
//
|
|
//*****************************************************************************
|
|
|
|
CWMILocFile::CWMILocFile(
|
|
ILocParser *pParentClass)
|
|
{
|
|
//
|
|
// C.O.M. initialization
|
|
//
|
|
m_pParentClass = pParentClass;
|
|
m_sCurrentNamespace = "";
|
|
m_ulRefCount = 0;
|
|
|
|
//
|
|
// WMI file initialization
|
|
//
|
|
m_uiLineNumber = 0;
|
|
m_pOpenSourceFile = NULL;
|
|
m_pOpenTargetFile = NULL;
|
|
|
|
AddRef();
|
|
IncrementClassCount();
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetFileDescriptions
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::GetFileDescriptions(
|
|
CEnumCallback &cb)
|
|
{
|
|
EnumInfo eiFileInfo;
|
|
CLString strDesc;
|
|
|
|
eiFileInfo.szAbbreviation = NULL;
|
|
|
|
LTVERIFY(strDesc.LoadString(g_hDll, IDS_WMI_DESC));
|
|
|
|
eiFileInfo.szDescription = (const TCHAR *)strDesc;
|
|
eiFileInfo.ulValue = ftWMIFileType;
|
|
|
|
cb.ProcessEnum(eiFileInfo);
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::AddRef
|
|
//
|
|
//*****************************************************************************
|
|
|
|
ULONG CWMILocFile::AddRef(void)
|
|
{
|
|
if (m_pParentClass != NULL)
|
|
{
|
|
m_pParentClass->AddRef();
|
|
}
|
|
|
|
return m_ulRefCount++;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::Release
|
|
//
|
|
//*****************************************************************************
|
|
|
|
ULONG CWMILocFile::Release(void)
|
|
{
|
|
LTASSERT(m_ulRefCount != 0);
|
|
|
|
if (m_pParentClass != NULL)
|
|
{
|
|
|
|
m_pParentClass->Release();
|
|
}
|
|
|
|
m_ulRefCount--;
|
|
if (m_ulRefCount == 0)
|
|
{
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
return m_ulRefCount;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::QueryInterface
|
|
//
|
|
//*****************************************************************************
|
|
|
|
HRESULT CWMILocFile::QueryInterface(
|
|
REFIID iid,
|
|
LPVOID *ppvObj)
|
|
{
|
|
if (m_pParentClass != NULL)
|
|
{
|
|
return m_pParentClass->QueryInterface(iid, ppvObj);
|
|
}
|
|
else
|
|
{
|
|
SCODE scRetVal = E_NOINTERFACE;
|
|
|
|
*ppvObj = NULL;
|
|
|
|
if (iid == IID_IUnknown)
|
|
{
|
|
*ppvObj = (IUnknown *)this;
|
|
scRetVal = S_OK;
|
|
}
|
|
else if (iid == IID_ILocFile)
|
|
{
|
|
*ppvObj = (ILocFile *)this;
|
|
scRetVal = S_OK;
|
|
}
|
|
|
|
if (scRetVal == S_OK)
|
|
{
|
|
AddRef();
|
|
}
|
|
return ResultFromScode(scRetVal);
|
|
}
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::AssertValidInterface
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::AssertValidInterface(void)
|
|
const
|
|
{
|
|
AssertValid();
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::OpenFile
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::OpenFile(
|
|
const CFileSpec &fsFile,
|
|
CReporter &Reporter)
|
|
{
|
|
LTTRACEPOINT("OpenFile()");
|
|
|
|
BOOL fRetCode;
|
|
|
|
LTASSERT(m_pOpenTargetFile == NULL);
|
|
fRetCode = FALSE;
|
|
|
|
m_didFileId = fsFile.GetFileId();
|
|
m_pstrFileName = fsFile.GetFileName();
|
|
|
|
if (m_pOpenSourceFile != NULL)
|
|
{
|
|
fclose(m_pOpenSourceFile);
|
|
m_pOpenSourceFile = NULL;
|
|
}
|
|
|
|
// We are just going to open the file.
|
|
// and save the handle.
|
|
// ===================================
|
|
|
|
try
|
|
{
|
|
|
|
m_pOpenSourceFile = fopen(_T(m_pstrFileName), "rb");
|
|
|
|
if (!m_pOpenSourceFile)
|
|
{
|
|
fclose(m_pOpenSourceFile);
|
|
m_pOpenSourceFile = NULL;
|
|
}
|
|
else
|
|
{
|
|
fRetCode = TRUE;
|
|
}
|
|
|
|
}
|
|
catch (CMemoryException *pMemoryException)
|
|
{
|
|
CLString strContext;
|
|
|
|
strContext.LoadString(g_hDll, IDS_WMI_GENERIC_LOCATION);
|
|
|
|
Reporter.IssueMessage(esError, strContext, g_hDll, IDS_WMI_NO_MEMORY,
|
|
g_locNull);
|
|
|
|
pMemoryException->Delete();
|
|
}
|
|
return fRetCode;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetFileType
|
|
//
|
|
//*****************************************************************************
|
|
|
|
FileType CWMILocFile::GetFileType(void)
|
|
const
|
|
{
|
|
//
|
|
// Just return some number that isn't ftUnknown...
|
|
//
|
|
return ftWMIFileType;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetFileTypeDescription
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::GetFileTypeDescription(
|
|
CLString &strDesc)
|
|
const
|
|
{
|
|
LTVERIFY(strDesc.LoadString(g_hDll, IDS_WMI_DESC));
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetAssociatedFiles
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::GetAssociatedFiles(
|
|
CStringList &lstFiles)
|
|
const
|
|
{
|
|
LTASSERT(lstFiles.GetCount() == 0);
|
|
|
|
lstFiles.RemoveAll();
|
|
return FALSE;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::EnumerateFile
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::EnumerateFile(
|
|
CLocItemHandler &ihItemHandler,
|
|
const CLocLangId &lid,
|
|
const DBID &dbidFileId)
|
|
{
|
|
BOOL bRet = TRUE;
|
|
DBID dbidThisId = dbidFileId;
|
|
|
|
LTTRACEPOINT("EnumerateFile()");
|
|
|
|
if (m_pOpenSourceFile == NULL)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
// Enumerate file will need to:
|
|
// * Parse the MOF.
|
|
// * Walk through all qualifiers. For each "Amended" qualifier,
|
|
// send back a CLocItem whose key is namespace, class, property and qualifier name.
|
|
// * Fail if the language ID does not match that of LocaleID.
|
|
// * Parent objects are namespaces, classes
|
|
// =============================================================
|
|
|
|
m_cpSource = lid.GetCodePage(cpAnsi);
|
|
m_wSourceId = lid.GetLanguageId();
|
|
|
|
ihItemHandler.SetProgressIndicator(0);
|
|
|
|
bRet = ReadLines(ihItemHandler, dbidFileId, FALSE);
|
|
|
|
return bRet;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GenerateFile
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::GenerateFile(
|
|
const CPascalString &pstrTargetFile,
|
|
CLocItemHandler &ihItemHandler,
|
|
const CLocLangId &lidSource,
|
|
const CLocLangId &lidTarget,
|
|
const DBID &dbidParent)
|
|
{
|
|
LTASSERT(m_pOpenTargetFile == NULL);
|
|
BOOL fRetVal = TRUE;
|
|
|
|
if (m_pOpenSourceFile== NULL)
|
|
{
|
|
return FALSE;
|
|
}
|
|
// Generate File needs to:
|
|
// * Parse the MOF.
|
|
// * Walk through all qualifiers. For each "Amended" qualifier,
|
|
// send back a CLocItem whose key is namespace, class, property and qualifier name.
|
|
// * Replace all Amended qualifiers with localized text
|
|
// * Replace all occurrences of the locale ID in namespaces and qualifiers
|
|
// with the new one.
|
|
// =================================================================================
|
|
|
|
m_cpSource = lidSource.GetCodePage(cpAnsi);
|
|
m_cpTarget = lidTarget.GetCodePage(cpAnsi);
|
|
|
|
m_wSourceId = lidSource.GetLanguageId();
|
|
m_wTargetId = lidTarget.GetLanguageId();
|
|
|
|
try
|
|
{
|
|
CFileException excFile;
|
|
fRetVal = FALSE;
|
|
|
|
if (m_pOpenTargetFile != NULL)
|
|
{
|
|
fclose(m_pOpenTargetFile);
|
|
m_pOpenTargetFile = NULL;
|
|
}
|
|
|
|
char FileName[255];
|
|
strcpy(FileName, _bstr_t(_T(pstrTargetFile)));
|
|
|
|
// This file must be in Unicode.
|
|
HANDLE hFile = CreateFile(FileName, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ, NULL,
|
|
CREATE_ALWAYS, 0, NULL);
|
|
if(hFile != INVALID_HANDLE_VALUE)
|
|
{
|
|
char cUnicodeHeader[2] = {0xff, 0xfe};
|
|
DWORD dwWrite;
|
|
WriteFile(hFile, cUnicodeHeader, 2, &dwWrite, NULL);
|
|
CloseHandle(hFile);
|
|
}
|
|
|
|
m_pOpenTargetFile = fopen(FileName, "ab");
|
|
|
|
if (!m_pOpenTargetFile)
|
|
{
|
|
fclose(m_pOpenTargetFile);
|
|
m_pOpenTargetFile = NULL;
|
|
}
|
|
else
|
|
{
|
|
fRetVal = TRUE;
|
|
}
|
|
}
|
|
catch (CMemoryException *pMemoryException)
|
|
{
|
|
CLString strContext;
|
|
GetFullContext(strContext);
|
|
|
|
ihItemHandler.IssueMessage(esError, strContext, g_hDll, IDS_WMI_NO_MEMORY,
|
|
g_locNull);
|
|
|
|
pMemoryException->Delete();
|
|
}
|
|
catch (CFileException *pFileException)
|
|
{
|
|
fclose(m_pOpenTargetFile);
|
|
fRetVal = FALSE;
|
|
|
|
ReportFileError((const WCHAR *)pstrTargetFile, m_didFileId, pFileException, ihItemHandler);
|
|
|
|
pFileException->Delete();
|
|
}
|
|
|
|
if (!fRetVal)
|
|
{
|
|
return fRetVal;
|
|
}
|
|
|
|
if (!GetParserOptionBool(g_puid, OPT_DISABLE_WATERMARKING))
|
|
{
|
|
WriteWaterMark();
|
|
}
|
|
|
|
fRetVal = ReadLines(ihItemHandler, dbidParent, TRUE);
|
|
|
|
fclose(m_pOpenTargetFile);
|
|
|
|
if (!fRetVal)
|
|
{
|
|
CLFile::Remove(pstrTargetFile);
|
|
}
|
|
|
|
return fRetVal;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GenerateItem
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::GenerateItem(
|
|
CLocItemHandler &ihItemHandler,
|
|
CLocItemSet &isItemSet,
|
|
wchar_t **pOutBuffer,
|
|
UINT &uiStartingPos)
|
|
{
|
|
|
|
BOOL fRetVal = TRUE;
|
|
UINT uiLength;
|
|
|
|
wchar_t *pTemp = *pOutBuffer;
|
|
|
|
_bstr_t sQualifierValue;
|
|
|
|
// If nothing has changed, we can just
|
|
// ignore this line.
|
|
|
|
fRetVal = GetQualifierValue(pTemp, uiStartingPos, sQualifierValue, uiLength);
|
|
if (fRetVal)
|
|
{
|
|
fRetVal = ihItemHandler.HandleItemSet(isItemSet);
|
|
if (fRetVal)
|
|
{
|
|
sQualifierValue = "";
|
|
for (int i = 0; i < isItemSet.GetSize(); i++)
|
|
{
|
|
CVC::ValidationCode vcRetVal;
|
|
CLocItem *pLocItem = isItemSet[i];
|
|
CLString strContext;
|
|
CLocation loc;
|
|
|
|
GetFullContext(strContext);
|
|
loc.SetGlobalId(
|
|
CGlobalId(pLocItem->GetMyDatabaseId(), otResource));
|
|
loc.SetView(vTransTab);
|
|
|
|
CPascalString pstrId, pstrText;
|
|
|
|
pLocItem->GetUniqueId().GetResId().GetId(pstrId);
|
|
pstrText = pLocItem->GetLocString().GetString();
|
|
|
|
if (i > 0)
|
|
sQualifierValue += L"\",\"";
|
|
|
|
sQualifierValue += (const wchar_t *)pstrText;
|
|
}
|
|
|
|
// Set it live in the buffer. We are not going to
|
|
// write it to the file until the very end.
|
|
|
|
fRetVal = SetQualifierValue(pTemp, pOutBuffer, uiStartingPos, sQualifierValue, uiLength);
|
|
pTemp = *pOutBuffer;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fRetVal;
|
|
}
|
|
|
|
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::EnumerateItem
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::EnumerateItem(
|
|
CLocItemHandler &ihItemHandler,
|
|
CLocItemSet &isItemSet)
|
|
{
|
|
BOOL fRetVal;
|
|
|
|
if (isItemSet.GetSize() != 0)
|
|
{
|
|
fRetVal = ihItemHandler.HandleItemSet(isItemSet);
|
|
}
|
|
else
|
|
{
|
|
fRetVal = TRUE;
|
|
}
|
|
|
|
return fRetVal;
|
|
}
|
|
|
|
|
|
|
|
#ifdef _DEBUG
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::AssertValid
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::AssertValid(void)
|
|
const
|
|
{
|
|
CLObject::AssertValid();
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::Dump
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::Dump(
|
|
CDumpContext &dc)
|
|
const
|
|
{
|
|
CLObject::Dump(dc);
|
|
}
|
|
|
|
#endif
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::~CWMILocFile
|
|
//
|
|
//*****************************************************************************
|
|
|
|
CWMILocFile::~CWMILocFile()
|
|
{
|
|
DEBUGONLY(AssertValid());
|
|
|
|
if (m_pOpenSourceFile != NULL)
|
|
{
|
|
fclose(m_pOpenSourceFile);
|
|
m_pOpenSourceFile = NULL;
|
|
}
|
|
|
|
DecrementClassCount();
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::SetFlags
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::SetFlags(
|
|
CLocItem *pItem,
|
|
CLocString &lsString)
|
|
const
|
|
{
|
|
ULONG ulItemType;
|
|
|
|
pItem->SetFDevLock(FALSE);
|
|
pItem->SetFUsrLock(FALSE);
|
|
pItem->SetFExpandable(FALSE);
|
|
|
|
LTVERIFY(pItem->GetUniqueId().GetTypeId().GetId(ulItemType));
|
|
|
|
switch (ulItemType)
|
|
{
|
|
case wltNamespaceName:
|
|
pItem->SetFDisplayable(TRUE);
|
|
pItem->SetFNoResTable(TRUE);
|
|
break;
|
|
|
|
case wltClassName:
|
|
case wltPropertyName:
|
|
pItem->SetFDisplayable(FALSE);
|
|
pItem->SetFNoResTable(FALSE);
|
|
lsString.SetCodePageType(cpAnsi);
|
|
lsString.SetStringType(CST::Text);
|
|
break;
|
|
|
|
default:
|
|
LTASSERT(FALSE && "Unexpected item type!");
|
|
}
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::ReadLines
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::ReadLines(
|
|
CLocItemHandler &ihItemHandler,
|
|
const DBID &dbidFileId,
|
|
BOOL fGenerating)
|
|
{
|
|
DBID dbidSectionId;
|
|
BOOL fRetVal = TRUE;
|
|
wchar_t *pstrNamespaceName;
|
|
_bstr_t pstrClassName;
|
|
UINT uiStartPos = 0;
|
|
|
|
UINT uiCommentNum;
|
|
UINT uiReadingOrder;
|
|
|
|
dbidSectionId = dbidFileId;
|
|
m_uiLineNumber = 0;
|
|
BOOL bPendingObj = FALSE;
|
|
|
|
|
|
try
|
|
{
|
|
UINT uiOldPercentage = 0, uiNewPercentage = 0;
|
|
UINT uiBytesRead, uiCurrPos = 1;
|
|
|
|
ihItemHandler.SetProgressIndicator(uiOldPercentage);
|
|
|
|
fseek(m_pOpenSourceFile, 0, SEEK_END);
|
|
long lSize = ftell(m_pOpenSourceFile) + 6;
|
|
fseek(m_pOpenSourceFile, 0, SEEK_SET);
|
|
|
|
// Check for UNICODE source file.
|
|
// ==============================
|
|
|
|
BYTE UnicodeSignature[2];
|
|
BOOL bUnicode = FALSE;
|
|
|
|
if (fread(UnicodeSignature, sizeof(BYTE), 2, m_pOpenSourceFile) != 2)
|
|
{
|
|
fRetVal = FALSE;
|
|
return fRetVal;
|
|
}
|
|
if ((UnicodeSignature[0] == 0xFF && UnicodeSignature[1] == 0xFE) ||
|
|
(UnicodeSignature[0] == 0xFE && UnicodeSignature[1] == 0xFF))
|
|
{
|
|
bUnicode = TRUE;
|
|
lSize *= 2;
|
|
}
|
|
if (!bUnicode)
|
|
fseek(m_pOpenSourceFile, 0, SEEK_SET);
|
|
|
|
wchar_t *pBuff = (wchar_t *)new wchar_t[lSize+1];
|
|
if (!pBuff)
|
|
{
|
|
fRetVal = FALSE;
|
|
return fRetVal;
|
|
}
|
|
memset(pBuff,0,lSize*sizeof(wchar_t));
|
|
|
|
// If this is not a Unicode file,
|
|
// we need to perform a conversion.
|
|
// =====================================
|
|
|
|
if (bUnicode)
|
|
uiBytesRead = fread(pBuff, sizeof(wchar_t), lSize, m_pOpenSourceFile);
|
|
else
|
|
{
|
|
char *pCharBuff = new char[lSize+1];
|
|
if (pCharBuff)
|
|
{
|
|
uiBytesRead = fread(pCharBuff, sizeof(char), lSize, m_pOpenSourceFile);
|
|
pCharBuff[lSize] = '\0';
|
|
swprintf(pBuff, L"%S", pCharBuff);
|
|
delete pCharBuff;
|
|
}
|
|
}
|
|
pBuff[lSize] = '\0';
|
|
|
|
if (uiBytesRead != 0)
|
|
{
|
|
wchar_t *pOutBuffer = NULL;
|
|
uiCurrPos += uiBytesRead;
|
|
fRetVal = TRUE;
|
|
|
|
pstrNamespaceName = NULL;
|
|
m_sCurrentNamespace = L"";
|
|
pstrClassName = L"";
|
|
|
|
uiCommentNum = 0;
|
|
uiReadingOrder = 1;
|
|
|
|
WMIFileError wmiRet;
|
|
CLocItemSet isItemSet;
|
|
UINT uiTemp = 0;
|
|
DWORD dwCount = 0;
|
|
|
|
// If we are generating a file, make a copy
|
|
// of the outbound buffer.
|
|
|
|
if (fGenerating)
|
|
pOutBuffer = pBuff;
|
|
|
|
while (GetNextQualifierPos(L"amended", pBuff, uiTemp, uiTemp) && !bPendingObj)
|
|
{
|
|
// If we have found the "amended" keyword,
|
|
// we want to find the namespace,
|
|
// class, and property if applicable, and
|
|
// generate the object as appropriate.
|
|
// ======================================
|
|
|
|
pstrNamespaceName = GetCurrentNamespace(pBuff, uiTemp);
|
|
if (!pstrNamespaceName || !wcslen(pstrNamespaceName))
|
|
{
|
|
delete pBuff;
|
|
return FALSE;
|
|
}
|
|
if (wcscmp(pstrNamespaceName, m_sCurrentNamespace))
|
|
{
|
|
// We need to generate this object,
|
|
// and set it up as the current parent.
|
|
// ====================================
|
|
|
|
CLocItem *pNewItem = new CLocItem;
|
|
CPascalString sId;
|
|
|
|
if (pNewItem)
|
|
{
|
|
|
|
CLocUniqueId uid;
|
|
sId = (const WCHAR *)pstrNamespaceName;
|
|
|
|
uid.GetResId().SetId(sId);
|
|
uid.GetTypeId().SetId(wltNamespaceName);
|
|
uid.SetParentId(dbidFileId);
|
|
pNewItem->SetUniqueId(uid);
|
|
|
|
CLocString lsString;
|
|
|
|
pNewItem->SetIconType(CIT::String);
|
|
CPascalString pstrComment, pstrText;
|
|
|
|
pNewItem->SetInstructions(pstrComment);
|
|
lsString.SetString(pstrText);
|
|
|
|
SetFlags(pNewItem, lsString);
|
|
pNewItem->SetLocString(lsString);
|
|
|
|
isItemSet.Add(pNewItem);
|
|
|
|
uiReadingOrder = (uiReadingOrder + 999)/1000*1000;
|
|
isItemSet[0]->SetDisplayOrder(uiReadingOrder);
|
|
uiReadingOrder++;
|
|
|
|
fRetVal = EnumerateItem(ihItemHandler,
|
|
isItemSet);
|
|
|
|
dbidSectionId.Clear();
|
|
dbidSectionId = isItemSet[0]->GetMyDatabaseId();
|
|
isItemSet.ClearItemSet();
|
|
uiTemp += 1;
|
|
|
|
}
|
|
|
|
m_sCurrentNamespace = pstrNamespaceName;
|
|
delete pstrNamespaceName;
|
|
|
|
|
|
}
|
|
|
|
// For the class name, this is trickier.
|
|
// If there are one or more qualifiers
|
|
// on the class itself, we need to read ahead
|
|
// to find the class name, and then
|
|
// generate all the qualifier objects at once.
|
|
// ==========================================
|
|
|
|
wmiRet = GetNextItemSet(dwCount, pBuff, isItemSet, dbidSectionId, uiStartPos);
|
|
while (wmiRet == WMINoError)
|
|
{
|
|
// For each item, we want to set its key,
|
|
// and push it or write it as appropriate.
|
|
// ========================================
|
|
|
|
dwCount++;
|
|
ULONG ulItemType;
|
|
CLocUniqueId &rUid = isItemSet[0]->GetUniqueId();
|
|
|
|
rUid.GetTypeId().GetId(ulItemType);
|
|
//if (ulItemType == wltClassName)
|
|
//{
|
|
// uiCommentNum = 0;
|
|
// uiReadingOrder = (uiReadingOrder + 999)/1000*1000;
|
|
//}
|
|
for (int i = 0; i < isItemSet.GetSize(); i++)
|
|
{
|
|
isItemSet[i]->SetDisplayOrder(uiReadingOrder);
|
|
uiReadingOrder++;
|
|
}
|
|
|
|
if (fGenerating)
|
|
{
|
|
fRetVal = GenerateItem(ihItemHandler,
|
|
isItemSet, &pOutBuffer, uiStartPos);
|
|
|
|
if (pBuff != pOutBuffer)
|
|
{
|
|
delete pBuff;
|
|
pBuff = NULL;
|
|
pBuff = pOutBuffer; // The old memory has already been deleted.
|
|
}
|
|
else
|
|
{
|
|
fRetVal = FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
fRetVal = EnumerateItem(ihItemHandler,
|
|
isItemSet);
|
|
}
|
|
|
|
isItemSet.ClearItemSet();
|
|
uiTemp += 1;
|
|
|
|
if (!fRetVal)
|
|
{
|
|
fRetVal = TRUE;
|
|
break;
|
|
}
|
|
|
|
wmiRet = GetNextItemSet(dwCount, pBuff, isItemSet, dbidSectionId, uiStartPos);
|
|
if (uiStartPos > uiTemp)
|
|
uiTemp = uiStartPos;
|
|
|
|
if (dwCount%20 == 0)
|
|
{
|
|
if (uiNewPercentage < 100)
|
|
uiNewPercentage++;
|
|
ihItemHandler.SetProgressIndicator(uiNewPercentage);
|
|
}
|
|
}
|
|
|
|
// If we were generating the file,
|
|
// we're done.
|
|
// ==============================
|
|
if (fGenerating)
|
|
break;
|
|
|
|
if (uiNewPercentage < 100)
|
|
uiNewPercentage++;
|
|
ihItemHandler.SetProgressIndicator(uiNewPercentage);
|
|
|
|
}
|
|
|
|
uiTemp = 0;
|
|
|
|
// Now, we get to search and replace the locale IDs,
|
|
// and actually write out the file.
|
|
// =================================================
|
|
|
|
if (fRetVal && fGenerating)
|
|
{
|
|
fRetVal = WriteNewFile(pOutBuffer);
|
|
}
|
|
|
|
}
|
|
|
|
if (pBuff)
|
|
delete pBuff;
|
|
|
|
ihItemHandler.SetProgressIndicator(100);
|
|
|
|
}
|
|
catch (CFileException *pFileException)
|
|
{
|
|
fRetVal = FALSE;
|
|
|
|
ReportFileError(m_pstrFileName, m_didFileId, pFileException, ihItemHandler);
|
|
|
|
pFileException->Delete();
|
|
}
|
|
catch (CUnicodeException *pUnicodeException)
|
|
{
|
|
CLocation loc;
|
|
|
|
loc.SetGlobalId(CGlobalId(m_didFileId, otFile));
|
|
loc.SetView(vProjWindow);
|
|
|
|
ReportUnicodeError(pUnicodeException, ihItemHandler, loc);
|
|
|
|
pUnicodeException->Delete();
|
|
fRetVal = FALSE;
|
|
}
|
|
catch (CMemoryException *pMemoryException)
|
|
{
|
|
CLString strContext;
|
|
|
|
ihItemHandler.IssueMessage(esError, strContext,
|
|
g_hDll, IDS_WMI_NO_MEMORY, g_locNull);
|
|
|
|
fRetVal = FALSE;
|
|
|
|
pMemoryException->Delete();
|
|
}
|
|
catch (CException *pException)
|
|
{
|
|
CLocation loc;
|
|
|
|
loc.SetGlobalId(CGlobalId(m_didFileId, otFile));
|
|
loc.SetView(vProjWindow);
|
|
|
|
ReportException(pException, ihItemHandler, loc);
|
|
|
|
pException->Delete();
|
|
fRetVal = FALSE;
|
|
}
|
|
return fRetVal;
|
|
}
|
|
|
|
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::WriteWaterMark
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::WriteWaterMark()
|
|
{
|
|
LTASSERT(NULL != m_pOpenTargetFile);
|
|
|
|
LTASSERT(NULL != m_pOpenSourceFile);
|
|
|
|
// Do we need to support this?
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetNextQualifierPos
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::GetNextQualifierPos(const wchar_t *wTmp, const wchar_t *pBuff, UINT &uiNewPos, UINT uiStartingPos)
|
|
{
|
|
BOOL bRet = FALSE;
|
|
UINT uiPos = uiStartingPos;
|
|
BOOL bComment = FALSE;
|
|
|
|
if (pBuff && wcslen(pBuff) < uiStartingPos)
|
|
return FALSE;
|
|
|
|
wchar_t *pTemp = (wchar_t *)pBuff;
|
|
pTemp += uiStartingPos;
|
|
|
|
while (TRUE)
|
|
{
|
|
wchar_t *pszTest2 = NULL;
|
|
|
|
pszTest2 = wcsstr(pTemp, L":");
|
|
if (pszTest2)
|
|
{
|
|
uiPos = pszTest2 - pBuff + 1;
|
|
|
|
// Look for the "amended" keyword.
|
|
// ==============================
|
|
|
|
WCHAR temp = pszTest2[0];
|
|
while(temp == L' ' || temp == L'\0' || temp == L':')
|
|
{
|
|
pszTest2++;
|
|
temp = pszTest2[0];
|
|
}
|
|
|
|
if (temp != L'\0')
|
|
{
|
|
wchar_t wTmp2[8];
|
|
wcsncpy(wTmp2, pszTest2, 7);
|
|
wTmp2[7] = '\0';
|
|
if (!wcsicmp(wTmp2, wTmp))
|
|
{
|
|
bRet = TRUE;
|
|
}
|
|
}
|
|
|
|
// If here, we found a non-match, so try again.
|
|
// ============================================
|
|
|
|
if (!bRet)
|
|
pTemp = pszTest2 + 1;
|
|
else
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (bRet)
|
|
uiNewPos = uiPos;
|
|
|
|
return bRet;
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetCurrentNamespace
|
|
//
|
|
//*****************************************************************************
|
|
|
|
wchar_t *CWMILocFile::GetCurrentNamespace(wchar_t *pstr, UINT uPos)
|
|
{
|
|
wchar_t *pTemp = pstr;
|
|
_bstr_t pstrNamespace = m_sCurrentNamespace;
|
|
UINT uiCurrPos = 0;
|
|
BOOL bComment = FALSE;
|
|
|
|
wchar_t wTmp[] = L"#pragma namespace";
|
|
int iHCLen = wcslen(wTmp);
|
|
|
|
// Find the first occurrence of the namespace
|
|
// before the current position.
|
|
|
|
if (pstrNamespace.length() > 0)
|
|
pTemp = wcsstr(pTemp, pstrNamespace); // Jump directly to the existing one.
|
|
|
|
while (uiCurrPos < uPos)
|
|
{
|
|
wchar_t *pszTest2 = NULL;
|
|
|
|
pszTest2 = wcsstr(pTemp, L"#");
|
|
if (pszTest2)
|
|
{
|
|
// First, go back and make sure this isn't a comment line.
|
|
// =======================================================
|
|
bComment = FALSE;
|
|
|
|
wchar_t *pComment = pszTest2;
|
|
while (pComment > pstr)
|
|
{
|
|
if (pComment[0] == L'\n' || pComment[0] == L'\r')
|
|
{
|
|
if (pComment[1] == L'/' && pComment[2] == L'/')
|
|
{
|
|
bComment = TRUE;
|
|
}
|
|
else
|
|
{
|
|
bComment = FALSE;
|
|
}
|
|
break;
|
|
}
|
|
pComment--;
|
|
}
|
|
|
|
if (!bComment)
|
|
{
|
|
|
|
wchar_t wTmp2[100];
|
|
wcsncpy(wTmp2, pszTest2, 17);
|
|
wTmp2[17] = '\0';
|
|
if (!wcsicmp(wTmp2, wTmp))
|
|
{
|
|
uiCurrPos += (pszTest2 - pTemp);
|
|
wchar_t *pszTest3 = wcschr(pszTest2, L')');
|
|
|
|
int iLen = (pszTest3 - pszTest2);
|
|
wchar_t *pszTmpNS = new wchar_t[iLen*2+1];
|
|
if (pszTmpNS)
|
|
{
|
|
pszTest2 += iHCLen + 2; // skip quote and open parent.
|
|
wcsncpy(pszTmpNS, pszTest2, iLen - 2); // strip quotes.
|
|
pszTmpNS[iLen-iHCLen-3] = '\0';
|
|
pstrNamespace = pszTmpNS;
|
|
|
|
pTemp = pszTest2 + 1;
|
|
delete pszTmpNS;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pTemp = pszTest2 + 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pTemp = pszTest2 + 1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
int iLen = wcslen(pstrNamespace) ;
|
|
|
|
wchar_t *pNew = new wchar_t[iLen*2+1];
|
|
if (pNew)
|
|
{
|
|
wcsncpy(pNew, (const wchar_t *)pstrNamespace, iLen);
|
|
pNew[iLen] = '\0';
|
|
}
|
|
|
|
return pNew;
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetNextItemSet
|
|
//
|
|
//*****************************************************************************
|
|
|
|
CWMILocFile::WMIFileError CWMILocFile::GetNextItemSet(
|
|
DWORD dwCurrPos,
|
|
const _bstr_t &pstrCurrentLine,
|
|
CLocItemSet &aNewItem,
|
|
const DBID &dbidSection,
|
|
UINT &uiStartPos)
|
|
{
|
|
|
|
// In this function, we know there is an
|
|
// "amended" keyword in here somewhere.
|
|
// We want to know to which class and/or
|
|
// property does it belong? If we don't
|
|
// have enough data to figure it out,
|
|
// we need to send back a WMIIncompleteObj
|
|
// code.
|
|
// ======================================
|
|
UINT uiCurrPos = 0;
|
|
WMIFileError feRetCode = WMINoError;
|
|
_bstr_t sQualifierName, sRawValue, sPropName, sClassName;
|
|
BOOL bClass = FALSE;
|
|
int iLen = pstrCurrentLine.length() + 1;
|
|
iLen *= 2;
|
|
|
|
// Get the position of the keyword
|
|
// "amended" in this chunk of text.
|
|
|
|
wchar_t *wTemp = new wchar_t[iLen+1];
|
|
if (!wTemp)
|
|
{
|
|
feRetCode = WMIOOM;
|
|
return feRetCode;
|
|
}
|
|
|
|
if (GetNextQualifierPos(L"amended", pstrCurrentLine, uiCurrPos, uiStartPos))
|
|
{
|
|
BOOL bArray = FALSE;
|
|
|
|
uiStartPos = uiCurrPos;
|
|
|
|
// Find the qualifier name and value.
|
|
// wTemp = Top of File
|
|
// wTmp2 = "Amended" keyword
|
|
// wQfrVal = Opening bracket
|
|
// wBkwd = floating pointer.
|
|
|
|
wchar_t *wTmp2 = NULL, *wBkwd = NULL, *wQfrVal = NULL;
|
|
|
|
wcscpy(wTemp, pstrCurrentLine);
|
|
wTemp[iLen] = '\0';
|
|
|
|
wTmp2 = wTemp;
|
|
wTmp2 += (uiCurrPos - 1); // the "Amended" keyword.
|
|
|
|
wQfrVal = FindTop(wTmp2, wTemp, bArray);
|
|
|
|
if (!wQfrVal) // Make sure we had an open parenth
|
|
{
|
|
feRetCode = WMISyntaxError;
|
|
delete wTemp;
|
|
return feRetCode;
|
|
}
|
|
|
|
// Find the beginning of the qualifier name.
|
|
wBkwd = wQfrVal;
|
|
|
|
while (wBkwd[0] != L',' && wBkwd[0] != L'[' && wBkwd >= wTemp)
|
|
{
|
|
wBkwd--;
|
|
}
|
|
|
|
if (wBkwd[0] != L',' && wBkwd[0] != L'[') // Make sure we had a valid qualifier name.
|
|
{
|
|
feRetCode = WMISyntaxError;
|
|
delete wTemp;
|
|
return feRetCode;
|
|
}
|
|
|
|
WCHAR *token;
|
|
UINT uiLen;
|
|
|
|
wBkwd += 1;
|
|
|
|
wchar_t wTmpBuff[256];
|
|
wcsncpy(wTmpBuff, wBkwd, wQfrVal - wBkwd);
|
|
wTmpBuff[wQfrVal - wBkwd] = '\0';
|
|
sQualifierName = wTmpBuff;
|
|
|
|
GetQualifierValue(wTemp, uiStartPos, sRawValue, uiLen);
|
|
|
|
// Finally, populate the CLocItem.
|
|
// ===============================
|
|
|
|
LTASSERT(aNewItem.GetSize() == 0);
|
|
|
|
if (feRetCode == WMINoError)
|
|
{
|
|
CLocItem *pNewItem;
|
|
|
|
try
|
|
{
|
|
// Now we have a value, but it may be an
|
|
// array. If so, we need to add one CLocItem
|
|
// for each value in the array.
|
|
|
|
VectorString arrValues;
|
|
if (bArray)
|
|
ParseArray(sRawValue, arrValues);
|
|
else
|
|
arrValues.push_back(sRawValue);
|
|
|
|
for (int i = 0; i < arrValues.size(); i++)
|
|
{
|
|
wchar_t szTmp[20];
|
|
swprintf(szTmp, L"%ld", dwCurrPos);
|
|
|
|
_bstr_t sValue = arrValues.at(i);
|
|
|
|
pNewItem = new CLocItem;
|
|
|
|
CLocUniqueId uid;
|
|
|
|
CPascalString sTempString;
|
|
|
|
sTempString = sQualifierName;
|
|
sTempString += szTmp;
|
|
|
|
uid.GetResId().SetId(sTempString) ;
|
|
|
|
if (bClass)
|
|
uid.GetTypeId().SetId(wltClassName);
|
|
else
|
|
uid.GetTypeId().SetId(wltPropertyName);
|
|
|
|
uid.SetParentId(dbidSection);
|
|
pNewItem->SetUniqueId(uid);
|
|
|
|
CLocString lsString;
|
|
CPascalString pstrComment, pstrText;
|
|
|
|
pstrText = sValue;
|
|
|
|
pNewItem->SetIconType(CIT::String);
|
|
pNewItem->SetInstructions(pstrComment);
|
|
|
|
lsString.SetString(pstrText);
|
|
SetFlags(pNewItem, lsString);
|
|
pNewItem->SetLocString(lsString);
|
|
|
|
aNewItem.Add(pNewItem);
|
|
}
|
|
|
|
}
|
|
catch (CMemoryException *pMemoryException)
|
|
{
|
|
feRetCode = WMIOOM;
|
|
|
|
pMemoryException->Delete();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
LTTRACE("Unable to process line '%ls'",
|
|
(const WCHAR *)pstrCurrentLine);
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
feRetCode = WMINoMore;
|
|
}
|
|
uiStartPos = uiCurrPos;
|
|
|
|
|
|
delete wTemp;
|
|
return feRetCode;
|
|
}
|
|
|
|
const UINT WMI_MAX_CONTEXT = 256;
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetFullContext
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::GetFullContext(
|
|
CLString &strContext)
|
|
const
|
|
{
|
|
CLString strFormat;
|
|
|
|
strFormat.LoadString(g_hDll, IDS_WMI_FULL_CONTEXT);
|
|
|
|
strContext.Empty();
|
|
|
|
_sntprintf(strContext.GetBuffer(WMI_MAX_CONTEXT), WMI_MAX_CONTEXT,
|
|
(const TCHAR *)strFormat,
|
|
(const WCHAR *)m_pstrFileName, (UINT)m_uiLineNumber);
|
|
strContext.ReleaseBuffer();
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::ReportFileError
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::ReportFileError(
|
|
const _bstr_t &pstrFileName,
|
|
const DBID &didFileId,
|
|
CFileException *pFileException,
|
|
CReporter &Reporter)
|
|
const
|
|
{
|
|
CLString strContext;
|
|
CLString strMessage;
|
|
const UINT MAX_MESSAGE = 256;
|
|
TCHAR szFileErrorMessage[MAX_MESSAGE];
|
|
CLocation loc;
|
|
|
|
pFileException->GetErrorMessage(szFileErrorMessage, MAX_MESSAGE);
|
|
|
|
strMessage.Format(g_hDll, IDS_WMI_BAD_FILE, (const WCHAR *)pstrFileName,
|
|
szFileErrorMessage);
|
|
|
|
GetFullContext(strContext);
|
|
loc.SetGlobalId(CGlobalId(didFileId, otFile));
|
|
loc.SetView(vProjWindow);
|
|
|
|
Reporter.IssueMessage(esError, strContext, strMessage, loc);
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::ReportUnicodeError
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::ReportUnicodeError(
|
|
CUnicodeException *pUnicodeException,
|
|
CReporter &Reporter,
|
|
const CLocation &Location)
|
|
const
|
|
{
|
|
CLString strContext;
|
|
CLString strMessage;
|
|
const UINT MAX_MESSAGE = 256;
|
|
TCHAR szUnicodeErrorMessage[MAX_MESSAGE];
|
|
CLocation loc;
|
|
|
|
pUnicodeException->GetErrorMessage(szUnicodeErrorMessage, MAX_MESSAGE);
|
|
|
|
strMessage.Format(g_hDll, IDS_WMI_UNICODE_ERROR, szUnicodeErrorMessage);
|
|
GetFullContext(strContext);
|
|
|
|
Reporter.IssueMessage(esError, strContext, strMessage, Location,
|
|
IDH_UNICODE_CONV);
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::ReportException
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::ReportException(
|
|
CException *pException,
|
|
CReporter &Reporter,
|
|
const CLocation &Location)
|
|
const
|
|
{
|
|
CLString strContext;
|
|
CLString strMessage;
|
|
const UINT MAX_MESSAGE = 256;
|
|
TCHAR szErrorMessage[MAX_MESSAGE];
|
|
|
|
pException->GetErrorMessage(szErrorMessage, MAX_MESSAGE);
|
|
|
|
strMessage.Format(g_hDll, IDS_WMI_EXCEPTION, szErrorMessage);
|
|
GetFullContext(strContext);
|
|
|
|
Reporter.IssueMessage(esError, strContext, strMessage, Location);
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::GetQualifierValue
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::GetQualifierValue(wchar_t *pBuffer, UINT &uiPos, _bstr_t &sValue, UINT &uiPhysLen)
|
|
{
|
|
|
|
// This needs to read up the text of the qualifier,
|
|
// strip out the quotes and carriage returns, and
|
|
// return it and its *physical* length-in-file.
|
|
|
|
BOOL fRetVal = FALSE;
|
|
BOOL bArray = FALSE;
|
|
|
|
wchar_t *pTemp = pBuffer;
|
|
|
|
pTemp += uiPos;
|
|
|
|
pTemp = FindTop(pTemp, pBuffer, bArray);
|
|
if (pTemp)
|
|
{
|
|
wchar_t tempBuff[4096];
|
|
int iCount = 0;
|
|
|
|
pTemp++; // Step past this character.
|
|
uiPhysLen = 0;
|
|
|
|
WCHAR *token = pTemp;
|
|
BOOL bEnd = FALSE;
|
|
while (!bEnd)
|
|
{
|
|
uiPhysLen++; // Count every character.
|
|
WCHAR *Test;
|
|
|
|
switch(*token)
|
|
{
|
|
case L'\0':
|
|
bEnd = TRUE;
|
|
break;
|
|
case L'\n':
|
|
case L'\r':
|
|
case L'\t':
|
|
break;
|
|
case L'\"':
|
|
if (!iCount)
|
|
break;
|
|
case L')':
|
|
case L'}':
|
|
Test = token - 1;
|
|
while (TRUE)
|
|
{
|
|
if (*Test == L' ' || *Test == L'\r' || *Test == L'\n' || *Test == L'\t')
|
|
{
|
|
Test--;
|
|
continue;
|
|
}
|
|
if (*Test == L'\"')
|
|
{
|
|
bEnd = TRUE;
|
|
}
|
|
else
|
|
{
|
|
tempBuff[iCount] = *token;
|
|
iCount++;
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
default:
|
|
tempBuff[iCount] = *token;
|
|
iCount++;
|
|
break;
|
|
|
|
}
|
|
token++;
|
|
}
|
|
if (tempBuff[iCount-1] == L'\"')
|
|
tempBuff[iCount-1] = '\0';
|
|
else
|
|
tempBuff[iCount] = '\0';
|
|
sValue = tempBuff;
|
|
fRetVal = TRUE;
|
|
}
|
|
uiPhysLen -= 1; // We want to keep the closing parenth.
|
|
|
|
return fRetVal;
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::SetQualifierValue
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::SetQualifierValue(wchar_t *pIn, wchar_t **pOut, UINT &uiPos, _bstr_t &sValue, UINT &uiLen, BOOL bQuotes)
|
|
{
|
|
// This needs to write the localized qualifier value
|
|
// and erase *uiLen* characters.
|
|
// uiPos will need to be updated with the *new*
|
|
// position of this qualifier.
|
|
|
|
BOOL fRetVal = FALSE;
|
|
wchar_t *pStart = pIn + uiPos;
|
|
BOOL bArray = FALSE;
|
|
|
|
pStart = FindTop(pStart, pIn, bArray);
|
|
if (pStart)
|
|
{
|
|
int iNewLen = wcslen(sValue);
|
|
int iLen = wcslen(pIn) + 3;
|
|
if (iNewLen > uiLen) // The length of the new buffer
|
|
iLen += (iNewLen - uiLen); // If the new value is longer, add it.
|
|
|
|
pStart++; // jump past the '(' character. uiLen starts now.
|
|
int iPos = pStart-pIn; // The current position.
|
|
|
|
iLen *= 2;
|
|
wchar_t *pNew = new wchar_t[iLen+3];
|
|
|
|
if (pNew)
|
|
{
|
|
int iTempPos = 0;
|
|
|
|
wcsncpy(pNew, pIn, iPos); // Copy the initial part of the file.
|
|
if (bQuotes)
|
|
pNew[iPos] = '\"';
|
|
pNew[iPos+1] = '\0'; // Null terminate
|
|
|
|
wcscat(pNew, sValue); // Add the new value.
|
|
|
|
iPos += 1 + wcslen(sValue); // Jump past the value
|
|
if (bQuotes)
|
|
pNew[iPos] = '\"';
|
|
pNew[iPos+1] = '\0'; // Null terminate the value.
|
|
|
|
pStart += uiLen; // Jump past the current value.
|
|
|
|
iTempPos = iPos;
|
|
iPos = wcslen(pIn) - (pStart-pIn); // Calculate the length of the rest of the file.
|
|
|
|
wcsncat(pNew, pStart, iPos); // Append the rest of the file to the new buffer.
|
|
|
|
pStart = pNew + iLen;
|
|
pStart = FindPrevious(pStart, L";", pNew);
|
|
pStart[1] = L'\r';
|
|
pStart[2] = L'\n';
|
|
pStart[3] = L'\0';
|
|
|
|
*pOut = pNew;
|
|
|
|
fRetVal = TRUE;
|
|
}
|
|
}
|
|
|
|
// Adjust the position.
|
|
|
|
int iNewLen = wcslen(sValue);
|
|
if (iNewLen < uiLen)
|
|
uiPos -= (uiLen - iNewLen);
|
|
else
|
|
uiPos += (iNewLen - uiLen);
|
|
uiPos += 3;
|
|
|
|
return fRetVal;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::WriteNewFile
|
|
//
|
|
//*****************************************************************************
|
|
|
|
BOOL CWMILocFile::WriteNewFile(wchar_t *pBuffer)
|
|
{
|
|
// This needs to seek and replace all instances of the
|
|
// original Locale with the new one.
|
|
// ===================================================
|
|
|
|
BOOL fRetVal = FALSE, fSuccess = TRUE;
|
|
UINT uiPos = 0, uiStartingPos = 0;
|
|
int uiLen = wcslen(pBuffer);
|
|
|
|
_bstr_t sThisLocale, sTargetLocale;
|
|
wchar_t wOldCodePage[30], wNewCodePage[30];
|
|
swprintf(wOldCodePage, L"_%X", m_wSourceId );
|
|
swprintf(wNewCodePage, L"_%X", m_wTargetId );
|
|
|
|
if (m_wSourceId != m_wTargetId)
|
|
{
|
|
|
|
wchar_t *pLocale = wcsstr(pBuffer, wOldCodePage);
|
|
while (pLocale != NULL)
|
|
{
|
|
for (int i = 0; i < wcslen(wOldCodePage); i++)
|
|
{
|
|
pLocale[i] = wNewCodePage[i];
|
|
}
|
|
|
|
pLocale = wcsstr(pLocale, wOldCodePage);
|
|
}
|
|
|
|
// Now look for the locale if
|
|
// it was converted to a decimal.
|
|
// ==============================
|
|
|
|
swprintf(wOldCodePage, L"(%ld)", m_wSourceId );
|
|
swprintf(wNewCodePage, L"(%ld)", m_wTargetId );
|
|
|
|
pLocale = wcsstr(pBuffer, wOldCodePage);
|
|
while (pLocale != NULL)
|
|
{
|
|
for (int i = 0; i < wcslen(wOldCodePage); i++)
|
|
{
|
|
pLocale[i] = wNewCodePage[i];
|
|
}
|
|
|
|
pLocale = wcsstr(pLocale, wOldCodePage);
|
|
}
|
|
}
|
|
|
|
if (fSuccess)
|
|
{
|
|
fRetVal = TRUE;
|
|
|
|
// Finally, write out the buffer to a brand new file
|
|
// =================================================
|
|
|
|
while (uiLen >= 0)
|
|
{
|
|
if (fwrite(pBuffer, sizeof(wchar_t), (uiLen > 4096) ? 4096: uiLen, m_pOpenTargetFile) < 0)
|
|
{
|
|
fRetVal = FALSE;
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
fRetVal = TRUE;
|
|
pBuffer += 4096;
|
|
uiLen -= 4096;
|
|
}
|
|
|
|
fflush(m_pOpenTargetFile);
|
|
}
|
|
}
|
|
|
|
return fRetVal;
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::FindPrevious
|
|
//
|
|
//*****************************************************************************
|
|
|
|
wchar_t *CWMILocFile::FindPrevious(wchar_t *pBuffer, const wchar_t *pFind, const wchar_t *pTop)
|
|
{
|
|
|
|
wchar_t *pRet = NULL;
|
|
WCHAR t1, t2;
|
|
int iLen = wcslen(pFind);
|
|
BOOL bFound = FALSE;
|
|
|
|
pRet = pBuffer;
|
|
while (pRet >= pTop)
|
|
{
|
|
t2 = pRet[0];
|
|
for (int i = 0; i < iLen; i++)
|
|
{
|
|
t1 = pFind[i];
|
|
|
|
if (t1 == t2)
|
|
{
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (bFound)
|
|
break;
|
|
|
|
pRet--;
|
|
}
|
|
|
|
if (pRet <= pTop)
|
|
pRet = NULL;
|
|
|
|
return pRet;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::FindTop
|
|
//
|
|
//*****************************************************************************
|
|
|
|
wchar_t *CWMILocFile::FindTop(wchar_t *wTmp2, wchar_t *wTop, BOOL &bArray)
|
|
{
|
|
|
|
wchar_t *wQfrVal = FindPrevious(wTmp2, L"({", wTop);
|
|
|
|
while (wQfrVal)
|
|
{
|
|
WCHAR *pQT = wQfrVal + 1;
|
|
BOOL bFound = FALSE;
|
|
|
|
while (TRUE)
|
|
{
|
|
if (*pQT != L' ' && *pQT != L'\t' && *pQT != L'\r' && *pQT != L'\n')
|
|
{
|
|
if (*pQT == L'\"')
|
|
{
|
|
bFound = TRUE;
|
|
}
|
|
break;
|
|
}
|
|
pQT++;
|
|
}
|
|
|
|
if (!bFound)
|
|
{
|
|
wQfrVal --;
|
|
wQfrVal = FindPrevious(wQfrVal, L"({", wTop);
|
|
}
|
|
else
|
|
break;
|
|
}
|
|
|
|
if (wQfrVal)
|
|
{
|
|
if (wQfrVal[0] == L'{')
|
|
bArray = TRUE;
|
|
}
|
|
|
|
return wQfrVal;
|
|
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CWMILocFile::ParseArray
|
|
//
|
|
//*****************************************************************************
|
|
|
|
void CWMILocFile::ParseArray(wchar_t *pIn, VectorString &arrOut)
|
|
{
|
|
|
|
wchar_t Buff[4096];
|
|
wchar_t *pLast = pIn;
|
|
if (*pLast == L'\"')
|
|
pLast++;
|
|
wchar_t *pFind = wcsstr(pIn, L"\",");
|
|
|
|
arrOut.clear();
|
|
|
|
while (pFind)
|
|
{
|
|
wchar_t temp = pFind[-1];
|
|
if (temp == '\\')
|
|
{
|
|
pFind++;
|
|
pFind = wcsstr(pFind, L"\",");
|
|
continue;
|
|
}
|
|
|
|
wcsncpy(Buff, pLast, pFind-pLast);
|
|
Buff[pFind-pLast] = '\0';
|
|
|
|
arrOut.push_back(_bstr_t(Buff));
|
|
|
|
// Now move pFind to the next valid char.
|
|
|
|
while (pFind[0] == L'\n' ||
|
|
pFind[0] == L'\r' ||
|
|
pFind[0] == L' ' ||
|
|
pFind[0] == L',' ||
|
|
pFind[0] == L'\"' )
|
|
pFind++;
|
|
|
|
pLast = pFind ;
|
|
pFind = wcsstr(pFind, L"\",");
|
|
}
|
|
|
|
wcscpy(Buff, pLast);
|
|
|
|
if (Buff[wcslen(Buff)-1] == L'\"')
|
|
Buff[wcslen(Buff)-1] = L'\0'; // strip off that trailing quote.
|
|
else
|
|
Buff[wcslen(Buff)] = L'\0'; // strip off that trailing quote.
|
|
arrOut.push_back(_bstr_t(Buff));
|
|
|
|
return;
|
|
}
|
|
|
|
//*****************************************************************************
|
|
//
|
|
// CVC::ValidateString
|
|
//
|
|
//*****************************************************************************
|
|
|
|
CVC::ValidationCode ValidateString(
|
|
const CLocTypeId &,
|
|
const CLocString &clsOutputLine,
|
|
CReporter &repReporter,
|
|
const CLocation &loc,
|
|
const CLString &strContext)
|
|
{
|
|
CVC::ValidationCode vcRetVal = CVC::NoError;
|
|
CLString strMyContext = strContext;
|
|
|
|
if (strMyContext.GetLength() == 0)
|
|
{
|
|
strMyContext.LoadString(g_hDll, IDS_WMI_GENERIC_LOCATION);
|
|
}
|
|
|
|
loc; repReporter; clsOutputLine;
|
|
|
|
/*
|
|
if (clsOutputLine.HasHotKey())
|
|
{
|
|
vcRetVal = CVC::UpgradeValue(vcRetVal, CVC::Warning);
|
|
repReporter.IssueMessage(esWarning, strMyContext, g_hDll,
|
|
IDS_WMI_VAL_HOTKEY, loc);
|
|
}
|
|
|
|
_bstr_t pstrBadChars;
|
|
UINT uiBadPos;
|
|
|
|
pstrBadChars.SetString(L"\n\ra", (UINT)3);
|
|
|
|
if (pstrOutput.FindOneOf(pstrBadChars, 0, uiBadPos))
|
|
{
|
|
vcRetVal = CVC::UpgradeValue(vcRetVal, CVC::Error);
|
|
|
|
repReporter.IssueMessage(esError, strMyContext, g_hDll,
|
|
IDS_WMI_VAL_BAD_CHARS, loc);
|
|
}
|
|
*/
|
|
return vcRetVal;
|
|
}
|