Unix Shell在C - 文件描述符问题中实现Cat

时间:2011-02-25 18:56:06

标签: c file shell unix file-descriptor

我已经完成了我的Unix shell实践实现,除非我遇到问题,当它输出到文件时实现cat; IE:cat foo.txt > bar.txt - 将foo的内容输出到bar

让我们从主要功能&然后我将定义子方法:

int main(int argc, char **argv)
{           
    printf("[MYSHELL] $ ");

    while (TRUE) {
        user_input = getchar();
        switch (user_input) {

            case EOF:
                exit(-1);

            case '\n':
                printf("[MYSHELL] $ ");
                break;

            default:
                // parse input into cmd_argv - store # commands in cmd_argc
                handle_user_input();

                //determine input and execute foreground/background process
                execute_command();
        }
        background = 0;
    }
    printf("\n[MYSHELL] $ ");
    return 0;    
}

handle_user_input 只需填充cmd_argv数组即可执行user_input,并删除>并设置output标记如果用户希望输出到文件。这是该方法的核心:

while (buffer_pointer != NULL) { 
        cmd_argv[cmd_argc] = buffer_pointer;
        buffer_pointer = strtok(NULL, " ");

        if(strcmp(cmd_argv[cmd_argc], ">") == 0){
            printf("\nThere was a '>' in %s @ index: %d for buffer_pointer: %s \n", *cmd_argv,cmd_argc,buffer_pointer);
            cmd_argv[cmd_argc] = strtok(NULL, " ");
            output = 1;
        }

        cmd_argc++;

        if(output){
            filename = buffer_pointer;
            printf("The return of handling input for filename %s =  %s + %s \n", buffer_pointer, cmd_argv[0], cmd_argv[1]); 
            return;
        }        
}

然后调用

execute_command ,解释现在填充的cmd_argv。只是为了让你了解大局。显然,这些情况都不匹配,并调用create_process方法:

int execute_command()
{
    if (strcmp("pwd", cmd_argv[0]) == 0){
        printf("%s\n",getenv("PATH"));  
        return 1;
    }
    else if(strcmp("cd", cmd_argv[0]) == 0){
        change_directory();
        return 1;        
    }
    else if (strcmp("jobs", cmd_argv[0]) == 0){
        display_job_list();
        return 1;   
    }
    else if (strcmp("kill", cmd_argv[0]) == 0){
        kill_job();
    }
    else if (strcmp("EOT", cmd_argv[0]) == 0){
        exit(1);
    }
    else if (strcmp("exit", cmd_argv[0]) == 0){
        exit(-1);
    }
    else{
        create_process();
        return;
    }
}

非常直接,对吧?


create_process 是我遇到问题的地方。

void create_process()
{
    status = 0;
    int pid = fork();
    background = 0;

    if (pid == 0) {
        // child process
        if(output){
            printf("Output set in create process to %d\n",output);
            output = 0;
            int output_fd = open(filename, O_RDONLY);
            printf("Output desc = %d\n",output_fd);
            if (output_fd > -1) {
                dup2(output_fd, STDOUT_FILENO);
                close(output_fd);
            } else {
                perror("open");
            }
        }
        printf("Executing command, but STDOUT writing to COMMAND PROMPT instead of FILE - as I get the 'open' error above \n");
        execvp(*cmd_argv,cmd_argv);
        // If an error occurs, print error and exit
        fprintf (stderr, "unknown command: %s\n", cmd_argv[0]);
        exit(0);
    } else {
        // parent process, waiting on child process
            waitpid(pid, &status, 0);
        if (status != 0)
            fprintf  (stderr, "error: %s exited with status code %d\n", cmd_argv[0], status);
    }
    return;
}

我已打印output_fd = -1,我设法将perror("open")置于其他内容中:open: No such file or directory。然后它显示它是"writing to COMMAND PROMPT instead of FILE",因为我显示到控制台。然后执行处理execvp的{​​{1}},但将其打印到控制台而不是文件。

我意识到它不应该在这一点上,因为cat foo.txt不合适并且应该返回另一个值;但是我无法弄清楚如何正确使用文件描述符以便用output_fd = -1打开一个新的/现有文件并写入它,就像WELL AS GET BACK回到命令行的stdin一样。

我已经设法输出到文件,但后来失去了正确的标准输入。有人可以指点我吗?我觉得我在愚蠢的事情上徘徊,我做错了或者看着。

非常感谢任何帮助。

2 个答案:

答案 0 :(得分:2)

如果要写入文件,为什么要使用O_RDONLY?我的猜测是你应该使用类似的东西:

int output_fd = open(filename, O_WRONLY|O_CREAT, 0666);

(0666是在创建时设置访问权限。)

显然,如果您无法打开已删除的文件,则不应启动该命令。

答案 1 :(得分:1)

首先,我注意到的显而易见的事情是你打开了O_RDONLY文件。输出效果不好!

其次,重定向输出的基本过程是:

  • 打开文件进行编写
  • dup stdout,以便您可以根据需要保留副本。与stderr相同,如果重定向。
  • 将您的副本fcntl发送到CLOEXEC(或者,使用dup3)
  • dup2 file to stdout
  • 执行命令

最后,你是否真的将命令名称作为全局变量传递?我认为一旦你尝试实施cat foo | ( cat bar; echo hi; cat ) > baz或某些事情,这将会困扰你。

相关问题