137 lines
3.3 KiB
C++
137 lines
3.3 KiB
C++
/*
|
|
* Copyright (c) 2004, Matthew P. Cashdollar <mpc@innographx.com>
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are
|
|
* met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* * Neither the name of the author nor the names of any contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
|
|
* IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
|
|
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
|
|
* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
|
* OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
// Modelled after JThread by Jori Liesenborgs
|
|
|
|
#include <cassert>
|
|
#include "platform.hpp"
|
|
#ifdef WINTHREAD
|
|
#include <windows.h>
|
|
#else
|
|
#include <pthread.h>
|
|
#endif
|
|
using namespace std;
|
|
#include "mutex.hpp"
|
|
using namespace Libsockthread;
|
|
|
|
/*
|
|
* Creates a mutex
|
|
*/
|
|
Mutex::Mutex(void)
|
|
{
|
|
#ifdef WINTHREAD
|
|
mutex = CreateMutex(0, false, 0);
|
|
assert(mutex != 0);
|
|
#else
|
|
int rc = pthread_mutex_init(&mutex, 0);
|
|
assert(rc == 0);
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* Destroys a mutex
|
|
*/
|
|
Mutex::~Mutex(void)
|
|
{
|
|
#ifdef WINTHREAD
|
|
BOOL rc = CloseHandle(mutex);
|
|
assert(rc);
|
|
#else
|
|
int rc = pthread_mutex_destroy(&mutex);
|
|
assert(rc == 0);
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* Locks the mutex
|
|
*/
|
|
void Mutex::lock(void)
|
|
{
|
|
#ifdef WINTHREAD
|
|
DWORD rc = WaitForSingleObject(mutex, INFINITE);
|
|
assert(rc != WAIT_FAILED);
|
|
#else
|
|
int rc = pthread_mutex_lock(&mutex);
|
|
assert(rc == 0);
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* Unlocks the mutex
|
|
*/
|
|
void Mutex::unlock(void)
|
|
{
|
|
#ifdef WINTHREAD
|
|
BOOL rc = ReleaseMutex(mutex);
|
|
assert(rc);
|
|
#else
|
|
int rc = pthread_mutex_unlock(&mutex);
|
|
assert(rc == 0);
|
|
#endif
|
|
}
|
|
|
|
#ifdef UNIT_TEST
|
|
// g++ -Wall -c thread.cpp -o thread.o
|
|
// g++ -Wall -DUNIT_TEST -c mutex.cpp -o mutex.o
|
|
// g++ -Wall -DUNIT_TEST mutex.o thread.o -o mutex -lpthread
|
|
#include <iostream>
|
|
#include "thread.hpp"
|
|
|
|
Mutex widget;
|
|
|
|
int main(void)
|
|
{
|
|
class Mutex_test : public Thread
|
|
{
|
|
public:
|
|
Mutex_test(int n)
|
|
: testval(n) {}
|
|
|
|
void* thread(void)
|
|
{
|
|
widget.lock();
|
|
cout << "I got it! thread #" << testval << '\n';
|
|
// If this works, only one thread should be able to lock the
|
|
// widget, since it is never unlocked
|
|
return 0;
|
|
}
|
|
private:
|
|
int testval;
|
|
};
|
|
Mutex_test t1(1);
|
|
Mutex_test t2(2);
|
|
Mutex_test t3(3);
|
|
t1.start(); t2.start(); t3.start();
|
|
while (true);
|
|
|
|
return 0;
|
|
}
|
|
#endif // UNIT_TEST
|