Martin Michelsen
38 lines
840 B
C++
38 lines
840 B
C++
#pragma once
|
|
|
|
#include <atomic>
|
|
#include <thread>
|
|
#include <string>
|
|
#include <set>
|
|
|
|
|
|
class DNSServer {
|
|
public:
|
|
DNSServer(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);
|
|
|
|
virtual void start();
|
|
virtual void schedule_stop();
|
|
virtual void wait_for_stop();
|
|
|
|
private:
|
|
std::atomic<bool> should_exit;
|
|
std::thread t;
|
|
|
|
std::set<int> fds;
|
|
|
|
uint32_t local_connect_address;
|
|
uint32_t external_connect_address;
|
|
|
|
void run_thread();
|
|
static std::string build_response(const std::string& input,
|
|
uint32_t connect_address);
|
|
};
|