mutex.cc

   1 /*
   2     Copyright 2004-2005 Chris Tallon
   3
   4     This file is part of VOMP.
   5
   6     VOMP is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     VOMP is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with VOMP; if not, write to the Free Software
  18     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19 */
  20 #include "mutex.h"
  21
  22
  23 #include <unistd.h>
  24 #include <sys/syscall.h>
  25
  26 Mutex::Mutex() {
  27 #ifndef WIN32
  28         pthread_mutex_init(&my_mutex, NULL);
  29 #else
  30          my_mutex=CreateMutex(NULL,FALSE,NULL);
  31 #endif
  32          my_cur_thread=0;
  33          num_locks=0;
  34 }
  35
  36 Mutex::~Mutex() {
  37 #ifdef WIN32
  38   CloseHandle(my_mutex);
  39 #endif
  40 }
  41
  42 void Mutex::Lock() {
  43         if (num_locks> 0) {
  44 #ifndef WIN32
  45                 if (my_cur_thread==(pid_t) syscall(SYS_gettid)) {
  46 #else
  47                 if (my_cur_thread==GetCurrentThreadId()) {
  48 #endif
  49                         num_locks++;
  50                         return;
  51                 }
  52         }
  53 #ifndef WIN32
  54   pthread_mutex_lock(&my_mutex);
  55   my_cur_thread=syscall(SYS_gettid);
  56 #else
  57    WaitForSingleObject(my_mutex, INFINITE );
  58    my_cur_thread=GetCurrentThreadId();
  59 #endif
  60    num_locks++;
  61
  62 }
  63
  64 void Mutex::Unlock() {
  65 #ifndef WIN32
  66         if (my_cur_thread==syscall(SYS_gettid)) num_locks--;
  67 #else
  68         if (my_cur_thread==GetCurrentThreadId()) num_locks--;
  69 #endif
  70         if (num_locks==0) {
  71                 my_cur_thread=0;
  72 #ifndef WIN32
  73                 pthread_mutex_unlock(&my_mutex);
  74 #else
  75                 ReleaseMutex(my_mutex);
  76 #endif
  77         }
  78 }
  79