| // Copyright 2021 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef NET_DNS_DNS_CONFIG_SERVICE_LINUX_H_ |
| #define NET_DNS_DNS_CONFIG_SERVICE_LINUX_H_ |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "base/check.h" |
| #include "base/gtest_prod_util.h" |
| #include "net/base/net_export.h" |
| #include "net/dns/dns_config_service.h" |
| #include "net/dns/nsswitch_reader.h" |
| #include "net/dns/public/resolv_reader.h" |
| |
| namespace net { |
| |
| // Use DnsConfigService::CreateSystemService to use it outside of tests. |
| namespace internal { |
| |
| // Service for reading and watching Linux DNS settings. |
| // This object is not thread-safe and methods may perform blocking I/O so |
| // methods must be called on a sequence that allows blocking (i.e. |
| // base::MayBlock). It may be constructed on a different sequence than which |
| // it's later called on. WatchConfig() must be called prior to ReadConfig(). |
| class NET_EXPORT_PRIVATE DnsConfigServiceLinux : public DnsConfigService { |
| public: |
| DnsConfigServiceLinux(); |
| ~DnsConfigServiceLinux() override; |
| |
| DnsConfigServiceLinux(const DnsConfigServiceLinux&) = delete; |
| DnsConfigServiceLinux& operator=(const DnsConfigServiceLinux&) = delete; |
| |
| void set_resolv_reader_for_testing( |
| std::unique_ptr<ResolvReader> resolv_reader) { |
| DCHECK(!config_reader_); // Need to call before first read. |
| DCHECK(resolv_reader); |
| resolv_reader_ = std::move(resolv_reader); |
| } |
| |
| void set_nsswitch_reader_for_testing( |
| std::unique_ptr<NsswitchReader> nsswitch_reader) { |
| DCHECK(!config_reader_); // Need to call before first read. |
| DCHECK(nsswitch_reader); |
| nsswitch_reader_ = std::move(nsswitch_reader); |
| } |
| |
| protected: |
| // DnsConfigService: |
| void ReadConfigNow() override; |
| bool StartWatching() override; |
| |
| // Create |config_reader_|. |
| void CreateReader(); |
| |
| private: |
| FRIEND_TEST_ALL_PREFIXES(DnsConfigServiceLinuxTest, |
| ChangeConfigMultipleTimes); |
| class Watcher; |
| class ConfigReader; |
| |
| std::unique_ptr<ResolvReader> resolv_reader_ = |
| std::make_unique<ResolvReader>(); |
| std::unique_ptr<NsswitchReader> nsswitch_reader_ = |
| std::make_unique<NsswitchReader>(); |
| |
| std::unique_ptr<Watcher> watcher_; |
| std::unique_ptr<ConfigReader> config_reader_; |
| }; |
| |
| } // namespace internal |
| |
| } // namespace net |
| |
| #endif // NET_DNS_DNS_CONFIG_SERVICE_LINUX_H_ |