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..8f2db715ff3f558a2dd713a4a75a997df9d12cb1 |
| --- /dev/null |
| +++ b/chrome/utility/image_writer/disk_unmounter_mac.h |
| @@ -0,0 +1,73 @@ |
| +// 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 |
|
Robert Sesek
2014/06/06 18:34:06
That's not quite accurate. You can only ever call
Drew Haven
2014/06/10 01:13:25
I agree. I followed up on that. Creation of the
|
| +// again before the continuation returns will cause undefined behavior. |
| +class DiskUnmounterMac { |
| + public: |
| + DiskUnmounterMac(base::WeakPtr<ImageWriter> image_writer); |
| + virtual ~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& 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 for sending an error back to the |image_writer_|. |
| + void Error(const std::string& message); |
| + |
| + base::WeakPtr<ImageWriter> image_writer_; |
| + base::Closure unmount_continuation_closure_; |
| + scoped_refptr<base::MessageLoopProxy> original_thread_; |
| + |
| + 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_ |