Subversion Repositories HelenOS-historic

Rev

Rev 842 | Rev 935 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

  1. /*
  2.  * Copyright (C) 2001-2004 Jakub Jermar
  3.  * All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  *
  9.  * - Redistributions of source code must retain the above copyright
  10.  *   notice, this list of conditions and the following disclaimer.
  11.  * - Redistributions in binary form must reproduce the above copyright
  12.  *   notice, this list of conditions and the following disclaimer in the
  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
  15.  *   derived from this software without specific prior written permission.
  16.  *
  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
  19.  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  20.  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  21.  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  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
  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
  26.  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27.  */
  28.  
  29. #include <proc/scheduler.h>
  30. #include <proc/thread.h>
  31. #include <proc/task.h>
  32. #include <mm/frame.h>
  33. #include <mm/page.h>
  34. #include <arch/asm.h>
  35. #include <arch.h>
  36. #include <synch/synch.h>
  37. #include <synch/spinlock.h>
  38. #include <synch/waitq.h>
  39. #include <synch/rwlock.h>
  40. #include <cpu.h>
  41. #include <func.h>
  42. #include <context.h>
  43. #include <adt/list.h>
  44. #include <typedefs.h>
  45. #include <time/clock.h>
  46. #include <adt/list.h>
  47. #include <config.h>
  48. #include <arch/interrupt.h>
  49. #include <smp/ipi.h>
  50. #include <arch/faddr.h>
  51. #include <arch/atomic.h>
  52. #include <memstr.h>
  53. #include <print.h>
  54. #include <mm/slab.h>
  55. #include <debug.h>
  56.  
  57. char *thread_states[] = {"Invalid", "Running", "Sleeping", "Ready", "Entering", "Exiting"}; /**< Thread states */
  58.  
  59. SPINLOCK_INITIALIZE(threads_lock);  /**< Lock protecting threads_head list. For locking rules, see declaration thereof. */
  60. LIST_INITIALIZE(threads_head);      /**< List of all threads. */
  61.  
  62. SPINLOCK_INITIALIZE(tidlock);
  63. __u32 last_tid = 0;
  64.  
  65. static slab_cache_t *thread_slab;
  66.  
  67.  
  68. /** Thread wrapper
  69.  *
  70.  * This wrapper is provided to ensure that every thread
  71.  * makes a call to thread_exit() when its implementing
  72.  * function returns.
  73.  *
  74.  * interrupts_disable() is assumed.
  75.  *
  76.  */
  77. static void cushion(void)
  78. {
  79.     void (*f)(void *) = THREAD->thread_code;
  80.     void *arg = THREAD->thread_arg;
  81.  
  82.     /* this is where each thread wakes up after its creation */
  83.     before_thread_runs();
  84.  
  85.     spinlock_unlock(&THREAD->lock);
  86.     interrupts_enable();
  87.  
  88.     f(arg);
  89.     thread_exit();
  90.     /* not reached */
  91. }
  92.  
  93. /** Initialization and allocation for thread_t structure */
  94. static int thr_constructor(void *obj, int kmflags)
  95. {
  96.     thread_t *t = (thread_t *)obj;
  97.     pfn_t pfn;
  98.     int status;
  99.  
  100.     spinlock_initialize(&t->lock, "thread_t_lock");
  101.     link_initialize(&t->rq_link);
  102.     link_initialize(&t->wq_link);
  103.     link_initialize(&t->th_link);
  104.     link_initialize(&t->threads_link);
  105.    
  106.     pfn = frame_alloc_rc(ONE_FRAME, FRAME_KA | kmflags,&status);
  107.     if (status)
  108.         return -1;
  109.     t->kstack = (__u8 *)PA2KA(PFN2ADDR(pfn));
  110.  
  111.     return 0;
  112. }
  113.  
  114. /** Destruction of thread_t object */
  115. static int thr_destructor(void *obj)
  116. {
  117.     thread_t *t = (thread_t *)obj;
  118.  
  119.     frame_free(ADDR2PFN(KA2PA(t->kstack)));
  120.     return 1; /* One page freed */
  121. }
  122.  
  123. /** Initialize threads
  124.  *
  125.  * Initialize kernel threads support.
  126.  *
  127.  */
  128. void thread_init(void)
  129. {
  130.     THREAD = NULL;
  131.     atomic_set(&nrdy,0);
  132.     thread_slab = slab_cache_create("thread_slab",
  133.                     sizeof(thread_t),0,
  134.                     thr_constructor, thr_destructor, 0);
  135. }
  136.  
  137.  
  138. /** Make thread ready
  139.  *
  140.  * Switch thread t to the ready state.
  141.  *
  142.  * @param t Thread to make ready.
  143.  *
  144.  */
  145. void thread_ready(thread_t *t)
  146. {
  147.     cpu_t *cpu;
  148.     runq_t *r;
  149.     ipl_t ipl;
  150.     int i, avg;
  151.  
  152.     ipl = interrupts_disable();
  153.  
  154.     spinlock_lock(&t->lock);
  155.  
  156.     i = (t->priority < RQ_COUNT -1) ? ++t->priority : t->priority;
  157.    
  158.     cpu = CPU;
  159.     if (t->flags & X_WIRED) {
  160.         cpu = t->cpu;
  161.     }
  162.     spinlock_unlock(&t->lock);
  163.    
  164.     /*
  165.      * Append t to respective ready queue on respective processor.
  166.      */
  167.     r = &cpu->rq[i];
  168.     spinlock_lock(&r->lock);
  169.     list_append(&t->rq_link, &r->rq_head);
  170.     r->n++;
  171.     spinlock_unlock(&r->lock);
  172.  
  173.     atomic_inc(&nrdy);
  174.     avg = atomic_get(&nrdy) / config.cpu_active;
  175.     atomic_inc(&cpu->nrdy);
  176.  
  177.     interrupts_restore(ipl);
  178. }
  179.  
  180.  
  181. /** Destroy thread memory structure
  182.  *
  183.  * Detach thread from all queues, cpus etc. and destroy it.
  184.  *
  185.  * Assume thread->lock is held!!
  186.  */
  187. void thread_destroy(thread_t *t)
  188. {
  189.     ASSERT(t->state == Exiting);
  190.     ASSERT(t->task);
  191.     ASSERT(t->cpu);
  192.  
  193.     spinlock_lock(&t->cpu->lock);
  194.     if(t->cpu->fpu_owner==t)
  195.         t->cpu->fpu_owner=NULL;
  196.     spinlock_unlock(&t->cpu->lock);
  197.  
  198.     /*
  199.      * Detach from the containing task.
  200.      */
  201.     spinlock_lock(&t->task->lock);
  202.     list_remove(&t->th_link);
  203.     spinlock_unlock(&t->task->lock);
  204.    
  205.     spinlock_unlock(&t->lock);
  206.    
  207.     spinlock_lock(&threads_lock);
  208.     list_remove(&t->threads_link);
  209.     spinlock_unlock(&threads_lock);
  210.    
  211.     slab_free(thread_slab, t);
  212. }
  213.  
  214.  
  215. /** Create new thread
  216.  *
  217.  * Create a new thread.
  218.  *
  219.  * @param func  Thread's implementing function.
  220.  * @param arg   Thread's implementing function argument.
  221.  * @param task  Task to which the thread belongs.
  222.  * @param flags Thread flags.
  223.  *
  224.  * @return New thread's structure on success, NULL on failure.
  225.  *
  226.  */
  227. thread_t *thread_create(void (* func)(void *), void *arg, task_t *task, int flags)
  228. {
  229.     thread_t *t;
  230.     ipl_t ipl;
  231.    
  232.     t = (thread_t *) slab_alloc(thread_slab, 0);
  233.     if (!t)
  234.         return NULL;
  235.    
  236.     /* Not needed, but good for debugging */
  237.     memsetb((__address)t->kstack, THREAD_STACK_SIZE, 0);
  238.    
  239.     ipl = interrupts_disable();
  240.     spinlock_lock(&tidlock);
  241.     t->tid = ++last_tid;
  242.     spinlock_unlock(&tidlock);
  243.     interrupts_restore(ipl);
  244.    
  245.     context_save(&t->saved_context);
  246.     context_set(&t->saved_context, FADDR(cushion), (__address) t->kstack, THREAD_STACK_SIZE);
  247.    
  248.     the_initialize((the_t *) t->kstack);
  249.    
  250.     ipl = interrupts_disable();
  251.     t->saved_context.ipl = interrupts_read();
  252.     interrupts_restore(ipl);
  253.    
  254.     t->thread_code = func;
  255.     t->thread_arg = arg;
  256.     t->ticks = -1;
  257.     t->priority = -1;       /* start in rq[0] */
  258.     t->cpu = NULL;
  259.     t->flags = 0;
  260.     t->state = Entering;
  261.     t->call_me = NULL;
  262.     t->call_me_with = NULL;
  263.    
  264.     timeout_initialize(&t->sleep_timeout);
  265.     t->sleep_queue = NULL;
  266.     t->timeout_pending = 0;
  267.    
  268.     t->rwlock_holder_type = RWLOCK_NONE;
  269.        
  270.     t->task = task;
  271.    
  272.     t->fpu_context_exists = 0;
  273.     t->fpu_context_engaged = 0;
  274.    
  275.     /*
  276.      * Register this thread in the system-wide list.
  277.      */
  278.     ipl = interrupts_disable();
  279.     spinlock_lock(&threads_lock);
  280.     list_append(&t->threads_link, &threads_head);
  281.     spinlock_unlock(&threads_lock);
  282.    
  283.     /*
  284.      * Attach to the containing task.
  285.      */
  286.     spinlock_lock(&task->lock);
  287.     list_append(&t->th_link, &task->th_head);
  288.     spinlock_unlock(&task->lock);
  289.    
  290.     interrupts_restore(ipl);
  291.    
  292.     return t;
  293. }
  294.  
  295.  
  296. /** Make thread exiting
  297.  *
  298.  * End current thread execution and switch it to the exiting
  299.  * state. All pending timeouts are executed.
  300.  *
  301.  */
  302. void thread_exit(void)
  303. {
  304.     ipl_t ipl;
  305.  
  306. restart:
  307.     ipl = interrupts_disable();
  308.     spinlock_lock(&THREAD->lock);
  309.     if (THREAD->timeout_pending) { /* busy waiting for timeouts in progress */
  310.         spinlock_unlock(&THREAD->lock);
  311.         interrupts_restore(ipl);
  312.         goto restart;
  313.     }
  314.     THREAD->state = Exiting;
  315.     spinlock_unlock(&THREAD->lock);
  316.     scheduler();
  317. }
  318.  
  319.  
  320. /** Thread sleep
  321.  *
  322.  * Suspend execution of the current thread.
  323.  *
  324.  * @param sec Number of seconds to sleep.
  325.  *
  326.  */
  327. void thread_sleep(__u32 sec)
  328. {
  329.     thread_usleep(sec*1000000);
  330. }
  331.  
  332.  
  333. /** Thread usleep
  334.  *
  335.  * Suspend execution of the current thread.
  336.  *
  337.  * @param usec Number of microseconds to sleep.
  338.  *
  339.  */
  340. void thread_usleep(__u32 usec)
  341. {
  342.     waitq_t wq;
  343.                  
  344.     waitq_initialize(&wq);
  345.  
  346.     (void) waitq_sleep_timeout(&wq, usec, SYNCH_NON_BLOCKING);
  347. }
  348.  
  349.  
  350. /** Register thread out-of-context invocation
  351.  *
  352.  * Register a function and its argument to be executed
  353.  * on next context switch to the current thread.
  354.  *
  355.  * @param call_me      Out-of-context function.
  356.  * @param call_me_with Out-of-context function argument.
  357.  *
  358.  */
  359. void thread_register_call_me(void (* call_me)(void *), void *call_me_with)
  360. {
  361.     ipl_t ipl;
  362.    
  363.     ipl = interrupts_disable();
  364.     spinlock_lock(&THREAD->lock);
  365.     THREAD->call_me = call_me;
  366.     THREAD->call_me_with = call_me_with;
  367.     spinlock_unlock(&THREAD->lock);
  368.     interrupts_restore(ipl);
  369. }
  370.  
  371. /** Print list of threads debug info */
  372. void thread_print_list(void)
  373. {
  374.     link_t *cur;
  375.     thread_t *t;
  376.     ipl_t ipl;
  377.    
  378.     /* Messing with thread structures, avoid deadlock */
  379.     ipl = interrupts_disable();
  380.     spinlock_lock(&threads_lock);
  381.  
  382.     for (cur=threads_head.next; cur!=&threads_head; cur=cur->next) {
  383.         t = list_get_instance(cur, thread_t, threads_link);
  384.         printf("Thr: %d(%s) ", t->tid, thread_states[t->state]);
  385.         if (t->cpu)
  386.             printf("cpu%d ", t->cpu->id);
  387.        
  388.         printf("\n");
  389.     }
  390.  
  391.     spinlock_unlock(&threads_lock);
  392.     interrupts_enable();
  393. }
  394.