add redirection of stdout
This commit is contained in:
parent
8a592d6e57
commit
ad5116fd27
64
src/loop.c
64
src/loop.c
@ -14,9 +14,12 @@
|
||||
*/
|
||||
|
||||
#define _GNU_SOURCE
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include "array.h"
|
||||
#include "builtins.h"
|
||||
#include "console_line.h"
|
||||
@ -64,10 +67,28 @@ void loop() {
|
||||
|
||||
StringArray *args = create_string_array();
|
||||
|
||||
char *output_file = NULL;
|
||||
bool get_output_file = false;
|
||||
bool output_append = false;
|
||||
|
||||
char *saveptr = NULL;
|
||||
char *token = strtok_r(line, " ", &saveptr);
|
||||
while (token) {
|
||||
if (get_output_file) {
|
||||
get_output_file = false;
|
||||
output_file = malloc((strlen(token) +1 ) * sizeof(char *));
|
||||
strcpy(output_file, token);
|
||||
}
|
||||
else if (strcmp(token, ">") == 0) {
|
||||
get_output_file = true;
|
||||
}
|
||||
else if (strcmp(token, ">>") == 0) {
|
||||
get_output_file = true;
|
||||
output_append = true;
|
||||
}
|
||||
else {
|
||||
insert_string_array(args, token);
|
||||
}
|
||||
token = strtok_r(NULL, " ", &saveptr);
|
||||
}
|
||||
if (line != NULL) {
|
||||
@ -81,6 +102,38 @@ void loop() {
|
||||
continue;
|
||||
}
|
||||
|
||||
int stdout_copy = -1;
|
||||
int fd = -1;
|
||||
|
||||
// redirect output to a file instead of stdout
|
||||
if (output_file != NULL) {
|
||||
stdout_copy = dup(1);
|
||||
if (stdout_copy == -1) {
|
||||
perror("dup");
|
||||
}
|
||||
else {
|
||||
if (output_append) {
|
||||
fd = open(output_file, O_WRONLY | O_CREAT | O_APPEND, 0664);
|
||||
}
|
||||
else {
|
||||
fd = open(output_file, O_WRONLY | O_CREAT | O_TRUNC, 0664);
|
||||
}
|
||||
if (fd == -1) {
|
||||
fprintf(stderr, "open: Could not open file %s: \"%s\"\n", output_file, strerror(errno));
|
||||
free(output_file);
|
||||
output_file = NULL;
|
||||
}
|
||||
else {
|
||||
if (close(1) == -1) {
|
||||
perror("close");
|
||||
}
|
||||
if (dup(fd) == -1) {
|
||||
perror("dup");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (is_builtin(args->array[0])) {
|
||||
run_builtin(args);
|
||||
}
|
||||
@ -88,6 +141,17 @@ void loop() {
|
||||
launch_program(args);
|
||||
}
|
||||
|
||||
if (output_file != NULL) {
|
||||
if (close(fd) == -1) {
|
||||
perror("close");
|
||||
}
|
||||
if (dup2(stdout_copy, 1) == -1) {
|
||||
perror("dup2");
|
||||
}
|
||||
free(output_file);
|
||||
output_file = NULL;
|
||||
}
|
||||
|
||||
free_string_array(args);
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user