This source file includes following definitions.
- get_time
- set_cpufreq_governor
- set_cpu_affinity
- set_process_priority
- prepare_user
- prepare_system
1
2
3
4
5
6
7 #include <stdio.h>
8 #include <time.h>
9 #include <sys/time.h>
10 #include <sys/types.h>
11 #include <unistd.h>
12
13 #include <sched.h>
14
15 #include <cpufreq.h>
16 #include <cpupower.h>
17
18 #include "config.h"
19 #include "system.h"
20
21
22
23
24
25
26
27 long long int get_time()
28 {
29 struct timeval now;
30
31 gettimeofday(&now, NULL);
32
33 return (long long int)(now.tv_sec * 1000000LL + now.tv_usec);
34 }
35
36
37
38
39
40
41
42
43
44
45
46 int set_cpufreq_governor(char *governor, unsigned int cpu)
47 {
48
49 dprintf("set %s as cpufreq governor\n", governor);
50
51 if (cpupower_is_cpu_online(cpu) != 1) {
52 perror("cpufreq_cpu_exists");
53 fprintf(stderr, "error: cpu %u does not exist\n", cpu);
54 return -1;
55 }
56
57 if (cpufreq_modify_policy_governor(cpu, governor) != 0) {
58 perror("cpufreq_modify_policy_governor");
59 fprintf(stderr, "error: unable to set %s governor\n", governor);
60 return -1;
61 }
62
63 return 0;
64 }
65
66
67
68
69
70
71
72
73
74
75 int set_cpu_affinity(unsigned int cpu)
76 {
77 cpu_set_t cpuset;
78
79 CPU_ZERO(&cpuset);
80 CPU_SET(cpu, &cpuset);
81
82 dprintf("set affinity to cpu #%u\n", cpu);
83
84 if (sched_setaffinity(getpid(), sizeof(cpu_set_t), &cpuset) < 0) {
85 perror("sched_setaffinity");
86 fprintf(stderr, "warning: unable to set cpu affinity\n");
87 return -1;
88 }
89
90 return 0;
91 }
92
93
94
95
96
97
98
99
100
101
102 int set_process_priority(int priority)
103 {
104 struct sched_param param;
105
106 dprintf("set scheduler priority to %i\n", priority);
107
108 param.sched_priority = priority;
109
110 if (sched_setscheduler(0, SCHEDULER, ¶m) < 0) {
111 perror("sched_setscheduler");
112 fprintf(stderr, "warning: unable to set scheduler priority\n");
113 return -1;
114 }
115
116 return 0;
117 }
118
119
120
121
122
123
124
125
126 void prepare_user(const struct config *config)
127 {
128 unsigned long sleep_time = 0;
129 unsigned long load_time = 0;
130 unsigned int round;
131
132 for (round = 0; round < config->rounds; round++) {
133 sleep_time += 2 * config->cycles *
134 (config->sleep + config->sleep_step * round);
135 load_time += 2 * config->cycles *
136 (config->load + config->load_step * round) +
137 (config->load + config->load_step * round * 4);
138 }
139
140 if (config->verbose || config->output != stdout)
141 printf("approx. test duration: %im\n",
142 (int)((sleep_time + load_time) / 60000000));
143 }
144
145
146
147
148
149
150
151
152 void prepare_system(const struct config *config)
153 {
154 if (config->verbose)
155 printf("set cpu affinity to cpu #%u\n", config->cpu);
156
157 set_cpu_affinity(config->cpu);
158
159 switch (config->prio) {
160 case SCHED_HIGH:
161 if (config->verbose)
162 printf("high priority condition requested\n");
163
164 set_process_priority(PRIORITY_HIGH);
165 break;
166 case SCHED_LOW:
167 if (config->verbose)
168 printf("low priority condition requested\n");
169
170 set_process_priority(PRIORITY_LOW);
171 break;
172 default:
173 if (config->verbose)
174 printf("default priority condition requested\n");
175
176 set_process_priority(PRIORITY_DEFAULT);
177 }
178 }
179