001:
002:
003:
004:
005:
006:
007:
008:
009:
010:
011:
012:
013:
014:
015:
016:
017:
018:
019:
020:
021:
022:
023:
024:
025:
026:
027:
028:
029:
030:
031:
032:
033:
034:
035:
036:
037:
038:
039:
040:
041:
042:
043:
044:
045:
046:
047:
048:
049:
050:
051:
052:
053:
054:
055:
056:
057:
058:
059:
060:
061:
062:
063:
064:
065:
066:
067:
068:
069:
070:
071:
072:
073:
074:
075:
076:
077:
078:
079:
080:
081:
082:
083:
084:
085:
086:
087:
088:
089:
090:
091:
092:
093:
094:
095:
096:
097:
098:
099:
100:
101:
102:
103:
104:
105:
106:
107:
108:
109:
110:
111:
112:
113:
114:
115:
116:
117:
118:
119:
120:
121:
122:
123:
124:
125:
126:
127:
128:
129:
130:
131:
132:
133:
134:
135:
136:
137:
138:
139:
140:
141:
142:
143:
144:
145:
146:
147:
148:
149:
150:
151:
152:
153:
154:
155:
156:
157:
158:
159:
160:
161:
162:
163:
164:
165:
166:
167:
168:
169:
170:
171:
172:
173:
174:
175:
176:
177:
178:
179:
180:
181:
182:
183:
184:
185:
186:
187:
188:
189:
190:
191:
192:
193:
194:
195:
196:
197:
198:
199:
200:
201:
202:
203:
204:
205:
206:
207:
208:
209:
210:
211:
212:
213:
214:
215:
216:
217:
218:
219:
220:
221:
222:
223:
224:
225:
226:
227:
228:
229:
230:
231:
232:
233:
234:
235:
236:
237:
238:
239:
240:
241:
242:
243:
244:
245:
246:
247:
248:
249:
250:
251:
252:
253:
254:
255:
256:
257:
258:
259:
260:
261:
262:
263:
264:
265:
266:
267:
268:
269:
270:
271:
272:
273:
274:
275:
276:
277:
278:
279:
280:
281:
282:
283:
284:
285:
286:
287:
288:
289:
290:
291:
292:
293:
294:
295:
296:
297:
298:
299:
300:
301:
302:
303:
304:
305:
306:
307:

/*
* Copyright (c), Zeriph Enterprises
* All rights reserved.
*
* Contributor(s):
* Zechariah Perez, omni (at) zeriph (dot) com
*
* THIS SOFTWARE IS PROVIDED BY ZERIPH 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 ZERIPH AND 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.
*/
#include <omni/sync/semaphore.hpp>
#include <omni/chrono/stopwatch.hpp> // for the wait timeout
#include <omni/system.hpp> // OMNI_GLE_PRNT
#include <omni/exception.hpp>
#if defined(OMNI_SAFE_SEMAPHORE)
#define OMNI_SAFE_SEM_MILST_FW , m_mtx()
#define OMNI_SAFE_SEM_INIT_FW omni::sync::mutex_init(this->m_mtx);
#define OMNI_SAFE_SEM_DTOR_FW omni::sync::mutex_destroy(this->m_mtx);
#define OMNI_SAFE_SEM_LOCK_FW omni::sync::mutex_lock(this->m_mtx);
#define OMNI_SAFE_SEM_UNLOCK_FW omni::sync::mutex_unlock(this->m_mtx);
#define OMNI_SAFE_SEM_ALOCK_FW omni::sync::scoped_lock<omni::sync::mutex_t> smal(&this->m_mtx);
#define OMNI_SAFE_SEM_EQU_FW(o) (&this->m_mtx == &o.m_mtx) &&
#else
#define OMNI_SAFE_SEM_MILST_FW
#define OMNI_SAFE_SEM_INIT_FW
#define OMNI_SAFE_SEM_DTOR_FW
#define OMNI_SAFE_SEM_LOCK_FW
#define OMNI_SAFE_SEM_UNLOCK_FW
#define OMNI_SAFE_SEM_ALOCK_FW
#define OMNI_SAFE_SEM_EQU_FW(o)
#endif
omni::sync::semaphore::semaphore() :
OMNI_CTOR_FW(omni::sync::semaphore)
m_cnt(0),
m_max(),
m_sem()
OMNI_SAFE_SEM_MILST_FW
{
this->_init(5);
}
omni::sync::semaphore::semaphore(uint32_t maxent) :
OMNI_CTOR_FW(omni::sync::semaphore)
m_cnt(0),
m_max(),
m_sem()
OMNI_SAFE_SEM_MILST_FW
{
this->_init(maxent);
}
omni::sync::semaphore::semaphore(uint32_t maxent, bool lock1st) :
OMNI_CTOR_FW(omni::sync::semaphore)
m_cnt(0),
m_max(),
m_sem()
OMNI_SAFE_SEM_MILST_FW
{
this->_init(maxent);
if (lock1st) { this->wait(); }
}
omni::sync::semaphore::~semaphore()
{
OMNI_TRY_FW
OMNI_DTOR_FW
this->_dispose();
OMNI_D5_FW("destroyed");
OMNI_CATCH_FW
}
omni::sync::semaphore_t omni::sync::semaphore::handle() const
{
return this->m_sem;
}
uint32_t omni::sync::semaphore::locked() const
{
OMNI_SAFE_SEM_ALOCK_FW
return this->m_cnt;
}
uint32_t omni::sync::semaphore::max_ent() const
{
OMNI_SAFE_SEM_ALOCK_FW
return this->m_max;
}
uint32_t omni::sync::semaphore::open() const
{
OMNI_SAFE_SEM_ALOCK_FW
return (this->m_max - this->m_cnt);
}
uint32_t omni::sync::semaphore::release_all()
{
OMNI_SAFE_SEM_LOCK_FW
uint32_t cnt = this->m_cnt;
OMNI_SAFE_SEM_UNLOCK_FW
return this->release(cnt);
}
uint32_t omni::sync::semaphore::release(uint32_t rcount)
{
OMNI_SAFE_SEM_ALOCK_FW
uint32_t pcnt = this->m_cnt;
if (rcount == 0 || rcount > pcnt) {
OMNI_ERR_RETV_FW("an invalid release count was specified", omni::exceptions::invalid_release_count(), 0)
}
#if defined(OMNI_OS_WIN)
if (::ReleaseSemaphore(this->m_sem, rcount, NULL) == 0) {
OMNI_ERRV_RETV_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT, omni::exceptions::semaphore_system_exception(OMNI_GLE_PRNT), 0)
} else {
this->m_cnt -= rcount;
}
#else
for (uint32_t i = rcount; i > 0; --i) {
#if defined(OMNI_OS_APPLE)
if (::semaphore_signal(this->m_sem) != KERN_SUCCESS)
#else
if (::sem_post(&this->m_sem) == -1)
#endif
{
OMNI_ERRV_RETV_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT, omni::exceptions::semaphore_system_exception(OMNI_GLE_PRNT), 0)
} else {
--this->m_cnt;
}
}
#endif
OMNI_DV3_FW("new count of ", this->m_cnt);
return pcnt; // return old count
}
bool omni::sync::semaphore::wait()
{
bool aq = false;
#if defined(OMNI_OS_WIN)
switch(::WaitForSingleObject(this->m_sem, INFINITE)) {
case WAIT_OBJECT_0: aq = true; break; // 0x00000000L
#if defined(OMNI_DBG_L3)
// If debug is not enabled the following flags are not relevant to this context
// WAIT_FAILED (0xFFFFFFFF) is something to note, but in this context, it has not effect
case WAIT_FAILED: OMNI_DBGEV("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT); break;
OMNI_DSW_FW(WAIT_ABANDONED); // 0x00000080L
OMNI_DSW_FW(WAIT_TIMEOUT); // 0x00000102L
#endif
default: break;
}
#else
#if defined(OMNI_OS_APPLE)
if (::semaphore_wait(this->m_sem) == KERN_SUCCESS)
#else
if (::sem_wait(&this->m_sem) != -1)
#endif
{ aq = true; }
#if defined(OMNI_DBG_L3)
else { OMNI_DV3_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT); }
#endif
#endif
if (aq) {
OMNI_SAFE_SEM_LOCK_FW
++this->m_cnt;
OMNI_DV4_FW("new count of ", this->m_cnt);
OMNI_SAFE_SEM_UNLOCK_FW
}
return aq;
}
bool omni::sync::semaphore::wait(uint32_t timeout_ms)
{
bool aq = false;
#if defined(OMNI_OS_WIN)
switch(::WaitForSingleObject(this->m_sem, timeout_ms)) {
case WAIT_OBJECT_0: aq = true; break; // 0x00000000L
#if defined(OMNI_DBG_L3)
case WAIT_FAILED: OMNI_DV3_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT); break;
OMNI_DSW_FW(WAIT_ABANDONED);
OMNI_DSW_FW(WAIT_TIMEOUT);
#endif
default: break;
}
#elif defined(OMNI_OS_APPLE)
mach_timespec_t tm;
if (timeout_ms > 0) {
if (timeout_ms >= 1000) { tm.tv_sec = (timeout_ms / 1000); }
tm.tv_nsec = ((timeout_ms % 1000) * 1000);
}
switch(::semaphore_timedwait(this->m_sem, tm)) {
case KERN_SUCCESS:
aq = true; break;
#if defined(OMNI_DBG_L3)
OMNI_DSW_FW(KERN_OPERATION_TIMED_OUT);
#endif
default:
OMNI_DV3_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT);
break;
}
#else
if (timeout_ms == 0) {
if (::sem_trywait(&this->m_sem) == 0) { aq = true; }
#if defined(OMNI_DBG_L3)
else { OMNI_DV3_FW("a system error occurred on the semaphore object : ", OMNI_GLE_PRNT); }
#endif
} else {
omni::chrono::tick_t start = omni::chrono::monotonic_tick();
do {
if (::sem_trywait(&this->m_sem) == 0) {
aq = true;
break;
}
#if defined(OMNI_DBG_L5) // normally L3 level, but inside loop == l5
else { OMNI_DV3_FW("a system error occurred on the semaphore object : ", OMNI_GLE_PRNT); }
#endif
} while (omni::chrono::elapsed_ms(start) <= timeout_ms);
}
#endif
if (aq) {
OMNI_SAFE_SEM_LOCK_FW
++this->m_cnt;
OMNI_DV3_FW("new count of ", this->m_cnt);
OMNI_SAFE_SEM_UNLOCK_FW
}
return aq;
}
bool omni::sync::semaphore::wait(const omni::chrono::unsigned_span_t& span)
{
return this->wait(static_cast<uint32_t>(span.total_milliseconds()));
}
bool omni::sync::semaphore::trywait()
{
return this->wait(0);
}
bool omni::sync::semaphore::operator==(const omni::sync::semaphore& o) const
{
if (this == &o) { return true; }
uint32_t ocnt = o.locked();
OMNI_SAFE_SEM_ALOCK_FW
return (OMNI_SAFE_SEM_EQU_FW(o)
this->m_cnt == ocnt &&
this->m_max == o.m_max &&
&this->m_sem == &o.m_sem)
OMNI_EQUAL_FW(o);
}
bool omni::sync::semaphore::operator!=(const omni::sync::semaphore& o) const
{
return !(*this == o);
}
///////// start private methods /////////
void omni::sync::semaphore::_dispose()
{
OMNI_SAFE_SEM_LOCK_FW
bool await = (this->m_cnt > 0);
OMNI_SAFE_SEM_UNLOCK_FW
OMNI_SAFE_SEM_DTOR_FW
if (await) {
OMNI_ERR_FW("the semaphore has active waits on call to dispose", omni::exceptions::active_wait_exception())
}
#if defined(OMNI_OS_WIN)
if (::CloseHandle(this->m_sem) == 0)
#elif defined(OMNI_OS_APPLE)
if (::semaphore_destroy(::mach_task_self(), this->m_sem) != KERN_SUCCESS)
#else
if (::sem_destroy(&this->m_sem) == -1)
#endif
{
OMNI_ERRV_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT, omni::exceptions::semaphore_system_exception(OMNI_GLE_PRNT))
}
}
void omni::sync::semaphore::_init(uint32_t maxent)
{
#if !defined(OMNI_CHRONO_AUTO_INIT_TICK)
omni::chrono::monotonic::initialize();
#endif
OMNI_SAFE_SEM_INIT_FW
#if defined(OMNI_OS_WIN)
this->m_sem = ::CreateSemaphore(NULL, maxent, maxent, NULL);
if (this->m_sem == NULL)
#elif defined(OMNI_OS_APPLE)
if (::semaphore_create(::mach_task_self(), &this->m_sem, SYNC_POLICY_FIFO, static_cast<int>(maxent)) != KERN_SUCCESS)
#else
if (::sem_init(&this->m_sem, 0, static_cast<int>(maxent)) == -1)
#endif
{
// this will hit as well if maxent > SEM_VALUE_MAX
OMNI_ERRV_FW("a system error occurred on the semaphore object: ", OMNI_GLE_PRNT, omni::exceptions::semaphore_system_exception(OMNI_GLE_PRNT))
}
this->m_max = maxent;
OMNI_D5_FW("created");
}