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
64
65
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef EXTENSIONS_BROWSER_API_DECLARATIVE_NET_REQUEST_GLOBAL_RULES_TRACKER_H_
#define EXTENSIONS_BROWSER_API_DECLARATIVE_NET_REQUEST_GLOBAL_RULES_TRACKER_H_
#include "base/memory/raw_ptr.h"
#include "extensions/common/extension_id.h"
namespace extensions {
class ExtensionPrefs;
class ExtensionRegistry;
namespace declarative_net_request {
class GlobalRulesTracker {
public:
explicit GlobalRulesTracker(ExtensionPrefs* extension_prefs,
ExtensionRegistry* extension_registry);
~GlobalRulesTracker();
GlobalRulesTracker(const GlobalRulesTracker& other) = delete;
GlobalRulesTracker& operator=(const GlobalRulesTracker& other) = delete;
// Called when an extension's allocated static rule count is updated.
// Returns whether the extension's new rule count will result in the total
// rule count staying within the global rule limit. The extension's allocated
// rule count is updated iff true is returned.
bool OnExtensionRuleCountUpdated(const ExtensionId& extension_id,
size_t new_rule_count);
// Returns the number of unallocated rules in the global pool.
size_t GetUnallocatedRuleCount() const;
// Returns the number of rules in the global pool available for the extension
// before the global limit is reached. This includes the extension's global
// pool allocation.
size_t GetAvailableAllocation(const ExtensionId& extension_id) const;
// Clears the extension's allocated rule count.
void ClearExtensionAllocation(const ExtensionId& extension_id);
// Returns `allocated_global_rule_count_`.
size_t GetAllocatedGlobalRuleCountForTesting() const;
private:
// Returns the number of rules in the global pool allocated for the extension.
// Note: Some of these allocated rules might not be enabled by the extension.
size_t GetAllocationInPrefs(const ExtensionId& extension_id) const;
// The number of static rules from all extensions which contribute to the
// global rule pool. Any enabled static rules for an extension past
// `GUARANTEED_MINIMUM_STATIC_RULES` count towards this. This value must never
// exceed `kMaxStaticRulesPerProfile`.
size_t allocated_global_rule_count_ = 0;
const raw_ptr<ExtensionPrefs> extension_prefs_;
const raw_ptr<ExtensionRegistry> extension_registry_;
};
} // namespace declarative_net_request
} // namespace extensions
#endif // EXTENSIONS_BROWSER_API_DECLARATIVE_NET_REQUEST_GLOBAL_RULES_TRACKER_H_
|
