| // Copyright 2019 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef SRC_SYS_APPMGR_STORAGE_WATCHDOG_H_ |
| #define SRC_SYS_APPMGR_STORAGE_WATCHDOG_H_ |
| |
| #include <fidl/fuchsia.io/cpp/wire.h> |
| #include <lib/async/dispatcher.h> |
| #include <lib/inspect/cpp/vmo/types.h> |
| #include <lib/zx/channel.h> |
| |
| #include <string> |
| |
| const inspect::StringReference kBytesUsed("byte_used"); |
| const inspect::StringReference kBytesAvailable("byte_available"); |
| |
| // StorageWatchdog can be used to observe the storage usage on a given |
| // partition, and when the storage usage reaches 95% the isolated component |
| // storage under a given path will be deleted. |
| class StorageWatchdog { |
| public: |
| StorageWatchdog(inspect::Node node, std::string path_to_watch, std::string path_to_clean) |
| : node_(std::move(node)), |
| bytes_used_(node_.CreateUint(kBytesUsed, 0)), |
| bytes_avail_(node_.CreateUint(kBytesAvailable, 0)), |
| path_to_watch_(std::move(path_to_watch)), |
| path_to_clean_(std::move(path_to_clean)) {} |
| |
| void Run(async_dispatcher_t* dispatcher); |
| struct StorageUsage { |
| size_t avail = 0; |
| size_t used = 0; |
| size_t percent() const { |
| if (avail > 0) { |
| return used * 100 / avail; |
| } |
| return 0; |
| } |
| }; |
| StorageUsage GetStorageUsage(); |
| void CheckStorage(async_dispatcher_t* dispatcher, |
| size_t threshold_purge_percent = kCachePurgeThresholdPct); |
| void PurgeCache(); |
| |
| static constexpr size_t kCachePurgeThresholdPct = 95; |
| |
| protected: |
| virtual zx_status_t GetFilesystemInfo(zx_handle_t directory, |
| fuchsia_io::wire::FilesystemInfo* out_info); |
| |
| private: |
| inspect::Node node_; |
| inspect::UintProperty bytes_used_; |
| inspect::UintProperty bytes_avail_; |
| const std::string path_to_watch_; |
| const std::string path_to_clean_; |
| }; |
| |
| #endif // SRC_SYS_APPMGR_STORAGE_WATCHDOG_H_ |