ACE/ace/TLI_Connector.cpp

   1 #include "ace/TLI_Connector.h"
   2
   3
   4
   5 #if defined (ACE_HAS_TLI)
   6
   7 #if !defined (__ACE_INLINE__)
   8 #include "ace/TLI_Connector.inl"
   9 #endif /* __ACE_INLINE__ */
  10
  11 #include "ace/Handle_Set.h"
  12 #include "ace/ACE.h"
  13 #include "ace/OS_NS_string.h"
  14 #include "ace/Time_Value.h"
  15
  16 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  17
  18 ACE_ALLOC_HOOK_DEFINE(ACE_TLI_Connector)
  19
  20 void
  21 ACE_TLI_Connector::dump (void) const
  22 {
  23 #if defined (ACE_HAS_DUMP)
  24   ACE_TRACE ("ACE_TLI_Connector::dump");
  25 #endif /* ACE_HAS_DUMP */
  26 }
  27
  28 ACE_TLI_Connector::ACE_TLI_Connector (void)
  29 {
  30   ACE_TRACE ("ACE_TLI_Connector::ACE_TLI_Connector");
  31 }
  32
  33 // Connect the <new_stream> to the <remote_sap>, waiting up to
  34 // <timeout> amount of time if necessary.  It's amazing how
  35 // complicated this is to do in TLI...
  36
  37 int
  38 ACE_TLI_Connector::connect (ACE_TLI_Stream &new_stream,
  39                             const ACE_Addr &remote_sap,
  40                             ACE_Time_Value *timeout,
  41                             const ACE_Addr &local_sap,
  42                             int reuse_addr,
  43                             int flags,
  44                             int /* perms */,
  45                             const char device[],
  46                             struct t_info *info,
  47                             int rwf,
  48                             struct netbuf *udata,
  49                             struct netbuf *opt)
  50 {
  51   ACE_TRACE ("ACE_TLI_Connector::connect");
  52   int result = 0;
  53
  54   // Only open a new endpoint if we don't already have a valid handle.
  55
  56   if (new_stream.get_handle () == ACE_INVALID_HANDLE
  57       && new_stream.open (device, flags, info) == ACE_INVALID_HANDLE)
  58     return -1;
  59
  60   if (local_sap != ACE_Addr::sap_any)
  61     {
  62       // Bind the local endpoint to a specific addr.
  63
  64       struct t_bind *localaddr;
  65
  66       localaddr = (struct t_bind *)
  67         ACE_OS::t_alloc (new_stream.get_handle (), T_BIND, T_ADDR);
  68
  69       if (localaddr == 0)
  70         result = -1;
  71       else
  72         {
  73           int one = 1;
  74 #if !defined (ACE_HAS_FORE_ATM_XTI)
  75           // Reusing the address causes problems with FORE's API. The
  76           // issue may be that t_optmgmt isn't fully supported by
  77           // FORE. t_errno is TBADOPT after the t_optmgmt call so
  78           // maybe options are configured differently for XTI than for
  79           // TLI (at least for FORE's implementation - XTI is supposed
  80           // to be a superset of TLI).
  81           if (reuse_addr
  82               && new_stream.set_option (SOL_SOCKET,
  83                                         SO_REUSEADDR,
  84                                         &one,
  85                                         sizeof one) == -1)
  86             result = -1;
  87           else
  88 #endif /* ACE_HAS_FORE_ATM_XTI */
  89             {
  90               void *addr_buf = local_sap.get_addr ();
  91               localaddr->addr.len = local_sap.get_size ();
  92               ACE_OS::memcpy(localaddr->addr.buf,
  93                              addr_buf,
  94                              localaddr->addr.len);
  95
  96               if (ACE_OS::t_bind (new_stream.get_handle (),
  97                                   localaddr,
  98                                   localaddr) == -1)
  99                 result = -1;
 100
 101               ACE_OS::t_free ((char *) localaddr,
 102                               T_BIND);
 103             }
 104         }
 105
 106       if (result == -1)
 107         {
 108           new_stream.close ();
 109           return -1;
 110         }
 111     }
 112   // Let TLI select the local endpoint addr.
 113   else if (ACE_OS::t_bind (new_stream.get_handle (), 0, 0) == -1)
 114     return -1;
 115
 116   struct t_call *callptr = 0;
 117
 118   callptr = (struct t_call *)
 119     ACE_OS::t_alloc (new_stream.get_handle (), T_CALL, T_ADDR);
 120
 121   if (callptr == 0)
 122     {
 123       new_stream.close ();
 124       return -1;
 125     }
 126
 127   void *addr_buf = remote_sap.get_addr ();
 128   callptr->addr.len = remote_sap.get_size ();
 129   ACE_OS::memcpy (callptr->addr.buf,
 130                   addr_buf,
 131                   callptr->addr.len);
 132   //callptr->addr.buf = (char *) remote_sap.get_addr ();
 133
 134   if (udata != 0)
 135     ACE_OS::memcpy ((void *) &callptr->udata, (void *) udata, sizeof *udata);
 136   if (opt != 0)
 137     ACE_OS::memcpy ((void *) &callptr->opt, (void *) opt, sizeof *opt);
 138
 139   // Connect to remote endpoint.
 140 #if defined (ACE_HAS_FORE_ATM_XTI)
 141   // FORE's XTI/ATM driver has problems with ioctl/fcntl calls so (at least
 142   // for now) always have blocking calls.
 143   timeout = 0;
 144 #endif /* ACE_HAS_FORE_ATM_XTI */
 145
 146   if (timeout != 0)   // Enable non-blocking, if required.
 147     {
 148       if (new_stream.enable (ACE_NONBLOCK) == -1)
 149         result = -1;
 150
 151       // Do a non-blocking connect.
 152       if (ACE_OS::t_connect (new_stream.get_handle (), callptr, 0) == -1)
 153         {
 154           result = -1;
 155
 156           // Check to see if we simply haven't connected yet on a
 157           // non-blocking handle or whether there's really an error.
 158           if (t_errno == TNODATA)
 159             {
 160               if (*timeout == ACE_Time_Value::zero)
 161                 errno = EWOULDBLOCK;
 162               else
 163                 result = this->complete (new_stream, 0, timeout);
 164             }
 165           else if (t_errno == TLOOK && new_stream.look () == T_DISCONNECT)
 166             new_stream.rcvdis ();
 167         }
 168     }
 169   // Do a blocking connect to the server.
 170   else if (ACE_OS::t_connect (new_stream.get_handle (), callptr, 0) == -1)
 171     result = -1;
 172
 173   if (result != -1)
 174     {
 175       new_stream.set_rwflag (rwf);
 176 #if defined (I_PUSH) && !defined (ACE_HAS_FORE_ATM_XTI)
 177       if (new_stream.get_rwflag ())
 178         result = ACE_OS::ioctl (new_stream.get_handle (),
 179                                 I_PUSH,
 180                                 const_cast<char *> ("tirdwr"));
 181 #endif /* I_PUSH */
 182     }
 183   else if (!(errno == EWOULDBLOCK || errno == ETIME))
 184     {
 185       // If things have gone wrong, close down and return an error.
 186       new_stream.close ();
 187       new_stream.set_handle (ACE_INVALID_HANDLE);
 188     }
 189
 190   if (ACE_OS::t_free ((char *) callptr, T_CALL) == -1)
 191     return -1;
 192   return result;
 193 }
 194
 195 // Try to complete a non-blocking connection.
 196
 197 int
 198 ACE_TLI_Connector::complete (ACE_TLI_Stream &new_stream,
 199                              ACE_Addr *remote_sap,
 200                              ACE_Time_Value *tv)
 201 {
 202   ACE_TRACE ("ACE_TLI_Connector::complete");
 203 #if defined (ACE_WIN32)
 204    if (WaitForSingleObject (new_stream.get_handle(), tv->msec()) == WAIT_OBJECT_0)
 205     {
 206       if (ACE_OS::t_look (new_stream.get_handle()) == T_CONNECT)
 207         return t_rcvconnect (new_stream.get_handle(), 0);
 208       else
 209         return -1;
 210     }
 211   else
 212     return -1;
 213 #else
 214   ACE_HANDLE h = ACE::handle_timed_complete (new_stream.get_handle (),
 215                                              tv,
 216                                              1);
 217   if (h == ACE_INVALID_HANDLE)
 218     {
 219       new_stream.close ();
 220       return -1;
 221     }
 222   else    // We've successfully connected!
 223     {
 224       if (remote_sap != 0)
 225         {
 226 #if defined (ACE_HAS_XTI) || defined (ACE_HAS_SVR4_TLI)
 227           struct netbuf name;
 228
 229           name.maxlen = remote_sap->get_size ();
 230           name.buf    = (char *) remote_sap->get_addr ();
 231
 232           if (ACE_OS::t_getname (new_stream.get_handle (),
 233                                  &name,
 234                                  REMOTENAME) == -1)
 235 #else /* SunOS4 */
 236           if (0)
 237 #endif /* ACE_HAS_XTI || ACE_HAS_SVR4_TLI */
 238             {
 239               new_stream.close ();
 240               return -1;
 241             }
 242         }
 243
 244       // Start out with non-blocking disabled on the <new_stream>.
 245       new_stream.disable (ACE_NONBLOCK);
 246
 247       return 0;
 248     }
 249 #endif /* ACE_WIN32 */
 250 }
 251
 252 ACE_END_VERSIONED_NAMESPACE_DECL
 253
 254 #endif /* ACE_HAS_TLI */