feat_expose_raw_response_headers_from_urlloader.patch 8.2 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 87868727a78223baef9ffd2591f49fced240ef4e..f6a6ff28f33e2d1c065f2abeb96733b2d28a6ea1 100644
  18. --- a/services/network/public/cpp/resource_request.cc
  19. +++ b/services/network/public/cpp/resource_request.cc
  20. @@ -233,6 +233,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 8b9a7b36f53a4cfcd159ac18c06d1724072013c8..ef0191bae8a07e531ae129cf32b22f4930c5e266 100644
  30. --- a/services/network/public/cpp/resource_request.h
  31. +++ b/services/network/public/cpp/resource_request.h
  32. @@ -156,6 +156,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 1fcf54cac11c38352e14774cd08bcaa162443e9c..5356da11391d52a8f9aaa57a27616cee6dc0f2b6 100644
  42. --- a/services/network/public/cpp/url_request_mojom_traits.cc
  43. +++ b/services/network/public/cpp/url_request_mojom_traits.cc
  44. @@ -209,6 +209,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 1b8dbc0538d0af843e40edc41505d08f9034f97b..270822eb756090f8a74f34823009942ed21e8616 100644
  54. --- a/services/network/public/cpp/url_request_mojom_traits.h
  55. +++ b/services/network/public/cpp/url_request_mojom_traits.h
  56. @@ -269,6 +269,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 79b5d03ded03ced9e6ff4d17d10935004bfb0062..923883fd010f9621c790dd5381a7e1f0cb36e740 100644
  68. --- a/services/network/public/mojom/url_request.mojom
  69. +++ b/services/network/public/mojom/url_request.mojom
  70. @@ -312,6 +312,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 4c4cc16db82d7434573f7740855fbe72d68815e6..f71290800b6bb51a39b1f86be36f02d602ac3397 100644
  81. --- a/services/network/public/mojom/url_response_head.mojom
  82. +++ b/services/network/public/mojom/url_response_head.mojom
  83. @@ -8,6 +8,7 @@ import "mojo/public/mojom/base/time.mojom";
  84. import "mojo/public/mojom/base/unguessable_token.mojom";
  85. import "services/network/public/mojom/fetch_api.mojom";
  86. import "services/network/public/mojom/ip_address_space.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. @@ -29,6 +30,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 1530c51f9090ce51de1f3bc495ca83590d6430ab..d053aa813ab53092dcefb81116ff850318f439c2 100644
  101. --- a/services/network/url_loader.cc
  102. +++ b/services/network/url_loader.cc
  103. @@ -470,6 +470,7 @@ URLLoader::URLLoader(
  104. mojo::SimpleWatcher::ArmingPolicy::MANUAL,
  105. base::SequencedTaskRunnerHandle::Get()),
  106. per_factory_corb_state_(context.GetMutableCorbState()),
  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. @@ -637,7 +638,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. @@ -1408,6 +1409,19 @@ void URLLoader::OnResponseStarted(net::URLRequest* url_request, int net_error) {
  120. }
  121. response_ = BuildResponseHead();
  122. + if (raw_response_headers_ && report_raw_headers_) {
  123. + std::vector<network::mojom::HttpRawHeaderPairPtr> header_array;
  124. + size_t iterator = 0;
  125. + std::string name, value;
  126. + while (raw_response_headers_->EnumerateHeaderLines(&iterator, &name, &value)) {
  127. + network::mojom::HttpRawHeaderPairPtr pair =
  128. + network::mojom::HttpRawHeaderPair::New();
  129. + pair->key = name;
  130. + pair->value = value;
  131. + header_array.push_back(std::move(pair));
  132. + }
  133. + response_->raw_response_headers = std::move(header_array);
  134. + }
  135. DispatchOnRawResponse();
  136. // Parse and remove the Trust Tokens response headers, if any are expected,
  137. diff --git a/services/network/url_loader.h b/services/network/url_loader.h
  138. index 53822d17b20f4e6d163d0dc84eea6c455254a5f7..c494a63b0c1dcb39120e8aabc84a60ebbf2ff224 100644
  139. --- a/services/network/url_loader.h
  140. +++ b/services/network/url_loader.h
  141. @@ -502,6 +502,8 @@ class COMPONENT_EXPORT(NETWORK_SERVICE) URLLoader
  142. std::unique_ptr<ResourceScheduler::ScheduledResourceRequest>
  143. resource_scheduler_request_handle_;
  144. + // Whether client requested raw headers.
  145. + bool report_raw_headers_ = false;
  146. bool enable_reporting_raw_headers_ = false;
  147. bool seen_raw_request_headers_ = false;
  148. scoped_refptr<const net::HttpResponseHeaders> raw_response_headers_;