toaruos/apps/tee.c

64 lines
1.2 KiB
C

/**
* @brief tee - copy stdin to stdout and to specified files
*
* @copyright
* This file is part of ToaruOS and is released under the terms
* of the NCSA / University of Illinois License - see LICENSE.md
* Copyright (C) 2018 K. Lange
*/
#include <stdio.h>
#include <unistd.h>
#include <getopt.h>
#include <signal.h>
#include <string.h>
#include <errno.h>
int main(int argc, char * argv[]) {
int append = 0;
int ret_val = 0;
int opt;
while ((opt = getopt(argc, argv, "ai")) != -1) {
switch (opt) {
case 'a':
append = 1;
break;
case 'i':
signal(SIGINT, SIG_IGN);
break;
}
}
int file_count = argc - optind;
FILE ** files = malloc(sizeof(FILE *) * file_count);
for (int i = 0, j = optind; j < argc && i < file_count; j++) {
files[i] = fopen(argv[j], append ? "a" : "w");
if (!files[i]) {
fprintf(stderr, "%s: %s: %s\n", argv[0], argv[j], strerror(errno));
ret_val = 1;
file_count--;
continue;
} else {
i++;
}
}
while (!feof(stdin)) {
int c = fgetc(stdin);
if (c >= 0) {
fputc(c, stdout);
for (int i = 0; i < file_count; ++i) {
fputc(c, files[i]);
}
}
}
for (int i = 0; i < file_count; ++i) {
fclose(files[i]);
}
return ret_val;
}