创作本文目的:记录自己的学习历程
一、基于UDP的网络聊天室
1.要求
- 如果有用户登录,其他用户可以收到这个人的登录信息
- 如果有人发送信息,其他用户可以收到这个人的群聊信息
- 如果有人下线,其他用户可以收到这个人的下线信息
- 服务器可以发送系统信息
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;
}