Chromium Code Reviews| Index: chrome/utility/image_writer/disk_unmounter_mac.h |
| diff --git a/chrome/utility/image_writer/disk_unmounter_mac.h b/chrome/utility/image_writer/disk_unmounter_mac.h |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..4757e1b3864295671fc35570df33db9c92f3cebc |
| --- /dev/null |
| +++ b/chrome/utility/image_writer/disk_unmounter_mac.h |
| @@ -0,0 +1,74 @@ |
| +// 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. |
| + |
| +#ifndef CHROME_UTILITY_IMAGE_WRITER_DISK_UNMOUNTER_MAC_H_ |
| +#define CHROME_UTILITY_IMAGE_WRITER_DISK_UNMOUNTER_MAC_H_ |
| + |
| +#include <CoreFoundation/CoreFoundation.h> |
| +#include <DiskArbitration/DiskArbitration.h> |
| + |
| +#include "base/bind.h" |
| +#include "base/callback.h" |
| +#include "base/mac/foundation_util.h" |
| +#include "base/memory/weak_ptr.h" |
| +#include "base/threading/thread.h" |
| + |
| +namespace image_writer { |
| + |
| +class ImageWriter; |
| + |
| +// Manages the unmounting of disks through Disk Arbitration. Disk Arbitration |
| +// has to be run on a thread with a CFRunLoop. In the utility process neither |
| +// the main or IO thread have one by default, so we need to manage a new thread |
| +// which will explicitly have a CFRunLoop-based message pump. Note that this |
| +// class can only handle one unmount operation at a time and calling Unmount |
| +// again before the continuation returns will cause undefined behavior. |
| +class DiskUnmounterMac { |
| + public: |
| + DiskUnmounterMac(); |
| + virtual ~DiskUnmounterMac(); |
|
Robert Sesek
2014/06/10 19:47:46
Need not be virtual.
Drew Haven
2014/06/12 02:24:26
Isn't it good practice to make a destructor virtua
|
| + |
| + // Claims and unmounts the device described by |device_path| and then calls |
| + // the |continuation| when complete. This can be called from any thread. |
| + // The continuation will be run on the thread this object was created on. |
| + void Unmount(const std::string& device_path, |
| + const base::Closure& success_continuation, |
| + const base::Closure& failure_continuation); |
| + |
| + private: |
| + // Handles disk-claimed callbacks. |
| + static void DiskClaimed(DADiskRef disk, |
| + DADissenterRef dissenter, |
| + void* context); |
| + // Handles when we fail to claim a disk. |
| + static DADissenterRef DiskClaimRevoked(DADiskRef disk, void* context); |
| + // Handles the disk-unmounted callback. |
| + static void DiskUnmounted(DADiskRef disk, |
| + DADissenterRef dissenter, |
| + void* context); |
| + |
| + // A |MessagePumpFactory| for creating the thread. |
| + static scoped_ptr<base::MessagePump> CreateMessagePump(); |
| + |
| + // Starts the unmount process. Should be posted to the |cf_thread_|. |
| + void UnmountOnWorker(const std::string& device_path); |
| + |
| + // A convenience method that triggers the failure continuation. |
| + void Error(); |
| + |
| + scoped_refptr<base::MessageLoopProxy> original_thread_; |
| + base::Closure success_continuation_; |
| + base::Closure failure_continuation_; |
| + |
| + base::ScopedCFTypeRef<DADiskRef> disk_; |
| + base::ScopedCFTypeRef<DASessionRef> session_; |
| + |
| + // Thread is last to ensure it is stopped before the data members are |
| + // destroyed. |
| + base::Thread cf_thread_; |
| +}; |
| + |
| +} // namespace image_writer |
| + |
| +#endif // CHROME_UTILITY_IMAGE_WRITER_DISK_UNMOUNTER_MAC_H_ |