我正在尝试使用Linux,而我目前正在编写一个模拟Linux shell的程序。
我有一个解析输入的主函数,而我的问题现在无关紧要。 在解析每一行之后,调用处理行方法来处理所有内容。目前我支持常规流程,后台流程,目前正在处理仅包含2个命令(cmd1 | cmd2)的未命名管道。
以下是代码:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/wait.h>
#include <unistd.h>
void wait_for_background_process(void* arg) {
int status;
pid_t* pid = (pid_t*) arg;
do {
printf("Waiting for %d\n", *pid);
waitpid(*pid, &status, WUNTRACED);
} while (!WIFEXITED(status) && !WIFSIGNALED(status));
}
/************************
function: void pipeCommand(char** cmd1, char** cmd2)
comment: This pipes the output of cmd1 into cmd2.
**************************/
void pipe_command(char** cmd1, char** cmd2) {
int fds[2]; // file descriptors
if(pipe(fds) < 0) {
perror("myShell");
exit(EXIT_FAILURE);
}
pid_t pid;
pid = fork();
if(pid == 0) {
dup2(fds[1], 1);
if(execvp(cmd1[0], cmd1) < 0) {
perror("myShell");
exit(EXIT_FAILURE);
}
execvp(cmd1[0], cmd1);
} else if(pid < 0) {
perror("myShell");
exit(EXIT_FAILURE);
} else {
wait(NULL);
dup2(fds[0], 0);
if(execvp(cmd2[0], cmd2)) {
perror("myShell");
exit(EXIT_FAILURE);
}
}
}
/*
* Checks if the command is pipe command, if so we will return the
* index of the pipe
*/
int is_pipe_command(char** arglist, int count) {
int i = 0;
for(i = 0; i < count; i++) {
if(strcmp(arglist[i], "|") == 0) {
return i;
}
}
return 0;
}
int process_arglist(int count, char** arglist) {
pid_t pid;
int pipe_index;
pid = fork();
if (pid == 0) {
// Child process
if(strcmp(arglist[count-1],"&") == 0) {
char** background_arglist = (char**) malloc((count)*sizeof(char*));
if(background_arglist == NULL) {
printf("malloc failed: %s\n", strerror(errno));
exit(EXIT_FAILURE);
}
int i = 0;
for(i = 0; i < count - 1; i++) {
background_arglist[i] = arglist[i];
}
background_arglist[count - 1] = NULL;
if (execvp(background_arglist[0], background_arglist) == -1) {
perror("myShell");
}
} else if(pipe_index = is_pipe_command(arglist, count)) {
char** cmd1 = (char**) malloc((pipe_index+1)*sizeof(char*));
if(cmd1 == NULL) {
printf("malloc failed: %s\n", strerror(errno));
exit(EXIT_FAILURE);
}
int i;
int cmd1index = 0;
for(i = 0; i < pipe_index; i++) {
cmd1[cmd1index] = arglist[i];
cmd1index++;
}
cmd1[pipe_index] = NULL;
char** cmd2 = (char**) malloc((count - pipe_index)*sizeof(char*));
if(cmd2 == NULL) {
printf("malloc failed: %s\n", strerror(errno));
exit(EXIT_FAILURE);
}
int cmd2index = 0;
for(i = pipe_index+1; i < count; i++) {
cmd2[cmd2index] = arglist[i];
cmd2index++;
}
cmd2[count-pipe_index-1] = NULL;
pipe_command(cmd1, cmd2);
} else {
if (execvp(arglist[0], arglist) == -1) {
perror("myShell");
}
}
exit(EXIT_FAILURE);
} else if (pid < 0) {
// Error forking
perror("myShell");
exit(EXIT_FAILURE);
} else {
// Parent process
if(strcmp(arglist[count-1],"&") == 0) {
// The child is a background process
pthread_t thread;
pthread_create(&thread, NULL, wait_for_background_process, &pid);
}
else {
// Regular process
}
}
return 1;
}
我们可以专注于pipe_command函数,它正确地获取2个命令,我无法理解为什么我没有获得任何输出来调用例如ls -l | sort
或ls -l | grep "a"
感谢。
答案 0 :(得分:2)
dup
后,您需要关闭fds[1]
。否则,第二个命令进程(示例中为sort
或grep
)将不会从其EOF
读取中获得stdin
。需要EOF
以便进程知道何时停止处理和退出。
明确地,else
中的pipe_command
块应该包含close
,如下所示:
} else {
wait(NULL);
dup2(fds[0], 0);
close(fds[1]); /* ADDED THIS LINE */
if(execvp(cmd2[0], cmd2)) {
perror("myShell");
exit(EXIT_FAILURE);
}
}
还有一点需要注意。通常,第一个命令进程也需要类似的close(fds[0]);
。但是在你的情况下不需要它,因为第二个进程在调用execvp
之前等待第一个进程退出,这导致第一个进程隐式关闭所有打开的文件描述符。