[linux-block.git] / tools / tracing / rtla / src / utils.h

// SPDX-License-Identifier: GPL-2.0

#include <stdint.h>
#include <time.h>
#include <sched.h>

/*
 * '18446744073709551615\0'
 */
#define BUFF_U64_STR_SIZE	24
#define MAX_PATH		1024
#define MAX_NICE		20
#define MIN_NICE		-19

#define container_of(ptr, type, member)({			\
	const typeof(((type *)0)->member) *__mptr = (ptr);	\
	(type *)((char *)__mptr - offsetof(type, member)) ; })

extern int config_debug;
void debug_msg(const char *fmt, ...);
void err_msg(const char *fmt, ...);

long parse_seconds_duration(char *val);
void get_duration(time_t start_time, char *output, int output_size);

int parse_cpu_list(char *cpu_list, char **monitored_cpus);
long long get_llong_from_str(char *start);

static inline void
update_min(unsigned long long *a, unsigned long long *b)
{
	if (*a > *b)
		*a = *b;
}

static inline void
update_max(unsigned long long *a, unsigned long long *b)
{
	if (*a < *b)
		*a = *b;
}

static inline void
update_sum(unsigned long long *a, unsigned long long *b)
{
	*a += *b;
}

#ifndef SCHED_ATTR_SIZE_VER0
struct sched_attr {
	uint32_t size;
	uint32_t sched_policy;
	uint64_t sched_flags;
	int32_t sched_nice;
	uint32_t sched_priority;
	uint64_t sched_runtime;
	uint64_t sched_deadline;
	uint64_t sched_period;
};
#endif /* SCHED_ATTR_SIZE_VER0 */

int parse_prio(char *arg, struct sched_attr *sched_param);
int parse_cpu_set(char *cpu_list, cpu_set_t *set);
int __set_sched_attr(int pid, struct sched_attr *attr);
int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr);
int set_comm_cgroup(const char *comm_prefix, const char *cgroup);
int set_pid_cgroup(pid_t pid, const char *cgroup);
int set_cpu_dma_latency(int32_t latency);
#ifdef HAVE_LIBCPUPOWER_SUPPORT
int save_cpu_idle_disable_state(unsigned int cpu);
int restore_cpu_idle_disable_state(unsigned int cpu);
void free_cpu_idle_disable_states(void);
int set_deepest_cpu_idle_state(unsigned int cpu, unsigned int state);
static inline int have_libcpupower_support(void) { return 1; }
#else
static inline int save_cpu_idle_disable_state(unsigned int cpu) { return -1; }
static inline int restore_cpu_idle_disable_state(unsigned int cpu) { return -1; }
static inline void free_cpu_idle_disable_states(void) { }
static inline int set_deepest_cpu_idle_state(unsigned int cpu, unsigned int state) { return -1; }
static inline int have_libcpupower_support(void) { return 0; }
#endif /* HAVE_LIBCPUPOWER_SUPPORT */
int auto_house_keeping(cpu_set_t *monitored_cpus);

#define ns_to_usf(x) (((double)x/1000))
#define ns_to_per(total, part) ((part * 100) / (double)total)

enum result {
	PASSED = 0, /* same as EXIT_SUCCESS */
	ERROR = 1,  /* same as EXIT_FAILURE, an error in arguments */
	FAILED = 2, /* test hit the stop tracing condition */
};
Commit	Line	Data
b1696371	1	// SPDX-License-Identifier: GPL-2.0
272ced25	2
b1696371 DBO	3	#include <stdint.h>
b1696371 DBO	4	#include <time.h>
272ced25	5	#include <sched.h>
b1696371 DBO	6
	7	/*
	8	* '18446744073709551615\0'
	9	*/
	10	#define BUFF_U64_STR_SIZE 24
dada03db	11	#define MAX_PATH 1024
14f08c97	12	#define MAX_NICE 20
14f08c97	13	#define MIN_NICE -19
b1696371 DBO	14
	15	#define container_of(ptr, type, member)({ \
	16	const typeof(((type )0)->member) __mptr = (ptr); \
	17	(type )((char )__mptr - offsetof(type, member)) ; })
	18
	19	extern int config_debug;
	20	void debug_msg(const char *fmt, ...);
	21	void err_msg(const char *fmt, ...);
	22
	23	long parse_seconds_duration(char *val);
	24	void get_duration(time_t start_time, char *output, int output_size);
	25
	26	int parse_cpu_list(char cpu_list, char *monitored_cpus);
	27	long long get_llong_from_str(char *start);
	28
	29	static inline void
	30	update_min(unsigned long long a, unsigned long long b)
	31	{
	32	if (a > b)
	33	a = b;
	34	}
	35
	36	static inline void
	37	update_max(unsigned long long a, unsigned long long b)
	38	{
	39	if (a < b)
	40	a = b;
	41	}
	42
	43	static inline void
	44	update_sum(unsigned long long a, unsigned long long b)
	45	{
	46	a += b;
	47	}
	48
0eecee34	49	#ifndef SCHED_ATTR_SIZE_VER0
b1696371 DBO	50	struct sched_attr {
	51	uint32_t size;
	52	uint32_t sched_policy;
	53	uint64_t sched_flags;
	54	int32_t sched_nice;
	55	uint32_t sched_priority;
	56	uint64_t sched_runtime;
	57	uint64_t sched_deadline;
	58	uint64_t sched_period;
	59	};
0eecee34	60	#endif /* SCHED_ATTR_SIZE_VER0 */
b1696371 DBO	61
b1696371 DBO	62	int parse_prio(char arg, struct sched_attr sched_param);
272ced25	63	int parse_cpu_set(char cpu_list, cpu_set_t set);
cdca4f4e	64	int __set_sched_attr(int pid, struct sched_attr *attr);
dada03db	65	int set_comm_sched_attr(const char comm_prefix, struct sched_attr attr);
a957cbc0	66	int set_comm_cgroup(const char comm_prefix, const char cgroup);
cdca4f4e	67	int set_pid_cgroup(pid_t pid, const char *cgroup);
7d0dc957	68	int set_cpu_dma_latency(int32_t latency);
083d29d3 TG	69	#ifdef HAVE_LIBCPUPOWER_SUPPORT
	70	int save_cpu_idle_disable_state(unsigned int cpu);
	71	int restore_cpu_idle_disable_state(unsigned int cpu);
	72	void free_cpu_idle_disable_states(void);
	73	int set_deepest_cpu_idle_state(unsigned int cpu, unsigned int state);
	74	static inline int have_libcpupower_support(void) { return 1; }
	75	#else
	76	static inline int save_cpu_idle_disable_state(unsigned int cpu) { return -1; }
	77	static inline int restore_cpu_idle_disable_state(unsigned int cpu) { return -1; }
	78	static inline void free_cpu_idle_disable_states(void) { }
	79	static inline int set_deepest_cpu_idle_state(unsigned int cpu, unsigned int state) { return -1; }
	80	static inline int have_libcpupower_support(void) { return 0; }
	81	#endif /* HAVE_LIBCPUPOWER_SUPPORT */
c58a3f8c	82	int auto_house_keeping(cpu_set_t *monitored_cpus);
27e348b2 DBO	83
	84	#define ns_to_usf(x) (((double)x/1000))
	85	#define ns_to_per(total, part) ((part * 100) / (double)total)
18682166 CS	86
	87	enum result {
	88	PASSED = 0, /* same as EXIT_SUCCESS */
	89	ERROR = 1, /* same as EXIT_FAILURE, an error in arguments */
	90	FAILED = 2, /* test hit the stop tracing condition */
	91	};