Martin Michelsen
38 lines
1.1 KiB
C++
38 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <event2/event.h>
|
|
|
|
#include <memory>
|
|
#include <set>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
class DNSServer {
|
|
public:
|
|
DNSServer(std::shared_ptr<struct event_base> base,
|
|
uint32_t local_connect_address, uint32_t external_connect_address);
|
|
DNSServer(const DNSServer&) = delete;
|
|
DNSServer(DNSServer&&) = delete;
|
|
virtual ~DNSServer();
|
|
|
|
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;
|
|
|
|
static void dispatch_on_receive_message(evutil_socket_t fd, short events,
|
|
void* ctx);
|
|
void on_receive_message(int fd, short event);
|
|
};
|