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.
104 lines
2.7 KiB
C++
104 lines
2.7 KiB
C++
#include "route_manager.hpp"
|
|
#include <llarp/handlers/tun.hpp>
|
|
#include <llarp/service/context.hpp>
|
|
#include <llarp.hpp>
|
|
#include <memory>
|
|
|
|
namespace llarp::apple
|
|
{
|
|
void
|
|
RouteManager::check_trampoline(bool enable)
|
|
{
|
|
if (trampoline_active == enable)
|
|
return;
|
|
auto router = context.router;
|
|
if (!router)
|
|
{
|
|
LogError("Cannot reconfigure to use DNS trampoline: no router");
|
|
return;
|
|
}
|
|
|
|
std::shared_ptr<llarp::handlers::TunEndpoint> tun;
|
|
router->hiddenServiceContext().ForEachService([&tun](const auto& /*name*/, const auto ep) {
|
|
tun = std::dynamic_pointer_cast<llarp::handlers::TunEndpoint>(ep);
|
|
return !tun;
|
|
});
|
|
|
|
if (!tun)
|
|
{
|
|
LogError("Cannot reconfigure to use DNS trampoline: no tun endpoint found (!?)");
|
|
return;
|
|
}
|
|
|
|
if (enable)
|
|
tun->ReconfigureDNS({SockAddr{127, 0, 0, 1, {dns_trampoline_port}}});
|
|
else
|
|
tun->ReconfigureDNS(router->GetConfig()->dns.m_upstreamDNS);
|
|
|
|
trampoline_active = enable;
|
|
}
|
|
|
|
void
|
|
RouteManager::AddDefaultRouteViaInterface(vpn::NetworkInterface&)
|
|
{
|
|
check_trampoline(true);
|
|
if (callback_context and route_callbacks.add_default_route)
|
|
route_callbacks.add_default_route(callback_context);
|
|
}
|
|
|
|
void
|
|
RouteManager::DelDefaultRouteViaInterface(vpn::NetworkInterface&)
|
|
{
|
|
check_trampoline(false);
|
|
if (callback_context and route_callbacks.del_default_route)
|
|
route_callbacks.del_default_route(callback_context);
|
|
}
|
|
|
|
void
|
|
RouteManager::AddRouteViaInterface(vpn::NetworkInterface&, IPRange range)
|
|
{
|
|
check_trampoline(true);
|
|
if (callback_context)
|
|
{
|
|
if (range.IsV4())
|
|
{
|
|
if (route_callbacks.add_ipv4_route)
|
|
route_callbacks.add_ipv4_route(
|
|
range.BaseAddressString().c_str(),
|
|
net::TruncateV6(range.netmask_bits).ToString().c_str(),
|
|
callback_context);
|
|
}
|
|
else
|
|
{
|
|
if (route_callbacks.add_ipv6_route)
|
|
route_callbacks.add_ipv6_route(
|
|
range.BaseAddressString().c_str(), range.HostmaskBits(), callback_context);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
RouteManager::DelRouteViaInterface(vpn::NetworkInterface&, IPRange range)
|
|
{
|
|
check_trampoline(false);
|
|
if (callback_context)
|
|
{
|
|
if (range.IsV4())
|
|
{
|
|
if (route_callbacks.del_ipv4_route)
|
|
route_callbacks.del_ipv4_route(
|
|
range.BaseAddressString().c_str(),
|
|
net::TruncateV6(range.netmask_bits).ToString().c_str(),
|
|
callback_context);
|
|
}
|
|
else
|
|
{
|
|
if (route_callbacks.del_ipv6_route)
|
|
route_callbacks.del_ipv6_route(
|
|
range.BaseAddressString().c_str(), range.HostmaskBits(), callback_context);
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace llarp::apple
|