Home | History | Annotate | Download | only in deutil
      1 /*-------------------------------------------------------------------------
      2  * drawElements Utility Library
      3  * ----------------------------
      4  *
      5  * Copyright 2014 The Android Open Source Project
      6  *
      7  * Licensed under the Apache License, Version 2.0 (the "License");
      8  * you may not use this file except in compliance with the License.
      9  * You may obtain a copy of the License at
     10  *
     11  *      http://www.apache.org/licenses/LICENSE-2.0
     12  *
     13  * Unless required by applicable law or agreed to in writing, software
     14  * distributed under the License is distributed on an "AS IS" BASIS,
     15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     16  * See the License for the specific language governing permissions and
     17  * limitations under the License.
     18  *
     19  *//*!
     20  * \file
     21  * \brief Socket abstraction.
     22  *//*--------------------------------------------------------------------*/
     23 
     24 #include "deSocket.h"
     25 #include "deMemory.h"
     26 #include "deMutex.h"
     27 #include "deInt32.h"
     28 
     29 #if (DE_OS == DE_OS_WIN32)
     30 #	define DE_USE_WINSOCK
     31 #elif (DE_OS == DE_OS_UNIX) || (DE_OS == DE_OS_OSX) || (DE_OS == DE_OS_IOS) || (DE_OS == DE_OS_ANDROID) || (DE_OS == DE_OS_SYMBIAN)
     32 #	define DE_USE_BERKELEY_SOCKETS
     33 #else
     34 #	error Implement deSocket for your OS.
     35 #endif
     36 
     37 /* Common utilities. */
     38 
     39 const char* deGetSocketResultName (deSocketResult result)
     40 {
     41 	switch (result)
     42 	{
     43 		case DE_SOCKETRESULT_SUCCESS:				return "DE_SOCKETRESULT_SUCCESS";
     44 		case DE_SOCKETRESULT_WOULD_BLOCK:			return "DE_SOCKETRESULT_WOULD_BLOCK";
     45 		case DE_SOCKETRESULT_CONNECTION_CLOSED:		return "DE_SOCKETRESULT_CONNECTION_CLOSED";
     46 		case DE_SOCKETRESULT_CONNECTION_TERMINATED:	return "DE_SOCKETRESULT_CONNECTION_TERMINATED";
     47 		case DE_SOCKETRESULT_ERROR:					return "DE_SOCKETRESULT_ERROR";
     48 		default:									return DE_NULL;
     49 	}
     50 }
     51 
     52 const char* deGetSocketFamilyName (deSocketFamily family)
     53 {
     54 	switch (family)
     55 	{
     56 		case DE_SOCKETFAMILY_INET4:		return "DE_SOCKETFAMILY_INET4";
     57 		case DE_SOCKETFAMILY_INET6:		return "DE_SOCKETFAMILY_INET6";
     58 		default:						return DE_NULL;
     59 	}
     60 }
     61 
     62 #if defined(DE_USE_WINSOCK) || defined(DE_USE_BERKELEY_SOCKETS)
     63 
     64 /* Common deSocketAddress implementation. */
     65 
     66 struct deSocketAddress_s
     67 {
     68 	char*				host;
     69 	int					port;
     70 	deSocketFamily		family;
     71 	deSocketType		type;
     72 	deSocketProtocol	protocol;
     73 };
     74 
     75 deSocketAddress* deSocketAddress_create (void)
     76 {
     77 	deSocketAddress* addr = (deSocketAddress*)deCalloc(sizeof(deSocketAddress));
     78 	if (!addr)
     79 		return addr;
     80 
     81 	/* Sane defaults. */
     82 	addr->family	= DE_SOCKETFAMILY_INET4;
     83 	addr->type		= DE_SOCKETTYPE_STREAM;
     84 	addr->protocol	= DE_SOCKETPROTOCOL_TCP;
     85 
     86 	return addr;
     87 }
     88 
     89 deBool deSocketAddress_setFamily (deSocketAddress* address, deSocketFamily family)
     90 {
     91 	address->family = family;
     92 	return DE_TRUE;
     93 }
     94 
     95 deSocketFamily deSocketAddress_getFamily (const deSocketAddress* address)
     96 {
     97 	return address->family;
     98 }
     99 
    100 void deSocketAddress_destroy (deSocketAddress* address)
    101 {
    102 	deFree(address->host);
    103 	deFree(address);
    104 }
    105 
    106 deBool deSocketAddress_setPort (deSocketAddress* address, int port)
    107 {
    108 	address->port = port;
    109 	return DE_TRUE;
    110 }
    111 
    112 int deSocketAddress_getPort (const deSocketAddress* address)
    113 {
    114 	return address->port;
    115 }
    116 
    117 deBool deSocketAddress_setHost (deSocketAddress* address, const char* host)
    118 {
    119 	if (address->host)
    120 	{
    121 		deFree(address->host);
    122 		address->host = DE_NULL;
    123 	}
    124 
    125 	address->host = deStrdup(host);
    126 	return address->host != DE_NULL;
    127 }
    128 
    129 const char* deSocketAddress_getHost (const deSocketAddress* address)
    130 {
    131 	return address->host;
    132 }
    133 
    134 
    135 deBool deSocketAddress_setType (deSocketAddress* address, deSocketType type)
    136 {
    137 	address->type = type;
    138 	return DE_TRUE;
    139 }
    140 
    141 deSocketType deSocketAddress_getType (const deSocketAddress* address)
    142 {
    143 	return address->type;
    144 }
    145 
    146 deBool deSocketAddress_setProtocol (deSocketAddress* address, deSocketProtocol protocol)
    147 {
    148 	address->protocol = protocol;
    149 	return DE_TRUE;
    150 }
    151 
    152 deSocketProtocol deSocketAddress_getProtocol (const deSocketAddress* address)
    153 {
    154 	return address->protocol;
    155 }
    156 
    157 #endif
    158 
    159 #if defined(DE_USE_WINSOCK)
    160 
    161 	/* WinSock spesific. */
    162 #	include <WinSock2.h>
    163 #	include <WS2tcpip.h>
    164 #	include <WinDef.h>
    165 
    166 static deBool initWinsock (void)
    167 {
    168 	WSADATA wsaData;
    169 	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
    170 		return DE_FALSE;
    171 
    172 	return DE_TRUE;
    173 }
    174 
    175 #elif defined(DE_USE_BERKELEY_SOCKETS)
    176 
    177 	/* Berkeley Socket includes. */
    178 #	include <sys/socket.h>
    179 #	include <netinet/in.h>
    180 #	include <netinet/tcp.h>
    181 #	include <arpa/inet.h>
    182 #	include <netdb.h>
    183 #	include <unistd.h>
    184 #	include <fcntl.h>
    185 #	include <errno.h>
    186 
    187 #endif
    188 
    189 /* Socket type. */
    190 #if defined(DE_USE_WINSOCK)
    191 	/* \note SOCKET is unsigned type! */
    192 	typedef SOCKET					deSocketHandle;
    193 	typedef int						NativeSocklen;
    194 	typedef int						NativeSize;
    195 #	define DE_INVALID_SOCKET_HANDLE	INVALID_SOCKET
    196 #else
    197 	typedef int						deSocketHandle;
    198 	typedef socklen_t				NativeSocklen;
    199 	typedef size_t					NativeSize;
    200 #	define DE_INVALID_SOCKET_HANDLE	(-1)
    201 #endif
    202 
    203 DE_INLINE deBool deSocketHandleIsValid (deSocketHandle handle)
    204 {
    205 	return handle != DE_INVALID_SOCKET_HANDLE;
    206 }
    207 
    208 #if defined(DE_USE_WINSOCK) || defined(DE_USE_BERKELEY_SOCKETS)
    209 
    210 /* Shared berkeley and winsock implementation. */
    211 
    212 struct deSocket_s
    213 {
    214 	deSocketHandle			handle;
    215 
    216 	deMutex					stateLock;
    217 	volatile deSocketState	state;
    218 	volatile deUint32		openChannels;
    219 };
    220 
    221 /* Common socket functions. */
    222 
    223 static deUint16 deHostToNetworkOrder16 (deUint16 v)
    224 {
    225 #if (DE_ENDIANNESS == DE_LITTLE_ENDIAN)
    226 	return deReverseBytes16(v);
    227 #else
    228 	return v;
    229 #endif
    230 }
    231 
    232 static deUint16 deNetworkToHostOrder16 (deUint16 v)
    233 {
    234 #if (DE_ENDIANNESS == DE_LITTLE_ENDIAN)
    235 	return deReverseBytes16(v);
    236 #else
    237 	return v;
    238 #endif
    239 }
    240 
    241 DE_STATIC_ASSERT(sizeof(((struct sockaddr_in*)DE_NULL)->sin_port) == sizeof(deUint16));
    242 DE_STATIC_ASSERT(sizeof(((struct sockaddr_in6*)DE_NULL)->sin6_port) == sizeof(deUint16));
    243 
    244 static int deSocketFamilyToBsdFamily (deSocketFamily family)
    245 {
    246 	switch (family)
    247 	{
    248 		case DE_SOCKETFAMILY_INET4:	return AF_INET;
    249 		case DE_SOCKETFAMILY_INET6:	return AF_INET6;
    250 		default:
    251 			DE_ASSERT(DE_FALSE);
    252 			return 0;
    253 	}
    254 }
    255 
    256 static int deSocketTypeToBsdType (deSocketType type)
    257 {
    258 	switch (type)
    259 	{
    260 		case DE_SOCKETTYPE_STREAM:		return SOCK_STREAM;
    261 		case DE_SOCKETTYPE_DATAGRAM:	return SOCK_DGRAM;
    262 		default:
    263 			DE_ASSERT(DE_FALSE);
    264 			return 0;
    265 	}
    266 }
    267 
    268 static int deSocketProtocolToBsdProtocol (deSocketProtocol protocol)
    269 {
    270 	switch (protocol)
    271 	{
    272 		case DE_SOCKETPROTOCOL_TCP:	return IPPROTO_TCP;
    273 		case DE_SOCKETPROTOCOL_UDP:	return IPPROTO_UDP;
    274 		default:
    275 			DE_ASSERT(DE_FALSE);
    276 			return 0;
    277 	}
    278 }
    279 
    280 static deBool deSocketAddressToBsdAddress (const deSocketAddress* address, size_t bsdAddrBufSize, struct sockaddr* bsdAddr, NativeSocklen* bsdAddrLen)
    281 {
    282 	deMemset(bsdAddr, 0, bsdAddrBufSize);
    283 
    284 	/* Resolve host. */
    285 	if (address->host != DE_NULL)
    286 	{
    287 		struct addrinfo*	result	= DE_NULL;
    288 		struct addrinfo		hints;
    289 
    290 		deMemset(&hints, 0, sizeof(hints));
    291 		hints.ai_family		= deSocketFamilyToBsdFamily(address->family);
    292 		hints.ai_socktype	= deSocketTypeToBsdType(address->type);
    293 		hints.ai_protocol	= deSocketProtocolToBsdProtocol(address->protocol);
    294 
    295 		if (getaddrinfo(address->host, DE_NULL, &hints, &result) != 0 || !result)
    296 		{
    297 			if (result)
    298 				freeaddrinfo(result);
    299 			return DE_FALSE;
    300 		}
    301 
    302 		/* \note Always uses first address. */
    303 
    304 		if (bsdAddrBufSize < (size_t)result->ai_addrlen)
    305 		{
    306 			DE_FATAL("Too small bsdAddr buffer");
    307 			freeaddrinfo(result);
    308 			return DE_FALSE;
    309 		}
    310 
    311 		*bsdAddrLen	= (NativeSocklen)result->ai_addrlen;
    312 
    313 		deMemcpy(bsdAddr, result->ai_addr, (size_t)result->ai_addrlen);
    314 		freeaddrinfo(result);
    315 
    316 		/* Add port. */
    317 		if (bsdAddr->sa_family == AF_INET)
    318 		{
    319 			if (*bsdAddrLen < (NativeSocklen)sizeof(struct sockaddr_in))
    320 				return DE_FALSE;
    321 			((struct sockaddr_in*)bsdAddr)->sin_port = deHostToNetworkOrder16((deUint16)address->port);
    322 		}
    323 		else if (bsdAddr->sa_family == AF_INET6)
    324 		{
    325 			if (*bsdAddrLen < (NativeSocklen)sizeof(struct sockaddr_in6))
    326 				return DE_FALSE;
    327 			((struct sockaddr_in6*)bsdAddr)->sin6_port = deHostToNetworkOrder16((deUint16)address->port);
    328 		}
    329 		else
    330 			return DE_FALSE;
    331 
    332 		return DE_TRUE;
    333 	}
    334 	else if (address->family == DE_SOCKETFAMILY_INET4)
    335 	{
    336 		struct sockaddr_in* addr4 = (struct sockaddr_in*)bsdAddr;
    337 
    338 		if (bsdAddrBufSize < sizeof(struct sockaddr_in))
    339 		{
    340 			DE_FATAL("Too small bsdAddr buffer");
    341 			return DE_FALSE;
    342 		}
    343 
    344 		addr4->sin_port			= deHostToNetworkOrder16((deUint16)address->port);
    345 		addr4->sin_family		= AF_INET;
    346 		addr4->sin_addr.s_addr	= INADDR_ANY;
    347 
    348 		*bsdAddrLen	= (NativeSocklen)sizeof(struct sockaddr_in);
    349 
    350 		return DE_TRUE;
    351 	}
    352 	else if (address->family == DE_SOCKETFAMILY_INET6)
    353 	{
    354 		struct sockaddr_in6* addr6 = (struct sockaddr_in6*)bsdAddr;
    355 
    356 		if (bsdAddrBufSize < sizeof(struct sockaddr_in6))
    357 		{
    358 			DE_FATAL("Too small bsdAddr buffer");
    359 			return DE_FALSE;
    360 		}
    361 
    362 		addr6->sin6_port	= deHostToNetworkOrder16((deUint16)address->port);
    363 		addr6->sin6_family	= AF_INET6;
    364 
    365 		*bsdAddrLen	= (NativeSocklen)sizeof(struct sockaddr_in6);
    366 
    367 		return DE_TRUE;
    368 	}
    369 	else
    370 		return DE_FALSE;
    371 }
    372 
    373 void deBsdAddressToSocketAddress (deSocketAddress* address, const struct sockaddr* bsdAddr, int addrLen)
    374 {
    375 	/* Decode client address info. */
    376 	if (bsdAddr->sa_family == AF_INET)
    377 	{
    378 		const struct sockaddr_in* addr4 = (const struct sockaddr_in*)bsdAddr;
    379 		DE_ASSERT(addrLen >= (int)sizeof(struct sockaddr_in));
    380 		DE_UNREF(addrLen);
    381 
    382 		deSocketAddress_setFamily(address, DE_SOCKETFAMILY_INET4);
    383 		deSocketAddress_setPort(address, (int)deNetworkToHostOrder16((deUint16)addr4->sin_port));
    384 
    385 		{
    386 			char buf[16]; /* Max valid address takes 3*4 + 3 = 15 chars */
    387 			inet_ntop(AF_INET, (void*)&addr4->sin_addr, buf, sizeof(buf));
    388 			deSocketAddress_setHost(address, buf);
    389 		}
    390 	}
    391 	else if (bsdAddr->sa_family == AF_INET6)
    392 	{
    393 		const struct sockaddr_in6* addr6 = (const struct sockaddr_in6*)bsdAddr;
    394 		DE_ASSERT(addrLen >= (int)sizeof(struct sockaddr_in6));
    395 		DE_UNREF(addrLen);
    396 
    397 		deSocketAddress_setFamily(address, DE_SOCKETFAMILY_INET6);
    398 		deSocketAddress_setPort(address, (int)deNetworkToHostOrder16((deUint16)addr6->sin6_port));
    399 
    400 		{
    401 			char buf[40]; /* Max valid address takes 8*4 + 7 = 39 chars */
    402 			inet_ntop(AF_INET6, (void*)&addr6->sin6_addr, buf, sizeof(buf));
    403 			deSocketAddress_setHost(address, buf);
    404 		}
    405 	}
    406 	else
    407 		DE_ASSERT(DE_FALSE);
    408 }
    409 
    410 deSocket* deSocket_create (void)
    411 {
    412 	deSocket* sock = (deSocket*)deCalloc(sizeof(deSocket));
    413 	if (!sock)
    414 		return sock;
    415 
    416 #if defined(DE_USE_WINSOCK)
    417 	/* Make sure WSA is up. */
    418 	if (!initWinsock())
    419 		return DE_NULL;
    420 #endif
    421 
    422 	sock->stateLock	= deMutex_create(0);
    423 	sock->handle	= DE_INVALID_SOCKET_HANDLE;
    424 	sock->state		= DE_SOCKETSTATE_CLOSED;
    425 
    426 	return sock;
    427 }
    428 
    429 void deSocket_destroy (deSocket* sock)
    430 {
    431 	if (sock->state != DE_SOCKETSTATE_CLOSED)
    432 		deSocket_close(sock);
    433 
    434 	deMutex_destroy(sock->stateLock);
    435 	deFree(sock);
    436 }
    437 
    438 deSocketState deSocket_getState (const deSocket* sock)
    439 {
    440 	return sock->state;
    441 }
    442 
    443 deUint32 deSocket_getOpenChannels (const deSocket* sock)
    444 {
    445 	return sock->openChannels;
    446 }
    447 
    448 deBool deSocket_setFlags (deSocket* sock, deUint32 flags)
    449 {
    450 	deSocketHandle fd = sock->handle;
    451 
    452 	if (sock->state == DE_SOCKETSTATE_CLOSED)
    453 		return DE_FALSE;
    454 
    455 	/* Keepalive. */
    456 	{
    457 		int mode = (flags & DE_SOCKET_KEEPALIVE) ? 1 : 0;
    458 		if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, (const char*)&mode, sizeof(mode)) != 0)
    459 			return DE_FALSE;
    460 	}
    461 
    462 	/* Nodelay. */
    463 	{
    464 		int mode = (flags & DE_SOCKET_NODELAY) ? 1 : 0;
    465 		if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (const char*)&mode, sizeof(mode)) != 0)
    466 			return DE_FALSE;
    467 	}
    468 
    469 	/* Non-blocking. */
    470 	{
    471 #if defined(DE_USE_WINSOCK)
    472 		u_long mode = (flags & DE_SOCKET_NONBLOCKING) ? 1 : 0;
    473 		if (ioctlsocket(fd, FIONBIO, &mode) != 0)
    474 			return DE_FALSE;
    475 #else
    476 		int oldFlags	= fcntl(fd, F_GETFL, 0);
    477 		int newFlags	= (flags & DE_SOCKET_NONBLOCKING) ? (oldFlags | O_NONBLOCK) : (oldFlags & ~O_NONBLOCK);
    478 		if (fcntl(fd, F_SETFL, newFlags) != 0)
    479 			return DE_FALSE;
    480 #endif
    481 	}
    482 
    483 	/* Close on exec. */
    484 	{
    485 #if defined(DE_USE_BERKELEY_SOCKETS)
    486 		int oldFlags = fcntl(fd, F_GETFD, 0);
    487 		int newFlags = (flags & DE_SOCKET_CLOSE_ON_EXEC) ? (oldFlags | FD_CLOEXEC) : (oldFlags & ~FD_CLOEXEC);
    488 		if (fcntl(fd, F_SETFD, newFlags) != 0)
    489 			return DE_FALSE;
    490 #endif
    491 	}
    492 
    493 	return DE_TRUE;
    494 }
    495 
    496 deBool deSocket_listen (deSocket* sock, const deSocketAddress* address)
    497 {
    498 	const int			backlogSize	= 4;
    499 	deUint8				bsdAddrBuf[sizeof(struct sockaddr_in6)];
    500 	struct sockaddr*	bsdAddr		= (struct sockaddr*)&bsdAddrBuf[0];
    501 	NativeSocklen		bsdAddrLen;
    502 
    503 	if (sock->state != DE_SOCKETSTATE_CLOSED)
    504 		return DE_FALSE;
    505 
    506 	/* Resolve address. */
    507 	if (!deSocketAddressToBsdAddress(address, sizeof(bsdAddrBuf), bsdAddr, &bsdAddrLen))
    508 		return DE_FALSE;
    509 
    510 	/* Create socket. */
    511 	sock->handle = socket(bsdAddr->sa_family, deSocketTypeToBsdType(address->type), deSocketProtocolToBsdProtocol(address->protocol));
    512 	if (!deSocketHandleIsValid(sock->handle))
    513 		return DE_FALSE;
    514 
    515 	sock->state = DE_SOCKETSTATE_DISCONNECTED;
    516 
    517 	/* Allow re-using address. */
    518 	{
    519 		int reuseVal = 1;
    520 		setsockopt(sock->handle, SOL_SOCKET, SO_REUSEADDR, (const char*)&reuseVal, (int)sizeof(reuseVal));
    521 	}
    522 
    523 	/* Bind to address. */
    524 	if (bind(sock->handle, bsdAddr, (NativeSocklen)bsdAddrLen) != 0)
    525 	{
    526 		deSocket_close(sock);
    527 		return DE_FALSE;
    528 	}
    529 
    530 	/* Start listening. */
    531 	if (listen(sock->handle, backlogSize) != 0)
    532 	{
    533 		deSocket_close(sock);
    534 		return DE_FALSE;
    535 	}
    536 
    537 	sock->state = DE_SOCKETSTATE_LISTENING;
    538 
    539 	return DE_TRUE;
    540 }
    541 
    542 deSocket* deSocket_accept (deSocket* sock, deSocketAddress* clientAddress)
    543 {
    544 	deSocketHandle		newFd		= DE_INVALID_SOCKET_HANDLE;
    545 	deSocket*			newSock		= DE_NULL;
    546 	deUint8				bsdAddrBuf[sizeof(struct sockaddr_in6)];
    547 	struct sockaddr*	bsdAddr		= (struct sockaddr*)&bsdAddrBuf[0];
    548 	NativeSocklen		bsdAddrLen	= (NativeSocklen)sizeof(bsdAddrBuf);
    549 
    550 	deMemset(bsdAddr, 0, (size_t)bsdAddrLen);
    551 
    552 	newFd = accept(sock->handle, bsdAddr, &bsdAddrLen);
    553 	if (!deSocketHandleIsValid(newFd))
    554 		return DE_NULL;
    555 
    556 	newSock = (deSocket*)deCalloc(sizeof(deSocket));
    557 	if (!newSock)
    558 	{
    559 #if defined(DE_USE_WINSOCK)
    560 		closesocket(newFd);
    561 #else
    562 		close(newFd);
    563 #endif
    564 		return DE_NULL;
    565 	}
    566 
    567 	newSock->stateLock		= deMutex_create(0);
    568 	newSock->handle			= newFd;
    569 	newSock->state			= DE_SOCKETSTATE_CONNECTED;
    570 	newSock->openChannels	= DE_SOCKETCHANNEL_BOTH;
    571 
    572 	if (clientAddress)
    573 		deBsdAddressToSocketAddress(clientAddress, bsdAddr, (int)bsdAddrLen);
    574 
    575 	return newSock;
    576 }
    577 
    578 deBool deSocket_connect (deSocket* sock, const deSocketAddress* address)
    579 {
    580 	deUint8				bsdAddrBuf[sizeof(struct sockaddr_in6)];
    581 	struct sockaddr*	bsdAddr		= (struct sockaddr*)&bsdAddrBuf[0];
    582 	NativeSocklen		bsdAddrLen;
    583 
    584 	/* Resolve address. */
    585 	if (!deSocketAddressToBsdAddress(address, sizeof(bsdAddrBuf), bsdAddr, &bsdAddrLen))
    586 		return DE_FALSE;
    587 
    588 	/* Create socket. */
    589 	sock->handle = socket(bsdAddr->sa_family, deSocketTypeToBsdType(address->type), deSocketProtocolToBsdProtocol(address->protocol));
    590 	if (!deSocketHandleIsValid(sock->handle))
    591 		return DE_FALSE;
    592 
    593 	/* Connect. */
    594 	if (connect(sock->handle, bsdAddr, bsdAddrLen) != 0)
    595 	{
    596 #if defined(DE_USE_WINSOCK)
    597 		closesocket(sock->handle);
    598 #else
    599 		close(sock->handle);
    600 #endif
    601 		sock->handle = DE_INVALID_SOCKET_HANDLE;
    602 		return DE_FALSE;
    603 	}
    604 
    605 	sock->state			= DE_SOCKETSTATE_CONNECTED;
    606 	sock->openChannels	= DE_SOCKETCHANNEL_BOTH;
    607 
    608 	return DE_TRUE;
    609 }
    610 
    611 deBool deSocket_shutdown (deSocket* sock, deUint32 channels)
    612 {
    613 	deUint32 closedChannels = 0;
    614 
    615 	deMutex_lock(sock->stateLock);
    616 
    617 	if (sock->state == DE_SOCKETSTATE_DISCONNECTED ||
    618 		sock->state == DE_SOCKETSTATE_CLOSED)
    619 	{
    620 		deMutex_unlock(sock->stateLock);
    621 		return DE_FALSE;
    622 	}
    623 
    624 	DE_ASSERT(channels != 0 && (channels & ~(deUint32)DE_SOCKETCHANNEL_BOTH) == 0);
    625 
    626 	/* Don't attempt to close already closed channels on partially open socket. */
    627 	channels &= sock->openChannels;
    628 
    629 	if (channels == 0)
    630 	{
    631 		deMutex_unlock(sock->stateLock);
    632 		return DE_FALSE;
    633 	}
    634 
    635 #if defined(DE_USE_WINSOCK)
    636 	{
    637 		int how = 0;
    638 
    639 		if ((channels & DE_SOCKETCHANNEL_BOTH) == DE_SOCKETCHANNEL_BOTH)
    640 			how = SD_BOTH;
    641 		else if (channels & DE_SOCKETCHANNEL_SEND)
    642 			how = SD_SEND;
    643 		else if (channels & DE_SOCKETCHANNEL_RECEIVE)
    644 			how = SD_RECEIVE;
    645 
    646 		if (shutdown(sock->handle, how) == 0)
    647 			closedChannels = channels;
    648 		else
    649 		{
    650 			int err = WSAGetLastError();
    651 
    652 			/* \note Due to asynchronous behavior certain errors are perfectly ok. */
    653 			if (err == WSAECONNABORTED || err == WSAECONNRESET || err == WSAENOTCONN)
    654 				closedChannels = DE_SOCKETCHANNEL_BOTH;
    655 			else
    656 			{
    657 				deMutex_unlock(sock->stateLock);
    658 				return DE_FALSE;
    659 			}
    660 		}
    661 	}
    662 #else
    663 	{
    664 		int how = 0;
    665 
    666 		if ((channels & DE_SOCKETCHANNEL_BOTH) == DE_SOCKETCHANNEL_BOTH)
    667 			how = SHUT_RDWR;
    668 		else if (channels & DE_SOCKETCHANNEL_SEND)
    669 			how = SHUT_WR;
    670 		else if (channels & DE_SOCKETCHANNEL_RECEIVE)
    671 			how = SHUT_RD;
    672 
    673 		if (shutdown(sock->handle, how) == 0)
    674 			closedChannels = channels;
    675 		else
    676 		{
    677 			if (errno == ENOTCONN)
    678 				closedChannels = DE_SOCKETCHANNEL_BOTH;
    679 			else
    680 			{
    681 				deMutex_unlock(sock->stateLock);
    682 				return DE_FALSE;
    683 			}
    684 		}
    685 	}
    686 #endif
    687 
    688 	sock->openChannels &= ~closedChannels;
    689 	if (sock->openChannels == 0)
    690 		sock->state = DE_SOCKETSTATE_DISCONNECTED;
    691 
    692 	deMutex_unlock(sock->stateLock);
    693 	return DE_TRUE;
    694 }
    695 
    696 deBool deSocket_close (deSocket* sock)
    697 {
    698 	deMutex_lock(sock->stateLock);
    699 
    700 	if (sock->state == DE_SOCKETSTATE_CLOSED)
    701 	{
    702 		deMutex_unlock(sock->stateLock);
    703 		return DE_FALSE;
    704 	}
    705 
    706 #if !defined(DE_USE_WINSOCK)
    707 	if (sock->state == DE_SOCKETSTATE_LISTENING)
    708 	{
    709 		/* There can be a thread blockin in accept(). Release it by calling shutdown. */
    710 		shutdown(sock->handle, SHUT_RDWR);
    711 	}
    712 #endif
    713 
    714 #if defined(DE_USE_WINSOCK)
    715 	if (closesocket(sock->handle) != 0)
    716 		return DE_FALSE;
    717 #else
    718 	if (close(sock->handle) != 0)
    719 		return DE_FALSE;
    720 #endif
    721 	sock->state			= DE_SOCKETSTATE_CLOSED;
    722 	sock->handle		= DE_INVALID_SOCKET_HANDLE;
    723 	sock->openChannels	= 0;
    724 
    725 	deMutex_unlock(sock->stateLock);
    726 	return DE_TRUE;
    727 }
    728 
    729 static deSocketResult mapSendRecvResult (int numBytes)
    730 {
    731 	if (numBytes > 0)
    732 		return DE_SOCKETRESULT_SUCCESS;
    733 	else if (numBytes == 0)
    734 		return DE_SOCKETRESULT_CONNECTION_CLOSED;
    735 	else
    736 	{
    737 		/* Other errors. */
    738 #if defined(DE_USE_WINSOCK)
    739 		int	error = WSAGetLastError();
    740 		switch (error)
    741 		{
    742 			case WSAEWOULDBLOCK:	return DE_SOCKETRESULT_WOULD_BLOCK;
    743 			case WSAENETDOWN:
    744 			case WSAENETRESET:
    745 			case WSAECONNABORTED:
    746 			case WSAECONNRESET:		return DE_SOCKETRESULT_CONNECTION_TERMINATED;
    747 			default:				return DE_SOCKETRESULT_ERROR;
    748 		}
    749 #else
    750 		switch (errno)
    751 		{
    752 			case EAGAIN:		return DE_SOCKETRESULT_WOULD_BLOCK;
    753 			case ECONNABORTED:
    754 			case ECONNRESET:	return DE_SOCKETRESULT_CONNECTION_TERMINATED;
    755 			default:			return DE_SOCKETRESULT_ERROR;
    756 		}
    757 #endif
    758 	}
    759 }
    760 
    761 DE_INLINE void deSocket_setChannelsClosed (deSocket* sock, deUint32 channels)
    762 {
    763 	deMutex_lock(sock->stateLock);
    764 
    765 	sock->openChannels &= ~channels;
    766 	if (sock->openChannels == 0)
    767 		sock->state = DE_SOCKETSTATE_DISCONNECTED;
    768 
    769 	deMutex_unlock(sock->stateLock);
    770 }
    771 
    772 deSocketResult deSocket_send (deSocket* sock, const void* buf, size_t bufSize, size_t* numSentPtr)
    773 {
    774 	int				numSent	= (int)send(sock->handle, (const char*)buf, (NativeSize)bufSize, 0);
    775 	deSocketResult	result	= mapSendRecvResult(numSent);
    776 
    777 	if (numSentPtr)
    778 		*numSentPtr = (numSent > 0) ? ((size_t)numSent) : (0);
    779 
    780 	/* Update state. */
    781 	if (result == DE_SOCKETRESULT_CONNECTION_CLOSED)
    782 		deSocket_setChannelsClosed(sock, DE_SOCKETCHANNEL_SEND);
    783 	else if (result == DE_SOCKETRESULT_CONNECTION_TERMINATED)
    784 		deSocket_setChannelsClosed(sock, DE_SOCKETCHANNEL_BOTH);
    785 
    786 	return result;
    787 }
    788 
    789 deSocketResult deSocket_receive (deSocket* sock, void* buf, size_t bufSize, size_t* numReceivedPtr)
    790 {
    791 	int				numRecv	= (int)recv(sock->handle, (char*)buf, (NativeSize)bufSize, 0);
    792 	deSocketResult	result	= mapSendRecvResult(numRecv);
    793 
    794 	if (numReceivedPtr)
    795 		*numReceivedPtr = (numRecv > 0) ? ((size_t)numRecv) : (0);
    796 
    797 	/* Update state. */
    798 	if (result == DE_SOCKETRESULT_CONNECTION_CLOSED)
    799 		deSocket_setChannelsClosed(sock, DE_SOCKETCHANNEL_RECEIVE);
    800 	else if (result == DE_SOCKETRESULT_CONNECTION_TERMINATED)
    801 		deSocket_setChannelsClosed(sock, DE_SOCKETCHANNEL_BOTH);
    802 
    803 	return result;
    804 }
    805 
    806 #endif
    807