// Copyright 2014 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. // StorageMonitorCros listens for mount point changes and notifies listeners // about the addition and deletion of media devices. This class lives on the // UI thread. #ifndef COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_CHROMEOS_H_ #define COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_CHROMEOS_H_ #include "build/chromeos_buildflags.h" #if !BUILDFLAG(IS_CHROMEOS_ASH) #error "Should only be used on ChromeOS." #endif #include #include #include #include "ash/components/disks/disk_mount_manager.h" #include "base/compiler_specific.h" #include "base/memory/weak_ptr.h" #include "build/build_config.h" #include "build/chromeos_buildflags.h" #include "components/storage_monitor/storage_monitor.h" #include "mojo/public/cpp/bindings/pending_remote.h" #include "mojo/public/cpp/bindings/remote.h" #include "services/device/public/mojom/mtp_manager.mojom.h" namespace storage_monitor { class MtpManagerClientChromeOS; class StorageMonitorCros : public StorageMonitor, public ash::disks::DiskMountManager::Observer { public: // Should only be called by browser start up code. // Use StorageMonitor::GetInstance() instead. StorageMonitorCros(); StorageMonitorCros(const StorageMonitorCros&) = delete; StorageMonitorCros& operator=(const StorageMonitorCros&) = delete; ~StorageMonitorCros() override; // Sets up disk listeners and issues notifications for any discovered // mount points. Sets up MTP manager and listeners. void Init() override; protected: void SetMediaTransferProtocolManagerForTest( mojo::PendingRemote test_manager); // ash::disks::DiskMountManager::Observer implementation. void OnBootDeviceDiskEvent(ash::disks::DiskMountManager::DiskEvent event, const ash::disks::Disk& disk) override; void OnMountEvent( ash::disks::DiskMountManager::MountEvent event, ash::MountError error_code, const ash::disks::DiskMountManager::MountPoint& mount_info) override; // StorageMonitor implementation. bool GetStorageInfoForPath(const base::FilePath& path, StorageInfo* device_info) const override; void EjectDevice(const std::string& device_id, base::OnceCallback callback) override; device::mojom::MtpManager* media_transfer_protocol_manager() override; private: // Mapping of mount path to removable mass storage info. typedef std::map MountMap; // Helper method that checks existing mount points to see if they are media // devices. Eventually calls AddMountedPath for all mount points. void CheckExistingMountPoints(); // Adds the mount point in |mount_info| to |mount_map_| and send a media // device attach notification. |has_dcim| is true if the attached device has // a DCIM folder. void AddMountedPath( const ash::disks::DiskMountManager::MountPoint& mount_info, bool has_dcim); // Adds the mount point in |disk| to |mount_map_| and send a device // attach notification. void AddFixedStorageDisk(const ash::disks::Disk& disk); // Removes the mount point in |disk| from |mount_map_| and send a device // detach notification. void RemoveFixedStorageDisk(const ash::disks::Disk& disk); // Mapping of relevant mount points and their corresponding mount devices. MountMap mount_map_; mojo::Remote mtp_device_manager_; std::unique_ptr mtp_manager_client_; base::WeakPtrFactory weak_ptr_factory_{this}; }; } // namespace storage_monitor #endif // COMPONENTS_STORAGE_MONITOR_STORAGE_MONITOR_CHROMEOS_H_