X-Git-Url: https://git.lizzy.rs/?a=blobdiff_plain;f=src%2Fsocket.cpp;h=df9c57c5f9223a06d511e83e2b2e1551ebd706dd;hb=dfe00abc5addd66dbf840d7e826690291f4afd93;hp=f4b8f442943aaf78834569547ab376763a65aef0;hpb=385dd9917fd4ced23a704f0566779d58f5a6b727;p=dragonfireclient.git diff --git a/src/socket.cpp b/src/socket.cpp index f4b8f4429..df9c57c5f 100644 --- a/src/socket.cpp +++ b/src/socket.cpp @@ -1,42 +1,79 @@ /* -Minetest-c55 -Copyright (C) 2010 celeron55, Perttu Ahola +Minetest +Copyright (C) 2013 celeron55, Perttu Ahola This program is free software; you can redistribute it and/or modify -it under the terms of the GNU General Public License as published by -the Free Software Foundation; either version 2 of the License, or +it under the terms of the GNU Lesser General Public License as published by +the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -GNU General Public License for more details. +GNU Lesser General Public License for more details. -You should have received a copy of the GNU General Public License along +You should have received a copy of the GNU Lesser General Public License along with this program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */ #include "socket.h" -#include "debug.h" + #include #include #include +#include #include +#include +#include +#include "util/string.h" +#include "util/numeric.h" +#include "constants.h" +#include "debug.h" +#include "settings.h" +#include "log.h" +#include "main.h" // for g_settings -// Debug printing options -#define DP 0 -#define DPS "" +#ifdef _WIN32 + #ifndef WIN32_LEAN_AND_MEAN + #define WIN32_LEAN_AND_MEAN + #endif + // Without this some of the network functions are not found on mingw + #ifndef _WIN32_WINNT + #define _WIN32_WINNT 0x0501 + #endif + #include + #include + #include + #ifdef _MSC_VER + #pragma comment(lib, "ws2_32.lib") + #endif +typedef SOCKET socket_t; +typedef int socklen_t; +#else + #include + #include + #include + #include + #include + #include + #include +typedef int socket_t; +#endif + +// Set to true to enable verbose debug output +bool socket_enable_debug_output = false; // yuck -bool g_sockets_initialized = false; +static bool g_sockets_initialized = false; +// Initialize sockets void sockets_init() { #ifdef _WIN32 + // Windows needs sockets to be initialized before use WSADATA WsaData; if(WSAStartup( MAKEWORD(2,2), &WsaData ) != NO_ERROR) throw SocketException("WSAStartup failed"); -#else #endif g_sockets_initialized = true; } @@ -44,124 +81,284 @@ void sockets_init() void sockets_cleanup() { #ifdef _WIN32 + // On Windows, cleanup sockets after use WSACleanup(); #endif } +/* + Address +*/ + Address::Address() { + m_addr_family = 0; + memset(&m_address, 0, sizeof(m_address)); + m_port = 0; } -Address::Address(unsigned int address, unsigned short port) +Address::Address(u32 address, u16 port) { - m_address = address; - m_port = port; + memset(&m_address, 0, sizeof(m_address)); + setAddress(address); + setPort(port); } -Address::Address(unsigned int a, unsigned int b, - unsigned int c, unsigned int d, - unsigned short port) +Address::Address(u8 a, u8 b, u8 c, u8 d, u16 port) { - m_address = (a<<24) | (b<<16) | ( c<<8) | d; - m_port = port; + memset(&m_address, 0, sizeof(m_address)); + setAddress(a, b, c, d); + setPort(port); } -bool Address::operator==(Address &address) +Address::Address(const IPv6AddressBytes *ipv6_bytes, u16 port) { - return (m_address == address.m_address - && m_port == address.m_port); + memset(&m_address, 0, sizeof(m_address)); + setAddress(ipv6_bytes); + setPort(port); } -bool Address::operator!=(Address &address) +// Equality (address family, address and port must be equal) +bool Address::operator==(const Address &address) +{ + if(address.m_addr_family != m_addr_family || address.m_port != m_port) + return false; + else if(m_addr_family == AF_INET) + { + return m_address.ipv4.sin_addr.s_addr == + address.m_address.ipv4.sin_addr.s_addr; + } + else if(m_addr_family == AF_INET6) + { + return memcmp(m_address.ipv6.sin6_addr.s6_addr, + address.m_address.ipv6.sin6_addr.s6_addr, 16) == 0; + } + else + return false; +} + +bool Address::operator!=(const Address &address) { return !(*this == address); } void Address::Resolve(const char *name) { - struct addrinfo *resolved; - int e = getaddrinfo(name, NULL, NULL, &resolved); + if (!name || name[0] == 0) { + if (m_addr_family == AF_INET) { + setAddress((u32) 0); + } else if (m_addr_family == AF_INET6) { + setAddress((IPv6AddressBytes*) 0); + } + return; + } + + struct addrinfo *resolved, hints; + memset(&hints, 0, sizeof(hints)); + + // Setup hints + hints.ai_socktype = 0; + hints.ai_protocol = 0; + hints.ai_flags = 0; + if(g_settings->getBool("enable_ipv6")) + { + // AF_UNSPEC allows both IPv6 and IPv4 addresses to be returned + hints.ai_family = AF_UNSPEC; + } + else + { + hints.ai_family = AF_INET; + } + + // Do getaddrinfo() + int e = getaddrinfo(name, NULL, &hints, &resolved); if(e != 0) + throw ResolveError(gai_strerror(e)); + + // Copy data + if(resolved->ai_family == AF_INET) + { + struct sockaddr_in *t = (struct sockaddr_in *) resolved->ai_addr; + m_addr_family = AF_INET; + m_address.ipv4 = *t; + } + else if(resolved->ai_family == AF_INET6) + { + struct sockaddr_in6 *t = (struct sockaddr_in6 *) resolved->ai_addr; + m_addr_family = AF_INET6; + m_address.ipv6 = *t; + } + else + { + freeaddrinfo(resolved); throw ResolveError(""); - /* - FIXME: This is an ugly hack; change the whole class - to store the address as sockaddr - */ - struct sockaddr_in *t = (struct sockaddr_in*)resolved->ai_addr; - m_address = ntohl(t->sin_addr.s_addr); + } freeaddrinfo(resolved); } -unsigned int Address::getAddress() const +// IP address -> textual representation +std::string Address::serializeString() const +{ +// windows XP doesnt have inet_ntop, maybe use better func +#ifdef _WIN32 + if(m_addr_family == AF_INET) + { + u8 a, b, c, d; + u32 addr; + addr = ntohl(m_address.ipv4.sin_addr.s_addr); + a = (addr & 0xFF000000) >> 24; + b = (addr & 0x00FF0000) >> 16; + c = (addr & 0x0000FF00) >> 8; + d = (addr & 0x000000FF); + return itos(a) + "." + itos(b) + "." + itos(c) + "." + itos(d); + } + else if(m_addr_family == AF_INET6) + { + std::ostringstream os; + for(int i = 0; i < 16; i += 2) + { + u16 section = + (m_address.ipv6.sin6_addr.s6_addr[i] << 8) | + (m_address.ipv6.sin6_addr.s6_addr[i + 1]); + os << std::hex << section; + if(i < 14) + os << ":"; + } + return os.str(); + } + else + return std::string(""); +#else + char str[INET6_ADDRSTRLEN]; + if (inet_ntop(m_addr_family, (m_addr_family == AF_INET) ? (void*)&(m_address.ipv4.sin_addr) : (void*)&(m_address.ipv6.sin6_addr), str, INET6_ADDRSTRLEN) == NULL) { + return std::string(""); + } + return std::string(str); +#endif +} + +struct sockaddr_in Address::getAddress() const +{ + return m_address.ipv4; // NOTE: NO PORT INCLUDED, use getPort() +} + +struct sockaddr_in6 Address::getAddress6() const { - return m_address; + return m_address.ipv6; // NOTE: NO PORT INCLUDED, use getPort() } -unsigned short Address::getPort() const +u16 Address::getPort() const { return m_port; } -void Address::setAddress(unsigned int address) +int Address::getFamily() const +{ + return m_addr_family; +} + +bool Address::isIPv6() const +{ + return m_addr_family == AF_INET6; +} + +bool Address::isZero() const +{ + if (m_addr_family == AF_INET) { + return m_address.ipv4.sin_addr.s_addr == 0; + } else if (m_addr_family == AF_INET6) { + static const char zero[16] = {0}; + return memcmp(m_address.ipv6.sin6_addr.s6_addr, + zero, 16) == 0; + } + return false; +} + +void Address::setAddress(u32 address) { - m_address = address; + m_addr_family = AF_INET; + m_address.ipv4.sin_family = AF_INET; + m_address.ipv4.sin_addr.s_addr = htonl(address); } -void Address::setPort(unsigned short port) +void Address::setAddress(u8 a, u8 b, u8 c, u8 d) +{ + m_addr_family = AF_INET; + m_address.ipv4.sin_family = AF_INET; + u32 addr = htonl((a << 24) | (b << 16) | (c << 8) | d); + m_address.ipv4.sin_addr.s_addr = addr; +} + +void Address::setAddress(const IPv6AddressBytes *ipv6_bytes) +{ + m_addr_family = AF_INET6; + m_address.ipv6.sin6_family = AF_INET6; + if (ipv6_bytes) + memcpy(m_address.ipv6.sin6_addr.s6_addr, ipv6_bytes->bytes, 16); + else + memset(m_address.ipv6.sin6_addr.s6_addr, 0, 16); +} + +void Address::setPort(u16 port) { m_port = port; } void Address::print(std::ostream *s) const { - (*s)<<((m_address>>24)&0xff)<<"." - <<((m_address>>16)&0xff)<<"." - <<((m_address>>8)&0xff)<<"." - <<((m_address>>0)&0xff)<<":" - < "; - destination.print(); - dstream<<", size="< "; + destination.print(&dstream); + dstream << ", size=" << size; + + // Print packet contents + dstream << ", data="; + for(int i = 0; i < size && i < 20; i++) { + if(i % 2 == 0) + dstream << " "; + unsigned int a = ((const unsigned char *)data)[i]; + dstream << std::hex << std::setw(2) << std::setfill('0') << a; } - if(size>20) - dstream<<"..."; + + if(size > 20) + dstream << "..."; + if(dumping_packet) - dstream<<" (DUMPED BY INTERNET_SIMULATOR)"; - dstream<20) - dstream<<"..."; - dstream< 20) + dstream << "..."; + + dstream << std::endl; } return received; @@ -301,40 +556,38 @@ bool UDPSocket::WaitData(int timeout_ms) struct timeval tv; tv.tv_sec = 0; tv.tv_usec = timeout_ms * 1000; + // select() result = select(m_handle+1, &readset, NULL, NULL, &tv); - if(result == 0){ - // Timeout - /*dstream<<"Select timed out (timeout_ms=" - <