| Index: third_party/libva/test/v4l_h264/encode/TCPSocketClient.cpp
|
| diff --git a/third_party/libva/test/v4l_h264/encode/TCPSocketClient.cpp b/third_party/libva/test/v4l_h264/encode/TCPSocketClient.cpp
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..93573a87673cf56f953b74c94a512376920228e2
|
| --- /dev/null
|
| +++ b/third_party/libva/test/v4l_h264/encode/TCPSocketClient.cpp
|
| @@ -0,0 +1,123 @@
|
| +/*
|
| + * 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 "TCPSocketClient.h"
|
| +
|
| +#include <sys/types.h>
|
| +#include <sys/socket.h>
|
| +#include <netinet/in.h>
|
| +#include <arpa/inet.h>
|
| +#include <unistd.h>
|
| +#include <netdb.h>
|
| +#include <fcntl.h>
|
| +#include <errno.h>
|
| +#include <cstring>
|
| +using std::string;
|
| +
|
| +TCPSocketClient::TCPSocketClient(const std::string &remoteAddr, const unsigned short &remotePort) throw(std::runtime_error) :
|
| +sockDesc(-1)
|
| +{
|
| + ::memset(&sockAddr, 0, sizeof(sockAddr));
|
| + sockAddr.sin_family = AF_INET;
|
| + struct hostent* hp;
|
| + if ((hp = gethostbyname(remoteAddr.c_str())) == NULL) {
|
| + throw std::runtime_error("Unknown host " + remoteAddr);
|
| + }
|
| + bcopy(hp->h_addr, &sockAddr.sin_addr, hp->h_length);
|
| + sockAddr.sin_port = htons(remotePort);
|
| + if ((sockDesc = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
|
| + throw std::runtime_error("Socket creation failed (socket())");
|
| + }
|
| + /* Try to connect */
|
| + if (connect(sockDesc, (struct sockaddr *) &sockAddr, sizeof(sockAddr)) < 0) {
|
| + throw std::runtime_error("Error connecting to remote host (connect())");
|
| + }
|
| +}
|
| +
|
| +/* Destructor */
|
| +TCPSocketClient::~TCPSocketClient()
|
| +{
|
| + if (sockDesc >= 0) {
|
| + ::close(sockDesc);
|
| + }
|
| +}
|
| +
|
| +/* Communication over socket */
|
| +/* Receive data */
|
| +ssize_t TCPSocketClient::recv(void *buffer, const size_t &bufferLen) throw (std::runtime_error)
|
| +{
|
| + if (sockDesc < 0) {
|
| + throw std::runtime_error("socket is not connected (recv())");
|
| + }
|
| + int rval = ::read(sockDesc, buffer, bufferLen);
|
| + if (rval <= 0) {
|
| + // EOF (connection closed by remote host) or error:
|
| + // reset state, so a new accept() call will succeed
|
| + sockDesc = -1;
|
| + ::memset(&sockAddr, 0, sizeof(sockAddr));
|
| + }
|
| + if (rval == -1) {
|
| + throw std::runtime_error("Error reading from socket (read())");
|
| + }
|
| + return rval;
|
| +}
|
| +
|
| +/* Send data */
|
| +ssize_t TCPSocketClient::send(const void *buffer, const int &bufferLen) throw (std::runtime_error)
|
| +{
|
| + if (sockDesc <= 0) {
|
| + throw std::runtime_error("socket is not connected (send())");
|
| + }
|
| +
|
| + int rval = ::write(sockDesc, buffer, bufferLen);
|
| +
|
| + if (rval <= 0) {
|
| + // EOF (connection closed by remote host) or error:
|
| + // reset state, so a new accept() call will succeed
|
| + sockDesc = -1;
|
| + ::memset(&sockAddr, 0, sizeof(sockAddr));
|
| + }
|
| +
|
| + if (rval == -1) {
|
| + throw std::runtime_error("Error reading from socket (read())");
|
| + }
|
| +
|
| + return rval;
|
| +}
|
| +
|
| +ssize_t TCPSocketClient::send(const string &message) throw (std::runtime_error)
|
| +{
|
| + return send(message.c_str(), message.length());
|
| +}
|
| +
|
| +
|
| +ssize_t TCPSocketClient::send(unsigned int val) throw (std::runtime_error)
|
| +{
|
| + return send(&val, 4);
|
| +}
|
| +
|
|
|