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 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         LIST_ASSERT_VALID(&ProcReadyList);
  63         CurrentProcess = (struct Process *)list_remHead(&ProcReadyList);
  64         LIST_ASSERT_VALID(&ProcReadyList);
  65         ASSERT2(CurrentProcess, "no idle proc?");
  66
  67         IRQ_ENABLE;
  68
  69         TRACEMSG("new proc: %p:%s", CurrentProcess, CurrentProcess ? CurrentProcess->monitor.name : "---");
  70         monitor_report();
  71 }
  72
  73 void proc_preempt_timer(UNUSED_ARG(void *, param))
  74 {
  75         /* Abort if task preemption is disabled */
  76         if (preempt_forbid_cnt)
  77                 return;
  78
  79         IRQ_DISABLE;
  80 /*
  81         if (!CurrentProcess->forbid_cnt)
  82         {
  83                 TRACEMSG("preempting %p:%s", CurrentProcess, CurrentProcess->monitor.name);
  84                 SCHED_ENQUEUE(CurrentProcess);
  85                 proc_preempt();
  86         }
  87 */
  88         IRQ_ENABLE;
  89
  90         timer_setDelay(&preempt_timer, CONFIG_KERN_QUANTUM);
  91         timer_add(&preempt_timer);
  92 }
  93
  94 void proc_schedule(void)
  95 {
  96         ATOMIC(LIST_ASSERT_VALID(&ProcReadyList));
  97         TRACEMSG("%p:%s", CurrentProcess, proc_currentName());
  98         ATOMIC(LIST_ASSERT_VALID(&ProcReadyList));
  99
 100         /* Sleeping with IRQs disabled or preemption forbidden is illegal */
 101         ASSERT_IRQ_ENABLED();
 102         ASSERT(preempt_forbid_cnt == 0);
 103
 104         // Will invoke proc_preempt() in interrupt context
 105         kill(0, SIGUSR1);
 106 }
 107
 108 void proc_yield(void)
 109 {
 110         TRACEMSG("%p:%s", CurrentProcess, proc_currentName());
 111
 112         IRQ_DISABLE;
 113         SCHED_ENQUEUE(CurrentProcess);
 114         IRQ_ENABLE;
 115
 116         proc_schedule();
 117 }
 118
 119 void proc_entry(void (*user_entry)(void))
 120 {
 121         user_entry();
 122         proc_exit();
 123 }
 124
 125 void preempt_init(void)
 126 {
 127         MOD_CHECK(irq);
 128         MOD_CHECK(timer);
 129
 130         irq_register(SIGUSR1, proc_preempt);
 131
 132         timer_setSoftint(&preempt_timer, proc_preempt_timer, NULL);
 133         timer_setDelay(&preempt_timer, CONFIG_KERN_QUANTUM);
 134         timer_add(&preempt_timer);
 135
 136         idle_init();
 137 }