ACE/ace/IOStream_T.inl

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