diff options
Diffstat (limited to 'ACE/ace/Pipe.cpp')
-rw-r--r-- | ACE/ace/Pipe.cpp | 344 |
1 files changed, 344 insertions, 0 deletions
diff --git a/ACE/ace/Pipe.cpp b/ACE/ace/Pipe.cpp new file mode 100644 index 00000000000..3d91ce9ecce --- /dev/null +++ b/ACE/ace/Pipe.cpp @@ -0,0 +1,344 @@ +// $Id$ + +#include "ace/Pipe.h" +#include "ace/SOCK_Acceptor.h" +#include "ace/SOCK_Connector.h" +#include "ace/Log_Msg.h" +#include "ace/OS_NS_sys_socket.h" +#include "ace/OS_Memory.h" + +#if defined (ACE_HAS_STREAM_PIPES) || defined (__QNX__) +# include "ace/OS_NS_unistd.h" +#endif // ACE_HAS_STREAM_PIPES || __QNX__ + +#include "ace/os_include/netinet/os_tcp.h" + +#if !defined (__ACE_INLINE__) +#include "ace/Pipe.inl" +#endif /* __ACE_INLINE__ */ + +ACE_RCSID(ace, Pipe, "$Id$") + +ACE_BEGIN_VERSIONED_NAMESPACE_DECL + +void +ACE_Pipe::dump (void) const +{ +#if defined (ACE_HAS_DUMP) + ACE_TRACE ("ACE_Pipe::dump"); + ACE_DEBUG ((LM_DEBUG, ACE_BEGIN_DUMP, this)); + ACE_DEBUG ((LM_DEBUG, ACE_LIB_TEXT ("handles_[0] = %d"), this->handles_[0])); + ACE_DEBUG ((LM_DEBUG, ACE_LIB_TEXT ("\nhandles_[1] = %d"), this->handles_[1])); + ACE_DEBUG ((LM_DEBUG, ACE_LIB_TEXT ("\n"))); + ACE_DEBUG ((LM_DEBUG, ACE_END_DUMP)); +#endif /* ACE_HAS_DUMP */ +} + +int +ACE_Pipe::open (int buffer_size) +{ + ACE_TRACE ("ACE_Pipe::open"); + +#if defined (ACE_LACKS_SOCKETPAIR) || defined (__Lynx__) + ACE_INET_Addr my_addr; + ACE_SOCK_Acceptor acceptor; + ACE_SOCK_Connector connector; + ACE_SOCK_Stream reader; + ACE_SOCK_Stream writer; + int result = 0; +# if defined (ACE_WIN32) + ACE_INET_Addr local_any (static_cast<u_short> (0), ACE_LOCALHOST); +# else + ACE_Addr local_any = ACE_Addr::sap_any; +# endif /* ACE_WIN32 */ + + // Bind listener to any port and then find out what the port was. + if (acceptor.open (local_any) == -1 + || acceptor.get_local_addr (my_addr) == -1) + result = -1; + else + { + ACE_INET_Addr sv_addr (my_addr.get_port_number (), + ACE_LOCALHOST); + + // Establish a connection within the same process. + if (connector.connect (writer, sv_addr) == -1) + result = -1; + else if (acceptor.accept (reader) == -1) + { + writer.close (); + result = -1; + } + } + + // Close down the acceptor endpoint since we don't need it anymore. + acceptor.close (); + if (result == -1) + return -1; + + this->handles_[0] = reader.get_handle (); + this->handles_[1] = writer.get_handle (); + +# if !defined (ACE_LACKS_TCP_NODELAY) + int one = 1; + + // Make sure that the TCP stack doesn't try to buffer small writes. + // Since this communication is purely local to the host it doesn't + // affect network performance. + + if (writer.set_option (ACE_IPPROTO_TCP, + TCP_NODELAY, + &one, + sizeof one) == -1) + { + this->close (); + return -1; + } +# endif /* ! ACE_LACKS_TCP_NODELAY */ + +# if defined (ACE_LACKS_SOCKET_BUFSIZ) + ACE_UNUSED_ARG (buffer_size); +# else /* ! ACE_LACKS_SOCKET_BUFSIZ */ + if (reader.set_option (SOL_SOCKET, + SO_RCVBUF, + reinterpret_cast <void *> (&buffer_size), + sizeof (buffer_size)) == -1 + && errno != ENOTSUP) + { + this->close (); + return -1; + } + else if (writer.set_option (SOL_SOCKET, + SO_SNDBUF, + reinterpret_cast <void *> (&buffer_size), + sizeof (buffer_size)) == -1 + && errno != ENOTSUP) + { + this->close (); + return -1; + } +# endif /* ! ACE_LACKS_SOCKET_BUFSIZ */ + +#elif defined (ACE_HAS_STREAM_PIPES) || defined (__QNX__) + ACE_UNUSED_ARG (buffer_size); + if (ACE_OS::pipe (this->handles_) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + ACE_LIB_TEXT ("%p\n"), + ACE_LIB_TEXT ("pipe")), + -1); + +#if !defined(__QNX__) + int arg = RMSGN; + + // Enable "msg no discard" mode, which ensures that record + // boundaries are maintained when messages are sent and received. + if (ACE_OS::ioctl (this->handles_[0], + I_SRDOPT, + (void *) arg) == -1 + || ACE_OS::ioctl (this->handles_[1], + I_SRDOPT, + (void *) arg) == -1) + { + this->close (); + ACE_ERROR_RETURN ((LM_ERROR, + ACE_LIB_TEXT ("%p\n"), + ACE_LIB_TEXT ("ioctl")), -1); + } +#endif /* __QNX__ */ + +#else /* ! ACE_LACKS_SOCKETPAIR && ! ACE_HAS_STREAM_PIPES */ + if (ACE_OS::socketpair (AF_UNIX, + SOCK_STREAM, + 0, + this->handles_) == -1) + ACE_ERROR_RETURN ((LM_ERROR, + ACE_LIB_TEXT ("%p\n"), + ACE_LIB_TEXT ("socketpair")), + -1); +# if defined (ACE_LACKS_SOCKET_BUFSIZ) + ACE_UNUSED_ARG (buffer_size); +# else /* ! ACE_LACKS_SOCKET_BUFSIZ */ + if (ACE_OS::setsockopt (this->handles_[0], + SOL_SOCKET, + SO_RCVBUF, + reinterpret_cast <const char *> (&buffer_size), + sizeof (buffer_size)) == -1 + && errno != ENOTSUP) + { + this->close (); + return -1; + } + if (ACE_OS::setsockopt (this->handles_[1], + SOL_SOCKET, + SO_SNDBUF, + reinterpret_cast <const char *> (&buffer_size), + sizeof (buffer_size)) == -1 + && errno != ENOTSUP) + { + this->close (); + return -1; + } +# endif /* ! ACE_LACKS_SOCKET_BUFSIZ */ +#endif /* ! ACE_LACKS_SOCKETPAIR && ! ACE_HAS_STREAM_PIPES */ + // Point both the read and write HANDLES to the appropriate socket + // HANDLEs. + + return 0; +} + +int +ACE_Pipe::open (ACE_HANDLE handles[2]) +{ + ACE_TRACE ("ACE_Pipe::open"); + + if (this->open () == -1) + return -1; + else + { + handles[0] = this->handles_[0]; + handles[1] = this->handles_[1]; + return 0; + } +} + +// Do nothing... + +ACE_Pipe::ACE_Pipe (void) +{ + ACE_TRACE ("ACE_Pipe::ACE_Pipe"); + + this->handles_[0] = ACE_INVALID_HANDLE; + this->handles_[1] = ACE_INVALID_HANDLE; +} + +ACE_Pipe::ACE_Pipe (ACE_HANDLE handles[2]) +{ + ACE_TRACE ("ACE_Pipe::ACE_Pipe"); + + if (this->open (handles) == -1) + ACE_ERROR ((LM_ERROR, + ACE_LIB_TEXT ("ACE_Pipe::ACE_Pipe"))); +} + +ACE_Pipe::ACE_Pipe (ACE_HANDLE read, + ACE_HANDLE write) +{ + ACE_TRACE ("ACE_Pipe::ACE_Pipe"); + this->handles_[0] = read; + this->handles_[1] = write; +} + +int +ACE_Pipe::close (void) +{ + ACE_TRACE ("ACE_Pipe::close"); + + int result = 0; + + // Note that the following will work even if we aren't closing down + // sockets because <ACE_OS::closesocket> will just call <::close> in + // that case! + + if (this->handles_[0] != ACE_INVALID_HANDLE) + result = ACE_OS::closesocket (this->handles_[0]); + this->handles_[0] = ACE_INVALID_HANDLE; + + if (this->handles_[1] != ACE_INVALID_HANDLE) + result |= ACE_OS::closesocket (this->handles_[1]); + this->handles_[1] = ACE_INVALID_HANDLE; + + return result; +} + +// Send N char *ptrs and int lengths. Note that the char *'s precede +// the ints (basically, an varargs version of writev). The count N is +// the *total* number of trailing arguments, *not* a couple of the +// number of tuple pairs! + +ssize_t +ACE_Pipe::send (size_t n, ...) const +{ + ACE_TRACE ("ACE_Pipe::send"); + va_list argp; + size_t total_tuples = n / 2; + iovec *iovp; +#if defined (ACE_HAS_ALLOCA) + iovp = (iovec *) alloca (total_tuples * sizeof (iovec)); +#else + ACE_NEW_RETURN (iovp, + iovec[total_tuples], + -1); +#endif /* !defined (ACE_HAS_ALLOCA) */ + + va_start (argp, n); + + for (size_t i = 0; i < total_tuples; i++) + { + iovp[i].iov_base = va_arg (argp, char *); + iovp[i].iov_len = va_arg (argp, int); + } + +#if defined (ACE_WIN32) + ssize_t result = ACE::sendv (this->write_handle (), + iovp, + total_tuples); +#else + ssize_t result = ACE_OS::writev (this->write_handle (), + iovp, + total_tuples); +#endif /* ACE_WIN32 */ + +#if !defined (ACE_HAS_ALLOCA) + delete [] iovp; +#endif /* !defined (ACE_HAS_ALLOCA) */ + va_end (argp); + return result; +} + +// This is basically an interface to ACE_OS::readv, that doesn't use +// the struct iovec explicitly. The ... can be passed as an arbitrary +// number of (char *ptr, int len) tuples. However, the count N is the +// *total* number of trailing arguments, *not* a couple of the number +// of tuple pairs! + +ssize_t +ACE_Pipe::recv (size_t n, ...) const +{ + ACE_TRACE ("ACE_Pipe::recv"); + va_list argp; + size_t total_tuples = n / 2; + iovec *iovp; +#if defined (ACE_HAS_ALLOCA) + iovp = (iovec *) alloca (total_tuples * sizeof (iovec)); +#else + ACE_NEW_RETURN (iovp, + iovec[total_tuples], + -1); +#endif /* !defined (ACE_HAS_ALLOCA) */ + + va_start (argp, n); + + for (size_t i = 0; i < total_tuples; i++) + { + iovp[i].iov_base = va_arg (argp, char *); + iovp[i].iov_len = va_arg (argp, int); + } + +#if defined (ACE_WIN32) + ssize_t result = ACE::recvv (this->read_handle (), + iovp, + total_tuples); +#else + ssize_t result = ACE_OS::readv (this->read_handle (), + iovp, + total_tuples); +#endif /* ACE_WIN32 */ + +#if !defined (ACE_HAS_ALLOCA) + delete [] iovp; +#endif /* !defined (ACE_HAS_ALLOCA) */ + va_end (argp); + return result; +} + +ACE_END_VERSIONED_NAMESPACE_DECL |