resources/disk_bfq/mngr/diskbfq_mngr.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 FRESCOR BFQ disk bandwidth reservation           */
  26 /*                                                                        */
  27 /* FWP 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.  FWP 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 FWP; 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 FWP header files in a file,          */
  39 /* instantiating FWP generics or templates, or linking other files        */
  40 /* with FWP 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 #include <frm_generic.h>
  47 #include <forb.h>
  48 #include <error.h>
  49 #include <errno.h>
  50 #include <fres_sa_scenario.h>
  51 #include <stdbool.h>
  52 #include <ul_log.h>
  53 #include <stdio.h>
  54 #include <math.h>
  55
  56 #include "res_disk.h"
  57 #include "diskbfq_th.h"
  58
  59 UL_LOG_CUST(ulogd_frm_diskbfq);
  60 ul_log_domain_t ulogd_fra_fwp = {UL_LOGL_DEB, "frm_diskbfq"};
  61
  62 struct disk_data {
  63         double throughput;
  64         int budget;
  65 };
  66
  67 static int bandwidth_to_weight(struct disk_data *data,
  68                                unsigned long budget,
  69                                unsigned long period)
  70 {
  71         float th, weight;
  72         //unsigned long q_i, p_i;
  73         int b, w_i;
  74
  75         th = data->throughput / 10E9;
  76         b = data->budget;
  77
  78         /* Convert to nanoseconds! */
  79         //q_i *= 1000;
  80         //p_i *= 1000;
  81
  82         if (period < b / th)
  83                 return -1;
  84
  85         weight = (DISKBFQ_WEIGHT_MAX * budget) / (period - (b / th));
  86         w_i = floor(weight);
  87         if (weight != w_i)
  88                 w_i++;
  89
  90         return w_i;
  91 }
  92
  93 static int weight_to_ioprio(int weight)
  94 {
  95         if (weight >= 100)
  96                 return 0;
  97         if (weight > 14)
  98                 return 1;
  99         if (weight == -1)
 100                 return -1;
 101
 102         return 8 - 1 - (weight / 2);
 103 }
 104
 105 static int diskbfq_admtest(struct fres_sa_scenario *scenario,
 106                            void *priv,
 107                            bool *schedulable)
 108 {
 109         struct fres_sa_contract *c;
 110         long int period, budget;
 111         long int sum_weight = 0;
 112         struct disk_data *data = priv;
 113
 114         fres_sa_scenario_for_each_contract(scenario, c) {
 115                 fres_block_basic *basic;
 116                 fres_block_resource *resource;
 117                 fres_block_disk_sched *disk_sched;
 118                 char id[40];
 119
 120                 fres_contract_id_to_string(id, &c->contract->id, sizeof(id));
 121                 basic = fres_contract_get_basic(c->contract);
 122                 resource = fres_contract_get_resource(c->contract);
 123
 124                 period = fosa_rel_time_to_nsec(basic->period);
 125                 budget = fosa_rel_time_to_nsec(basic->budget);
 126                 printf("processing: id=%s, period=%ld ms, budget=%ld ms\n",
 127                        id, period, budget);
 128
 129                 if (c->contract == c->new) {
 130                         int ret;
 131
 132                         /* Calc weight and IO prio for the new contracts */
 133                         disk_sched = malloc(sizeof(*disk_sched));
 134                         if (!disk_sched) return -1;
 135
 136                         disk_sched->weight = bandwidth_to_weight(data,
 137                                                                  budget,
 138                                                                  period);
 139                         disk_sched->ioprio =
 140                                 weight_to_ioprio(disk_sched->weight);
 141                         ret = fres_contract_add_disk_sched(c->contract,
 142                                                            disk_sched);
 143                         if (ret) {
 144                                 fprintf(stderr,
 145                                         "Cannot add disk_sched block\n");
 146                                 free(disk_sched);
 147                                 return -1;
 148                         }
 149                 } else {
 150                         disk_sched = fres_contract_get_disk_sched(c->contract);
 151                         if (!disk_sched) {
 152                                 fprintf(stderr,
 153                                         "disk_sched is not present\n");
 154                                 return -1;
 155                         }
 156                 }
 157                 sum_weight += disk_sched->weight;
 158         }
 159
 160         printf("sum_weight=%ld, max_weight=%d\n",
 161                sum_weight,
 162                DISKBFQ_WEIGHT_MAX);
 163         *schedulable = sum_weight < DISKBFQ_WEIGHT_MAX;
 164         printf("=>%s\n", (*schedulable) ? "schedulable" : "not schedulable");
 165
 166         return 0;
 167 }
 168
 169 struct disk_data disk;
 170
 171 static struct fres_res_manager frm = {
 172         .res_type = FRSH_RT_DISK,
 173         .res_id = 0,
 174         .name = "Disk BFQ",
 175         .admission_test = diskbfq_admtest,
 176         .priv = &disk
 177 };
 178
 179 int main(int argc, char *argv[])
 180 {
 181         forb_init_attr_t attr = { .orb_id = "org.frescor.frm.diskbfq" };
 182         forb_orb orb;
 183         FILE* fd;
 184         int readers = 4;
 185         char disk_dev[40], *disk_name = disk_dev;
 186         char path[128], scheduler[128];
 187         int i, opt, ret;
 188
 189         if (argc < 3) {
 190 err_usage:
 191                 fprintf(stderr, "Usage:\n %s "
 192                         "-i disk_id -d disk_device [-n readers]\n",
 193                         argv[0]);
 194                 error(1, EINVAL, "frm_diskbfq");
 195         }
 196
 197         while ((opt = getopt(argc, argv, "i:d:n:")) != -1) {
 198                 switch (opt) {
 199                         case 'i':
 200                                 frm.res_id = atoi(optarg);
 201                                 break;
 202                         case 'd':
 203                                 strncpy(disk_dev,
 204                                         optarg, sizeof(disk_dev));
 205
 206                                 for (i = strlen(disk_dev) - 1; i >= 0; i--) {
 207                                         if (disk_dev[i] == '/') {
 208                                                 disk_name = disk_dev + i + 1;
 209                                                 break;
 210                                         }
 211                                 }
 212                                 break;
 213                         case 'n':
 214                                 readers = atoi(optarg);
 215                                 break;
 216                         default:
 217                                 goto err_usage;
 218                 }
 219         }
 220
 221         orb = forb_init(&argc, &argv, &attr);
 222         if (!orb) error(1, errno, "forb_init");
 223
 224         /*
 225          * Check if BFQ I/O scheduler is available
 226          *  in the system.
 227          */
 228         snprintf(path, 128,
 229                  SYSFS_BFQ_SCHED_PATH,
 230                  disk_name);
 231         if (!(fd = fopen(path, "rw")))
 232                 error(1, errno, "frm_generic_run");
 233
 234         ret = fscanf(fd, "%128[^\n]", scheduler);
 235         if (ret == 0 || ret == EOF)
 236                 error(1, errno, "frm_generic_run");
 237         if (!strstr(scheduler, SYSFS_BFQ_NAME))
 238                 error(1, errno, "frm_generic_run");
 239
 240         /**
 241          * Now check if BFQ is the default I/O scheduler
 242          *  for the specific disk and set it to so if not.
 243          **/
 244         else if (!strstr(scheduler, "[" SYSFS_BFQ_NAME "]")) {
 245                 rewind(fd);
 246                 ret = fprintf(fd, SYSFS_BFQ_NAME);
 247                 if (ret && ret != EOF)
 248                         ret = fflush(fd);
 249                 if (ret == 0 || ret == EOF)
 250                         error(1, errno, "frm_generic_run");
 251         }
 252         fclose(fd);
 253
 254         snprintf(path, sizeof(path),
 255                  SYSFS_BFQ_BUDGET_PATH,
 256                  disk_name);
 257         if (!(fd = fopen(path, "r")))
 258                 error(1, errno, "frm_generic_run");
 259
 260         ret = fscanf(fd, "%d", &disk.budget);
 261         if (ret == 0 || ret == EOF)
 262                 error(1, errno, "frm_generic_run");
 263         fclose(fd);
 264
 265         /**
 266          * Estimate the disk throughput in order to achieve
 267          *  effective weight assignement during runtime.
 268          **/
 269         disk.throughput = estimate_throughput(disk_dev, readers);
 270         if (disk.throughput < 0.0)
 271                 error(1, errno, "frm_generic_run");
 272
 273         fprintf(stdout, "disk name: %s\n"
 274                 "scheduler: %s\n"
 275                 "disk aggregate throughput: %f\n"
 276                 "scheduling budget: %d\n",
 277                 disk_name, SYSFS_BFQ_NAME, disk.throughput, disk.budget);
 278
 279         /* Register fres_block_disk_sched to contract handling
 280          *  functions */
 281         fres_block_register_disk_sched();
 282
 283         ret = frm_register_and_run(orb, &frm);
 284         if (ret) error(1, errno, "frm_generic_run");
 285
 286         return 0;
 287 }
 288