129 lines
3.1 KiB
C++
129 lines
3.1 KiB
C++
/*
|
|
** Command & Conquer Renegade(tm)
|
|
** Copyright 2025 Electronic Arts Inc.
|
|
**
|
|
** 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 3 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, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "critsec.h"
|
|
#include <assert.h>
|
|
#include "wlib/wdebug.h"
|
|
|
|
|
|
|
|
CritSec::CritSec()
|
|
{
|
|
#ifdef _UNIX
|
|
pthread_mutex_init(&Mutex_, NULL);
|
|
RefCount_ = 0;
|
|
#elif defined(_WIN32)
|
|
InitializeCriticalSection(&CritSec_);
|
|
#endif
|
|
}
|
|
|
|
CritSec::~CritSec()
|
|
{
|
|
#ifdef _UNIX
|
|
pthread_mutex_destroy(&Mutex_);
|
|
#elif defined(_WIN32)
|
|
DeleteCriticalSection(&CritSec_);
|
|
#endif
|
|
}
|
|
|
|
// The "lock" function blocks until the mutex is available.
|
|
// Returns 0 on success, error code on error.
|
|
//
|
|
// A thread that already has a lock will increment a reference count if it calls
|
|
// lock again. It must then call unlock() enough times to get the reference to 0.
|
|
//
|
|
// If refcount is not null you can get the current ref counter after the lock.
|
|
//
|
|
sint32 CritSec::lock(int *refcount) RO
|
|
{
|
|
#ifdef _UNIX
|
|
sint32 status;
|
|
|
|
// I TRY to get the lock. IF I succeed, OR if I fail because
|
|
// I already have the lock, I just increment the reference
|
|
// count and return.
|
|
if (((status = pthread_mutex_trylock(&Mutex_)) == 0) ||
|
|
((status == EBUSY) && (ThreadId_ == pthread_self())))
|
|
{
|
|
ThreadId_ = pthread_self();
|
|
RefCount_++;
|
|
if (refcount)
|
|
*refcount=RefCount_;
|
|
return(0);
|
|
}
|
|
|
|
// Otherwise, I wait for the lock.
|
|
if ((status = pthread_mutex_lock(&Mutex_)) == 0)
|
|
{
|
|
assert(RefCount_ == 0);
|
|
ThreadId_ = pthread_self();
|
|
RefCount_++;
|
|
}
|
|
else
|
|
{
|
|
ERRMSG("pthread_mutex_lock: " << strerror(errno));
|
|
}
|
|
|
|
if (refcount)
|
|
*refcount=RefCount_;
|
|
|
|
return(status);
|
|
#elif defined(_WIN32)
|
|
// TOFIX update the refcount
|
|
EnterCriticalSection(&CritSec_);
|
|
return(0);
|
|
#else
|
|
#error Must define either _WIN32 or _UNIX
|
|
#endif
|
|
}
|
|
|
|
// The "unlock" function release the critical section.
|
|
sint32 CritSec::unlock(void) RO
|
|
{
|
|
#ifdef _UNIX
|
|
sint32 status = 0;
|
|
|
|
assert(RefCount_ >= 0);
|
|
if (RefCount_ <= 0)
|
|
{
|
|
//ERRMSG("unlocking unlocked mutex!");
|
|
return(-1);
|
|
}
|
|
|
|
///assert(ThreadId_ == pthread_self());
|
|
if (ThreadId_ != pthread_self())
|
|
{
|
|
WRNMSG("tried to unlock a mutex not owned by self");
|
|
return(-1);
|
|
}
|
|
|
|
if (--RefCount_ == 0)
|
|
{
|
|
// Set thread id to zero -- we're going to release mutex
|
|
ThreadId_ = (pthread_t)0;
|
|
|
|
// Unlock the mutex.
|
|
if ((status = pthread_mutex_unlock(&Mutex_)) != 0)
|
|
ERRMSG("pthread_mutex_lock: " << strerror(errno));
|
|
}
|
|
return status;
|
|
#elif defined(_WIN32)
|
|
LeaveCriticalSection(&CritSec_);
|
|
return(0);
|
|
#endif
|
|
}
|