aboutsummaryrefslogtreecommitdiff
path: root/nxcomp/Transport.cpp
diff options
context:
space:
mode:
authorReinhard Tartler <siretart@tauware.de>2011-11-13 09:27:51 +0100
committerReinhard Tartler <siretart@tauware.de>2011-11-13 09:27:51 +0100
commit3e924126b56c4d421e8263d25f6b14aa4ceed047 (patch)
tree921403de7b834fb1ac89f97d84ce105e0b487051 /nxcomp/Transport.cpp
parenta840692edc9c6d19cd7c057f68e39c7d95eb767d (diff)
downloadnx-libs-3e924126b56c4d421e8263d25f6b14aa4ceed047.tar.gz
nx-libs-3e924126b56c4d421e8263d25f6b14aa4ceed047.tar.bz2
nx-libs-3e924126b56c4d421e8263d25f6b14aa4ceed047.zip
Imported nxcomp-3.1.0-4.tar.gznxcomp/3.1.0-4
Summary: Imported nxcomp-3.1.0-4.tar.gz Keywords: Imported nxcomp-3.1.0-4.tar.gz into Git repository
Diffstat (limited to 'nxcomp/Transport.cpp')
-rwxr-xr-xnxcomp/Transport.cpp3056
1 files changed, 3056 insertions, 0 deletions
diff --git a/nxcomp/Transport.cpp b/nxcomp/Transport.cpp
new file mode 100755
index 000000000..db57c890e
--- /dev/null
+++ b/nxcomp/Transport.cpp
@@ -0,0 +1,3056 @@
+/**************************************************************************/
+/* */
+/* Copyright (c) 2001, 2007 NoMachine, http://www.nomachine.com/. */
+/* */
+/* NXCOMP, NX protocol compression and NX extensions to this software */
+/* are copyright of NoMachine. Redistribution and use of the present */
+/* software is allowed according to terms specified in the file LICENSE */
+/* which comes in the source distribution. */
+/* */
+/* Check http://www.nomachine.com/licensing.html for applicability. */
+/* */
+/* NX and NoMachine are trademarks of NoMachine S.r.l. */
+/* */
+/* All rights reserved. */
+/* */
+/**************************************************************************/
+
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <signal.h>
+#include <sys/socket.h>
+
+#include "Transport.h"
+
+#include "Statistics.h"
+
+//
+// Set the verbosity level. You also
+// need to define DUMP in Misc.cpp
+// if DUMP is defined here.
+//
+
+#define PANIC
+#define WARNING
+#undef TEST
+#undef DEBUG
+#undef INSPECT
+#undef DUMP
+
+//
+// Used to lock and unlock the transport
+// buffers before they are accessed by
+// different threads.
+//
+
+#undef THREADS
+
+//
+// Define this to get logging all the
+// operations performed by the parent
+// thread, the one that enqueues and
+// dequeues data.
+//
+
+#define PARENT
+
+//
+// Define this to know when a channel
+// is created or destroyed.
+//
+
+#undef REFERENCES
+
+//
+// Reference count for allocated buffers.
+//
+
+#ifdef REFERENCES
+
+int Transport::references_;
+int ProxyTransport::references_;
+int InternalTransport::references_;
+
+#endif
+
+//
+// This is the base class providing methods for read
+// and write buffering.
+//
+
+Transport::Transport(int fd) : fd_(fd)
+{
+ #ifdef TEST
+ *logofs << "Transport: Going to create base transport "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ type_ = transport_base;
+
+ //
+ // Set up the write buffer.
+ //
+
+ w_buffer_.length_ = 0;
+ w_buffer_.start_ = 0;
+
+ initialSize_ = TRANSPORT_BUFFER_DEFAULT_SIZE;
+ thresholdSize_ = TRANSPORT_BUFFER_DEFAULT_SIZE << 1;
+ maximumSize_ = TRANSPORT_BUFFER_DEFAULT_SIZE << 4;
+
+ w_buffer_.data_.resize(initialSize_);
+
+ //
+ // Set non-blocking IO on socket.
+ //
+
+ SetNonBlocking(fd_, 1);
+
+ blocked_ = 0;
+ finish_ = 0;
+
+ #ifdef REFERENCES
+ *logofs << "Transport: Created new object at "
+ << this << " out of " << ++references_
+ << " allocated references.\n" << logofs_flush;
+ #endif
+}
+
+Transport::~Transport()
+{
+ #ifdef TEST
+ *logofs << "Transport: Going to destroy base class "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ ::close(fd_);
+
+ #ifdef REFERENCES
+ *logofs << "Transport: Deleted object at "
+ << this << " out of " << --references_
+ << " allocated references.\n" << logofs_flush;
+ #endif
+}
+
+//
+// Read data from its file descriptor.
+//
+
+int Transport::read(unsigned char *data, unsigned int size)
+{
+ #ifdef DEBUG
+ *logofs << "Transport: Going to read " << size << " bytes from "
+ << "FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ //
+ // Read the available data from the socket.
+ //
+
+ int result = ::read(fd_, data, size);
+
+ //
+ // Update the current timestamp as the read
+ // can have scheduled some other process.
+ //
+
+ getNewTimestamp();
+
+ if (result < 0)
+ {
+ if (EGET() == EAGAIN)
+ {
+ #ifdef TEST
+ *logofs << "Transport: WARNING! Read of " << size << " bytes from "
+ << "FD#" << fd_ << " would block.\n" << logofs_flush;
+ #endif
+
+ return 0;
+ }
+ else if (EGET() == EINTR)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Read of " << size << " bytes from "
+ << "FD#" << fd_ << " was interrupted.\n"
+ << logofs_flush;
+ #endif
+
+ return 0;
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Error reading from "
+ << "FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ finish();
+
+ return -1;
+ }
+ }
+ else if (result == 0)
+ {
+ #ifdef TEST
+ *logofs << "Transport: No data read from "
+ << "FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ finish();
+
+ return -1;
+ }
+
+ #ifdef TEST
+ *logofs << "Transport: Read " << result << " bytes out of "
+ << size << " from FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef DUMP
+
+ *logofs << "Transport: Dumping content of read data.\n"
+ << logofs_flush;
+
+ DumpData(data, result);
+
+ #endif
+
+ return result;
+}
+
+//
+// Write as many bytes as possible to socket.
+// Append the remaining data bytes to the end
+// of the buffer and update length to reflect
+// changes.
+//
+
+int Transport::write(T_write type, const unsigned char *data, const unsigned int size)
+{
+ //
+ // If an immediate write was requested then
+ // flush the enqueued data first.
+ //
+ // Alternatively may try to write only if
+ // the socket is not blocked.
+ //
+ // if (w_buffer_.length_ > 0 && blocked_ == 0 &&
+ // type == write_immediate)
+ // {
+ // ...
+ // }
+ //
+
+ if (w_buffer_.length_ > 0 && type == write_immediate)
+
+ {
+ #ifdef TEST
+ *logofs << "Transport: Writing " << w_buffer_.length_
+ << " bytes of previous data to FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ int result = Transport::flush();
+
+ if (result < 0)
+ {
+ return -1;
+ }
+ }
+
+ //
+ // If nothing is remained, write immediately
+ // to the socket.
+ //
+
+ unsigned int written = 0;
+
+ if (w_buffer_.length_ == 0 && blocked_ == 0 &&
+ type == write_immediate)
+ {
+ //
+ // Limit the amount of data sent.
+ //
+
+ unsigned int toWrite = size;
+
+ #ifdef DUMP
+
+ *logofs << "Transport: Going to write " << toWrite
+ << " bytes to FD#" << fd_ << " with checksum ";
+
+ DumpChecksum(data, size);
+
+ *logofs << ".\n" << logofs_flush;
+
+ #endif
+
+ T_timestamp writeTs;
+
+ int diffTs;
+
+ while (written < toWrite)
+ {
+ //
+ // Trace system time spent writing data.
+ //
+
+ writeTs = getTimestamp();
+
+ int result = ::write(fd_, data + written, toWrite - written);
+
+ diffTs = diffTimestamp(writeTs, getNewTimestamp());
+
+ statistics -> addWriteTime(diffTs);
+
+ if (result <= 0)
+ {
+ if (EGET() == EAGAIN)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write of " << toWrite - written
+ << " bytes on FD#" << fd_ << " would block.\n"
+ << logofs_flush;
+ #endif
+
+ blocked_ = 1;
+
+ break;
+ }
+ else if (EGET() == EINTR)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write of " << toWrite - written
+ << " bytes on FD#" << fd_ << " was interrupted.\n"
+ << logofs_flush;
+ #endif
+
+ continue;
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write to " << "FD#"
+ << fd_ << " failed.\n" << logofs_flush;
+ #endif
+
+ finish();
+
+ return -1;
+ }
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Immediately written " << result
+ << " bytes on " << "FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ written += result;
+ }
+ }
+
+ #ifdef DUMP
+
+ if (written > 0)
+ {
+ *logofs << "Transport: Dumping content of immediately written data.\n"
+ << logofs_flush;
+
+ DumpData(data, written);
+ }
+
+ #endif
+ }
+
+ if (written == size)
+ {
+ //
+ // We will not affect the write buffer.
+ //
+
+ return written;
+ }
+
+ #ifdef DEBUG
+ *logofs << "Transport: Going to append " << size - written
+ << " bytes to write buffer for " << "FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ if (resize(w_buffer_, size - written) < 0)
+ {
+ return -1;
+ }
+
+ memmove(w_buffer_.data_.begin() + w_buffer_.start_ + w_buffer_.length_,
+ data + written, size - written);
+
+ w_buffer_.length_ += size - written;
+
+ #ifdef TEST
+ *logofs << "Transport: Write buffer for FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "Transport: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " size is "
+ << w_buffer_.data_.size() << " capacity is "
+ << w_buffer_.data_.capacity() << ".\n"
+ << logofs_flush;
+ #endif
+
+ //
+ // Note that this function always returns the whole
+ // size of buffer that was provided, either if not
+ // all the data could be actually written.
+ //
+
+ return size;
+}
+
+//
+// Write pending data to its file descriptor.
+//
+
+int Transport::flush()
+{
+ if (w_buffer_.length_ == 0)
+ {
+ #ifdef TEST
+ *logofs << "Transport: No data to flush on "
+ << "FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef WARNING
+ if (blocked_ != 0)
+ {
+ *logofs << "Transport: Blocked flag is " << blocked_
+ << " with no data to flush on FD#" << fd_
+ << ".\n" << logofs_flush;
+ }
+ #endif
+
+ return 0;
+ }
+
+ //
+ // It's time to move data from the
+ // write buffer to the real link.
+ //
+
+ int written = 0;
+
+ int toWrite = w_buffer_.length_;
+
+ //
+ // We will do our best to write any available
+ // data to the socket, so let's say we start
+ // from a clean state.
+ //
+
+ blocked_ = 0;
+
+ #ifdef TEST
+ *logofs << "Transport: Going to flush " << toWrite
+ << " bytes on FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ T_timestamp writeTs;
+
+ int diffTs;
+
+ while (written < toWrite)
+ {
+ writeTs = getTimestamp();
+
+ int result = ::write(fd_, w_buffer_.data_.begin() + w_buffer_.start_ +
+ written, toWrite - written);
+
+ diffTs = diffTimestamp(writeTs, getNewTimestamp());
+
+ statistics -> addWriteTime(diffTs);
+
+ if (result <= 0)
+ {
+ if (EGET() == EAGAIN)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write of " << toWrite - written
+ << " bytes on FD#" << fd_ << " would block.\n"
+ << logofs_flush;
+ #endif
+
+ blocked_ = 1;
+
+ break;
+ }
+ else if (EGET() == EINTR)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write of " << toWrite - written
+ << " bytes on FD#" << fd_ << " was interrupted.\n"
+ << logofs_flush;
+ #endif
+
+ continue;
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write to " << "FD#"
+ << fd_ << " failed.\n" << logofs_flush;
+ #endif
+
+ finish();
+
+ return -1;
+ }
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Flushed " << result << " bytes on "
+ << "FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ written += result;
+ }
+ }
+
+ if (written > 0)
+ {
+ #ifdef DUMP
+
+ *logofs << "Transport: Dumping content of flushed data.\n"
+ << logofs_flush;
+
+ DumpData(w_buffer_.data_.begin() + w_buffer_.start_, written);
+
+ #endif
+
+ //
+ // Update the buffer status.
+ //
+
+ w_buffer_.length_ -= written;
+
+ if (w_buffer_.length_ == 0)
+ {
+ w_buffer_.start_ = 0;
+ }
+ else
+ {
+ w_buffer_.start_ += written;
+ }
+ }
+
+ //
+ // It can be that we wrote less bytes than
+ // available because of the write limit.
+ //
+
+ if (w_buffer_.length_ > 0)
+ {
+ #ifdef TEST
+ *logofs << "Transport: There are still " << w_buffer_.length_
+ << " bytes in write buffer for " << "FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ blocked_ = 1;
+ }
+
+ #ifdef TEST
+ *logofs << "Transport: Write buffer for FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "Transport: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " size is "
+ << w_buffer_.data_.size() << " capacity is "
+ << w_buffer_.data_.capacity() << ".\n"
+ << logofs_flush;
+ #endif
+
+ //
+ // No new data was produced for the link except
+ // any outstanding data from previous writes.
+ //
+
+ return 0;
+}
+
+int Transport::drain(int limit, int timeout)
+{
+ if (w_buffer_.length_ <= limit)
+ {
+ return 1;
+ }
+
+ //
+ // Write the data accumulated in the write
+ // buffer until it is below the limit or
+ // the timeout is elapsed.
+ //
+
+ int toWrite = w_buffer_.length_;
+
+ int written = 0;
+
+ #ifdef TEST
+ *logofs << "Transport: Draining " << toWrite - limit
+ << " bytes on FD#" << fd_ << " with limit set to "
+ << limit << ".\n" << logofs_flush;
+ #endif
+
+ T_timestamp startTs = getNewTimestamp();
+
+ T_timestamp selectTs;
+ T_timestamp writeTs;
+ T_timestamp idleTs;
+
+ T_timestamp nowTs = startTs;
+
+ int diffTs;
+
+ fd_set writeSet;
+ fd_set readSet;
+
+ FD_ZERO(&writeSet);
+ FD_ZERO(&readSet);
+
+ int result;
+ int ready;
+
+ while (w_buffer_.length_ - written > limit)
+ {
+ nowTs = getNewTimestamp();
+
+ //
+ // Wait for descriptor to become
+ // readable or writable.
+ //
+
+ FD_SET(fd_, &writeSet);
+ FD_SET(fd_, &readSet);
+
+ setTimestamp(selectTs, timeout / 2);
+
+ idleTs = nowTs;
+
+ result = select(fd_ + 1, &readSet, &writeSet, NULL, &selectTs);
+
+ nowTs = getNewTimestamp();
+
+ diffTs = diffTimestamp(idleTs, nowTs);
+
+ statistics -> addIdleTime(diffTs);
+
+ statistics -> subReadTime(diffTs);
+
+ if (result < 0)
+ {
+ if (EGET() == EINTR)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Select on FD#" << fd_
+ << " was interrupted.\n" << logofs_flush;
+ #endif
+
+ continue;
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Select on FD#" << fd_
+ << " failed.\n" << logofs_flush;
+ #endif
+
+ finish();
+
+ return -1;
+ }
+ }
+ else if (result > 0)
+ {
+ ready = result;
+
+ if (FD_ISSET(fd_, &writeSet))
+ {
+ writeTs = getNewTimestamp();
+
+ result = ::write(fd_, w_buffer_.data_.begin() + w_buffer_.start_ +
+ written, toWrite - written);
+
+ nowTs = getNewTimestamp();
+
+ diffTs = diffTimestamp(writeTs, nowTs);
+
+ statistics -> addWriteTime(diffTs);
+
+ if (result > 0)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Forced flush of " << result
+ << " bytes on " << "FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ written += result;
+ }
+ else if (result < 0 && EGET() == EINTR)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write to FD#" << fd_
+ << " was interrupted.\n" << logofs_flush;
+ #endif
+
+ continue;
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Write to FD#" << fd_
+ << " failed.\n" << logofs_flush;
+ #endif
+
+ finish();
+
+ return -1;
+ }
+
+ ready--;
+ }
+
+ if (ready > 0)
+ {
+ if (FD_ISSET(fd_, &readSet))
+ {
+ #ifdef TEST
+ *logofs << "Transport: Not draining further "
+ << "due to data readable on FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ break;
+ }
+ }
+ }
+ #ifdef TEST
+ else
+ {
+ *logofs << "Transport: Timeout encountered "
+ << "waiting for FD#" << fd_ << ".\n"
+ << logofs_flush;
+ }
+ #endif
+
+ nowTs = getNewTimestamp();
+
+ diffTs = diffTimestamp(startTs, nowTs);
+
+ if (diffTs >= timeout)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Not draining further "
+ << "due to the timeout on FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ break;
+ }
+ }
+
+ if (written > 0)
+ {
+ #ifdef DUMP
+
+ *logofs << "Transport: Dumping content of flushed data.\n"
+ << logofs_flush;
+
+ DumpData(w_buffer_.data_.begin() + w_buffer_.start_, written);
+
+ #endif
+
+ //
+ // Update the buffer status.
+ //
+
+ w_buffer_.length_ -= written;
+
+ if (w_buffer_.length_ == 0)
+ {
+ w_buffer_.start_ = 0;
+
+ blocked_ = 0;
+ }
+ else
+ {
+ w_buffer_.start_ += written;
+
+ #ifdef TEST
+ *logofs << "Transport: There are still " << w_buffer_.length_
+ << " bytes in write buffer for " << "FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ blocked_ = 1;
+ }
+ }
+ #ifdef TEST
+ else
+ {
+ *logofs << "Transport: WARNING! No data written to FD#" << fd_
+ << " with " << toWrite << " bytes to drain and limit "
+ << "set to " << limit << ".\n" << logofs_flush;
+ }
+ #endif
+
+ #ifdef TEST
+ *logofs << "Transport: Write buffer for FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "Transport: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " size is "
+ << w_buffer_.data_.size() << " capacity is "
+ << w_buffer_.data_.capacity() << ".\n"
+ << logofs_flush;
+ #endif
+
+ return (w_buffer_.length_ <= limit);
+}
+
+int Transport::wait(int timeout) const
+{
+ T_timestamp startTs = getNewTimestamp();
+
+ T_timestamp idleTs;
+ T_timestamp selectTs;
+
+ T_timestamp nowTs = startTs;
+
+ long available = 0;
+ int result = 0;
+
+ int diffTs;
+
+ fd_set readSet;
+
+ FD_ZERO(&readSet);
+ FD_SET(fd_, &readSet);
+
+ for (;;)
+ {
+ available = readable();
+
+ diffTs = diffTimestamp(startTs, nowTs);
+
+ if (available != 0 || timeout == 0 ||
+ (diffTs + (timeout / 10)) >= timeout)
+ {
+ #ifdef TEST
+ *logofs << "Transport: There are " << available
+ << " bytes on FD#" << fd_ << " after "
+ << diffTs << " Ms.\n" << logofs_flush;
+ #endif
+
+ return available;
+ }
+ else if (available == 0 && result > 0)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Read on " << "FD#"
+ << fd_ << " failed.\n" << logofs_flush;
+ #endif
+
+ return -1;
+ }
+
+ //
+ // TODO: Should subtract the time
+ // already spent in select.
+ //
+
+ selectTs.tv_sec = 0;
+ selectTs.tv_usec = timeout * 1000;
+
+ idleTs = nowTs;
+
+ //
+ // Wait for descriptor to become readable.
+ //
+
+ result = select(fd_ + 1, &readSet, NULL, NULL, &selectTs);
+
+ nowTs = getNewTimestamp();
+
+ diffTs = diffTimestamp(idleTs, nowTs);
+
+ statistics -> addIdleTime(diffTs);
+
+ statistics -> subReadTime(diffTs);
+
+ if (result < 0)
+ {
+ if (EGET() == EINTR)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Select on FD#" << fd_
+ << " was interrupted.\n" << logofs_flush;
+ #endif
+
+ continue;
+ }
+ else
+ {
+ #ifdef TEST
+ *logofs << "Transport: Select on " << "FD#"
+ << fd_ << " failed.\n" << logofs_flush;
+ #endif
+
+ return -1;
+ }
+ }
+ #ifdef TEST
+ else if (result == 0)
+ {
+ *logofs << "Transport: No data available on FD#" << fd_
+ << " after " << diffTimestamp(startTs, nowTs)
+ << " Ms.\n" << logofs_flush;
+ }
+ else
+ {
+ *logofs << "Transport: Data became available on FD#" << fd_
+ << " after " << diffTimestamp(startTs, nowTs)
+ << " Ms.\n" << logofs_flush;
+ }
+ #endif
+ }
+}
+
+void Transport::setSize(unsigned int initialSize, unsigned int thresholdSize,
+ unsigned int maximumSize)
+{
+ initialSize_ = initialSize;
+ thresholdSize_ = thresholdSize;
+ maximumSize_ = maximumSize;
+
+ #ifdef TEST
+ *logofs << "Transport: Set buffer sizes for FD#" << fd_
+ << " to " << initialSize_ << "/" << thresholdSize_
+ << "/" << maximumSize_ << ".\n" << logofs_flush;
+ #endif
+}
+
+void Transport::fullReset()
+{
+ blocked_ = 0;
+ finish_ = 0;
+
+ fullReset(w_buffer_);
+}
+
+int Transport::resize(T_buffer &buffer, const int &size)
+{
+ if ((int) buffer.data_.size() >= (buffer.length_ + size) &&
+ (buffer.start_ + buffer.length_ + size) >
+ (int) buffer.data_.size())
+ {
+ if (buffer.length_ > 0)
+ {
+ //
+ // There is enough space in buffer but we need
+ // to move existing data at the beginning.
+ //
+
+ #ifdef TEST
+ *logofs << "Transport: Moving " << buffer.length_
+ << " bytes of data for " << "FD#" << fd_
+ << " to make room in the buffer.\n"
+ << logofs_flush;
+ #endif
+
+ memmove(buffer.data_.begin(), buffer.data_.begin() +
+ buffer.start_, buffer.length_);
+ }
+
+ buffer.start_ = 0;
+
+ #ifdef DEBUG
+ *logofs << "Transport: Made room for "
+ << buffer.data_.size() - buffer.start_
+ << " bytes in buffer for " << "FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+ }
+ else if ((buffer.length_ + size) > (int) buffer.data_.size())
+ {
+ //
+ // Not enough space, so increase
+ // the size of the buffer.
+ //
+
+ if (buffer.start_ != 0 && buffer.length_ > 0)
+ {
+ #ifdef TEST
+ *logofs << "Transport: Moving " << buffer.length_
+ << " bytes of data for " << "FD#" << fd_
+ << " to resize the buffer.\n"
+ << logofs_flush;
+ #endif
+
+ memmove(buffer.data_.begin(), buffer.data_.begin() +
+ buffer.start_, buffer.length_);
+ }
+
+ buffer.start_ = 0;
+
+ unsigned int newSize = thresholdSize_;
+
+ while (newSize < (unsigned int) buffer.length_ + size)
+ {
+ newSize <<= 1;
+
+ if (newSize >= maximumSize_)
+ {
+ newSize = buffer.length_ + size + initialSize_;
+ }
+ }
+
+ #ifdef DEBUG
+ *logofs << "Transport: Buffer for " << "FD#" << fd_
+ << " will be enlarged from " << buffer.data_.size()
+ << " to at least " << buffer.length_ + size
+ << " bytes.\n" << logofs_flush;
+ #endif
+
+ buffer.data_.resize(newSize);
+
+ #ifdef TEST
+ if (newSize >= maximumSize_)
+ {
+ *logofs << "Transport: WARNING! Buffer for FD#" << fd_
+ << " grown to reach size of " << newSize
+ << " bytes.\n" << logofs_flush;
+ }
+ #endif
+
+ #ifdef TEST
+ *logofs << "Transport: Data buffer for " << "FD#"
+ << fd_ << " has now size " << buffer.data_.size()
+ << " and capacity " << buffer.data_.capacity()
+ << ".\n" << logofs_flush;
+ #endif
+ }
+
+ return (buffer.length_ + size);
+}
+
+void Transport::fullReset(T_buffer &buffer)
+{
+ //
+ // Force deallocation and allocation
+ // of the initial size.
+ //
+
+ #ifdef TEST
+ *logofs << "Transport: Resetting buffer for " << "FD#"
+ << fd_ << " with size " << buffer.data_.size()
+ << " and capacity " << buffer.data_.capacity()
+ << ".\n" << logofs_flush;
+ #endif
+
+ buffer.start_ = 0;
+ buffer.length_ = 0;
+
+ if (buffer.data_.size() > (unsigned int) initialSize_ &&
+ buffer.data_.capacity() > (unsigned int) initialSize_)
+ {
+ buffer.data_.clear();
+
+ buffer.data_.resize(initialSize_);
+
+ #ifdef TEST
+ *logofs << "Transport: Data buffer for " << "FD#"
+ << fd_ << " shrunk to size " << buffer.data_.size()
+ << " and capacity " << buffer.data_.capacity()
+ << ".\n" << logofs_flush;
+ #endif
+ }
+}
+
+ProxyTransport::ProxyTransport(int fd) : Transport(fd)
+{
+ #ifdef TEST
+ *logofs << "ProxyTransport: Going to create proxy transport "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ type_ = transport_proxy;
+
+ //
+ // Set up the read buffer.
+ //
+
+ r_buffer_.length_ = 0;
+ r_buffer_.start_ = 0;
+
+ r_buffer_.data_.resize(initialSize_);
+
+ //
+ // For now we own the buffer.
+ //
+
+ owner_ = 1;
+
+ //
+ // Set up ZLIB compression.
+ //
+
+ int result;
+
+ r_stream_.zalloc = NULL;
+ r_stream_.zfree = NULL;
+ r_stream_.opaque = NULL;
+
+ r_stream_.next_in = NULL;
+ r_stream_.avail_in = 0;
+
+ if ((result = inflateInit2(&r_stream_, 15)) != Z_OK)
+ {
+ #ifdef PANIC
+ *logofs << "ProxyTransport: PANIC! Failed initialization of ZLIB read stream. "
+ << "Error is '" << zError(result) << "'.\n" << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Failed initialization of ZLIB read stream. "
+ << "Error is '" << zError(result) << "'.\n";
+
+ HandleCleanup();
+ }
+
+ if (control -> LocalStreamCompression)
+ {
+ w_stream_.zalloc = NULL;
+ w_stream_.zfree = NULL;
+ w_stream_.opaque = NULL;
+
+ if ((result = deflateInit2(&w_stream_, control -> LocalStreamCompressionLevel, Z_DEFLATED,
+ 15, 9, Z_DEFAULT_STRATEGY)) != Z_OK)
+ {
+ #ifdef PANIC
+ *logofs << "ProxyTransport: PANIC! Failed initialization of ZLIB write stream. "
+ << "Error is '" << zError(result) << "'.\n" << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Failed initialization of ZLIB write stream. "
+ << "Error is '" << zError(result) << "'.\n";
+
+ HandleCleanup();
+ }
+ }
+
+ //
+ // No ZLIB stream to flush yet.
+ //
+
+ flush_ = 0;
+
+ #ifdef REFERENCES
+ *logofs << "ProxyTransport: Created new object at "
+ << this << " out of " << ++references_
+ << " allocated references.\n" << logofs_flush;
+ #endif
+}
+
+ProxyTransport::~ProxyTransport()
+{
+ #ifdef TEST
+ *logofs << "ProxyTransport: Going to destroy derived class "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ //
+ // Deallocate the ZLIB stream state.
+ //
+
+ inflateEnd(&r_stream_);
+
+ if (control -> LocalStreamCompression)
+ {
+ deflateEnd(&w_stream_);
+ }
+
+ #ifdef REFERENCES
+ *logofs << "ProxyTransport: Deleted object at "
+ << this << " out of " << --references_
+ << " allocated references.\n" << logofs_flush;
+ #endif
+}
+
+//
+// Read data from its file descriptor.
+//
+
+int ProxyTransport::read(unsigned char *data, unsigned int size)
+{
+ //
+ // If the remote peer is not compressing
+ // the stream then just return any byte
+ // read from the socket.
+ //
+
+ if (control -> RemoteStreamCompression == 0)
+ {
+ int result = Transport::read(data, size);
+
+ if (result <= 0)
+ {
+ return result;
+ }
+
+ statistics -> addBytesIn(result);
+
+ return result;
+ }
+
+ //
+ // Return any pending data first.
+ //
+
+ if (r_buffer_.length_ > 0)
+ {
+ //
+ // If the size of the buffer doesn't
+ // match the amount of data pending,
+ // force the caller to retry.
+ //
+
+ if ((int) size < r_buffer_.length_)
+ {
+ #ifdef TEST
+ *logofs << "ProxyTransport: WARNING! Forcing a retry with "
+ << r_buffer_.length_ << " bytes pending and "
+ << size << " in the buffer.\n"
+ << logofs_flush;
+ #endif
+
+ ESET(EAGAIN);
+
+ return -1;
+ }
+
+ int copied = (r_buffer_.length_ > ((int) size) ?
+ ((int) size) : r_buffer_.length_);
+
+ memcpy(data, r_buffer_.data_.begin() + r_buffer_.start_, copied);
+
+ //
+ // Update the buffer status.
+ //
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Going to immediately return " << copied
+ << " bytes from proxy FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ r_buffer_.length_ -= copied;
+
+ if (r_buffer_.length_ == 0)
+ {
+ r_buffer_.start_ = 0;
+ }
+ else
+ {
+ r_buffer_.start_ += copied;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: There are still " << r_buffer_.length_
+ << " bytes in read buffer for proxy " << "FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+ }
+
+ return copied;
+ }
+
+ //
+ // Read data in the user buffer.
+ //
+
+ int result = Transport::read(data, size);
+
+ if (result <= 0)
+ {
+ return result;
+ }
+
+ statistics -> addBytesIn(result);
+
+ //
+ // Decompress the data into the read
+ // buffer.
+ //
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Going to decompress data for "
+ << "proxy FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ int saveTotalIn = r_stream_.total_in;
+ int saveTotalOut = r_stream_.total_out;
+
+ int oldTotalIn = saveTotalIn;
+ int oldTotalOut = saveTotalOut;
+
+ int diffTotalIn;
+ int diffTotalOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: oldTotalIn = " << oldTotalIn
+ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: oldTotalOut = " << oldTotalOut
+ << ".\n" << logofs_flush;
+ #endif
+
+ r_stream_.next_in = (Bytef *) data;
+ r_stream_.avail_in = result;
+
+ //
+ // Let ZLIB use all the space already
+ // available in the buffer.
+ //
+
+ unsigned int newAvailOut = r_buffer_.data_.size() - r_buffer_.start_ -
+ r_buffer_.length_;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Initial decompress buffer is "
+ << newAvailOut << " bytes.\n" << logofs_flush;
+ #endif
+
+ for (;;)
+ {
+ #ifdef INSPECT
+ *logofs << "\nProxyTransport: Running the decompress loop.\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_buffer_.length_ = " << r_buffer_.length_
+ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_buffer_.data_.size() = " << r_buffer_.data_.size()
+ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: newAvailOut = " << newAvailOut
+ << ".\n" << logofs_flush;
+ #endif
+
+ if (resize(r_buffer_, newAvailOut) < 0)
+ {
+ return -1;
+ }
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_buffer_.data_.size() = "
+ << r_buffer_.data_.size() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.next_in = "
+ << (void *) r_stream_.next_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.avail_in = "
+ << r_stream_.avail_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ r_stream_.next_out = r_buffer_.data_.begin() + r_buffer_.start_ +
+ r_buffer_.length_;
+
+ r_stream_.avail_out = newAvailOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.next_out = "
+ << (void *) r_stream_.next_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.avail_out = "
+ << r_stream_.avail_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ int result = inflate(&r_stream_, Z_SYNC_FLUSH);
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: Called inflate() result is "
+ << result << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.avail_in = "
+ << r_stream_.avail_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.avail_out = "
+ << r_stream_.avail_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.total_in = "
+ << r_stream_.total_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_stream_.total_out = "
+ << r_stream_.total_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ diffTotalIn = r_stream_.total_in - oldTotalIn;
+ diffTotalOut = r_stream_.total_out - oldTotalOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: diffTotalIn = "
+ << diffTotalIn << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: diffTotalOut = "
+ << diffTotalOut << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_buffer_.length_ = "
+ << r_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ r_buffer_.length_ += diffTotalOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: r_buffer_.length_ = "
+ << r_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ oldTotalIn = r_stream_.total_in;
+ oldTotalOut = r_stream_.total_out;
+
+ if (result == Z_OK)
+ {
+ if (r_stream_.avail_in == 0 && r_stream_.avail_out > 0)
+ {
+ break;
+ }
+ }
+ else if (result == Z_BUF_ERROR && r_stream_.avail_out > 0 &&
+ r_stream_.avail_in == 0)
+ {
+ #ifdef TEST
+ *logofs << "ProxyTransport: WARNING! Raised Z_BUF_ERROR decompressing data.\n"
+ << logofs_flush;
+ #endif
+
+ break;
+ }
+ else
+ {
+ #ifdef PANIC
+ *logofs << "ProxyTransport: PANIC! Decompression of data failed. "
+ << "Error is '" << zError(result) << "'.\n"
+ << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Decompression of data failed. Error is '"
+ << zError(result) << "'.\n";
+
+ finish();
+
+ return -1;
+ }
+
+ //
+ // Add more bytes to the buffer.
+ //
+
+ if (newAvailOut < thresholdSize_)
+ {
+ newAvailOut = thresholdSize_;
+ }
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Need to add " << newAvailOut
+ << " bytes to the decompress buffer in read.\n"
+ << logofs_flush;
+ #endif
+ }
+
+ diffTotalIn = r_stream_.total_in - saveTotalIn;
+ diffTotalOut = r_stream_.total_out - saveTotalOut;
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Decompressed data from "
+ << diffTotalIn << " to " << diffTotalOut
+ << " bytes.\n" << logofs_flush;
+ #endif
+
+ statistics -> addDecompressedBytes(diffTotalIn, diffTotalOut);
+
+ //
+ // Check if the size of the buffer
+ // matches the produced data.
+ //
+
+ if ((int) size < r_buffer_.length_)
+ {
+ #ifdef TEST
+ *logofs << "ProxyTransport: WARNING! Forcing a retry with "
+ << r_buffer_.length_ << " bytes pending and "
+ << size << " in the buffer.\n"
+ << logofs_flush;
+ #endif
+
+ ESET(EAGAIN);
+
+ return -1;
+ }
+
+ //
+ // Copy the decompressed data to the
+ // provided buffer.
+ //
+
+ int copied = (r_buffer_.length_ > ((int) size) ?
+ ((int) size) : r_buffer_.length_);
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Going to return " << copied
+ << " bytes from proxy FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ memcpy(data, r_buffer_.data_.begin() + r_buffer_.start_, copied);
+
+ //
+ // Update the buffer status.
+ //
+
+ r_buffer_.length_ -= copied;
+
+ if (r_buffer_.length_ == 0)
+ {
+ r_buffer_.start_ = 0;
+ }
+ else
+ {
+ r_buffer_.start_ += copied;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: There are still " << r_buffer_.length_
+ << " bytes in read buffer for proxy FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+ }
+
+ return copied;
+}
+
+//
+// If required compress data, else write it to socket.
+//
+
+int ProxyTransport::write(T_write type, const unsigned char *data, const unsigned int size)
+{
+ #ifdef TEST
+ if (size == 0)
+ {
+ *logofs << "ProxyTransport: WARNING! Write called for FD#"
+ << fd_ << " without any data to write.\n"
+ << logofs_flush;
+
+ return 0;
+ }
+ #endif
+
+ //
+ // If there is no compression revert to
+ // plain socket management.
+ //
+
+ if (control -> LocalStreamCompression == 0)
+ {
+ int result = Transport::write(type, data, size);
+
+ if (result <= 0)
+ {
+ return result;
+ }
+
+ statistics -> addBytesOut(result);
+
+ statistics -> updateBitrate(result);
+
+ FlushCallback(result);
+
+ return result;
+ }
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Going to compress " << size
+ << " bytes to write buffer for proxy FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ //
+ // Compress data into the write buffer.
+ //
+
+ int saveTotalIn = w_stream_.total_in;
+ int saveTotalOut = w_stream_.total_out;
+
+ int oldTotalIn = saveTotalIn;
+ int oldTotalOut = saveTotalOut;
+
+ int diffTotalIn;
+ int diffTotalOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: oldTotalIn = " << oldTotalIn
+ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: oldTotalOut = " << oldTotalOut
+ << ".\n" << logofs_flush;
+ #endif
+
+ w_stream_.next_in = (Bytef *) data;
+ w_stream_.avail_in = size;
+
+ //
+ // Let ZLIB use all the space already
+ // available in the buffer.
+ //
+
+ unsigned int newAvailOut = w_buffer_.data_.size() - w_buffer_.start_ -
+ w_buffer_.length_;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Initial compress buffer is "
+ << newAvailOut << " bytes.\n" << logofs_flush;
+ #endif
+
+ for (;;)
+ {
+ #ifdef INSPECT
+ *logofs << "\nProxyTransport: Running the compress loop.\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.length_ = "
+ << w_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.data_.size() = "
+ << w_buffer_.data_.size() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: newAvailOut = "
+ << newAvailOut << ".\n"
+ << logofs_flush;
+ #endif
+
+ if (resize(w_buffer_, newAvailOut) < 0)
+ {
+ return -1;
+ }
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.data_.size() = "
+ << w_buffer_.data_.size() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.next_in = "
+ << (void *) w_stream_.next_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_in = "
+ << w_stream_.avail_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ w_stream_.next_out = w_buffer_.data_.begin() + w_buffer_.start_ +
+ w_buffer_.length_;
+
+ w_stream_.avail_out = newAvailOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.next_out = "
+ << (void *) w_stream_.next_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_out = "
+ << w_stream_.avail_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ int result = deflate(&w_stream_, (type == write_delayed ?
+ Z_NO_FLUSH : Z_SYNC_FLUSH));
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: Called deflate() result is "
+ << result << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_in = "
+ << w_stream_.avail_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_out = "
+ << w_stream_.avail_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.total_in = "
+ << w_stream_.total_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.total_out = "
+ << w_stream_.total_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ diffTotalOut = w_stream_.total_out - oldTotalOut;
+ diffTotalIn = w_stream_.total_in - oldTotalIn;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: diffTotalIn = "
+ << diffTotalIn << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: diffTotalOut = "
+ << diffTotalOut << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.length_ = "
+ << w_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ w_buffer_.length_ += diffTotalOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.length_ = "
+ << w_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ oldTotalOut = w_stream_.total_out;
+ oldTotalIn = w_stream_.total_in;
+
+ if (result == Z_OK)
+ {
+ if (w_stream_.avail_in == 0 && w_stream_.avail_out > 0)
+ {
+ break;
+ }
+ }
+ else if (result == Z_BUF_ERROR && w_stream_.avail_out > 0 &&
+ w_stream_.avail_in == 0)
+ {
+ #ifdef TEST
+ *logofs << "ProxyTransport: WARNING! Raised Z_BUF_ERROR compressing data.\n"
+ << logofs_flush;
+ #endif
+
+ break;
+ }
+ else
+ {
+ #ifdef PANIC
+ *logofs << "ProxyTransport: PANIC! Compression of data failed. "
+ << "Error is '" << zError(result) << "'.\n"
+ << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Compression of data failed. Error is '"
+ << zError(result) << "'.\n";
+
+ finish();
+
+ return -1;
+ }
+
+ //
+ // Add more bytes to the buffer.
+ //
+
+ if (newAvailOut < thresholdSize_)
+ {
+ newAvailOut = thresholdSize_;
+ }
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Need to add " << newAvailOut
+ << " bytes to the compress buffer in write.\n"
+ << logofs_flush;
+ #endif
+ }
+
+ diffTotalIn = w_stream_.total_in - saveTotalIn;
+ diffTotalOut = w_stream_.total_out - saveTotalOut;
+
+ #ifdef TEST
+
+ *logofs << "ProxyTransport: Compressed data from "
+ << diffTotalIn << " to " << diffTotalOut
+ << " bytes.\n" << logofs_flush;
+
+ if (diffTotalIn != (int) size)
+ {
+ #ifdef PANIC
+ *logofs << "ProxyTransport: PANIC! Bytes provided to ZLIB stream "
+ << "should be " << size << " but they look to be "
+ << diffTotalIn << ".\n" << logofs_flush;
+ #endif
+ }
+
+ #endif
+
+ //
+ // Find out what we have to do with the
+ // produced data.
+ //
+
+ if (type == write_immediate)
+ {
+ //
+ // If user requested an immediate write we
+ // flushed the ZLIB buffer. We can now reset
+ // the counter and write data to socket.
+ //
+
+ flush_ = 0;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Write buffer for proxy FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "ProxyTransport: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " flush is "
+ << flush_ << " size is " << w_buffer_.data_.size()
+ << " capacity is " << w_buffer_.data_.capacity()
+ << ".\n" << logofs_flush;
+ #endif
+
+ //
+ // Alternatively may try to write only if
+ // the socket is not blocked.
+ //
+ // if (w_buffer_.length_ > 0 && blocked_ == 0)
+ // {
+ // ...
+ // }
+ //
+
+ if (w_buffer_.length_ > 0)
+
+ {
+ #ifdef TEST
+ *logofs << "ProxyTransport: Writing " << w_buffer_.length_
+ << " bytes of produced data to FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ int result = Transport::flush();
+
+ if (result < 0)
+ {
+ return -1;
+ }
+ }
+ }
+ else
+ {
+ //
+ // We haven't flushed the ZLIB compression
+ // buffer, so user will have to call proxy
+ // transport's flush explicitly.
+ //
+
+ flush_ += diffTotalIn;
+ }
+
+ //
+ // We either wrote the data or added it to the
+ // write buffer. It's convenient to update the
+ // counters at this stage to get the current
+ // bitrate earlier.
+ //
+
+ statistics -> addCompressedBytes(diffTotalIn, diffTotalOut);
+
+ statistics -> addBytesOut(diffTotalOut);
+
+ statistics -> updateBitrate(diffTotalOut);
+
+ FlushCallback(diffTotalOut);
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Write buffer for proxy FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "ProxyTransport: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " flush is "
+ << flush_ << " size is " << w_buffer_.data_.size()
+ << " capacity is " << w_buffer_.data_.capacity()
+ << ".\n" << logofs_flush;
+ #endif
+
+ return size;
+}
+
+//
+// Write data to its file descriptor.
+//
+
+int ProxyTransport::flush()
+{
+ //
+ // If there is no compression or we already compressed
+ // outgoing data and just need to write it to socket
+ // because of previous incomplete writes then revert
+ // to plain socket management.
+ //
+
+ if (flush_ == 0 || control -> LocalStreamCompression == 0)
+ {
+ int result = Transport::flush();
+
+ if (result < 0)
+ {
+ return -1;
+ }
+
+ return result;
+ }
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Going to flush compression on "
+ << "proxy FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Flush counter for proxy FD#" << fd_
+ << " is " << flush_ << " bytes.\n" << logofs_flush;
+ #endif
+
+ //
+ // Flush ZLIB stream into the write buffer.
+ //
+
+ int saveTotalIn = w_stream_.total_in;
+ int saveTotalOut = w_stream_.total_out;
+
+ int oldTotalIn = saveTotalIn;
+ int oldTotalOut = saveTotalOut;
+
+ int diffTotalOut;
+ int diffTotalIn;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: oldTotalIn = " << oldTotalIn
+ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: oldTotalOut = " << oldTotalOut
+ << ".\n" << logofs_flush;
+ #endif
+
+ w_stream_.next_in = w_buffer_.data_.begin() + w_buffer_.start_ + w_buffer_.length_;
+ w_stream_.avail_in = 0;
+
+ //
+ // Let ZLIB use all the space already
+ // available in the buffer.
+ //
+
+ unsigned int newAvailOut = w_buffer_.data_.size() - w_buffer_.start_ -
+ w_buffer_.length_;
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Initial flush buffer is "
+ << newAvailOut << " bytes.\n" << logofs_flush;
+ #endif
+
+ for (;;)
+ {
+ #ifdef INSPECT
+ *logofs << "\nProxyTransport: Running the flush loop.\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.length_ = "
+ << w_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.data_.size() = "
+ << w_buffer_.data_.size() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: newAvailOut = "
+ << newAvailOut << ".\n"
+ << logofs_flush;
+ #endif
+
+ if (resize(w_buffer_, newAvailOut) < 0)
+ {
+ return -1;
+ }
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.data_.size() = "
+ << w_buffer_.data_.size() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.next_in = "
+ << (void *) w_stream_.next_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_in = "
+ << w_stream_.avail_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ w_stream_.next_out = w_buffer_.data_.begin() + w_buffer_.start_ +
+ w_buffer_.length_;
+
+ w_stream_.avail_out = newAvailOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.next_out = "
+ << (void *) w_stream_.next_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_out = "
+ << w_stream_.avail_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ int result = deflate(&w_stream_, Z_SYNC_FLUSH);
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: Called deflate() result is "
+ << result << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_in = "
+ << w_stream_.avail_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.avail_out = "
+ << w_stream_.avail_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.total_in = "
+ << w_stream_.total_in << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_stream_.total_out = "
+ << w_stream_.total_out << ".\n"
+ << logofs_flush;
+ #endif
+
+ diffTotalOut = w_stream_.total_out - oldTotalOut;
+ diffTotalIn = w_stream_.total_in - oldTotalIn;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: diffTotalIn = "
+ << diffTotalIn << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: diffTotalOut = "
+ << diffTotalOut << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.length_ = "
+ << w_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ w_buffer_.length_ += diffTotalOut;
+
+ #ifdef INSPECT
+ *logofs << "ProxyTransport: w_buffer_.length_ = "
+ << w_buffer_.length_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ oldTotalOut = w_stream_.total_out;
+ oldTotalIn = w_stream_.total_in;
+
+ if (result == Z_OK)
+ {
+ if (w_stream_.avail_in == 0 && w_stream_.avail_out > 0)
+ {
+ break;
+ }
+ }
+ else if (result == Z_BUF_ERROR && w_stream_.avail_out > 0 &&
+ w_stream_.avail_in == 0)
+ {
+ #ifdef TEST
+ *logofs << "ProxyTransport: WARNING! Raised Z_BUF_ERROR flushing data.\n"
+ << logofs_flush;
+ #endif
+
+ break;
+ }
+ else
+ {
+ #ifdef PANIC
+ *logofs << "ProxyTransport: PANIC! Flush of compressed data failed. "
+ << "Error is '" << zError(result) << "'.\n"
+ << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Flush of compressed data failed. Error is '"
+ << zError(result) << "'.\n";
+
+ finish();
+
+ return -1;
+ }
+
+ //
+ // Add more bytes to the buffer.
+ //
+
+ if (newAvailOut < thresholdSize_)
+ {
+ newAvailOut = thresholdSize_;
+ }
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Need to add " << newAvailOut
+ << " bytes to the compress buffer in flush.\n"
+ << logofs_flush;
+ #endif
+ }
+
+ diffTotalIn = w_stream_.total_in - saveTotalIn;
+ diffTotalOut = w_stream_.total_out - saveTotalOut;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Compressed flush data from "
+ << diffTotalIn << " to " << diffTotalOut
+ << " bytes.\n" << logofs_flush;
+ #endif
+
+ //
+ // Time to move data from the write
+ // buffer to the real link.
+ //
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Reset flush counter for proxy FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ flush_ = 0;
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: Write buffer for proxy FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "ProxyTransport: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " flush is "
+ << flush_ << " size is " << w_buffer_.data_.size()
+ << " capacity is " << w_buffer_.data_.capacity()
+ << ".\n" << logofs_flush;
+ #endif
+
+ int result = Transport::flush();
+
+ if (result < 0)
+ {
+ return -1;
+ }
+
+ //
+ // Update all the counters.
+ //
+
+ statistics -> addCompressedBytes(diffTotalIn, diffTotalOut);
+
+ statistics -> addBytesOut(diffTotalOut);
+
+ statistics -> updateBitrate(diffTotalOut);
+
+ FlushCallback(diffTotalOut);
+
+ return result;
+}
+
+unsigned int ProxyTransport::getPending(unsigned char *&data)
+{
+ //
+ // Return a pointer to the data in the
+ // read buffer. It is up to the caller
+ // to ensure that the data is consumed
+ // before the read buffer is reused.
+ //
+
+ if (r_buffer_.length_ > 0)
+ {
+ unsigned int size = r_buffer_.length_;
+
+ data = r_buffer_.data_.begin() + r_buffer_.start_;
+
+ #ifdef DEBUG
+ *logofs << "ProxyTransport: Returning " << size
+ << " pending bytes from proxy FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ r_buffer_.length_ = 0;
+ r_buffer_.start_ = 0;
+
+ //
+ // Prevent the deletion of the buffer.
+ //
+
+ owner_ = 0;
+
+ return size;
+ }
+
+ #ifdef TEST
+ *logofs << "ProxyTransport: WARNING! No pending data "
+ << "for proxy FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ data = NULL;
+
+ return 0;
+}
+
+void ProxyTransport::fullReset()
+{
+ blocked_ = 0;
+ finish_ = 0;
+ flush_ = 0;
+
+ if (control -> RemoteStreamCompression)
+ {
+ inflateReset(&r_stream_);
+ }
+
+ if (control -> LocalStreamCompression)
+ {
+ deflateReset(&w_stream_);
+ }
+
+ if (owner_ == 1)
+ {
+ Transport::fullReset(r_buffer_);
+ }
+
+ Transport::fullReset(w_buffer_);
+}
+
+AgentTransport::AgentTransport(int fd) : Transport(fd)
+{
+ #ifdef TEST
+ *logofs << "AgentTransport: Going to create agent transport "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ type_ = transport_agent;
+
+ //
+ // Set up the read buffer.
+ //
+
+ r_buffer_.length_ = 0;
+ r_buffer_.start_ = 0;
+
+ r_buffer_.data_.resize(initialSize_);
+
+ //
+ // For now we own the buffer.
+ //
+
+ owner_ = 1;
+
+ //
+ // Set up the mutexes.
+ //
+
+ #ifdef THREADS
+
+ pthread_mutexattr_t m_attributes;
+
+ pthread_mutexattr_init(&m_attributes);
+
+ //
+ // Interfaces in pthread to handle mutex
+ // type do not work in current version.
+ //
+
+ m_attributes.__mutexkind = PTHREAD_MUTEX_ERRORCHECK_NP;
+
+ if (pthread_mutex_init(&m_read_, &m_attributes) != 0)
+ {
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: Creation of read mutex failed. "
+ << "Error is " << EGET() << " '" << ESTR()
+ << "'.\n" << logofs_flush;
+ #endif
+ }
+
+ if (pthread_mutex_init(&m_write_, &m_attributes) != 0)
+ {
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: Creation of write mutex failed. "
+ << "Error is " << EGET() << " '" << ESTR()
+ << "'.\n" << logofs_flush;
+ #endif
+ }
+
+ #endif
+
+ #ifdef REFERENCES
+ *logofs << "AgentTransport: Child: Created new object at "
+ << this << " out of " << ++references_
+ << " allocated references.\n" << logofs_flush;
+ #endif
+}
+
+AgentTransport::~AgentTransport()
+{
+ #ifdef TEST
+ *logofs << "AgentTransport: Going to destroy derived class "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ //
+ // Unlock and free all mutexes.
+ //
+
+ #ifdef THREADS
+
+ pthread_mutex_unlock(&m_read_);
+ pthread_mutex_unlock(&m_write_);
+
+ pthread_mutex_destroy(&m_read_);
+ pthread_mutex_destroy(&m_write_);
+
+ #endif
+
+ #ifdef REFERENCES
+ *logofs << "AgentTransport: Child: Deleted object at "
+ << this << " out of " << --references_
+ << " allocated references.\n" << logofs_flush;
+ #endif
+}
+
+//
+// Read data enqueued by the other thread.
+//
+
+int AgentTransport::read(unsigned char *data, unsigned int size)
+{
+ #ifdef THREADS
+
+ lockRead();
+
+ #endif
+
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Child: Going to read " << size
+ << " bytes from " << "FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ int copied = -1;
+
+ if (r_buffer_.length_ > 0)
+ {
+ if ((int) size < r_buffer_.length_)
+ {
+ #ifdef TEST
+ *logofs << "AgentTransport: WARNING! Forcing a retry with "
+ << r_buffer_.length_ << " bytes pending and "
+ << size << " in the buffer.\n"
+ << logofs_flush;
+ #endif
+
+ ESET(EAGAIN);
+ }
+ else
+ {
+ copied = (r_buffer_.length_ > ((int) size) ?
+ ((int) size) : r_buffer_.length_);
+
+ memcpy(data, r_buffer_.data_.begin() + r_buffer_.start_, copied);
+
+ //
+ // Update the buffer status.
+ //
+
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: Going to immediately return "
+ << copied << " bytes from FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef DUMP
+
+ *logofs << "AgentTransport: Child: Dumping content of read data.\n"
+ << logofs_flush;
+
+ DumpData(data, copied);
+
+ #endif
+
+ r_buffer_.length_ -= copied;
+
+ if (r_buffer_.length_ == 0)
+ {
+ r_buffer_.start_ = 0;
+ }
+ else
+ {
+ r_buffer_.start_ += copied;
+
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: There are still "
+ << r_buffer_.length_ << " bytes in read buffer for "
+ << "FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+ }
+ }
+ }
+ else
+ {
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Child: No data can be got "
+ << "from read buffer for FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ ESET(EAGAIN);
+ }
+
+ #ifdef THREADS
+
+ unlockRead();
+
+ #endif
+
+ return copied;
+}
+
+//
+// Write data to buffer so that the other
+// thread can get it.
+//
+
+int AgentTransport::write(T_write type, const unsigned char *data, const unsigned int size)
+{
+ #ifdef THREADS
+
+ lockWrite();
+
+ #endif
+
+ //
+ // Just append data to socket's write buffer.
+ // Note that we don't care if buffer exceeds
+ // the size limits set for this type of
+ // transport.
+ //
+
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: Going to append " << size
+ << " bytes to write buffer for " << "FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ int copied = -1;
+
+ if (resize(w_buffer_, size) < 0)
+ {
+ finish();
+
+ ESET(EPIPE);
+ }
+ else
+ {
+ memmove(w_buffer_.data_.begin() + w_buffer_.start_ + w_buffer_.length_, data, size);
+
+ w_buffer_.length_ += size;
+
+ #ifdef DUMP
+
+ *logofs << "AgentTransport: Child: Dumping content of written data.\n"
+ << logofs_flush;
+
+ DumpData(data, size);
+
+ #endif
+
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: Write buffer for FD#" << fd_
+ << " has data for " << w_buffer_.length_ << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "AgentTransport: Child: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " size is "
+ << w_buffer_.data_.size() << " capacity is "
+ << w_buffer_.data_.capacity() << ".\n"
+ << logofs_flush;
+ #endif
+
+ copied = size;
+ }
+
+ //
+ // Let child access again the read buffer.
+ //
+
+ #ifdef THREADS
+
+ unlockWrite();
+
+ #endif
+
+ return copied;
+}
+
+int AgentTransport::flush()
+{
+ //
+ // In case of memory-to-memory transport
+ // this function should never be called.
+ //
+
+ #ifdef PANIC
+ *logofs << "AgentTransport: Child: PANIC! Called flush() for "
+ << "memory to memory transport on " << "FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Called flush() for "
+ << "memory to memory transport on " << "FD#"
+ << fd_ << ".\n";
+
+ HandleAbort();
+}
+
+int AgentTransport::drain(int limit, int timeout)
+{
+ //
+ // We can't drain the channel in the case
+ // of the memory-to-memory transport. Data
+ // is enqueued for the agent to read but
+ // the agent could require multiple loops
+ // to read it all.
+ //
+
+ //
+ // In case of memory-to-memory transport
+ // this function should never be called.
+ //
+
+ #ifdef PANIC
+ *logofs << "AgentTransport: Child: PANIC! Called drain() for "
+ << "memory to memory transport on " << "FD#"
+ << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ cerr << "Error" << ": Called drain() for "
+ << "memory to memory transport on " << "FD#"
+ << fd_ << ".\n";
+
+ HandleAbort();
+}
+
+unsigned int AgentTransport::getPending(unsigned char *&data)
+{
+ #ifdef THREADS
+
+ lockRead();
+
+ #endif
+
+ if (r_buffer_.length_ > 0)
+ {
+ unsigned int size = r_buffer_.length_;
+
+ data = r_buffer_.data_.begin() + r_buffer_.start_;
+
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Child: Returning " << size
+ << " pending bytes from FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ r_buffer_.length_ = 0;
+ r_buffer_.start_ = 0;
+
+ #ifdef THREADS
+
+ unlockRead();
+
+ #endif
+
+ //
+ // Prevent the deletion of the buffer.
+ //
+
+ owner_ = 0;
+
+ return size;
+ }
+
+ #ifdef TEST
+ *logofs << "AgentTransport: WARNING! No pending data "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ #ifdef THREADS
+
+ unlockRead();
+
+ #endif
+
+ data = NULL;
+
+ return 0;
+}
+
+void AgentTransport::fullReset()
+{
+ #ifdef THREADS
+
+ lockRead();
+ lockWrite();
+
+ #endif
+
+ #ifdef TEST
+ *logofs << "AgentTransport: Child: Resetting transport "
+ << "for FD#" << fd_ << ".\n" << logofs_flush;
+ #endif
+
+ blocked_ = 0;
+ finish_ = 0;
+
+ if (owner_ == 1)
+ {
+ Transport::fullReset(r_buffer_);
+ }
+
+ Transport::fullReset(w_buffer_);
+}
+
+int AgentTransport::enqueue(const char *data, const int size)
+{
+ #ifdef THREADS
+
+ lockRead();
+
+ #endif
+
+ if (finish_ == 1)
+ {
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Returning EPIPE in "
+ << "write for finishing FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ ESET(EPIPE);
+
+ return -1;
+ }
+
+ //
+ // Always allow the agent to write
+ // all its data.
+ //
+
+ int toPut = size;
+
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Going to put " << toPut
+ << " bytes into read buffer for FD#" << fd_
+ << ". Buffer length is " << r_buffer_.length_
+ << ".\n" << logofs_flush;
+ #endif
+
+ if (resize(r_buffer_, toPut) < 0)
+ {
+ finish();
+
+ #ifdef THREADS
+
+ unlockRead();
+
+ #endif
+
+ return -1;
+ }
+
+ memcpy(r_buffer_.data_.begin() + r_buffer_.start_ + r_buffer_.length_, data, toPut);
+
+ r_buffer_.length_ += toPut;
+
+ #if defined(DUMP) && defined(PARENT)
+
+ *logofs << "AgentTransport: Parent: Dumping content of enqueued data.\n"
+ << logofs_flush;
+
+ DumpData((const unsigned char *) data, toPut);
+
+ #endif
+
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Read buffer for FD#" << fd_
+ << " has now data for " << r_buffer_.length_
+ << " bytes.\n" << logofs_flush;
+
+ *logofs << "AgentTransport: Parent: Start is " << r_buffer_.start_
+ << " length is " << r_buffer_.length_ << " size is "
+ << r_buffer_.data_.size() << " capacity is "
+ << r_buffer_.data_.capacity() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef THREADS
+
+ unlockRead();
+
+ #endif
+
+ return toPut;
+}
+
+int AgentTransport::dequeue(char *data, int size)
+{
+ #ifdef THREADS
+
+ lockWrite();
+
+ #endif
+
+ if (w_buffer_.length_ == 0)
+ {
+ if (finish_ == 1)
+ {
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Returning 0 in read "
+ << "for finishing FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ return 0;
+ }
+
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: No data can be read "
+ << "from write buffer for FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ ESET(EAGAIN);
+
+ #ifdef THREADS
+
+ unlockWrite();
+
+ #endif
+
+ return -1;
+ }
+
+ //
+ // Return as many bytes as possible.
+ //
+
+ int toGet = ((int) size > w_buffer_.length_ ? w_buffer_.length_ : size);
+
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Going to get " << toGet
+ << " bytes from write buffer for FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ memcpy(data, w_buffer_.data_.begin() + w_buffer_.start_, toGet);
+
+ w_buffer_.start_ += toGet;
+ w_buffer_.length_ -= toGet;
+
+ #if defined(DUMP) && defined(PARENT)
+
+ *logofs << "AgentTransport: Parent: Dumping content of dequeued data.\n"
+ << logofs_flush;
+
+ DumpData((const unsigned char *) data, toGet);
+
+ #endif
+
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Write buffer for FD#" << fd_
+ << " has now data for " << length() << " bytes.\n"
+ << logofs_flush;
+
+ *logofs << "AgentTransport: Parent: Start is " << w_buffer_.start_
+ << " length is " << w_buffer_.length_ << " size is "
+ << w_buffer_.data_.size() << " capacity is "
+ << w_buffer_.data_.capacity() << ".\n"
+ << logofs_flush;
+ #endif
+
+ #ifdef THREADS
+
+ unlockWrite();
+
+ #endif
+
+ return toGet;
+}
+
+int AgentTransport::dequeuable()
+{
+ if (finish_ == 1)
+ {
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Returning EPIPE in "
+ << "readable for finishing FD#" << fd_
+ << ".\n" << logofs_flush;
+ #endif
+
+ ESET(EPIPE);
+
+ return -1;
+ }
+
+ #if defined(PARENT) && defined(TEST)
+ *logofs << "AgentTransport: Parent: Returning "
+ << w_buffer_.length_ << " as data readable "
+ << "from read buffer for FD#" << fd_ << ".\n"
+ << logofs_flush;
+ #endif
+
+ return w_buffer_.length_;
+}
+
+#ifdef THREADS
+
+int AgentTransport::lockRead()
+{
+ for (;;)
+ {
+ int result = pthread_mutex_lock(&m_read_);
+
+ if (result == 0)
+ {
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Read mutex locked by thread id "
+ << pthread_self() << ".\n" << logofs_flush;
+ #endif
+
+ return 0;
+ }
+ else if (EGET() == EINTR)
+ {
+ continue;
+ }
+ else
+ {
+ #ifdef WARNING
+ *logofs << "AgentTransport: WARNING! Locking of read mutex by thread id "
+ << pthread_self() << " returned " << result << ". Error is '"
+ << ESTR() << "'.\n" << logofs_flush;
+ #endif
+
+ return result;
+ }
+ }
+}
+
+int AgentTransport::lockWrite()
+{
+ for (;;)
+ {
+ int result = pthread_mutex_lock(&m_write_);
+
+ if (result == 0)
+ {
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Write mutex locked by thread id "
+ << pthread_self() << ".\n" << logofs_flush;
+ #endif
+
+ return 0;
+ }
+ else if (EGET() == EINTR)
+ {
+ continue;
+ }
+ else
+ {
+ #ifdef WARNING
+ *logofs << "AgentTransport: WARNING! Locking of write mutex by thread id "
+ << pthread_self() << " returned " << result << ". Error is '"
+ << ESTR() << "'.\n" << logofs_flush;
+ #endif
+
+ return result;
+ }
+ }
+}
+
+int AgentTransport::unlockRead()
+{
+ for (;;)
+ {
+ int result = pthread_mutex_unlock(&m_read_);
+
+ if (result == 0)
+ {
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Read mutex unlocked by thread id "
+ << pthread_self() << ".\n" << logofs_flush;
+ #endif
+
+ return 0;
+ }
+ else if (EGET() == EINTR)
+ {
+ continue;
+ }
+ else
+ {
+ #ifdef WARNING
+ *logofs << "AgentTransport: WARNING! Unlocking of read mutex by thread id "
+ << pthread_self() << " returned " << result << ". Error is '"
+ << ESTR() << "'.\n" << logofs_flush;
+ #endif
+
+ return result;
+ }
+ }
+}
+
+int AgentTransport::unlockWrite()
+{
+ for (;;)
+ {
+ int result = pthread_mutex_unlock(&m_write_);
+
+ if (result == 0)
+ {
+ #ifdef DEBUG
+ *logofs << "AgentTransport: Write mutex unlocked by thread id "
+ << pthread_self() << ".\n" << logofs_flush;
+ #endif
+
+ return 0;
+ }
+ else if (EGET() == EINTR)
+ {
+ continue;
+ }
+ else
+ {
+ #ifdef WARNING
+ *logofs << "AgentTransport: WARNING! Unlocking of write mutex by thread id "
+ << pthread_self() << " returned " << result << ". Error is '"
+ << ESTR() << "'.\n" << logofs_flush;
+ #endif
+
+ return result;
+ }
+ }
+}
+
+#endif