diff options
Diffstat (limited to 'profiling/common')
-rw-r--r-- | profiling/common/include/NetworkSockets.hpp | 2 | ||||
-rw-r--r-- | profiling/common/include/SocketConnectionException.hpp | 10 | ||||
-rw-r--r-- | profiling/common/src/NetworkSockets.cpp | 4 |
3 files changed, 10 insertions, 6 deletions
diff --git a/profiling/common/include/NetworkSockets.hpp b/profiling/common/include/NetworkSockets.hpp index b9e58aac1d..77507644e6 100644 --- a/profiling/common/include/NetworkSockets.hpp +++ b/profiling/common/include/NetworkSockets.hpp @@ -34,6 +34,8 @@ using PollFd = pollfd; using Socket = SOCKET; using PollFd = WSAPOLLFD; +using nfds_t = int; +using socklen_t = int; #define SOCK_CLOEXEC 0 #endif diff --git a/profiling/common/include/SocketConnectionException.hpp b/profiling/common/include/SocketConnectionException.hpp index 58b8a14d6d..fceaa0f697 100644 --- a/profiling/common/include/SocketConnectionException.hpp +++ b/profiling/common/include/SocketConnectionException.hpp @@ -8,6 +8,8 @@ #include <stdexcept> #include <string> +#include "NetworkSockets.hpp" + namespace armnnProfiling { @@ -15,11 +17,11 @@ namespace armnnProfiling class SocketConnectionException : public std::exception { public: - explicit SocketConnectionException(const std::string& message, int socket) + explicit SocketConnectionException(const std::string& message, armnnUtils::Sockets::Socket socket) : m_Message(message), m_Socket(socket), m_ErrNo(-1) {}; - explicit SocketConnectionException(const std::string& message, int socket, int errNo) + explicit SocketConnectionException(const std::string& message, armnnUtils::Sockets::Socket socket, int errNo) : m_Message(message), m_Socket(socket), m_ErrNo(errNo) {}; @@ -31,7 +33,7 @@ public: /// @return - Socket File Descriptor of SocketProfilingConnection /// or '-1', an invalid file descriptor - int GetSocketFd() const noexcept + armnnUtils::Sockets::Socket GetSocketFd() const noexcept { return m_Socket; } @@ -44,7 +46,7 @@ public: private: std::string m_Message; - int m_Socket; + armnnUtils::Sockets::Socket m_Socket; int m_ErrNo; }; diff --git a/profiling/common/src/NetworkSockets.cpp b/profiling/common/src/NetworkSockets.cpp index 8ce5f197c1..1e1f70112d 100644 --- a/profiling/common/src/NetworkSockets.cpp +++ b/profiling/common/src/NetworkSockets.cpp @@ -52,7 +52,7 @@ long Write(Socket s, const void* buf, size_t len) #if defined(__unix__) return write(s, buf, len); #elif defined(_MSC_VER) - return send(s, static_cast<const char*>(buf), len, 0); + return send(s, static_cast<const char*>(buf), static_cast<int>(len), 0); #endif } @@ -62,7 +62,7 @@ long Read(Socket s, void* buf, size_t len) #if defined(__unix__) return read(s, buf, len); #elif defined(_MSC_VER) - return recv(s, static_cast<char*>(buf), len, 0); + return recv(s, static_cast<char*>(buf), static_cast<int>(len), 0); #endif } |