Linux下p2p的聊天功能实现
Linux下p2p的聊天功能实现细节
Do one thing at a time, and do well.
今天闲着没事，写一个P2P的点对点的聊天功能的小程序，我觉得对网络编程初学者的学习很有用的。二话不说，我先贴代码吧。有几个地方需要考虑清楚。我会在代码的后面写出来。代码的下载文章的末尾。
server.c
#include <stdio.h> #include <sys/types.h> #include <sys/socket.h> #include <stdlib.h> #include <netinet/in.h> #include <arpa/inet.h> #include <unistd.h> #include <string.h> #include <signal.h>
#define ERR_EXIT(m) \ do \ { \ perror(m); \ exit(EXIT_FAILURE); \ } while(0) void do_something(int conn) { char recvbuf[1024]; for(;;) { memset(recvbuf,0,sizeof(recvbuf)); int ret = read(conn,recvbuf,sizeof(recvbuf)); if(ret == 0) { printf("client closed!\n"); break; } else if(ret == -1) { ERR_EXIT("read"); } fputs(recvbuf,stdout); write(conn,recvbuf,ret); } } void handler(int sig) { printf("recv a sig = %d\n",sig); exit(EXIT_SUCCESS); } int main() { int listenfd; if((listenfd = socket(AF_INET,SOCK_STREAM,0)) < 0) ERR_EXIT("socket"); int on = 1; int ret = setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on) ); struct sockaddr_in servaddr; memset(&servaddr,0,sizeof(servaddr)); servaddr.sin_family = AF_INET; servaddr.sin_port = htons(10001); servaddr.sin_addr.s_addr = htonl(INADDR_ANY); 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 socklen = sizeof(peeraddr); int conn; pid_t pid; if((conn = accept(listenfd,(struct sockaddr*)&peeraddr,&socklen)) < 0)// 获得到是主动套接字 ERR_EXIT("accept"); printf("ip:%s port:%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port)); pid = fork(); char sendbuf[1024] = {0}; if(pid == -1) ERR_EXIT("pid"); if(pid == 0) { signal(SIGUSR1,handler); while(fgets(sendbuf,sizeof(sendbuf),stdin) != NULL) { write(conn,sendbuf,strlen(sendbuf)); memset(sendbuf,0,sizeof(sendbuf)); } printf("child closed\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; } fputs(recvbuf,stdout); } printf("kill parent!\n"); kill(pid,SIGUSR1); exit(EXIT_SUCCESS); //do_something(conn); } close(conn); close(listenfd); exit(0); }
#include <stdio.h> #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <string.h> #include <unistd.h> #include <stdlib.h> #include <arpa/inet.h> #include <signal.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(int argc,char *argv[]) { int sockfd; if((sockfd = socket(AF_INET,SOCK_STREAM,0)) < 0) ERR_EXIT("socket"); struct sockaddr_in servaddr; memset(&servaddr,0,sizeof(servaddr)); servaddr.sin_family = AF_INET; servaddr.sin_port = htons(10001); servaddr.sin_addr.s_addr = inet_addr("127.0.0.1"); if(connect(sockfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0) ERR_EXIT("connect"); char sendbuf[1024] = {0}; char recvbuf[1024] = {0}; pid_t pid; pid = fork(); if(pid == -1) ERR_EXIT("fork"); if(pid == 0) { while(1) { memset(recvbuf,0,sizeof(recvbuf)); int ret = read(sockfd,recvbuf,sizeof(recvbuf)); if(ret == -1) ERR_EXIT("read"); else if (ret == 0) { printf("peer closed\n"); break; } fputs(recvbuf,stdout); } printf("child close\n"); kill(pid,SIGUSR1); exit(EXIT_SUCCESS); } else { signal(SIGUSR1,handler); while(fgets(sendbuf,sizeof(sendbuf),stdin) != NULL) { write (sockfd,sendbuf,strlen(sendbuf)); memset(sendbuf,0,sizeof(sendbuf)); } printf("parent close!\n"); } close(sockfd); exit(0); }
实现的功能很简单，但是需要注意的几个细节：
1、C和S连接以后，当S关闭后，C仍然没有关闭，我用到了信号的功能。
实现的方法：当父进程关闭的时候，子进程也关闭，当子进程关闭的时候，把父进程也关闭了。
2、一个线程用来监听，一个线程用来等待输入。那么，这是一个多线程的小程序。
创建一个子进程。C端，子进程进程监听，父进程等待输入。S端相反。
3、信号量的问题：
SIGUSR1：用户自定义信号量，函数handler用来杀死进程，实现退出。
程序的测试：
总结：总的来说，这个还是很简单的，也就是几个函数是否灵活应用。注意，read函数如果没有读到，就会进入阻塞，如果收到一个0，代表对方关闭了程序，则退出程序。
代码下载：GitHub
声明：水平有限，如果有什么地方写错了或者理解有误，希望广大网友指正。