Index: third_party/libva/test/v4l_h264/decode/TCPSocketServer.cpp |
diff --git a/third_party/libva/test/v4l_h264/decode/TCPSocketServer.cpp b/third_party/libva/test/v4l_h264/decode/TCPSocketServer.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..ad33d77bbf77f7ce6ebc53528cade65c8e11c7de |
--- /dev/null |
+++ b/third_party/libva/test/v4l_h264/decode/TCPSocketServer.cpp |
@@ -0,0 +1,192 @@ |
+/* |
+ * Copyright (c) 2012 Intel Corporation. All Rights Reserved. |
+ * |
+ * Permission is hereby granted, free of charge, to any person obtaining a |
+ * copy of this software and associated documentation files (the |
+ * "Software"), to deal in the Software without restriction, including |
+ * without limitation the rights to use, copy, modify, merge, publish, |
+ * distribute, sub license, and/or sell copies of the Software, and to |
+ * permit persons to whom the Software is furnished to do so, subject to |
+ * the following conditions: |
+ * |
+ * The above copyright notice and this permission notice (including the |
+ * next paragraph) shall be included in all copies or substantial portions |
+ * of the Software. |
+ * |
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES |
+ * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. |
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, |
+ * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
+ * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
+ * USE OR OTHER DEALINGS IN THE SOFTWARE. |
+ */ |
+ |
+/* |
+* C++ wrapper around an TCP socket |
+*/ |
+ |
+#include <stdio.h> |
+#include <stdlib.h> |
+#include <sys/types.h> // for data types |
+#include <sys/socket.h> // for socket(), connect(), send(), recv() |
+#include <netinet/in.h> // for IPPROTO_TCP, sockadd_in |
+#include <arpa/inet.h> // for inet_ntoa() |
+#include <unistd.h> // for close() |
+#include <netdb.h> // for hostent, gethostbyname() |
+#include <fcntl.h> // for fcntl() |
+#include <errno.h> |
+ |
+#include <cstring> // for memset |
+ |
+#include "TCPSocketServer.h" |
+ |
+using std::string; |
+ |
+ |
+TCPSocketServer::TCPSocketServer(unsigned short localPort) throw(std::runtime_error) : |
+sockDesc(-1), |
+ connSockDesc(-1) |
+{ |
+ // create new socket |
+ if ((sockDesc = socket(PF_INET, SOCK_STREAM, 0)) < 0) { |
+ throw std::runtime_error("Socket creation failed (socket())"); |
+ } |
+ |
+ ::memset(&sockAddr, 0, sizeof(sockAddr)); |
+ ::memset(&connSockAddr, 0, sizeof(connSockAddr)); |
+ |
+ sockAddr.sin_family = AF_INET; |
+ sockAddr.sin_addr.s_addr = htonl(INADDR_ANY); |
+ sockAddr.sin_port = htons(localPort); |
+ |
+ if (bind(sockDesc, (sockaddr *) &sockAddr, sizeof(sockAddr)) < 0) |
+ { |
+ throw std::runtime_error("Bind to local port failed (bind())"); |
+ } |
+ |
+ if (listen(sockDesc, 2) < 0) |
+ { |
+ throw std::runtime_error("Socket initialization failed (listen())"); |
+ } |
+} |
+ |
+ |
+/* Destructor */ |
+TCPSocketServer::~TCPSocketServer() |
+{ |
+ if (connSockDesc > 0) { |
+ ::close(connSockDesc); |
+ } |
+ |
+ ::close(sockDesc); |
+} |
+ |
+ |
+/* Listen for an incoming connection */ |
+void TCPSocketServer::accept(string &remoteAddr, unsigned short &remotePort) throw (std::runtime_error) |
+{ |
+ if (connSockDesc > 0) { |
+ throw std::runtime_error("accept() called, but socket is already connected"); |
+ } |
+ |
+ socklen_t connSockLen = sizeof(connSockAddr); |
+ if ((connSockDesc = ::accept(sockDesc, (sockaddr *) &connSockAddr, &connSockLen)) < 0) { |
+ throw std::runtime_error("Connection accept failed (accept())"); |
+ } |
+ |
+ remoteAddr = inet_ntoa(connSockAddr.sin_addr); |
+ remotePort = ntohs(connSockAddr.sin_port); |
+} |
+ |
+ |
+/* Communication over socket */ |
+/* Receive data */ |
+ssize_t TCPSocketServer::recv(void *buffer, const size_t &bufferLen) throw (std::runtime_error) |
+{ |
+ if (connSockDesc <= 0) { |
+ throw std::runtime_error("recv() called, but socket is not connected. Call accept() first"); |
+ } |
+ |
+ int rval = ::read(connSockDesc, buffer, bufferLen); |
+ |
+ if (rval <= 0) { |
+ // EOF (connection closed by remote host) or error: |
+ // reset state, so a new accept() call will succeed |
+ connSockDesc = -1; |
+ ::memset(&connSockAddr, 0, sizeof(connSockAddr)); |
+ } |
+ |
+ if (rval == -1) { |
+ throw std::runtime_error("Error reading from socket (read())"); |
+ } |
+ |
+ return rval; |
+} |
+ |
+ |
+/* Send data */ |
+ssize_t TCPSocketServer::send(const void *buffer, const int &bufferLen) throw (std::runtime_error) |
+{ |
+ if (connSockDesc <= 0) { |
+ throw std::runtime_error("send() called, but socket is not connected. Call accept() first"); |
+ } |
+ |
+ int rval = ::write(connSockDesc, buffer, bufferLen); |
+ |
+ if (rval <= 0) { |
+ // EOF (connection closed by remote host) or error: |
+ // reset state, so a new accept() call will succeed |
+ connSockDesc = -1; |
+ ::memset(&connSockAddr, 0, sizeof(connSockAddr)); |
+ } |
+ |
+ if (rval == -1) { |
+ throw std::runtime_error("Error reading from socket (read())"); |
+ } |
+ |
+ return rval; |
+} |
+ |
+ssize_t TCPSocketServer::send(const string &message) throw (std::runtime_error) |
+{ |
+ send(message.c_str(), message.length()); |
+} |
+ |
+ |
+int TCPSocketServer::recv_data(unsigned char *data, int size) |
+{ |
+ int total = 0; |
+ try { |
+ int pos = 0; |
+ int recvMsgSize; |
+ for (;;) { |
+ // try to receive a message |
+ recvMsgSize = recv(&data[pos], size); |
+ if ((recvMsgSize < 0) && (errno == EWOULDBLOCK)) { |
+ // no data received on non-blocking socket |
+ usleep(100); |
+ } else { |
+ total += recvMsgSize; |
+ if (recvMsgSize != size) { |
+ pos += recvMsgSize; |
+ size -= recvMsgSize; |
+ } else { |
+ break; |
+ } |
+ } |
+ } |
+ } |
+ catch (const std::exception& e) { |
+ printf("%s\n", e.what()); |
+ exit(1); |
+ } |
+ return total; |
+} |
+ |
+unsigned int TCPSocketServer::recv_uint32() |
+{ |
+ unsigned int buffer; |
+ recv_data((unsigned char*)&buffer, 4); |
+ return buffer; |
+} |