network_service_unittest.cc 64 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "services/network/network_service.h"
  5. #include <memory>
  6. #include <utility>
  7. #include "base/base64.h"
  8. #include "base/bind.h"
  9. #include "base/command_line.h"
  10. #include "base/containers/span.h"
  11. #include "base/files/file_util.h"
  12. #include "base/files/scoped_temp_dir.h"
  13. #include "base/json/json_file_value_serializer.h"
  14. #include "base/path_service.h"
  15. #include "base/ranges/algorithm.h"
  16. #include "base/run_loop.h"
  17. #include "base/strings/escape.h"
  18. #include "base/strings/string_util.h"
  19. #include "base/strings/stringprintf.h"
  20. #include "base/test/bind.h"
  21. #include "base/test/scoped_feature_list.h"
  22. #include "base/test/task_environment.h"
  23. #include "base/threading/thread_task_runner_handle.h"
  24. #include "build/build_config.h"
  25. #include "build/chromeos_buildflags.h"
  26. #include "net/base/ip_address.h"
  27. #include "net/base/ip_endpoint.h"
  28. #include "net/base/mock_network_change_notifier.h"
  29. #include "net/base/url_util.h"
  30. #include "net/dns/dns_client.h"
  31. #include "net/dns/dns_config.h"
  32. #include "net/dns/dns_config_service.h"
  33. #include "net/dns/dns_test_util.h"
  34. #include "net/dns/host_resolver.h"
  35. #include "net/dns/host_resolver_manager.h"
  36. #include "net/dns/public/dns_over_https_config.h"
  37. #include "net/dns/public/dns_protocol.h"
  38. #include "net/dns/public/doh_provider_entry.h"
  39. #include "net/http/http_auth_handler_factory.h"
  40. #include "net/http/http_auth_scheme.h"
  41. #include "net/http/http_network_session.h"
  42. #include "net/http/http_transaction_factory.h"
  43. #include "net/http/transport_security_state.h"
  44. #include "net/net_buildflags.h"
  45. #include "net/socket/client_socket_pool_manager.h"
  46. #include "net/test/embedded_test_server/embedded_test_server.h"
  47. #include "net/test/embedded_test_server/http_request.h"
  48. #include "net/test/embedded_test_server/http_response.h"
  49. #include "net/test/spawned_test_server/spawned_test_server.h"
  50. #include "net/test/test_data_directory.h"
  51. #include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
  52. #include "net/url_request/url_request_context.h"
  53. #include "services/network/network_context.h"
  54. #include "services/network/public/cpp/features.h"
  55. #include "services/network/public/cpp/network_switches.h"
  56. #include "services/network/public/mojom/net_log.mojom.h"
  57. #include "services/network/public/mojom/network_change_manager.mojom.h"
  58. #include "services/network/public/mojom/network_context.mojom.h"
  59. #include "services/network/public/mojom/network_service.mojom.h"
  60. #include "services/network/test/fake_test_cert_verifier_params_factory.h"
  61. #include "services/network/test/test_network_context_client.h"
  62. #include "services/network/test/test_url_loader_client.h"
  63. #include "services/network/test/test_url_loader_network_observer.h"
  64. #include "services/network/test/test_utils.h"
  65. #include "testing/gmock/include/gmock/gmock.h"
  66. #include "testing/gtest/include/gtest/gtest.h"
  67. #include "third_party/abseil-cpp/absl/types/optional.h"
  68. #include "url/gurl.h"
  69. #if BUILDFLAG(USE_KERBEROS)
  70. #include "net/http/http_auth_handler_negotiate.h"
  71. #endif
  72. namespace network {
  73. namespace {
  74. const base::FilePath::CharType kServicesTestData[] =
  75. FILE_PATH_LITERAL("services/test/data");
  76. // Returns a new URL with key=value pair added to the query.
  77. GURL AddQuery(const GURL& url,
  78. const std::string& key,
  79. const std::string& value) {
  80. return GURL(url.spec() + (url.has_query() ? "&" : "?") + key + "=" +
  81. base::EscapeQueryParamValue(value, false));
  82. }
  83. mojom::NetworkContextParamsPtr CreateContextParams() {
  84. mojom::NetworkContextParamsPtr params =
  85. CreateNetworkContextParamsForTesting();
  86. // Use a dummy CertVerifier that always passes cert verification, since
  87. // these unittests don't need to test CertVerifier behavior.
  88. params->cert_verifier_params =
  89. FakeTestCertVerifierParamsFactory::GetCertVerifierParams();
  90. return params;
  91. }
  92. class NetworkServiceTest : public testing::Test {
  93. public:
  94. explicit NetworkServiceTest(
  95. base::test::TaskEnvironment::TimeSource time_source =
  96. base::test::TaskEnvironment::TimeSource::MOCK_TIME)
  97. : task_environment_(base::test::TaskEnvironment::MainThreadType::IO,
  98. time_source),
  99. service_(NetworkService::CreateForTesting()) {}
  100. ~NetworkServiceTest() override {}
  101. base::test::TaskEnvironment* task_environment() { return &task_environment_; }
  102. NetworkService* service() const { return service_.get(); }
  103. void DestroyService() { service_.reset(); }
  104. private:
  105. base::test::TaskEnvironment task_environment_;
  106. std::unique_ptr<NetworkService> service_;
  107. };
  108. // Test shutdown in the case a NetworkContext is destroyed before the
  109. // NetworkService.
  110. TEST_F(NetworkServiceTest, CreateAndDestroyContext) {
  111. mojo::Remote<mojom::NetworkContext> network_context;
  112. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  113. CreateContextParams());
  114. network_context.reset();
  115. // Make sure the NetworkContext is destroyed.
  116. base::RunLoop().RunUntilIdle();
  117. }
  118. // Test shutdown in the case there is still a live NetworkContext when the
  119. // NetworkService is destroyed. The service should destroy the NetworkContext
  120. // itself.
  121. TEST_F(NetworkServiceTest, DestroyingServiceDestroysContext) {
  122. mojo::Remote<mojom::NetworkContext> network_context;
  123. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  124. CreateContextParams());
  125. base::RunLoop run_loop;
  126. network_context.set_disconnect_handler(run_loop.QuitClosure());
  127. DestroyService();
  128. // Destroying the service should destroy the context, causing a connection
  129. // error.
  130. run_loop.Run();
  131. }
  132. TEST_F(NetworkServiceTest, CreateContextWithoutChannelID) {
  133. mojom::NetworkContextParamsPtr params = CreateContextParams();
  134. mojo::Remote<mojom::NetworkContext> network_context;
  135. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  136. std::move(params));
  137. network_context.reset();
  138. // Make sure the NetworkContext is destroyed.
  139. base::RunLoop().RunUntilIdle();
  140. }
  141. TEST_F(NetworkServiceTest, AuthDefaultParams) {
  142. mojo::Remote<mojom::NetworkContext> network_context_remote;
  143. NetworkContext network_context(
  144. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  145. CreateContextParams());
  146. net::HttpAuthHandlerRegistryFactory* auth_handler_factory =
  147. reinterpret_cast<net::HttpAuthHandlerRegistryFactory*>(
  148. network_context.url_request_context()->http_auth_handler_factory());
  149. ASSERT_TRUE(auth_handler_factory);
  150. // These three factories should always be created by default. Negotiate may
  151. // or may not be created, depending on other build flags.
  152. EXPECT_TRUE(
  153. auth_handler_factory->IsSchemeAllowedForTesting(net::kBasicAuthScheme));
  154. EXPECT_TRUE(
  155. auth_handler_factory->IsSchemeAllowedForTesting(net::kDigestAuthScheme));
  156. EXPECT_TRUE(
  157. auth_handler_factory->IsSchemeAllowedForTesting(net::kNtlmAuthScheme));
  158. #if BUILDFLAG(USE_KERBEROS) && !BUILDFLAG(IS_ANDROID)
  159. ASSERT_TRUE(auth_handler_factory->IsSchemeAllowedForTesting(
  160. net::kNegotiateAuthScheme));
  161. #if BUILDFLAG(IS_POSIX) && !BUILDFLAG(IS_CHROMEOS_ASH)
  162. EXPECT_EQ("", auth_handler_factory->GetNegotiateLibraryNameForTesting());
  163. #endif
  164. #endif // BUILDFLAG(USE_KERBEROS) && !BUILDFLAG(IS_ANDROID)
  165. EXPECT_FALSE(auth_handler_factory->http_auth_preferences()
  166. ->NegotiateDisableCnameLookup());
  167. EXPECT_FALSE(
  168. auth_handler_factory->http_auth_preferences()->NegotiateEnablePort());
  169. #if BUILDFLAG(IS_POSIX) || BUILDFLAG(IS_FUCHSIA)
  170. EXPECT_TRUE(auth_handler_factory->http_auth_preferences()->NtlmV2Enabled());
  171. #endif // BUILDFLAG(IS_POSIX) || BUILDFLAG(IS_FUCHSIA)
  172. #if BUILDFLAG(IS_ANDROID)
  173. EXPECT_EQ("", auth_handler_factory->http_auth_preferences()
  174. ->AuthAndroidNegotiateAccountType());
  175. #endif // BUILDFLAG(IS_ANDROID)
  176. }
  177. TEST_F(NetworkServiceTest, AuthSchemesDynamicallyChanging) {
  178. service()->SetUpHttpAuth(mojom::HttpAuthStaticParams::New());
  179. mojo::Remote<mojom::NetworkContext> network_context_remote;
  180. NetworkContext network_context(
  181. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  182. CreateContextParams());
  183. net::HttpAuthHandlerRegistryFactory* auth_handler_factory =
  184. reinterpret_cast<net::HttpAuthHandlerRegistryFactory*>(
  185. network_context.url_request_context()->http_auth_handler_factory());
  186. ASSERT_TRUE(auth_handler_factory);
  187. EXPECT_TRUE(
  188. auth_handler_factory->IsSchemeAllowedForTesting(net::kBasicAuthScheme));
  189. EXPECT_TRUE(
  190. auth_handler_factory->IsSchemeAllowedForTesting(net::kDigestAuthScheme));
  191. EXPECT_TRUE(
  192. auth_handler_factory->IsSchemeAllowedForTesting(net::kNtlmAuthScheme));
  193. #if BUILDFLAG(USE_KERBEROS) && !BUILDFLAG(IS_ANDROID)
  194. EXPECT_TRUE(auth_handler_factory->IsSchemeAllowedForTesting(
  195. net::kNegotiateAuthScheme));
  196. #else
  197. EXPECT_FALSE(auth_handler_factory->IsSchemeAllowedForTesting(
  198. net::kNegotiateAuthScheme));
  199. #endif
  200. {
  201. mojom::HttpAuthDynamicParamsPtr auth_params =
  202. mojom::HttpAuthDynamicParams::New();
  203. auth_params->allowed_schemes = std::vector<std::string>{};
  204. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  205. EXPECT_FALSE(
  206. auth_handler_factory->IsSchemeAllowedForTesting(net::kBasicAuthScheme));
  207. EXPECT_FALSE(auth_handler_factory->IsSchemeAllowedForTesting(
  208. net::kDigestAuthScheme));
  209. EXPECT_FALSE(
  210. auth_handler_factory->IsSchemeAllowedForTesting(net::kNtlmAuthScheme));
  211. EXPECT_FALSE(auth_handler_factory->IsSchemeAllowedForTesting(
  212. net::kNegotiateAuthScheme));
  213. }
  214. {
  215. mojom::HttpAuthDynamicParamsPtr auth_params =
  216. mojom::HttpAuthDynamicParams::New();
  217. auth_params->allowed_schemes =
  218. std::vector<std::string>{net::kDigestAuthScheme, net::kNtlmAuthScheme};
  219. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  220. EXPECT_FALSE(
  221. auth_handler_factory->IsSchemeAllowedForTesting(net::kBasicAuthScheme));
  222. EXPECT_TRUE(auth_handler_factory->IsSchemeAllowedForTesting(
  223. net::kDigestAuthScheme));
  224. EXPECT_TRUE(
  225. auth_handler_factory->IsSchemeAllowedForTesting(net::kNtlmAuthScheme));
  226. EXPECT_FALSE(auth_handler_factory->IsSchemeAllowedForTesting(
  227. net::kNegotiateAuthScheme));
  228. }
  229. {
  230. mojom::HttpAuthDynamicParamsPtr auth_params =
  231. mojom::HttpAuthDynamicParams::New();
  232. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  233. EXPECT_TRUE(
  234. auth_handler_factory->IsSchemeAllowedForTesting(net::kBasicAuthScheme));
  235. EXPECT_TRUE(auth_handler_factory->IsSchemeAllowedForTesting(
  236. net::kDigestAuthScheme));
  237. EXPECT_TRUE(
  238. auth_handler_factory->IsSchemeAllowedForTesting(net::kNtlmAuthScheme));
  239. #if BUILDFLAG(USE_KERBEROS) && !BUILDFLAG(IS_ANDROID)
  240. EXPECT_TRUE(auth_handler_factory->IsSchemeAllowedForTesting(
  241. net::kNegotiateAuthScheme));
  242. #else
  243. EXPECT_FALSE(auth_handler_factory->IsSchemeAllowedForTesting(
  244. net::kNegotiateAuthScheme));
  245. #endif
  246. }
  247. }
  248. TEST_F(NetworkServiceTest, AuthSchemesNone) {
  249. service()->SetUpHttpAuth(mojom::HttpAuthStaticParams::New());
  250. mojo::Remote<mojom::NetworkContext> network_context_remote;
  251. NetworkContext network_context(
  252. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  253. CreateContextParams());
  254. net::HttpAuthHandlerRegistryFactory* auth_handler_factory =
  255. reinterpret_cast<net::HttpAuthHandlerRegistryFactory*>(
  256. network_context.url_request_context()->http_auth_handler_factory());
  257. ASSERT_TRUE(auth_handler_factory);
  258. // An empty list means to support no schemes.
  259. mojom::HttpAuthDynamicParamsPtr auth_params =
  260. mojom::HttpAuthDynamicParams::New();
  261. auth_params->allowed_schemes = std::vector<std::string>{};
  262. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  263. EXPECT_FALSE(
  264. auth_handler_factory->IsSchemeAllowedForTesting(net::kBasicAuthScheme));
  265. EXPECT_FALSE(
  266. auth_handler_factory->IsSchemeAllowedForTesting(net::kDigestAuthScheme));
  267. EXPECT_FALSE(
  268. auth_handler_factory->IsSchemeAllowedForTesting(net::kNtlmAuthScheme));
  269. }
  270. #if BUILDFLAG(USE_EXTERNAL_GSSAPI)
  271. TEST_F(NetworkServiceTest, AuthGssapiLibraryName) {
  272. const std::string kGssapiLibraryName = "Jim";
  273. mojom::HttpAuthStaticParamsPtr static_auth_params =
  274. mojom::HttpAuthStaticParams::New();
  275. static_auth_params->gssapi_library_name = kGssapiLibraryName;
  276. service()->SetUpHttpAuth(std::move(static_auth_params));
  277. mojom::HttpAuthDynamicParamsPtr dynamic_auth_params =
  278. mojom::HttpAuthDynamicParams::New();
  279. dynamic_auth_params->allowed_schemes =
  280. std::vector<std::string>{net::kNegotiateAuthScheme};
  281. service()->ConfigureHttpAuthPrefs(std::move(dynamic_auth_params));
  282. mojo::Remote<mojom::NetworkContext> network_context_remote;
  283. NetworkContext network_context(
  284. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  285. CreateContextParams());
  286. net::HttpAuthHandlerRegistryFactory* auth_handler_factory =
  287. reinterpret_cast<net::HttpAuthHandlerRegistryFactory*>(
  288. network_context.url_request_context()->http_auth_handler_factory());
  289. ASSERT_TRUE(auth_handler_factory->IsSchemeAllowedForTesting(
  290. net::kNegotiateAuthScheme));
  291. EXPECT_EQ(kGssapiLibraryName,
  292. auth_handler_factory->GetNegotiateLibraryNameForTesting());
  293. }
  294. #endif // BUILDFLAG(USE_EXTERNAL_GSSAPI)
  295. TEST_F(NetworkServiceTest, AuthServerAllowlist) {
  296. // Add one server to the allowlist before creating any NetworkContexts.
  297. mojom::HttpAuthDynamicParamsPtr auth_params =
  298. mojom::HttpAuthDynamicParams::New();
  299. auth_params->server_allowlist = "server1";
  300. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  301. // Create a network context, which should reflect the allowlist.
  302. mojo::Remote<mojom::NetworkContext> network_context_remote;
  303. NetworkContext network_context(
  304. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  305. CreateContextParams());
  306. net::HttpAuthHandlerFactory* auth_handler_factory =
  307. network_context.url_request_context()->http_auth_handler_factory();
  308. ASSERT_TRUE(auth_handler_factory);
  309. ASSERT_TRUE(auth_handler_factory->http_auth_preferences());
  310. EXPECT_TRUE(
  311. auth_handler_factory->http_auth_preferences()->CanUseDefaultCredentials(
  312. url::SchemeHostPort(GURL("https://server1/"))));
  313. EXPECT_FALSE(
  314. auth_handler_factory->http_auth_preferences()->CanUseDefaultCredentials(
  315. url::SchemeHostPort(GURL("https://server2/"))));
  316. // Change allowlist to only have a different server on it. The pre-existing
  317. // NetworkContext should be using the new list.
  318. auth_params = mojom::HttpAuthDynamicParams::New();
  319. auth_params->server_allowlist = "server2";
  320. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  321. EXPECT_FALSE(
  322. auth_handler_factory->http_auth_preferences()->CanUseDefaultCredentials(
  323. url::SchemeHostPort(GURL("https://server1/"))));
  324. EXPECT_TRUE(
  325. auth_handler_factory->http_auth_preferences()->CanUseDefaultCredentials(
  326. url::SchemeHostPort(GURL("https://server2/"))));
  327. // Change allowlist to have multiple servers. The pre-existing NetworkContext
  328. // should be using the new list.
  329. auth_params = mojom::HttpAuthDynamicParams::New();
  330. auth_params->server_allowlist = "server1,server2";
  331. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  332. EXPECT_TRUE(
  333. auth_handler_factory->http_auth_preferences()->CanUseDefaultCredentials(
  334. url::SchemeHostPort(GURL("https://server1/"))));
  335. EXPECT_TRUE(
  336. auth_handler_factory->http_auth_preferences()->CanUseDefaultCredentials(
  337. url::SchemeHostPort(GURL("https://server2/"))));
  338. }
  339. TEST_F(NetworkServiceTest, AuthDelegateAllowlist) {
  340. using DelegationType = net::HttpAuth::DelegationType;
  341. // Add one server to the allowlist before creating any NetworkContexts.
  342. mojom::HttpAuthDynamicParamsPtr auth_params =
  343. mojom::HttpAuthDynamicParams::New();
  344. auth_params->delegate_allowlist = "server1";
  345. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  346. // Create a network context, which should reflect the allowlist.
  347. mojo::Remote<mojom::NetworkContext> network_context_remote;
  348. NetworkContext network_context(
  349. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  350. CreateContextParams());
  351. net::HttpAuthHandlerFactory* auth_handler_factory =
  352. network_context.url_request_context()->http_auth_handler_factory();
  353. ASSERT_TRUE(auth_handler_factory);
  354. const net::HttpAuthPreferences* auth_prefs =
  355. auth_handler_factory->http_auth_preferences();
  356. ASSERT_TRUE(auth_prefs);
  357. EXPECT_EQ(DelegationType::kUnconstrained,
  358. auth_prefs->GetDelegationType(
  359. url::SchemeHostPort(GURL("https://server1/"))));
  360. EXPECT_EQ(DelegationType::kNone,
  361. auth_prefs->GetDelegationType(
  362. url::SchemeHostPort(GURL("https://server2/"))));
  363. // Change allowlist to only have a different server on it. The pre-existing
  364. // NetworkContext should be using the new list.
  365. auth_params = mojom::HttpAuthDynamicParams::New();
  366. auth_params->delegate_allowlist = "server2";
  367. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  368. EXPECT_EQ(DelegationType::kNone,
  369. auth_prefs->GetDelegationType(
  370. url::SchemeHostPort(GURL("https://server1/"))));
  371. EXPECT_EQ(DelegationType::kUnconstrained,
  372. auth_prefs->GetDelegationType(
  373. url::SchemeHostPort(GURL("https://server2/"))));
  374. // Change allowlist to have multiple servers. The pre-existing NetworkContext
  375. // should be using the new list.
  376. auth_params = mojom::HttpAuthDynamicParams::New();
  377. auth_params->delegate_allowlist = "server1,server2";
  378. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  379. EXPECT_EQ(DelegationType::kUnconstrained,
  380. auth_prefs->GetDelegationType(
  381. url::SchemeHostPort(GURL("https://server1/"))));
  382. EXPECT_EQ(DelegationType::kUnconstrained,
  383. auth_prefs->GetDelegationType(
  384. url::SchemeHostPort(GURL("https://server2/"))));
  385. }
  386. TEST_F(NetworkServiceTest, DelegateByKdcPolicy) {
  387. // Create a network context, which should use default value.
  388. mojo::Remote<mojom::NetworkContext> network_context_remote;
  389. NetworkContext network_context(
  390. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  391. CreateContextParams());
  392. net::HttpAuthHandlerFactory* auth_handler_factory =
  393. network_context.url_request_context()->http_auth_handler_factory();
  394. ASSERT_TRUE(auth_handler_factory);
  395. ASSERT_TRUE(auth_handler_factory->http_auth_preferences());
  396. EXPECT_FALSE(
  397. auth_handler_factory->http_auth_preferences()->delegate_by_kdc_policy());
  398. // Change allowlist to only have a different server on it. The pre-existing
  399. // NetworkContext should be using the new list.
  400. mojom::HttpAuthDynamicParamsPtr auth_params =
  401. mojom::HttpAuthDynamicParams::New();
  402. auth_params->delegate_by_kdc_policy = true;
  403. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  404. EXPECT_TRUE(
  405. auth_handler_factory->http_auth_preferences()->delegate_by_kdc_policy());
  406. }
  407. TEST_F(NetworkServiceTest, AuthNegotiateCnameLookup) {
  408. // Set |negotiate_disable_cname_lookup| to true before creating any
  409. // NetworkContexts.
  410. mojom::HttpAuthDynamicParamsPtr auth_params =
  411. mojom::HttpAuthDynamicParams::New();
  412. auth_params->negotiate_disable_cname_lookup = true;
  413. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  414. // Create a network context, which should reflect the setting.
  415. mojo::Remote<mojom::NetworkContext> network_context_remote;
  416. NetworkContext network_context(
  417. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  418. CreateContextParams());
  419. net::HttpAuthHandlerFactory* auth_handler_factory =
  420. network_context.url_request_context()->http_auth_handler_factory();
  421. ASSERT_TRUE(auth_handler_factory);
  422. ASSERT_TRUE(auth_handler_factory->http_auth_preferences());
  423. EXPECT_TRUE(auth_handler_factory->http_auth_preferences()
  424. ->NegotiateDisableCnameLookup());
  425. // Set it to false. The pre-existing NetworkContext should be using the new
  426. // setting.
  427. auth_params = mojom::HttpAuthDynamicParams::New();
  428. auth_params->negotiate_disable_cname_lookup = false;
  429. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  430. EXPECT_FALSE(auth_handler_factory->http_auth_preferences()
  431. ->NegotiateDisableCnameLookup());
  432. // Set it back to true. The pre-existing NetworkContext should be using the
  433. // new setting.
  434. auth_params = mojom::HttpAuthDynamicParams::New();
  435. auth_params->negotiate_disable_cname_lookup = true;
  436. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  437. EXPECT_TRUE(auth_handler_factory->http_auth_preferences()
  438. ->NegotiateDisableCnameLookup());
  439. }
  440. TEST_F(NetworkServiceTest, AuthEnableNegotiatePort) {
  441. // Set |enable_negotiate_port| to true before creating any NetworkContexts.
  442. mojom::HttpAuthDynamicParamsPtr auth_params =
  443. mojom::HttpAuthDynamicParams::New();
  444. auth_params->enable_negotiate_port = true;
  445. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  446. // Create a network context, which should reflect the setting.
  447. mojo::Remote<mojom::NetworkContext> network_context_remote;
  448. NetworkContext network_context(
  449. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  450. CreateContextParams());
  451. net::HttpAuthHandlerFactory* auth_handler_factory =
  452. network_context.url_request_context()->http_auth_handler_factory();
  453. ASSERT_TRUE(auth_handler_factory);
  454. ASSERT_TRUE(auth_handler_factory->http_auth_preferences());
  455. EXPECT_TRUE(
  456. auth_handler_factory->http_auth_preferences()->NegotiateEnablePort());
  457. // Set it to false. The pre-existing NetworkContext should be using the new
  458. // setting.
  459. auth_params = mojom::HttpAuthDynamicParams::New();
  460. auth_params->enable_negotiate_port = false;
  461. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  462. EXPECT_FALSE(
  463. auth_handler_factory->http_auth_preferences()->NegotiateEnablePort());
  464. // Set it back to true. The pre-existing NetworkContext should be using the
  465. // new setting.
  466. auth_params = mojom::HttpAuthDynamicParams::New();
  467. auth_params->enable_negotiate_port = true;
  468. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  469. EXPECT_TRUE(
  470. auth_handler_factory->http_auth_preferences()->NegotiateEnablePort());
  471. }
  472. // DnsClient isn't supported on iOS.
  473. #if !BUILDFLAG(IS_IOS)
  474. TEST_F(NetworkServiceTest, DnsClientEnableDisable) {
  475. // Create valid DnsConfig.
  476. net::DnsConfig config;
  477. config.nameservers.push_back(net::IPEndPoint());
  478. auto dns_client = std::make_unique<net::MockDnsClient>(
  479. std::move(config), net::MockDnsClientRuleList());
  480. dns_client->set_ignore_system_config_changes(true);
  481. net::DnsClient* dns_client_ptr = dns_client.get();
  482. service()->host_resolver_manager()->SetDnsClientForTesting(
  483. std::move(dns_client));
  484. service()->ConfigureStubHostResolver(
  485. /*insecure_dns_client_enabled=*/true, net::SecureDnsMode::kOff,
  486. /*dns_over_https_config=*/{},
  487. /*additional_dns_types_enabled=*/true);
  488. EXPECT_TRUE(dns_client_ptr->CanUseInsecureDnsTransactions());
  489. EXPECT_EQ(net::SecureDnsMode::kOff,
  490. dns_client_ptr->GetEffectiveConfig()->secure_dns_mode);
  491. service()->ConfigureStubHostResolver(
  492. /*insecure_dns_client_enabled=*/false, net::SecureDnsMode::kOff,
  493. /*dns_over_https_config=*/{},
  494. /*additional_dns_types_enabled=*/true);
  495. EXPECT_FALSE(dns_client_ptr->CanUseInsecureDnsTransactions());
  496. EXPECT_EQ(net::SecureDnsMode::kOff,
  497. dns_client_ptr->GetEffectiveConfig()->secure_dns_mode);
  498. service()->ConfigureStubHostResolver(
  499. /*insecure_dns_client_enabled=*/false, net::SecureDnsMode::kAutomatic,
  500. /*dns_over_https_config=*/{},
  501. /*additional_dns_types_enabled=*/true);
  502. EXPECT_FALSE(dns_client_ptr->CanUseInsecureDnsTransactions());
  503. EXPECT_EQ(net::SecureDnsMode::kAutomatic,
  504. dns_client_ptr->GetEffectiveConfig()->secure_dns_mode);
  505. service()->ConfigureStubHostResolver(
  506. /*insecure_dns_client_enabled=*/false, net::SecureDnsMode::kAutomatic,
  507. *net::DnsOverHttpsConfig::FromString("https://foo/"),
  508. /*additional_dns_types_enabled=*/true);
  509. EXPECT_FALSE(dns_client_ptr->CanUseInsecureDnsTransactions());
  510. EXPECT_EQ(net::SecureDnsMode::kAutomatic,
  511. dns_client_ptr->GetEffectiveConfig()->secure_dns_mode);
  512. }
  513. TEST_F(NetworkServiceTest, HandlesAdditionalDnsQueryTypesEnableDisable) {
  514. // Create valid DnsConfig.
  515. net::DnsConfig config;
  516. config.nameservers.emplace_back();
  517. auto dns_client = std::make_unique<net::MockDnsClient>(
  518. std::move(config), net::MockDnsClientRuleList());
  519. dns_client->set_ignore_system_config_changes(true);
  520. const net::DnsClient* dns_client_ptr = dns_client.get();
  521. service()->host_resolver_manager()->SetDnsClientForTesting(
  522. std::move(dns_client));
  523. service()->ConfigureStubHostResolver(
  524. /*insecure_dns_client_enabled=*/true, net::SecureDnsMode::kOff,
  525. /*dns_over_https_config=*/{},
  526. /*additional_dns_types_enabled=*/true);
  527. EXPECT_TRUE(dns_client_ptr->CanQueryAdditionalTypesViaInsecureDns());
  528. service()->ConfigureStubHostResolver(
  529. /*insecure_dns_client_enabled=*/true, net::SecureDnsMode::kOff,
  530. /*dns_over_https_config=*/{},
  531. /*additional_dns_types_enabled=*/false);
  532. EXPECT_FALSE(dns_client_ptr->CanQueryAdditionalTypesViaInsecureDns());
  533. }
  534. TEST_F(NetworkServiceTest, DnsOverHttpsEnableDisable) {
  535. const auto kConfig1 = *net::DnsOverHttpsConfig::FromString("https://foo/");
  536. const auto kConfig2 = *net::DnsOverHttpsConfig::FromString(
  537. "https://bar/dns-query{?dns} https://grapefruit/resolver/query{?dns}");
  538. // Create valid DnsConfig.
  539. net::DnsConfig config;
  540. config.nameservers.push_back(net::IPEndPoint());
  541. auto dns_client = std::make_unique<net::MockDnsClient>(
  542. std::move(config), net::MockDnsClientRuleList());
  543. dns_client->set_ignore_system_config_changes(true);
  544. net::MockDnsClient* dns_client_ptr = dns_client.get();
  545. service()->host_resolver_manager()->SetDnsClientForTesting(
  546. std::move(dns_client));
  547. // Enable DNS over HTTPS for one server.
  548. service()->ConfigureStubHostResolver(
  549. /*insecure_dns_client_enabled=*/false, net::SecureDnsMode::kAutomatic,
  550. kConfig1,
  551. /*additional_dns_types_enabled=*/true);
  552. EXPECT_TRUE(
  553. service()->host_resolver_manager()->GetDnsConfigAsValue().is_dict());
  554. EXPECT_EQ(kConfig1, dns_client_ptr->GetEffectiveConfig()->doh_config);
  555. // Enable DNS over HTTPS for two servers.
  556. service()->ConfigureStubHostResolver(
  557. /*insecure_dns_client_enabled=*/true, net::SecureDnsMode::kSecure,
  558. kConfig2,
  559. /*additional_dns_types_enabled=*/true);
  560. EXPECT_TRUE(
  561. service()->host_resolver_manager()->GetDnsConfigAsValue().is_dict());
  562. EXPECT_EQ(kConfig2, dns_client_ptr->GetEffectiveConfig()->doh_config);
  563. }
  564. TEST_F(NetworkServiceTest, DisableDohUpgradeProviders) {
  565. auto FindProviderFeature = [](base::StringPiece provider) -> base::Feature {
  566. const auto it =
  567. base::ranges::find(net::DohProviderEntry::GetList(), provider,
  568. &net::DohProviderEntry::provider);
  569. CHECK(it != net::DohProviderEntry::GetList().end())
  570. << "Provider named \"" << provider
  571. << "\" not found in DoH provider list.";
  572. return (*it)->feature;
  573. };
  574. base::test::ScopedFeatureList scoped_features;
  575. scoped_features.InitWithFeatures(
  576. /*enabled_features=*/{features::kDnsOverHttpsUpgrade},
  577. /*disabled_features=*/{FindProviderFeature("CleanBrowsingSecure"),
  578. FindProviderFeature("Cloudflare")});
  579. service()->ConfigureStubHostResolver(
  580. /*insecure_dns_client_enabled=*/true, net::SecureDnsMode::kAutomatic,
  581. /*dns_over_https_config=*/{},
  582. /*additional_dns_types_enabled=*/true);
  583. // Set valid DnsConfig.
  584. net::DnsConfig config;
  585. // Cloudflare upgradeable IPs
  586. net::IPAddress dns_ip0(1, 0, 0, 1);
  587. net::IPAddress dns_ip1;
  588. EXPECT_TRUE(dns_ip1.AssignFromIPLiteral("2606:4700:4700::1111"));
  589. // CleanBrowsing family filter upgradeable IP
  590. net::IPAddress dns_ip2;
  591. EXPECT_TRUE(dns_ip2.AssignFromIPLiteral("2a0d:2a00:2::"));
  592. // CleanBrowsing security filter upgradeable IP
  593. net::IPAddress dns_ip3(185, 228, 169, 9);
  594. // Non-upgradeable IP
  595. net::IPAddress dns_ip4(1, 2, 3, 4);
  596. config.nameservers.emplace_back(dns_ip0, net::dns_protocol::kDefaultPort);
  597. config.nameservers.emplace_back(dns_ip1, net::dns_protocol::kDefaultPort);
  598. config.nameservers.emplace_back(dns_ip2, 54);
  599. config.nameservers.emplace_back(dns_ip3, net::dns_protocol::kDefaultPort);
  600. config.nameservers.emplace_back(dns_ip4, net::dns_protocol::kDefaultPort);
  601. auto dns_client = net::DnsClient::CreateClient(nullptr /* net_log */);
  602. dns_client->SetSystemConfig(config);
  603. net::DnsClient* dns_client_ptr = dns_client.get();
  604. service()->host_resolver_manager()->SetDnsClientForTesting(
  605. std::move(dns_client));
  606. auto expected_doh_config = *net::DnsOverHttpsConfig::FromString(
  607. "https://doh.cleanbrowsing.org/doh/family-filter{?dns}");
  608. EXPECT_TRUE(dns_client_ptr->GetEffectiveConfig());
  609. EXPECT_EQ(expected_doh_config,
  610. dns_client_ptr->GetEffectiveConfig()->doh_config);
  611. }
  612. TEST_F(NetworkServiceTest, DohProbe) {
  613. mojom::NetworkContextParamsPtr context_params = CreateContextParams();
  614. mojo::Remote<mojom::NetworkContext> network_context;
  615. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  616. std::move(context_params));
  617. net::DnsConfig config;
  618. config.nameservers.push_back(net::IPEndPoint());
  619. config.doh_config =
  620. *net::DnsOverHttpsConfig::FromString("https://example.com/");
  621. auto dns_client = std::make_unique<net::MockDnsClient>(
  622. std::move(config), net::MockDnsClientRuleList());
  623. dns_client->set_ignore_system_config_changes(true);
  624. net::MockDnsClient* dns_client_ptr = dns_client.get();
  625. service()->host_resolver_manager()->SetDnsClientForTesting(
  626. std::move(dns_client));
  627. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  628. task_environment()->FastForwardBy(NetworkService::kInitialDohProbeTimeout);
  629. EXPECT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  630. }
  631. TEST_F(NetworkServiceTest, DohProbe_MultipleContexts) {
  632. mojom::NetworkContextParamsPtr context_params1 = CreateContextParams();
  633. mojo::Remote<mojom::NetworkContext> network_context1;
  634. service()->CreateNetworkContext(network_context1.BindNewPipeAndPassReceiver(),
  635. std::move(context_params1));
  636. net::DnsConfig config;
  637. config.nameservers.push_back(net::IPEndPoint());
  638. config.doh_config =
  639. *net::DnsOverHttpsConfig::FromString("https://example.com/");
  640. auto dns_client = std::make_unique<net::MockDnsClient>(
  641. std::move(config), net::MockDnsClientRuleList());
  642. dns_client->set_ignore_system_config_changes(true);
  643. net::MockDnsClient* dns_client_ptr = dns_client.get();
  644. service()->host_resolver_manager()->SetDnsClientForTesting(
  645. std::move(dns_client));
  646. task_environment()->FastForwardBy(NetworkService::kInitialDohProbeTimeout);
  647. ASSERT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  648. mojom::NetworkContextParamsPtr context_params2 = CreateContextParams();
  649. mojo::Remote<mojom::NetworkContext> network_context2;
  650. service()->CreateNetworkContext(network_context2.BindNewPipeAndPassReceiver(),
  651. std::move(context_params2));
  652. EXPECT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  653. network_context2.reset();
  654. task_environment()->FastForwardUntilNoTasksRemain();
  655. EXPECT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  656. network_context1.reset();
  657. task_environment()->FastForwardUntilNoTasksRemain();
  658. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  659. }
  660. TEST_F(NetworkServiceTest, DohProbe_ContextAddedBeforeTimeout) {
  661. net::DnsConfig config;
  662. config.nameservers.push_back(net::IPEndPoint());
  663. config.doh_config =
  664. *net::DnsOverHttpsConfig::FromString("https://example.com/");
  665. auto dns_client = std::make_unique<net::MockDnsClient>(
  666. std::move(config), net::MockDnsClientRuleList());
  667. dns_client->set_ignore_system_config_changes(true);
  668. net::MockDnsClient* dns_client_ptr = dns_client.get();
  669. service()->host_resolver_manager()->SetDnsClientForTesting(
  670. std::move(dns_client));
  671. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  672. mojom::NetworkContextParamsPtr context_params = CreateContextParams();
  673. mojo::Remote<mojom::NetworkContext> network_context;
  674. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  675. std::move(context_params));
  676. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  677. task_environment()->FastForwardBy(NetworkService::kInitialDohProbeTimeout);
  678. EXPECT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  679. }
  680. TEST_F(NetworkServiceTest, DohProbe_ContextAddedAfterTimeout) {
  681. net::DnsConfig config;
  682. config.nameservers.push_back(net::IPEndPoint());
  683. config.doh_config =
  684. *net::DnsOverHttpsConfig::FromString("https://example.com/");
  685. auto dns_client = std::make_unique<net::MockDnsClient>(
  686. std::move(config), net::MockDnsClientRuleList());
  687. dns_client->set_ignore_system_config_changes(true);
  688. net::MockDnsClient* dns_client_ptr = dns_client.get();
  689. service()->host_resolver_manager()->SetDnsClientForTesting(
  690. std::move(dns_client));
  691. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  692. task_environment()->FastForwardBy(NetworkService::kInitialDohProbeTimeout);
  693. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  694. mojom::NetworkContextParamsPtr context_params = CreateContextParams();
  695. mojo::Remote<mojom::NetworkContext> network_context;
  696. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  697. std::move(context_params));
  698. EXPECT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  699. }
  700. TEST_F(NetworkServiceTest, DohProbe_ContextRemovedBeforeTimeout) {
  701. mojom::NetworkContextParamsPtr context_params = CreateContextParams();
  702. mojo::Remote<mojom::NetworkContext> network_context;
  703. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  704. std::move(context_params));
  705. net::DnsConfig config;
  706. config.nameservers.push_back(net::IPEndPoint());
  707. config.doh_config =
  708. *net::DnsOverHttpsConfig::FromString("https://example.com/");
  709. auto dns_client = std::make_unique<net::MockDnsClient>(
  710. std::move(config), net::MockDnsClientRuleList());
  711. dns_client->set_ignore_system_config_changes(true);
  712. net::MockDnsClient* dns_client_ptr = dns_client.get();
  713. service()->host_resolver_manager()->SetDnsClientForTesting(
  714. std::move(dns_client));
  715. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  716. network_context.reset();
  717. task_environment()->FastForwardUntilNoTasksRemain();
  718. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  719. task_environment()->FastForwardBy(NetworkService::kInitialDohProbeTimeout);
  720. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  721. }
  722. TEST_F(NetworkServiceTest, DohProbe_ContextRemovedAfterTimeout) {
  723. mojom::NetworkContextParamsPtr context_params = CreateContextParams();
  724. mojo::Remote<mojom::NetworkContext> network_context;
  725. service()->CreateNetworkContext(network_context.BindNewPipeAndPassReceiver(),
  726. std::move(context_params));
  727. net::DnsConfig config;
  728. config.nameservers.push_back(net::IPEndPoint());
  729. config.doh_config =
  730. *net::DnsOverHttpsConfig::FromString("https://example.com/");
  731. auto dns_client = std::make_unique<net::MockDnsClient>(
  732. std::move(config), net::MockDnsClientRuleList());
  733. dns_client->set_ignore_system_config_changes(true);
  734. net::MockDnsClient* dns_client_ptr = dns_client.get();
  735. service()->host_resolver_manager()->SetDnsClientForTesting(
  736. std::move(dns_client));
  737. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  738. task_environment()->FastForwardBy(NetworkService::kInitialDohProbeTimeout);
  739. EXPECT_TRUE(dns_client_ptr->factory()->doh_probes_running());
  740. network_context.reset();
  741. task_environment()->FastForwardUntilNoTasksRemain();
  742. EXPECT_FALSE(dns_client_ptr->factory()->doh_probes_running());
  743. }
  744. #endif // !BUILDFLAG(IS_IOS)
  745. // |ntlm_v2_enabled| is only supported on POSIX platforms.
  746. #if BUILDFLAG(IS_POSIX)
  747. TEST_F(NetworkServiceTest, AuthNtlmV2Enabled) {
  748. // Set |ntlm_v2_enabled| to false before creating any NetworkContexts.
  749. mojom::HttpAuthDynamicParamsPtr auth_params =
  750. mojom::HttpAuthDynamicParams::New();
  751. auth_params->ntlm_v2_enabled = false;
  752. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  753. // Create a network context, which should reflect the setting.
  754. mojo::Remote<mojom::NetworkContext> network_context_remote;
  755. NetworkContext network_context(
  756. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  757. CreateContextParams());
  758. net::HttpAuthHandlerFactory* auth_handler_factory =
  759. network_context.url_request_context()->http_auth_handler_factory();
  760. ASSERT_TRUE(auth_handler_factory);
  761. ASSERT_TRUE(auth_handler_factory->http_auth_preferences());
  762. EXPECT_FALSE(auth_handler_factory->http_auth_preferences()->NtlmV2Enabled());
  763. // Set it to true. The pre-existing NetworkContext should be using the new
  764. // setting.
  765. auth_params = mojom::HttpAuthDynamicParams::New();
  766. auth_params->ntlm_v2_enabled = true;
  767. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  768. EXPECT_TRUE(auth_handler_factory->http_auth_preferences()->NtlmV2Enabled());
  769. // Set it back to false. The pre-existing NetworkContext should be using the
  770. // new setting.
  771. auth_params = mojom::HttpAuthDynamicParams::New();
  772. auth_params->ntlm_v2_enabled = false;
  773. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  774. EXPECT_FALSE(auth_handler_factory->http_auth_preferences()->NtlmV2Enabled());
  775. }
  776. #endif // BUILDFLAG(IS_POSIX)
  777. // |android_negotiate_account_type| is only supported on Android.
  778. #if BUILDFLAG(IS_ANDROID)
  779. TEST_F(NetworkServiceTest, AuthAndroidNegotiateAccountType) {
  780. const char kInitialAccountType[] = "Scorpio";
  781. const char kFinalAccountType[] = "Pisces";
  782. // Set |android_negotiate_account_type| to before creating any
  783. // NetworkContexts.
  784. mojom::HttpAuthDynamicParamsPtr auth_params =
  785. mojom::HttpAuthDynamicParams::New();
  786. auth_params->android_negotiate_account_type = kInitialAccountType;
  787. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  788. // Create a network context, which should reflect the setting.
  789. mojo::Remote<mojom::NetworkContext> network_context_remote;
  790. NetworkContext network_context(
  791. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  792. CreateContextParams());
  793. net::HttpAuthHandlerFactory* auth_handler_factory =
  794. network_context.url_request_context()->http_auth_handler_factory();
  795. ASSERT_TRUE(auth_handler_factory);
  796. ASSERT_TRUE(auth_handler_factory->http_auth_preferences());
  797. EXPECT_EQ(kInitialAccountType, auth_handler_factory->http_auth_preferences()
  798. ->AuthAndroidNegotiateAccountType());
  799. // Change |android_negotiate_account_type|. The pre-existing NetworkContext
  800. // should be using the new setting.
  801. auth_params = mojom::HttpAuthDynamicParams::New();
  802. auth_params->android_negotiate_account_type = kFinalAccountType;
  803. service()->ConfigureHttpAuthPrefs(std::move(auth_params));
  804. EXPECT_EQ(kFinalAccountType, auth_handler_factory->http_auth_preferences()
  805. ->AuthAndroidNegotiateAccountType());
  806. }
  807. #endif // BUILDFLAG(IS_ANDROID)
  808. static int GetGlobalMaxConnectionsPerProxy() {
  809. return net::ClientSocketPoolManager::max_sockets_per_proxy_server(
  810. net::HttpNetworkSession::NORMAL_SOCKET_POOL);
  811. }
  812. // Tests that NetworkService::SetMaxConnectionsPerProxy() (1) modifies globals
  813. // in net::ClientSocketPoolManager (2) saturates out of bound values.
  814. TEST_F(NetworkServiceTest, SetMaxConnectionsPerProxy) {
  815. const int kDefault = net::kDefaultMaxSocketsPerProxyServer;
  816. const int kMin = 6;
  817. const int kMax = 99;
  818. // Starts off at default value.
  819. EXPECT_EQ(net::kDefaultMaxSocketsPerProxyServer,
  820. GetGlobalMaxConnectionsPerProxy());
  821. // Anything less than kMin saturates to kMin.
  822. service()->SetMaxConnectionsPerProxy(kMin - 1);
  823. EXPECT_EQ(kMin, GetGlobalMaxConnectionsPerProxy());
  824. // Anything larger than kMax saturates to kMax
  825. service()->SetMaxConnectionsPerProxy(kMax + 1);
  826. EXPECT_EQ(kMax, GetGlobalMaxConnectionsPerProxy());
  827. // Anything in between kMin and kMax should be set exactly.
  828. service()->SetMaxConnectionsPerProxy(58);
  829. EXPECT_EQ(58, GetGlobalMaxConnectionsPerProxy());
  830. // Negative values select the default.
  831. service()->SetMaxConnectionsPerProxy(-2);
  832. EXPECT_EQ(kDefault, GetGlobalMaxConnectionsPerProxy());
  833. // Restore the default value to minize sideffects.
  834. service()->SetMaxConnectionsPerProxy(kDefault);
  835. }
  836. #if BUILDFLAG(IS_CT_SUPPORTED)
  837. // Tests that disabling CT enforcement disables the feature for both existing
  838. // and new network contexts.
  839. TEST_F(NetworkServiceTest, DisableCTEnforcement) {
  840. mojo::Remote<mojom::NetworkContext> network_context_remote;
  841. NetworkContext network_context(
  842. service(), network_context_remote.BindNewPipeAndPassReceiver(),
  843. CreateContextParams());
  844. net::TransportSecurityState* transport_security_state =
  845. network_context.url_request_context()->transport_security_state();
  846. EXPECT_FALSE(
  847. transport_security_state->is_ct_emergency_disabled_for_testing());
  848. base::RunLoop run_loop;
  849. service()->SetCtEnforcementEnabled(false, run_loop.QuitClosure());
  850. run_loop.Run();
  851. EXPECT_TRUE(transport_security_state->is_ct_emergency_disabled_for_testing());
  852. mojo::Remote<mojom::NetworkContext> new_network_context_remote;
  853. NetworkContext new_network_context(
  854. service(), new_network_context_remote.BindNewPipeAndPassReceiver(),
  855. CreateContextParams());
  856. transport_security_state =
  857. new_network_context.url_request_context()->transport_security_state();
  858. EXPECT_TRUE(transport_security_state->is_ct_emergency_disabled_for_testing());
  859. }
  860. #endif // BUILDFLAG(IS_CT_SUPPORTED)
  861. class NetworkServiceTestWithService : public testing::Test {
  862. public:
  863. NetworkServiceTestWithService()
  864. : task_environment_(base::test::TaskEnvironment::MainThreadType::IO) {}
  865. NetworkServiceTestWithService(const NetworkServiceTestWithService&) = delete;
  866. NetworkServiceTestWithService& operator=(
  867. const NetworkServiceTestWithService&) = delete;
  868. ~NetworkServiceTestWithService() override {}
  869. void SetUp() override {
  870. test_server_.AddDefaultHandlers(base::FilePath(kServicesTestData));
  871. ASSERT_TRUE(test_server_.Start());
  872. service_ = NetworkService::CreateForTesting();
  873. service_->Bind(network_service_.BindNewPipeAndPassReceiver());
  874. }
  875. void CreateNetworkContext() {
  876. mojom::NetworkContextParamsPtr context_params =
  877. mojom::NetworkContextParams::New();
  878. // Use a dummy CertVerifier that always passes cert verification, since
  879. // these unittests don't need to test CertVerifier behavior.
  880. context_params->cert_verifier_params =
  881. FakeTestCertVerifierParamsFactory::GetCertVerifierParams();
  882. network_service_->CreateNetworkContext(
  883. network_context_.BindNewPipeAndPassReceiver(),
  884. std::move(context_params));
  885. }
  886. void LoadURL(const GURL& url, int options = mojom::kURLLoadOptionNone) {
  887. ResourceRequest request;
  888. request.url = url;
  889. request.method = "GET";
  890. request.request_initiator = url::Origin();
  891. StartLoadingURL(request, 0 /* process_id */, options);
  892. client_->RunUntilComplete();
  893. }
  894. void StartLoadingURL(const ResourceRequest& request,
  895. uint32_t process_id,
  896. int options = mojom::kURLLoadOptionNone) {
  897. client_ = std::make_unique<TestURLLoaderClient>();
  898. mojo::Remote<mojom::URLLoaderFactory> loader_factory;
  899. mojom::URLLoaderFactoryParamsPtr params =
  900. mojom::URLLoaderFactoryParams::New();
  901. params->process_id = process_id;
  902. params->request_initiator_origin_lock =
  903. url::Origin::Create(GURL("https://initiator.example.com"));
  904. params->is_corb_enabled = false;
  905. network_context_->CreateURLLoaderFactory(
  906. loader_factory.BindNewPipeAndPassReceiver(), std::move(params));
  907. loader_.reset();
  908. loader_factory->CreateLoaderAndStart(
  909. loader_.BindNewPipeAndPassReceiver(), 1, options, request,
  910. client_->CreateRemote(),
  911. net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
  912. }
  913. void Shutdown() { service_.reset(); }
  914. net::EmbeddedTestServer* test_server() { return &test_server_; }
  915. TestURLLoaderClient* client() { return client_.get(); }
  916. mojom::URLLoader* loader() { return loader_.get(); }
  917. mojom::NetworkService* service() { return network_service_.get(); }
  918. mojom::NetworkContext* context() { return network_context_.get(); }
  919. protected:
  920. base::test::TaskEnvironment task_environment_;
  921. std::unique_ptr<NetworkService> service_;
  922. net::EmbeddedTestServer test_server_;
  923. std::unique_ptr<TestURLLoaderClient> client_;
  924. mojo::Remote<mojom::NetworkService> network_service_;
  925. mojo::Remote<mojom::NetworkContext> network_context_;
  926. mojo::Remote<mojom::URLLoader> loader_;
  927. base::test::ScopedFeatureList scoped_features_;
  928. };
  929. // Verifies that loading a URL through the network service's mojo interface
  930. // works.
  931. TEST_F(NetworkServiceTestWithService, Basic) {
  932. CreateNetworkContext();
  933. LoadURL(test_server()->GetURL("/echo"));
  934. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  935. }
  936. // Verifies that a passed net log file is successfully opened and sane data
  937. // written to it.
  938. TEST_F(NetworkServiceTestWithService, StartsNetLog) {
  939. base::ScopedTempDir temp_dir;
  940. ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
  941. base::FilePath log_dir = temp_dir.GetPath();
  942. base::FilePath log_path = log_dir.Append(FILE_PATH_LITERAL("test_log.json"));
  943. base::Value::Dict dict;
  944. dict.Set("amiatest", "iamatest");
  945. base::File log_file(log_path,
  946. base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE);
  947. network_service_->StartNetLog(
  948. std::move(log_file), net::NetLogCaptureMode::kDefault, std::move(dict));
  949. CreateNetworkContext();
  950. LoadURL(test_server()->GetURL("/echo"));
  951. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  952. // |log_file| is closed on destruction of the NetworkService.
  953. Shutdown();
  954. // |log_file| is closed on another thread, so have to wait for that to happen.
  955. task_environment_.RunUntilIdle();
  956. JSONFileValueDeserializer deserializer(log_path);
  957. std::unique_ptr<base::Value> log_dict =
  958. deserializer.Deserialize(nullptr, nullptr);
  959. ASSERT_TRUE(log_dict);
  960. ASSERT_TRUE(log_dict->is_dict());
  961. ASSERT_EQ(*log_dict->GetDict().FindStringByDottedPath("constants.amiatest"),
  962. "iamatest");
  963. }
  964. // Verifies that raw headers are only reported if requested.
  965. TEST_F(NetworkServiceTestWithService, RawRequestHeadersAbsent) {
  966. CreateNetworkContext();
  967. ResourceRequest request;
  968. request.url = test_server()->GetURL("/server-redirect?/echo");
  969. request.method = "GET";
  970. request.request_initiator = url::Origin();
  971. StartLoadingURL(request, 0);
  972. client()->RunUntilRedirectReceived();
  973. EXPECT_TRUE(client()->has_received_redirect());
  974. loader()->FollowRedirect({}, {}, {}, absl::nullopt);
  975. client()->RunUntilComplete();
  976. }
  977. class NetworkServiceTestWithResolverMap : public NetworkServiceTestWithService {
  978. void SetUp() override {
  979. base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
  980. network::switches::kHostResolverRules, "MAP *.test 127.0.0.1");
  981. NetworkServiceTestWithService::SetUp();
  982. }
  983. };
  984. TEST_F(NetworkServiceTestWithService, SetNetworkConditions) {
  985. const base::UnguessableToken profile_id = base::UnguessableToken::Create();
  986. CreateNetworkContext();
  987. mojom::NetworkConditionsPtr network_conditions =
  988. mojom::NetworkConditions::New();
  989. network_conditions->offline = true;
  990. context()->SetNetworkConditions(profile_id, std::move(network_conditions));
  991. ResourceRequest request;
  992. request.url = test_server()->GetURL("/nocache.html");
  993. request.request_initiator =
  994. url::Origin::Create(GURL("https://initiator.example.com"));
  995. request.method = "GET";
  996. StartLoadingURL(request, 0);
  997. client()->RunUntilComplete();
  998. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  999. request.throttling_profile_id = profile_id;
  1000. StartLoadingURL(request, 0);
  1001. client()->RunUntilComplete();
  1002. EXPECT_EQ(net::ERR_INTERNET_DISCONNECTED,
  1003. client()->completion_status().error_code);
  1004. network_conditions = mojom::NetworkConditions::New();
  1005. network_conditions->offline = false;
  1006. context()->SetNetworkConditions(profile_id, std::move(network_conditions));
  1007. StartLoadingURL(request, 0);
  1008. client()->RunUntilComplete();
  1009. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  1010. network_conditions = mojom::NetworkConditions::New();
  1011. network_conditions->offline = true;
  1012. context()->SetNetworkConditions(profile_id, std::move(network_conditions));
  1013. request.throttling_profile_id = profile_id;
  1014. StartLoadingURL(request, 0);
  1015. client()->RunUntilComplete();
  1016. EXPECT_EQ(net::ERR_INTERNET_DISCONNECTED,
  1017. client()->completion_status().error_code);
  1018. context()->SetNetworkConditions(profile_id, nullptr);
  1019. StartLoadingURL(request, 0);
  1020. client()->RunUntilComplete();
  1021. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  1022. }
  1023. // Integration test confirming that the SetTrustTokenKeyCommitments IPC is wired
  1024. // up correctly by verifying that it's possible to read a value previously
  1025. // passed to the setter.
  1026. TEST_F(NetworkServiceTestWithService, SetsTrustTokenKeyCommitments) {
  1027. ASSERT_TRUE(service_->trust_token_key_commitments());
  1028. auto expectation = mojom::TrustTokenKeyCommitmentResult::New();
  1029. expectation->protocol_version =
  1030. mojom::TrustTokenProtocolVersion::kTrustTokenV3Pmb;
  1031. expectation->id = 1;
  1032. expectation->batch_size = 5;
  1033. base::RunLoop run_loop;
  1034. network_service_->SetTrustTokenKeyCommitments(
  1035. R"( { "https://issuer.example": { "TrustTokenV3PMB": {
  1036. "protocol_version": "TrustTokenV3PMB", "id": 1, "batchsize": 5 } } } )",
  1037. run_loop.QuitClosure());
  1038. run_loop.Run();
  1039. mojom::TrustTokenKeyCommitmentResultPtr result;
  1040. bool ran = false;
  1041. service_->trust_token_key_commitments()->Get(
  1042. *SuitableTrustTokenOrigin::Create(GURL("https://issuer.example")),
  1043. base::BindLambdaForTesting(
  1044. [&](mojom::TrustTokenKeyCommitmentResultPtr ptr) {
  1045. result = std::move(ptr);
  1046. ran = true;
  1047. }));
  1048. ASSERT_TRUE(ran);
  1049. EXPECT_TRUE(result.Equals(expectation));
  1050. }
  1051. TEST_F(NetworkServiceTestWithService, GetDnsConfigChangeManager) {
  1052. mojo::Remote<mojom::DnsConfigChangeManager> remote;
  1053. ASSERT_FALSE(remote.is_bound());
  1054. network_service_->GetDnsConfigChangeManager(
  1055. remote.BindNewPipeAndPassReceiver());
  1056. EXPECT_TRUE(remote.is_bound());
  1057. }
  1058. TEST_F(NetworkServiceTestWithService, GetNetworkList) {
  1059. base::RunLoop run_loop;
  1060. network_service_->GetNetworkList(
  1061. net::INCLUDE_HOST_SCOPE_VIRTUAL_INTERFACES,
  1062. base::BindLambdaForTesting(
  1063. [&](const absl::optional<std::vector<net::NetworkInterface>>& list) {
  1064. EXPECT_NE(absl::nullopt, list);
  1065. for (auto it = list->begin(); it != list->end(); ++it) {
  1066. // Verify that names are not empty.
  1067. EXPECT_FALSE(it->name.empty());
  1068. EXPECT_FALSE(it->friendly_name.empty());
  1069. // Verify that the address is correct.
  1070. EXPECT_TRUE(it->address.IsValid());
  1071. EXPECT_FALSE(it->address.IsZero());
  1072. EXPECT_GT(it->prefix_length, 1u);
  1073. EXPECT_LE(it->prefix_length, it->address.size() * 8);
  1074. }
  1075. run_loop.Quit();
  1076. }));
  1077. run_loop.Run();
  1078. }
  1079. class TestNetworkChangeManagerClient
  1080. : public mojom::NetworkChangeManagerClient {
  1081. public:
  1082. explicit TestNetworkChangeManagerClient(
  1083. mojom::NetworkService* network_service)
  1084. : connection_type_(mojom::ConnectionType::CONNECTION_UNKNOWN) {
  1085. mojo::Remote<mojom::NetworkChangeManager> manager_remote;
  1086. network_service->GetNetworkChangeManager(
  1087. manager_remote.BindNewPipeAndPassReceiver());
  1088. mojo::PendingRemote<mojom::NetworkChangeManagerClient> client_remote;
  1089. receiver_.Bind(client_remote.InitWithNewPipeAndPassReceiver());
  1090. manager_remote->RequestNotifications(std::move(client_remote));
  1091. }
  1092. TestNetworkChangeManagerClient(const TestNetworkChangeManagerClient&) =
  1093. delete;
  1094. TestNetworkChangeManagerClient& operator=(
  1095. const TestNetworkChangeManagerClient&) = delete;
  1096. ~TestNetworkChangeManagerClient() override {}
  1097. // NetworkChangeManagerClient implementation:
  1098. void OnInitialConnectionType(mojom::ConnectionType type) override {
  1099. if (type == connection_type_)
  1100. run_loop_.Quit();
  1101. }
  1102. void OnNetworkChanged(mojom::ConnectionType type) override {
  1103. if (type == connection_type_)
  1104. run_loop_.Quit();
  1105. }
  1106. // Waits for the desired |connection_type| notification.
  1107. void WaitForNotification(mojom::ConnectionType type) {
  1108. connection_type_ = type;
  1109. run_loop_.Run();
  1110. }
  1111. void Flush() { receiver_.FlushForTesting(); }
  1112. private:
  1113. base::RunLoop run_loop_;
  1114. mojom::ConnectionType connection_type_;
  1115. mojo::Receiver<mojom::NetworkChangeManagerClient> receiver_{this};
  1116. };
  1117. class NetworkChangeTest : public testing::Test {
  1118. public:
  1119. NetworkChangeTest()
  1120. : task_environment_(base::test::TaskEnvironment::MainThreadType::IO),
  1121. network_change_notifier_(
  1122. net::NetworkChangeNotifier::CreateMockIfNeeded()),
  1123. service_(NetworkService::CreateForTesting()) {}
  1124. ~NetworkChangeTest() override {}
  1125. NetworkService* service() const { return service_.get(); }
  1126. private:
  1127. base::test::TaskEnvironment task_environment_;
  1128. std::unique_ptr<net::NetworkChangeNotifier> network_change_notifier_;
  1129. std::unique_ptr<NetworkService> service_;
  1130. };
  1131. // mojom:NetworkChangeManager isn't supported on iOS.
  1132. // See the same ifdef in CreateNetworkChangeNotifierIfNeeded.
  1133. #if BUILDFLAG(IS_IOS)
  1134. #define MAYBE_NetworkChangeManagerRequest DISABLED_NetworkChangeManagerRequest
  1135. #else
  1136. #define MAYBE_NetworkChangeManagerRequest NetworkChangeManagerRequest
  1137. #endif
  1138. TEST_F(NetworkChangeTest, MAYBE_NetworkChangeManagerRequest) {
  1139. TestNetworkChangeManagerClient manager_client(service());
  1140. net::NetworkChangeNotifier::NotifyObserversOfNetworkChangeForTests(
  1141. net::NetworkChangeNotifier::CONNECTION_3G);
  1142. manager_client.WaitForNotification(mojom::ConnectionType::CONNECTION_3G);
  1143. }
  1144. class NetworkServiceNetworkChangeTest : public testing::Test {
  1145. public:
  1146. NetworkServiceNetworkChangeTest()
  1147. : task_environment_(base::test::TaskEnvironment::MainThreadType::IO),
  1148. network_change_notifier_(
  1149. net::NetworkChangeNotifier::CreateMockIfNeeded()),
  1150. service_(NetworkService::CreateForTesting()) {
  1151. service_->Bind(network_service_.BindNewPipeAndPassReceiver());
  1152. }
  1153. NetworkServiceNetworkChangeTest(const NetworkServiceNetworkChangeTest&) =
  1154. delete;
  1155. NetworkServiceNetworkChangeTest& operator=(
  1156. const NetworkServiceNetworkChangeTest&) = delete;
  1157. ~NetworkServiceNetworkChangeTest() override {}
  1158. mojom::NetworkService* service() { return network_service_.get(); }
  1159. private:
  1160. base::test::TaskEnvironment task_environment_;
  1161. std::unique_ptr<net::NetworkChangeNotifier> network_change_notifier_;
  1162. mojo::Remote<mojom::NetworkService> network_service_;
  1163. std::unique_ptr<NetworkService> service_;
  1164. };
  1165. TEST_F(NetworkServiceNetworkChangeTest, MAYBE_NetworkChangeManagerRequest) {
  1166. TestNetworkChangeManagerClient manager_client(service());
  1167. // Wait for the NetworkChangeManagerClient registration to be processed within
  1168. // the NetworkService impl before simulating a change. Flushing guarantees
  1169. // end-to-end connection of the client interface.
  1170. manager_client.Flush();
  1171. net::NetworkChangeNotifier::NotifyObserversOfNetworkChangeForTests(
  1172. net::NetworkChangeNotifier::CONNECTION_3G);
  1173. manager_client.WaitForNotification(mojom::ConnectionType::CONNECTION_3G);
  1174. }
  1175. class NetworkServiceNetworkDelegateTest : public NetworkServiceTest {
  1176. public:
  1177. NetworkServiceNetworkDelegateTest()
  1178. : NetworkServiceTest(
  1179. base::test::TaskEnvironment::TimeSource::SYSTEM_TIME) {}
  1180. NetworkServiceNetworkDelegateTest(const NetworkServiceNetworkDelegateTest&) =
  1181. delete;
  1182. NetworkServiceNetworkDelegateTest& operator=(
  1183. const NetworkServiceNetworkDelegateTest&) = delete;
  1184. ~NetworkServiceNetworkDelegateTest() override = default;
  1185. void CreateNetworkContext() {
  1186. mojom::NetworkContextParamsPtr context_params =
  1187. mojom::NetworkContextParams::New();
  1188. // Use a dummy CertVerifier that always passes cert verification, since
  1189. // these unittests don't need to test CertVerifier behavior.
  1190. context_params->cert_verifier_params =
  1191. FakeTestCertVerifierParamsFactory::GetCertVerifierParams();
  1192. service()->CreateNetworkContext(
  1193. network_context_.BindNewPipeAndPassReceiver(),
  1194. std::move(context_params));
  1195. }
  1196. void LoadURL(const GURL& url,
  1197. int options = mojom::kURLLoadOptionNone,
  1198. mojo::PendingRemote<mojom::URLLoaderNetworkServiceObserver>
  1199. url_loader_network_observer = mojo::NullRemote()) {
  1200. ResourceRequest request;
  1201. request.url = url;
  1202. request.method = "GET";
  1203. request.request_initiator = url::Origin();
  1204. StartLoadingURL(request, 0 /* process_id */, options,
  1205. std::move(url_loader_network_observer));
  1206. client_->RunUntilComplete();
  1207. }
  1208. void StartLoadingURL(
  1209. const ResourceRequest& request,
  1210. uint32_t process_id,
  1211. int options = mojom::kURLLoadOptionNone,
  1212. mojo::PendingRemote<mojom::URLLoaderNetworkServiceObserver>
  1213. url_loader_network_observer = mojo::NullRemote()) {
  1214. client_ = std::make_unique<TestURLLoaderClient>();
  1215. mojo::Remote<mojom::URLLoaderFactory> loader_factory;
  1216. mojom::URLLoaderFactoryParamsPtr params =
  1217. mojom::URLLoaderFactoryParams::New();
  1218. params->process_id = process_id;
  1219. params->is_corb_enabled = false;
  1220. params->url_loader_network_observer =
  1221. std::move(url_loader_network_observer);
  1222. network_context_->CreateURLLoaderFactory(
  1223. loader_factory.BindNewPipeAndPassReceiver(), std::move(params));
  1224. loader_.reset();
  1225. loader_factory->CreateLoaderAndStart(
  1226. loader_.BindNewPipeAndPassReceiver(), 1, options, request,
  1227. client_->CreateRemote(),
  1228. net::MutableNetworkTrafficAnnotationTag(TRAFFIC_ANNOTATION_FOR_TESTS));
  1229. }
  1230. net::EmbeddedTestServer* https_server() { return https_server_.get(); }
  1231. TestURLLoaderClient* client() { return client_.get(); }
  1232. protected:
  1233. void SetUp() override {
  1234. // Set up HTTPS server.
  1235. https_server_ = std::make_unique<net::EmbeddedTestServer>(
  1236. net::test_server::EmbeddedTestServer::TYPE_HTTPS);
  1237. https_server_->SetSSLConfig(net::EmbeddedTestServer::CERT_OK);
  1238. https_server_->RegisterRequestHandler(base::BindRepeating(
  1239. &NetworkServiceNetworkDelegateTest::HandleHTTPSRequest,
  1240. base::Unretained(this)));
  1241. ASSERT_TRUE(https_server_->Start());
  1242. }
  1243. // Responds with the header "<header>" if we have "header"=<header> query
  1244. // parameters in the url.
  1245. std::unique_ptr<net::test_server::HttpResponse> HandleHTTPSRequest(
  1246. const net::test_server::HttpRequest& request) {
  1247. std::string header;
  1248. if (net::GetValueForKeyInQuery(request.GetURL(), "header", &header)) {
  1249. std::unique_ptr<net::test_server::RawHttpResponse> response(
  1250. new net::test_server::RawHttpResponse("HTTP/1.1 200 OK\r\n", ""));
  1251. // Newlines are encoded as '%0A' in URLs.
  1252. const std::string newline_escape = "%0A";
  1253. std::size_t pos = header.find(newline_escape);
  1254. while (pos != std::string::npos) {
  1255. header.replace(pos, newline_escape.length(), "\r\n");
  1256. pos = header.find(newline_escape);
  1257. }
  1258. response->AddHeader(header);
  1259. return response;
  1260. }
  1261. return nullptr;
  1262. }
  1263. std::unique_ptr<net::EmbeddedTestServer> https_server_;
  1264. std::unique_ptr<TestURLLoaderClient> client_;
  1265. mojo::Remote<mojom::NetworkContext> network_context_;
  1266. mojo::Remote<mojom::URLLoader> loader_;
  1267. };
  1268. class ClearSiteDataAuthCertObserver : public TestURLLoaderNetworkObserver {
  1269. public:
  1270. ClearSiteDataAuthCertObserver() = default;
  1271. ~ClearSiteDataAuthCertObserver() override = default;
  1272. void OnClearSiteData(
  1273. const GURL& url,
  1274. const std::string& header_value,
  1275. int load_flags,
  1276. const absl::optional<net::CookiePartitionKey>& cookie_partition_key,
  1277. OnClearSiteDataCallback callback) override {
  1278. ++on_clear_site_data_counter_;
  1279. last_on_clear_site_data_header_value_ = header_value;
  1280. std::move(callback).Run();
  1281. }
  1282. int on_clear_site_data_counter() const { return on_clear_site_data_counter_; }
  1283. const std::string& last_on_clear_site_data_header_value() const {
  1284. return last_on_clear_site_data_header_value_;
  1285. }
  1286. void ClearOnClearSiteDataCounter() {
  1287. on_clear_site_data_counter_ = 0;
  1288. last_on_clear_site_data_header_value_.clear();
  1289. }
  1290. private:
  1291. int on_clear_site_data_counter_ = 0;
  1292. std::string last_on_clear_site_data_header_value_;
  1293. };
  1294. // Check that |NetworkServiceNetworkDelegate| handles Clear-Site-Data header
  1295. // w/ and w/o |NetworkContextCient|.
  1296. TEST_F(NetworkServiceNetworkDelegateTest, ClearSiteDataObserver) {
  1297. const char kClearCookiesHeader[] = "Clear-Site-Data: \"cookies\"";
  1298. CreateNetworkContext();
  1299. // Null |url_loader_network_observer|. The request should complete without
  1300. // being deferred.
  1301. GURL url = https_server()->GetURL("/foo");
  1302. url = AddQuery(url, "header", kClearCookiesHeader);
  1303. LoadURL(url);
  1304. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  1305. // With |url_loader_network_observer|. The request should go through
  1306. // |ClearSiteDataAuthCertObserver| and complete.
  1307. ClearSiteDataAuthCertObserver clear_site_observer;
  1308. url = https_server()->GetURL("/bar");
  1309. url = AddQuery(url, "header", kClearCookiesHeader);
  1310. EXPECT_EQ(0, clear_site_observer.on_clear_site_data_counter());
  1311. LoadURL(url, mojom::kURLLoadOptionNone, clear_site_observer.Bind());
  1312. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  1313. EXPECT_EQ(1, clear_site_observer.on_clear_site_data_counter());
  1314. }
  1315. // Check that headers are handled and passed to the client correctly.
  1316. TEST_F(NetworkServiceNetworkDelegateTest, HandleClearSiteDataHeaders) {
  1317. const char kClearCookiesHeaderValue[] = "\"cookies\"";
  1318. const char kClearCookiesHeader[] = "Clear-Site-Data: \"cookies\"";
  1319. CreateNetworkContext();
  1320. ClearSiteDataAuthCertObserver clear_site_observer;
  1321. // |passed_header_value| are only checked if |should_call_client| is true.
  1322. const struct TestCase {
  1323. std::string response_headers;
  1324. bool should_call_client;
  1325. std::string passed_header_value;
  1326. } kTestCases[] = {
  1327. // The throttle does not defer requests if there are no interesting
  1328. // response headers.
  1329. {"", false, ""},
  1330. {"Set-Cookie: abc=123;", false, ""},
  1331. {"Content-Type: image/png;", false, ""},
  1332. // Both malformed and valid Clear-Site-Data headers will defer requests
  1333. // and be passed to the client. It's client's duty to detect malformed
  1334. // headers.
  1335. {"Clear-Site-Data: cookies", true, "cookies"},
  1336. {"Clear-Site-Data: \"unknown type\"", true, "\"unknown type\""},
  1337. {"Clear-Site-Data: \"cookies\", \"unknown type\"", true,
  1338. "\"cookies\", \"unknown type\""},
  1339. {kClearCookiesHeader, true, kClearCookiesHeaderValue},
  1340. {base::StringPrintf("Content-Type: image/png;\n%s", kClearCookiesHeader),
  1341. true, kClearCookiesHeaderValue},
  1342. {base::StringPrintf("%s\nContent-Type: image/png;", kClearCookiesHeader),
  1343. true, kClearCookiesHeaderValue},
  1344. // Multiple instances of the header will be parsed correctly.
  1345. {base::StringPrintf("%s\n%s", kClearCookiesHeader, kClearCookiesHeader),
  1346. true, "\"cookies\", \"cookies\""},
  1347. };
  1348. for (const TestCase& test_case : kTestCases) {
  1349. SCOPED_TRACE(
  1350. base::StringPrintf("Headers:\n%s", test_case.response_headers.c_str()));
  1351. GURL url = https_server()->GetURL("/foo");
  1352. url = AddQuery(url, "header", test_case.response_headers);
  1353. EXPECT_EQ(0, clear_site_observer.on_clear_site_data_counter());
  1354. LoadURL(url, mojom::kURLLoadOptionNone, clear_site_observer.Bind());
  1355. EXPECT_EQ(net::OK, client()->completion_status().error_code);
  1356. if (test_case.should_call_client) {
  1357. EXPECT_EQ(1, clear_site_observer.on_clear_site_data_counter());
  1358. EXPECT_EQ(test_case.passed_header_value,
  1359. clear_site_observer.last_on_clear_site_data_header_value());
  1360. } else {
  1361. EXPECT_EQ(0, clear_site_observer.on_clear_site_data_counter());
  1362. }
  1363. clear_site_observer.ClearOnClearSiteDataCounter();
  1364. }
  1365. }
  1366. } // namespace
  1367. } // namespace network