dep/ACE_wrappers/ace/ICMP_Socket.cpp

   1 // $Id: ICMP_Socket.cpp 80826 2008-03-04 14:51:23Z wotte $
   2
   3 #include "ace/ICMP_Socket.h"
   4
   5 #if defined (ACE_HAS_ICMP_SUPPORT) && (ACE_HAS_ICMP_SUPPORT == 1)
   6
   7 #include "ace/ACE.h"
   8 #include "ace/Log_Msg.h"
   9 #include "ace/OS_NS_netdb.h"
  10 #include "ace/OS_NS_sys_socket.h"
  11
  12
  13 ACE_RCSID (ace,
  14            ICMP_Socket,
  15            "$Id: ICMP_Socket.cpp 80826 2008-03-04 14:51:23Z wotte $")
  16
  17
  18 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  19
  20 ACE_ALLOC_HOOK_DEFINE (ACE_ICMP_Socket)
  21
  22
  23 void
  24 ACE_ICMP_Socket::dump (void) const
  25 {
  26   ACE_TRACE ("ACE_ICMP_Socket::dump");
  27 }
  28
  29 ACE_ICMP_Socket::ACE_ICMP_Socket (void)
  30 {
  31   ACE_TRACE ("ACE_ICMP_Socket::ACE_ICMP_Socket");
  32 }
  33
  34 ssize_t
  35 ACE_ICMP_Socket::send (void const * buf,
  36                        size_t n,
  37                        ACE_Addr const & addr,
  38                        int flags) const
  39 {
  40   ACE_TRACE ("ACE_ICMP_Socket::send");
  41
  42   return ACE_OS::sendto (this->get_handle (),
  43                          (char const *) buf,
  44                          n,
  45                          flags,
  46                          (sockaddr const *) addr.get_addr (),
  47                          addr.get_size ());
  48 }
  49
  50 ssize_t
  51 ACE_ICMP_Socket::recv (void * buf,
  52                        size_t n,
  53                        ACE_Addr & addr,
  54                        int flags) const
  55 {
  56   ACE_TRACE ("ACE_ICMP_Socket::recv");
  57
  58   int addr_len = addr.get_size ();
  59   ssize_t status = ACE_OS::recvfrom (this->get_handle (),
  60                                      (char *) buf,
  61                                      n,
  62                                      flags,
  63                                      (sockaddr *) addr.get_addr (),
  64                                      (int*) &addr_len);
  65   addr.set_size (addr_len);
  66
  67   return status;
  68 }
  69
  70 ssize_t
  71 ACE_ICMP_Socket::recv (void * buf,
  72                        size_t n,
  73                        int flags,
  74                        ACE_Time_Value const * timeout) const
  75 {
  76   ACE_TRACE ("ACE_ICMP_Socket::recv");
  77
  78   return ACE::recv (this->get_handle (),
  79                     buf,
  80                     n,
  81                     flags,
  82                     timeout);
  83 }
  84
  85 int
  86 ACE_ICMP_Socket::open (ACE_Addr const & local,
  87                        int protocol,
  88                        int reuse_addr)
  89 {
  90   ACE_TRACE ("ACE_ICMP_Socket::open");
  91
  92   // Check if icmp protocol is supported on this host
  93   int proto_number = -1;
  94   protoent *proto;
  95
  96   if (! (proto = getprotobyname ("icmp")))
  97     {
  98       ACE_ERROR_RETURN
  99         ((LM_ERROR,
 100           ACE_TEXT ("(%P|%t) ACE_ICMP_Socket::open: %p; %s\n"),
 101           ACE_TEXT ("getprotobyname"),
 102           ACE_TEXT ("ICMP protocol is not properly configured ")
 103           ACE_TEXT ("or not supported.")),
 104          -1);
 105     }
 106   proto_number = proto->p_proto;
 107
 108   if (proto_number != IPPROTO_ICMP || proto_number != protocol)
 109     {
 110       ACE_ERROR_RETURN ((LM_ERROR,
 111                          ACE_TEXT ("(%P|%t) ACE::ICMP_Socket::open - ")
 112                          ACE_TEXT ("only IPPROTO_ICMP protocol is ")
 113                          ACE_TEXT ("currently supported.\n")),
 114                         -1);
 115     }
 116
 117   if (ACE_SOCK::open (SOCK_RAW,
 118                       AF_INET,
 119                       protocol,
 120                       reuse_addr) == -1)
 121     {
 122       return -1;
 123     }
 124
 125   return this->shared_open (local);
 126 }
 127
 128 int
 129 ACE_ICMP_Socket::shared_open (ACE_Addr const & local)
 130 {
 131   ACE_TRACE ("ACE_ICMP_Socket::shared_open");
 132
 133   int error = 0;
 134   if (local == ACE_Addr::sap_any)
 135     {
 136       if (ACE::bind_port (this->get_handle ()) == -1)
 137         {
 138           error = 1;
 139         }
 140     }
 141   else if (ACE_OS::bind (this->get_handle (),
 142                          reinterpret_cast<sockaddr *> (local.get_addr ()),
 143                          local.get_size ()) == -1)
 144     {
 145       error = 1;
 146     }
 147
 148   if (error != 0)
 149     {
 150       this->close ();
 151     }
 152
 153   return error ? -1 : 0;
 154 }
 155
 156 unsigned short
 157 ACE_ICMP_Socket::calculate_checksum (unsigned short * paddress,
 158                                      int len)
 159 {
 160   int nleft = len;
 161   int sum = 0;
 162   unsigned short * w = paddress;
 163   unsigned short answer = 0;
 164   while (nleft > 1)
 165     {
 166       sum += *w++;
 167       nleft -= 2;
 168     }
 169
 170   if (nleft == 1)
 171     {
 172       *((unsigned char *) &answer) = *((unsigned char *) w);
 173       sum += answer;
 174     }
 175
 176   // add back carry outs from top 16 bits to low 16 bits
 177   sum = (sum >> 16) + (sum & 0xffff); // add hi 16 to low 16
 178   sum += (sum >> 16);                 // add carry
 179   answer = ~sum;                      // truncate to 16 bits
 180
 181   return (answer);
 182 }
 183
 184 ACE_END_VERSIONED_NAMESPACE_DECL
 185
 186 #endif  /* ACE_HAS_ICMP_SUPPORT == 1 */