Loading...
pthreads/pthread_mutex.c Libc-320 Libc-262.3.2
--- Libc/Libc-320/pthreads/pthread_mutex.c
+++ Libc/Libc-262.3.2/pthreads/pthread_mutex.c
@@ -1,27 +1,3 @@
-/*
- * Copyright (c) 2000-2003 Apple Computer, Inc. All rights reserved.
- *
- * @APPLE_LICENSE_HEADER_START@
- * 
- * Copyright (c) 1999-2003 Apple Computer, Inc.  All Rights Reserved.
- * 
- * 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. 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_LICENSE_HEADER_END@
- */
 /*
  * Copyright 1996 1995 by Open Software Foundation, Inc. 1997 1996 1995 1994 1993 1992 1991
  *              All Rights Reserved
@@ -60,65 +36,49 @@
 int
 pthread_mutex_destroy(pthread_mutex_t *mutex)
 {
-	int res;
-
-	LOCK(mutex->lock);
-	if (mutex->sig == _PTHREAD_MUTEX_SIG)
-	{
-		if (mutex->owner == (pthread_t)NULL &&
-		    mutex->busy == (pthread_cond_t *)NULL)
-		{
-			mutex->sig = _PTHREAD_NO_SIG;
-			res = ESUCCESS;
-		}
+        if (mutex->sig != _PTHREAD_MUTEX_SIG)
+                return (EINVAL);
+        if ((mutex->owner != (pthread_t)NULL) ||
+            (mutex->busy != (pthread_cond_t *)NULL))
+                return (EBUSY);
+        mutex->sig = _PTHREAD_NO_SIG;
+	return (ESUCCESS);
+}
+
+/*
+ * Initialize a mutex variable, possibly with additional attributes.
+ */
+int
+pthread_mutex_init(pthread_mutex_t *mutex, const pthread_mutexattr_t *attr)
+{
+        LOCK_INIT(mutex->lock);
+        mutex->sig = _PTHREAD_MUTEX_SIG;
+        if (attr)
+        {
+                if (attr->sig != _PTHREAD_MUTEX_ATTR_SIG)
+                        return (EINVAL);
+                mutex->prioceiling = attr->prioceiling;
+                mutex->protocol = attr->protocol;
+                mutex->type = attr->type;
+		if ((mutex->type == PTHREAD_MUTEX_DEFAULT) || (mutex->type == PTHREAD_MUTEX_NORMAL)) 
+			mutex->def = 1;
 		else
-			res = EBUSY;
-	}
-	else
-		res = EINVAL;
-	UNLOCK(mutex->lock);
-	return (res);
-}
-
-/*
- * Initialize a mutex variable, possibly with additional attributes.
- */
-static int
-_pthread_mutex_init(pthread_mutex_t *mutex, const pthread_mutexattr_t *attr)
-{
-	if (attr)
-	{
-		if (attr->sig != _PTHREAD_MUTEX_ATTR_SIG)
-			return (EINVAL);
-		mutex->prioceiling = attr->prioceiling;
-		mutex->protocol = attr->protocol;
-		mutex->type = attr->type;
-	} else {
-		mutex->prioceiling = _PTHREAD_DEFAULT_PRIOCEILING;
-		mutex->protocol = _PTHREAD_DEFAULT_PROTOCOL;
-		mutex->type = PTHREAD_MUTEX_DEFAULT;
-	}
-	mutex->lock_count = 0;
-	mutex->owner = (pthread_t)NULL;
-	mutex->next = (pthread_mutex_t *)NULL;
-	mutex->prev = (pthread_mutex_t *)NULL;
-	mutex->busy = (pthread_cond_t *)NULL;
+			mutex->def = 0;
+        } else {
+                mutex->prioceiling = _PTHREAD_DEFAULT_PRIOCEILING;
+                mutex->protocol = _PTHREAD_DEFAULT_PROTOCOL;
+                mutex->type = PTHREAD_MUTEX_DEFAULT;
+		mutex->def = 1;
+        }
+        mutex->lock_count = 0;
+        mutex->owner = (pthread_t)NULL;
+        mutex->next = (pthread_mutex_t *)NULL;
+        mutex->prev = (pthread_mutex_t *)NULL;
+        mutex->busy = (pthread_cond_t *)NULL;
 	mutex->waiters = 0;
-	mutex->sem = SEMAPHORE_NULL;
-	mutex->order = SEMAPHORE_NULL;
-	mutex->sig = _PTHREAD_MUTEX_SIG;
+	mutex->cond_lock = 0;
+	mutex->sem = MACH_PORT_NULL;
 	return (ESUCCESS);
-}
-
-/*
- * Initialize a mutex variable, possibly with additional attributes.
- * Public interface - so don't trust the lock - initialize it first.
- */
-int
-pthread_mutex_init(pthread_mutex_t *mutex, const pthread_mutexattr_t *attr)
-{
-	LOCK_INIT(mutex->lock);
-	return (_pthread_mutex_init(mutex, attr));
 }
 
 /*
@@ -129,19 +89,18 @@
 _pthread_mutex_add(pthread_mutex_t *mutex, pthread_t self)
 {
         pthread_mutex_t *m;
-	if (self != (pthread_t)0)
-	{
-		if ((m = self->mutexes) != (pthread_mutex_t *)NULL)
+        if (self != (pthread_t)0) {
+            if ((m = self->mutexes) != (pthread_mutex_t *)NULL)
                 { /* Add to list */
-			m->prev = mutex;
-                }
-		mutex->next = m;
-		mutex->prev = (pthread_mutex_t *)NULL;
-		self->mutexes = mutex;
+                	m->prev = mutex;
+                }
+            mutex->next = m;
+            mutex->prev = (pthread_mutex_t *)NULL;
+            self->mutexes = mutex;
 	}
 }
 
-__private_extern__ void
+static void
 _pthread_mutex_remove(pthread_mutex_t *mutex, pthread_t self)
 {
         pthread_mutex_t *n, *prev;
@@ -154,9 +113,9 @@
                 prev->next = mutex->next;
         } else
         { /* This is the first in the list */
-		if (self != (pthread_t)0) {
-			self->mutexes = n;
-		}
+            if (self != (pthread_t)0) {
+                self->mutexes = n;
+            }
         }
 }
 #endif
@@ -168,170 +127,141 @@
 int
 pthread_mutex_lock(pthread_mutex_t *mutex)
 {
-	kern_return_t kern_res;
-	pthread_t self;
-	int sig = mutex->sig; 
-
-	/* To provide backwards compat for apps using mutex incorrectly */
-	if ((sig != _PTHREAD_MUTEX_SIG) && (sig != _PTHREAD_MUTEX_SIG_init))
-		return(EINVAL);
-	LOCK(mutex->lock);
-	if (mutex->sig != _PTHREAD_MUTEX_SIG)
+        kern_return_t kern_res;
+        pthread_t self;
+	int slowpath;
+
+        if (mutex->sig == _PTHREAD_MUTEX_SIG_init)
+        {
+                int res;
+                if (res = pthread_mutex_init(mutex, NULL))
+                        return (res);
+        }
+        if (mutex->sig != _PTHREAD_MUTEX_SIG)
+                return (EINVAL);        /* Not a mutex variable */
+
+#if !defined(DEBUG)
+	if (mutex->def) {
+		slowpath = 0;
+		self = (pthread_t)0x12141968;
+	} else 
+#endif /* DEBUG */
 	{
-		if (mutex->sig != _PTHREAD_MUTEX_SIG_init)
-		{
-			UNLOCK(mutex->lock);
-			return (EINVAL);
+		slowpath = 1;
+		self = pthread_self();
+	}
+
+        LOCK(mutex->lock);
+                
+	if (mutex->waiters || (mutex->owner != (pthread_t)NULL))
+	{
+        	if(slowpath && (mutex->owner == self)) {
+            		if(mutex->type == PTHREAD_MUTEX_ERRORCHECK ) {
+               			UNLOCK(mutex->lock);
+               			return(EDEADLK);
+            		} else if (mutex->type == PTHREAD_MUTEX_RECURSIVE ) {
+				if (mutex->lock_count >= USHRT_MAX){
+               				UNLOCK(mutex->lock);
+					return(EAGAIN);
+				}
+                		mutex->lock_count++;
+               			UNLOCK(mutex->lock);
+               			return(ESUCCESS);
+            		}
+        	} 
+                mutex->waiters++;
+                if (mutex->sem == MACH_PORT_NULL) {
+			mutex->sem = new_sem_from_pool();
 		}
-		_pthread_mutex_init(mutex, NULL);
-		self = _PTHREAD_MUTEX_OWNER_SELF;
-	} 
-	else if (mutex->type != PTHREAD_MUTEX_NORMAL)
+                UNLOCK(mutex->lock);
+		do {
+			PTHREAD_MACH_CALL(semaphore_wait(mutex->sem), kern_res);
+		} while (kern_res == KERN_ABORTED);
+                LOCK(mutex->lock);
+		mutex->waiters--;
+		if (mutex->waiters == 0) {
+			restore_sem_to_pool(mutex->sem);
+			mutex->sem = MACH_PORT_NULL;
+		}
+                if (mutex->cond_lock) {
+                    mutex->cond_lock = 0;
+                }
+        }
+#if  defined(DEBUG) 
+        _pthread_mutex_add(mutex, self);
+#endif
+        mutex->owner = self;
+        if (slowpath && (mutex->type == PTHREAD_MUTEX_RECURSIVE))
+                mutex->lock_count = 1;
+        UNLOCK(mutex->lock);
+        return (ESUCCESS);
+}
+
+/*
+ * Attempt to lock a mutex, but don't block if this isn't possible.
+ */
+int
+pthread_mutex_trylock(pthread_mutex_t *mutex)
+{
+        kern_return_t kern_res;
+        pthread_t self;
+	int slowpath;
+	
+	if (mutex->sig == _PTHREAD_MUTEX_SIG_init)
+        {
+                int res;
+                if (res = pthread_mutex_init(mutex, NULL))
+                        return (res);
+        }
+        if (mutex->sig != _PTHREAD_MUTEX_SIG)
+                return (EINVAL);        /* Not a mutex variable */
+        
+#if !defined(DEBUG)
+	if (mutex->def) {
+		slowpath = 0;
+		self = (pthread_t)0x12141968;
+	} else 
+#endif /* DEBUG */
 	{
+		slowpath = 1;
 		self = pthread_self();
-		if (mutex->owner == self)
-		{
-			int res;
-
-			if (mutex->type == PTHREAD_MUTEX_RECURSIVE)
-			{
-				if (mutex->lock_count < USHRT_MAX)
-				{
-					mutex->lock_count++;
-					res = ESUCCESS;
-				} else
-					res = EAGAIN;
-			} else	/* PTHREAD_MUTEX_ERRORCHECK */
-				res = EDEADLK;
-			UNLOCK(mutex->lock);
-			return (res);
+	}
+
+        if (!TRY_LOCK(mutex->lock)) {
+            return (EBUSY);
+        }
+
+        if(slowpath && (mutex->owner == self) && (mutex->type == PTHREAD_MUTEX_RECURSIVE )) {
+		if (mutex->lock_count >= USHRT_MAX) {
+               		UNLOCK(mutex->lock);
+			return(EAGAIN);
 		}
-	} else 
-		self = _PTHREAD_MUTEX_OWNER_SELF;
-
-	if (mutex->owner != (pthread_t)NULL) {
-		if (mutex->waiters || mutex->owner != _PTHREAD_MUTEX_OWNER_SWITCHING)
-		{
-			semaphore_t sem, order;
-
-			if (++mutex->waiters == 1)
-			{
-				mutex->sem = sem = new_sem_from_pool();
-				mutex->order = order = new_sem_from_pool();
-			}
-			else
-			{
-				sem = mutex->sem;
-				order = mutex->order;
-				do {
-					PTHREAD_MACH_CALL(semaphore_wait(order), kern_res);
-				} while (kern_res == KERN_ABORTED);
-			} 
-			UNLOCK(mutex->lock);
-
-			PTHREAD_MACH_CALL(semaphore_wait_signal(sem, order), kern_res);
-			while (kern_res == KERN_ABORTED)
-			{
-				PTHREAD_MACH_CALL(semaphore_wait(sem), kern_res);
-			} 
-
-			LOCK(mutex->lock);
-			if (--mutex->waiters == 0)
-			{
-				PTHREAD_MACH_CALL(semaphore_wait(order), kern_res);
-				mutex->sem = mutex->order = SEMAPHORE_NULL;
-				restore_sem_to_pool(order);
-				restore_sem_to_pool(sem);
-			}
-		} 
-		else if (mutex->owner == _PTHREAD_MUTEX_OWNER_SWITCHING)
-		{
-			semaphore_t sem = mutex->sem;
-			do {
-				PTHREAD_MACH_CALL(semaphore_wait(sem), kern_res);
-			} while (kern_res == KERN_ABORTED);
-			mutex->sem = SEMAPHORE_NULL;
-			restore_sem_to_pool(sem);
+               mutex->lock_count++;
+               UNLOCK(mutex->lock);
+               return(ESUCCESS);
+        } 
+        
+        if (mutex->waiters ||
+		((mutex->owner != (pthread_t)NULL) && (mutex->cond_lock == 0)))
+        {
+                UNLOCK(mutex->lock);
+                return (EBUSY);
+        } else {
+#if defined(DEBUG)
+                _pthread_mutex_add(mutex, self);
+#endif
+                mutex->owner = (pthread_t)self;
+		if (mutex->cond_lock) {
+                    PTHREAD_MACH_CALL(semaphore_wait(mutex->sem), kern_res);
+                    mutex->cond_lock = 0;
+                    restore_sem_to_pool(mutex->sem);
+                    mutex->sem = MACH_PORT_NULL;
 		}
-	}
-
-	mutex->lock_count = 1;
-	mutex->owner = self;
-#if defined(DEBUG)
-	_pthread_mutex_add(mutex, self);
-#endif
-	UNLOCK(mutex->lock);
-	return (ESUCCESS);
-}
-
-/*
- * Attempt to lock a mutex, but don't block if this isn't possible.
- */
-int
-pthread_mutex_trylock(pthread_mutex_t *mutex)
-{
-	kern_return_t kern_res;
-	pthread_t self;
-
-	LOCK(mutex->lock);
-	if (mutex->sig != _PTHREAD_MUTEX_SIG)
-	{
-		if (mutex->sig != _PTHREAD_MUTEX_SIG_init)
-		{
-			UNLOCK(mutex->lock);
-			return (EINVAL);
-		}
-		_pthread_mutex_init(mutex, NULL);
-		self = _PTHREAD_MUTEX_OWNER_SELF;
-	}
-	else if (mutex->type != PTHREAD_MUTEX_NORMAL)
-	{
-		self = pthread_self();
-		if (mutex->type == PTHREAD_MUTEX_RECURSIVE)
-		{
-			if (mutex->owner == self)
-			{
-				int res;
-
-				if (mutex->lock_count < USHRT_MAX)
-				{
-					mutex->lock_count++;
-					res = ESUCCESS;
-				} else
-					res = EAGAIN;
-				UNLOCK(mutex->lock);
-				return (res);
-			}
-		}
-	} else
-		self = _PTHREAD_MUTEX_OWNER_SELF;
-
-	if (mutex->owner != (pthread_t)NULL)
-	{
-		if (mutex->waiters || mutex->owner != _PTHREAD_MUTEX_OWNER_SWITCHING)
-		{
-			UNLOCK(mutex->lock);
-			return (EBUSY);
-		}
-		else if (mutex->owner == _PTHREAD_MUTEX_OWNER_SWITCHING)
-		{
-			semaphore_t sem = mutex->sem;
-
-			do {
-				PTHREAD_MACH_CALL(semaphore_wait(sem), kern_res);
-			} while (kern_res == KERN_ABORTED);
-			restore_sem_to_pool(sem);
-			mutex->sem = SEMAPHORE_NULL;
-		}
-	}
-
-	mutex->lock_count = 1;
-	mutex->owner = self;
-#if defined(DEBUG)
-	_pthread_mutex_add(mutex, self);
-#endif
-	UNLOCK(mutex->lock);
-	return (ESUCCESS);
+                if (slowpath && (mutex->type == PTHREAD_MUTEX_RECURSIVE))
+                    mutex->lock_count = 1;
+                UNLOCK(mutex->lock);
+                return (ESUCCESS);
+        }
 }
 
 /*
@@ -341,63 +271,67 @@
 int
 pthread_mutex_unlock(pthread_mutex_t *mutex)
 {
-	kern_return_t kern_res;
-	int waiters;
-	int sig = mutex->sig; 
-
-	/* To provide backwards compat for apps using mutex incorrectly */
-	
-	if ((sig != _PTHREAD_MUTEX_SIG) && (sig != _PTHREAD_MUTEX_SIG_init))
-		return(EINVAL);
-	LOCK(mutex->lock);
-	if (mutex->sig != _PTHREAD_MUTEX_SIG)
+        kern_return_t kern_res;
+        int waiters;
+        pthread_t self;
+	int slowpath;
+        
+        if (mutex->sig == _PTHREAD_MUTEX_SIG_init)
+        {
+                int res;
+                if (res = pthread_mutex_init(mutex, NULL))
+                        return (res);
+        }
+        if (mutex->sig != _PTHREAD_MUTEX_SIG)
+                return (EINVAL);        /* Not a mutex variable */
+
+#if !defined(DEBUG)
+	if (mutex->def) {
+		slowpath = 0;
+		self = (pthread_t)0x12141968;
+	} else 
+#endif /* DEBUG */
 	{
-		if (mutex->sig != _PTHREAD_MUTEX_SIG_init)
-		{
-			UNLOCK(mutex->lock);
-			return (EINVAL);        /* Not a mutex variable */
-		}
-		_pthread_mutex_init(mutex, NULL);
-	} else
-
-#if !defined(DEBUG)
-	if (mutex->type != PTHREAD_MUTEX_NORMAL)
+		slowpath = 1;
+		self = pthread_self();
+	}
+
+        LOCK(mutex->lock);
+
+#if defined(DEBUG)
+	if (mutex->owner != self)
+#else
+        if (slowpath 
+		&& ((mutex->type == PTHREAD_MUTEX_ERRORCHECK ) 
+             		|| (mutex->type == PTHREAD_MUTEX_RECURSIVE )) 
+             	&& (mutex->owner != self)) 
+#endif /* DEBUG */
+	{
+               UNLOCK(mutex->lock);
+#if defined(DEBUG)
+		abort();
 #endif
-	{
-		pthread_t self = pthread_self();
-		if (mutex->owner != self)
-		{
+               return(EPERM);
+        }
+        
+        if (slowpath && (mutex->type == PTHREAD_MUTEX_RECURSIVE)  && --mutex->lock_count) {
+                UNLOCK(mutex->lock);
+		return (ESUCCESS);
+            
+        } else {
 #if defined(DEBUG)
-			abort();
-#endif
-			UNLOCK(mutex->lock);
-			return EPERM;
-		} else if (mutex->type == PTHREAD_MUTEX_RECURSIVE &&
-		    --mutex->lock_count)
-		{
-			UNLOCK(mutex->lock);
-			return ESUCCESS;
-		}
-	}
-
-	mutex->lock_count = 0;
-#if defined(DEBUG)
-	_pthread_mutex_remove(mutex, mutex->owner);
+                _pthread_mutex_remove(mutex, mutex->owner);
 #endif /* DEBUG */
-
-	waiters = mutex->waiters;
-	if (waiters)
-	{
-		mutex->owner = _PTHREAD_MUTEX_OWNER_SWITCHING;
+                waiters = mutex->waiters;
+                mutex->owner = (pthread_t)NULL;
 		UNLOCK(mutex->lock);
-		PTHREAD_MACH_CALL(semaphore_signal(mutex->sem), kern_res);
-	}
-	else
-	{
-		mutex->owner = (pthread_t)NULL;
-		UNLOCK(mutex->lock);
-	}
-	return (ESUCCESS);
+                if (waiters)
+                {
+		    PTHREAD_MACH_CALL(semaphore_signal(mutex->sem), kern_res);
+                }
+                return (ESUCCESS);
+        
+        }
 }
 
 /*
@@ -408,17 +342,14 @@
 pthread_mutex_getprioceiling(const pthread_mutex_t *mutex,
                              int *prioceiling)
 {
-	int res;
-
-	LOCK(mutex->lock);
         if (mutex->sig == _PTHREAD_MUTEX_SIG)
         {
                 *prioceiling = mutex->prioceiling;
-                res = ESUCCESS;
-        } else
-                res = EINVAL; /* Not an initialized 'attribute' structure */
-	UNLOCK(mutex->lock);
-	return (res);
+                return (ESUCCESS);
+        } else
+        {
+                return (EINVAL); /* Not an initialized 'attribute' structure */
+        }
 }
 
 /*
@@ -430,9 +361,6 @@
                              int prioceiling,
                              int *old_prioceiling)
 {
-	int res;
-
-	LOCK(mutex->lock);
         if (mutex->sig == _PTHREAD_MUTEX_SIG)
         {
                 if ((prioceiling >= -999) ||
@@ -440,13 +368,15 @@
                 {
                         *old_prioceiling = mutex->prioceiling;
                         mutex->prioceiling = prioceiling;
-                        res = ESUCCESS;
+                        return (ESUCCESS);
                 } else
-                        res = EINVAL; /* Invalid parameter */
-        } else
-                res = EINVAL; /* Not an initialized 'attribute' structure */
-	UNLOCK(mutex->lock);
-	return (res);
+                {
+                        return (EINVAL); /* Invalid parameter */
+                }
+        } else
+        {
+                return (EINVAL); /* Not an initialized 'attribute' structure */
+        }
 }
 
 /*