summaryrefslogtreecommitdiff
path: root/ace/SOCK_Dgram_Mcast.cpp
blob: 756d9ca5183c46b6f0a892384da671a8fce934bb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
// SOCK_Dgram_Mcast.cpp
// $Id$

#define ACE_BUILD_DLL
#include "ace/SOCK_Dgram_Mcast.h"
#include "ace/INET_Addr.h"

ACE_ALLOC_HOOK_DEFINE(ACE_SOCK_Dgram_Mcast)

void
ACE_SOCK_Dgram_Mcast::dump (void) const
{
  ACE_TRACE ("ACE_SOCK_Dgram_Mcast::dump");
}

// Dummy default constructor...

ACE_SOCK_Dgram_Mcast::ACE_SOCK_Dgram_Mcast (void)
{
  ACE_TRACE ("ACE_SOCK_Dgram_Mcast::ACE_SOCK_Dgram_Mcast");
}

int
ACE_SOCK_Dgram_Mcast::subscribe (const ACE_INET_Addr &mcast_addr,
				 int reuse_addr,
				 const char *net_if,
				 int protocol_family,
				 int protocol)
{
  ACE_TRACE ("ACE_SOCK_Dgram_Mcast::subscribe");
  // make a local copy of address to use in sends
  this->mcast_addr_.set (mcast_addr);

  if (ACE_SOCK::open (SOCK_DGRAM, protocol_family, 
		      protocol, reuse_addr) == -1)
    return -1;

  // Create multicast request.
  if (this->make_multicast_address (this->mcast_addr_, net_if) == -1)
    return -1;
  
  int one = 1;
  if (reuse_addr 
      && this->ACE_SOCK::set_option (SOL_SOCKET,
				     SO_REUSEADDR,
				     &one,
				     sizeof one) == -1)
    return -1;
#if defined (SO_REUSEPORT)
  if (this->ACE_SOCK::set_option (SOL_SOCKET,
				  SO_REUSEPORT,
				  &one,
				  sizeof one) == -1)
    return -1;
#endif /* SO_REUSEPORT */

  // Create an address to bind the socket to.
  
  ACE_INET_Addr local;
  
  if (local.set (this->mcast_addr_.get_port_number ()) == -1)
    return -1;
  else if (ACE_SOCK_Dgram::shared_open (local, protocol_family) == -1)
    return -1;

  // Tell network device driver to read datagrams with a
  // multicast_address address.
  else if (this->ACE_SOCK::set_option (IPPROTO_IP, 
				       IP_ADD_MEMBERSHIP,
				       &multicast_address_, 
				       sizeof multicast_address_) == -1)
    return -1;
  return 0;
}

int
ACE_SOCK_Dgram_Mcast::unsubscribe (void)
{
  ACE_TRACE ("ACE_SOCK_Dgram_Mcast::unsubscribe");
  return this->ACE_SOCK::set_option (IPPROTO_IP, 
				     IP_DROP_MEMBERSHIP,
				     &multicast_address_, 
				     sizeof multicast_address_);
}

int
ACE_SOCK_Dgram_Mcast::make_multicast_address (const ACE_INET_Addr &mcast_addr,
					      const char *net_if)
{
  ACE_TRACE ("ACE_SOCK_Dgram_Mcast::make_multicast_address");

  if (net_if != 0)
    {
#if !defined (ACE_WIN32)
      struct ifreq if_address;

      ACE_OS::strcpy (if_address.ifr_name, net_if);

      if (ACE_OS::ioctl (this->get_handle (), SIOCGIFADDR, &if_address) == -1)
	return -1;

      struct sockaddr_in *socket_address;
      socket_address = (sockaddr_in *) &if_address.ifr_addr;
      multicast_address_.imr_interface.s_addr = socket_address->sin_addr.s_addr;
#else
      // This port number is not necessary, just convenient
      ACE_INET_Addr interface_addr;
      if (interface_addr.set (mcast_addr.get_port_number (),
			      net_if) == -1)
	return -1;
      multicast_address_.imr_interface.s_addr = 
	htonl (interface_addr.get_ip_address ());      
#endif /* ACE_WIN32 */      
    }
  else
    multicast_address_.imr_interface.s_addr = INADDR_ANY;
  
  multicast_address_.imr_multiaddr.s_addr = htonl (mcast_addr.get_ip_address ());
  return 0;
}