One Hat Cyber Team

View File Name : apc_lock.c

/*
  +----------------------------------------------------------------------+
  | APCu                                                                 |
  +----------------------------------------------------------------------+
  | Copyright (c) 2013 The PHP Group                                |
  +----------------------------------------------------------------------+
  | This source file is subject to version 3.01 of the PHP license,      |
  | that is bundled with this package in the file LICENSE, and is        |
  | available through the world-wide-web at the following url:           |
  | http://www.php.net/license/3_01.txt                                  |
  | If you did not receive a copy of the PHP license and are unable to   |
  | obtain it through the world-wide-web, please send a note to          |
  | license@php.net so we can mail you a copy immediately.               |
  +----------------------------------------------------------------------+
  | Author: Joe Watkins <joe.watkins@live.co.uk>                        |
  +----------------------------------------------------------------------+
 */
#ifndef HAVE_APC_LOCK
#define HAVE_APC_LOCK

#ifndef HAVE_APC_LOCK_H
# include "apc_lock.h"
#endif

/*
 * While locking calls should never fail, apcu checks for the success of write-lock
 * acquisitions, to prevent more damage when a deadlock is detected.
 */

/* {{{ There's very little point in initializing a billion sets of attributes */
#ifndef PHP_WIN32
# ifndef APC_SPIN_LOCK
#   ifndef APC_FCNTL_LOCK
#       ifdef APC_LOCK_RECURSIVE
			static pthread_mutexattr_t apc_lock_attr;
#       else
			static pthread_rwlockattr_t apc_lock_attr;
#       endif
#   else
#       include <unistd.h>
#       include <fcntl.h>

		static int apc_fcntl_call(int fd, int cmd, int type, off_t offset, int whence, off_t len) {
			int ret;
			struct flock lock;

			lock.l_type = type;
			lock.l_start = offset;
			lock.l_whence = whence;
			lock.l_len = len;
			lock.l_pid = 0;

			do {
				ret = fcntl(fd, cmd, &lock) ;
			} while(ret < 0 && errno == EINTR);

			return(ret);
		}
#   endif
# else
PHP_APCU_API int apc_lock_init(apc_lock_t* lock)
{
	lock->state = 0;
}

PHP_APCU_API int apc_lock_try(apc_lock_t* lock)
{
	int failed = 1;

	asm volatile
	(
		"xchgl %0, 0(%1)" :
		"=r" (failed) : "r" (&lock->state),
		"0" (failed)
	);

	return failed;
}

PHP_APCU_API int apc_lock_get(apc_lock_t* lock)
{
	int failed = 1;

	do {
		failed = apc_lock_try(
			lock);
#ifdef APC_LOCK_NICE
		usleep(0);
#endif
	} while (failed);

	return failed;
}

PHP_APCU_API int apc_lock_release(apc_lock_t* lock)
{
	int released = 0;

	asm volatile (
		"xchg %0, 0(%1)" : "=r" (released) : "r" (&lock->state),
		"0" (released)
	);

	return !released;
}
# endif
static zend_bool apc_lock_ready = 0;
#endif /* }}} */

/* {{{ Initialize the global lock attributes */
PHP_APCU_API zend_bool apc_lock_init() {
#ifndef PHP_WIN32
	if (apc_lock_ready)
		return 1;

	/* once per process please */
	apc_lock_ready = 1;

#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		if (pthread_mutexattr_init(&apc_lock_attr) == SUCCESS) {
			if (pthread_mutexattr_setpshared(&apc_lock_attr, PTHREAD_PROCESS_SHARED) == SUCCESS) {
				pthread_mutexattr_settype(&apc_lock_attr, PTHREAD_MUTEX_RECURSIVE);
				return 1;
			}
		}
#   else
		if (pthread_rwlockattr_init(&apc_lock_attr) == SUCCESS) {
			if (pthread_rwlockattr_setpshared(&apc_lock_attr, PTHREAD_PROCESS_SHARED) == SUCCESS) {
				return 1;
			}
		}
#   endif
# endif
#endif
	return 0;
#else
	return 1;
#endif
} /* }}} */

/* {{{ Cleanup attributes and statics */
PHP_APCU_API void apc_lock_cleanup() {
#ifndef PHP_WIN32
	if (!apc_lock_ready)
		return;

	/* once per process please */
	apc_lock_ready = 0;

#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		pthread_mutexattr_destroy(&apc_lock_attr);
#   else
		pthread_rwlockattr_destroy(&apc_lock_attr);
#   endif
# endif
#endif
#endif
} /* }}} */

PHP_APCU_API zend_bool apc_lock_create(apc_lock_t *lock) {
#ifndef PHP_WIN32
# ifndef APC_SPIN_LOCK
#   ifndef APC_FCNTL_LOCK
#       ifdef APC_LOCK_RECURSIVE
			{
				pthread_mutex_init(lock, &apc_lock_attr);
				return 1;
			}
#       else
			{
				/* Native */
				return (pthread_rwlock_init(lock, &apc_lock_attr)==SUCCESS);
			}
#       endif
# else
	{
		/* FCNTL */
		char lock_path[] = "/tmp/.apc.XXXXXX";
		mktemp(lock_path);
		(*lock) = open(lock_path, O_RDWR|O_CREAT, 0666);
		if((*lock) > 0 ) {
			unlink(lock_path);
			return 1;
		} else {
			return 0;
		}
	}
# endif
#else
	{
		/* SPIN */
		lock->state = 0;
		return 1;
	}

#endif
#else
	lock = (apc_lock_t *)apc_windows_cs_create((apc_windows_cs_rwlock_t *)lock);

	return (NULL != lock);
#endif
}

PHP_APCU_API zend_bool apc_lock_rlock(apc_lock_t *lock) {
#ifndef PHP_WIN32
#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		pthread_mutex_lock(lock);
#   else
		pthread_rwlock_rdlock(lock);
#   endif
# else
	{
		/* FCNTL */
		apc_fcntl_call((*lock), F_SETLKW, F_RDLCK, 0, SEEK_SET, 0);
	}
# endif
#else
	{
		/* SPIN */
		apc_lock_get(lock);
	}
#endif
#else
	apc_windows_cs_rdlock((apc_windows_cs_rwlock_t *)lock);
#endif
	return 1;
}

static inline zend_bool apc_lock_wlock_impl(apc_lock_t *lock) {
#ifndef PHP_WIN32
#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		return pthread_mutex_lock(lock) == 0;
#   else
		return pthread_rwlock_wrlock(lock) == 0;
#   endif
# else
	{
		/* FCNTL */
		apc_fcntl_call((*lock), F_SETLKW, F_WRLCK, 0, SEEK_SET, 0);
	}
# endif
#else
	{
		/* SPIN */
		apc_lock_get(lock);
	}
#endif
#else
	apc_windows_cs_lock((apc_windows_cs_rwlock_t *)lock);
#endif
	return 1;
}

PHP_APCU_API zend_bool apc_lock_wlock(apc_lock_t *lock) {
	HANDLE_BLOCK_INTERRUPTIONS();
	if (apc_lock_wlock_impl(lock)) {
		return 1;
	}

	HANDLE_UNBLOCK_INTERRUPTIONS();
	apc_warning("Failed to acquire write lock");
	return 0;
}

PHP_APCU_API zend_bool apc_lock_wunlock(apc_lock_t *lock) {
#ifndef PHP_WIN32
#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		pthread_mutex_unlock(lock);
#   else
		pthread_rwlock_unlock(lock);
#   endif
# else
	{
		/* FCNTL */
		apc_fcntl_call((*lock), F_SETLKW, F_UNLCK, 0, SEEK_SET, 0);
	}
# endif
#else
	{
		/* SPIN */
		apc_lock_release(lock);
	}
#endif
#else
	apc_windows_cs_unlock_wr((apc_windows_cs_rwlock_t *)lock);
#endif
	return 1;
}

PHP_APCU_API zend_bool apc_lock_runlock(apc_lock_t *lock) {
#ifndef PHP_WIN32
#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		pthread_mutex_unlock(lock);
#   else
		pthread_rwlock_unlock(lock);
#   endif
# else
	{
		/* FCNTL */
		apc_fcntl_call((*lock), F_SETLKW, F_UNLCK, 0, SEEK_SET, 0);
	}
# endif
#else
	{
		/* SPIN */
		apc_lock_release(lock);
	}
#endif
#else
	apc_windows_cs_unlock_rd((apc_windows_cs_rwlock_t *)lock);
#endif
	return 1;
}

PHP_APCU_API void apc_lock_destroy(apc_lock_t *lock) {
#ifndef PHP_WIN32
#ifndef APC_SPIN_LOCK
# ifndef APC_FCNTL_LOCK
#   ifdef APC_LOCK_RECURSIVE
		/* nothing */
#   else
		/* nothing */
#   endif
# else
	{
		/* FCNTL */
		close((*lock));
	}
# endif
#endif
#else
	apc_windows_cs_destroy((apc_windows_cs_rwlock_t *)lock);
#endif
}
#endif