UNPv1第五章：TCP客户服务器程序实例

现在写一个完整的TCP客户-服务器程序例子

这个简单的例子是完成下述功能的一个回射服务器：
1. 客户从标准输入读一行文本，写到服务器上
2. 服务器读入此行，并回射给客户
3. 客户读此回射行写到标准输出

客户端代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define SRV_PORT 8888
#define MAXLINE 4096
void str_cli(FILE *fp, int sockfd);
int main(int argc, char **argv)
{
    if(argc != 2)
    {
        printf("usage:tcpcli <ip address>\n");
        exit(0);
    }

    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd < 0)
    {
        perror("create socket error.");
    }

    struct sockaddr_in srvaddr;
    bzero(&srvaddr, sizeof(srvaddr));
    srvaddr.sin_family = AF_INET;
    srvaddr.sin_port = htons(SRV_PORT);

    if(inet_pton(AF_INET, argv[1], &srvaddr.sin_addr) <= 0)
    {
        printf("address error %s\n", argv[1]);
        exit(0);
    }

    if(connect(sockfd, (struct sockaddr *)&srvaddr, sizeof(srvaddr)) < 0 )
    {
        perror("connect error");
    }

    str_cli(stdin, sockfd);
    exit(0);
}

void str_cli(FILE *fp, int sockfd)
{
    char sendline[MAXLINE];
    char readline[MAXLINE];

    while(fgets(sendline, MAXLINE, fp))
    {
        if( write(sockfd, sendline, strlen(sendline)) != strlen(sendline) )
        {
            perror("send data error");
        }
        if( read(sockfd, readline, MAXLINE) == 0)
        {
            perror("recv data error");
        }
        fputs(readline,stdout);
    }
}

服务器端代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define SRV_PORT 8888
#define MAXLINE 4096
void str_echo(int fd);
int main(int argc, char **argv)
{
    int listenfd = socket(AF_INET, SOCK_STREAM, 0);
    if(listenfd < 0)
    {
        perror("create socket error.");
    }

    struct sockaddr_in srvaddr;
    bzero(&srvaddr, sizeof(srvaddr));
    srvaddr.sin_family = AF_INET;
    srvaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    srvaddr.sin_port = htons(SRV_PORT);

    if(bind(listenfd, (struct sockaddr*)&srvaddr, sizeof(srvaddr)) < 0)
    {
        perror("bind error.");
    }

    if(listen(listenfd, 1023) < 0)
    {
        perror("listen error.");
    }

    struct sockaddr_in cliaddr;

    for(; ;)
    {
        socklen_t clilen = sizeof(cliaddr);
        int connfd = accept(listenfd, (struct sockaddr*)&cliaddr, &clilen);
        if(connfd < 0)
        {
            perror("accept error.");
        }
        pid_t childpid;
        if( (childpid = fork()) == 0 )
        {
            close(listenfd);
            str_echo(connfd);
            exit(0);
        }
        close(connfd);
    }

    return 0;
}

void str_echo(int sockfd)
{
    char line[MAXLINE];

    while(read(sockfd, line, MAXLINE) != 0)
    {
        if(write(sockfd, line, strlen(line)) != strlen(line))
        {
            perror("write error");
        }
    }
}

accept返回前客户/服务器的状态

accept返回后客户/服务器的状态
(connfd = Accept(listenfd, (SA *) &cliaddr, &clilen);)

fork返回后客户/服务器的状态.
注意，此时listen和connfd这两个描述符都在父进程和子进程共享

在下一步是由父进程关闭已连接套接字，由子进程关闭监听套接字
父子进程关闭相应套接字后客户/服务器的状态

wait和waitpid

可以调用如下两个函数处理已终止的子进程

#include <sys/wait.h>
pid_t wait (int *statloc);
pid_t waitpid (pid_t pid, int *statloc, int options);
//返回值：成功返回进程ID，出错返回返回0或-1;

对于参数pid 想等待的进程ID号。-1表示等待第一个结束的子进程，options附加选项，常用的是WNOHANG，告知内核在没有以终止子进程时不要阻塞
函数wait和waitpid均返回两个值: 函数的返回值是终止子进程的进程ID号，子进程的终止状态(一个整数)则是通过指针statloc返回的。
wait和waitpid的区别: wait 等待第一个结束的子进程，如果没有结束的子进程，wait将阻塞。waitpid 通过参数设置，可以在没有子进程结束时waitpid不阻塞。