ACE/ace/FIFO_Recv_Msg.inl

   1 // -*- C++ -*-
   2 #include "ace/Min_Max.h"
   3 #include "ace/OS_NS_stropts.h"
   4 #include "ace/Truncate.h"
   5
   6 #if !defined (ACE_HAS_STREAM_PIPES)
   7 #include "ace/OS_NS_unistd.h"
   8 #endif
   9
  10 ACE_BEGIN_VERSIONED_NAMESPACE_DECL
  11
  12 ACE_INLINE ssize_t
  13 ACE_FIFO_Recv_Msg::recv (ACE_Str_Buf &recv_msg)
  14 {
  15   ACE_TRACE ("ACE_FIFO_Recv_Msg::recv");
  16 #if defined (ACE_HAS_STREAM_PIPES)
  17   int i = 0;
  18   if (ACE_OS::getmsg (this->get_handle (),
  19                       (strbuf *) 0,
  20                       (strbuf *) &recv_msg,
  21                       &i) == -1)
  22     {
  23       return -1;
  24     }
  25   else
  26     {
  27       return recv_msg.len;
  28     }
  29 #else /* Do the ol' 2-read trick... */
  30   if (ACE_OS::read (this->get_handle (),
  31                     (char *) &recv_msg.len,
  32                     sizeof recv_msg.len) != (ssize_t) sizeof recv_msg.len)
  33     {
  34       return -1;
  35     }
  36   else
  37     {
  38       size_t remaining = static_cast<size_t> (recv_msg.len);
  39       size_t requested = static_cast<size_t> (recv_msg.maxlen);
  40       ssize_t recv_len = ACE_OS::read (this->get_handle (),
  41                                        (char *) recv_msg.buf,
  42                                        ACE_MIN (remaining, requested));
  43
  44       if (recv_len == -1)
  45         {
  46           return -1;
  47         }
  48
  49       // Tell caller what's really in the buffer.
  50       recv_msg.len = static_cast<int> (recv_len);
  51
  52       // If there are more bytes remaining in the message, read them and
  53       // throw them away. Leaving them in the FIFO would make it difficult
  54       // to find the start of the next message in the fifo.
  55       // Since the ACE_HAS_STREAM_PIPES version of this method doesn't
  56       // return getmsg()'s indication of "data remaining", don't worry about
  57       // saving the indication here either to read the remainder later.
  58       size_t total_msg_size = remaining;
  59       remaining -= recv_len;
  60
  61       while (remaining > 0)
  62         {
  63           const size_t throw_away = 1024;
  64           char dev_null[throw_away];
  65           recv_len = ACE_OS::read (this->get_handle (),
  66                                    dev_null,
  67                                    ACE_MIN (remaining, throw_away));
  68
  69           if (recv_len == -1)
  70             {
  71               break;
  72             }
  73
  74           remaining -= recv_len;
  75         }
  76
  77       return ACE_Utils::truncate_cast<ssize_t> (total_msg_size);
  78     }
  79 #endif /* ACE_HAS_STREAM_PIPES */
  80 }
  81
  82 ACE_INLINE ssize_t
  83 ACE_FIFO_Recv_Msg::recv (void *buf, size_t max_len)
  84 {
  85   ACE_TRACE ("ACE_FIFO_Recv_Msg::recv");
  86   ACE_Str_Buf recv_msg ((char *) buf, 0, static_cast<int> (max_len));
  87
  88   return this->recv (recv_msg);
  89 }
  90
  91 #if defined (ACE_HAS_STREAM_PIPES)
  92 ACE_INLINE ssize_t
  93 ACE_FIFO_Recv_Msg::recv (ACE_Str_Buf *data,
  94                          ACE_Str_Buf *cntl,
  95                          int *flags)
  96 {
  97   ACE_TRACE ("ACE_FIFO_Recv_Msg::recv");
  98   if (ACE_OS::getmsg (this->get_handle (),
  99                       (strbuf *) cntl,
 100                       (strbuf *) data,
 101                       flags) == -1)
 102     {
 103       return -1;
 104     }
 105   else
 106     {
 107       return (cntl == 0 ? 0 : cntl->len) + (data == 0 ? 0 : data->len);
 108     }
 109 }
 110
 111 ACE_INLINE ssize_t
 112 ACE_FIFO_Recv_Msg::recv (int *band,
 113                          ACE_Str_Buf *data,
 114                          ACE_Str_Buf *cntl,
 115                          int *flags)
 116 {
 117   ACE_TRACE ("ACE_FIFO_Recv_Msg::recv");
 118
 119   if (ACE_OS::getpmsg (this->get_handle (),
 120                        (strbuf *) cntl,
 121                        (strbuf *) data,
 122                        band,
 123                        flags) == -1)
 124     {
 125       return -1;
 126     }
 127   else
 128     {
 129       return (cntl == 0 ? 0 : cntl->len) + (data == 0 ? 0 : data->len);
 130     }
 131 }
 132 #endif /* ACE_HAS_STREAM_PIPES */
 133
 134 ACE_END_VERSIONED_NAMESPACE_DECL