statsmgr.cpp
上传用户:zhongxx05
上传日期:2007-06-06
资源大小:33641k
文件大小:11k
源码类别:

Symbian

开发平台:

C/C++

  1. /* ***** BEGIN LICENSE BLOCK ***** 
  2.  * Version: RCSL 1.0/RPSL 1.0 
  3.  *  
  4.  * Portions Copyright (c) 1995-2002 RealNetworks, Inc. All Rights Reserved. 
  5.  *      
  6.  * The contents of this file, and the files included with this file, are 
  7.  * subject to the current version of the RealNetworks Public Source License 
  8.  * Version 1.0 (the "RPSL") available at 
  9.  * http://www.helixcommunity.org/content/rpsl unless you have licensed 
  10.  * the file under the RealNetworks Community Source License Version 1.0 
  11.  * (the "RCSL") available at http://www.helixcommunity.org/content/rcsl, 
  12.  * in which case the RCSL will apply. You may also obtain the license terms 
  13.  * directly from RealNetworks.  You may not use this file except in 
  14.  * compliance with the RPSL or, if you have a valid RCSL with RealNetworks 
  15.  * applicable to this file, the RCSL.  Please see the applicable RPSL or 
  16.  * RCSL for the rights, obligations and limitations governing use of the 
  17.  * contents of the file.  
  18.  *  
  19.  * This file is part of the Helix DNA Technology. RealNetworks is the 
  20.  * developer of the Original Code and owns the copyrights in the portions 
  21.  * it created. 
  22.  *  
  23.  * This file, and the files included with this file, is distributed and made 
  24.  * available on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 
  25.  * EXPRESS OR IMPLIED, AND REALNETWORKS HEREBY DISCLAIMS ALL SUCH WARRANTIES, 
  26.  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS 
  27.  * FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 
  28.  * 
  29.  * Technology Compatibility Kit Test Suite(s) Location: 
  30.  *    http://www.helixcommunity.org/content/tck 
  31.  * 
  32.  * Contributor(s): 
  33.  *  
  34.  * ***** END LICENSE BLOCK ***** */ 
  35. #include "hxtypes.h"
  36. #include "hxcom.h"
  37. #include <stdio.h>
  38. #include <stdlib.h>
  39. #include "hxcomm.h"
  40. #include "hxengin.h"
  41. #include "hxmon.h"
  42. #include "hxclreg.h"
  43. #include "chxpckts.h"
  44. #include "hxslist.h"
  45. #include "hxmap.h"
  46. #include "statsmgr.h"
  47. #include "hxstrutl.h"
  48. #include "hxheap.h"
  49. #ifdef _DEBUG
  50. #undef HX_THIS_FILE
  51. static const char HX_THIS_FILE[] = __FILE__;
  52. #endif
  53. StatsManager::StatsManager(HXClientRegistry* pRegistry,
  54.    UINT32 ulRegistryID,
  55.    UINT32 ulRepeatedRegistryID)
  56. {
  57.     m_pStatsMap = new CHXMapLongToObj;
  58.     
  59.     HX_RESULT     hr = HXR_OK;
  60.     IHXBuffer*     pBuffer = NULL;
  61.     m_lRefCount = 0;
  62.     if (!pRegistry)
  63.     {
  64. hr = HXR_FAILED;
  65. goto cleanup;
  66.     }
  67.     m_pRegistry = pRegistry;
  68.     m_pRegistry->AddRef();
  69.     if (HXR_OK == m_pRegistry->GetPropName(ulRepeatedRegistryID, pBuffer))
  70.     {
  71. m_ulOffset = pBuffer->GetSize();
  72.     }
  73.     HX_RELEASE(pBuffer);
  74.     if (HXR_OK == m_pRegistry->GetPropName(ulRegistryID, pBuffer))
  75.     {
  76. m_pRegistryName = new char[pBuffer->GetSize() + 1];
  77. strcpy(m_pRegistryName, (const char*)pBuffer->GetBuffer()); /* Flawfinder: ignore */
  78.     }
  79.     HX_RELEASE(pBuffer);
  80.     m_ulRegistryID = ulRegistryID;
  81.     m_ulRepeatedRegistryID = m_ulRepeatedRegistryID;
  82.     m_pPropWatchList = new CHXSimpleList();
  83.     if (HXR_OK != SetWatch(ulRepeatedRegistryID))
  84.     {
  85. hr = HXR_UNEXPECTED;
  86. goto cleanup;
  87.     }
  88. cleanup:
  89.     return;
  90. }
  91. StatsManager::~StatsManager()
  92. {
  93.     HX_VECTOR_DELETE(m_pRegistryName);
  94.     HX_RELEASE(m_pRegistry);
  95.     HX_DELETE(m_pStatsMap);
  96. }
  97. /////////////////////////////////////////////////////////////////////////
  98. //  Method:
  99. //      HXRegistry::QueryInterface
  100. //  Purpose:
  101. //      Implement this to export the interfaces supported by your
  102. //      object.
  103. //
  104. STDMETHODIMP
  105. StatsManager::QueryInterface(REFIID riid, void** ppvObj)
  106. {
  107.     QInterfaceList qiList[] =
  108.         {
  109.             { GET_IIDHANDLE(IID_IHXPropWatchResponse), (IHXPropWatchResponse*)this },
  110.             { GET_IIDHANDLE(IID_IUnknown), (IUnknown*)this },
  111.         };
  112.     
  113.     return ::QIFind(qiList, QILISTSIZE(qiList), riid, ppvObj);
  114. }   
  115. /////////////////////////////////////////////////////////////////////////
  116. //  Method:
  117. //      HXRegistry::AddRef
  118. //  Purpose:
  119. //      Everyone usually implements this the same... feel free to use
  120. //      this implementation.
  121. //
  122. STDMETHODIMP_(ULONG32)
  123. StatsManager::AddRef()
  124. {
  125.     return InterlockedIncrement(&m_lRefCount);
  126. }   
  127. /////////////////////////////////////////////////////////////////////////
  128. //  Method:
  129. //      HXRegistry::Release
  130. //  Purpose:
  131. //      Everyone usually implements this the same... feel free to use
  132. //      this implementation.
  133. //
  134. STDMETHODIMP_(ULONG32)
  135. StatsManager::Release()
  136. {
  137.     if (InterlockedDecrement(&m_lRefCount) > 0)
  138.     {
  139.         return m_lRefCount;
  140.     }
  141.     
  142.     delete this;
  143.     return 0;
  144. }   
  145. HX_RESULT
  146. StatsManager::DoCleanup()
  147. {
  148.     HX_RESULT hr = HXR_OK;
  149.     CHXSimpleList::Iterator i;
  150.     CHXMapLongToObj::Iterator j;
  151.     if (m_pPropWatchList)
  152.     {
  153. PropWatchEntry*  pPropWatchEntry = NULL;
  154. i = m_pPropWatchList->Begin();
  155. for (; i != m_pPropWatchList->End(); ++i)
  156. {
  157.     pPropWatchEntry = (PropWatchEntry*)(*i);
  158.     
  159.     pPropWatchEntry->pPropWatch->ClearWatchById(pPropWatchEntry->ulPropID);
  160.     HX_RELEASE(pPropWatchEntry->pPropWatch);
  161.     HX_DELETE(pPropWatchEntry);
  162. }
  163. HX_DELETE(m_pPropWatchList);
  164.     }
  165.     StatsMapEntry* pEntry = NULL;
  166.     j = m_pStatsMap->Begin();
  167.     for (; j != m_pStatsMap->End(); ++j)
  168.     {
  169. pEntry = (StatsMapEntry*)(*j);
  170. HX_DELETE(pEntry);
  171.     }
  172.     m_pStatsMap->RemoveAll();
  173.     return hr;
  174. }
  175. HX_RESULT
  176. StatsManager::Copy()
  177. {
  178.     HX_RESULT     hr = HXR_OK;
  179.     INT32     lValue = 0;
  180.     IHXBuffer*     pName = NULL;
  181.     IHXBuffer*     pBuffer = NULL;
  182.     StatsMapEntry*  pEntry = NULL;
  183.     CHXMapLongToObj::Iterator i;
  184.     i = m_pStatsMap->Begin();
  185.     for (; i != m_pStatsMap->End(); ++i)
  186.     {
  187. pEntry = (StatsMapEntry*)(*i);
  188. switch(pEntry->type)
  189. {
  190.     case PT_INTEGER:
  191. m_pRegistry->GetIntById(pEntry->ulFrom, lValue);     
  192. m_pRegistry->SetIntById(pEntry->ulTo, lValue);
  193. break;
  194.     case PT_INTREF:
  195. m_pRegistry->GetIntById(pEntry->ulFrom, lValue);
  196. m_pRegistry->GetPropName(pEntry->ulTo, pName);
  197. m_pRegistry->AddIntRef((const char*)pName->GetBuffer(), &lValue);
  198. HX_RELEASE(pName);
  199. break;
  200.     case PT_STRING:
  201. if (HXR_OK == m_pRegistry->GetStrById(pEntry->ulFrom, pBuffer) &&
  202.     pBuffer)
  203. {
  204.     m_pRegistry->SetStrById(pEntry->ulTo, pBuffer);
  205. }
  206. HX_RELEASE(pBuffer);
  207. break;
  208.     case PT_BUFFER:
  209. if (HXR_OK == m_pRegistry->GetBufById(pEntry->ulFrom, pBuffer) &&
  210.     pBuffer)
  211. {
  212.     m_pRegistry->SetBufById(pEntry->ulTo, pBuffer);
  213. }
  214. HX_RELEASE(pBuffer);
  215. break;
  216.     default:
  217. break;
  218. }
  219.     }
  220.     return hr;
  221. }
  222.     
  223. HX_RESULT
  224. StatsManager::SetWatch(UINT32 ulRegistryID)
  225. {
  226.     HX_RESULT     hr = HXR_OK;
  227.     PropWatchEntry* pPropWatchEntry = NULL;
  228.     IHXPropWatch*  pPropWatch = NULL;
  229.     if (HXR_OK != m_pRegistry->CreatePropWatch(pPropWatch))
  230.     {
  231. hr = HXR_FAILED;
  232. goto cleanup;
  233.     }
  234.     if (HXR_OK != pPropWatch->Init((IHXPropWatchResponse*)this))
  235.     {
  236. hr = HXR_FAILED;
  237. goto cleanup;
  238.     }
  239.     pPropWatch->SetWatchById(ulRegistryID);
  240.     pPropWatchEntry = new PropWatchEntry;
  241.     pPropWatchEntry->ulPropID = ulRegistryID;
  242.     pPropWatchEntry->pPropWatch = pPropWatch;
  243.     m_pPropWatchList->AddTail(pPropWatchEntry);
  244. cleanup:
  245.     if (HXR_OK != hr)
  246.     {
  247. HX_RELEASE(pPropWatch);
  248. HX_DELETE(pPropWatchEntry);
  249.     }
  250.     return hr;
  251. }
  252. STDMETHODIMP
  253. StatsManager::AddedProp(const UINT32 ulHash,
  254. const HXPropType type,
  255. const UINT32 ulParentHash)
  256. {
  257.     HX_RESULT hr = HXR_OK;
  258.     INT32 lValue = 0;
  259.     UINT32 ulRegistryID = 0;
  260.     char szRegKeyName[MAX_DISPLAY_NAME] = {0}; /* Flawfinder: ignore */
  261.     HXPropType theType = PT_UNKNOWN;
  262.     IHXBuffer* pBuffer = NULL;
  263.     if (HXR_OK == m_pRegistry->GetPropName(ulHash, pBuffer))
  264.     {
  265. // apparently the type returned from IHXPropWatchReponse is always
  266. // PT_UNKNOWN !!
  267. theType = m_pRegistry->GetTypeById(ulHash);
  268. // exclude leading "Repeat.*."
  269. SafeSprintf(szRegKeyName, MAX_DISPLAY_NAME, "%s.%s", m_pRegistryName, (const char*)(pBuffer->GetBuffer() + m_ulOffset));
  270. HX_RELEASE(pBuffer);
  271. ulRegistryID = m_pRegistry->GetId(szRegKeyName);
  272. if (!ulRegistryID)
  273. {
  274.     switch (theType)
  275.     {
  276.     case PT_INTEGER:
  277. m_pRegistry->GetIntById(ulHash, lValue);     
  278. ulRegistryID = m_pRegistry->AddInt(szRegKeyName, lValue);
  279. break;
  280.     case PT_INTREF:
  281. m_pRegistry->GetIntById(ulHash, lValue);
  282. ulRegistryID = m_pRegistry->AddIntRef(szRegKeyName, &lValue);
  283. break;
  284.     case PT_STRING:
  285. m_pRegistry->GetStrById(ulHash, pBuffer);
  286. ulRegistryID = m_pRegistry->AddStr(szRegKeyName, pBuffer);
  287. HX_RELEASE(pBuffer);
  288. break;
  289.     case PT_BUFFER:
  290. m_pRegistry->GetBufById(ulHash, pBuffer);
  291. ulRegistryID = m_pRegistry->AddBuf(szRegKeyName, pBuffer);
  292. HX_RELEASE(pBuffer);
  293. break;
  294.     case PT_COMPOSITE:
  295. hr = m_pRegistry->AddComp(szRegKeyName);
  296. break;
  297.     default:
  298. break;
  299.     }
  300. }
  301. if (PT_COMPOSITE != theType)
  302. {
  303.     StatsMapEntry* pEntry = new StatsMapEntry;
  304.     pEntry->ulFrom = ulHash;
  305.     pEntry->ulTo = ulRegistryID;
  306.     pEntry->type = theType;
  307.     m_pStatsMap->SetAt((INT32)ulHash, pEntry);
  308. }
  309. else
  310. {
  311.     hr = SetWatch(ulHash);
  312. }
  313.     }
  314.     return hr;
  315. }
  316. STDMETHODIMP
  317. StatsManager::ModifiedProp(const UINT32 ulHash,
  318.    const HXPropType type,
  319.    const UINT32 ulParentHash)
  320. {
  321.     return HXR_OK;
  322. }
  323. STDMETHODIMP
  324. StatsManager::DeletedProp(const UINT32 ulHash,
  325.   const UINT32 ulParentHash)
  326. {
  327.     HX_RESULT hr = HXR_OK;
  328.     UINT32 ulRegistryID = 0;
  329.     char szRegKeyName[MAX_DISPLAY_NAME] = {0}; /* Flawfinder: ignore */
  330.     IHXBuffer* pBuffer = NULL;
  331.     if (HXR_OK == m_pRegistry->GetPropName(ulHash, pBuffer))
  332.     {
  333. // exclude leading "Repeat.*."
  334. SafeSprintf(szRegKeyName, MAX_DISPLAY_NAME, "%s.%s", m_pRegistryName, (const char*)(pBuffer->GetBuffer() + m_ulOffset));
  335. HX_RELEASE(pBuffer);
  336. ulRegistryID = m_pRegistry->GetId(szRegKeyName);
  337. if (ulRegistryID)
  338. {
  339.     m_pRegistry->DeleteById(ulRegistryID);
  340.     StatsMapEntry* pEntry = NULL;
  341.     if (m_pStatsMap->Lookup((INT32)ulHash, (void*&)pEntry))
  342.     {
  343. HX_DELETE(pEntry);
  344. m_pStatsMap->RemoveKey((INT32)ulHash);
  345.     }
  346. }
  347.     }
  348.     
  349.     return HXR_OK;
  350. }
  351. HX_RESULT
  352. StatsManager::UpdateRegistry(UINT32 ulRegistryID)
  353. {
  354.     IHXBuffer* pBuffer = NULL;
  355.     if (HXR_OK == m_pRegistry->GetPropName(ulRegistryID, pBuffer))
  356.     {
  357. HX_VECTOR_DELETE(m_pRegistryName);
  358. m_pRegistryName = new char[pBuffer->GetSize() + 1];
  359. strcpy(m_pRegistryName, (const char*)pBuffer->GetBuffer()); /* Flawfinder: ignore */
  360.     }
  361.     HX_RELEASE(pBuffer);
  362.     m_ulRegistryID = ulRegistryID;
  363.     return HXR_OK;
  364. }