dump

networking/include/AsyncSslTransport.hpp

@@ -0,0 +1,62 @@
+//
+// Created by Stepan Usatiuk on 14.12.2024.
+//
+
+#ifndef MESSAGEPUMP_HPP
+#define MESSAGEPUMP_HPP
+
+#include <condition_variable>
+#include <deque>
+#include <future>
+#include <mutex>
+#include <thread>
+#include <unordered_map>
+
+#include <openssl/ssl.h>
+
+#include "Helpers.hpp"
+
+class AsyncSslTransport {
+public:
+    AsyncSslTransport(SSL_CTX* ssl_ctx, int fd);
+    virtual ~AsyncSslTransport() = 0;
+    void run();
+
+    void send_message(std::shared_ptr<MsgWrapper> msg);
+
+    bool is_failed() const { return _failed; }
+    bool is_stopped() const { return _stopped; }
+
+    void stop();
+
+protected:
+    virtual void handle_message(std::shared_ptr<MsgWrapper> msg) = 0;
+    virtual void handle_fail()                                   = 0;
+
+    virtual void before_entry() {}
+
+    std::unique_ptr<SSL, decltype(&SSL_free)> _ssl{nullptr, &SSL_free};
+    int                                       _fd;
+
+private:
+    void thread_entry();
+
+    std::atomic<bool>       _stopped = 0;
+    std::mutex              _stopped_mutex;
+    std::condition_variable _stopped_condition;
+
+    std::thread _thread;
+
+    std::mutex                              _to_send_mutex;
+    int                                     _to_send_notif_pipe[2];
+    std::deque<std::shared_ptr<MsgWrapper>> _to_send;
+
+    std::atomic<bool> _failed;
+
+    AsyncSslTransport(const AsyncSslTransport& other)                = delete;
+    AsyncSslTransport(AsyncSslTransport&& other) noexcept            = delete;
+    AsyncSslTransport& operator=(const AsyncSslTransport& other)     = delete;
+    AsyncSslTransport& operator=(AsyncSslTransport&& other) noexcept = delete;
+};
+
+#endif // MESSAGEPUMP_HPP