mirror of
https://github.com/oxen-io/lokinet
synced 2023-12-14 06:53:00 +01:00
- ReconfigureDNS wasn't returning the old servers; made it void instead (the Apple code can just store a copy of the original upstream servers instead). - Reconfiguring DNS reset the unbound context but didn't replace it, so a Down()/Up() would crash. - Simplify Resolver() destructor to just call Down(), and make it final just so that no one tries to inherit from us (so that calling a virtual function from the destructor is safe). - Rename CancelPendingQueries() to Down(); the former cancelled but also shut down the object, so the name seemed a bit misleading. - Rename SetInternalState in Resolver_Base to ResetResolver, so that we aren't conflicting with ResetInternalState from Endpoint (which was a problem because TunEndpoint inherited from both; it could be resolved through the different argument type if we removed the default, but that seems gross). - Make Resolver use a bare unbound context pointer rather than a shared_ptr; since Resolver (now) entirely manages it already we don't need an extra management layer, and it saves a bunch of `.get()`s.
102 lines
2.7 KiB
C++
102 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
|