AutoLock.h
上传用户:liguizhu
上传日期:2015-11-01
资源大小:2422k
文件大小:2k
源码类别:

P2P编程

开发平台:

Visual C++

  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. #ifndef __AUTOLOCK_H_
  20. #define __AUTOLOCK_H_
  21. class CCritSec
  22. {
  23.     CCritSec(const CCritSec &refCritSec);
  24.     CCritSec &operator=(const CCritSec &refCritSec);
  25.     CRITICAL_SECTION m_CritSec;
  26. public:
  27.     CCritSec() { InitializeCriticalSection(&m_CritSec); }
  28.     ~CCritSec() { DeleteCriticalSection(&m_CritSec); }
  29.     void Lock() { EnterCriticalSection(&m_CritSec); }
  30.     void Unlock() { LeaveCriticalSection(&m_CritSec); }
  31. };
  32. class CAutoLock
  33. {
  34.     CAutoLock(const CAutoLock &refAutoLock);
  35.     CAutoLock &operator=(const CAutoLock &refAutoLock);
  36. protected:
  37.     CCritSec * m_pLock;
  38. public:
  39.     CAutoLock(CCritSec *plock)
  40.     {
  41.         m_pLock = plock;
  42.         m_pLock->Lock();
  43.     }
  44.     ~CAutoLock() { m_pLock->Unlock(); }
  45. };
  46. #endif