X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/07bd133ff9a290ff7475c94c1578db3d0d91b853..3cc6c36fcf3f0c6449debeb56e53c2ede03efc26:/projects/sgauthstress/proto.h

diff --git a/projects/sgauthstress/proto.h b/projects/sgauthstress/proto.h
index 05fe4722..a63a4f90 100644
--- a/projects/sgauthstress/proto.h
+++ b/projects/sgauthstress/proto.h
@@ -31,7 +31,7 @@ class PROTO {
 
         const std::string GetStrError() const { return errorStr; }
 
-        void AddUser(const USER & user);
+        void AddUser(const USER & user, bool connect = false);
 
         bool Connect(uint32_t ip);
         bool Disconnect(uint32_t ip);
@@ -74,6 +74,9 @@ class PROTO {
         bool Send_DISCONN_SYN(USER * user);
         bool Send_DISCONN_ACK(USER * user);
         bool Send_ALIVE_ACK(USER * user);
+
+        bool RealConnect(USER * user);
+        bool RealDisconnect(USER * user);
 };
 
 #endif