☰

自行创建的linux shell中的重定向

tag-icon • tag-icon shell io-redirection fork shell-builtin
自行创建的linux shell中的重定向

我正在写我自己的shell。我想实现redirection（>和>>）。为此我使用了dup2()系统调用。但是，如果我输入的命令具有重定向，则即使我不使用>或>>在其中，另一个命令也会遵循先前的重定向。我是否缺少关闭以前的文件描述符？此外，即使我使用 >，输出也始终附加到文件中。

以下是我的代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/wait.h>
#include <setjmp.h>
#include <signal.h>
#include <fcntl.h>
#include <stdbool.h>

#define MAXLINE 259
#define PROMPT "> "
#define MAX_ARG_LIST    200
//-Wno-write-strings to g++ compiler to supress warnings

extern char **environ;

#define MAX_CMD_SIZE    50
#define SEARCH_FOR_CMD  -1
typedef void (*buildInFunc) (char **);
typedef struct {
    char cmd[MAX_CMD_SIZE];
    buildInFunc func;
} builtInCmd;

// built-in commands
void execExit(char *cmd[]);
void execCd(char *cmd[]);
builtInCmd builtInCmds[] = {
        {"exit",    execExit  },
        {"cd",  execCd }
};
int builtInCnt = sizeof(builtInCmds)/sizeof(builtInCmd);
int isBuiltIn(char *cmd);

void execBuiltIn(int i, char *cmd[]);

// capture SIG_INT and recover
sigjmp_buf ctrlc_buf;
void ctrl_hndlr(int signo) {
    siglongjmp(ctrlc_buf, 1);
}

int main(int argc, char *argv[]) {

    char line[MAXLINE];
    pid_t childPID;
    int argn; char *args[MAX_ARG_LIST];
    int cmdn;
        char *in_file,*out_file;
        char *gt=">";       //truncate redirection char pointer
        char *gtgt=">>";    //append redirection char pointer        
        int in_fd,out_fd;   //
        bool out_fd_present=false;  //checks if > or >> string is present

    // setup SIG_INT handler
    if (signal(SIGINT, ctrl_hndlr) == SIG_ERR)
        fputs("ERROR: failed to register interrupts in kernel.\n", stderr);

    // setup longjmp buffer
    while (sigsetjmp(ctrlc_buf, 1) != 0) ;

    for(;;) {
    // prompt and get commandline
        fputs(PROMPT, stdout);
        fgets(line, MAXLINE, stdin);
        if (feof(stdin)) break; // exit on end of input

    // process commandline
        if (line[strlen(line)-1] == '\n')
            line[strlen(line)-1] = '\0';
        // build argument list
        args[argn=0]=strtok(line, " \t");
                while(args[argn]!=NULL && argn<MAX_ARG_LIST){

                        args[++argn]=strtok(NULL," \t");
                        //if append >> redirection present
                        if(strcmp( args[argn-1],gtgt ) == 0){
                            out_fd_present=true;
                            out_file=args[argn];
                            argn=argn-2;
                            out_fd = open(out_file, O_WRONLY | O_APPEND | O_CREAT,S_IRWXG | S_IRWXO | S_IRWXU);
                            //printf("\n >> found\n");                            
                        }
                        //if trncate > redirection present
                        else if(strcmp( args[argn-1],gt ) == 0){
                            out_fd_present=true;
                            out_file=args[argn];
                            argn=argn-2;
                            out_fd = open(out_file, O_WRONLY | O_CREAT | O_TRUNC, S_IRWXG | S_IRWXO | S_IRWXU);
                            //printf("\n > found\n");
                        }
                        printf("args[%d]=%s\n",argn-1,args[argn-1]);
                }


    // execute commandline
    if ((cmdn = isBuiltIn(args[0]))>-1) {
        execBuiltIn(cmdn, args);
    } else {
        childPID = fork();
                int save_out;
        if (childPID == 0) {
                    if(out_fd_present){
                        int a;
                        save_out = dup(STDOUT_FILENO);
                        dup2(out_fd,1);
                        close(out_fd);                        
                    }
                        execvp(args[0], args);                        
            fputs("ERROR: can't execute command.\n", stderr);
            _exit(1);
        } else {
            waitpid(childPID, NULL, 0);
        }
    }

    // cleanup
                fflush(stderr);
                fflush(stdout);
    }

    return 0;
}

int isBuiltIn(char *cmd) {
    int i = 0;        
    while (i<builtInCnt) {
        if (strcmp(cmd,builtInCmds[i].cmd)==0)
            break;
        ++i;
    }
    return i<builtInCnt?i:-1;
}

void execBuiltIn(int i, char *cmd[]) {
    if (i == SEARCH_FOR_CMD)
        i = isBuiltIn(cmd[0]);
    if (i >= 0 && i < builtInCnt)
        builtInCmds[i].func(cmd);
    else
        fprintf(stderr, "ERROR: unknown built-in command\n");
}

void execExit(char *cmd[]) {        
    exit(0);
}
void execCd(char *cmd[]){
    chdir(cmd[1]);
}

答案1

您有一个简单的逻辑错误：当您检测到 或 时out_fd_present，您永远不会将其重置为（例如在循环的顶部）。true>>>false

虽然我猜你的“外壳”是一个玩具外壳，但我不得不提到你还有很多其他问题。例如，您out_fd在父进程和save_out子进程中泄漏（这是做什么用的？）。您有一个错误，您可以调用strcmp字符串NULL。还有另一张为chdir.您的语法和功能也不同于 shell“应该”执行的操作（重定向运算符的位置、umask、间距和标记化等）。更不用说风格不一致了（丑陋的骆驼箱和蛇箱都用过）。要实现外壳，即使是玩具外壳，您可能应该首先从良好的设计开始。

相关内容