src/Utils/RWSection.h

   1 // TortoiseSVN - a Windows shell extension for easy version control\r
   2 \r
   3 // Copyright (C) 2003-2008 - TortoiseSVN\r
   4 \r
   5 // This program is free software; you can redistribute it and/or\r
   6 // modify it under the terms of the GNU General Public License\r
   7 // as published by the Free Software Foundation; either version 2\r
   8 // of the License, or (at your option) any later version.\r
   9 \r
  10 // This program is distributed in the hope that it will be useful,\r
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  13 // GNU General Public License for more details.\r
  14 \r
  15 // You should have received a copy of the GNU General Public License\r
  16 // along with this program; if not, write to the Free Software Foundation,\r
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.\r
  18 //\r
  19 #pragma once\r
  20 \r
  21 /**\r
  22  * \ingroup Utils\r
  23  * Implements a reader-writer lock.\r
  24  */\r
  25 class CRWSection\r
  26 {\r
  27 public:\r
  28         CRWSection();\r
  29         ~CRWSection();\r
  30 \r
  31         bool WaitToRead(DWORD waitTime = INFINITE);\r
  32         bool WaitToWrite(DWORD waitTime = INFINITE);\r
  33         void Done();\r
  34         bool IsWriter() {return ((m_nWaitingWriters > 0) || (m_nActive < 0));}\r
  35 #if defined (DEBUG) || defined (_DEBUG)\r
  36         void AssertLock() {ATLASSERT(m_nActive);}\r
  37         void AssertWriting() {ATLASSERT((m_nWaitingWriters || (m_nActive < 0)));}\r
  38 #else\r
  39         void AssertLock() {;}\r
  40         void AssertWriting() {;}\r
  41 #endif\r
  42 private:\r
  43         int                                     m_nWaitingReaders;      // Number of readers waiting for access\r
  44         int                                     m_nWaitingWriters;      // Number of writers waiting for access\r
  45         int                                     m_nActive;                      // Number of threads accessing the section\r
  46         CRITICAL_SECTION        m_cs;\r
  47         HANDLE                          m_hReaders;\r
  48         HANDLE                          m_hWriters;\r
  49 };\r