socket程式設計readline函式實現回射伺服器/客戶端
阿新 • • 發佈：2018-11-01
為了解決粘包問題，reanline函式使用recv函式進行窺探，不清除內容
#include <unistd.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <stdlib.h> #include <stdio.h> #include <errno.h> #include <string.h> #define ERR_EXIT(m) \ do \ { \ perror(m); \ exit(EXIT_FAILURE);\ }while(0) //讀確定大小的包 //ssize_t有符號數，size_t為無符號數 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)//訊號中斷 { continue; } return -1;//否則出錯 } else if(nread==0)//表示對方關閉傳送 { 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) { //recv函式只用於套介面 //recv函式讀取後，不將資料在緩衝區清除 int ret= recv(sockfd, buf, len, MSG_PEEK); if(ret == -1 && errno == EINTR) continue; return ret; } } ssize_t readline(int sockfd, void *buf, size_t maxline) { int ret;//設定窺探返回值 int nread;//設定已窺探字元數 char *bufp = (char*)buf;//快取buf int nleft = maxline;//設定maxline為包最大長度，nleft為剩餘需讀取字元數 while(1) { ret=recv_peek(sockfd, bufp, nleft); if(ret<0) { return ret; } else if(ret=0) { return ret;//對方終止了傳送 } nread=ret; int i;//檢測有沒有‘\n’字元，有則讀取 for(i=0; i<nread,i++) { if(bufp[i]=='\n') { ret= readn(sockfd,bufp, i+1) if(ret != i+1)//已經窺探到有i+1字元，如果沒有則錯誤 exit(EXIT_FAILURE); return ret； } } //如果沒有讀到‘\n’，則將訊息讀入，直到最大包 nleft -= nread; ret= readn(sockfd, bufp, nread); if(ret != nread)//已經窺探到有nread個字元，如果readn函式不能讀取這麼多，則錯誤 { exit(EXIT_FAILURE); } bufp += nread; } return -1; } void do_service(int conn)//執行連線建立後，內容讀取，和返回內容 { char recvbuf[1024]; int n; while(1) { memset(recvbuf, 0, sizeof(recvbuf)); int ret = readline(conn, recvbuf,1024);//先接受包頭 if(ret==-1)//讀取失敗則退出 ERR_EXIT("readline"); else if(ret==0)//客戶端關閉 { printf("client_close\n"); break; } fputs(recvbuf,stdout);//列印標準輸出 writen(conn,recvbuf, strlen(recvbuf)); } } int main(void) { int listenfd;//建立套接字 if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0) /* if((listenfd=socket(PF_INET,SOCK_STREAM<0))<0)*/ ERR_EXIT("socket"); //建立伺服器地址 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);*/ int on = 1; //為了使伺服器處於TIME_WATTING狀態，能再次繫結地址 if(setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))<0) ERR_EXIT("setsockopt"); //繫結地址 if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0) ERR_EXIT("bind"); //監聽套接字 if(listen(listenfd,SOMAXCONN)<0)//套接字變為被動套接字 ERR_EXIT("listen"); //存放客戶端地址 struct sockaddr_in peeraddr; socklen_t peerlen =sizeof(peeraddr); int conn;//接受accept的套接字 //建立子程序，以便於多個連線進入伺服器 pid_t pid; while(1) { if((conn=accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen))<0)//套接字變為主動，且以後使用accept返回的套接字; ERR_EXIT("accept"); 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); exit(EXIT_SUCCESS); } else close(conn); } return 0; } #include <unistd.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <stdlib.h> #include <stdio.h> #include <errno.h> #include <string.h> #define ERR_EXIT(m) \ do \ { \ perror(m); \ exit(EXIT_FAILURE); \ }while(0) //讀確定大小的包 //ssize_t有符號數，size_t為無符號數 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)//訊號中斷 { continue; } return -1;//否則出錯 } else if(nread==0)//表示對方關閉傳送 { 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) { //recv函式只用於套介面 //recv函式讀取後，不將資料在緩衝區清除 int ret= recv(sockfd, buf, len, MSG_PEEK); if(ret == -1 && errno == EINTR) continue; return ret; } } ssize_t readline(int sockfd, void *buf, size_t maxline) { int ret; int nread; char *bufp = (char*)buf; int nleft = maxline; 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]=='\n') { ret= readn(sockfd,bufp, i+1) if(ret != i+1)//已經窺探到有i+1字元，如果沒有則錯誤 exit(EXIT_FAILURE); return ret； } } //如果沒有讀到‘\n’，則將訊息讀入 nleft -= nread; ret= readn(sockfd, bufp, nread); if(ret != nread)//已經窺探到有nread個字元，如果readn函式不能讀取這麼多，則錯誤 { exit(EXIT_FAILURE); } bufp += nread; } return -1; } int main(void) { int sock; if((sock=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0) /* if((listenfd=socket(PF_INET,SOCK_STREAM<0))<0)*/ ERR_EXIT("socket"); 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);*/ //發起連線 if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0) ERR_EXIT("connect"); char recvbuf[1024]; char sendbuf[1024]; memset(sendbuf, 0, sizeof(sendbuf)); memset(recvbuf, 0, sizeof(recvbuf)); while(fgets(sendbuf,sizeof(sendbuf), stdin) !=NULL) { writen(sock, sendbuf, strlen(sendbuf)); int ret = readline(sock, recvbuf,sizeof(recvbuf));//先接受包頭 if(ret==-1)//讀取失敗則退出 ERR_EXIT("readn"); else if(ret==0)//客戶端關閉 { printf("client_close\n"); break; } fputs(recvbuf.buf,stdout);//列印標準輸出 memset(&sendbuf,0, sizeof(sendbuf)); memset(&recvbuf, 0, sizeof(recvbuf)); } close(sock); return 0; }