select 使用难题

lpsir

一个简单的select学习　但是在服务端或者客户端断开的时候
另一个就会不断打印　上次接受到的内容　本人很菜　怎么也看不出来问题
还请高手指教

1. 
   
2. server.c
   
3. #include    <math.h>
   
4. #include    <stdio.h>
   
5. #include    <stdlib.h>
   
6. #include    <string.h>
   
7. #include    <strings.h>
   
8. #include    <fcntl.h>
   
9. #include    <dirent.h>
   
10. #include    <sys/types.h>
    
11. #include    <sys/stat.h>
    
12. #include    <errno.h>
    
13. #include    <sys/msg.h>
    
14. #include    <unistd.h>
    
15. #include    <time.h>
    
16. #include    <sys/socket.h>
    
17. #include    <netinet/in.h>
    
18. #include    <arpa/inet.h>
    
19. #include    <sys/msg.h>
    
20. #include    <signal.h>
    
21. #include    <sys/uio.h>
    
22. #include    <sys/ipc.h>
    
23. #include    <netdb.h>
    
24. #include    <sys/wait.h>
    
25. #include    <sys/time.h>
    
26. 
    
27. void SetExitFlag(int signo);
    
28. 
    
29. int flag=1;
    
30. 
    
31. int main(){
    
32.        
    
33.         struct sockaddr_in servaddr,cliaddr;
    
34.         struct hostent *lhost;
    
35.         int sfd;
    
36.         int reuseaddr = 1;
    
37.         int recvfd;
    
38.         int clilen=sizeof(struct sockaddr_in);
    
39.         int recvlen;
    
40.         char buf[1024];
    
41.         int sendlen;
    
42.         struct                 sigaction act,oact;
    
43.         uint16_t port;
    
44.         char addr[17];
    
45.         fd_set readset,writeset;
    
46.        
    
47.         memset(buf,0,sizeof(buf));
    
48.         memset(addr,0,sizeof(addr));
    
49.        
    
50.         /**信号设置**/
    
51.         act.sa_handler=SetExitFlag;
    
52.         sigemptyset(&act.sa_mask);
    
53.         act.sa_flags=0;
    
54.         if(sigaction(SIGINT,&act,&oact)!=0)
    
55.         {
    
56.                 perror("sigaction");
    
57.                 exit(1);
    
58.         }
    
59.         signal(SIGTERM, SetExitFlag);
    
60.         signal(SIGKILL, SetExitFlag);
    
61.         signal(SIGSTOP, SetExitFlag);
    
62.         signal(SIGQUIT, SetExitFlag);
    
63.        
    
64.         /*lhost=gethostbyaddr("dctest1");
    
65.         printf("主机名:[%s][%s]\n",lhost->h_name,lhost->h_addr);*/
    
66.        
    
67.         sfd = socket(AF_INET, SOCK_STREAM, 0);
    
68.         if(sfd<0)
    
69.                 perror("socket ");
    
70.         if(setsockopt(sfd,SOL_SOCKET, SO_REUSEADDR, &reuseaddr, sizeof(reuseaddr)) == -1)
    
71.                 perror("reuse addr ");
    
72.         if(setsockopt(sfd,SOL_SOCKET, SO_REUSEPORT, &reuseaddr, sizeof(reuseaddr)) == -1)
    
73.                 perror("reuse port");
    
74.                        
    
75.         bzero((char *)&servaddr,sizeof(servaddr));
    
76.         servaddr.sin_family=AF_INET;
    
77.         servaddr.sin_addr.s_addr=htonl(INADDR_ANY);
    
78.         servaddr.sin_port=htons(10006);
    
79.        
    
80.         if(bind(sfd,(struct sockaddr *)&servaddr,sizeof(struct sockaddr_in)) == -1)
    
81.                 perror("bind ");
    
82.        
    
83.         if(listen(sfd,10) == -1)
    
84.                 perror("listen ");
    
85.        
    
86.        
    
87.         while(flag)
    
88.         {
    
89.                
    
90.                 if((recvfd=accept(sfd,(struct sockaddr *)&cliaddr,&clilen)) == -1)
    
91.                 {       
    
92.                         perror("accept ");
    
93.                         break;
    
94.                 }
    
95.                 port=ntohs(cliaddr.sin_port);
    
96.                 strcpy(addr,inet_ntoa(cliaddr.sin_addr));
    
97.                 printf("接收到连接　端口:[%d]ip[%s]\n",port,addr);
    
98.                 /*FD_ZERO(&readset);
    
99.                 FD_SET(0,&readset);
    
100.                 FD_SET(recvfd,&readset);
     
101.                 if(select(recvfd+1,&readset,NULL,NULL,NULL)<0)
     
102.                         perror("select ");*/
     
103.                 while(flag)
     
104.                 {
     
105.                         FD_ZERO(&readset);
     
106.                         FD_SET(0,&readset);
     
107.                         FD_SET(recvfd,&readset);
     
108.                         if(select(recvfd+1,&readset,NULL,NULL,NULL)<0)
     
109.                                 perror("select ");
     
110.                         if(FD_ISSET(0,&readset))
     
111.                         {       
     
112.                                 gets(buf);
     
113.                                 sendlen=write(recvfd,buf,strlen(buf));
     
114.                                 if(sendlen < 0 )
     
115.                                 {
     
116.                                         perror("send ");
     
117.                                         close(recvfd);       
     
118.                                         break;
     
119.                                 }
     
120.                         }
     
121.                         if(FD_ISSET(recvfd,&readset))
     
122.                         {       
     
123.                                 recvlen=read(recvfd,buf,1024);
     
124.                                 if(recvlen < 0)
     
125.                                 {       
     
126.                                         perror("recv ");
     
127.                                         close(recvfd);       
     
128.                                         break;
     
129.                                 }       
     
130.                                 else
     
131.                                         printf("recv:[%s]\n",buf);
     
132.                         }       
     
133.                                
     
134.                 }
     
135.                 close(recvfd);
     
136.         }
     
137. }
     
138. 
     
139. void SetExitFlag(int signo)
     
140. {
     
141.         flag=0;
     
142.         printf("接收到退出信号\n");
     
143. }
     
144. 
     
145. client.c
     
146. #include    <math.h>
     
147. #include    <stdio.h>
     
148. #include    <stdlib.h>
     
149. #include    <string.h>
     
150. #include    <strings.h>
     
151. #include    <fcntl.h>
     
152. #include    <dirent.h>
     
153. #include    <sys/types.h>
     
154. #include    <sys/stat.h>
     
155. #include    <errno.h>
     
156. #include    <sys/msg.h>
     
157. #include    <unistd.h>
     
158. #include    <time.h>
     
159. #include    <sys/socket.h>
     
160. #include    <netinet/in.h>
     
161. #include    <arpa/inet.h>
     
162. #include    <sys/msg.h>
     
163. #include    <signal.h>
     
164. #include    <sys/uio.h>
     
165. #include    <sys/ipc.h>
     
166. #include    <netdb.h>
     
167. #include    <sys/wait.h>
     
168. #include    <sys/time.h>
     
169. 
     
170. 
     
171. void SetExitFlag(int signo);
     
172. 
     
173. int flag=1;
     
174. 
     
175. int main(){
     
176.        
     
177.         struct sockaddr_in conaddr;
     
178.         int cfd;
     
179.         char buf[1024];
     
180.         int sendlen;
     
181.         int recvlen;
     
182.         fd_set readset,writeset;
     
183.         struct                 sigaction act,oact;
     
184.        
     
185.         memset(buf,0,1024);
     
186.         /**信号设置**/
     
187.         act.sa_handler=SetExitFlag;
     
188.         sigemptyset(&act.sa_mask);
     
189.         act.sa_flags=0;
     
190.         if(sigaction(SIGINT,&act,&oact)!=0)
     
191.         {
     
192.                 perror("sigaction");
     
193.                 exit(1);
     
194.         }
     
195.         signal(SIGTERM, SetExitFlag);
     
196.         signal(SIGKILL, SetExitFlag);
     
197.         signal(SIGSTOP, SetExitFlag);
     
198.         signal(SIGQUIT, SetExitFlag);
     
199.        
     
200.         cfd=socket(AF_INET,SOCK_STREAM,0);
     
201.         if(cfd == -1)
     
202.                 perror("socket ");
     
203.        
     
204.         bzero((char *)&conaddr,sizeof(conaddr));
     
205.        
     
206.         conaddr.sin_family=AF_INET;
     
207.         conaddr.sin_addr.s_addr=inet_addr("172.16.227.11");
     
208.         conaddr.sin_port=htons(10006);
     
209.        
     
210.         if(connect(cfd,(struct sockaddr *)&conaddr,sizeof(struct sockaddr_in)) == -1)
     
211.         {       
     
212.                 perror("bind ");
     
213.                 close(cfd);
     
214.                 exit(1);
     
215.         }       
     
216.        
     
217.         /*FD_ZERO(&readset);
     
218.         FD_SET(0,&readset);
     
219.         FD_SET(cfd,&readset);
     
220.         if(select(cfd+1,&readset,NULL,NULL,NULL)<0)
     
221.                 perror("select ");*/
     
222.         while(flag)
     
223.         {       
     
224.                 FD_ZERO(&readset);
     
225.                 FD_SET(0,&readset);
     
226.                 FD_SET(cfd,&readset);
     
227.                 if(select(cfd+1,&readset,NULL,NULL,NULL)<0)
     
228.                         perror("select ");
     
229.                 memset(buf,0,sizeof(buf));
     
230.                 if(FD_ISSET(0,&readset))
     
231.                 {       
     
232.                         gets(buf);
     
233.                         sendlen=write(cfd,buf,sizeof(buf));
     
234.                         if(sendlen<0)
     
235.                         {
     
236.                                 perror("send ");
     
237.                                 break;
     
238.                         }
     
239.                 }
     
240.                 if(FD_ISSET(cfd,&readset))
     
241.                 {       
     
242.                         recvlen=read(cfd,buf,1024);
     
243.                         if(recvlen<0)
     
244.                         {
     
245.                                 perror("recv ");
     
246.                                 break;
     
247.                         }
     
248.                         else       
     
249.                                 printf("接受:[%s]\n",buf);
     
250.                 }       
     
251.         }
     
252.                
     
253.         close(cfd);               
     
254.        
     
255. }       
     
256. 
     
257. void SetExitFlag(int signo)
     
258. {
     
259.         flag=0;
     
260.         printf("接收到退出信号\n");
     
261. }
     

复制代码

x11

你把recvlen打印出来看看

lpsir

什么意思呢
就是对方连接断了　是不是我判断recvlen　错了呢
把recvlen改成＜＝０的判断就可以了
perror　的输出是　Invalid argument

这个判断就可以了吧　谢谢了

realtang

select为－1时，程序应该exit了吧，都出错了。当然如果errno为EINTR还是可以继续让它做下去的。

lpsir

你理解错我的意思了
比如说　客户端断开　　这时候　服务端会不断打印东西
这个时候服务端的select不会报错的
呵呵

realtang

这时你应该判断write的返回值吧，为-1就意味着出错，程序也该exit才是。

lpsir

情况是这样的
客户端退出
但是服务端并没有向ｓｏｃｋｅｔ写东西
所以取法判断　write的返回值的
只能用read来做判断