home *** CD-ROM | disk | FTP | other *** search
/ Tricks of the Windows Gam…ming Gurus (2nd Edition) / Disc2.iso / vc98 / mfc / src / occlock.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  1998-06-16  |  3.4 KB  |  150 lines
  1. // This is a part of the Microsoft Foundation Classes C++ library.
  2. // Copyright (C) 1992-1998 Microsoft Corporation
  3. // All rights reserved.
  4. //
  5. // This source code is only intended as a supplement to the
  6. // Microsoft Foundation Classes Reference and related
  7. // electronic documentation provided with the library.
  8. // See these sources for detailed information regarding the
  9. // Microsoft Foundation Classes product.
  10.  
  11. #include "stdafx.h"
  12. #include "occimpl.h"
  13.  
  14. #ifdef AFX_OCC_SEG
  15. #pragma code_seg(AFX_OCC_SEG)
  16. #endif
  17.  
  18. #ifdef _DEBUG
  19. #undef THIS_FILE
  20. static char THIS_FILE[] = __FILE__;
  21. #endif
  22.  
  23. #define new DEBUG_NEW
  24.  
  25. /////////////////////////////////////////////////////////////////////////////
  26.  
  27. COleControlLock::COleControlLock(REFCLSID clsid)
  28. {
  29.     // initialize the object
  30.     m_pNextLock = NULL;
  31.     m_clsid = clsid;
  32.     m_pClassFactory = NULL;
  33.  
  34.     // initialize OLE, if necessary
  35.     _AFX_THREAD_STATE* pState = AfxGetThreadState();
  36.     if (!pState->m_bNeedTerm && !AfxOleInit())
  37.         return;
  38.  
  39.     // attempt to lock the class factory of the control
  40.     if (SUCCEEDED(::CoGetClassObject(clsid,
  41.         CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER, NULL, IID_IClassFactory,
  42.         (void**)&m_pClassFactory)))
  43.     {
  44.         ASSERT(m_pClassFactory != NULL);
  45.         if (FAILED(m_pClassFactory->LockServer(TRUE)))
  46.         {
  47.             m_pClassFactory->Release();
  48.             m_pClassFactory = NULL;
  49.         }
  50.     }
  51. }
  52.  
  53. COleControlLock::~COleControlLock()
  54. {
  55.     if (m_pClassFactory != NULL)
  56.     {
  57.         m_pClassFactory->LockServer(FALSE);
  58.         m_pClassFactory->Release();
  59.         m_pClassFactory = NULL;
  60.     }
  61. }
  62.  
  63. BOOL AFXAPI AfxOleLockControl(REFCLSID clsid)
  64. {
  65.     COleControlLock* pLock = NULL;
  66.  
  67.     TRY
  68.     {
  69.         pLock = new COleControlLock(clsid);
  70.         if (pLock->m_pClassFactory == NULL)
  71.         {
  72.             delete pLock;
  73.             pLock = NULL;
  74.         }
  75.         else
  76.         {
  77.             AFX_MODULE_STATE* pModuleState = AfxGetModuleState();
  78.             AfxLockGlobals(CRIT_CTLLOCKLIST);
  79.             pModuleState->m_lockList.AddHead(pLock);
  80.             AfxUnlockGlobals(CRIT_CTLLOCKLIST);
  81.         }
  82.     }
  83.     CATCH_ALL (e)
  84.     {
  85.         // Note: DELETE_EXCEPTION(e) not necessary
  86.         pLock = NULL;
  87.     }
  88.     END_CATCH_ALL
  89.  
  90.     return pLock != NULL;
  91. }
  92.  
  93. BOOL AFXAPI AfxOleLockControl(LPCTSTR lpszProgID)
  94. {
  95.     // map prog id to CLSID
  96.     CLSID clsid;
  97.     HRESULT hr = AfxGetClassIDFromString(lpszProgID, &clsid);
  98.     if (FAILED(hr))
  99.         return FALSE;
  100.  
  101.     return AfxOleLockControl(clsid);
  102. }
  103.  
  104. BOOL AFXAPI AfxOleUnlockControl(LPCTSTR lpszProgID)
  105. {
  106.     // map prog id to CLSID
  107.     CLSID clsid;
  108.     HRESULT hr = AfxGetClassIDFromString(lpszProgID, &clsid);
  109.     if (FAILED(hr))
  110.         return FALSE;
  111.  
  112.     return AfxOleUnlockControl(clsid);
  113. }
  114.  
  115. BOOL AFXAPI AfxOleUnlockControl(REFCLSID clsid)
  116. {
  117.     AFX_MODULE_STATE* pModuleState = AfxGetModuleState();
  118.     AfxLockGlobals(CRIT_CTLLOCKLIST);
  119.     COleControlLock* pLock = pModuleState->m_lockList;
  120.     BOOL bResult = FALSE;
  121.     while (pLock != NULL)
  122.     {
  123.         COleControlLock* pNext = pLock->m_pNextLock;
  124.         if (clsid == pLock->m_clsid)
  125.         {
  126.             // remove lock from list and delete it
  127.             pModuleState->m_lockList.Remove(pLock);
  128.             delete pLock;
  129.             bResult = TRUE;
  130.         }
  131.         pLock = pNext;
  132.     }
  133.     AfxUnlockGlobals(CRIT_CTLLOCKLIST);
  134.     return bResult;
  135. }
  136.  
  137. void AFXAPI AfxOleUnlockAllControls()
  138. {
  139.     AFX_MODULE_STATE* pModuleState = AfxGetModuleState();
  140.     COleControlLock* pLock;
  141.     AfxLockGlobals(CRIT_CTLLOCKLIST);
  142.     while ((pLock = pModuleState->m_lockList) != NULL)
  143.     {
  144.         pModuleState->m_lockList.Remove(pLock);
  145.         delete pLock;
  146.     }
  147.     ASSERT(pModuleState->m_lockList.IsEmpty());
  148.     AfxUnlockGlobals(CRIT_CTLLOCKLIST);
  149. }
  150.