1 // Copyright 2020 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 
5 #include "extensions/common/identifiability_metrics.h"
6 
7 #include "extensions/common/extension_set.h"
8 #include "services/metrics/public/cpp/ukm_recorder.h"
9 #include "services/metrics/public/cpp/ukm_source_id.h"
10 #include "third_party/blink/public/common/privacy_budget/identifiability_metric_builder.h"
11 
12 namespace extensions {
13 
SurfaceForExtension(blink::IdentifiableSurface::Type type,const ExtensionId & extension_id)14 blink::IdentifiableSurface SurfaceForExtension(
15     blink::IdentifiableSurface::Type type,
16     const ExtensionId& extension_id) {
17   return blink::IdentifiableSurface::FromTypeAndToken(
18       type, base::as_bytes(base::make_span(extension_id)));
19 }
20 
RecordExtensionResourceAccessResult(ukm::SourceIdObj ukm_source_id,const GURL & gurl,ExtensionResourceAccessResult result)21 void RecordExtensionResourceAccessResult(ukm::SourceIdObj ukm_source_id,
22                                          const GURL& gurl,
23                                          ExtensionResourceAccessResult result) {
24   if (ukm_source_id == ukm::kInvalidSourceIdObj)
25     return;
26 
27   ExtensionId extension_id = ExtensionSet::GetExtensionIdByURL(gurl);
28   blink::IdentifiabilityMetricBuilder(ukm_source_id)
29       .Set(SurfaceForExtension(
30                blink::IdentifiableSurface::Type::kExtensionFileAccess,
31                extension_id),
32            result)
33       .Record(ukm::UkmRecorder::Get());
34 }
35 
RecordContentScriptInjection(ukm::SourceIdObj ukm_source_id,const ExtensionId & extension_id)36 void RecordContentScriptInjection(ukm::SourceIdObj ukm_source_id,
37                                   const ExtensionId& extension_id) {
38   if (ukm_source_id == ukm::kInvalidSourceIdObj)
39     return;
40   blink::IdentifiabilityMetricBuilder(ukm_source_id)
41       .Set(SurfaceForExtension(
42                blink::IdentifiableSurface::Type::kExtensionContentScript,
43                extension_id),
44            /* Succeeded= */ true)
45       .Record(ukm::UkmRecorder::Get());
46 }
47 
RecordNetworkRequestBlocked(ukm::SourceIdObj ukm_source_id,const ExtensionId & extension_id)48 void RecordNetworkRequestBlocked(ukm::SourceIdObj ukm_source_id,
49                                  const ExtensionId& extension_id) {
50   if (ukm_source_id == ukm::kInvalidSourceIdObj)
51     return;
52   blink::IdentifiabilityMetricBuilder(ukm_source_id)
53       .Set(SurfaceForExtension(
54                blink::IdentifiableSurface::Type::kExtensionCancelRequest,
55                extension_id),
56            /* Succeeded= */ true)
57       .Record(ukm::UkmRecorder::Get());
58 }
59 
60 }  // namespace extensions
61