OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "components/data_use_measurement/content/content_url_request_classifier
.h" | 5 #include "components/data_use_measurement/content/content_url_request_classifier
.h" |
6 | 6 |
| 7 #include <string> |
| 8 |
| 9 #include "base/strings/string_util.h" |
7 #include "content/public/browser/resource_request_info.h" | 10 #include "content/public/browser/resource_request_info.h" |
| 11 #include "content/public/common/resource_type.h" |
| 12 #include "net/http/http_response_headers.h" |
8 #include "net/url_request/url_request.h" | 13 #include "net/url_request/url_request.h" |
9 | 14 |
10 namespace data_use_measurement { | 15 namespace data_use_measurement { |
11 | 16 |
12 bool IsUserRequest(const net::URLRequest& request) { | 17 bool IsUserRequest(const net::URLRequest& request) { |
13 // The presence of ResourecRequestInfo in |request| implies that this request | 18 // The presence of ResourecRequestInfo in |request| implies that this request |
14 // was created for a content::WebContents. For now we could add a condition to | 19 // was created for a content::WebContents. For now we could add a condition to |
15 // check ProcessType in info is content::PROCESS_TYPE_RENDERER, but it won't | 20 // check ProcessType in info is content::PROCESS_TYPE_RENDERER, but it won't |
16 // be compatible with upcoming PlzNavigate architecture. So just existence of | 21 // be compatible with upcoming PlzNavigate architecture. So just existence of |
17 // ResourceRequestInfo is verified, and the current check should be compatible | 22 // ResourceRequestInfo is verified, and the current check should be compatible |
18 // with upcoming changes in PlzNavigate. | 23 // with upcoming changes in PlzNavigate. |
19 // TODO(rajendrant): Verify this condition for different use cases. See | 24 // TODO(rajendrant): Verify this condition for different use cases. See |
20 // crbug.com/626063. | 25 // crbug.com/626063. |
21 return content::ResourceRequestInfo::ForRequest(&request) != nullptr; | 26 return content::ResourceRequestInfo::ForRequest(&request) != nullptr; |
22 } | 27 } |
23 | 28 |
24 bool ContentURLRequestClassifier::IsUserRequest( | 29 bool ContentURLRequestClassifier::IsUserRequest( |
25 const net::URLRequest& request) const { | 30 const net::URLRequest& request) const { |
26 return data_use_measurement::IsUserRequest(request); | 31 return data_use_measurement::IsUserRequest(request); |
27 } | 32 } |
28 | 33 |
| 34 DataUseUserData::DataUseContentType ContentURLRequestClassifier::GetContentType( |
| 35 const net::URLRequest& request, |
| 36 const net::HttpResponseHeaders& response_headers) const { |
| 37 const content::ResourceRequestInfo* request_info = |
| 38 content::ResourceRequestInfo::ForRequest(&request); |
| 39 std::string mime_type; |
| 40 if (response_headers.GetMimeType(&mime_type)) { |
| 41 if (mime_type == "text/html" && request_info && |
| 42 request_info->GetResourceType() == |
| 43 content::ResourceType::RESOURCE_TYPE_MAIN_FRAME) { |
| 44 return DataUseUserData::MAIN_FRAME_HTML; |
| 45 } else if (mime_type == "text/html") { |
| 46 return DataUseUserData::NON_MAIN_FRAME_HTML; |
| 47 } else if (mime_type == "text/css") { |
| 48 return DataUseUserData::CSS; |
| 49 } else if (base::StartsWith(mime_type, "image/", |
| 50 base::CompareCase::SENSITIVE)) { |
| 51 return DataUseUserData::IMAGE; |
| 52 } else if (base::EndsWith(mime_type, "javascript", |
| 53 base::CompareCase::SENSITIVE) || |
| 54 base::EndsWith(mime_type, "ecmascript", |
| 55 base::CompareCase::SENSITIVE)) { |
| 56 return DataUseUserData::JAVASCRIPT; |
| 57 } else if (mime_type.find("font") != std::string::npos) { |
| 58 return DataUseUserData::FONT; |
| 59 } else if (base::StartsWith(mime_type, "audio/", |
| 60 base::CompareCase::SENSITIVE)) { |
| 61 return DataUseUserData::AUDIO; |
| 62 } else if (base::StartsWith(mime_type, "video/", |
| 63 base::CompareCase::SENSITIVE)) { |
| 64 return DataUseUserData::VIDEO; |
| 65 } |
| 66 } |
| 67 return DataUseUserData::OTHER; |
| 68 } |
| 69 |
29 } // namespace data_use_measurement | 70 } // namespace data_use_measurement |
OLD | NEW |