363 lines
9.7 KiB
C++
363 lines
9.7 KiB
C++
|
#include "ace/Pipe.h"
|
||
|
|
#include "ace/SOCK_Acceptor.h"
|
||
|
|
#include "ace/SOCK_Connector.h"
|
||
|
|
#include "ace/Log_Category.h"
|
||
|
|
#include "ace/OS_NS_sys_socket.h"
|
||
|
|
#include "ace/OS_Memory.h"
|
||
|
|
#include "ace/Truncate.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_BEGIN_VERSIONED_NAMESPACE_DECL
|
||
|
|
|
||
|
|
void
|
||
|
|
ACE_Pipe::dump (void) const
|
||
|
|
{
|
||
|
|
#if defined (ACE_HAS_DUMP)
|
||
|
|
ACE_TRACE ("ACE_Pipe::dump");
|
||
|
|
ACELIB_DEBUG ((LM_DEBUG, ACE_BEGIN_DUMP, this));
|
||
|
|
ACELIB_DEBUG ((LM_DEBUG, ACE_TEXT ("handles_[0] = %d"), this->handles_[0]));
|
||
|
|
ACELIB_DEBUG ((LM_DEBUG, ACE_TEXT ("\nhandles_[1] = %d\n"), this->handles_[1]));
|
||
|
|
ACELIB_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)
|
||
|
|
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_SO_RCVBUF) && defined (ACE_LACKS_SO_SNDBUF)
|
||
|
|
ACE_UNUSED_ARG (buffer_size);
|
||
|
|
# endif
|
||
|
|
# if !defined (ACE_LACKS_SO_RCVBUF)
|
||
|
|
if (reader.set_option (SOL_SOCKET,
|
||
|
|
SO_RCVBUF,
|
||
|
|
reinterpret_cast <void *> (&buffer_size),
|
||
|
|
sizeof (buffer_size)) == -1
|
||
|
|
&& errno != ENOTSUP)
|
||
|
|
{
|
||
|
|
this->close ();
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
# endif /* !ACE_LACKS_SO_RCVBUF */
|
||
|
|
# if !defined (ACE_LACKS_SO_SNDBUF)
|
||
|
|
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_SO_SNDBUF */
|
||
|
|
|
||
|
|
#elif defined (ACE_HAS_STREAM_PIPES) || defined (__QNX__)
|
||
|
|
ACE_UNUSED_ARG (buffer_size);
|
||
|
|
if (ACE_OS::pipe (this->handles_) == -1)
|
||
|
|
ACELIB_ERROR_RETURN ((LM_ERROR,
|
||
|
|
ACE_TEXT ("%p\n"),
|
||
|
|
ACE_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 ();
|
||
|
|
ACELIB_ERROR_RETURN ((LM_ERROR,
|
||
|
|
ACE_TEXT ("%p\n"),
|
||
|
|
ACE_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)
|
||
|
|
ACELIB_ERROR_RETURN ((LM_ERROR,
|
||
|
|
ACE_TEXT ("%p\n"),
|
||
|
|
ACE_TEXT ("socketpair")),
|
||
|
|
-1);
|
||
|
|
# if defined (ACE_LACKS_SO_SNDBUF) && defined (ACE_LACKS_SO_RCVBUF)
|
||
|
|
ACE_UNUSED_ARG (buffer_size);
|
||
|
|
# endif
|
||
|
|
# if !defined (ACE_LACKS_SO_RCVBUF)
|
||
|
|
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;
|
||
|
|
}
|
||
|
|
# endif
|
||
|
|
# if !defined (ACE_LACKS_SO_SNDBUF)
|
||
|
|
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_SO_SNDBUF */
|
||
|
|
# if defined (ACE_OPENVMS) && !defined (ACE_LACKS_TCP_NODELAY)
|
||
|
|
int one = 1;
|
||
|
|
// OpenVMS implements socketpair(AF_UNIX...) by returning AF_INET sockets.
|
||
|
|
// Since these are plagued by Nagle as any other INET socket we need to set
|
||
|
|
// TCP_NODELAY on the write handle.
|
||
|
|
if (ACE_OS::setsockopt (this->handles_[1],
|
||
|
|
ACE_IPPROTO_TCP,
|
||
|
|
TCP_NODELAY,
|
||
|
|
reinterpret_cast <const char *> (&one),
|
||
|
|
sizeof (one)) == -1)
|
||
|
|
{
|
||
|
|
this->close ();
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
# endif /* ACE_OPENVMS && !ACE_LACKS_TCP_NODELAY */
|
||
|
|
#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)
|
||
|
|
ACELIB_ERROR ((LM_ERROR,
|
||
|
|
ACE_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 = this->close_read ();
|
||
|
|
result |= this->close_write ();
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
ACE_Pipe::close_read (void)
|
||
|
|
{
|
||
|
|
return this->close_handle (0);
|
||
|
|
}
|
||
|
|
|
||
|
|
int ACE_Pipe::close_write (void)
|
||
|
|
{
|
||
|
|
return this->close_handle (1);
|
||
|
|
}
|
||
|
|
|
||
|
|
// 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;
|
||
|
|
int total_tuples = ACE_Utils::truncate_cast<int> (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 (int 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;
|
||
|
|
int total_tuples = ACE_Utils::truncate_cast<int> (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 (int 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 const result = ACE::recvv (this->read_handle (),
|
||
|
|
iovp,
|
||
|
|
total_tuples);
|
||
|
|
#else
|
||
|
|
ssize_t const 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
|