RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 17:57:35 +00:00
Code Issues Activity Actions

LookupServer: Watch /etc/hosts for changes during runtime

Browse source 
This adds a FileWatcher to the LookupServer which watches '/etc/hosts'
for changes during runtime and reloads its contents. If the file is
deleted, m_etc_hosts will be cleared.

Since we now need to access '/etc/hosts' later during runtime, it needs
to be unveiled with read permissions.
This commit is contained in:
Max Wipfli 2021-06-07 20:26:09 +02:00 • committed by Ali Mohammad Pur
parent 7b39db3bf4
commit 4efccbd030
3 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

24
Userland/Services/LookupServer/LookupServer.cpp
View file

@ -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()) {