X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/3271a865ea0cf4a74d41b97148ccc2356be756b7..39c6313308afb248b140d98b8052f1b484549df2:/projects/sgauthstress/user.h

diff --git a/projects/sgauthstress/user.h b/projects/sgauthstress/user.h
index f494e5db..ef63a907 100644
--- a/projects/sgauthstress/user.h
+++ b/projects/sgauthstress/user.h
@@ -5,29 +5,34 @@
 #include <string>
 
 #include "stg/os_int.h"
-#include "stg/ia.h"
+#include "stg/blowfish.h"
 
 class USER {
     public:
         USER(const std::string & login,
              const std::string & password,
              uint32_t ip);
+        USER(const USER & rvalue);
         ~USER();
 
+        USER & operator=(const USER & rvalue);
+
+        bool InitNetwork();
+
         const std::string & GetLogin() const { return login; }
         uint32_t GetIP() const { return ip; }
         uint32_t GetAliveTimeout() const { return aliveTimeout; }
         uint32_t GetUserTimeout() const { return userTimeout; }
         int GetPhase() const { return phase; }
-        int GetRnd() const { return rnd; }
+        uint32_t GetRnd() const { return rnd; }
         int GetSocket() const { return sock; }
         time_t GetPhaseChangeTime() const { return phaseChangeTime; }
 
         BLOWFISH_CTX * GetCtx() { return &ctx; }
 
         void SetPhase(int p) { phase = p; time(&phaseChangeTime); }
-        void SetRnd(int r) { rnd = r; }
-        int IncRnd() { return ++rnd; }
+        void SetRnd(uint32_t r) { rnd = r; }
+        uint32_t IncRnd() { return ++rnd; }
         void SetAliveTimeout(uint32_t timeout) { aliveTimeout = timeout; }
         void SetUserTimeout(uint32_t timeout) { userTimeout = timeout; }
 
@@ -39,7 +44,7 @@ class USER {
         uint32_t userTimeout;
         int phase;
         time_t phaseChangeTime;
-        int rnd;
+        uint32_t rnd;
         int sock;
         BLOWFISH_CTX ctx;
 };