mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-07 15:20:31 +00:00
4c630e0437
- Previous android java and jni code updated to work, but with much love still needed to make it work nicely, e.g. handling when the VPN is turned off. - DNS handling refactored to allow android to intercept and handle DNS requests as we can't set the system DNS to use a high port (and apparently Chrome ignores system DNS settings anyway) - add packet router structure to allow separate handling of specific intercepted traffic, e.g. UDP traffic to port 53 gets handled by our DNS handler rather than being naively forwarded as exit traffic. - For now, android lokinet is exit-only and hard-coded to use exit.loki as its exit. The exit will be configurable before release, but allowing to not use exit-only mode is more of a challenge. - some old gitignore remnants which were matching to things we don't want them to (and are no longer relevant) removed - some minor changes to CI configuration
97 lines
2.3 KiB
C++
97 lines
2.3 KiB
C++
#ifndef LLARP_DNS_SERVER_HPP
|
|
#define LLARP_DNS_SERVER_HPP
|
|
|
|
#include <dns/message.hpp>
|
|
#include <ev/ev.h>
|
|
#include <net/net.hpp>
|
|
#include <util/thread/logic.hpp>
|
|
#include <dns/unbound_resolver.hpp>
|
|
|
|
#include <unordered_map>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace dns
|
|
{
|
|
/// handler of dns query hooking
|
|
struct IQueryHandler
|
|
{
|
|
virtual ~IQueryHandler() = default;
|
|
|
|
/// return true if we should hook this message
|
|
virtual bool
|
|
ShouldHookDNSMessage(const Message& msg) const = 0;
|
|
|
|
/// handle a hooked message
|
|
virtual bool
|
|
HandleHookedDNSMessage(Message query, std::function<void(Message)> sendReply) = 0;
|
|
};
|
|
|
|
struct PacketHandler : public std::enable_shared_from_this<PacketHandler>
|
|
{
|
|
using Logic_ptr = std::shared_ptr<Logic>;
|
|
using Buffer_t = std::vector<uint8_t>;
|
|
|
|
explicit PacketHandler(Logic_ptr logic, IQueryHandler* handler);
|
|
|
|
virtual ~PacketHandler() = default;
|
|
|
|
virtual bool
|
|
Start(SockAddr localaddr, std::vector<IpAddress> upstreamResolvers);
|
|
|
|
void
|
|
Stop();
|
|
|
|
void
|
|
Restart();
|
|
|
|
void
|
|
HandlePacket(SockAddr resolver, SockAddr from, Buffer_t buf);
|
|
|
|
bool
|
|
ShouldHandlePacket(SockAddr to, SockAddr from, Buffer_t buf) const;
|
|
|
|
protected:
|
|
virtual void
|
|
SendServerMessageBufferTo(SockAddr from, SockAddr to, Buffer_t buf) = 0;
|
|
|
|
private:
|
|
void
|
|
HandleUpstreamFailure(SockAddr from, SockAddr to, Message msg);
|
|
|
|
bool
|
|
SetupUnboundResolver(std::vector<IpAddress> resolvers);
|
|
|
|
IQueryHandler* const m_QueryHandler;
|
|
std::set<IpAddress> m_Resolvers;
|
|
std::shared_ptr<UnboundResolver> m_UnboundResolver;
|
|
Logic_ptr m_Logic;
|
|
};
|
|
|
|
struct Proxy : public PacketHandler
|
|
{
|
|
using Logic_ptr = std::shared_ptr<Logic>;
|
|
explicit Proxy(llarp_ev_loop_ptr loop, Logic_ptr logic, IQueryHandler* handler);
|
|
|
|
bool
|
|
Start(SockAddr localaddr, std::vector<IpAddress> resolvers) override;
|
|
|
|
using Buffer_t = std::vector<uint8_t>;
|
|
|
|
protected:
|
|
void
|
|
SendServerMessageBufferTo(SockAddr from, SockAddr to, Buffer_t buf) override;
|
|
|
|
private:
|
|
static void
|
|
HandleUDP(llarp_udp_io*, const SockAddr&, ManagedBuffer);
|
|
|
|
private:
|
|
llarp_udp_io m_Server;
|
|
llarp_ev_loop_ptr m_Loop;
|
|
};
|
|
} // namespace dns
|
|
} // namespace llarp
|
|
|
|
#endif
|