8 #include "common/include/SocketConnectionException.hpp" 24 memset(m_Socket, 0,
sizeof(m_Socket));
26 m_Socket[0].fd = socket(PF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0);
27 if (m_Socket[0].fd == -1)
29 throw armnnProfiling::SocketConnectionException(
30 std::string(
"SocketProfilingConnection: Socket construction failed: ") + strerror(errno),
37 memset(&server, 0,
sizeof(sockaddr_un));
39 memcpy(server.sun_path, m_GatorNamespace, strlen(m_GatorNamespace + 1) + 1);
40 server.sun_family = AF_UNIX;
41 if (0 != connect(m_Socket[0].fd, reinterpret_cast<const sockaddr*>(&server),
sizeof(sockaddr_un)))
44 throw armnnProfiling::SocketConnectionException(
45 std::string(
"SocketProfilingConnection: Cannot connect to stream socket: ") + strerror(errno),
51 m_Socket[0].events = POLLIN;
57 throw armnnProfiling::SocketConnectionException(
58 std::string(
"SocketProfilingConnection: Failed to set socket as non blocking: ") + strerror(errno),
66 return m_Socket[0].fd > 0;
73 throw armnnProfiling::SocketConnectionException(
74 std::string(
"SocketProfilingConnection: Cannot close stream socket: ") + strerror(errno),
79 memset(m_Socket, 0,
sizeof(m_Socket));
84 if (buffer ==
nullptr || length == 0)
95 int bytes_available = 0;
97 if (bytes_available >= 8)
100 return ReceivePacket();
104 int pollResult =
Sockets::Poll(&m_Socket[0], 1, static_cast<int>(timeout));
109 throw armnnProfiling::SocketConnectionException(
110 std::string(
"SocketProfilingConnection: Error occured while reading from socket: ") + strerror(errno),
115 throw TimeoutException(
"SocketProfilingConnection: Timeout while reading from socket");
119 if (m_Socket[0].revents & (POLLNVAL | POLLERR | POLLHUP))
121 if (m_Socket[0].revents == POLLNVAL)
125 throw armnnProfiling::SocketConnectionException(
126 std::string(
"SocketProfilingConnection: Error occured while polling receiving socket: POLLNVAL."),
129 if (m_Socket[0].revents == POLLERR)
131 throw armnnProfiling::SocketConnectionException(
133 "SocketProfilingConnection: Error occured while polling receiving socket: POLLERR: ")
138 if (m_Socket[0].revents == POLLHUP)
142 throw armnnProfiling::SocketConnectionException(
143 std::string(
"SocketProfilingConnection: Connection closed by remote client: POLLHUP."),
149 if (!(m_Socket[0].revents & (POLLIN)))
154 "SocketProfilingConnection: File descriptor was polled but no data was available to receive.");
157 return ReceivePacket();
161 Packet SocketProfilingConnection::ReceivePacket()
164 long receiveResult =
Sockets::Read(m_Socket[0].fd, &header,
sizeof(header));
166 switch( receiveResult )
171 throw armnnProfiling::SocketConnectionException(
172 std::string(
"SocketProfilingConnection: Remote socket has closed the connection."),
177 throw armnnProfiling::SocketConnectionException(
178 std::string(
"SocketProfilingConnection: Error occured while reading the packet: ") + strerror(errno),
182 if (receiveResult < 8)
184 throw armnnProfiling::SocketConnectionException(
186 "SocketProfilingConnection: The received packet did not contains a valid PIPE header."),
193 uint32_t metadataIdentifier = 0;
194 std::memcpy(&metadataIdentifier, header,
sizeof(metadataIdentifier));
197 uint32_t dataLength = 0;
198 std::memcpy(&dataLength, header + 4u,
sizeof(dataLength));
200 std::unique_ptr<unsigned char[]> packetData;
203 packetData = std::make_unique<unsigned char[]>(dataLength);
204 long receivedLength =
Sockets::Read(m_Socket[0].fd, packetData.get(), dataLength);
205 if (receivedLength < 0)
207 throw armnnProfiling::SocketConnectionException(
208 std::string(
"SocketProfilingConnection: Error occured while reading the packet: ") + strerror(errno),
212 if (dataLength != static_cast<uint32_t>(receivedLength))
215 throw armnnProfiling::SocketConnectionException(
216 std::string(
"SocketProfilingConnection: Invalid PIPE packet."),
221 return Packet(metadataIdentifier, dataLength, packetData);
bool IsOpen() const final
int Poll(PollFd *fds, nfds_t numFds, int timeout)
Copyright (c) 2020 ARM Limited.
int Ioctl(Socket s, unsigned long int cmd, void *arg)
bool Initialize()
Performs any required one-time setup.
bool SetNonBlocking(Socket s)
long Write(Socket s, const void *buf, size_t len)
long Read(Socket s, void *buf, size_t len)
SocketProfilingConnection()
bool WritePacket(const unsigned char *buffer, uint32_t length) final
Packet ReadPacket(uint32_t timeout) final