diff options
-rw-r--r-- | Userland/Services/LookupServer/LookupServer.cpp | 24 | ||||
-rw-r--r-- | Userland/Services/LookupServer/LookupServer.h | 2 | ||||
-rw-r--r-- | Userland/Services/LookupServer/main.cpp | 5 |
3 files changed, 31 insertions, 0 deletions
diff --git a/Userland/Services/LookupServer/LookupServer.cpp b/Userland/Services/LookupServer/LookupServer.cpp index 692b885d9a..f0e294e908 100644 --- a/Userland/Services/LookupServer/LookupServer.cpp +++ b/Userland/Services/LookupServer/LookupServer.cpp @@ -45,6 +45,29 @@ LookupServer::LookupServer() load_etc_hosts(); + auto maybe_file_watcher = Core::FileWatcher::create(); + // NOTE: If this happens during startup, something is very wrong. + if (maybe_file_watcher.is_error()) { + dbgln("Core::FileWatcher::create(): {}", maybe_file_watcher.error()); + VERIFY_NOT_REACHED(); + } + m_file_watcher = maybe_file_watcher.release_value(); + + m_file_watcher->on_change = [this](auto&) { + dbgln("Reloading '/etc/hosts' because it was changed."); + load_etc_hosts(); + }; + + auto result = m_file_watcher->add_watch("/etc/hosts", Core::FileWatcherEvent::Type::ContentModified | Core::FileWatcherEvent::Type::Deleted); + // NOTE: If this happens during startup, something is very wrong. + if (result.is_error()) { + dbgln("Core::FileWatcher::add_watch(): {}", result.error()); + VERIFY_NOT_REACHED(); + } else if (!result.value()) { + dbgln("Core::FileWatcher::add_watch(): {}", result.value()); + VERIFY_NOT_REACHED(); + } + if (config->read_bool_entry("DNS", "EnableServer")) { m_dns_server = DNSServer::construct(this); // TODO: drop root privileges here. @@ -68,6 +91,7 @@ LookupServer::LookupServer() void LookupServer::load_etc_hosts() { + m_etc_hosts.clear(); auto add_answer = [this](const DNSName& name, DNSRecordType record_type, String data) { auto it = m_etc_hosts.find(name); if (it == m_etc_hosts.end()) { diff --git a/Userland/Services/LookupServer/LookupServer.h b/Userland/Services/LookupServer/LookupServer.h index c948f2aa07..2e42d5e290 100644 --- a/Userland/Services/LookupServer/LookupServer.h +++ b/Userland/Services/LookupServer/LookupServer.h @@ -10,6 +10,7 @@ #include "DNSPacket.h" #include "DNSServer.h" #include "MulticastDNS.h" +#include <LibCore/FileWatcher.h> #include <LibCore/Object.h> namespace LookupServer { @@ -35,6 +36,7 @@ private: RefPtr<DNSServer> m_dns_server; RefPtr<MulticastDNS> m_mdns; Vector<String> m_nameservers; + RefPtr<Core::FileWatcher> m_file_watcher; HashMap<DNSName, Vector<DNSAnswer>, DNSName::Traits> m_etc_hosts; HashMap<DNSName, Vector<DNSAnswer>, DNSName::Traits> m_lookup_cache; }; diff --git a/Userland/Services/LookupServer/main.cpp b/Userland/Services/LookupServer/main.cpp index d4f0a89bc4..574b441b9d 100644 --- a/Userland/Services/LookupServer/main.cpp +++ b/Userland/Services/LookupServer/main.cpp @@ -30,6 +30,11 @@ int main([[maybe_unused]] int argc, [[maybe_unused]] char** argv) return 1; } + if (unveil("/etc/hosts", "r") < 0) { + perror("unveil"); + return 1; + } + if (unveil(nullptr, nullptr) < 0) { perror("unveil"); return 1; |