core: stop ownership circles in connection

1 files changed, 12 insertions, 1 deletions

diff --git a/src/control_interface.hpp b/src/control_interface.hpp
index c69aeb4..51d8c62 100644
--- a/src/control_interface.hpp
+++ b/src/control_interface.hpp
@@ -7,11 +7,14 @@
 #include <boost/asio.hpp>
 #include <boost/system/error_code.hpp>
 
+#include <cstddef>
 #include <filesystem>
 #include <istream>
+#include <map>
 #include <memory>
 #include <set>
 #include <string>
+#include <type_traits>
 
 namespace wanda
 {
@@ -33,9 +36,16 @@ struct control_interface : control_connection::listener, keyed<control_interface
     boost::system::error_code start();
     boost::system::error_code shutdown();
 
-    void on_close(control_connection::pointer) override;
+    void on_close(control_connection::pointer connection) override;
+    void on_received(control_connection::pointer connection, message message) override;
 
   private:
+    enum struct state : std::underlying_type_t<std::byte>
+    {
+        fresh,
+        greeted,
+    };
+
     void perform_accept();
 
     friend pointer make_interface(boost::asio::io_service &service, std::filesystem::path file);
@@ -46,6 +56,7 @@ struct control_interface : control_connection::listener, keyed<control_interface
     protocol::acceptor m_acceptor;
     socket_deleter m_deleter{m_endpoint.path()};
     std::set<control_connection::pointer> m_connections;
+    std::map<control_connection::pointer, state> m_states;
 };
 
 control_interface::pointer make_interface(boost::asio::io_service &service, std::filesystem::path file);