1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
/*
*
* test self pipe
$ gcc \
module-test/self_pipe.c \
-std=gnu23 \
-O0 -ggdb3 -fno-omit-frame-pointer -fno-inline \
-o module-test/self_pipe.out
*/
#include <stdio.h>
#include <sys/select.h>
#include <unistd.h>
#include <signal.h>
#define TMP_SIZE 1024
typedef void handler_t(int);
handler_t *Signal(int signum, handler_t *handler);
void sigusr1_handler(int signum);
int my_read(char *s, int len);
int self_pipe_fd[2] = {-1, -1};
void sigusr1_handler(int signum)
{
char tmp = 'A';
write(self_pipe_fd[1], &tmp, 1);
}
int my_read(char *s, int len)
{
int max_fd;
if (STDIN_FILENO > self_pipe_fd[0]) max_fd = STDIN_FILENO;
else max_fd = self_pipe_fd[0];
fd_set rset;
FD_ZERO(&rset);
FD_SET(STDIN_FILENO, &rset);
FD_SET(self_pipe_fd[0], &rset);
select(max_fd+1, &rset, nullptr, nullptr, nullptr);
if (FD_ISSET(self_pipe_fd[0], &rset))
{
return read(self_pipe_fd[0], s, len);
}
else if (FD_ISSET(STDIN_FILENO, &rset))
{
return read(STDIN_FILENO, s, len);
}
return -1;
}
handler_t *Signal(int signum, handler_t *handler)
{
struct sigaction action, old_action;
action.sa_handler = handler;
sigemptyset(&action.sa_mask); /* block sigs of type being handled */
action.sa_flags = SA_RESTART; /* restart syscalls if possible */
sigaction(signum, &action, &old_action);
return (old_action.sa_handler);
}
int main()
{
printf("Do not enter anything\n");
Signal(SIGUSR1, sigusr1_handler);
pipe(self_pipe_fd);
int pid = getpid();
printf("Running PID: %d\n", pid);
printf("Do not enter anything\n");
char tmp[TMP_SIZE];
while (my_read(tmp, TMP_SIZE))
printf("Hi\n");
printf("Finished\n");
return 0;
}
|
