source: cheroxy/trunk/src/CRXSocket.cpp@ 56

Last change on this file since 56 was 38, checked in by cheese, 11 years ago

#1 more mutex interface and add debugging code to check connection port number
File size: 9.4 KB
RevLine 
[6]1/**
2 * CRXSocket.cpp
3 */
4#include "CRXSocket.h"
5
6#include <string.h>
7
8#ifdef _WIN32
9# pragma comment (lib, "ws2_32.lib")
10# define close(__socket) closesocket(__socket)
11#endif
12
13bool CRXSocket::mInitialized = false;
14
15CRXSocket::CRXSocket (int aSocket)
16 : mSocket (aSocket)
17{
18 /*----------------------------------------------------------------*/
[24]19 CRXSocket::Initialize ();
20
[14]21 memset ((void *)&mAddress, 0x0, sizeof (struct sockaddr_in));
[6]22 /*----------------------------------------------------------------*/
23}
24
25CRXSocket::~CRXSocket (void)
26{
27 /*----------------------------------------------------------------*/
28 Close ();
29 /*----------------------------------------------------------------*/
30}
31
32void
33CRXSocket::Attach (int aSocket)
34{
35 /*----------------------------------------------------------------*/
36 mSocket = aSocket;
37 /*----------------------------------------------------------------*/
38}
39
40int
41CRXSocket::Detach (void)
42{
43 int aSocket = mSocket;
44
45 /*----------------------------------------------------------------*/
46 mSocket = 0;
47
48 /*----------------------------------------------------------------*/
49 return aSocket;
50}
51
52void
53CRXSocket::Close (void)
54{
55 /*----------------------------------------------------------------*/
56 if (mSocket <= 0)
57 return ;
58
59 close (mSocket);
60 mSocket = 0;
61 /*----------------------------------------------------------------*/
62}
63
64CRXSocket::operator int (void) const
65{
66 /*----------------------------------------------------------------*/
67 /*----------------------------------------------------------------*/
68 return mSocket;
69}
70
71CRXSocket &
72CRXSocket::operator = (int aSocket)
73{
74 /*----------------------------------------------------------------*/
75 Attach (aSocket);
76 /*----------------------------------------------------------------*/
77 return *this;
78}
79
80int
[24]81CRXSocket::SetTimeout (const int aTimeout)
82{
83 int aResult = 0;
84
85#ifndef _WIN32
86 struct timeval aTimeVal;
87 aTimeVal.tv_sec = aTimeout;
88 aTimeVal.tv_usec= 0;
89#else
90 int aTimeMilliSec = aTimeout * 1000;
91#endif
92
93 /*----------------------------------------------------------------*/
94 if (!IsReady ())
95 return ERROR_TCPSOCKET_NOT_READY;
96
97 if (aTimeout == TCPSOCKET_NO_TIMEOUT)
98 return 0;
99
100 aResult = setsockopt (mSocket,
101 SOL_SOCKET,
102 SO_RCVTIMEO,
103#ifdef _WIN32
104 (char *) &aTimeMilliSec,
105 sizeof (aTimeMilliSec));
106#else
107 &aTimeVal,
108 (socklen_t) sizeof (aTimeVal));
109#endif
110 if (aResult < 0)
111 {
112 aResult = ERROR_TCPSOCKET_FAILED_TO_SETSOCKOPT;
[35]113 CRX_ERROR_SET (aResult, "Failed to set socket option.");
[24]114 }
115 /*----------------------------------------------------------------*/
116
117 return aResult;
118}
119
120int
[6]121CRXSocket::Initialize (void)
122{
123 int aResult = 0;
124
125 /*----------------------------------------------------------------*/
126 if (CRXSocket::IsInitialized ())
127 return aResult;
128
129#ifdef WIN32
130 WSADATA aWinSockData;
131
132 aResult = WSAStartup (MAKEWORD (2, 0), &aWinSockData);
133 if (aResult != 0)
134 {
[18]135 aResult = ERROR_TCPSOCKET_FAILED_TO_INITIALIZE;
136 return aResult;
[6]137 }
138#endif
139
140 mInitialized = true;
141
142 /*----------------------------------------------------------------*/
143 return aResult;
144}
145
146void
147CRXSocket::Finalize (void)
148{
149 /*----------------------------------------------------------------*/
150#ifdef WIN32
151 WSACleanup ();
152#endif
153 /*----------------------------------------------------------------*/
154}
155
156bool
157CRXSocket::IsInitialized (void)
158{
159 /*----------------------------------------------------------------*/
160 /*----------------------------------------------------------------*/
161 return mInitialized;
162}
163
164bool
165CRXSocket::IsReady (void) const
166{
167 /*----------------------------------------------------------------*/
168 /*----------------------------------------------------------------*/
169 return IsInitialized () && IsCreated ();
170}
171
172bool
173CRXSocket::IsCreated (void) const
174{
175 /*----------------------------------------------------------------*/
176 /*----------------------------------------------------------------*/
177 return mSocket > 0 ? true : false;
178}
179
180int
181CRXSocket::CreateSocket (void)
182{
183 int aResult = -1;
184
185 /*----------------------------------------------------------------*/
186 if (!IsInitialized () || IsCreated ())
[18]187 {
188 aResult = ERROR_TCPSOCKET_ALREADY_IN_USE;
[35]189 CRX_ERROR_SET (aResult, "Already in use.");
[18]190 return aResult;
191 }
[6]192
193 aResult = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP);
194 if (aResult < 0)
195 {
[18]196 aResult = ERROR_TCPSOCKET_FAILED_TO_CREATE_SOCKET;
[35]197 CRX_ERROR_SET (aResult, "Failed to create socket(%d).", GetSystemErrorCode ());
[18]198 return aResult;
[6]199 }
200
201 mSocket = aResult;
202
203 /*----------------------------------------------------------------*/
204 return aResult;
205}
206
207int
[14]208CRXSocket::Connect (const std::string aUrl,
209 const unsigned short aPort,
210 const int aTimeout)
[6]211{
212 int aResult = -1;
213 struct hostent * aHostEnt;
214
215 /*----------------------------------------------------------------*/
216 aResult = CreateSocket ();
217 if (aResult < 0)
218 {
[24]219 aResult = ERROR_TCPSOCKET_FAILED_TO_CREATE_SOCKET;
[35]220 CRX_ERROR_SET (aResult, "Failed to create socket(%d).", GetSystemErrorCode ());
[6]221 return aResult;
222 }
223
224 mAddress.sin_family = AF_INET;
225 mAddress.sin_port = htons (aPort);
226 mAddress.sin_addr.s_addr = inet_addr (aUrl.c_str ());
227
228 if (mAddress.sin_addr.s_addr == (unsigned int)-1)
229 {
230 aHostEnt = gethostbyname (aUrl.c_str ());
231 if (aHostEnt == NULL)
232 {
[18]233 aResult = ERROR_TCPSOCKET_FAILED_TO_GET_HOSTNAME;
[35]234 CRX_ERROR_SET (aResult, "Failed to get hostname.");
[18]235 return aResult;
[6]236 }
237 mAddress.sin_family = aHostEnt->h_addrtype;
238 memcpy (&(mAddress.sin_addr.s_addr), aHostEnt->h_addr, aHostEnt->h_length);
239 }
240
[24]241 aResult = SetTimeout (aTimeout);
242 if (aResult < 0)
[14]243 {
[24]244 aResult = ERROR_TCPSOCKET_FAILED_TO_SETSOCKOPT;
[35]245 CRX_ERROR_SET (aResult, "Failed to set timeout (%d).", GetSystemErrorCode ());
[24]246 return aResult;
[14]247 }
248
249 aResult = connect (mSocket, (struct sockaddr*) &mAddress, sizeof (mAddress));
[6]250 if (aResult < 0)
251 {
[18]252 aResult = ERROR_TCPSOCKET_FAILED_TO_CONNECT;
[35]253 CRX_ERROR_SET (aResult, "Failed to connect (%d).", GetSystemErrorCode ());
[6]254 }
255
256 /*----------------------------------------------------------------*/
257 return aResult;
258}
259
260int
[14]261CRXSocket::CreateServer (const unsigned short aPort,
262 const int aBacklog,
263 struct sockaddr_in * aAddress)
[6]264{
265 int aResult = -1;
266
267 /*----------------------------------------------------------------*/
268 aResult = CreateSocket ();
269 if (aResult < 0)
270 {
[24]271 aResult = ERROR_TCPSOCKET_FAILED_TO_CREATE_SOCKET;
[35]272 CRX_ERROR_SET (aResult, "Failed to create socket(%d).", GetSystemErrorCode ());
[6]273 return aResult;
274 }
275
276 mAddress.sin_family = AF_INET;
277 mAddress.sin_addr.s_addr= htonl (INADDR_ANY);
278 mAddress.sin_port = htons (aPort);
279
[14]280 aResult = bind (mSocket, (struct sockaddr *)&mAddress, sizeof (struct sockaddr));
[6]281 if (aResult < 0)
282 {
[18]283 aResult = ERROR_TCPSOCKET_FAILED_TO_BIND;
[35]284 CRX_ERROR_SET (aResult, "Failed to bind(%d).", GetSystemErrorCode ());
[18]285 return aResult;
[6]286 }
287
288 aResult = listen (mSocket, aBacklog);
289 if (aResult < 0)
290 {
[18]291 aResult = ERROR_TCPSOCKET_FAILED_TO_LISTEN;
[35]292 CRX_ERROR_SET (aResult, "Failed to listen(%d).", GetSystemErrorCode ());
[18]293 return aResult;
[6]294 }
295
296 if (aAddress != NULL)
297 {
[14]298 memset ((void *)aAddress, 0x0, sizeof (struct sockaddr_in));
299 memcpy ((void *)aAddress, (void *)&mAddress, sizeof (struct sockaddr_in));
[6]300 }
301
302 /*----------------------------------------------------------------*/
303 return aResult;
304}
305
306int
[14]307CRXSocket::Accept (struct sockaddr_in * aRemoteAddress,
308 int * aAddressLength)
[6]309{
[14]310 int aResult = -1;
[6]311
[14]312 struct sockaddr_in aAddress;
313 socklen_t aLength = sizeof (aAddress);
[6]314
315 /*----------------------------------------------------------------*/
[24]316 if (!IsReady ())
317 return ERROR_TCPSOCKET_NOT_READY;
[6]318
[14]319 aResult = accept (mSocket, (struct sockaddr *) &aAddress, &aLength);
[6]320 if (aResult < 0)
321 {
[18]322 aResult = ERROR_TCPSOCKET_FAILED_TO_ACCEPT;
[35]323 CRX_ERROR_SET (aResult, "Failed to accept(%d).", GetSystemErrorCode ());
[6]324 return aResult;
325 }
326
327 if (aRemoteAddress != NULL)
328 {
[14]329 memset ((void *)aRemoteAddress, 0x0, sizeof (struct sockaddr_in));
330 memcpy ((void *)aRemoteAddress, (void *)&aAddress, sizeof (struct sockaddr_in));
[6]331 }
332
333 if (aAddressLength != NULL)
334 *aAddressLength = aLength;
335
336 /*----------------------------------------------------------------*/
337 return aResult;
338}
339
340int
[24]341CRXSocket::Send (const char * aBuffer,
342 int aSize)
[6]343{
344 int aResult = -1;
345
346 /*----------------------------------------------------------------*/
[24]347 if (!IsReady ())
348 return ERROR_TCPSOCKET_NOT_READY;
[6]349
350 aResult = send (mSocket, aBuffer, aSize, 0);
351 if (aResult != aSize)
352 {
[18]353 aResult = ERROR_TCPSOCKET_FAILED_TO_SEND;
[35]354 CRX_ERROR_SET (aResult, "Failed to send(%d).", GetSystemErrorCode ());
[18]355 return aResult;
[6]356 }
357
358 /*----------------------------------------------------------------*/
359 return aResult;
360}
361
362int
363CRXSocket::Receive (char * aBuffer,
364 int aSize)
365{
366 int aResult = -1;
367
368 /*----------------------------------------------------------------*/
[24]369 if (!IsReady ())
370 return ERROR_TCPSOCKET_NOT_READY;
[6]371
372 aResult = recv (mSocket, aBuffer, aSize, 0);
373 if (aResult < 0)
374 {
[18]375 aResult = ERROR_TCPSOCKET_FAILED_TO_RECEIVE;
[35]376 CRX_ERROR_SET (aResult, "Failed to receive(%d).", GetSystemErrorCode ());
[18]377 return aResult;
[6]378 }
379
380 /*----------------------------------------------------------------*/
381 return aResult;
382}
Note: See TracBrowser for help on using the repository browser.