#include <sys/param.h>
#include <sys/systm.h>
#include <sys/clockintr.h>
#include <sys/proc.h>
#include <sys/kernel.h>
#include <sys/malloc.h>
#include <sys/resourcevar.h>
#include <uvm/uvm_extern.h>
#include <sys/sched.h>
#include <sys/timeout.h>
#include <sys/smr.h>
#include <sys/tracepoint.h>
#ifdef KTRACE
#include <sys/ktrace.h>
#endif
uint64_t roundrobin_period;
struct mutex sched_lock;
void update_loadavg(void *);
void schedcpu(void *);
uint32_t decay_aftersleep(uint32_t, uint32_t);
extern struct cpuset sched_idle_cpus;
extern struct cpuset sched_all_cpus;
static const fixpt_t cexp[3] = {
0.9200444146293232 * FSCALE,
0.9834714538216174 * FSCALE,
0.9944598480048967 * FSCALE,
};
struct loadavg averunnable;
void
roundrobin(struct clockrequest *cr, void *cf, void *arg)
{
uint64_t count;
struct cpu_info *ci = curcpu();
struct schedstate_percpu *spc = &ci->ci_schedstate;
count = clockrequest_advance(cr, roundrobin_period);
if (ci->ci_curproc != NULL) {
if (spc->spc_schedflags & SPCF_SEENRR || count >= 2) {
atomic_setbits_int(&spc->spc_schedflags,
SPCF_SEENRR | SPCF_SHOULDYIELD);
} else {
atomic_setbits_int(&spc->spc_schedflags,
SPCF_SEENRR);
}
}
if (spc->spc_nrun || spc->spc_schedflags & SPCF_SHOULDYIELD)
need_resched(ci);
}
void
update_loadavg(void *unused)
{
static struct timeout to = TIMEOUT_INITIALIZER(update_loadavg, NULL);
CPU_INFO_ITERATOR cii;
struct cpu_info *ci;
struct cpuset set;
u_int i, nrun;
cpuset_complement(&set, &sched_idle_cpus, &sched_all_cpus);
nrun = cpuset_cardinality(&set);
CPU_INFO_FOREACH(cii, ci) {
nrun += ci->ci_schedstate.spc_nrun;
}
for (i = 0; i < 3; i++) {
averunnable.ldavg[i] = (cexp[i] * averunnable.ldavg[i] +
nrun * FSCALE * (FSCALE - cexp[i])) >> FSHIFT;
}
timeout_add_sec(&to, 5);
}
#define loadfactor(loadav) (2 * (loadav))
#define decay_cpu(loadfac, cpu) (((loadfac) * (cpu)) / ((loadfac) + FSCALE))
fixpt_t ccpu = 0.95122942450071400909 * FSCALE;
#define CCPU_SHIFT 11
void
schedcpu(void *unused)
{
static struct timeout to = TIMEOUT_INITIALIZER(schedcpu, NULL);
fixpt_t loadfac = loadfactor(averunnable.ldavg[0]), pctcpu;
struct proc *p;
unsigned int newcpu, cpt;
LIST_FOREACH(p, &allproc, p_list) {
if (p->p_cpu != NULL &&
p->p_cpu->ci_schedstate.spc_idleproc == p)
continue;
if (p->p_stat == SSLEEP || p->p_stat == SSTOP)
p->p_slptime++;
pctcpu = (p->p_pctcpu * ccpu) >> FSHIFT;
if (p->p_slptime > 1) {
p->p_pctcpu = pctcpu;
continue;
}
SCHED_LOCK();
cpt = READ_ONCE(p->p_cpticks);
#if (FSHIFT >= CCPU_SHIFT)
pctcpu += (stathz == 100) ?
(cpt - p->p_cpticks2) << (FSHIFT - CCPU_SHIFT) :
100 * ((cpt - p->p_cpticks2)
<< (FSHIFT - CCPU_SHIFT)) / stathz;
#else
pctcpu += ((FSCALE - ccpu) *
((cpt - p->p_cpticks2) * FSCALE / stathz)) >> FSHIFT;
#endif
p->p_pctcpu = pctcpu;
p->p_cpticks2 = cpt;
newcpu = (u_int) decay_cpu(loadfac, p->p_estcpu);
setpriority(p, newcpu, p->p_p->ps_nice);
if (p->p_stat == SRUN &&
(p->p_runpri / SCHED_PPQ) != (p->p_usrpri / SCHED_PPQ)) {
remrunqueue(p);
setrunqueue(p->p_cpu, p, p->p_usrpri);
}
SCHED_UNLOCK();
}
timeout_add_sec(&to, 1);
}
uint32_t
decay_aftersleep(uint32_t estcpu, uint32_t slptime)
{
fixpt_t loadfac = loadfactor(averunnable.ldavg[0]);
uint32_t newcpu;
if (slptime > 5 * loadfac)
newcpu = 0;
else {
newcpu = estcpu;
slptime--;
while (newcpu && --slptime)
newcpu = decay_cpu(loadfac, newcpu);
}
return (newcpu);
}
void
yield(void)
{
struct proc *p = curproc;
SCHED_LOCK();
setrunqueue(p->p_cpu, p, p->p_usrpri);
p->p_ru.ru_nvcsw++;
mi_switch();
}
void
preempt(void)
{
struct proc *p = curproc;
SCHED_LOCK();
setrunqueue(p->p_cpu, p, p->p_usrpri);
p->p_ru.ru_nivcsw++;
mi_switch();
}
void
mi_switch(void)
{
struct schedstate_percpu *spc = &curcpu()->ci_schedstate;
struct proc *p = curproc;
struct proc *nextproc;
int oldipl;
#ifdef MULTIPROCESSOR
int hold_count;
#endif
KASSERT(p->p_stat != SONPROC);
SCHED_ASSERT_LOCKED();
#ifdef MULTIPROCESSOR
if (_kernel_lock_held())
hold_count = __mp_release_all(&kernel_lock);
else
hold_count = 0;
#endif
tuagg_add_runtime();
if (ISSET(spc->spc_schedflags, SPCF_ITIMER)) {
atomic_clearbits_int(&spc->spc_schedflags, SPCF_ITIMER);
clockintr_cancel(&spc->spc_itimer);
}
if (ISSET(spc->spc_schedflags, SPCF_PROFCLOCK)) {
atomic_clearbits_int(&spc->spc_schedflags, SPCF_PROFCLOCK);
clockintr_cancel(&spc->spc_profclock);
}
atomic_clearbits_int(&spc->spc_schedflags, SPCF_SWITCHCLEAR);
nextproc = sched_chooseproc();
oldipl = MUTEX_OLDIPL(&sched_lock);
if (p != nextproc) {
uvmexp.swtch++;
TRACEPOINT(sched, off__cpu, nextproc->p_tid + THREAD_PID_OFFSET,
nextproc->p_p->ps_pid);
cpu_switchto(p, nextproc);
TRACEPOINT(sched, on__cpu, NULL);
} else {
TRACEPOINT(sched, remain__cpu, NULL);
p->p_stat = SONPROC;
}
clear_resched(curcpu());
SCHED_ASSERT_LOCKED();
MUTEX_OLDIPL(&sched_lock) = oldipl;
SCHED_UNLOCK();
SCHED_ASSERT_UNLOCKED();
assertwaitok();
smr_idle();
KASSERT(p->p_cpu == curcpu());
spc = &p->p_cpu->ci_schedstate;
if (ISSET(p->p_p->ps_flags, PS_ITIMER)) {
atomic_setbits_int(&spc->spc_schedflags, SPCF_ITIMER);
clockintr_advance(&spc->spc_itimer, hardclock_period);
}
if (ISSET(p->p_p->ps_flags, PS_PROFIL)) {
atomic_setbits_int(&spc->spc_schedflags, SPCF_PROFCLOCK);
clockintr_advance(&spc->spc_profclock, profclock_period);
}
nanouptime(&spc->spc_runtime);
#ifdef MULTIPROCESSOR
if (hold_count)
__mp_acquire_count(&kernel_lock, hold_count);
#endif
}
void
setrunnable(struct proc *p)
{
struct process *pr = p->p_p;
u_char prio;
SCHED_ASSERT_LOCKED();
switch (p->p_stat) {
case 0:
case SRUN:
case SONPROC:
case SDEAD:
case SIDL:
default:
panic("setrunnable");
case SSTOP:
prio = p->p_usrpri;
TRACEPOINT(sched, unstop, p->p_tid + THREAD_PID_OFFSET,
p->p_p->ps_pid, CPU_INFO_UNIT(p->p_cpu));
if (ISSET(p->p_flag, P_INSCHED)) {
if (p->p_wchan != NULL)
p->p_stat = SSLEEP;
else
p->p_stat = SONPROC;
return;
}
setrunqueue(NULL, p, prio);
break;
case SSLEEP:
prio = p->p_slppri;
TRACEPOINT(sched, wakeup, p->p_tid + THREAD_PID_OFFSET,
p->p_p->ps_pid, CPU_INFO_UNIT(p->p_cpu));
if (ISSET(p->p_flag, P_INSCHED))
return;
setrunqueue(NULL, p, prio);
break;
}
if (p->p_slptime > 1) {
uint32_t newcpu;
newcpu = decay_aftersleep(p->p_estcpu, p->p_slptime);
setpriority(p, newcpu, pr->ps_nice);
}
p->p_slptime = 0;
}
void
setpriority(struct proc *p, uint32_t newcpu, uint8_t nice)
{
unsigned int newprio;
newprio = min((PUSER + newcpu + NICE_WEIGHT * (nice - NZERO)), MAXPRI);
SCHED_ASSERT_LOCKED();
p->p_estcpu = newcpu;
p->p_usrpri = newprio;
}
void
schedclock(struct proc *p)
{
struct cpu_info *ci = curcpu();
struct schedstate_percpu *spc = &ci->ci_schedstate;
uint32_t newcpu;
if (p == spc->spc_idleproc || spc->spc_spinning)
return;
SCHED_LOCK();
newcpu = ESTCPULIM(p->p_estcpu + 1);
setpriority(p, newcpu, p->p_p->ps_nice);
SCHED_UNLOCK();
}
void (*cpu_setperf)(int);
#define PERFPOL_MANUAL 0
#define PERFPOL_AUTO 1
#define PERFPOL_HIGH 2
int perflevel = 100;
int perfpolicy_on_ac = PERFPOL_HIGH;
int perfpolicy_on_battery = PERFPOL_AUTO;
#ifndef SMALL_KERNEL
#include <sys/sysctl.h>
void setperf_auto(void *);
struct timeout setperf_to = TIMEOUT_INITIALIZER(setperf_auto, NULL);
extern int hw_power;
static inline int
perfpolicy_dynamic(void)
{
return (perfpolicy_on_ac == PERFPOL_AUTO ||
perfpolicy_on_battery == PERFPOL_AUTO);
}
static inline int
current_perfpolicy(void)
{
return (hw_power) ? perfpolicy_on_ac : perfpolicy_on_battery;
}
void
setperf_auto(void *v)
{
static uint64_t *idleticks, *totalticks;
static int downbeats;
int i, j = 0;
int speedup = 0;
CPU_INFO_ITERATOR cii;
struct cpu_info *ci;
uint64_t idle, total, allidle = 0, alltotal = 0;
unsigned int gen;
if (!perfpolicy_dynamic())
return;
if (cpu_setperf == NULL)
return;
if (current_perfpolicy() == PERFPOL_HIGH) {
speedup = 1;
goto faster;
}
if (!idleticks)
if (!(idleticks = mallocarray(ncpusfound, sizeof(*idleticks),
M_DEVBUF, M_NOWAIT | M_ZERO)))
return;
if (!totalticks)
if (!(totalticks = mallocarray(ncpusfound, sizeof(*totalticks),
M_DEVBUF, M_NOWAIT | M_ZERO))) {
free(idleticks, M_DEVBUF,
sizeof(*idleticks) * ncpusfound);
return;
}
CPU_INFO_FOREACH(cii, ci) {
struct schedstate_percpu *spc;
if (!cpu_is_online(ci))
continue;
spc = &ci->ci_schedstate;
pc_cons_enter(&spc->spc_cp_time_lock, &gen);
do {
total = 0;
for (i = 0; i < CPUSTATES; i++) {
total += spc->spc_cp_time[i];
}
idle = spc->spc_cp_time[CP_IDLE];
} while (pc_cons_leave(&spc->spc_cp_time_lock, &gen) != 0);
total -= totalticks[j];
idle -= idleticks[j];
if (idle < total / 3)
speedup = 1;
alltotal += total;
allidle += idle;
idleticks[j] += idle;
totalticks[j] += total;
j++;
}
if (allidle < alltotal / 2)
speedup = 1;
if (speedup && downbeats < 5)
downbeats++;
if (speedup && perflevel != 100) {
faster:
perflevel = 100;
cpu_setperf(perflevel);
} else if (!speedup && perflevel != 0 && --downbeats <= 0) {
perflevel = 0;
cpu_setperf(perflevel);
}
timeout_add_msec(&setperf_to, 100);
}
int
sysctl_hwsetperf(void *oldp, size_t *oldlenp, void *newp, size_t newlen)
{
int err;
if (!cpu_setperf)
return EOPNOTSUPP;
if (perfpolicy_on_ac != PERFPOL_MANUAL)
return sysctl_rdint(oldp, oldlenp, newp, perflevel);
err = sysctl_int_bounded(oldp, oldlenp, newp, newlen,
&perflevel, 0, 100);
if (err)
return err;
if (newp != NULL)
cpu_setperf(perflevel);
return 0;
}
int
sysctl_hwperfpolicy(void *oldp, size_t *oldlenp, void *newp, size_t newlen)
{
char policy[32];
char *policy_on_battery;
int err, perfpolicy;
if (!cpu_setperf)
return EOPNOTSUPP;
switch (current_perfpolicy()) {
case PERFPOL_MANUAL:
strlcpy(policy, "manual", sizeof(policy));
break;
case PERFPOL_AUTO:
strlcpy(policy, "auto", sizeof(policy));
break;
case PERFPOL_HIGH:
strlcpy(policy, "high", sizeof(policy));
break;
default:
strlcpy(policy, "unknown", sizeof(policy));
break;
}
if (newp == NULL)
return sysctl_rdstring(oldp, oldlenp, newp, policy);
err = sysctl_string(oldp, oldlenp, newp, newlen, policy, sizeof(policy));
if (err)
return err;
policy_on_battery = strchr(policy, ',');
if (policy_on_battery != NULL) {
*policy_on_battery = '\0';
policy_on_battery++;
}
if (strcmp(policy, "manual") == 0)
perfpolicy = PERFPOL_MANUAL;
else if (strcmp(policy, "auto") == 0)
perfpolicy = PERFPOL_AUTO;
else if (strcmp(policy, "high") == 0)
perfpolicy = PERFPOL_HIGH;
else
return EINVAL;
if (policy_on_battery == NULL)
perfpolicy_on_battery = perfpolicy_on_ac = perfpolicy;
else {
if (strcmp(policy_on_battery, "manual") == 0 ||
perfpolicy == PERFPOL_MANUAL) {
return EINVAL;
}
if (strcmp(policy_on_battery, "auto") == 0)
perfpolicy_on_battery = PERFPOL_AUTO;
else if (strcmp(policy_on_battery, "high") == 0)
perfpolicy_on_battery = PERFPOL_HIGH;
else
return EINVAL;
perfpolicy_on_ac = perfpolicy;
}
if (current_perfpolicy() == PERFPOL_HIGH) {
perflevel = 100;
cpu_setperf(perflevel);
}
if (perfpolicy_dynamic())
timeout_add_msec(&setperf_to, 200);
return 0;
}
#endif
void
scheduler_start(void)
{
schedcpu(NULL);
update_loadavg(NULL);
#ifndef SMALL_KERNEL
if (perfpolicy_dynamic())
timeout_add_msec(&setperf_to, 200);
#endif
}