- 一、客户端
- 二、服务器
- 1.多进程
- 2.多线程
- 总结
一、客户端
代码如下：
// TCP通信客户端 #include二、服务器 1.多进程#include #include #include #include int main() { //1.创建套接字 int fd = socket(AF_INET, SOCK_STREAM, 0); if(fd==-1) { perror("socket"); exit(-1); } //2.连接服务器 struct sockaddr_in saddr; inet_pton(AF_INET, "182.61.11.197", &saddr.sin_addr.s_addr); saddr.sin_family = AF_INET; saddr.sin_port = htons(9999); int ret = connect(fd, (struct sockaddr *)&saddr, sizeof(saddr)); if(ret==-1) { perror("connet"); exit(-1); } //3.通信 char recvbuf[1024] = {0}; int i = 0; while (1) { sprintf(recvbuf, "%dn", i++); write(fd, recvbuf, strlen(recvbuf)+1); int lens = read(fd, recvbuf, sizeof(recvbuf)); if(lens==-1) { perror("read"); exit(-1); }else if(lens>0) { printf("recv server data: %sn", recvbuf); }else if(lens==0) {//表示客户端断开连接 printf("server closed..."); } sleep(1); } //关闭连接 close(fd); return 0; }
代码如下：
#include2.多线程#include #include #include #include #include #include #include #include void recyleChild(int arg) { while(1) { int ret = waitpid(-1, NULL, WNOHANG); if(ret == -1) { //所有的子进程都回收了 break; }else if(ret == 0) { //还有子进程活着 break; }else if(ret > 0) { //被回收了 printf("子进程 %d 被回收了n", ret); } } } int main() { struct sigaction act; act.sa_flags = 0; sigemptyset(&act.sa_mask); act.sa_handler = recyleChild; //注册信号捕捉 sigaction(SIGCHLD, &act, NULL); //创建socket int lfd = socket(PF_INET, SOCK_STREAM, 0); if(lfd == -1) { perror("socket"); exit(-1); } struct sockaddr_in saddr; saddr.sin_family = AF_INET; saddr.sin_port = htons(9999); saddr.sin_addr.s_addr = INADDR_ANY; //绑定 int ret = bind(lfd, (struct sockaddr *)&saddr, sizeof(saddr)); if(ret==-1) { perror("bind"); exit(-1); } //监听 ret = listen(lfd, 5); if(ret==-1) { perror("listen"); exit(-1); } //不断循环等待客户端连接 while(1) { struct sockaddr_in caddr; int len = sizeof(caddr); //接受连接 int cfd = accept(lfd, (struct sockaddr *)&caddr, &len); if(cfd==-1) { if(errno == EINTR) { continue; } perror("accept"); exit(-1); } //每一个连接进来，创建一个子进程跟客户端通信 pid_t pid = fork(); if(pid==0) { //子进程 //获取客户端的信息 char cip[16]; inet_ntop(AF_INET, &caddr.sin_addr.s_addr, cip, sizeof(cip)); unsigned short cport = ntohs(caddr.sin_port); printf("client ip is :%s, port is %dn", cip, cport); //接受客户端发来的数据 char recvbuf[1024]; while(1) { int len = read(cfd, &recvbuf, sizeof(recvbuf)); if(len == -1) { perror("read"); exit(-1); }else if(len > 0) { printf("recv client data: %sn", recvbuf); }else { printf("client closed...n"); break; } write(cfd, recvbuf, strlen(recvbuf)+1); } close(cfd); exit(0); //退出当前子进程 } } close(lfd); return 0; }
代码如下：
#include#include #include #include #include #include #include struct sockInfo{ int fd;//通信的文件描述符 pthread_t tid;//线程号 struct sockaddr_in addr; }; struct sockInfo sockinfos[128]; void * working(void *arg) { //子线程和客户端通信 cfd 客户端信息 线程号 //获取客户端的信息 struct sockInfo* pinfo = (struct sockInfo*)arg; char cip[16]; inet_ntop(AF_INET, &pinfo->addr.sin_addr.s_addr, cip, sizeof(cip)); unsigned short cport = ntohs(pinfo->addr.sin_port); printf("client ip is :%s, port is %dn", cip, cport); //接受客户端发来的数据 char recvbuf[1024]; while(1) { int len = read(pinfo->fd, &recvbuf, sizeof(recvbuf)); if(len == -1) { perror("read"); exit(-1); }else if(len > 0) { printf("recv client data: %sn", recvbuf); }else { printf("client closed...n"); break; } write(pinfo->fd, recvbuf, strlen(recvbuf)+1); } close(pinfo->fd); } int main() { //创建socket int lfd = socket(PF_INET, SOCK_STREAM, 0); if(lfd == -1) { perror("socket"); exit(-1); } struct sockaddr_in saddr; saddr.sin_family = AF_INET; saddr.sin_port = htons(9999); saddr.sin_addr.s_addr = INADDR_ANY; //绑定 int ret = bind(lfd, (struct sockaddr *)&saddr, sizeof(saddr)); if(ret==-1) { perror("bind"); exit(-1); } //监听 ret = listen(lfd, 5); if(ret==-1) { perror("listen"); exit(-1); } //初始化数据 int max = sizeof(sockinfos) / sizeof(sockinfos[0]); for (int i = 0; i < max;i++) { bzero(&sockinfos[i], sizeof(sockinfos[i])); sockinfos[i].fd = -1; sockinfos[i].tid = -1; } //循环等待客户端连接，一旦一个客户端连接起来，就创建一个子线程进行通信 while (1) { struct sockaddr_in caddr; int len = sizeof(caddr); //接受连接 int cfd = accept(lfd, (struct sockaddr *)&caddr, &len); struct sockInfo *pinfo; for (int i = 0; i < max;i++) {//从这个数组中找到一个可以用sockInfo元素 if(sockinfos[i].fd==-1) { pinfo = &sockinfos[i]; break; } if(i == max-1) { sleep(1); i=-1; } } pinfo->fd=cfd; memcpy(&pinfo->addr, &caddr, len); //创建子线程 pthread_create(&pinfo->tid, NULL, working, pinfo); pthread_detach(pinfo->tid); } close(lfd); return 0; }
总结
本节主要介绍了TCP实现并发服务器。主要运用了多线程和多进程的知识点。
