folly::BlockingWriteServer Class Reference

#include <AsyncSSLSocketTest.h>

Inheritance diagram for folly::BlockingWriteServer:

Public Member Functions
	BlockingWriteServer (AsyncSSLSocket::UniquePtr socket)
void	checkBuffer (struct iovec *iov, uint32_t count) const

Private Member Functions
void	handshakeSuc (AsyncSSLSocket *) noexceptoverride
void	handshakeErr (AsyncSSLSocket *, const AsyncSocketException &ex) noexceptoverride
void	getReadBuffer (void **bufReturn, size_t *lenReturn) override
void	readDataAvailable (size_t len) noexceptoverride
void	readEOF () noexceptoverride
void	readErr (const AsyncSocketException &ex) noexceptoverride
Private Member Functions inherited from folly::AsyncSSLSocket::HandshakeCB
virtual	~HandshakeCB ()=default
virtual bool	handshakeVer (AsyncSSLSocket *, bool preverifyOk, X509_STORE_CTX *) noexcept
Private Member Functions inherited from folly::AsyncReader::ReadCallback
virtual	~ReadCallback ()=default
virtual bool	isBufferMovable () noexcept
virtual size_t	maxBufferSize () const
virtual void	readBufferAvailable (std::unique_ptr< IOBuf >) noexcept

Private Attributes
AsyncSSLSocket::UniquePtr	socket_
uint32_t	bufSize_
uint32_t	bytesRead_
std::unique_ptr< uint8_t[]>	buf_

Detailed Description

Definition at line 848 of file AsyncSSLSocketTest.h.

Constructor & Destructor Documentation

folly::BlockingWriteServer::BlockingWriteServer ( AsyncSSLSocket::UniquePtr  socket )

inlineexplicit

Definition at line 851 of file AsyncSSLSocketTest.h.

References folly::SendMsgParamsCallbackBase::socket_.

      : socket_(std::move(socket)), bufSize_(2500 * 2000), bytesRead_(0) {
    buf_ = std::make_unique<uint8_t[]>(bufSize_);
    socket_->sslAccept(this, std::chrono::milliseconds(100));
  }

Member Function Documentation

void folly::BlockingWriteServer::checkBuffer ( struct iovec *  iov, uint32_t  count  ) const

inline

Definition at line 857 of file AsyncSSLSocketTest.h.

References ADD_FAILURE, count, FAIL, min, and uint32_t.

                                                            {
    uint32_t idx = 0;
    for (uint32_t n = 0; n < count; ++n) {
      size_t bytesLeft = bytesRead_ - idx;
      int rc = memcmp(
          buf_.get() + idx,
          iov[n].iov_base,
          std::min(iov[n].iov_len, bytesLeft));
      if (rc != 0) {
        FAIL() << "buffer mismatch at iovec " << n << "/" << count
               << ": rc=" << rc;
      }
      if (iov[n].iov_len > bytesLeft) {
        FAIL() << "server did not read enough data: "
               << "ended at byte " << bytesLeft << "/" << iov[n].iov_len
               << " in iovec " << n << "/" << count;
      }

      idx += iov[n].iov_len;
    }
    if (idx != bytesRead_) {
      ADD_FAILURE() << "server read extra data: " << bytesRead_
                    << " bytes read; expected " << idx;
    }
  }

void folly::BlockingWriteServer::getReadBuffer ( void **  bufReturn, size_t *  lenReturn  )

inlineoverrideprivatevirtual

When data becomes available, getReadBuffer() will be invoked to get the buffer into which data should be read.

This method allows the ReadCallback to delay buffer allocation until data becomes available. This allows applications to manage large numbers of idle connections, without having to maintain a separate read buffer for each idle connection.

It is possible that in some cases, getReadBuffer() may be called multiple times before readDataAvailable() is invoked. In this case, the data will be written to the buffer returned from the most recent call to readDataAvailable(). If the previous calls to readDataAvailable() returned different buffers, the ReadCallback is responsible for ensuring that they are not leaked.

If getReadBuffer() throws an exception, returns a nullptr buffer, or returns a 0 length, the ReadCallback will be uninstalled and its readError() method will be invoked.

getReadBuffer() is not allowed to change the transport state before it returns. (For example, it should never uninstall the read callback, or set a different read callback.)

Parameters

bufReturn	getReadBuffer() should update *bufReturn to contain the address of the read buffer. This parameter will never be nullptr.
lenReturn	getReadBuffer() should update *lenReturn to contain the maximum number of bytes that may be written to the read buffer. This parameter will never be nullptr.

Implements folly::AsyncReader::ReadCallback.

Definition at line 894 of file AsyncSSLSocketTest.h.

                                                                   {
    *bufReturn = buf_.get() + bytesRead_;
    *lenReturn = bufSize_ - bytesRead_;
  }

void folly::BlockingWriteServer::handshakeErr ( AsyncSSLSocket *  sock, const AsyncSocketException &  ex  )

inlineoverrideprivatevirtualnoexcept

handshakeErr() is called if an error occurs while establishing the SSL connection.

The HandshakeCB will be uninstalled before handshakeErr() is called.

Parameters

sock	SSL socket on which the handshake was initiated
ex	An exception representing the error.

Implements folly::AsyncSSLSocket::HandshakeCB.

Definition at line 889 of file AsyncSSLSocketTest.h.

References ADD_FAILURE.

                                                        {
    ADD_FAILURE() << "server handshake error: " << ex.what();
  }

void folly::BlockingWriteServer::handshakeSuc ( AsyncSSLSocket *  sock )

inlineoverrideprivatevirtualnoexcept

handshakeSuc() is called when a new SSL connection is established, i.e., after SSL_accept/connect() returns successfully.

The HandshakeCB will be uninstalled before handshakeSuc() is called.

Parameters

sock	SSL socket on which the handshake was initiated

Implements folly::AsyncSSLSocket::HandshakeCB.

Definition at line 884 of file AsyncSSLSocketTest.h.

References folly::SendMsgParamsCallbackBase::socket_.

                                                       {
    // Wait 10ms before reading, so the client's writes will initially block.
    socket_->getEventBase()->tryRunAfterDelay(
        [this] { socket_->setReadCB(this); }, 10);
  }

void folly::BlockingWriteServer::readDataAvailable ( size_t  len )

inlineoverrideprivatevirtualnoexcept

readDataAvailable() will be invoked when data has been successfully read into the buffer returned by the last call to getReadBuffer().

The read callback remains installed after readDataAvailable() returns. It must be explicitly uninstalled to stop receiving read events. getReadBuffer() will be called at least once before each call to readDataAvailable(). getReadBuffer() will also be called before any call to readEOF().

Parameters

len	The number of bytes placed in the buffer.

Implements folly::AsyncReader::ReadCallback.

Definition at line 898 of file AsyncSSLSocketTest.h.

References folly::SendMsgParamsCallbackBase::socket_.

                                                       {
    bytesRead_ += len;
    socket_->setReadCB(nullptr);
    socket_->getEventBase()->tryRunAfterDelay(
        [this] { socket_->setReadCB(this); }, 2);
  }

void folly::BlockingWriteServer::readEOF ( )

inlineoverrideprivatevirtualnoexcept

readEOF() will be invoked when the transport is closed.

The read callback will be automatically uninstalled immediately before readEOF() is invoked.

Implements folly::AsyncReader::ReadCallback.

Definition at line 904 of file AsyncSSLSocketTest.h.

References folly::SendMsgParamsCallbackBase::socket_.

                                   {
    socket_->close();
  }

void folly::BlockingWriteServer::readErr ( const AsyncSocketException &  ex )

inlineoverrideprivatevirtualnoexcept

readError() will be invoked if an error occurs reading from the transport.

The read callback will be automatically uninstalled immediately before readError() is invoked.

Parameters

ex	An exception describing the error that occurred.

Implements folly::AsyncReader::ReadCallback.

Definition at line 907 of file AsyncSSLSocketTest.h.

References ADD_FAILURE.

                                                                 {
    ADD_FAILURE() << "server read error: " << ex.what();
  }

Member Data Documentation

std::unique_ptr<uint8_t[]> folly::BlockingWriteServer::buf_

private

Definition at line 914 of file AsyncSSLSocketTest.h.

uint32_t folly::BlockingWriteServer::bufSize_

private

Definition at line 912 of file AsyncSSLSocketTest.h.

uint32_t folly::BlockingWriteServer::bytesRead_

private

Definition at line 913 of file AsyncSSLSocketTest.h.

AsyncSSLSocket::UniquePtr folly::BlockingWriteServer::socket_

private

Definition at line 911 of file AsyncSSLSocketTest.h.

---

The documentation for this class was generated from the following file:

• proxygen/folly/folly/io/async/test/AsyncSSLSocketTest.h