forked from gentilkiwi/mimikatz
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathkull_m_xml.c
113 lines (107 loc) · 2.81 KB
/
kull_m_xml.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/* Benjamin DELPY `gentilkiwi`
http://blog.gentilkiwi.com
Licence : https://creativecommons.org/licenses/by/4.0/
*/
#include "kull_m_xml.h"
IXMLDOMDocument * kull_m_xml_CreateAndInitDOM()
{
IXMLDOMDocument *pDoc = NULL;
HRESULT hr = CoCreateInstance(&CLSID_DOMDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IXMLDOMDocument, (void **) &pDoc);
if(hr == S_OK)
{
IXMLDOMDocument_put_async(pDoc, VARIANT_FALSE);
IXMLDOMDocument_put_validateOnParse(pDoc, VARIANT_FALSE);
IXMLDOMDocument_put_resolveExternals(pDoc, VARIANT_FALSE);
IXMLDOMDocument_put_preserveWhiteSpace(pDoc, VARIANT_FALSE);
}
else PRINT_ERROR(L"CoCreateInstance: 0x%08x\n", hr);
return pDoc;
}
void kull_m_xml_ReleaseDom(IXMLDOMDocument *pDoc)
{
if(pDoc)
IXMLDOMDocument_Release(pDoc);
}
BOOL kull_m_xml_LoadXMLFile(IXMLDOMDocument *pXMLDom, PCWSTR filename)
{
BOOL status = FALSE;
VARIANT varFileName;
VARIANT_BOOL varStatus;
BSTR bFilename;
HRESULT hr;
if(filename)
{
if(bFilename = SysAllocString(filename))
{
VariantInit(&varFileName);
V_VT(&varFileName) = VT_BSTR;
V_BSTR(&varFileName) = bFilename;
hr = IXMLDOMDocument_load(pXMLDom, varFileName, &varStatus);
status = (hr == S_OK);
if(!status)
PRINT_ERROR(L"IXMLDOMDocument_load: 0x%08x\n", hr);
SysFreeString(bFilename);
}
}
return status;
}
BOOL kull_m_xml_SaveXMLFile(IXMLDOMDocument *pXMLDom, PCWSTR filename)
{
BOOL status = FALSE;
VARIANT varFileName;
BSTR bFilename;
HRESULT hr;
if(filename)
{
if(bFilename = SysAllocString(filename))
{
VariantInit(&varFileName);
V_VT(&varFileName) = VT_BSTR;
V_BSTR(&varFileName) = bFilename;
hr = IXMLDOMDocument_save(pXMLDom, varFileName);
status = (hr == S_OK);
if(!status)
PRINT_ERROR(L"IXMLDOMDocument_save: 0x%08x\n", hr);
SysFreeString(bFilename);
}
}
return status;
}
wchar_t * kull_m_xml_getAttribute(IXMLDOMNode *pNode, PCWSTR name)
{
wchar_t *result = NULL;
IXMLDOMNamedNodeMap *map;
IXMLDOMNode *nAttr;
BSTR bstrGeneric;
long length, i;
BOOL isMatch = FALSE;
if(IXMLDOMNode_get_attributes(pNode, &map) == S_OK)
{
if(IXMLDOMNamedNodeMap_get_length(map, &length) == S_OK)
{
for(i = 0; (i < length) && !isMatch; i++)
{
if(IXMLDOMNamedNodeMap_get_item(map, i, &nAttr) == S_OK)
{
if(IXMLDOMNode_get_nodeName(nAttr, &bstrGeneric) == S_OK)
{
isMatch = (_wcsicmp(name, bstrGeneric) == 0);
SysFreeString(bstrGeneric);
if(isMatch)
{
if(IXMLDOMNode_get_text(nAttr, &bstrGeneric) == S_OK)
{
kull_m_string_copy(&result, bstrGeneric);
SysFreeString(bstrGeneric);
}
}
}
IXMLDOMNode_Release(nAttr);
}
}
}
IXMLDOMNamedNodeMap_Release(map);
}
return result;
}