Index: chrome/utility/image_writer/image_writer_handler.cc |
diff --git a/chrome/utility/image_writer/image_writer_handler.cc b/chrome/utility/image_writer/image_writer_handler.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..471d0de00d84295accda673d17b336366929db2d |
--- /dev/null |
+++ b/chrome/utility/image_writer/image_writer_handler.cc |
@@ -0,0 +1,84 @@ |
+// Copyright 2013 The Chromium Authors. All rights reserved. |
+// Use of this image code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/common/chrome_utility_messages.h" |
+#include "chrome/utility/image_writer/error_messages.h" |
+#include "chrome/utility/image_writer/image_writer.h" |
+#include "chrome/utility/image_writer/image_writer_handler.h" |
+#include "content/public/utility/utility_thread.h" |
+ |
+namespace chrome { |
+namespace image_writer { |
+ |
+ImageWriterHandler::ImageWriterHandler() : weak_factory_(this) {} |
+ |
+ImageWriterHandler::~ImageWriterHandler() {} |
+ |
+void ImageWriterHandler::SendSucceeded() { |
+ Send(new ChromeUtilityHostMsg_ImageWriter_Succeeded()); |
+ image_writer_ = NULL; |
+ content::UtilityThread::Get()->ReleaseProcessIfNeeded(); |
+} |
+ |
+void ImageWriterHandler::SendCancelled() { |
+ Send(new ChromeUtilityHostMsg_ImageWriter_Cancelled()); |
+ image_writer_ = NULL; |
+ content::UtilityThread::Get()->ReleaseProcessIfNeeded(); |
+} |
+ |
+void ImageWriterHandler::SendFailed(const std::string& message) { |
+ Send(new ChromeUtilityHostMsg_ImageWriter_Failed(message)); |
+ image_writer_ = NULL; |
+ content::UtilityThread::Get()->ReleaseProcessIfNeeded(); |
+} |
+ |
+void ImageWriterHandler::SendProgress(int64 progress) { |
+ Send(new ChromeUtilityHostMsg_ImageWriter_Progress(progress)); |
+} |
+ |
+void ImageWriterHandler::Send(IPC::Message* msg) { |
+ content::UtilityThread::Get()->Send(msg); |
+} |
+ |
+bool ImageWriterHandler::OnMessageReceived(const IPC::Message& message) { |
+ bool handled = true; |
+ IPC_BEGIN_MESSAGE_MAP(ImageWriterHandler, message) |
+ IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ImageWriter_Write, OnWriteStart) |
+ IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ImageWriter_Verify, OnVerifyStart) |
+ IPC_MESSAGE_HANDLER(ChromeUtilityMsg_ImageWriter_Cancel, OnCancel) |
+ IPC_MESSAGE_UNHANDLED(handled = false) |
+ IPC_END_MESSAGE_MAP() |
+ return handled; |
+} |
+ |
+void ImageWriterHandler::OnWriteStart(const base::FilePath& image, |
+ const base::FilePath& device) { |
+ if (image_writer_ != NULL) { |
+ SendFailed(error::kOperationAlreadyInProgress); |
Matt Perry
2014/02/14 00:16:23
This will cancel the in-progress operation. Would
Drew Haven
2014/02/14 02:32:56
Good catch. I completely missed that!
|
+ return; |
+ } |
+ image_writer_ = new ImageWriter(weak_factory_.GetWeakPtr()); |
+ image_writer_->Write(image, device); |
+} |
+ |
+void ImageWriterHandler::OnVerifyStart(const base::FilePath& image, |
+ const base::FilePath& device) { |
+ if (image_writer_ != NULL) { |
+ SendFailed(error::kOperationAlreadyInProgress); |
+ return; |
+ } |
+ image_writer_ = new ImageWriter(weak_factory_.GetWeakPtr()); |
+ image_writer_->Verify(image, device); |
+} |
+ |
+void ImageWriterHandler::OnCancel() { |
+ if (image_writer_) { |
+ image_writer_->Cancel(); |
+ } else { |
+ SendCancelled(); |
+ } |
+} |
+ |
+} // namespace image_writer |
+} // namespace chrome |