chrome/utility/image_writer/image_writer.h

   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
  20 namespace image_writer {
  21
  22 class ImageWriterHandler;
  23 #if defined(OS_MACOSX)
  24 class 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.
  30 class 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_