rres_init.c

Go to the documentation of this file.

/*
 * Copyright (C) 2002 Luca Abeni, 2003 Luca Marzario
 * This is Free Software; see GPL.txt for details
 */
#ifdef QOS_KS
//#include <linux/autoconf.h>
  #ifndef CONFIG_GENERIC_SCHEDULER
    #error "The Generic Scheduler is not enabled !"
  #endif
//#ifdef CONFIG_SMP
//#  error "AQuoSA is not (yet) ready for an SMP-enabled kernel !"
//#endif
  #include <linux/kernel.h>
  #include <linux/version.h>
  #include <linux/module.h>
  #include <linux/moduleparam.h>
  #include <linux/spinlock.h>
  #include <linux/sched.h>
#else
  #include "kal_generic.h"
  #define module_init(func) int main_func(void) { return func(); }
  #define module_exit(x) 
  #define MODULE_LICENSE(x)
#endif

#include "rres_config.h"
#define QOS_DEBUG_LEVEL RRES_MOD_DEBUG_LEVEL
#include "qos_debug.h"

#include "rres.h"
#include "rres_hooks.h"
#include "rres_proc_fs.h"

#include <linux/version.h>

#ifdef GENSCHED_VERSION_CODE
/* Check gensched patch version with GENSCHED_VERSION macro */
# if GENSCHED_VERSION_CODE < GENSCHED_VERSION(2,0)
#  error "Need a kernel patched with a generic scheduler patch with version 2.x"
# endif
#else
/* Get UTS_RELEASE definition */
# if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,21)
#  include <linux/utsrelease.h>
# endif
#endif

int rres_mod_removed = 0;

int rres_init_module(void) {
  unsigned long flags;
  int err = 0;
/*   kal_task_t *t; */

#ifndef GENSCHED_VERSION_CODE
  if (strstr(UTS_RELEASE, "gs-2.") == NULL) {
    qos_log_crit("Need a kernel patched with a generic scheduler patch with version gs-2.x");
    return -1;
  }
#endif

/* #ifdef QOS_KS */
/*   read_lock(&tasklist_lock); */
/*   for_each_task(t) { */
/*     if (rres_find_by_task(t) != NULL) { */
/*       qos_log_err("Some tasks have private_data != NULL !!!"); */
/*       err = 1; */
/*       goto out_for_each; */
/*     } */
/*   } */
/*  out_for_each: */
/*   read_unlock(&tasklist_lock); */
/*   if (err) */
/*     return -1; */
/* #endif */

  kal_spin_lock_irqsave(rres_get_spinlock(), &flags);

  qos_log_debug("Initializing rres module... ");
  rres_sample_time();
  if (rres_init() != QOS_OK) {
    qos_log_err("rres_init() failed");
    err = 1;
  }

  kal_spin_unlock_irqrestore(rres_get_spinlock(), &flags);

  if (err)
    return -1;

  write_lock(&hook_lock);
  if (! check_gensched_hooks()) {
    qos_log_crit("Hooks already initialized !!!");
    err = 1;
    goto out_hooks;
  }
  set_gensched_hooks();
 out_hooks:
  write_unlock(&hook_lock);

  if (err) {
    kal_spin_lock_irqsave(rres_get_spinlock(), &flags);
    rres_cleanup();
    kal_spin_unlock_irqrestore(rres_get_spinlock(), &flags);
    return -1;
  }

  qos_log_debug("Hooks initialized");

  if (rres_proc_register())
    qos_log_err("Failed to register rres module in /proc filesystem");

  qos_log_debug("rres module initialization finished");

  return 0;
}

void rres_cleanup_module(void) {
  unsigned long flags;

  rres_mod_removed = 1;
  rres_proc_unregister();

  qos_log_debug("Unsetting hooks... ");
  write_lock(&hook_lock);
  unset_gensched_hooks();
  write_unlock(&hook_lock);

  qos_log_debug("Cleaning up module... ");
  kal_spin_lock_irqsave(rres_get_spinlock(), &flags);
  qos_chk_ok(rres_cleanup());

  if (! qos_mem_clean()) {
    qos_log_err("MEMORY NOT CLEAN AFTER MODULE CLEANUP");
  } else {
    qos_log_info("Memory clean");
  }

  kal_spin_unlock_irqrestore(rres_get_spinlock(), &flags);
  qos_log_debug("rres module removed from kernel");
}

module_init(rres_init_module);
module_exit(rres_cleanup_module);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("ReTiS Lab");
MODULE_DESCRIPTION("AQuoSA Resource Reservation - Core CPU Scheduler");

---