1 /**************************************************************************/
2 /* ---------------------------------------------------------------------- */
3 /* Copyright (C) 2006 - 2008 FRESCOR consortium partners: */
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 */
12 /* Thales Communication S.A. FRANCE */
13 /* Visual Tools S.A. SPAIN */
14 /* Rapita Systems Ltd UK */
17 /* See http://www.frescor.org for a link to partners' websites */
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. */
25 /* This file is part of FRESCOR BFQ disk bandwidth reservation */
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. */
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>
50 #include <fres_sa_scenario.h>
53 #include <ul_logreg.h>
58 #include "diskbfq_th.h"
60 UL_LOG_CUST(ulogd_frm_diskbfq);
61 ul_log_domain_t ulogd_fra_fwp = {UL_LOGL_DEB, "frm_diskbfq"};
68 static int bandwidth_to_weight(struct disk_data *data,
73 //unsigned long q_i, p_i;
76 th = data->throughput / 10E9;
79 /* Convert to nanoseconds! */
86 weight = (DISKBFQ_WEIGHT_MAX * budget) / (period - (b / th));
94 static int weight_to_ioprio(int weight)
103 return 8 - 1 - (weight / 2);
106 static int diskbfq_admtest(struct fres_sa_scenario *scenario,
110 struct fres_sa_contract *c;
111 long int period, budget;
112 long int sum_weight = 0;
113 struct disk_data *data = priv;
115 fres_sa_scenario_for_each_no_cancel_contract(scenario, c) {
116 fres_block_basic *basic;
117 fres_block_resource *resource;
118 fres_block_disk_sched *disk_sched;
121 fres_contract_id_to_string(id, &c->contract->id, sizeof(id));
122 basic = fres_contract_get_basic(c->contract);
123 resource = fres_contract_get_resource(c->contract);
125 period = fosa_rel_time_to_nsec(basic->period);
126 budget = fosa_rel_time_to_nsec(basic->budget);
127 printf("processing: id=%s, period=%ld ms, budget=%ld ms\n",
130 if (c->contract == c->new) {
133 /* Calc weight and IO prio for the new contracts */
134 disk_sched = malloc(sizeof(*disk_sched));
135 if (!disk_sched) return -1;
137 disk_sched->weight = bandwidth_to_weight(data,
141 weight_to_ioprio(disk_sched->weight);
142 ret = fres_contract_add_disk_sched(c->contract,
146 "Cannot add disk_sched block\n");
151 disk_sched = fres_contract_get_disk_sched(c->contract);
154 "disk_sched is not present\n");
158 sum_weight += disk_sched->weight;
161 printf("sum_weight=%ld, max_weight=%d\n",
164 *schedulable = sum_weight < DISKBFQ_WEIGHT_MAX;
165 printf("=>%s\n", (*schedulable) ? "schedulable" : "not schedulable");
170 struct disk_data disk;
172 static struct fres_res_manager frm = {
173 .res_type = FRSH_RT_DISK,
176 .admission_test = diskbfq_admtest,
180 int main(int argc, char *argv[])
182 forb_init_attr_t attr = { .orb_id = "org.frescor.frm.diskbfq" };
186 char disk_dev[40], *disk_name = disk_dev;
187 char path[128], scheduler[128];
190 ul_logreg_domain(&ulogd_frm_diskbfq);
194 fprintf(stderr, "Usage:\n %s "
195 "-i disk_id -d disk_device [-n readers]\n",
197 error(1, EINVAL, "frm_diskbfq");
200 while ((opt = getopt(argc, argv, "i:d:n:")) != -1) {
203 frm.res_id = atoi(optarg);
207 optarg, sizeof(disk_dev));
209 for (i = strlen(disk_dev) - 1; i >= 0; i--) {
210 if (disk_dev[i] == '/') {
211 disk_name = disk_dev + i + 1;
217 readers = atoi(optarg);
224 orb = forb_init(&argc, &argv, &attr);
225 if (!orb) error(1, errno, "forb_init");
228 * Check if BFQ I/O scheduler is available
232 SYSFS_BFQ_SCHED_PATH,
234 if (!(fd = fopen(path, "r+")))
235 error(1, errno, "frm_generic_run");
237 ret = fscanf(fd, "%128[^\n]", scheduler);
238 if (ret == 0 || ret == EOF)
239 error(1, errno, "frm_generic_run");
240 if (!strstr(scheduler, SYSFS_BFQ_NAME))
241 error(1, errno, "frm_generic_run");
244 * Now check if BFQ is the default I/O scheduler
245 * for the specific disk and set it to so if not.
247 else if (!strstr(scheduler, "[" SYSFS_BFQ_NAME "]")) {
249 ret = fprintf(fd, SYSFS_BFQ_NAME);
250 if (ret && ret != EOF)
252 if (ret == 0 || ret == EOF)
253 error(1, errno, "frm_generic_run");
257 snprintf(path, sizeof(path),
258 SYSFS_BFQ_BUDGET_PATH,
260 if (!(fd = fopen(path, "r+")))
261 error(1, errno, "frm_generic_run");
263 ret = fprintf(fd, "256");
265 error(1, errno, "frm_generic_run");
268 ret = fscanf(fd, "%d", &disk.budget);
269 if (ret == 0 || ret == EOF)
270 error(1, errno, "frm_generic_run");
274 * Estimate the disk throughput in order to achieve
275 * effective weight assignement during runtime.
277 disk.throughput = estimate_throughput(disk_dev, readers);
278 if (disk.throughput < 0.0)
279 error(1, errno, "frm_generic_run");
281 fprintf(stdout, "disk name: %s\n"
283 "disk aggregate throughput: %f\n"
284 "scheduling budget: %d\n",
285 disk_name, SYSFS_BFQ_NAME, disk.throughput, disk.budget);
287 /* Register fres_block_disk_sched to contract handling
289 fres_block_register_disk_sched();
291 ret = frm_register_and_run(orb, &frm);
292 if (ret) error(1, errno, "frm_generic_run");