• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright (c) 2012 The Chromium 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 #include "chromeos/disks/disk_mount_manager.h"
6 
7 #include <map>
8 #include <set>
9 
10 #include "base/bind.h"
11 #include "base/memory/weak_ptr.h"
12 #include "base/observer_list.h"
13 #include "base/stl_util.h"
14 #include "base/strings/string_util.h"
15 #include "chromeos/dbus/dbus_thread_manager.h"
16 
17 namespace chromeos {
18 namespace disks {
19 
20 namespace {
21 
22 const char kDeviceNotFound[] = "Device could not be found";
23 
24 DiskMountManager* g_disk_mount_manager = NULL;
25 
26 // The DiskMountManager implementation.
27 class DiskMountManagerImpl : public DiskMountManager {
28  public:
DiskMountManagerImpl()29   DiskMountManagerImpl() :
30     already_refreshed_(false),
31     weak_ptr_factory_(this) {
32     DBusThreadManager* dbus_thread_manager = DBusThreadManager::Get();
33     DCHECK(dbus_thread_manager);
34     cros_disks_client_ = dbus_thread_manager->GetCrosDisksClient();
35     DCHECK(cros_disks_client_);
36     cros_disks_client_->SetMountEventHandler(
37         base::Bind(&DiskMountManagerImpl::OnMountEvent,
38                    weak_ptr_factory_.GetWeakPtr()));
39     cros_disks_client_->SetMountCompletedHandler(
40         base::Bind(&DiskMountManagerImpl::OnMountCompleted,
41                    weak_ptr_factory_.GetWeakPtr()));
42     cros_disks_client_->SetFormatCompletedHandler(
43         base::Bind(&DiskMountManagerImpl::OnFormatCompleted,
44                    weak_ptr_factory_.GetWeakPtr()));
45   }
46 
~DiskMountManagerImpl()47   virtual ~DiskMountManagerImpl() {
48     STLDeleteContainerPairSecondPointers(disks_.begin(), disks_.end());
49   }
50 
51   // DiskMountManager override.
AddObserver(Observer * observer)52   virtual void AddObserver(Observer* observer) OVERRIDE {
53     observers_.AddObserver(observer);
54   }
55 
56   // DiskMountManager override.
RemoveObserver(Observer * observer)57   virtual void RemoveObserver(Observer* observer) OVERRIDE {
58     observers_.RemoveObserver(observer);
59   }
60 
61   // DiskMountManager override.
MountPath(const std::string & source_path,const std::string & source_format,const std::string & mount_label,MountType type)62   virtual void MountPath(const std::string& source_path,
63                          const std::string& source_format,
64                          const std::string& mount_label,
65                          MountType type) OVERRIDE {
66     // Hidden and non-existent devices should not be mounted.
67     if (type == MOUNT_TYPE_DEVICE) {
68       DiskMap::const_iterator it = disks_.find(source_path);
69       if (it == disks_.end() || it->second->is_hidden()) {
70         OnMountCompleted(MountEntry(MOUNT_ERROR_INTERNAL, source_path, type,
71                                     ""));
72         return;
73       }
74     }
75     cros_disks_client_->Mount(
76         source_path,
77         source_format,
78         mount_label,
79         // When succeeds, OnMountCompleted will be called by
80         // "MountCompleted" signal instead.
81         base::Bind(&base::DoNothing),
82         base::Bind(&DiskMountManagerImpl::OnMountCompleted,
83                    weak_ptr_factory_.GetWeakPtr(),
84                    MountEntry(MOUNT_ERROR_INTERNAL, source_path, type, "")));
85   }
86 
87   // DiskMountManager override.
UnmountPath(const std::string & mount_path,UnmountOptions options,const UnmountPathCallback & callback)88   virtual void UnmountPath(const std::string& mount_path,
89                            UnmountOptions options,
90                            const UnmountPathCallback& callback) OVERRIDE {
91     UnmountChildMounts(mount_path);
92     cros_disks_client_->Unmount(mount_path, options,
93                                 base::Bind(&DiskMountManagerImpl::OnUnmountPath,
94                                            weak_ptr_factory_.GetWeakPtr(),
95                                            callback,
96                                            true,
97                                            mount_path),
98                                 base::Bind(&DiskMountManagerImpl::OnUnmountPath,
99                                            weak_ptr_factory_.GetWeakPtr(),
100                                            callback,
101                                            false,
102                                            mount_path));
103   }
104 
105   // DiskMountManager override.
FormatMountedDevice(const std::string & mount_path)106   virtual void FormatMountedDevice(const std::string& mount_path) OVERRIDE {
107     MountPointMap::const_iterator mount_point = mount_points_.find(mount_path);
108     if (mount_point == mount_points_.end()) {
109       LOG(ERROR) << "Mount point with path \"" << mount_path << "\" not found.";
110       OnFormatCompleted(FORMAT_ERROR_UNKNOWN, mount_path);
111       return;
112     }
113 
114     std::string device_path = mount_point->second.source_path;
115     DiskMap::const_iterator disk = disks_.find(device_path);
116     if (disk == disks_.end()) {
117       LOG(ERROR) << "Device with path \"" << device_path << "\" not found.";
118       OnFormatCompleted(FORMAT_ERROR_UNKNOWN, device_path);
119       return;
120     }
121 
122     UnmountPath(disk->second->mount_path(),
123                 UNMOUNT_OPTIONS_NONE,
124                 base::Bind(&DiskMountManagerImpl::OnUnmountPathForFormat,
125                            weak_ptr_factory_.GetWeakPtr(),
126                            device_path));
127   }
128 
129   // DiskMountManager override.
UnmountDeviceRecursively(const std::string & device_path,const UnmountDeviceRecursivelyCallbackType & callback)130   virtual void UnmountDeviceRecursively(
131       const std::string& device_path,
132       const UnmountDeviceRecursivelyCallbackType& callback) OVERRIDE {
133     std::vector<std::string> devices_to_unmount;
134 
135     // Get list of all devices to unmount.
136     int device_path_len = device_path.length();
137     for (DiskMap::iterator it = disks_.begin(); it != disks_.end(); ++it) {
138       if (!it->second->mount_path().empty() &&
139           strncmp(device_path.c_str(), it->second->device_path().c_str(),
140                   device_path_len) == 0) {
141         devices_to_unmount.push_back(it->second->mount_path());
142       }
143     }
144 
145     // We should detect at least original device.
146     if (devices_to_unmount.empty()) {
147       if (disks_.find(device_path) == disks_.end()) {
148         LOG(WARNING) << "Unmount recursive request failed for device "
149                      << device_path << ", with error: " << kDeviceNotFound;
150         callback.Run(false);
151         return;
152       }
153 
154       // Nothing to unmount.
155       callback.Run(true);
156       return;
157     }
158 
159     // We will send the same callback data object to all Unmount calls and use
160     // it to synchronize callbacks.
161     // Note: this implementation has a potential memory leak issue. For
162     // example if this instance is destructed before all the callbacks for
163     // Unmount are invoked, the memory pointed by |cb_data| will be leaked.
164     // It is because the UnmountDeviceRecursivelyCallbackData keeps how
165     // many times OnUnmountDeviceRecursively callback is called and when
166     // all the callbacks are called, |cb_data| will be deleted in the method.
167     // However destructing the instance before all callback invocations will
168     // cancel all pending callbacks, so that the |cb_data| would never be
169     // deleted.
170     // Fortunately, in the real scenario, the instance will be destructed
171     // only for ShutDown. So, probably the memory would rarely be leaked.
172     // TODO(hidehiko): Fix the issue.
173     UnmountDeviceRecursivelyCallbackData* cb_data =
174         new UnmountDeviceRecursivelyCallbackData(
175             callback, devices_to_unmount.size());
176     for (size_t i = 0; i < devices_to_unmount.size(); ++i) {
177       cros_disks_client_->Unmount(
178           devices_to_unmount[i],
179           UNMOUNT_OPTIONS_NONE,
180           base::Bind(&DiskMountManagerImpl::OnUnmountDeviceRecursively,
181                      weak_ptr_factory_.GetWeakPtr(),
182                      cb_data,
183                      true,
184                      devices_to_unmount[i]),
185           base::Bind(&DiskMountManagerImpl::OnUnmountDeviceRecursively,
186                      weak_ptr_factory_.GetWeakPtr(),
187                      cb_data,
188                      false,
189                      devices_to_unmount[i]));
190     }
191   }
192 
193   // DiskMountManager override.
EnsureMountInfoRefreshed(const EnsureMountInfoRefreshedCallback & callback)194   virtual void EnsureMountInfoRefreshed(
195       const EnsureMountInfoRefreshedCallback& callback) OVERRIDE {
196     if (already_refreshed_) {
197       callback.Run(true);
198       return;
199     }
200 
201     refresh_callbacks_.push_back(callback);
202     if (refresh_callbacks_.size() == 1) {
203       // If there's no in-flight refreshing task, start it.
204       cros_disks_client_->EnumerateAutoMountableDevices(
205           base::Bind(&DiskMountManagerImpl::RefreshAfterEnumerateDevices,
206                      weak_ptr_factory_.GetWeakPtr()),
207           base::Bind(&DiskMountManagerImpl::RefreshCompleted,
208                      weak_ptr_factory_.GetWeakPtr(), false));
209     }
210   }
211 
212   // DiskMountManager override.
disks() const213   virtual const DiskMap& disks() const OVERRIDE { return disks_; }
214 
215   // DiskMountManager override.
FindDiskBySourcePath(const std::string & source_path) const216   virtual const Disk* FindDiskBySourcePath(const std::string& source_path)
217       const OVERRIDE {
218     DiskMap::const_iterator disk_it = disks_.find(source_path);
219     return disk_it == disks_.end() ? NULL : disk_it->second;
220   }
221 
222   // DiskMountManager override.
mount_points() const223   virtual const MountPointMap& mount_points() const OVERRIDE {
224     return mount_points_;
225   }
226 
227   // DiskMountManager override.
AddDiskForTest(Disk * disk)228   virtual bool AddDiskForTest(Disk* disk) OVERRIDE {
229     if (disks_.find(disk->device_path()) != disks_.end()) {
230       LOG(ERROR) << "Attempt to add a duplicate disk";
231       return false;
232     }
233 
234     disks_.insert(std::make_pair(disk->device_path(), disk));
235     return true;
236   }
237 
238   // DiskMountManager override.
239   // Corresponding disk should be added to the manager before this is called.
AddMountPointForTest(const MountPointInfo & mount_point)240   virtual bool AddMountPointForTest(
241       const MountPointInfo& mount_point) OVERRIDE {
242     if (mount_points_.find(mount_point.mount_path) != mount_points_.end()) {
243       LOG(ERROR) << "Attempt to add a duplicate mount point";
244       return false;
245     }
246     if (mount_point.mount_type == chromeos::MOUNT_TYPE_DEVICE &&
247         disks_.find(mount_point.source_path) == disks_.end()) {
248       LOG(ERROR) << "Device mount points must have a disk entry.";
249       return false;
250     }
251 
252     mount_points_.insert(std::make_pair(mount_point.mount_path, mount_point));
253     return true;
254   }
255 
256  private:
257   struct UnmountDeviceRecursivelyCallbackData {
UnmountDeviceRecursivelyCallbackDatachromeos::disks::__anon97f72b290111::DiskMountManagerImpl::UnmountDeviceRecursivelyCallbackData258     UnmountDeviceRecursivelyCallbackData(
259         const UnmountDeviceRecursivelyCallbackType& in_callback,
260         int in_num_pending_callbacks)
261         : callback(in_callback),
262           num_pending_callbacks(in_num_pending_callbacks) {
263     }
264 
265     const UnmountDeviceRecursivelyCallbackType callback;
266     size_t num_pending_callbacks;
267   };
268 
269   // Unmounts all mount points whose source path is transitively parented by
270   // |mount_path|.
UnmountChildMounts(const std::string & mount_path_in)271   void UnmountChildMounts(const std::string& mount_path_in) {
272     std::string mount_path = mount_path_in;
273     // Let's make sure mount path has trailing slash.
274     if (mount_path[mount_path.length() - 1] != '/')
275       mount_path += '/';
276 
277     for (MountPointMap::iterator it = mount_points_.begin();
278          it != mount_points_.end();
279          ++it) {
280       if (StartsWithASCII(it->second.source_path, mount_path,
281                           true /*case sensitive*/)) {
282         // TODO(tbarzic): Handle the case where this fails.
283         UnmountPath(it->second.mount_path,
284                     UNMOUNT_OPTIONS_NONE,
285                     UnmountPathCallback());
286       }
287     }
288   }
289 
290   // Callback for UnmountDeviceRecursively.
OnUnmountDeviceRecursively(UnmountDeviceRecursivelyCallbackData * cb_data,bool success,const std::string & mount_path)291   void OnUnmountDeviceRecursively(
292       UnmountDeviceRecursivelyCallbackData* cb_data,
293       bool success,
294       const std::string& mount_path) {
295     if (success) {
296       // Do standard processing for Unmount event.
297       OnUnmountPath(UnmountPathCallback(), true, mount_path);
298       VLOG(1) << mount_path <<  " unmounted.";
299     }
300     // This is safe as long as all callbacks are called on the same thread as
301     // UnmountDeviceRecursively.
302     cb_data->num_pending_callbacks--;
303 
304     if (cb_data->num_pending_callbacks == 0) {
305       // This code has a problem that the |success| status used here is for the
306       // last "unmount" callback, but not whether all unmounting is succeeded.
307       // TODO(hidehiko): Fix the issue.
308       cb_data->callback.Run(success);
309       delete cb_data;
310     }
311   }
312 
313   // Callback to handle MountCompleted signal and Mount method call failure.
OnMountCompleted(const MountEntry & entry)314   void OnMountCompleted(const MountEntry& entry) {
315     MountCondition mount_condition = MOUNT_CONDITION_NONE;
316     if (entry.mount_type() == MOUNT_TYPE_DEVICE) {
317       if (entry.error_code() == MOUNT_ERROR_UNKNOWN_FILESYSTEM) {
318         mount_condition = MOUNT_CONDITION_UNKNOWN_FILESYSTEM;
319       }
320       if (entry.error_code() == MOUNT_ERROR_UNSUPPORTED_FILESYSTEM) {
321         mount_condition = MOUNT_CONDITION_UNSUPPORTED_FILESYSTEM;
322       }
323     }
324     const MountPointInfo mount_info(entry.source_path(),
325                                     entry.mount_path(),
326                                     entry.mount_type(),
327                                     mount_condition);
328 
329     NotifyMountStatusUpdate(MOUNTING, entry.error_code(), mount_info);
330 
331     // If the device is corrupted but it's still possible to format it, it will
332     // be fake mounted.
333     if ((entry.error_code() == MOUNT_ERROR_NONE ||
334          mount_info.mount_condition) &&
335         mount_points_.find(mount_info.mount_path) == mount_points_.end()) {
336       mount_points_.insert(MountPointMap::value_type(mount_info.mount_path,
337                                                      mount_info));
338     }
339     if ((entry.error_code() == MOUNT_ERROR_NONE ||
340          mount_info.mount_condition) &&
341         mount_info.mount_type == MOUNT_TYPE_DEVICE &&
342         !mount_info.source_path.empty() &&
343         !mount_info.mount_path.empty()) {
344       DiskMap::iterator iter = disks_.find(mount_info.source_path);
345       if (iter == disks_.end()) {
346         // disk might have been removed by now?
347         return;
348       }
349       Disk* disk = iter->second;
350       DCHECK(disk);
351       disk->set_mount_path(mount_info.mount_path);
352     }
353   }
354 
355   // Callback for UnmountPath.
OnUnmountPath(const UnmountPathCallback & callback,bool success,const std::string & mount_path)356   void OnUnmountPath(const UnmountPathCallback& callback,
357                      bool success,
358                      const std::string& mount_path) {
359     MountPointMap::iterator mount_points_it = mount_points_.find(mount_path);
360     if (mount_points_it == mount_points_.end()) {
361       // The path was unmounted, but not as a result of this unmount request,
362       // so return error.
363       if (!callback.is_null())
364         callback.Run(MOUNT_ERROR_INTERNAL);
365       return;
366     }
367 
368     NotifyMountStatusUpdate(
369         UNMOUNTING,
370         success ? MOUNT_ERROR_NONE : MOUNT_ERROR_INTERNAL,
371         MountPointInfo(mount_points_it->second.source_path,
372                        mount_points_it->second.mount_path,
373                        mount_points_it->second.mount_type,
374                        mount_points_it->second.mount_condition));
375 
376     std::string path(mount_points_it->second.source_path);
377     if (success)
378       mount_points_.erase(mount_points_it);
379 
380     DiskMap::iterator disk_iter = disks_.find(path);
381     if (disk_iter != disks_.end()) {
382       DCHECK(disk_iter->second);
383       if (success)
384         disk_iter->second->clear_mount_path();
385     }
386 
387     if (!callback.is_null())
388       callback.Run(success ? MOUNT_ERROR_NONE : MOUNT_ERROR_INTERNAL);
389   }
390 
OnUnmountPathForFormat(const std::string & device_path,MountError error_code)391   void OnUnmountPathForFormat(const std::string& device_path,
392                               MountError error_code) {
393     if (error_code == MOUNT_ERROR_NONE &&
394         disks_.find(device_path) != disks_.end()) {
395       FormatUnmountedDevice(device_path);
396     } else {
397       OnFormatCompleted(FORMAT_ERROR_UNKNOWN, device_path);
398     }
399   }
400 
401   // Starts device formatting.
FormatUnmountedDevice(const std::string & device_path)402   void FormatUnmountedDevice(const std::string& device_path) {
403     DiskMap::const_iterator disk = disks_.find(device_path);
404     DCHECK(disk != disks_.end() && disk->second->mount_path().empty());
405 
406     const char kFormatVFAT[] = "vfat";
407     cros_disks_client_->Format(
408         device_path,
409         kFormatVFAT,
410         base::Bind(&DiskMountManagerImpl::OnFormatStarted,
411                    weak_ptr_factory_.GetWeakPtr(),
412                    device_path),
413         base::Bind(&DiskMountManagerImpl::OnFormatCompleted,
414                    weak_ptr_factory_.GetWeakPtr(),
415                    FORMAT_ERROR_UNKNOWN,
416                    device_path));
417   }
418 
419   // Callback for Format.
OnFormatStarted(const std::string & device_path)420   void OnFormatStarted(const std::string& device_path) {
421     NotifyFormatStatusUpdate(FORMAT_STARTED, FORMAT_ERROR_NONE, device_path);
422   }
423 
424   // Callback to handle FormatCompleted signal and Format method call failure.
OnFormatCompleted(FormatError error_code,const std::string & device_path)425   void OnFormatCompleted(FormatError error_code,
426                          const std::string& device_path) {
427     NotifyFormatStatusUpdate(FORMAT_COMPLETED, error_code, device_path);
428   }
429 
430   // Callback for GetDeviceProperties.
OnGetDeviceProperties(const DiskInfo & disk_info)431   void OnGetDeviceProperties(const DiskInfo& disk_info) {
432     // TODO(zelidrag): Find a better way to filter these out before we
433     // fetch the properties:
434     // Ignore disks coming from the device we booted the system from.
435     if (disk_info.on_boot_device())
436       return;
437 
438     LOG(WARNING) << "Found disk " << disk_info.device_path();
439     // Delete previous disk info for this path:
440     bool is_new = true;
441     DiskMap::iterator iter = disks_.find(disk_info.device_path());
442     if (iter != disks_.end()) {
443       delete iter->second;
444       disks_.erase(iter);
445       is_new = false;
446     }
447     Disk* disk = new Disk(disk_info.device_path(),
448                           disk_info.mount_path(),
449                           disk_info.system_path(),
450                           disk_info.file_path(),
451                           disk_info.label(),
452                           disk_info.drive_label(),
453                           disk_info.vendor_id(),
454                           disk_info.vendor_name(),
455                           disk_info.product_id(),
456                           disk_info.product_name(),
457                           disk_info.uuid(),
458                           FindSystemPathPrefix(disk_info.system_path()),
459                           disk_info.device_type(),
460                           disk_info.total_size_in_bytes(),
461                           disk_info.is_drive(),
462                           disk_info.is_read_only(),
463                           disk_info.has_media(),
464                           disk_info.on_boot_device(),
465                           disk_info.on_removable_device(),
466                           disk_info.is_hidden());
467     disks_.insert(std::make_pair(disk_info.device_path(), disk));
468     NotifyDiskStatusUpdate(is_new ? DISK_ADDED : DISK_CHANGED, disk);
469   }
470 
471   // Part of EnsureMountInfoRefreshed(). Called after the list of devices are
472   // enumerated.
RefreshAfterEnumerateDevices(const std::vector<std::string> & devices)473   void RefreshAfterEnumerateDevices(const std::vector<std::string>& devices) {
474     std::set<std::string> current_device_set(devices.begin(), devices.end());
475     for (DiskMap::iterator iter = disks_.begin(); iter != disks_.end(); ) {
476       if (current_device_set.find(iter->first) == current_device_set.end()) {
477         delete iter->second;
478         disks_.erase(iter++);
479       } else {
480         ++iter;
481       }
482     }
483     RefreshDeviceAtIndex(devices, 0);
484   }
485 
486   // Part of EnsureMountInfoRefreshed(). Called for each device to refresh info.
RefreshDeviceAtIndex(const std::vector<std::string> & devices,size_t index)487   void RefreshDeviceAtIndex(const std::vector<std::string>& devices,
488                             size_t index) {
489     if (index == devices.size()) {
490       // All devices info retrieved. Proceed to enumerate mount point info.
491       cros_disks_client_->EnumerateMountEntries(
492           base::Bind(&DiskMountManagerImpl::RefreshAfterEnumerateMountEntries,
493                      weak_ptr_factory_.GetWeakPtr()),
494           base::Bind(&DiskMountManagerImpl::RefreshCompleted,
495                      weak_ptr_factory_.GetWeakPtr(), false));
496       return;
497     }
498 
499     cros_disks_client_->GetDeviceProperties(
500         devices[index],
501         base::Bind(&DiskMountManagerImpl::RefreshAfterGetDeviceProperties,
502                    weak_ptr_factory_.GetWeakPtr(), devices, index + 1),
503         base::Bind(&DiskMountManagerImpl::RefreshCompleted,
504                    weak_ptr_factory_.GetWeakPtr(), false));
505   }
506 
507   // Part of EnsureMountInfoRefreshed().
RefreshAfterGetDeviceProperties(const std::vector<std::string> & devices,size_t next_index,const DiskInfo & disk_info)508   void RefreshAfterGetDeviceProperties(const std::vector<std::string>& devices,
509                                        size_t next_index,
510                                        const DiskInfo& disk_info) {
511     OnGetDeviceProperties(disk_info);
512     RefreshDeviceAtIndex(devices, next_index);
513   }
514 
515   // Part of EnsureMountInfoRefreshed(). Called after mount entries are listed.
RefreshAfterEnumerateMountEntries(const std::vector<MountEntry> & entries)516   void RefreshAfterEnumerateMountEntries(
517       const std::vector<MountEntry>& entries) {
518     for (size_t i = 0; i < entries.size(); ++i)
519       OnMountCompleted(entries[i]);
520     RefreshCompleted(true);
521   }
522 
523   // Part of EnsureMountInfoRefreshed(). Called when the refreshing is done.
RefreshCompleted(bool success)524   void RefreshCompleted(bool success) {
525     already_refreshed_ = true;
526     for (size_t i = 0; i < refresh_callbacks_.size(); ++i)
527       refresh_callbacks_[i].Run(success);
528     refresh_callbacks_.clear();
529   }
530 
531   // Callback to handle mount event signals.
OnMountEvent(MountEventType event,const std::string & device_path_arg)532   void OnMountEvent(MountEventType event, const std::string& device_path_arg) {
533     // Take a copy of the argument so we can modify it below.
534     std::string device_path = device_path_arg;
535     switch (event) {
536       case CROS_DISKS_DISK_ADDED: {
537         cros_disks_client_->GetDeviceProperties(
538             device_path,
539             base::Bind(&DiskMountManagerImpl::OnGetDeviceProperties,
540                        weak_ptr_factory_.GetWeakPtr()),
541             base::Bind(&base::DoNothing));
542         break;
543       }
544       case CROS_DISKS_DISK_REMOVED: {
545         // Search and remove disks that are no longer present.
546         DiskMountManager::DiskMap::iterator iter = disks_.find(device_path);
547         if (iter != disks_.end()) {
548           Disk* disk = iter->second;
549           NotifyDiskStatusUpdate(DISK_REMOVED, disk);
550           delete iter->second;
551           disks_.erase(iter);
552         }
553         break;
554       }
555       case CROS_DISKS_DEVICE_ADDED: {
556         system_path_prefixes_.insert(device_path);
557         NotifyDeviceStatusUpdate(DEVICE_ADDED, device_path);
558         break;
559       }
560       case CROS_DISKS_DEVICE_REMOVED: {
561         system_path_prefixes_.erase(device_path);
562         NotifyDeviceStatusUpdate(DEVICE_REMOVED, device_path);
563         break;
564       }
565       case CROS_DISKS_DEVICE_SCANNED: {
566         NotifyDeviceStatusUpdate(DEVICE_SCANNED, device_path);
567         break;
568       }
569       default: {
570         LOG(ERROR) << "Unknown event: " << event;
571       }
572     }
573   }
574 
575   // Notifies all observers about disk status update.
NotifyDiskStatusUpdate(DiskEvent event,const Disk * disk)576   void NotifyDiskStatusUpdate(DiskEvent event,
577                               const Disk* disk) {
578     FOR_EACH_OBSERVER(Observer, observers_, OnDiskEvent(event, disk));
579   }
580 
581   // Notifies all observers about device status update.
NotifyDeviceStatusUpdate(DeviceEvent event,const std::string & device_path)582   void NotifyDeviceStatusUpdate(DeviceEvent event,
583                                 const std::string& device_path) {
584     FOR_EACH_OBSERVER(Observer, observers_, OnDeviceEvent(event, device_path));
585   }
586 
587   // Notifies all observers about mount completion.
NotifyMountStatusUpdate(MountEvent event,MountError error_code,const MountPointInfo & mount_info)588   void NotifyMountStatusUpdate(MountEvent event,
589                                MountError error_code,
590                                const MountPointInfo& mount_info) {
591     FOR_EACH_OBSERVER(Observer, observers_,
592                       OnMountEvent(event, error_code, mount_info));
593   }
594 
NotifyFormatStatusUpdate(FormatEvent event,FormatError error_code,const std::string & device_path)595   void NotifyFormatStatusUpdate(FormatEvent event,
596                                 FormatError error_code,
597                                 const std::string& device_path) {
598     FOR_EACH_OBSERVER(Observer, observers_,
599                       OnFormatEvent(event, error_code, device_path));
600   }
601 
602   // Finds system path prefix from |system_path|.
FindSystemPathPrefix(const std::string & system_path)603   const std::string& FindSystemPathPrefix(const std::string& system_path) {
604     if (system_path.empty())
605       return base::EmptyString();
606     for (SystemPathPrefixSet::const_iterator it = system_path_prefixes_.begin();
607          it != system_path_prefixes_.end();
608          ++it) {
609       const std::string& prefix = *it;
610       if (StartsWithASCII(system_path, prefix, true))
611         return prefix;
612     }
613     return base::EmptyString();
614   }
615 
616   // Mount event change observers.
617   ObserverList<Observer> observers_;
618 
619   CrosDisksClient* cros_disks_client_;
620 
621   // The list of disks found.
622   DiskMountManager::DiskMap disks_;
623 
624   DiskMountManager::MountPointMap mount_points_;
625 
626   typedef std::set<std::string> SystemPathPrefixSet;
627   SystemPathPrefixSet system_path_prefixes_;
628 
629   bool already_refreshed_;
630   std::vector<EnsureMountInfoRefreshedCallback> refresh_callbacks_;
631 
632   base::WeakPtrFactory<DiskMountManagerImpl> weak_ptr_factory_;
633 
634   DISALLOW_COPY_AND_ASSIGN(DiskMountManagerImpl);
635 };
636 
637 }  // namespace
638 
Disk(const std::string & device_path,const std::string & mount_path,const std::string & system_path,const std::string & file_path,const std::string & device_label,const std::string & drive_label,const std::string & vendor_id,const std::string & vendor_name,const std::string & product_id,const std::string & product_name,const std::string & fs_uuid,const std::string & system_path_prefix,DeviceType device_type,uint64 total_size_in_bytes,bool is_parent,bool is_read_only,bool has_media,bool on_boot_device,bool on_removable_device,bool is_hidden)639 DiskMountManager::Disk::Disk(const std::string& device_path,
640                              const std::string& mount_path,
641                              const std::string& system_path,
642                              const std::string& file_path,
643                              const std::string& device_label,
644                              const std::string& drive_label,
645                              const std::string& vendor_id,
646                              const std::string& vendor_name,
647                              const std::string& product_id,
648                              const std::string& product_name,
649                              const std::string& fs_uuid,
650                              const std::string& system_path_prefix,
651                              DeviceType device_type,
652                              uint64 total_size_in_bytes,
653                              bool is_parent,
654                              bool is_read_only,
655                              bool has_media,
656                              bool on_boot_device,
657                              bool on_removable_device,
658                              bool is_hidden)
659     : device_path_(device_path),
660       mount_path_(mount_path),
661       system_path_(system_path),
662       file_path_(file_path),
663       device_label_(device_label),
664       drive_label_(drive_label),
665       vendor_id_(vendor_id),
666       vendor_name_(vendor_name),
667       product_id_(product_id),
668       product_name_(product_name),
669       fs_uuid_(fs_uuid),
670       system_path_prefix_(system_path_prefix),
671       device_type_(device_type),
672       total_size_in_bytes_(total_size_in_bytes),
673       is_parent_(is_parent),
674       is_read_only_(is_read_only),
675       has_media_(has_media),
676       on_boot_device_(on_boot_device),
677       on_removable_device_(on_removable_device),
678       is_hidden_(is_hidden) {
679 }
680 
~Disk()681 DiskMountManager::Disk::~Disk() {}
682 
AddDiskForTest(Disk * disk)683 bool DiskMountManager::AddDiskForTest(Disk* disk) {
684   return false;
685 }
686 
AddMountPointForTest(const MountPointInfo & mount_point)687 bool DiskMountManager::AddMountPointForTest(const MountPointInfo& mount_point) {
688   return false;
689 }
690 
691 // static
MountConditionToString(MountCondition condition)692 std::string DiskMountManager::MountConditionToString(MountCondition condition) {
693   switch (condition) {
694     case MOUNT_CONDITION_NONE:
695       return "";
696     case MOUNT_CONDITION_UNKNOWN_FILESYSTEM:
697       return "unknown_filesystem";
698     case MOUNT_CONDITION_UNSUPPORTED_FILESYSTEM:
699       return "unsupported_filesystem";
700     default:
701       NOTREACHED();
702   }
703   return "";
704 }
705 
706 // static
DeviceTypeToString(DeviceType type)707 std::string DiskMountManager::DeviceTypeToString(DeviceType type) {
708   switch (type) {
709     case DEVICE_TYPE_USB:
710       return "usb";
711     case DEVICE_TYPE_SD:
712       return "sd";
713     case DEVICE_TYPE_OPTICAL_DISC:
714       return "optical";
715     case DEVICE_TYPE_MOBILE:
716       return "mobile";
717     default:
718       return "unknown";
719   }
720 }
721 
722 // static
Initialize()723 void DiskMountManager::Initialize() {
724   if (g_disk_mount_manager) {
725     LOG(WARNING) << "DiskMountManager was already initialized";
726     return;
727   }
728   g_disk_mount_manager = new DiskMountManagerImpl();
729   VLOG(1) << "DiskMountManager initialized";
730 }
731 
732 // static
InitializeForTesting(DiskMountManager * disk_mount_manager)733 void DiskMountManager::InitializeForTesting(
734     DiskMountManager* disk_mount_manager) {
735   if (g_disk_mount_manager) {
736     LOG(WARNING) << "DiskMountManager was already initialized";
737     return;
738   }
739   g_disk_mount_manager = disk_mount_manager;
740   VLOG(1) << "DiskMountManager initialized";
741 }
742 
743 // static
Shutdown()744 void DiskMountManager::Shutdown() {
745   if (!g_disk_mount_manager) {
746     LOG(WARNING) << "DiskMountManager::Shutdown() called with NULL manager";
747     return;
748   }
749   delete g_disk_mount_manager;
750   g_disk_mount_manager = NULL;
751   VLOG(1) << "DiskMountManager Shutdown completed";
752 }
753 
754 // static
GetInstance()755 DiskMountManager* DiskMountManager::GetInstance() {
756   return g_disk_mount_manager;
757 }
758 
759 }  // namespace disks
760 }  // namespace chromeos
761