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
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
#include "./providerLoader.h"
#include "./providerEnumerator.h"
#include "./xmlInt32Parser.h"
#include "../api.h"
#include "../../xml/obj_xml.h"
#include <api/service/waservicefactory.h>
LoginProviderLoader::LoginProviderLoader()
{
}
LoginProviderLoader::~LoginProviderLoader()
{
}
HRESULT LoginProviderLoader::ReadXml(LPCWSTR pszPath, LoginProviderEnumerator **enumerator, INT *prefVisible)
{
if (NULL == enumerator) return E_POINTER;
*enumerator = NULL;
if (NULL == pszPath || L'\0' == *pszPath)
return E_INVALIDARG;
HANDLE hFile = CreateFile(pszPath, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
if (INVALID_HANDLE_VALUE == hFile)
{
DWORD error = GetLastError();
return HRESULT_FROM_WIN32(error);
}
HRESULT hr;
if (NULL != WASABI_API_SVC)
{
waServiceFactory *sf = WASABI_API_SVC->service_getServiceByGuid(obj_xmlGUID);
obj_xml *reader = (NULL != sf) ? (obj_xml*)sf->getInterface() : NULL;
if (NULL != reader)
{
if (OBJ_XML_SUCCESS == reader->xmlreader_open())
{
providerList.clear();
parser.SetReader(reader);
XmlInt32Parser visibleParser;
reader->xmlreader_registerCallback(L"loginProviders\fprovider", this);
if (NULL != prefVisible)
reader->xmlreader_registerCallback(L"loginProviders\fvisibleProviders", &visibleParser);
hr = FeedFile(reader, hFile, 8192);
reader->xmlreader_close();
parser.SetReader(NULL);
if (SUCCEEDED(hr))
hr = LoginProviderEnumerator::CreateInstance(providerList.begin(), providerList.size(), enumerator);
if (NULL != prefVisible && FAILED(visibleParser.GetValue(prefVisible)))
*prefVisible = 0;
}
else
hr = E_FAIL;
sf->releaseInterface(reader);
}
else
hr = E_FAIL;
}
else
hr = E_UNEXPECTED;
CloseHandle(hFile);
return hr;
}
HRESULT LoginProviderLoader::FeedFile(obj_xml *reader, HANDLE hFile, DWORD bufferSize)
{
if (NULL == reader || INVALID_HANDLE_VALUE == hFile || 0 == bufferSize)
return E_INVALIDARG;
BYTE *buffer = (BYTE*)calloc(bufferSize, sizeof(BYTE));
if (NULL == buffer) return E_OUTOFMEMORY;
HRESULT hr;
INT readerCode = OBJ_XML_SUCCESS;
for(;;)
{
DWORD read = 0;
if (FALSE == ReadFile(hFile, buffer, bufferSize, &read, NULL) || 0 == read)
{
DWORD errorCode = GetLastError();
hr = HRESULT_FROM_WIN32(errorCode);
if (0 == read && OBJ_XML_SUCCESS == readerCode)
reader->xmlreader_feed(0, 0);
break;
}
readerCode = reader->xmlreader_feed(buffer, read);
if (OBJ_XML_SUCCESS != readerCode)
{
hr = E_FAIL;
break;
}
}
free(buffer);
return hr;
}
void LoginProviderLoader::Event_XmlStartElement(const wchar_t *xmlpath, const wchar_t *xmltag, ifc_xmlreaderparams *params)
{
parser.Begin(params);
}
void LoginProviderLoader::Event_XmlEndElement(const wchar_t *xmlpath, const wchar_t *xmltag)
{
LoginProvider *provider;
if (SUCCEEDED(parser.End(&provider)))
{
providerList.push_back(provider);
}
}
void LoginProviderLoader::Event_XmlError(int linenum, int errcode, const wchar_t *errstr)
{
parser.End(NULL);
}
#define CBCLASS LoginProviderLoader
START_DISPATCH;
VCB(ONSTARTELEMENT, Event_XmlStartElement)
VCB(ONENDELEMENT, Event_XmlEndElement)
VCB(ONERROR, Event_XmlError)
END_DISPATCH;
#undef CBCLASS
|
