dep/ACE_wrappers/ace/IOStream_T.inl

   1 // -*- C++ -*-
   2 //
   3 // $Id: IOStream_T.inl 80826 2008-03-04 14:51:23Z wotte $
   4
   5 #include "ace/Handle_Set.h"
   6 #include "ace/OS_NS_errno.h"
   7 #include "ace/OS_NS_sys_select.h"
   8
   9 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  10
  11 template <class STREAM> ACE_INLINE int
  12 ACE_IOStream<STREAM>::eof (void) const
  13 {
  14   // Get the timeout value of the streambuf
  15   ACE_Time_Value *timeout = this->streambuf_->recv_timeout (0);
  16
  17   // Reset the timeout value of the streambuf.
  18   (void) this->streambuf_->recv_timeout (timeout);
  19
  20   char c;
  21   int rval = this->streambuf_->recv_n (&c,
  22                                        sizeof c,
  23                                        MSG_PEEK,
  24                                        timeout);
  25
  26   // Timeout, not an eof
  27   if (this->streambuf_->timeout())
  28     return 0;
  29
  30   // No timeout, got enough data:  not eof
  31   if (rval == sizeof(char))
  32     return 0;
  33
  34   // No timeout, not enough data:  definately eof
  35   return 1;
  36 }
  37
  38 template <class STREAM> ACE_INLINE
  39 ACE_SOCK_Dgram_SC<STREAM>::ACE_SOCK_Dgram_SC (void)
  40 {
  41 }
  42
  43 template <class STREAM> ACE_INLINE
  44 ACE_SOCK_Dgram_SC<STREAM>::ACE_SOCK_Dgram_SC (STREAM &source,
  45                                               ACE_INET_Addr &dest)
  46   : STREAM (source),
  47     peer_ (dest)
  48 {
  49 }
  50
  51 template <class STREAM> ACE_INLINE ssize_t
  52 ACE_SOCK_Dgram_SC<STREAM>::send_n (char *buf,
  53                                    ssize_t len)
  54 {
  55   return STREAM::send (buf, len, peer_);
  56 }
  57
  58 template <class STREAM> ACE_INLINE ssize_t
  59 ACE_SOCK_Dgram_SC<STREAM>::recv (char *buf,
  60                                  ssize_t len,
  61                                  ACE_Time_Value *tv)
  62 {
  63   //FUZZ: disable check_for_lack_ACE_OS
  64   return recv (buf, len, 0, tv);
  65   //FUZZ: enable check_for_lack_ACE_OS
  66 }
  67
  68 template <class STREAM> ACE_INLINE ssize_t
  69 ACE_SOCK_Dgram_SC<STREAM>::recv (char *buf,
  70                                  ssize_t len,
  71                                  int flags,
  72                                  ACE_Time_Value *tv)
  73 {
  74   if (tv != 0)
  75     {
  76       ACE_HANDLE handle = this->get_handle ();
  77       ACE_Handle_Set handle_set;
  78
  79       handle_set.set_bit (handle);
  80
  81       switch (ACE_OS::select (int (handle) + 1,
  82                               (fd_set *) handle_set, // read_fds.
  83                               (fd_set *) 0,          // write_fds.
  84                               (fd_set *) 0,          // exception_fds.
  85                               tv))
  86         {
  87         case 0:
  88           errno = ETIME;
  89         case -1:
  90           return -1;
  91         default:
  92           ;     // Do the 'recv' below
  93         }
  94     }
  95
  96   int rval = STREAM::recv (buf, len, peer_, flags);
  97 #if defined (ACE_WIN32)
  98   if (rval == SOCKET_ERROR)
  99     if (::WSAGetLastError () == WSAEMSGSIZE)
 100       if (ACE_BIT_ENABLED (flags, MSG_PEEK))
 101         rval = len;
 102 #endif /* ACE_WIN32 */
 103   return rval < len ? rval : len;
 104 }
 105
 106 template <class STREAM> ACE_INLINE ssize_t
 107 ACE_SOCK_Dgram_SC<STREAM>::recv_n (char *buf,
 108                                    ssize_t len,
 109                                    int flags,
 110                                    ACE_Time_Value *tv)
 111 {
 112   int rval = this->recv (buf, len, flags, tv);
 113   return rval;
 114 }
 115
 116 template <class STREAM> ACE_INLINE int
 117 ACE_SOCK_Dgram_SC<STREAM>::get_remote_addr (ACE_INET_Addr &addr) const
 118 {
 119   addr = peer_;
 120   return 0;
 121 }
 122
 123 ACE_END_VERSIONED_NAMESPACE_DECL