/* * Copyright (c) 1998-2012 Apple Inc. All rights reserved. * * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ * * This file contains Original Code and/or Modifications of Original Code * as defined in and that are subject to the Apple Public Source License * Version 2.0 (the 'License'). You may not use this file except in * compliance with the License. The rights granted to you under the License * may not be used to create, or enable the creation or redistribution of, * unlawful or unlicensed copies of an Apple operating system, or to * circumvent, violate, or enable the circumvention or violation of, any * terms of an Apple operating system software license agreement. * * Please obtain a copy of the License at * http://www.opensource.apple.com/apsl/ and read it before using this file. * * The Original Code and all software distributed under the License are * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. * Please see the License for the specific language governing rights and * limitations under the License. * * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ */ /* * */ #ifndef __IOKIT_IOLOCKS_H #define __IOKIT_IOLOCKS_H #ifndef KERNEL #error IOLocks.h is for kernel use only #endif #include <sys/appleapiopts.h> #include <sys/cdefs.h> #include <IOKit/system.h> #include <IOKit/IOReturn.h> #include <IOKit/IOTypes.h> #ifdef __cplusplus extern "C" { #endif #include <libkern/locks.h> #include <machine/machine_routines.h> /*! @var IOLockGroup * Global lock group used by all IOKit locks. To simplify kext debugging and lock-heat analysis, consider using lck_* locks with a per-driver lock group, as defined in kern/locks.h. */ extern lck_grp_t *IOLockGroup; #if defined(XNU_KERNEL_PRIVATE) #define IOLOCKS_INLINE 1 #endif /* * Mutex lock operations */ #ifdef IOLOCKS_INLINE typedef lck_mtx_t IOLock; #else typedef struct _IOLock IOLock; #endif /* IOLOCKS_INLINE */ /*! @function IOLockAlloc * @abstract Allocates and initializes a mutex. * @discussion Allocates a mutex in general purpose memory, and initializes it. Mutexes are general purpose blocking mutual exclusion locks, supplied by libkern/locks.h. This function may block and so should not be called from interrupt level or while a spin lock is held. IOLocks use the global IOKit lock group, IOLockGroup. To simplify kext debugging and lock-heat analysis, consider using lck_* locks with a per-driver lock group, as defined in kern/locks.h. * @result Pointer to the allocated lock, or zero on failure. */ IOLock * IOLockAlloc( void ); /*! @function IOLockFree * @abstract Frees a mutex. * @discussion Frees a lock allocated with IOLockAlloc. Mutex should be unlocked with no waiters. * @param lock Pointer to the allocated lock. */ void IOLockFree( IOLock * lock); /*! @function IOLockGetMachLock * @abstract Accessor to a Mach mutex. * @discussion Accessor to the Mach mutex. * @param lock Pointer to the allocated lock. */ lck_mtx_t * IOLockGetMachLock( IOLock * lock); /*! @function IOLockLock * @abstract Lock a mutex. * @discussion Lock the mutex. If the lock is held by any thread, block waiting for its unlock. This function may block and so should not be called from interrupt level or while a spin lock is held. Locking the mutex recursively from one thread will result in deadlock. * @param lock Pointer to the allocated lock. */ #ifdef IOLOCKS_INLINE #define IOLockLock(l) lck_mtx_lock(l) #else void IOLockLock( IOLock * lock); #endif /* !IOLOCKS_INLINE */ /*! @function IOLockTryLock * @abstract Attempt to lock a mutex. * @discussion Lock the mutex if it is currently unlocked, and return true. If the lock is held by any thread, return false. * @param lock Pointer to the allocated lock. * @result True if the mutex was unlocked and is now locked by the caller, otherwise false. */ #ifdef IOLOCKS_INLINE #define IOLockTryLock(l) lck_mtx_try_lock(l) #else boolean_t IOLockTryLock( IOLock * lock); #endif /* !IOLOCKS_INLINE */ /*! @function IOLockUnlock * @abstract Unlock a mutex. * @discussion Unlock the mutex and wake any blocked waiters. Results are undefined if the caller has not locked the mutex. This function may block and so should not be called from interrupt level or while a spin lock is held. * @param lock Pointer to the allocated lock. */ #ifdef IOLOCKS_INLINE #define IOLockUnlock(l) lck_mtx_unlock(l) #else void IOLockUnlock( IOLock * lock); #endif /* !IOLOCKS_INLINE */ /*! @function IOLockSleep * @abstract Sleep with mutex unlock and relock * @discussion Prepare to sleep,unlock the mutex, and re-acquire it on wakeup. Results are undefined if the caller has not locked the mutex. This function may block and so should not be called from interrupt level or while a spin lock is held. * @param lock Pointer to the locked lock. * @param event The event to sleep on. Must be non-NULL. * @param interType How can the sleep be interrupted. * @result The wait-result value indicating how the thread was awakened.*/ int IOLockSleep( IOLock * lock, void *event, UInt32 interType) __DARWIN14_ALIAS(IOLockSleep); int IOLockSleepDeadline( IOLock * lock, void *event, AbsoluteTime deadline, UInt32 interType) __DARWIN14_ALIAS(IOLockSleepDeadline); void IOLockWakeup(IOLock * lock, void *event, bool oneThread) __DARWIN14_ALIAS(IOLockWakeup); #ifdef XNU_KERNEL_PRIVATE /*! @enum IOLockAssertState * @abstract Used with IOLockAssert to assert the state of a lock. */ typedef enum { kIOLockAssertOwned = LCK_ASSERT_OWNED, kIOLockAssertNotOwned = LCK_ASSERT_NOTOWNED } IOLockAssertState; #ifdef IOLOCKS_INLINE #define IOLockAssert(l, type) LCK_MTX_ASSERT(l, type) #else /*! @function IOLockAssert * @abstract Assert that lock is either held or not held by current thread. * @discussion Call with either kIOLockAssertOwned or kIOLockAssertNotOwned. * Panics the kernel if the lock is not owned if called with kIOLockAssertOwned, * and vice-versa. */ void IOLockAssert(IOLock * lock, IOLockAssertState type); #endif /* !IOLOCKS_INLINE */ #endif /* !XNU_KERNEL_PRIVATE */ #ifdef __APPLE_API_OBSOLETE /* The following API is deprecated */ typedef enum { kIOLockStateUnlocked = 0, kIOLockStateLocked = 1 } IOLockState; void IOLockInitWithState( IOLock * lock, IOLockState state); #define IOLockInit( l ) IOLockInitWithState( l, kIOLockStateUnlocked); static __inline__ void IOTakeLock( IOLock * lock) { IOLockLock(lock); } static __inline__ boolean_t IOTryLock( IOLock * lock) { return IOLockTryLock(lock); } static __inline__ void IOUnlock( IOLock * lock) { IOLockUnlock(lock); } #endif /* __APPLE_API_OBSOLETE */ /* * Recursive lock operations */ typedef struct _IORecursiveLock IORecursiveLock; /*! @function IORecursiveLockAlloc * @abstract Allocates and initializes an recursive lock. * @discussion Allocates a recursive lock in general purpose memory, and initializes it. Recursive locks function identically to mutexes but allow one thread to lock more than once, with balanced unlocks. IORecursiveLocks use the global IOKit lock group, IOLockGroup. To simplify kext debugging and lock-heat analysis, consider using lck_* locks with a per-driver lock group, as defined in kern/locks.h. * @result Pointer to the allocated lock, or zero on failure. */ IORecursiveLock * IORecursiveLockAlloc( void ); /*! @function IORecursiveLockFree * @abstract Frees a recursive lock. * @discussion Frees a lock allocated with IORecursiveLockAlloc. Lock should be unlocked with no waiters. * @param lock Pointer to the allocated lock. */ void IORecursiveLockFree( IORecursiveLock * lock); /*! @function IORecursiveLockGetMachLock * @abstract Accessor to a Mach mutex. * @discussion Accessor to the Mach mutex. * @param lock Pointer to the allocated lock. */ lck_mtx_t * IORecursiveLockGetMachLock( IORecursiveLock * lock); /*! @function IORecursiveLockLock * @abstract Lock a recursive lock. * @discussion Lock the recursive lock. If the lock is held by another thread, block waiting for its unlock. This function may block and so should not be called from interrupt level or while a spin lock is held. The lock may be taken recursively by the same thread, with a balanced number of calls to IORecursiveLockUnlock. * @param lock Pointer to the allocated lock. */ void IORecursiveLockLock( IORecursiveLock * lock); /*! @function IORecursiveLockTryLock * @abstract Attempt to lock a recursive lock. * @discussion Lock the lock if it is currently unlocked, or held by the calling thread, and return true. If the lock is held by another thread, return false. Successful calls to IORecursiveLockTryLock should be balanced with calls to IORecursiveLockUnlock. * @param lock Pointer to the allocated lock. * @result True if the lock is now locked by the caller, otherwise false. */ boolean_t IORecursiveLockTryLock( IORecursiveLock * lock); /*! @function IORecursiveLockUnlock * @abstract Unlock a recursive lock. * @discussion Undo one call to IORecursiveLockLock, if the lock is now unlocked wake any blocked waiters. Results are undefined if the caller does not balance calls to IORecursiveLockLock with IORecursiveLockUnlock. This function may block and so should not be called from interrupt level or while a spin lock is held. * @param lock Pointer to the allocated lock. */ void IORecursiveLockUnlock( IORecursiveLock * lock); /*! @function IORecursiveLockHaveLock * @abstract Check if a recursive lock is held by the calling thread. * @discussion If the lock is held by the calling thread, return true, otherwise the lock is unlocked, or held by another thread and false is returned. * @param lock Pointer to the allocated lock. * @result True if the calling thread holds the lock otherwise false. */ boolean_t IORecursiveLockHaveLock( const IORecursiveLock * lock); extern int IORecursiveLockSleep( IORecursiveLock *_lock, void *event, UInt32 interType); extern int IORecursiveLockSleepDeadline( IORecursiveLock * _lock, void *event, AbsoluteTime deadline, UInt32 interType); extern void IORecursiveLockWakeup( IORecursiveLock *_lock, void *event, bool oneThread); /* * Complex (read/write) lock operations */ #ifdef IOLOCKS_INLINE typedef lck_rw_t IORWLock; #else typedef struct _IORWLock IORWLock; #endif /* IOLOCKS_INLINE */ /*! @function IORWLockAlloc * @abstract Allocates and initializes a read/write lock. * @discussion Allocates and initializes a read/write lock in general purpose memory. Read/write locks provide for multiple readers, one exclusive writer, and are supplied by libkern/locks.h. This function may block and so should not be called from interrupt level or while a spin lock is held. IORWLocks use the global IOKit lock group, IOLockGroup. To simplify kext debugging and lock-heat analysis, consider using lck_* locks with a per-driver lock group, as defined in kern/locks.h. * @result Pointer to the allocated lock, or zero on failure. */ IORWLock * IORWLockAlloc( void ); /*! @function IORWLockFree * @abstract Frees a read/write lock. * @discussion Frees a lock allocated with IORWLockAlloc. Lock should be unlocked with no waiters. * @param lock Pointer to the allocated lock. */ void IORWLockFree( IORWLock * lock); /*! @function IORWLockGetMachLock * @abstract Accessor to a Mach read/write lock. * @discussion Accessor to the Mach read/write lock. * @param lock Pointer to the allocated lock. */ lck_rw_t * IORWLockGetMachLock( IORWLock * lock); /*! @function IORWLockRead * @abstract Lock a read/write lock for read. * @discussion Lock the lock for read, allowing multiple readers when there are no writers. If the lock is held for write, block waiting for its unlock. This function may block and so should not be called from interrupt level or while a spin lock is held. Locking the lock recursively from one thread, for read or write, can result in deadlock. * @param lock Pointer to the allocated lock. */ #ifdef IOLOCKS_INLINE #define IORWLockRead(l) lck_rw_lock_shared(l) #else void IORWLockRead(IORWLock * lock); #endif /* !IOLOCKS_INLINE */ /*! @function IORWLockWrite * @abstract Lock a read/write lock for write. * @discussion Lock the lock for write, allowing one writer exlusive access. If the lock is held for read or write, block waiting for its unlock. This function may block and so should not be called from interrupt level or while a spin lock is held. Locking the lock recursively from one thread, for read or write, can result in deadlock. * @param lock Pointer to the allocated lock. */ #ifdef IOLOCKS_INLINE #define IORWLockWrite(l) lck_rw_lock_exclusive(l) #else void IORWLockWrite( IORWLock * lock); #endif /* !IOLOCKS_INLINE */ /*! @function IORWLockUnlock * @abstract Unlock a read/write lock. * @discussion Undo one call to IORWLockRead or IORWLockWrite. Results are undefined if the caller has not locked the lock. This function may block and so should not be called from interrupt level or while a spin lock is held. * @param lock Pointer to the allocated lock. */ #ifdef IOLOCKS_INLINE #define IORWLockUnlock(l) lck_rw_done(l) #else void IORWLockUnlock( IORWLock * lock); #endif /* !IOLOCKS_INLINE */ #ifdef XNU_KERNEL_PRIVATE /*! @enum IORWLockAssertState * @abstract Used with IORWLockAssert to assert the state of a lock. */ typedef enum { kIORWLockAssertRead = LCK_RW_ASSERT_SHARED, kIORWLockAssertWrite = LCK_RW_ASSERT_EXCLUSIVE, kIORWLockAssertHeld = LCK_RW_ASSERT_HELD, kIORWLockAssertNotHeld = LCK_RW_ASSERT_NOTHELD } IORWLockAssertState; #ifdef IOLOCKS_INLINE #define IORWLockAssert(l, type) LCK_RW_ASSERT(l, type) #else /*! @function IORWLockAssert * @abstract Assert that a reader-writer lock is either held or not held * by the current thread. * @discussion Call with a value defined by the IORWLockAssertState type. * If the specified lock is not in the state specified by the type argument, * then the kernel will panic. */ void IORWLockAssert(IORWLock * lock, IORWLockAssertState type); #endif /* !IOLOCKS_INLINE */ #endif /* !XNU_KERNEL_PRIVATE */ #ifdef __APPLE_API_OBSOLETE /* The following API is deprecated */ static __inline__ void IOReadLock( IORWLock * lock) { IORWLockRead(lock); } static __inline__ void IOWriteLock( IORWLock * lock) { IORWLockWrite(lock); } static __inline__ void IORWUnlock( IORWLock * lock) { IORWLockUnlock(lock); } #endif /* __APPLE_API_OBSOLETE */ /* * Simple locks. Cannot block while holding a simple lock. */ #ifdef IOLOCKS_INLINE typedef lck_spin_t IOSimpleLock; #else typedef struct _IOSimpleLock IOSimpleLock; #endif /* IOLOCKS_INLINE */ /*! @function IOSimpleLockAlloc * @abstract Allocates and initializes a spin lock. * @discussion Allocates and initializes a spin lock in general purpose memory. Spin locks provide non-blocking mutual exclusion for synchronization between thread context and interrupt context, or for multiprocessor synchronization, and are supplied by libkern/locks.h. This function may block and so should not be called from interrupt level or while a spin lock is held. IOSimpleLocks use the global IOKit lock group, IOLockGroup. To simplify kext debugging and lock-heat analysis, consider using lck_* locks with a per-driver lock group, as defined in kern/locks.h. * @result Pointer to the allocated lock, or zero on failure. */ IOSimpleLock * IOSimpleLockAlloc( void ); /*! @function IOSimpleLockFree * @abstract Frees a spin lock. * @discussion Frees a lock allocated with IOSimpleLockAlloc. * @param lock Pointer to the lock. */ void IOSimpleLockFree( IOSimpleLock * lock ); /*! @function IOSimpleLockGetMachLock * @abstract Accessor to a Mach spin lock. * @discussion Accessor to the Mach spin lock. * @param lock Pointer to the allocated lock. */ lck_spin_t * IOSimpleLockGetMachLock( IOSimpleLock * lock); /*! @function IOSimpleLockInit * @abstract Initialize a spin lock. * @discussion Initialize a non heap allocated spin lock to the unlocked state. Use this function when your lock is, for example, a member variable. You will need to call IOSimpleLockDestroy when you are finished with the lock to avoid lock group refcount leaks. * @param lock Pointer to the lock. */ void IOSimpleLockInit( IOSimpleLock * lock ); /*! @function IOSimpleLockDestroy * @abstract De-initializes (destroys) a spin lock initialized with IOSimpleLockInit * @discussion Destroy / De-initialize a non heap allocated spin lock, releasing any system resources such as lock group refcounts. * @param lock Pointer to the lock. */ void IOSimpleLockDestroy( IOSimpleLock * lock ); /*! @function IOSimpleLockLock * @abstract Lock a spin lock. * @discussion Lock the spin lock. If the lock is held, spin waiting for its unlock. Spin locks disable preemption, cannot be held across any blocking operation, and should be held for very short periods. When used to synchronize between interrupt context and thread context they should be locked with interrupts disabled - IOSimpleLockLockDisableInterrupt() will do both. Locking the lock recursively from one thread will result in deadlock. * @param lock Pointer to the lock. */ #ifdef IOLOCKS_INLINE #define IOSimpleLockLock(l) lck_spin_lock(l) #else void IOSimpleLockLock( IOSimpleLock * lock ); #endif /* !IOLOCKS_INLINE */ /*! @function IOSimpleLockTryLock * @abstract Attempt to lock a spin lock. * @discussion Lock the spin lock if it is currently unlocked, and return true. If the lock is held, return false. Successful calls to IOSimpleLockTryLock should be balanced with calls to IOSimpleLockUnlock. * @param lock Pointer to the lock. * @result True if the lock was unlocked and is now locked by the caller, otherwise false. */ #ifdef IOLOCKS_INLINE #define IOSimpleLockTryLock(l) lck_spin_try_lock(l) #else boolean_t IOSimpleLockTryLock( IOSimpleLock * lock ); #endif /* !IOLOCKS_INLINE */ /*! @function IOSimpleLockUnlock * @abstract Unlock a spin lock. * @discussion Unlock the lock, and restore preemption. Results are undefined if the caller has not locked the lock. * @param lock Pointer to the lock. */ #ifdef IOLOCKS_INLINE #define IOSimpleLockUnlock(l) lck_spin_unlock(l) #else void IOSimpleLockUnlock( IOSimpleLock * lock ); #endif /* !IOLOCKS_INLINE */ #ifdef XNU_KERNEL_PRIVATE /*! @enum IOSimpleLockAssertState * @abstract Used with IOSimpleLockAssert to assert the state of a lock. */ typedef enum { kIOSimpleLockAssertOwned = LCK_ASSERT_OWNED, kIOSimpleLockAssertNotOwned = LCK_ASSERT_NOTOWNED } IOSimpleLockAssertState; #ifdef IOLOCKS_INLINE #define IOSimpleLockAssert(l, type) LCK_SPIN_ASSERT(l, type) #else /*! @function IOSimpleLockAssert * @abstract Assert that spinlock is either held or not held by current thread. * @discussion Call with either kIOSimpleLockAssertOwned or kIOSimpleLockAssertNotOwned. * Panics the kernel if the lock is not owned if called with * kIOSimpleLockAssertOwned and vice-versa. */ void IOSimpleLockAssert(IOSimpleLock *lock, IOSimpleLockAssertState type); #endif /* !IOLOCKS_INLINE */ #endif /* !XNU_KERNEL_PRIVATE */ #if __LP64__ typedef boolean_t IOInterruptState; #else typedef long int IOInterruptState; #endif /*! @function IOSimpleLockLockDisableInterrupt * @abstract Lock a spin lock. * @discussion Lock the spin lock. If the lock is held, spin waiting for its unlock. Simple locks disable preemption, cannot be held across any blocking operation, and should be held for very short periods. When used to synchronize between interrupt context and thread context they should be locked with interrupts disabled - IOSimpleLockLockDisableInterrupt() will do both. Locking the lock recursively from one thread will result in deadlock. * @param lock Pointer to the lock. */ static __inline__ IOInterruptState IOSimpleLockLockDisableInterrupt( IOSimpleLock * lock ) { IOInterruptState state = ml_set_interrupts_enabled( false ); IOSimpleLockLock( lock ); return state; } /*! @function IOSimpleLockUnlockEnableInterrupt * @abstract Unlock a spin lock, and restore interrupt state. * @discussion Unlock the lock, and restore preemption and interrupts to the state as they were when the lock was taken. Results are undefined if the caller has not locked the lock. * @param lock Pointer to the lock. * @param state The interrupt state returned by IOSimpleLockLockDisableInterrupt() */ static __inline__ void IOSimpleLockUnlockEnableInterrupt( IOSimpleLock * lock, IOInterruptState state ) { IOSimpleLockUnlock( lock ); ml_set_interrupts_enabled( state ); } #ifdef __cplusplus } /* extern "C" */ #endif #endif /* !__IOKIT_IOLOCKS_H */