基于UDP的网络聊天室
项目需求
1.如果有用户登录,其他用户可以收到这个人的登录信息
2.如果有人发送信息,其他用户可以收到这个人的群聊信息
3.如果有人下线,其他用户可以收到这个人的下线信息
4.服务器可以发送系统信息
写项目的方法
1.画流程图
2.根据流程图写框架
3.将每个功能实现
流程图
服务器
客户端
实现代码
服务器
#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 <stdlib.h>
#include <pthread.h>
#include "linklist.h"
//打印错误新的宏函数
#define ERR_MSG(msg) do{\
fprintf(stderr, " __%d__ ", __LINE__);\
perror(msg);\
}while(0)
int sfd;
void* login(void* arg)
{
Linklist *L=(Linklist*)arg;
struct sockaddr_in cin; //存储接收到数据包来自哪里
char buf[128]="";
socklen_t addrlen=sizeof(cin);
Linklist *p=L->next;
while(1)
{
p=L->next;
//接收
bzero(buf,sizeof(buf));
if(recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,&addrlen)<0)
{
ERR_MSG("recfrom");
return NULL;
}
if(buf[0]=='L')
{
list_insert_tail(L,cin);
p=L->next;
char str[100]="";
while(p!=NULL)
{
//发送
bzero(str,sizeof(str));
sprintf(str,"%c---%s已登录---\n",1,buf+1);
if(sendto(sfd, str, sizeof(str), 0, (struct sockaddr*)&(p->data), sizeof(cin)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
p=p->next;
}
printf("%s\t[%s:%d]\t%s\n",buf+1, inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),"登录成功");
}
else if(buf[0]=='Q')
{
p=L->next;
char str[100]="";
while(p!=NULL)
{
if(p->data.sin_family==cin.sin_family&&p->data.sin_port==cin.sin_port&&p->data.sin_addr.s_addr==cin.sin_addr.s_addr)
{
p=p->next;
continue;
}
bzero(str,sizeof(str));
sprintf(str,"%c---%s已下线---\n",1,buf+1);
if(sendto(sfd, str, sizeof(str), 0, (struct sockaddr*)&(p->data), sizeof(cin)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
p=p->next;
}
printf("%s\t[%s:%d]\t%s\n",buf+1, inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),"退出成功");
list_delete_tail(L);
}
else if(buf[0]=='C'||buf[0]=='S')
{
while(p!=NULL)
{
if(p->data.sin_family==cin.sin_family&&p->data.sin_port==cin.sin_port&&p->data.sin_addr.s_addr==cin.sin_addr.s_addr)
{
p=p->next;
continue;
}
if(sendto(sfd,buf, sizeof(buf), 0, (struct sockaddr*)&(p->data), sizeof(cin)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
p=p->next;
}
if(buf[0]=='C')
{
printf("%s\t[%s:%d]\t%s : %s\n",buf+1, inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),"chat成功", buf+2+strlen(buf+1));
}
if(buf[0]=='S')
{
printf("%s\t[%s:%d]\t%s\n",buf+1, inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),"System信息发送成功");
}
}
else
{
printf("error");
return NULL;
}
}
pthread_exit(NULL);
}
void* snd(void* arg)
{
struct sockaddr_in sin = *(struct sockaddr_in*)arg;
char str[100] = "";
char buf[128] = "";
while(1)
{
bzero(str,sizeof(str));
bzero(buf,sizeof(buf));
fgets(str, sizeof(str), stdin);
str[strlen(str)-1] = 0;
int size=sprintf(buf,"%c%s%c%s%c",'S',"System",0,str,0);
if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
}
pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
if(argc < 3)
{
fprintf(stderr, "请输入IP port\n");
return -1;
}
//将获取到的端口号字符串,转换成整形
int port = atoi(argv[2]);
//创建报式套接字
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("connect");
return -1;
}
printf("bind success\n");
Linklist *L=list_create();
if(NULL==L)
{
return -1;
}
pthread_t tid1,tid2;
//创建线程
if(pthread_create(&tid1, NULL, login, (void*)L) != 0)
{
perror("pthread_create");
return -1;
}
if(pthread_create(&tid2, NULL,snd, (void*)&sin) != 0)
{
perror("pthread_create");
return -1;
}
pthread_join(tid1,NULL);
pthread_join(tid2,NULL);
//关闭套接字
close(sfd);
return 0;
}
客户端
#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 <pthread.h>
#include <netinet/in.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
//打印错误新的宏函数
#define ERR_MSG(msg) do{\
fprintf(stderr, " __%d__ ", __LINE__);\
perror(msg);\
}while(0)
char name[20] = "";
int sfd;
struct sockaddr_in sin1;
void* snd(void* arg)
{
char *buf=(char*)arg;
while(1)
{
bzero(buf, 128);
char str[100];
fgets(str, sizeof(str), stdin);
str[strlen(str)-1] = 0;
int size=sprintf(buf,"%c%s%c%s",'C',name,'\0',str);
if(strcmp(str,"quit")==0)
{
buf[0]='Q';
}
//将数据包发送给服务器,所以地址信息结构体需要填服务器的信息
if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin1, sizeof(sin1)) < 0)
{
ERR_MSG("sendto");
return NULL;
}
if(strcmp(str,"quit")==0)
{
exit(0);
}
}
}
void* rcv(void* arg)
{
char *buf=(char*)arg;
struct sockaddr_in rcv_addrmsg; //存储接收到的数据包来自哪里
socklen_t addrlen= sizeof(rcv_addrmsg);
while(1)
{
//接收
bzero(buf, 128);
//接收服务器发送过来的数据包
if(recvfrom(sfd, buf,128, 0, (struct sockaddr*)&rcv_addrmsg, &addrlen) < 0)
{
ERR_MSG("recvfrom");
return NULL;
}
if(buf[0]==1) //登录时的数据包
{
printf("%s",buf+1);
}
else //聊天时的数据包
{
printf("%s:%s\n",buf+1,buf+strlen(buf+1)+1+1);
}
}
}
int main(int argc, const char *argv[])
{
if(argc < 3)
{
fprintf(stderr, "请输入IP port\n");
return -1;
}
//将获取到的端口号字符串,转换成整形
int port = atoi(argv[2]);
//创建报式套接字
sfd = socket(AF_INET, SOCK_DGRAM, 0);
if(sfd < 0)
{
ERR_MSG("socket");
return -1;
}
//绑定客户端自身的地址信息结构体 ---> 非必须绑定
//填充服务器的IP地址以及端口号 -->因为客户端要主动发送数据包给服务器
sin1.sin_family = AF_INET;
sin1.sin_port = htons(port);
sin1.sin_addr.s_addr = inet_addr(argv[1]);
char buf[128] = "";
printf("请输入姓名>>>");
fgets(name, sizeof(name), stdin);
name[strlen(name)-1] = 0;
int size=sprintf(buf,"%c%s%c",'L',name,'\0');
//将数据包发送给服务器,所以地址信息结构体需要填服务器的信息
if(sendto(sfd, buf, size, 0, (struct sockaddr*)&sin1, sizeof(sin1)) < 0)
{
ERR_MSG("sendto");
return -1;
}
//创建线程
pthread_t tid1,tid2;
if(pthread_create(&tid1, NULL, snd, (void*)buf) != 0)
{
perror("pthread_create");
return -1;
}
if(pthread_create(&tid2, NULL,rcv, (void*)buf) != 0)
{
perror("pthread_create");
return -1;
}
pthread_join(tid1,NULL);
pthread_join(tid2,NULL);
//关闭套接字
close(sfd);
return 0;
}