WebSVN – HelenOS – Diff – /branches/tracing/kernel/generic/src/proc/thread.c


/*
 * 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.
 */
 
/** @addtogroup genericproc
 * @{
 */
 
/**
 * @file
 * @brief   Thread management functions.
 */
 
#include <proc/scheduler.h>
#include <proc/thread.h>
#include <proc/task.h>
#include <proc/uarg.h>
#include <mm/frame.h>
#include <mm/page.h>
#include <arch/asm.h>
#include <arch/cycle.h>
#include <arch.h>
#include <synch/synch.h>
#include <synch/spinlock.h>
#include <synch/waitq.h>
#include <synch/rwlock.h>
#include <cpu.h>
#include <func.h>
#include <context.h>
#include <adt/avl.h>
#include <adt/list.h>
#include <time/clock.h>
#include <time/timeout.h>
#include <config.h>
#include <arch/interrupt.h>
#include <smp/ipi.h>
#include <arch/faddr.h>
#include <atomic.h>
#include <memstr.h>
#include <print.h>
#include <mm/slab.h>
#include <debug.h>
#include <main/uinit.h>
#include <syscall/copy.h>
#include <errno.h>
 
 
#ifndef LOADED_PROG_STACK_PAGES_NO
#define LOADED_PROG_STACK_PAGES_NO 1
#endif
 
 
/** Thread states */
char *thread_states[] = {
    "Invalid",
    "Running",
    "Sleeping",
    "Ready",
    "Entering",
    "Exiting",
    "Lingering"
};
 
/** Lock protecting the threads_tree AVL tree.
 *
 * For locking rules, see declaration thereof.
 */
SPINLOCK_INITIALIZE(threads_lock);
 
/** AVL tree of all threads.
 *
 * When a thread is found in the threads_tree AVL tree, it is guaranteed to
 * exist as long as the threads_lock is held.
 */
avltree_t threads_tree;    
 
SPINLOCK_INITIALIZE(tidlock);
thread_id_t last_tid = 0;
 
static slab_cache_t *thread_slab;
#ifdef ARCH_HAS_FPU
slab_cache_t *fpu_context_slab;
#endif
 
/** Thread wrapper.
 *
 * This wrapper is provided to ensure that every thread makes a call to
 * thread_exit() when its implementing function returns.
 *
 * interrupts_disable() is assumed.
 *
 */
static void cushion(void)
{
    void (*f)(void *) = THREAD->thread_code;
    void *arg = THREAD->thread_arg;
    THREAD->last_cycle = get_cycle();
 
    /* This is where each thread wakes up after its creation */
    spinlock_unlock(&THREAD->lock);
    interrupts_enable();
 
    f(arg);
   
    /* Accumulate accounting to the task */
    ipl_t ipl = interrupts_disable();
   
    spinlock_lock(&THREAD->lock);
    if (!THREAD->uncounted) {
        thread_update_accounting();
        uint64_t cycles = THREAD->cycles;
        THREAD->cycles = 0;
        spinlock_unlock(&THREAD->lock);
       
        spinlock_lock(&TASK->lock);
        TASK->cycles += cycles;
        spinlock_unlock(&TASK->lock);
    } else
        spinlock_unlock(&THREAD->lock);
   
    interrupts_restore(ipl);
   
    thread_exit();
    /* not reached */
}
 
/** Initialization and allocation for thread_t structure */
static int thr_constructor(void *obj, int kmflags)
{
    thread_t *t = (thread_t *) obj;
 
    spinlock_initialize(&t->lock, "thread_t_lock");
    link_initialize(&t->rq_link);
    link_initialize(&t->wq_link);
    link_initialize(&t->th_link);
 
    /* call the architecture-specific part of the constructor */
    thr_constructor_arch(t);
   
#ifdef ARCH_HAS_FPU
#ifdef CONFIG_FPU_LAZY
    t->saved_fpu_context = NULL;
#else
    t->saved_fpu_context = slab_alloc(fpu_context_slab, kmflags);
    if (!t->saved_fpu_context)
        return -1;
#endif
#endif  
 
    t->kstack = (uint8_t *) frame_alloc(STACK_FRAMES, FRAME_KA | kmflags);
    if (!t->kstack) {
#ifdef ARCH_HAS_FPU
        if (t->saved_fpu_context)
            slab_free(fpu_context_slab, t->saved_fpu_context);
#endif
        return -1;
    }
 
    mutex_initialize(&t->udebug.lock);
 
    return 0;
}
 
/** Destruction of thread_t object */
static int thr_destructor(void *obj)
{
    thread_t *t = (thread_t *) obj;
 
    /* call the architecture-specific part of the destructor */
    thr_destructor_arch(t);
 
    frame_free(KA2PA(t->kstack));
#ifdef ARCH_HAS_FPU
    if (t->saved_fpu_context)
        slab_free(fpu_context_slab, t->saved_fpu_context);
#endif
    return 1; /* One page freed */
}
 
/** Initialize threads
 *
 * Initialize kernel threads support.
 *
 */
void thread_init(void)
{
    THREAD = NULL;
    atomic_set(&nrdy,0);
    thread_slab = slab_cache_create("thread_slab", sizeof(thread_t), 0,
        thr_constructor, thr_destructor, 0);
 
#ifdef ARCH_HAS_FPU
    fpu_context_slab = slab_cache_create("fpu_slab", sizeof(fpu_context_t),
        FPU_CONTEXT_ALIGN, NULL, NULL, 0);
#endif
 
    avltree_create(&threads_tree);
}
 
/** Make thread ready
 *
 * Switch thread t to the ready state.
 *
 * @param t Thread to make ready.
 *
 */
void thread_ready(thread_t *t)
{
    cpu_t *cpu;
    runq_t *r;
    ipl_t ipl;
    int i, avg;
 
    ipl = interrupts_disable();
 
    spinlock_lock(&t->lock);
 
    ASSERT(!(t->state == Ready));
 
    i = (t->priority < RQ_COUNT - 1) ? ++t->priority : t->priority;
   
    cpu = CPU;
    if (t->flags & THREAD_FLAG_WIRED) {
        ASSERT(t->cpu != NULL);
        cpu = t->cpu;
    }
    t->state = Ready;
    spinlock_unlock(&t->lock);
   
    /*
     * Append t to respective ready queue on respective processor.
     */
    r = &cpu->rq[i];
    spinlock_lock(&r->lock);
    list_append(&t->rq_link, &r->rq_head);
    r->n++;
    spinlock_unlock(&r->lock);
 
    atomic_inc(&nrdy);
    avg = atomic_get(&nrdy) / config.cpu_active;
    atomic_inc(&cpu->nrdy);
 
    interrupts_restore(ipl);
}
 
/** Create new thread
 *
 * Create a new thread.
 *
 * @param func      Thread's implementing function.
 * @param arg       Thread's implementing function argument.
 * @param task      Task to which the thread belongs. The caller must
 *          guarantee that the task won't cease to exist during the
 *          call. The task's lock may not be held.
 * @param flags     Thread flags.
 * @param name      Symbolic name.
 * @param uncounted Thread's accounting doesn't affect accumulated task
 *          accounting.
 *
 * @return      New thread's structure on success, NULL on failure.
 *
 */
thread_t *thread_create(void (* func)(void *), void *arg, task_t *task,
    int flags, char *name, bool uncounted)
{
    thread_t *t;
    ipl_t ipl;
   
    t = (thread_t *) slab_alloc(thread_slab, 0);
    if (!t)
        return NULL;
   
    /* Not needed, but good for debugging */
    memsetb(t->kstack, THREAD_STACK_SIZE * 1 << STACK_FRAMES, 0);
   
    ipl = interrupts_disable();
    spinlock_lock(&tidlock);
    t->tid = ++last_tid;
    spinlock_unlock(&tidlock);
    interrupts_restore(ipl);
   
    context_save(&t->saved_context);
    context_set(&t->saved_context, FADDR(cushion), (uintptr_t) t->kstack,
        THREAD_STACK_SIZE);
   
    the_initialize((the_t *) t->kstack);
   
    ipl = interrupts_disable();
    t->saved_context.ipl = interrupts_read();
    interrupts_restore(ipl);
   
    memcpy(t->name, name, THREAD_NAME_BUFLEN);
   
    t->thread_code = func;
    t->thread_arg = arg;
    t->ticks = -1;
    t->cycles = 0;
    t->uncounted = uncounted;
    t->priority = -1;       /* start in rq[0] */
    t->cpu = NULL;
    t->flags = flags;
    t->state = Entering;
    t->call_me = NULL;
    t->call_me_with = NULL;
   
    timeout_initialize(&t->sleep_timeout);
    t->sleep_interruptible = false;
    t->sleep_queue = NULL;
    t->timeout_pending = 0;
 
    t->in_copy_from_uspace = false;
    t->in_copy_to_uspace = false;
 
    t->interrupted = false;
    t->detached = false;
    waitq_initialize(&t->join_wq);
   
    t->rwlock_holder_type = RWLOCK_NONE;
       
    t->task = task;
   
    t->fpu_context_exists = 0;
    t->fpu_context_engaged = 0;
 
    avltree_node_initialize(&t->threads_tree_node);
    t->threads_tree_node.key = (uintptr_t) t;
   
    /* Init debugging stuff */
    udebug_thread_initialize(&t->udebug);
 
    /* might depend on previous initialization */
    thread_create_arch(t); 
 
    if (!(flags & THREAD_FLAG_NOATTACH))
        thread_attach(t, task);
 
    return t;
}
 
/** Destroy thread memory structure
 *
 * Detach thread from all queues, cpus etc. and destroy it.
 *
 * Assume thread->lock is held!!
 */
void thread_destroy(thread_t *t)
{
    ASSERT(t->state == Exiting || t->state == Lingering);
    ASSERT(t->task);
    ASSERT(t->cpu);
 
    spinlock_lock(&t->cpu->lock);
    if (t->cpu->fpu_owner == t)
        t->cpu->fpu_owner = NULL;
    spinlock_unlock(&t->cpu->lock);
 
    spinlock_unlock(&t->lock);
 
    spinlock_lock(&threads_lock);
    avltree_delete(&threads_tree, &t->threads_tree_node);
    spinlock_unlock(&threads_lock);
 
    /*
     * Detach from the containing task.
     */
    spinlock_lock(&t->task->lock);
    list_remove(&t->th_link);
    spinlock_unlock(&t->task->lock);   
 
    /*
     * t is guaranteed to be the very last thread of its task.
     * It is safe to destroy the task.
     */
    if (atomic_predec(&t->task->refcount) == 0)
        task_destroy(t->task);
   
    slab_free(thread_slab, t);
}
 
/** Make the thread visible to the system.
 *
 * Attach the thread structure to the current task and make it visible in the
 * threads_tree.
 *
 * @param t Thread to be attached to the task.
 * @param task  Task to which the thread is to be attached.
 */
void thread_attach(thread_t *t, task_t *task)
{
    ipl_t ipl;
 
    /*
     * Attach to the specified task.
     */
    ipl = interrupts_disable();
    spinlock_lock(&task->lock);
    atomic_inc(&task->refcount);
    /* Must not count kbox thread into lifecount */
    if (t->flags & THREAD_FLAG_USPACE)
        atomic_inc(&task->lifecount);
    list_append(&t->th_link, &task->th_head);
    spinlock_unlock(&task->lock);
 
    /*
     * Register this thread in the system-wide list.
     */
    spinlock_lock(&threads_lock);
    avltree_insert(&threads_tree, &t->threads_tree_node);
    spinlock_unlock(&threads_lock);
   
    interrupts_restore(ipl);
}
 
/** Terminate thread.
 *
 * End current thread execution and switch it to the exiting state. All pending
 * timeouts are executed.
 */
void thread_exit(void)
{
    ipl_t ipl;
 
    if (THREAD->flags & THREAD_FLAG_USPACE) {
        /* Generate udebug THREAD_E event */
        udebug_thread_e_event();
 
        if (atomic_predec(&TASK->lifecount) == 0) {
            /*
             * We are the last userspace thread in the task that
             * still has not exited. With the exception of the
             * moment the task was created, new userspace threads
             * can only be created by threads of the same task.
             * We are safe to perform cleanup.
             */
            ipc_cleanup();
            futex_cleanup();
            LOG("Cleanup of task %" PRIu64" completed.", TASK->taskid);
        }
    }
 
restart:
    ipl = interrupts_disable();
    spinlock_lock(&THREAD->lock);
    if (THREAD->timeout_pending) {
        /* busy waiting for timeouts in progress */
        spinlock_unlock(&THREAD->lock);
        interrupts_restore(ipl);
        goto restart;
    }
   
    THREAD->state = Exiting;
    spinlock_unlock(&THREAD->lock);
    scheduler();
 
    /* Not reached */
    while (1)
        ;
}
 
 
/** Thread sleep
 *
 * Suspend execution of the current thread.
 *
 * @param sec Number of seconds to sleep.
 *
 */
void thread_sleep(uint32_t sec)
{
    thread_usleep(sec * 1000000);
}
 
/** Wait for another thread to exit.
 *
 * @param t Thread to join on exit.
 * @param usec Timeout in microseconds.
 * @param flags Mode of operation.
 *
 * @return An error code from errno.h or an error code from synch.h.
 */
int thread_join_timeout(thread_t *t, uint32_t usec, int flags)
{
    ipl_t ipl;
    int rc;
 
    if (t == THREAD)
        return EINVAL;
 
    /*
     * Since thread join can only be called once on an undetached thread,
     * the thread pointer is guaranteed to be still valid.
     */
   
    ipl = interrupts_disable();
    spinlock_lock(&t->lock);
    ASSERT(!t->detached);
    spinlock_unlock(&t->lock);
    interrupts_restore(ipl);
   
    rc = waitq_sleep_timeout(&t->join_wq, usec, flags);
   
    return rc; 
}
 
/** Detach thread.
 *
 * Mark the thread as detached, if the thread is already in the Lingering
 * state, deallocate its resources.
 *
 * @param t Thread to be detached.
 */
void thread_detach(thread_t *t)
{
    ipl_t ipl;
 
    /*
     * Since the thread is expected not to be already detached,
     * pointer to it must be still valid.
     */
    ipl = interrupts_disable();
    spinlock_lock(&t->lock);
    ASSERT(!t->detached);
    if (t->state == Lingering) {
        thread_destroy(t);  /* unlocks &t->lock */
        interrupts_restore(ipl);
        return;
    } else {
        t->detached = true;
    }
    spinlock_unlock(&t->lock);
    interrupts_restore(ipl);
}
 
/** Thread usleep
 *
 * Suspend execution of the current thread.
 *
 * @param usec Number of microseconds to sleep.
 *
 */
void thread_usleep(uint32_t usec)
{
    waitq_t wq;
                 
    waitq_initialize(&wq);
 
    (void) waitq_sleep_timeout(&wq, usec, SYNCH_FLAGS_NON_BLOCKING);
}
 
/** Register thread out-of-context invocation
 *
 * Register a function and its argument to be executed
 * on next context switch to the current thread.
 *
 * @param call_me      Out-of-context function.
 * @param call_me_with Out-of-context function argument.
 *
 */
void thread_register_call_me(void (* call_me)(void *), void *call_me_with)
{
    ipl_t ipl;
   
    ipl = interrupts_disable();
    spinlock_lock(&THREAD->lock);
    THREAD->call_me = call_me;
    THREAD->call_me_with = call_me_with;
    spinlock_unlock(&THREAD->lock);
    interrupts_restore(ipl);
}
 
static bool thread_walker(avltree_node_t *node, void *arg)
{
    thread_t *t = avltree_get_instance(node, thread_t, threads_tree_node);
   
    uint64_t cycles;
    char suffix;
    order(t->cycles, &cycles, &suffix);
 
#ifdef __32_BITS__
    printf("%-6" PRIu64" %-10s %10p %-8s %10p %-3" PRIu32 " %10p %10p %9" PRIu64 "%c ",
        t->tid, t->name, t, thread_states[t->state], t->task,
        t->task->context, t->thread_code, t->kstack, cycles, suffix);
#endif
 
#ifdef __64_BITS__
    printf("%-6" PRIu64" %-10s %18p %-8s %18p %-3" PRIu32 " %18p %18p %9" PRIu64 "%c ",
        t->tid, t->name, t, thread_states[t->state], t->task,
        t->task->context, t->thread_code, t->kstack, cycles, suffix);
#endif
           
    if (t->cpu)
        printf("%-4u", t->cpu->id);
    else
        printf("none");
           
    if (t->state == Sleeping) {
#ifdef __32_BITS__
        printf(" %10p", t->sleep_queue);
#endif
 
#ifdef __64_BITS__
        printf(" %18p", t->sleep_queue);
#endif
    }
           
    printf("\n");
 
    return true;
}
 
/** Print list of threads debug info */
void thread_print_list(void)
{
    ipl_t ipl;
   
    /* Messing with thread structures, avoid deadlock */
    ipl = interrupts_disable();
    spinlock_lock(&threads_lock);
 
#ifdef __32_BITS__  
    printf("tid    name       address    state    task       "
        "ctx code       stack      cycles     cpu  "
        "waitqueue\n");
    printf("------ ---------- ---------- -------- ---------- "
        "--- ---------- ---------- ---------- ---- "
        "----------\n");
#endif
 
#ifdef __64_BITS__
    printf("tid    name       address            state    task               "
        "ctx code               stack              cycles     cpu  "
        "waitqueue\n");
    printf("------ ---------- ------------------ -------- ------------------ "
        "--- ------------------ ------------------ ---------- ---- "
        "------------------\n");
#endif
 
    avltree_walk(&threads_tree, thread_walker, NULL);
 
    spinlock_unlock(&threads_lock);
    interrupts_restore(ipl);
}
 
/** Check whether thread exists.
 *
 * Note that threads_lock must be already held and
 * interrupts must be already disabled.
 *
 * @param t Pointer to thread.
 *
 * @return True if thread t is known to the system, false otherwise.
 */
bool thread_exists(thread_t *t)
{
    avltree_node_t *node;
 
    node = avltree_search(&threads_tree, (avltree_key_t) ((uintptr_t) t));
   
    return node != NULL;
}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
/** Update accounting of current thread.
 *
 * Note that thread_lock on THREAD must be already held and
 * interrupts must be already disabled.
 *
 */
void thread_update_accounting(void)
{
    uint64_t time = get_cycle();
    THREAD->cycles += time - THREAD->last_cycle;
    THREAD->last_cycle = time;
}
 
/** Process syscall to create new thread.
 *
 */
unative_t sys_thread_create(uspace_arg_t *uspace_uarg, char *uspace_name,
    thread_id_t *uspace_thread_id)
{
    thread_t *t;
    char namebuf[THREAD_NAME_BUFLEN];
    uspace_arg_t *kernel_uarg;
    int rc;
 
    rc = copy_from_uspace(namebuf, uspace_name, THREAD_NAME_BUFLEN);
    if (rc != 0)
        return (unative_t) rc;
 
    /*
     * In case of failure, kernel_uarg will be deallocated in this function.
     * In case of success, kernel_uarg will be freed in uinit().
     */
    kernel_uarg = (uspace_arg_t *) malloc(sizeof(uspace_arg_t), 0);
   
    rc = copy_from_uspace(kernel_uarg, uspace_uarg, sizeof(uspace_arg_t));
    if (rc != 0) {
        free(kernel_uarg);
        return (unative_t) rc;
    }
 
    t = thread_create(uinit, kernel_uarg, TASK,
        THREAD_FLAG_USPACE | THREAD_FLAG_NOATTACH, namebuf, false);
    if (t) {
        if (uspace_thread_id != NULL) {
            int rc;
 
            rc = copy_to_uspace(uspace_thread_id, &t->tid,
                sizeof(t->tid));
            if (rc != 0) {
                /*
                 * We have encountered a failure, but the thread
                 * has already been created. We need to undo its
                 * creation now.
                 */
 
                /*
                 * The new thread structure is initialized, but
                 * is still not visible to the system.
                 * We can safely deallocate it.
                 */
                slab_free(thread_slab, t);
                free(kernel_uarg);
 
                return (unative_t) rc;
             }
        }
        thread_attach(t, TASK);
        thread_ready(t);
 
        /* Generate udebug THREAD_B event */
        udebug_thread_b_event(t);
 
        return 0;
    } else
        free(kernel_uarg);
 
    return (unative_t) ENOMEM;
}
 
/** Process syscall to terminate thread.
 *
 */
unative_t sys_thread_exit(int uspace_status)
{
    thread_exit();
    /* Unreachable */
    return 0;
}
 
/** Syscall for getting TID.
 *
 * @param uspace_thread_id Userspace address of 8-byte buffer where to store
 * current thread ID.
 *
 * @return 0 on success or an error code from @ref errno.h.
 */
unative_t sys_thread_get_id(thread_id_t *uspace_thread_id)
{
    /*
     * No need to acquire lock on THREAD because tid
     * remains constant for the lifespan of the thread.
     */
    return (unative_t) copy_to_uspace(uspace_thread_id, &THREAD->tid,
        sizeof(THREAD->tid));
}
 
/** @}
 */
 

Rev 3424	Rev 3425
1	/*	1	/*
2	* Copyright (c) 2001-2004 Jakub Jermar	2	* Copyright (c) 2001-2004 Jakub Jermar
3	* All rights reserved.	3	* All rights reserved.
4	*	4	*
5	* Redistribution and use in source and binary forms, with or without	5	* Redistribution and use in source and binary forms, with or without
6	* modification, are permitted provided that the following conditions	6	* modification, are permitted provided that the following conditions
7	* are met:	7	* are met:
8	*	8	*
9	* - Redistributions of source code must retain the above copyright	9	* - Redistributions of source code must retain the above copyright
10	* notice, this list of conditions and the following disclaimer.	10	* notice, this list of conditions and the following disclaimer.
11	* - Redistributions in binary form must reproduce the above copyright	11	* - Redistributions in binary form must reproduce the above copyright
12	* notice, this list of conditions and the following disclaimer in the	12	* notice, this list of conditions and the following disclaimer in the
13	* documentation and/or other materials provided with the distribution.	13	* documentation and/or other materials provided with the distribution.
14	* - The name of the author may not be used to endorse or promote products	14	* - The name of the author may not be used to endorse or promote products
15	* derived from this software without specific prior written permission.	15	* derived from this software without specific prior written permission.
16	*	16	*
17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR	17	* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES	18	* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.	19	* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,	20	* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT	21	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,	22	* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY	23	* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT	24	* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF	25	* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.	26	* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27	*/	27	*/
28		28
29	/** @addtogroup genericproc	29	/** @addtogroup genericproc
30	* @{	30	* @{
31	*/	31	*/
32		32
33	/**	33	/**
34	* @file	34	* @file
35	* @brief Thread management functions.	35	* @brief Thread management functions.
36	*/	36	*/
37		37
38	#include <proc/scheduler.h>	38	#include <proc/scheduler.h>
39	#include <proc/thread.h>	39	#include <proc/thread.h>
40	#include <proc/task.h>	40	#include <proc/task.h>
41	#include <proc/uarg.h>	41	#include <proc/uarg.h>
42	#include <mm/frame.h>	42	#include <mm/frame.h>
43	#include <mm/page.h>	43	#include <mm/page.h>
44	#include <arch/asm.h>	44	#include <arch/asm.h>
45	#include <arch/cycle.h>	45	#include <arch/cycle.h>
46	#include <arch.h>	46	#include <arch.h>
47	#include <synch/synch.h>	47	#include <synch/synch.h>
48	#include <synch/spinlock.h>	48	#include <synch/spinlock.h>
49	#include <synch/waitq.h>	49	#include <synch/waitq.h>
50	#include <synch/rwlock.h>	50	#include <synch/rwlock.h>
51	#include <cpu.h>	51	#include <cpu.h>
52	#include <func.h>	52	#include <func.h>
53	#include <context.h>	53	#include <context.h>
54	#include <adt/avl.h>	54	#include <adt/avl.h>
55	#include <adt/list.h>	55	#include <adt/list.h>
56	#include <time/clock.h>	56	#include <time/clock.h>
57	#include <time/timeout.h>	57	#include <time/timeout.h>
58	#include <config.h>	58	#include <config.h>
59	#include <arch/interrupt.h>	59	#include <arch/interrupt.h>
60	#include <smp/ipi.h>	60	#include <smp/ipi.h>
61	#include <arch/faddr.h>	61	#include <arch/faddr.h>
62	#include <atomic.h>	62	#include <atomic.h>
63	#include <memstr.h>	63	#include <memstr.h>
64	#include <print.h>	64	#include <print.h>
65	#include <mm/slab.h>	65	#include <mm/slab.h>
66	#include <debug.h>	66	#include <debug.h>
67	#include <main/uinit.h>	67	#include <main/uinit.h>
68	#include <syscall/copy.h>	68	#include <syscall/copy.h>
69	#include <errno.h>	69	#include <errno.h>
70		70
71		71
72	#ifndef LOADED_PROG_STACK_PAGES_NO	72	#ifndef LOADED_PROG_STACK_PAGES_NO
73	#define LOADED_PROG_STACK_PAGES_NO 1	73	#define LOADED_PROG_STACK_PAGES_NO 1
74	#endif	74	#endif
75		75
76		76
77	/** Thread states */	77	/** Thread states */
78	char *thread_states[] = {	78	char *thread_states[] = {
79	"Invalid",	79	"Invalid",
80	"Running",	80	"Running",
81	"Sleeping",	81	"Sleeping",
82	"Ready",	82	"Ready",
83	"Entering",	83	"Entering",
84	"Exiting",	84	"Exiting",
85	"Lingering"	85	"Lingering"
86	};	86	};
87		87
88	/** Lock protecting the threads_tree AVL tree.	88	/** Lock protecting the threads_tree AVL tree.
89	*	89	*
90	* For locking rules, see declaration thereof.	90	* For locking rules, see declaration thereof.
91	*/	91	*/
92	SPINLOCK_INITIALIZE(threads_lock);	92	SPINLOCK_INITIALIZE(threads_lock);
93		93
94	/** AVL tree of all threads.	94	/** AVL tree of all threads.
95	*	95	*
96	* When a thread is found in the threads_tree AVL tree, it is guaranteed to	96	* When a thread is found in the threads_tree AVL tree, it is guaranteed to
97	* exist as long as the threads_lock is held.	97	* exist as long as the threads_lock is held.
98	*/	98	*/
99	avltree_t threads_tree;	99	avltree_t threads_tree;
100		100
101	SPINLOCK_INITIALIZE(tidlock);	101	SPINLOCK_INITIALIZE(tidlock);
102	thread_id_t last_tid = 0;	102	thread_id_t last_tid = 0;
103		103
104	static slab_cache_t *thread_slab;	104	static slab_cache_t *thread_slab;
105	#ifdef ARCH_HAS_FPU	105	#ifdef ARCH_HAS_FPU
106	slab_cache_t *fpu_context_slab;	106	slab_cache_t *fpu_context_slab;
107	#endif	107	#endif
108		108
109	/** Thread wrapper.	109	/** Thread wrapper.
110	*	110	*
111	* This wrapper is provided to ensure that every thread makes a call to	111	* This wrapper is provided to ensure that every thread makes a call to
112	* thread_exit() when its implementing function returns.	112	* thread_exit() when its implementing function returns.
113	*	113	*
114	* interrupts_disable() is assumed.	114	* interrupts_disable() is assumed.
115	*	115	*
116	*/	116	*/
117	static void cushion(void)	117	static void cushion(void)
118	{	118	{
119	void (f)(void ) = THREAD->thread_code;	119	void (f)(void ) = THREAD->thread_code;
120	void *arg = THREAD->thread_arg;	120	void *arg = THREAD->thread_arg;
121	THREAD->last_cycle = get_cycle();	121	THREAD->last_cycle = get_cycle();
122		122
123	/* This is where each thread wakes up after its creation */	123	/* This is where each thread wakes up after its creation */
124	spinlock_unlock(&THREAD->lock);	124	spinlock_unlock(&THREAD->lock);
125	interrupts_enable();	125	interrupts_enable();
126		126
127	f(arg);	127	f(arg);
128		128
129	/* Accumulate accounting to the task */	129	/* Accumulate accounting to the task */
130	ipl_t ipl = interrupts_disable();	130	ipl_t ipl = interrupts_disable();
131		131
132	spinlock_lock(&THREAD->lock);	132	spinlock_lock(&THREAD->lock);
133	if (!THREAD->uncounted) {	133	if (!THREAD->uncounted) {
134	thread_update_accounting();	134	thread_update_accounting();
135	uint64_t cycles = THREAD->cycles;	135	uint64_t cycles = THREAD->cycles;
136	THREAD->cycles = 0;	136	THREAD->cycles = 0;
137	spinlock_unlock(&THREAD->lock);	137	spinlock_unlock(&THREAD->lock);
138		138
139	spinlock_lock(&TASK->lock);	139	spinlock_lock(&TASK->lock);
140	TASK->cycles += cycles;	140	TASK->cycles += cycles;
141	spinlock_unlock(&TASK->lock);	141	spinlock_unlock(&TASK->lock);
142	} else	142	} else
143	spinlock_unlock(&THREAD->lock);	143	spinlock_unlock(&THREAD->lock);
144		144
145	interrupts_restore(ipl);	145	interrupts_restore(ipl);
146		146
147	thread_exit();	147	thread_exit();
148	/* not reached */	148	/* not reached */
149	}	149	}
150		150
151	/** Initialization and allocation for thread_t structure */	151	/** Initialization and allocation for thread_t structure */
152	static int thr_constructor(void *obj, int kmflags)	152	static int thr_constructor(void *obj, int kmflags)
153	{	153	{
154	thread_t t = (thread_t ) obj;	154	thread_t t = (thread_t ) obj;
155		155
156	spinlock_initialize(&t->lock, "thread_t_lock");	156	spinlock_initialize(&t->lock, "thread_t_lock");
157	link_initialize(&t->rq_link);	157	link_initialize(&t->rq_link);
158	link_initialize(&t->wq_link);	158	link_initialize(&t->wq_link);
159	link_initialize(&t->th_link);	159	link_initialize(&t->th_link);
160		160
161	/* call the architecture-specific part of the constructor */	161	/* call the architecture-specific part of the constructor */
162	thr_constructor_arch(t);	162	thr_constructor_arch(t);
163		163
164	#ifdef ARCH_HAS_FPU	164	#ifdef ARCH_HAS_FPU
165	#ifdef CONFIG_FPU_LAZY	165	#ifdef CONFIG_FPU_LAZY
166	t->saved_fpu_context = NULL;	166	t->saved_fpu_context = NULL;
167	#else	167	#else
168	t->saved_fpu_context = slab_alloc(fpu_context_slab, kmflags);	168	t->saved_fpu_context = slab_alloc(fpu_context_slab, kmflags);
169	if (!t->saved_fpu_context)	169	if (!t->saved_fpu_context)
170	return -1;	170	return -1;
171	#endif	171	#endif
172	#endif	172	#endif
173		173
174	t->kstack = (uint8_t *) frame_alloc(STACK_FRAMES, FRAME_KA \| kmflags);	174	t->kstack = (uint8_t *) frame_alloc(STACK_FRAMES, FRAME_KA \| kmflags);
175	if (!t->kstack) {	175	if (!t->kstack) {
176	#ifdef ARCH_HAS_FPU	176	#ifdef ARCH_HAS_FPU
177	if (t->saved_fpu_context)	177	if (t->saved_fpu_context)
178	slab_free(fpu_context_slab, t->saved_fpu_context);	178	slab_free(fpu_context_slab, t->saved_fpu_context);
179	#endif	179	#endif
180	return -1;	180	return -1;
181	}	181	}
182		182
183	mutex_initialize(&t->udebug.lock);	183	mutex_initialize(&t->udebug.lock);
184		184
185	return 0;	185	return 0;
186	}	186	}
187		187
188	/** Destruction of thread_t object */	188	/** Destruction of thread_t object */
189	static int thr_destructor(void *obj)	189	static int thr_destructor(void *obj)
190	{	190	{
191	thread_t t = (thread_t ) obj;	191	thread_t t = (thread_t ) obj;
192		192
193	/* call the architecture-specific part of the destructor */	193	/* call the architecture-specific part of the destructor */
194	thr_destructor_arch(t);	194	thr_destructor_arch(t);
195		195
196	frame_free(KA2PA(t->kstack));	196	frame_free(KA2PA(t->kstack));
197	#ifdef ARCH_HAS_FPU	197	#ifdef ARCH_HAS_FPU
198	if (t->saved_fpu_context)	198	if (t->saved_fpu_context)
199	slab_free(fpu_context_slab, t->saved_fpu_context);	199	slab_free(fpu_context_slab, t->saved_fpu_context);
200	#endif	200	#endif
201	return 1; /* One page freed */	201	return 1; /* One page freed */
202	}	202	}
203		203
204	/** Initialize threads	204	/** Initialize threads
205	*	205	*
206	* Initialize kernel threads support.	206	* Initialize kernel threads support.
207	*	207	*
208	*/	208	*/
209	void thread_init(void)	209	void thread_init(void)
210	{	210	{
211	THREAD = NULL;	211	THREAD = NULL;
212	atomic_set(&nrdy,0);	212	atomic_set(&nrdy,0);
213	thread_slab = slab_cache_create("thread_slab", sizeof(thread_t), 0,	213	thread_slab = slab_cache_create("thread_slab", sizeof(thread_t), 0,
214	thr_constructor, thr_destructor, 0);	214	thr_constructor, thr_destructor, 0);
215		215
216	#ifdef ARCH_HAS_FPU	216	#ifdef ARCH_HAS_FPU
217	fpu_context_slab = slab_cache_create("fpu_slab", sizeof(fpu_context_t),	217	fpu_context_slab = slab_cache_create("fpu_slab", sizeof(fpu_context_t),
218	FPU_CONTEXT_ALIGN, NULL, NULL, 0);	218	FPU_CONTEXT_ALIGN, NULL, NULL, 0);
219	#endif	219	#endif
220		220
221	avltree_create(&threads_tree);	221	avltree_create(&threads_tree);
222	}	222	}
223		223
224	/** Make thread ready	224	/** Make thread ready
225	*	225	*
226	* Switch thread t to the ready state.	226	* Switch thread t to the ready state.
227	*	227	*
228	* @param t Thread to make ready.	228	* @param t Thread to make ready.
229	*	229	*
230	*/	230	*/
231	void thread_ready(thread_t *t)	231	void thread_ready(thread_t *t)
232	{	232	{
233	cpu_t *cpu;	233	cpu_t *cpu;
234	runq_t *r;	234	runq_t *r;
235	ipl_t ipl;	235	ipl_t ipl;
236	int i, avg;	236	int i, avg;
237		237
238	ipl = interrupts_disable();	238	ipl = interrupts_disable();
239		239
240	spinlock_lock(&t->lock);	240	spinlock_lock(&t->lock);
241		241
242	ASSERT(!(t->state == Ready));	242	ASSERT(!(t->state == Ready));
243		243
244	i = (t->priority < RQ_COUNT - 1) ? ++t->priority : t->priority;	244	i = (t->priority < RQ_COUNT - 1) ? ++t->priority : t->priority;
245		245
246	cpu = CPU;	246	cpu = CPU;
247	if (t->flags & THREAD_FLAG_WIRED) {	247	if (t->flags & THREAD_FLAG_WIRED) {
248	ASSERT(t->cpu != NULL);	248	ASSERT(t->cpu != NULL);
249	cpu = t->cpu;	249	cpu = t->cpu;
250	}	250	}
251	t->state = Ready;	251	t->state = Ready;
252	spinlock_unlock(&t->lock);	252	spinlock_unlock(&t->lock);
253		253
254	/*	254	/*
255	* Append t to respective ready queue on respective processor.	255	* Append t to respective ready queue on respective processor.
256	*/	256	*/
257	r = &cpu->rq[i];	257	r = &cpu->rq[i];
258	spinlock_lock(&r->lock);	258	spinlock_lock(&r->lock);
259	list_append(&t->rq_link, &r->rq_head);	259	list_append(&t->rq_link, &r->rq_head);
260	r->n++;	260	r->n++;
261	spinlock_unlock(&r->lock);	261	spinlock_unlock(&r->lock);
262		262
263	atomic_inc(&nrdy);	263	atomic_inc(&nrdy);
264	avg = atomic_get(&nrdy) / config.cpu_active;	264	avg = atomic_get(&nrdy) / config.cpu_active;
265	atomic_inc(&cpu->nrdy);	265	atomic_inc(&cpu->nrdy);
266		266
267	interrupts_restore(ipl);	267	interrupts_restore(ipl);
268	}	268	}
269		269
270	/** Create new thread	270	/** Create new thread
271	*	271	*
272	* Create a new thread.	272	* Create a new thread.
273	*	273	*
274	* @param func Thread's implementing function.	274	* @param func Thread's implementing function.
275	* @param arg Thread's implementing function argument.	275	* @param arg Thread's implementing function argument.
276	* @param task Task to which the thread belongs. The caller must	276	* @param task Task to which the thread belongs. The caller must
277	* guarantee that the task won't cease to exist during the	277	* guarantee that the task won't cease to exist during the
278	* call. The task's lock may not be held.	278	* call. The task's lock may not be held.
279	* @param flags Thread flags.	279	* @param flags Thread flags.
280	* @param name Symbolic name.	280	* @param name Symbolic name.
281	* @param uncounted Thread's accounting doesn't affect accumulated task	281	* @param uncounted Thread's accounting doesn't affect accumulated task
282	* accounting.	282	* accounting.
283	*	283	*
284	* @return New thread's structure on success, NULL on failure.	284	* @return New thread's structure on success, NULL on failure.
285	*	285	*
286	*/	286	*/
287	thread_t thread_create(void ( func)(void ), void arg, task_t *task,	287	thread_t thread_create(void ( func)(void ), void arg, task_t *task,
288	int flags, char *name, bool uncounted)	288	int flags, char *name, bool uncounted)
289	{	289	{
290	thread_t *t;	290	thread_t *t;
291	ipl_t ipl;	291	ipl_t ipl;
292		292
293	t = (thread_t *) slab_alloc(thread_slab, 0);	293	t = (thread_t *) slab_alloc(thread_slab, 0);
294	if (!t)	294	if (!t)
295	return NULL;	295	return NULL;
296		296
297	/* Not needed, but good for debugging */	297	/* Not needed, but good for debugging */
298	memsetb(t->kstack, THREAD_STACK_SIZE * 1 << STACK_FRAMES, 0);	298	memsetb(t->kstack, THREAD_STACK_SIZE * 1 << STACK_FRAMES, 0);
299		299
300	ipl = interrupts_disable();	300	ipl = interrupts_disable();
301	spinlock_lock(&tidlock);	301	spinlock_lock(&tidlock);
302	t->tid = ++last_tid;	302	t->tid = ++last_tid;
303	spinlock_unlock(&tidlock);	303	spinlock_unlock(&tidlock);
304	interrupts_restore(ipl);	304	interrupts_restore(ipl);
305		305
306	context_save(&t->saved_context);	306	context_save(&t->saved_context);
307	context_set(&t->saved_context, FADDR(cushion), (uintptr_t) t->kstack,	307	context_set(&t->saved_context, FADDR(cushion), (uintptr_t) t->kstack,
308	THREAD_STACK_SIZE);	308	THREAD_STACK_SIZE);
309		309
310	the_initialize((the_t *) t->kstack);	310	the_initialize((the_t *) t->kstack);
311		311
312	ipl = interrupts_disable();	312	ipl = interrupts_disable();
313	t->saved_context.ipl = interrupts_read();	313	t->saved_context.ipl = interrupts_read();
314	interrupts_restore(ipl);	314	interrupts_restore(ipl);
315		315
316	memcpy(t->name, name, THREAD_NAME_BUFLEN);	316	memcpy(t->name, name, THREAD_NAME_BUFLEN);
317		317
318	t->thread_code = func;	318	t->thread_code = func;
319	t->thread_arg = arg;	319	t->thread_arg = arg;
320	t->ticks = -1;	320	t->ticks = -1;
321	t->cycles = 0;	321	t->cycles = 0;
322	t->uncounted = uncounted;	322	t->uncounted = uncounted;
323	t->priority = -1; /* start in rq[0] */	323	t->priority = -1; /* start in rq[0] */
324	t->cpu = NULL;	324	t->cpu = NULL;
325	t->flags = flags;	325	t->flags = flags;
326	t->state = Entering;	326	t->state = Entering;
327	t->call_me = NULL;	327	t->call_me = NULL;
328	t->call_me_with = NULL;	328	t->call_me_with = NULL;
329		329
330	timeout_initialize(&t->sleep_timeout);	330	timeout_initialize(&t->sleep_timeout);
331	t->sleep_interruptible = false;	331	t->sleep_interruptible = false;
332	t->sleep_queue = NULL;	332	t->sleep_queue = NULL;
333	t->timeout_pending = 0;	333	t->timeout_pending = 0;
334		334
335	t->in_copy_from_uspace = false;	335	t->in_copy_from_uspace = false;
336	t->in_copy_to_uspace = false;	336	t->in_copy_to_uspace = false;
337		337
338	t->interrupted = false;	338	t->interrupted = false;
339	t->detached = false;	339	t->detached = false;
340	waitq_initialize(&t->join_wq);	340	waitq_initialize(&t->join_wq);
341		341
342	t->rwlock_holder_type = RWLOCK_NONE;	342	t->rwlock_holder_type = RWLOCK_NONE;
343		343
344	t->task = task;	344	t->task = task;
345		345
346	t->fpu_context_exists = 0;	346	t->fpu_context_exists = 0;
347	t->fpu_context_engaged = 0;	347	t->fpu_context_engaged = 0;
348		348
349	avltree_node_initialize(&t->threads_tree_node);	349	avltree_node_initialize(&t->threads_tree_node);
350	t->threads_tree_node.key = (uintptr_t) t;	350	t->threads_tree_node.key = (uintptr_t) t;
351		351
352	/* Init debugging stuff */	352	/* Init debugging stuff */
353	udebug_thread_initialize(&t->udebug);	353	udebug_thread_initialize(&t->udebug);
354		354
355	/* might depend on previous initialization */	355	/* might depend on previous initialization */
356	thread_create_arch(t);	356	thread_create_arch(t);
357		357
358	if (!(flags & THREAD_FLAG_NOATTACH))	358	if (!(flags & THREAD_FLAG_NOATTACH))
359	thread_attach(t, task);	359	thread_attach(t, task);
360		360
361	return t;	361	return t;
362	}	362	}
363		363
364	/** Destroy thread memory structure	364	/** Destroy thread memory structure
365	*	365	*
366	* Detach thread from all queues, cpus etc. and destroy it.	366	* Detach thread from all queues, cpus etc. and destroy it.
367	*	367	*
368	* Assume thread->lock is held!!	368	* Assume thread->lock is held!!
369	*/	369	*/
370	void thread_destroy(thread_t *t)	370	void thread_destroy(thread_t *t)
371	{	371	{
372	ASSERT(t->state == Exiting \|\| t->state == Lingering);	372	ASSERT(t->state == Exiting \|\| t->state == Lingering);
373	ASSERT(t->task);	373	ASSERT(t->task);
374	ASSERT(t->cpu);	374	ASSERT(t->cpu);
375		375
376	spinlock_lock(&t->cpu->lock);	376	spinlock_lock(&t->cpu->lock);
377	if (t->cpu->fpu_owner == t)	377	if (t->cpu->fpu_owner == t)
378	t->cpu->fpu_owner = NULL;	378	t->cpu->fpu_owner = NULL;
379	spinlock_unlock(&t->cpu->lock);	379	spinlock_unlock(&t->cpu->lock);
380		380
381	spinlock_unlock(&t->lock);	381	spinlock_unlock(&t->lock);
382		382
383	spinlock_lock(&threads_lock);	383	spinlock_lock(&threads_lock);
384	avltree_delete(&threads_tree, &t->threads_tree_node);	384	avltree_delete(&threads_tree, &t->threads_tree_node);
385	spinlock_unlock(&threads_lock);	385	spinlock_unlock(&threads_lock);
386		386
387	/*	387	/*
388	* Detach from the containing task.	388	* Detach from the containing task.
389	*/	389	*/
390	spinlock_lock(&t->task->lock);	390	spinlock_lock(&t->task->lock);
391	list_remove(&t->th_link);	391	list_remove(&t->th_link);
392	spinlock_unlock(&t->task->lock);	392	spinlock_unlock(&t->task->lock);
393		393
394	/*	394	/*
395	* t is guaranteed to be the very last thread of its task.	395	* t is guaranteed to be the very last thread of its task.
396	* It is safe to destroy the task.	396	* It is safe to destroy the task.
397	*/	397	*/
398	if (atomic_predec(&t->task->refcount) == 0)	398	if (atomic_predec(&t->task->refcount) == 0)
399	task_destroy(t->task);	399	task_destroy(t->task);
400		400
401	slab_free(thread_slab, t);	401	slab_free(thread_slab, t);
402	}	402	}
403		403
404	/** Make the thread visible to the system.	404	/** Make the thread visible to the system.
405	*	405	*
406	* Attach the thread structure to the current task and make it visible in the	406	* Attach the thread structure to the current task and make it visible in the
407	* threads_tree.	407	* threads_tree.
408	*	408	*
409	* @param t Thread to be attached to the task.	409	* @param t Thread to be attached to the task.
410	* @param task Task to which the thread is to be attached.	410	* @param task Task to which the thread is to be attached.
411	*/	411	*/
412	void thread_attach(thread_t t, task_t task)	412	void thread_attach(thread_t t, task_t task)
413	{	413	{
414	ipl_t ipl;	414	ipl_t ipl;
415		415
416	/*	416	/*
417	* Attach to the specified task.	417	* Attach to the specified task.
418	*/	418	*/
419	ipl = interrupts_disable();	419	ipl = interrupts_disable();
420	spinlock_lock(&task->lock);	420	spinlock_lock(&task->lock);
421	atomic_inc(&task->refcount);	421	atomic_inc(&task->refcount);
422	/* Must not count kbox thread into lifecount */	422	/* Must not count kbox thread into lifecount */
423	if (t->flags & THREAD_FLAG_USPACE)	423	if (t->flags & THREAD_FLAG_USPACE)
424	atomic_inc(&task->lifecount);	424	atomic_inc(&task->lifecount);
425	list_append(&t->th_link, &task->th_head);	425	list_append(&t->th_link, &task->th_head);
426	spinlock_unlock(&task->lock);	426	spinlock_unlock(&task->lock);
427		427
428	/*	428	/*
429	* Register this thread in the system-wide list.	429	* Register this thread in the system-wide list.
430	*/	430	*/
431	spinlock_lock(&threads_lock);	431	spinlock_lock(&threads_lock);
432	avltree_insert(&threads_tree, &t->threads_tree_node);	432	avltree_insert(&threads_tree, &t->threads_tree_node);
433	spinlock_unlock(&threads_lock);	433	spinlock_unlock(&threads_lock);
434		434
435	interrupts_restore(ipl);	435	interrupts_restore(ipl);
436	}	436	}
437		437
438	/** Terminate thread.	438	/** Terminate thread.
439	*	439	*
440	* End current thread execution and switch it to the exiting state. All pending	440	* End current thread execution and switch it to the exiting state. All pending
441	* timeouts are executed.	441	* timeouts are executed.
442	*/	442	*/
443	void thread_exit(void)	443	void thread_exit(void)
444	{	444	{
445	ipl_t ipl;	445	ipl_t ipl;
446		446
447	if (THREAD->flags & THREAD_FLAG_USPACE) {	447	if (THREAD->flags & THREAD_FLAG_USPACE) {
448	/* Generate udebug THREAD_E event */	448	/* Generate udebug THREAD_E event */
449	udebug_thread_e_event();	449	udebug_thread_e_event();
450		450
451	if (atomic_predec(&TASK->lifecount) == 0) {	451	if (atomic_predec(&TASK->lifecount) == 0) {
452	/*	452	/*
453	* We are the last userspace thread in the task that	453	* We are the last userspace thread in the task that
454	* still has not exited. With the exception of the	454	* still has not exited. With the exception of the
455	* moment the task was created, new userspace threads	455	* moment the task was created, new userspace threads
456	* can only be created by threads of the same task.	456	* can only be created by threads of the same task.
457	* We are safe to perform cleanup.	457	* We are safe to perform cleanup.
458	*/	458	*/
459	ipc_cleanup();	459	ipc_cleanup();
460	futex_cleanup();	460	futex_cleanup();
461	LOG("Cleanup of task %" PRIu64" completed.", TASK->taskid);	461	LOG("Cleanup of task %" PRIu64" completed.", TASK->taskid);
462	}	462	}
463	}	463	}
464		464
465	restart:	465	restart:
466	ipl = interrupts_disable();	466	ipl = interrupts_disable();
467	spinlock_lock(&THREAD->lock);	467	spinlock_lock(&THREAD->lock);
468	if (THREAD->timeout_pending) {	468	if (THREAD->timeout_pending) {
469	/* busy waiting for timeouts in progress */	469	/* busy waiting for timeouts in progress */
470	spinlock_unlock(&THREAD->lock);	470	spinlock_unlock(&THREAD->lock);
471	interrupts_restore(ipl);	471	interrupts_restore(ipl);
472	goto restart;	472	goto restart;
473	}	473	}
474		474
475	THREAD->state = Exiting;	475	THREAD->state = Exiting;
476	spinlock_unlock(&THREAD->lock);	476	spinlock_unlock(&THREAD->lock);
477	scheduler();	477	scheduler();
478		478
479	/* Not reached */	479	/* Not reached */
480	while (1)	480	while (1)
481	;	481	;
482	}	482	}
483		483
484		484
485	/** Thread sleep	485	/** Thread sleep
486	*	486	*
487	* Suspend execution of the current thread.	487	* Suspend execution of the current thread.
488	*	488	*
489	* @param sec Number of seconds to sleep.	489	* @param sec Number of seconds to sleep.
490	*	490	*
491	*/	491	*/
492	void thread_sleep(uint32_t sec)	492	void thread_sleep(uint32_t sec)
493	{	493	{
494	thread_usleep(sec * 1000000);	494	thread_usleep(sec * 1000000);
495	}	495	}
496		496
497	/** Wait for another thread to exit.	497	/** Wait for another thread to exit.
498	*	498	*
499	* @param t Thread to join on exit.	499	* @param t Thread to join on exit.
500	* @param usec Timeout in microseconds.	500	* @param usec Timeout in microseconds.

Subversion Repositories HelenOS

(root)/branches/tracing/kernel/generic/src/proc/thread.c @ 3042 – Rev 3424 → 3425