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_IMAGE_WRITER_H_
6#define CHROME_UTILITY_IMAGE_WRITER_IMAGE_WRITER_H_
7
8#include <vector>
9
10#include "base/bind.h"
11#include "base/callback.h"
12#include "base/files/file.h"
13#include "base/files/file_path.h"
14#include "base/memory/weak_ptr.h"
15
16#if defined(OS_WIN)
17#include <windows.h>
18#endif
19
20namespace image_writer {
21
22class ImageWriterHandler;
23#if defined(OS_MACOSX)
24class DiskUnmounterMac;
25#endif
26
27// Manages a write within the utility thread.  This class holds all the state
28// around the writing and communicates with the ImageWriterHandler to dispatch
29// messages.
30class ImageWriter : public base::SupportsWeakPtr<ImageWriter> {
31 public:
32  explicit ImageWriter(ImageWriterHandler* handler,
33                       const base::FilePath& image_path,
34                       const base::FilePath& device_path);
35  virtual ~ImageWriter();
36
37  // Starts a write from |image_path_| to |device_path_|.
38  void Write();
39  // Starts verifying that |image_path_| and |device_path_| have the same size
40  // and contents.
41  void Verify();
42  // Cancels any pending writes or verifications.
43  void Cancel();
44
45  // Returns whether an operation is in progress.
46  bool IsRunning() const;
47  // Checks if a path is a valid target device.
48  // This method has OS-specific implementations.
49  bool IsValidDevice();
50  // Unmounts all volumes on the target device.
51  // This method has OS-specific implementations.
52  void UnmountVolumes(const base::Closure& continuation);
53
54  // Return the current image path.
55  const base::FilePath& GetImagePath();
56  // Return the current device path.
57  const base::FilePath& GetDevicePath();
58
59 private:
60  // Convenience wrappers.
61  void PostTask(const base::Closure& task);
62  void PostProgress(int64 progress);
63  void Error(const std::string& message);
64
65  // Initializes the files.
66  bool InitializeFiles();
67  bool OpenDevice();
68
69  // Work loops.
70  void WriteChunk();
71  void VerifyChunk();
72
73  base::FilePath image_path_;
74  base::FilePath device_path_;
75
76  base::File image_file_;
77  base::File device_file_;
78  int64 bytes_processed_;
79  bool running_;
80
81#if defined(OS_WIN)
82  std::vector<HANDLE> volume_handles_;
83#endif
84
85#if defined(OS_MACOSX)
86  friend class DiskUnmounterMac;
87  scoped_ptr<DiskUnmounterMac> unmounter_;
88#endif
89
90  ImageWriterHandler* handler_;
91};
92
93}  // namespace image_writer
94
95#endif  // CHROME_UTILITY_IMAGE_WRITER_IMAGE_WRITER_H_
96