add redirection of stdout
This commit is contained in:
parent
8a592d6e57
commit
ad5116fd27
66
src/loop.c
66
src/loop.c
@ -14,9 +14,12 @@
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
#define _GNU_SOURCE
|
#define _GNU_SOURCE
|
||||||
|
#include <errno.h>
|
||||||
|
#include <fcntl.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
#include "array.h"
|
#include "array.h"
|
||||||
#include "builtins.h"
|
#include "builtins.h"
|
||||||
#include "console_line.h"
|
#include "console_line.h"
|
||||||
@ -64,10 +67,28 @@ void loop() {
|
|||||||
|
|
||||||
StringArray *args = create_string_array();
|
StringArray *args = create_string_array();
|
||||||
|
|
||||||
|
char *output_file = NULL;
|
||||||
|
bool get_output_file = false;
|
||||||
|
bool output_append = false;
|
||||||
|
|
||||||
char *saveptr = NULL;
|
char *saveptr = NULL;
|
||||||
char *token = strtok_r(line, " ", &saveptr);
|
char *token = strtok_r(line, " ", &saveptr);
|
||||||
while (token) {
|
while (token) {
|
||||||
insert_string_array(args, 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);
|
token = strtok_r(NULL, " ", &saveptr);
|
||||||
}
|
}
|
||||||
if (line != NULL) {
|
if (line != NULL) {
|
||||||
@ -81,6 +102,38 @@ void loop() {
|
|||||||
continue;
|
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])) {
|
if (is_builtin(args->array[0])) {
|
||||||
run_builtin(args);
|
run_builtin(args);
|
||||||
}
|
}
|
||||||
@ -88,6 +141,17 @@ void loop() {
|
|||||||
launch_program(args);
|
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);
|
free_string_array(args);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user