网络编程3

#include <stdio.h>
#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <pthread.h>

/*
struct ip_mreq
{
	struct in_addr imr_multiaddr;	//组播地址
	struct in_addr imr_interface;       //本机地址
}

struct in_addr imr_multiaddr
{
	s_addr;//组播地址
}

struct in_addr imr_interface
{
	s_addr;//本机地址
}
*/
int sock_fd = -1;
void *to_multi(void *arg)
{
	
	struct ip_mreq m = *(struct ip_mreq *)arg;
	//往组播地址发消息
	
	struct sockaddr_in multi; 
	multi.sin_family = AF_INET;
	multi.sin_port = htons(9999);
	multi.sin_addr.s_addr = inet_addr("224.10.10.20");
	
	socklen_t multi_len = sizeof(multi);
	char *buf = malloc(100);
	
	while(1)
	{
		memset(buf,0,100);
		fgets(buf,100,stdin);
		sendto(sock_fd,buf,100,0,(struct sockaddr *)&multi,multi_len);
		if(strcmp(buf,"bye\n") == 0)
		{
			//退出组播组
			//发送广播一条消息
			setsockopt(sock_fd,IPPROTO_IP,IP_DROP_MEMBERSHIP,&m ,m_len );
			break;
		}
	}
	
	//？？怎么退出主线程
}


int main()
{
	//1.创建套接字
	sock_fd = socket(AF_INET,SOCK_DGRAM,0);
	
	//2.加入组播组
	struct ip_mreq m;
	m.imr_multiaddr.s_addr = inet_addr("224.10.10.20");//224.0.0.1-------239.255.255.255
	m.imr_interface.s_addr = INADDR_ANY;
	
	socklen_t m_len = sizeof(m);
	
	setsockopt(sock_fd,IPPROTO_IP,IP_ADD_MEMBERSHIP,&m ,m_len );
	
	//3.绑定地址
	struct sockaddr_in local,opposite,broadcast;
	local.sin_family = AF_INET;
	local.sin_port = htons(9999);
	local.sin_addr.s_addr = INADDR_ANY;
	
	socklen_t local_len = sizeof(local);
	socklen_t opposite_len = sizeof(opposite);
	
	bind(sock_fd,(struct sockaddr *)&local,local_len);
	
	//发送一条广播消息
	int bt = 1;
	socklen_t len = sizeof(bt);
	int ret = setsockopt(sock_fd, SOL_SOCKET, SO_BROADCAST, &bt, len);
					//选项级别  //级别中的选项名 //选项值 //选项值的大小
	if(-1 == ret)
	{
		perror("set socket failed");
		return -1;
	}		
	char *buf = malloc(100);
	
	memset(buf,0,100);
	
	snprintf(buf,100,"大爷已加入组播，快来和我聊天吧");
	
	sendto(sock_fd,buf,100,0,(struct sockaddr *)&broadcast,broadcast_len);
	
	
	pthread_t tid;
	pthread_create(&tid,NULL,to_multi,(void *)&m);
	
	
	while(1)
	{
		memset(buf,0,100);
		recvfrom(sock_fd,buf,100,0,(struct sockaddr *)&opposite,&opposite_len);
		printf("%s\n",buf);
	}
	
}