fixed transmission control

[blank.git] / src / client / MasterState.hpp

diff --git a/src/client/MasterState.hpp b/src/client/MasterState.hpp
index af2e6aac8710a7b92e4af91566766e8387a43a0c..6bdb671836a9193a54068b67132a1284dad0e4c4 100644 (file)
--- a/src/client/MasterState.hpp
+++ b/src/client/MasterState.hpp
@@ -5,7 +5,7 @@
 #include "InteractiveState.hpp"
 #include "../app/State.hpp"
 #include "../net/Client.hpp"
-#include "../net/PacketHandler.hpp"
+#include "../net/ConnectionHandler.hpp"
 
 #include <memory>
 
@@ -20,7 +20,7 @@ class InteractiveState;
 
 class MasterState
 : public State
-, public PacketHandler {
+, public ConnectionHandler {
 
 public:
        MasterState(
@@ -46,6 +46,9 @@ public:
        void Update(int dt) override;
        void Render(Viewport &) override;
 
+       void OnPacketLost(std::uint16_t) override;
+       void OnTimeout() override;
+
        void On(const Packet::Join &) override;
        void On(const Packet::Part &) override;
 
@@ -59,6 +62,8 @@ private:
 
        InitialState init_state;
 
+       int login_packet;
+
 };
 
 }