X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Fsocket.cpp;h=e82052f7728083430f32e3ed2bdd2d914045dc99;hb=c6766b275f25368e78eeffc68719b0ca1979c819;hp=f8afed229add6f4093c72ea66684a2e6a6a182a3;hpb=d0ea6f9920d30f46d1f5d44e8823a8d932f9f29d;p=oweals%2Fminetest.git diff --git a/src/socket.cpp b/src/socket.cpp index f8afed229..e82052f77 100644 --- a/src/socket.cpp +++ b/src/socket.cpp @@ -1,6 +1,6 @@ /* -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 Lesser General Public License as published by @@ -19,8 +19,24 @@ with this program; if not, write to the Free Software Foundation, Inc., #include "socket.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" + #ifdef _WIN32 - #define WIN32_LEAN_AND_MEAN + #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 @@ -28,9 +44,6 @@ with this program; if not, write to the Free Software Foundation, Inc., #include #include #include - #ifdef _MSC_VER - #pragma comment(lib, "ws2_32.lib") - #endif typedef SOCKET socket_t; typedef int socklen_t; #else @@ -44,29 +57,19 @@ typedef int socklen_t; typedef int socket_t; #endif -#include "constants.h" -#include "debug.h" -#include -#include -#include -#include -#include "util/string.h" -#include "util/numeric.h" - -bool socket_enable_debug_output = false; -#define DP socket_enable_debug_output -// This is prepended to everything printed here -#define DPS "" +// 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; } @@ -74,142 +77,284 @@ void sockets_init() void sockets_cleanup() { #ifdef _WIN32 + // On Windows, cleanup sockets after use WSACleanup(); #endif } +/* + Address +*/ + Address::Address() { - m_address = 0; + 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); +} + +// 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!=(Address &address) +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); } +// IP address -> textual representation std::string Address::serializeString() const { - unsigned int a, b, c, d; - a = (m_address & 0xFF000000)>>24; - b = (m_address & 0x00FF0000)>>16; - c = (m_address & 0x0000FF00)>>8; - d = (m_address & 0x000000FF); - return itos(a)+"."+itos(b)+"."+itos(c)+"."+itos(d); +// 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() } -unsigned int Address::getAddress() const +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 { - m_address = address; + return m_addr_family; } -void Address::setAddress(unsigned int a, unsigned int b, - unsigned int c, unsigned int d) +bool Address::isIPv6() const { - m_address = (a<<24) | (b<<16) | ( c<<8) | d; + 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::setPort(unsigned short port) +void Address::setAddress(u32 address) +{ + m_addr_family = AF_INET; + m_address.ipv4.sin_family = AF_INET; + m_address.ipv4.sin_addr.s_addr = htonl(address); +} + +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; @@ -351,43 +552,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=" - <