Linux下组播&广播消息小例子

组播：

// 发送端
#include <iostream>
#include <stdio.h>
#include <sys/socket.h>
#include <unistd.h>
#include <sys/types.h>
#include <netdb.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>


using namespace std;

int main()
{
	setvbuf(stdout, NULL, _IONBF, 0); 
	fflush(stdout); 

	int sock = -1;
	if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) 
	{   
		cout<<"socket error"<<endl;	
		return false;
	}   
	
	const int opt = 1;
	//设置该套接字为广播类型，
	int nb = 0;
	nb = setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char *)&opt, sizeof(opt));
	if(nb == -1)
	{
		cout<<"set socket error..."<<endl;
		return false;
	}

	struct sockaddr_in addrto;
	bzero(&addrto, sizeof(struct sockaddr_in));
	addrto.sin_family=AF_INET;
	addrto.sin_addr.s_addr=htonl(INADDR_BROADCAST);
	addrto.sin_port=htons(6000);
	int nlen=sizeof(addrto);

	while(1)
	{
		sleep(1);
		//从广播地址发送消息
		char smsg[] = {"abcdef"};
		int ret=sendto(sock, smsg, strlen(smsg), 0, (sockaddr*)&addrto, nlen);
		if(ret<0)
		{
			cout<<"send error...."<<ret<<endl;
		}
		else
		{		
			printf("ok ");	
		}
	}

	return 0;
}

// 接收端
#include <iostream>
#include <stdio.h>
#include <sys/socket.h>
#include <unistd.h>
#include <sys/types.h>
#include <netdb.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>


using namespace std;

int main()
{
	setvbuf(stdout, NULL, _IONBF, 0); 
	fflush(stdout); 

	// 绑定地址
	struct sockaddr_in addrto;
	bzero(&addrto, sizeof(struct sockaddr_in));
	addrto.sin_family = AF_INET;
	addrto.sin_addr.s_addr = htonl(INADDR_ANY);
	addrto.sin_port = htons(6000);
	
	// 广播地址
	struct sockaddr_in from;
	bzero(&from, sizeof(struct sockaddr_in));
	from.sin_family = AF_INET;
	from.sin_addr.s_addr = htonl(INADDR_ANY);
	from.sin_port = htons(6000);
	
	int sock = -1;
	if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) 
	{   
		cout<<"socket error"<<endl;	
		return false;
	}   

	const int opt = 1;
	//设置该套接字为广播类型，
	int nb = 0;
	nb = setsockopt(sock, SOL_SOCKET, SO_BROADCAST, (char *)&opt, sizeof(opt));
	if(nb == -1)
	{
		cout<<"set socket error..."<<endl;
		return false;
	}

	if(bind(sock,(struct sockaddr *)&(addrto), sizeof(struct sockaddr_in)) == -1) 
	{   
		cout<<"bind error..."<<endl;
		return false;
	}

	int len = sizeof(sockaddr_in);
	char smsg[100] = {0};

	while(1)
	{
		//从广播地址接受消息
		int ret=recvfrom(sock, smsg, 100, 0, (struct sockaddr*)&from,(socklen_t*)&len);
		if(ret<=0)
		{
			cout<<"read error...."<<sock<<endl;
		}
		else
		{		
			printf("%s\t", smsg);	
		}

		sleep(1);
	}

	return 0;
}

Sender.cpp

[c-sharp]  view plain copy print ?

1. #include <sys/types.h>    
2. #include <sys/socket.h>    
3. #include <netinet/in.h>    
4. #include <arpa/inet.h>    
5. #include <time.h>    
6. #include <string.h>    
7. #include <stdio.h>    
8. #include <unistd.h>    
9. #include <stdlib.h>    
10. #define HELLO_PORT 12345    
11. #define HELLO_GROUP "225.0.0.37"    
12. int main(int argc, char *argv[])    
13. {    
14.     struct sockaddr_in addr;    
15.     int fd, cnt;    
16.     struct ip_mreq mreq;    
17.     char *message="Hello, World!";    
18.     /* create what looks like an ordinary UDP socket */    
19.     if ((fd=socket(AF_INET,SOCK_DGRAM,0)) < 0)     
20.     {    
21.         perror("socket");    
22.         exit(1);    
23.     }    
24.     /* set up destination address */    
25.     memset(&addr,0,sizeof(addr));    
26.     addr.sin_family=AF_INET;    
27.     addr.sin_addr.s_addr=inet_addr(HELLO_GROUP);    
28.     addr.sin_port=htons(HELLO_PORT);    
29.     /* now just sendto() our destination! */    
30.     while (1)    
31.     {    
32.         if (sendto(fd,message, strlen(message), 0, (struct sockaddr *) &addr, sizeof(addr)) < 0)     
33.         {    
34.             perror("sendto");    
35.             exit(1);    
36.         }    
37.         sleep(1);    
38.     }    
39.     return 0;    
40. }    

 

Recver.cpp

[c-sharp]  view plain copy print ?

1. #include <sys/types.h>    
2. #include <sys/socket.h>    
3. #include <netinet/in.h>    
4. #include <arpa/inet.h>    
5. #include <time.h>    
6. #include <string.h>    
7. #include <stdio.h>    
8. #include <unistd.h>    
9. #include <stdlib.h>    
10. #define HELLO_PORT  12345    
11. #define HELLO_GROUP "225.0.0.37"    
12. #define MSGBUFSIZE 256    
13. int main(int argc, char *argv[])    
14. {    
15.     struct sockaddr_in addr;    
16.     int fd, nbytes,addrlen;    
17.     struct ip_mreq mreq;    
18.     char msgbuf[MSGBUFSIZE];    
19.     u_int yes=1; /*** MODIFICATION TO ORIGINAL */    
20.     /* create what looks like an ordinary UDP socket */    
21.     if ((fd=socket(AF_INET,SOCK_DGRAM,0)) < 0)     
22.     {    
23.         perror("socket");    
24.         exit(1);    
25.     }    
26.     /**** MODIFICATION TO ORIGINAL */    
27.     /* allow multiple sockets to use the same PORT number */    
28.     if (setsockopt(fd,SOL_SOCKET,SO_REUSEADDR,&yes,sizeof(yes)) < 0)     
29.     {    
30.         perror("Reusing ADDR failed");    
31.         exit(1);    
32.     }    
33.     /*** END OF MODIFICATION TO ORIGINAL */    
34.     /* set up destination address */    
35.     memset(&addr,0,sizeof(addr));    
36.     addr.sin_family=AF_INET;    
37.     addr.sin_addr.s_addr=htonl(INADDR_ANY); /* N.B.: differs from sender */    
38.     addr.sin_port=htons(HELLO_PORT);    
39.     /* bind to receive address */    
40.     if (bind(fd,(struct sockaddr *) &addr,sizeof(addr)) < 0)    
41.     {    
42.         perror("bind");    
43.         exit(1);    
44.     }    
45.     /* use setsockopt() to request that the kernel join a multicast group */    
46.     mreq.imr_multiaddr.s_addr=inet_addr(HELLO_GROUP);    
47.     mreq.imr_interface.s_addr=htonl(INADDR_ANY);    
48.     if (setsockopt(fd,IPPROTO_IP,IP_ADD_MEMBERSHIP,&mreq,sizeof(mreq)) < 0)     
49.     {    
50.         perror("setsockopt");    
51.         exit(1);    
52.     }    
53.     /* now just enter a read-print loop */    
54.     while (1)     
55.     {    
56.         //ssize_t recvfrom(int s, void *buf, size_t len, int flags, struct sockaddr *from, socklen_t *fromlen);    
57.         addrlen=sizeof(addr);    
58.         if ((nbytes=recvfrom(fd, msgbuf, MSGBUFSIZE, 0, (struct sockaddr *) &addr, (socklen_t *)&addrlen)) < 0)     
59.         {    
60.             perror("recvfrom");    
61.             exit(1);    
62.         }    
63.         puts(msgbuf);    
64.     }    
65.     return 0;    
66. }    

 

Makefile

[cpp]  view plain copy print ?

1. CC = gcc    
2. CXX = g++    
3. CFLAGS = -Wall -pipe -D_DEBUG -DDEBUG -g -O0    
4. LDFLAGS = -lstdc++    
5. RM = /bin/rm -f    
6. MODULE_INC = -I../curl-7.21.3/include -I../boost_1_45_0 -I./    
7. MODULE_LIB = -L../boost_1_45_0/stage/lib    
8. CFLAGS  += $(MODULE_INC)    
9. LDFLAGS += $(MODULE_LIB)    
10. LIBOBJS = Sender.o Recver.o    
11. TARGET = sender recver    
12. all: $(TARGET)    
13. sender: Sender.o    
14.     $(CXX) -o $@ $^ $(LDFLAGS)    
15. recver: Recver.o    
16.     $(CXX) -o $@ $^ $(LDFLAGS)    
17. clean:     
18.     rm -f *.o    
19.     rm -f $(TARGET)    
20. # make rule    
21. %.o : %.c    
22.     $(CC) $(CFLAGS) -c $^ -o $@     
23. %.o : %.cpp    
24.     $(CC) $(CFLAGS) -c $^ -o $@     
25. install:    
26.     cp -f $(TARGET) ../bin/