服务端
#include<myhead.h>
#define ERR_MSG(msg) do{\
fprintf(stderr,"__%d__:",__LINE__);\
perror(msg);\
}while(0)
#define PORT 5555 //端口号,范围1024~49151
#define PORT_R 6666
#define GRP_IP "224.32.32.32" //组播IP
#define LOL_IP "192.168.114.140" //本机IP
struct msg
{
char type;
char name[20];
char text[128];
};
//回收僵尸进程
void handler(int sig)
{
while(waitpid(-1,NULL,WNOHANG)>0);
}
int main(int argc, const char *argv[])
{
//捕获17)SIGCHLD信号,为信号注册新的处理函数
if(signal(17,handler)==SIG_ERR)
{
ERR_MSG("signal");
return -1;
}
//创建报式套接字 socket
int sfd=socket(AF_INET,SOCK_DGRAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("socket create success sfd=%d\n",sfd);
//加入多播组
struct ip_mreqn mq;
mq.imr_multiaddr.s_addr=inet_addr(GRP_IP); //要加入的组播IP
mq.imr_address.s_addr=inet_addr(LOL_IP); //本机IP
mq.imr_ifindex=2; //网络设备索引号
if(setsockopt(sfd,IPPROTO_IP,IP_ADD_MEMBERSHIP,&mq,sizeof(mq))<0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("加入多播组[%s:%d]成功\n",GRP_IP,PORT);
//填充服务器的地址信息结构体,给bing函数使用
//真实的地址信息结构体根据地址族指定,AF_INET:man 7 IP
struct sockaddr_in sin;
sin.sin_family=AF_INET; //必须填AF_INET
sin.sin_port=htons(PORT); //端口号的网络字节序
sin.sin_addr.s_addr=inet_addr(GRP_IP); //组播IP
//绑定服务器的地址信息————>必须绑定 bind
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//填充接收方的地址信息结构体,给sendto函数使用
//真实的地址信息结构体根据地址族指定,AF_INET:man 7 IP
struct sockaddr_in cin;
cin.sin_family=AF_INET; //必须填AF_INET
cin.sin_port=htons(PORT_R); //接收方绑定的端口号的网络字节序
cin.sin_addr.s_addr=inet_addr(GRP_IP); //接收方绑定的组播IP
char buf[128]="";
struct msg m; //存储客户端发送的信息结构体
struct sockaddr_in rcvaddr; //存储客户端的地址信息
socklen_t addrlen=sizeof(rcvaddr); //真实的地址信息结构体的大小
pid_t pid=-1;
while(1)
{
pid=fork();
if(pid>0)
{
//父进程
//父进程负责接收
if(recvfrom(sfd,&m,sizeof(m),0,(struct sockaddr*)&rcvaddr,&addrlen)<0)
{
ERR_MSG("recv");
return -1;
}
if(m.type='L')
{
bzero(buf,sizeof(buf));
strcpy(buf,"登录成功");
//发送数据
if(sendto(sfd,buf,sizeof(buf),0,(struct sockaddr*)&rcvaddr,sizeof(rcvaddr))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("sendto success\n");
}
else if(m.type='C')
{
//发送数据
if(sendto(sfd,&m,sizeof(m),0,(struct sockaddr*)&cin,sizeof(cin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("sendto success\n");
}
else if(m.type='Q')
{
strcpy(m.text,"已下线");
if(sendto(sfd,&m,sizeof(m),0,(struct sockaddr*)&cin,sizeof(cin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("sendto success\n");
}
}
else if(pid==0)
{
//子进程
//清空字符串
bzero(buf,sizeof(buf)); //或memset函数
printf("请输入: ");
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]='\0';
//发送数据
if(sendto(sfd,buf,sizeof(buf),0,(struct sockaddr*)&cin,sizeof(cin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("sendto success\n");
exit(0);
}
else
{
ERR_MSG("fork");
return -1;
}
}
//关闭文件名描述符
if(close(sfd)<0)
{
ERR_MSG("close");
return -1;
}
return 0;
}
客户端
#include<myhead.h>
#define ERR_MSG(msg) do{\
fprintf(stderr,"__%d__:",__LINE__);\
perror(msg);\
}while(0)
#define PORT_R 5555 //接收方绑定的端口号
#define PORT 6666
#define GRP_IP "224.32.32.32" //接收方绑定的组播IP
#define LOL_IP "192.168.114.140" //本机IP
struct msg
{
char type;
char name[20];
char text[128];
};
//回收僵尸进程
void handler(int sig)
{
while(waitpid(-1,NULL,WNOHANG)>0);
}
int main(int argc, const char *argv[])
{
//捕获17)SIGCHLD信号,为信号注册新的处理函数
if(signal(17,handler)==SIG_ERR)
{
ERR_MSG("signal");
return -1;
}
//创建报式套接字 socket
int cfd=socket(AF_INET,SOCK_DGRAM,0);
if(cfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("socket create success cfd=%d\n",cfd);
//加入多播组
struct ip_mreqn mq;
mq.imr_multiaddr.s_addr=inet_addr(GRP_IP); //要加入的组播IP
mq.imr_address.s_addr=inet_addr(LOL_IP); //本机IP
mq.imr_ifindex=2; //网络设备索引号
//填定客户端的地址信息结构体,给bing函数使用
//真实的地址信息结构体根据地址族指定,AF_INET:man 7 IP
struct sockaddr_in cin;
cin.sin_family=AF_INET; //必须填AF_INET
cin.sin_port=htons(PORT); //端口号的网络字节序
cin.sin_addr.s_addr=inet_addr(GRP_IP); //组播IP
//绑定客户端的地址信息————>必须绑定 bind
if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))<0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//填充接收方的地址信息结构体,给sendto函数使用
//真实的地址信息结构体根据地址族指定,AF_INET:man 7 IP
struct sockaddr_in sin;
sin.sin_family=AF_INET; //必须填AF_INET
sin.sin_port=htons(PORT_R); //接收方绑定的端口号的网络字节序
sin.sin_addr.s_addr=inet_addr(GRP_IP); //接收方绑定的组播IP
char buf[128]="";
struct msg m;
struct sockaddr_in rcvaddr; //存储服务端的地址信息
socklen_t addrlen=sizeof(rcvaddr); //真实的地址信息结构体的大小
while(1)
{
//登录请求
m.type='L';
printf("请输入用户名>>>");
scanf("%s",m.name);
while(getchar()!=10);
//发送数据
if(sendto(cfd,&m,sizeof(m),0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("sendto success\n");
bzero(buf,sizeof(buf));
//接收
if(recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&rcvaddr,&addrlen)<0)
{
ERR_MSG("recv");
return -1;
}
if(strcmp(buf,"登录成功")==0)
{
break;
}
}
pid_t pid;
while(1)
{
pid=fork();
if(pid>0)
{
//父进程负责接收
if(recvfrom(cfd,buf,sizeof(buf),0,(struct sockaddr*)&rcvaddr,&addrlen)<0)
{
ERR_MSG("recv");
return -1;
}
printf("[%s]:%s\n",m.name,m.text);
if(strcmp(m.text,"已下线"))
{
break;
}
}
else if(pid==0)
{
//子进程
m.type='C';
bzero(m.text,sizeof(m.text));
printf("请输入: ");
fgets(m.text,sizeof(m.text),stdin);
m.text[strlen(m.text)-1]='\0';
//发送数据
if(sendto(cfd,&m,sizeof(m),0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
return -1;
}
printf("sendto success\n");
if(strcmp(m.text,"quit")==0)
{
m.type='Q';
bzero(m.text,sizeof(m.text));
//发送数据
if(sendto(cfd,&m,sizeof(m),0,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("sendto");
return -1;
}
}
exit(0);
}
else
{
ERR_MSG("fork");
return -1;
}
}
//关闭文件名描述符
if(close(cfd)<0)
{
ERR_MSG("close");
return -1;
}
return 0;
}