c语言简单实现telnet客户端

#include 

 

#define DO 0xfd

#define WONT 0xfc

#define WILL 0xfb

#define DONT 0xfe

#define CMD 0xff

#define CMD_ECHO 1

#define CMD_WINDOW_SIZE 31

#define IAC 255

#define SB 250

#define SE 240

#define BUFLEN 200

#define ESCAPE 27

 

void negotiate(int sock, unsigned char *buf, int len) {

    int i;

    const char* option_code[350];

    option_code[00] = "TRANSMIT-BINARY";

    option_code[01] = "ECHO";

    option_code[03] = "SUPPRESS-GO-AHEAD";

    option_code[05] = "STATUS";

    option_code[06] = "TIMING-MARK";

    option_code[10] = "NAOCRD";

    option_code[11] = "NAOHTS";

    option_code[12] = "NAOHTD";

    option_code[13] = "NAOFFD";

    option_code[14] = "NAOVTS";

    option_code[15] = "NAOVTD";

    option_code[16] = "NAOLFD";

    option_code[17] = "EXTEND-ASCII";

    option_code[18] = "LOGOUT";

    option_code[19] = "BM";

    option_code[20] = "DET";

    option_code[23] = "SEND-LOCATION";

    option_code[24] = "TERMINAL-TYPE";

    option_code[25] = "END-OF-RECORD";

    option_code[26] = "TUID";

    option_code[27] = "OUTMRK";

    option_code[28] = "TTYLOC";

    option_code[29] = "3270-REGIME";

    option_code[30] = "X.3-PAD";

    option_code[31] = "NAWS";

    option_code[32] = "TERMINAL-SPEED";

    option_code[33] = "TOGGLE-FLOW-CONTROL";

    option_code[34] = "LINEMODE";

    option_code[35] = "X-DISPLAY-LOCATION";

    option_code[36] = "ENVIRON";

    option_code[37] = "AUTHENTICATION";

    option_code[38] = "ENCRYPT";

    option_code[39] = "NEW-ENVIRON";

    option_code[40] = "TN3270E";

    option_code[42] = "CHARSET";

    option_code[44] = "COM-PORT-OPTION";

    option_code[47] = "KERMIT";

    option_code[250] = "SB";

    option_code[240] = "SE";

    option_code[251] = "WILL";

    option_code[252] = "WONT";

    option_code[253] = "DO";

    option_code[254] = "DONT";

    option_code[255] = "IAC";

    if (buf[1] == DO && buf[2] == CMD_WINDOW_SIZE) {

        unsigned char tmp1[10] = { IAC, WILL, CMD_WINDOW_SIZE };

        if (send(sock, tmp1, 3, 0) < 0)

            exit(1);

 

        unsigned char tmp2[10] = { IAC, SB, CMD_WINDOW_SIZE, 0, 80, 0, 24, IAC,

        SE };

        if (send(sock, tmp2, 9, 0) < 0)

            exit(1);

        return;

    }

 

    /*

     * To avoid any problems, we reply to every DO with a WONT and every

     * WILL with a DO. Both of these are not required, but are included to prevent

     * any problems with most telnet servers.

     *

     * Echoing back the same characters to the server works on Linux machines,

     * but may not work on other terminals.

     *

     */

    for (i = 1; i < len; i++) {

        if (buf[i] == DO) {

            buf[i] = WONT;

        } else if (buf[i] == WILL) {

            buf[i] = DO;

        }

    }

 

    if (send(sock, buf, len, 0) < 0)

        exit(1);

}

 

static struct termios tin;

 

static void terminal_set(void) {

    // save terminal configuration

    tcgetattr(STDIN_FILENO, &tin);

 

    static struct termios tlocal;

    memcpy(&tlocal, &tin, sizeof(tin));

    // The file descriptor which has to be turned to raw mode is the standard

    // input of the parent process

    cfmakeraw(&tlocal);

    tcsetattr(STDIN_FILENO, TCSANOW, &tlocal);

}

 

static void terminal_reset(void) {

    // restore terminal upon exit

    tcsetattr(STDIN_FILENO, TCSANOW, &tin);

}

 

void connect_to_server(int sock, int port, char* address) {

    struct sockaddr_in server;

    server.sin_addr.s_addr = inet_addr(address);

    server.sin_family = AF_INET;

    server.sin_port = htons(port);

 

    //Connect to remote server

    if (connect(sock, (struct sockaddr *) &server, sizeof(server)) < 0) {

        perror("connect failed. Error");

        exit(1);

    }

}

 

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

    int sock;

    unsigned char buf[BUFLEN + 1];

    int len;

    int i;

    int port = 23;

 

    if (argc < 2 || argc > 3) {

        printf("Usage: %s address [port]\n", argv[0]);

        return 1;

    }

    

    if (argc == 3)

        port = atoi(argv[2]);

 

    //Create socket

    sock = socket(AF_INET, SOCK_STREAM, 0);

    if (sock == -1) {

        perror("Could not create socket. Error");

        return 1;

    }

    printf("Trying %s...", argv[1]);

 

    connect_to_server(sock, port, argv[1]);

 

    printf("Connected to %s\n", argv[1]);

    puts("Escape character is '^]'.");

 

    // set terminal

    terminal_set();

    atexit(terminal_reset);

 

    struct timeval ts;

    ts.tv_sec = 1; // 1 second

    ts.tv_usec = 0;

 

    while (1) {

        // select setup

        fd_set fds;

        FD_ZERO(&fds);

        if (sock != 0)

            FD_SET(sock, &fds);

        FD_SET(0, &fds);

 

        // wait for data

        int nready = select(sock + 1, &fds, (fd_set *) 0, (fd_set *) 0, &ts);

        if (nready < 0) {

            perror("select. Error");

            return 1;

        } else if (nready == 0) {

            ts.tv_sec = 1;

            ts.tv_usec = 0;

        } else if (sock != 0 && FD_ISSET(sock, &fds)) {

            // start by reading a single byte

            int rv;

            if ((rv = recv(sock, buf, 1, 0)) < 0)

                return 1;

            else if (rv == 0) {

                printf("Connection closed by the remote end\n\r");

                return 0;

            }

 

            if (buf[0] == CMD) {

                // read 2 more bytes

                len = recv(sock, buf + 1, 2, 0);

                if (len < 0)

                    return 1;

                else if (len == 0) {

                    printf("Connection closed by the remote end\n\r");

                    return 0;

                }

                negotiate(sock, buf, 3);

            } else {

                len = 1;

                buf[len] = '\0';

                printf("%s", buf);

                fflush(stdout);

            }

        }

 

        else if (FD_ISSET(0, &fds)) {

            static char crlf[] = { '\r', '\n' };

            buf[0] = getc(stdin); //fgets(buf, 1, stdin);

            if (buf[0] == '\n') { // with the terminal in raw mode we need to force a LF

                if (send(sock, crlf, 1, 0) < 0) {

                    return 1;

                }

            } else if (buf[0] == ESCAPE) {

                printf("Connection closed by the client end\n\r");

                return 0;

            } else {

                if (send(sock, buf, 1, 0) < 0)

                    return 1;

            }

        }

    }

    close(sock);

    return 0;

}