开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > ICQ/即时通讯 > 查看源码
XMLParser.cpp
资源名称：QQ2004_Source_0.01.rar　[点击查看]
上传用户：hysujiao87
上传日期：2007-12-02
资源大小：156k
文件大小：9k
源码类别：

ICQ/即时通讯

开发平台：

C/C++

XMLParser.cpp：源码内容
  1. // XMLParser.cpp: implementation of the CXMLParser class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  4. #include "stdafx.h"
  6. #include "XMLParser.h"
  8. //////////////////////////////////////////////////////////////////////
  9. // Construction/Destruction
  10. //////////////////////////////////////////////////////////////////////
  11. #define TEMP_BUF_LEN 1024
  13. CXMLParser::CXMLParser() :
  14. m_pos(0),
  15. m_root(NULL)
  16. {
  17. m_version = "xml version="1.0"";
  18. }
  20. CXMLParser::~CXMLParser()
  21. {
  22. FreeXML();
  23. FreeTree();
  24. }
  26. int CXMLParser::LoadXML(BYTE* pBuf, DWORD nSize)
  27. {
  28. _ASSERTE(pBuf != NULL);
  29. if (pBuf == NULL)
  30. return -1;
  32. FreeTree();
  34. LPTSTR xmlBuffer = m_xml.GetBuffer(nSize + 1);
  35. memcpy((void*)xmlBuffer, pBuf, nSize);
  36. xmlBuffer[nSize] = '';
  37. m_xml.ReleaseBuffer();
  39. int result = ParseVersion();
  40. if (result != 0)
  41. return result;
  43. return ParseRoot();
  44. }
  46. int CXMLParser::LoadXML(LPCTSTR pathXML)
  47. {
  48. CByteArray xmlBuffer;
  49. try{
  50. CFile file(pathXML, CFile::modeRead | CFile::typeBinary);
  51. int length = (int)file.GetLength();
  52. if (length < 0)
  53. {
  54. TRACE("ERROR: Try to load a too big XML file.");
  55. file.Close();
  56. return -1;
  57. }
  58. xmlBuffer.SetSize(length);
  60. BYTE *xmlData = xmlBuffer.GetData();
  61. _ASSERTE(xmlData != NULL);
  62. if (xmlData == NULL)
  63. {
  64. TRACE("ERROR: Memory not enought.");
  65. file.Close();
  66. return -1;
  67. }
  69. file.Read(xmlData, length);
  70. file.Close();
  71. }
  72. catch (CFileException *fe) {
  73. fe->ReportError();
  74. fe->Delete();
  75. }
  77. return LoadXML(xmlBuffer.GetData(), (DWORD)xmlBuffer.GetSize());
  78. }
  80. int CXMLParser::ParseVersion()
  81. {
  82. //Try to find the start of the version tag.
  83. int version_start = m_xml.Find(_T("<?"), 0);
  84. if (version_start == -1)
  85. return -1;
  86. m_pos = version_start + 2;
  88. int version_finish = m_xml.Find(_T("?>"), m_pos);
  89. if (version_finish == -1)
  90. return -1;
  92. m_version = m_xml.Mid(m_pos, version_finish - m_pos);
  93. m_pos = version_finish + 2;
  95. return 0;
  96. }
  98. int CXMLParser::BuildVersion()
  99. {
  100. CString version_str = "<?" + m_version + "?>rn";
  101. m_xml += version_str;
  103. return 0;
  104. }
  106. int CXMLParser::BuildXML()
  107. {
  108. FreeXML();
  110. int ret = 0;
  111. if ( (ret = BuildVersion()) != 0)
  112. return ret;
  114. return m_root->BuildElement(m_xml);
  115. }
  117. /*BOOL CXMLParser::ParseEntities(CCString& value)
  118. {
  119. static TCHAR AMP_REF[] = _T("&amp;");
  120. static TCHAR LT_REF[] = _T("&lt;");
  121. static TCHAR GT_REF[] = _T("&gt;");
  122. static TCHAR QUOT_REF[] = _T("&quot;");
  123. static TCHAR APOS_REF[] = _T("&apos;");
  125. static TCHAR AMP[] = _T("&");
  126. static TCHAR LT[] = _T("<");
  127. static TCHAR GT[] = _T(">");
  128. static TCHAR QUOT[] = _T(""");
  129. static TCHAR APOS[] = _T("");
  131. static LPCTSTR szEntitiesRef[] = {AMP_REF, LT_REF, GT_REF, QUOT_REF, APOS_REF};
  132. static LPCTSTR szEntities[] = {AMP, LT, GT, QUOT, APOS};
  134. for(int i = 0; i < sizeof(szEntitiesRef) / sizeof(LPCTSTR); i++)
  135. {
  136. int nPos = 0;
  137. do
  138. {
  139. // Try to find the entities reference.
  140. nPos = value.Find(szEntitiesRef[i], nPos);
  141. if (nPos != -1)
  142. {
  143. // Replace the reference with the entities.
  144. value.Delete(nPos, _tcslen(szEntitiesRef[i]));
  145. value.Insert(nPos, szEntities[i]);
  146. nPos += _tcslen(szEntities[i]);
  147. }
  148. }while(nPos != -1);
  149. }
  150. return TRUE;
  151. }
  152. */
  154. int CXMLParser::FreeTree()
  155. {
  156. if (m_root)
  157. delete m_root;
  158. m_root = NULL;
  160. m_version = "xml version="1.0"";
  162. m_pos = 0;
  164. return 0;
  165. }
  167. int CXMLParser::FreeXML()
  168. {
  169. m_xml.Empty();
  170. m_pos = 0;
  172. return 0;
  173. }
  175. int CXMLParser::findNextTag(CString &tagName, CString &beforeTag)
  176. {
  177. tagName.Empty(); beforeTag.Empty();
  179. int beforeTagStart = findFirstNotOf(m_xml, _T("trn "), m_pos, 4);
  180. if (beforeTagStart == -1)
  181. return -1;
  183. int beforeTagFinish = m_xml.Find("<", beforeTagStart);
  184. if (beforeTagFinish == -1)
  185. return -1;
  187. if (beforeTagFinish - beforeTagStart > 0)
  188. {
  189. beforeTag = m_xml.Mid(beforeTagStart, 
  190.   beforeTagFinish - beforeTagStart);
  191. beforeTag.TrimRight();
  192. }
  194. // Find the tag finish position.
  195. int tag_start = beforeTagFinish;
  196. int tag_finish = m_xml.Find(">", tag_start);
  197. if (tag_finish == -1)
  198. return -1;
  200. tagName = m_xml.Mid(tag_start + 1, tag_finish - tag_start - 1);
  201. m_pos = tag_finish + 1;
  203. return 0;
  204. }
  206. int CXMLParser::ParseElement(CElement* element)
  207. {
  208. _ASSERTE(element != NULL);
  209. if (element == NULL)
  210. return -1;
  212. while(m_pos < m_xml.GetLength())
  213. {
  214. CString tagName;
  215. CString beforeTag;
  216. if (findNextTag(tagName, beforeTag) == -1)
  217. {
  218. TRACE("Error: without a finish tag for <%s>.n", element->get_tag());
  219. return -1;
  220. }
  222. if (element->get_content().IsEmpty())
  223. element->set_content(beforeTag);
  225. // Is a end-tag?
  226. if (tagName[0] == '/')
  227. {
  228. // Remove the '/' character.
  229. CString end_tagName;
  231. end_tagName = tagName;
  232. end_tagName.Delete(0, 1);
  234. if (element->get_tag() != end_tagName)
  235. {
  236. TRACE("Can't find a end tag for <%s>.n", element->get_tag());
  237. return -1;
  238. }
  240. return 0;
  241. }
  243. CElement* sub_element = new CElement(element);
  244. if (!sub_element)
  245. return -1;
  247. // Is it a empty tag?
  248. if (tagName[tagName.GetLength() - 1] == '/')
  249. {
  250. tagName.Delete(tagName.GetLength() - 1, 1);
  252. CString actualTagName;
  253. CMapStringToString attribute;
  254. if (ParseAttribute(tagName, actualTagName, sub_element->_attribute) != 0)
  255. return -1;
  257. sub_element->set_tag(actualTagName);
  258. continue ;
  259. }
  261. CString actualTagName;
  262. CMapStringToString attribute;
  263. if (ParseAttribute(tagName, actualTagName, sub_element->_attribute) != 0)
  264. return -1;
  266. sub_element->set_tag(actualTagName);
  267. if (ParseElement(sub_element) != 0)
  268. return -1;
  269. }
  271. TRACE("Error: without a finish tag for <%s>.n", element->get_tag());
  272. return -1;
  273. }
  275. int CXMLParser::EnumerateElements(LPENUM_CALLBACK_FUNC pFunc)
  276. {
  277. _ASSERTE(pFunc != NULL);
  278. if (!pFunc)
  279. return -1;
  281. return m_root->Enumerate(pFunc);
  282. }
  284. int CXMLParser::findFirstNotOf(CString &str, LPCTSTR array, int start, int arraySize)
  285. {
  286. int result = start;
  287. while(result < str.GetLength())
  288. {
  289. BOOL bEqual = FALSE;
  290. for (int i = 0; i < arraySize; i++)
  291. {
  292. if (str[result] == array[i])
  293. bEqual = TRUE;
  294. }
  295. if (bEqual == FALSE)
  296. return result;
  298. result++;
  299. }
  301. return -1;
  302. }
  304. int CXMLParser::ParseRoot()
  305. {
  306. // Find the root start tag.
  307. int tag_start = findFirstNotOf(m_xml, _T("trn "), m_pos, 4);
  309. // Find the next tag start position (it is the content part finish also).
  310. char temp = m_xml[tag_start];
  311. if (m_xml[tag_start] != _T('<'))
  312. {
  313. TRACE("Invalid XML format.n");
  314. return -1;
  315. }
  317. int tag_finish = m_xml.Find(_T('>'), tag_start);
  318. if (tag_finish == -1 || tag_finish - tag_start <= 1)
  319. {
  320. TRACE("Invalid root tag name.n");
  321. return -1;
  322. }
  324. CString tagName = m_xml.Mid(tag_start + 1, tag_finish - tag_start - 1);
  325. set_root(new CElement(NULL));
  326. get_root()->set_tag(tagName);
  328. m_pos = tag_finish + 1;
  330. return ParseElement(m_root);
  331. }
  333. int CXMLParser::ParseAttribute(LPCTSTR tagString, CString &tagName, CMapStringToString &attribute)
  334. {
  335. _ASSERTE(tagString != NULL);
  336. if (tagString == NULL)
  337. return -1;
  339. attribute.RemoveAll();
  341. CString tempSting = tagString;
  342. tempSting.TrimLeft(); tempSting.TrimRight();
  343. int firstSpacePos = tempSting.FindOneOf(_T(" trn"));
  344. if (firstSpacePos == -1)
  345. {
  346. tagName = tempSting;
  347. return 0;
  348. }
  349. tagName = tempSting.Left(firstSpacePos);
  351. int startPos = firstSpacePos + 1;
  352. int equalPos = tempSting.Find(_T('='), startPos);
  353. while (equalPos != -1)
  354. {
  355. CString attributeName = tempSting.Mid(startPos, equalPos - startPos);
  356. attributeName.TrimLeft(); attributeName.TrimRight();
  357. int leftInvertedCommasPos = tempSting.Find(_T('"'), equalPos + 1);
  358. if (leftInvertedCommasPos == -1)
  359. return -1;
  361. int rightInvertedCommasPos = tempSting.Find(_T('"'), leftInvertedCommasPos + 1);
  362. if (rightInvertedCommasPos == -1)
  363. return -1;
  365. CString attributeValue = tempSting.Mid(leftInvertedCommasPos + 1, rightInvertedCommasPos - leftInvertedCommasPos - 1);
  366. attributeValue.TrimLeft(); attributeValue.TrimRight();
  368. attribute.SetAt(attributeName, attributeValue);
  370. startPos = rightInvertedCommasPos + 1;
  371. equalPos = tempSting.Find(_T('='), startPos);
  372. }
  374. if (startPos <= tempSting.GetLength() - 1)
  375. return -1;
  377. return 0;
  378. }
  380. CElement *CXMLParser::createElement(LPCTSTR tag, LPCTSTR content)
  381. {
  382. _ASSERTE(tag != NULL && content != NULL);
  383. if (tag == NULL || content == NULL)
  384. return NULL;
  386. CElement *element = new CElement(NULL);
  387. if (element == NULL)
  388. return NULL;
  390. element->set_tag(tag);
  391. element->set_content(content);
  392. return element;
  393. }