1 /*服務器:
 2 1.客戶端關閉后，服務器再向客戶端發送信息，第一次會收到一個RST復位報文，第二次會收到SIGPIPE信號，導致服務器關閉，必須對這個信號進行處理：
 3     1.在服務器對read返回值為0的情況進行處理，不向客戶端發送信息
 4     2.signal函數: signal(SIGPIPE, handle) 或者直接忽略signal(SIGPIPE, SIG_IGN)
 5 */
 6 #include <stdio.h>
 7 #include <stdlib.h>
 8 #include <string.h>
 9 #include <unistd.h>
10 #include <errno.h>
11 #include <sys/types.h>
12 #include <sys/stat.h>
13 #include <sys/socket.h>
14 #include <arpa/inet.h>
15 #include <signal.h>
16 
17 #define ERR_EXIT(m) \
18     do { \
19         perror(m);\
20         exit(EXIT_FAILURE);\
21     }while(0)//宏定義的錯誤處理
22 
23 static void do_service(int fd)
24 {
25     char recvbuf[1024] = {0};
26     int ret;
27     while(1)
28     {
29         memset(recvbuf, 0, sizeof recvbuf);
30         ret = read(&rt, recvbuf, 1024);
31         sleep(3);
32         if(ret == 0)//關閉客戶端時會導致ret?== 0
33             printf("no message\n");
34         else if(ret == -1)
35         {
36             if(errno == EINTR)
37                 continue;
38             return ;
39         }   
40         rio_writen(fd, recvbuf, 5);
41     }
42 }
43 
44 void handle(int signum)//SIGPIPE信號處理函數
45 {
46 printf("hello\n");
47 }
48 
49 int main(int argc, const char *argv[])
50 {
51     if(signal(SIGPIPE, handle) == SIG_ERR)
52         ERR_EXIT("signal");    
53 
54     int listenfd = socket(AF_INET, SOCK_STREAM, 0);//準備一個socketfd
55     if(listenfd == -1 )
56         ERR_EXIT("listen");
57 
58     int on = 1;
59     if(setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1)//setsockopt設置端口復用
60     {
61         close(listenfd);
62         ERR_EXIT("setsockopt");
63     }
64 
65     struct sockaddr_in seraddr;
66     seraddr.sin_family = AF_INET;
67     seraddr.sin_port = htons(8888);
68     seraddr.sin_addr.s_addr = htonl(INADDR_ANY);
69     socklen_t len = sizeof(seraddr);
70     if(bind(listenfd, (struct sockaddr*)&seraddr, len) == -1)//監聽socket端口，
71     {
72         close(listenfd);
73         ERR_EXIT("bind");
74     }
75?//上述過程可以封裝到起來，不用全部寫入main
76     if(listen(listenfd, 6) == -1)
77     {
78         close(listenfd);
79         ERR_EXIT("listen");
80     }
81 
82     struct sockaddr_in cliaddr;
83     bzero(&cliaddr, sizeof(cliaddr));
84     socklen_t cli_len = sizeof cliaddr;
85 
86     int clientfd;
87     clientfd = accept(listenfd, (struct sockaddr*)&cliaddr, &cli_len);
88 
89     if(clientfd == -1)  
90     {
91         close(listenfd);
92         ERR_EXIT("accept");
93     }
94     do_service(clientfd);
95 
96     close(clientfd);
97     close(listenfd);
98     return 0;
99 }

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3 #include <string.h>
 4 #include <errno.h>
 5 #include <sys/types.h>
 6 #include <sys/socket.h>
 7 #include <netinet/in.h>
 8 #include <arpa/inet.h>
 9 #include <sys/select.h>
10 
11 #define ERR_EXIT(m) \
12     do { \
13         perror(m);\
14         exit(EXIT_FAILURE);\
15     }while(0)
16 
17 static void do_client(int fd)
18 {
19     char recvbuf[MAXLINE + 1] = {0};
20     char sendbuf[MAXLINE + 1] = {0};
21 
22     fd_set reade, ready;//將ready集合放入select輪詢，每次輪詢前將reade賦值給ready
23     FD_ZERO(&reade);//清空reade集合
24     int fd_stdin = fileno(stdin);
25     FD_SET(fd_stdin, &reade);
26     FD_SET(fd, &reade);//將需要監聽的文件描述符加入集合
27     int fd_max = (fd_stdin > fd) ? fd_stdin : fd;//select輪詢的最大文件描述符
28 
29     int ret;
30     while(1)
31     {
32         ready = reade;
33         ret = select( fd_max+1, &ready, NULL, NULL, NULL);//輪詢，最后一個參數struct *timeval設置為NULL表示即時輪詢（每次輪詢間沒時間間隔）
34         if(ret < 0)
35         {
36             if(errno == EINTR)
37                 continue;
38             ERR_EXIT("select");
39         }else if(ret ==  0)//監聽的文件描述符狀態未發生變化
40         {
41             continue;
42         }
43 
44         if(FD_ISSET(fd_stdin, &ready))
45         {
46             if(fgets(sendbuf, sizeof(sendbuf), stdin) == NULL)
47             {
48                 shutdown(fd, SHUT_WR);//關閉fd的寫端，close關閉時關閉整個描述符,若再寫的話會導致write返回-1，系統會向客戶端進程發送一個SIGPIPE信號，
               導致關閉，可以對信號進行signal(SIGPIPE, SIG_IGN)處理，也可將fd_stdin從監聽隊列刪除
49                // close(fd);
50                 exit(EXIT_SUCCESS);
51                ?//break;
52                   }else
53                 write(fd, sendbuf, strlen(sendbuf));           
54         }
55 
56         if(FD_ISSET(fd, &ready))
57         {
58             int nread = read(fd, recvbuf, MAXLINE);
59             if(nread < 0)
60                 ERR_EXIT("read");
61             if(nread == 0)//如果沒接收到消息，打印關閉描述符，退出循環
62             {
63                 fprintf(stdout, "fd close\n");
64                 break;
65             }
66             fprintf(stdout, "receive:%s\n", recvbuf);//注意要刷新
67         }
68         memset(recvbuf, 0, sizeof recvbuf);
69         memset(sendbuf, 0, sizeof sendbuf);
70     }
71 }
72 
73 int main(int argc, const char *argv[])
74 {
75     int fd = socket(AF_INET, SOCK_STREAM, 0);
76     if(fd < 0)
77         ERR_EXIT("socket");
78 
79     struct sockaddr_in cliaddr;
80     cliaddr.sin_family = AF_INET;
81     cliaddr.sin_port = htons(8888);
82     cliaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
83     socklen_t len = sizeof cliaddr;
84 
85     int ret ;
86     if((ret = connect(fd, (struct sockaddr*)&cliaddr, len)) == -1)
87     {
88         close(fd);
89         ERR_EXIT("connect");
90     }
91     do_client(fd);
92     close(fd);
93     printf("123\n");
94     return 0;
95 }