第十篇：基于TCP的一对回射客户/服务器程序及其运行过程分析( 上 )

前言

       本文将讲解一对经典的客户/服务器回射程序，感受网络编程的大致框架( 该程序稍作改装即可演变成各种提供其他服务的程序 )；同时，还将对其运行过程加以分析，观察程序背后协议的执行细节，学习调试网络程序的技巧。

客户端

#include    "unp.h"

void str_cli(FILE *fp, int sockfd);

int
main(int argc, char **argv)
{
    int                    sockfd;
    struct sockaddr_in    servaddr;

    if (argc != 2)
        err_quit("usage: tcpcli <IPaddress>");

    sockfd = Socket(AF_INET, SOCK_STREAM, 0);

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    Inet_pton(AF_INET, argv[1], &servaddr.sin_addr);

    Connect(sockfd, (SA *) &servaddr, sizeof(servaddr));

    str_cli(stdin, sockfd);    

    exit(0);
}

/*
 * 事务函数
*/
void
str_cli(FILE *fp, int sockfd)
{
    char    sendline[MAXLINE], recvline[MAXLINE];

    // 从标准输入读取字符串( 阻塞于用户 ）    
    while (Fgets(sendline, MAXLINE, fp) != NULL) {

        // 往缓冲区写入字符串
        Writen(sockfd, sendline, strlen(sendline));

        // 从缓冲区读取字符串( 阻塞于服务器传回的数据 )
        if (Readline(sockfd, recvline, MAXLINE) == 0)
            err_quit("str_cli: server terminated prematurely");

        // 读取从服务器回射的消息并在终端打印
        Fputs(recvline, stdout);
    }
}

服务器端

#include    "unp.h"

void str_echo(int sockfd);

int
main(int argc, char **argv)
{
    int                    listenfd, connfd;
    pid_t                childpid;
    socklen_t            clilen;
    struct sockaddr_in    cliaddr, servaddr;

    listenfd = Socket(AF_INET, SOCK_STREAM, 0);

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family      = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port        = htons(SERV_PORT);

    Bind(listenfd, (SA *) &servaddr, sizeof(servaddr));

    Listen(listenfd, LISTENQ);

    for ( ; ; ) {
        clilen = sizeof(cliaddr);
        connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);

        if ( (childpid = Fork()) == 0) {    /* 子进程处理段 */
            Close(listenfd);    /* 关闭监听套接字 */
            str_echo(connfd);    /* 事务处理 */
            exit(0);
        }
        Close(connfd);            /* 父进程关闭连接套接字 */
    }
}

/*
 * 事务处理函数
*/
void
str_echo(int sockfd)
{
    ssize_t        n;
    char        buf[MAXLINE];

again:
    while ( (n = read(sockfd, buf, MAXLINE)) > 0)
        Writen(sockfd, buf, n);

    if (n < 0 && errno == EINTR)
        goto again;
    else if (n < 0)
        err_sys("str_echo: read error");
}