tcp/ip通信第5期之服务器端程序

/*
此程序是tcp/ip通信服务器端程序，测试运行在redhat5上
重构readline函数，解决粘包问题——利用“
”识别一个消息边界
*/

#include<stdio.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<fcntl.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<sys/socket.h>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<signal.h>
#include<errno.h>

ssize_t readn(int fd, void *buf, size_t count)
{
    size_t nleft = count;//还留下多少字节没有读
    ssize_t nread; //已经读了多少字节
    char *bufp = (char *)buf;
    while (nleft > 0)
            {
                if ((nread = read(fd, bufp, nleft)) < 0)
                {
                    if (errno == EINTR)
 //被信号中断，errno这个全局变量的值就会等于EINTR。
                        continue;
                    return -1;
                }
                else if (nread == 0) //对方关闭或者已经读到eof
                    return count - nleft;
                bufp += nread;
                nleft -= nread;
            }
            return count;
}
     
ssize_t writen(int fd, const void * buf, size_t count) 
{
        size_t nleft = count;
        ssize_t nwritten;
        char *bufp = (char *)buf;
        while (nleft > 0)
            {
                if ((nwritten = write(fd, bufp, nleft)) < 0)
                {
                    if (errno == EINTR)
                        continue;
 //要保证读取的字节数为指定字节数，所以继续
                    return -1;
                }
                else if (nwritten == 0)        
                    continue;
 //由于其他原因引起的什么都没有写进，则继续操作，保证指定字节数
                bufp += nwritten;
                nleft -= nwritten;
            }
            return count;
}

ssize_t recv_peek(int sockfd,void *buf,size_t len)
{
    while(1)
    {
        int ret=recv(sockfd,buf,len,MSG_PEEK);
        if(ret==-1&&errno==EINTR)
            continue;
        return ret;
    }
}

ssize_t recv_line(int sockfd,void *buf,size_t len)
{
    int ret;//记录函数返回值
    int nread;//已经读到的字节数
    char *bufp=buf;
    int nleft=len;
    while(1)
    {
        ret=recv_peek(sockfd,bufp,nleft);
        if(ret<0)
            return ret;
        else if(ret==0)
            return ret;
        nread=ret;
        int i;
        for(i=0;i<nread;i++)
        {
            if(bufp[i]=='
')
            {
                ret=readn(sockfd,bufp,i+1);
                if(ret!=i+1)
                    exit(1);
                return ret;
            }
        }
        if(nread>nleft)
            exit(1);
        nleft -= nread;
        ret=readn(sockfd,bufp,nread);
        if(ret!=nread)
            exit(0);
        bufp+=nread;
    }
    return -1;
}


#define port 5188
int main()
{
    int listenfd;
    //*****创建套接字*******
    if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
    /*if((listenfd=socket(PF_INET,SOCK_STREAM,0))<0)*/
        perror("error");
        
    //*******ipv4地址结构**********
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr)); //清空结构体变量
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(port); //使用端口号：5188
    servaddr.sin_addr.s_addr=htonl(INADDR_ANY);//INADDR_ANY表示使用本机的任意可用ip地址,转换成网络地址序
    /*servaddr.sin_addr.s_addr = inet_addr("127.168.0.12");*/
    /*inet_aton("127.168.0.12",&servaddr.sin_addr);*/
    
    //*******绑定套接字和本机地址***********
    //1、设置REUSEADDR选项
    int N=1;
    if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&N,sizeof(N))<0)
        perror("error");
    //2、进行绑定
    if(bind(listenfd,(struct sockaddr*)(&servaddr),sizeof(struct sockaddr))<0)
        perror("error");
    
    //********将绑定的套接字转换为监听状态********
    if(listen(listenfd,SOMAXCONN)<0) //SOMAXCONN这个宏表示最大队列值
        perror("error");
    /*一旦调用listen函数，那么这个套接字就变成了被动套接字（只能被动接受连接——accept，
      不能发起连接——connect），否则还是主动套接字（可以发起连接——connect）*/
    
    //********接收对方的连接请求**************
    struct sockaddr_in peeraddr;  //定义对方地址
    socklen_t peerlen=sizeof(peeraddr);
    int con;
    if((con=accept(listenfd,(struct sockaddr*)(&peeraddr),&peerlen))<0)
        perror("error");
    else
        printf("client_ip=%s,client_port=%d
",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
    
    //*******数据通信过程***********
    char recvbuf[1024];
    while(1)
    {
        memset(recvbuf,0,sizeof(recvbuf));
        int ret=recv_line(con,recvbuf,sizeof(recvbuf));//con这里是已连接套接字，不再是被动套接字，而是主动套接字了
        if(ret==-1)
            exit(1);
        if(ret==0)
        {
            printf("client_port is closed.
");
            break;
        }
        fputs(recvbuf,stdout);
        writen(con,recvbuf,strlen(recvbuf));
        memset(recvbuf,0,sizeof(recvbuf));
    }
    close(con);
    close(listenfd);
    return 0;
    
}