server
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.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)
void handler(int sig){
printf("recv a sig=%d\n",sig);
exit(EXIT_SUCCESS);
}
int main(void){
int listenfd;
if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<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;
if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
ERR_EXIT("setsockop");
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(peerlen);
int conn;
if((conn=accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen))<0)
ERR_EXIT("accept");
printf("ip=%s port=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
pid_t pid;
pid=fork();
if(pid==-1)
ERR_EXIT("fork");
if(pid==0){
signal(SIGUSR1,handler);
char sendbuf[1024]={0};
while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL){
write(conn,sendbuf,strlen(sendbuf));
memset(sendbuf,0,sizeof(sendbuf));
}
printf("child close\n");
exit(EXIT_SUCCESS);
}else{
char recvbuf[1024];
while(1){
memset(recvbuf,0,sizeof(recvbuf));
int ret=read(conn,recvbuf,sizeof(recvbuf));
if(ret==-1)
ERR_EXIT("read");
else if(ret==0){
printf("peer close\n");
break;
}
puts("client say:");
fputs(recvbuf,stdout);
}
printf("parent close\n");
kill(pid,SIGUSR1);
exit(EXIT_SUCCESS);
}
return 0;
}
client:
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.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)
void handler(int sig){
printf("recv a sig=%d\n",sig);
exit(EXIT_SUCCESS);
}
int main(void){
int sock;
if((sock=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<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=inet_addr("127.0.0.1");
if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
ERR_EXIT("connect");
pid_t pid;
pid=fork();
if(pid==-1)
ERR_EXIT("fork");
if(pid==0){
char recvbuf[1024];
while(1){
memset(recvbuf,0,sizeof(recvbuf));
int ret=read(sock,recvbuf,sizeof(recvbuf));
if(ret==-1)
ERR_EXIT("read");
else if(ret==0){
printf("peer close\n");
break;
}
puts("server say:");
fputs(recvbuf,stdout);
}
close(sock);
kill(getppid(),SIGUSR1);
}else{
signal(SIGUSR1,handler);
char sendbuf[1024]={0};
while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL){
write(sock,sendbuf,strlen(sendbuf));
memset(sendbuf,0,sizeof(sendbuf));
}
close(sock);
}
return 0;
}
运行: