1// Copyright 2014 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#ifndef CHROME_UTILITY_IMAGE_WRITER_DISK_UNMOUNTER_MAC_H_
6#define CHROME_UTILITY_IMAGE_WRITER_DISK_UNMOUNTER_MAC_H_
7
8#include <CoreFoundation/CoreFoundation.h>
9#include <DiskArbitration/DiskArbitration.h>
10
11#include "base/bind.h"
12#include "base/callback.h"
13#include "base/mac/foundation_util.h"
14#include "base/memory/weak_ptr.h"
15#include "base/threading/thread.h"
16
17namespace image_writer {
18
19class ImageWriter;
20
21// Manages the unmounting of disks through Disk Arbitration.  Disk Arbitration
22// has to be run on a thread with a CFRunLoop.  In the utility process neither
23// the main or IO thread have one by default, so we need to manage a new thread
24// which will explicitly have a CFRunLoop-based message pump.  Note that this
25// class can only handle one unmount operation at a time and calling Unmount
26// again before the continuation returns will cause undefined behavior.
27class DiskUnmounterMac {
28 public:
29  DiskUnmounterMac();
30  ~DiskUnmounterMac();
31
32  // Claims and unmounts the device described by |device_path| and then calls
33  // the |continuation| when complete.  This can be called from any thread.
34  // The continuation will be run on the thread this object was created on.
35  void Unmount(const std::string& device_path,
36               const base::Closure& success_continuation,
37               const base::Closure& failure_continuation);
38
39 private:
40  // Handles disk-claimed callbacks.
41  static void DiskClaimed(DADiskRef disk,
42                          DADissenterRef dissenter,
43                          void* context);
44  // Handles when we fail to claim a disk.
45  static DADissenterRef DiskClaimRevoked(DADiskRef disk, void* context);
46  // Handles the disk-unmounted callback.
47  static void DiskUnmounted(DADiskRef disk,
48                            DADissenterRef dissenter,
49                            void* context);
50
51  // A |MessagePumpFactory| for creating the thread.
52  static scoped_ptr<base::MessagePump> CreateMessagePump();
53
54  // Starts the unmount process.  Should be posted to the |cf_thread_|.
55  void UnmountOnWorker(const std::string& device_path);
56
57  // A convenience method that triggers the failure continuation.
58  void Error();
59
60  scoped_refptr<base::MessageLoopProxy> original_thread_;
61  base::Closure success_continuation_;
62  base::Closure failure_continuation_;
63
64  base::ScopedCFTypeRef<DADiskRef> disk_;
65  base::ScopedCFTypeRef<DASessionRef> session_;
66
67  // Thread is last to ensure it is stopped before the data members are
68  // destroyed.
69  base::Thread cf_thread_;
70};
71
72}  // namespace image_writer
73
74#endif  // CHROME_UTILITY_IMAGE_WRITER_DISK_UNMOUNTER_MAC_H_
75