Subversion Repositories shark

/*
 * Project: S.Ha.R.K.
 *
 * Coordinators:
 *   Giorgio Buttazzo    <giorgio@sssup.it>
 *   Paolo Gai           <pj@gandalf.sssup.it>
 *
 * Authors     :
 *   Paolo Gai           <pj@gandalf.sssup.it>
 *   (see the web pages for full authors list)
 *
 * ReTiS Lab (Scuola Superiore S.Anna - Pisa - Italy)
 *
 * http://www.sssup.it
 * http://retis.sssup.it
 * http://shark.sssup.it
 */

/*
 * Copyright (C) 2001 Paolo Gai
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "edfstar.h"
#include <ll/stdio.h>
#include <ll/string.h>
#include <kernel/model.h>
#include <kernel/descr.h>
#include <kernel/var.h>
#include <kernel/func.h>
#include <kernel/trace.h>


#define EDFSTAR_CHANGE_LEVEL 1

/* for iqueues */
/* #include "iqueue.h" Now iqueues are the only queue type available
   into the kernel */
#include <kernel/iqueue.h>

/* for BUDGET_TASK_MODEL */
#include "fsf_server.h"

/*
 * DEBUG stuffs begin
 */

//#define EDFSTAR_DEBUG

#ifdef EDFSTAR_DEBUG

static __inline__ fake_printf(char *fmt, ...) {}

//#define edfstar_printf fake_printf
//#define edfstar_printf2 fake_printf
//#define edfstar_printf3 fake_printf

#define edfstar_printf kern_printf
#define edfstar_printf2 kern_printf
#define edfstar_printf3 kern_printf
#endif

/*
 * DEBUG stuffs end
 */

/* Status used in the level */
#define EDFSTAR_READY         MODULE_STATUS_BASE    /* - Ready status        */
#define EDFSTAR_IDLE          MODULE_STATUS_BASE+4  /* to wait the deadline  */

/* flags */
#define EDFSTAR_FLAG_NORAISEEXC  2

/* the level redefinition for the Earliest Deadline First level */
typedef struct {
  level_des l;     /* the standard level descriptor          */

  TIME period[MAX_PROC]; /* The task periods; the deadlines are
                       stored in the priority field           */
  int deadline_timer[MAX_PROC];
                   /* The task deadline timers               */

  struct timespec deadline_timespec[MAX_PROC];

  int dline_miss[MAX_PROC]; /* Deadline miss counter */
  int wcet_miss[MAX_PROC];  /* Wcet miss counter */

  int nact[MAX_PROC];       /* Wcet miss counter */

  int flag[MAX_PROC];
                   /* used to manage the JOB_TASK_MODEL and the
                       periodicity                            */

  IQUEUE ready;     /* the ready queue                        */

  PID activated;   /* the task that has been inserted into the
                       master module */

  int budget[MAX_PROC];

  int scheduling_level;

  int new_level[MAX_PROC];
  int wcet[MAX_PROC]; /* save the wcet fields */

} EDFSTAR_level_des;

static void EDFSTAR_check_preemption(EDFSTAR_level_des *lev)
{
  PID first;

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:chk)");
#endif

  if ((first = iq_query_first(&lev->ready)) != lev->activated) {
    if (lev->activated != NIL)
      level_table[ lev->scheduling_level ]->
        private_extract(lev->scheduling_level, lev->activated);

    lev->activated = first;

    if (first != NIL) {
      BUDGET_TASK_MODEL b;
      budget_task_default_model(b, lev->budget[first]);

      level_table[ lev->scheduling_level ]->
        private_insert(lev->scheduling_level, first, (TASK_MODEL *)&b);
    }
  }
}

static void EDFSTAR_timer_deadline(void *par);

static void EDFSTAR_internal_activate(EDFSTAR_level_des *lev, PID p,
                                      struct timespec *t)
{
#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:iact)");
#endif

  ADDUSEC2TIMESPEC(lev->period[p], t);

  *iq_query_timespec(p, &lev->ready) = *t;
  lev->deadline_timespec[p] = *t;

  /* Insert task in the correct position */
  proc_table[p].status = EDFSTAR_READY;
  iq_timespec_insert(p,&lev->ready);

  /* needed because when there is a wcet miss I disable CONTROL_CAP */
  proc_table[p].control |= CONTROL_CAP;

  /* check for preemption */
  EDFSTAR_check_preemption(lev);
}

static void EDFSTAR_timer_deadline(void *par)
{
  PID p = (PID) par;
  EDFSTAR_level_des *lev;

#ifdef EDFSTAR_DEBUG
//  edfstar_printf("(E:tdl ");
#endif

  lev = (EDFSTAR_level_des *)level_table[proc_table[p].task_level];

  switch (proc_table[p].status) {
    case EDFSTAR_IDLE:
#ifdef EDFSTAR_DEBUG
//      edfstar_printf2("I%d",p);
#endif
      /* set the request time */
      EDFSTAR_internal_activate(lev,p,iq_query_timespec(p, &lev->ready));

      event_need_reschedule();
      break;

    default:
  #ifdef EDFSTAR_DEBUG
      kern_printf("(E:Dl:%d)",p);
  #endif
      /* else, a deadline miss occurred!!! */
      lev->dline_miss[p]++;

      /* the task is into another state */
      lev->nact[p]++;

      /* Set the deadline timer */
      ADDUSEC2TIMESPEC(lev->period[p], &lev->deadline_timespec[p]);
  }

  /* Set the deadline timer */
  lev->deadline_timer[p] = kern_event_post(&lev->deadline_timespec[p],
                                           EDFSTAR_timer_deadline,
                                           (void *)p);

#ifdef EDFSTAR_DEBUG
//  edfstar_printf(")");
#endif
}

static void EDFSTAR_timer_guest_deadline(void *par)
{
  PID p = (PID) par;

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:gdl)");
#endif

  kern_raise(XDEADLINE_MISS,p);
}

static int EDFSTAR_public_create(LEVEL l, PID p, TASK_MODEL *m)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  /* if the EDFSTAR_task_create is called, then the pclass must be a
     valid pclass. */
  HARD_TASK_MODEL *h;

  if (m->pclass != HARD_PCLASS) return -1;
  if (m->level != 0 && m->level != l) return -1;
  h = (HARD_TASK_MODEL *)m;
  if (!h->wcet || !h->mit || h->periodicity != PERIODIC) return -1;
  /* now we know that m is a valid model */


#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:tcr)");
#endif

  lev->period[p] = h->mit;

  lev->flag[p] = 0;
  lev->deadline_timer[p] = -1;
  lev->dline_miss[p]     = 0;
  lev->wcet_miss[p]      = 0;
  lev->nact[p]           = 0;

  /* Enable wcet check */
  proc_table[p].avail_time = h->wcet;
  proc_table[p].wcet       = h->wcet;
  proc_table[p].control |= CONTROL_CAP;

  return 0; /* OK, also if the task cannot be guaranteed... */
}

static int EDFSTAR_public_eligible(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  edfstar_printf2("(E:eli)");
#endif

  return level_table[ lev->scheduling_level ]->
    private_eligible(lev->scheduling_level,p);
}

static void EDFSTAR_public_dispatch(LEVEL l, PID p, int nostop)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:dis)");

  edfstar_printf3("(%d %d)",
                  iq_query_timespec(p, &lev->ready)->tv_nsec/1000000,
                  schedule_time.tv_nsec/1000000);
#endif

  level_table[ lev->scheduling_level ]->
    private_dispatch(lev->scheduling_level,p,nostop);
}

static void EDFSTAR_public_epilogue(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:epi ");
#endif

  /* check if the wcet is finished... */
  if (proc_table[p].avail_time <= 0 && proc_table[p].control&CONTROL_CAP) {
    /* wcet finished: disable wcet event and count wcet miss */
#ifdef EDFSTAR_DEBUG
    edfstar_printf2("W%d",p);
#endif
    proc_table[p].control &= ~CONTROL_CAP;
    lev->wcet_miss[p]++;
  }
#ifdef EDFSTAR_DEBUG
  edfstar_printf(")");
#endif

  level_table[ lev->scheduling_level ]->
    private_epilogue(lev->scheduling_level,p);

  proc_table[p].status = EDFSTAR_READY;
}

static void EDFSTAR_public_activate(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  struct timespec t;

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:act:%d)",p);
  #endif

  /* Test if we are trying to activate a non sleeping task    */
  /* save activation (only if needed... */
  if (proc_table[p].status != SLEEP) {
    /* a periodic task cannot be activated when it is already active */
    kern_raise(XACTIVATION,p);
    return;
  }

  kern_gettime(&t);

  EDFSTAR_internal_activate(lev,p, &t);

  /* Set the deadline timer */
  lev->deadline_timer[p] = kern_event_post(&lev->deadline_timespec[p],
                                           EDFSTAR_timer_deadline,
                                           (void *)p);

}

static void EDFSTAR_public_unblock(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:ins)");
#endif

  /* Insert task in the correct position */
  proc_table[p].status = EDFSTAR_READY;
  iq_timespec_insert(p,&lev->ready);

  /* and check for preemption! */
  EDFSTAR_check_preemption(lev);
}

static void EDFSTAR_public_block(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:ext)");
#endif

  /* the task is blocked on a synchronization primitive. we have to
     remove it from the master module -and- from the local queue! */
  iq_extract(p,&lev->ready);

  /* and finally, a preemption check! (it will also call guest_end) */
  EDFSTAR_check_preemption(lev);
}

static int EDFSTAR_public_message(LEVEL l, PID p, void *m)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  struct timespec temp;

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:ecy ");
#endif

  switch ((long)(m)) {

    /* Task EndCycle */
  case (long)(NULL):

    /* we call guest_end directly here because the same task may
       be reinserted in the queue before calling the preemption check! */
    level_table[ lev->scheduling_level ]->
      private_extract(lev->scheduling_level,p);
    lev->activated = NIL;
 
    iq_extract(p,&lev->ready);
 
    /* we reset the capacity counters... */
    proc_table[p].avail_time = proc_table[p].wcet;
 
    if (lev->nact[p] > 0) {
   
      #ifdef EDFSTAR_DEBUG
        edfstar_printf2("E%d",p);
      #endif
     
      /* Pending activation: reactivate the thread!!! */
      lev->nact[p]--;
     
      /* see also EDFSTAR_timer_deadline */
      kern_gettime(&temp);
     
      EDFSTAR_internal_activate(lev,p, &temp);
     
      /* check if the deadline has already expired */
      temp = *iq_query_timespec(p, &lev->ready);
      if (TIMESPEC_A_LT_B(&temp, &schedule_time)) {
          /* count the deadline miss */
          lev->dline_miss[p]++;
          kern_event_delete(lev->deadline_timer[p]);
      }
     
    } else {

      #ifdef EDFSTAR_DEBUG
        edfstar_printf("e%d",p);
      #endif
     
      /* the task has terminated his job before it consume the wcet. All OK! */
      proc_table[p].status = EDFSTAR_IDLE;
     
      /* and finally, a preemption check! */
      EDFSTAR_check_preemption(lev);
     
      /* when the deadline timer fire, it recognize the situation and set
         correctly all the stuffs (like reactivation, etc... ) */
    }
#ifdef EDFSTAR_DEBUG
    edfstar_printf(")");
#endif
   
    jet_update_endcycle(); /* Update the Jet data... */
    trc_logevent(TRC_ENDCYCLE,&exec_shadow); /* tracer stuff */
    break;

  default:

    break;
     
  }
  return 0;
}

static void EDFSTAR_public_end(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  edfstar_printf("(E:end)");
#endif

  iq_extract(p,&lev->ready);

  /* we finally put the task in the ready queue */
  proc_table[p].status = FREE;
 
  iq_insertfirst(p,&freedesc);
 
  if (lev->deadline_timer[p] != -1) {
    kern_event_delete(lev->deadline_timer[p]);
  }

  /* and finally, a preemption check! (it will also call guest_end) */
  EDFSTAR_check_preemption(lev);
}

/* Guest Functions
   These functions manages a JOB_TASK_MODEL, that is used to put
   a guest task in the EDFSTAR ready queue. */

static void EDFSTAR_private_insert(LEVEL l, PID p, TASK_MODEL *m)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  JOB_TASK_MODEL *job;

  if (m->pclass != JOB_PCLASS || (m->level != 0 && m->level != l) ) {
    kern_raise(XINVALID_TASK, p);
    return;
  }

  job = (JOB_TASK_MODEL *)m;

  /* if the EDFSTAR_guest_create is called, then the pclass must be a
     valid pclass. */

  *iq_query_timespec(p, &lev->ready) = job->deadline;
 
  lev->deadline_timer[p] = -1;
  lev->dline_miss[p]     = 0;
  lev->wcet_miss[p]      = 0;
  lev->nact[p]           = 0;

  if (job->noraiseexc)
    lev->flag[p] |= EDFSTAR_FLAG_NORAISEEXC;
  else {
    lev->flag[p] &= ~EDFSTAR_FLAG_NORAISEEXC;
    lev->deadline_timer[p] = kern_event_post(iq_query_timespec(p, &lev->ready),
                                             EDFSTAR_timer_guest_deadline,
                                             (void *)p);
  }

  lev->period[p] = job->period;

  /* Insert task in the correct position */
  iq_timespec_insert(p,&lev->ready);
  proc_table[p].status = EDFSTAR_READY;

  /* check for preemption */
  EDFSTAR_check_preemption(lev);

  /* there is no bandwidth guarantee at this level, it is performed
     by the level that inserts guest tasks... */
}

static void EDFSTAR_private_dispatch(LEVEL l, PID p, int nostop)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  level_table[ lev->scheduling_level ]->
    private_dispatch(lev->scheduling_level,p,nostop);
}

static void EDFSTAR_private_epilogue(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  /* the task has been preempted. it returns into the ready queue... */
  level_table[ lev->scheduling_level ]->
    private_epilogue(lev->scheduling_level,p);

  proc_table[p].status = EDFSTAR_READY;
}

static void EDFSTAR_private_extract(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  //kern_printf("EDFSTAR_guest_end: dline timer %d\n",lev->deadline_timer[p]);
#endif

  iq_extract(p, &lev->ready);

  /* we remove the deadline timer, because the slice is finished */
  if (lev->deadline_timer[p] != NIL) {
#ifdef EDFSTAR_DEBUG
//    kern_printf("EDFSTAR_guest_end: dline timer %d\n",lev->deadline_timer[p]);
#endif
    kern_event_delete(lev->deadline_timer[p]);
    lev->deadline_timer[p] = NIL;
  }

  /* and finally, a preemption check! (it will also call guest_end() */
  EDFSTAR_check_preemption(lev);
}

/* Registration functions */

/* Registration function:
    int flags                 the init flags ... see EDFSTAR.h */

LEVEL EDFSTAR_register_level(int master)
{
  LEVEL l;            /* the level that we register */
  EDFSTAR_level_des *lev;  /* for readableness only */
  PID i;              /* a counter */

#ifdef EDFSTAR_DEBUG
  printk("EDFSTAR_register_level\n");
#endif

  /* request an entry in the level_table */
  l = level_alloc_descriptor(sizeof(EDFSTAR_level_des));

  lev = (EDFSTAR_level_des *)level_table[l];

  printk("    lev=%d\n",(int)lev);

  /* fill the standard descriptor */
  lev->l.private_insert   = EDFSTAR_private_insert;
  lev->l.private_extract  = EDFSTAR_private_extract;
  lev->l.private_dispatch = EDFSTAR_private_dispatch;
  lev->l.private_epilogue = EDFSTAR_private_epilogue;

  lev->l.public_guarantee = NULL;
  lev->l.public_eligible  = EDFSTAR_public_eligible;
  lev->l.public_create    = EDFSTAR_public_create;
  lev->l.public_end       = EDFSTAR_public_end;
  lev->l.public_dispatch  = EDFSTAR_public_dispatch;
  lev->l.public_epilogue  = EDFSTAR_public_epilogue;
  lev->l.public_activate  = EDFSTAR_public_activate;
  lev->l.public_unblock   = EDFSTAR_public_unblock;
  lev->l.public_block     = EDFSTAR_public_block;
  lev->l.public_message   = EDFSTAR_public_message;

  /* fill the EDFSTAR descriptor part */
  for(i=0; i<MAX_PROC; i++) {
    lev->period[i]         = 0;
    lev->deadline_timer[i] = -1;
    lev->flag[i]           = 0;
    lev->dline_miss[i]     = 0;
    lev->wcet_miss[i]      = 0;
    lev->nact[i]           = 0;
    lev->budget[i]         = NIL;
    lev->wcet[i]           = 0;
  }

  iq_init(&lev->ready, NULL, IQUEUE_NO_PRIORITY);
  lev->activated = NIL;

  lev->scheduling_level = master;

  return l;
}

int EDFSTAR_get_dline_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
 
  return lev->dline_miss[p];
}

int EDFSTAR_get_wcet_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  return lev->wcet_miss[p];
}

int EDFSTAR_get_nact(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  return lev->nact[p];
}

int EDFSTAR_reset_dline_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  lev->dline_miss[p] = 0;
  return 0;
}

int EDFSTAR_reset_wcet_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  lev->wcet_miss[p] = 0;
  return 0;
}

int EDFSTAR_setbudget(LEVEL l, PID p, int budget)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
 
  lev->budget[p] = budget;

  return 0;

}

int EDFSTAR_getbudget(LEVEL l, PID p)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  return lev->budget[p];

}

int EDFSTAR_budget_has_thread(LEVEL l, int budget)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  int i;

  for(i = 0; i< MAX_PROC; i++)
    if (lev->budget[i] == budget) return 1;

  return 0;

}