Add support for sending data streams via callback (TcpConnection::sendStream()) (an-tao#196)

Author: Greisby

CMakeLists.txt

@@ -148,6 +148,9 @@ if(HAVE_C-ARES)
   set(private_headers
       ${private_headers}
       trantor/net/inner/AresResolver.h)
+  if(APPLE)
+      target_link_libraries(${PROJECT_NAME} PRIVATE resolv)
+  endif()
 else()
   set(TRANTOR_SOURCES
       ${TRANTOR_SOURCES}

trantor/net/TcpConnection.h

@@ -76,6 +76,19 @@ class TRANTOR_EXPORT TcpConnection
     virtual void sendFile(const wchar_t *fileName,
                           size_t offset = 0,
                           size_t length = 0) = 0;
+    /**
+     * @brief Send a stream to the peer.
+     *
+     * @param callback function to retrieve the stream data (stream ends when a
+     * zero size is returned) the callback will be called with nullptr when the
+     * send is finished/interrupted, so that it cleans up any internal data (ex:
+     * close file).
+     * @warning The buffer size should be >= 10 to allow http chunked-encoding
+     * data stream
+     */
+    virtual void sendStream(std::function<std::size_t(char *, std::size_t)>
+                                callback) = 0;  // (buffer, buffer size) -> size
+                                                // of data put in buffer
 
     /**
      * @brief Get the local address of the connection.

trantor/net/inner/Socket.cc

@@ -228,7 +228,11 @@ int Socket::getSocketError()
 
     if (::getsockopt(sockFd_, SOL_SOCKET, SO_ERROR, &optval, &optlen) < 0)
     {
+#ifdef _WIN32
+        return ::WSAGetLastError();
+#else
         return errno;
+#endif
     }
     else
     {