题目:基于UDP搭建群聊
代码:
wechart_ser.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <strings.h>
#include <stdlib.h>
#define ERR_MSG(msg) do{\
fprintf(stderr, "line:%d\n", __LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.1.27" //填你们自己的本机IP ifconfig查看
#define PORT 6666 //1024~49151
struct msg
{
char type;
char name[20];
char text[128];
};
struct account
{
struct sockaddr_in cin;
char name[20];
};
struct account zin[100];
struct sockaddr_in rcvAddr; //存储接收到的数据包的地址信息
socklen_t addrlen = sizeof(rcvAddr);
int num=0;
void *call_r(void *arg)
{
int sfd=*((int*)arg);
struct msg cli;
bzero(&cli,sizeof(cli));
// char buf[128] = "";
ssize_t res = 0;
while(1)
{
//接收
bzero(&cli,sizeof(cli));
res = recvfrom(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&rcvAddr, &addrlen);
if(res < 0)
{
ERR_MSG("recvfrom");
break;
}
if(cli.type=='L')
{
int i;
for(i=0;i<num;i++)
{
if(strcmp(zin[i].name,cli.name)==0)
{
break;
}
}
if(i==num)
{
cli.type='C';
strcpy(cli.text,"登录成功");
for(int j=0;j<num;j++)
{
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&zin[j].cin, sizeof(zin[j].cin)) < 0)
{
ERR_MSG("sendto");
break;
}
}
strcpy(inet_ntoa(zin[num].cin.sin_addr),inet_ntoa(rcvAddr.sin_addr));
zin[num].cin.sin_port=rcvAddr.sin_port;
strcpy(zin[num].name,cli.name);
num++;
}
else
{
cli.type='C';
strcpy(cli.text,"账号已登录,不可重复登录");
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&rcvAddr, sizeof(rcvAddr)) < 0)
{
ERR_MSG("sendto");
break;
}
}
}
else if(cli.type=='C')
{
cli.type='C';
for(int i=0;i<num;i++)
{
if(strcmp(zin[i].name,cli.name)==0)
{
continue;
}
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&zin[i].cin, sizeof(zin[i].cin)) < 0)
{
ERR_MSG("sendto");
break;
}
}
}
else if(cli.type=='Q')
{
int quit;
cli.type='C';
strcpy(cli.text,"账号已退出");
for(int i=0;i<num;i++)
{
if(strcmp(zin[i].name,cli.name)==0)
{
quit=i;
continue;
}
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&zin[i].cin, sizeof(zin[i].cin)) < 0)
{
ERR_MSG("sendto");
break;
}
}
for(int i=quit;i<(num-1);i++)
{
zin[i]=zin[i+1];
}
num--;
}
printf("[%c:%s:%s]\n",cli.type,cli.name,cli.text);
}
}
void *call_w(void *arg)
{
int sfd=*((int*)arg);
struct msg cli;
bzero(&cli,sizeof(cli));
//char buf[128] = "";
//ssize_t res = 0;
while(1)
{
bzero(&cli, sizeof(cli));
scanf("%c %s %s",&cli.type,cli.name,cli.text);
while(getchar()!=10);
if(cli.type=='Q'&&strcmp(cli.text,"quit")==0)
{
strcpy(cli.text,"账号已退出");
for(int i=0;i<num;i++)
{
if(strcmp(cli.name,zin[i].name)==0)
{
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&zin[i].cin, sizeof(zin[i].cin)) < 0)
{
ERR_MSG("sendto");
break;
}
continue;
}
cli.type='C';
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&zin[i].cin, sizeof(zin[i].cin)) < 0)
{
ERR_MSG("sendto");
break;
}
cli.type='Q';
}
}
else if(cli.type=='C')
{
for(int i=0;i<num;i++)
{
if(strcmp(cli.name,zin[i].name)==0)
{
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&zin[i].cin, sizeof(zin[i].cin)) < 0)
{
ERR_MSG("sendto");
break;
}
break;
}
}
}
}
}
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd = socket(AF_INET, SOCK_DGRAM, 0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
//允许端口快速重用--->允许程序退出后,端口号能被新的进程快速覆盖
//必须放在bind之前
int reuse = 1;
if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) <0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("允许端口快速重用成功\n");
//填充服务器自身的地址信息结构体,AF_INET: man 7 ip
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
//绑定服务器的地址信息结构体
if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
for(int i=0;i<100;i++)
{
bzero(inet_ntoa(zin[i].cin.sin_addr),sizeof(zin[i].cin.sin_addr));
zin[i].cin.sin_port=-1;
bzero(zin[i].name,sizeof(zin[i].name));
}
pthread_t tid_r,tid_w;
pthread_create(&tid_r,NULL,call_r,&sfd);
pthread_create(&tid_w,NULL,call_w,&sfd);
pthread_join(tid_r,NULL);
pthread_join(tid_w,NULL);
//关闭套接字
close(sfd);
return 0;
}
wechart_cli.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#define ERR_MSG(msg) do{\
fprintf(stderr, "line:%d\n", __LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.1.27" //填你们自己的本机IP ifconfig查看
#define PORT 6666 //1024~49151
pthread_t tid_r,tid_w;
struct msg
{
char type;
char name[20];
char text[128];
};
void *call_r(void *arg)
{
int sfd=*((int*)arg);
struct sockaddr_in rcvAddr; //存储接收到的数据包的地址信息
socklen_t addrlen = sizeof(rcvAddr);
char buf[128] = "";
ssize_t res = 0;
struct msg cli;
while(1)
{
//接收
bzero(&cli, sizeof(cli));
res = recvfrom(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&rcvAddr, &addrlen);
if(res < 0)
{
ERR_MSG("recvfrom");
break;
}
if(cli.type=='Q')
{
printf("[%s:账号已退出]\n",cli.name);
break;
}
printf("[%s:%s]\n",cli.name,cli.text);
}
pthread_cancel(tid_r);
pthread_cancel(tid_w);
}
void *call_w(void *arg)
{
int sfd=*((int*)arg);
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
struct msg cli;
bzero(&cli,sizeof(cli));
printf("请输入姓名:");
cli.type='L';
scanf("%s",cli.name);
while(getchar()!=10);
if(sendto(sfd,&cli,sizeof(cli), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("sendto");
}
while(1)
{
bzero(cli.text, sizeof(cli.text));
fgets(cli.text,sizeof(cli.text), stdin);
cli.text[strlen(cli.text)-1] = 0;
if(strcmp(cli.text,"quit")==0)
{
cli.type='Q';
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("sendto");
break;
}
break;
}
cli.type='C';
if(sendto(sfd,&cli, sizeof(cli), 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("sendto");
break;
}
//printf("sendto success\n");
}
pthread_cancel(tid_w);
pthread_cancel(tid_r);
}
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd = socket(AF_INET, SOCK_DGRAM, 0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
pthread_create(&tid_r,NULL,call_r,&sfd);
pthread_create(&tid_w,NULL,call_w,&sfd);
pthread_join(tid_r,NULL);
pthread_join(tid_w,NULL);
//关闭套接字
close(sfd);
return 0;
}
运行结果:
![](https://i-blog.csdnimg.cn/blog_migrate/a4948ce2639dfb56d6c56d62d997b0b8.png)