读-写共享独占锁源码实现-C++实现

#include "StdAfx.h" #include "MySRWLockManage.h" CMySRWLockManage::CMySRWLockManage(void) { m_pfnMyInitializeSRWLockFunc = NULL; m_pfnMyAcquireSRWLockExclusiveFunc = NULL; m_pfnMyReleaseSRWLockExclusiveFunc = NULL; m_pfnMyAcquireSRWLockSharedFunc = NULL; m_pfnMyReleaseSRWLockSharedFunc = NULL; } CMySRWLockManage::~CMySRWLockManage(void) { } BOOL CMySRWLockManage::InitSRWStation() { BOOL bResult = FALSE; m_pfnMyInitializeSRWLockFunc = (MyInitializeSRWLockFunc)MyGetProcAddress(L"kernel32.dll", "InitializeSRWLock"); m_pfnMyAcquireSRWLockExclusiveFunc = (MyAcquireSRWLockExclusiveFunc)MyGetProcAddress(L"kernel32.dll", "AcquireSRWLockExclusive"); m_pfnMyReleaseSRWLockExclusiveFunc = (MyReleaseSRWLockExclusiveFunc)MyGetProcAddress(L"kernel32.dll", "ReleaseSRWLockExclusive"); m_pfnMyAcquireSRWLockSharedFunc = (MyAcquireSRWLockSharedFunc)GetProcAddress(GetModuleHandle(L"kernel32.dll"), "AcquireSRWLockShared"); m_pfnMyReleaseSRWLockSharedFunc = (MyReleaseSRWLockSharedFunc)MyGetProcAddress(L"kernel32.dll", "ReleaseSRWLockShared"); if(m_pfnMyInitializeSRWLockFunc == NULL || m_pfnMyAcquireSRWLockExclusiveFunc == NULL || m_pfnMyReleaseSRWLockExclusiveFunc == NULL || m_pfnMyAcquireSRWLockSharedFunc == NULL || m_pfnMyReleaseSRWLockSharedFunc == NULL) goto out; m_pfnMyInitializeSRWLockFunc(&m_pSRWLock); bResult = TRUE; out: return bResult; } void CMySRWLockManage::MyAcquireSRWLockExclusive() { if(m_pfnMyAcquireSRWLockExclusiveFunc) m_pfnMyAcquireSRWLockExclusiveFunc(&m_pSRWLock); return ; } void CMySRWLockManage::MyReleaseSRWLockExclusive() { if(m_pfnMyReleaseSRWLockExclusiveFunc) m_pfnMyReleaseSRWLockExclusiveFunc(&m_pSRWLock); return ; } void CMySRWLockManage::MyAcquireSRWLockShared() { if(m_pfnMyAcquireSRWLockSharedFunc) m_pfnMyAcquireSRWLockSharedFunc(&m_pSRWLock); return ; } void CMySRWLockManage::MyReleaseSRWLockShared() { if(m_pfnMyReleaseSRWLockSharedFunc) m_pfnMyReleaseSRWLockSharedFunc(&m_pSRWLock); return ; } FORCEINLINE PVOID CMySRWLockManage::MyGetProcAddress( __in PTSTR LibraryName, __in PSTR ProcName ) { return GetProcAddress(GetModuleHandle(LibraryName), ProcName); }