IXMLDOMNode* SettingsXML::FindItem(IXMLDOMNode* apFrom, const wchar_t* asType, const wchar_t* asName, bool abAllowCreate)
{
HRESULT hr = S_OK;
IXMLDOMNodeList* pList = NULL;
IXMLDOMNode* pChild = NULL;
IXMLDOMNamedNodeMap* pAttrs = NULL;
IXMLDOMAttribute *pIXMLDOMAttribute = NULL;
IXMLDOMNode *pIXMLDOMNode = NULL;
IXMLDOMNode *pName = NULL;
BSTR bsText = NULL;
BSTR bsCheck = NULL;
DOMNodeType nodeTypeCheck = NODE_INVALID;
BOOL lbEmpty = TRUE;
int iLastIndent = 1;
// Получить все дочерние элементы нужного типа
if (apFrom == NULL)
{
hr = S_FALSE;
}
else
{
long lFound = 0;
// key[@name="abc"], but it is case-sensitive, and may fails in theory
bsText = lstrmerge(asType, L"[@name=\"", asName, L"\"]");
hr = apFrom->selectNodes(bsText, &pList);
if (SUCCEEDED(hr) && pList)
{
hr = pList->get_length(&lFound);
if (FAILED(hr) || (lFound < 1))
{
SafeRelease(pList);
}
}
SafeFree(bsText);
// May be case-insensitive search will be succeeded?
// However, it is very slow
if (!pList)
{
bsText = ::SysAllocString(asType);
hr = apFrom->selectNodes(bsText, &pList);
::SysFreeString(bsText); bsText = NULL;
}
}
if (SUCCEEDED(hr) && pList)
{
hr = pList->reset();
while ((hr = pList->nextNode(&pIXMLDOMNode)) == S_OK && pIXMLDOMNode)
{
lbEmpty = FALSE;
hr = pIXMLDOMNode->get_attributes(&pAttrs);
if (SUCCEEDED(hr) && pAttrs)
{
bsText = GetAttr(pIXMLDOMNode, pAttrs, L"name");
if (bsText)
{
if (lstrcmpi(bsText, asName) == 0)
{
::SysFreeString(bsText); bsText = NULL;
pChild = pIXMLDOMNode; pIXMLDOMNode = NULL;
break;
}
::SysFreeString(bsText); bsText = NULL;
}
}
pIXMLDOMNode->Release(); pIXMLDOMNode = NULL;
}
pList->Release();
//pList = NULL; -- для отладки
}
if (lbEmpty && abAllowCreate && (asType[0] == L'k'))
{
bsText = ::SysAllocString(L"value");
hr = apFrom->selectNodes(bsText, &pList);
::SysFreeString(bsText); bsText = NULL;
if (SUCCEEDED(hr) && pList)
{
hr = pList->reset();
if ((hr = pList->nextNode(&pIXMLDOMNode)) == S_OK && pIXMLDOMNode)
{
lbEmpty = FALSE;
pIXMLDOMNode->Release(); pIXMLDOMNode = NULL;
}
pList->Release();
//pList = NULL; -- для отладки
}
}
if (!pChild && abAllowCreate)
{
if (asType[0] == L'k')
{
hr = apFrom->get_lastChild(&pChild);
if (SUCCEEDED(hr) && pChild)
{
hr = pChild->get_nodeType(&nodeTypeCheck);
if (SUCCEEDED(hr) && (nodeTypeCheck == NODE_TEXT))
{
hr = pChild->get_text(&bsCheck);
if (SUCCEEDED(hr) && bsCheck)
{
iLastIndent = 0;
LPCWSTR pszTabs = bsCheck;
while (*pszTabs)
{
if (*(pszTabs++) == L'\t')
iLastIndent++;
}
::SysFreeString(bsCheck); bsCheck = NULL;
}
}
}
SafeRelease(pChild);
}
VARIANT vtType; vtType.vt = VT_I4;
vtType.lVal = NODE_ELEMENT;
bsText = ::SysAllocString(asType);
hr = mp_File->createNode(vtType, bsText, L"", &pChild);
::SysFreeString(bsText); bsText = NULL;
if (SUCCEEDED(hr) && pChild)
{
if (SetAttr(pChild, L"name", asName))
{
if (asType[0] == L'k')
{
AppendNewLine(pChild);
mb_KeyEmpty = true;
TouchKey(pChild);
}
if (asType[0] == L'k')
{
//if (mb_KeyEmpty)
//AppendIndent(apFrom, lbEmpty ? (mi_Level-1) : mi_Level);
AppendIndent(apFrom, (mi_Level-iLastIndent));
}
else if (mb_KeyEmpty)
{
AppendIndent(apFrom, !lbEmpty ? (mi_Level-1) : mi_Level);
}
else
{
AppendIndent(apFrom, 1);
}
hr = apFrom->appendChild(pChild, &pIXMLDOMNode);
pChild->Release(); pChild = NULL;
if (FAILED(hr))
{
pAttrs->Release(); pAttrs = NULL;
}
else
{
pChild = pIXMLDOMNode;
pIXMLDOMNode = NULL;
}
AppendNewLine(apFrom);
AppendIndent(apFrom, mi_Level-1);
if ((asType[0] != L'k') && mb_KeyEmpty)
mb_KeyEmpty = false;
}
else
{
pChild->Release(); pChild = NULL;
}
}
}
return pChild;
}
/**Read the XML config file. Currently contains keyboard choices.*/
void read_config_file()
{
TrainerConfig *result = new TrainerConfig();
CoInitialize(NULL);
//read XML
MSXML2::IXMLDOMDocumentPtr spXMLDoc;
spXMLDoc.CreateInstance(__uuidof(MSXML2::DOMDocument60));
if (!spXMLDoc->load("ent-config.xml"))
{
write_text_to_log_file("No config found, using defaults");
config = result; //the default config
}
IXMLDOMNodeListPtr nodes = spXMLDoc->selectNodes(L"//ent-config/keys/key");
long length;
nodes->get_length(&length);
for (int i = 0; i < length; i++)
{
IXMLDOMNode *node;
nodes->get_item(i, &node);
IXMLDOMNamedNodeMap *attribs;
node->get_attributes(&attribs);
long length_attribs;
attribs->get_length(&length_attribs);
char *attrib_key_func = NULL;
char *attrib_key_value = NULL;
for (long j = 0; j < length_attribs; j++)
{
IXMLDOMNode *attribNode;
attribs->get_item(j, &attribNode);
attribNode->get_nodeName(&bstr);
if (wcscmp(bstr, L"function") == 0)
{
VARIANT var;
VariantInit(&var);
attribNode->get_nodeValue(&var);
attrib_key_func = _com_util::ConvertBSTRToString(V_BSTR(&var));
}
else if (wcscmp(bstr, L"value") == 0)
{
VARIANT var;
VariantInit(&var);
attribNode->get_nodeValue(&var);
attrib_key_value = _com_util::ConvertBSTRToString(V_BSTR(&var));
}
SysFreeString(bstr);
attribNode->Release();
}
if (attrib_key_func != NULL && attrib_key_value != NULL)
{
result->get_key_config()->set_key(attrib_key_func, attrib_key_value);
}
delete attrib_key_func;
delete attrib_key_value;
attribs->Release();
node->Release();
}
//nodes->Release(); //don't do this, it crashes on exit
spXMLDoc.Release();
CoUninitialize();
config = result;
}
IXMLDOMNode* SettingsXML::FindItem(IXMLDOMNode* apFrom, const wchar_t* asType, const wchar_t* asName, bool abAllowCreate)
{
HRESULT hr = S_OK;
IXMLDOMNodeList* pList = NULL;
IXMLDOMNode* pChild = NULL;
IXMLDOMNamedNodeMap* pAttrs = NULL;
IXMLDOMAttribute *pIXMLDOMAttribute = NULL;
IXMLDOMNode *pIXMLDOMNode = NULL;
IXMLDOMNode *pName = NULL;
BSTR bsText = NULL;
BOOL lbEmpty = TRUE;
// Получить все дочерние элементы нужного типа
if (apFrom == NULL)
{
hr = S_FALSE;
}
else
{
bsText = ::SysAllocString(asType);
hr = apFrom->selectNodes(bsText, &pList);
::SysFreeString(bsText); bsText = NULL;
}
if (SUCCEEDED(hr) && pList)
{
hr = pList->reset();
while((hr = pList->nextNode(&pIXMLDOMNode)) == S_OK && pIXMLDOMNode)
{
lbEmpty = FALSE;
hr = pIXMLDOMNode->get_attributes(&pAttrs);
if (SUCCEEDED(hr) && pAttrs)
{
bsText = GetAttr(pIXMLDOMNode, pAttrs, L"name");
if (bsText)
{
if (lstrcmpi(bsText, asName) == 0)
{
::SysFreeString(bsText); bsText = NULL;
pChild = pIXMLDOMNode; pIXMLDOMNode = NULL;
break;
}
::SysFreeString(bsText); bsText = NULL;
}
}
pIXMLDOMNode->Release(); pIXMLDOMNode = NULL;
}
pList->Release();
//pList = NULL; -- для отладки
}
if (lbEmpty && abAllowCreate && (asType[0] == L'k'))
{
bsText = ::SysAllocString(L"value");
hr = apFrom->selectNodes(bsText, &pList);
::SysFreeString(bsText); bsText = NULL;
if (SUCCEEDED(hr) && pList)
{
hr = pList->reset();
if ((hr = pList->nextNode(&pIXMLDOMNode)) == S_OK && pIXMLDOMNode)
{
lbEmpty = FALSE;
pIXMLDOMNode->Release(); pIXMLDOMNode = NULL;
}
pList->Release();
//pList = NULL; -- для отладки
}
}
if (!pChild && abAllowCreate)
{
VARIANT vtType; vtType.vt = VT_I4;
vtType.lVal = NODE_ELEMENT;
bsText = ::SysAllocString(asType);
hr = mp_File->createNode(vtType, bsText, L"", &pChild);
::SysFreeString(bsText); bsText = NULL;
if (SUCCEEDED(hr) && pChild)
{
if (SetAttr(pChild, L"name", asName))
{
if (asType[0] == L'k')
{
AppendNewLine(pChild);
mb_KeyEmpty = true;
}
if (asType[0] == L'k')
{
//if (mb_KeyEmpty)
//AppendIndent(apFrom, lbEmpty ? (mi_Level-1) : mi_Level);
AppendIndent(apFrom, (mi_Level-1));
}
else if (mb_KeyEmpty)
{
AppendIndent(apFrom, !lbEmpty ? (mi_Level-1) : mi_Level);
}
else
{
AppendIndent(apFrom, 1);
}
hr = apFrom->appendChild(pChild, &pIXMLDOMNode);
pChild->Release(); pChild = NULL;
if (FAILED(hr))
{
pAttrs->Release(); pAttrs = NULL;
}
else
{
pChild = pIXMLDOMNode;
pIXMLDOMNode = NULL;
}
AppendNewLine(apFrom);
AppendIndent(apFrom, mi_Level-1);
if ((asType[0] != L'k') && mb_KeyEmpty)
mb_KeyEmpty = false;
}
else
{
pChild->Release(); pChild = NULL;
}
}
}
return pChild;
}
//void SettingsXML::Save(const wchar_t *regName, const wchar_t *value)
//{
// if (!value) value = L""; // сюда мог придти и NULL
//
// Save(regName, (LPCBYTE)value, REG_SZ, (_tcslen(value)+1)*sizeof(wchar_t));
//}
void SettingsXML::Save(const wchar_t *regName, LPCBYTE value, DWORD nType, DWORD nSize)
{
HRESULT hr = S_OK;
IXMLDOMNamedNodeMap* pAttrs = NULL;
IXMLDOMNodeList* pList = NULL;
IXMLDOMAttribute *pIXMLDOMAttribute = NULL;
IXMLDOMNode *pNode = NULL;
IXMLDOMNode* pChild = NULL;
IXMLDOMNode *pNodeRmv = NULL;
BSTR bsValue = NULL;
BSTR bsType = NULL;
bool bNeedSetType = false;
// nType:
// REG_DWORD: сохранение числа в 16-ричном или 10-чном формате, в зависимости от того, что сейчас указано в xml ("dword"/"ulong"/"long")
// REG_BINARY: строго в hex (FF,FF,...)
// REG_SZ: ASCIIZ строка, можно проконтролировать, чтобы nSize/2 не был меньше длины строки
// REG_MULTI_SZ: ASCIIZZ. При формировании <list...> нужно убедиться, что мы не вылезли за пределы nSize
pChild = FindItem(mp_Key, L"value", regName, true); // создать, если его еще нету
if (!pChild)
goto wrap;
hr = pChild->get_attributes(&pAttrs);
if (FAILED(hr) || !pAttrs)
goto wrap;
bsType = GetAttr(pChild, pAttrs, L"type");
switch(nType)
{
case REG_DWORD:
{
wchar_t szValue[32];
if (bsType && (bsType[0] == L'u' || bsType[0] == L'U'))
{
_wsprintf(szValue, SKIPLEN(countof(szValue)) L"%u", *(LPDWORD)value);
}
else if (bsType && (bsType[0] == L'l' || bsType[0] == L'L'))
{
_wsprintf(szValue, SKIPLEN(countof(szValue)) L"%i", *(int*)value);
}
else
{
_wsprintf(szValue, SKIPLEN(countof(szValue)) L"%08x", *(LPDWORD)value);
if (bsType) ::SysFreeString(bsType);
// нужно добавить/установить тип
bsType = ::SysAllocString(L"dword"); bNeedSetType = true;
}
bsValue = ::SysAllocString(szValue);
} break;
case REG_BINARY:
{
if (nSize == 1 && bsType && (bsType[0] == L'u' || bsType[0] == L'U'))
{
wchar_t szValue[4];
BYTE bt = *value;
_wsprintf(szValue, SKIPLEN(countof(szValue)) L"%u", (DWORD)bt);
bsValue = ::SysAllocString(szValue);
}
else if (nSize == 1 && bsType && (bsType[0] == L'l' || bsType[0] == L'L'))
{
wchar_t szValue[4];
char bt = *value;
_wsprintf(szValue, SKIPLEN(countof(szValue)) L"%i", (int)bt);
bsValue = ::SysAllocString(szValue);
}
else
{
DWORD nLen = nSize*2 + (nSize-1); // по 2 символа на байт + ',' между ними
bsValue = ::SysAllocStringLen(NULL, nLen);
nLen ++; // Чтобы далее не добавлять WCHAR на '\0'
wchar_t* psz = (wchar_t*)bsValue;
LPCBYTE ptr = value;
while(nSize)
{
_wsprintf(psz, SKIPLEN(nLen-(psz-bsValue)) L"%02x", (DWORD)*ptr);
ptr++; nSize--; psz+=2;
if (nSize)
*(psz++) = L',';
}
if (bsType && lstrcmp(bsType, L"hex"))
{
// Допустим только "hex"
::SysFreeString(bsType); bsType = NULL;
}
if (!bsType)
{
// нужно добавить/установить тип
bsType = ::SysAllocString(L"hex"); bNeedSetType = true;
}
}
} break;
case REG_SZ:
{
wchar_t* psz = (wchar_t*)value;
bsValue = ::SysAllocString(psz);
if (bsType && lstrcmp(bsType, L"string"))
{
// Допустим только "string"
::SysFreeString(bsType); bsType = NULL;
}
if (!bsType)
{
// нужно добавить/установить тип
bsType = ::SysAllocString(L"string"); bNeedSetType = true;
}
} break;
case REG_MULTI_SZ:
{
if (bsType && lstrcmp(bsType, L"multi"))
{
// Допустим только "multi"
::SysFreeString(bsType); bsType = NULL;
}
if (!bsType)
{
// нужно добавить/установить тип
bsType = ::SysAllocString(L"multi"); bNeedSetType = true;
}
} break;
default:
goto wrap; // не поддерживается
}
if (bNeedSetType)
{
_ASSERTE(bsType!=NULL);
SetAttr(pChild, pAttrs, L"type", bsType);
::SysFreeString(bsType); bsType = NULL;
}
// Теперь собственно значение
if (nType != REG_MULTI_SZ)
{
_ASSERTE(bsValue != NULL);
SetAttr(pChild, pAttrs, L"data", bsValue);
::SysFreeString(bsValue); bsValue = NULL;
}
else // Тут нужно формировать список элементов <list>
{
VARIANT_BOOL bHasChild = VARIANT_FALSE;
DOMNodeType nodeType = NODE_INVALID;
// Если ранее был параметр "data" - удалить его из списка атрибутов
hr = pAttrs->getNamedItem(L"data", &pNode);
if (SUCCEEDED(hr) && pNode)
{
hr = pChild->removeChild(pNode, &pNodeRmv);
pNode->Release(); pNode = NULL;
if (pNodeRmv) { pNodeRmv->Release(); pNodeRmv = NULL; }
}
//TODO: может оставить перевод строки?
// Сначала почистим
#ifdef _DEBUG
hr = pChild->get_nodeType(&nodeType);
#endif
hr = pChild->hasChildNodes(&bHasChild);
if (bHasChild)
{
while((hr = pChild->get_firstChild(&pNode)) == S_OK && pNode)
{
hr = pNode->get_nodeType(&nodeType);
#ifdef _DEBUG
BSTR bsDebug = NULL;
pNode->get_text(&bsDebug);
if (bsDebug) ::SysFreeString(bsDebug); bsDebug = NULL;
#endif
hr = pChild->removeChild(pNode, &pNodeRmv);
if (pNodeRmv) { pNodeRmv->Release(); pNodeRmv = NULL; }
pNode->Release(); pNode = NULL;
}
}
// Теперь - добавляем список
wchar_t* psz = (wchar_t*)value;
BSTR bsNodeType = ::SysAllocString(L"line");
VARIANT vtType; vtType.vt = VT_I4; vtType.lVal = NODE_ELEMENT;
long nAllLen = nSize/2; // длина в wchar_t
long nLen = 0;
while(psz && *psz && nAllLen > 0)
{
hr = mp_File->createNode(vtType, bsNodeType, L"", &pNode);
if (FAILED(hr) || !pNode)
break;
if (!SetAttr(pNode, L"data", psz))
break;
hr = pChild->appendChild(pNode, &pNodeRmv);
pNode->Release(); pNode = NULL;
if (pNodeRmv) { pNodeRmv->Release(); pNodeRmv = NULL; }
if (FAILED(hr))
break;
nLen = _tcslen(psz)+1;
psz += nLen;
nAllLen -= nLen;
}
_ASSERTE(nAllLen <= 1);
}
mb_Modified = true;
wrap:
if (pIXMLDOMAttribute) { pIXMLDOMAttribute->Release(); pIXMLDOMAttribute = NULL; }
if (pNode) { pNode->Release(); pNode = NULL; }
if (pNodeRmv) { pNodeRmv->Release(); pNodeRmv = NULL; }
if (pChild) { pChild->Release(); pChild = NULL; }
if (pAttrs) { pAttrs->Release(); pAttrs = NULL; }
if (bsValue) { ::SysFreeString(bsValue); bsValue = NULL; }
if (bsType) { ::SysFreeString(bsType); bsType = NULL; }
}
bool SettingsXML::Load(const wchar_t *regName, LPBYTE value, DWORD nSize)
{
bool lbRc = false;
HRESULT hr = S_OK;
IXMLDOMNode* pChild = NULL;
IXMLDOMNamedNodeMap* pAttrs = NULL;
IXMLDOMAttribute *pIXMLDOMAttribute = NULL;
IXMLDOMNode *pNode = NULL;
BSTR bsType = NULL;
BSTR bsData = NULL;
if (!value || !nSize)
return false;
if (mp_Key)
pChild = FindItem(mp_Key, L"value", regName, false);
if (!pChild)
return false;
hr = pChild->get_attributes(&pAttrs);
if (SUCCEEDED(hr) && pAttrs)
{
bsType = GetAttr(pChild, pAttrs, L"type");
}
if (SUCCEEDED(hr) && bsType)
{
bsData = GetAttr(pChild, pAttrs, L"data");
}
if (SUCCEEDED(hr) && bsData)
{
if (!lstrcmpi(bsType, L"string"))
{
#ifdef _DEBUG
DWORD nLen = _tcslen(bsData) + 1;
#endif
DWORD nMaxLen = nSize / 2;
lstrcpyn((wchar_t*)value, bsData, nMaxLen);
lbRc = true;
}
else if (!lstrcmpi(bsType, L"ulong"))
{
wchar_t* pszEnd = NULL;
DWORD lVal = wcstoul(bsData, &pszEnd, 10);
if (nSize > 4) nSize = 4;
if (pszEnd && pszEnd != bsData)
{
memmove(value, &lVal, nSize);
lbRc = true;
}
}
else if (!lstrcmpi(bsType, L"long"))
{
wchar_t* pszEnd = NULL;
int lVal = wcstol(bsData, &pszEnd, 10);
if (nSize > 4) nSize = 4;
if (pszEnd && pszEnd != bsData)
{
memmove(value, &lVal, nSize);
lbRc = true;
}
}
else if (!lstrcmpi(bsType, L"dword"))
{
wchar_t* pszEnd = NULL;
DWORD lVal = wcstoul(bsData, &pszEnd, 16);
if (nSize > 4) nSize = 4;
if (pszEnd && pszEnd != bsData)
{
memmove(value, &lVal, nSize);
lbRc = true;
}
}
else if (!lstrcmpi(bsType, L"hex"))
{
wchar_t* pszCur = bsData;
wchar_t* pszEnd = NULL;
LPBYTE pCur = value;
wchar_t cHex;
DWORD lVal = 0;
lbRc = true;
while (*pszCur && nSize)
{
lVal = 0;
cHex = *(pszCur++);
if (cHex >= L'0' && cHex <= L'9')
{
lVal = cHex - L'0';
}
else if (cHex >= L'a' && cHex <= L'f')
{
lVal = cHex - L'a' + 10;
}
else if (cHex >= L'A' && cHex <= L'F')
{
lVal = cHex - L'A' + 10;
}
else
{
lbRc = false; break;
}
cHex = *(pszCur++);
if (cHex && cHex != L',')
{
lVal = lVal << 4;
if (cHex >= L'0' && cHex <= L'9')
{
lVal |= cHex - L'0';
}
else if (cHex >= L'a' && cHex <= L'f')
{
lVal |= cHex - L'a' + 10;
}
else if (cHex >= L'A' && cHex <= L'F')
{
lVal |= cHex - L'A' + 10;
}
else
{
lbRc = false; break;
}
cHex = *(pszCur++);
}
*pCur = (BYTE)lVal;
pCur++; nSize--;
if (cHex != L',')
{
break;
}
}
while(nSize--) // очистить хвост
*(pCur++) = 0;
}
}
// Остальные типы (строки) - не интересуют
if (bsType) { ::SysFreeString(bsType); bsType = NULL; }
if (bsData) { ::SysFreeString(bsData); bsData = NULL; }
if (pChild) { pChild->Release(); pChild = NULL; }
if (pAttrs) { pAttrs->Release(); pAttrs = NULL; }
return lbRc;
}
// эта функция, если значения нет (или тип некорректный) *value НЕ трогает
bool SettingsXML::Load(const wchar_t *regName, wchar_t **value)
{
bool lbRc = false;
HRESULT hr = S_OK;
IXMLDOMNode* pChild = NULL;
IXMLDOMNamedNodeMap* pAttrs = NULL;
IXMLDOMAttribute *pIXMLDOMAttribute = NULL;
IXMLDOMNode *pNode = NULL;
IXMLDOMNodeList* pList = NULL;
BSTR bsType = NULL;
BSTR bsData = NULL;
size_t nLen = 0;
//if (*value) {free(*value); *value = NULL;}
if (mp_Key)
pChild = FindItem(mp_Key, L"value", regName, false);
if (!pChild)
return false;
hr = pChild->get_attributes(&pAttrs);
if (SUCCEEDED(hr) && pAttrs)
{
bsType = GetAttr(pChild, pAttrs, L"type");
}
if (SUCCEEDED(hr) && bsType)
{
if (!lstrcmpi(bsType, L"multi"))
{
// Тут значения хранятся так:
//<value name="CmdLineHistory" type="multi">
// <line data="C:\Far\Far.exe"/>
// <line data="cmd"/>
//</value>
wchar_t *pszData = NULL, *pszCur = NULL;
size_t nMaxLen = 0, nCurLen = 0;
long nCount = 0;
if (pAttrs) { pAttrs->Release(); pAttrs = NULL; }
// Получить все дочерние элементы нужного типа
bsData = ::SysAllocString(L"line");
hr = pChild->selectNodes(bsData, &pList);
::SysFreeString(bsData); bsData = NULL;
if (SUCCEEDED(hr) && pList)
{
hr = pList->get_length(&nCount);
if (SUCCEEDED(hr) && nCount > 0)
{
HEAPVAL;
nMaxLen = ((MAX_PATH+1) * nCount) + 1;
pszData = (wchar_t*)malloc(nMaxLen * sizeof(wchar_t));
pszCur = pszData;
pszCur[0] = 0; pszCur[1] = 0;
nCurLen = 2; // сразу посчитать DoubleZero
HEAPVAL;
}
}
if (SUCCEEDED(hr) && pList)
{
hr = pList->reset();
while((hr = pList->nextNode(&pNode)) == S_OK && pNode)
{
bsData = GetAttr(pNode, L"data");
pNode->Release(); pNode = NULL;
if (SUCCEEDED(hr) && bsData)
{
nLen = _tcslen(bsData) + 1;
if ((nCurLen + nLen) > nMaxLen)
{
// Нужно пересоздать!
nMaxLen = nCurLen + nLen + MAX_PATH + 1;
wchar_t *psz = (wchar_t*)malloc(nMaxLen * sizeof(wchar_t));
_ASSERTE(psz);
if (!psz) break; // Не удалось выделить память!
wmemmove(psz, pszData, nCurLen);
pszCur = psz + (pszCur - pszData);
HEAPVAL;
free(pszData);
pszData = psz;
HEAPVAL;
}
lstrcpy(pszCur, bsData);
pszCur += nLen; // указатель - на место для следующей строки
nCurLen += nLen;
*pszCur = 0; // ASCIIZZ
HEAPVAL;
::SysFreeString(bsData); bsData = NULL;
}
}
pList->Release(); pList = NULL;
}
// значит что-то прочитать удалось
if (pszData)
{
if (*value) {free(*value); *value = NULL;}
*value = pszData;
lbRc = true;
}
}
else if (!lstrcmpi(bsType, L"string"))
{
bsData = GetAttr(pChild, pAttrs, L"data");
if (SUCCEEDED(hr) && bsData)
{
nLen = _tcslen(bsData);
if (*value) {free(*value); *value = NULL;}
*value = (wchar_t*)malloc((nLen+2)*sizeof(wchar_t));
lstrcpy(*value, bsData);
(*value)[nLen] = 0; // уже должен быть после lstrcpy
(*value)[nLen+1] = 0; // ASCIIZZ
lbRc = true;
}
}
// Все остальные типы - не интересуют. Нам нужны только строки
}
if (bsType) { ::SysFreeString(bsType); bsType = NULL; }
if (bsData) { ::SysFreeString(bsData); bsData = NULL; }
if (pChild) { pChild->Release(); pChild = NULL; }
if (pAttrs) { pAttrs->Release(); pAttrs = NULL; }
//if (!lbRc)
//{
// _ASSERTE(*value == NULL);
// *value = (wchar_t*)malloc(sizeof(wchar_t)*2);
// (*value)[0] = 0; (*value)[1] = 0; // На случай REG_MULTI_SZ
//}
return lbRc;
}
