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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#include <poll.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/wait.h>
#include <unistd.h>
int
filter(char *input, size_t inlen, char *cmd, char **outputo, size_t *outleno)
{
char *output;
ssize_t outlen;
ssize_t outalloc = 4096;
pid_t pid;
output = malloc(outalloc);
outlen = 0;
int pipe0[2];
int pipe1[2];
if (pipe(pipe0) != 0 || pipe(pipe1) != 0)
goto fail;
char *argv[] = { "/bin/sh", "-c", cmd, (char *)0 };
if (!(pid = fork())) {
dup2(pipe0[0], 0);
close(pipe0[1]);
close(pipe0[0]);
dup2(pipe1[1], 1);
close(pipe1[0]);
close(pipe1[1]);
execvp(argv[0], argv);
exit(-1);
}
close(pipe0[0]);
close(pipe1[1]);
if (pid < 0) {
close(pipe0[1]);
close(pipe1[0]);
goto fail;
}
struct pollfd fds[2];
fds[0].fd = pipe1[0];
fds[0].events = POLLIN | POLLHUP;
fds[1].fd = pipe0[1];
fds[1].events = POLLOUT;
while ((fds[0].fd >= 0 || fds[1].fd >= 0) &&
poll(fds, sizeof fds / sizeof fds[0], -1) >= 0) {
if (fds[0].revents & POLLIN) {
if (outlen + 512 > outalloc) {
outalloc *= 2;
if (outalloc < 0)
exit(-1);
output = realloc(output, outalloc);
if (!output)
exit(-1);
}
ssize_t ret = read(fds[0].fd, output + outlen, 512);
if (ret > 0)
outlen += ret;
else
close(fds[0].fd);
} else if (fds[0].revents & (POLLERR | POLLHUP | POLLNVAL)) {
fds[0].fd = -1;
}
if (fds[1].revents & POLLOUT) {
ssize_t ret = write(fds[1].fd, input, inlen);
if (ret > 0) {
input += ret;
inlen -= ret;
}
if (ret <= 0 || inlen == 0)
close(fds[1].fd);
} else if (fds[1].revents & (POLLERR | POLLHUP | POLLNVAL)) {
fds[1].fd = -1;
}
}
// ok to fail when closed already
close(pipe0[1]);
close(pipe1[0]);
int status;
waitpid(pid, &status, 0);
*outputo = output;
*outleno = outlen;
return WEXITSTATUS(status);
fail:
*outputo = 0;
*outleno = 0;
return -1;
}
#ifdef TEST
int
main()
{
char *input = "foo\nbar\nbaz";
int e;
char *output;
size_t outlen;
e = filter(input, strlen(input), "rev;exit 2", &output, &outlen);
fwrite(output, 1, outlen, stdout);
printf("%ld -> %d\n", outlen, e);
return 0;
}
#endif
|
