基于UDP的网络聊天室

news/2024/5/11 16:57:45

1、如果有用户登录,其他用户可以收到这个人的登录信息

2、如果有人发送信息,其他用户可以收到这个人的群聊信息

3、如果有人下线,其他用户可以收到这个人的下线信息

4、服务器可以发送系统信息
服务器

#include<myhead.h>typedef struct Node
{char usrName[20];//用户名struct sockaddr_in cin;struct Node *next; //指针
}__link;//收发信息结构体
typedef struct msgtyp
{int type;char name[20];char text[128];
}msg;
int do_lodin(int sfd,__link* head,msg recv_msg,struct sockaddr_in cin)
{printf("%s[%s:%d]登陆成功\n",recv_msg.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));sprintf(recv_msg.text,"----%s登陆成功----",recv_msg.name);while(head->next!=NULL){head=head->next;if(sendto(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&(head->cin),sizeof(head->cin))==-1){perror("sendto");return -1;}}//添加客户到链表__link *new=(__link*)malloc(sizeof(__link));new->cin=cin;new->next=NULL;head->next=new;return 0;
}
int do_chat(int sfd,__link* head, msg recv_msg, struct sockaddr_in cin)
{printf("%s[%s:%d]chat成功\n",recv_msg.name,inet_ntoa(cin.sin_addr),ntohs(cin.sin_port));char buf[258]="";sprintf(buf,"%s:%s",recv_msg.name,recv_msg.text);strcpy(recv_msg.text,buf);//给其他客户端发送消息while(head->next!=NULL){head=head->next;head->cin=cin;if(memcmp(&cin, &head->cin, sizeof(cin)) != 0){if(sendto(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&(head->cin),sizeof(head->cin))==-1){perror("sendto");return -1;}}}return 0;
}
int do_quit(int sfd,__link* head, msg recv_msg, struct sockaddr_in cin)
{sprintf(recv_msg.text, "-----%s 已下线-----\n", recv_msg.name);while(head->next != NULL){if(memcmp(&cin, &head->next->cin, sizeof(cin)) == 0){//删除__link* temp = head->next;head->next = temp->next;free(temp);}else{head = head->next;if(sendto(sfd, &recv_msg, sizeof(recv_msg),0, (struct sockaddr*)&(head->cin), sizeof(head->cin))==-1){perror("sendto");return -1;}}}return 0;
}int main(int argc, char *argv[])
{if(argc<3){printf("请输入ip和端口号\n");return -1;}int sfd=socket(AF_INET,SOCK_DGRAM,0);if(sfd==-1){perror("socket");return -1;}int reuse=-1;if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1){perror("setsockopt");return -1;}struct sockaddr_in sin;sin.sin_family=AF_INET;sin.sin_port=htons(atoi(argv[2]));sin.sin_addr.s_addr=inet_addr(argv[1]);if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1){perror("bind");return -1;}//printf("bind success\n");int res=fork();//父进程接收消息if(res>0){__link *head=(__link*)malloc(sizeof(__link));head->next=NULL;msg recv_msg;int recv_len=0;struct sockaddr_in cin;socklen_t len=sizeof(cin);while(1){recv_len=recvfrom(sfd,&recv_msg,sizeof(recv_msg),0,(struct sockaddr*)&cin,&len);if(recv_len<0){perror("recvfrom");return -1;}int type=ntohl(recv_msg.type);switch(type){case 1:{do_lodin(sfd,head,recv_msg,cin);}break;case 2:{do_chat(sfd,head,recv_msg,cin);}break;case 3:{do_quit(sfd,head,recv_msg,cin);}break;}}}else if(res==0){//子进程,发送系统消息msg sys_msg={htonl(1),"**system**"};while(1){bzero(sys_msg.text,1024);fgets(sys_msg.text,1024,stdin);sys_msg.text[strlen(sys_msg.text)-1]=0;if(sendto(sfd,&sys_msg,sizeof(sys_msg),0,(struct sockaddr*)&sin,sizeof(sin))==-1){perror("sendto");return -1;}}printf("系统消息发送成功\n");}close(sfd);return 0;
}

客户端代码

#include<myhead.h>
#define LOGIN 	1
#define CHAT 	2
#define QUIT 	3
#define ERR_LOG(msg) do{\printf("%d %s %s \n", __LINE__, __func__, __FILE__);\perror(msg);\
}while(0)//收发信息的结构体
typedef struct
{int type;char name[20];char text[128] ;}MSG;typedef void (*sighandler_t)(int);int do_chat(int sfd, MSG msg, struct sockaddr_in sin)
{while(1){//从终端获取聊天信息bzero(msg.text, 128);fgets(msg.text, 128,stdin);msg.text[strlen(msg.text)-1] = 0;if(strncasecmp(msg.text, "QUIT",4) == 0){msg.type = htonl(QUIT);}else{msg.type = htonl(CHAT);}//发送 if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0){ERR_LOG("sendto");return -1;}//如果是退出信息,则客户端要退出if(msg.type == htonl(QUIT)){exit(0); 		//退出子进程;}}
}int do_recv(int sfd)
{MSG rcv_msg ;while(1){if(recvfrom(sfd, &rcv_msg, sizeof(rcv_msg), 0, NULL, NULL) < 0){perror("recvfrom");return -1;}printf("%s\n",rcv_msg.text);}
}void handler(int sig)
{//回收子进程资源并退出while(waitpid(-1,NULL, WNOHANG)>0);exit(0);
}int main(int argc, char *argv[])
{if(argc < 3){   printf("请输入 ip 和端口号\n");return -1; } sighandler_t s = signal(SIGCHLD, handler);if(s == SIG_ERR){ERR_LOG("signal");return -1;}int sfd = socket(AF_INET, SOCK_DGRAM, 0);if(sfd < 0){perror("socket");return -1;}//2.填充服务器ip和端口号	struct sockaddr_in sin;sin.sin_family=AF_INET;sin.sin_port=htons(atoi(argv[2]));sin.sin_addr.s_addr=inet_addr(argv[1]);socklen_t addrlen = sizeof(sin);//登录协议MSG msg ;msg.type = htonl(1);printf("请输入姓名>>>");fgets(msg.name, 20, stdin);msg.name[strlen(msg.name)-1] = 0;if(sendto(sfd, &msg, sizeof(msg), 0, (void*)&sin, sizeof(sin)) < 0){ERR_LOG("sendto");	return -1;}pid_t pid = fork();if(pid > 0){//父进程获取信息do_recv(sfd);}else if(0 == pid){//子进程发送信息do_chat(sfd, msg, sin);}//4.关闭套接字close(sfd);return 0;
}

效果

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.tangninghui.cn.cn/item-12148.htm

如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈,一经查实,立即删除!

相关文章

Linux 进程信号:产生信号

目录 一、通过终端按键产生信号 1、signal()函数 2、核心转储 3、ulmit命令 二、调用系统函数向进程发信号 1、kill()函数 2、raise()函数 3、abort()函数 三、发送信号的过程 读端关闭、写端继续写入的情况 如何理解软件条件给进程发送信号: 四、软件条件产生信…

几个常用的控件(2)

目录 一、单选按钮Radiobutton和RadioButtonList 1、Radiobutton控件 &#xff08;1&#xff09;button控制方式 &#xff08;2&#xff09;Radiobutton控制方式 2、RadiobuttonList控件 二、列表框ListBox和下拉列表DropdownList 1、ListBox 2、DropdownList 三、面板…

第八章贪心算法——理论基础,分发饼干题目

目录 概念 什么时候使用 题目举例 分发饼干 力扣题号&#xff1b;455. 分发饼干 - 力扣&#xff08;LeetCode&#xff09; 题目描述 示例 1: 示例 2: 解法一&#xff1a;排序暴力 解法二&#xff1a;贪心 思路 代码实现 总结 概念 贪心算法是一种在每一步选择中都采…

什么算法可以进行小语种的OCR?

对于小语种的OCR识别&#xff0c;可以采用以下算法和技术&#xff1a; 1. 迁移学习&#xff08;Transfer Learning&#xff09;&#xff1a;使用在大语种上预训练好的OCR模型&#xff0c;并通过迁移学习的方式对小语种进行微调。这样可以利用大语种上已有的丰富数据和知识&…

HAL STM32G4 +ADC手动触发采集+各种滤波算法实现

HAL STM32G4 ADC手动触发采集各种滤波算法实现 &#x1f4cd;相关篇《HAL STM32G4 TIM1 3路PWM互补输出VOFA波形演示》 ✨本篇内容也是继欧拉电子相关无刷电机驱动控制学习的相关基础内容。仅作为个人笔记记录使用。 &#x1f4cd;感谢网友提供的相关内容《基于STM32的ADC采样及…

刚刚,百度和苹果宣布联名

百度 Apple 就在刚刚&#xff0c;财联社报道&#xff0c;百度将为苹果今年发布的 iPhone16、Mac 系统和 iOS18 提供 AI 功能。 苹果曾与阿里以及另外一家国产大模型公司进行过洽谈&#xff0c;最后确定由百度提供这项服务&#xff0c;苹果预计采取 API 接口的方式计费。 苹果将…