TCP回射客户服务器模型(02 设置套接字选项、处理多并发)

　　int setsockopt(int sockfd, int level, int optname, const void *optval, socklen_t optlen);  //设置套接字选项


　　回射服务器编写，服务器关闭后处于TIME_WAIT状态，过一段时间才可以绑定刚才的端口。
在绑定服务器之前尽可能调用setsockopt来设置REUSEADDR套接字选项，使用REUSEADDR选项可以使得不必等待TIME_WAIT状态消失就可以重启服务器
　　服务器处理多并发，客户端短连接过来就创建子进程通信，客户端关闭后，子进程要退出.父进程一直处于监听。
　　

　　服务器进程程序：

#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<stdlib.h>
#include<stdio.h>
#include<errno.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#define ERR_EXIT(m)\
    do\
    {\
        perror(m);\
        exit(EXIT_FAILURE);\
    }while(0)
void do_service(int conn)
{
        char recvbuf[1024];
        while(1)
        {
            memset(recvbuf,0,sizeof(recvbuf));
            int ret=read(conn,recvbuf,sizeof(recvbuf));
            //捕捉客户端关闭            
            if(ret==0)
            {
                printf("client close\n");
                break;//不用继续循环等待客户端数据
            }
            else if(ret==-1)
                ERR_EXIT("read error");
            fputs(recvbuf,stdout);
            write(conn,recvbuf,ret);
        }
}
int main(void)
{
    int listenfd;
    if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
        ERR_EXIT("socket error");
    //if((listenfd=socket(PF_INET,SOCK_STREAM,0))<0)
    

    //本地协议地址赋给一个套接字
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(5188);
    servaddr.sin_addr.s_addr=htonl(INADDR_ANY);//表示本机地址
    //servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    //inet_aton("127.0.0.1",&servaddr.sin_addr);
    
    //绑定之前，开启地址重复使用,关闭服务器再打开不用等待TIME_WAIT
    int on=1;
    if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
        ERR_EXIT("setsockopt error");
    //绑定本地套接字
    if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
        ERR_EXIT("bind error");
    if(listen(listenfd,SOMAXCONN)<0)//设置监听套接字(被动套接字)
        ERR_EXIT("listen error");
    
    struct sockaddr_in peeraddr;//对方套接字地址
    socklen_t peerlen=sizeof(peeraddr);/一定要初始化
    int conn;//已连接套接字(主动套接字)
    pid_t pid;
　　//原先的程序为何不能处理多客户端连接：因为之前的服务器程序，一旦客户端请求连接，服务器就会一直处于while循环，处理客户端请求，不再处理accept
    while(1){
        if((conn=accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen))<0)
            ERR_EXIT("accept error");//accept本质是到已连接队列中，取第一个连接。
        //连接好之后就构成连接，端口是客户端的。peeraddr是对端
        printf("ip=%s port=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
        pid=fork();
        if(pid==-1)
            ERR_EXIT("fork");
        if(pid==0){    
                close(listenfd);//父子进程共享描述符
                do_service(conn);
                //do_service返回，某个客户端关闭，结束该子进程,否则子进程也去接受连接（accept）。
                exit(EXIT_SUCCESS);
        }else     
            close(conn);
    }
    return 0;
}

 

　　客户端程序:

#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<stdlib.h>
#include<stdio.h>
#include<errno.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#define ERR_EXIT(m)\
    do\
    {\
        perror(m);\
        exit(EXIT_FAILURE);\
    }while(0)
int main(void)
{
    int sock;//客户端创建套接字
    if((sock=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
        ERR_EXIT("socket error");
    
    struct sockaddr_in servaddr;//本地协议地址赋给一个套接字
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(5188);
    
    servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");//服务器段地址
    //inet_aton("127.0.0.1",&servaddr.sin_addr);
    
    if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
        ERR_EXIT("connect");
    char sendbuf[1024]={0};
    char recvbuf[1024]={0};
    while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)//默认有换行符
    {
        write(sock,sendbuf,strlen(sendbuf));
        read(sock,recvbuf,sizeof(recvbuf));
        fputs(recvbuf,stdout);

        memset(sendbuf,0,sizeof(sendbuf));
        memset(recvbuf,0,sizeof(recvbuf));
    }
    close(sock);
    
    return 0;
}