OLD | NEW |
(Empty) | |
| 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 s
hort &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 (s
td::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 |
OLD | NEW |