server.cpp
#include <iostream>
using namespace std;
#include <stdlib.h>
#include <sys/socket.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/wait.h>
#include <signal.h>
void fun(int sig){
pid_t pid;
while(waitpid(-1,NULL,WNOHANG) > 0){
// pid = waitpid(-1,NULL,WNOHANG) ;
// if(pid <= 0 ) break;//小于0全部退出,==0表示没有退出
//else
cout<<"回收到进程:"<<sig<<endl;
}
}
int main(){
sigset_t set;
sigemptyset(&set);
sigaddset(&set,SIGCHLD);
sigprocmask(SIG_BLOCK,&set,NULL);
//创建套接字并绑定
int socfd = socket(AF_INET,SOCK_STREAM,0);
struct sockaddr_in ser;
memset((sockaddr*)&ser,0,sizeof(ser));
ser.sin_family = AF_INET;
ser.sin_addr.s_addr = inet_addr("127.0.0.1");
ser.sin_port = htons(10000);
bind(socfd,(sockaddr*)&ser,sizeof(ser));
//监听
listen(socfd,128);
//start
struct sockaddr_in cli;
memset((sockaddr*)&ser,0,sizeof(ser));
socklen_t len = sizeof(cli);
a:
while(1){
//memset((sockaddr*)&cli,0,sizeof(cli));
//bzero(&cli,sizeof(cli));
int clifd = accept(socfd,(sockaddr*)&cli,&len);
if(clifd < 0 ) goto a;
cout<<"new client"<<endl;
cout<<"ip:"<<inet_ntoa(cli.sin_addr)<<endl;
cout<<"port:"<<ntohs(cli.sin_port)<<endl;
//创建子进程
auto pid = fork();
if(pid == -1){cout<<"pid error"<<endl;exit(0);}
else if(pid == 0){//子进程
close(socfd);
while(1){
char buf[1024];
memset(buf,0,1024);
int ret = -1;
ret = read(clifd,buf,sizeof(buf));
if(ret < 0){cout<<"read error"<<endl;close(clifd);exit(0);}
else if(ret == 0){cout<<ntohs(cli.sin_port)<<" client disconnect"<<endl;close(clifd);exit(0);}
else{
cout<<ntohs(cli.sin_port)<<" say "<<buf<<endl;
memset(buf,0,sizeof(buf));
strcpy(buf,"2023-03-01");
write(clifd,buf,sizeof(buf));
cout<<"已向"<<ntohs(cli.sin_port)<<"回复"<<endl;
}
}
}
else{//父进程
close(clifd);
sigprocmask(SIG_UNBLOCK,&set,NULL);
struct sigaction act;
act.sa_flags = 0;
act.sa_handler = fun;
sigemptyset(&act.sa_mask);
sigaction(SIGCHLD,&act,NULL);
//阿里云服务器加了这个代码会乱码
}
}
//close(socfd);
}
client.cpp
#include <iostream>
using namespace std;
#include <sys/socket.h>
#include <arpa/inet.h>
#include <cstring>
#include <unistd.h>
int main(){
int sockfd = socket(AF_INET,SOCK_STREAM,0);
struct sockaddr_in cli;//创建结构体
bzero(&cli,sizeof(cli));
cli.sin_family = AF_INET;
cli.sin_addr.s_addr = inet_addr("127.0.0.1");
cli.sin_port = htons(10000);
connect(sockfd,(sockaddr*)&cli,sizeof(cli));
while(1){
//string str="";
char buf[1024];
bzero(&buf,sizeof(buf));
cin>>buf;
ssize_t size = write(sockfd,buf,sizeof(buf));
//cout<<"客户机说:"<<buf<<endl;
if(size == -1 ){
cout<<"写程序出错"<<endl;
}
bzero(&buf,sizeof(buf));
ssize_t read_size = read(sockfd,buf,sizeof(buf));
if(size > 0 ){
cout<<"服务器说:"<<buf<<endl;
}
else if(size == 0){
cout<<"服务器没有连接"<<endl;
break;
}
else if(size == -1){
cout<<"socket出错"<<endl;
close(sockfd);
}
}
close(sockfd);
return 0;
}
结果展示
暂时存在的问题
问题1
在阻塞信号集,对子进程退出信号阻塞,后续解锁使用SIG_UNLOCK服务器陷入死循环?
解决方法是在accept时进行检查,如果是<0的重新进入循环就会把错误数据过滤掉