PDSos161/kern/thread/synch.c

/*
 * Copyright (c) 2000, 2001, 2002, 2003, 2004, 2005, 2008, 2009
 *	The President and Fellows of Harvard College.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE UNIVERSITY OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

/*
 * Synchronization primitives.
 * The specifications of the functions are in synch.h.
 */

#include <types.h>
#include <lib.h>
#include <spinlock.h>
#include <wchan.h>
#include <thread.h>
#include <current.h>
#include <synch.h>

////////////////////////////////////////////////////////////
//
// Semaphore.

struct semaphore *
sem_create(const char *name, unsigned initial_count)
{
        struct semaphore *sem;

        sem = kmalloc(sizeof(*sem));
        if (sem == NULL) {
                return NULL;
        }

        sem->sem_name = kstrdup(name);
        if (sem->sem_name == NULL) {
                kfree(sem);
                return NULL;
        }

	sem->sem_wchan = wchan_create(sem->sem_name);
	if (sem->sem_wchan == NULL) {
		kfree(sem->sem_name);
		kfree(sem);
		return NULL;
	}

	spinlock_init(&sem->sem_lock);
        sem->sem_count = initial_count;

        return sem;
}

void
sem_destroy(struct semaphore *sem)
{
        KASSERT(sem != NULL);

	/* wchan_cleanup will assert if anyone's waiting on it */
	spinlock_cleanup(&sem->sem_lock);
	wchan_destroy(sem->sem_wchan);
        kfree(sem->sem_name);
        kfree(sem);
}

void
P(struct semaphore *sem)
{
        KASSERT(sem != NULL);

        /*
         * May not block in an interrupt handler.
         *
         * For robustness, always check, even if we can actually
         * complete the P without blocking.
         */
        KASSERT(curthread->t_in_interrupt == false);

	/* Use the semaphore spinlock to protect the wchan as well. */
	spinlock_acquire(&sem->sem_lock);
        while (sem->sem_count == 0) {
		/*
		 *
		 * Note that we don't maintain strict FIFO ordering of
		 * threads going through the semaphore; that is, we
		 * might "get" it on the first try even if other
		 * threads are waiting. Apparently according to some
		 * textbooks semaphores must for some reason have
		 * strict ordering. Too bad. :-)
		 *
		 * Exercise: how would you implement strict FIFO
		 * ordering?
		 */
		wchan_sleep(sem->sem_wchan, &sem->sem_lock);
        }
        KASSERT(sem->sem_count > 0);
        sem->sem_count--;
	spinlock_release(&sem->sem_lock);
}

void
V(struct semaphore *sem)
{
        KASSERT(sem != NULL);

	spinlock_acquire(&sem->sem_lock);

        sem->sem_count++;
        KASSERT(sem->sem_count > 0);
	wchan_wakeone(sem->sem_wchan, &sem->sem_lock);

	spinlock_release(&sem->sem_lock);
}

////////////////////////////////////////////////////////////
//
// Lock.

struct lock *
lock_create(const char *name)
{
        struct lock *lock;

        lock = kmalloc(sizeof(*lock));
        if (lock == NULL) {
                return NULL;
        }

        lock->lk_name = kstrdup(name);
        if (lock->lk_name == NULL) {
                kfree(lock);
                return NULL;
        }

        // add stuff here as needed

#if OPT_SEMLOCK
        lock->sem = sem_create(name, 1);
        if (lock->sem == NULL) {
		kfree(lock->lk_name);
		kfree(lock);
                return NULL;
        }
#elif OPT_WCHANLOCK
	lock->wchan = wchan_create(lock->lk_name);
	if (lock->wchan == NULL) {
		kfree(lock->lk_name);
		kfree(lock);
		return NULL;
	}
        spinlock_init(&lock->slk);
#endif

        lock->owner = NULL;
        return lock;
}

void
lock_destroy(struct lock *lock)
{
        KASSERT(lock != NULL);

        // add stuff here as needed

        kfree(lock->lk_name);
#if OPT_SEMLOCK
        sem_destroy(lock->sem);
#elif OPT_WCHANLOCK
        wchan_destroy(lock->wchan);
        spinlock_cleanup(&lock->slk);
#endif
        kfree(lock);
}

void
lock_acquire(struct lock *lock)
{
        // Write this

#if OPT_SEMLOCK
        P(lock->sem);
        lock->owner = curthread;
#elif OPT_WCHANLOCK
        spinlock_acquire(&lock->slk);
        while (lock->owner != NULL) {
		wchan_sleep(lock->wchan, &lock->slk);
        }
        lock->owner = curthread;
        spinlock_release(&lock->slk);
#else
        lock->owner = curthread;
#endif
}

void
lock_release(struct lock *lock)
{
        // Write this

        KASSERT(lock != NULL);
        KASSERT(lock_do_i_hold(lock));
#if OPT_SEMLOCK
        lock->owner = NULL;
        V(lock->sem);
#elif OPT_WCHANLOCK
        spinlock_acquire(&lock->slk);
        lock->owner = NULL;
	wchan_wakeone(lock->wchan, &lock->slk);
        spinlock_release(&lock->slk);
#else
        lock->owner = NULL;
#endif
}

bool
lock_do_i_hold(struct lock *lock)
{
        // Write this
        KASSERT(lock != NULL);
#if OPT_WCHANLOCK
        bool ret = true;
        spinlock_acquire(&lock->slk);
        ret = lock->owner == curthread;
        spinlock_release(&lock->slk);
        return ret; // dummy until code gets written
#else
        return lock->owner == curthread;
#endif

}

////////////////////////////////////////////////////////////
//
// CV


struct cv *
cv_create(const char *name)
{
        struct cv *cv;

        cv = kmalloc(sizeof(*cv));
        if (cv == NULL) {
                return NULL;
        }

        cv->cv_name = kstrdup(name);
        if (cv->cv_name==NULL) {
                kfree(cv);
                return NULL;
        }

#if OPT_WCHANLOCK
        cv->wchan = wchan_create(name);
        if (cv->wchan == NULL) {
                kfree(cv->cv_name);
                kfree(cv);
                return NULL;
        }
        spinlock_init(&cv->slk);
#endif
        return cv;
}

void
cv_destroy(struct cv *cv)
{
        KASSERT(cv != NULL);

#if OPT_WCHANLOCK
        spinlock_cleanup(&cv->slk);
        wchan_destroy(cv->wchan);
#endif
        kfree(cv->cv_name);
        kfree(cv);
}

void
cv_wait(struct cv *cv, struct lock *lock)
{
#if OPT_WCHANLOCK
        KASSERT(cv != NULL);
        KASSERT(lock_do_i_hold(lock));

        spinlock_acquire(&cv->slk);

        lock_release(lock);
        wchan_sleep(cv->wchan,&cv->slk);
        spinlock_release(&cv->slk);
        lock_acquire(lock);
#else
        (void)cv;    // suppress warning until code gets written
        (void)lock;  // suppress warning until code gets written
#endif
}

void
cv_signal(struct cv *cv, struct lock *lock)
{
#if OPT_WCHANLOCK
        KASSERT(cv != NULL);
        KASSERT(lock_do_i_hold(lock));

        spinlock_acquire(&cv->slk);
        wchan_wakeone(cv->wchan, &cv->slk);
        spinlock_release(&cv->slk);
#else
	(void)cv;    // suppress warning until code gets written
	(void)lock;  // suppress warning until code gets written
#endif
}

void
cv_broadcast(struct cv *cv, struct lock *lock)
{
#if OPT_WCHANLOCK
        KASSERT(cv != NULL);
	KASSERT(lock_do_i_hold(lock));

        spinlock_acquire(&cv->slk);
        wchan_wakeall(cv->wchan, &cv->slk);
        spinlock_release(&cv->slk);
#else
	(void)cv;    // suppress warning until code gets written
	(void)lock;  // suppress warning until code gets written
#endif
}