Add proper header to use struct sockaddr_in in sgauthstress

[stg.git] / projects / stargazer / plugins / other / rscript / send_functor.h

diff --git a/projects/stargazer/plugins/other/rscript/send_functor.h b/projects/stargazer/plugins/other/rscript/send_functor.h
index 479cdc3b84275b6fc67ae32a0f32b393b7ac46c2..b9b789eaf1556ed034ae350b29b6479d96714b63 100644 (file)
--- a/projects/stargazer/plugins/other/rscript/send_functor.h
+++ b/projects/stargazer/plugins/other/rscript/send_functor.h
@@ -32,25 +32,24 @@
 
 #include <functional>
 
 
 #include <functional>
 

-#include "os_int.h"
+#include "stg/os_int.h"

 
 

-class PacketSender : public std::unary_function<uint32_t, void> {
+class PacketSender : public std::unary_function<uint32_t, int> {

     public:
         PacketSender(int s, char * b, int l, uint16_t p)
             : sock(s),
               buffer(b),
               length(l),
               port(p) {};
     public:
         PacketSender(int s, char * b, int l, uint16_t p)
             : sock(s),
               buffer(b),
               length(l),
               port(p) {};

-        void operator() (uint32_t ip)
+        int operator() (uint32_t ip)

         {
         {

-        int res;

         struct sockaddr_in sendAddr;
 
         sendAddr.sin_family = AF_INET;
         sendAddr.sin_port = port;
         sendAddr.sin_addr.s_addr = ip;
 
         struct sockaddr_in sendAddr;
 
         sendAddr.sin_family = AF_INET;
         sendAddr.sin_port = port;
         sendAddr.sin_addr.s_addr = ip;
 

-        res = sendto(sock, buffer, length, 0, (struct sockaddr*)&sendAddr, sizeof(sendAddr));
+        return sendto(sock, buffer, length, 0, (struct sockaddr*)&sendAddr, sizeof(sendAddr));

         }
     private:
         int sock;
         }
     private:
         int sock;