fixes to make enum_net.cpp actually work

This commit is contained in:
Arvid Norberg
2008-01-11 07:56:52 +00:00
parent 72edfc2ee7
commit 1f99a1d98d

View File

@@ -43,6 +43,7 @@ POSSIBILITY OF SUCH DAMAGE.
#endif #endif
#include "libtorrent/enum_net.hpp" #include "libtorrent/enum_net.hpp"
#include <asio/ip/host_name.hpp>
namespace libtorrent namespace libtorrent
{ {
@@ -96,8 +97,9 @@ namespace libtorrent
std::vector<ip_interface> enum_net_interfaces(asio::io_service& ios, asio::error_code& ec) std::vector<ip_interface> enum_net_interfaces(asio::io_service& ios, asio::error_code& ec)
{ {
std::vector<ip_interface> ret; std::vector<ip_interface> ret;
// covers linux, MacOS X and BSD distributions
#if defined __linux__ || defined BSD #if defined __linux__ || (defined __APPLE__ && __MACH__) || defined __FreeBSD__ || defined __NetBSD__ \
|| defined __OpenBSD__ || defined __bsdi__ || defined __DragonFly__
int s = socket(AF_INET, SOCK_DGRAM, 0); int s = socket(AF_INET, SOCK_DGRAM, 0);
if (s < 0) if (s < 0)
{ {
@@ -130,16 +132,28 @@ namespace libtorrent
ifreq netmask = item; ifreq netmask = item;
if (ioctl(s, SIOCGIFNETMASK, &netmask) < 0) if (ioctl(s, SIOCGIFNETMASK, &netmask) < 0)
{
if (iface.interface_address.is_v6())
{
// this is expected to fail (at least on MacOS X)
iface.netmask = address_v6::any();
}
else
{ {
close(s); close(s);
ec = asio::error_code(errno, asio::error::system_category); ec = asio::error_code(errno, asio::error::system_category);
return ret; return ret;
} }
}
else
{
iface.netmask = sockaddr_to_address(&netmask.ifr_addr); iface.netmask = sockaddr_to_address(&netmask.ifr_addr);
}
ret.push_back(iface); ret.push_back(iface);
} }
#if defined BSD #if (defined __APPLE__ && __MACH__) || defined __FreeBSD__ || defined __NetBSD__ \
|| defined __OpenBSD__ || defined __bsdi__ || defined __DragonFly__
int current_size = item.ifr_addr.sa_len + IFNAMSIZ; int current_size = item.ifr_addr.sa_len + IFNAMSIZ;
#elif defined __linux__ #elif defined __linux__
int current_size = sizeof(ifreq); int current_size = sizeof(ifreq);
@@ -154,7 +168,7 @@ namespace libtorrent
SOCKET s = socket(AF_INET, SOCK_DGRAM, 0); SOCKET s = socket(AF_INET, SOCK_DGRAM, 0);
if (s == SOCKET_ERROR) if (s == SOCKET_ERROR)
{ {
ec = asio::error::fault; ec = asio::error_code(WSAGetLastError(), asio::error::system_category);
return ret; return ret;
} }
@@ -164,8 +178,8 @@ namespace libtorrent
if (WSAIoctl(s, SIO_GET_INTERFACE_LIST, 0, 0, buffer, if (WSAIoctl(s, SIO_GET_INTERFACE_LIST, 0, 0, buffer,
sizeof(buffer), &size, 0, 0) != 0) sizeof(buffer), &size, 0, 0) != 0)
{ {
ec = asio::error_code(WSAGetLastError(), asio::error::system_category);
closesocket(s); closesocket(s);
ec = asio::error::fault;
return ret; return ret;
} }
closesocket(s); closesocket(s);
@@ -185,13 +199,14 @@ namespace libtorrent
#warning THIS OS IS NOT RECOGNIZED, enum_net_interfaces WILL PROBABLY NOT WORK #warning THIS OS IS NOT RECOGNIZED, enum_net_interfaces WILL PROBABLY NOT WORK
// make a best guess of the interface we're using and its IP // make a best guess of the interface we're using and its IP
udp::resolver r(ios); udp::resolver r(ios);
udp::resolver::iterator i = r.resolve(udp::resolver::query(asio::ip::host_name(), "0")); udp::resolver::iterator i = r.resolve(udp::resolver::query(asio::ip::host_name(ec), "0"), ec);
if (ec) return ret;
ip_interface iface; ip_interface iface;
for (;i != udp::resolver_iterator(); ++i) for (;i != udp::resolver_iterator(); ++i)
{ {
iface.interface_address = i->endpoint().address(); iface.interface_address = i->endpoint().address();
if (iface.interface_address.is_v4()) if (iface.interface_address.is_v4())
iface.netmask = iface.interface_address.to_v4().netmask(); iface.netmask = address_v4::netmask(iface.interface_address.to_v4());
ret.push_back(iface); ret.push_back(iface);
} }
#endif #endif