277 lines
7.9 KiB
C
277 lines
7.9 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <err.h>
|
|
#include <math.h>
|
|
#include <sys/types.h>
|
|
#include <sys/sysctl.h>
|
|
#include <time.h>
|
|
#include <sys/sched.h>
|
|
#include <signal.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
int min, batt_min, wall_min;
|
|
int max, batt_max, wall_max;
|
|
int threshold, batt_threshold, wall_threshold;
|
|
int down_step, batt_down_step, wall_down_step;
|
|
int inertia, batt_inertia, wall_inertia;
|
|
int step, batt_step, wall_step;
|
|
int timefreq, batt_timefreq, wall_timefreq;
|
|
|
|
|
|
/* define the policy to auto or manual */
|
|
void set_policy(char* policy) {
|
|
int mib[2];
|
|
size_t len;
|
|
|
|
mib[0] = CTL_HW;
|
|
mib[1] = HW_PERFPOLICY;
|
|
len = sizeof(policy);
|
|
if (sysctl(mib, 2, NULL, 0, policy, len) == -1)
|
|
err(1, "sysctl");
|
|
}
|
|
|
|
/* restore policy auto upon exit */
|
|
void quit_gracefully() {
|
|
set_policy("auto");
|
|
exit(0);
|
|
}
|
|
|
|
void usage() {
|
|
printf("obsdfreqd [-h] [-q] [-i cycles] [-l min_freq] [-m max_freq] [-d percent_down_freq_step] [-r threshold] [-s percent_freq_step] [-t milliseconds]\n");
|
|
}
|
|
|
|
// switch to wall profile
|
|
void switch_wall() {
|
|
min = wall_min;
|
|
max = wall_max;
|
|
threshold = wall_threshold;
|
|
down_step = wall_down_step;
|
|
inertia = wall_inertia;
|
|
step = wall_step;
|
|
timefreq = wall_timefreq;
|
|
}
|
|
|
|
// switch to battery profile
|
|
void switch_batt() {
|
|
min = batt_min;
|
|
max = batt_max;
|
|
threshold = batt_threshold;
|
|
down_step = batt_down_step;
|
|
inertia = batt_inertia;
|
|
step = batt_step;
|
|
timefreq = batt_timefreq;
|
|
}
|
|
|
|
// assign values to variable if comma separated
|
|
// if not, assign value to two variables
|
|
void assign_values_from_param(char* parameter, int* charging, int* battery) {
|
|
int count = 0;
|
|
char *token = strtok(parameter, ",");
|
|
|
|
while (token != NULL) {
|
|
if(count == 0)
|
|
*charging = atoi(token);
|
|
if(count == 1)
|
|
*battery = atoi(token);
|
|
token = strtok(NULL, ",");
|
|
count++;
|
|
if(count > 1)
|
|
break;
|
|
}
|
|
|
|
if(count == 0) {
|
|
*charging = atoi(parameter);
|
|
*battery = *charging;
|
|
}
|
|
|
|
if(count == 1)
|
|
*battery = *charging;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
|
|
int opt;
|
|
int mib_perf[2];
|
|
int mib_powerplug[2];
|
|
int mib_load[2];
|
|
long cpu[CPUSTATES], cpu_previous[CPUSTATES];
|
|
int frequency = 0;
|
|
int quiet = 0;
|
|
int value, current_frequency, inertia_timer = 0;
|
|
int cpu_usage_percent = 0, cpu_usage;
|
|
size_t len, len_cpu;
|
|
|
|
min = batt_min= wall_min = 0;
|
|
max = batt_max= wall_max = 100;
|
|
threshold = batt_threshold= wall_threshold = 30;
|
|
down_step = batt_down_step= wall_down_step = 100;
|
|
inertia = batt_inertia= wall_inertia = 0;
|
|
step = batt_step= wall_step = 10;
|
|
timefreq = batt_timefreq= wall_timefreq = 300;
|
|
|
|
unveil("/", "r");
|
|
unveil(NULL, NULL);
|
|
|
|
while((opt = getopt(argc, argv, "d:hi:l:m:qr:s:t:")) != -1) {
|
|
switch(opt) {
|
|
case 'd':
|
|
assign_values_from_param(optarg, &wall_down_step, &batt_down_step);
|
|
if(down_step > 100 || down_step <= 0)
|
|
err(1, "decay step must be positive and up to 100");
|
|
break;
|
|
case 'i':
|
|
assign_values_from_param(optarg, &wall_inertia, &batt_inertia);
|
|
if(inertia < 0)
|
|
err(1, "inertia must be positive");
|
|
break;
|
|
case 'l':
|
|
assign_values_from_param(optarg, &wall_min, &batt_min);
|
|
if(min > 100 || min < 0)
|
|
err(1, "minimum frequency must be between 0 and 100");
|
|
break;
|
|
case 'm':
|
|
assign_values_from_param(optarg, &wall_max, &batt_max);
|
|
if(max > 100 || max < 0)
|
|
err(1, "maximum frequency must be between 0 and 100");
|
|
break;
|
|
case 'q':
|
|
quiet = 1;
|
|
break;
|
|
case 'r':
|
|
assign_values_from_param(optarg, &wall_threshold, &batt_threshold);
|
|
if(threshold < 0)
|
|
err(1, "CPU use threshold must be positive");
|
|
break;
|
|
case 's':
|
|
assign_values_from_param(optarg, &wall_step, &batt_step);
|
|
if(step > 100 || step <= 0)
|
|
err(1, "step must be positive and up to 100");
|
|
break;
|
|
case 't':
|
|
assign_values_from_param(optarg, &wall_timefreq, &batt_timefreq);
|
|
if(wall_timefreq <= 0 || batt_timefreq <= 0)
|
|
err(1, "time frequency must be positive");
|
|
break;
|
|
case 'h':
|
|
default:
|
|
usage();
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
mib_perf[0] = CTL_HW;
|
|
mib_perf[1] = HW_SETPERF;
|
|
|
|
mib_powerplug[0] = CTL_HW;
|
|
mib_powerplug[1] = HW_POWER;
|
|
|
|
mib_load[0] = CTL_KERN;
|
|
mib_load[1] = KERN_CPTIME;
|
|
|
|
len = sizeof(value);
|
|
len_cpu = sizeof(cpu);
|
|
|
|
signal(SIGINT, quit_gracefully);
|
|
signal(SIGTERM, quit_gracefully);
|
|
set_policy("manual");
|
|
|
|
// avoid weird reading for first delta
|
|
if (sysctl(mib_load, 2, &cpu_previous, &len_cpu, NULL, 0) == -1)
|
|
err(1, "sysctl");
|
|
usleep(1000*500);
|
|
|
|
/* main loop */
|
|
for(;;) {
|
|
// get if using power plug or not
|
|
if (sysctl(mib_powerplug, 2, &value, &len, NULL, 0) == -1)
|
|
err(1, "sysctl");
|
|
|
|
if(quiet == 0) printf("power: %i |", value);
|
|
if(value ==0)
|
|
switch_batt();
|
|
else
|
|
switch_wall();
|
|
|
|
|
|
// get current frequency
|
|
if (sysctl(mib_perf, 2, ¤t_frequency, &len, NULL, 0) == -1)
|
|
err(1, "sysctl");
|
|
if(quiet == 0) printf("perf: %3i |", current_frequency);
|
|
|
|
// get where the CPU time is spent, last field is IDLE
|
|
if (sysctl(mib_load, 2, &cpu, &len_cpu, NULL, 0) == -1)
|
|
err(1, "sysctl");
|
|
|
|
// calculate delta between old and last cpu readings
|
|
cpu_usage = cpu[0]-cpu_previous[0] +
|
|
cpu[1]-cpu_previous[1] +
|
|
cpu[2]-cpu_previous[2] +
|
|
cpu[3]-cpu_previous[3] +
|
|
cpu[4]-cpu_previous[4] +
|
|
cpu[5]-cpu_previous[5];
|
|
|
|
// debug
|
|
//if(quiet == 0) printf("\nDEBUG: User: %3i\tNice: %3i\t Sys: %3i\tSpin: %3i\t Intr: %3i\tIdle: %3i\n",
|
|
// cpu[0]-cpu_previous[0],
|
|
// cpu[1]-cpu_previous[1],
|
|
// cpu[2]-cpu_previous[2],
|
|
// cpu[3]-cpu_previous[3],
|
|
// cpu[4]-cpu_previous[4],
|
|
// cpu[5]-cpu_previous[5]);
|
|
//if(quiet == 0) printf("cpu usage = %i et idle = %i\n", cpu_usage, cpu[5] - cpu_previous[5]);
|
|
|
|
cpu_usage_percent = 100-round(100*(cpu[5]-cpu_previous[5])/cpu_usage);
|
|
memcpy(cpu_previous, cpu, sizeof(cpu));
|
|
if(quiet == 0) printf("usage: %3i%% |", cpu_usage_percent);
|
|
|
|
// change frequency
|
|
len = sizeof(frequency);
|
|
|
|
// small brain condition to increase CPU
|
|
if(cpu_usage_percent > threshold) {
|
|
|
|
// increase frequency by step if under max
|
|
if(frequency+step < max)
|
|
frequency = frequency + step;
|
|
else
|
|
frequency = max;
|
|
|
|
// don't try to set frequency more than 100%
|
|
if( frequency > 100 )
|
|
frequency = 100;
|
|
|
|
inertia_timer = inertia;
|
|
|
|
if (sysctl(mib_perf, 2, NULL, 0, &frequency, len) == -1)
|
|
err(1, "sysctl");
|
|
}
|
|
else {
|
|
|
|
if(inertia_timer == 0) {
|
|
// keep frequency more than min
|
|
if(frequency-down_step < min)
|
|
frequency = min;
|
|
else
|
|
frequency = frequency - down_step;
|
|
|
|
// don't try to set frequency below 0%
|
|
if (frequency < 0 )
|
|
frequency = 0;
|
|
|
|
if (sysctl(mib_perf, 2, NULL, 0, &frequency, len) == -1)
|
|
err(1, "sysctl");
|
|
} else {
|
|
inertia_timer--;
|
|
}
|
|
}
|
|
|
|
if(quiet == 0) printf("inertia: %2i |new freq: %3i", inertia_timer, frequency);
|
|
|
|
if(quiet == 0) printf("\n");
|
|
usleep(1000*timefreq);
|
|
}
|
|
|
|
return(0);
|
|
}
|