| //===-- PThreadEvent.cpp ----------------------------------------*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // Created by Greg Clayton on 6/16/07. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "PThreadEvent.h" |
| #include "errno.h" |
| #include "DNBLog.h" |
| |
| PThreadEvent::PThreadEvent(uint32_t bits, uint32_t validBits) : |
| m_mutex(), |
| m_set_condition(), |
| m_reset_condition(), |
| m_bits(bits), |
| m_validBits(validBits), |
| m_reset_ack_mask(0) |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x, 0x%8.8x)", this, __FUNCTION__, bits, validBits); |
| } |
| |
| PThreadEvent::~PThreadEvent() |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p %s", this, __PRETTY_FUNCTION__); |
| } |
| |
| |
| uint32_t |
| PThreadEvent::NewEventBit() |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p %s", this, __PRETTY_FUNCTION__); |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| uint32_t mask = 1; |
| while (mask & m_validBits) |
| mask <<= 1; |
| m_validBits |= mask; |
| return mask; |
| } |
| |
| void |
| PThreadEvent::FreeEventBits(const uint32_t mask) |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x)", this, __FUNCTION__, mask); |
| if (mask) |
| { |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| m_bits &= ~mask; |
| m_validBits &= ~mask; |
| } |
| } |
| |
| |
| uint32_t |
| PThreadEvent::GetEventBits() const |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p %s", this, __PRETTY_FUNCTION__); |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| uint32_t bits = m_bits; |
| return bits; |
| } |
| |
| // Replace the event bits with a new bitmask value |
| void |
| PThreadEvent::ReplaceEventBits(const uint32_t bits) |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x)", this, __FUNCTION__, bits); |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| // Make sure we have some bits and that they aren't already set... |
| if (m_bits != bits) |
| { |
| // Figure out which bits are changing |
| uint32_t changed_bits = m_bits ^ bits; |
| // Set the new bit values |
| m_bits = bits; |
| // If any new bits are set, then broadcast |
| if (changed_bits & m_bits) |
| m_set_condition.Broadcast(); |
| } |
| } |
| |
| // Set one or more event bits and broadcast if any new event bits get set |
| // that weren't already set. |
| |
| void |
| PThreadEvent::SetEvents(const uint32_t mask) |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x)", this, __FUNCTION__, mask); |
| // Make sure we have some bits to set |
| if (mask) |
| { |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| // Save the old event bit state so we can tell if things change |
| uint32_t old = m_bits; |
| // Set the all event bits that are set in 'mask' |
| m_bits |= mask; |
| // Broadcast only if any extra bits got set. |
| if (old != m_bits) |
| m_set_condition.Broadcast(); |
| } |
| } |
| |
| // Reset one or more event bits |
| void |
| PThreadEvent::ResetEvents(const uint32_t mask) |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x)", this, __FUNCTION__, mask); |
| if (mask) |
| { |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| |
| // Save the old event bit state so we can tell if things change |
| uint32_t old = m_bits; |
| // Clear the all event bits that are set in 'mask' |
| m_bits &= ~mask; |
| // Broadcast only if any extra bits got reset. |
| if (old != m_bits) |
| m_reset_condition.Broadcast(); |
| } |
| } |
| |
| //---------------------------------------------------------------------- |
| // Wait until 'timeout_abstime' for any events that are set in |
| // 'mask'. If 'timeout_abstime' is NULL, then wait forever. |
| //---------------------------------------------------------------------- |
| uint32_t |
| PThreadEvent::WaitForSetEvents(const uint32_t mask, const struct timespec *timeout_abstime) const |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x, %p)", this, __FUNCTION__, mask, timeout_abstime); |
| int err = 0; |
| // pthread_cond_timedwait() or pthread_cond_wait() will atomically |
| // unlock the mutex and wait for the condition to be set. When either |
| // function returns, they will re-lock the mutex. We use an auto lock/unlock |
| // class (PThreadMutex::Locker) to allow us to return at any point in this |
| // function and not have to worry about unlocking the mutex. |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| do |
| { |
| // Check our predicate (event bits) in case any are already set |
| if (mask & m_bits) |
| { |
| uint32_t bits_set = mask & m_bits; |
| // Our PThreadMutex::Locker will automatically unlock our mutex |
| return bits_set; |
| } |
| if (timeout_abstime) |
| { |
| // Wait for condition to get broadcast, or for a timeout. If we get |
| // a timeout we will drop out of the do loop and return false which |
| // is what we want. |
| err = ::pthread_cond_timedwait (m_set_condition.Condition(), m_mutex.Mutex(), timeout_abstime); |
| // Retest our predicate in case of a race condition right at the end |
| // of the timeout. |
| if (err == ETIMEDOUT) |
| { |
| uint32_t bits_set = mask & m_bits; |
| return bits_set; |
| } |
| } |
| else |
| { |
| // Wait for condition to get broadcast. The only error this function |
| // should return is if |
| err = ::pthread_cond_wait (m_set_condition.Condition(), m_mutex.Mutex()); |
| } |
| } while (err == 0); |
| return 0; |
| } |
| |
| //---------------------------------------------------------------------- |
| // Wait until 'timeout_abstime' for any events in 'mask' to reset. |
| // If 'timeout_abstime' is NULL, then wait forever. |
| //---------------------------------------------------------------------- |
| uint32_t |
| PThreadEvent::WaitForEventsToReset(const uint32_t mask, const struct timespec *timeout_abstime) const |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x, %p)", this, __FUNCTION__, mask, timeout_abstime); |
| int err = 0; |
| // pthread_cond_timedwait() or pthread_cond_wait() will atomically |
| // unlock the mutex and wait for the condition to be set. When either |
| // function returns, they will re-lock the mutex. We use an auto lock/unlock |
| // class (PThreadMutex::Locker) to allow us to return at any point in this |
| // function and not have to worry about unlocking the mutex. |
| PTHREAD_MUTEX_LOCKER (locker, m_mutex); |
| do |
| { |
| // Check our predicate (event bits) each time through this do loop |
| if ((mask & m_bits) == 0) |
| { |
| // All the bits requested have been reset, return zero indicating |
| // which bits from the mask were still set (none of them) |
| return 0; |
| } |
| if (timeout_abstime) |
| { |
| // Wait for condition to get broadcast, or for a timeout. If we get |
| // a timeout we will drop out of the do loop and return false which |
| // is what we want. |
| err = ::pthread_cond_timedwait (m_reset_condition.Condition(), m_mutex.Mutex(), timeout_abstime); |
| } |
| else |
| { |
| // Wait for condition to get broadcast. The only error this function |
| // should return is if |
| err = ::pthread_cond_wait (m_reset_condition.Condition(), m_mutex.Mutex()); |
| } |
| } while (err == 0); |
| // Return a mask indicating which bits (if any) were still set |
| return mask & m_bits; |
| } |
| |
| uint32_t |
| PThreadEvent::WaitForResetAck (const uint32_t mask, const struct timespec *timeout_abstime) const |
| { |
| if (mask & m_reset_ack_mask) |
| { |
| // DNBLogThreadedIf(LOG_EVENTS, "%p PThreadEvent::%s (0x%8.8x, %p)", this, __FUNCTION__, mask, timeout_abstime); |
| return WaitForEventsToReset (mask & m_reset_ack_mask, timeout_abstime); |
| } |
| return 0; |
| } |