Skip to content

Commit

Permalink
DEV9: Improve logic for getting MacAddress
Browse files Browse the repository at this point in the history
  • Loading branch information
fjtrujy committed Apr 23, 2024
1 parent c4190df commit 2edf250
Showing 1 changed file with 63 additions and 20 deletions.
83 changes: 63 additions & 20 deletions pcsx2/DEV9/AdapterUtils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -5,25 +5,28 @@

#include "common/Assertions.h"
#include "common/Console.h"
#include "common/ScopedGuard.h"
#include "common/StringUtil.h"

#ifdef __POSIX__
#include <unistd.h>
#include <vector>
#include <fstream>
#include <net/if.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#ifdef __linux__
#include <unistd.h>
#include <sys/ioctl.h>
#endif
#include <string.h>

#if defined(__FreeBSD__) || (__APPLE__)
#include <sys/types.h>
#include <net/if_dl.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <sys/sockio.h>
#include <net/route.h>
#include <net/if_var.h>
#endif
#endif

Expand Down Expand Up @@ -244,30 +247,70 @@ std::optional<MAC_Address> AdapterUtils::GetAdapterMAC(Adapter* adapter)

return std::nullopt;
}
#elif defined(__POSIX__)
#ifdef __linux__
#else
std::optional<MAC_Address> AdapterUtils::GetAdapterMAC(Adapter* adapter)
{
struct ifreq ifr;
strcpy(ifr.ifr_name, adapter->ifa_name);
MAC_Address macAddr = {};
#if defined(AF_LINK)
ifaddrs *adapterInfo;
const int error = getifaddrs(&adapterInfo);
if (error)
{
Console.Error("DEV9: Failed to get adapter information %s", error);
return std::nullopt;
}

int fd = socket(AF_INET, SOCK_DGRAM, 0);
int ret = ioctl(fd, SIOCGIFHWADDR, &ifr);
close(fd);
const ScopedGuard adapterInfoGuard = [&adapterInfo]() {
freeifaddrs(adapterInfo);
};

if (ret == 0)
return *(MAC_Address*)ifr.ifr_hwaddr.sa_data;
for (const ifaddrs *po = adapterInfo; po; po = po->ifa_next)
{
if (strcmp(po->ifa_name, adapter->ifa_name))
continue;

return std::nullopt;
}
if (po->ifa_addr->sa_family != AF_LINK)
continue;

// We have a valid MAC address.
std::memcpy(&macAddr, LLADDR(reinterpret_cast<sockaddr_dl*>(po->ifa_addr)), sizeof(macAddr));
return macAddr;
}
freeifaddrs(adapterInfo);
Console.Error("DEV9: Failed to get MAC address for adapter using AF_LINK");
#else
std::optional<MAC_Address> AdapterUtils::GetAdapterMAC(Adapter* adapter)
{
Console.Error("DEV9: Unsupported OS, can't get MAC address");
const int sd = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
if (sd < 0)
{
Console.Error("DEV9: Failed to open socket");
return std::nullopt;
}
struct ifreq ifr = {};
StringUtil::Strlcpy(ifr.ifr_name, adapter->ifa_name, std::size(ifr.ifr_name));
#if defined(SIOCGIFHWADDR)
if (ioctl(sd, SIOCGIFHWADDR, &ifr) < 0)
{
close(sd);
Console.Error("DEV9: Failed to get MAC address for adapter using SIOCGIFHWADDR");
return std::nullopt;
}
std::memcpy(&macAddr, &ifr.ifr_hwaddr.sa_data, sizeof(macAddr));
#elif defined(SIOCGENADDR)
if (ioctl(sd, SIOCGENADDR, &ifr) < 0)
{
close(sd);
Console.Error("DEV9: Failed to get MAC address for adapter using SIOCGENADDR");
return std::nullopt;
}
std::memcpy(&macAddr, &ifr.ifr_enaddr, sizeof(macAddr));
#endif
close(sd);
return macAddr;
#endif
Console.Error("DEV9: Unsupported OS, can't get network features");
return std::nullopt;
}
#endif
#endif

// AdapterIP.
#ifdef _WIN32
Expand Down Expand Up @@ -376,7 +419,7 @@ std::vector<IP_Address> AdapterUtils::GetGateways(Adapter* adapter)
}
return collection;
}
#elif defined(__FreeBSD__) || (__APPLE__)
#elif defined(__FreeBSD__) || defined(__APPLE__)
std::vector<IP_Address> AdapterUtils::GetGateways(Adapter* adapter)
{
if (adapter == nullptr)
Expand Down

0 comments on commit 2edf250

Please sign in to comment.