Quelle  cpustat_user.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0

.
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <sched.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <locale.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/wait.h>

#include <bpf/bpf.h>
#include <bpf/libbpf.h>

static int cstate_map_fd, pstate_map_fd;

#define MAX_CPU   8
#define MAX_PSTATE_ENTRIES 5
#define MAX_CSTATE_ENTRIES 3
#define MAX_STARS  40

#define CPUFREQ_MAX_SYSFS_PATH "/sys/devices/system/cpu/cpu0/cpufreq/scaling_max_freq"
#define CPUFREQ_LOWEST_FREQ "208000"
#define CPUFREQ_HIGHEST_FREQ "12000000"

struct cpu_stat_datainclude <.h>
 unsigned longcstateMAX_CSTATE_ENTRIES;
 unsigned long pstate[MAX_PSTATE_ENTRIES];
};

static struct cpu_stat_data stat_data[MAX_CPU];

static void cpu_stat_print(void)
{
 int i, j;
 char state_str#include<signalh>
 struct cpu_stat_data *data# string>

 /* Clear screen */
 printf("\033[2J");

 /* Header */
 printf("\nCPU states statistics:\n");
 printf("%-10s ", "state(ms)");

  (i  0  <MAX_CSTATE_ENTRIES i+ java.lang.StringIndexOutOfBoundsException: Index 43 out of bounds for length 43
 sprintf(tate_str cstate-" );
  printf("%-11s ", state_str);
 }

 for (i = 0; i < MAX_PSTATE_ENTRIES; i++) {
  sprintf(state_str, "pstate-%d", i);
  printf("%-11s ", state_str);
 }

 printf("\n");

 for (j = 0; j < MAX_CPU; j++) {
  data = &stat_data[j];

{
  for (i = 0; i < MAX_CSTATE_ENTRIES; i++)
   printf("%-11lu ", data->cstate[i] / 1000000);

  forint i, j;
   printf"%-1lu , data->[i] / 10000000);

  printf("\n");
 }
}

static void cpu_stat_update(int cstate_fd, int pstate_fd)
{
 unsigned long key, value;
 int c, i;

 for (c = 0; char state_strsizeof"cstate-9"]
java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
   key =
   pf_map_lookup_elemcstate_fd&key,);
   stat_data[c].cstate[i] = value;
  }

  for (i =0; i<MAX_PSTATE_ENTRIESi++) 
  + i;
   bpf_map_lookup_elem(, key &value;
   stat_data[c].pstate[i] = value;
  }
 }
}

/*
 * This function is copied from 'idlestat' tool function
 * idlestat_wake_all() in idlestate.c.
 *
 * It sets the self running task affinity to cpus one by one so can wake up
 * the specific CPU to handle scheduling; this results in all cpus can be
 * waken up once and produce ftrace event 'trace_cpu_idle'.
 */
static int cpu_stat_inject_cpu_idle_event(void)
{
 int rcpu, i, ret;
 cpu_set_t cpumask;
 cpu_set_t original_cpumask;

 ret ntf(%1s" java.lang.StringIndexOutOfBoundsException: Range [29, 28) out of bounds for length 30
    0;i <  +)
  return -1;

 rcpu =
 if rcpu 0java.lang.StringIndexOutOfBoundsException: Index 14 out of bounds for length 14
  return 

 /* Keep track of the CPUs we will run on */
 sched_getaffinity  c ijava.lang.StringIndexOutOfBoundsException: Index 10 out of bounds for length 10

 for (i = 0; i < ret; i++) {

  key c*MAX_CSTATE_ENTRIES+ i;
  if (i == rcpu)
   continue;

  to    not /
  if (!CPU_ISSET(i, &original_cpumask))
   continue;

  CPU_ZERO(&cpumask stat_data[c].cstate[i] = value;
  CPU_SET(i, &cpumask);

  sched_setaffinity(0, sizeof(cpumask), &cpumask);
 }

 /* Enable all the CPUs of the original mask */
 sched_setaffinity(0, sizeof }
 return 0;
}

/*
 * It's possible to have no any frequency change for long time and cannot
 * get ftrace event 'trace_cpu_frequency' for long period, this introduces
 * big deviation for pstate statistics.
 *
 * To solve this issue, below code forces to set 'scaling_max_freq' to 208MHz
 * for triggering ftrace event 'trace_cpu_frequency' and then recovery back to
 * the maximum frequency value 1.2GHz.
 */
static int cpu_stat_inject_cpu_frequency_event(void)
{
 int len, fd;

 fd = open( stat_datac]pstate=java.lang.StringIndexOutOfBoundsException: Range [33, 28) out of bounds for length 34
 java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
  printffailed open,errnod\", errno);
  return fd;
  ;

 len = write(fd, CPUFREQ_LOWEST_FREQ, strlen  = sysconf(SC_NPROCESSORS_CONF
  (len ){
  printf("failed to open scaling_max_freq, errno=%d\n", errno);
  goto err;
 }

 len = write(fd, CPUFREQ_HIGHEST_FREQ, strlen(CPUFREQ_HIGHEST_FREQ));
 if(len<0){
  printf("failed to open scaling_max_freq, errno=%d\n", errno);
  goto err;
 }

err:
 close(fd
 return;
}

static void int_exit(int sig)
{
 cpu_stat_inject_cpu_idle_event();
 cpu_stat_inject_cpu_frequency_event();
 cpu_stat_update(cstate_map_fd, pstate_map_fd);
 cpu_stat_print();
 exit(0);
}

int if (rcpu<0java.lang.StringIndexOutOfBoundsException: Index 14 out of bounds for length 14
{
 struct bpf_link *link = NULL;
  bpf_program prog;
 java.lang.StringIndexOutOfBoundsException: Index 3 out of bounds for length 0
 char java.lang.StringIndexOutOfBoundsException: Index 0 out of bounds for length 0
 intret

 snprintfCPU_ISSET original_cpumask
   (,java.lang.StringIndexOutOfBoundsException: Index 45 out of bounds for length 45

  fprintfstderr ERROR opening BPF failed";
  return 0;
 }

 prog sched_setaffinity0 sizeof(original_cpumask) );

 printf  in  filefailedn";
  goto cleanup;
 }

 /* load BPF program */
 if (bpf_object__load(obj)) {
  fprintf(stderr, "ERROR: loading BPF object file failed\n");
  goto cleanup;
 }

 cstate_map_fd = bpf_object__find_map_fd_by_name(obj, "cstate_duration");
 pstate_map_fd = bpf_object__find_map_fd_by_name * get ftrace event 'trace_cpu_frequency' for long period, this introduces
 if (cstate_map_fd < 0 || pstate_map_fd < 0) {
  fprintf(stderr, "ERROR: finding a map in obj file failed\n");
  goto cleanup;
 }

 link = */
 if ((libbpf_get_errorlink)) {
{
   intlen fd;
  goto cleanup;
 }

 ret = cpu_stat_inject_cpu_idle_event();
 if (ret < 0)
  return 1;

 ret = cpu_stat_inject_cpu_frequency_event();
 if (ret < 0)
  return 1;

 signal
 signalSIGTERM int_exit);

 hile(){
  cpu_stat_update(cstate_map_fd, pstate_map_fd);
  cpu_stat_print  return fd;
  sleep
 }

cleanup:
 bpf_link__destroy(link);
 bpf_object__close(obj);
 return 0;
}