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
103
104
105
106
107
108
109
110
111
112
113
|
// Copyright 2022 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_ASH_ARC_SESSION_ARC_DISK_SPACE_MONITOR_H_
#define CHROME_BROWSER_ASH_ARC_SESSION_ARC_DISK_SPACE_MONITOR_H_
#include <optional>
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "base/timer/timer.h"
#include "chrome/browser/ash/arc/session/arc_session_manager_observer.h"
namespace arc {
// These thresholds are chosen based on UMA stats. (go/arcvm-virtio-blk-sparse)
// Show a pre-stop warning notification if free disk space is lower than this.
inline constexpr int64_t kDiskSpaceThresholdForPreStopNotification =
1LL << 30; // 1GB
// Stop ARC and show a post-stop warning notification if free disk space is
// lower than this.
//
// 256MB
inline constexpr int64_t kDiskSpaceThresholdForStoppingArc = 256LL << 20;
// TODO(b/233030867): Choose these values based on some logic
// instead of deciding them on a hunch.
// Disk space check interval used when free disk space is lower than
// kDiskSpaceThresholdForPreStopNotification.
inline constexpr base::TimeDelta kDiskSpaceCheckIntervalShort =
base::Seconds(1);
// Disk space check interval used when free disk space is higher than
// kDiskSpaceThresholdForPreStopNotification.
inline constexpr base::TimeDelta kDiskSpaceCheckIntervalLong =
base::Seconds(10);
// A pre-stop warning notification should not be shown more than once within
// this interval.
// TODO(b/237040345): Finalize the value.
inline constexpr base::TimeDelta kPreStopNotificationReshowInterval =
base::Minutes(2);
// Notifier ID of ArcDiskSpaceMonitor.
inline constexpr char kDiskSpaceMonitorNotifierId[] = "arc_disk_space_monitor";
// Notification ID of the pre-stop warning notification.
inline constexpr char kLowDiskSpacePreStopNotificationId[] =
"arc_low_disk_space_pre_stop";
// Notification ID of the post-stop warning notification.
inline constexpr char kLowDiskSpacePostStopNotificationId[] =
"arc_low_disk_space_post_stop";
// Monitors disk usage. Requests stopping ARC and/or shows a warning
// notification when device's free disk space becomes lower than a threshold.
// Used when arcvm_virtio_blk_data is enabled. (go/arcvm-virtio-blk-sparse)
// TODO(b/233030867): Delete this after we switch ARCVM to using Storage
// Balloons for disk space management.
class ArcDiskSpaceMonitor : public ArcSessionManagerObserver {
public:
ArcDiskSpaceMonitor();
~ArcDiskSpaceMonitor() override;
ArcDiskSpaceMonitor(const ArcDiskSpaceMonitor&) = delete;
ArcDiskSpaceMonitor& operator=(const ArcDiskSpaceMonitor&) = delete;
bool IsTimerRunningForTesting() { return timer_.IsRunning(); }
void SetOnGetFreeDiskSpaceCallbackForTesting(
base::OnceCallback<void()> callback) {
on_get_free_disk_space_callback_for_testing_ = std::move(callback);
}
base::TimeDelta GetTimerCurrentDelayForTesting() {
return timer_.GetCurrentDelay();
}
// ArcSessionManagerObserver overrides.
void OnArcStarted() override;
void OnArcSessionStopped(ArcStopReason stop_reason) override;
private:
// Schedules calling CheckDiskSpace().
void ScheduleCheckDiskSpace(base::TimeDelta delay);
// Checks disk usage, requests stopping ARC and/or shows a warning
// notification based on the free disk space.
void CheckDiskSpace();
// Used as a callback function.
void OnGetFreeDiskSpace(std::optional<int64_t> reply);
// Shows a pre-stop warning notification if |is_pre_stop| is true and the
// same notification was not shown within kPreStopNotificationReshowInterval.
// Always shows a post-stop warning notification if |is_pre_stop| is false.
void MaybeShowNotification(bool is_pre_stop);
// The last time when a pre-stop warning notification was shown.
base::Time pre_stop_notification_last_shown_time_;
// Used for periodically calling CheckDiskSpace().
base::OneShotTimer timer_;
base::OnceCallback<void()> on_get_free_disk_space_callback_for_testing_;
// WeakPtrFactory to use callbacks.
base::WeakPtrFactory<ArcDiskSpaceMonitor> weak_ptr_factory_{this};
};
} // namespace arc
#endif // CHROME_BROWSER_ASH_ARC_SESSION_ARC_DISK_SPACE_MONITOR_H_
|
