You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
141 lines
3.0 KiB
141 lines
3.0 KiB
/* |
|
* ZeroTier One - Global Peer to Peer Ethernet |
|
* Copyright (C) 2011-2014 ZeroTier Networks LLC |
|
* |
|
* 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/>. |
|
* |
|
* -- |
|
* |
|
* ZeroTier may be used and distributed under the terms of the GPLv3, which |
|
* are available at: http://www.gnu.org/licenses/gpl-3.0.html |
|
* |
|
* If you would like to embed ZeroTier into a commercial application or |
|
* redistribute it in a modified binary form, please contact ZeroTier Networks |
|
* LLC. Start here: http://www.zerotier.com/ |
|
*/ |
|
|
|
#ifndef ZT_CONDITION_HPP |
|
#define ZT_CONDITION_HPP |
|
|
|
#include "Constants.hpp" |
|
#include "NonCopyable.hpp" |
|
|
|
#ifdef __WINDOWS__ |
|
|
|
#include <Windows.h> |
|
#include <stdlib.h> |
|
|
|
#include "Utils.hpp" |
|
|
|
namespace ZeroTier { |
|
|
|
class Condition : NonCopyable |
|
{ |
|
public: |
|
Condition() |
|
throw() |
|
{ |
|
_sem = CreateSemaphore(NULL,0,1,NULL); |
|
} |
|
|
|
~Condition() |
|
{ |
|
CloseHandle(_sem); |
|
} |
|
|
|
inline void wait() const |
|
throw() |
|
{ |
|
WaitForSingleObject(_sem,INFINITE); |
|
} |
|
|
|
inline void wait(unsigned long ms) const |
|
throw() |
|
{ |
|
WaitForSingleObject(_sem,(DWORD)ms); |
|
} |
|
|
|
inline void signal() const |
|
throw() |
|
{ |
|
ReleaseSemaphore(_sem,1,NULL); |
|
} |
|
|
|
private: |
|
HANDLE _sem; |
|
}; |
|
|
|
} // namespace ZeroTier |
|
|
|
#else // !__WINDOWS__ |
|
|
|
#include <time.h> |
|
#include <stdlib.h> |
|
#include <pthread.h> |
|
#include "Utils.hpp" |
|
|
|
namespace ZeroTier { |
|
|
|
class Condition : NonCopyable |
|
{ |
|
public: |
|
Condition() |
|
throw() |
|
{ |
|
pthread_mutex_init(&_mh,(const pthread_mutexattr_t *)0); |
|
pthread_cond_init(&_cond,(const pthread_condattr_t *)0); |
|
} |
|
|
|
~Condition() |
|
{ |
|
pthread_cond_destroy(&_cond); |
|
pthread_mutex_destroy(&_mh); |
|
} |
|
|
|
inline void wait() const |
|
throw() |
|
{ |
|
pthread_mutex_lock(const_cast <pthread_mutex_t *>(&_mh)); |
|
pthread_cond_wait(const_cast <pthread_cond_t *>(&_cond),const_cast <pthread_mutex_t *>(&_mh)); |
|
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh)); |
|
} |
|
|
|
inline void wait(unsigned long ms) const |
|
throw() |
|
{ |
|
uint64_t when = Utils::now() + (uint64_t)ms; |
|
struct timespec ts; |
|
ts.tv_sec = (unsigned long)(when / 1000); |
|
ts.tv_nsec = (unsigned long)(when % 1000) * 1000000; |
|
pthread_mutex_lock(const_cast <pthread_mutex_t *>(&_mh)); |
|
pthread_cond_timedwait(const_cast <pthread_cond_t *>(&_cond),const_cast <pthread_mutex_t *>(&_mh),&ts); |
|
pthread_mutex_unlock(const_cast <pthread_mutex_t *>(&_mh)); |
|
} |
|
|
|
inline void signal() const |
|
throw() |
|
{ |
|
pthread_cond_signal(const_cast <pthread_cond_t *>(&_cond)); |
|
} |
|
|
|
private: |
|
pthread_cond_t _cond; |
|
pthread_mutex_t _mh; |
|
}; |
|
|
|
} // namespace ZeroTier |
|
|
|
#endif // !__WINDOWS__ |
|
|
|
#endif
|
|
|