回射客户-服务器模型（3）

前面连接的C/S模型只是单方面的客户端发送数据，服务器端接收数据。那么本节将根据前两节已有的经验知识来实现一个简单的点对点聊天程序。

注：这里只实现了以个服务器与一个客户端的相互收发数据。

下面这篇文章对整个回射客户服务器的模型建立过程有一个说明，我这里就不再做过多讲解。由于刚刚接触网络编程，中间可能会有纰漏，还请大家多多指正。

点击打开链接

服务器端：p2psrv.c

#include <stdio.h>#include <stdlib.h>#include <errno.h>#include <arpa/inet.h>#include <sys/types.h>#include <sys/socket.h>#include <unistd.h>#include <signal.h>#include <string.h>void ERR_EXIT(const char* err){perror(err);exit(EXIT_SUCCESS);}void handler(int sig){printf("recv a sig = %d\n", sig);exit(EXIT_SUCCESS);}int main(void){//1.创建套接字int listenfd;if((listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)ERR_EXIT("socket create error");//2.初始化地址struct sockaddr_in servaddr;memset(&servaddr, 0, sizeof(servaddr));servaddr.sin_family = AF_INET;servaddr.sin_port = htons(5188);servaddr.sin_addr.s_addr = INADDR_ANY;//3.在绑定之前先初始化SO_REUSEADDR选项//目的是让服务器在TIME_WAIT状态下仍能开启int on;if(setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)ERR_EXIT("setsockopt error");//4.绑定if(bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)ERR_EXIT("bind error");//5.监听if(listen(listenfd, SOMAXCONN) < 0)ERR_EXIT("listen error");//6.接收连接请求//先给对方指定一个地址结构struct sockaddr_in peeraddr;socklen_t peerlen = sizeof(peeraddr);memset(&peeraddr, 0, sizeof(peeraddr));int conn;if((conn = accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen)) < 0)ERR_EXIT("accept error");printf("IP=%s port=%d\n", inet_ntoa(peeraddr.sin_addr), peeraddr.sin_port);//7.双方开始进行通信//这里我们需要fork出一个子进程，使得父进程用来接收数据，子进程用来发送数据close(listenfd);pid_t pid = fork();if(pid < 0){ERR_EXIT("fork error");}else 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));}close(conn);exit(EXIT_SUCCESS);}else{//父进程处理，接收数据char recvbuf[1024];while(1){memset(recvbuf, 0, sizeof(recvbuf));//要考虑到客户端关闭的情况int ret = read(conn, recvbuf, sizeof(recvbuf));if(ret == -1)perror("read error");else if(ret == 0){printf("peer close\n");break;}fputs(recvbuf, stdout);}//父进程退出的同时也要通知子进程也退出//给子进程一个信号，退出close(conn);kill(pid, SIGUSR1);usleep(1);exit(EXIT_SUCCESS);}return 0;}

客户端：p2pcli.c

#include <stdio.h>#include <stdlib.h>#include <string.h>#include <errno.h>#include <arpa/inet.h>#include <sys/types.h>#include <sys/socket.h>#include <unistd.h>#include <signal.h>void ERR_EXIT(const char* err){perror(err);exit(EXIT_SUCCESS);}void handler(int sig){printf("recv a sig = %d\n", sig);exit(EXIT_SUCCESS);}int main(){//1.创建套接字int sock;if((sock = socket(AF_INET, SOCK_STREAM, 0)) < 0)ERR_EXIT("socket create error");//2.指定对等方的地址struct sockaddr_in servaddr;memset(&servaddr, 0, sizeof(servaddr));servaddr.sin_family = AF_INET;servaddr.sin_port = htons(5188);inet_aton("127.0.0.1", &servaddr.sin_addr);//3.建立连接if(connect(sock, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)ERR_EXIT("connect error");//4.双方开始进行通信//需要fork出一个子进程用于接收数据pid_t pid = fork();if(pid < 0){ERR_EXIT("fork error");}else 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 error");else if(ret == 0){printf("peer close\n");break;}fputs(recvbuf, stdout);}close(sock);//给父进程一个信号，退出kill(getppid(), SIGUSR1);exit(EXIT_SUCCESS);}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);exit(EXIT_SUCCESS);}return 0;}