1 // SPDX-License-Identifier: GPL-2.0
2
3 #include <stdint.h>
4 #include <time.h>
5 #include <sched.h>
6
7 /*
8 * '18446744073709551615\0'
9 */
10 #define BUFF_U64_STR_SIZE 24
11 #define MAX_PATH 1024
12 #define MAX_NICE 20
13 #define MIN_NICE -19
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
update_min(unsigned long long * a,unsigned long long * b)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
update_max(unsigned long long * a,unsigned long long * b)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
update_sum(unsigned long long * a,unsigned long long * b)44 update_sum(unsigned long long *a, unsigned long long *b)
45 {
46 *a += *b;
47 }
48
49 #ifndef SCHED_ATTR_SIZE_VER0
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 };
60 #endif /* SCHED_ATTR_SIZE_VER0 */
61
62 int parse_prio(char *arg, struct sched_attr *sched_param);
63 int parse_cpu_set(char *cpu_list, cpu_set_t *set);
64 int __set_sched_attr(int pid, struct sched_attr *attr);
65 int set_comm_sched_attr(const char *comm_prefix, struct sched_attr *attr);
66 int set_comm_cgroup(const char *comm_prefix, const char *cgroup);
67 int set_pid_cgroup(pid_t pid, const char *cgroup);
68 int set_cpu_dma_latency(int32_t latency);
69 int auto_house_keeping(cpu_set_t *monitored_cpus);
70
71 #define ns_to_usf(x) (((double)x/1000))
72 #define ns_to_per(total, part) ((part * 100) / (double)total)
73