1b2441318SGreg Kroah-Hartman /* SPDX-License-Identifier: GPL-2.0 */
2108fc825SYoshihiro YUNOMAE #ifndef __TRACE_AGENT_H__
3108fc825SYoshihiro YUNOMAE #define __TRACE_AGENT_H__
4108fc825SYoshihiro YUNOMAE #include <pthread.h>
5108fc825SYoshihiro YUNOMAE #include <stdbool.h>
6108fc825SYoshihiro YUNOMAE 
7108fc825SYoshihiro YUNOMAE #define MAX_CPUS	256
8108fc825SYoshihiro YUNOMAE #define PIPE_INIT       (1024*1024)
9108fc825SYoshihiro YUNOMAE 
10108fc825SYoshihiro YUNOMAE /*
11108fc825SYoshihiro YUNOMAE  * agent_info - structure managing total information of guest agent
12108fc825SYoshihiro YUNOMAE  * @pipe_size:	size of pipe (default 1MB)
13108fc825SYoshihiro YUNOMAE  * @use_stdout:	set to true when o option is added (default false)
14108fc825SYoshihiro YUNOMAE  * @cpus:	total number of CPUs
15108fc825SYoshihiro YUNOMAE  * @ctl_fd:	fd of control path, /dev/virtio-ports/agent-ctl-path
16108fc825SYoshihiro YUNOMAE  * @rw_ti:	structure managing information of read/write threads
17108fc825SYoshihiro YUNOMAE  */
18108fc825SYoshihiro YUNOMAE struct agent_info {
19108fc825SYoshihiro YUNOMAE 	unsigned long pipe_size;
20108fc825SYoshihiro YUNOMAE 	bool use_stdout;
21108fc825SYoshihiro YUNOMAE 	int cpus;
22108fc825SYoshihiro YUNOMAE 	int ctl_fd;
23108fc825SYoshihiro YUNOMAE 	struct rw_thread_info *rw_ti[MAX_CPUS];
24108fc825SYoshihiro YUNOMAE };
25108fc825SYoshihiro YUNOMAE 
26108fc825SYoshihiro YUNOMAE /*
27108fc825SYoshihiro YUNOMAE  * rw_thread_info - structure managing a read/write thread a cpu
28108fc825SYoshihiro YUNOMAE  * @cpu_num:	cpu number operating this read/write thread
29108fc825SYoshihiro YUNOMAE  * @in_fd:	fd of reading trace data path in cpu_num
30108fc825SYoshihiro YUNOMAE  * @out_fd:	fd of writing trace data path in cpu_num
31108fc825SYoshihiro YUNOMAE  * @read_pipe:	fd of read pipe
32108fc825SYoshihiro YUNOMAE  * @write_pipe:	fd of write pipe
33108fc825SYoshihiro YUNOMAE  * @pipe_size:	size of pipe (default 1MB)
34108fc825SYoshihiro YUNOMAE  */
35108fc825SYoshihiro YUNOMAE struct rw_thread_info {
36108fc825SYoshihiro YUNOMAE 	int cpu_num;
37108fc825SYoshihiro YUNOMAE 	int in_fd;
38108fc825SYoshihiro YUNOMAE 	int out_fd;
39108fc825SYoshihiro YUNOMAE 	int read_pipe;
40108fc825SYoshihiro YUNOMAE 	int write_pipe;
41108fc825SYoshihiro YUNOMAE 	unsigned long pipe_size;
42108fc825SYoshihiro YUNOMAE };
43108fc825SYoshihiro YUNOMAE 
44108fc825SYoshihiro YUNOMAE /* use for stopping rw threads */
45108fc825SYoshihiro YUNOMAE extern bool global_sig_receive;
46108fc825SYoshihiro YUNOMAE 
47108fc825SYoshihiro YUNOMAE /* use for notification */
48108fc825SYoshihiro YUNOMAE extern bool global_run_operation;
49108fc825SYoshihiro YUNOMAE extern pthread_mutex_t mutex_notify;
50108fc825SYoshihiro YUNOMAE extern pthread_cond_t cond_wakeup;
51108fc825SYoshihiro YUNOMAE 
52108fc825SYoshihiro YUNOMAE /* for controller of read/write threads */
53108fc825SYoshihiro YUNOMAE extern int rw_ctl_init(const char *ctl_path);
54108fc825SYoshihiro YUNOMAE extern void *rw_ctl_loop(int ctl_fd);
55108fc825SYoshihiro YUNOMAE 
56108fc825SYoshihiro YUNOMAE /* for trace read/write thread */
57108fc825SYoshihiro YUNOMAE extern void *rw_thread_info_new(void);
58108fc825SYoshihiro YUNOMAE extern void *rw_thread_init(int cpu, const char *in_path, const char *out_path,
59108fc825SYoshihiro YUNOMAE 			bool stdout_flag, unsigned long pipe_size,
60108fc825SYoshihiro YUNOMAE 			struct rw_thread_info *rw_ti);
61108fc825SYoshihiro YUNOMAE extern pthread_t rw_thread_run(struct rw_thread_info *rw_ti);
62108fc825SYoshihiro YUNOMAE 
zalloc(size_t size)63108fc825SYoshihiro YUNOMAE static inline void *zalloc(size_t size)
64108fc825SYoshihiro YUNOMAE {
65108fc825SYoshihiro YUNOMAE 	return calloc(1, size);
66108fc825SYoshihiro YUNOMAE }
67108fc825SYoshihiro YUNOMAE 
68108fc825SYoshihiro YUNOMAE #define pr_err(format, ...) fprintf(stderr, format, ## __VA_ARGS__)
69108fc825SYoshihiro YUNOMAE #define pr_info(format, ...) fprintf(stdout, format, ## __VA_ARGS__)
70108fc825SYoshihiro YUNOMAE #ifdef DEBUG
71108fc825SYoshihiro YUNOMAE #define pr_debug(format, ...) fprintf(stderr, format, ## __VA_ARGS__)
72108fc825SYoshihiro YUNOMAE #else
73108fc825SYoshihiro YUNOMAE #define pr_debug(format, ...) do {} while (0)
74108fc825SYoshihiro YUNOMAE #endif
75108fc825SYoshihiro YUNOMAE 
76108fc825SYoshihiro YUNOMAE #endif /*__TRACE_AGENT_H__*/
77