~misterc/libva/Trunk

« back to all changes in this revision

Viewing changes to test/v4l_h264/encode/TCPSocketClient.cpp

  • Committer: Sean V Kelley
  • Date: 2017-02-18 23:19:05 UTC
  • Revision ID: git-v1:7b8cc07dc7a9b954b9a8c9bde5091fb2e8d443dc
PROJECT HAS MOVED

See https://github.com/01org/libva

Signed-off-by: Sean V Kelley <seanvk@posteo.de>

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*
2
 
 * Copyright (c) 2012 Intel Corporation. All Rights Reserved.
3
 
 *
4
 
 * Permission is hereby granted, free of charge, to any person obtaining a
5
 
 * copy of this software and associated documentation files (the
6
 
 * "Software"), to deal in the Software without restriction, including
7
 
 * without limitation the rights to use, copy, modify, merge, publish,
8
 
 * distribute, sub license, and/or sell copies of the Software, and to
9
 
 * permit persons to whom the Software is furnished to do so, subject to
10
 
 * the following conditions:
11
 
 *
12
 
 * The above copyright notice and this permission notice (including the
13
 
 * next paragraph) shall be included in all copies or substantial portions
14
 
 * of the Software.
15
 
 *
16
 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 
17
 
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES 
18
 
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. 
19
 
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, 
20
 
 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR 
21
 
 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE 
22
 
 * USE OR OTHER DEALINGS IN THE SOFTWARE.
23
 
 */
24
 
 
25
 
/*
26
 
* C++ wrapper around an TCP socket
27
 
*/
28
 
 
29
 
#include "TCPSocketClient.h"
30
 
 
31
 
#include <sys/types.h>
32
 
#include <sys/socket.h>
33
 
#include <netinet/in.h>
34
 
#include <arpa/inet.h>
35
 
#include <unistd.h>
36
 
#include <netdb.h>
37
 
#include <fcntl.h>
38
 
#include <errno.h>
39
 
#include <cstring>
40
 
using std::string;
41
 
 
42
 
TCPSocketClient::TCPSocketClient(const std::string &remoteAddr, const unsigned short &remotePort) throw(std::runtime_error) :
43
 
sockDesc(-1)
44
 
{
45
 
    ::memset(&sockAddr, 0, sizeof(sockAddr));
46
 
    sockAddr.sin_family = AF_INET;
47
 
    struct hostent* hp;
48
 
    if ((hp = gethostbyname(remoteAddr.c_str())) == NULL) {
49
 
        throw std::runtime_error("Unknown host " + remoteAddr);
50
 
    }
51
 
    bcopy(hp->h_addr, &sockAddr.sin_addr, hp->h_length);
52
 
    sockAddr.sin_port = htons(remotePort);
53
 
    if ((sockDesc = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
54
 
        throw std::runtime_error("Socket creation failed (socket())");
55
 
    }
56
 
    /* Try to connect */
57
 
    if (connect(sockDesc, (struct sockaddr *) &sockAddr, sizeof(sockAddr)) < 0) {
58
 
        throw std::runtime_error("Error connecting to remote host (connect())");
59
 
    }
60
 
}
61
 
 
62
 
/* Destructor */
63
 
TCPSocketClient::~TCPSocketClient()
64
 
{
65
 
    if (sockDesc >= 0) {
66
 
        ::close(sockDesc);
67
 
    }
68
 
}
69
 
 
70
 
/* Communication over socket */
71
 
/* Receive data */
72
 
ssize_t TCPSocketClient::recv(void *buffer, const size_t &bufferLen) throw (std::runtime_error)
73
 
{
74
 
    if (sockDesc < 0) {
75
 
        throw std::runtime_error("socket is not connected (recv())");
76
 
    }
77
 
    int rval = ::read(sockDesc, buffer, bufferLen);
78
 
    if (rval <= 0) {
79
 
        // EOF (connection closed by remote host) or error:
80
 
        // reset state, so a new accept() call will succeed
81
 
        sockDesc = -1;
82
 
        ::memset(&sockAddr, 0, sizeof(sockAddr));
83
 
    }
84
 
    if (rval == -1) {
85
 
        throw std::runtime_error("Error reading from socket (read())");
86
 
    }
87
 
    return rval;
88
 
}
89
 
 
90
 
/* Send data */
91
 
ssize_t TCPSocketClient::send(const void *buffer, const int &bufferLen) throw (std::runtime_error)
92
 
{
93
 
    if (sockDesc <= 0) {
94
 
        throw std::runtime_error("socket is not connected (send())");
95
 
    }
96
 
 
97
 
    int rval = ::write(sockDesc, buffer, bufferLen);
98
 
 
99
 
    if (rval <= 0) {
100
 
        // EOF (connection closed by remote host) or error:
101
 
        // reset state, so a new accept() call will succeed
102
 
        sockDesc = -1;
103
 
        ::memset(&sockAddr, 0, sizeof(sockAddr));
104
 
    }
105
 
 
106
 
    if (rval == -1) {
107
 
        throw std::runtime_error("Error reading from socket (read())");
108
 
    }
109
 
 
110
 
    return rval;
111
 
}
112
 
 
113
 
ssize_t TCPSocketClient::send(const string &message) throw (std::runtime_error)
114
 
{
115
 
    return send(message.c_str(), message.length());
116
 
}
117
 
 
118
 
 
119
 
ssize_t TCPSocketClient::send(unsigned int val) throw (std::runtime_error)
120
 
{
121
 
    return send(&val, 4);
122
 
}
123