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..616aa39c24566488884fec6092455ba88f30d328 |
--- /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(); |
+ ~DiskUnmounterMac(); |
+ |
+ // 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_ |