NetString::WriteStringToStream(): add Boost ASIO overload

diff --git a/lib/base/netstring.cpp b/lib/base/netstring.cpp
index 7fad3453cb8297062506fb8ff874fc5973e62355..debed2bda8dfb0e38619afd681d45c09b80a322a 100644 (file)
--- a/lib/base/netstring.cpp
+++ b/lib/base/netstring.cpp
@@ -2,7 +2,12 @@
 
 #include "base/netstring.hpp"
 #include "base/debug.hpp"
+#include "base/tlsstream.hpp"
+#include <memory>
 #include <sstream>
+#include <boost/asio/buffer.hpp>
+#include <boost/asio/spawn.hpp>
+#include <boost/asio/write.hpp>
 
 using namespace icinga;
 
@@ -110,6 +115,29 @@ size_t NetString::WriteStringToStream(const Stream::Ptr& stream, const String& s
        return msg.GetLength();
 }
 
+/**
+ * Writes data into a stream using the netstring format and returns bytes written.
+ *
+ * @param stream The stream.
+ * @param str The String that is to be written.
+ *
+ * @return The amount of bytes written.
+ */
+size_t NetString::WriteStringToStream(const std::shared_ptr<AsioTlsStream>& stream, const String& str, boost::asio::yield_context yc)
+{
+       namespace asio = boost::asio;
+
+       std::ostringstream msgbuf;
+       WriteStringToStream(msgbuf, str);
+
+       String msg = msgbuf.str();
+       asio::const_buffer msgBuf (msg.CStr(), msg.GetLength());
+
+       asio::async_write(*stream, msgBuf, yc);
+
+       return msg.GetLength();
+}
+
 /**
  * Writes data into a stream using the netstring format.
  *

diff --git a/lib/base/netstring.hpp b/lib/base/netstring.hpp
index 10ff4e2fc06eb3c44b3a758373bab86364010a45..f54d70c1721ac0a08035c9438806d4b948fdba29 100644 (file)
--- a/lib/base/netstring.hpp
+++ b/lib/base/netstring.hpp
@@ -5,6 +5,9 @@
 
 #include "base/i2-base.hpp"
 #include "base/stream.hpp"
+#include "base/tlsstream.hpp"
+#include <memory>
+#include <boost/asio/spawn.hpp>
 
 namespace icinga
 {
@@ -24,6 +27,7 @@ public:
        static StreamReadStatus ReadStringFromStream(const Stream::Ptr& stream, String *message, StreamReadContext& context,
                bool may_wait = false, ssize_t maxMessageLength = -1);
        static size_t WriteStringToStream(const Stream::Ptr& stream, const String& message);
+       static size_t WriteStringToStream(const std::shared_ptr<AsioTlsStream>& stream, const String& message, boost::asio::yield_context yc);
        static void WriteStringToStream(std::ostream& stream, const String& message);
 
 private: