Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 1 | // Copyright 2017 The Chromium OS Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #ifndef CROS_DISKS_RENAME_MANAGER_H_ |
| 6 | #define CROS_DISKS_RENAME_MANAGER_H_ |
| 7 | |
| 8 | #include <map> |
| 9 | #include <string> |
| 10 | |
Ben Chan | 445852f | 2017-10-02 23:00:16 -0700 | [diff] [blame] | 11 | #include <base/memory/weak_ptr.h> |
Simon Glass | 2b1da09 | 2020-05-21 12:24:16 -0600 | [diff] [blame] | 12 | #include <brillo/process/process_reaper.h> |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 13 | #include <chromeos/dbus/service_constants.h> |
| 14 | #include <gtest/gtest_prod.h> |
| 15 | |
| 16 | #include "cros-disks/sandboxed_process.h" |
| 17 | |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 18 | namespace cros_disks { |
| 19 | |
| 20 | class Platform; |
| 21 | class RenameManagerObserverInterface; |
| 22 | |
| 23 | class RenameManager { |
| 24 | public: |
Ben Chan | 445852f | 2017-10-02 23:00:16 -0700 | [diff] [blame] | 25 | RenameManager(Platform* platform, brillo::ProcessReaper* process_reaper); |
Qijiang Fan | 6bc59e1 | 2020-11-11 02:51:06 +0900 | [diff] [blame] | 26 | RenameManager(const RenameManager&) = delete; |
| 27 | RenameManager& operator=(const RenameManager&) = delete; |
| 28 | |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 29 | ~RenameManager(); |
| 30 | |
| 31 | // Starts a renaming process of a given device. |
| 32 | RenameErrorType StartRenaming(const std::string& device_path, |
| 33 | const std::string& device_file, |
| 34 | const std::string& volume_name, |
| 35 | const std::string& filesystem_type); |
| 36 | |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 37 | void set_observer(RenameManagerObserverInterface* observer) { |
| 38 | observer_ = observer; |
| 39 | } |
| 40 | |
| 41 | private: |
| 42 | FRIEND_TEST(RenameManagerTest, CanRename); |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 43 | |
Ben Chan | 445852f | 2017-10-02 23:00:16 -0700 | [diff] [blame] | 44 | void OnRenameProcessTerminated(const std::string& device_path, |
| 45 | const siginfo_t& info); |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 46 | |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 47 | // Returns true if renaming |source_path| is supported. |
| 48 | bool CanRename(const std::string& source_path) const; |
| 49 | |
Ben Chan | 445852f | 2017-10-02 23:00:16 -0700 | [diff] [blame] | 50 | // Platform service |
| 51 | Platform* platform_; |
| 52 | |
| 53 | brillo::ProcessReaper* process_reaper_; |
| 54 | |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 55 | // A list of outstanding renaming processes indexed by device path. |
| 56 | std::map<std::string, SandboxedProcess> rename_process_; |
| 57 | |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 58 | RenameManagerObserverInterface* observer_; |
| 59 | |
Ben Chan | 445852f | 2017-10-02 23:00:16 -0700 | [diff] [blame] | 60 | base::WeakPtrFactory<RenameManager> weak_ptr_factory_; |
Klemen Kozjek | b065885 | 2017-08-15 13:03:48 +0900 | [diff] [blame] | 61 | }; |
| 62 | |
| 63 | } // namespace cros_disks |
| 64 | |
| 65 | #endif // CROS_DISKS_RENAME_MANAGER_H_ |