waitq.c

Go to the documentation of this file.

/*
 * Copyright (C) 2001-2004 Jakub Jermar
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - 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.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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.
 */

#include <synch/waitq.h>
#include <synch/synch.h>
#include <synch/spinlock.h>
#include <proc/thread.h>
#include <proc/scheduler.h>
#include <arch/asm.h>
#include <arch/types.h>
#include <typedefs.h>
#include <time/timeout.h>
#include <arch.h>
#include <context.h>
#include <adt/list.h>

static void waitq_timeouted_sleep(void *data);

void waitq_initialize(waitq_t *wq)
{
        spinlock_initialize(&wq->lock, "waitq_lock");
        list_initialize(&wq->head);
        wq->missed_wakeups = 0;
}

void waitq_timeouted_sleep(void *data)
{
        thread_t *t = (thread_t *) data;
        waitq_t *wq;
        bool do_wakeup = false;

        spinlock_lock(&threads_lock);
        if (!thread_exists(t))
                goto out;

grab_locks:
        spinlock_lock(&t->lock);
        if ((wq = t->sleep_queue)) {            /* assignment */
                if (!spinlock_trylock(&wq->lock)) {
                        spinlock_unlock(&t->lock);
                        goto grab_locks;        /* avoid deadlock */
                }

                list_remove(&t->wq_link);
                t->saved_context = t->sleep_timeout_context;
                do_wakeup = true;
                t->sleep_queue = NULL;
                spinlock_unlock(&wq->lock);
        }
        
        t->timeout_pending = false;
        spinlock_unlock(&t->lock);
        
        if (do_wakeup)
                thread_ready(t);

out:
        spinlock_unlock(&threads_lock);
}

void waitq_interrupt_sleep(thread_t *t)
{
        waitq_t *wq;
        bool do_wakeup = false;
        ipl_t ipl;

        ipl = interrupts_disable();
        spinlock_lock(&threads_lock);
        if (!thread_exists(t))
                goto out;

grab_locks:
        spinlock_lock(&t->lock);
        if ((wq = t->sleep_queue)) {            /* assignment */
                if (!(t->sleep_interruptible)) {
                        /*
                         * The sleep cannot be interrupted.
                         */
                        spinlock_unlock(&t->lock);
                        goto out;
                }
                        
                if (!spinlock_trylock(&wq->lock)) {
                        spinlock_unlock(&t->lock);
                        goto grab_locks;        /* avoid deadlock */
                }

                if (t->timeout_pending && timeout_unregister(&t->sleep_timeout))
                        t->timeout_pending = false;

                list_remove(&t->wq_link);
                t->saved_context = t->sleep_interruption_context;
                do_wakeup = true;
                t->sleep_queue = NULL;
                spinlock_unlock(&wq->lock);
        }
        spinlock_unlock(&t->lock);

        if (do_wakeup)
                thread_ready(t);

out:
        spinlock_unlock(&threads_lock);
        interrupts_restore(ipl);
}

int waitq_sleep_timeout(waitq_t *wq, __u32 usec, int flags)
{
        ipl_t ipl;
        int rc;
        
        ipl = waitq_sleep_prepare(wq);
        rc = waitq_sleep_timeout_unsafe(wq, usec, flags);
        waitq_sleep_finish(wq, rc, ipl);
        return rc;
}

ipl_t waitq_sleep_prepare(waitq_t *wq)
{
        ipl_t ipl;
        
restart:
        ipl = interrupts_disable();

        if (THREAD) {   /* needed during system initiailzation */
                /*
                 * Busy waiting for a delayed timeout.
                 * This is an important fix for the race condition between
                 * a delayed timeout and a next call to waitq_sleep_timeout().
                 * Simply, the thread is not allowed to go to sleep if
                 * there are timeouts in progress.
                 */
                spinlock_lock(&THREAD->lock);
                if (THREAD->timeout_pending) {
                        spinlock_unlock(&THREAD->lock);
                        interrupts_restore(ipl);
                        goto restart;
                }
                spinlock_unlock(&THREAD->lock);
        }
                                                                                                        
        spinlock_lock(&wq->lock);
        return ipl;
}

void waitq_sleep_finish(waitq_t *wq, int rc, ipl_t ipl)
{
        switch (rc) {
        case ESYNCH_WOULD_BLOCK:
        case ESYNCH_OK_ATOMIC:
                spinlock_unlock(&wq->lock);
                break;
        default:
                break;
        }
        interrupts_restore(ipl);
}

int waitq_sleep_timeout_unsafe(waitq_t *wq, __u32 usec, int flags)
{
        /* checks whether to go to sleep at all */
        if (wq->missed_wakeups) {
                wq->missed_wakeups--;
                return ESYNCH_OK_ATOMIC;
        }
        else {
                if ((flags & SYNCH_FLAGS_NON_BLOCKING) && (usec == 0)) {
                        /* return immediatelly instead of going to sleep */
                        return ESYNCH_WOULD_BLOCK;
                }
        }
        
        /*
         * Now we are firmly decided to go to sleep.
         */
        spinlock_lock(&THREAD->lock);

        if (flags & SYNCH_FLAGS_INTERRUPTIBLE) {

                /*
                 * If the thread was already interrupted,
                 * don't go to sleep at all.
                 */
                if (THREAD->interrupted) {
                        spinlock_unlock(&THREAD->lock);
                        spinlock_unlock(&wq->lock);
                        return ESYNCH_INTERRUPTED;
                }

                /*
                 * Set context that will be restored if the sleep
                 * of this thread is ever interrupted.
                 */
                THREAD->sleep_interruptible = true;
                if (!context_save(&THREAD->sleep_interruption_context)) {
                        /* Short emulation of scheduler() return code. */
                        spinlock_unlock(&THREAD->lock);
                        return ESYNCH_INTERRUPTED;
                }

        } else {
                THREAD->sleep_interruptible = false;
        }

        if (usec) {
                /* We use the timeout variant. */
                if (!context_save(&THREAD->sleep_timeout_context)) {
                        /* Short emulation of scheduler() return code. */
                        spinlock_unlock(&THREAD->lock);
                        return ESYNCH_TIMEOUT;
                }
                THREAD->timeout_pending = true;
                timeout_register(&THREAD->sleep_timeout, (__u64) usec, waitq_timeouted_sleep, THREAD);
        }

        list_append(&THREAD->wq_link, &wq->head);

        /*
         * Suspend execution.
         */
        THREAD->state = Sleeping;
        THREAD->sleep_queue = wq;

        spinlock_unlock(&THREAD->lock);

        scheduler();    /* wq->lock is released in scheduler_separated_stack() */
        
        return ESYNCH_OK_BLOCKED;
}


void waitq_wakeup(waitq_t *wq, bool all)
{
        ipl_t ipl;

        ipl = interrupts_disable();
        spinlock_lock(&wq->lock);

        _waitq_wakeup_unsafe(wq, all);

        spinlock_unlock(&wq->lock);     
        interrupts_restore(ipl);        
}

void _waitq_wakeup_unsafe(waitq_t *wq, bool all)
{
        thread_t *t;

loop:   
        if (list_empty(&wq->head)) {
                wq->missed_wakeups++;
                if (all)
                        wq->missed_wakeups = 0;
                return;
        }

        t = list_get_instance(wq->head.next, thread_t, wq_link);
        
        /*
         * Lock the thread prior to removing it from the wq.
         * This is not necessary because of mutual exclusion
         * (the link belongs to the wait queue), but because
         * of synchronization with waitq_timeouted_sleep()
         * and waitq_interrupt_sleep().
         *
         * In order for these two functions to work, the following
         * invariant must hold:
         *
         * t->sleep_queue != NULL <=> t sleeps in a wait queue
         *
         * For an observer who locks the thread, the invariant
         * holds only when the lock is held prior to removing
         * it from the wait queue.
         */
        spinlock_lock(&t->lock);
        list_remove(&t->wq_link);
        
        if (t->timeout_pending && timeout_unregister(&t->sleep_timeout))
                t->timeout_pending = false;
        t->sleep_queue = NULL;
        spinlock_unlock(&t->lock);

        thread_ready(t);

        if (all)
                goto loop;
}

---