Martin Michelsen
45 lines
1.4 KiB
C++
45 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include <event2/event.h>
|
|
|
|
#include <memory>
|
|
#include <set>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
#include "IPV4RangeSet.hh"
|
|
|
|
class DNSServer {
|
|
public:
|
|
DNSServer(
|
|
std::shared_ptr<struct event_base> base,
|
|
uint32_t local_connect_address,
|
|
uint32_t external_connect_address,
|
|
std::shared_ptr<const IPV4RangeSet> banned_ipv4_ranges);
|
|
DNSServer(const DNSServer&) = delete;
|
|
DNSServer(DNSServer&&) = delete;
|
|
virtual ~DNSServer();
|
|
|
|
inline void set_banned_ipv4_ranges(std::shared_ptr<const IPV4RangeSet> banned_ipv4_ranges) {
|
|
this->banned_ipv4_ranges = banned_ipv4_ranges;
|
|
}
|
|
|
|
void listen(const std::string& socket_path);
|
|
void listen(const std::string& addr, int port);
|
|
void listen(int port);
|
|
void add_socket(int fd);
|
|
|
|
static std::string response_for_query(const void* vdata, size_t size, uint32_t resolved_address);
|
|
static std::string response_for_query(const std::string& query, uint32_t resolved_address);
|
|
|
|
private:
|
|
std::shared_ptr<struct event_base> base;
|
|
std::unordered_map<int, std::unique_ptr<struct event, void (*)(struct event*)>> fd_to_receive_event;
|
|
uint32_t local_connect_address;
|
|
uint32_t external_connect_address;
|
|
std::shared_ptr<const IPV4RangeSet> banned_ipv4_ranges;
|
|
|
|
static void dispatch_on_receive_message(evutil_socket_t fd, short events, void* ctx);
|
|
void on_receive_message(int fd, short event);
|
|
};
|