diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/Makefile.am | 2 | ||||
-rw-r--r-- | src/wye.c | 93 |
2 files changed, 95 insertions, 0 deletions
diff --git a/src/Makefile.am b/src/Makefile.am new file mode 100644 index 0000000..8490c6e --- /dev/null +++ b/src/Makefile.am @@ -0,0 +1,2 @@ +bin_PROGRAMS = wye +wye_SOURCES = wye.c diff --git a/src/wye.c b/src/wye.c new file mode 100644 index 0000000..17cbc3b --- /dev/null +++ b/src/wye.c @@ -0,0 +1,93 @@ +/* + * Copyright (C) 2020 - Vito Caputo - <vcaputo@pengaru.com> + * + * This program is free software: you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 3 as published + * by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include <assert.h> +#include <errno.h> +#include <fcntl.h> +#include <limits.h> +#include <poll.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/stat.h> +#include <sys/types.h> +#include <unistd.h> + +int main(int argc, char *argv[]) +{ + int n = argc; + struct pollfd *pfds; + + assert(argc >= 1); + + pfds = calloc(n, sizeof(*pfds)); + if (!pfds) { + fprintf(stderr, "Unable to allocate struct pollfd array: %s\n", + strerror(errno)); + return EXIT_FAILURE; + } + + for (int i = 0; i < n; i++) { + if (i > 0) { + const char *path = argv[i]; + + pfds[i].fd = open(path, O_RDONLY); + if (pfds[i].fd == -1) { + fprintf(stderr, "Unable to open \"%s\": %s\n", + path, strerror(errno)); + return EXIT_FAILURE; + } + } + pfds[i].events = POLLIN; + } + + while (poll(pfds, n, -1) > 0) { + for (int i = 0; i < n; i++) { + char buf[PIPE_BUF]; + ssize_t rr, rw; + + if (pfds[i].revents & POLLHUP) { + fprintf(stderr, "fd %i hangup, quitting\n", + pfds[i].fd); + return EXIT_SUCCESS; + } + + if (!(pfds[i].revents & POLLIN)) + continue; + + rr = read(pfds[i].fd, buf, sizeof(buf)); + if (rr < 0) { + fprintf(stderr, "Error reading fd %i: %s\n", + pfds[i].fd, strerror(errno)); + return EXIT_FAILURE; + } + + rw = write(1, buf, rr); + if (rw < 0) { + fprintf(stderr, "Error writing: %s\n", + strerror(errno)); + return EXIT_FAILURE; + } + + if (rw != rr) { + fprintf(stderr, "Short write\n"); + return EXIT_FAILURE; + } + } + } + + return EXIT_FAILURE; +} |