X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/34ef822e81b9f236b2f5edf52d351a0f82d59a0c..20a6f8130099fb69d67e7a58fa9935caa9be32c5:/projects/sgauthstress/proto.h?ds=sidebyside

diff --git a/projects/sgauthstress/proto.h b/projects/sgauthstress/proto.h
index 879e7a35..05fe4722 100644
--- a/projects/sgauthstress/proto.h
+++ b/projects/sgauthstress/proto.h
@@ -6,6 +6,7 @@
 #include <poll.h>
 
 #include <string>
+#include <vector>
 #include <map>
 
 #include "stg/os_int.h"
@@ -34,13 +35,15 @@ class PROTO {
 
         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;
         struct sockaddr_in serverAddr;
         int timeout;
 
-        std::map<uint32_t, USER> users;
+        std::vector<std::pair<uint32_t, USER> > users;
         std::vector<struct pollfd> pollFds;
 
         bool running;