Locking.h

Ugrás a fájl dokumentációjához.

/*
    Locking
    Copyright (C) 2006  giszo

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/
#ifndef LOCKING_H
#define LOCKING_H
#ifndef _WIN32
#include <pthread.h>
#endif

class CriticalSection {
#ifdef _WIN32
  public:
    void enter() throw() {
      EnterCriticalSection( &cs );
    }

    void leave() throw() {
      LeaveCriticalSection( &cs );
    }

    CriticalSection() throw() {
      InitializeCriticalSection( &cs );
    }

    ~CriticalSection() throw() {
      DeleteCriticalSection( &cs );
    }

  private:
    CRITICAL_SECTION cs;
#else
  public:
    CriticalSection() throw() {
      pthread_mutexattr_init( &mtx_attr );
      pthread_mutexattr_settype( &mtx_attr, PTHREAD_MUTEX_RECURSIVE );

      pthread_mutex_init( &mtx, &mtx_attr );
    };
    ~CriticalSection() throw() {
      pthread_mutex_destroy( &mtx );
      pthread_mutexattr_destroy( &mtx_attr );
    };
    void enter() throw() { pthread_mutex_lock( &mtx ); };
    void leave() throw() { pthread_mutex_unlock( &mtx ); };
    pthread_mutex_t& getMutex() { return mtx; };

  private:
    pthread_mutex_t mtx;
    pthread_mutexattr_t mtx_attr;
#endif
};

template<class T>
class LockBase {
  public:
    LockBase( T& aCs ) throw() : cs( aCs )  { cs.enter(); };
    ~LockBase() throw() { cs.leave(); };

  private:
    T& cs;
};

typedef LockBase<CriticalSection> Lock;

#endif