多线程TCP网络聊天室

我们都知道TCP是一种基于连接的传输协议,使用多线程来实现其实步骤很简单。
首先说明一下项目:服务端开启后,直接运行客户端可以加入聊天室,同时其他客户端可以接收到某某客户端进入聊天室的消息,如果有个客户端主动退出或者意外退出,其他客户端同样会收到某某客户端退出的消息,除了这两个注意的点以外其他就是正常聊天。
注:项目只是完成基本功能,涉及一些网络安全和一些网络加密方面我也不太懂。

客户端

主程序 main函数内部

首先创建网络套接字sockfd,sockfd我创建到全局变量里方便线程使用

sockfd = socket(AF_INET, SOCK_STREAM, 0); // 创建网络套接字

创建server的网络信息结构体并填充

struct sockaddr_in serverSock;
socklen_t serverLen = sizeof(struct sockaddr_in);
//地址为回环地址,如果想要使用其他地址请把htonl(INADDR_ANY)修改为inet_addr(IP),IP为自己地址
serverSock.sin_addr.s_addr = htonl(INADDR_ANY);
// 端口自行设置
serverSock.sin_port = htons(PORT);
serverSock.sin_family = AF_INET;

连接服务端

connect(sockfd, (struct sockaddr *)&serverSock, serverLen);

接下来就是打印下说明,好让人知道已经连接到服务端了

printf("=====欢迎加入聊天室=====\n");
printf("  退出服务器请输入quit  \n");

接下来就是创建两个线程:接收消息和发送消息

pthread_t pthreadSend;
pthread_create(&pthreadSend, NULL, pthread_send, NULL);
pthread_t pthreadRecv;
pthread_create(&pthreadRecv, NULL, pthread_recv, NULL);

在线程创建完成之后我们不要让这个程序结束,否则主程序结束线程也就跟着结束了。

while(!quit);
return 0;

quit变量是一个全局变量,是用在发送端主动输入quit的时候传递变量来结束主程序。
到这里主程序已经结束,接下来就是两个线程的内部功能

接收线程

void *pthread_recv(void *arg)
{msg_t msg; // 接收消息的结构体,我会放到head.h文件中解释bzero(&msg, 0); // 初始化while (1) //循环{/* 阻塞接收函数,ret返回值用处来表明接收消息是否正常 *//* 同时也能知道其他客户端是否还在不在聊天室,比如下文的等于0 */int ret = recv(sockfd, &msg, sizeof(msg), 0); if (ret > 0){/* type变量是一个区分,1为正常消息,0为不正常,通常是有客户端主动正常退出 */if (msg.type == 1 && strlen(msg.message) > 0){/* 打印消息同时打印是哪个客户端打印的 */printf("客户端(%s:%d):%s\n", inet_ntoa(msg.cli.client.sin_addr), ntohs(msg.cli.client.sin_port), msg.message);}else{printf("%s\n", msg.message);}}/* 为0就是某某客户端断开连接 */else if (ret == 0){printf("%s\n", msg.message);}}
}

发送线程

void *pthread_send(void *arg)
{msg_t msg;bzero(&msg, 0);while (1){/* 读取输入 */scanf("%s", msg.message);/* 防止读取到一个回车就发送 */if (getchar() == '\n' && msg.message[0] != '\n'){/* 安全起先还是加了一个\0 */int len = strlen(msg.message);msg.message[len + 1] = '\0';send(sockfd, &msg, sizeof(msg), 0);/* 如果是quit的话令quit变量为true,这样主程序会结束 */if (!strcmp(msg.message, "quit")){quit = true;break;}memset(msg.message, 0, sizeof(msg.message));}}
}

到这里客户端代码结束。

服务端

服务端采用多线程的话,思路是这样的

服务端 main函数内部

int main()
{sockfd = socket(AF_INET, SOCK_STREAM, 0);struct sockaddr_in serverSock;socklen_t serverLen = sizeof(struct sockaddr_in);serverSock.sin_addr.s_addr = htonl(INADDR_ANY);serverSock.sin_port = htons(PORT);serverSock.sin_family = AF_INET;bind(sockfd, (struct sockaddr *)&serverSock, serverLen);listen(sockfd, 0);pthread_t acceptPthread;pthread_create(&acceptPthread, NULL, pthread_accept, NULL);while (1);return 0;
}

main函数整体思路跟客户端是一致的,唯一区别就是只有一个接收线程,同时while(1)死循环,只是为了实现客户端主动退出,并未考虑服务端主动关闭。

服务端接收线程函数

void *pthread_accept(void *arg)
{struct sockaddr_in clientSock;socklen_t clientLen = sizeof(struct sockaddr_in);printf("===========聊天室已创建===========\n");while (1){int acceptFd = accept(sockfd, (struct sockaddr *)&clientSock, &clientLen);/*pNode保存客户端信息,并将这个结构体传递给common的线程*//*common线程是把客户端加入到链表中之后再创建的用于接收和转发聊天*/cli_t *pNode = (cli_t *)malloc(sizeof(cli_t));pNode->client = clientSock;pNode->fd = acceptFd;pNode->next = NULL;/*采用头插法*/if (phead.size > 0){cli_t *p = phead.cli.next;phead.cli.next = pNode;pNode->next = p;}else if (phead.size == 0){phead.cli.next = pNode;}phead.size++;pthread_t commonPthread;/*创建聊天线程*/pthread_create(&commonPthread, NULL, pthread_common, (void *)pNode);pthread_detach(commonPthread);}
}

服务端聊天线程

void *pthread_common(void *arg)
{/*传递过来的pNode结构体中包含客户端的信息*/cli_t *pNode = (cli_t *)arg;/*打印给服务端显示*/printf("客户端(%s:%d)加入聊天室\n", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));/*如果客户端链表大于1就转发某某加入聊天室的信息,并通知其他客户端*/if (phead.size > 1){cli_t *p = &phead.cli;msg_t msg;bzero(&msg, 0);msg.cli = *pNode;msg.type = 0;sprintf(msg.message, "客户端(%s:%d)加入聊天室", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));while (p->next != NULL){p = p->next;/*除自己外不用发*/if (p->fd == pNode->fd)continue;send(p->fd, &msg, sizeof(msg), 0);}}/*死循环,接收和转发消息或者处理客户端消息*/while (1){msg_t msgs;bzero(&msgs, 0);int ret = recv(pNode->fd, &msgs, sizeof(msgs), 0);/* 这里对客户端链表上锁,以防止两个或多个客户端同时退出造成 *//* 服务端崩溃或者转发消息失败 */pthread_mutex_lock(&phead.lock);if (ret > 0){/*先对发来quit消息的客户端进行处理*/if (!strcmp("quit", msgs.message)){phead.size--;msgs.type = 0;printf("客户端(%s:%d)退出聊天室\n", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));sprintf(msgs.message, "客户端(%s:%d)退出聊天室", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));cli_t *p = &phead.cli;while (p->next != NULL){p = p->next;if (p->fd == pNode->fd){continue;}send(p->fd, &msgs, sizeof(msgs), 0);}/* 待消息转发结束之后解锁链表 */pthread_mutex_unlock(&phead.lock);/* 从链表中删除此客户端信息 */delNodeFromList(&phead, pNode->fd);/* 关闭此套接字 */close(pNode->fd);/* 线程结束 */pthread_cancel(pthread_self());bzero(&msgs, 0);break;}else // 如果不是quit就正常转发此客户端发来的消息{msgs.cli = *pNode;cli_t *p = &phead.cli;/* type = 1就是为了能够客户端那边打印客户端消息 */msgs.type = 1; while (p->next != NULL){p = p->next;if (p->fd == pNode->fd)continue;send(p->fd, &msgs, sizeof(msgs), 0);}/* 正常发送消息结束后解锁链表 */pthread_mutex_unlock(&phead.lock);}}/* 这就是ret小于等于0的情况了,表示客户端意外退出 *//* 其处理方法和上边quit的处理相同,不过就是发送消息改为了 *//* 意外退出而已 */else{phead.size--;msgs.type = 0;printf("客户端(%s:%d)意外退出聊天室\n", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));sprintf(msgs.message, "客户端(%s:%d)意外退出聊天室", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));cli_t *p = &phead.cli;while (p->next != NULL){p = p->next;if (p->fd == pNode->fd){continue;}send(p->fd, &msgs, sizeof(msgs), 0);}pthread_mutex_unlock(&phead.lock);delNodeFromList(&phead, pNode->fd);pthread_cancel(pthread_self());close(pNode->fd);bzero(&msgs, 0);break;}}
}

服务端 链表删除客户端节点函数

void delNodeFromList(msg_t *phead, int fd)
{if (phead == NULL)return;cli_t *p = &phead->cli;cli_t *pDel = NULL;/* 遍历链表找到要删除客户端的节点令pDel指向它 */while (p->next != NULL){if (p->next->fd == fd){pDel = p->next;break;}p = p->next;}p->next = pDel->next;pDel->next = NULL;free(pDel);pDel = NULL;
}

服务端server.c文件

#include "../head.h"
msg_t phead;
int sockfd;void delNodeFromList(msg_t *phead, int fd)
{if (phead == NULL)return;cli_t *p = &phead->cli;cli_t *pDel = NULL;/* 遍历链表找到要删除客户端的节点令pDel指向它 */while (p->next != NULL){if (p->next->fd == fd){pDel = p->next;break;}p = p->next;}p->next = pDel->next;pDel->next = NULL;free(pDel);pDel = NULL;
}void *pthread_common(void *arg)
{/*传递过来的pNode结构体中包含客户端的信息*/cli_t *pNode = (cli_t *)arg;/*打印给服务端显示*/printf("客户端(%s:%d)加入聊天室\n", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));/*如果客户端链表大于1就转发某某加入聊天室的信息,并通知其他客户端*/if (phead.size > 1){cli_t *p = &phead.cli;msg_t msg;bzero(&msg, 0);msg.cli = *pNode;msg.type = 0;sprintf(msg.message, "客户端(%s:%d)加入聊天室", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));while (p->next != NULL){p = p->next;/*除自己外不用发*/if (p->fd == pNode->fd)continue;send(p->fd, &msg, sizeof(msg), 0);}}/*死循环,接收和转发消息或者处理客户端消息*/while (1){msg_t msgs;bzero(&msgs, 0);int ret = recv(pNode->fd, &msgs, sizeof(msgs), 0);/* 这里对客户端链表上锁,以防止两个或多个客户端同时退出造成 *//* 服务端崩溃或者转发消息失败 */pthread_mutex_lock(&phead.lock);if (ret > 0){/*先对发来quit消息的客户端进行处理*/if (!strcmp("quit", msgs.message)){phead.size--;msgs.type = 0;printf("客户端(%s:%d)退出聊天室\n", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));sprintf(msgs.message, "客户端(%s:%d)退出聊天室", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));cli_t *p = &phead.cli;while (p->next != NULL){p = p->next;if (p->fd == pNode->fd){continue;}send(p->fd, &msgs, sizeof(msgs), 0);}/* 待消息转发结束之后解锁链表 */pthread_mutex_unlock(&phead.lock);/* 从链表中删除此客户端信息 */delNodeFromList(&phead, pNode->fd);/* 关闭此套接字 */close(pNode->fd);/* 线程结束 */pthread_cancel(pthread_self());bzero(&msgs, 0);break;}else // 如果不是quit就正常转发此客户端发来的消息{msgs.cli = *pNode;cli_t *p = &phead.cli;/* type = 1就是为了能够客户端那边打印客户端消息 */msgs.type = 1; while (p->next != NULL){p = p->next;if (p->fd == pNode->fd)continue;send(p->fd, &msgs, sizeof(msgs), 0);}/* 正常发送消息结束后解锁链表 */pthread_mutex_unlock(&phead.lock);}}/* 这就是ret小于等于0的情况了,表示客户端意外退出 *//* 其处理方法和上边quit的处理相同,不过就是发送消息改为了 *//* 意外退出而已 */else{phead.size--;msgs.type = 0;printf("客户端(%s:%d)意外退出聊天室\n", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));sprintf(msgs.message, "客户端(%s:%d)意外退出聊天室", inet_ntoa(pNode->client.sin_addr), ntohs(pNode->client.sin_port));cli_t *p = &phead.cli;while (p->next != NULL){p = p->next;if (p->fd == pNode->fd){continue;}send(p->fd, &msgs, sizeof(msgs), 0);}pthread_mutex_unlock(&phead.lock);delNodeFromList(&phead, pNode->fd);pthread_cancel(pthread_self());close(pNode->fd);bzero(&msgs, 0);break;}}
}void *pthread_accept(void *arg)
{struct sockaddr_in clientSock;socklen_t clientLen = sizeof(struct sockaddr_in);printf("===========聊天室已创建===========\n");while (1){int acceptFd = accept(sockfd, (struct sockaddr *)&clientSock, &clientLen);/*pNode保存客户端信息,并将这个结构体传递给common的线程*//*common线程是把客户端加入到链表中之后再创建的用于接收和转发聊天*/cli_t *pNode = (cli_t *)malloc(sizeof(cli_t));pNode->client = clientSock;pNode->fd = acceptFd;pNode->next = NULL;/*采用头插法*/if (phead.size > 0){cli_t *p = phead.cli.next;phead.cli.next = pNode;pNode->next = p;}else if (phead.size == 0){phead.cli.next = pNode;}phead.size++;pthread_t commonPthread;/*创建聊天线程*/pthread_create(&commonPthread, NULL, pthread_common, (void *)pNode);pthread_detach(commonPthread);}
}
int main()
{sockfd = socket(AF_INET, SOCK_STREAM, 0);struct sockaddr_in serverSock;socklen_t serverLen = sizeof(struct sockaddr_in);serverSock.sin_addr.s_addr = htonl(INADDR_ANY);serverSock.sin_port = htons(PORT);serverSock.sin_family = AF_INET;bind(sockfd, (struct sockaddr *)&serverSock, serverLen);listen(sockfd, 0);pthread_t acceptPthread;pthread_create(&acceptPthread, NULL, pthread_accept, NULL);while (1);return 0;
}

客户端client.c文件

#include "../head.h"
int sockfd;
int quit = 0;
void *pthread_recv(void *arg)
{msg_t msg; // 接收消息的结构体,我会放到head.h文件中解释bzero(&msg, 0); // 初始化while (1) //循环{/* 阻塞接收函数,ret返回值用处来表明接收消息是否正常 *//* 同时也能知道其他客户端是否还在不在聊天室,比如下文的等于0 */int ret = recv(sockfd, &msg, sizeof(msg), 0); if (ret > 0){/* type变量是一个区分,1为正常消息,0为不正常,通常是有客户端主动正常退出 */if (msg.type == 1 && strlen(msg.message) > 0){/* 打印消息同时打印是哪个客户端打印的 */printf("客户端(%s:%d):%s\n", inet_ntoa(msg.cli.client.sin_addr), ntohs(msg.cli.client.sin_port), msg.message);}else{printf("%s\n", msg.message);}}/* 为0就是某某客户端断开连接 */else if (ret == 0){printf("%s\n", msg.message);}}
}void *pthread_send(void *arg)
{msg_t msg;bzero(&msg, 0);while (1){/* 读取输入 */scanf("%s", msg.message);/* 防止读取到一个回车就发送 */if (getchar() == '\n' && msg.message[0] != '\n'){/* 安全起先还是加了一个\0 */int len = strlen(msg.message);msg.message[len + 1] = '\0';send(sockfd, &msg, sizeof(msg), 0);/* 如果是quit的话令quit变量为true,这样主程序会结束 */if (!strcmp(msg.message, "quit")){quit = true;break;}memset(msg.message, 0, sizeof(msg.message));}}
}int main()
{sockfd = socket(AF_INET, SOCK_STREAM, 0);struct sockaddr_in serverSock;socklen_t serverLen = sizeof(struct sockaddr_in);serverSock.sin_addr.s_addr = htonl(INADDR_ANY);serverSock.sin_port = htons(PORT);serverSock.sin_family = AF_INET;connect(sockfd, (struct sockaddr *)&serverSock, serverLen);printf("=====欢迎加入聊天室=====\n");printf("  退出服务器请输入quit  \n");msg_t msg;bzero(&msg, 0);pthread_t pthreadSend;pthread_create(&pthreadSend, NULL, pthread_send, NULL);pthread_t pthreadRecv;pthread_create(&pthreadRecv, NULL, pthread_recv, NULL);while (!quit);return 0;
}

head.h头文件

#ifndef __HEAD_H__
#define __HEAD_H__
#include <stdio.h>
#include <string.h>
#include <stdlib.h>#include <sys/types.h>
#include <sys/socket.h>#include <unistd.h>
#include <arpa/inet.h>
#include <netinet/ip.h>
#include <netinet/ip.h>
#include <pthread.h>
// 使用htonl(INADDR_ANY)的话下边的IP用不到了
// 当然可以修改为自己的IP,那就把htonl改为inet_addr(IP)
#define IP "127.0.0.1"
#define PORT 5212typedef struct client
{struct sockaddr_in client;int fd;struct client *next;
}cli_t;typedef struct msg
{pthread_mutex_t lock; // 锁char message[512]; // 发送和接收数据int type; // 类型cli_t cli; // 客户端信息结构体int size; // 链表大小
}msg_t;#endif

Makefile文件编译两个可执行文件,cli和ser

all:gcc client.c -o cli -lpthreadgcc server.c -o ser -lpthread
clean:rm -rf cli ser

最后效果

正常加入聊天室和聊天的效果

主动退出和意外退出的效果

Linux C/TCP多线程网络聊天室相关推荐

  1. 基于TCP的网络聊天室实现(C语言)

    基于TCP的网络聊天室实现(C语言) 一.网络聊天室的功能 二.网络聊天室的结果展示 三.实现思路及流程 四.代码及说明 1.LinkList.h 2.LinkList.c 3.client.c 4. ...

  2. Linux C多人网络聊天室

    经过好几天的日夜奋斗,总算把这个聊天室给做出来了,虽然说不上多好,但也是这几天从早到晚劳动的成功,所以就写这篇博文来记录一下啦.别的不敢说,确保能用就是了,完整代码在最后哦~ 当然啦,如果有幸被转发, ...

  3. QT学习:基于TCP的网络聊天室程序

    TCP与UDP的差别如图: 一.TCP工作原理 如下图所示,TCP能够为应用程序提供可靠的通信连接,使一台计算机发出的字节流无差错 地送达网络上的其他计算机.因此,对可靠性要求高的数据通信系统往往使用 ...

  4. 基于TCP的网络聊天室

    综合分析: 基于TCP的聊天室,支持多个用户同时登陆服务器进行聊天.(相当于群) socket编程,做一对多的通信,必然要用到多线程,保证多个客户端(并行)登陆服务器时同时进行聊天. 项目要求: 利用 ...

  5. c++ socket 多线程 网络聊天室

    文章目录 1. 实现图解 2. 聊天室服务端:TCP_Server_Chat.cpp 3. 聊天室客户端:TCP_Client_Chat.cpp 1. 实现图解 2. 聊天室服务端:TCP_Serve ...

  6. 基于linux网络聊天室的设计,参考基于linux网络聊天室的设计.doc

    参考基于linux网络聊天室的设计 长沙理工大学<高级操作系统>课程设计报告学 院 计算机与通信工程 专 业 计算机科学与技术 班 级 学 号 学生姓名 指导教师 课程成绩 完成日期 课程 ...

  7. [NUAA]Python用UDP协议建立带有私聊功能的网络聊天室-建立聊天工具

    文章目录 前言 1.网络聊天室的基本架构是什么? 1.1 客户端和服务器的架构 1.2 通信协议的选择以及多线程通信 1.2.1 多线程通信 1.2.2 通信协议选择 1.3 前后端功能设计思路 1. ...

  8. linux点对点聊天室的实现与设计心得,基于Socket接口的Linux与Windows网络聊天室设计与实现...

    陈洁 孟晓景 摘要:为了实现Linux与Windows跨平台通信,及时共享信息,构建了一个适用于跨平台的网络聊天室通信程序.先搭建跨平台通信环境,然后使用Socket套接字网络编程接口实现通信.整个系 ...

  9. 利用多线程实现linux下C语言的聊天室程序:

    转载:http://www.360doc.com/content/16/0421/11/478627_552531090.shtml 利用多线程实现linux下C语言的聊天室程序: 客户端代码: th ...

最新文章

  1. GD32F207移植FreeRtos,程序运行vTaskStartScheduler后进入HardFault_Handler死循环解决方法
  2. C#中使用WebClient下载指定url的网络照片
  3. centos apt-get java_CentOS安装jdk的三种方法
  4. 如何用好埋点中的数据
  5. C#设计模式系列 8 ----Builder 生成器模式之--发工资了,带老婆到 岗顶百脑汇配置电脑...
  6. 数据结构之字符串反转
  7. jQuery Mobile中弹窗popup的data-*选项
  8. mac os x 安装mysql遇到 Access denied for user 'root'@'localhost' (using password: YES)的解决方法...
  9. BZOJ3343 洛谷2801:教主的魔法——题解
  10. Jquery 数组操作(转)
  11. ROMS简单应用——绘制区域温度图
  12. windows系统下,如何将dmg文件转化为cdr文件
  13. js将数组转换为带间隔符的字符串
  14. Java Idea设置运行内存大小(开发必备)
  15. ios iPhone的 自带输入法emoji编程
  16. 如何从 Github 中删除提交
  17. 【身份证识别】基于形态学实现二代身份证号码识别系统matlab源码含GUI
  18. 合作师专计算机培训,2016年西北师范大学校企合作计算机联合办学建设方案.pdf...
  19. JAVA多小区物业管理系统源码,Java物业系统源码
  20. php和吉他哪个难学,良心点评解析红棉和雅马哈的古典吉他有区别没有?哪个好?老手吐槽大实话...

热门文章

  1. 水很深的深度学习-卷积神经网络篇
  2. (附源码)SSM校园新闻发布系统JAVA计算机毕业设计项目
  3. 计算机中丢失crashrpt,crashrpt
  4. javascript判断是否为移动端
  5. c语言坐标海伦公式,C语言:用海伦公式求三角形面积 , C语言编程问题,利用海伦公式求三角形面积...
  6. 全世界最酷的软件!谷歌地球重磅更新
  7. 360手机助手 无法android 4.2手机软件移动sd卡,360手机助手为什么不能移至sd卡
  8. 计算机二级在线模拟考试免费,全国计算机等级考试模拟试题集(二级QBASIC).pdf...
  9. 淘宝直播回放如何下载
  10. 【AT91SAM9261EK】 编译运行Linux Kernel 2.6.39