123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422 |
- /**
- @file win32.c
- @brief ENet Win32 system specific functions
- */
- #ifdef _WIN32
- #define ENET_BUILDING_LIB 1
- #include "enet/enet.h"
- #include <windows.h>
- #include <mmsystem.h>
- static enet_uint32 timeBase = 0;
- int
- enet_initialize (void)
- {
- WORD versionRequested = MAKEWORD (1, 1);
- WSADATA wsaData;
-
- if (WSAStartup (versionRequested, & wsaData))
- return -1;
- if (LOBYTE (wsaData.wVersion) != 1||
- HIBYTE (wsaData.wVersion) != 1)
- {
- WSACleanup ();
-
- return -1;
- }
- timeBeginPeriod (1);
- return 0;
- }
- void
- enet_deinitialize (void)
- {
- timeEndPeriod (1);
- WSACleanup ();
- }
- enet_uint32
- enet_host_random_seed (void)
- {
- return (enet_uint32) timeGetTime ();
- }
- enet_uint32
- enet_time_get (void)
- {
- return (enet_uint32) timeGetTime () - timeBase;
- }
- void
- enet_time_set (enet_uint32 newTimeBase)
- {
- timeBase = (enet_uint32) timeGetTime () - newTimeBase;
- }
- int
- enet_address_set_host (ENetAddress * address, const char * name)
- {
- struct hostent * hostEntry;
- hostEntry = gethostbyname (name);
- if (hostEntry == NULL ||
- hostEntry -> h_addrtype != AF_INET)
- {
- unsigned long host = inet_addr (name);
- if (host == INADDR_NONE)
- return -1;
- address -> host = host;
- return 0;
- }
- address -> host = * (enet_uint32 *) hostEntry -> h_addr_list [0];
- return 0;
- }
- int
- enet_address_get_host_ip (const ENetAddress * address, char * name, size_t nameLength)
- {
- char * addr = inet_ntoa (* (struct in_addr *) & address -> host);
- if (addr == NULL)
- return -1;
- else
- {
- size_t addrLen = strlen(addr);
- if (addrLen >= nameLength)
- return -1;
- memcpy (name, addr, addrLen + 1);
- }
- return 0;
- }
- int
- enet_address_get_host (const ENetAddress * address, char * name, size_t nameLength)
- {
- struct in_addr in;
- struct hostent * hostEntry;
-
- in.s_addr = address -> host;
-
- hostEntry = gethostbyaddr ((char *) & in, sizeof (struct in_addr), AF_INET);
- if (hostEntry == NULL)
- return enet_address_get_host_ip (address, name, nameLength);
- else
- {
- size_t hostLen = strlen (hostEntry -> h_name);
- if (hostLen >= nameLength)
- return -1;
- memcpy (name, hostEntry -> h_name, hostLen + 1);
- }
- return 0;
- }
- int
- enet_socket_bind (ENetSocket socket, const ENetAddress * address)
- {
- struct sockaddr_in sin;
- memset (& sin, 0, sizeof (struct sockaddr_in));
- sin.sin_family = AF_INET;
- if (address != NULL)
- {
- sin.sin_port = ENET_HOST_TO_NET_16 (address -> port);
- sin.sin_addr.s_addr = address -> host;
- }
- else
- {
- sin.sin_port = 0;
- sin.sin_addr.s_addr = INADDR_ANY;
- }
- return bind (socket,
- (struct sockaddr *) & sin,
- sizeof (struct sockaddr_in)) == SOCKET_ERROR ? -1 : 0;
- }
- int
- enet_socket_get_address (ENetSocket socket, ENetAddress * address)
- {
- struct sockaddr_in sin;
- int sinLength = sizeof (struct sockaddr_in);
- if (getsockname (socket, (struct sockaddr *) & sin, & sinLength) == -1)
- return -1;
- address -> host = (enet_uint32) sin.sin_addr.s_addr;
- address -> port = ENET_NET_TO_HOST_16 (sin.sin_port);
- return 0;
- }
- int
- enet_socket_listen (ENetSocket socket, int backlog)
- {
- return listen (socket, backlog < 0 ? SOMAXCONN : backlog) == SOCKET_ERROR ? -1 : 0;
- }
- ENetSocket
- enet_socket_create (ENetSocketType type)
- {
- return socket (PF_INET, type == ENET_SOCKET_TYPE_DATAGRAM ? SOCK_DGRAM : SOCK_STREAM, 0);
- }
- int
- enet_socket_set_option (ENetSocket socket, ENetSocketOption option, int value)
- {
- int result = SOCKET_ERROR;
- switch (option)
- {
- case ENET_SOCKOPT_NONBLOCK:
- {
- u_long nonBlocking = (u_long) value;
- result = ioctlsocket (socket, FIONBIO, & nonBlocking);
- break;
- }
- case ENET_SOCKOPT_BROADCAST:
- result = setsockopt (socket, SOL_SOCKET, SO_BROADCAST, (char *) & value, sizeof (int));
- break;
- case ENET_SOCKOPT_REUSEADDR:
- result = setsockopt (socket, SOL_SOCKET, SO_REUSEADDR, (char *) & value, sizeof (int));
- break;
- case ENET_SOCKOPT_RCVBUF:
- result = setsockopt (socket, SOL_SOCKET, SO_RCVBUF, (char *) & value, sizeof (int));
- break;
- case ENET_SOCKOPT_SNDBUF:
- result = setsockopt (socket, SOL_SOCKET, SO_SNDBUF, (char *) & value, sizeof (int));
- break;
- case ENET_SOCKOPT_RCVTIMEO:
- result = setsockopt (socket, SOL_SOCKET, SO_RCVTIMEO, (char *) & value, sizeof (int));
- break;
- case ENET_SOCKOPT_SNDTIMEO:
- result = setsockopt (socket, SOL_SOCKET, SO_SNDTIMEO, (char *) & value, sizeof (int));
- break;
- case ENET_SOCKOPT_NODELAY:
- result = setsockopt (socket, IPPROTO_TCP, TCP_NODELAY, (char *) & value, sizeof (int));
- break;
- default:
- break;
- }
- return result == SOCKET_ERROR ? -1 : 0;
- }
- int
- enet_socket_get_option (ENetSocket socket, ENetSocketOption option, int * value)
- {
- int result = SOCKET_ERROR, len;
- switch (option)
- {
- case ENET_SOCKOPT_ERROR:
- len = sizeof(int);
- result = getsockopt (socket, SOL_SOCKET, SO_ERROR, (char *) value, & len);
- break;
- default:
- break;
- }
- return result == SOCKET_ERROR ? -1 : 0;
- }
- int
- enet_socket_connect (ENetSocket socket, const ENetAddress * address)
- {
- struct sockaddr_in sin;
- int result;
- memset (& sin, 0, sizeof (struct sockaddr_in));
- sin.sin_family = AF_INET;
- sin.sin_port = ENET_HOST_TO_NET_16 (address -> port);
- sin.sin_addr.s_addr = address -> host;
- result = connect (socket, (struct sockaddr *) & sin, sizeof (struct sockaddr_in));
- if (result == SOCKET_ERROR && WSAGetLastError () != WSAEWOULDBLOCK)
- return -1;
- return 0;
- }
- ENetSocket
- enet_socket_accept (ENetSocket socket, ENetAddress * address)
- {
- SOCKET result;
- struct sockaddr_in sin;
- int sinLength = sizeof (struct sockaddr_in);
- result = accept (socket,
- address != NULL ? (struct sockaddr *) & sin : NULL,
- address != NULL ? & sinLength : NULL);
- if (result == INVALID_SOCKET)
- return ENET_SOCKET_NULL;
- if (address != NULL)
- {
- address -> host = (enet_uint32) sin.sin_addr.s_addr;
- address -> port = ENET_NET_TO_HOST_16 (sin.sin_port);
- }
- return result;
- }
- int
- enet_socket_shutdown (ENetSocket socket, ENetSocketShutdown how)
- {
- return shutdown (socket, (int) how) == SOCKET_ERROR ? -1 : 0;
- }
- void
- enet_socket_destroy (ENetSocket socket)
- {
- if (socket != INVALID_SOCKET)
- closesocket (socket);
- }
- int
- enet_socket_send (ENetSocket socket,
- const ENetAddress * address,
- const ENetBuffer * buffers,
- size_t bufferCount)
- {
- struct sockaddr_in sin;
- DWORD sentLength;
- if (address != NULL)
- {
- memset (& sin, 0, sizeof (struct sockaddr_in));
- sin.sin_family = AF_INET;
- sin.sin_port = ENET_HOST_TO_NET_16 (address -> port);
- sin.sin_addr.s_addr = address -> host;
- }
- if (WSASendTo (socket,
- (LPWSABUF) buffers,
- (DWORD) bufferCount,
- & sentLength,
- 0,
- address != NULL ? (struct sockaddr *) & sin : NULL,
- address != NULL ? sizeof (struct sockaddr_in) : 0,
- NULL,
- NULL) == SOCKET_ERROR)
- {
- if (WSAGetLastError () == WSAEWOULDBLOCK)
- return 0;
- return -1;
- }
- return (int) sentLength;
- }
- int
- enet_socket_receive (ENetSocket socket,
- ENetAddress * address,
- ENetBuffer * buffers,
- size_t bufferCount)
- {
- INT sinLength = sizeof (struct sockaddr_in);
- DWORD flags = 0,
- recvLength;
- struct sockaddr_in sin;
- if (WSARecvFrom (socket,
- (LPWSABUF) buffers,
- (DWORD) bufferCount,
- & recvLength,
- & flags,
- address != NULL ? (struct sockaddr *) & sin : NULL,
- address != NULL ? & sinLength : NULL,
- NULL,
- NULL) == SOCKET_ERROR)
- {
- switch (WSAGetLastError ())
- {
- case WSAEWOULDBLOCK:
- case WSAECONNRESET:
- return 0;
- }
- return -1;
- }
- if (flags & MSG_PARTIAL)
- return -1;
- if (address != NULL)
- {
- address -> host = (enet_uint32) sin.sin_addr.s_addr;
- address -> port = ENET_NET_TO_HOST_16 (sin.sin_port);
- }
- return (int) recvLength;
- }
- int
- enet_socketset_select (ENetSocket maxSocket, ENetSocketSet * readSet, ENetSocketSet * writeSet, enet_uint32 timeout)
- {
- struct timeval timeVal;
- timeVal.tv_sec = timeout / 1000;
- timeVal.tv_usec = (timeout % 1000) * 1000;
- return select (maxSocket + 1, readSet, writeSet, NULL, & timeVal);
- }
- int
- enet_socket_wait (ENetSocket socket, enet_uint32 * condition, enet_uint32 timeout)
- {
- fd_set readSet, writeSet;
- struct timeval timeVal;
- int selectCount;
-
- timeVal.tv_sec = timeout / 1000;
- timeVal.tv_usec = (timeout % 1000) * 1000;
-
- FD_ZERO (& readSet);
- FD_ZERO (& writeSet);
- if (* condition & ENET_SOCKET_WAIT_SEND)
- FD_SET (socket, & writeSet);
- if (* condition & ENET_SOCKET_WAIT_RECEIVE)
- FD_SET (socket, & readSet);
- selectCount = select (socket + 1, & readSet, & writeSet, NULL, & timeVal);
- if (selectCount < 0)
- return -1;
- * condition = ENET_SOCKET_WAIT_NONE;
- if (selectCount == 0)
- return 0;
- if (FD_ISSET (socket, & writeSet))
- * condition |= ENET_SOCKET_WAIT_SEND;
-
- if (FD_ISSET (socket, & readSet))
- * condition |= ENET_SOCKET_WAIT_RECEIVE;
- return 0;
- }
- #endif
|