udp实现多人聊天室

最新推荐文章于 2023-06-26 16:03:23 发布

vivisiLi

最新推荐文章于 2023-06-26 16:03:23 发布

阅读量152

点赞数

文章标签： udp 链表网络协议

本文链接：https://blog.csdn.net/lsw2604190117/article/details/130879046

版权

该代码实现了一个UDP服务器，处理客户端的登录（L）、群聊（C）和退出（Q）请求。服务器使用链表存储在线用户，接收到消息后转发给其他在线用户。客户端则发送登录请求，接收并显示服务器的群聊消息，可以输入文本参与群聊或选择退出。

摘要由CSDN通过智能技术生成

服务器

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>                                               
#include <stdlib.h>


#define ERR_MSG(msg) do{\
    fprintf(stderr, "line:%d ", __LINE__);\
    perror(msg);\
}while(0)

#define PORT 6666               //1024~49151
#define IP  "192.168.136.128"    //IP地址，本机IP ifconfig 

typedef struct cli_msg
{
	char type; 		//L C Q
	char name[20];
	char text[128];
}Msg;

//存储登录消息
typedef struct node{
	union{
		struct sockaddr_in logcin;//登录的客户端
		int len;
	};
	struct node*next;
}seqlist;

int do_quit(struct sockaddr_in cin,Msg rcvbuf,seqlist*L,int sfd);
int do_chat(struct sockaddr_in cin,Msg rcvbuf,seqlist*L,int sfd );
int do_login(struct sockaddr_in cin,Msg rcvbuf,seqlist*L,int sfd);
void showlink(seqlist*L);

//创建链表
seqlist *Creatseqlist()
{
	seqlist*L=(seqlist*)malloc(sizeof(seqlist));
	if(NULL==L){
		printf("创建链表失败\n");
		return NULL;
	}
	L->len=0;
	L->next=NULL;
	return L;
}
//创建普通节点
seqlist*CommLinklistCreatNode()
{
	seqlist*L=(seqlist*)malloc(sizeof(seqlist));
	if(NULL==L){
		printf("创建链表失败\n");
		return NULL;
	}
	L->next=NULL;
	return L;
}
int main(int argc, const char *argv[])
{
	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success  sfd=%d __%d__\n", sfd, __LINE__);


	//允许端口快速被复用
	

	//填充服务器自身的地址信息结构体 AF_INET : man 7 IP
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 	//必须填AF_INET;
	sin.sin_port 		= htons(PORT); 	//服务器绑定的端口，网络字节序
	sin.sin_addr.s_addr = inet_addr(IP); 	//服务器绑定的IP，本机IP ifconfig

	//绑定--->必须绑定
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n", __LINE__);

	char buf[128] = "";
	ssize_t res = 0;

	struct sockaddr_in cin; 	//存储数据包从谁哪里过来
	socklen_t addrlen = sizeof(cin);
	//创建链表
	seqlist *L=Creatseqlist();
	if(L==NULL){
		printf("创建链表失败\n");
		return -1;
	}
	pid_t cpid = fork();
	if(cpid > 0)
	{
		//接收数据
		Msg rcvbuf;
		while(1)
		{
			res = recvfrom(sfd, &rcvbuf, sizeof(rcvbuf), 0, (struct sockaddr*)&cin, &addrlen);
			if(res < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
	//		printf("type = %c\tname=%s\n", rcvbuf.type,rcvbuf.name);
			switch(rcvbuf.type)
			{
			case 'L': 	
				do_login(cin,rcvbuf,L,sfd);
				break;
			case 'C':
				do_chat(cin,rcvbuf,L,sfd);
				break;
			case 'Q':
				do_quit(cin,rcvbuf,L,sfd);
				break;
			default:
				printf("协议 %c 错误 __%s__ __%d__\n", rcvbuf.type, __FILE__, __LINE__);
			}
		}
	}
	else if(0 == cpid)
	{
		Msg sndbuf;
		sndbuf.type = 'C'; 
		strcpy(sndbuf.name,"server");
		while(1){
			bzero(sndbuf.text, sizeof(sndbuf.text));
			fgets(sndbuf.text, sizeof(sndbuf.text), stdin);
			sndbuf.text[strlen(sndbuf.text)-1] = '\0';
			if(sendto(sfd, &sndbuf, sizeof(sndbuf), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
			{
				ERR_MSG("sendto"); 
				return -1;
			}
		}
	}
	else 
	{
		ERR_MSG("fork");
		return -1;
	}
	//关闭所有文件描述符
	close(sfd);
	return 0;
}
//链表遍历
void showlink(seqlist*L)
{
	seqlist*p=L;
	//printf("%d\n",L->len);
	for(int i=0;i<L->len;i++){
		p=p->next;
	//	printf("%d\n",p->logcin.sin_port);
	}
}
int do_login(struct sockaddr_in cin,Msg rcvbuf,seqlist*L,int sfd )
{
	//转发登录信息
	seqlist*p=L;
	printf("%s上线...\n",rcvbuf.name);   
	while(p->next!=NULL){
		p=p->next;
		if(sendto(sfd,&rcvbuf, sizeof(rcvbuf), 0, (struct sockaddr*)&(p->logcin), sizeof(p->logcin)) < 0)   
		{
			ERR_MSG("sendto");
			return -1;
		}
	}
	//将当前登录的用户的地址信息结构体存储起来
	seqlist *q=CommLinklistCreatNode();
	if(NULL==q){
		printf("创建普通节点失败\n");
		return -1;
	}
	q->logcin=cin;
	//插入链表头插
	q->next=L->next;
	L->next=q;
	L->len++;
	//遍历链表
	showlink(L);
	return 0;
}

int do_chat(struct sockaddr_in cin,Msg rcvbuf,seqlist*L,int sfd )
{
	//转发群聊信息
	seqlist*p=L;
	printf("%s:%s\n",rcvbuf.name,rcvbuf.text);
	while(p->next!=NULL){
		p=p->next;
	//跳过自己
		if(p->logcin.sin_port==cin.sin_port){
			continue;
		}
		if(sendto(sfd,&rcvbuf, sizeof(rcvbuf), 0, (struct sockaddr*)&(p->logcin), sizeof(p->logcin)) < 0)   
		{
			ERR_MSG("sendto");
			return -1;
		}
	//	printf("发送信息给%s成功\n",rcvbuf.name);
	}
	return 0;
}

int do_quit(struct sockaddr_in cin,Msg rcvbuf,seqlist*L,int sfd)
{
	//将当前下线客户端的地址信息结构体从存储位置删除
	seqlist*q=L;
	while(q->next!=NULL){
		seqlist*ago=q;
		q=q->next;
		if(q->logcin.sin_port==cin.sin_port){
			ago->next=q->next;
			free(q);
		}
	}
	//转发下线信息
	seqlist*p=L;
	printf("%s退出群聊\n",rcvbuf.name);
	while(p->next!=NULL){
		p=p->next;
		if(sendto(sfd,&rcvbuf, sizeof(rcvbuf), 0, (struct sockaddr*)&(p->logcin), sizeof(p->logcin)) < 0)   
		{
			ERR_MSG("sendto");
			return -1;
		}
	}
	return 0;
}

客户端

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>                                               
#include <signal.h>

#define ERR_MSG(msg) do{\
    fprintf(stderr, "line:%d ", __LINE__);\
    perror(msg);\
}while(0)

#define PORT 6666               //1024~49151
#define IP  "192.168.136.128"    //IP地址，本机IP ifconfig 

struct cli_msg
{
    char type;      // L登录C群聊 Q下线
    char name[20];
    char text[128];
};


int main(int argc, const char *argv[])
{
	//创建报式套接字	

	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success  sfd=%d __%d__\n", sfd, __LINE__);

	//绑定--->非必须绑定
	//如果不绑定，操作系统会自动绑定本机IP，从49152~65535范围内随机一个未被使用的端口号
	
	//填充服务器的地址信息结构体
	//供给下面的sendto使用，
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET; 	//必须填AF_INET;
	sin.sin_port 		= htons(PORT); 	//服务器绑定的端口，网络字节序
	sin.sin_addr.s_addr = inet_addr(IP); 	//服务器绑定的IP，本机IP ifconfig 

	printf("请输入名字>>>");
	char name[20] = "";
	scanf("%s", name);
	getchar();

	struct cli_msg sndbuf;
	sndbuf.type = 'L';
	strcpy(sndbuf.name, name);
	//发送登录请求
	//想要发送给谁，就填谁的地址信息结构体
	if(sendto(sfd, &sndbuf, sizeof(sndbuf), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}
	printf("发送登录信息成功\n");
	ssize_t res = 0;
	struct sockaddr_in rcvAddr; 	//存储获取到的数据包是从谁那里来的
	socklen_t addrlen = sizeof(rcvAddr);
	pid_t pid=fork();
	if(pid==0){
		//接收数据系统消息和登录退出的消息
		while(1){
			res = recvfrom(sfd, &sndbuf, sizeof(sndbuf), 0, (struct sockaddr*)&rcvAddr, &addrlen);
			if(res < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			//接收登录消息
			switch(sndbuf.type)
			{
			case 'L':
				printf("%s上线...\n",sndbuf.name);
				break;
			case 'C':
				//接收群聊消息
				printf("%s:%s\n",sndbuf.name,sndbuf.text);
				break;
			case 'Q':
				printf("%s退出群聊\n",sndbuf.name);
				break;
			default:
				printf("协议 %c 错误 __%s__ __%d__\n", sndbuf.type, __FILE__, __LINE__);
				break;
			}
		}
	}
	else if(pid>0){
		//发送聊天数据
		while(1){
			sndbuf.type = 'C';
			bzero(sndbuf.text, sizeof(sndbuf.text));
			fgets(sndbuf.text, sizeof(sndbuf.text), stdin);
			sndbuf.text[strlen(sndbuf.text)-1] = '\0';
			//下线
		
			if(strcmp(sndbuf.text,"quit")==0){
				sndbuf.type = 'Q';
				if(sendto(sfd, &sndbuf, sizeof(sndbuf), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
					return -1;
				}
				break;
			}					
			if(sendto(sfd, &sndbuf, sizeof(sndbuf), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
			{
				ERR_MSG("sendto");
				return -1;
			}
		}
	}
	else{
		ERR_MSG("fork");
		return -1;
	}
	close(sfd);
	return 0;
}