| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/browser/webui/web_ui_data_source_impl.h" | 5 #include "content/browser/webui/web_ui_data_source_impl.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 | 8 |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 67 return URLDataSource::GetContentSecurityPolicyObjectSrc(); | 67 return URLDataSource::GetContentSecurityPolicyObjectSrc(); |
| 68 } | 68 } |
| 69 std::string GetContentSecurityPolicyChildSrc() const override { | 69 std::string GetContentSecurityPolicyChildSrc() const override { |
| 70 if (parent_->frame_src_set_) | 70 if (parent_->frame_src_set_) |
| 71 return parent_->frame_src_; | 71 return parent_->frame_src_; |
| 72 return URLDataSource::GetContentSecurityPolicyChildSrc(); | 72 return URLDataSource::GetContentSecurityPolicyChildSrc(); |
| 73 } | 73 } |
| 74 bool ShouldDenyXFrameOptions() const override { | 74 bool ShouldDenyXFrameOptions() const override { |
| 75 return parent_->deny_xframe_options_; | 75 return parent_->deny_xframe_options_; |
| 76 } | 76 } |
| 77 std::vector<std::string> GetPreloadHeaders() const override { |
| 78 return parent_->preload_headers_; |
| 79 } |
| 77 | 80 |
| 78 private: | 81 private: |
| 79 WebUIDataSourceImpl* parent_; | 82 WebUIDataSourceImpl* parent_; |
| 80 }; | 83 }; |
| 81 | 84 |
| 82 WebUIDataSourceImpl::WebUIDataSourceImpl(const std::string& source_name) | 85 WebUIDataSourceImpl::WebUIDataSourceImpl(const std::string& source_name) |
| 83 : URLDataSourceImpl(source_name, new InternalDataSource(this)), | 86 : URLDataSourceImpl(source_name, new InternalDataSource(this)), |
| 84 source_name_(source_name), | 87 source_name_(source_name), |
| 85 default_resource_(-1), | 88 default_resource_(-1), |
| 86 add_csp_(true), | 89 add_csp_(true), |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 148 | 151 |
| 149 void WebUIDataSourceImpl::SetDefaultResource(int resource_id) { | 152 void WebUIDataSourceImpl::SetDefaultResource(int resource_id) { |
| 150 default_resource_ = resource_id; | 153 default_resource_ = resource_id; |
| 151 } | 154 } |
| 152 | 155 |
| 153 void WebUIDataSourceImpl::SetRequestFilter( | 156 void WebUIDataSourceImpl::SetRequestFilter( |
| 154 const WebUIDataSource::HandleRequestCallback& callback) { | 157 const WebUIDataSource::HandleRequestCallback& callback) { |
| 155 filter_callback_ = callback; | 158 filter_callback_ = callback; |
| 156 } | 159 } |
| 157 | 160 |
| 161 void WebUIDataSourceImpl::AddPreloadHeader(const std::string& preload_header) { |
| 162 preload_headers_.push_back(preload_header); |
| 163 } |
| 164 |
| 158 void WebUIDataSourceImpl::DisableReplaceExistingSource() { | 165 void WebUIDataSourceImpl::DisableReplaceExistingSource() { |
| 159 replace_existing_source_ = false; | 166 replace_existing_source_ = false; |
| 160 } | 167 } |
| 161 | 168 |
| 162 void WebUIDataSourceImpl::DisableContentSecurityPolicy() { | 169 void WebUIDataSourceImpl::DisableContentSecurityPolicy() { |
| 163 add_csp_ = false; | 170 add_csp_ = false; |
| 164 } | 171 } |
| 165 | 172 |
| 166 void WebUIDataSourceImpl::OverrideContentSecurityPolicyObjectSrc( | 173 void WebUIDataSourceImpl::OverrideContentSecurityPolicyObjectSrc( |
| 167 const std::string& data) { | 174 const std::string& data) { |
| 168 object_src_set_ = true; | 175 object_src_set_ = true; |
| 169 object_src_ = data; | 176 object_src_ = data; |
| 170 } | 177 } |
| 171 | 178 |
| 172 void WebUIDataSourceImpl::OverrideContentSecurityPolicyChildSrc( | 179 void WebUIDataSourceImpl::OverrideContentSecurityPolicyChildSrc( |
| 173 const std::string& data) { | 180 const std::string& data) { |
| 174 frame_src_set_ = true; | 181 frame_src_set_ = true; |
| 175 frame_src_ = data; | 182 frame_src_ = data; |
| 176 } | 183 } |
| 177 | 184 |
| 178 void WebUIDataSourceImpl::DisableDenyXFrameOptions() { | 185 void WebUIDataSourceImpl::DisableDenyXFrameOptions() { |
| 179 deny_xframe_options_ = false; | 186 deny_xframe_options_ = false; |
| 180 } | 187 } |
| 181 | 188 |
| 189 const std::map<std::string, int>& WebUIDataSourceImpl::path_to_idr_map() const { |
| 190 return path_to_idr_map_; |
| 191 } |
| 192 |
| 182 std::string WebUIDataSourceImpl::GetSource() const { | 193 std::string WebUIDataSourceImpl::GetSource() const { |
| 183 return source_name_; | 194 return source_name_; |
| 184 } | 195 } |
| 185 | 196 |
| 186 std::string WebUIDataSourceImpl::GetMimeType(const std::string& path) const { | 197 std::string WebUIDataSourceImpl::GetMimeType(const std::string& path) const { |
| 187 // Remove the query string for to determine the mime type. | 198 // Remove the query string for to determine the mime type. |
| 188 std::string file_path = path.substr(0, path.find_first_of('?')); | 199 std::string file_path = path.substr(0, path.find_first_of('?')); |
| 189 | 200 |
| 190 if (base::EndsWith(file_path, ".css", base::CompareCase::INSENSITIVE_ASCII)) | 201 if (base::EndsWith(file_path, ".css", base::CompareCase::INSENSITIVE_ASCII)) |
| 191 return "text/css"; | 202 return "text/css"; |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 252 } | 263 } |
| 253 | 264 |
| 254 void WebUIDataSourceImpl::SendLocalizedStringsAsJSON( | 265 void WebUIDataSourceImpl::SendLocalizedStringsAsJSON( |
| 255 const URLDataSource::GotDataCallback& callback) { | 266 const URLDataSource::GotDataCallback& callback) { |
| 256 std::string template_data; | 267 std::string template_data; |
| 257 webui::AppendJsonJS(&localized_strings_, &template_data); | 268 webui::AppendJsonJS(&localized_strings_, &template_data); |
| 258 callback.Run(base::RefCountedString::TakeString(&template_data)); | 269 callback.Run(base::RefCountedString::TakeString(&template_data)); |
| 259 } | 270 } |
| 260 | 271 |
| 261 } // namespace content | 272 } // namespace content |
| OLD | NEW |