izackp/Apollo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Try fix get_local_ip_for_gateway for Linux

Browse Source
This commit is contained in:
Yukino Song
2024-09-14 04:18:23 +08:00
parent 2617b17aab
commit 80ea11b815
1 changed files with 42 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/platform/linux/misc.cpp
View File

@@ -241,11 +241,11 @@ namespace platf {
return "00:00:00:00:00:00"s; return "00:00:00:00:00:00"s;
} }
std::string std::string
get_local_ip_for_gateway() { get_local_ip_for_gateway() {
int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE); int fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
if (fd < 0) { if (fd < 0) {
BOOST_LOG(warning) << "Socket creation failed"; BOOST_LOG(warning) << "Socket creation failed: " << strerror(errno);
return ""; return "";
} }
@@ -263,21 +263,21 @@ namespace platf {
nlMsg->nlmsg_pid = getpid(); nlMsg->nlmsg_pid = getpid();
if (send(fd, nlMsg, nlMsg->nlmsg_len, 0) < 0) { if (send(fd, nlMsg, nlMsg->nlmsg_len, 0) < 0) {
BOOST_LOG(warning) << "Send message failed: " << strerror(errno);
close(fd); close(fd);
BOOST_LOG(warning) << "Send message failed";
return "";
}
len = recv(fd, nlMsg, sizeof(buffer), 0);
if (len < 0) {
close(fd);
BOOST_LOG(warning) << "Receive message failed";
return ""; return "";
} }
std::string local_ip; std::string local_ip;
bool found = false;
while ((len = recv(fd, nlMsg, sizeof(buffer), 0)) > 0) {
for (; NLMSG_OK(nlMsg, len); nlMsg = NLMSG_NEXT(nlMsg, len)) { for (; NLMSG_OK(nlMsg, len); nlMsg = NLMSG_NEXT(nlMsg, len)) {
if (nlMsg->nlmsg_type == NLMSG_DONE) {
found = true;
break;
}
rtMsg = (struct rtmsg *)NLMSG_DATA(nlMsg); rtMsg = (struct rtmsg *)NLMSG_DATA(nlMsg);
if (rtMsg->rtm_family != AF_INET || rtMsg->rtm_table != RT_TABLE_MAIN) if (rtMsg->rtm_family != AF_INET || rtMsg->rtm_table != RT_TABLE_MAIN)
continue; continue;
@@ -305,10 +305,14 @@ namespace platf {
if (gateway.s_addr != 0 && local.s_addr != 0) { if (gateway.s_addr != 0 && local.s_addr != 0) {
local_ip = inet_ntoa(local); local_ip = inet_ntoa(local);
found = true;
break; break;
} }
} }
if (found) break;
}
close(fd); close(fd);
if (local_ip.empty()) { if (local_ip.empty()) {