代码:
服务端:存储客户端端口,IP;作为中转将某个用户发送的信息发给其他用户,并且可以指定用户退出;
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define ERR_MSG(msg){\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0);
#define IP "192.168.8.98"
#define PORT 6666
//协议结构体
struct mesg{
char flag;//'d'登录信息 'c'聊天信息 'q'退出信息
char name[20];
char txt[128];
};
//定义存储客户端IP与端口的链表
typedef struct sockaddr_in datatype;
typedef struct Node
{
union{
datatype data;
int len;
};
struct Node *next;
}linklist;
//创建链表
linklist *link_create();
//创建节点
linklist *get_node(datatype e);
//添加节点(头插)
int link_add(linklist *L,datatype e);
//按值进行查找
int list_find(linklist *L,datatype e);
//删除节点(按值查找删除)
int link_dele(linklist *L,datatype e);
linklist *L;
void *ser_rcv(void *arg)
{
pthread_detach(pthread_self());
struct mesg msg;
int sfd = *(int *)arg;
ssize_t res = 0;
struct sockaddr_in sin;
socklen_t addrlen = sizeof(sin);
while(1)
{
//清空上一次接收的姓名与信息,防止本次接收受到干扰
bzero(msg.txt,sizeof(msg.txt));
bzero(msg.name,sizeof(msg.name));
//接收来自客户端的数据包
res = recvfrom(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&sin,&addrlen);
if(res < 0)
{
ERR_MSG("recvfrom");
break;
}
linklist *p = L;
//判断发送的数据包是哪类信息
switch(msg.flag)
{
case 'd'://发送该用户登录信息给其他用户,再存储该用户的IP与端口
strcpy(msg.txt,">>>>>该新用户加入聊天!");
for(int i = 1;i <= L->len;i++)
{
p = p->next;
if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&p->data,sizeof(p->data)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
}
printf("serve sendto [%s] load message success\n",msg.name);
printf("[%s %d]%s %s\n",inet_ntoa(sin.sin_addr),\
ntohs(sin.sin_port),msg.name,msg.txt);//显示客户端IP,端口,姓名,信息
link_add(L,sin);//存储新用户IP与端口
break;
case 'c'://发送该名用户姓名以及信息内容给其他用户
for(int i = 1;i <= L->len;i++)
{
p = p->next;
if((p->data.sin_port == sin.sin_port) &&(p->data.sin_addr.s_addr == sin.sin_addr.s_addr))//本人跳过
continue;
if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&p->data,sizeof(p->data)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
}
printf("serve send [%s] chat message success\n",msg.name);
break;
case 'q'://服务器收到该用户的退出信息,删除用户的地址节点,向其他用户发送该用户的退出信息
link_dele(L,sin);
strcpy(msg.txt,">>>>>该用户退出聊天!");
for(int i = 1;i <= L->len;i++)
{
p = p->next;
if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&p->data,sizeof(p->data)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
}
printf("serve sendto [%s] quit message success\n",msg.name);
break;
}
}
}
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd = socket(AF_INET,SOCK_DGRAM,0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
printf("serve socket success\n");
//填写信息结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
socklen_t addrlen = sizeof(sin);
//绑定服务器端口与IP
if(bind(sfd,(struct sockaddr*)&sin,addrlen) < 0)
{
ERR_MSG("bind");
return -1;
}
printf("sreve bind success\n");
//多线程实现同时收发
pthread_t tid_rcv;//分支线程接收
L = link_create();
if(NULL == L)
{
printf("链表创建失败\n");
return -1;
}
if(pthread_create(&tid_rcv,NULL,ser_rcv,&sfd) != 0)
{
fprintf(stderr,"pthread_create ser_rcv failed!\n");
return -1;
}
//主线程只需要发送服务器给所有客户端的信息
struct mesg msg;
bzero(msg.name,sizeof(msg.name));
strcpy(msg.name,"system");
while(1)//主线程发送
{
bzero(msg.txt,sizeof(msg.txt));
fgets(msg.txt,sizeof(msg.txt),stdin);
msg.txt[strlen(msg.txt) - 1] = 0;
//发送的内容包括操作码flag,姓名name,信息txt
//name由分支线程的recvfrom填充,flag内容无意义
linklist *p = L;
for(int i = 1;i <= L->len;i++)
{
p = p->next;
if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&p->data,sizeof(p->data)) < 0)
{
ERR_MSG("sendto");
return -1;
}
}
printf("serve sendto success\n");
}
return 0;
}
//创建链表头结点
linklist *link_create()
{
linklist *L = (linklist *)malloc(sizeof(linklist));
if(NULL == L)
{
printf("链表创建失败\n");
return NULL;
}
L->len = 0;
L->next = NULL;
return L;
}
//向链表内添加节点,节点存储客户端IP与端口,节点数据域类型
//为地址信息结构体类型
linklist *get_node(datatype e)
{
linklist *p = (linklist*)malloc(sizeof(linklist));
if(NULL == p)
{
printf("创建节点失败\n");
return NULL;
}
p->data = e;
p->next = NULL;
return p;
}
//头插,向头节点添加节点
int link_add(linklist *L,datatype e)
{
if(NULL == L)
{
printf("链表不合法\n");
return 0;
}
linklist *p = get_node(e);
p->next = L->next;
L->next = p;
L->len++;
return 0;
}
//按值进行查找,查找对应数据在链表的位置
int link_find(linklist *L,datatype e)
{
if(NULL == L)
{
printf("链表不合法\n");
return -1;
}
linklist *p = L->next;
for(int i = 1;i <= L->len;i++)
{
if((p->data.sin_port == e.sin_port) &&(p->data.sin_addr.s_addr == e.sin_addr.s_addr))
return i;//当端口相同的时候,返回位置
p = p->next;
}
return 0;
}
//按值进行删除
int link_dele(linklist *L,datatype e)
{
if(NULL == L)
{
printf("链表不合法\n");
return -1;
}
int pos = link_find(L,e);
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--;
return 0;
}
客户端:发送用户信息,接收其他用户的信息,并且可以自己退出;
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define ERR_MSG(msg){\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0);
#define IP "192.168.8.98"
#define PORT 6666
struct mesg{
char flag;
char name[20];
char txt[128];
};
struct mesg msg_rcv;
struct mesg msg;
pthread_t tid_rcv,tid_snd;//线程接收,发送
void *cli_snd(void *arg)
{
int sfd = *(int*)arg;
//填写信息结构体
struct sockaddr_in cin;
cin.sin_family = AF_INET;
cin.sin_port = htons(PORT);
cin.sin_addr.s_addr = inet_addr(IP);
socklen_t addrlen = sizeof(cin);
bzero(msg.txt,sizeof(msg.txt));//清空发送信息数组,防止随机值干扰
//登录协议数据包发送
printf("请输入登录名>>");
fgets(msg.name,sizeof(msg.name),stdin);
msg.name[strlen(msg.name) - 1] = 0;
msg.flag = 'd';
if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&cin,addrlen) < 0)
{
ERR_MSG("sendto");
exit(0);
}
printf("登录信息发送成功\n");
//群聊信息发送
while(1)//主线程发送
{
bzero(msg.txt,sizeof(msg.txt));
//输入发送信息
fgets(msg.txt,sizeof(msg.txt),stdin);
msg.txt[strlen(msg.txt) - 1] = 0;
//判断是否是退出信息
if(strcmp(msg.txt,"quit") == 0)
msg.flag = 'q';
else
msg.flag = 'c';
if(sendto(sfd,&msg,sizeof(msg),0,(struct sockaddr*)&cin,addrlen) < 0)
{
ERR_MSG("sendto");
break;
}
printf("信息发送成功\n");
//告诉服务器本客户端退出后退出客户端
if(msg.flag == 'q')
{
printf("%s退出聊天\n",msg.name);
break;
}
}
pthread_cancel(tid_rcv);
pthread_exit(NULL);
}
void *cli_rcv(void *arg)
{
int sfd = *(int *)arg;
ssize_t res = 0;
while(1)
{
res = recvfrom(sfd,&msg_rcv,sizeof(msg_rcv),0,NULL,NULL);//不需要接收服务端端口与IP,填空
if(res < 0)
{
ERR_MSG("recvfrom");
break;
}
if(strcmp(msg.name,msg_rcv.txt) == 0)
{
printf("本用户被强制下线\n");
break;
}
printf("cli [%s] recvfrom success\n",msg.name);
printf("%s>>%s\n",msg_rcv.name,msg_rcv.txt);
}
pthread_cancel(tid_snd);
pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd = socket(AF_INET,SOCK_DGRAM,0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
//多线程实现同时收发
if(pthread_create(&tid_rcv,NULL,cli_rcv,&sfd) != 0)
{
fprintf(stderr,"pthread_create cli_rcv failed!\n");
return -1;
}
if(pthread_create(&tid_snd,NULL,cli_snd,&sfd) != 0)
{
fprintf(stderr,"pthread_create cli_snd failed!\n");
return -1;
}
pthread_join(tid_rcv,NULL);
pthread_join(tid_snd,NULL);
return 0;
}
实现效果:
不足:1.没有避免用户名重名,可以将用户名存储到对应链表节点的数据域中,定义一个结构体作为数据域存储用户的IP与端口和用户名,当收到登录请求时,遍历链表来查重;
2.服务器没有考虑退出,可以在服务器主线程发送端定义退出条件strcmp(msg.txt,"quit")比较条件后退出服务器;
3.服务器容易退出,但是当服务器退出后,由于UDP无连接,没有应答机制,客户端接收不到服务器的退出消息,就需要仿照TCP做UDP的应答机制。