// Written by Forest Hale 2003-06-15 and placed into public domain. #ifdef WIN32 #ifdef _MSC_VER #pragma comment(lib, "ws2_32.lib") #endif # ifdef SUPPORTIPV6 // Windows XP or higher is required for getaddrinfo, but the inclusion of wspiapi provides fallbacks for older versions # define _WIN32_WINNT 0x0501 # endif # include # include # ifdef USE_WSPIAPI_H # include # endif #endif #ifndef STANDALONETEST #include "quakedef.h" #endif #include #include #include #include #ifndef WIN32 #include #include #include #include #include #include #include #include #ifdef SUPPORTIPV6 #include #endif #endif #ifdef __MORPHOS__ #include #endif // for Z_Malloc/Z_Free in quake #ifndef STANDALONETEST #include "zone.h" #include "sys.h" #include "netconn.h" #else #define Con_Print printf #define Con_Printf printf #define Z_Malloc malloc #define Z_Free free #endif #include "lhnet.h" #if defined(WIN32) #define EWOULDBLOCK WSAEWOULDBLOCK #define ECONNREFUSED WSAECONNREFUSED #define SOCKETERRNO WSAGetLastError() #define IOC_VENDOR 0x18000000 #define _WSAIOW(x,y) (IOC_IN|(x)|(y)) #define SIO_UDP_CONNRESET _WSAIOW(IOC_VENDOR,12) #define SOCKLEN_T int #elif defined(__MORPHOS__) #define ioctlsocket IoctlSocket #define closesocket CloseSocket #define SOCKETERRNO Errno() #define SOCKLEN_T int #else #define ioctlsocket ioctl #define closesocket close #define SOCKETERRNO errno #define SOCKLEN_T socklen_t #endif #ifdef MSG_DONTWAIT #define LHNET_RECVFROM_FLAGS MSG_DONTWAIT #define LHNET_SENDTO_FLAGS 0 #else #define LHNET_RECVFROM_FLAGS 0 #define LHNET_SENDTO_FLAGS 0 #endif typedef struct lhnetaddressnative_s { lhnetaddresstype_t addresstype; int port; union { struct sockaddr sock; struct sockaddr_in in; #ifdef SUPPORTIPV6 struct sockaddr_in6 in6; #endif } addr; } lhnetaddressnative_t; // to make LHNETADDRESS_FromString resolve repeated hostnames faster, cache them #define MAX_NAMECACHE 64 static struct namecache_s { lhnetaddressnative_t address; double expirationtime; char name[64]; } namecache[MAX_NAMECACHE]; static int namecacheposition = 0; int LHNETADDRESS_FromPort(lhnetaddress_t *vaddress, lhnetaddresstype_t addresstype, int port) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; if (!address) return 0; switch(addresstype) { default: break; case LHNETADDRESSTYPE_LOOP: // local:port (loopback) memset(address, 0, sizeof(*address)); address->addresstype = LHNETADDRESSTYPE_LOOP; address->port = port; return 1; case LHNETADDRESSTYPE_INET4: // 0.0.0.0:port (INADDR_ANY, binds to all interfaces) memset(address, 0, sizeof(*address)); address->addresstype = LHNETADDRESSTYPE_INET4; address->port = port; address->addr.in.sin_family = AF_INET; address->addr.in.sin_port = htons((unsigned short)port); return 1; #ifdef SUPPORTIPV6 case LHNETADDRESSTYPE_INET6: // [0:0:0:0:0:0:0:0]:port (IN6ADDR_ANY, binds to all interfaces) memset(address, 0, sizeof(*address)); address->addresstype = LHNETADDRESSTYPE_INET6; address->port = port; address->addr.in6.sin6_family = AF_INET6; address->addr.in6.sin6_port = htons((unsigned short)port); return 1; #endif } return 0; } #ifdef SUPPORTIPV6 int LHNETADDRESS_Resolve(lhnetaddressnative_t *address, const char *name, int port) { char port_buff [16]; struct addrinfo hints; struct addrinfo* addrinf; int err; dpsnprintf (port_buff, sizeof (port_buff), "%d", port); port_buff[sizeof (port_buff) - 1] = '\0'; memset(&hints, 0, sizeof (hints)); hints.ai_family = AF_UNSPEC; hints.ai_socktype = SOCK_DGRAM; //hints.ai_flags = AI_PASSIVE; err = getaddrinfo(name, port_buff, &hints, &addrinf); if (err != 0 || addrinf == NULL) return 0; if (addrinf->ai_addr->sa_family != AF_INET6 && addrinf->ai_addr->sa_family != AF_INET) { freeaddrinfo (addrinf); return 0; } // great it worked if (addrinf->ai_addr->sa_family == AF_INET6) { address->addresstype = LHNETADDRESSTYPE_INET6; memcpy(&address->addr.in6, addrinf->ai_addr, sizeof(address->addr.in6)); } else { address->addresstype = LHNETADDRESSTYPE_INET4; memcpy(&address->addr.in, addrinf->ai_addr, sizeof(address->addr.in)); } address->port = port; freeaddrinfo (addrinf); return 1; } int LHNETADDRESS_FromString(lhnetaddress_t *vaddress, const char *string, int defaultport) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; int i, port, d1, d2, d3, d4, resolved; size_t namelen; unsigned char *a; char name[128]; #ifdef STANDALONETEST char string2[128]; #endif const char* addr_start; const char* addr_end = NULL; const char* port_name = NULL; int addr_family = AF_UNSPEC; if (!address || !string || !*string) return 0; memset(address, 0, sizeof(*address)); address->addresstype = LHNETADDRESSTYPE_NONE; port = 0; // If it's a bracketed IPv6 address if (string[0] == '[') { const char* end_bracket = strchr(string, ']'); if (end_bracket == NULL) return 0; if (end_bracket[1] == ':') port_name = end_bracket + 2; else if (end_bracket[1] != '\0') return 0; addr_family = AF_INET6; addr_start = &string[1]; addr_end = end_bracket; } else { const char* first_colon; addr_start = string; // If it's a numeric non-bracket IPv6 address (-> no port), // or it's a numeric IPv4 address, or a name, with a port first_colon = strchr(string, ':'); if (first_colon != NULL) { const char* last_colon = strrchr(first_colon + 1, ':'); // If it's an numeric IPv4 address, or a name, with a port if (last_colon == NULL) { addr_end = first_colon; port_name = first_colon + 1; } else addr_family = AF_INET6; } } if (addr_end != NULL) namelen = addr_end - addr_start; else namelen = strlen (addr_start); if (namelen >= sizeof(name)) namelen = sizeof(name) - 1; memcpy (name, addr_start, namelen); name[namelen] = 0; if (port_name) port = atoi(port_name); if (port == 0) port = defaultport; // handle loopback if (!strcmp(name, "local")) { address->addresstype = LHNETADDRESSTYPE_LOOP; address->port = port; return 1; } // try to parse as dotted decimal ipv4 address first // note this supports partial ip addresses d1 = d2 = d3 = d4 = 0; #if _MSC_VER >= 1400 #define sscanf sscanf_s #endif if (addr_family != AF_INET6 && sscanf(name, "%d.%d.%d.%d", &d1, &d2, &d3, &d4) >= 1 && (unsigned int)d1 < 256 && (unsigned int)d2 < 256 && (unsigned int)d3 < 256 && (unsigned int)d4 < 256) { // parsed a valid ipv4 address address->addresstype = LHNETADDRESSTYPE_INET4; address->port = port; address->addr.in.sin_family = AF_INET; address->addr.in.sin_port = htons((unsigned short)port); a = (unsigned char *)&address->addr.in.sin_addr; a[0] = d1; a[1] = d2; a[2] = d3; a[3] = d4; #ifdef STANDALONETEST LHNETADDRESS_ToString(address, string2, sizeof(string2), 1); printf("manual parsing of ipv4 dotted decimal address \"%s\" successful: %s\n", string, string2); #endif return 1; } for (i = 0;i < MAX_NAMECACHE;i++) if (!strcmp(namecache[i].name, name)) break; #ifdef STANDALONETEST if (i < MAX_NAMECACHE) #else if (i < MAX_NAMECACHE && realtime < namecache[i].expirationtime) #endif { *address = namecache[i].address; address->port = port; if (address->addresstype == LHNETADDRESSTYPE_INET6) { address->addr.in6.sin6_port = htons((unsigned short)port); return 1; } else if (address->addresstype == LHNETADDRESSTYPE_INET4) { address->addr.in.sin_port = htons((unsigned short)port); return 1; } return 0; } for (i = 0;i < (int)sizeof(namecache[namecacheposition].name)-1 && name[i];i++) namecache[namecacheposition].name[i] = name[i]; namecache[namecacheposition].name[i] = 0; #ifndef STANDALONETEST namecache[namecacheposition].expirationtime = realtime + 12 * 3600; // 12 hours #endif // try resolving the address (handles dns and other ip formats) resolved = LHNETADDRESS_Resolve(address, name, port); if (resolved) { #ifdef STANDALONETEST const char *protoname; switch (address->addresstype) { case LHNETADDRESSTYPE_INET6: protoname = "ipv6"; break; case LHNETADDRESSTYPE_INET4: protoname = "ipv4"; break; default: protoname = "UNKNOWN"; break; } LHNETADDRESS_ToString(vaddress, string2, sizeof(string2), 1); Con_Printf("LHNETADDRESS_Resolve(\"%s\") returned %s address %s\n", string, protoname, string2); #endif namecache[namecacheposition].address = *address; } else { #ifdef STANDALONETEST printf("name resolution failed on address \"%s\"\n", name); #endif namecache[namecacheposition].address.addresstype = LHNETADDRESSTYPE_NONE; } namecacheposition = (namecacheposition + 1) % MAX_NAMECACHE; return resolved; } #else int LHNETADDRESS_FromString(lhnetaddress_t *vaddress, const char *string, int defaultport) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; int i, port, namelen, d1, d2, d3, d4; struct hostent *hostentry; unsigned char *a; const char *colon; char name[128]; #ifdef STANDALONETEST char string2[128]; #endif if (!address || !string || !*string) return 0; memset(address, 0, sizeof(*address)); address->addresstype = LHNETADDRESSTYPE_NONE; port = 0; colon = strrchr(string, ':'); if (colon) port = atoi(colon + 1); else colon = string + strlen(string); if (port == 0) port = defaultport; namelen = colon - string; if (namelen > 127) namelen = 127; if (string[0] == '[' && namelen > 0 && string[namelen-1] == ']') // ipv6 { string++; namelen -= 2; } memcpy(name, string, namelen); name[namelen] = 0; // handle loopback if (!strcmp(name, "local")) { address->addresstype = LHNETADDRESSTYPE_LOOP; address->port = port; return 1; } // try to parse as dotted decimal ipv4 address first // note this supports partial ip addresses d1 = d2 = d3 = d4 = 0; #if _MSC_VER >= 1400 #define sscanf sscanf_s #endif if (sscanf(name, "%d.%d.%d.%d", &d1, &d2, &d3, &d4) >= 1 && (unsigned int)d1 < 256 && (unsigned int)d2 < 256 && (unsigned int)d3 < 256 && (unsigned int)d4 < 256) { // parsed a valid ipv4 address address->addresstype = LHNETADDRESSTYPE_INET4; address->port = port; address->addr.in.sin_family = AF_INET; address->addr.in.sin_port = htons((unsigned short)port); a = (unsigned char *)&address->addr.in.sin_addr; a[0] = d1; a[1] = d2; a[2] = d3; a[3] = d4; #ifdef STANDALONETEST LHNETADDRESS_ToString(address, string2, sizeof(string2), 1); printf("manual parsing of ipv4 dotted decimal address \"%s\" successful: %s\n", string, string2); #endif return 1; } for (i = 0;i < MAX_NAMECACHE;i++) if (!strcmp(namecache[i].name, name)) break; #ifdef STANDALONETEST if (i < MAX_NAMECACHE) #else if (i < MAX_NAMECACHE && realtime < namecache[i].expirationtime) #endif { *address = namecache[i].address; address->port = port; if (address->addresstype == LHNETADDRESSTYPE_INET6) { #ifdef SUPPORTIPV6 address->addr.in6.sin6_port = htons((unsigned short)port); return 1; #endif } else if (address->addresstype == LHNETADDRESSTYPE_INET4) { address->addr.in.sin_port = htons((unsigned short)port); return 1; } return 0; } // try gethostbyname (handles dns and other ip formats) hostentry = gethostbyname(name); if (hostentry) { if (hostentry->h_addrtype == AF_INET6) { #ifdef SUPPORTIPV6 // great it worked address->addresstype = LHNETADDRESSTYPE_INET6; address->port = port; address->addr.in6.sin6_family = hostentry->h_addrtype; address->addr.in6.sin6_port = htons((unsigned short)port); memcpy(&address->addr.in6.sin6_addr, hostentry->h_addr_list[0], sizeof(address->addr.in6.sin6_addr)); for (i = 0;i < (int)sizeof(namecache[namecacheposition].name)-1 && name[i];i++) namecache[namecacheposition].name[i] = name[i]; namecache[namecacheposition].name[i] = 0; #ifndef STANDALONETEST namecache[namecacheposition].expirationtime = realtime + 12 * 3600; // 12 hours #endif namecache[namecacheposition].address = *address; namecacheposition = (namecacheposition + 1) % MAX_NAMECACHE; #ifdef STANDALONETEST LHNETADDRESS_ToString(address, string2, sizeof(string2), 1); printf("gethostbyname(\"%s\") returned ipv6 address %s\n", string, string2); #endif return 1; #endif } else if (hostentry->h_addrtype == AF_INET) { // great it worked address->addresstype = LHNETADDRESSTYPE_INET4; address->port = port; address->addr.in.sin_family = hostentry->h_addrtype; address->addr.in.sin_port = htons((unsigned short)port); memcpy(&address->addr.in.sin_addr, hostentry->h_addr_list[0], sizeof(address->addr.in.sin_addr)); for (i = 0;i < (int)sizeof(namecache[namecacheposition].name)-1 && name[i];i++) namecache[namecacheposition].name[i] = name[i]; namecache[namecacheposition].name[i] = 0; #ifndef STANDALONETEST namecache[namecacheposition].expirationtime = realtime + 12 * 3600; // 12 hours #endif namecache[namecacheposition].address = *address; namecacheposition = (namecacheposition + 1) % MAX_NAMECACHE; #ifdef STANDALONETEST LHNETADDRESS_ToString(address, string2, sizeof(string2), 1); printf("gethostbyname(\"%s\") returned ipv4 address %s\n", string, string2); #endif return 1; } } #ifdef STANDALONETEST printf("gethostbyname failed on address \"%s\"\n", name); #endif for (i = 0;i < (int)sizeof(namecache[namecacheposition].name)-1 && name[i];i++) namecache[namecacheposition].name[i] = name[i]; namecache[namecacheposition].name[i] = 0; #ifndef STANDALONETEST namecache[namecacheposition].expirationtime = realtime + 12 * 3600; // 12 hours #endif namecache[namecacheposition].address.addresstype = LHNETADDRESSTYPE_NONE; namecacheposition = (namecacheposition + 1) % MAX_NAMECACHE; return 0; } #endif int LHNETADDRESS_ToString(const lhnetaddress_t *vaddress, char *string, int stringbuffersize, int includeport) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; const unsigned char *a; *string = 0; if (!address || !string || stringbuffersize < 1) return 0; switch(address->addresstype) { default: break; case LHNETADDRESSTYPE_LOOP: if (includeport) { if (stringbuffersize >= 12) { dpsnprintf(string, stringbuffersize, "local:%d", address->port); return 1; } } else { if (stringbuffersize >= 6) { memcpy(string, "local", 6); return 1; } } break; case LHNETADDRESSTYPE_INET4: a = (const unsigned char *)(&address->addr.in.sin_addr); if (includeport) { if (stringbuffersize >= 22) { dpsnprintf(string, stringbuffersize, "%d.%d.%d.%d:%d", a[0], a[1], a[2], a[3], address->port); return 1; } } else { if (stringbuffersize >= 16) { dpsnprintf(string, stringbuffersize, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]); return 1; } } break; #ifdef SUPPORTIPV6 case LHNETADDRESSTYPE_INET6: a = (const unsigned char *)(&address->addr.in6.sin6_addr); if (includeport) { if (stringbuffersize >= 88) { dpsnprintf(string, stringbuffersize, "[%x:%x:%x:%x:%x:%x:%x:%x]:%d", a[0] * 256 + a[1], a[2] * 256 + a[3], a[4] * 256 + a[5], a[6] * 256 + a[7], a[8] * 256 + a[9], a[10] * 256 + a[11], a[12] * 256 + a[13], a[14] * 256 + a[15], address->port); return 1; } } else { if (stringbuffersize >= 80) { dpsnprintf(string, stringbuffersize, "%x:%x:%x:%x:%x:%x:%x:%x", a[0] * 256 + a[1], a[2] * 256 + a[3], a[4] * 256 + a[5], a[6] * 256 + a[7], a[8] * 256 + a[9], a[10] * 256 + a[11], a[12] * 256 + a[13], a[14] * 256 + a[15]); return 1; } } break; #endif } return 0; } int LHNETADDRESS_GetAddressType(const lhnetaddress_t *address) { if (address) return address->addresstype; else return LHNETADDRESSTYPE_NONE; } const char *LHNETADDRESS_GetInterfaceName(const lhnetaddress_t *vaddress) { #ifdef SUPPORTIPV6 lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; if (address && address->addresstype == LHNETADDRESSTYPE_INET6) { #ifndef _WIN32 static char ifname [IF_NAMESIZE]; if (if_indextoname(address->addr.in6.sin6_scope_id, ifname) == ifname) return ifname; #else // The Win32 API doesn't have if_indextoname() until Windows Vista, // but luckily it just uses the interface ID as the interface name static char ifname [16]; if (dpsnprintf(ifname, sizeof(ifname), "%lu", address->addr.in6.sin6_scope_id) > 0) return ifname; #endif } #endif return NULL; } int LHNETADDRESS_GetPort(const lhnetaddress_t *address) { if (!address) return -1; return address->port; } int LHNETADDRESS_SetPort(lhnetaddress_t *vaddress, int port) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; if (!address) return 0; address->port = port; switch(address->addresstype) { case LHNETADDRESSTYPE_LOOP: return 1; case LHNETADDRESSTYPE_INET4: address->addr.in.sin_port = htons((unsigned short)port); return 1; #ifdef SUPPORTIPV6 case LHNETADDRESSTYPE_INET6: address->addr.in6.sin6_port = htons((unsigned short)port); return 1; #endif default: return 0; } } int LHNETADDRESS_Compare(const lhnetaddress_t *vaddress1, const lhnetaddress_t *vaddress2) { lhnetaddressnative_t *address1 = (lhnetaddressnative_t *)vaddress1; lhnetaddressnative_t *address2 = (lhnetaddressnative_t *)vaddress2; if (!address1 || !address2) return 1; if (address1->addresstype != address2->addresstype) return 1; switch(address1->addresstype) { case LHNETADDRESSTYPE_LOOP: if (address1->port != address2->port) return -1; return 0; case LHNETADDRESSTYPE_INET4: if (address1->addr.in.sin_family != address2->addr.in.sin_family) return 1; if (memcmp(&address1->addr.in.sin_addr, &address2->addr.in.sin_addr, sizeof(address1->addr.in.sin_addr))) return 1; if (address1->port != address2->port) return -1; return 0; #ifdef SUPPORTIPV6 case LHNETADDRESSTYPE_INET6: if (address1->addr.in6.sin6_family != address2->addr.in6.sin6_family) return 1; if (memcmp(&address1->addr.in6.sin6_addr, &address2->addr.in6.sin6_addr, sizeof(address1->addr.in6.sin6_addr))) return 1; if (address1->port != address2->port) return -1; return 0; #endif default: return 1; } } typedef struct lhnetpacket_s { void *data; int length; int sourceport; int destinationport; time_t timeout; #ifndef STANDALONETEST double sentdoubletime; #endif struct lhnetpacket_s *next, *prev; } lhnetpacket_t; static int lhnet_active; static lhnetsocket_t lhnet_socketlist; static lhnetpacket_t lhnet_packetlist; #ifdef WIN32 static int lhnet_didWSAStartup = 0; static WSADATA lhnet_winsockdata; #endif void LHNET_Init(void) { if (lhnet_active) return; lhnet_socketlist.next = lhnet_socketlist.prev = &lhnet_socketlist; lhnet_packetlist.next = lhnet_packetlist.prev = &lhnet_packetlist; lhnet_active = 1; #ifdef WIN32 lhnet_didWSAStartup = !WSAStartup(MAKEWORD(1, 1), &lhnet_winsockdata); if (!lhnet_didWSAStartup) Con_Print("LHNET_Init: WSAStartup failed, networking disabled\n"); #endif } void LHNET_Shutdown(void) { lhnetpacket_t *p; if (!lhnet_active) return; while (lhnet_socketlist.next != &lhnet_socketlist) LHNET_CloseSocket(lhnet_socketlist.next); while (lhnet_packetlist.next != &lhnet_packetlist) { p = lhnet_packetlist.next; p->prev->next = p->next; p->next->prev = p->prev; Z_Free(p); } #ifdef WIN32 if (lhnet_didWSAStartup) { lhnet_didWSAStartup = 0; WSACleanup(); } #endif lhnet_active = 0; } static const char *LHNETPRIVATE_StrError(void) { #ifdef WIN32 int i = WSAGetLastError(); switch (i) { case WSAEINTR: return "WSAEINTR"; case WSAEBADF: return "WSAEBADF"; case WSAEACCES: return "WSAEACCES"; case WSAEFAULT: return "WSAEFAULT"; case WSAEINVAL: return "WSAEINVAL"; case WSAEMFILE: return "WSAEMFILE"; case WSAEWOULDBLOCK: return "WSAEWOULDBLOCK"; case WSAEINPROGRESS: return "WSAEINPROGRESS"; case WSAEALREADY: return "WSAEALREADY"; case WSAENOTSOCK: return "WSAENOTSOCK"; case WSAEDESTADDRREQ: return "WSAEDESTADDRREQ"; case WSAEMSGSIZE: return "WSAEMSGSIZE"; case WSAEPROTOTYPE: return "WSAEPROTOTYPE"; case WSAENOPROTOOPT: return "WSAENOPROTOOPT"; case WSAEPROTONOSUPPORT: return "WSAEPROTONOSUPPORT"; case WSAESOCKTNOSUPPORT: return "WSAESOCKTNOSUPPORT"; case WSAEOPNOTSUPP: return "WSAEOPNOTSUPP"; case WSAEPFNOSUPPORT: return "WSAEPFNOSUPPORT"; case WSAEAFNOSUPPORT: return "WSAEAFNOSUPPORT"; case WSAEADDRINUSE: return "WSAEADDRINUSE"; case WSAEADDRNOTAVAIL: return "WSAEADDRNOTAVAIL"; case WSAENETDOWN: return "WSAENETDOWN"; case WSAENETUNREACH: return "WSAENETUNREACH"; case WSAENETRESET: return "WSAENETRESET"; case WSAECONNABORTED: return "WSAECONNABORTED"; case WSAECONNRESET: return "WSAECONNRESET"; case WSAENOBUFS: return "WSAENOBUFS"; case WSAEISCONN: return "WSAEISCONN"; case WSAENOTCONN: return "WSAENOTCONN"; case WSAESHUTDOWN: return "WSAESHUTDOWN"; case WSAETOOMANYREFS: return "WSAETOOMANYREFS"; case WSAETIMEDOUT: return "WSAETIMEDOUT"; case WSAECONNREFUSED: return "WSAECONNREFUSED"; case WSAELOOP: return "WSAELOOP"; case WSAENAMETOOLONG: return "WSAENAMETOOLONG"; case WSAEHOSTDOWN: return "WSAEHOSTDOWN"; case WSAEHOSTUNREACH: return "WSAEHOSTUNREACH"; case WSAENOTEMPTY: return "WSAENOTEMPTY"; case WSAEPROCLIM: return "WSAEPROCLIM"; case WSAEUSERS: return "WSAEUSERS"; case WSAEDQUOT: return "WSAEDQUOT"; case WSAESTALE: return "WSAESTALE"; case WSAEREMOTE: return "WSAEREMOTE"; case WSAEDISCON: return "WSAEDISCON"; case 0: return "no error"; default: return "unknown WSAE error"; } #else return strerror(errno); #endif } void LHNET_SleepUntilPacket_Microseconds(int microseconds) { #ifdef FD_SET fd_set fdreadset; struct timeval tv; int lastfd; lhnetsocket_t *s; FD_ZERO(&fdreadset); lastfd = 0; for (s = lhnet_socketlist.next;s != &lhnet_socketlist;s = s->next) { if (s->address.addresstype == LHNETADDRESSTYPE_INET4 || s->address.addresstype == LHNETADDRESSTYPE_INET6) { if (lastfd < s->inetsocket) lastfd = s->inetsocket; #ifdef WIN32 FD_SET((int)s->inetsocket, &fdreadset); #else FD_SET((unsigned int)s->inetsocket, &fdreadset); #endif } } tv.tv_sec = microseconds / 1000000; tv.tv_usec = microseconds % 1000000; select(lastfd + 1, &fdreadset, NULL, NULL, &tv); #else Sys_Sleep(microseconds); #endif } lhnetsocket_t *LHNET_OpenSocket_Connectionless(lhnetaddress_t *address) { lhnetsocket_t *lhnetsocket, *s; if (!address) return NULL; lhnetsocket = (lhnetsocket_t *)Z_Malloc(sizeof(*lhnetsocket)); if (lhnetsocket) { memset(lhnetsocket, 0, sizeof(*lhnetsocket)); lhnetsocket->address = *address; switch(lhnetsocket->address.addresstype) { case LHNETADDRESSTYPE_LOOP: if (lhnetsocket->address.port == 0) { // allocate a port dynamically // this search will always terminate because there is never // an allocated socket with port 0, so if the number wraps it // will find the port is unused, and then refuse to use port // 0, causing an intentional failure condition lhnetsocket->address.port = 1024; for (;;) { for (s = lhnet_socketlist.next;s != &lhnet_socketlist;s = s->next) if (s->address.addresstype == lhnetsocket->address.addresstype && s->address.port == lhnetsocket->address.port) break; if (s == &lhnet_socketlist) break; lhnetsocket->address.port++; } } // check if the port is available for (s = lhnet_socketlist.next;s != &lhnet_socketlist;s = s->next) if (s->address.addresstype == lhnetsocket->address.addresstype && s->address.port == lhnetsocket->address.port) break; if (s == &lhnet_socketlist && lhnetsocket->address.port != 0) { lhnetsocket->next = &lhnet_socketlist; lhnetsocket->prev = lhnetsocket->next->prev; lhnetsocket->next->prev = lhnetsocket; lhnetsocket->prev->next = lhnetsocket; return lhnetsocket; } break; case LHNETADDRESSTYPE_INET4: #ifdef SUPPORTIPV6 case LHNETADDRESSTYPE_INET6: #endif #ifdef WIN32 if (lhnet_didWSAStartup) { #endif #ifdef SUPPORTIPV6 if ((lhnetsocket->inetsocket = socket(address->addresstype == LHNETADDRESSTYPE_INET6 ? PF_INET6 : PF_INET, SOCK_DGRAM, IPPROTO_UDP)) != -1) #else if ((lhnetsocket->inetsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) != -1) #endif { #ifdef WIN32 u_long _false = 0; #endif #ifdef MSG_DONTWAIT if (1) #else #ifdef WIN32 u_long _true = 1; #else char _true = 1; #endif if (ioctlsocket(lhnetsocket->inetsocket, FIONBIO, &_true) != -1) #endif { #ifdef IPV6_V6ONLY // We need to set this flag to tell the OS that we only listen on IPv6. If we don't // most OSes will create a dual-protocol socket that also listens on IPv4. In this case // if an IPv4 socket is already bound to the port we want, our bind() call will fail. int ipv6_only = 1; if (address->addresstype != LHNETADDRESSTYPE_INET6 || setsockopt (lhnetsocket->inetsocket, IPPROTO_IPV6, IPV6_V6ONLY, (const char *)&ipv6_only, sizeof(ipv6_only)) == 0 #ifdef WIN32 // The Win32 API only supports IPV6_V6ONLY since Windows Vista, but fortunately // the default value is what we want on Win32 anyway (IPV6_V6ONLY = true) || SOCKETERRNO == WSAENOPROTOOPT #endif ) #endif { lhnetaddressnative_t *localaddress = (lhnetaddressnative_t *)&lhnetsocket->address; SOCKLEN_T namelen; int bindresult; #ifdef SUPPORTIPV6 if (address->addresstype == LHNETADDRESSTYPE_INET6) { namelen = sizeof(localaddress->addr.in6); bindresult = bind(lhnetsocket->inetsocket, &localaddress->addr.sock, namelen); if (bindresult != -1) getsockname(lhnetsocket->inetsocket, &localaddress->addr.sock, &namelen); } else #endif { namelen = sizeof(localaddress->addr.in); bindresult = bind(lhnetsocket->inetsocket, &localaddress->addr.sock, namelen); if (bindresult != -1) getsockname(lhnetsocket->inetsocket, &localaddress->addr.sock, &namelen); } if (bindresult != -1) { int i = 1; // enable broadcast on this socket setsockopt(lhnetsocket->inetsocket, SOL_SOCKET, SO_BROADCAST, (char *)&i, sizeof(i)); lhnetsocket->next = &lhnet_socketlist; lhnetsocket->prev = lhnetsocket->next->prev; lhnetsocket->next->prev = lhnetsocket; lhnetsocket->prev->next = lhnetsocket; #ifdef WIN32 if (ioctlsocket(lhnetsocket->inetsocket, SIO_UDP_CONNRESET, &_false) == -1) Con_DPrintf("LHNET_OpenSocket_Connectionless: ioctlsocket SIO_UDP_CONNRESET returned error: %s\n", LHNETPRIVATE_StrError()); #endif return lhnetsocket; } else Con_Printf("LHNET_OpenSocket_Connectionless: bind returned error: %s\n", LHNETPRIVATE_StrError()); } #ifdef IPV6_V6ONLY else Con_Printf("LHNET_OpenSocket_Connectionless: setsockopt(IPV6_V6ONLY) returned error: %s\n", LHNETPRIVATE_StrError()); #endif } else Con_Printf("LHNET_OpenSocket_Connectionless: ioctlsocket returned error: %s\n", LHNETPRIVATE_StrError()); closesocket(lhnetsocket->inetsocket); } else Con_Printf("LHNET_OpenSocket_Connectionless: socket returned error: %s\n", LHNETPRIVATE_StrError()); #ifdef WIN32 } else Con_Print("LHNET_OpenSocket_Connectionless: can't open a socket (WSAStartup failed during LHNET_Init)\n"); #endif break; default: break; } Z_Free(lhnetsocket); } return NULL; } void LHNET_CloseSocket(lhnetsocket_t *lhnetsocket) { if (lhnetsocket) { // unlink from socket list if (lhnetsocket->next == NULL) return; // invalid! lhnetsocket->next->prev = lhnetsocket->prev; lhnetsocket->prev->next = lhnetsocket->next; lhnetsocket->next = NULL; lhnetsocket->prev = NULL; // no special close code for loopback, just inet if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_INET4 || lhnetsocket->address.addresstype == LHNETADDRESSTYPE_INET6) { closesocket(lhnetsocket->inetsocket); } Z_Free(lhnetsocket); } } lhnetaddress_t *LHNET_AddressFromSocket(lhnetsocket_t *sock) { if (sock) return &sock->address; else return NULL; } int LHNET_Read(lhnetsocket_t *lhnetsocket, void *content, int maxcontentlength, lhnetaddress_t *vaddress) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; int value = 0; if (!lhnetsocket || !address || !content || maxcontentlength < 1) return -1; if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_LOOP) { time_t currenttime; lhnetpacket_t *p, *pnext; // scan for any old packets to timeout while searching for a packet // that is waiting to be delivered to this socket currenttime = time(NULL); for (p = lhnet_packetlist.next;p != &lhnet_packetlist;p = pnext) { pnext = p->next; if (p->timeout < currenttime) { // unlink and free p->next->prev = p->prev; p->prev->next = p->next; Z_Free(p); continue; } #ifndef STANDALONETEST if (cl_netlocalping.value && (realtime - cl_netlocalping.value * (1.0 / 2000.0)) < p->sentdoubletime) continue; #endif if (value == 0 && p->destinationport == lhnetsocket->address.port) { if (p->length <= maxcontentlength) { lhnetaddressnative_t *localaddress = (lhnetaddressnative_t *)&lhnetsocket->address; *address = *localaddress; address->port = p->sourceport; memcpy(content, p->data, p->length); value = p->length; } else value = -1; // unlink and free p->next->prev = p->prev; p->prev->next = p->next; Z_Free(p); } } } else if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_INET4) { SOCKLEN_T inetaddresslength; address->addresstype = LHNETADDRESSTYPE_NONE; inetaddresslength = sizeof(address->addr.in); value = recvfrom(lhnetsocket->inetsocket, (char *)content, maxcontentlength, LHNET_RECVFROM_FLAGS, &address->addr.sock, &inetaddresslength); if (value > 0) { address->addresstype = LHNETADDRESSTYPE_INET4; address->port = ntohs(address->addr.in.sin_port); return value; } else if (value < 0) { int e = SOCKETERRNO; if (e == EWOULDBLOCK) return 0; switch (e) { case ECONNREFUSED: Con_Print("Connection refused\n"); return 0; } Con_Printf("LHNET_Read: recvfrom returned error: %s\n", LHNETPRIVATE_StrError()); } } #ifdef SUPPORTIPV6 else if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_INET6) { SOCKLEN_T inetaddresslength; address->addresstype = LHNETADDRESSTYPE_NONE; inetaddresslength = sizeof(address->addr.in6); value = recvfrom(lhnetsocket->inetsocket, (char *)content, maxcontentlength, 0, &address->addr.sock, &inetaddresslength); if (value > 0) { address->addresstype = LHNETADDRESSTYPE_INET6; address->port = ntohs(address->addr.in6.sin6_port); return value; } else if (value == -1) { int e = SOCKETERRNO; if (e == EWOULDBLOCK) return 0; switch (e) { case ECONNREFUSED: Con_Print("Connection refused\n"); return 0; } Con_Printf("LHNET_Read: recvfrom returned error: %s\n", LHNETPRIVATE_StrError()); } } #endif return value; } int LHNET_Write(lhnetsocket_t *lhnetsocket, const void *content, int contentlength, const lhnetaddress_t *vaddress) { lhnetaddressnative_t *address = (lhnetaddressnative_t *)vaddress; int value = -1; if (!lhnetsocket || !address || !content || contentlength < 1) return -1; if (lhnetsocket->address.addresstype != address->addresstype) return -1; if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_LOOP) { lhnetpacket_t *p; p = (lhnetpacket_t *)Z_Malloc(sizeof(*p) + contentlength); p->data = (void *)(p + 1); memcpy(p->data, content, contentlength); p->length = contentlength; p->sourceport = lhnetsocket->address.port; p->destinationport = address->port; p->timeout = time(NULL) + 10; p->next = &lhnet_packetlist; p->prev = p->next->prev; p->next->prev = p; p->prev->next = p; #ifndef STANDALONETEST p->sentdoubletime = realtime; #endif value = contentlength; } else if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_INET4) { value = sendto(lhnetsocket->inetsocket, (char *)content, contentlength, LHNET_SENDTO_FLAGS, (struct sockaddr *)&address->addr.in, sizeof(struct sockaddr_in)); if (value == -1) { if (SOCKETERRNO == EWOULDBLOCK) return 0; Con_Printf("LHNET_Write: sendto returned error: %s\n", LHNETPRIVATE_StrError()); } } #ifdef SUPPORTIPV6 else if (lhnetsocket->address.addresstype == LHNETADDRESSTYPE_INET6) { value = sendto(lhnetsocket->inetsocket, (char *)content, contentlength, 0, (struct sockaddr *)&address->addr.in6, sizeof(struct sockaddr_in6)); if (value == -1) { if (SOCKETERRNO == EWOULDBLOCK) return 0; Con_Printf("LHNET_Write: sendto returned error: %s\n", LHNETPRIVATE_StrError()); } } #endif return value; } #ifdef STANDALONETEST int main(int argc, char **argv) { #if 1 char *buffer = "test", buffer2[1024]; int blen = strlen(buffer); int b2len = 1024; lhnetsocket_t *sock1; lhnetsocket_t *sock2; lhnetaddress_t myaddy1; lhnetaddress_t myaddy2; lhnetaddress_t myaddy3; lhnetaddress_t localhostaddy1; lhnetaddress_t localhostaddy2; int test1; int test2; printf("calling LHNET_Init\n"); LHNET_Init(); printf("calling LHNET_FromPort twice to create two local addresses\n"); LHNETADDRESS_FromPort(&myaddy1, LHNETADDRESSTYPE_INET4, 4000); LHNETADDRESS_FromPort(&myaddy2, LHNETADDRESSTYPE_INET4, 4001); LHNETADDRESS_FromString(&localhostaddy1, "127.0.0.1", 4000); LHNETADDRESS_FromString(&localhostaddy2, "127.0.0.1", 4001); printf("calling LHNET_OpenSocket_Connectionless twice to create two local sockets\n"); sock1 = LHNET_OpenSocket_Connectionless(&myaddy1); sock2 = LHNET_OpenSocket_Connectionless(&myaddy2); printf("calling LHNET_Write to send a packet from the first socket to the second socket\n"); test1 = LHNET_Write(sock1, buffer, blen, &localhostaddy2); printf("sleeping briefly\n"); #ifdef WIN32 Sleep (100); #else usleep (100000); #endif printf("calling LHNET_Read on the second socket to read the packet sent from the first socket\n"); test2 = LHNET_Read(sock2, buffer2, b2len - 1, &myaddy3); if (test2 > 0) Con_Printf("socket to socket test succeeded\n"); else Con_Printf("socket to socket test failed\n"); #ifdef WIN32 printf("press any key to exit\n"); getchar(); #endif printf("calling LHNET_Shutdown\n"); LHNET_Shutdown(); printf("exiting\n"); return 0; #else lhnetsocket_t *sock[16], *sendsock; int i; int numsockets; int count; int length; int port; time_t oldtime; time_t newtime; char *sendmessage; int sendmessagelength; lhnetaddress_t destaddress; lhnetaddress_t receiveaddress; lhnetaddress_t sockaddress[16]; char buffer[1536], addressstring[128], addressstring2[128]; if ((argc == 2 || argc == 5) && (port = atoi(argv[1])) >= 1 && port < 65535) { printf("calling LHNET_Init()\n"); LHNET_Init(); numsockets = 0; LHNETADDRESS_FromPort(&sockaddress[numsockets++], LHNETADDRESSTYPE_LOOP, port); LHNETADDRESS_FromPort(&sockaddress[numsockets++], LHNETADDRESSTYPE_INET4, port); LHNETADDRESS_FromPort(&sockaddress[numsockets++], LHNETADDRESSTYPE_INET6, port+1); sendsock = NULL; sendmessage = NULL; sendmessagelength = 0; for (i = 0;i < numsockets;i++) { LHNETADDRESS_ToString(&sockaddress[i], addressstring, sizeof(addressstring), 1); printf("calling LHNET_OpenSocket_Connectionless(<%s>)\n", addressstring); if ((sock[i] = LHNET_OpenSocket_Connectionless(&sockaddress[i]))) { LHNETADDRESS_ToString(LHNET_AddressFromSocket(sock[i]), addressstring2, sizeof(addressstring2), 1); printf("opened socket successfully (address \"%s\")\n", addressstring2); } else { printf("failed to open socket\n"); if (i == 0) { LHNET_Shutdown(); return -1; } } } count = 0; if (argc == 5) { count = atoi(argv[2]); if (LHNETADDRESS_FromString(&destaddress, argv[3], -1)) { sendmessage = argv[4]; sendmessagelength = strlen(sendmessage); sendsock = NULL; for (i = 0;i < numsockets;i++) if (sock[i] && LHNETADDRESS_GetAddressType(&destaddress) == LHNETADDRESS_GetAddressType(&sockaddress[i])) sendsock = sock[i]; if (sendsock == NULL) { printf("Could not find an open socket matching the addresstype (%i) of destination address, switching to listen only mode\n", LHNETADDRESS_GetAddressType(&destaddress)); argc = 2; } } else { printf("LHNETADDRESS_FromString did not like the address \"%s\", switching to listen only mode\n", argv[3]); argc = 2; } } printf("started, now listening for \"exit\" on the opened sockets\n"); oldtime = time(NULL); for(;;) { #ifdef WIN32 Sleep(1); #else usleep(1); #endif for (i = 0;i < numsockets;i++) { if (sock[i]) { length = LHNET_Read(sock[i], buffer, sizeof(buffer), &receiveaddress); if (length < 0) printf("localsock read error: length < 0"); else if (length > 0 && length < (int)sizeof(buffer)) { buffer[length] = 0; LHNETADDRESS_ToString(&receiveaddress, addressstring, sizeof(addressstring), 1); LHNETADDRESS_ToString(LHNET_AddressFromSocket(sock[i]), addressstring2, sizeof(addressstring2), 1); printf("received message \"%s\" from \"%s\" on socket \"%s\"\n", buffer, addressstring, addressstring2); if (!strcmp(buffer, "exit")) break; } } } if (i < numsockets) break; if (argc == 5 && count > 0) { newtime = time(NULL); if (newtime != oldtime) { LHNETADDRESS_ToString(&destaddress, addressstring, sizeof(addressstring), 1); LHNETADDRESS_ToString(LHNET_AddressFromSocket(sendsock), addressstring2, sizeof(addressstring2), 1); printf("calling LHNET_Write(<%s>, \"%s\", %i, <%s>)\n", addressstring2, sendmessage, sendmessagelength, addressstring); length = LHNET_Write(sendsock, sendmessage, sendmessagelength, &destaddress); if (length == sendmessagelength) printf("sent successfully\n"); else printf("LH_Write failed, returned %i (length of message was %i)\n", length, strlen(argv[4])); oldtime = newtime; count--; if (count <= 0) printf("Done sending, still listening for \"exit\"\n"); } } } for (i = 0;i < numsockets;i++) { if (sock[i]) { LHNETADDRESS_ToString(LHNET_AddressFromSocket(sock[i]), addressstring2, sizeof(addressstring2), 1); printf("calling LHNET_CloseSocket(<%s>)\n", addressstring2); LHNET_CloseSocket(sock[i]); } } printf("calling LHNET_Shutdown()\n"); LHNET_Shutdown(); return 0; } printf("Testing code for lhnet.c\nusage: lhnettest [ ]\n"); return -1; #endif } #endif