2022/8/12---网络编程day3

创作本文目的:记录自己的学习历程


一、基于UDP的网络聊天室

1.要求

  1. 如果有用户登录,其他用户可以收到这个人的登录信息
  2. 如果有人发送信息,其他用户可以收到这个人的群聊信息
  3. 如果有人下线,其他用户可以收到这个人的下线信息
  4. 服务器可以发送系统信息

2.服务器代码

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

//全局变量取链表头节点的地址
Doublelink **D_global;

//分支线程(发送系统消息给所有成员)
pthread_t thread; 

//接收客户端发送的地址信息结构体
struct sockaddr_in cli;
socklen_t addrlen = sizeof(cli);

void *callback(void *arg)
{
	datatype tmp;

	//发送消息给所有成员
	while(1)
	{
		tmp.stc.type='4';          							//填写消息类型
		bzero(tmp.stc.mtext,sizeof(tmp.stc.mtext));
		fgets(tmp.stc.mtext,sizeof(tmp.stc.mtext),stdin);
		tmp.stc.mtext[strlen(tmp.stc.mtext)-1]=0;
		//转发给链表中所有成员
		list_show(*D_global,*(int *)arg,tmp,0);      
	}
	pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
	int pos;                //存储节点位置
	ssize_t res;            //接收函数返回值
	char buf[128]="";       //存储客户端返回信息
	datatype tmp;           //存储客户端返回信息
	Doublelink *bak;        //接收返回的节点

	//控制外部传参
	if(argc<3)
	{
		printf("请输入IP和端口\n");
		return -1;
	}

	//创建链表
	Doublelink *D = list_create();
	if(NULL==D)
	{
		return -1;
	}
	D_global=&D;

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

	//填写服务器地址信息结构体
	struct sockaddr_in ser;
	ser.sin_family=AF_INET;
	ser.sin_port=htons(atoi(argv[2]));
	ser.sin_addr.s_addr=inet_addr(argv[1]);

	//绑定IP地址和端口号
	if(bind(sfd,(struct sockaddr *)&ser,sizeof(ser))<0)
	{
		ERR_MSG("bind");
		return -1;
	}

	//创建分支线程
	if(pthread_create(&thread,NULL,callback,(void *)&sfd)!=0)
	{
		ERR_MSG("pthread_create");
		return -1;
	}
	pthread_detach(thread);          //分离线程,退出自动回收资源

	while(1)
	{
		//接收数据
		bzero(buf,sizeof(buf));
		res=recvfrom(sfd,(void *)buf,sizeof(buf),0,(struct sockaddr *)&cli,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return -1;
		}
		bzero(&tmp,sizeof(tmp));
		tmp.cin=cli;	                 	//存储接收的地址信息结构体
		pos=list_search_value(D,tmp);       //找到用户在链表中的位置
		if(pos==0)
		{
			//登录信息
			sprintf(tmp.stc.name,"%s",buf);      //填写用户名
			tmp.stc.type='1';                    //填写消息类型
			list_insert_head(D,tmp);             //头插
			list_show(D,sfd,tmp,1);              //转发给链表中其他成员
			
			//在服务器显示信息
			printf("用户:%s [ip:%s  port:%d]上线\n",buf,inet_ntoa(cli.sin_addr),ntohs(cli.sin_port));
		}
		else
		{	
			if(strcmp(buf,"quit")==0)
			{
				//退出信息
				bak=list_search_pos(D,pos);    					    //找出链表中对应的节点
				sprintf(tmp.stc.name,"%s",bak->data.stc.name);      //填写用户名
				tmp.stc.type='3';         					//填写消息类型
				list_show(D,sfd,tmp,1);           				//转发给链表中其他成员
				//在服务器显示信息
				printf("用户:%s [ip:%s  port:%d]下线\n",bak->data.stc.name,inet_ntoa(cli.sin_addr),ntohs(cli.sin_port));
				list_delete_anywhere(D,pos);        //从链表中删除节点
			}
			else
			{
				//聊天信息
				bak=list_search_pos(D,pos);                			//找出链表中对应的节点
				sprintf(tmp.stc.name,"%s",bak->data.stc.name);      //填写用户名
				tmp.stc.type='2';                    				//填写消息类型
				sprintf(tmp.stc.mtext,"%s",buf);     				//填写聊天信息
				list_show(D,sfd,tmp,1);                      		//转发给链表中的其他成员
		        //在服务器显示信息
				printf("用户:%s [ip:%s  port:%d]:%s\n",bak->data.stc.name,inet_ntoa(cli.sin_addr),ntohs(cli.sin_port),buf);
			}
		}
	}

	//关闭套接字
	close(sfd);
	//释放链表
	list_free(D);

	return 0;
}

3.客户端代码

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

//打印报错信息
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)

//分支线程(接收系统消息)
pthread_t thread;   

//填写服务器的地址信息结构体
struct sockaddr_in ser;
socklen_t addrlen =sizeof(ser);

//线程任务体
void *callback(void *arg)
{
	msg tmp;
	ssize_t res;

	while(1)
	{
		//接收系统信息
		bzero(&tmp,sizeof(tmp));
		res=recvfrom(*(int *)arg,(void *)&tmp,sizeof(tmp),0,(struct sockaddr *)&ser,&addrlen);
		if(res<0)
		{
			ERR_MSG("recvfrom");
			return NULL;
		}
		switch(tmp.type)
		{
			case '1':
				{
					printf("-----%s上线啦-----\n",tmp.name);
				}break;
			case '2':
				{
					printf("%s:%s\n",tmp.name,tmp.mtext);
				}break;
			case '3':
				{
					printf("-----%s下线啦-----\n",tmp.name);
				}break;
			case '4':
				{
					printf("Server Message:%s\n",tmp.mtext);
				}break;
			default:printf("消息种类不正确:%c\n",tmp.type);
		}
	}

	pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
	ssize_t res;
	char buf[128]="";

	//控制外部传参
	if(argc<3)
	{
		printf("请输入IP和端口\n");
		return -1;
	}

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

	//绑定自身IP和端口(非必须)

	//填写服务器地址信息结构体
	ser.sin_family=AF_INET;
	ser.sin_port=htons(atoi(argv[2]));
	ser.sin_addr.s_addr=inet_addr(argv[1]);
	
	//创建分支线程
	if(pthread_create(&thread,NULL,callback,(void *)&sfd)!=0)
	{
		ERR_MSG("pthread_create");
		return -1;
	}
	pthread_detach(thread);          //分离线程,退出自动回收资源

	//发送用户名
	bzero(buf,sizeof(buf));
	printf("请输入用户名:");
	fgets(buf,sizeof(buf),stdin);
	buf[strlen(buf)-1]=0;
	res=sendto(sfd,(void *)buf,sizeof(buf),0,(struct sockaddr *)&ser,addrlen);
	if(res<0)
	{
		ERR_MSG("sendto");
		return -1;
	}

	while(1)
	{
		//发送聊天消息
		bzero(buf,sizeof(buf));
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1]=0;
		res=sendto(sfd,(void *)buf,sizeof(buf),0,(struct sockaddr *)&ser,addrlen);
		if(res<0)
		{
			ERR_MSG("sendto");
			return -1;
		}
		if(strcasecmp(buf,"quit")==0)
		{
			break;
		}
	}

	//关闭套接字
	close(sfd);

	return 0;
}

4.功能代码

doublelink.h

#ifndef __DOUBLELINK_H__
#define __DOUBLELINK_H__

//打印报错信息
#define ERR_MSG(msg) do{\
	fprintf(stderr,"__%d__",__LINE__);\
	perror(msg);\
}while(0)

//服务器消息包结构体
typedef struct
{
	char type;           //登录消息1,聊天消息2,退出消息3,服务器消息4
	char name[20];       //用户名
	char mtext[128];     //聊天信息
}msg;

typedef struct
{
	struct sockaddr_in cin;
	msg stc;
}datatype;

typedef struct Node
{
	union
	{
		int len;          //存储头节点,表示节点个数
		datatype data;    //存储普通节点,表示数据
	};
	struct Node *next;    //指向下一个节点
	struct Node *prev;    //指向前一个节点
}Doublelink;

//创建
Doublelink *list_create();
//判空
int list_empty(Doublelink *);
//申请节点函数
Doublelink *node_apply(datatype );
//头插
int list_insert_head(Doublelink *,datatype );
//转发客户端信息
void list_show(Doublelink *,int ,datatype ,int );
//按位置查找返回节点
Doublelink *list_search_pos(Doublelink *, int );
//头删
int list_delete_head(Doublelink *);
//任意删
int list_delete_anywhere(Doublelink *,int );
//释放
void list_free(Doublelink *);
//按值查找返回节点
int list_search_value(Doublelink *,datatype );

#endif

doublelink.c

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

//创建
Doublelink *list_create()
{
	Doublelink *D = (Doublelink *)malloc(sizeof(Doublelink));
	if(NULL==D)
	{
		printf("创建失败\n");
		return NULL;
	}

	D->len=0;
	D->prev=NULL;
	D->prev=NULL;

	//printf("创建成功\n");
	return D;
}

//判空
int list_empty(Doublelink *D)
{
	if(NULL==D)
	{
		printf("所给链表不合法\n");
		return -1;
	}

	return D->next == NULL ? 1:0;  //1表示空 0表示非空
}

//申请节点函数
Doublelink *node_apply(datatype e)
{
	Doublelink *p=(Doublelink *)malloc(sizeof(Doublelink));
	if(NULL==p)
	{
		printf("节点申请失败\n");
		return NULL;
	}

	p->data = e;
	p->prev = NULL;
	p->next = NULL;
	return p;
}

//头插
int list_insert_head(Doublelink *D,datatype e)
{
	if(NULL==D)
	{
		printf("所给链表不合法\n");
		return -1;
	}
	Doublelink *S = node_apply(e);
	if(list_empty(D))
	{
		S->prev=D;
		D->next=S;
	}
	else
	{
		S->prev=D;
		S->next=D->next;
		D->next->prev=S;
		D->next=S;
	}

	D->len++;

	//printf("头插成功\n");
	return 0;
}

//转发客户端信息 flag非0向除自己外的所有成员发送信息,0则所有
void list_show(Doublelink *D,int sfd,datatype tmp,int flag)
{
	ssize_t res;
	if(NULL==D || list_empty(D))
	{
		printf("遍历失败\n");
		return;
	}

	Doublelink *q=D->next;
	while(q!=NULL)
	{
		if(q->data.cin.sin_port!=tmp.cin.sin_port && flag)
		{
			res = sendto(sfd,(void *)&tmp.stc,sizeof(tmp.stc),0,(struct sockaddr *)&q->data.cin,sizeof(q->data.cin));
			if(res<0)
			{
				ERR_MSG("sendto");
				return;
			}
		}
		else if(!flag)
		{
			res = sendto(sfd,(void *)&tmp.stc,sizeof(tmp.stc),0,(struct sockaddr *)&q->data.cin,sizeof(q->data.cin));
			if(res<0)
			{
				ERR_MSG("sendto");
				return;
			}
		}
		q=q->next;
	}
}

//按位置查找返回节点
Doublelink *list_search_pos(Doublelink *D, int pos)
{
	if(NULL==D)
	{
		printf("定位失败\n");
		return NULL;
	}
	else if(pos<1 || pos>D->len)	
	{
		printf("位置不正确\n");
		return NULL;
	}
	Doublelink *q = D;
	for(int i=1;i<=pos;i++)
	{
		q=q->next;
	}

	return q;
}
//尾插
int list_insert_tail(Doublelink *D,datatype e)
{
	if(NULL==D)
	{
		printf("所给链表不合法\n");
		return -1;
	}
	Doublelink *p = node_apply(e);
	Doublelink *q=D;
	while(q->next!=NULL)
	{
		q=q->next;
	}
	q->next = p;
	p->prev = q;
	D->len++;
	//printf("尾插成功\n");
	return 0;
}
//头删
int list_delete_head(Doublelink *D)
{
	if(NULL==D || list_empty(D))
	{
		printf("所给链表不合法\n");
		return -1;
	}
	Doublelink *p=D->next;
	p->prev=NULL;
	D->next=p->next;
	free(p);
	p=NULL;

	D->len--;
	//printf("头删成功\n");
	return 0;
}
//任意删
int list_delete_anywhere(Doublelink *D,int pos)
{
	if(NULL==D || list_empty(D))
	{
		printf("所给链表不合法\n");
		return -1;
	}
	else if(pos<1 || pos>D->len)
	{
		printf("位置不正确\n");
		return -2;
	}

	Doublelink *p=list_search_pos(D,pos);
	if(p->next==NULL)
	{
		p->prev->next=NULL;
		free(p);
		p=NULL;
	}
	else
	{
		p->prev->next=p->next;
		p->next->prev=p->prev;
		free(p);
		p=NULL;
	}
	
	D->len--;
	//printf("删除成功\n");
	return 0;
}
//释放
void list_free(Doublelink *D)
{
	if(NULL==D)
	{
		printf("不合法\n");
		return;
	}
	
	while(D->next!=NULL)
	{
		list_delete_head(D);
	}
	free(D);
	D=NULL;
	printf("释放成功\n");
}
//按值查找返回节点位置
int list_search_value(Doublelink *D,datatype e)
{
	//判断逻辑
	if(NULL==D)
	{
		printf("查找失败\n");
		return -1;
	}
	//查找逻辑
	Doublelink *q = D->next;
	for(int i=1;i<=D->len;i++)
	{
		if(q->data.cin.sin_port==e.cin.sin_port)
		{
			return i;
		}
		q=q->next;
	}
	//printf("未查找到\n");
	return 0;
}

5.运行效果

在这里插入图片描述


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值