bertos/kern/preempt.c

   1 /**
   2  * \file
   3  * <!--
   4  * This file is part of BeRTOS.
   5  *
   6  * Bertos is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  * As a special exception, you may use this file as part of a free software
  21  * library without restriction.  Specifically, if other files instantiate
  22  * templates or use macros or inline functions from this file, or you compile
  23  * this file and link it with other files to produce an executable, this
  24  * file does not by itself cause the resulting executable to be covered by
  25  * the GNU General Public License.  This exception does not however
  26  * invalidate any other reasons why the executable file might be covered by
  27  * the GNU General Public License.
  28  *
  29  * Copyright 2008 Bernie Innocenti <bernie@codewiz.org>
  30  * -->
  31  *
  32  * \brief Simple preemptive multitasking scheduler.
  33  *
  34  * \version $Id: proc.c 1616 2008-08-10 19:41:26Z bernie $
  35  * \author Bernie Innocenti <bernie@codewiz.org>
  36  */
  37
  38 #include "proc_p.h"
  39 #include "proc.h"
  40
  41 #include <kern/irq.h>
  42 #include <kern/monitor.h>
  43 #include <cpu/frame.h> // CPU_IDLE
  44 #include <cpu/irq.h>   // IRQ_DISABLE()...
  45 #include <drv/timer.h>
  46 #include <cfg/module.h>
  47
  48
  49 int preempt_forbid_cnt;
  50
  51 static Timer preempt_timer;
  52
  53
  54 // fwd decl from idle.c
  55 void idle_init(void);
  56
  57
  58 void proc_preempt(void)
  59 {
  60         IRQ_DISABLE;
  61
  62         ASSERT(preempt_forbid_cnt == 0);
  63         LIST_ASSERT_VALID(&ProcReadyList);
  64         CurrentProcess = (struct Process *)list_remHead(&ProcReadyList);
  65         ASSERT2(CurrentProcess, "no idle proc?");
  66
  67         IRQ_ENABLE;
  68
  69         TRACEMSG("launching %p:%s", CurrentProcess, proc_currentName());
  70 }
  71
  72 void proc_preempt_timer(UNUSED_ARG(void *, param))
  73 {
  74         if (!preempt_forbid_cnt)
  75         {
  76                 IRQ_DISABLE;
  77                 TRACEMSG("preempting %p:%s", CurrentProcess, proc_currentName());
  78 #if 0
  79                 SCHED_ENQUEUE(CurrentProcess);
  80                 proc_preempt();
  81 #endif
  82                 IRQ_ENABLE;
  83         }
  84
  85         timer_setDelay(&preempt_timer, CONFIG_KERN_QUANTUM);
  86         timer_add(&preempt_timer);
  87 }
  88
  89 void proc_schedule(void)
  90 {
  91         ATOMIC(LIST_ASSERT_VALID(&ProcReadyList));
  92         TRACEMSG("%p:%s", CurrentProcess, proc_currentName());
  93         ATOMIC(LIST_ASSERT_VALID(&ProcReadyList));
  94
  95         /* Sleeping with IRQs disabled or preemption forbidden is illegal */
  96         ASSERT_IRQ_ENABLED();
  97         ASSERT(preempt_forbid_cnt == 0);
  98
  99         // Will invoke proc_preempt() in interrupt context
 100         kill(0, SIGUSR1);
 101 }
 102
 103 void proc_yield(void)
 104 {
 105         TRACEMSG("%p:%s", CurrentProcess, proc_currentName());
 106
 107         IRQ_DISABLE;
 108         SCHED_ENQUEUE(CurrentProcess);
 109         IRQ_ENABLE;
 110
 111         proc_schedule();
 112 }
 113
 114 void proc_entry(void (*user_entry)(void))
 115 {
 116         user_entry();
 117         proc_exit();
 118 }
 119
 120 void preempt_init(void)
 121 {
 122         MOD_CHECK(irq);
 123         MOD_CHECK(timer);
 124
 125         irq_register(SIGUSR1, proc_preempt);
 126
 127         timer_setSoftint(&preempt_timer, proc_preempt_timer, NULL);
 128         timer_setDelay(&preempt_timer, CONFIG_KERN_QUANTUM);
 129         timer_add(&preempt_timer);
 130
 131         idle_init();
 132 }