WebSVN – HelenOS – Diff – /branches/dynload/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;
    }
 
    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((uintptr_t) 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;
 
    /* 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 current task.
     */
    ipl = interrupts_disable();
    spinlock_lock(&task->lock);
    atomic_inc(&task->refcount);
    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 (atomic_predec(&TASK->lifecount) == 0) {
        /*
         * We are the last thread in the task that still has not exited.
         * With the exception of the moment the task was created, new
         * threads can only be created by threads of the same task.
         * We are safe to perform cleanup.
         */
        if (THREAD->flags & THREAD_FLAG_USPACE) {
            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;
}
 
 
/** Create new user task with 1 thread from image
 *
 * @param program_addr Address of program executable image.
 * @param name Program name.
 *
 * @return Initialized main thread of the task or NULL on error.
 */
thread_t *thread_create_program(void *program_addr, char *name)
{
    as_t *as;
    as_area_t *area;
    unsigned int rc;
    task_t *task;
    uspace_arg_t *kernel_uarg;
   
    kernel_uarg = (uspace_arg_t *) malloc(sizeof(uspace_arg_t), 0);
    if (kernel_uarg == NULL)
        return NULL;
   
    kernel_uarg->uspace_entry =
        (void *) ((elf_header_t *) program_addr)->e_entry;
    kernel_uarg->uspace_stack = (void *) USTACK_ADDRESS;
    kernel_uarg->uspace_thread_function = NULL;
    kernel_uarg->uspace_thread_arg = NULL;
    kernel_uarg->uspace_uarg = NULL;
 
    as = as_create(0);
    if (as == NULL) {
        free(kernel_uarg);
        return NULL;
    }
 
    rc = elf_load((elf_header_t *) program_addr, as);
    if (rc != EE_OK) {
        free(kernel_uarg);
        as_destroy(as);
        return NULL;
    }
   
    /*
     * Create the data as_area.
     */
    area = as_area_create(as,
        AS_AREA_READ | AS_AREA_WRITE | AS_AREA_CACHEABLE,
        LOADED_PROG_STACK_PAGES_NO * PAGE_SIZE, USTACK_ADDRESS,
        AS_AREA_ATTR_NONE, &anon_backend, NULL);
    if (area == NULL) {
        free(kernel_uarg);
        as_destroy(as);
        return NULL;
    }
   
    task = task_create(as, name);
    if (task == NULL) {
        free(kernel_uarg);
        as_destroy(as);
        return NULL;
    }
   
    /*
     * Create the main thread.
     */
    return thread_create(uinit, kernel_uarg, task, THREAD_FLAG_USPACE,
        "uinit", false);
}
 
 
/** 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);
 
        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 2927	Rev 3149
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
-		71
70	#include <console/klog.h>	72	#ifndef LOADED_PROG_STACK_PAGES_NO
-		73	#define LOADED_PROG_STACK_PAGES_NO 1
-		74	#endif
71		75
72		76
73	/** Thread states */	77	/** Thread states */
74	char *thread_states[] = {	78	char *thread_states[] = {
75	"Invalid",	79	"Invalid",
76	"Running",	80	"Running",
77	"Sleeping",	81	"Sleeping",
78	"Ready",	82	"Ready",
79	"Entering",	83	"Entering",
80	"Exiting",	84	"Exiting",
81	"Lingering"	85	"Lingering"
82	};	86	};
83		87
84	/** Lock protecting the threads_tree AVL tree.	88	/** Lock protecting the threads_tree AVL tree.
85	*	89	*
86	* For locking rules, see declaration thereof.	90	* For locking rules, see declaration thereof.
87	*/	91	*/
88	SPINLOCK_INITIALIZE(threads_lock);	92	SPINLOCK_INITIALIZE(threads_lock);
89		93
90	/** AVL tree of all threads.	94	/** AVL tree of all threads.
91	*	95	*
92	* 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
93	* exist as long as the threads_lock is held.	97	* exist as long as the threads_lock is held.
94	*/	98	*/
95	avltree_t threads_tree;	99	avltree_t threads_tree;
96		100
97	SPINLOCK_INITIALIZE(tidlock);	101	SPINLOCK_INITIALIZE(tidlock);
98	thread_id_t last_tid = 0;	102	thread_id_t last_tid = 0;
99		103
100	static slab_cache_t *thread_slab;	104	static slab_cache_t *thread_slab;
101	#ifdef ARCH_HAS_FPU	105	#ifdef ARCH_HAS_FPU
102	slab_cache_t *fpu_context_slab;	106	slab_cache_t *fpu_context_slab;
103	#endif	107	#endif
104		108
105	/** Thread wrapper.	109	/** Thread wrapper.
106	*	110	*
107	* 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
108	* thread_exit() when its implementing function returns.	112	* thread_exit() when its implementing function returns.
109	*	113	*
110	* interrupts_disable() is assumed.	114	* interrupts_disable() is assumed.
111	*	115	*
112	*/	116	*/
113	static void cushion(void)	117	static void cushion(void)
114	{	118	{
115	void (f)(void ) = THREAD->thread_code;	119	void (f)(void ) = THREAD->thread_code;
116	void *arg = THREAD->thread_arg;	120	void *arg = THREAD->thread_arg;
117	THREAD->last_cycle = get_cycle();	121	THREAD->last_cycle = get_cycle();
118		122
119	/* This is where each thread wakes up after its creation */	123	/* This is where each thread wakes up after its creation */
120	spinlock_unlock(&THREAD->lock);	124	spinlock_unlock(&THREAD->lock);
121	interrupts_enable();	125	interrupts_enable();
122		126
123	f(arg);	127	f(arg);
124		128
125	/* Accumulate accounting to the task */	129	/* Accumulate accounting to the task */
126	ipl_t ipl = interrupts_disable();	130	ipl_t ipl = interrupts_disable();
127		131
128	spinlock_lock(&THREAD->lock);	132	spinlock_lock(&THREAD->lock);
129	if (!THREAD->uncounted) {	133	if (!THREAD->uncounted) {
130	thread_update_accounting();	134	thread_update_accounting();
131	uint64_t cycles = THREAD->cycles;	135	uint64_t cycles = THREAD->cycles;
132	THREAD->cycles = 0;	136	THREAD->cycles = 0;
133	spinlock_unlock(&THREAD->lock);	137	spinlock_unlock(&THREAD->lock);
134		138
135	spinlock_lock(&TASK->lock);	139	spinlock_lock(&TASK->lock);
136	TASK->cycles += cycles;	140	TASK->cycles += cycles;
137	spinlock_unlock(&TASK->lock);	141	spinlock_unlock(&TASK->lock);
138	} else	142	} else
139	spinlock_unlock(&THREAD->lock);	143	spinlock_unlock(&THREAD->lock);
140		144
141	interrupts_restore(ipl);	145	interrupts_restore(ipl);
142		146
143	thread_exit();	147	thread_exit();
144	/* not reached */	148	/* not reached */
145	}	149	}
146		150
147	/** Initialization and allocation for thread_t structure */	151	/** Initialization and allocation for thread_t structure */
148	static int thr_constructor(void *obj, int kmflags)	152	static int thr_constructor(void *obj, int kmflags)
149	{	153	{
150	thread_t t = (thread_t ) obj;	154	thread_t t = (thread_t ) obj;
151		155
152	spinlock_initialize(&t->lock, "thread_t_lock");	156	spinlock_initialize(&t->lock, "thread_t_lock");
153	link_initialize(&t->rq_link);	157	link_initialize(&t->rq_link);
154	link_initialize(&t->wq_link);	158	link_initialize(&t->wq_link);
155	link_initialize(&t->th_link);	159	link_initialize(&t->th_link);
156		160
157	/* call the architecture-specific part of the constructor */	161	/* call the architecture-specific part of the constructor */
158	thr_constructor_arch(t);	162	thr_constructor_arch(t);
159		163
160	#ifdef ARCH_HAS_FPU	164	#ifdef ARCH_HAS_FPU
161	#ifdef CONFIG_FPU_LAZY	165	#ifdef CONFIG_FPU_LAZY
162	t->saved_fpu_context = NULL;	166	t->saved_fpu_context = NULL;
163	#else	167	#else
164	t->saved_fpu_context = slab_alloc(fpu_context_slab, kmflags);	168	t->saved_fpu_context = slab_alloc(fpu_context_slab, kmflags);
165	if (!t->saved_fpu_context)	169	if (!t->saved_fpu_context)
166	return -1;	170	return -1;
167	#endif	171	#endif
168	#endif	172	#endif
169		173
170	t->kstack = (uint8_t *) frame_alloc(STACK_FRAMES, FRAME_KA \| kmflags);	174	t->kstack = (uint8_t *) frame_alloc(STACK_FRAMES, FRAME_KA \| kmflags);
171	if (!t->kstack) {	175	if (!t->kstack) {
172	#ifdef ARCH_HAS_FPU	176	#ifdef ARCH_HAS_FPU
173	if (t->saved_fpu_context)	177	if (t->saved_fpu_context)
174	slab_free(fpu_context_slab, t->saved_fpu_context);	178	slab_free(fpu_context_slab, t->saved_fpu_context);
175	#endif	179	#endif
176	return -1;	180	return -1;
177	}	181	}
178		182
179	return 0;	183	return 0;
180	}	184	}
181		185
182	/** Destruction of thread_t object */	186	/** Destruction of thread_t object */
183	static int thr_destructor(void *obj)	187	static int thr_destructor(void *obj)
184	{	188	{
185	thread_t t = (thread_t ) obj;	189	thread_t t = (thread_t ) obj;
186		190
187	/* call the architecture-specific part of the destructor */	191	/* call the architecture-specific part of the destructor */
188	thr_destructor_arch(t);	192	thr_destructor_arch(t);
189		193
190	frame_free(KA2PA(t->kstack));	194	frame_free(KA2PA(t->kstack));
191	#ifdef ARCH_HAS_FPU	195	#ifdef ARCH_HAS_FPU
192	if (t->saved_fpu_context)	196	if (t->saved_fpu_context)
193	slab_free(fpu_context_slab, t->saved_fpu_context);	197	slab_free(fpu_context_slab, t->saved_fpu_context);
194	#endif	198	#endif
195	return 1; /* One page freed */	199	return 1; /* One page freed */
196	}	200	}
197		201
198	/** Initialize threads	202	/** Initialize threads
199	*	203	*
200	* Initialize kernel threads support.	204	* Initialize kernel threads support.
201	*	205	*
202	*/	206	*/
203	void thread_init(void)	207	void thread_init(void)
204	{	208	{
205	THREAD = NULL;	209	THREAD = NULL;
206	atomic_set(&nrdy,0);	210	atomic_set(&nrdy,0);
207	thread_slab = slab_cache_create("thread_slab", sizeof(thread_t), 0,	211	thread_slab = slab_cache_create("thread_slab", sizeof(thread_t), 0,
208	thr_constructor, thr_destructor, 0);	212	thr_constructor, thr_destructor, 0);
209		213
210	#ifdef ARCH_HAS_FPU	214	#ifdef ARCH_HAS_FPU
211	fpu_context_slab = slab_cache_create("fpu_slab", sizeof(fpu_context_t),	215	fpu_context_slab = slab_cache_create("fpu_slab", sizeof(fpu_context_t),
212	FPU_CONTEXT_ALIGN, NULL, NULL, 0);	216	FPU_CONTEXT_ALIGN, NULL, NULL, 0);
213	#endif	217	#endif
214		218
215	avltree_create(&threads_tree);	219	avltree_create(&threads_tree);
216	}	220	}
217		221
218	/** Make thread ready	222	/** Make thread ready
219	*	223	*
220	* Switch thread t to the ready state.	224	* Switch thread t to the ready state.
221	*	225	*
222	* @param t Thread to make ready.	226	* @param t Thread to make ready.
223	*	227	*
224	*/	228	*/
225	void thread_ready(thread_t *t)	229	void thread_ready(thread_t *t)
226	{	230	{
227	cpu_t *cpu;	231	cpu_t *cpu;
228	runq_t *r;	232	runq_t *r;
229	ipl_t ipl;	233	ipl_t ipl;
230	int i, avg;	234	int i, avg;
231		235
232	ipl = interrupts_disable();	236	ipl = interrupts_disable();
233		237
234	spinlock_lock(&t->lock);	238	spinlock_lock(&t->lock);
235		239
236	ASSERT(!(t->state == Ready));	240	ASSERT(!(t->state == Ready));
237		241
238	i = (t->priority < RQ_COUNT - 1) ? ++t->priority : t->priority;	242	i = (t->priority < RQ_COUNT - 1) ? ++t->priority : t->priority;
239		243
240	cpu = CPU;	244	cpu = CPU;
241	if (t->flags & THREAD_FLAG_WIRED) {	245	if (t->flags & THREAD_FLAG_WIRED) {
242	ASSERT(t->cpu != NULL);	246	ASSERT(t->cpu != NULL);
243	cpu = t->cpu;	247	cpu = t->cpu;
244	}	248	}
245	t->state = Ready;	249	t->state = Ready;
246	spinlock_unlock(&t->lock);	250	spinlock_unlock(&t->lock);
247		251
248	/*	252	/*
249	* Append t to respective ready queue on respective processor.	253	* Append t to respective ready queue on respective processor.
250	*/	254	*/
251	r = &cpu->rq[i];	255	r = &cpu->rq[i];
252	spinlock_lock(&r->lock);	256	spinlock_lock(&r->lock);
253	list_append(&t->rq_link, &r->rq_head);	257	list_append(&t->rq_link, &r->rq_head);
254	r->n++;	258	r->n++;
255	spinlock_unlock(&r->lock);	259	spinlock_unlock(&r->lock);
256		260
257	atomic_inc(&nrdy);	261	atomic_inc(&nrdy);
258	avg = atomic_get(&nrdy) / config.cpu_active;	262	avg = atomic_get(&nrdy) / config.cpu_active;
259	atomic_inc(&cpu->nrdy);	263	atomic_inc(&cpu->nrdy);
260		264
261	interrupts_restore(ipl);	265	interrupts_restore(ipl);
262	}	266	}
263		267
264	/** Create new thread	268	/** Create new thread
265	*	269	*
266	* Create a new thread.	270	* Create a new thread.
267	*	271	*
268	* @param func Thread's implementing function.	272	* @param func Thread's implementing function.
269	* @param arg Thread's implementing function argument.	273	* @param arg Thread's implementing function argument.
270	* @param task Task to which the thread belongs. The caller must	274	* @param task Task to which the thread belongs. The caller must
271	* guarantee that the task won't cease to exist during the	275	* guarantee that the task won't cease to exist during the
272	* call. The task's lock may not be held.	276	* call. The task's lock may not be held.
273	* @param flags Thread flags.	277	* @param flags Thread flags.
274	* @param name Symbolic name.	278	* @param name Symbolic name.
275	* @param uncounted Thread's accounting doesn't affect accumulated task	279	* @param uncounted Thread's accounting doesn't affect accumulated task
276	* accounting.	280	* accounting.
277	*	281	*
278	* @return New thread's structure on success, NULL on failure.	282	* @return New thread's structure on success, NULL on failure.
279	*	283	*
280	*/	284	*/
281	thread_t thread_create(void ( func)(void ), void arg, task_t *task,	285	thread_t thread_create(void ( func)(void ), void arg, task_t *task,
282	int flags, char *name, bool uncounted)	286	int flags, char *name, bool uncounted)
283	{	287	{
284	thread_t *t;	288	thread_t *t;
285	ipl_t ipl;	289	ipl_t ipl;
286		290
287	t = (thread_t *) slab_alloc(thread_slab, 0);	291	t = (thread_t *) slab_alloc(thread_slab, 0);
288	if (!t)	292	if (!t)
289	return NULL;	293	return NULL;
290		294
291	/* Not needed, but good for debugging */	295	/* Not needed, but good for debugging */
292	memsetb((uintptr_t) t->kstack, THREAD_STACK_SIZE * 1 << STACK_FRAMES,	296	memsetb((uintptr_t) t->kstack, THREAD_STACK_SIZE * 1 << STACK_FRAMES,
293	0);	297	0);
294		298
295	ipl = interrupts_disable();	299	ipl = interrupts_disable();
296	spinlock_lock(&tidlock);	300	spinlock_lock(&tidlock);
297	t->tid = ++last_tid;	301	t->tid = ++last_tid;
298	spinlock_unlock(&tidlock);	302	spinlock_unlock(&tidlock);
299	interrupts_restore(ipl);	303	interrupts_restore(ipl);
300		304
301	context_save(&t->saved_context);	305	context_save(&t->saved_context);
302	context_set(&t->saved_context, FADDR(cushion), (uintptr_t) t->kstack,	306	context_set(&t->saved_context, FADDR(cushion), (uintptr_t) t->kstack,
303	THREAD_STACK_SIZE);	307	THREAD_STACK_SIZE);
304		308
305	the_initialize((the_t *) t->kstack);	309	the_initialize((the_t *) t->kstack);
306		310
307	ipl = interrupts_disable();	311	ipl = interrupts_disable();
308	t->saved_context.ipl = interrupts_read();	312	t->saved_context.ipl = interrupts_read();
309	interrupts_restore(ipl);	313	interrupts_restore(ipl);
310		314
311	memcpy(t->name, name, THREAD_NAME_BUFLEN);	315	memcpy(t->name, name, THREAD_NAME_BUFLEN);
312		316
313	t->thread_code = func;	317	t->thread_code = func;
314	t->thread_arg = arg;	318	t->thread_arg = arg;
315	t->ticks = -1;	319	t->ticks = -1;
316	t->cycles = 0;	320	t->cycles = 0;
317	t->uncounted = uncounted;	321	t->uncounted = uncounted;
318	t->priority = -1; /* start in rq[0] */	322	t->priority = -1; /* start in rq[0] */
319	t->cpu = NULL;	323	t->cpu = NULL;
320	t->flags = flags;	324	t->flags = flags;
321	t->state = Entering;	325	t->state = Entering;
322	t->call_me = NULL;	326	t->call_me = NULL;
323	t->call_me_with = NULL;	327	t->call_me_with = NULL;
324		328
325	timeout_initialize(&t->sleep_timeout);	329	timeout_initialize(&t->sleep_timeout);
326	t->sleep_interruptible = false;	330	t->sleep_interruptible = false;
327	t->sleep_queue = NULL;	331	t->sleep_queue = NULL;
328	t->timeout_pending = 0;	332	t->timeout_pending = 0;
329		333
330	t->in_copy_from_uspace = false;	334	t->in_copy_from_uspace = false;
331	t->in_copy_to_uspace = false;	335	t->in_copy_to_uspace = false;
332		336
333	t->interrupted = false;	337	t->interrupted = false;
334	t->detached = false;	338	t->detached = false;
335	waitq_initialize(&t->join_wq);	339	waitq_initialize(&t->join_wq);
336		340
337	t->rwlock_holder_type = RWLOCK_NONE;	341	t->rwlock_holder_type = RWLOCK_NONE;
338		342
339	t->task = task;	343	t->task = task;
340		344
341	t->fpu_context_exists = 0;	345	t->fpu_context_exists = 0;
342	t->fpu_context_engaged = 0;	346	t->fpu_context_engaged = 0;
343		347
344	avltree_node_initialize(&t->threads_tree_node);	348	avltree_node_initialize(&t->threads_tree_node);
345	t->threads_tree_node.key = (uintptr_t) t;	349	t->threads_tree_node.key = (uintptr_t) t;
346		350
347	/* might depend on previous initialization */	351	/* might depend on previous initialization */
348	thread_create_arch(t);	352	thread_create_arch(t);
349		353
350	if (!(flags & THREAD_FLAG_NOATTACH))	354	if (!(flags & THREAD_FLAG_NOATTACH))
351	thread_attach(t, task);	355	thread_attach(t, task);
352		356
353	return t;	357	return t;
354	}	358	}
355		359
356	/** Destroy thread memory structure	360	/** Destroy thread memory structure
357	*	361	*
358	* Detach thread from all queues, cpus etc. and destroy it.	362	* Detach thread from all queues, cpus etc. and destroy it.
359	*	363	*
360	* Assume thread->lock is held!!	364	* Assume thread->lock is held!!
361	*/	365	*/
362	void thread_destroy(thread_t *t)	366	void thread_destroy(thread_t *t)
363	{	367	{
364	ASSERT(t->state == Exiting \|\| t->state == Lingering);	368	ASSERT(t->state == Exiting \|\| t->state == Lingering);
365	ASSERT(t->task);	369	ASSERT(t->task);
366	ASSERT(t->cpu);	370	ASSERT(t->cpu);
367		371
368	spinlock_lock(&t->cpu->lock);	372	spinlock_lock(&t->cpu->lock);
369	if (t->cpu->fpu_owner == t)	373	if (t->cpu->fpu_owner == t)
370	t->cpu->fpu_owner = NULL;	374	t->cpu->fpu_owner = NULL;
371	spinlock_unlock(&t->cpu->lock);	375	spinlock_unlock(&t->cpu->lock);
372		376
373	spinlock_unlock(&t->lock);	377	spinlock_unlock(&t->lock);
374		378
375	spinlock_lock(&threads_lock);	379	spinlock_lock(&threads_lock);
376	avltree_delete(&threads_tree, &t->threads_tree_node);	380	avltree_delete(&threads_tree, &t->threads_tree_node);
377	spinlock_unlock(&threads_lock);	381	spinlock_unlock(&threads_lock);
378		382
379	/*	383	/*
380	* Detach from the containing task.	384	* Detach from the containing task.
381	*/	385	*/
382	spinlock_lock(&t->task->lock);	386	spinlock_lock(&t->task->lock);
383	list_remove(&t->th_link);	387	list_remove(&t->th_link);
384	spinlock_unlock(&t->task->lock);	388	spinlock_unlock(&t->task->lock);
385		389
386	/*	390	/*
387	* t is guaranteed to be the very last thread of its task.	391	* t is guaranteed to be the very last thread of its task.
388	* It is safe to destroy the task.	392	* It is safe to destroy the task.
389	*/	393	*/
390	if (atomic_predec(&t->task->refcount) == 0)	394	if (atomic_predec(&t->task->refcount) == 0)
391	task_destroy(t->task);	395	task_destroy(t->task);
392		396
393	slab_free(thread_slab, t);	397	slab_free(thread_slab, t);
394	}	398	}
395		399
396	/** Make the thread visible to the system.	400	/** Make the thread visible to the system.
397	*	401	*
398	* Attach the thread structure to the current task and make it visible in the	402	* Attach the thread structure to the current task and make it visible in the
399	* threads_tree.	403	* threads_tree.
400	*	404	*
401	* @param t Thread to be attached to the task.	405	* @param t Thread to be attached to the task.
402	* @param task Task to which the thread is to be attached.	406	* @param task Task to which the thread is to be attached.
403	*/	407	*/
404	void thread_attach(thread_t t, task_t task)	408	void thread_attach(thread_t t, task_t task)
405	{	409	{
406	ipl_t ipl;	410	ipl_t ipl;
407		411
408	/*	412	/*
409	* Attach to the current task.	413	* Attach to the current task.
410	*/	414	*/
411	ipl = interrupts_disable();	415	ipl = interrupts_disable();
412	spinlock_lock(&task->lock);	416	spinlock_lock(&task->lock);
413	atomic_inc(&task->refcount);	417	atomic_inc(&task->refcount);
414	atomic_inc(&task->lifecount);	418	atomic_inc(&task->lifecount);
415	list_append(&t->th_link, &task->th_head);	419	list_append(&t->th_link, &task->th_head);
416	spinlock_unlock(&task->lock);	420	spinlock_unlock(&task->lock);
417		421
418	/*	422	/*
419	* Register this thread in the system-wide list.	423	* Register this thread in the system-wide list.
420	*/	424	*/
421	spinlock_lock(&threads_lock);	425	spinlock_lock(&threads_lock);
422	avltree_insert(&threads_tree, &t->threads_tree_node);	426	avltree_insert(&threads_tree, &t->threads_tree_node);
423	spinlock_unlock(&threads_lock);	427	spinlock_unlock(&threads_lock);
424		428
425	interrupts_restore(ipl);	429	interrupts_restore(ipl);
426	}	430	}
427		431
428	/** Terminate thread.	432	/** Terminate thread.
429	*	433	*
430	* End current thread execution and switch it to the exiting state. All pending	434	* End current thread execution and switch it to the exiting state. All pending
431	* timeouts are executed.	435	* timeouts are executed.
432	*/	436	*/
433	void thread_exit(void)	437	void thread_exit(void)
434	{	438	{
435	ipl_t ipl;	439	ipl_t ipl;
436		440
437	if (atomic_predec(&TASK->lifecount) == 0) {	441	if (atomic_predec(&TASK->lifecount) == 0) {
438	/*	442	/*
439	* We are the last thread in the task that still has not exited.	443	* We are the last thread in the task that still has not exited.
440	* With the exception of the moment the task was created, new	444	* With the exception of the moment the task was created, new
441	* threads can only be created by threads of the same task.	445	* threads can only be created by threads of the same task.
442	* We are safe to perform cleanup.	446	* We are safe to perform cleanup.
443	*/	447	*/
444	if (THREAD->flags & THREAD_FLAG_USPACE) {	448	if (THREAD->flags & THREAD_FLAG_USPACE) {
445	ipc_cleanup();	449	ipc_cleanup();
446	futex_cleanup();	450	futex_cleanup();
447	klog_printf("Cleanup of task %llu completed.",	451	LOG("Cleanup of task %" PRIu64" completed.", TASK->taskid);
448	TASK->taskid);	-
449	}	452	}
450	}	453	}
451		454
452	restart:	455	restart:
453	ipl = interrupts_disable();	456	ipl = interrupts_disable();
454	spinlock_lock(&THREAD->lock);	457	spinlock_lock(&THREAD->lock);
455	if (THREAD->timeout_pending) {	458	if (THREAD->timeout_pending) {
456	/* busy waiting for timeouts in progress */	459	/* busy waiting for timeouts in progress */
457	spinlock_unlock(&THREAD->lock);	460	spinlock_unlock(&THREAD->lock);
458	interrupts_restore(ipl);	461	interrupts_restore(ipl);
459	goto restart;	462	goto restart;
460	}	463	}
461		464
462	THREAD->state = Exiting;	465	THREAD->state = Exiting;
463	spinlock_unlock(&THREAD->lock);	466	spinlock_unlock(&THREAD->lock);
464	scheduler();	467	scheduler();
465		468
466	/* Not reached */	469	/* Not reached */
467	while (1)	470	while (1)
468	;	471	;
469	}	472	}
470		473
471		474
472	/** Thread sleep	475	/** Thread sleep
473	*	476	*
474	* Suspend execution of the current thread.	477	* Suspend execution of the current thread.
475	*	478	*
476	* @param sec Number of seconds to sleep.	479	* @param sec Number of seconds to sleep.
477	*	480	*
478	*/	481	*/
479	void thread_sleep(uint32_t sec)	482	void thread_sleep(uint32_t sec)
480	{	483	{
481	thread_usleep(sec * 1000000);	484	thread_usleep(sec * 1000000);
482	}	485	}
483		486
484	/** Wait for another thread to exit.	487	/** Wait for another thread to exit.
485	*	488	*
486	* @param t Thread to join on exit.	489	* @param t Thread to join on exit.
487	* @param usec Timeout in microseconds.	490	* @param usec Timeout in microseconds.
488	* @param flags Mode of operation.	491	* @param flags Mode of operation.
489	*	492	*
490	* @return An error code from errno.h or an error code from synch.h.	493	* @return An error code from errno.h or an error code from synch.h.
491	*/	494	*/
492	int thread_join_timeout(thread_t *t, uint32_t usec, int flags)	495	int thread_join_timeout(thread_t *t, uint32_t usec, int flags)
493	{	496	{
494	ipl_t ipl;	497	ipl_t ipl;
495	int rc;	498	int rc;
496		499

Subversion Repositories HelenOS

(root)/branches/dynload/kernel/generic/src/proc/thread.c – Rev 2927 → 3149