Mercurial > hg > openttd

changeset 9844:6611f2b7b31f draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
(svn r13988) -Codechange: move the to IP resolving functions to a separate file.
author rubidium <rubidium@openttd.org>
date Mon, 04 Aug 2008 12:56:38 +0000
parents afd919aa07c3
children ee10319a40b6
files projects/openttd_vs80.vcproj projects/openttd_vs90.vcproj source.list src/network/core/host.cpp src/network/core/host.h src/network/network.cpp src/network/network_internal.h src/network/network_udp.cpp
diffstat 8 files changed, 280 insertions(+), 223 deletions(-) [+]
line wrap: on
line diff
--- a/projects/openttd_vs80.vcproj
+++ b/projects/openttd_vs80.vcproj
@@ -2452,6 +2452,14 @@
 				>
 			</File>
 			<File
+				RelativePath=".\..\src\network\core\host.cpp"
+				>
+			</File>
+			<File
+				RelativePath=".\..\src\network\core\host.h"
+				>
+			</File>
+			<File
 				RelativePath=".\..\src\network\core\os_abstraction.h"
 				>
 			</File>
--- a/projects/openttd_vs90.vcproj
+++ b/projects/openttd_vs90.vcproj
@@ -2449,6 +2449,14 @@
 				>
 			</File>
 			<File
+				RelativePath=".\..\src\network\core\host.cpp"
+				>
+			</File>
+			<File
+				RelativePath=".\..\src\network\core\host.h"
+				>
+			</File>
+			<File
 				RelativePath=".\..\src\network\core\os_abstraction.h"
 				>
 			</File>
--- a/source.list
+++ b/source.list
@@ -566,6 +566,8 @@
 network/core/core.cpp
 network/core/core.h
 network/core/game.h
+network/core/host.cpp
+network/core/host.h
 network/core/os_abstraction.h
 network/core/packet.cpp
 network/core/packet.h
new file mode 100644
--- /dev/null
+++ b/src/network/core/host.cpp
@@ -0,0 +1,247 @@
+/* $Id$ */
+
+/** @file host.cpp Functions related to getting host specific data (IPs). */
+
+#ifdef ENABLE_NETWORK
+
+#include "../../stdafx.h"
+#include "../../debug.h"
+#include "os_abstraction.h"
+
+/**
+ * Internal implementation for finding the broadcast IPs.
+ * This function is implemented multiple times for multiple targets.
+ * @param broadcast the list of broadcasts to write into.
+ * @param limit     the maximum number of items to add.
+ */
+static int NetworkFindBroadcastIPsInternal(uint32 *broadcast, int limit);
+
+#if defined(PSP)
+static int NetworkFindBroadcastIPsInternal(uint32 *broadcast, int limit) // PSP implementation
+{
+	return 0;
+}
+
+#elif defined(BEOS_NET_SERVER) /* doesn't have neither getifaddrs or net/if.h */
+/* Based on Andrew Bachmann's netstat+.c. Big thanks to him! */
+int _netstat(int fd, char **output, int verbose);
+
+int seek_past_header(char **pos, const char *header)
+{
+	char *new_pos = strstr(*pos, header);
+	if (new_pos == 0) {
+		return B_ERROR;
+	}
+	*pos += strlen(header) + new_pos - *pos + 1;
+	return B_OK;
+}
+
+static int NetworkFindBroadcastIPsInternal(uint32 *broadcast, int limit) // BEOS implementation
+{
+	int sock = socket(AF_INET, SOCK_DGRAM, 0);
+
+	if (sock < 0) {
+		DEBUG(net, 0, "[core] error creating socket");
+		return;
+	}
+
+	char *output_pointer = NULL;
+	int output_length = _netstat(sock, &output_pointer, 1);
+	if (output_length < 0) {
+		DEBUG(net, 0, "[core] error running _netstat");
+		return;
+	}
+
+	int index;
+	char **output = &output_pointer;
+	if (seek_past_header(output, "IP Interfaces:") == B_OK) {
+		while (index != limit) {
+
+			uint32 n, fields, read;
+			uint8 i1, i2, i3, i4, j1, j2, j3, j4;
+			struct in_addr inaddr;
+			uint32 ip;
+			uint32 netmask;
+
+			fields = sscanf(*output, "%u: %hhu.%hhu.%hhu.%hhu, netmask %hhu.%hhu.%hhu.%hhu%n",
+												&n, &i1, &i2, &i3, &i4, &j1, &j2, &j3, &j4, &read);
+			read += 1;
+			if (fields != 9) {
+				break;
+			}
+
+			ip      = (uint32)i1 << 24 | (uint32)i2 << 16 | (uint32)i3 << 8 | (uint32)i4;
+			netmask = (uint32)j1 << 24 | (uint32)j2 << 16 | (uint32)j3 << 8 | (uint32)j4;
+
+			if (ip != INADDR_LOOPBACK && ip != INADDR_ANY) {
+				inaddr.s_addr = htonl(ip | ~netmask);
+				broadcast[index] = inaddr.s_addr;
+				index++;
+			}
+			if (read < 0) {
+				break;
+			}
+			*output += read;
+		}
+		closesocket(sock);
+	}
+
+	return index;
+}
+
+#elif defined(HAVE_GETIFADDRS)
+static int NetworkFindBroadcastIPsInternal(uint32 *broadcast, int limit) // GETIFADDRS implementation
+{
+	struct ifaddrs *ifap, *ifa;
+
+	if (getifaddrs(&ifap) != 0) return 0;
+
+	int index = 0;
+	for (ifa = ifap; ifa != NULL && index != limit; ifa = ifa->ifa_next) {
+		if (!(ifa->ifa_flags & IFF_BROADCAST)) continue;
+		if (ifa->ifa_broadaddr == NULL) continue;
+		if (ifa->ifa_broadaddr->sa_family != AF_INET) continue;
+
+		broadcast[index] = ((struct sockaddr_in*)ifa->ifa_broadaddr)->sin_addr.s_addr;
+		index++;
+	}
+	freeifaddrs(ifap);
+
+	return index;
+}
+
+#elif defined(WIN32)
+static int NetworkFindBroadcastIPsInternal(uint32 *broadcast, int limit) // Win32 implementation
+{
+	SOCKET sock = socket(AF_INET, SOCK_DGRAM, 0);
+	if (sock == INVALID_SOCKET) return 0;
+
+	DWORD len = 0;
+	INTERFACE_INFO ifo[MAX_INTERFACES];
+	memset(&ifo[0], 0, sizeof(ifo));
+	if ((WSAIoctl(sock, SIO_GET_INTERFACE_LIST, NULL, 0, &ifo[0], sizeof(ifo), &len, NULL, NULL)) != 0) {
+		closesocket(sock);
+		return;
+	}
+
+	int index = 0;
+	for (int j = 0; j < len / sizeof(*ifo) && index != limit; j++) {
+		if (ifo[j].iiFlags & IFF_LOOPBACK) continue;
+		if (!(ifo[j].iiFlags & IFF_BROADCAST)) continue;
+
+		/* iiBroadcast is unusable, because it always seems to be set to 255.255.255.255. */
+		broadcast[index++] = ifo[j].iiAddress.AddressIn.sin_addr.s_addr | ~ifo[j].iiNetmask.AddressIn.sin_addr.s_addr;
+	}
+
+	closesocket(sock);
+	return index;
+}
+
+#else /* not HAVE_GETIFADDRS */
+static int NetworkFindBroadcastIPsInternal(uint32 *broadcast, int limit) // !GETIFADDRS implementation
+{
+	SOCKET sock = socket(AF_INET, SOCK_DGRAM, 0);
+	if (sock == INVALID_SOCKET) return 0;
+
+	char buf[4 * 1024]; // Arbitrary buffer size
+	struct ifconf ifconf;
+
+	ifconf.ifc_len = sizeof(buf);
+	ifconf.ifc_buf = buf;
+	if (ioctl(sock, SIOCGIFCONF, &ifconf) == -1) {
+		closesocket(sock);
+		return 0;
+	}
+
+	const char *buf_end = buf + ifconf.ifc_len;
+	int index;
+	for (const char *p = buf; p < buf_end && index != 0;) {
+		const struct ifreq* req = (const struct ifreq*)p;
+
+		if (req->ifr_addr.sa_family == AF_INET) {
+			struct ifreq r;
+
+			strncpy(r.ifr_name, req->ifr_name, lengthof(r.ifr_name));
+			if (ioctl(sock, SIOCGIFFLAGS, &r) != -1 &&
+					r.ifr_flags & IFF_BROADCAST &&
+					ioctl(sock, SIOCGIFBRDADDR, &r) != -1) {
+				broadcast[index++] = ((struct sockaddr_in*)&r.ifr_broadaddr)->sin_addr.s_addr;
+			}
+		}
+
+		p += sizeof(struct ifreq);
+#if defined(AF_LINK) && !defined(SUNOS)
+		p += req->ifr_addr.sa_len - sizeof(struct sockaddr);
+#endif
+	}
+
+	closesocket(sock);
+}
+#endif /* all NetworkFindBroadcastIPsInternals */
+
+/**
+ * Find the IPs to broadcast.
+ * @param broadcast the list of broadcasts to write into.
+ * @param limit     the maximum number of items to add.
+ */
+void NetworkFindBroadcastIPs(uint32 *broadcast, int limit)
+{
+	int count = NetworkFindBroadcastIPsInternal(broadcast, limit);
+
+	/* Make sure the list is terminated. */
+	broadcast[count] = 0;
+
+	/* Now display to the debug all the detected ips */
+	DEBUG(net, 3, "Detected broadcast addresses:");
+	for (int i = 0; broadcast[i] != 0; i++) {
+		DEBUG(net, 3, "%d) %s", i, inet_ntoa(*(struct in_addr *)&broadcast[i])); //inet_ntoa(inaddr));
+	}
+}
+
+
+/**
+ * Resolve a hostname to an ip.
+ * @param hsotname the hostname to resolve
+ * @return the IP belonging to that hostname, or 0 on failure.
+ */
+uint32 NetworkResolveHost(const char *hostname)
+{
+	/* Is this an IP address? */
+	in_addr_t ip = inet_addr(hostname);
+
+	if (ip != INADDR_NONE) return ip;
+
+	/* No, try to resolve the name */
+	struct in_addr addr;
+#if !defined(PSP)
+	struct hostent *he = gethostbyname(hostname);
+	if (he == NULL) {
+		DEBUG(net, 0, "[NET] Cannot resolve %s", hostname);
+		return 0;
+	}
+	addr = *(struct in_addr *)he->h_addr_list[0];
+#else
+	int rid = -1;
+	char buf[1024];
+
+	/* Create a resolver */
+	if (sceNetResolverCreate(&rid, buf, sizeof(buf)) < 0) {
+		DEBUG(net, 0, "[NET] Error connecting resolver");
+		return 0;
+	}
+
+	/* Try to resolve the name */
+	if (sceNetResolverStartNtoA(rid, hostname, &addr, 2, 3) < 0) {
+		DEBUG(net, 0, "[NET] Cannot resolve %s", hostname);
+		sceNetResolverDelete(rid);
+		return 0;
+	}
+	sceNetResolverDelete(rid);
+#endif /* PSP */
+
+	DEBUG(net, 1, "[NET] Resolved %s to %s", hostname, inet_ntoa(addr));
+	ip = addr.s_addr;
+	return ip;
+}
+
+#endif /* ENABLE_NETWORK */
new file mode 100644
--- /dev/null
+++ b/src/network/core/host.h
@@ -0,0 +1,12 @@
+/* $Id$ */
+
+/**
+ * @file host.h Resolving of hostnames/IPs
+ */
+
+#ifndef NETWORK_CORE_HOST_H
+
+void NetworkFindBroadcastIPs(uint32 *broadcast, int limit);
+uint32 NetworkResolveHost(const char *hostname);
+
+#endif /* NETWORK_CORE_HOST_H */
--- a/src/network/network.cpp
+++ b/src/network/network.cpp
@@ -23,6 +23,7 @@
 #include "core/udp.h"
 #include "core/tcp.h"
 #include "core/core.h"
+#include "core/host.h"
 #include "network_gui.h"
 #include "../console_func.h"
 #include <stdarg.h> /* va_list */
@@ -370,227 +371,6 @@
 	}
 }
 
-// Find all IP-aliases for this host
-static void NetworkFindIPs()
-{
-#if !defined(PSP)
-	int i;
-
-#if defined(BEOS_NET_SERVER) /* doesn't have neither getifaddrs or net/if.h */
-	/* Based on Andrew Bachmann's netstat+.c. Big thanks to him! */
-	int _netstat(int fd, char **output, int verbose);
-
-	int seek_past_header(char **pos, const char *header) {
-		char *new_pos = strstr(*pos, header);
-		if (new_pos == 0) {
-			return B_ERROR;
-		}
-		*pos += strlen(header) + new_pos - *pos + 1;
-		return B_OK;
-	}
-
-	int output_length;
-	char *output_pointer = NULL;
-	char **output;
-	int sock = socket(AF_INET, SOCK_DGRAM, 0);
-	i = 0;
-
-	// If something fails, make sure the list is empty
-	_broadcast_list[0] = 0;
-
-	if (sock < 0) {
-		DEBUG(net, 0, "[core] error creating socket");
-		return;
-	}
-
-	output_length = _netstat(sock, &output_pointer, 1);
-	if (output_length < 0) {
-		DEBUG(net, 0, "[core] error running _netstat");
-		return;
-	}
-
-	output = &output_pointer;
-	if (seek_past_header(output, "IP Interfaces:") == B_OK) {
-		for (;;) {
-			uint32 n, fields, read;
-			uint8 i1, i2, i3, i4, j1, j2, j3, j4;
-			struct in_addr inaddr;
-			uint32 ip;
-			uint32 netmask;
-
-			fields = sscanf(*output, "%u: %hhu.%hhu.%hhu.%hhu, netmask %hhu.%hhu.%hhu.%hhu%n",
-												&n, &i1, &i2, &i3, &i4, &j1, &j2, &j3, &j4, &read);
-			read += 1;
-			if (fields != 9) {
-				break;
-			}
-
-			ip      = (uint32)i1 << 24 | (uint32)i2 << 16 | (uint32)i3 << 8 | (uint32)i4;
-			netmask = (uint32)j1 << 24 | (uint32)j2 << 16 | (uint32)j3 << 8 | (uint32)j4;
-
-			if (ip != INADDR_LOOPBACK && ip != INADDR_ANY) {
-				inaddr.s_addr = htonl(ip | ~netmask);
-				_broadcast_list[i] = inaddr.s_addr;
-				i++;
-				if (i == MAX_INTERFACES) break;
-			}
-			if (read < 0) {
-				break;
-			}
-			*output += read;
-		}
-		/* XXX - Using either one of these crashes openttd heavily? - wber */
-		/*free(output_pointer);*/
-		/*free(output);*/
-		closesocket(sock);
-	}
-#elif defined(HAVE_GETIFADDRS)
-	struct ifaddrs *ifap, *ifa;
-
-	// If something fails, make sure the list is empty
-	_broadcast_list[0] = 0;
-
-	if (getifaddrs(&ifap) != 0)
-		return;
-
-	i = 0;
-	for (ifa = ifap; ifa != NULL; ifa = ifa->ifa_next) {
-		if (!(ifa->ifa_flags & IFF_BROADCAST)) continue;
-		if (ifa->ifa_broadaddr == NULL) continue;
-		if (ifa->ifa_broadaddr->sa_family != AF_INET) continue;
-		_broadcast_list[i] = ((struct sockaddr_in*)ifa->ifa_broadaddr)->sin_addr.s_addr;
-		i++;
-		if (i == MAX_INTERFACES) break;
-	}
-	freeifaddrs(ifap);
-
-#else /* not HAVE_GETIFADDRS */
-	SOCKET sock;
-#ifdef WIN32
-	DWORD len = 0;
-	INTERFACE_INFO ifo[MAX_INTERFACES];
-	uint j;
-#else
-	char buf[4 * 1024]; // Arbitrary buffer size
-	struct ifconf ifconf;
-	const char* buf_end;
-	const char* p;
-#endif
-
-	// If something fails, make sure the list is empty
-	_broadcast_list[0] = 0;
-
-	sock = socket(AF_INET, SOCK_DGRAM, 0);
-	if (sock == INVALID_SOCKET) return;
-
-#ifdef WIN32
-	memset(&ifo[0], 0, sizeof(ifo));
-	if ((WSAIoctl(sock, SIO_GET_INTERFACE_LIST, NULL, 0, &ifo[0], sizeof(ifo), &len, NULL, NULL)) != 0) {
-		closesocket(sock);
-		return;
-	}
-
-	i = 0;
-	for (j = 0; j < len / sizeof(*ifo); j++) {
-		if (ifo[j].iiFlags & IFF_LOOPBACK) continue;
-		if (!(ifo[j].iiFlags & IFF_BROADCAST)) continue;
-		/* iiBroadcast is unusable, because it always seems to be set to
-		 * 255.255.255.255.
-		 */
-		_broadcast_list[i++] =
-			 ifo[j].iiAddress.AddressIn.sin_addr.s_addr |
-			~ifo[j].iiNetmask.AddressIn.sin_addr.s_addr;
-		if (i == MAX_INTERFACES) break;
-	}
-#else
-	ifconf.ifc_len = sizeof(buf);
-	ifconf.ifc_buf = buf;
-	if (ioctl(sock, SIOCGIFCONF, &ifconf) == -1) {
-		closesocket(sock);
-		return;
-	}
-
-	i = 0;
-	buf_end = buf + ifconf.ifc_len;
-	for (p = buf; p < buf_end;) {
-		const struct ifreq* req = (const struct ifreq*)p;
-
-		if (req->ifr_addr.sa_family == AF_INET) {
-			struct ifreq r;
-
-			strncpy(r.ifr_name, req->ifr_name, lengthof(r.ifr_name));
-			if (ioctl(sock, SIOCGIFFLAGS, &r) != -1 &&
-					r.ifr_flags & IFF_BROADCAST &&
-					ioctl(sock, SIOCGIFBRDADDR, &r) != -1) {
-				_broadcast_list[i++] =
-					((struct sockaddr_in*)&r.ifr_broadaddr)->sin_addr.s_addr;
-				if (i == MAX_INTERFACES) break;
-			}
-		}
-
-		p += sizeof(struct ifreq);
-#if defined(AF_LINK) && !defined(SUNOS)
-		p += req->ifr_addr.sa_len - sizeof(struct sockaddr);
-#endif
-	}
-#endif
-
-	closesocket(sock);
-#endif /* not HAVE_GETIFADDRS */
-
-	_broadcast_list[i] = 0;
-
-	DEBUG(net, 3, "Detected broadcast addresses:");
-	// Now display to the debug all the detected ips
-	for (i = 0; _broadcast_list[i] != 0; i++) {
-		DEBUG(net, 3, "%d) %s", i, inet_ntoa(*(struct in_addr *)&_broadcast_list[i]));//inet_ntoa(inaddr));
-	}
-#endif /* PSP */
-}
-
-// Resolve a hostname to a inet_addr
-unsigned long NetworkResolveHost(const char *hostname)
-{
-	in_addr_t ip;
-
-	// First try: is it an ip address?
-	ip = inet_addr(hostname);
-
-	// If not try to resolve the name
-	if (ip == INADDR_NONE) {
-		struct in_addr addr;
-#if !defined(PSP)
-		struct hostent *he = gethostbyname(hostname);
-		if (he == NULL) {
-			DEBUG(net, 0, "Cannot resolve '%s'", hostname);
-			return ip;
-		}
-		addr = *(struct in_addr *)he->h_addr_list[0];
-#else
-		int rid = -1;
-		char buf[1024];
-
-		/* Create a resolver */
-		if (sceNetResolverCreate(&rid, buf, sizeof(buf)) < 0) {
-			DEBUG(net, 0, "[NET] Error connecting resolver");
-			return ip;
-		}
-
-		/* Try to resolve the name */
-		if (sceNetResolverStartNtoA(rid, hostname, &addr, 2, 3) < 0) {
-			DEBUG(net, 0, "[NET] Cannot resolve %s", hostname);
-			sceNetResolverDelete(rid);
-			return ip;
-		}
-		sceNetResolverDelete(rid);
-#endif /* PSP */
-
-		DEBUG(net, 1, "[NET] Resolved %s to %s", hostname, inet_ntoa(addr));
-		ip = addr.s_addr;
-	}
-	return ip;
-}
-
 /** Converts a string to ip/port/player
  *  Format: IP#player:port
  *
@@ -1431,7 +1211,7 @@
 
 	NetworkInitialize();
 	DEBUG(net, 3, "[core] network online, multiplayer available");
-	NetworkFindIPs();
+	NetworkFindBroadcastIPs(_broadcast_list, MAX_INTERFACES);
 }
 
 /** This shuts the network down */
--- a/src/network/network_internal.h
+++ b/src/network/network_internal.h
@@ -147,7 +147,6 @@
 uint NetworkCalculateLag(const NetworkTCPSocketHandler *cs);
 byte NetworkGetCurrentLanguageIndex();
 NetworkTCPSocketHandler *NetworkFindClientStateFromIndex(uint16 client_index);
-unsigned long NetworkResolveHost(const char *hostname);
 char* GetNetworkErrorMsg(char* buf, NetworkErrorCode err, const char* last);
 bool NetworkFindName(char new_name[NETWORK_CLIENT_NAME_LENGTH]);
 
--- a/src/network/network_udp.cpp
+++ b/src/network/network_udp.cpp
@@ -16,6 +16,7 @@
 #include "network_gamelist.h"
 #include "network_udp.h"
 #include "network_internal.h"
+#include "core/host.h"
 #include "../variables.h"
 #include "../newgrf_config.h"
 #include "../core/endian_func.hpp"