Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1188)

Unified Diff: chrome/browser/media_gallery/linux/read_mtp_file_worker.cc

Issue 11348337: Move MTPDeviceDelegateImplLinux worker classes to its own files. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: '' Created 8 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: chrome/browser/media_gallery/linux/read_mtp_file_worker.cc
diff --git a/chrome/browser/media_gallery/linux/read_mtp_file_worker.cc b/chrome/browser/media_gallery/linux/read_mtp_file_worker.cc
new file mode 100644
index 0000000000000000000000000000000000000000..bea731b853fc31ac65d6288ee41884a4eadcfa66
--- /dev/null
+++ b/chrome/browser/media_gallery/linux/read_mtp_file_worker.cc
@@ -0,0 +1,112 @@
+// Copyright (c) 2012 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.
+
+#include "chrome/browser/media_gallery/linux/read_mtp_file_worker.h"
+
+#include <fcntl.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
+#include "base/bind.h"
+#include "base/file_util.h"
+#include "base/sequenced_task_runner.h"
+#include "base/synchronization/waitable_event.h"
+#include "chrome/browser/media_transfer_protocol/media_transfer_protocol_manager.h"
+#include "chrome/browser/media_transfer_protocol/mtp_file_entry.pb.h"
+#include "content/public/browser/browser_thread.h"
+#include "third_party/cros_system_api/dbus/service_constants.h"
+
+using content::BrowserThread;
+
+namespace chrome {
+
+ReadMTPFileWorker::ReadMTPFileWorker(const std::string& handle,
+ const std::string& src_path,
+ uint32 total_size,
+ const FilePath& dest_path,
+ base::SequencedTaskRunner* task_runner,
+ base::WaitableEvent* task_completed_event,
+ base::WaitableEvent* shutdown_event)
+ : device_handle_(handle),
+ src_path_(src_path),
+ total_bytes_(total_size),
+ dest_path_(dest_path),
+ bytes_read_(0),
+ error_occurred_(false),
+ media_task_runner_(task_runner),
+ on_task_completed_event_(task_completed_event),
+ on_shutdown_event_(shutdown_event) {
+ DCHECK(on_task_completed_event_);
+ DCHECK(on_shutdown_event_);
+}
+
+void ReadMTPFileWorker::Run() {
+ if (on_shutdown_event_->IsSignaled()) {
+ // Process is in shutdown mode.
+ // Do not post any task on |media_task_runner_|.
+ return;
+ }
+
+ int dest_fd = open(dest_path_.value().c_str(), O_WRONLY);
+ if (dest_fd < 0)
+ return;
+ file_util::ScopedFD dest_fd_scoper(&dest_fd);
+
+ while (bytes_read_ < total_bytes_) {
+ BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
+ base::Bind(&ReadMTPFileWorker::DoWorkOnUIThread,
+ this));
+ on_task_completed_event_->Wait();
+ if (error_occurred_)
+ break;
+ if (on_shutdown_event_->IsSignaled()) {
+ cancel_tasks_flag_.Set();
+ break;
+ }
+
+ int bytes_written =
+ file_util::WriteFileDescriptor(dest_fd, data_.data(), data_.size());
+ if (static_cast<int>(data_.size()) != bytes_written)
+ break;
+
+ bytes_read_ += data_.size();
+ }
+}
+
+bool ReadMTPFileWorker::Succeeded() const {
+ return (!error_occurred_ && (bytes_read_ == total_bytes_));
+}
+
+ReadMTPFileWorker::~ReadMTPFileWorker() {
+ // This object must be destructed on |media_task_runner_|.
+}
+
+void ReadMTPFileWorker::DoWorkOnUIThread() {
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+ if (cancel_tasks_flag_.IsSet())
+ return;
+
+ GetMediaTransferProtocolManager()->ReadFileChunkByPath(
+ device_handle_, src_path_, bytes_read_, BytesToRead(),
+ base::Bind(&ReadMTPFileWorker::OnDidWorkOnUIThread, this));
+}
+
+void ReadMTPFileWorker::OnDidWorkOnUIThread(const std::string& data,
+ bool error) {
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+ if (cancel_tasks_flag_.IsSet())
+ return;
+ error_occurred_ = (error || (data.size() != BytesToRead()));
Lei Zhang 2012/12/03 22:23:54 nit: don't go overboard with parenthesis.
+ if (!error_occurred_)
+ data_ = data;
+ on_task_completed_event_->Signal();
+}
+
+uint32 ReadMTPFileWorker::BytesToRead() const {
+ // Read data in 1 MB chunks.
+ static const uint32 kReadChunkSize = 1024 * 1024;
+ return std::min(kReadChunkSize, total_bytes_ - bytes_read_);
+}
+
+} // namespace chrome

Powered by Google App Engine
This is Rietveld 408576698