Swiftdiff options
| author | Remko Tronçon <git@el-tramo.be> | 2011-09-26 20:50:17 (GMT) |
|---|---|---|
| committer | Remko Tronçon <git@el-tramo.be> | 2011-09-26 20:59:50 (GMT) |
| commit | 694cc7318b92aed079badb3cc22428ab108591cf (patch) | |
| tree | 6b41862164d7e65fef8818778b42fa9f46fc81c4 /Swiften/Network | |
| parent | 4b61e7363dbc0e8bf595da06840db4175e3b86c3 (diff) | |
| download | swift-contrib-694cc7318b92aed079badb3cc22428ab108591cf.zip swift-contrib-694cc7318b92aed079badb3cc22428ab108591cf.tar.bz2 | |
Windows NetworkEnvironment fixes.
Diffstat (limited to 'Swiften/Network')
| -rw-r--r-- | Swiften/Network/NetworkInterface.h | 12 | ||||
| -rw-r--r-- | Swiften/Network/WindowsNetworkEnvironment.cpp | 47 |
2 files changed, 17 insertions, 42 deletions
diff --git a/Swiften/Network/NetworkInterface.h b/Swiften/Network/NetworkInterface.h index 6395758..1d302cb 100644 --- a/Swiften/Network/NetworkInterface.h +++ b/Swiften/Network/NetworkInterface.h @@ -7,28 +7,19 @@ #pragma once #include <vector> #include <Swiften/Network/HostAddress.h> namespace Swift { class NetworkInterface { public: - enum Type { - Unknown, - WLAN, - Ethernet, - Mobile, - VPN, - }; - - public: - NetworkInterface(const std::string& name, bool loopback, Type type = Unknown) : name(name), loopback(loopback), type(type) { + NetworkInterface(const std::string& name, bool loopback) : name(name), loopback(loopback) { } void addAddress(const HostAddress& address) { addresses.push_back(address); } const std::vector<HostAddress>& getAddresses() const { return addresses; } @@ -38,13 +29,12 @@ namespace Swift { } bool isLoopback() const { return loopback; } private: std::string name; bool loopback; - Type type; std::vector<HostAddress> addresses; }; } diff --git a/Swiften/Network/WindowsNetworkEnvironment.cpp b/Swiften/Network/WindowsNetworkEnvironment.cpp index 0eedea0..5df04a7 100644 --- a/Swiften/Network/WindowsNetworkEnvironment.cpp +++ b/Swiften/Network/WindowsNetworkEnvironment.cpp @@ -3,45 +3,30 @@ * Licensed under the simplified BSD license. * See Documentation/Licenses/BSD-simplified.txt for more information. */ #include <Swiften/Network/WindowsNetworkEnvironment.h> #include <string> #include <vector> #include <map> - -#include <boost/shared_ptr.hpp> -#include <boost/smart_ptr/make_shared.hpp> -#include <Swiften/Base/boost_bsignals.h> +#include <boost/optional.hpp> #include <Swiften/Network/HostAddress.h> #include <Swiften/Network/NetworkInterface.h> #include <winsock2.h> #include <iphlpapi.h> namespace Swift { -std::string winSocketAddressToStdString(const SOCKET_ADDRESS& socketAddress) { - char text[46]; - ULONG bufferSize = sizeof(text); - std::string result; - - int ret = WSAAddressToString(socketAddress.lpSockaddr, socketAddress.iSockaddrLength, NULL, text, &bufferSize); - if (ret == 0) { - result.assign(text, sizeof(text)); - } - return result; -} - -std::vector<NetworkInterface> WindowsNetworkEnvironment::getNetworkInterfaces() { +std::vector<NetworkInterface> WindowsNetworkEnvironment::getNetworkInterfaces() const { std::vector<NetworkInterface> result; - std::map<std::string, WindowsNetworkInterface> interfaces; + std::map<std::string,NetworkInterface> interfaces; IP_ADAPTER_ADDRESSES preBuffer[5]; PIP_ADAPTER_ADDRESSES adapterStart = preBuffer; ULONG bufferSize = sizeof(preBuffer); ULONG flags = GAA_FLAG_INCLUDE_ALL_INTERFACES | GAA_FLAG_INCLUDE_PREFIX | GAA_FLAG_SKIP_MULTICAST | GAA_FLAG_SKIP_DNS_SERVER; ULONG ret = GetAdaptersAddresses( AF_UNSPEC, flags, NULL, adapterStart, &bufferSize); @@ -62,36 +47,36 @@ std::vector<NetworkInterface> WindowsNetworkEnvironment::getNetworkInterfaces() for (PIP_ADAPTER_ADDRESSES adapter = adapterStart; adapter; adapter = adapter->Next) { std::string name(adapter->AdapterName); if (adapter->OperStatus != IfOperStatusUp) { continue; } // iterate over addresses for (PIP_ADAPTER_UNICAST_ADDRESS address = adapter->FirstUnicastAddress; address; address = address->Next) { - std::string ip; - - if (address->Address.lpSockaddr->sa_family == PF_INET || - address->Address.lpSockaddr->sa_family == PF_INET6) { - ip = winSocketAddressToStdString(address->Address); - if (!ip.empty()) { - if (interfaces.find(name) == interfaces.end()) { - interfaces[name] = boost::make_shared<WindowsNetworkInterface>(); - interfaces[name]->setName(name); - } - interfaces[name]->addAddress(HostAddress(ip)); - } + boost::optional<HostAddress> hostAddress; + if (address->Address.lpSockaddr->sa_family == PF_INET) { + sockaddr_in* sa = reinterpret_cast<sockaddr_in*>(address->Address.lpSockaddr); + hostAddress = HostAddress(reinterpret_cast<const unsigned char*>(&(sa->sin_addr)), 4); + } + else if (address->Address.lpSockaddr->sa_family == PF_INET6) { + sockaddr_in6* sa = reinterpret_cast<sockaddr_in6*>(address->Address.lpSockaddr); + hostAddress = HostAddress(reinterpret_cast<const unsigned char*>(&(sa->sin6_addr)), 16); + } + if (hostAddress) { + std::map<std::string, NetworkInterface>::iterator i = interfaces.insert(std::make_pair(name, NetworkInterface(name, false))).first; + i->second.addAddress(*hostAddress); } } } if (adapterStart != preBuffer) { //delete adapterStart; } - for (std::map<std::string, WindowsNetworkInterface>::const_iterator i = interfaces.begin(); i != interfaces.end(); ++i) { + for (std::map<std::string,NetworkInterface>::const_iterator i = interfaces.begin(); i != interfaces.end(); ++i) { result.push_back(i->second); } return result; } } |