initial commit

DNSServer.hh

@@ -0,0 +1,37 @@
+#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);
+};