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