1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Controller of read/write threads for virtio-trace
4  *
5  * Copyright (C) 2012 Hitachi, Ltd.
6  * Created by Yoshihiro Yunomae <yoshihiro.yunomae.ez@hitachi.com>
7  *            Masami Hiramatsu <masami.hiramatsu.pt@hitachi.com>
8  */
9 
10 #define _GNU_SOURCE
11 #include <fcntl.h>
12 #include <poll.h>
13 #include <signal.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <unistd.h>
17 #include "trace-agent.h"
18 
19 #define HOST_MSG_SIZE		256
20 #define EVENT_WAIT_MSEC		100
21 
22 static volatile sig_atomic_t global_signal_val;
23 bool global_sig_receive;	/* default false */
24 bool global_run_operation;	/* default false*/
25 
26 /* Handle SIGTERM/SIGINT/SIGQUIT to exit */
27 static void signal_handler(int sig)
28 {
29 	global_signal_val = sig;
30 }
31 
32 int rw_ctl_init(const char *ctl_path)
33 {
34 	int ctl_fd;
35 
36 	ctl_fd = open(ctl_path, O_RDONLY);
37 	if (ctl_fd == -1) {
38 		pr_err("Cannot open ctl_fd\n");
39 		goto error;
40 	}
41 
42 	return ctl_fd;
43 
44 error:
45 	exit(EXIT_FAILURE);
46 }
47 
48 static int wait_order(int ctl_fd)
49 {
50 	struct pollfd poll_fd;
51 	int ret = 0;
52 
53 	while (!global_sig_receive) {
54 		poll_fd.fd = ctl_fd;
55 		poll_fd.events = POLLIN;
56 
57 		ret = poll(&poll_fd, 1, EVENT_WAIT_MSEC);
58 
59 		if (global_signal_val) {
60 			global_sig_receive = true;
61 			pr_info("Receive interrupt %d\n", global_signal_val);
62 
63 			/* Wakes rw-threads when they are sleeping */
64 			if (!global_run_operation)
65 				pthread_cond_broadcast(&cond_wakeup);
66 
67 			ret = -1;
68 			break;
69 		}
70 
71 		if (ret < 0) {
72 			pr_err("Polling error\n");
73 			goto error;
74 		}
75 
76 		if (ret)
77 			break;
78 	}
79 
80 	return ret;
81 
82 error:
83 	exit(EXIT_FAILURE);
84 }
85 
86 /*
87  * contol read/write threads by handling global_run_operation
88  */
89 void *rw_ctl_loop(int ctl_fd)
90 {
91 	ssize_t rlen;
92 	char buf[HOST_MSG_SIZE];
93 	int ret;
94 
95 	/* Setup signal handlers */
96 	signal(SIGTERM, signal_handler);
97 	signal(SIGINT, signal_handler);
98 	signal(SIGQUIT, signal_handler);
99 
100 	while (!global_sig_receive) {
101 
102 		ret = wait_order(ctl_fd);
103 		if (ret < 0)
104 			break;
105 
106 		rlen = read(ctl_fd, buf, sizeof(buf));
107 		if (rlen < 0) {
108 			pr_err("read data error in ctl thread\n");
109 			goto error;
110 		}
111 
112 		if (rlen == 2 && buf[0] == '1') {
113 			/*
114 			 * If host writes '1' to a control path,
115 			 * this controller wakes all read/write threads.
116 			 */
117 			global_run_operation = true;
118 			pthread_cond_broadcast(&cond_wakeup);
119 			pr_debug("Wake up all read/write threads\n");
120 		} else if (rlen == 2 && buf[0] == '0') {
121 			/*
122 			 * If host writes '0' to a control path, read/write
123 			 * threads will wait for notification from Host.
124 			 */
125 			global_run_operation = false;
126 			pr_debug("Stop all read/write threads\n");
127 		} else
128 			pr_info("Invalid host notification: %s\n", buf);
129 	}
130 
131 	return NULL;
132 
133 error:
134 	exit(EXIT_FAILURE);
135 }
136