summaryrefslogtreecommitdiff
path: root/TAO/orbsvcs/orbsvcs/IOR_Multicast.cpp
blob: 898b98e781ec1a5597da8ff1dd8310046ea65139 (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
121
122
123
124
125
// $Id$

#include "IOR_Multicast.h"

ACE_RCSID(orbsvcs, IOR_Multicast, "$Id$")

ACE_HANDLE
TAO_IOR_Multicast::get_handle (void) const
{
  return this->mcast_dgram_.get_handle ();
}

TAO_IOR_Multicast::TAO_IOR_Multicast (void)
  : service_id_ ((TAO_Service_ID) 0),
    ior_ (0)    
{
}  

TAO_IOR_Multicast::TAO_IOR_Multicast (const char *ior,
                                      u_short port,
                                      const char *mcast_addr,
                                      TAO_Service_ID service_id)
{
  if (this->init (ior, port, mcast_addr, service_id) == -1)
    ACE_ERROR ((LM_ERROR, ASYS_TEXT ("%p\n"), ASYS_TEXT ("TAO_IOR_Multicast")));
}

// destructor

TAO_IOR_Multicast::~TAO_IOR_Multicast (void)
{
  this->mcast_dgram_.unsubscribe ();
}

int
TAO_IOR_Multicast::init (const char *ior,
                         u_short port,
                         const char *mcast_addr,
                         TAO_Service_ID service_id)
{
  this->service_id_ = service_id;
  this->mcast_addr_.set (port, mcast_addr);
  this->ior_ = ior;
  this->response_addr_.set ((u_short) 0);
  this->response_.open (this->response_addr_);

  // Use ACE_SOCK_Dgram_Mcast factory to subscribe to multicast group.
  if (this->mcast_dgram_.subscribe (this->mcast_addr_) == -1)
    ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "subscribe"), -1);

  return 0;
}

int
TAO_IOR_Multicast::handle_timeout (const ACE_Time_Value &,
                               const void *)
{
  return 0;
}

int
TAO_IOR_Multicast::handle_input (ACE_HANDLE)
{  
  struct
  {
    u_short reply_port;
    CORBA::Short service_id;
  } mcast_info;


  ACE_INET_Addr remote_addr;
  ssize_t retcode =
    this->mcast_dgram_.recv (&mcast_info,
                             sizeof (mcast_info),
                             remote_addr);

  if (retcode == -1)
    return -1;

  if (TAO_debug_level > 0)
    ACE_DEBUG ((LM_DEBUG, "(%P|%t) Received multicast.\n"));

  // @@ validate data string received is from a valid client here
  // @@ Probably not needed

  if (retcode != sizeof (mcast_info))
    ACE_ERROR_RETURN ((LM_ERROR,
                       "Reply to multicast not sent. Received %d bytes, expected %d.",
                       retcode,
                       sizeof (mcast_info)),
                      -1);

  // Confirm that we were meant to respond to this request.
  mcast_info.service_id = ntohs (mcast_info.service_id);
  if (mcast_info.service_id == this->service_id_)
    {
      // Convert port number received to network byte order and set port
      // number to reply;
      mcast_info.reply_port = ntohs (mcast_info.reply_port);
      remote_addr.set_port_number (mcast_info.reply_port);

      // send the object reference for the naming service
      retcode = response_.send (this->ior_,
                                ACE_OS::strlen (this->ior_) + 1,
                                remote_addr,
                                0);
      if (TAO_debug_level > 0)
	ACE_DEBUG ((LM_DEBUG,
		    "(%P|%t) ior_: <%s>\n"
		    " sent through port %u.\n"
		    "retcode=%d\n",
		    this->ior_,
		    remote_addr.get_port_number (),
		    retcode));

      if (retcode == -1)
        return -1;
    }
  else
    {
      ACE_DEBUG ((LM_DEBUG, "(%P|%t) Multicast was not for us."));
    }

  return 0;
}