基于UDP的网络聊天室

最新推荐文章于 2024-08-12 14:30:00 发布

@Groot

最新推荐文章于 2024-08-12 14:30:00 发布

阅读量39

点赞数

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

本文链接：https://blog.csdn.net/weixin_45566329/article/details/130815578

版权

服务器

//main.c
#include "fun.h"

int main(int argc,const char * argv[])
{
	int sfd = socket(AF_INET, SOCK_DGRAM,0);
	if(sfd < 0)
	{
		perror("socket");
		return -1;
	}

	//允许端口快速被重用，快速被复用
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0)
	{
		perror("setsockopt");
		return -1;
	}

	//填充服务器自身的地址信息结构体
	//真实的地址信息结构体根据地址族制定 AF_INET: man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET; //必须填充AF_INET
	sin.sin_port = htons(PORT); //端口号的网络字节序　1024~49151
	sin.sin_addr.s_addr = inet_addr(IP); //ifconfig查看本机IP

	//将本机IP地址和端口号绑定到指定的套接字文件描述符上
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		perror("bind");
		return -1;
	}

	struct sockaddr_in cli;
	socklen_t len = sizeof(cli);
	int res = 0;
	pthread_t tid;
	
	linklist l = LinklistCreateHead();
	struct cli_msg buf;
	struct sin_msg info;
	info.sfd = sfd;
	info.savesin = sin;

	if(pthread_create(&tid,NULL,del_sin_msg,&info) != 0)
	{
		fprintf(stderr,"pthread_create failed __%d__\n",__LINE__);
		return -1;
	}
	pthread_detach(tid);	
	
	while(1)
	{
		bzero(buf.name,sizeof(buf.name));
		bzero(buf.message,sizeof(buf.message));
		res = recvfrom(sfd,&buf,sizeof(buf),0,(struct sockaddr *)&cli,&len);
		if(res < 0)
		{
			perror("recvfrom");
			return -1;
		}
		
		switch(buf.type)
		{
			case 'L':
				do_login(sfd,l,cli,buf);
				break;
			case 'C':
				do_chat(sfd,l,buf);
				break;
			case 'Q':
				do_quit(sfd,l,buf,cli);
				break;
			default:
				break;
		}
	}
	if(close(sfd) < 0)
	{
		perror("close");
		return -1;
	}
}
void *del_sin_msg(void *arg) //void *arg = &info
{
	struct cli_msg buf;
	int sfd = ((struct sin_msg *)arg)->sfd;
	struct sockaddr_in sin = ((struct sin_msg*)arg)->savesin;
	buf.type = 'C';
	bzero(buf.name,sizeof(buf.name));
	strcpy(buf.name,"系统公告");
	bzero(buf.message,sizeof(buf.message));
	fgets(buf.message,sizeof(buf.message),stdin);
	buf.message[strlen(buf.message)-1] = 0;
	if(sendto(sfd,&buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		perror("sendto");
		return NULL;
	}
}

//fun.c
#include "fun.h"
//创建头结点
linklist LinklistCreateHead()
{
	linklist l = (linklist)malloc(sizeof(Linklist));
	if(l==NULL)
		return NULL;
	l->len = 0;//链表长度为0
	l->next = NULL;
	return l;
}
//创建普通节点
linklist LinklistCreateNode()
{
	linklist p = (linklist)malloc(sizeof(Linklist));
	if(p==NULL)
		return NULL;
	p->next = NULL;
	return p;
}
//头插
void Insert(linklist l,struct sockaddr_in cli)
{
	linklist p =LinklistCreateNode();
	if(p==NULL)
		return ;
	p->data = cli;
	p->next = l->next;
	l->next = p;
	l->len++;
}

int deletedata(linklist l,struct sockaddr_in cli)
{
	if(NULL == l)
		return -1;
	linklist p = l;
	int len = l->len;
	for(int i = 0;i<=len;i++)
	{
		p = p->next;
		if(p->data.sin_port == cli.sin_port && p->data.sin_addr.s_addr == cli.sin_addr.s_addr)
		{
			deletepos(i,l);
			break;
		}
	}
	return 0;
}

int deletepos(int pos,linklist l)
{
	if(NULL == l)
		return -1;
	linklist p = l;
	for(int i = 0;i<pos-1;i++)
	{
		p = p->next;
	}
	linklist q = p->next;
	p->next = q->next;
	free(q);
	l->len--;
}

int do_login(int sfd,linklist l,struct sockaddr_in cli,struct cli_msg buf)
{
	char sendbuf[128] = "";
	strcpy(sendbuf,buf.name);
	strcat(sendbuf,"已上线");
	printf("%s\n",sendbuf);

	if(l->next)
	{
		linklist p =l->next;
		while(p)
		{
			if(sendto(sfd,sendbuf,sizeof(sendbuf),0,(struct sockaddr*)&(p->data),sizeof(p->data)) < 0)
			{
				perror("sendto");
				return -1;
			}
		}
		p = p->next;
	}
	Insert(l,cli);
	return 0;
}

int do_chat(int sfd,linklist l,struct cli_msg buf)
{
	char coze[128] = "";
	strcpy(coze,buf.name);
	strcat(coze,":");
	strcat(coze,buf.message);
	linklist p = l->next;
	while(p)
	{
		if(sendto(sfd,coze,sizeof(coze),0,(struct sockaddr *)&(p->data),sizeof(p->data)) < 0)
		{
			perror("sendto");
			return -1;
		}
		p = p->next;
	}
}

int do_quit(int sfd,linklist l,struct cli_msg buf,struct sockaddr_in cli)
{
	char sendbuf[128] = "";
	strcpy(sendbuf,buf.name);
	strcat(sendbuf,"已下线");
	printf("%s\n",sendbuf);
	linklist p = l->next;
	while(p)
	{
		if(sendto(sfd,sendbuf,sizeof(sendbuf),0,(struct sockaddr *)&(p->data),sizeof(p->data)) < 0)
		{
			perror("sendto");
			return -1;
		}
		p = p->next;
	}
	deletedata(l,cli);
	return 0;
}

//fun.h
#ifndef __FUN_H__
#define __FUN_H__

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

#define IP "192.168.10.10"
#define PORT 8888

struct sin_msg
{
	int sfd;
	struct sockaddr_in savesin;
};

struct cli_msg
{
	char type;
	char name[20];
	char message[128];
};

typedef struct Node
{
	union
	{
		int len;
		struct sockaddr_in data;
	};
	struct Node * next;
}*linklist,Linklist;

linklist LinklistCreateHead();
linklist LinklistCreateNode();
void Insert(linklist l,struct sockaddr_in cli);
int deletedata(linklist l,struct sockaddr_in cli);
int deletepos(int pos,linklist l);
int do_login(int sfd,linklist l,struct sockaddr_in cli,struct cli_msg buf);
int do_chat(int sfd,linklist l,struct cli_msg buf);
int do_quit(int sfd,linklist l,struct cli_msg buf,struct sockaddr_in cli);
void *del_sin_msg(void *arg);
#endif

客户端

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

#define IP "192.168.10.10"
#define PORT 8888

struct cli_msg
{
	char type;
	char name[20];
	char message[128];
};

struct sin_msg
{
	int cfd;
	struct sockaddr_in savesin;
};

void *del_sin_msg(void *arg);
int main(int argc, const char *argv[])
{
	int cfd = socket(AF_INET, SOCK_DGRAM,0);
	if(cfd < 0)
	{
		perror("socket");
		return -1;
	}

	//允许端口快速被重用，快速被复用
	int reuse = 1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0)
	{
		perror("setsockopt");
		return -1;
	}

	//填充服务器自身的地址信息结构体
	//真实的地址信息结构体根据地址族制定 AF_INET: man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET; //必须填充AF_INET
	sin.sin_port = htons(PORT); //端口号的网络字节序　1024~49151
	sin.sin_addr.s_addr = inet_addr(IP); //ifconfig查看本机IP
	socklen_t len_sin = sizeof(sin);

	struct sockaddr_in cli;
	socklen_t len = sizeof(cli);

	struct cli_msg buf;
	pthread_t tid;

	struct sin_msg info;
	char id[20] = "";;
	info.cfd = cfd;
	info.savesin = sin;
	char data[128] = "";

	printf("请输入登录姓名>");
	fgets(id,sizeof(id),stdin);
	id[strlen(id)-1] = 0;
	buf.type = 'L';
	strcpy(buf.name,id);
	if(sendto(cfd,&buf,sizeof(buf),0,(struct sockaddr *)&sin,sizeof(sin)) < 0)
	{
		perror("sendto");
		return -1;
	}

	pthread_detach(tid);
	if(pthread_create(&tid,NULL,del_sin_msg,&info) != 0)
	{
		fprintf(stderr,"pthread_create failed __%d__\n",__LINE__);
		return -1;
	}
	ssize_t res = 0;
	while(1)
	{
		//ssize_t res = recvfrom(cfd,data,sizeof(data),0,(struct sockaddr *)&cli,&len);
		res = recv(cfd,data,sizeof(data),0);
		if(res < 0)
		{
			perror("recvfrom");
			return -1;
		}
		printf("%s\n",data);

	}
	return 0;
}

void *del_sin_msg(void *arg)
{
	struct cli_msg buf;
	int cfd = ((struct sin_msg*)arg)->cfd;
	struct sockaddr_in sin = ((struct sin_msg*)arg)->savesin;

	int flag = 1;
	while(1)
	{
		bzero(buf.message,sizeof(buf.message));
		buf.type = 'C';
		fgets(buf.message,sizeof(buf.message),stdin);
		bzero(buf.message,sizeof(buf.message));
		flag = strcmp(buf.message,"quit");
		if(0 == flag)
		{
			buf.type = 'Q';
		}
		if(sendto(cfd,&buf,sizeof(buf),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
		{
			perror("sendto");
			return NULL;
		}
		if(0 == flag)
			break;
	}
	return NULL;
}