TCP回射客户服务器模型(02 设置套接字选项、处理多并发)

时间:2023-03-09 05:13:40
TCP回射客户服务器模型(02 设置套接字选项、处理多并发)

  int setsockopt(int sockfd, int level, int optname, const void *optval, socklen_t optlen);  //设置套接字选项

  回射服务器编写,服务器关闭后处于TIME_WAIT状态,过一段时间才可以绑定刚才的端口。
在绑定服务器之前尽可能调用setsockopt来设置REUSEADDR套接字选项,使用REUSEADDR选项可以使得不必等待TIME_WAIT状态消失就可以重启服务器
  服务器处理多并发,客户端短连接过来就创建子进程通信,客户端关闭后,子进程要退出.父进程一直处于监听。
  TCP回射客户服务器模型(02 设置套接字选项、处理多并发)

  服务器进程程序:

#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<stdlib.h>
#include<stdio.h>
#include<errno.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#define ERR_EXIT(m)\
do\
{\
perror(m);\
exit(EXIT_FAILURE);\
}while(0)
void do_service(int conn)
{
char recvbuf[1024];
while(1)
{
memset(recvbuf,0,sizeof(recvbuf));
int ret=read(conn,recvbuf,sizeof(recvbuf));
//捕捉客户端关闭
if(ret==0)
{
printf("client close\n");
break;//不用继续循环等待客户端数据
}
else if(ret==-1)
ERR_EXIT("read error");
fputs(recvbuf,stdout);
write(conn,recvbuf,ret);
}
}
int main(void)
{
int listenfd;
if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
ERR_EXIT("socket error");
//if((listenfd=socket(PF_INET,SOCK_STREAM,0))<0) //本地协议地址赋给一个套接字
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); //绑定之前,开启地址重复使用,关闭服务器再打开不用等待TIME_WAIT
int on=1;
if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
ERR_EXIT("setsockopt error");
//绑定本地套接字
if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
ERR_EXIT("bind error");
if(listen(listenfd,SOMAXCONN)<0)//设置监听套接字(被动套接字)
ERR_EXIT("listen error"); struct sockaddr_in peeraddr;//对方套接字地址
socklen_t peerlen=sizeof(peeraddr);/一定要初始化
int conn;//已连接套接字(主动套接字)
pid_t pid;
  //原先的程序为何不能处理多客户端连接:因为之前的服务器程序,一旦客户端请求连接,服务器就会一直处于while循环,处理客户端请求,不再处理accept
while(1){
if((conn=accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen))<0)
ERR_EXIT("accept error");//accept本质是到已连接队列中,取第一个连接。
//连接好之后就构成连接,端口是客户端的。peeraddr是对端
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);
//do_service返回,某个客户端关闭,结束该子进程,否则子进程也去接受连接(accept)。
exit(EXIT_SUCCESS);
}else
close(conn);
}
return 0;
}

  客户端程序:

 1 #include<unistd.h>
2 #include<sys/types.h>
3 #include<sys/socket.h>
4 #include<string.h>
5 #include<stdlib.h>
6 #include<stdio.h>
7 #include<errno.h>
8 #include<netinet/in.h>
9 #include<arpa/inet.h>
10 #define ERR_EXIT(m)\
11 do\
12 {\
13 perror(m);\
14 exit(EXIT_FAILURE);\
15 }while(0)
16 int main(void)
17 {
18 int sock;//客户端创建套接字
19 if((sock=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
20 ERR_EXIT("socket error");
21
22 struct sockaddr_in servaddr;//本地协议地址赋给一个套接字
23 memset(&servaddr,0,sizeof(servaddr));
24 servaddr.sin_family=AF_INET;
25 servaddr.sin_port=htons(5188);
26
27 servaddr.sin_addr.s_addr=inet_addr("127.0.0.1");//服务器段地址
28 //inet_aton("127.0.0.1",&servaddr.sin_addr);
29
30 if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
31 ERR_EXIT("connect");
32 char sendbuf[1024]={0};
33 char recvbuf[1024]={0};
34 while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)//默认有换行符
35 {
36 write(sock,sendbuf,strlen(sendbuf));
37 read(sock,recvbuf,sizeof(recvbuf));
38 fputs(recvbuf,stdout);
39
40 memset(sendbuf,0,sizeof(sendbuf));
41 memset(recvbuf,0,sizeof(recvbuf));
42 }
43 close(sock);
44
45 return 0;
46 }