1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_SAFE_BROWSING_EXTENSION_TELEMETRY_EXTENSION_JS_CALLSTACKS_H_
#define CHROME_BROWSER_SAFE_BROWSING_EXTENSION_TELEMETRY_EXTENSION_JS_CALLSTACKS_H_
#include "base/containers/flat_map.h"
#include "components/safe_browsing/core/common/proto/csd.pb.h"
#include "extensions/common/stack_frame.h"
namespace safe_browsing {
using ::google::protobuf::RepeatedPtrField;
using SignalInfoJSCallStackFrame =
ExtensionTelemetryReportRequest::SignalInfo::JSCallStackFrame;
using SignalInfoJSCallStack =
ExtensionTelemetryReportRequest::SignalInfo::JSCallStack;
// This is a helper class used to store and retrieve JS callstacks
// for extension API signals. A signal processor class, for e.g.
// TabsApiSignalProcessor, includes this class as a member to store
// callstacks for corresponding (for e.g, chrome.tabs) API calls.
class ExtensionJSCallStacks {
public:
ExtensionJSCallStacks();
~ExtensionJSCallStacks();
ExtensionJSCallStacks(const ExtensionJSCallStacks&);
// Stores the input `callstack`. Does nothing if:
// - the callstack being added is empty
// - max number of callstacks already stored
// - the callstack is a duplicate of one already stored
//
void Add(const extensions::StackTrace& callstack);
// Returns all the callstacks stored in the SignalInfo protobuf format.
RepeatedPtrField<SignalInfoJSCallStack> GetAll();
// Returns the current number of callstacks stored.
unsigned int NumCallStacks() { return siginfo_callstack_map_.size(); }
// Returns the maximum number of distinct callstacks that
// will be stored. Attempts to add more will be ignored.
static unsigned int MaxCallStacks();
// Helpers.
static SignalInfoJSCallStack ToSignalInfoJSCallStack(
const extensions::StackTrace& stack_trace);
static extensions::StackTrace ToExtensionsStackTrace(
const SignalInfoJSCallStack& callstack);
static std::string GetUniqueId(const SignalInfoJSCallStack& callstack);
// Used for debug logging only.
static std::string SignalInfoJSCallStackAsString(
const SignalInfoJSCallStack& callstack);
protected:
base::flat_map<std::string, SignalInfoJSCallStack> siginfo_callstack_map_;
};
} // namespace safe_browsing
#endif // CHROME_BROWSER_SAFE_BROWSING_EXTENSION_TELEMETRY_EXTENSION_JS_CALLSTACKS_H_
|
