diff --git a/userspace/tee/main.cpp b/userspace/tee/main.cpp index 31b15cea..bf533e18 100644 --- a/userspace/tee/main.cpp +++ b/userspace/tee/main.cpp @@ -9,56 +9,56 @@ int main(int argc, char** argv) { - int files[MAX_FILES] {}; - size_t file_count = 0; + int files[MAX_FILES] {}; + size_t file_count = 0; - int arg = 1; + int arg = 1; - int oflag = O_WRONLY | O_CREAT; - if (arg < argc && strcmp(argv[arg], "-a") == 0) - { - oflag |= O_APPEND; - arg++; - } - else - { - oflag |= O_TRUNC; - } + int oflag = O_WRONLY | O_CREAT; + if (arg < argc && strcmp(argv[arg], "-a") == 0) + { + oflag |= O_APPEND; + arg++; + } + else + { + oflag |= O_TRUNC; + } - for (int i = arg; i < argc; i++) - { - files[file_count] = open(argv[i], oflag, 0644); - if (files[file_count] == -1) - perror(argv[i]); - else - file_count++; - - if (file_count >= MAX_FILES) - { - fprintf(stderr, "only up to %d files are supported\n", MAX_FILES); - break; - } - } + for (int i = arg; i < argc; i++) + { + files[file_count] = open(argv[i], oflag, 0644); + if (files[file_count] == -1) + perror(argv[i]); + else + file_count++; + + if (file_count >= MAX_FILES) + { + fprintf(stderr, "only up to %d files are supported\n", MAX_FILES); + break; + } + } - char* buffer = (char*)malloc(BUF_SIZE); - for (;;) - { - ssize_t nread = read(STDIN_FILENO, buffer, BUF_SIZE); - if (nread == -1) - perror("stdin"); - if (nread <= 0) - break; - write(STDOUT_FILENO, buffer, nread); - for (size_t i = 0; i < file_count; i++) - write(files[i], buffer, nread); - } - free(buffer); + char* buffer = (char*)malloc(BUF_SIZE); + for (;;) + { + ssize_t nread = read(STDIN_FILENO, buffer, BUF_SIZE); + if (nread == -1) + perror("stdin"); + if (nread <= 0) + break; + write(STDOUT_FILENO, buffer, nread); + for (size_t i = 0; i < file_count; i++) + write(files[i], buffer, nread); + } + free(buffer); - if (ferror(stdin)) - perror("stdin"); + if (ferror(stdin)) + perror("stdin"); - for (size_t i = 0; i < file_count; i++) - close(files[i]); + for (size_t i = 0; i < file_count; i++) + close(files[i]); - return 0; + return 0; }