Telnet处理wc和从键盘读取输入的任何命令

时间:2013-03-15 08:43:28

标签: c unix telnet systems-programming unix-socket

我是UNIX的新手,也是使用C语言编写的新手。

我想要做的是telnet的简单实现。

我在双方都做了所有必需的连接(服务器和客户端)。

在服务器上:我收到的任何请求,fork处理它的新进程

void createProcess() {
    int pid = fork();
    if( pid == CHILD ) {
        transferHandling(s);
        close(s);
        _exit(0);
    }
}

此方法用于处理来自客户端的命令:

int transferHandling(int socket) {
    char buf[20];
    char command[1024];
    int n, pid, fd[2];
    char *shell, *ps1;
    if( checkLogin(socket) == 0 ) { // Used to check user account
        send(socket, "Login Succeeded\n", 16, 0);
        if( setupENV(info->name) == 0 ) {  // setuid and setgid for the user
            if( pipe(fd) == ERROR ) {
                send(socket, "Server Busy\n", 12, 0);
                return 1;
            }
                    // Here I make a process to run the default shell for the user
                    // and make pipe between them
            if((pid = fork()) > 0) {
                dup2(fd[0], STDOUT_FILENO);
                close(fd[1]);

                //read commands from socket
                while(1) {
                    n = recv(socket, command, 1024, 0);
                                    // send the commands received to the child which contains the default shell
                    write(fd[0], command, n);
                }
            } else {
                dup2(fd[1], STDIN_FILENO);
                dup2(socket, STDOUT_FILENO);
                dup2(socket, STDERR_FILENO);
                close(fd[0]);

                //run default shell
                shell = getpwnam(info->name)->pw_shell;
                execlp(shell, shell, NULL);
            }
        } else {
            return 1; // setup ENV error
        }
    } else {
        send(socket, "Login Failed\n", 13, 0);
    }
}

此程序可以使用任何命令成功运行,但是当客户端编写任何命令时:wc, cd, bash (to run any other shell)。该计划被绞死。

这是建立连接后的客户端。

if(login(soc)) { // send the user information for the server
        char cmd[2024], cc[1024];
        int n;
        while(1) {
            printf("Reading\n");
            write(1, "username@hostname> ", 19);
            n = read(0, cmd, 2024);  // read the command from the STDIN, and send it to server
            send(soc, cmd, n, 0);

            printf("Receiving Now\n");
            n = recv(soc, cmd, 2024, 0); // getting back the command output
            write(1, cmd, n);
        }
    }

1 个答案:

答案 0 :(得分:0)

您的管道文件描述符错误。你应该:

write(fd[1], ...);

dup2(fd[0], STDIN_FILENO);
顺便说一下,中间进程不需要从套接字读取命令并通过管道将它们发送到shell。您可以直接从套接字重定向shell的输入。而且,如果直接写入管道,则不需要dup2()。最后,执行此操作的无限循环似乎是一个非常糟糕的主意,因为它会在shell死亡时旋转。