123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557 |
- /**
- @file unix.c
- @brief ENet Unix system specific functions
- */
- #ifndef _WIN32
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <sys/ioctl.h>
- #include <sys/time.h>
- #include <arpa/inet.h>
- #include <netinet/tcp.h>
- #include <netdb.h>
- #include <unistd.h>
- #include <string.h>
- #include <errno.h>
- #include <time.h>
- #define ENET_BUILDING_LIB 1
- #include "enet/enet.h"
- #ifdef __APPLE__
- #ifdef HAS_POLL
- #undef HAS_POLL
- #endif
- #ifndef HAS_FCNTL
- #define HAS_FCNTL 1
- #endif
- #ifndef HAS_INET_PTON
- #define HAS_INET_PTON 1
- #endif
- #ifndef HAS_INET_NTOP
- #define HAS_INET_NTOP 1
- #endif
- #ifndef HAS_MSGHDR_FLAGS
- #define HAS_MSGHDR_FLAGS 1
- #endif
- #ifndef HAS_SOCKLEN_T
- #define HAS_SOCKLEN_T 1
- #endif
- #endif
- #ifdef HAS_FCNTL
- #include <fcntl.h>
- #endif
- #ifdef HAS_POLL
- #include <sys/poll.h>
- #endif
- #ifndef HAS_SOCKLEN_T
- typedef int socklen_t;
- #endif
- #ifndef MSG_NOSIGNAL
- #define MSG_NOSIGNAL 0
- #endif
- static enet_uint32 timeBase = 0;
- int
- enet_initialize (void)
- {
- return 0;
- }
- void
- enet_deinitialize (void)
- {
- }
- enet_uint32
- enet_host_random_seed (void)
- {
- return (enet_uint32) time (NULL);
- }
- enet_uint32
- enet_time_get (void)
- {
- struct timeval timeVal;
- gettimeofday (& timeVal, NULL);
- return timeVal.tv_sec * 1000 + timeVal.tv_usec / 1000 - timeBase;
- }
- void
- enet_time_set (enet_uint32 newTimeBase)
- {
- struct timeval timeVal;
- gettimeofday (& timeVal, NULL);
-
- timeBase = timeVal.tv_sec * 1000 + timeVal.tv_usec / 1000 - newTimeBase;
- }
- int
- enet_address_set_host (ENetAddress * address, const char * name)
- {
- struct hostent * hostEntry = NULL;
- #ifdef HAS_GETHOSTBYNAME_R
- struct hostent hostData;
- char buffer [2048];
- int errnum;
- #if defined(linux) || defined(__linux) || defined(__linux__) || defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
- gethostbyname_r (name, & hostData, buffer, sizeof (buffer), & hostEntry, & errnum);
- #else
- hostEntry = gethostbyname_r (name, & hostData, buffer, sizeof (buffer), & errnum);
- #endif
- #else
- hostEntry = gethostbyname (name);
- #endif
- if (hostEntry == NULL ||
- hostEntry -> h_addrtype != AF_INET)
- {
- #ifdef HAS_INET_PTON
- if (! inet_pton (AF_INET, name, & address -> host))
- #else
- if (! inet_aton (name, (struct in_addr *) & address -> host))
- #endif
- return -1;
- 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)
- {
- #ifdef HAS_INET_NTOP
- if (inet_ntop (AF_INET, & address -> host, name, nameLength) == NULL)
- #else
- char * addr = inet_ntoa (* (struct in_addr *) & address -> host);
- if (addr != NULL)
- {
- size_t addrLen = strlen(addr);
- if (addrLen >= nameLength)
- return -1;
- memcpy (name, addr, addrLen + 1);
- }
- else
- #endif
- return -1;
- return 0;
- }
- int
- enet_address_get_host (const ENetAddress * address, char * name, size_t nameLength)
- {
- struct in_addr in;
- struct hostent * hostEntry = NULL;
- #ifdef HAS_GETHOSTBYADDR_R
- struct hostent hostData;
- char buffer [2048];
- int errnum;
- in.s_addr = address -> host;
- #if defined(linux) || defined(__linux) || defined(__linux__) || defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
- gethostbyaddr_r ((char *) & in, sizeof (struct in_addr), AF_INET, & hostData, buffer, sizeof (buffer), & hostEntry, & errnum);
- #else
- hostEntry = gethostbyaddr_r ((char *) & in, sizeof (struct in_addr), AF_INET, & hostData, buffer, sizeof (buffer), & errnum);
- #endif
- #else
- in.s_addr = address -> host;
- hostEntry = gethostbyaddr ((char *) & in, sizeof (struct in_addr), AF_INET);
- #endif
- 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));
- }
- int
- enet_socket_get_address (ENetSocket socket, ENetAddress * address)
- {
- struct sockaddr_in sin;
- socklen_t 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);
- }
- 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 = -1;
- switch (option)
- {
- case ENET_SOCKOPT_NONBLOCK:
- #ifdef HAS_FCNTL
- result = fcntl (socket, F_SETFL, (value ? O_NONBLOCK : 0) | (fcntl (socket, F_GETFL) & ~O_NONBLOCK));
- #else
- result = ioctl (socket, FIONBIO, & value);
- #endif
- 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:
- {
- struct timeval timeVal;
- timeVal.tv_sec = value / 1000;
- timeVal.tv_usec = (value % 1000) * 1000;
- result = setsockopt (socket, SOL_SOCKET, SO_RCVTIMEO, (char *) & timeVal, sizeof (struct timeval));
- break;
- }
- case ENET_SOCKOPT_SNDTIMEO:
- {
- struct timeval timeVal;
- timeVal.tv_sec = value / 1000;
- timeVal.tv_usec = (value % 1000) * 1000;
- result = setsockopt (socket, SOL_SOCKET, SO_SNDTIMEO, (char *) & timeVal, sizeof (struct timeval));
- break;
- }
- case ENET_SOCKOPT_NODELAY:
- result = setsockopt (socket, IPPROTO_TCP, TCP_NODELAY, (char *) & value, sizeof (int));
- break;
- default:
- break;
- }
- return result == -1 ? -1 : 0;
- }
- int
- enet_socket_get_option (ENetSocket socket, ENetSocketOption option, int * value)
- {
- int result = -1;
- socklen_t len;
- switch (option)
- {
- case ENET_SOCKOPT_ERROR:
- len = sizeof (int);
- result = getsockopt (socket, SOL_SOCKET, SO_ERROR, value, & len);
- break;
- default:
- break;
- }
- return result == -1 ? -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 == -1 && errno == EINPROGRESS)
- return 0;
- return result;
- }
- ENetSocket
- enet_socket_accept (ENetSocket socket, ENetAddress * address)
- {
- int result;
- struct sockaddr_in sin;
- socklen_t sinLength = sizeof (struct sockaddr_in);
- result = accept (socket,
- address != NULL ? (struct sockaddr *) & sin : NULL,
- address != NULL ? & sinLength : NULL);
-
- if (result == -1)
- 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);
- }
- void
- enet_socket_destroy (ENetSocket socket)
- {
- if (socket != -1)
- close (socket);
- }
- int
- enet_socket_send (ENetSocket socket,
- const ENetAddress * address,
- const ENetBuffer * buffers,
- size_t bufferCount)
- {
- struct msghdr msgHdr;
- struct sockaddr_in sin;
- int sentLength;
- memset (& msgHdr, 0, sizeof (struct msghdr));
- 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;
- msgHdr.msg_name = & sin;
- msgHdr.msg_namelen = sizeof (struct sockaddr_in);
- }
- msgHdr.msg_iov = (struct iovec *) buffers;
- msgHdr.msg_iovlen = bufferCount;
- sentLength = sendmsg (socket, & msgHdr, MSG_NOSIGNAL);
-
- if (sentLength == -1)
- {
- if (errno == EWOULDBLOCK)
- return 0;
- return -1;
- }
- return sentLength;
- }
- int
- enet_socket_receive (ENetSocket socket,
- ENetAddress * address,
- ENetBuffer * buffers,
- size_t bufferCount)
- {
- struct msghdr msgHdr;
- struct sockaddr_in sin;
- int recvLength;
- memset (& msgHdr, 0, sizeof (struct msghdr));
- if (address != NULL)
- {
- msgHdr.msg_name = & sin;
- msgHdr.msg_namelen = sizeof (struct sockaddr_in);
- }
- msgHdr.msg_iov = (struct iovec *) buffers;
- msgHdr.msg_iovlen = bufferCount;
- recvLength = recvmsg (socket, & msgHdr, MSG_NOSIGNAL);
- if (recvLength == -1)
- {
- if (errno == EWOULDBLOCK)
- return 0;
- return -1;
- }
- #ifdef HAS_MSGHDR_FLAGS
- if (msgHdr.msg_flags & MSG_TRUNC)
- return -1;
- #endif
- if (address != NULL)
- {
- address -> host = (enet_uint32) sin.sin_addr.s_addr;
- address -> port = ENET_NET_TO_HOST_16 (sin.sin_port);
- }
- return 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)
- {
- #ifdef HAS_POLL
- struct pollfd pollSocket;
- int pollCount;
-
- pollSocket.fd = socket;
- pollSocket.events = 0;
- if (* condition & ENET_SOCKET_WAIT_SEND)
- pollSocket.events |= POLLOUT;
- if (* condition & ENET_SOCKET_WAIT_RECEIVE)
- pollSocket.events |= POLLIN;
- pollCount = poll (& pollSocket, 1, timeout);
- if (pollCount < 0)
- {
- if (errno == EINTR && * condition & ENET_SOCKET_WAIT_INTERRUPT)
- {
- * condition = ENET_SOCKET_WAIT_INTERRUPT;
- return 0;
- }
- return -1;
- }
- * condition = ENET_SOCKET_WAIT_NONE;
- if (pollCount == 0)
- return 0;
- if (pollSocket.revents & POLLOUT)
- * condition |= ENET_SOCKET_WAIT_SEND;
-
- if (pollSocket.revents & POLLIN)
- * condition |= ENET_SOCKET_WAIT_RECEIVE;
- return 0;
- #else
- 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)
- {
- if (errno == EINTR && * condition & ENET_SOCKET_WAIT_INTERRUPT)
- {
- * condition = ENET_SOCKET_WAIT_INTERRUPT;
- return 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
- }
- #endif
|