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