Linux/Unix编程

开发平台：
Unix_Linux

rwsem.h：源码内容
							/* rwsem.h: R/W semaphores implemented using XADD/CMPXCHG for x86_64+
 *
 * Written by David Howells (dhowells@redhat.com).
 * Ported by Andi Kleen <ak@suse.de> to x86-64.
 *
 * Derived from asm-i386/semaphore.h and asm-i386/rwsem.h
 *
 * Trylock by Brian Watson (Brian.J.Watson@compaq.com).
 *
 * The MSW of the count is the negated number of active writers and waiting
 * lockers, and the LSW is the total number of active locks
 *
 * The lock count is initialized to 0 (no active and no waiting lockers).
 *
 * When a writer subtracts WRITE_BIAS, it'll get 0xffff0001 for the case of an
 * uncontended lock. This can be determined because XADD returns the old value.
 * Readers increment by 1 and see a positive value when uncontended, negative
 * if there are writers (and maybe) readers waiting (in which case it goes to
 * sleep).
 *
 * The value of WAITING_BIAS supports up to 32766 waiting processes. This can
 * be extended to 65534 by manually checking the whole MSW rather than relying
 * on the S flag.
 *
 * The value of ACTIVE_BIAS supports up to 65535 active processes.
 *
 * This should be totally fair - if anything is waiting, a process that wants a
 * lock will go to the back of the queue. When the currently active lock is
 * released, if there's a writer at the front of the queue, then that and only
 * that will be woken up; if there's a bunch of consequtive readers at the
 * front, then they'll all be woken up, but no other readers will be.
 */
#ifndef _X8664_RWSEM_H
#define _X8664_RWSEM_H
#ifndef _LINUX_RWSEM_H
#error please dont include asm/rwsem.h directly, use linux/rwsem.h instead
#endif
#ifdef __KERNEL__
#include <linux/list.h>
#include <linux/spinlock.h>
struct rwsem_waiter;
extern struct rw_semaphore *rwsem_down_read_failed(struct rw_semaphore *sem);
extern struct rw_semaphore *rwsem_down_write_failed(struct rw_semaphore *sem);
extern struct rw_semaphore *rwsem_wake(struct rw_semaphore *);
/*
 * the semaphore definition
 */
struct rw_semaphore {
	signed int		count;
#define RWSEM_UNLOCKED_VALUE		0x00000000
#define RWSEM_ACTIVE_BIAS		0x00000001
#define RWSEM_ACTIVE_MASK		0x0000ffff
#define RWSEM_WAITING_BIAS		(-0x00010000)
#define RWSEM_ACTIVE_READ_BIAS		RWSEM_ACTIVE_BIAS
#define RWSEM_ACTIVE_WRITE_BIAS		(RWSEM_WAITING_BIAS + RWSEM_ACTIVE_BIAS)
	spinlock_t		wait_lock;
	struct list_head	wait_list;
#if RWSEM_DEBUG
	int			debug;
#endif
};
/*
 * initialisation
 */
#if RWSEM_DEBUG
#define __RWSEM_DEBUG_INIT      , 0
#else
#define __RWSEM_DEBUG_INIT	/* */
#endif
#define __RWSEM_INITIALIZER(name) 
{ RWSEM_UNLOCKED_VALUE, SPIN_LOCK_UNLOCKED, LIST_HEAD_INIT((name).wait_list) 
	__RWSEM_DEBUG_INIT }
#define DECLARE_RWSEM(name) 
	struct rw_semaphore name = __RWSEM_INITIALIZER(name)
static inline void init_rwsem(struct rw_semaphore *sem)
{
	sem->count = RWSEM_UNLOCKED_VALUE;
	spin_lock_init(&sem->wait_lock);
	INIT_LIST_HEAD(&sem->wait_list);
#if RWSEM_DEBUG
	sem->debug = 0;
#endif
}
/*
 * lock for reading
 */
static inline void __down_read(struct rw_semaphore *sem)
{
	__asm__ __volatile__(
		"# beginning down_readnt"
LOCK_PREFIX	"  incl      (%%rdi)nt" /* adds 0x00000001, returns the old value */
		"  js        2fnt" /* jump if we weren't granted the lock */
		"1:nt"
		LOCK_SECTION_START("") 
		"2:nt"
		"  call      rwsem_down_read_failed_thunknt"
		"  jmp       1bn"
		LOCK_SECTION_END 
		"# ending down_readnt"
		: "+m"(sem->count)
		: "D"(sem)
		: "memory", "cc");
}
/*
 * trylock for reading -- returns 1 if successful, 0 if contention
 */
static inline int __down_read_trylock(struct rw_semaphore *sem)
{
	__s32 result, tmp;
	__asm__ __volatile__(
		"# beginning __down_read_trylocknt"
		"  movl      %0,%1nt"
		"1:nt"
		"  movl	     %1,%2nt"
		"  addl      %3,%2nt"
		"  jle	     2fnt"
LOCK_PREFIX	"  cmpxchgl  %2,%0nt"
		"  jnz	     1bnt"
		"2:nt"
		"# ending __down_read_trylocknt"
		: "+m"(sem->count), "=&a"(result), "=&r"(tmp)
		: "i"(RWSEM_ACTIVE_READ_BIAS)
		: "memory", "cc");
	return result>=0 ? 1 : 0;
}
/*
 * lock for writing
 */
static inline void __down_write(struct rw_semaphore *sem)
{
	int tmp;
	tmp = RWSEM_ACTIVE_WRITE_BIAS;
	__asm__ __volatile__(
		"# beginning down_writent"
LOCK_PREFIX	"  xaddl      %0,(%%rdi)nt" /* subtract 0x0000ffff, returns the old value */
		"  testl     %0,%0nt" /* was the count 0 before? */
		"  jnz       2fnt" /* jump if we weren't granted the lock */
		"1:nt"
		LOCK_SECTION_START("")
		"2:nt"
		"  call      rwsem_down_write_failed_thunknt"
		"  jmp       1bn"
		LOCK_SECTION_END
		"# ending down_write"
		: "=&r" (tmp) 
		: "0"(tmp), "D"(sem)
		: "memory", "cc");
}
/*
 * trylock for writing -- returns 1 if successful, 0 if contention
 */
static inline int __down_write_trylock(struct rw_semaphore *sem)
{
	signed long ret = cmpxchg(&sem->count,
				  RWSEM_UNLOCKED_VALUE, 
				  RWSEM_ACTIVE_WRITE_BIAS);
	if (ret == RWSEM_UNLOCKED_VALUE)
		return 1;
	return 0;
}
/*
 * unlock after reading
 */
static inline void __up_read(struct rw_semaphore *sem)
{
	__s32 tmp = -RWSEM_ACTIVE_READ_BIAS;
	__asm__ __volatile__(
		"# beginning __up_readnt"
LOCK_PREFIX	"  xaddl      %%edx,(%%rdi)nt" /* subtracts 1, returns the old value */
		"  js        2fnt" /* jump if the lock is being waited upon */
		"1:nt"
		LOCK_SECTION_START("")
		"2:nt"
		"  decw      %%dxnt" /* do nothing if still outstanding active readers */
		"  jnz       1bnt"
		"  call      rwsem_wake_thunknt"
		"  jmp       1bn"
		LOCK_SECTION_END
		"# ending __up_readn"
		: "+m"(sem->count), "+d"(tmp)
		: "D"(sem)
		: "memory", "cc");
}
/*
 * unlock after writing
 */
static inline void __up_write(struct rw_semaphore *sem)
{
	__asm__ __volatile__(
		"# beginning __up_writent"
		"  movl      %2,%%edxnt"
LOCK_PREFIX	"  xaddl     %%edx,(%%rdi)nt" /* tries to transition 0xffff0001 -> 0x00000000 */
		"  jnz       2fnt" /* jump if the lock is being waited upon */
		"1:nt"
		LOCK_SECTION_START("")
		"2:nt"
		"  decw      %%dxnt" /* did the active count reduce to 0? */
		"  jnz       1bnt" /* jump back if not */
		"  call      rwsem_wake_thunknt"
		"  jmp       1bn"
		LOCK_SECTION_END
		"# ending __up_writen"
		: "+m"(sem->count)
		: "D"(sem), "i"(-RWSEM_ACTIVE_WRITE_BIAS)
		: "memory", "cc", "rdx");
}
/*
 * implement atomic add functionality
 */
static inline void rwsem_atomic_add(int delta, struct rw_semaphore *sem)
{
	__asm__ __volatile__(
LOCK_PREFIX	"addl %1,%0"
		:"=m"(sem->count)
		:"ir"(delta), "m"(sem->count));
}
/*
 * implement exchange and add functionality
 */
static inline int rwsem_atomic_update(int delta, struct rw_semaphore *sem)
{
	int tmp = delta;
	__asm__ __volatile__(
LOCK_PREFIX	"xaddl %0,(%2)"
		: "=r"(tmp), "=m"(sem->count)
		: "r"(sem), "m"(sem->count), "0" (tmp)
		: "memory");
	return tmp+delta;
}
#endif /* __KERNEL__ */
#endif /* _X8664_RWSEM_H */

						