feat_expose_raw_response_headers_from_urlloader.patch 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  1. From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
  2. From: deepak1556 <[email protected]>
  3. Date: Tue, 17 Aug 2021 22:42:42 -0700
  4. Subject: feat: expose raw response headers from URLLoader
  5. With https://chromium-review.googlesource.com/c/chromium/src/+/2856099
  6. unfiltered response headers are only available via trusted channel
  7. through //services/network/public/mojom/devtools_observer.mojom.
  8. https://github.com/electron/electron/pull/30503/commits/28f4da1582d046e96cb58f3cbb590503e89dfd0d
  9. was an attempt to use this interface but given the original response is
  10. signalled on a different interface via URLLoaderClient::OnReceiveResponse
  11. it is harder to sync these data from two different channels for a synchronous
  12. event emitted on the SimpleURLLoaderWrapper::OnResponseStarted.
  13. This patch does the minimal approach to add back the raw response
  14. headers, moving forward we should find a way in upstream to provide
  15. access to these headers for loader clients created on the browser process.
  16. diff --git a/services/network/public/cpp/resource_request.cc b/services/network/public/cpp/resource_request.cc
  17. index f14c99863eb3f31ebc9ffd30a55f7536585fb561..d1f11d2d13445c9814d2e7227e5e02001377329b 100644
  18. --- a/services/network/public/cpp/resource_request.cc
  19. +++ b/services/network/public/cpp/resource_request.cc
  20. @@ -234,6 +234,7 @@ bool ResourceRequest::EqualsForTesting(const ResourceRequest& request) const {
  21. do_not_prompt_for_login == request.do_not_prompt_for_login &&
  22. is_main_frame == request.is_main_frame &&
  23. transition_type == request.transition_type &&
  24. + report_raw_headers == request.report_raw_headers &&
  25. previews_state == request.previews_state &&
  26. upgrade_if_insecure == request.upgrade_if_insecure &&
  27. is_revalidating == request.is_revalidating &&
  28. diff --git a/services/network/public/cpp/resource_request.h b/services/network/public/cpp/resource_request.h
  29. index 7c835d244420afe1c9ab6eac267c8789fb2afb01..564caebc55e1338e7124731d639cd6f01aebf595 100644
  30. --- a/services/network/public/cpp/resource_request.h
  31. +++ b/services/network/public/cpp/resource_request.h
  32. @@ -157,6 +157,7 @@ struct COMPONENT_EXPORT(NETWORK_CPP_BASE) ResourceRequest {
  33. bool do_not_prompt_for_login = false;
  34. bool is_main_frame = false;
  35. int transition_type = 0;
  36. + bool report_raw_headers = false;
  37. int previews_state = 0;
  38. bool upgrade_if_insecure = false;
  39. bool is_revalidating = false;
  40. diff --git a/services/network/public/cpp/url_request_mojom_traits.cc b/services/network/public/cpp/url_request_mojom_traits.cc
  41. index 9fefb0ca05e8f461d22b7f9875cc722232ccaed4..9c020bd1272ad2894f0b5af4c504d551b9927af0 100644
  42. --- a/services/network/public/cpp/url_request_mojom_traits.cc
  43. +++ b/services/network/public/cpp/url_request_mojom_traits.cc
  44. @@ -210,6 +210,7 @@ bool StructTraits<
  45. out->do_not_prompt_for_login = data.do_not_prompt_for_login();
  46. out->is_main_frame = data.is_main_frame();
  47. out->transition_type = data.transition_type();
  48. + out->report_raw_headers = data.report_raw_headers();
  49. out->previews_state = data.previews_state();
  50. out->upgrade_if_insecure = data.upgrade_if_insecure();
  51. out->is_revalidating = data.is_revalidating();
  52. diff --git a/services/network/public/cpp/url_request_mojom_traits.h b/services/network/public/cpp/url_request_mojom_traits.h
  53. index 53b1cd261ec419dd68906186a3da651fe792ede0..5945e6d9749ad2a888edfa0c4b00377a7190881b 100644
  54. --- a/services/network/public/cpp/url_request_mojom_traits.h
  55. +++ b/services/network/public/cpp/url_request_mojom_traits.h
  56. @@ -272,6 +272,9 @@ struct COMPONENT_EXPORT(NETWORK_CPP_BASE)
  57. static int32_t transition_type(const network::ResourceRequest& request) {
  58. return request.transition_type;
  59. }
  60. + static bool report_raw_headers(const network::ResourceRequest& request) {
  61. + return request.report_raw_headers;
  62. + }
  63. static int32_t previews_state(const network::ResourceRequest& request) {
  64. return request.previews_state;
  65. }
  66. diff --git a/services/network/public/mojom/url_request.mojom b/services/network/public/mojom/url_request.mojom
  67. index 27477d595fd22dc49329631a0b52ebdaafe1edb1..a4fd22381738648040cd72dce747285f5fcb698a 100644
  68. --- a/services/network/public/mojom/url_request.mojom
  69. +++ b/services/network/public/mojom/url_request.mojom
  70. @@ -318,6 +318,9 @@ struct URLRequest {
  71. // about this.
  72. int32 transition_type;
  73. + // Whether to provide unfiltered response headers.
  74. + bool report_raw_headers;
  75. +
  76. // Whether or not to request a Preview version of the resource or let the
  77. // browser decide.
  78. // Note: this is an enum of type PreviewsState.
  79. diff --git a/services/network/public/mojom/url_response_head.mojom b/services/network/public/mojom/url_response_head.mojom
  80. index 80f26b62b25bd6aa46886e3112b2ffd2698693fe..e72cab84db36eec86309160d4f7416ee431f5e34 100644
  81. --- a/services/network/public/mojom/url_response_head.mojom
  82. +++ b/services/network/public/mojom/url_response_head.mojom
  83. @@ -7,6 +7,7 @@ module network.mojom;
  84. import "mojo/public/mojom/base/time.mojom";
  85. import "mojo/public/mojom/base/unguessable_token.mojom";
  86. import "services/network/public/mojom/fetch_api.mojom";
  87. +import "services/network/public/mojom/http_raw_headers.mojom";
  88. import "services/network/public/mojom/ip_endpoint.mojom";
  89. import "services/network/public/mojom/load_timing_info.mojom";
  90. import "services/network/public/mojom/network_param.mojom";
  91. @@ -28,6 +29,9 @@ struct URLResponseHead {
  92. // The response headers or NULL if the URL type does not support headers.
  93. HttpResponseHeaders headers;
  94. + // Actual response headers, as obtained from the network stack.
  95. + array<HttpRawHeaderPair> raw_response_headers;
  96. +
  97. // The mime type of the response. This may be a derived value.
  98. string mime_type;
  99. diff --git a/services/network/url_loader.cc b/services/network/url_loader.cc
  100. index 59a89d471326069e183cdcc42af862fe19d7448e..39cd96d5e5785879bd3d82fb4c71fdf80c48e5ee 100644
  101. --- a/services/network/url_loader.cc
  102. +++ b/services/network/url_loader.cc
  103. @@ -541,6 +541,7 @@ URLLoader::URLLoader(
  104. peer_closed_handle_watcher_(FROM_HERE,
  105. mojo::SimpleWatcher::ArmingPolicy::MANUAL,
  106. base::SequencedTaskRunnerHandle::Get()),
  107. + report_raw_headers_(request.report_raw_headers),
  108. devtools_request_id_(request.devtools_request_id),
  109. request_mode_(request.mode),
  110. request_credentials_mode_(request.credentials_mode),
  111. @@ -699,7 +700,7 @@ URLLoader::URLLoader(
  112. url_request_->SetRequestHeadersCallback(base::BindRepeating(
  113. &URLLoader::SetRawRequestHeadersAndNotify, base::Unretained(this)));
  114. - if (devtools_request_id()) {
  115. + if (devtools_request_id() || report_raw_headers_) {
  116. url_request_->SetResponseHeadersCallback(base::BindRepeating(
  117. &URLLoader::SetRawResponseHeaders, base::Unretained(this)));
  118. }
  119. @@ -1349,6 +1350,19 @@ void URLLoader::OnResponseStarted(net::URLRequest* url_request, int net_error) {
  120. response_ = network::mojom::URLResponseHead::New();
  121. PopulateResourceResponse(url_request_.get(), is_load_timing_enabled_,
  122. options_, response_.get());
  123. + if (raw_response_headers_ && report_raw_headers_) {
  124. + std::vector<network::mojom::HttpRawHeaderPairPtr> header_array;
  125. + size_t iterator = 0;
  126. + std::string name, value;
  127. + while (raw_response_headers_->EnumerateHeaderLines(&iterator, &name, &value)) {
  128. + network::mojom::HttpRawHeaderPairPtr pair =
  129. + network::mojom::HttpRawHeaderPair::New();
  130. + pair->key = name;
  131. + pair->value = value;
  132. + header_array.push_back(std::move(pair));
  133. + }
  134. + response_->raw_response_headers = std::move(header_array);
  135. + }
  136. DispatchOnRawResponse();
  137. // Parse and remove the Trust Tokens response headers, if any are expected,
  138. diff --git a/services/network/url_loader.h b/services/network/url_loader.h
  139. index fcc47a2389c5172e3c674e413baa857183e0f2f8..af952b03ca87c4022400ba93d2e1fb8d3759960a 100644
  140. --- a/services/network/url_loader.h
  141. +++ b/services/network/url_loader.h
  142. @@ -503,6 +503,8 @@ class COMPONENT_EXPORT(NETWORK_SERVICE) URLLoader
  143. std::unique_ptr<ResourceScheduler::ScheduledResourceRequest>
  144. resource_scheduler_request_handle_;
  145. + // Whether client requested raw headers.
  146. + bool report_raw_headers_ = false;
  147. bool enable_reporting_raw_headers_ = false;
  148. bool seen_raw_request_headers_ = false;
  149. scoped_refptr<const net::HttpResponseHeaders> raw_response_headers_;