forked from dan/liminix
94 lines
2.1 KiB
C
94 lines
2.1 KiB
C
#include <poll.h>
|
|
#include <sys/timerfd.h>
|
|
#include <time.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <error.h>
|
|
#include <signal.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
|
|
int open_shipper_socket(char *pathname) {
|
|
int fd;
|
|
|
|
struct sockaddr_un sa = {
|
|
.sun_family = AF_LOCAL
|
|
};
|
|
strncpy(sa.sun_path, pathname, sizeof(sa.sun_path) - 1);
|
|
|
|
fd = socket(AF_LOCAL, SOCK_STREAM, 0);
|
|
if(fd >= 0) {
|
|
if(connect(fd, (struct sockaddr *) &sa, sizeof sa)) {
|
|
perror("connect socket");
|
|
return -1;
|
|
}
|
|
int flags = fcntl(fd, F_GETFL);
|
|
fcntl(fd, F_SETFL, flags | O_NONBLOCK);
|
|
}
|
|
return fd;
|
|
}
|
|
|
|
|
|
int main(int argc, char * argv[]) {
|
|
struct pollfd fds[] = {
|
|
{ .fd = 0, .events = POLLIN },
|
|
{ .fd = 1, .events = POLLERR },
|
|
{ .fd = -1, .events = POLLERR },
|
|
};
|
|
|
|
char * buf = malloc(8192);
|
|
int out_bytes = 0;
|
|
int tee_bytes = 0;
|
|
|
|
if(argc > 1 && (strlen(argv[1]) > 108)) {
|
|
error(1, 0, "socket pathname \"%s\" is too long, max 108 bytes",
|
|
argv[1]);
|
|
};
|
|
signal(SIGPIPE, SIG_IGN);
|
|
|
|
char * start_cookie = "COOKIE-START\n";
|
|
char * stop_cookie = "COOKIE-STOP\n";
|
|
|
|
int flags = fcntl(STDOUT_FILENO, F_GETFL);
|
|
fcntl(STDOUT_FILENO, F_SETFL, flags | O_NONBLOCK);
|
|
|
|
while(1) {
|
|
int nfds = poll(fds, 3, 2000);
|
|
if(nfds > 0) {
|
|
if((fds[0].revents & (POLLIN|POLLHUP)) &&
|
|
(out_bytes == 0) &&
|
|
(tee_bytes == 0)) {
|
|
out_bytes = read(fds[0].fd, buf, 8192);
|
|
if(out_bytes == 0) {
|
|
exit(0);
|
|
};
|
|
if(fds[2].fd >= 0) tee_bytes = out_bytes;
|
|
};
|
|
|
|
if(out_bytes) {
|
|
out_bytes -= write(fds[1].fd, buf, out_bytes);
|
|
};
|
|
if(fds[1].revents & (POLLERR|POLLHUP)) {
|
|
exit(1); // can't even log an error if the logging stream fails
|
|
};
|
|
if(fds[2].fd >= 0) {
|
|
if(tee_bytes) {
|
|
tee_bytes -= write(fds[2].fd, buf, tee_bytes);
|
|
};
|
|
if(fds[2].revents & (POLLERR|POLLHUP)) {
|
|
close(fds[2].fd);
|
|
fds[2].fd = -1;
|
|
(void) write(1, stop_cookie , strlen(stop_cookie));
|
|
};
|
|
};
|
|
} else {
|
|
if(fds[2].fd < 0) {
|
|
if(argc>1) fds[2].fd = open_shipper_socket(argv[1]);
|
|
}
|
|
};
|
|
};
|
|
}
|