mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
411 lines
10 KiB
C++
411 lines
10 KiB
C++
#include "route.hpp"
|
|
|
|
#ifdef __linux__
|
|
#include <string.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <net/if.h>
|
|
#include <arpa/inet.h>
|
|
#include <sys/socket.h>
|
|
#include <linux/rtnetlink.h>
|
|
#endif
|
|
|
|
#include <sstream>
|
|
|
|
namespace llarp::net
|
|
{
|
|
#ifndef __linux__
|
|
void
|
|
Execute(std::string cmd)
|
|
{
|
|
std::cout << cmd << std::endl;
|
|
#ifdef _WIN32
|
|
system(cmd.c_str());
|
|
#else
|
|
std::vector<std::string> parts_str;
|
|
std::vector<const char*> parts_raw;
|
|
std::stringstream in(cmd);
|
|
for (std::string part; std::getline(in, part, ' ');)
|
|
{
|
|
if (part.empty())
|
|
continue;
|
|
parts_str.push_back(part);
|
|
}
|
|
for (const auto& part : parts_str)
|
|
{
|
|
parts_raw.push_back(part.c_str());
|
|
}
|
|
parts_raw.push_back(nullptr);
|
|
const auto pid = fork();
|
|
if (pid == -1)
|
|
{
|
|
throw std::runtime_error("failed to fork");
|
|
}
|
|
else if (pid == 0)
|
|
{
|
|
char* const* args = (char* const*)parts_raw.data();
|
|
const auto result = execv(parts_raw[0], args);
|
|
if (result)
|
|
{
|
|
std::cout << "failed: " << result << std::endl;
|
|
}
|
|
else
|
|
{
|
|
std::cout << "ok" << std::endl;
|
|
}
|
|
exit(result);
|
|
}
|
|
else
|
|
{
|
|
waitpid(pid, 0, 0);
|
|
}
|
|
#endif
|
|
}
|
|
#endif
|
|
|
|
#ifdef __linux__
|
|
struct NLSocket
|
|
{
|
|
NLSocket() : fd(socket(AF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE))
|
|
{
|
|
if (fd == -1)
|
|
throw std::runtime_error("failed to make netlink socket");
|
|
}
|
|
|
|
~NLSocket()
|
|
{
|
|
if (fd != -1)
|
|
close(fd);
|
|
}
|
|
|
|
const int fd;
|
|
};
|
|
|
|
/* Helper structure for ip address data and attributes */
|
|
typedef struct
|
|
{
|
|
char family;
|
|
char bitlen;
|
|
unsigned char data[sizeof(struct in6_addr)];
|
|
} _inet_addr;
|
|
|
|
/* */
|
|
|
|
#define NLMSG_TAIL(nmsg) ((struct rtattr*)(((void*)(nmsg)) + NLMSG_ALIGN((nmsg)->nlmsg_len)))
|
|
|
|
/* Add new data to rtattr */
|
|
int
|
|
rtattr_add(struct nlmsghdr* n, int maxlen, int type, const void* data, int alen)
|
|
{
|
|
int len = RTA_LENGTH(alen);
|
|
struct rtattr* rta;
|
|
|
|
if (NLMSG_ALIGN(n->nlmsg_len) + RTA_ALIGN(len) > maxlen)
|
|
{
|
|
fprintf(stderr, "rtattr_add error: message exceeded bound of %d\n", maxlen);
|
|
return -1;
|
|
}
|
|
|
|
rta = NLMSG_TAIL(n);
|
|
rta->rta_type = type;
|
|
rta->rta_len = len;
|
|
|
|
if (alen)
|
|
{
|
|
memcpy(RTA_DATA(rta), data, alen);
|
|
}
|
|
|
|
n->nlmsg_len = NLMSG_ALIGN(n->nlmsg_len) + RTA_ALIGN(len);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
do_route(int sock, int cmd, int flags, _inet_addr* dst, _inet_addr* gw, int def_gw, int if_idx)
|
|
{
|
|
struct
|
|
{
|
|
struct nlmsghdr n;
|
|
struct rtmsg r;
|
|
char buf[4096];
|
|
} nl_request{};
|
|
|
|
/* Initialize request structure */
|
|
nl_request.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
|
|
nl_request.n.nlmsg_flags = NLM_F_REQUEST | flags;
|
|
nl_request.n.nlmsg_type = cmd;
|
|
nl_request.r.rtm_family = dst->family;
|
|
nl_request.r.rtm_table = RT_TABLE_MAIN;
|
|
nl_request.r.rtm_scope = RT_SCOPE_NOWHERE;
|
|
|
|
/* Set additional flags if NOT deleting route */
|
|
if (cmd != RTM_DELROUTE)
|
|
{
|
|
nl_request.r.rtm_protocol = RTPROT_BOOT;
|
|
nl_request.r.rtm_type = RTN_UNICAST;
|
|
}
|
|
|
|
nl_request.r.rtm_family = dst->family;
|
|
nl_request.r.rtm_dst_len = dst->bitlen;
|
|
|
|
/* Select scope, for simplicity we supports here only IPv6 and IPv4 */
|
|
if (nl_request.r.rtm_family == AF_INET6)
|
|
{
|
|
nl_request.r.rtm_scope = RT_SCOPE_UNIVERSE;
|
|
}
|
|
else
|
|
{
|
|
nl_request.r.rtm_scope = RT_SCOPE_LINK;
|
|
}
|
|
|
|
/* Set gateway */
|
|
if (gw->bitlen != 0)
|
|
{
|
|
rtattr_add(&nl_request.n, sizeof(nl_request), RTA_GATEWAY, &gw->data, gw->bitlen / 8);
|
|
nl_request.r.rtm_scope = 0;
|
|
nl_request.r.rtm_family = gw->family;
|
|
}
|
|
|
|
/* Don't set destination and interface in case of default gateways */
|
|
if (!def_gw)
|
|
{
|
|
/* Set destination network */
|
|
rtattr_add(
|
|
&nl_request.n, sizeof(nl_request), /*RTA_NEWDST*/ RTA_DST, &dst->data, dst->bitlen / 8);
|
|
|
|
/* Set interface */
|
|
rtattr_add(&nl_request.n, sizeof(nl_request), RTA_OIF, &if_idx, sizeof(int));
|
|
}
|
|
|
|
/* Send message to the netlink */
|
|
return send(sock, &nl_request, sizeof(nl_request), 0);
|
|
}
|
|
|
|
int
|
|
read_addr(char* addr, _inet_addr* res)
|
|
{
|
|
if (strchr(addr, ':'))
|
|
{
|
|
res->family = AF_INET6;
|
|
res->bitlen = 128;
|
|
}
|
|
else
|
|
{
|
|
res->family = AF_INET;
|
|
res->bitlen = 32;
|
|
}
|
|
return inet_pton(res->family, addr, res->data);
|
|
}
|
|
|
|
#endif
|
|
|
|
void
|
|
AddRoute(std::string ip, std::string gateway)
|
|
{
|
|
#ifdef __linux__
|
|
NLSocket sock;
|
|
int default_gw = 0;
|
|
int if_idx = 0;
|
|
_inet_addr to_addr{};
|
|
_inet_addr gw_addr{};
|
|
int nl_cmd = RTM_NEWROUTE;
|
|
int nl_flags = NLM_F_CREATE | NLM_F_EXCL;
|
|
read_addr(gateway.c_str(), &gw_addr);
|
|
read_addr(ip.c_str(), &to_addr);
|
|
do_route(sock.fd, nl_cmd, nl_flags, &to_addr, &gw_addr, default_gw, if_idx);
|
|
#else
|
|
std::stringstream ss;
|
|
#if _WIN32
|
|
ss << "route ADD " << ip << " MASK 255.255.255.255 " << gateway << " METRIC 2";
|
|
#elif __APPLE__
|
|
ss << "route -n add -host " << ip << " " << gateway;
|
|
#else
|
|
#error unsupported platform
|
|
#endif
|
|
Execute(ss.str());
|
|
#endif
|
|
}
|
|
|
|
void
|
|
DelRoute(std::string ip, std::string gateway)
|
|
{
|
|
#ifdef __linux__
|
|
NLSocket sock;
|
|
int default_gw = 0;
|
|
int if_idx = 0;
|
|
_inet_addr to_addr{};
|
|
_inet_addr gw_addr{};
|
|
int nl_cmd = RTM_DELROUTE;
|
|
int nl_flags = 0;
|
|
read_addr(gateway.c_str(), &gw_addr);
|
|
read_addr(ip.c_str(), &to_addr);
|
|
do_route(sock.fd, nl_cmd, nl_flags, &to_addr, &gw_addr, default_gw, if_idx);
|
|
#else
|
|
std::stringstream ss;
|
|
#if _WIN32
|
|
ss << "route DELETE " << ip << " MASK 255.255.255.255 " << gateway << " METRIC 2";
|
|
#elif __APPLE__
|
|
ss << "route -n delete -host " << ip << " " << gateway;
|
|
#else
|
|
#error unsupported platform
|
|
#endif
|
|
Execute(ss.str());
|
|
#endif
|
|
}
|
|
|
|
void
|
|
AddDefaultRouteViaInterface(std::string ifname)
|
|
{
|
|
#ifdef __linux__
|
|
NLSocket sock;
|
|
int default_gw = 1;
|
|
int if_idx = if_nametoindex(ifname.c_str());
|
|
_inet_addr to_addr{};
|
|
_inet_addr gw_addr{};
|
|
int nl_cmd = RTM_NEWROUTE;
|
|
int nl_flags = NLM_F_CREATE | NLM_F_EXCL;
|
|
do_route(sock.fd, nl_cmd, nl_flags, &to_addr, &gw_addr, default_gw, if_idx);
|
|
#elif _WIN32
|
|
ifname.back()++;
|
|
Execute("route ADD 0.0.0.0 MASK 128.0.0.0 " + ifname);
|
|
Execute("route ADD 128.0.0.0 MASK 128.0.0.0 " + ifname);
|
|
#elif __APPLE__
|
|
Execute("route -cloning add -net 0.0.0.0 -netmask 0.0.0.0 -interface " + ifname);
|
|
#else
|
|
#error unsupported platform
|
|
#endif
|
|
}
|
|
|
|
void
|
|
DelDefaultRouteViaInterface(std::string ifname)
|
|
{
|
|
#ifdef __linux__
|
|
|
|
NLSocket sock;
|
|
int default_gw = 1;
|
|
int if_idx = if_nametoindex(ifname.c_str());
|
|
_inet_addr to_addr{};
|
|
_inet_addr gw_addr{};
|
|
int nl_cmd = RTM_DELROUTE;
|
|
int nl_flags = 0;
|
|
do_route(sock.fd, nl_cmd, nl_flags, &to_addr, &gw_addr, default_gw, if_idx);
|
|
#elif _WIN32
|
|
ifname.back()++;
|
|
Execute("route DELETE 0.0.0.0 MASK 128.0.0.0 " + ifname);
|
|
Execute("route DELETE 128.0.0.0 MASK 128.0.0.0 " + ifname);
|
|
#elif __APPLE__
|
|
Execute("route -cloning delete -net 0.0.0.0 -netmask 0.0.0.0 -interface " + ifname);
|
|
#else
|
|
#error unsupported platform
|
|
#endif
|
|
}
|
|
|
|
std::vector<std::string>
|
|
GetGatewaysNotOnInterface(std::string ifname)
|
|
{
|
|
std::vector<std::string> gateways;
|
|
#ifdef __linux__
|
|
|
|
FILE* p = popen("ip route", "r");
|
|
if (p == nullptr)
|
|
return gateways;
|
|
char* line = nullptr;
|
|
size_t len = 0;
|
|
ssize_t read = 0;
|
|
while ((read = getline(&line, &len, p)) != -1)
|
|
{
|
|
std::string line_str(line, len);
|
|
std::vector<std::string> words;
|
|
std::istringstream instr(line_str);
|
|
for (std::string word; std::getline(instr, word, ' ');)
|
|
{
|
|
words.emplace_back(std::move(word));
|
|
}
|
|
if (words[0] == "default" and words[1] == "via" and words[3] == "dev" and words[4] != ifname)
|
|
{
|
|
gateways.emplace_back(std::move(words[2]));
|
|
}
|
|
}
|
|
pclose(p);
|
|
return gateways;
|
|
#elif _WIN32
|
|
#define MALLOC(x) HeapAlloc(GetProcessHeap(), 0, (x))
|
|
#define FREE(x) HeapFree(GetProcessHeap(), 0, (x))
|
|
PMIB_IPFORWARDTABLE pIpForwardTable;
|
|
DWORD dwSize = 0;
|
|
DWORD dwRetVal = 0;
|
|
|
|
pIpForwardTable = (MIB_IPFORWARDTABLE*)MALLOC(sizeof(MIB_IPFORWARDTABLE));
|
|
if (pIpForwardTable == nullptr)
|
|
return gateways;
|
|
|
|
if (GetIpForwardTable(pIpForwardTable, &dwSize, 0) == ERROR_INSUFFICIENT_BUFFER)
|
|
{
|
|
FREE(pIpForwardTable);
|
|
pIpForwardTable = (MIB_IPFORWARDTABLE*)MALLOC(dwSize);
|
|
if (pIpForwardTable == nullptr)
|
|
{
|
|
return gateways;
|
|
}
|
|
}
|
|
|
|
if ((dwRetVal = GetIpForwardTable(pIpForwardTable, &dwSize, 0)) == NO_ERROR)
|
|
{
|
|
for (int i = 0; i < (int)pIpForwardTable->dwNumEntries; i++)
|
|
{
|
|
struct in_addr gateway, interface_addr;
|
|
gateway.S_un.S_addr = (u_long)pIpForwardTable->table[i].dwForwardDest;
|
|
interface_addr.S_un.S_addr = (u_long)pIpForwardTable->table[i].dwForwardNextHop;
|
|
std::array<char, 128> interface_str{};
|
|
StringCchCopy(interface_str.data(), interface_str.size(), inet_ntoa(interface_addr));
|
|
std::string interface_name{interface_str.data()};
|
|
if ((!gateway.S_un.S_addr) and interface_name != ifname)
|
|
{
|
|
gateways.push_back(std::move(interface_name));
|
|
}
|
|
}
|
|
}
|
|
FREE(pIpForwardTable);
|
|
#undef MALLOC
|
|
#undef FREE
|
|
return gateways;
|
|
#elif __APPLE__
|
|
const auto maybe = llarp::GetIFAddr(ifname);
|
|
if (not maybe.has_value())
|
|
return gateways;
|
|
const auto interface = maybe->toString();
|
|
// mac os is so godawful man
|
|
FILE* p = popen("netstat -rn -f inet", "r");
|
|
if (p == nullptr)
|
|
return gateways;
|
|
char* line = nullptr;
|
|
size_t len = 0;
|
|
ssize_t read = 0;
|
|
while ((read = getline(&line, &len, p)) != -1)
|
|
{
|
|
std::string line_str(line, len);
|
|
if (line_str.find("default") == 0)
|
|
{
|
|
line_str = line_str.substr(7);
|
|
while (line_str[0] == ' ')
|
|
{
|
|
line_str = line_str.substr(1);
|
|
}
|
|
const auto pos = line_str.find(" ");
|
|
if (pos != std::string::npos)
|
|
{
|
|
auto gateway = line_str.substr(0, pos);
|
|
if (gateway != interface)
|
|
gateways.emplace_back(std::move(gateway));
|
|
}
|
|
}
|
|
}
|
|
pclose(p);
|
|
return gateways;
|
|
#else
|
|
#error unsupported platform
|
|
#endif
|
|
}
|
|
|
|
} // namespace llarp::net
|