Implement connecting and disconnecting users

[stg.git] / projects / sgauthstress / proto.h

diff --git a/projects/sgauthstress/proto.h b/projects/sgauthstress/proto.h
index 0a2fb2de7648ba3f3675d2bbb9f59aa7ddb56b2e..a63a4f90c3e6e7bf5df600363d5509fc62b9f6e2 100644 (file)
--- a/projects/sgauthstress/proto.h
+++ b/projects/sgauthstress/proto.h
@@ -31,10 +31,12 @@ 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);
+
+        size_t UserCount() const { return users.size(); }
     private:
         BLOWFISH_CTX ctx;
         struct sockaddr_in localAddr;
@@ -72,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