PurpleI2P
/
i2pd
mirror of https://github.com/PurpleI2P/i2pd.git
1
0
Fork 0
Code Issues Releases Activity
Browse Source

backport GetMTU

pull/304/head
orignal 9 years ago
parent
c42636b0ee
commit
a8f2239495
1 changed files with 191 additions and 174 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 359
      util.cpp

359
util.cpp
Unescape View File

@ -506,200 +506,217 @@ namespace http
namespace net namespace net
{ {
int GetMTU (const boost::asio::ip::address& localAddress)
{
#if defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__APPLE__) || defined(__OpenBSD__) #if defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__APPLE__) || defined(__OpenBSD__)
ifaddrs * ifaddr, * ifa = nullptr;
if (getifaddrs(&ifaddr) == -1) int GetMTUUnix(const boost::asio::ip::address& localAddress, int fallback)
{
ifaddrs* ifaddr, *ifa = nullptr;
if(getifaddrs(&ifaddr) == -1)
{ {
LogPrint (eLogError, "Can't excute getifaddrs"); LogPrint(eLogError, "Can't excute getifaddrs");
return 0; return fallback;
} }
int family = 0;
// loook for interface matching local address int family = 0;
for (ifa = ifaddr; ifa != nullptr; ifa = ifa->ifa_next) // look for interface matching local address
for(ifa = ifaddr; ifa != nullptr; ifa = ifa->ifa_next)
{ {
if (!ifa->ifa_addr) continue; if(!ifa->ifa_addr)
family = ifa->ifa_addr->sa_family; continue;
if (family == AF_INET && localAddress.is_v4 ())
family = ifa->ifa_addr->sa_family;
if(family == AF_INET && localAddress.is_v4())
{ {
sockaddr_in * sa = (sockaddr_in *)ifa->ifa_addr; sockaddr_in* sa = (sockaddr_in*) ifa->ifa_addr;
if (!memcmp (&sa->sin_addr, localAddress.to_v4 ().to_bytes ().data (), 4)) if(!memcmp(&sa->sin_addr, localAddress.to_v4().to_bytes().data(), 4))
break; // address matches break; // address matches
} }
else if (family == AF_INET6 && localAddress.is_v6 ()) else if(family == AF_INET6 && localAddress.is_v6())
{ {
sockaddr_in6 * sa = (sockaddr_in6 *)ifa->ifa_addr; sockaddr_in6* sa = (sockaddr_in6*) ifa->ifa_addr;
if (!memcmp (&sa->sin6_addr, localAddress.to_v6 ().to_bytes ().data (), 16)) if(!memcmp(&sa->sin6_addr, localAddress.to_v6().to_bytes().data(), 16))
break; // address matches break; // address matches
} }
} }
int mtu = 0; int mtu = fallback;
if (ifa && family) // interface found? if(ifa && family)
{ { // interface found?
int fd = socket (family, SOCK_DGRAM, 0); int fd = socket(family, SOCK_DGRAM, 0);
if (fd > 0) if(fd > 0)
{ {
ifreq ifr; ifreq ifr;
strncpy (ifr.ifr_name, ifa->ifa_name, IFNAMSIZ); // set interface for query strncpy(ifr.ifr_name, ifa->ifa_name, IFNAMSIZ); // set interface for query
if (ioctl (fd, SIOCGIFMTU, &ifr) >= 0) if(ioctl(fd, SIOCGIFMTU, &ifr) >= 0)
mtu = ifr.ifr_mtu; // MTU mtu = ifr.ifr_mtu; // MTU
else else
LogPrint (eLogError, "Failed to run ioctl"); LogPrint (eLogError, "Failed to run ioctl");
close (fd); close(fd);
} }
else else
LogPrint (eLogError, "Failed to create datagram socket"); LogPrint(eLogError, "Failed to create datagram socket");
} }
else else
LogPrint (eLogWarning, "Interface for local address", localAddress.to_string (), " not found"); LogPrint(eLogWarning, "Interface for local address", localAddress.to_string(), " not found");
freeifaddrs(ifaddr);
freeifaddrs (ifaddr); return mtu;
return mtu; }
#elif defined(WIN32)
int result = 576; // fallback MTU #elif defined(_WIN32)
int GetMTUWindowsIpv4(sockaddr_in inputAddress, int fallback)
{
ULONG outBufLen = 0;
PIP_ADAPTER_ADDRESSES pAddresses = nullptr;
PIP_ADAPTER_ADDRESSES pCurrAddresses = nullptr;
PIP_ADAPTER_UNICAST_ADDRESS pUnicast = nullptr;
DWORD dwRetVal = 0;
ULONG outBufLen = 0;
PIP_ADAPTER_ADDRESSES pAddresses = nullptr;
PIP_ADAPTER_ADDRESSES pCurrAddresses = nullptr;
PIP_ADAPTER_UNICAST_ADDRESS pUnicast = nullptr;
#ifdef UNICODE if(GetAdaptersAddresses(AF_INET, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen)
string localAddress_temporary = localAddress.to_string(); == ERROR_BUFFER_OVERFLOW) {
wstring localAddressUniversal (localAddress_temporary.begin(), localAddress_temporary.end()); FREE(pAddresses);
#else pAddresses = (IP_ADAPTER_ADDRESSES*) MALLOC(outBufLen);
std::string localAddressUniversal = localAddress.to_string(); }
#endif
if (localAddress.is_v4()) DWORD dwRetVal = GetAdaptersAddresses(
{ AF_INET, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen
struct sockaddr_in inputAddress; );
inet_pton(AF_INET, localAddressUniversal.c_str(), &(inputAddress.sin_addr));
if (GetAdaptersAddresses (AF_INET, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen) if(dwRetVal != NO_ERROR) {
== ERROR_BUFFER_OVERFLOW) LogPrint(
{ eLogError, "GetMTU() has failed: enclosed GetAdaptersAddresses() call has failed"
FREE (pAddresses); );
pAddresses = (IP_ADAPTER_ADDRESSES *)MALLOC (outBufLen); FREE(pAddresses);
} return fallback;
}
dwRetVal = GetAdaptersAddresses (AF_INET, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen); pCurrAddresses = pAddresses;
if (dwRetVal == NO_ERROR) while(pCurrAddresses) {
{ PIP_ADAPTER_UNICAST_ADDRESS firstUnicastAddress = pCurrAddresses->FirstUnicastAddress;
pCurrAddresses = pAddresses;
while (pCurrAddresses) pUnicast = pCurrAddresses->FirstUnicastAddress;
{ if(pUnicast == nullptr) {
PIP_ADAPTER_UNICAST_ADDRESS firstUnicastAddress = pCurrAddresses->FirstUnicastAddress; LogPrint(
eLogError, "GetMTU() has failed: not a unicast ipv4 address, this is not supported"
pUnicast = pCurrAddresses->FirstUnicastAddress; );
if (pUnicast != nullptr) }
{ for(int i = 0; pUnicast != nullptr; ++i) {
for (int i = 0; pUnicast != nullptr; ++i) LPSOCKADDR lpAddr = pUnicast->Address.lpSockaddr;
{ sockaddr_in* localInterfaceAddress = (sockaddr_in*) lpAddr;
LPSOCKADDR lpAddr = pUnicast->Address.lpSockaddr; if(localInterfaceAddress->sin_addr.S_un.S_addr == inputAddress.sin_addr.S_un.S_addr) {
struct sockaddr_in *localInterfaceAddress = (struct sockaddr_in*) lpAddr; auto result = pAddresses->Mtu;
if (localInterfaceAddress->sin_addr.S_un.S_addr == inputAddress.sin_addr.S_un.S_addr) FREE(pAddresses);
{ return result;
result = pAddresses->Mtu; }
FREE (pAddresses); pUnicast = pUnicast->Next;
pAddresses = nullptr; }
return result; pCurrAddresses = pCurrAddresses->Next;
} }
pUnicast = pUnicast->Next;
}
}
else
{
LogPrint (eLogError, "GetMTU() has failed: not a unicast ipv4 address, this is not supported");
}
pCurrAddresses = pCurrAddresses->Next;
}
} LogPrint(eLogError, "GetMTU() error: no usable unicast ipv4 addresses found");
else FREE(pAddresses);
{ return fallback;
LogPrint (eLogError, "GetMTU() has failed: enclosed GetAdaptersAddresses() call has failed"); }
}
int GetMTUWindowsIpv6(sockaddr_in6 inputAddress, int fallback)
{
ULONG outBufLen = 0;
PIP_ADAPTER_ADDRESSES pAddresses = nullptr;
PIP_ADAPTER_ADDRESSES pCurrAddresses = nullptr;
PIP_ADAPTER_UNICAST_ADDRESS pUnicast = nullptr;
if(GetAdaptersAddresses(AF_INET6, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen)
== ERROR_BUFFER_OVERFLOW) {
FREE(pAddresses);
pAddresses = (IP_ADAPTER_ADDRESSES*) MALLOC(outBufLen);
}
} DWORD dwRetVal = GetAdaptersAddresses(
else if (localAddress.is_v6()) AF_INET6, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen
{ );
struct sockaddr_in6 inputAddress;
inet_pton(AF_INET6, localAddressUniversal.c_str(), &(inputAddress.sin6_addr)); if(dwRetVal != NO_ERROR) {
LogPrint(
eLogError,
"GetMTU() has failed: enclosed GetAdaptersAddresses() call has failed"
);
FREE(pAddresses);
return fallback;
}
if (GetAdaptersAddresses(AF_INET6, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen) bool found_address = false;
== ERROR_BUFFER_OVERFLOW) pCurrAddresses = pAddresses;
{ while(pCurrAddresses) {
FREE (pAddresses); PIP_ADAPTER_UNICAST_ADDRESS firstUnicastAddress = pCurrAddresses->FirstUnicastAddress;
pAddresses = (IP_ADAPTER_ADDRESSES *)MALLOC (outBufLen); pUnicast = pCurrAddresses->FirstUnicastAddress;
} if(pUnicast == nullptr) {
LogPrint(
eLogError,
"GetMTU() has failed: not a unicast ipv6 address, this is not supported"
);
}
for(int i = 0; pUnicast != nullptr; ++i) {
LPSOCKADDR lpAddr = pUnicast->Address.lpSockaddr;
sockaddr_in6 *localInterfaceAddress = (sockaddr_in6*) lpAddr;
for (int j = 0; j != 8; ++j) {
if (localInterfaceAddress->sin6_addr.u.Word[j] != inputAddress.sin6_addr.u.Word[j]) {
break;
} else {
found_address = true;
}
} if (found_address) {
auto result = pAddresses->Mtu;
FREE(pAddresses);
pAddresses = nullptr;
return result;
}
pUnicast = pUnicast->Next;
}
pCurrAddresses = pCurrAddresses->Next;
}
dwRetVal = GetAdaptersAddresses (AF_INET6, GAA_FLAG_INCLUDE_PREFIX, nullptr, pAddresses, &outBufLen); LogPrint(eLogError, "GetMTU() error: no usable unicast ipv6 addresses found");
if (dwRetVal == NO_ERROR) FREE(pAddresses);
{ return fallback;
bool found_address = false; }
pCurrAddresses = pAddresses;
while (pCurrAddresses)
{
PIP_ADAPTER_UNICAST_ADDRESS firstUnicastAddress = pCurrAddresses->FirstUnicastAddress;
pUnicast = pCurrAddresses->FirstUnicastAddress;
if (pUnicast != nullptr)
{
for (int i = 0; pUnicast != nullptr; ++i)
{
LPSOCKADDR lpAddr = pUnicast->Address.lpSockaddr;
struct sockaddr_in6 *localInterfaceAddress = (struct sockaddr_in6*) lpAddr;
for (int j = 0; j != 8; ++j)
{
if (localInterfaceAddress->sin6_addr.u.Word[j] != inputAddress.sin6_addr.u.Word[j])
{
break;
}
else
{
found_address = true;
}
}
if (found_address)
{
result = pAddresses->Mtu;
FREE (pAddresses);
pAddresses = nullptr;
return result;
}
pUnicast = pUnicast->Next;
}
}
else
{
LogPrint (eLogError, "GetMTU() has failed: not a unicast ipv6 address, this is not supported");
}
pCurrAddresses = pCurrAddresses->Next;
}
} int GetMTUWindows(const boost::asio::ip::address& localAddress, int fallback)
else {
{ #ifdef UNICODE
LogPrint (eLogError, "GetMTU() has failed: enclosed GetAdaptersAddresses() call has failed"); string localAddress_temporary = localAddress.to_string();
} wstring localAddressUniversal(localAddress_temporary.begin(), localAddress_temporary.end());
} #else
else std::string localAddressUniversal = localAddress.to_string();
{ #endif
LogPrint (eLogError, "GetMTU() has failed: address family is not supported");
}
FREE (pAddresses); if(localAddress.is_v4()) {
pAddresses = nullptr; sockaddr_in inputAddress;
LogPrint(eLogError, "GetMTU() error: control flow should never reach this line"); inet_pton(AF_INET, localAddressUniversal.c_str(), &(inputAddress.sin_addr));
return result; return GetMTUWindowsIpv4(inputAddress, fallback);
} else if(localAddress.is_v6()) {
sockaddr_in6 inputAddress;
inet_pton(AF_INET6, localAddressUniversal.c_str(), &(inputAddress.sin6_addr));
return GetMTUWindowsIpv6(inputAddress, fallback);
} else {
LogPrint(eLogError, "GetMTU() has failed: address family is not supported");
return fallback;
}
}
#endif // WIN32
int GetMTU(const boost::asio::ip::address& localAddress)
{
const int fallback = 576; // fallback MTU
#if defined(__linux__) || defined(__FreeBSD_kernel__) || defined(__APPLE__) || defined(__OpenBSD__)
return GetMTUUnix(localAddress, fallback);
#elif defined(WIN32)
return GetMTUWindows(localAddress, fallback);
#endif #endif
} return fallback;
}
} }
} // util } // util

Write Preview
Loading…
Cancel
Save