src/frsh/resources/dummy/frm_dummy.c

   1 /**************************************************************************/
   2 /* ---------------------------------------------------------------------- */
   3 /* Copyright (C) 2006 - 2008 FRESCOR consortium partners:                 */
   4 /*                                                                        */
   5 /*   Universidad de Cantabria,              SPAIN                         */
   6 /*   University of York,                    UK                            */
   7 /*   Scuola Superiore Sant'Anna,            ITALY                         */
   8 /*   Kaiserslautern University,             GERMANY                       */
   9 /*   Univ. Politécnica  Valencia,           SPAIN                        */
  10 /*   Czech Technical University in Prague,  CZECH REPUBLIC                */
  11 /*   ENEA                                   SWEDEN                        */
  12 /*   Thales Communication S.A.              FRANCE                        */
  13 /*   Visual Tools S.A.                      SPAIN                         */
  14 /*   Rapita Systems Ltd                     UK                            */
  15 /*   Evidence                               ITALY                         */
  16 /*                                                                        */
  17 /*   See http://www.frescor.org for a link to partners' websites          */
  18 /*                                                                        */
  19 /*          FRESCOR project (FP6/2005/IST/5-034026) is funded             */
  20 /*       in part by the European Union Sixth Framework Programme          */
  21 /*       The European Union is not liable of any use that may be          */
  22 /*       made of this code.                                               */
  23 /*                                                                        */
  24 /*                                                                        */
  25 /*  This file is part of FRSH (FRescor ScHeduler)                         */
  26 /*                                                                        */
  27 /* FRSH is free software; you can redistribute it and/or modify it        */
  28 /* under terms of the GNU General Public License as published by the      */
  29 /* Free Software Foundation; either version 2, or (at your option) any    */
  30 /* later version.  FRSH is distributed in the hope that it will be        */
  31 /* useful, but WITHOUT ANY WARRANTY; without even the implied warranty    */
  32 /* of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU    */
  33 /* General Public License for more details. You should have received a    */
  34 /* copy of the GNU General Public License along with FRSH; see file       */
  35 /* COPYING. If not, write to the Free Software Foundation, 675 Mass Ave,  */
  36 /* Cambridge, MA 02139, USA.                                              */
  37 /*                                                                        */
  38 /* As a special exception, including FRSH header files in a file,         */
  39 /* instantiating FRSH generics or templates, or linking other files       */
  40 /* with FRSH objects to produce an executable application, does not       */
  41 /* by itself cause the resulting executable application to be covered     */
  42 /* by the GNU General Public License. This exception does not             */
  43 /* however invalidate any other reasons why the executable file might be  */
  44 /* covered by the GNU Public License.                                     */
  45 /**************************************************************************/
  46
  47 /**
  48  * @file   frm_dummy.c
  49  * @author Michal Sojka <sojkam1@fel.cvut.cz>
  50  * @date   Wed Feb 18 16:06:04 2009
  51  *
  52  * @brief  Dummy resource manager implementation
  53  *
  54  *
  55  */
  56
  57
  58 #include <frm_generic.h>
  59 #include <forb.h>
  60 #include <error.h>
  61 #include <errno.h>
  62 #include <getopt.h>
  63 #include <fres_sa_scenario.h>
  64 #include <stdbool.h>
  65 #include <ul_log.h>
  66 #include <ul_logreg.h>
  67 #include "res_dummy.h"
  68 #include <stdio.h>
  69 #include "dummy_config.h"
  70
  71 bool opt_accept_all = false;
  72 bool opt_daemon = false;
  73 char *opt_pidfile = NULL;
  74 int opt_res_id = DUMMY_RESOURCE_ID;
  75
  76 struct dummy_data {
  77         int some_data;
  78 };
  79
  80 int admission_test(struct fres_sa_scenario *scenario, void *priv, bool *schedulable)
  81 {
  82         struct dummy_data *data = priv;
  83         struct fres_sa_contract *c;
  84         int ret;
  85         long int utilization = 0;
  86
  87 #ifdef CONFIG_RESOURCE_DUMMY_VERBOSE
  88         printf("Admission test:\n");
  89 #endif
  90
  91         if (opt_accept_all) {
  92                 *schedulable = true;
  93                 return 0;
  94         }
  95
  96         data->some_data++;
  97
  98         fres_sa_scenario_for_each_no_cancel_contract(scenario, c) {
  99                 fres_block_basic *basic;
 100                 fres_block_dummy_sched *dummy_sched;
 101                 char id[40];
 102                 fres_contract_id_to_string(id, &c->id, sizeof(id));
 103 #ifdef CONFIG_RESOURCE_DUMMY_VERBOSE
 104                 printf("  %s contract: id=%s num_blocks=%d\n",
 105                        c->contract == c->new ? "new" : "old", id,
 106                        fres_contract_get_num_blocks(c->contract));
 107 #endif
 108
 109                 basic = fres_contract_get_basic(c->contract);
 110                 if (!basic) {
 111                         fprintf(stderr, "No basic block present\n");
 112                         return -1;
 113                 }
 114
 115                 if (c->contract == c->new) {
 116                         /* Add data for scheduler to the new contracts */
 117                         dummy_sched = malloc(sizeof(*dummy_sched));
 118                         if (!dummy_sched) return -1;
 119                         dummy_sched->priority = 100 - fosa_rel_time_to_msec(basic->budget);
 120                         ret = fres_contract_add_dummy_sched(c->contract, dummy_sched);
 121                         if (ret) {
 122                                 fprintf(stderr, "Cannot add dummy_sched block\n");
 123                                 return -1;
 124                         }
 125                 } else {
 126                         dummy_sched = fres_contract_get_dummy_sched(c->contract);
 127                         if (!dummy_sched) {
 128                                 fprintf(stderr, "Dummy_sched is not present\n");
 129                                 return -1;
 130                         }
 131                 }
 132 #ifdef CONFIG_RESOURCE_DUMMY_VERBOSE
 133                 printf("                period=%ld ms, budget=%ld ms, priority=%d\n",
 134                        fosa_rel_time_to_msec(basic->period),
 135                        fosa_rel_time_to_msec(basic->budget), dummy_sched->priority);
 136 #endif
 137                 utilization +=
 138                         1000*fosa_rel_time_to_msec(basic->budget) /
 139                         fosa_rel_time_to_msec(basic->period);
 140         }
 141         *schedulable = utilization < 1000;
 142         if (*schedulable) {
 143                 scenario->utilization = utilization/10;
 144         }
 145
 146 #ifdef CONFIG_RESOURCE_DUMMY_VERBOSE
 147         printf("utilization=%ld.%03ld => %s\n", utilization/1000, utilization%1000,
 148                *schedulable?"schedulable":"not schedulable");
 149 #endif
 150
 151         return 0;
 152 }
 153
 154 struct dummy_data dummy_data;
 155
 156 static struct fres_res_manager frm = {
 157         .res_type = DUMMY_RESOURCE_TYPE,
 158         .res_id = DUMMY_RESOURCE_ID,
 159         .admission_test = admission_test,
 160         .priv = &dummy_data,
 161         .name = "Dummy resource",
 162 };
 163
 164 static struct option long_opts[] = {
 165     { "loglevel", 1, 0, 'l' },
 166     { "id",       required_argument, 0, 'i' },
 167     { "accept-all", 0, 0, 'a' },
 168     { "help", 0, 0, 'h' },
 169     { 0, 0, 0, 0}
 170 };
 171
 172 static void
 173 usage(void)
 174 {
 175         printf("usage: frm_dummy [ options ]\n");
 176         printf("  -a, --accept-all   Accepts all contracts\n");
 177         printf("  -d, --daemon [pid-file]   go to background after FORB initialization\n");
 178         printf("  -i, --id <resource id>    Use a different id that the default (%d)\n", DUMMY_RESOURCE_ID);
 179         printf("  -h, --help         Display this help\n");
 180         printf("  -l, --loglevel <number>|<domain>=<number>,...\n");
 181 }
 182
 183 int forb_main(forb_orb orb, int argc, char *argv[])
 184 {
 185         int ret;
 186         int  opt;
 187
 188         while ((opt = getopt_long(argc, argv, "ai:l:h", &long_opts[0], NULL)) != EOF) {
 189                 switch (opt) {
 190                         case 'a':
 191                                 opt_accept_all = true;
 192                         case 'd':
 193                                 opt_daemon = true;
 194                                 opt_pidfile = optarg;
 195                                 break;
 196                         case 'i':
 197                                 opt_res_id = atoi(optarg);
 198                                 break;
 199                         case 'l':
 200                                 ul_log_domain_arg2levels(optarg);
 201                                 break;
 202                         case 'h':
 203                         /*default:*/
 204                                 usage();
 205                                 exit(opt == 'h' ? 0 : 1);
 206                 }
 207         }
 208
 209         if (opt_daemon)
 210                 forb_daemon_prepare(opt_pidfile);
 211
 212         /* Register fres_block_dummy_sched to contract handling
 213          * functions */
 214         fres_block_register_dummy();
 215
 216 #ifndef CONFIG_RESOURCE_DUMMY_VERBOSE
 217         ulogd_frm_generic.level = UL_LOGL_ERR;
 218 #endif
 219
 220         frm.res_id = opt_res_id;
 221         ret = frm_register_and_run(orb, &frm);
 222
 223         if (ret != 0) {
 224                 error(1, errno, "frm_generic_run failed");
 225         }
 226
 227         return 0;
 228 }