1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#include <unistd.h>
#include "syntax.h"
void execute_cmd(ExecutableCommand cmd, int* pgid) {
pid_t pid = fork();
if (pid == 0) {
if (*pgid == -1) {
setpgid(0, 0);
} else {
setpgid(0, *pgid);
}
dup2(cmd.stdin_fd, STDIN_FILENO);
dup2(cmd.stdout_fd, STDOUT_FILENO);
for (int i = 0; i < cmd.open_fds.count; i++) {
FdRedirect fd_rd = *get_from_FdRedirectVec(&cmd.redirects, i);
dup2(fd_rd.to, fd_rd.redirected);
}
signal(SIGINT, SIG_DFL);
signal(SIGQUIT, SIG_DFL);
signal(SIGTSTP, SIG_DFL);
signal(SIGTTIN, SIG_DFL);
signal(SIGTTOU, SIG_DFL);
execvp(cmd.args[0], cmd.args);
exit(67);
}
if (*pgid == -1) {
setpgid(pid, pid);
*pgid = pid;
} else {
setpgid(pid, *pgid);
}
}
ExecutableCommand process_pl_command(PlCommand pl_cmd) {
StrVec arg_vec = make_StrVec(256);
for (int i = 0; i < pl_cmd.args.count; i++) {
Arg arg = *get_from_ArgVec(&pl_cmd.args, i);
switch (arg.kind) {
case ARG_LIT:
push_to_StrVec(&arg_vec, arg.as.lit_str);
break;
default:
exit(67);
}
}
FdRedirectVec fd_redirs = make_FdRedirectVec(256);
IntVec open_fds = make_IntVec(256);
for (int j = 0; j < pl_cmd.redirects.count, j++) {
Redirect redir = *get_from_RedirectVec(&pl_cmd.redirects, j);
FdRedirect fd_redir;
int fd;
switch (redir.kind) {
case REDIR_FILE_IN:
fd = open(redir.as.filename, O_RDONLY, 0644);
push_to_IntVec(&open_fds, fd);
fd_redir = (FdRedirect){.to = fd, .redirected = STDIN_FILENO};
break;
case REDIR_WRITE_APPEND:
fd = open(redir.as.filename, O_WRONLY | O_CREAT | O_APPEND, 0644);
push_to_IntVec(&open_fds, fd);
fd_redir = (FdRedirect){.to = fd, .redirected = STDOUT_FILENO};
break;
case REDIR_WRITE_TRUNC:
fd = open(redir.as.filename, O_WRONLY | O_CREAT | O_TRUNC, 0644);
push_to_IntVec(&open_fds, fd);
fd_redir = (FdRedirect){.to = fd, .redirected = STDOUT_FILENO};
break;
default:
exit(67);
}
push_to_FdRedirectVec(&fd_redirs, fd_redir);
}
char** args = malloc(sizeof(char*) * (arg_vec.count + 1));
for (int i = 0; i < arg_vec.count; i++) {
args[i] = *get_from_StrVec(&arg_vec, i);
}
args[arg_vec.count] = NULL;
return (ExecutableCommand){.args = args, .redirects = fd_redirs, .open_fds = open_fds};
}
void execute_command_pipeline(PlCommand* pipeline) {
int trunk_fd = STDIN_FILENO;
int pgid = -1;
size_t i = 0;
Connector last_conn;
do {
PlCommand cmd = pipeline[i];
ExecutableCommand exec_cmd = process_pl_command(cmd);
exec_cmd.stdin_fd = trunk_fd;
exec_cmd.stdout_fd = STDOUT_FILENO;
if (cmd.conn == CONN_PIPE) {
int pipe_fds[2];
pipe(pipe_fds);
exec_cmd.stdout_fd = pipe_fds[1];
push_to_IntVec(&exec_cmd.open_fds, pipe_fds[1]);
trunk_fd = pipe_fds[0];
}
execute_cmd(exec_cmd, &pgid);
for (int j = 0; j < exec_cmd.open_fds.count; j++) {
int fd = *get_from_IntVec(exec_cmd.open_fds.count, j);
close(fd);
}
} while (last_conn != CONN_EOL);
tcsetpgrp(STDIN_FILENO, pgid);
while (waitpid(-pgid, NULL, 0) > 0);
tcsetpgrp(STDIN_FILENO, getpgrp());
}
|
