1
2
3
4
5
6
7#include <stdio.h>
8#include <stdlib.h>
9#include <string.h>
10#include <unistd.h>
11#include <getopt.h>
12#include <errno.h>
13
14#include "config.h"
15#include "system.h"
16#include "benchmark.h"
17
18static struct option long_options[] = {
19 {"output", 1, 0, 'o'},
20 {"sleep", 1, 0, 's'},
21 {"load", 1, 0, 'l'},
22 {"verbose", 0, 0, 'v'},
23 {"cpu", 1, 0, 'c'},
24 {"governor", 1, 0, 'g'},
25 {"prio", 1, 0, 'p'},
26 {"file", 1, 0, 'f'},
27 {"cycles", 1, 0, 'n'},
28 {"rounds", 1, 0, 'r'},
29 {"load-step", 1, 0, 'x'},
30 {"sleep-step", 1, 0, 'y'},
31 {"help", 0, 0, 'h'},
32 {0, 0, 0, 0}
33};
34
35
36
37
38
39void usage()
40{
41 printf("usage: ./bench\n");
42 printf("Options:\n");
43 printf(" -l, --load=<long int>\t\tinitial load time in us\n");
44 printf(" -s, --sleep=<long int>\t\tinitial sleep time in us\n");
45 printf(" -x, --load-step=<long int>\ttime to be added to load time, in us\n");
46 printf(" -y, --sleep-step=<long int>\ttime to be added to sleep time, in us\n");
47 printf(" -c, --cpu=<cpu #>\t\t\tCPU Nr. to use, starting at 0\n");
48 printf(" -p, --prio=<priority>\t\t\tscheduler priority, HIGH, LOW or DEFAULT\n");
49 printf(" -g, --governor=<governor>\t\tcpufreq governor to test\n");
50 printf(" -n, --cycles=<int>\t\t\tload/sleep cycles\n");
51 printf(" -r, --rounds<int>\t\t\tload/sleep rounds\n");
52 printf(" -f, --file=<configfile>\t\tconfig file to use\n");
53 printf(" -o, --output=<dir>\t\t\toutput path. Filename will be OUTPUTPATH/benchmark_TIMESTAMP.log\n");
54 printf(" -v, --verbose\t\t\t\tverbose output on/off\n");
55 printf(" -h, --help\t\t\t\tPrint this help screen\n");
56 exit(1);
57}
58
59
60
61
62
63int main(int argc, char **argv)
64{
65 int c;
66 int option_index = 0;
67 struct config *config = NULL;
68
69 config = prepare_default_config();
70
71 if (config == NULL)
72 return EXIT_FAILURE;
73
74 while (1) {
75 c = getopt_long (argc, argv, "hg:o:s:l:vc:p:f:n:r:x:y:",
76 long_options, &option_index);
77 if (c == -1)
78 break;
79
80 switch (c) {
81 case 'o':
82 if (config->output != NULL)
83 fclose(config->output);
84
85 config->output = prepare_output(optarg);
86
87 if (config->output == NULL)
88 return EXIT_FAILURE;
89
90 dprintf("user output path -> %s\n", optarg);
91 break;
92 case 's':
93 sscanf(optarg, "%li", &config->sleep);
94 dprintf("user sleep time -> %s\n", optarg);
95 break;
96 case 'l':
97 sscanf(optarg, "%li", &config->load);
98 dprintf("user load time -> %s\n", optarg);
99 break;
100 case 'c':
101 sscanf(optarg, "%u", &config->cpu);
102 dprintf("user cpu -> %s\n", optarg);
103 break;
104 case 'g':
105 strncpy(config->governor, optarg, 14);
106 dprintf("user governor -> %s\n", optarg);
107 break;
108 case 'p':
109 if (string_to_prio(optarg) != SCHED_ERR) {
110 config->prio = string_to_prio(optarg);
111 dprintf("user prio -> %s\n", optarg);
112 } else {
113 if (config != NULL) {
114 if (config->output != NULL)
115 fclose(config->output);
116 free(config);
117 }
118 usage();
119 }
120 break;
121 case 'n':
122 sscanf(optarg, "%u", &config->cycles);
123 dprintf("user cycles -> %s\n", optarg);
124 break;
125 case 'r':
126 sscanf(optarg, "%u", &config->rounds);
127 dprintf("user rounds -> %s\n", optarg);
128 break;
129 case 'x':
130 sscanf(optarg, "%li", &config->load_step);
131 dprintf("user load_step -> %s\n", optarg);
132 break;
133 case 'y':
134 sscanf(optarg, "%li", &config->sleep_step);
135 dprintf("user sleep_step -> %s\n", optarg);
136 break;
137 case 'f':
138 if (prepare_config(optarg, config))
139 return EXIT_FAILURE;
140 break;
141 case 'v':
142 config->verbose = 1;
143 dprintf("verbose output enabled\n");
144 break;
145 case 'h':
146 case '?':
147 default:
148 if (config != NULL) {
149 if (config->output != NULL)
150 fclose(config->output);
151 free(config);
152 }
153 usage();
154 }
155 }
156
157 if (config->verbose) {
158 printf("starting benchmark with parameters:\n");
159 printf("config:\n\t"
160 "sleep=%li\n\t"
161 "load=%li\n\t"
162 "sleep_step=%li\n\t"
163 "load_step=%li\n\t"
164 "cpu=%u\n\t"
165 "cycles=%u\n\t"
166 "rounds=%u\n\t"
167 "governor=%s\n\n",
168 config->sleep,
169 config->load,
170 config->sleep_step,
171 config->load_step,
172 config->cpu,
173 config->cycles,
174 config->rounds,
175 config->governor);
176 }
177
178 prepare_user(config);
179 prepare_system(config);
180 start_benchmark(config);
181
182 if (config->output != stdout)
183 fclose(config->output);
184
185 free(config);
186
187 return EXIT_SUCCESS;
188}
189
190
