Fixed compilation with new TRAFF_TYPE.

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

diff --git a/projects/sgauthstress/proto.h b/projects/sgauthstress/proto.h
index ed17f876d6544c8f40310a5066ef61d87c5dc4eb..746622b77b40a1c9794b430f6770513a2828731b 100644 (file)
--- a/projects/sgauthstress/proto.h
+++ b/projects/sgauthstress/proto.h
@@ -1,6 +1,9 @@
 #ifndef __PROTO_H__
 #define __PROTO_H__
 
+#ifndef LINUX
+#include <netinet/in_systm.h>
+#endif
 #include <netinet/ip.h>
 #include <pthread.h>
 #include <poll.h>
@@ -60,6 +63,7 @@ class PROTO {
         static void * Runner(void * data);
 
         void Run();
+        void CheckTimeouts();
         bool RecvPacket();
         bool SendPacket(const void * buffer, size_t length, USER * user);
         bool HandlePacket(const char * buffer, size_t length, USER * user);