Index: chrome/browser/extensions/api/image_writer_private/image_writer_utility_client.h |
diff --git a/chrome/browser/extensions/api/image_writer_private/image_writer_utility_client.h b/chrome/browser/extensions/api/image_writer_private/image_writer_utility_client.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..ed997f0fa292b057ec847163a675d684963d9501 |
--- /dev/null |
+++ b/chrome/browser/extensions/api/image_writer_private/image_writer_utility_client.h |
@@ -0,0 +1,86 @@ |
+// 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_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_IMAGE_WRITER_UTILITY_CLIENT_H_ |
+#define CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_IMAGE_WRITER_UTILITY_CLIENT_H_ |
+ |
+#include "base/files/file_path.h" |
+#include "base/memory/weak_ptr.h" |
+#include "base/threading/sequenced_worker_pool.h" |
+#include "content/public/browser/utility_process_host.h" |
+#include "content/public/browser/utility_process_host_client.h" |
+ |
+// Writes a disk image to a device inside the utility process. |
+class ImageWriterUtilityClient : public content::UtilityProcessHostClient { |
+ public: |
+ typedef base::Callback<void()> CancelCallback; |
+ typedef base::Callback<void()> SuccessCallback; |
+ typedef base::Callback<void(int64)> ProgressCallback; |
+ typedef base::Callback<void(const std::string&)> ErrorCallback; |
+ |
+ ImageWriterUtilityClient(); |
+ |
+ // Starts the write. |
+ // |progress_callback|: Called periodically with the count of bytes processed. |
+ // |success_callback|: Called at successful completion. |
+ // |error_callback|: Called with an error message on failure. |
+ // |source|: The path to the source file to read data from. |
+ // |target|: The path to the target device to write the image file to. |
+ void Write(const ProgressCallback& progress_callback, |
+ const SuccessCallback& success_callback, |
+ const ErrorCallback& error_callback, |
+ const base::FilePath& source, |
+ const base::FilePath& target); |
+ |
+ // Starts a verification. |
+ // |progress_callback|: Called periodically with the count of bytes processed. |
+ // |success_callback|: Called at successful completion. |
+ // |error_callback|: Called with an error message on failure. |
+ // |source|: The path to the source file to read data from. |
+ // |target|: The path to the target device to write the image file to. |
+ void Verify(const ProgressCallback& progress_callback, |
+ const SuccessCallback& success_callback, |
+ const ErrorCallback& error_callback, |
+ const base::FilePath& source, |
+ const base::FilePath& target); |
+ |
+ // Cancels any pending write or verification. |
+ // |cancel_callback|: Called when the cancel has actually occurred. |
+ void Cancel(const CancelCallback& cancel_callback); |
+ |
+ // Shuts down the Utility thread that may have been created. |
+ void Shutdown(); |
+ |
+ private: |
+ // Ensures the UtilityProcessHost has been created. |
+ void StartHost(); |
+ |
+ // UtilityProcessHostClient implementation. |
+ virtual void OnProcessCrashed(int exit_code) OVERRIDE; |
+ virtual void OnProcessLaunchFailed() OVERRIDE; |
+ virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
+ virtual bool Send(IPC::Message* msg); |
+ |
+ // It's a reference-counted object, so destructor is private. |
+ virtual ~ImageWriterUtilityClient(); |
+ |
+ // IPC message handlers. |
+ void OnWriteImageSucceeded(); |
+ void OnWriteImageCancelled(); |
+ void OnWriteImageFailed(const std::string& message); |
+ void OnWriteImageProgress(int64 progress); |
+ |
+ CancelCallback cancel_callback_; |
+ ProgressCallback progress_callback_; |
+ SuccessCallback success_callback_; |
+ ErrorCallback error_callback_; |
+ |
+ base::WeakPtr<content::UtilityProcessHost> utility_process_host_; |
+ |
+ scoped_refptr<base::MessageLoopProxy> message_loop_proxy_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(ImageWriterUtilityClient); |
+}; |
+ |
+#endif // CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_IMAGE_WRITER_UTILITY_CLIENT_H_ |