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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
// Copyright 2015 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_SOCKET_APP_FIREWALL_HOLE_MANAGER_H_
#define EXTENSIONS_BROWSER_API_SOCKET_APP_FIREWALL_HOLE_MANAGER_H_
#include <stdint.h>
#include <map>
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/scoped_observation.h"
#include "chromeos/components/firewall_hole/firewall_hole.h"
#include "extensions/browser/app_window/app_window_registry.h"
#include "extensions/common/extension_id.h"
namespace content {
class BrowserContext;
} // namespace content
namespace extensions {
class AppFirewallHoleManager;
// Represents an open port in the system firewall that will be opened and closed
// automatically when the application has a visible window or not. The hole is
// closed on destruction.
class AppFirewallHole {
public:
~AppFirewallHole();
const ExtensionId& extension_id() const { return extension_id_; }
private:
friend class AppFirewallHoleManager;
AppFirewallHole(const base::WeakPtr<AppFirewallHoleManager>& manager,
chromeos::FirewallHole::PortType type,
uint16_t port,
const ExtensionId& extension_id);
void SetVisible(bool app_visible);
void OnFirewallHoleOpened(
std::unique_ptr<chromeos::FirewallHole> firewall_hole);
chromeos::FirewallHole::PortType type_;
uint16_t port_;
ExtensionId extension_id_;
bool app_visible_ = false;
base::WeakPtr<AppFirewallHoleManager> manager_;
// This will hold the FirewallHole object if one is opened.
std::unique_ptr<chromeos::FirewallHole> firewall_hole_;
base::WeakPtrFactory<AppFirewallHole> weak_factory_{this};
};
// Tracks ports in the system firewall opened by an application so that they
// may be automatically opened and closed only when the application has a
// visible window.
class AppFirewallHoleManager : public KeyedService,
public AppWindowRegistry::Observer {
public:
explicit AppFirewallHoleManager(content::BrowserContext* context);
~AppFirewallHoleManager() override;
// Returns the instance for a given browser context, or NULL if none.
static AppFirewallHoleManager* Get(content::BrowserContext* context);
// Opens a port on the system firewall if the associated application is
// currently visible.
std::unique_ptr<AppFirewallHole> Open(chromeos::FirewallHole::PortType type,
uint16_t port,
const ExtensionId& extension_id);
static void EnsureFactoryBuilt();
private:
friend class AppFirewallHole;
void Close(AppFirewallHole* hole);
// AppWindowRegistry::Observer
void OnAppWindowRemoved(AppWindow* app_window) override;
void OnAppWindowHidden(AppWindow* app_window) override;
void OnAppWindowShown(AppWindow* app_window, bool was_hidden) override;
raw_ptr<content::BrowserContext> context_;
base::ScopedObservation<AppWindowRegistry, AppWindowRegistry::Observer>
observation_{this};
std::multimap<ExtensionId, raw_ptr<AppFirewallHole, CtnExperimental>>
tracked_holes_;
base::WeakPtrFactory<AppFirewallHoleManager> weak_factory_{this};
};
} // namespace extensions
#endif // EXTENSIONS_BROWSER_API_SOCKET_APP_FIREWALL_HOLE_MANAGER_H_
|
