Commit | Line | Data |
---|---|---|
0e51267b LOK |
1 | #pragma once |
2 | /* | |
3 | * This file is part of the libCEC(R) library. | |
4 | * | |
b492c10e | 5 | * libCEC(R) is Copyright (C) 2011-2012 Pulse-Eight Limited. All rights reserved. |
0e51267b LOK |
6 | * libCEC(R) is an original work, containing original code. |
7 | * | |
8 | * libCEC(R) is a trademark of Pulse-Eight Limited. | |
9 | * | |
10 | * This program is dual-licensed; you can redistribute it and/or modify | |
11 | * it under the terms of the GNU General Public License as published by | |
12 | * the Free Software Foundation; either version 2 of the License, or | |
13 | * (at your option) any later version. | |
14 | * | |
15 | * This program is distributed in the hope that it will be useful, | |
16 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
17 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
18 | * GNU General Public License for more details. | |
19 | * | |
20 | * You should have received a copy of the GNU General Public License | |
21 | * along with this program; if not, write to the Free Software | |
22 | * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
23 | * | |
24 | * | |
25 | * Alternatively, you can license this library under a commercial license, | |
26 | * please contact Pulse-Eight Licensing for more information. | |
27 | * | |
28 | * For more information contact: | |
29 | * Pulse-Eight Licensing <license@pulse-eight.com> | |
30 | * http://www.pulse-eight.com/ | |
31 | * http://www.pulse-eight.net/ | |
32 | */ | |
33 | ||
34 | #include "socket.h" | |
35 | ||
36 | #if defined(__WINDOWS__) | |
37 | #include "../windows/os-tcp.h" | |
38 | #else | |
39 | #include "../posix/os-tcp.h" | |
40 | #endif | |
41 | ||
cb2397e9 LOK |
42 | using namespace std; |
43 | ||
0e51267b LOK |
44 | namespace PLATFORM |
45 | { | |
46 | class CTcpSocket : public CSocket | |
47 | { | |
48 | public: | |
cb2397e9 | 49 | CTcpSocket(void) {}; |
0e51267b LOK |
50 | virtual ~CTcpSocket(void) {} |
51 | ||
52 | virtual bool Open(const CStdString &strHostname, uint16_t iPort, uint64_t nTimeout) | |
53 | { | |
54 | bool bReturn(false); | |
cb2397e9 LOK |
55 | struct addrinfo *address(NULL), *addr(NULL); |
56 | CLockObject lock(m_mutex); | |
57 | if (!TcpResolveAddress(strHostname.c_str(), iPort, &m_iError, &address)) | |
0e51267b | 58 | { |
cb2397e9 | 59 | m_strError = strerror(m_iError); |
0e51267b LOK |
60 | return bReturn; |
61 | } | |
62 | ||
63 | for(addr = address; !bReturn && addr; addr = addr->ai_next) | |
64 | { | |
cb2397e9 | 65 | m_socket = TcpCreateSocket(addr, &m_iError, nTimeout); |
0e51267b LOK |
66 | if (m_socket != INVALID_SOCKET && m_socket != SOCKET_ERROR) |
67 | bReturn = true; | |
68 | else | |
cb2397e9 | 69 | m_strError = strerror(m_iError); |
0e51267b LOK |
70 | } |
71 | ||
72 | freeaddrinfo(address); | |
73 | return bReturn; | |
74 | } | |
75 | ||
cb2397e9 LOK |
76 | virtual void Shutdown(void) |
77 | { | |
78 | CLockObject lock(m_mutex); | |
79 | if (m_socket != INVALID_SOCKET && m_socket != SOCKET_ERROR) | |
80 | TcpShutdownSocket(m_socket); | |
81 | m_socket = INVALID_SOCKET; | |
82 | m_strError = ""; | |
83 | } | |
84 | ||
0e51267b LOK |
85 | protected: |
86 | virtual socket_t TcpCreateSocket(struct addrinfo* addr, int* iError, uint64_t iTimeout) | |
87 | { | |
88 | socket_t fdSock = socket(addr->ai_family, addr->ai_socktype, addr->ai_protocol); | |
cb2397e9 | 89 | if (fdSock == INVALID_SOCKET || fdSock == SOCKET_ERROR) |
0e51267b | 90 | { |
cb2397e9 | 91 | *iError = errno; |
0e51267b LOK |
92 | return (socket_t)SOCKET_ERROR; |
93 | } | |
94 | ||
cb2397e9 | 95 | if (!TcpConnectSocket(fdSock, addr, iError, iTimeout)) |
0e51267b | 96 | { |
cb2397e9 | 97 | SocketClose(fdSock); |
0e51267b LOK |
98 | return (socket_t)SOCKET_ERROR; |
99 | } | |
100 | ||
101 | TcpSetNoDelay(fdSock); | |
102 | ||
103 | return fdSock; | |
104 | } | |
105 | }; | |
106 | }; |