基于UDP的网络聊天室网络编程0811作业-洪庆乐

已于 2022-08-12 11:22:36 修改
阅读量192 收藏
点赞数 1
文章标签： 网络 链表 数据结构
于 2022-08-11 21:47:06 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_57325286/article/details/126293786
版权

项目：基于UDP的网络聊天室

功能：1.服务器日志系统，且可以查看（实现）

           2.有用户，其他用户收到这个人登入信息（实现）

           3.群发消息（实现）

           4.如果有用户下线，其他用户收到下线消息（实现）

           5.服务器发送系统消息 （实现）

           6.踢出某个客户端（还没实现）

           7.禁言客户端（还未实现

 服务器：

#include "test.h"

int main(int argc, const char *argv[])
{
	//打开日志文件
	FILE *fw = fopen("./logfile","w+");
	if(fw == NULL)
	{
		ERR_MSG("fopen");
		return -1;
	}
	printf("服务器日志文件初始化成功\n");

	if(argc < 3)
	{
		fprintf(stderr, "请输入IP port\n");
		return -1;
	}
	//将获取到的端口号字符串，转换成整形
	int port = atoi(argv[2]);
	if(port < 1024 || port > 49151)
	{
		fprintf(stderr, "port %d input error!! 1024~49151\n", port);
		return -1;
	}

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}


	//填充服务器的IP地址以及端口号
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(port);
	sin.sin_addr.s_addr = inet_addr(argv[1]);


	//绑定服务器的地址信息结构体
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)        
	{
		ERR_MSG("bind");
		return -1;
	}


	//客户端网络结构体定义初始化
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);  

	//创建父子进程
	pid_t pid = fork();
	if(pid > 0)         //父进程负责服务器发送给客户端系统消息
	{
		printf("服务器功能模块初始化中\n");
		sleep(1);
		khd_msg msg;
		int num;
		while(1)
		{
			gnmenu();
			printf("请选择功能:");
			scanf("%d",&num);
			getchar();
			if(num == 1)
			{
				printf("请输入要群发的内容:");
				fgets(msg.text,128,stdin);               
				msg.text[strlen(msg.text)-1] = 0;
				msg.type = 2;
				strcpy(msg.username,"server");
				if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
					return -1;
				}
			}
			else if(num == 5)   //查看日志文件
			{
				FILE* fr = fopen("./logfile", "r");
				fseek(fr,0,SEEK_SET);
				if(NULL == fr)
				{   
					perror("fopen");
					return -1;
				}
				char c[10];
				while(1)
				{   
					//对于字符串的操作，请养成习惯在使用前或者使用后，清空字符串
					bzero(c, sizeof(c));    //清空字符串
					if(fread(c, sizeof(c)-1, 1, fr) == 0)
						break;
					printf("%s", c);
				}
				printf("\n");

				//关闭文件
				fclose(fr);
			}
		}
	}
	else if(pid == 0)  //子进程负责接收处理客户端发来的消息
	{
		//创建存储客户端信息的链表
		linkList* M = (linkList*)malloc(sizeof(linkList));
		if(NULL == M)
		{
			printf("创建失败\n");
			return -1;
		}
		//初始化
		M->next = NULL;
		printf("客户端信息链表创建成功\n");

		khd_msg msg;
		while(1)
		{
			bzero(&msg,sizeof(msg));

			//接受客户端信息
			if(recvfrom(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&cin,&addrlen) < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}

			if(msg.type == 1)
			{
				list_show(M,msg,sfd);
				list_intsrt_head(M,cin,msg,fw);		
			}
			else if(msg.type == 2)
			{
				list_msg(M,msg,sfd,cin,fw);
			}
			else if(msg.type == 3)
			{
				list_del(M,msg,sfd,cin,fw);
			}

		}
		
	}

	return 0;
}

客户端 ：

#include "test.h"

int main(int argc, const char *argv[])
{
	//打开日志文件
	FILE *fw = fopen("./logfile","w+");
	if(fw == NULL)
	{
		ERR_MSG("fopen");
		return -1;
	}
	printf("服务器日志文件初始化成功\n");

	if(argc < 3)
	{
		fprintf(stderr, "请输入IP port\n");
		return -1;
	}
	//将获取到的端口号字符串，转换成整形
	int port = atoi(argv[2]);
	if(port < 1024 || port > 49151)
	{
		fprintf(stderr, "port %d input error!! 1024~49151\n", port);
		return -1;
	}

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}


	//填充服务器的IP地址以及端口号
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(port);
	sin.sin_addr.s_addr = inet_addr(argv[1]);


	//绑定服务器的地址信息结构体
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)        
	{
		ERR_MSG("bind");
		return -1;
	}


	//客户端网络结构体定义初始化
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);  

	//创建父子进程
	pid_t pid = fork();
	if(pid > 0)         //父进程负责服务器发送给客户端系统消息
	{
		printf("服务器功能模块初始化中\n");
		sleep(1);
		khd_msg msg;
		int num;
		while(1)
		{
			gnmenu();
			printf("请选择功能:");
			scanf("%d",&num);
			getchar();
			if(num == 1)
			{
				printf("请输入要群发的内容:");
				fgets(msg.text,128,stdin);               
				msg.text[strlen(msg.text)-1] = 0;
				msg.type = 2;
				strcpy(msg.username,"server");
				if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&sin,sizeof(sin)) < 0)
				{
					ERR_MSG("sendto");
					return -1;
				}
			}
			else if(num == 5)   //查看日志文件
			{
				FILE* fr = fopen("./logfile", "r");
				fseek(fr,0,SEEK_SET);
				if(NULL == fr)
				{   
					perror("fopen");
					return -1;
				}
				char c[10];
				while(1)
				{   
					//对于字符串的操作，请养成习惯在使用前或者使用后，清空字符串
					bzero(c, sizeof(c));    //清空字符串
					if(fread(c, sizeof(c)-1, 1, fr) == 0)
						break;
					printf("%s", c);
				}
				printf("\n");

				//关闭文件
				fclose(fr);
			}
		}
	}
	else if(pid == 0)  //子进程负责接收处理客户端发来的消息
	{
		//创建存储客户端信息的链表
		linkList* M = (linkList*)malloc(sizeof(linkList));
		if(NULL == M)
		{
			printf("创建失败\n");
			return -1;
		}
		//初始化
		M->next = NULL;
		printf("客户端信息链表创建成功\n");

		khd_msg msg;
		while(1)
		{
			bzero(&msg,sizeof(msg));

			//接受客户端信息
			if(recvfrom(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&cin,&addrlen) < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}

			if(msg.type == 1)
			{
				list_show(M,msg,sfd);
				list_intsrt_head(M,cin,msg,fw);		
			}
			else if(msg.type == 2)
			{
				list_msg(M,msg,sfd,cin,fw);
			}
			else if(msg.type == 3)
			{
				list_del(M,msg,sfd,cin,fw);
			}

		}
		
	}

	return 0;
}
ubuntu@ubuntu:test$ 
ubuntu@ubuntu:test$ cat 4_udptClitext.c 
#include "test.h"


int main(int argc, const char *argv[])
{
	if(argc < 3)
	{
		fprintf(stderr, "请输入IP port\n");
		return -1;
	}
	//将获取到的端口号字符串，转换成整形
	int port = atoi(argv[2]);
	if(port < 1024 || port > 49151)
	{
		fprintf(stderr, "port %d input error!! 1024~49151\n", port);
		return -1;
	}

	//创建报式套接字
	int sfd = socket(AF_INET, SOCK_DGRAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//绑定客户端自身的地址信息结构体 ---> 非必须绑定 
	
	//填充服务器的IP地址以及端口号 -->因为客户端要主动发送数据包给服务器
	struct sockaddr_in sin;
	sin.sin_family 		= AF_INET;
	sin.sin_port 		= htons(port);
	sin.sin_addr.s_addr = inet_addr(argv[1]);


	//connect函数
	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("连接成功\n");


	//输入用户名
	khd_msg msg;
	printf("请输入用户名>>>");
	fgets(msg.username,15, stdin);
	msg.username[strlen(msg.username)-1] = 0;
	msg.type = 1;

	//向服务器发送加入包
	if(sendto(sfd,&msg,sizeof(msg),0,NULL,0) < 0)
	{
		ERR_MSG("sendto");
		return -1;
	}


	//创建父子进程
	pid_t pid = fork();
	if(pid > 0)
	{
		//父进程负责发送
		while(1)
		{
			//向包内写入信息
			fgets(msg.text,128,stdin);
			msg.text[strlen(msg.text)-1] = 0;

			//判断是否是退出包
			if(strcasecmp(msg.text,"quit"))
			{
				//如果不是退出包变成信息包
				msg.type = 2;
			}
			else
			{
				//如果是退出包将类型变成3
				msg.type = 3;
			}

			//向服务器发送我的消息包
			if(sendto(sfd,&msg,sizeof(msg),0,NULL,0) < 0)	
			{
				ERR_MSG("sendto");
				return -1;
			}

			//如果quit直接退出子进程
			if(strcasecmp(msg.text,"quit"))
			{
			}
			else
			{
				kill(pid,2);
				break;
			}
		}
		close(sfd);
	}
	else if(pid == 0)
	{
		while(1)
		{
			//接收
			bzero(&msg, sizeof(msg));
			//接收服务器发送过来的数据包
			if(recvfrom(sfd, &msg, sizeof(msg), 0, NULL, NULL) < 0)
			{
				ERR_MSG("recvfrom");
				return -1;
			}
			printf("%s：\t%s\n",msg.username,msg.text);
		}
	}
	else
	{
		ERR_MSG("fork");
		return -1;
	}

	//关闭套接字
	close(sfd);
	return 0;
}

 text.h:

#ifndef __TEXT_H__
#define __TEXT_H__

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <signal.h>
#include <time.h>
//打印错误新的宏函数
#define ERR_MSG(msg1)  do{\
	fprintf(stderr, " __%d__ ", __LINE__);\
	perror(msg1);\
}while(0)


//存储客户端地址信息的结构体说明
typedef struct Node
{
	struct sockaddr_in cin_addr; 
    struct Node *next;         //指针域，记录下个结点的地址
}linkList;

//定义发送包的结构体
typedef struct MSG
{
	char type;       //1加入  2消息  3退出 
	char username[15];
	char text[128];
}khd_msg;

//服务器功能列表
void gnmenu();

//创建并头插函数
int list_intsrt_head(linkList*M, struct sockaddr_in cin_addr,khd_msg msg,FILE* fw);

//遍历并发送加入信息
void list_show(linkList *M,khd_msg msg,int sfd);

//遍历并发送给所有客户端信息
void list_msg(linkList *M,khd_msg msg,int sfd,struct sockaddr_in cin,FILE* fw);

//遍历并发送给所有客户端退出信息
void list_del(linkList *M,khd_msg msg,int sfd,struct sockaddr_in cin,FILE* fw);
//链表判空
int list_empty(linkList *M);




#endif

 text.c:

#include "test.h"

//创建并头插函数
int list_intsrt_head(linkList*M, struct sockaddr_in cin_addr,khd_msg msg,FILE *fw)
{
	//判断条件
	if(NULL==M)
	{
		printf("所给链表不合法\n");
		return -1;
	}

	//申请节点
	linkList *p = (linkList*)malloc(sizeof(linkList));
	if(NULL==p)
	{
		printf("节点申请失败\n");
		return -1;
	}

	//将要插入的数据放入节点
	p->cin_addr = cin_addr;
	p->next = NULL;

	//完成头插
	p->next = M->next;
	M->next = p;

	time_t nowtime = time(NULL);
	struct tm *info = localtime(&nowtime);
	fprintf(fw,"[%d-%02d-%02d %02d:%02d:%02d] ", \
			info->tm_year+1900, info->tm_mon+1, info->tm_mday,\
			info->tm_hour, info->tm_min, info->tm_sec);
	fprintf(fw,"[%s:%d] %s 群发成功\n",inet_ntoa(cin_addr.sin_addr),cin_addr.sin_port,msg.username);
	fflush(fw);
	return 0;
}



//遍历并发送加入信息
void list_show(linkList *M,khd_msg msg,int sfd)
{
	//遍历逻辑
	linkList *q = M;
	
	//发送
	sprintf(msg.text,"------%s-------","加入聊天");
	while(q->next != NULL)
	{
		q = q->next;
		if(sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&(q->cin_addr), sizeof(q->cin_addr)) < 0)
		{                                                                                
			ERR_MSG("sendto");
			return ;
		}                                               
	}
}




//遍历并发送给所有客户端信息
void list_msg(linkList *M,khd_msg msg,int sfd,struct sockaddr_in cin,FILE *fw)
{
	//遍历逻辑
	linkList *q = M->next;

	while(q != NULL )
	{
		if(memcmp(&(q->cin_addr),&cin,sizeof(q->cin_addr)))
		{
			if(sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&(q->cin_addr), sizeof(q->cin_addr)) < 0)
			{                                                                                
				ERR_MSG("sendto");
				return ;
			}
		}
		q = q->next;
	}

	//将信息写入日志
	time_t nowtime = time(NULL);
	struct tm *info = localtime(&nowtime);
	fprintf(fw,"[%d-%02d-%02d %02d:%02d:%02d] ", \
			info->tm_year+1900, info->tm_mon+1, info->tm_mday,\
			info->tm_hour, info->tm_min, info->tm_sec);
	fprintf(fw,"[%s:%d] %s 群发成功\n",inet_ntoa(cin.sin_addr),cin.sin_port,msg.username);
	fflush(fw);
	return;
}

//遍历并发送给所有客户端退出信息
void list_del(linkList *M,khd_msg msg,int sfd,struct sockaddr_in cin,FILE* fw)
{
	//遍历逻辑
	linkList *q = M;
	sprintf(msg.text,"------%s-------","退出聊天");
	while(q->next != NULL )
	{
		if(memcmp(&(q->next->cin_addr),&cin,sizeof(cin)))
		{
			if(sendto(sfd, &msg, sizeof(msg), 0, (struct sockaddr*)&(q->next->cin_addr), sizeof(q->next->cin_addr)) < 0)
			{                                                                                
				ERR_MSG("sendto");
				return ;
			}
			q = q->next;
		}
		else 
		{
			//删除逻辑
			linkList *p = q->next;
			q->next = p->next;
			free(p);
			p = NULL;
		}
	}

	//将信息写入日志
	time_t nowtime = time(NULL);
	struct tm *info = localtime(&nowtime);
	fprintf(fw,"[%d-%02d-%02d %02d:%02d:%02d] ", \
			info->tm_year+1900, info->tm_mon+1, info->tm_mday,\
			info->tm_hour, info->tm_min, info->tm_sec);
	fprintf(fw,"[%s:%d] %s 群发成功\n",inet_ntoa(cin.sin_addr),cin.sin_port,msg.username);
	fflush(fw);
	return;

}

//服务器功能列表
void gnmenu()
{
	printf("****************************************\n");
	printf("*---------服务器功能对应包列表---------*\n");
	printf("*********1.群发服务器系统信息***********\n");
	printf("*********2.踢出客户端*******************\n");
	printf("*********3.禁言客户端*******************\n");
	printf("*********4.查看聊天室所有客户端信息*****\n");
	printf("*********5.查看日志文件*****************\n");
	printf("****************************************\n");   
}

洪乐乐
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值