162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Memory bandwidth monitoring and allocation library
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2018 Intel Corporation
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Authors:
862306a36Sopenharmony_ci *    Sai Praneeth Prakhya <sai.praneeth.prakhya@intel.com>,
962306a36Sopenharmony_ci *    Fenghua Yu <fenghua.yu@intel.com>
1062306a36Sopenharmony_ci */
1162306a36Sopenharmony_ci#include "resctrl.h"
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#define UNCORE_IMC		"uncore_imc"
1462306a36Sopenharmony_ci#define READ_FILE_NAME		"events/cas_count_read"
1562306a36Sopenharmony_ci#define WRITE_FILE_NAME		"events/cas_count_write"
1662306a36Sopenharmony_ci#define DYN_PMU_PATH		"/sys/bus/event_source/devices"
1762306a36Sopenharmony_ci#define SCALE			0.00006103515625
1862306a36Sopenharmony_ci#define MAX_IMCS		20
1962306a36Sopenharmony_ci#define MAX_TOKENS		5
2062306a36Sopenharmony_ci#define READ			0
2162306a36Sopenharmony_ci#define WRITE			1
2262306a36Sopenharmony_ci#define CON_MON_MBM_LOCAL_BYTES_PATH				\
2362306a36Sopenharmony_ci	"%s/%s/mon_groups/%s/mon_data/mon_L3_%02d/mbm_local_bytes"
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci#define CON_MBM_LOCAL_BYTES_PATH		\
2662306a36Sopenharmony_ci	"%s/%s/mon_data/mon_L3_%02d/mbm_local_bytes"
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci#define MON_MBM_LOCAL_BYTES_PATH		\
2962306a36Sopenharmony_ci	"%s/mon_groups/%s/mon_data/mon_L3_%02d/mbm_local_bytes"
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci#define MBM_LOCAL_BYTES_PATH			\
3262306a36Sopenharmony_ci	"%s/mon_data/mon_L3_%02d/mbm_local_bytes"
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci#define CON_MON_LCC_OCCUP_PATH		\
3562306a36Sopenharmony_ci	"%s/%s/mon_groups/%s/mon_data/mon_L3_%02d/llc_occupancy"
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci#define CON_LCC_OCCUP_PATH		\
3862306a36Sopenharmony_ci	"%s/%s/mon_data/mon_L3_%02d/llc_occupancy"
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci#define MON_LCC_OCCUP_PATH		\
4162306a36Sopenharmony_ci	"%s/mon_groups/%s/mon_data/mon_L3_%02d/llc_occupancy"
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci#define LCC_OCCUP_PATH			\
4462306a36Sopenharmony_ci	"%s/mon_data/mon_L3_%02d/llc_occupancy"
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_cistruct membw_read_format {
4762306a36Sopenharmony_ci	__u64 value;         /* The value of the event */
4862306a36Sopenharmony_ci	__u64 time_enabled;  /* if PERF_FORMAT_TOTAL_TIME_ENABLED */
4962306a36Sopenharmony_ci	__u64 time_running;  /* if PERF_FORMAT_TOTAL_TIME_RUNNING */
5062306a36Sopenharmony_ci	__u64 id;            /* if PERF_FORMAT_ID */
5162306a36Sopenharmony_ci};
5262306a36Sopenharmony_ci
5362306a36Sopenharmony_cistruct imc_counter_config {
5462306a36Sopenharmony_ci	__u32 type;
5562306a36Sopenharmony_ci	__u64 event;
5662306a36Sopenharmony_ci	__u64 umask;
5762306a36Sopenharmony_ci	struct perf_event_attr pe;
5862306a36Sopenharmony_ci	struct membw_read_format return_value;
5962306a36Sopenharmony_ci	int fd;
6062306a36Sopenharmony_ci};
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_cistatic char mbm_total_path[1024];
6362306a36Sopenharmony_cistatic int imcs;
6462306a36Sopenharmony_cistatic struct imc_counter_config imc_counters_config[MAX_IMCS][2];
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_civoid membw_initialize_perf_event_attr(int i, int j)
6762306a36Sopenharmony_ci{
6862306a36Sopenharmony_ci	memset(&imc_counters_config[i][j].pe, 0,
6962306a36Sopenharmony_ci	       sizeof(struct perf_event_attr));
7062306a36Sopenharmony_ci	imc_counters_config[i][j].pe.type = imc_counters_config[i][j].type;
7162306a36Sopenharmony_ci	imc_counters_config[i][j].pe.size = sizeof(struct perf_event_attr);
7262306a36Sopenharmony_ci	imc_counters_config[i][j].pe.disabled = 1;
7362306a36Sopenharmony_ci	imc_counters_config[i][j].pe.inherit = 1;
7462306a36Sopenharmony_ci	imc_counters_config[i][j].pe.exclude_guest = 0;
7562306a36Sopenharmony_ci	imc_counters_config[i][j].pe.config =
7662306a36Sopenharmony_ci		imc_counters_config[i][j].umask << 8 |
7762306a36Sopenharmony_ci		imc_counters_config[i][j].event;
7862306a36Sopenharmony_ci	imc_counters_config[i][j].pe.sample_type = PERF_SAMPLE_IDENTIFIER;
7962306a36Sopenharmony_ci	imc_counters_config[i][j].pe.read_format =
8062306a36Sopenharmony_ci		PERF_FORMAT_TOTAL_TIME_ENABLED | PERF_FORMAT_TOTAL_TIME_RUNNING;
8162306a36Sopenharmony_ci}
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_civoid membw_ioctl_perf_event_ioc_reset_enable(int i, int j)
8462306a36Sopenharmony_ci{
8562306a36Sopenharmony_ci	ioctl(imc_counters_config[i][j].fd, PERF_EVENT_IOC_RESET, 0);
8662306a36Sopenharmony_ci	ioctl(imc_counters_config[i][j].fd, PERF_EVENT_IOC_ENABLE, 0);
8762306a36Sopenharmony_ci}
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_civoid membw_ioctl_perf_event_ioc_disable(int i, int j)
9062306a36Sopenharmony_ci{
9162306a36Sopenharmony_ci	ioctl(imc_counters_config[i][j].fd, PERF_EVENT_IOC_DISABLE, 0);
9262306a36Sopenharmony_ci}
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci/*
9562306a36Sopenharmony_ci * get_event_and_umask:	Parse config into event and umask
9662306a36Sopenharmony_ci * @cas_count_cfg:	Config
9762306a36Sopenharmony_ci * @count:		iMC number
9862306a36Sopenharmony_ci * @op:			Operation (read/write)
9962306a36Sopenharmony_ci */
10062306a36Sopenharmony_civoid get_event_and_umask(char *cas_count_cfg, int count, bool op)
10162306a36Sopenharmony_ci{
10262306a36Sopenharmony_ci	char *token[MAX_TOKENS];
10362306a36Sopenharmony_ci	int i = 0;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	strcat(cas_count_cfg, ",");
10662306a36Sopenharmony_ci	token[0] = strtok(cas_count_cfg, "=,");
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci	for (i = 1; i < MAX_TOKENS; i++)
10962306a36Sopenharmony_ci		token[i] = strtok(NULL, "=,");
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	for (i = 0; i < MAX_TOKENS; i++) {
11262306a36Sopenharmony_ci		if (!token[i])
11362306a36Sopenharmony_ci			break;
11462306a36Sopenharmony_ci		if (strcmp(token[i], "event") == 0) {
11562306a36Sopenharmony_ci			if (op == READ)
11662306a36Sopenharmony_ci				imc_counters_config[count][READ].event =
11762306a36Sopenharmony_ci				strtol(token[i + 1], NULL, 16);
11862306a36Sopenharmony_ci			else
11962306a36Sopenharmony_ci				imc_counters_config[count][WRITE].event =
12062306a36Sopenharmony_ci				strtol(token[i + 1], NULL, 16);
12162306a36Sopenharmony_ci		}
12262306a36Sopenharmony_ci		if (strcmp(token[i], "umask") == 0) {
12362306a36Sopenharmony_ci			if (op == READ)
12462306a36Sopenharmony_ci				imc_counters_config[count][READ].umask =
12562306a36Sopenharmony_ci				strtol(token[i + 1], NULL, 16);
12662306a36Sopenharmony_ci			else
12762306a36Sopenharmony_ci				imc_counters_config[count][WRITE].umask =
12862306a36Sopenharmony_ci				strtol(token[i + 1], NULL, 16);
12962306a36Sopenharmony_ci		}
13062306a36Sopenharmony_ci	}
13162306a36Sopenharmony_ci}
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_cistatic int open_perf_event(int i, int cpu_no, int j)
13462306a36Sopenharmony_ci{
13562306a36Sopenharmony_ci	imc_counters_config[i][j].fd =
13662306a36Sopenharmony_ci		perf_event_open(&imc_counters_config[i][j].pe, -1, cpu_no, -1,
13762306a36Sopenharmony_ci				PERF_FLAG_FD_CLOEXEC);
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	if (imc_counters_config[i][j].fd == -1) {
14062306a36Sopenharmony_ci		fprintf(stderr, "Error opening leader %llx\n",
14162306a36Sopenharmony_ci			imc_counters_config[i][j].pe.config);
14262306a36Sopenharmony_ci
14362306a36Sopenharmony_ci		return -1;
14462306a36Sopenharmony_ci	}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_ci	return 0;
14762306a36Sopenharmony_ci}
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci/* Get type and config (read and write) of an iMC counter */
15062306a36Sopenharmony_cistatic int read_from_imc_dir(char *imc_dir, int count)
15162306a36Sopenharmony_ci{
15262306a36Sopenharmony_ci	char cas_count_cfg[1024], imc_counter_cfg[1024], imc_counter_type[1024];
15362306a36Sopenharmony_ci	FILE *fp;
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	/* Get type of iMC counter */
15662306a36Sopenharmony_ci	sprintf(imc_counter_type, "%s%s", imc_dir, "type");
15762306a36Sopenharmony_ci	fp = fopen(imc_counter_type, "r");
15862306a36Sopenharmony_ci	if (!fp) {
15962306a36Sopenharmony_ci		perror("Failed to open imc counter type file");
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci		return -1;
16262306a36Sopenharmony_ci	}
16362306a36Sopenharmony_ci	if (fscanf(fp, "%u", &imc_counters_config[count][READ].type) <= 0) {
16462306a36Sopenharmony_ci		perror("Could not get imc type");
16562306a36Sopenharmony_ci		fclose(fp);
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci		return -1;
16862306a36Sopenharmony_ci	}
16962306a36Sopenharmony_ci	fclose(fp);
17062306a36Sopenharmony_ci
17162306a36Sopenharmony_ci	imc_counters_config[count][WRITE].type =
17262306a36Sopenharmony_ci				imc_counters_config[count][READ].type;
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_ci	/* Get read config */
17562306a36Sopenharmony_ci	sprintf(imc_counter_cfg, "%s%s", imc_dir, READ_FILE_NAME);
17662306a36Sopenharmony_ci	fp = fopen(imc_counter_cfg, "r");
17762306a36Sopenharmony_ci	if (!fp) {
17862306a36Sopenharmony_ci		perror("Failed to open imc config file");
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_ci		return -1;
18162306a36Sopenharmony_ci	}
18262306a36Sopenharmony_ci	if (fscanf(fp, "%s", cas_count_cfg) <= 0) {
18362306a36Sopenharmony_ci		perror("Could not get imc cas count read");
18462306a36Sopenharmony_ci		fclose(fp);
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_ci		return -1;
18762306a36Sopenharmony_ci	}
18862306a36Sopenharmony_ci	fclose(fp);
18962306a36Sopenharmony_ci
19062306a36Sopenharmony_ci	get_event_and_umask(cas_count_cfg, count, READ);
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	/* Get write config */
19362306a36Sopenharmony_ci	sprintf(imc_counter_cfg, "%s%s", imc_dir, WRITE_FILE_NAME);
19462306a36Sopenharmony_ci	fp = fopen(imc_counter_cfg, "r");
19562306a36Sopenharmony_ci	if (!fp) {
19662306a36Sopenharmony_ci		perror("Failed to open imc config file");
19762306a36Sopenharmony_ci
19862306a36Sopenharmony_ci		return -1;
19962306a36Sopenharmony_ci	}
20062306a36Sopenharmony_ci	if  (fscanf(fp, "%s", cas_count_cfg) <= 0) {
20162306a36Sopenharmony_ci		perror("Could not get imc cas count write");
20262306a36Sopenharmony_ci		fclose(fp);
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_ci		return -1;
20562306a36Sopenharmony_ci	}
20662306a36Sopenharmony_ci	fclose(fp);
20762306a36Sopenharmony_ci
20862306a36Sopenharmony_ci	get_event_and_umask(cas_count_cfg, count, WRITE);
20962306a36Sopenharmony_ci
21062306a36Sopenharmony_ci	return 0;
21162306a36Sopenharmony_ci}
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci/*
21462306a36Sopenharmony_ci * A system can have 'n' number of iMC (Integrated Memory Controller)
21562306a36Sopenharmony_ci * counters, get that 'n'. For each iMC counter get it's type and config.
21662306a36Sopenharmony_ci * Also, each counter has two configs, one for read and the other for write.
21762306a36Sopenharmony_ci * A config again has two parts, event and umask.
21862306a36Sopenharmony_ci * Enumerate all these details into an array of structures.
21962306a36Sopenharmony_ci *
22062306a36Sopenharmony_ci * Return: >= 0 on success. < 0 on failure.
22162306a36Sopenharmony_ci */
22262306a36Sopenharmony_cistatic int num_of_imcs(void)
22362306a36Sopenharmony_ci{
22462306a36Sopenharmony_ci	char imc_dir[512], *temp;
22562306a36Sopenharmony_ci	unsigned int count = 0;
22662306a36Sopenharmony_ci	struct dirent *ep;
22762306a36Sopenharmony_ci	int ret;
22862306a36Sopenharmony_ci	DIR *dp;
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_ci	dp = opendir(DYN_PMU_PATH);
23162306a36Sopenharmony_ci	if (dp) {
23262306a36Sopenharmony_ci		while ((ep = readdir(dp))) {
23362306a36Sopenharmony_ci			temp = strstr(ep->d_name, UNCORE_IMC);
23462306a36Sopenharmony_ci			if (!temp)
23562306a36Sopenharmony_ci				continue;
23662306a36Sopenharmony_ci
23762306a36Sopenharmony_ci			/*
23862306a36Sopenharmony_ci			 * imc counters are named as "uncore_imc_<n>", hence
23962306a36Sopenharmony_ci			 * increment the pointer to point to <n>. Note that
24062306a36Sopenharmony_ci			 * sizeof(UNCORE_IMC) would count for null character as
24162306a36Sopenharmony_ci			 * well and hence the last underscore character in
24262306a36Sopenharmony_ci			 * uncore_imc'_' need not be counted.
24362306a36Sopenharmony_ci			 */
24462306a36Sopenharmony_ci			temp = temp + sizeof(UNCORE_IMC);
24562306a36Sopenharmony_ci
24662306a36Sopenharmony_ci			/*
24762306a36Sopenharmony_ci			 * Some directories under "DYN_PMU_PATH" could have
24862306a36Sopenharmony_ci			 * names like "uncore_imc_free_running", hence, check if
24962306a36Sopenharmony_ci			 * first character is a numerical digit or not.
25062306a36Sopenharmony_ci			 */
25162306a36Sopenharmony_ci			if (temp[0] >= '0' && temp[0] <= '9') {
25262306a36Sopenharmony_ci				sprintf(imc_dir, "%s/%s/", DYN_PMU_PATH,
25362306a36Sopenharmony_ci					ep->d_name);
25462306a36Sopenharmony_ci				ret = read_from_imc_dir(imc_dir, count);
25562306a36Sopenharmony_ci				if (ret) {
25662306a36Sopenharmony_ci					closedir(dp);
25762306a36Sopenharmony_ci
25862306a36Sopenharmony_ci					return ret;
25962306a36Sopenharmony_ci				}
26062306a36Sopenharmony_ci				count++;
26162306a36Sopenharmony_ci			}
26262306a36Sopenharmony_ci		}
26362306a36Sopenharmony_ci		closedir(dp);
26462306a36Sopenharmony_ci		if (count == 0) {
26562306a36Sopenharmony_ci			perror("Unable find iMC counters!\n");
26662306a36Sopenharmony_ci
26762306a36Sopenharmony_ci			return -1;
26862306a36Sopenharmony_ci		}
26962306a36Sopenharmony_ci	} else {
27062306a36Sopenharmony_ci		perror("Unable to open PMU directory!\n");
27162306a36Sopenharmony_ci
27262306a36Sopenharmony_ci		return -1;
27362306a36Sopenharmony_ci	}
27462306a36Sopenharmony_ci
27562306a36Sopenharmony_ci	return count;
27662306a36Sopenharmony_ci}
27762306a36Sopenharmony_ci
27862306a36Sopenharmony_cistatic int initialize_mem_bw_imc(void)
27962306a36Sopenharmony_ci{
28062306a36Sopenharmony_ci	int imc, j;
28162306a36Sopenharmony_ci
28262306a36Sopenharmony_ci	imcs = num_of_imcs();
28362306a36Sopenharmony_ci	if (imcs <= 0)
28462306a36Sopenharmony_ci		return imcs;
28562306a36Sopenharmony_ci
28662306a36Sopenharmony_ci	/* Initialize perf_event_attr structures for all iMC's */
28762306a36Sopenharmony_ci	for (imc = 0; imc < imcs; imc++) {
28862306a36Sopenharmony_ci		for (j = 0; j < 2; j++)
28962306a36Sopenharmony_ci			membw_initialize_perf_event_attr(imc, j);
29062306a36Sopenharmony_ci	}
29162306a36Sopenharmony_ci
29262306a36Sopenharmony_ci	return 0;
29362306a36Sopenharmony_ci}
29462306a36Sopenharmony_ci
29562306a36Sopenharmony_ci/*
29662306a36Sopenharmony_ci * get_mem_bw_imc:	Memory band width as reported by iMC counters
29762306a36Sopenharmony_ci * @cpu_no:		CPU number that the benchmark PID is binded to
29862306a36Sopenharmony_ci * @bw_report:		Bandwidth report type (reads, writes)
29962306a36Sopenharmony_ci *
30062306a36Sopenharmony_ci * Memory B/W utilized by a process on a socket can be calculated using
30162306a36Sopenharmony_ci * iMC counters. Perf events are used to read these counters.
30262306a36Sopenharmony_ci *
30362306a36Sopenharmony_ci * Return: = 0 on success. < 0 on failure.
30462306a36Sopenharmony_ci */
30562306a36Sopenharmony_cistatic int get_mem_bw_imc(int cpu_no, char *bw_report, float *bw_imc)
30662306a36Sopenharmony_ci{
30762306a36Sopenharmony_ci	float reads, writes, of_mul_read, of_mul_write;
30862306a36Sopenharmony_ci	int imc, j, ret;
30962306a36Sopenharmony_ci
31062306a36Sopenharmony_ci	/* Start all iMC counters to log values (both read and write) */
31162306a36Sopenharmony_ci	reads = 0, writes = 0, of_mul_read = 1, of_mul_write = 1;
31262306a36Sopenharmony_ci	for (imc = 0; imc < imcs; imc++) {
31362306a36Sopenharmony_ci		for (j = 0; j < 2; j++) {
31462306a36Sopenharmony_ci			ret = open_perf_event(imc, cpu_no, j);
31562306a36Sopenharmony_ci			if (ret)
31662306a36Sopenharmony_ci				return -1;
31762306a36Sopenharmony_ci		}
31862306a36Sopenharmony_ci		for (j = 0; j < 2; j++)
31962306a36Sopenharmony_ci			membw_ioctl_perf_event_ioc_reset_enable(imc, j);
32062306a36Sopenharmony_ci	}
32162306a36Sopenharmony_ci
32262306a36Sopenharmony_ci	sleep(1);
32362306a36Sopenharmony_ci
32462306a36Sopenharmony_ci	/* Stop counters after a second to get results (both read and write) */
32562306a36Sopenharmony_ci	for (imc = 0; imc < imcs; imc++) {
32662306a36Sopenharmony_ci		for (j = 0; j < 2; j++)
32762306a36Sopenharmony_ci			membw_ioctl_perf_event_ioc_disable(imc, j);
32862306a36Sopenharmony_ci	}
32962306a36Sopenharmony_ci
33062306a36Sopenharmony_ci	/*
33162306a36Sopenharmony_ci	 * Get results which are stored in struct type imc_counter_config
33262306a36Sopenharmony_ci	 * Take over flow into consideration before calculating total b/w
33362306a36Sopenharmony_ci	 */
33462306a36Sopenharmony_ci	for (imc = 0; imc < imcs; imc++) {
33562306a36Sopenharmony_ci		struct imc_counter_config *r =
33662306a36Sopenharmony_ci			&imc_counters_config[imc][READ];
33762306a36Sopenharmony_ci		struct imc_counter_config *w =
33862306a36Sopenharmony_ci			&imc_counters_config[imc][WRITE];
33962306a36Sopenharmony_ci
34062306a36Sopenharmony_ci		if (read(r->fd, &r->return_value,
34162306a36Sopenharmony_ci			 sizeof(struct membw_read_format)) == -1) {
34262306a36Sopenharmony_ci			perror("Couldn't get read b/w through iMC");
34362306a36Sopenharmony_ci
34462306a36Sopenharmony_ci			return -1;
34562306a36Sopenharmony_ci		}
34662306a36Sopenharmony_ci
34762306a36Sopenharmony_ci		if (read(w->fd, &w->return_value,
34862306a36Sopenharmony_ci			 sizeof(struct membw_read_format)) == -1) {
34962306a36Sopenharmony_ci			perror("Couldn't get write bw through iMC");
35062306a36Sopenharmony_ci
35162306a36Sopenharmony_ci			return -1;
35262306a36Sopenharmony_ci		}
35362306a36Sopenharmony_ci
35462306a36Sopenharmony_ci		__u64 r_time_enabled = r->return_value.time_enabled;
35562306a36Sopenharmony_ci		__u64 r_time_running = r->return_value.time_running;
35662306a36Sopenharmony_ci
35762306a36Sopenharmony_ci		if (r_time_enabled != r_time_running)
35862306a36Sopenharmony_ci			of_mul_read = (float)r_time_enabled /
35962306a36Sopenharmony_ci					(float)r_time_running;
36062306a36Sopenharmony_ci
36162306a36Sopenharmony_ci		__u64 w_time_enabled = w->return_value.time_enabled;
36262306a36Sopenharmony_ci		__u64 w_time_running = w->return_value.time_running;
36362306a36Sopenharmony_ci
36462306a36Sopenharmony_ci		if (w_time_enabled != w_time_running)
36562306a36Sopenharmony_ci			of_mul_write = (float)w_time_enabled /
36662306a36Sopenharmony_ci					(float)w_time_running;
36762306a36Sopenharmony_ci		reads += r->return_value.value * of_mul_read * SCALE;
36862306a36Sopenharmony_ci		writes += w->return_value.value * of_mul_write * SCALE;
36962306a36Sopenharmony_ci	}
37062306a36Sopenharmony_ci
37162306a36Sopenharmony_ci	for (imc = 0; imc < imcs; imc++) {
37262306a36Sopenharmony_ci		close(imc_counters_config[imc][READ].fd);
37362306a36Sopenharmony_ci		close(imc_counters_config[imc][WRITE].fd);
37462306a36Sopenharmony_ci	}
37562306a36Sopenharmony_ci
37662306a36Sopenharmony_ci	if (strcmp(bw_report, "reads") == 0) {
37762306a36Sopenharmony_ci		*bw_imc = reads;
37862306a36Sopenharmony_ci		return 0;
37962306a36Sopenharmony_ci	}
38062306a36Sopenharmony_ci
38162306a36Sopenharmony_ci	if (strcmp(bw_report, "writes") == 0) {
38262306a36Sopenharmony_ci		*bw_imc = writes;
38362306a36Sopenharmony_ci		return 0;
38462306a36Sopenharmony_ci	}
38562306a36Sopenharmony_ci
38662306a36Sopenharmony_ci	*bw_imc = reads + writes;
38762306a36Sopenharmony_ci	return 0;
38862306a36Sopenharmony_ci}
38962306a36Sopenharmony_ci
39062306a36Sopenharmony_civoid set_mbm_path(const char *ctrlgrp, const char *mongrp, int resource_id)
39162306a36Sopenharmony_ci{
39262306a36Sopenharmony_ci	if (ctrlgrp && mongrp)
39362306a36Sopenharmony_ci		sprintf(mbm_total_path, CON_MON_MBM_LOCAL_BYTES_PATH,
39462306a36Sopenharmony_ci			RESCTRL_PATH, ctrlgrp, mongrp, resource_id);
39562306a36Sopenharmony_ci	else if (!ctrlgrp && mongrp)
39662306a36Sopenharmony_ci		sprintf(mbm_total_path, MON_MBM_LOCAL_BYTES_PATH, RESCTRL_PATH,
39762306a36Sopenharmony_ci			mongrp, resource_id);
39862306a36Sopenharmony_ci	else if (ctrlgrp && !mongrp)
39962306a36Sopenharmony_ci		sprintf(mbm_total_path, CON_MBM_LOCAL_BYTES_PATH, RESCTRL_PATH,
40062306a36Sopenharmony_ci			ctrlgrp, resource_id);
40162306a36Sopenharmony_ci	else if (!ctrlgrp && !mongrp)
40262306a36Sopenharmony_ci		sprintf(mbm_total_path, MBM_LOCAL_BYTES_PATH, RESCTRL_PATH,
40362306a36Sopenharmony_ci			resource_id);
40462306a36Sopenharmony_ci}
40562306a36Sopenharmony_ci
40662306a36Sopenharmony_ci/*
40762306a36Sopenharmony_ci * initialize_mem_bw_resctrl:	Appropriately populate "mbm_total_path"
40862306a36Sopenharmony_ci * @ctrlgrp:			Name of the control monitor group (con_mon grp)
40962306a36Sopenharmony_ci * @mongrp:			Name of the monitor group (mon grp)
41062306a36Sopenharmony_ci * @cpu_no:			CPU number that the benchmark PID is binded to
41162306a36Sopenharmony_ci * @resctrl_val:		Resctrl feature (Eg: mbm, mba.. etc)
41262306a36Sopenharmony_ci */
41362306a36Sopenharmony_cistatic void initialize_mem_bw_resctrl(const char *ctrlgrp, const char *mongrp,
41462306a36Sopenharmony_ci				      int cpu_no, char *resctrl_val)
41562306a36Sopenharmony_ci{
41662306a36Sopenharmony_ci	int resource_id;
41762306a36Sopenharmony_ci
41862306a36Sopenharmony_ci	if (get_resource_id(cpu_no, &resource_id) < 0) {
41962306a36Sopenharmony_ci		perror("Could not get resource_id");
42062306a36Sopenharmony_ci		return;
42162306a36Sopenharmony_ci	}
42262306a36Sopenharmony_ci
42362306a36Sopenharmony_ci	if (!strncmp(resctrl_val, MBM_STR, sizeof(MBM_STR)))
42462306a36Sopenharmony_ci		set_mbm_path(ctrlgrp, mongrp, resource_id);
42562306a36Sopenharmony_ci
42662306a36Sopenharmony_ci	if (!strncmp(resctrl_val, MBA_STR, sizeof(MBA_STR))) {
42762306a36Sopenharmony_ci		if (ctrlgrp)
42862306a36Sopenharmony_ci			sprintf(mbm_total_path, CON_MBM_LOCAL_BYTES_PATH,
42962306a36Sopenharmony_ci				RESCTRL_PATH, ctrlgrp, resource_id);
43062306a36Sopenharmony_ci		else
43162306a36Sopenharmony_ci			sprintf(mbm_total_path, MBM_LOCAL_BYTES_PATH,
43262306a36Sopenharmony_ci				RESCTRL_PATH, resource_id);
43362306a36Sopenharmony_ci	}
43462306a36Sopenharmony_ci}
43562306a36Sopenharmony_ci
43662306a36Sopenharmony_ci/*
43762306a36Sopenharmony_ci * Get MBM Local bytes as reported by resctrl FS
43862306a36Sopenharmony_ci * For MBM,
43962306a36Sopenharmony_ci * 1. If con_mon grp and mon grp are given, then read from con_mon grp's mon grp
44062306a36Sopenharmony_ci * 2. If only con_mon grp is given, then read from con_mon grp
44162306a36Sopenharmony_ci * 3. If both are not given, then read from root con_mon grp
44262306a36Sopenharmony_ci * For MBA,
44362306a36Sopenharmony_ci * 1. If con_mon grp is given, then read from it
44462306a36Sopenharmony_ci * 2. If con_mon grp is not given, then read from root con_mon grp
44562306a36Sopenharmony_ci */
44662306a36Sopenharmony_cistatic int get_mem_bw_resctrl(unsigned long *mbm_total)
44762306a36Sopenharmony_ci{
44862306a36Sopenharmony_ci	FILE *fp;
44962306a36Sopenharmony_ci
45062306a36Sopenharmony_ci	fp = fopen(mbm_total_path, "r");
45162306a36Sopenharmony_ci	if (!fp) {
45262306a36Sopenharmony_ci		perror("Failed to open total bw file");
45362306a36Sopenharmony_ci
45462306a36Sopenharmony_ci		return -1;
45562306a36Sopenharmony_ci	}
45662306a36Sopenharmony_ci	if (fscanf(fp, "%lu", mbm_total) <= 0) {
45762306a36Sopenharmony_ci		perror("Could not get mbm local bytes");
45862306a36Sopenharmony_ci		fclose(fp);
45962306a36Sopenharmony_ci
46062306a36Sopenharmony_ci		return -1;
46162306a36Sopenharmony_ci	}
46262306a36Sopenharmony_ci	fclose(fp);
46362306a36Sopenharmony_ci
46462306a36Sopenharmony_ci	return 0;
46562306a36Sopenharmony_ci}
46662306a36Sopenharmony_ci
46762306a36Sopenharmony_cipid_t bm_pid, ppid;
46862306a36Sopenharmony_ci
46962306a36Sopenharmony_civoid ctrlc_handler(int signum, siginfo_t *info, void *ptr)
47062306a36Sopenharmony_ci{
47162306a36Sopenharmony_ci	/* Only kill child after bm_pid is set after fork() */
47262306a36Sopenharmony_ci	if (bm_pid)
47362306a36Sopenharmony_ci		kill(bm_pid, SIGKILL);
47462306a36Sopenharmony_ci	umount_resctrlfs();
47562306a36Sopenharmony_ci	tests_cleanup();
47662306a36Sopenharmony_ci	ksft_print_msg("Ending\n\n");
47762306a36Sopenharmony_ci
47862306a36Sopenharmony_ci	exit(EXIT_SUCCESS);
47962306a36Sopenharmony_ci}
48062306a36Sopenharmony_ci
48162306a36Sopenharmony_ci/*
48262306a36Sopenharmony_ci * Register CTRL-C handler for parent, as it has to kill
48362306a36Sopenharmony_ci * child process before exiting.
48462306a36Sopenharmony_ci */
48562306a36Sopenharmony_ciint signal_handler_register(void)
48662306a36Sopenharmony_ci{
48762306a36Sopenharmony_ci	struct sigaction sigact = {};
48862306a36Sopenharmony_ci	int ret = 0;
48962306a36Sopenharmony_ci
49062306a36Sopenharmony_ci	bm_pid = 0;
49162306a36Sopenharmony_ci
49262306a36Sopenharmony_ci	sigact.sa_sigaction = ctrlc_handler;
49362306a36Sopenharmony_ci	sigemptyset(&sigact.sa_mask);
49462306a36Sopenharmony_ci	sigact.sa_flags = SA_SIGINFO;
49562306a36Sopenharmony_ci	if (sigaction(SIGINT, &sigact, NULL) ||
49662306a36Sopenharmony_ci	    sigaction(SIGTERM, &sigact, NULL) ||
49762306a36Sopenharmony_ci	    sigaction(SIGHUP, &sigact, NULL)) {
49862306a36Sopenharmony_ci		perror("# sigaction");
49962306a36Sopenharmony_ci		ret = -1;
50062306a36Sopenharmony_ci	}
50162306a36Sopenharmony_ci	return ret;
50262306a36Sopenharmony_ci}
50362306a36Sopenharmony_ci
50462306a36Sopenharmony_ci/*
50562306a36Sopenharmony_ci * Reset signal handler to SIG_DFL.
50662306a36Sopenharmony_ci * Non-Value return because the caller should keep
50762306a36Sopenharmony_ci * the error code of other path even if sigaction fails.
50862306a36Sopenharmony_ci */
50962306a36Sopenharmony_civoid signal_handler_unregister(void)
51062306a36Sopenharmony_ci{
51162306a36Sopenharmony_ci	struct sigaction sigact = {};
51262306a36Sopenharmony_ci
51362306a36Sopenharmony_ci	sigact.sa_handler = SIG_DFL;
51462306a36Sopenharmony_ci	sigemptyset(&sigact.sa_mask);
51562306a36Sopenharmony_ci	if (sigaction(SIGINT, &sigact, NULL) ||
51662306a36Sopenharmony_ci	    sigaction(SIGTERM, &sigact, NULL) ||
51762306a36Sopenharmony_ci	    sigaction(SIGHUP, &sigact, NULL)) {
51862306a36Sopenharmony_ci		perror("# sigaction");
51962306a36Sopenharmony_ci	}
52062306a36Sopenharmony_ci}
52162306a36Sopenharmony_ci
52262306a36Sopenharmony_ci/*
52362306a36Sopenharmony_ci * print_results_bw:	the memory bandwidth results are stored in a file
52462306a36Sopenharmony_ci * @filename:		file that stores the results
52562306a36Sopenharmony_ci * @bm_pid:		child pid that runs benchmark
52662306a36Sopenharmony_ci * @bw_imc:		perf imc counter value
52762306a36Sopenharmony_ci * @bw_resc:		memory bandwidth value
52862306a36Sopenharmony_ci *
52962306a36Sopenharmony_ci * Return:		0 on success. non-zero on failure.
53062306a36Sopenharmony_ci */
53162306a36Sopenharmony_cistatic int print_results_bw(char *filename,  int bm_pid, float bw_imc,
53262306a36Sopenharmony_ci			    unsigned long bw_resc)
53362306a36Sopenharmony_ci{
53462306a36Sopenharmony_ci	unsigned long diff = fabs(bw_imc - bw_resc);
53562306a36Sopenharmony_ci	FILE *fp;
53662306a36Sopenharmony_ci
53762306a36Sopenharmony_ci	if (strcmp(filename, "stdio") == 0 || strcmp(filename, "stderr") == 0) {
53862306a36Sopenharmony_ci		printf("Pid: %d \t Mem_BW_iMC: %f \t ", bm_pid, bw_imc);
53962306a36Sopenharmony_ci		printf("Mem_BW_resc: %lu \t Difference: %lu\n", bw_resc, diff);
54062306a36Sopenharmony_ci	} else {
54162306a36Sopenharmony_ci		fp = fopen(filename, "a");
54262306a36Sopenharmony_ci		if (!fp) {
54362306a36Sopenharmony_ci			perror("Cannot open results file");
54462306a36Sopenharmony_ci
54562306a36Sopenharmony_ci			return errno;
54662306a36Sopenharmony_ci		}
54762306a36Sopenharmony_ci		if (fprintf(fp, "Pid: %d \t Mem_BW_iMC: %f \t Mem_BW_resc: %lu \t Difference: %lu\n",
54862306a36Sopenharmony_ci			    bm_pid, bw_imc, bw_resc, diff) <= 0) {
54962306a36Sopenharmony_ci			fclose(fp);
55062306a36Sopenharmony_ci			perror("Could not log results.");
55162306a36Sopenharmony_ci
55262306a36Sopenharmony_ci			return errno;
55362306a36Sopenharmony_ci		}
55462306a36Sopenharmony_ci		fclose(fp);
55562306a36Sopenharmony_ci	}
55662306a36Sopenharmony_ci
55762306a36Sopenharmony_ci	return 0;
55862306a36Sopenharmony_ci}
55962306a36Sopenharmony_ci
56062306a36Sopenharmony_cistatic void set_cmt_path(const char *ctrlgrp, const char *mongrp, char sock_num)
56162306a36Sopenharmony_ci{
56262306a36Sopenharmony_ci	if (strlen(ctrlgrp) && strlen(mongrp))
56362306a36Sopenharmony_ci		sprintf(llc_occup_path,	CON_MON_LCC_OCCUP_PATH,	RESCTRL_PATH,
56462306a36Sopenharmony_ci			ctrlgrp, mongrp, sock_num);
56562306a36Sopenharmony_ci	else if (!strlen(ctrlgrp) && strlen(mongrp))
56662306a36Sopenharmony_ci		sprintf(llc_occup_path,	MON_LCC_OCCUP_PATH, RESCTRL_PATH,
56762306a36Sopenharmony_ci			mongrp, sock_num);
56862306a36Sopenharmony_ci	else if (strlen(ctrlgrp) && !strlen(mongrp))
56962306a36Sopenharmony_ci		sprintf(llc_occup_path,	CON_LCC_OCCUP_PATH, RESCTRL_PATH,
57062306a36Sopenharmony_ci			ctrlgrp, sock_num);
57162306a36Sopenharmony_ci	else if (!strlen(ctrlgrp) && !strlen(mongrp))
57262306a36Sopenharmony_ci		sprintf(llc_occup_path, LCC_OCCUP_PATH,	RESCTRL_PATH, sock_num);
57362306a36Sopenharmony_ci}
57462306a36Sopenharmony_ci
57562306a36Sopenharmony_ci/*
57662306a36Sopenharmony_ci * initialize_llc_occu_resctrl:	Appropriately populate "llc_occup_path"
57762306a36Sopenharmony_ci * @ctrlgrp:			Name of the control monitor group (con_mon grp)
57862306a36Sopenharmony_ci * @mongrp:			Name of the monitor group (mon grp)
57962306a36Sopenharmony_ci * @cpu_no:			CPU number that the benchmark PID is binded to
58062306a36Sopenharmony_ci * @resctrl_val:		Resctrl feature (Eg: cat, cmt.. etc)
58162306a36Sopenharmony_ci */
58262306a36Sopenharmony_cistatic void initialize_llc_occu_resctrl(const char *ctrlgrp, const char *mongrp,
58362306a36Sopenharmony_ci					int cpu_no, char *resctrl_val)
58462306a36Sopenharmony_ci{
58562306a36Sopenharmony_ci	int resource_id;
58662306a36Sopenharmony_ci
58762306a36Sopenharmony_ci	if (get_resource_id(cpu_no, &resource_id) < 0) {
58862306a36Sopenharmony_ci		perror("# Unable to resource_id");
58962306a36Sopenharmony_ci		return;
59062306a36Sopenharmony_ci	}
59162306a36Sopenharmony_ci
59262306a36Sopenharmony_ci	if (!strncmp(resctrl_val, CMT_STR, sizeof(CMT_STR)))
59362306a36Sopenharmony_ci		set_cmt_path(ctrlgrp, mongrp, resource_id);
59462306a36Sopenharmony_ci}
59562306a36Sopenharmony_ci
59662306a36Sopenharmony_cistatic int
59762306a36Sopenharmony_cimeasure_vals(struct resctrl_val_param *param, unsigned long *bw_resc_start)
59862306a36Sopenharmony_ci{
59962306a36Sopenharmony_ci	unsigned long bw_resc, bw_resc_end;
60062306a36Sopenharmony_ci	float bw_imc;
60162306a36Sopenharmony_ci	int ret;
60262306a36Sopenharmony_ci
60362306a36Sopenharmony_ci	/*
60462306a36Sopenharmony_ci	 * Measure memory bandwidth from resctrl and from
60562306a36Sopenharmony_ci	 * another source which is perf imc value or could
60662306a36Sopenharmony_ci	 * be something else if perf imc event is not available.
60762306a36Sopenharmony_ci	 * Compare the two values to validate resctrl value.
60862306a36Sopenharmony_ci	 * It takes 1sec to measure the data.
60962306a36Sopenharmony_ci	 */
61062306a36Sopenharmony_ci	ret = get_mem_bw_imc(param->cpu_no, param->bw_report, &bw_imc);
61162306a36Sopenharmony_ci	if (ret < 0)
61262306a36Sopenharmony_ci		return ret;
61362306a36Sopenharmony_ci
61462306a36Sopenharmony_ci	ret = get_mem_bw_resctrl(&bw_resc_end);
61562306a36Sopenharmony_ci	if (ret < 0)
61662306a36Sopenharmony_ci		return ret;
61762306a36Sopenharmony_ci
61862306a36Sopenharmony_ci	bw_resc = (bw_resc_end - *bw_resc_start) / MB;
61962306a36Sopenharmony_ci	ret = print_results_bw(param->filename, bm_pid, bw_imc, bw_resc);
62062306a36Sopenharmony_ci	if (ret)
62162306a36Sopenharmony_ci		return ret;
62262306a36Sopenharmony_ci
62362306a36Sopenharmony_ci	*bw_resc_start = bw_resc_end;
62462306a36Sopenharmony_ci
62562306a36Sopenharmony_ci	return 0;
62662306a36Sopenharmony_ci}
62762306a36Sopenharmony_ci
62862306a36Sopenharmony_ci/*
62962306a36Sopenharmony_ci * resctrl_val:	execute benchmark and measure memory bandwidth on
63062306a36Sopenharmony_ci *			the benchmark
63162306a36Sopenharmony_ci * @benchmark_cmd:	benchmark command and its arguments
63262306a36Sopenharmony_ci * @param:		parameters passed to resctrl_val()
63362306a36Sopenharmony_ci *
63462306a36Sopenharmony_ci * Return:		0 on success. non-zero on failure.
63562306a36Sopenharmony_ci */
63662306a36Sopenharmony_ciint resctrl_val(const char * const *benchmark_cmd, struct resctrl_val_param *param)
63762306a36Sopenharmony_ci{
63862306a36Sopenharmony_ci	char *resctrl_val = param->resctrl_val;
63962306a36Sopenharmony_ci	unsigned long bw_resc_start = 0;
64062306a36Sopenharmony_ci	struct sigaction sigact;
64162306a36Sopenharmony_ci	int ret = 0, pipefd[2];
64262306a36Sopenharmony_ci	char pipe_message = 0;
64362306a36Sopenharmony_ci	union sigval value;
64462306a36Sopenharmony_ci
64562306a36Sopenharmony_ci	if (strcmp(param->filename, "") == 0)
64662306a36Sopenharmony_ci		sprintf(param->filename, "stdio");
64762306a36Sopenharmony_ci
64862306a36Sopenharmony_ci	if (!strncmp(resctrl_val, MBA_STR, sizeof(MBA_STR)) ||
64962306a36Sopenharmony_ci	    !strncmp(resctrl_val, MBM_STR, sizeof(MBM_STR))) {
65062306a36Sopenharmony_ci		ret = validate_bw_report_request(param->bw_report);
65162306a36Sopenharmony_ci		if (ret)
65262306a36Sopenharmony_ci			return ret;
65362306a36Sopenharmony_ci	}
65462306a36Sopenharmony_ci
65562306a36Sopenharmony_ci	/*
65662306a36Sopenharmony_ci	 * If benchmark wasn't successfully started by child, then child should
65762306a36Sopenharmony_ci	 * kill parent, so save parent's pid
65862306a36Sopenharmony_ci	 */
65962306a36Sopenharmony_ci	ppid = getpid();
66062306a36Sopenharmony_ci
66162306a36Sopenharmony_ci	if (pipe(pipefd)) {
66262306a36Sopenharmony_ci		perror("# Unable to create pipe");
66362306a36Sopenharmony_ci
66462306a36Sopenharmony_ci		return -1;
66562306a36Sopenharmony_ci	}
66662306a36Sopenharmony_ci
66762306a36Sopenharmony_ci	/*
66862306a36Sopenharmony_ci	 * Fork to start benchmark, save child's pid so that it can be killed
66962306a36Sopenharmony_ci	 * when needed
67062306a36Sopenharmony_ci	 */
67162306a36Sopenharmony_ci	fflush(stdout);
67262306a36Sopenharmony_ci	bm_pid = fork();
67362306a36Sopenharmony_ci	if (bm_pid == -1) {
67462306a36Sopenharmony_ci		perror("# Unable to fork");
67562306a36Sopenharmony_ci
67662306a36Sopenharmony_ci		return -1;
67762306a36Sopenharmony_ci	}
67862306a36Sopenharmony_ci
67962306a36Sopenharmony_ci	if (bm_pid == 0) {
68062306a36Sopenharmony_ci		/*
68162306a36Sopenharmony_ci		 * Mask all signals except SIGUSR1, parent uses SIGUSR1 to
68262306a36Sopenharmony_ci		 * start benchmark
68362306a36Sopenharmony_ci		 */
68462306a36Sopenharmony_ci		sigfillset(&sigact.sa_mask);
68562306a36Sopenharmony_ci		sigdelset(&sigact.sa_mask, SIGUSR1);
68662306a36Sopenharmony_ci
68762306a36Sopenharmony_ci		sigact.sa_sigaction = run_benchmark;
68862306a36Sopenharmony_ci		sigact.sa_flags = SA_SIGINFO;
68962306a36Sopenharmony_ci
69062306a36Sopenharmony_ci		/* Register for "SIGUSR1" signal from parent */
69162306a36Sopenharmony_ci		if (sigaction(SIGUSR1, &sigact, NULL))
69262306a36Sopenharmony_ci			PARENT_EXIT("Can't register child for signal");
69362306a36Sopenharmony_ci
69462306a36Sopenharmony_ci		/* Tell parent that child is ready */
69562306a36Sopenharmony_ci		close(pipefd[0]);
69662306a36Sopenharmony_ci		pipe_message = 1;
69762306a36Sopenharmony_ci		if (write(pipefd[1], &pipe_message, sizeof(pipe_message)) <
69862306a36Sopenharmony_ci		    sizeof(pipe_message)) {
69962306a36Sopenharmony_ci			perror("# failed signaling parent process");
70062306a36Sopenharmony_ci			close(pipefd[1]);
70162306a36Sopenharmony_ci			return -1;
70262306a36Sopenharmony_ci		}
70362306a36Sopenharmony_ci		close(pipefd[1]);
70462306a36Sopenharmony_ci
70562306a36Sopenharmony_ci		/* Suspend child until delivery of "SIGUSR1" from parent */
70662306a36Sopenharmony_ci		sigsuspend(&sigact.sa_mask);
70762306a36Sopenharmony_ci
70862306a36Sopenharmony_ci		PARENT_EXIT("Child is done");
70962306a36Sopenharmony_ci	}
71062306a36Sopenharmony_ci
71162306a36Sopenharmony_ci	ksft_print_msg("Benchmark PID: %d\n", bm_pid);
71262306a36Sopenharmony_ci
71362306a36Sopenharmony_ci	/*
71462306a36Sopenharmony_ci	 * The cast removes constness but nothing mutates benchmark_cmd within
71562306a36Sopenharmony_ci	 * the context of this process. At the receiving process, it becomes
71662306a36Sopenharmony_ci	 * argv, which is mutable, on exec() but that's after fork() so it
71762306a36Sopenharmony_ci	 * doesn't matter for the process running the tests.
71862306a36Sopenharmony_ci	 */
71962306a36Sopenharmony_ci	value.sival_ptr = (void *)benchmark_cmd;
72062306a36Sopenharmony_ci
72162306a36Sopenharmony_ci	/* Taskset benchmark to specified cpu */
72262306a36Sopenharmony_ci	ret = taskset_benchmark(bm_pid, param->cpu_no);
72362306a36Sopenharmony_ci	if (ret)
72462306a36Sopenharmony_ci		goto out;
72562306a36Sopenharmony_ci
72662306a36Sopenharmony_ci	/* Write benchmark to specified control&monitoring grp in resctrl FS */
72762306a36Sopenharmony_ci	ret = write_bm_pid_to_resctrl(bm_pid, param->ctrlgrp, param->mongrp,
72862306a36Sopenharmony_ci				      resctrl_val);
72962306a36Sopenharmony_ci	if (ret)
73062306a36Sopenharmony_ci		goto out;
73162306a36Sopenharmony_ci
73262306a36Sopenharmony_ci	if (!strncmp(resctrl_val, MBM_STR, sizeof(MBM_STR)) ||
73362306a36Sopenharmony_ci	    !strncmp(resctrl_val, MBA_STR, sizeof(MBA_STR))) {
73462306a36Sopenharmony_ci		ret = initialize_mem_bw_imc();
73562306a36Sopenharmony_ci		if (ret)
73662306a36Sopenharmony_ci			goto out;
73762306a36Sopenharmony_ci
73862306a36Sopenharmony_ci		initialize_mem_bw_resctrl(param->ctrlgrp, param->mongrp,
73962306a36Sopenharmony_ci					  param->cpu_no, resctrl_val);
74062306a36Sopenharmony_ci	} else if (!strncmp(resctrl_val, CMT_STR, sizeof(CMT_STR)))
74162306a36Sopenharmony_ci		initialize_llc_occu_resctrl(param->ctrlgrp, param->mongrp,
74262306a36Sopenharmony_ci					    param->cpu_no, resctrl_val);
74362306a36Sopenharmony_ci
74462306a36Sopenharmony_ci	/* Parent waits for child to be ready. */
74562306a36Sopenharmony_ci	close(pipefd[1]);
74662306a36Sopenharmony_ci	while (pipe_message != 1) {
74762306a36Sopenharmony_ci		if (read(pipefd[0], &pipe_message, sizeof(pipe_message)) <
74862306a36Sopenharmony_ci		    sizeof(pipe_message)) {
74962306a36Sopenharmony_ci			perror("# failed reading message from child process");
75062306a36Sopenharmony_ci			close(pipefd[0]);
75162306a36Sopenharmony_ci			goto out;
75262306a36Sopenharmony_ci		}
75362306a36Sopenharmony_ci	}
75462306a36Sopenharmony_ci	close(pipefd[0]);
75562306a36Sopenharmony_ci
75662306a36Sopenharmony_ci	/* Signal child to start benchmark */
75762306a36Sopenharmony_ci	if (sigqueue(bm_pid, SIGUSR1, value) == -1) {
75862306a36Sopenharmony_ci		perror("# sigqueue SIGUSR1 to child");
75962306a36Sopenharmony_ci		ret = errno;
76062306a36Sopenharmony_ci		goto out;
76162306a36Sopenharmony_ci	}
76262306a36Sopenharmony_ci
76362306a36Sopenharmony_ci	/* Give benchmark enough time to fully run */
76462306a36Sopenharmony_ci	sleep(1);
76562306a36Sopenharmony_ci
76662306a36Sopenharmony_ci	/* Test runs until the callback setup() tells the test to stop. */
76762306a36Sopenharmony_ci	while (1) {
76862306a36Sopenharmony_ci		ret = param->setup(param);
76962306a36Sopenharmony_ci		if (ret == END_OF_TESTS) {
77062306a36Sopenharmony_ci			ret = 0;
77162306a36Sopenharmony_ci			break;
77262306a36Sopenharmony_ci		}
77362306a36Sopenharmony_ci		if (ret < 0)
77462306a36Sopenharmony_ci			break;
77562306a36Sopenharmony_ci
77662306a36Sopenharmony_ci		if (!strncmp(resctrl_val, MBM_STR, sizeof(MBM_STR)) ||
77762306a36Sopenharmony_ci		    !strncmp(resctrl_val, MBA_STR, sizeof(MBA_STR))) {
77862306a36Sopenharmony_ci			ret = measure_vals(param, &bw_resc_start);
77962306a36Sopenharmony_ci			if (ret)
78062306a36Sopenharmony_ci				break;
78162306a36Sopenharmony_ci		} else if (!strncmp(resctrl_val, CMT_STR, sizeof(CMT_STR))) {
78262306a36Sopenharmony_ci			sleep(1);
78362306a36Sopenharmony_ci			ret = measure_cache_vals(param, bm_pid);
78462306a36Sopenharmony_ci			if (ret)
78562306a36Sopenharmony_ci				break;
78662306a36Sopenharmony_ci		}
78762306a36Sopenharmony_ci	}
78862306a36Sopenharmony_ci
78962306a36Sopenharmony_ciout:
79062306a36Sopenharmony_ci	kill(bm_pid, SIGKILL);
79162306a36Sopenharmony_ci
79262306a36Sopenharmony_ci	return ret;
79362306a36Sopenharmony_ci}
794