WebSVN – HelenOS – Diff – /trunk/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.
 */
 
/**
 * @file    thread.c
 * @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.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/btree.h>
#include <adt/list.h>
#include <typedefs.h>
#include <time/clock.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>
 
 
/** Thread states */
char *thread_states[] = {
    "Invalid",
    "Running",
    "Sleeping",
    "Ready",
    "Entering",
    "Exiting",
    "Undead"
};
 
/** Lock protecting threads_head list. For locking rules, see declaration thereof. */
SPINLOCK_INITIALIZE(threads_lock);
btree_t threads_btree;          /**< B+tree of all threads. */
 
SPINLOCK_INITIALIZE(tidlock);
__u32 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;
 
    /* this is where each thread wakes up after its creation */
    spinlock_unlock(&THREAD->lock);
    interrupts_enable();
 
    f(arg);
    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;
    pfn_t pfn;
    int status;
 
    spinlock_initialize(&t->lock, "thread_t_lock");
    link_initialize(&t->rq_link);
    link_initialize(&t->wq_link);
    link_initialize(&t->th_link);
   
#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  
 
    pfn = frame_alloc_rc(STACK_FRAMES, FRAME_KA | kmflags,&status);
    if (status) {
#ifdef ARCH_HAS_FPU
        if (t->saved_fpu_context)
            slab_free(fpu_context_slab,t->saved_fpu_context);
#endif
        return -1;
    }
    t->kstack = (__u8 *)PA2KA(PFN2ADDR(pfn));
 
    return 0;
}
 
/** Destruction of thread_t object */
static int thr_destructor(void *obj)
{
    thread_t *t = (thread_t *)obj;
 
    frame_free(ADDR2PFN(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
 
    btree_create(&threads_btree);
}
 
/** 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 & X_WIRED) {
        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);
}
 
/** 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)
{
    bool destroy_task = false; 
 
    ASSERT(t->state == Exiting || t->state == Undead);
    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);
    btree_remove(&threads_btree, (btree_key_t) ((__address ) t), NULL);
    spinlock_unlock(&threads_lock);
 
    /*
     * Detach from the containing task.
     */
    spinlock_lock(&t->task->lock);
    list_remove(&t->th_link);
    if (--t->task->refcount == 0) {
        t->task->accept_new_threads = false;
        destroy_task = true;
    }
    spinlock_unlock(&t->task->lock);   
   
    if (destroy_task)
        task_destroy(t->task);
   
    slab_free(thread_slab, t);
}
 
/** 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.
 * @param flags Thread flags.
 * @param name  Symbolic name.
 *
 * @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)
{
    thread_t *t;
    ipl_t ipl;
   
    t = (thread_t *) slab_alloc(thread_slab, 0);
    if (!t)
        return NULL;
 
    thread_create_arch(t);
   
    /* Not needed, but good for debugging */
    memsetb((__address)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), (__address) 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->priority = -1;       /* start in rq[0] */
    t->cpu = NULL;
    t->flags = 0;
    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;
   
    /*
     * Attach to the containing task.
     */
    spinlock_lock(&task->lock);
    if (!task->accept_new_threads) {
        spinlock_unlock(&task->lock);
        slab_free(thread_slab, t);
        return NULL;
    }
    list_append(&t->th_link, &task->th_head);
    task->refcount++;
    spinlock_unlock(&task->lock);
 
    /*
     * Register this thread in the system-wide list.
     */
    ipl = interrupts_disable();
    spinlock_lock(&threads_lock);
    btree_insert(&threads_btree, (btree_key_t) ((__address) t), (void *) t, NULL);
    spinlock_unlock(&threads_lock);
   
    interrupts_restore(ipl);
   
    return t;
}
 
/** Make thread exiting
 *
 * End current thread execution and switch it to the exiting
 * state. All pending timeouts are executed.
 *
 */
void thread_exit(void)
{
    ipl_t ipl;
 
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();
}
 
 
/** Thread sleep
 *
 * Suspend execution of the current thread.
 *
 * @param sec Number of seconds to sleep.
 *
 */
void thread_sleep(__u32 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, __u32 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);
   
    (void) waitq_sleep_prepare(&t->join_wq);
    spinlock_unlock(&t->lock);
   
    rc = waitq_sleep_timeout_unsafe(&t->join_wq, usec, flags);
   
    waitq_sleep_finish(&t->join_wq, rc, ipl);
   
    return rc; 
}
 
/** Detach thread.
 *
 * Mark the thread as detached, if the thread is already in the Undead state,
 * deallocate its resources.
 *
 * @param t Thread to be detached.
 */
void thread_detach(thread_t *t)
{
    ipl_t ipl;
 
    /*
     * Since the thread is expected to not be already detached,
     * pointer to it must be still valid.
     */
   
    ipl = interrupts_disable();
    spinlock_lock(&t->lock);
    ASSERT(!t->detached);
    if (t->state == Undead) {
        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(__u32 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);
}
 
/** Print list of threads debug info */
void thread_print_list(void)
{
    link_t *cur;
    ipl_t ipl;
   
    /* Messing with thread structures, avoid deadlock */
    ipl = interrupts_disable();
    spinlock_lock(&threads_lock);
 
    for (cur = threads_btree.leaf_head.next; cur != &threads_btree.leaf_head; cur = cur->next) {
        btree_node_t *node;
        int i;
 
        node = list_get_instance(cur, btree_node_t, leaf_link);
        for (i = 0; i < node->keys; i++) {
            thread_t *t;
       
            t = (thread_t *) node->value[i];
            printf("%s: address=%#zX, tid=%zd, state=%s, task=%#zX, code=%#zX, stack=%#zX, cpu=",
                t->name, t, t->tid, thread_states[t->state], t->task, t->thread_code, t->kstack);
            if (t->cpu)
                printf("cpu%zd", t->cpu->id);
            else
                printf("none");
            if (t->state == Sleeping) {
                printf(", kst=%#zX", t->kstack);
                printf(", wq=%#zX", t->sleep_queue);
            }
            printf("\n");
        }
    }
 
    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)
{
    btree_node_t *leaf;
   
    return btree_search(&threads_btree, (btree_key_t) ((__address) t), &leaf) != NULL;
}
 
/** Process syscall to create new thread.
 *
 */
__native sys_thread_create(uspace_arg_t *uspace_uarg, char *uspace_name)
{
    thread_t *t;
    char namebuf[THREAD_NAME_BUFLEN];
    uspace_arg_t *kernel_uarg;
    __u32 tid;
    int rc;
 
    rc = copy_from_uspace(namebuf, uspace_name, THREAD_NAME_BUFLEN);
    if (rc != 0)
        return (__native) rc;
 
    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 (__native) rc;
    }
 
    if ((t = thread_create(uinit, kernel_uarg, TASK, 0, namebuf))) {
        tid = t->tid;
        thread_ready(t);
        return (__native) tid;
    } else {
        free(kernel_uarg);
    }
 
    return (__native) ENOMEM;
}
 
/** Process syscall to terminate thread.
 *
 */
__native sys_thread_exit(int uspace_status)
{
    thread_exit();
    /* Unreachable */
    return 0;
}
 

Rev 1579	Rev 1581
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	/**	29	/**
30	* @file thread.c	30	* @file thread.c
31	* @brief Thread management functions.	31	* @brief Thread management functions.
32	*/	32	*/
33		33
34	#include <proc/scheduler.h>	34	#include <proc/scheduler.h>
35	#include <proc/thread.h>	35	#include <proc/thread.h>
36	#include <proc/task.h>	36	#include <proc/task.h>
37	#include <proc/uarg.h>	37	#include <proc/uarg.h>
38	#include <mm/frame.h>	38	#include <mm/frame.h>
39	#include <mm/page.h>	39	#include <mm/page.h>
40	#include <arch/asm.h>	40	#include <arch/asm.h>
41	#include <arch.h>	41	#include <arch.h>
42	#include <synch/synch.h>	42	#include <synch/synch.h>
43	#include <synch/spinlock.h>	43	#include <synch/spinlock.h>
44	#include <synch/waitq.h>	44	#include <synch/waitq.h>
45	#include <synch/rwlock.h>	45	#include <synch/rwlock.h>
46	#include <cpu.h>	46	#include <cpu.h>
47	#include <func.h>	47	#include <func.h>
48	#include <context.h>	48	#include <context.h>
49	#include <adt/btree.h>	49	#include <adt/btree.h>
50	#include <adt/list.h>	50	#include <adt/list.h>
51	#include <typedefs.h>	51	#include <typedefs.h>
52	#include <time/clock.h>	52	#include <time/clock.h>
53	#include <config.h>	53	#include <config.h>
54	#include <arch/interrupt.h>	54	#include <arch/interrupt.h>
55	#include <smp/ipi.h>	55	#include <smp/ipi.h>
56	#include <arch/faddr.h>	56	#include <arch/faddr.h>
57	#include <atomic.h>	57	#include <atomic.h>
58	#include <memstr.h>	58	#include <memstr.h>
59	#include <print.h>	59	#include <print.h>
60	#include <mm/slab.h>	60	#include <mm/slab.h>
61	#include <debug.h>	61	#include <debug.h>
62	#include <main/uinit.h>	62	#include <main/uinit.h>
63	#include <syscall/copy.h>	63	#include <syscall/copy.h>
64	#include <errno.h>	64	#include <errno.h>
65		65
66		66
67	/** Thread states */	67	/** Thread states */
68	char *thread_states[] = {	68	char *thread_states[] = {
69	"Invalid",	69	"Invalid",
70	"Running",	70	"Running",
71	"Sleeping",	71	"Sleeping",
72	"Ready",	72	"Ready",
73	"Entering",	73	"Entering",
74	"Exiting",	74	"Exiting",
75	"Undead"	75	"Undead"
76	};	76	};
77		77
78	/** Lock protecting threads_head list. For locking rules, see declaration thereof. */	78	/** Lock protecting threads_head list. For locking rules, see declaration thereof. */
79	SPINLOCK_INITIALIZE(threads_lock);	79	SPINLOCK_INITIALIZE(threads_lock);
80	btree_t threads_btree; /*< B+tree of all threads. /	80	btree_t threads_btree; /*< B+tree of all threads. /
81		81
82	SPINLOCK_INITIALIZE(tidlock);	82	SPINLOCK_INITIALIZE(tidlock);
83	__u32 last_tid = 0;	83	__u32 last_tid = 0;
84		84
85	static slab_cache_t *thread_slab;	85	static slab_cache_t *thread_slab;
86	#ifdef ARCH_HAS_FPU	86	#ifdef ARCH_HAS_FPU
87	slab_cache_t *fpu_context_slab;	87	slab_cache_t *fpu_context_slab;
88	#endif	88	#endif
89		89
90	/** Thread wrapper	90	/** Thread wrapper
91	*	91	*
92	* This wrapper is provided to ensure that every thread	92	* This wrapper is provided to ensure that every thread
93	* makes a call to thread_exit() when its implementing	93	* makes a call to thread_exit() when its implementing
94	* function returns.	94	* function returns.
95	*	95	*
96	* interrupts_disable() is assumed.	96	* interrupts_disable() is assumed.
97	*	97	*
98	*/	98	*/
99	static void cushion(void)	99	static void cushion(void)
100	{	100	{
101	void (f)(void ) = THREAD->thread_code;	101	void (f)(void ) = THREAD->thread_code;
102	void *arg = THREAD->thread_arg;	102	void *arg = THREAD->thread_arg;
103		103
104	/* this is where each thread wakes up after its creation */	104	/* this is where each thread wakes up after its creation */
105	spinlock_unlock(&THREAD->lock);	105	spinlock_unlock(&THREAD->lock);
106	interrupts_enable();	106	interrupts_enable();
107		107
108	f(arg);	108	f(arg);
109	thread_exit();	109	thread_exit();
110	/* not reached */	110	/* not reached */
111	}	111	}
112		112
113	/** Initialization and allocation for thread_t structure */	113	/** Initialization and allocation for thread_t structure */
114	static int thr_constructor(void *obj, int kmflags)	114	static int thr_constructor(void *obj, int kmflags)
115	{	115	{
116	thread_t t = (thread_t )obj;	116	thread_t t = (thread_t )obj;
117	pfn_t pfn;	117	pfn_t pfn;
118	int status;	118	int status;
119		119
120	spinlock_initialize(&t->lock, "thread_t_lock");	120	spinlock_initialize(&t->lock, "thread_t_lock");
121	link_initialize(&t->rq_link);	121	link_initialize(&t->rq_link);
122	link_initialize(&t->wq_link);	122	link_initialize(&t->wq_link);
123	link_initialize(&t->th_link);	123	link_initialize(&t->th_link);
124		124
125	#ifdef ARCH_HAS_FPU	125	#ifdef ARCH_HAS_FPU
126	# ifdef CONFIG_FPU_LAZY	126	# ifdef CONFIG_FPU_LAZY
127	t->saved_fpu_context = NULL;	127	t->saved_fpu_context = NULL;
128	# else	128	# else
129	t->saved_fpu_context = slab_alloc(fpu_context_slab,kmflags);	129	t->saved_fpu_context = slab_alloc(fpu_context_slab,kmflags);
130	if (!t->saved_fpu_context)	130	if (!t->saved_fpu_context)
131	return -1;	131	return -1;
132	# endif	132	# endif
133	#endif	133	#endif
134		134
135	pfn = frame_alloc_rc(STACK_FRAMES, FRAME_KA \| kmflags,&status);	135	pfn = frame_alloc_rc(STACK_FRAMES, FRAME_KA \| kmflags,&status);
136	if (status) {	136	if (status) {
137	#ifdef ARCH_HAS_FPU	137	#ifdef ARCH_HAS_FPU
138	if (t->saved_fpu_context)	138	if (t->saved_fpu_context)
139	slab_free(fpu_context_slab,t->saved_fpu_context);	139	slab_free(fpu_context_slab,t->saved_fpu_context);
140	#endif	140	#endif
141	return -1;	141	return -1;
142	}	142	}
143	t->kstack = (__u8 *)PA2KA(PFN2ADDR(pfn));	143	t->kstack = (__u8 *)PA2KA(PFN2ADDR(pfn));
144		144
145	return 0;	145	return 0;
146	}	146	}
147		147
148	/** Destruction of thread_t object */	148	/** Destruction of thread_t object */
149	static int thr_destructor(void *obj)	149	static int thr_destructor(void *obj)
150	{	150	{
151	thread_t t = (thread_t )obj;	151	thread_t t = (thread_t )obj;
152		152
153	frame_free(ADDR2PFN(KA2PA(t->kstack)));	153	frame_free(ADDR2PFN(KA2PA(t->kstack)));
154	#ifdef ARCH_HAS_FPU	154	#ifdef ARCH_HAS_FPU
155	if (t->saved_fpu_context)	155	if (t->saved_fpu_context)
156	slab_free(fpu_context_slab,t->saved_fpu_context);	156	slab_free(fpu_context_slab,t->saved_fpu_context);
157	#endif	157	#endif
158	return 1; /* One page freed */	158	return 1; /* One page freed */
159	}	159	}
160		160
161	/** Initialize threads	161	/** Initialize threads
162	*	162	*
163	* Initialize kernel threads support.	163	* Initialize kernel threads support.
164	*	164	*
165	*/	165	*/
166	void thread_init(void)	166	void thread_init(void)
167	{	167	{
168	THREAD = NULL;	168	THREAD = NULL;
169	atomic_set(&nrdy,0);	169	atomic_set(&nrdy,0);
170	thread_slab = slab_cache_create("thread_slab",	170	thread_slab = slab_cache_create("thread_slab",
171	sizeof(thread_t),0,	171	sizeof(thread_t),0,
172	thr_constructor, thr_destructor, 0);	172	thr_constructor, thr_destructor, 0);
173	#ifdef ARCH_HAS_FPU	173	#ifdef ARCH_HAS_FPU
174	fpu_context_slab = slab_cache_create("fpu_slab",	174	fpu_context_slab = slab_cache_create("fpu_slab",
175	sizeof(fpu_context_t),	175	sizeof(fpu_context_t),
176	FPU_CONTEXT_ALIGN,	176	FPU_CONTEXT_ALIGN,
177	NULL, NULL, 0);	177	NULL, NULL, 0);
178	#endif	178	#endif
179		179
180	btree_create(&threads_btree);	180	btree_create(&threads_btree);
181	}	181	}
182		182
183	/** Make thread ready	183	/** Make thread ready
184	*	184	*
185	* Switch thread t to the ready state.	185	* Switch thread t to the ready state.
186	*	186	*
187	* @param t Thread to make ready.	187	* @param t Thread to make ready.
188	*	188	*
189	*/	189	*/
190	void thread_ready(thread_t *t)	190	void thread_ready(thread_t *t)
191	{	191	{
192	cpu_t *cpu;	192	cpu_t *cpu;
193	runq_t *r;	193	runq_t *r;
194	ipl_t ipl;	194	ipl_t ipl;
195	int i, avg;	195	int i, avg;
196		196
197	ipl = interrupts_disable();	197	ipl = interrupts_disable();
198		198
199	spinlock_lock(&t->lock);	199	spinlock_lock(&t->lock);
200		200
201	ASSERT(! (t->state == Ready));	201	ASSERT(! (t->state == Ready));
202		202
203	i = (t->priority < RQ_COUNT -1) ? ++t->priority : t->priority;	203	i = (t->priority < RQ_COUNT -1) ? ++t->priority : t->priority;
204		204
205	cpu = CPU;	205	cpu = CPU;
206	if (t->flags & X_WIRED) {	206	if (t->flags & X_WIRED) {
207	cpu = t->cpu;	207	cpu = t->cpu;
208	}	208	}
209	t->state = Ready;	209	t->state = Ready;
210	spinlock_unlock(&t->lock);	210	spinlock_unlock(&t->lock);
211		211
212	/*	212	/*
213	* Append t to respective ready queue on respective processor.	213	* Append t to respective ready queue on respective processor.
214	*/	214	*/
215	r = &cpu->rq[i];	215	r = &cpu->rq[i];
216	spinlock_lock(&r->lock);	216	spinlock_lock(&r->lock);
217	list_append(&t->rq_link, &r->rq_head);	217	list_append(&t->rq_link, &r->rq_head);
218	r->n++;	218	r->n++;
219	spinlock_unlock(&r->lock);	219	spinlock_unlock(&r->lock);
220		220
221	atomic_inc(&nrdy);	221	atomic_inc(&nrdy);
222	avg = atomic_get(&nrdy) / config.cpu_active;	222	avg = atomic_get(&nrdy) / config.cpu_active;
223	atomic_inc(&cpu->nrdy);	223	atomic_inc(&cpu->nrdy);
224		224
225	interrupts_restore(ipl);	225	interrupts_restore(ipl);
226	}	226	}
227		227
228	/** Destroy thread memory structure	228	/** Destroy thread memory structure
229	*	229	*
230	* Detach thread from all queues, cpus etc. and destroy it.	230	* Detach thread from all queues, cpus etc. and destroy it.
231	*	231	*
232	* Assume thread->lock is held!!	232	* Assume thread->lock is held!!
233	*/	233	*/
234	void thread_destroy(thread_t *t)	234	void thread_destroy(thread_t *t)
235	{	235	{
236	bool destroy_task = false;	236	bool destroy_task = false;
237		237
238	ASSERT(t->state == Exiting);	238	ASSERT(t->state == Exiting \|\| t->state == Undead);
239	ASSERT(t->task);	239	ASSERT(t->task);
240	ASSERT(t->cpu);	240	ASSERT(t->cpu);
241		241
242	spinlock_lock(&t->cpu->lock);	242	spinlock_lock(&t->cpu->lock);
243	if(t->cpu->fpu_owner==t)	243	if(t->cpu->fpu_owner==t)
244	t->cpu->fpu_owner=NULL;	244	t->cpu->fpu_owner=NULL;
245	spinlock_unlock(&t->cpu->lock);	245	spinlock_unlock(&t->cpu->lock);
246		246
247	spinlock_unlock(&t->lock);	247	spinlock_unlock(&t->lock);
248		248
249	spinlock_lock(&threads_lock);	249	spinlock_lock(&threads_lock);
250	btree_remove(&threads_btree, (btree_key_t) ((__address ) t), NULL);	250	btree_remove(&threads_btree, (btree_key_t) ((__address ) t), NULL);
251	spinlock_unlock(&threads_lock);	251	spinlock_unlock(&threads_lock);
252		252
253	/*	253	/*
254	* Detach from the containing task.	254	* Detach from the containing task.
255	*/	255	*/
256	spinlock_lock(&t->task->lock);	256	spinlock_lock(&t->task->lock);
257	list_remove(&t->th_link);	257	list_remove(&t->th_link);
258	if (--t->task->refcount == 0) {	258	if (--t->task->refcount == 0) {
259	t->task->accept_new_threads = false;	259	t->task->accept_new_threads = false;
260	destroy_task = true;	260	destroy_task = true;
261	}	261	}
262	spinlock_unlock(&t->task->lock);	262	spinlock_unlock(&t->task->lock);
263		263
264	if (destroy_task)	264	if (destroy_task)
265	task_destroy(t->task);	265	task_destroy(t->task);
266		266
267	slab_free(thread_slab, t);	267	slab_free(thread_slab, t);
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.	276	* @param task Task to which the thread belongs.
277	* @param flags Thread flags.	277	* @param flags Thread flags.
278	* @param name Symbolic name.	278	* @param name Symbolic name.
279	*	279	*
280	* @return New thread's structure on success, NULL on failure.	280	* @return New thread's structure on success, NULL on failure.
281	*	281	*
282	*/	282	*/
283	thread_t thread_create(void ( func)(void ), void arg, task_t task, int flags, char name)	283	thread_t thread_create(void ( func)(void ), void arg, task_t task, int flags, char name)
284	{	284	{
285	thread_t *t;	285	thread_t *t;
286	ipl_t ipl;	286	ipl_t ipl;
287		287
288	t = (thread_t *) slab_alloc(thread_slab, 0);	288	t = (thread_t *) slab_alloc(thread_slab, 0);
289	if (!t)	289	if (!t)
290	return NULL;	290	return NULL;
291		291
292	thread_create_arch(t);	292	thread_create_arch(t);
293		293
294	/* Not needed, but good for debugging */	294	/* Not needed, but good for debugging */
295	memsetb((__address)t->kstack, THREAD_STACK_SIZE * 1<<STACK_FRAMES, 0);	295	memsetb((__address)t->kstack, THREAD_STACK_SIZE * 1<<STACK_FRAMES, 0);
296		296
297	ipl = interrupts_disable();	297	ipl = interrupts_disable();
298	spinlock_lock(&tidlock);	298	spinlock_lock(&tidlock);
299	t->tid = ++last_tid;	299	t->tid = ++last_tid;
300	spinlock_unlock(&tidlock);	300	spinlock_unlock(&tidlock);
301	interrupts_restore(ipl);	301	interrupts_restore(ipl);
302		302
303	context_save(&t->saved_context);	303	context_save(&t->saved_context);
304	context_set(&t->saved_context, FADDR(cushion), (__address) t->kstack, THREAD_STACK_SIZE);	304	context_set(&t->saved_context, FADDR(cushion), (__address) t->kstack, THREAD_STACK_SIZE);
305		305
306	the_initialize((the_t *) t->kstack);	306	the_initialize((the_t *) t->kstack);
307		307
308	ipl = interrupts_disable();	308	ipl = interrupts_disable();
309	t->saved_context.ipl = interrupts_read();	309	t->saved_context.ipl = interrupts_read();
310	interrupts_restore(ipl);	310	interrupts_restore(ipl);
311		311
312	memcpy(t->name, name, THREAD_NAME_BUFLEN);	312	memcpy(t->name, name, THREAD_NAME_BUFLEN);
313		313
314	t->thread_code = func;	314	t->thread_code = func;
315	t->thread_arg = arg;	315	t->thread_arg = arg;
316	t->ticks = -1;	316	t->ticks = -1;
317	t->priority = -1; /* start in rq[0] */	317	t->priority = -1; /* start in rq[0] */
318	t->cpu = NULL;	318	t->cpu = NULL;
319	t->flags = 0;	319	t->flags = 0;
320	t->state = Entering;	320	t->state = Entering;
321	t->call_me = NULL;	321	t->call_me = NULL;
322	t->call_me_with = NULL;	322	t->call_me_with = NULL;
323		323
324	timeout_initialize(&t->sleep_timeout);	324	timeout_initialize(&t->sleep_timeout);
325	t->sleep_interruptible = false;	325	t->sleep_interruptible = false;
326	t->sleep_queue = NULL;	326	t->sleep_queue = NULL;
327	t->timeout_pending = 0;	327	t->timeout_pending = 0;
328		328
329	t->in_copy_from_uspace = false;	329	t->in_copy_from_uspace = false;
330	t->in_copy_to_uspace = false;	330	t->in_copy_to_uspace = false;
331		331
332	t->interrupted = false;	332	t->interrupted = false;
333	t->detached = false;	333	t->detached = false;
334	waitq_initialize(&t->join_wq);	334	waitq_initialize(&t->join_wq);
335		335
336	t->rwlock_holder_type = RWLOCK_NONE;	336	t->rwlock_holder_type = RWLOCK_NONE;
337		337
338	t->task = task;	338	t->task = task;
339		339
340	t->fpu_context_exists = 0;	340	t->fpu_context_exists = 0;
341	t->fpu_context_engaged = 0;	341	t->fpu_context_engaged = 0;
342		342
343	/*	343	/*
344	* Attach to the containing task.	344	* Attach to the containing task.
345	*/	345	*/
346	spinlock_lock(&task->lock);	346	spinlock_lock(&task->lock);
347	if (!task->accept_new_threads) {	347	if (!task->accept_new_threads) {
348	spinlock_unlock(&task->lock);	348	spinlock_unlock(&task->lock);
349	slab_free(thread_slab, t);	349	slab_free(thread_slab, t);
350	return NULL;	350	return NULL;
351	}	351	}
352	list_append(&t->th_link, &task->th_head);	352	list_append(&t->th_link, &task->th_head);
353	task->refcount++;	353	task->refcount++;
354	spinlock_unlock(&task->lock);	354	spinlock_unlock(&task->lock);
355		355
356	/*	356	/*
357	* Register this thread in the system-wide list.	357	* Register this thread in the system-wide list.
358	*/	358	*/
359	ipl = interrupts_disable();	359	ipl = interrupts_disable();
360	spinlock_lock(&threads_lock);	360	spinlock_lock(&threads_lock);
361	btree_insert(&threads_btree, (btree_key_t) ((__address) t), (void *) t, NULL);	361	btree_insert(&threads_btree, (btree_key_t) ((__address) t), (void *) t, NULL);
362	spinlock_unlock(&threads_lock);	362	spinlock_unlock(&threads_lock);
363		363
364	interrupts_restore(ipl);	364	interrupts_restore(ipl);
365		365
366	return t;	366	return t;
367	}	367	}
368		368
369	/** Make thread exiting	369	/** Make thread exiting
370	*	370	*
371	* End current thread execution and switch it to the exiting	371	* End current thread execution and switch it to the exiting
372	* state. All pending timeouts are executed.	372	* state. All pending timeouts are executed.
373	*	373	*
374	*/	374	*/
375	void thread_exit(void)	375	void thread_exit(void)
376	{	376	{
377	ipl_t ipl;	377	ipl_t ipl;
378		378
379	restart:	379	restart:
380	ipl = interrupts_disable();	380	ipl = interrupts_disable();
381	spinlock_lock(&THREAD->lock);	381	spinlock_lock(&THREAD->lock);
382	if (THREAD->timeout_pending) { /* busy waiting for timeouts in progress */	382	if (THREAD->timeout_pending) { /* busy waiting for timeouts in progress */
383	spinlock_unlock(&THREAD->lock);	383	spinlock_unlock(&THREAD->lock);
384	interrupts_restore(ipl);	384	interrupts_restore(ipl);
385	goto restart;	385	goto restart;
386	}	386	}
387	THREAD->state = Exiting;	387	THREAD->state = Exiting;
388	spinlock_unlock(&THREAD->lock);	388	spinlock_unlock(&THREAD->lock);
389	scheduler();	389	scheduler();
390	}	390	}
391		391
392		392
393	/** Thread sleep	393	/** Thread sleep
394	*	394	*
395	* Suspend execution of the current thread.	395	* Suspend execution of the current thread.
396	*	396	*
397	* @param sec Number of seconds to sleep.	397	* @param sec Number of seconds to sleep.
398	*	398	*
399	*/	399	*/
400	void thread_sleep(__u32 sec)	400	void thread_sleep(__u32 sec)
401	{	401	{
402	thread_usleep(sec*1000000);	402	thread_usleep(sec*1000000);
403	}	403	}
404		404
405	/** Wait for another thread to exit.	405	/** Wait for another thread to exit.
406	*	406	*
407	* @param t Thread to join on exit.	407	* @param t Thread to join on exit.
408	* @param usec Timeout in microseconds.	408	* @param usec Timeout in microseconds.
409	* @param flags Mode of operation.	409	* @param flags Mode of operation.
410	*	410	*
411	* @return An error code from errno.h or an error code from synch.h.	411	* @return An error code from errno.h or an error code from synch.h.
412	*/	412	*/
413	int thread_join_timeout(thread_t *t, __u32 usec, int flags)	413	int thread_join_timeout(thread_t *t, __u32 usec, int flags)
414	{	414	{
415	ipl_t ipl;	415	ipl_t ipl;
416	int rc;	416	int rc;
417		417
418	if (t == THREAD)	418	if (t == THREAD)
419	return EINVAL;	419	return EINVAL;
420		420
421	/*	421	/*
422	* Since thread join can only be called once on an undetached thread,	422	* Since thread join can only be called once on an undetached thread,
423	* the thread pointer is guaranteed to be still valid.	423	* the thread pointer is guaranteed to be still valid.
424	*/	424	*/
425		425
426	ipl = interrupts_disable();	426	ipl = interrupts_disable();
427	spinlock_lock(&t->lock);	427	spinlock_lock(&t->lock);
428		428
429	ASSERT(!t->detached);	429	ASSERT(!t->detached);
430		430
431	(void) waitq_sleep_prepare(&t->join_wq);	431	(void) waitq_sleep_prepare(&t->join_wq);
432	spinlock_unlock(&t->lock);	432	spinlock_unlock(&t->lock);
433		433
434	rc = waitq_sleep_timeout_unsafe(&t->join_wq, usec, flags);	434	rc = waitq_sleep_timeout_unsafe(&t->join_wq, usec, flags);
435		435
436	waitq_sleep_finish(&t->join_wq, rc, ipl);	436	waitq_sleep_finish(&t->join_wq, rc, ipl);
437		437
438	return rc;	438	return rc;
439	}	439	}
440		440
441	/** Detach thread.	441	/** Detach thread.
442	*	442	*
443	* Mark the thread as detached, if the thread is already in the Undead state,	443	* Mark the thread as detached, if the thread is already in the Undead state,
444	* deallocate its resources.	444	* deallocate its resources.
445	*	445	*
446	* @param t Thread to be detached.	446	* @param t Thread to be detached.
447	*/	447	*/
448	void thread_detach(thread_t *t)	448	void thread_detach(thread_t *t)
449	{	449	{
450	ipl_t ipl;	450	ipl_t ipl;
451		451
452	/*	452	/*
453	* Since the thread is expected to not be already detached,	453	* Since the thread is expected to not be already detached,
454	* pointer to it must be still valid.	454	* pointer to it must be still valid.
455	*/	455	*/
456		456
457	ipl = interrupts_disable();	457	ipl = interrupts_disable();
458	spinlock_lock(&t->lock);	458	spinlock_lock(&t->lock);
459	ASSERT(!t->detached);	459	ASSERT(!t->detached);
460	if (t->state == Undead) {	460	if (t->state == Undead) {
461	thread_destroy(t); /* unlocks &t->lock */	461	thread_destroy(t); /* unlocks &t->lock */
462	interrupts_restore(ipl);	462	interrupts_restore(ipl);
463	return;	463	return;
464	} else {	464	} else {
465	t->detached = true;	465	t->detached = true;
466	}	466	}
467	spinlock_unlock(&t->lock);	467	spinlock_unlock(&t->lock);
468	interrupts_restore(ipl);	468	interrupts_restore(ipl);
469	}	469	}
470		470
471	/** Thread usleep	471	/** Thread usleep
472	*	472	*
473	* Suspend execution of the current thread.	473	* Suspend execution of the current thread.
474	*	474	*
475	* @param usec Number of microseconds to sleep.	475	* @param usec Number of microseconds to sleep.
476	*	476	*
477	*/	477	*/
478	void thread_usleep(__u32 usec)	478	void thread_usleep(__u32 usec)
479	{	479	{
480	waitq_t wq;	480	waitq_t wq;
481		481
482	waitq_initialize(&wq);	482	waitq_initialize(&wq);
483		483
484	(void) waitq_sleep_timeout(&wq, usec, SYNCH_FLAGS_NON_BLOCKING);	484	(void) waitq_sleep_timeout(&wq, usec, SYNCH_FLAGS_NON_BLOCKING);
485	}	485	}
486		486
487	/** Register thread out-of-context invocation	487	/** Register thread out-of-context invocation
488	*	488	*
489	* Register a function and its argument to be executed	489	* Register a function and its argument to be executed
490	* on next context switch to the current thread.	490	* on next context switch to the current thread.
491	*	491	*
492	* @param call_me Out-of-context function.	492	* @param call_me Out-of-context function.
493	* @param call_me_with Out-of-context function argument.	493	* @param call_me_with Out-of-context function argument.
494	*	494	*
495	*/	495	*/
496	void thread_register_call_me(void (* call_me)(void ), void call_me_with)	496	void thread_register_call_me(void (* call_me)(void ), void call_me_with)
497	{	497	{
498	ipl_t ipl;	498	ipl_t ipl;
499		499
500	ipl = interrupts_disable();	500	ipl = interrupts_disable();

Subversion Repositories HelenOS

(root)/trunk/kernel/generic/src/proc/thread.c @ 2436 – Rev 1579 → 1581