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

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

 #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 > )
             {
                 if ((nread = read(fd, bufp, nleft)) < )
                 {
                     if (errno == EINTR)
  //被信号中断，errno这个全局变量的值就会等于EINTR。
                         continue;
                     return -;
                 }
                 else if (nread == ) //对方关闭或者已经读到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 > )
             {
                 if ((nwritten = write(fd, bufp, nleft)) < )
                 {
                     if (errno == EINTR)
                         continue;
  //要保证读取的字节数为指定字节数，所以继续
                     return -;
                 }
                 else if (nwritten == )
                     continue;
  //由于其他原因引起的什么都没有写进，则继续操作，保证指定字节数
                 bufp += nwritten;
                 nleft -= nwritten;
             }
             return count;
 }

 ssize_t recv_peek(int sockfd,void *buf,size_t len)
 {
     while()
     {
         int ret=recv(sockfd,buf,len,MSG_PEEK);
         if(ret==-&&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()
     {
         ret=recv_peek(sockfd,bufp,nleft);
         if(ret<)
             return ret;
         else if(ret==)
             return ret;
         nread=ret;
         int i;
         for(i=;i<nread;i++)
         {
             if(bufp[i]=='\n')
             {
                 ret=readn(sockfd,bufp,i+);
                 if(ret!=i+)
                     exit();
                 return ret;
             }
         }
         if(nread>nleft)
             exit();
         nleft -= nread;
         ret=readn(sockfd,bufp,nread);
         if(ret!=nread)
             exit();
         bufp+=nread;
     }
     return -;
 }

 #define port 5188
 int main()
 {
     int listenfd;
     //*****创建套接字*******
     if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<)
     /*if((listenfd=socket(PF_INET,SOCK_STREAM,0))<0)*/
         perror("error");

     //*******ipv4地址结构**********
     struct sockaddr_in servaddr;
     memset(&servaddr,,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=;
     if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&N,sizeof(N))<)
         perror("error");
     //2、进行绑定
     if(bind(listenfd,(struct sockaddr*)(&servaddr),sizeof(struct sockaddr))<)
         perror("error");

     //********将绑定的套接字转换为监听状态********
     if(listen(listenfd,SOMAXCONN)<) //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))<)
         perror("error");
     else
         printf("client_ip=%s,client_port=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));

     //*******数据通信过程***********
     char recvbuf[];
     while()
     {
         memset(recvbuf,,sizeof(recvbuf));
         int ret=recv_line(con,recvbuf,sizeof(recvbuf));//con这里是已连接套接字，不再是被动套接字，而是主动套接字了
         if(ret==-)
             exit();
         if(ret==)
         {
             printf("client_port is closed.\n");
             break;
         }
         fputs(recvbuf,stdout);
         writen(con,recvbuf,strlen(recvbuf));
         memset(recvbuf,,sizeof(recvbuf));
     }
     close(con);
     close(listenfd);
     return ;

 }